TheArchive/yggdrasil-go
1
0
Fork 0
mirror of https://github.com/yggdrasil-network/yggdrasil-go.git synced 2024-11-23 18:15:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use crypto.GetNodeID instead of sha512 directly

Browse Source
This commit is contained in:
Neil Alexander 2019-11-11 09:40:25 +00:00
parent e3a5e4f3b7
commit e310a25e59
No known key found for this signature in database
GPG Key ID: A02A2019A2BB0944
1 changed files with 18 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
cmd/yggdrasil/main.go
View File

@ -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