Merge 5b77793ba6 into 3ed4a92288

cmd/genkeys/main.go

@@ -13,12 +13,14 @@ package main
 import (
 	"crypto/ed25519"
 	"encoding/hex"
+	"flag"
 	"fmt"
 	"net"
 	"runtime"
 	"time"
 
 	"github.com/yggdrasil-network/yggdrasil-go/src/address"
+	"github.com/yggdrasil-network/yggdrasil-go/src/config"
 )
 
 type keySet struct {
@@ -27,9 +29,22 @@ type keySet struct {
 }
 
 func main() {
+	security := flag.Int("security", 0, "generates a key with a specific amount of security bits. defaults to 0 which continuously generates more keys")
+	flag.Parse()
+
+	start := time.Now()
+	if (*security > 0) {
+		// If higher than 0, generates a key with the set amount of security bits
+		var secureKey keySet
+		secureKey.priv, secureKey.pub = config.NewSecureKeyPair(*security)
+		fmt.Println("-----", time.Since(start))
+		fmt.Println("Priv:", hex.EncodeToString(secureKey.priv))
+		fmt.Println("Pub:", hex.EncodeToString(secureKey.pub))
+		addr := address.AddrForKey(secureKey.pub)
+		fmt.Println("IP:", net.IP(addr[:]).String())
+	} else {
 		threads := runtime.GOMAXPROCS(0)
 		fmt.Println("Threads:", threads)
-	start := time.Now()
 		var currentBest ed25519.PublicKey
 		newKeys := make(chan keySet, threads)
 		for i := 0; i < threads; i++ {
@@ -47,6 +62,7 @@ func main() {
 			}
 		}
 	}
+}
 
 func isBetter(oldPub, newPub ed25519.PublicKey) bool {
 	for idx := range oldPub {

cmd/yggdrasil/main.go

@@ -40,6 +40,7 @@ type node struct {
 // The main function is responsible for configuring and starting Yggdrasil.
 func main() {
 	genconf := flag.Bool("genconf", false, "print a new config to stdout")
+	security := flag.Int("security", 0, "use in combination with either -genconf or -autoconf, generates a higher security address up to the security bits desired")
 	useconf := flag.Bool("useconf", false, "read HJSON/JSON config from stdin")
 	useconffile := flag.String("useconffile", "", "read HJSON/JSON config from specified file path")
 	normaliseconf := flag.Bool("normaliseconf", false, "use in combination with either -useconf or -useconffile, outputs your configuration normalised")
@@ -88,6 +89,11 @@ func main() {
 	}
 
 	cfg := config.GenerateConfig()
+	if (*security > 0) {
+		// Checks if the security flag is set, and generates a key with that many security bits
+		newKey, _ := config.NewSecureKeyPair(*security)
+		cfg.PrivateKey = []byte(newKey)
+	}
 	var err error
 	switch {
 	case *ver:

src/config/config.go

@@ -30,6 +30,8 @@ import (
 	"io"
 	"math/big"
 	"os"
+	"runtime"
+	"strings"
 	"time"
 
 	"github.com/hjson/hjson-go/v4"
@@ -208,6 +210,41 @@ func (cfg *NodeConfig) NewPrivateKey() {
 	cfg.PrivateKey = KeyBytes(spriv)
 }
 
+func NewSecureKeyPair(bits int) (priv ed25519.PrivateKey, pub ed25519.PublicKey) {
+	// Generates a key pair with a prescribed number of security bits.
+	threads := runtime.GOMAXPROCS(0)
+	if (bits > 64) {
+		bits = 64
+		// Bounding the maximum number of security bits to the maximum public key length of 64.
+	}
+	type keySet struct {
+		priv ed25519.PrivateKey
+		pub  ed25519.PublicKey
+	}
+	expected :=  strings.Repeat("0", bits)
+	// Generates the expected security substring in advance
+	newKeys := make(chan keySet, threads)
+	for i := 0; i < threads; i++ {
+		go func(out chan<- keySet) {
+			for {
+				pub, priv, err := ed25519.GenerateKey(nil)
+				if err != nil {
+					panic(err)
+				}
+				if !(hex.EncodeToString(pub)[0:bits] == expected) {
+					// Checks if the public key contains the expected security substring
+					continue
+				}
+				out <- keySet{priv, pub}
+			}
+		}(newKeys)
+	}
+	for {
+		newKey := <-newKeys
+		return newKey.priv, newKey.pub
+	}
+}
+
 func (cfg *NodeConfig) MarshalPEMPrivateKey() ([]byte, error) {
 	b, err := x509.MarshalPKCS8PrivateKey(ed25519.PrivateKey(cfg.PrivateKey))
 	if err != nil {