diff options
| author | zelig <viktor.tron@gmail.com> | 2015-01-18 17:44:49 +0800 |
|---|---|---|
| committer | Felix Lange <fjl@twurst.com> | 2015-02-06 07:00:34 +0800 |
| commit | d227f6184e9d8ce21d40d032dedc910b2bd25f89 (patch) | |
| tree | 2a60ae163076bf7b96a31fd801f401542333683c /p2p | |
| parent | 88167f39a6437e282ff75a6ec30e8081d6d50441 (diff) | |
| download | dexon-d227f6184e9d8ce21d40d032dedc910b2bd25f89.tar dexon-d227f6184e9d8ce21d40d032dedc910b2bd25f89.tar.gz dexon-d227f6184e9d8ce21d40d032dedc910b2bd25f89.tar.bz2 dexon-d227f6184e9d8ce21d40d032dedc910b2bd25f89.tar.lz dexon-d227f6184e9d8ce21d40d032dedc910b2bd25f89.tar.xz dexon-d227f6184e9d8ce21d40d032dedc910b2bd25f89.tar.zst dexon-d227f6184e9d8ce21d40d032dedc910b2bd25f89.zip | |
fix protocol to accomodate privkey
Diffstat (limited to 'p2p')
| -rw-r--r-- | p2p/protocol.go | 4 | ||||
| -rw-r--r-- | p2p/protocol_test.go | 11 |
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 } |