forked from cerc-io/plugeth
add initial peer level test (failing)
This commit is contained in:
parent
4afde4e738
commit
1f2adb05b5
@ -6,6 +6,7 @@ import (
|
||||
// "crypto/elliptic"
|
||||
// "crypto/rand"
|
||||
"fmt"
|
||||
"net"
|
||||
"testing"
|
||||
|
||||
"github.com/ethereum/go-ethereum/crypto"
|
||||
@ -114,7 +115,9 @@ func TestCryptoHandshake(t *testing.T) {
|
||||
t.Errorf("%v", err)
|
||||
}
|
||||
|
||||
fmt.Printf("\nauth %x\ninitNonce %x\nresponse%x\nremoteRecNonce %x\nremoteInitNonce %x\nremoteRandomPubKey %x\nrecNonce %x\nremoteInitRandomPubKey %x\ninitSessionToken %x\n\n", auth, initNonce, response, remoteRecNonce, remoteInitNonce, remoteRandomPubKey, recNonce, remoteInitRandomPubKey, initSessionToken)
|
||||
fmt.Printf("\nauth (%v) %x\n\nresp (%v) %x\n\n", len(auth), auth, len(response), response)
|
||||
|
||||
// fmt.Printf("\nauth %x\ninitNonce %x\nresponse%x\nremoteRecNonce %x\nremoteInitNonce %x\nremoteRandomPubKey %x\nrecNonce %x\nremoteInitRandomPubKey %x\ninitSessionToken %x\n\n", auth, initNonce, response, remoteRecNonce, remoteInitNonce, remoteRandomPubKey, recNonce, remoteInitRandomPubKey, initSessionToken)
|
||||
|
||||
if !bytes.Equal(initNonce, remoteInitNonce) {
|
||||
t.Errorf("nonces do not match")
|
||||
@ -140,3 +143,51 @@ func TestCryptoHandshake(t *testing.T) {
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestPeersHandshake(t *testing.T) {
|
||||
defer testlog(t).detach()
|
||||
var err error
|
||||
// var sessionToken []byte
|
||||
prv0, _ := crypto.GenerateKey() // = ecdsa.GenerateKey(crypto.S256(), rand.Reader)
|
||||
pub0 := &prv0.PublicKey
|
||||
prv1, _ := crypto.GenerateKey()
|
||||
pub1 := &prv1.PublicKey
|
||||
|
||||
prv0s := crypto.FromECDSA(prv0)
|
||||
pub0s := crypto.FromECDSAPub(pub0)
|
||||
prv1s := crypto.FromECDSA(prv1)
|
||||
pub1s := crypto.FromECDSAPub(pub1)
|
||||
|
||||
conn1, conn2 := net.Pipe()
|
||||
initiator := newPeer(conn1, []Protocol{}, nil)
|
||||
receiver := newPeer(conn2, []Protocol{}, nil)
|
||||
initiator.dialAddr = &peerAddr{IP: net.ParseIP("1.2.3.4"), Port: 2222, Pubkey: pub1s[1:]}
|
||||
initiator.ourID = &peerId{prv0s, pub0s}
|
||||
|
||||
// this is cheating. identity of initiator/dialler not available to listener/receiver
|
||||
// its public key should be looked up based on IP address
|
||||
receiver.identity = initiator.ourID
|
||||
receiver.ourID = &peerId{prv1s, pub1s}
|
||||
|
||||
initiator.pubkeyHook = func(*peerAddr) error { return nil }
|
||||
receiver.pubkeyHook = func(*peerAddr) error { return nil }
|
||||
|
||||
initiator.cryptoHandshake = true
|
||||
receiver.cryptoHandshake = true
|
||||
errc0 := make(chan error, 1)
|
||||
errc1 := make(chan error, 1)
|
||||
go func() {
|
||||
_, err := initiator.loop()
|
||||
errc0 <- err
|
||||
}()
|
||||
go func() {
|
||||
_, err := receiver.loop()
|
||||
errc1 <- err
|
||||
}()
|
||||
select {
|
||||
case err = <-errc0:
|
||||
t.Errorf("peer 0 quit with error: %v", err)
|
||||
case err = <-errc1:
|
||||
t.Errorf("peer 1 quit with error: %v", err)
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user