From 51ef7f499e3f76a11c7f9bca7fab356831dacb7b Mon Sep 17 00:00:00 2001 From: John Hilliard Date: Thu, 22 Sep 2022 20:57:20 -0400 Subject: [PATCH] feat: adding ipfs seed feature Matching the `ipfs_seed` implementation from the avail LC --- cmd/nodekey.go | 46 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 46 insertions(+) diff --git a/cmd/nodekey.go b/cmd/nodekey.go index 1b672ffd..2e16140b 100644 --- a/cmd/nodekey.go +++ b/cmd/nodekey.go @@ -17,8 +17,10 @@ along with this program. If not, see . package cmd import ( + "bytes" "crypto/ed25519" "crypto/rand" + "encoding/binary" "encoding/hex" "encoding/json" "fmt" @@ -53,6 +55,7 @@ var ( inputNodeKeyUDP *int inputNodeKeyFile *string inputNodeKeySign *bool + inputNodeKeySeed *uint64 ) // nodekeyCmd represents the nodekey command @@ -70,6 +73,10 @@ implemented devp2p because that's what we needed first. if *inputNodeKeyProtocol == "libp2p" { return generateLibp2pNodeKey() } + if *inputNodeKeyProtocol == "seed-libp2p" { + return generateSeededLibp2pNodeKey() + } + return fmt.Errorf("%s is not implemented yet", *inputNodeKeyProtocol) }, } @@ -80,6 +87,7 @@ type ( PrivateKey string FullPrivateKey string `json:",omitempty"` ENR string `json:",omitempty"` + Seed uint64 `json:",omitempty"` } ) @@ -159,6 +167,43 @@ func generateETHNodeKey() error { return nil } +func generateSeededLibp2pNodeKey() error { + seedValue := *inputNodeKeySeed + seedData := make([]byte, 32, 32) + binary.BigEndian.PutUint64(seedData, seedValue) + buf := bytes.NewBuffer(seedData) + rand32 := io.LimitReader(buf, 32) + prvKey, _, err := libp2pcrypto.GenerateEd25519Key(rand32) + if err != nil { + return err + } + + rawPrvKey, err := prvKey.Raw() + if err != nil { + return err + } + + id, err := libp2ppeer.IDFromPrivateKey(prvKey) + if err != nil { + return err + } + nko := nodeKeyOut{ + PublicKey: id.String(), + // half of the private key is the public key. Substrate doesn't handle this well and need just the 32 byte seed/private key + PrivateKey: hex.EncodeToString(rawPrvKey[0:ed25519.PublicKeySize]), + FullPrivateKey: hex.EncodeToString(rawPrvKey), + Seed: seedValue, + } + + out, err := json.Marshal(nko) + if err != nil { + return fmt.Errorf("could not json marshel the key data %v", err) + } + + fmt.Println(string(out)) + + return nil +} func init() { rootCmd.AddCommand(nodekeyCmd) @@ -168,6 +213,7 @@ func init() { inputNodeKeyTCP = nodekeyCmd.PersistentFlags().IntP("tcp", "t", 30303, "The tcp Port to be associated with this address") inputNodeKeyUDP = nodekeyCmd.PersistentFlags().IntP("udp", "u", 0, "The udp Port to be associated with this address") inputNodeKeySign = nodekeyCmd.PersistentFlags().BoolP("sign", "s", false, "Should the node record be signed?") + inputNodeKeySeed = nodekeyCmd.PersistentFlags().Uint64P("seed", "S", 271828, "A numeric seed value") inputNodeKeyFile = nodekeyCmd.PersistentFlags().StringP("file", "f", "", "A file with the private nodekey in hex format")