From e310a25e5987a2400bbdbb765b14bb1967ca02e2 Mon Sep 17 00:00:00 2001
From: Neil Alexander <neilalexander@users.noreply.github.com>
Date: Mon, 11 Nov 2019 09:40:25 +0000
Subject: [PATCH] Use crypto.GetNodeID instead of sha512 directly

---
 cmd/yggdrasil/main.go | 28 ++++++++++++++++++----------
 1 file changed, 18 insertions(+), 10 deletions(-)

diff --git a/cmd/yggdrasil/main.go b/cmd/yggdrasil/main.go
index 6158fd9d..981c6efc 100644
--- a/cmd/yggdrasil/main.go
+++ b/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 {
-			nodeid := sha512.Sum512(pubkey)
-			fromnodeid := address.AddrForNodeID((*crypto.NodeID)(&nodeid))
-			fmt.Println(net.IP(fromnodeid[:]).String())
+		if nodeid := getNodeID(); nodeid != nil {
+			addr := *address.AddrForNodeID(nodeid)
+			ip := net.IP(addr[:])
+			fmt.Println(ip.String())
 		}
 		return
 	case *getsnet:
-		if pubkey, err := hex.DecodeString(cfg.EncryptionPublicKey); err == nil {
-			nodeid := sha512.Sum512(pubkey)
-			fromnodeid := address.SubnetForNodeID((*crypto.NodeID)(&nodeid))
-			subnet := append(fromnodeid[:], 0, 0, 0, 0, 0, 0, 0, 0)
-			ipnet := net.IPNet{IP: subnet, Mask: net.CIDRMask(64, 128)}
+		if nodeid := getNodeID(); nodeid != nil {
+			snet := *address.SubnetForNodeID(nodeid)
+			ipnet := net.IPNet{
+				IP:   append(snet[:], 0, 0, 0, 0, 0, 0, 0, 0),
+				Mask: net.CIDRMask(len(snet)*8, 128),
+			}
 			fmt.Println(ipnet.String())
 		}
 		return