aboutsummaryrefslogtreecommitdiffstats
path: root/p2p
diff options
context:
space:
mode:
authorzelig <viktor.tron@gmail.com>2015-01-18 17:44:49 +0800
committerFelix Lange <fjl@twurst.com>2015-02-06 07:00:34 +0800
commitd227f6184e9d8ce21d40d032dedc910b2bd25f89 (patch)
tree2a60ae163076bf7b96a31fd801f401542333683c /p2p
parent88167f39a6437e282ff75a6ec30e8081d6d50441 (diff)
downloadgo-tangerine-d227f6184e9d8ce21d40d032dedc910b2bd25f89.tar.gz
go-tangerine-d227f6184e9d8ce21d40d032dedc910b2bd25f89.tar.zst
go-tangerine-d227f6184e9d8ce21d40d032dedc910b2bd25f89.zip
fix protocol to accomodate privkey
Diffstat (limited to 'p2p')
-rw-r--r--p2p/protocol.go4
-rw-r--r--p2p/protocol_test.go11
2 files changed, 14 insertions, 1 deletions
diff --git a/p2p/protocol.go b/p2p/protocol.go
index 1d121a885..62ada929d 100644
--- a/p2p/protocol.go
+++ b/p2p/protocol.go
@@ -64,6 +64,10 @@ func (h *handshake) Pubkey() []byte {
return h.NodeID
}
+func (h *handshake) PrivKey() []byte {
+ return nil
+}
+
// Cap is the structure of a peer capability.
type Cap struct {
Name string
diff --git a/p2p/protocol_test.go b/p2p/protocol_test.go
index b1d10ac53..6804a9d40 100644
--- a/p2p/protocol_test.go
+++ b/p2p/protocol_test.go
@@ -11,7 +11,7 @@ import (
)
type peerId struct {
- pubkey []byte
+ privKey, pubkey []byte
}
func (self *peerId) String() string {
@@ -27,6 +27,15 @@ func (self *peerId) Pubkey() (pubkey []byte) {
return
}
+func (self *peerId) PrivKey() (privKey []byte) {
+ privKey = self.privKey
+ if len(privKey) == 0 {
+ privKey = crypto.GenerateNewKeyPair().PublicKey
+ self.privKey = privKey
+ }
+ return
+}
+
func newTestPeer() (peer *Peer) {
peer = NewPeer(&peerId{}, []Cap{})
peer.pubkeyHook = func(*peerAddr) error { return nil }