Use the same server as in addr for NTP.

mchackorg · Dec 11, 2018 · a59c3ff · a59c3ff
1 parent 5d0da97
commit a59c3ff
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/client/client.go b/client/client.go
@@ -4,6 +4,7 @@ import (
 	"crypto/tls"
 	"flag"
 	"fmt"
+	"strings"
 
 	"../ntske"
 	"github.com/beevik/ntp"
@@ -60,7 +61,10 @@ func main() {
 
 	fmt.Printf("NTS-KE negotiated data: %#v\n", ke.Meta)
 
-	ntpTime, err := ntp.Time("localhost")
+	// TODO use the negotiated NTP server in ke.Meta.Server if any
+	addrNoPort := addr[:strings.IndexByte(addr, ':')]
+
+	ntpTime, err := ntp.Time(addrNoPort)
 	if err != nil {
 		fmt.Println(err)
 	}