add Node Key func

Author: wenchaopeng

config/config.go

@@ -1,6 +1,9 @@
 package config
 
 import (
+	"os/user"
+	"path"
+
 	"github.com/qlcchain/go-qlc/common"
 )
 
@@ -14,8 +17,9 @@ type RPC struct {
 	Port   uint   `json:"Port"`
 }
 type Node struct {
-	Version uint   `json:"Version"`
-	Network string `json:"Network"`
+	Version        uint   `json:"Version"`
+	Network        string `json:"Network"`
+	PrivateKeyPath string `json:"PrivateKeyPath"`
 }
 type Config struct {
 	*RPC  `json:"RPC"`
@@ -37,17 +41,31 @@ var (
 			Port:   29735,
 		},
 		Node: &Node{
-			Version: 1,
-			Network: "testnet",
+			Version:        1,
+			Network:        "testnet",
+			PrivateKeyPath: "",
 		},
 	}
 )
 
+func privateKeyPath() (string, error) {
+	usr, err := user.Current()
+	if err != nil {
+		return "", err
+	}
+	dir := path.Join(usr.HomeDir, ".qlcchain")
+	return path.Join(dir, "network.key"), nil
+}
+
 var log = common.NewLogger("config")
 
 func init() {
 	cfg := DefaultlConfig
-	var err error
+	privateKeyPath, err := privateKeyPath()
+	if err != nil {
+		log.Errorf("privateKeyPath error: %s", err)
+	}
+	cfg.PrivateKeyPath = privateKeyPath
 	if qlccfg, err = NewCfgManager("", ""); err != nil {
 		log.Errorf("config manager error: %s", err)
 	}

go.mod

@@ -3,26 +3,29 @@ module github.com/qlcchain/go-qlc
 require (
 	github.com/AndreasBriese/bbloom v0.0.0-20180913140656-343706a395b7 // indirect
 	github.com/BurntSushi/toml v0.3.1 // indirect
+	github.com/agl/ed25519 v0.0.0-20170116200512-5312a6153412 // indirect
+	github.com/btcsuite/btcd v0.0.0-20181013004428-67e573d211ac // indirect
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/dgraph-io/badger v1.5.4
 	github.com/dgryski/go-farm v0.0.0-20180109070241-2de33835d102 // indirect
-	github.com/golang/protobuf v1.2.0
+	github.com/gogo/protobuf v1.1.1 // indirect
+	github.com/golang/protobuf v1.2.0 // indirect
 	github.com/kr/pretty v0.1.0 // indirect
-	github.com/kr/pty v1.1.3 // indirect
+	github.com/libp2p/go-libp2p-crypto v1.5.0
 	github.com/mitchellh/go-homedir v1.0.0
 	github.com/philhofer/fwd v1.0.0 // indirect
 	github.com/pkg/errors v0.8.0 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
 	github.com/shopspring/decimal v0.0.0-20180709203117-cd690d0c9e24
 	github.com/stretchr/testify v1.2.2 // indirect
 	github.com/tinylib/msgp v1.0.2
-	go.uber.org/atomic v1.3.2
-	go.uber.org/multierr v1.1.0
+	go.uber.org/atomic v1.3.2 // indirect
+	go.uber.org/multierr v1.1.0 // indirect
 	go.uber.org/zap v1.9.1
 	golang.org/x/crypto v0.0.0-20180927165925-5295e8364332
 	golang.org/x/net v0.0.0-20180926154720-4dfa2610cdf3 // indirect
 	golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f // indirect
-	golang.org/x/sys v0.0.0-20180928133829-e4b3c5e90611
+	golang.org/x/sys v0.0.0-20180928133829-e4b3c5e90611 // indirect
 	gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 // indirect
 	gopkg.in/natefinch/lumberjack.v2 v2.0.0
 	gopkg.in/yaml.v2 v2.2.1 // indirect

p2p/key.go

@@ -0,0 +1,63 @@
+package p2p
+
+import (
+	"encoding/base64"
+	"io/ioutil"
+	mrand "math/rand"
+	"time"
+
+	crypto "github.com/libp2p/go-libp2p-crypto"
+)
+
+// LoadNetworkKeyFromFile load network priv key from file.
+func loadNetworkKeyFromFile(path string) (crypto.PrivKey, error) {
+	data, err := ioutil.ReadFile(path)
+	if err != nil {
+		key, err := generateEd25519Key()
+		if err != nil {
+			return nil, err
+		}
+
+		str, _ := marshalNetworkKey(key)
+
+		ioutil.WriteFile(path, []byte(str), 0600)
+
+		return key, err
+	}
+	return unmarshalNetworkKey(string(data))
+}
+
+// LoadNetworkKeyFromFileOrCreateNew load network priv key from file or create new one.
+func LoadNetworkKeyFromFileOrCreateNew(path string) (crypto.PrivKey, error) {
+	if path == "" {
+		return generateEd25519Key()
+	}
+	return loadNetworkKeyFromFile(path)
+}
+
+// UnmarshalNetworkKey unmarshal network key.
+func unmarshalNetworkKey(data string) (crypto.PrivKey, error) {
+	binaryData, err := base64.StdEncoding.DecodeString(data)
+	if err != nil {
+		return nil, err
+	}
+
+	return crypto.UnmarshalPrivateKey(binaryData)
+}
+
+// MarshalNetworkKey marshal network key.
+func marshalNetworkKey(key crypto.PrivKey) (string, error) {
+	binaryData, err := crypto.MarshalPrivateKey(key)
+	if err != nil {
+		return "", err
+	}
+
+	return base64.StdEncoding.EncodeToString(binaryData), nil
+}
+
+// GenerateEd25519Key return a new generated Ed22519 Private key.
+func generateEd25519Key() (crypto.PrivKey, error) {
+	r := mrand.New(mrand.NewSource(time.Now().UnixNano()))
+	key, _, err := crypto.GenerateKeyPairWithReader(crypto.RSA, 2048, r)
+	return key, err
+}

p2p/key_test.go

@@ -0,0 +1,12 @@
+package p2p
+
+import (
+	"testing"
+
+	"github.com/qlcchain/go-qlc/config"
+)
+
+func TestLoadNetworkKeyFromFileOrCreateNew(t *testing.T) {
+	var cfg = config.DefaultlConfig
+	LoadNetworkKeyFromFileOrCreateNew(cfg.PrivateKeyPath)
+}