WIP

peer/crypto.go

@@ -1,30 +1,15 @@
 package peer
 
 import (
-	"crypto/rand"
 	"log"
 
-	"golang.org/x/crypto/nacl/box"
-	"golang.org/x/crypto/nacl/sign"
+	"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
 )
 
-type cryptoKeys struct {
-	PubKey      []byte
-	PrivKey     []byte
-	PubSignKey  []byte
-	PrivSignKey []byte
-}
-
-func generateKeys() cryptoKeys {
-	pubKey, privKey, err := box.GenerateKey(rand.Reader)
+func generateWGKey() wgtypes.Key {
+	key, err := wgtypes.GeneratePrivateKey()
 	if err != nil {
-		log.Fatalf("Failed to generate encryption keys: %v", err)
+		log.Fatalf("Failed to generate WireGuard private key: %v", err)
 	}
-
-	pubSignKey, privSignKey, err := sign.GenerateKey(rand.Reader)
-	if err != nil {
-		log.Fatalf("Failed to generate signing keys: %v", err)
-	}
-
-	return cryptoKeys{pubKey[:], privKey[:], pubSignKey[:], privSignKey[:]}
+	return key
 }