Ethan Buchman
7 years ago
6 changed files with 204 additions and 18 deletions
Split View
Diff Options
-
+9 -0config/config.go
-
+12 -4node/node.go
-
+111 -0p2p/key.go
-
+49 -0p2p/key_test.go
-
+21 -12p2p/switch.go
-
+2 -2p2p/switch_test.go
+ 9
- 0
config/config.go
View File
+ 12
- 4
node/node.go
View File
+ 111
- 0
p2p/key.go
View File
| @ -0,0 +1,111 @@ | |||
| package p2p | |||
| import ( | |||
| "bytes" | |||
| "encoding/json" | |||
| "fmt" | |||
| "io/ioutil" | |||
| "math/big" | |||
| crypto "github.com/tendermint/go-crypto" | |||
| cmn "github.com/tendermint/tmlibs/common" | |||
| ) | |||
| //------------------------------------------------------------------------------ | |||
| // Persistent peer ID | |||
| // TODO: encrypt on disk | |||
| // NodeKey is the persistent peer key. | |||
| // It contains the nodes private key for authentication. | |||
| type NodeKey struct { | |||
| PrivKey crypto.PrivKey `json:"priv_key"` // our priv key | |||
| } | |||
| // ID returns the peer's canonical ID - the hash of its public key. | |||
| func (nodeKey *NodeKey) ID() []byte { | |||
| return nodeKey.PrivKey.PubKey().Address() | |||
| } | |||
| // PubKey returns the peer's PubKey | |||
| func (nodeKey *NodeKey) PubKey() crypto.PubKey { | |||
| return nodeKey.PrivKey.PubKey() | |||
| } | |||
| // LoadOrGenNodeKey attempts to load the NodeKey from the given filePath, | |||
| // and checks that the corresponding ID is less than the target. | |||
| // If the file does not exist, it generates and saves a new NodeKey | |||
| // with ID less than target. | |||
| func LoadOrGenNodeKey(filePath string, target []byte) (*NodeKey, error) { | |||
| if cmn.FileExists(filePath) { | |||
| nodeKey, err := loadNodeKey(filePath) | |||
| if err != nil { | |||
| return nil, err | |||
| } | |||
| if bytes.Compare(nodeKey.ID(), target) >= 0 { | |||
| return nil, fmt.Errorf("Loaded ID (%X) does not satisfy target (%X)", nodeKey.ID(), target) | |||
| } | |||
| return nodeKey, nil | |||
| } else { | |||
| return genNodeKey(filePath, target) | |||
| } | |||
| } | |||
| // MakePoWTarget returns a 20 byte target byte array. | |||
| func MakePoWTarget(difficulty uint8) []byte { | |||
| zeroPrefixLen := (int(difficulty) / 8) | |||
| prefix := bytes.Repeat([]byte{0}, zeroPrefixLen) | |||
| mod := (difficulty % 8) | |||
| if mod > 0 { | |||
| nonZeroPrefix := byte(1 << (8 - mod)) | |||
| prefix = append(prefix, nonZeroPrefix) | |||
| } | |||
| return append(prefix, bytes.Repeat([]byte{255}, 20-len(prefix))...) | |||
| } | |||
| func loadNodeKey(filePath string) (*NodeKey, error) { | |||
| jsonBytes, err := ioutil.ReadFile(filePath) | |||
| if err != nil { | |||
| return nil, err | |||
| } | |||
| nodeKey := new(NodeKey) | |||
| err = json.Unmarshal(jsonBytes, nodeKey) | |||
| if err != nil { | |||
| return nil, fmt.Errorf("Error reading NodeKey from %v: %v\n", filePath, err) | |||
| } | |||
| return nodeKey, nil | |||
| } | |||
| func genNodeKey(filePath string, target []byte) (*NodeKey, error) { | |||
| privKey := genPrivKeyEd25519PoW(target).Wrap() | |||
| nodeKey := &NodeKey{ | |||
| PrivKey: privKey, | |||
| } | |||
| jsonBytes, err := json.Marshal(nodeKey) | |||
| if err != nil { | |||
| return nil, err | |||
| } | |||
| err = ioutil.WriteFile(filePath, jsonBytes, 0600) | |||
| if err != nil { | |||
| return nil, err | |||
| } | |||
| return nodeKey, nil | |||
| } | |||
| // generate key with address satisfying the difficult target | |||
| func genPrivKeyEd25519PoW(target []byte) crypto.PrivKeyEd25519 { | |||
| secret := crypto.CRandBytes(32) | |||
| var privKey crypto.PrivKeyEd25519 | |||
| for i := 0; ; i++ { | |||
| privKey = crypto.GenPrivKeyEd25519FromSecret(secret) | |||
| if bytes.Compare(privKey.PubKey().Address(), target) < 0 { | |||
| break | |||
| } | |||
| z := new(big.Int) | |||
| z.SetBytes(secret) | |||
| z = z.Add(z, big.NewInt(1)) | |||
| secret = z.Bytes() | |||
| } | |||
| return privKey | |||
| } | |||
+ 49
- 0
p2p/key_test.go
View File
| @ -0,0 +1,49 @@ | |||
| package p2p | |||
| import ( | |||
| "bytes" | |||
| "os" | |||
| "path/filepath" | |||
| "testing" | |||
| "github.com/stretchr/testify/assert" | |||
| cmn "github.com/tendermint/tmlibs/common" | |||
| ) | |||
| func TestLoadOrGenNodeKey(t *testing.T) { | |||
| filePath := filepath.Join(os.TempDir(), cmn.RandStr(12)+"_peer_id.json") | |||
| target := MakePoWTarget(2) | |||
| nodeKey, err := LoadOrGenNodeKey(filePath, target) | |||
| assert.Nil(t, err) | |||
| nodeKey2, err := LoadOrGenNodeKey(filePath, target) | |||
| assert.Nil(t, err) | |||
| assert.Equal(t, nodeKey, nodeKey2) | |||
| } | |||
| func repeatBytes(val byte, n int) []byte { | |||
| return bytes.Repeat([]byte{val}, n) | |||
| } | |||
| func TestPoWTarget(t *testing.T) { | |||
| cases := []struct { | |||
| difficulty uint8 | |||
| target []byte | |||
| }{ | |||
| {0, bytes.Repeat([]byte{255}, 20)}, | |||
| {1, append([]byte{128}, repeatBytes(255, 19)...)}, | |||
| {8, append([]byte{0}, repeatBytes(255, 19)...)}, | |||
| {9, append([]byte{0, 128}, repeatBytes(255, 18)...)}, | |||
| {10, append([]byte{0, 64}, repeatBytes(255, 18)...)}, | |||
| {16, append([]byte{0, 0}, repeatBytes(255, 18)...)}, | |||
| {17, append([]byte{0, 0, 128}, repeatBytes(255, 17)...)}, | |||
| } | |||
| for _, c := range cases { | |||
| assert.Equal(t, MakePoWTarget(c.difficulty), c.target) | |||
| } | |||
| } | |||