lbc/yggdrasil/keys.go

package yggdrasil

import (
	"crypto/ed25519"
	"encoding/hex"
	"fmt"
	"os"
	"strings"

	"github.com/gologme/log"
	"github.com/spf13/viper"

	yggConfig "github.com/yggdrasil-network/yggdrasil-go/src/config"
	"github.com/yggdrasil-network/yggdrasil-go/src/core"
)

func GeneratePrivateKey() yggConfig.KeyBytes {
	return yggConfig.GenerateConfig().PrivateKey
}

func GetPublicKey(keyPath string) (ed25519.PublicKey, error) {
	data, err := os.ReadFile(keyPath)
	if err != nil {
		return ed25519.PublicKey{}, err
	}

	decoded, err := hex.DecodeString(strings.TrimSpace(string(data)))
	if err != nil {
		return ed25519.PublicKey{}, err
	}

	if len(decoded) != ed25519.PrivateKeySize {
		return ed25519.PublicKey{}, fmt.Errorf("invalid private key size: %d", len(decoded))
	}

	privateKey := ed25519.PrivateKey(decoded)
	return privateKey.Public().(ed25519.PublicKey), nil
}

func GetYggdrasilAddress(config *viper.Viper) string {
	//var remoteTcp types.TCPRemoteMappings
	ygg := config.Sub("yggdrasil")
	if ygg == nil {
		return ""
	}

	//laddr := config.Sub("p2p").GetString("laddr")
	//remoteTcp.Set(laddr)

	cfg := yggConfig.GenerateConfig()

	cfg.AdminListen = ygg.GetString("admin_listen")
	cfg.Listen = ygg.GetStringSlice("listen")
	cfg.Peers = ygg.GetStringSlice("peers")
	cfg.AllowedPublicKeys = ygg.GetStringSlice("allowed-public-keys")
	cfg.PrivateKeyPath = ygg.GetString("private-key-file")

	logger := log.Default()

	n := &node{}

	// Setup the Yggdrasil node itself.
	{
		options := []core.SetupOption{
			core.NodeInfo(cfg.NodeInfo),
			core.NodeInfoPrivacy(cfg.NodeInfoPrivacy),
		}
		for _, addr := range cfg.Listen {
			options = append(options, core.ListenAddress(addr))
		}
		for _, peer := range cfg.Peers {
			options = append(options, core.Peer{URI: peer})
		}
		for intf, peers := range cfg.InterfacePeers {
			for _, peer := range peers {
				options = append(options, core.Peer{URI: peer, SourceInterface: intf})
			}
		}
		for _, allowed := range cfg.AllowedPublicKeys {
			k, err := hex.DecodeString(allowed)
			if err != nil {
				panic(err)
			}
			options = append(options, core.AllowedPublicKey(k[:]))
		}

		var err error
		if n.core, err = core.New(cfg.Certificate, logger, options...); err != nil {
			panic(err)
		}

		address := n.core.Address()
		n.core.Stop()
		return address.String()
	}

}