Use crypto.GetNodeID instead of sha512 directly

cmd/yggdrasil/main.go

@@ -2,7 +2,6 @@ package main
 
 import (
 	"bytes"
-	"crypto/sha512"
 	"encoding/hex"
 	"encoding/json"
 	"flag"
@@ -194,20 +193,29 @@ func main() {
 		return
 	}
 	// Have we been asked for the node address yet? If so, print it and then stop.
+	getNodeID := func() *crypto.NodeID {
+		if pubkey, err := hex.DecodeString(cfg.EncryptionPublicKey); err == nil {
+			var box crypto.BoxPubKey
+			copy(box[:], pubkey[:])
+			return crypto.GetNodeID(&box)
+		}
+		return nil
+	}
 	switch {
 	case *getaddr:
-		if pubkey, err := hex.DecodeString(cfg.EncryptionPublicKey); err == nil {
+		if nodeid := getNodeID(); nodeid != nil {
-			nodeid := sha512.Sum512(pubkey)
+			addr := *address.AddrForNodeID(nodeid)
-			fromnodeid := address.AddrForNodeID((*crypto.NodeID)(&nodeid))
+			ip := net.IP(addr[:])
-			fmt.Println(net.IP(fromnodeid[:]).String())
+			fmt.Println(ip.String())
 		}
 		return
 	case *getsnet:
-		if pubkey, err := hex.DecodeString(cfg.EncryptionPublicKey); err == nil {
+		if nodeid := getNodeID(); nodeid != nil {
-			nodeid := sha512.Sum512(pubkey)
+			snet := *address.SubnetForNodeID(nodeid)
-			fromnodeid := address.SubnetForNodeID((*crypto.NodeID)(&nodeid))
+			ipnet := net.IPNet{
-			subnet := append(fromnodeid[:], 0, 0, 0, 0, 0, 0, 0, 0)
+				IP:   append(snet[:], 0, 0, 0, 0, 0, 0, 0, 0),
-			ipnet := net.IPNet{IP: subnet, Mask: net.CIDRMask(64, 128)}
+				Mask: net.CIDRMask(len(snet)*8, 128),
+			}
 			fmt.Println(ipnet.String())
 		}
 		return