package network import ( "net" "testing" "github.com/nspcc-dev/neo-go/pkg/network/payload" "github.com/stretchr/testify/require" ) func connReadStub(conn net.Conn) { b := make([]byte, 1024) var err error for ; err == nil; _, err = conn.Read(b) { } } func TestPeerHandshake(t *testing.T) { server, client := net.Pipe() tcpS := NewTCPPeer(server, newTestServer(t, ServerConfig{})) tcpC := NewTCPPeer(client, newTestServer(t, ServerConfig{})) // Something should read things written into the pipe. go connReadStub(tcpS.conn) go connReadStub(tcpC.conn) // No handshake yet. require.Equal(t, false, tcpS.Handshaked()) require.Equal(t, false, tcpC.Handshaked()) // No ordinary messages can be written. require.Error(t, tcpS.EnqueueP2PMessage(&Message{})) require.Error(t, tcpC.EnqueueP2PMessage(&Message{})) // Try to mess with VersionAck on both client and server, it should fail. require.Error(t, tcpS.SendVersionAck(&Message{})) require.Error(t, tcpS.HandleVersionAck()) require.Error(t, tcpC.SendVersionAck(&Message{})) require.Error(t, tcpC.HandleVersionAck()) // No handshake yet. require.Equal(t, false, tcpS.Handshaked()) require.Equal(t, false, tcpC.Handshaked()) // Now send and handle versions, but in a different order on client and // server. require.NoError(t, tcpC.SendVersion()) require.Error(t, tcpC.HandleVersionAck()) // Didn't receive version yet. require.NoError(t, tcpS.HandleVersion(&payload.Version{})) require.Error(t, tcpS.SendVersionAck(&Message{})) // Didn't send version yet. require.NoError(t, tcpC.HandleVersion(&payload.Version{})) require.NoError(t, tcpS.SendVersion()) // No handshake yet. require.Equal(t, false, tcpS.Handshaked()) require.Equal(t, false, tcpC.Handshaked()) // These are sent/received and should fail now. require.Error(t, tcpC.SendVersion()) require.Error(t, tcpS.HandleVersion(&payload.Version{})) require.Error(t, tcpC.HandleVersion(&payload.Version{})) require.Error(t, tcpS.SendVersion()) // Now send and handle ACK, again in a different order on client and // server. require.NoError(t, tcpC.SendVersionAck(&Message{})) require.NoError(t, tcpS.HandleVersionAck()) require.NoError(t, tcpC.HandleVersionAck()) require.NoError(t, tcpS.SendVersionAck(&Message{})) // Handshaked now. require.Equal(t, true, tcpS.Handshaked()) require.Equal(t, true, tcpC.Handshaked()) // Subsequent ACKing should fail. require.Error(t, tcpC.SendVersionAck(&Message{})) require.Error(t, tcpS.HandleVersionAck()) require.Error(t, tcpC.HandleVersionAck()) require.Error(t, tcpS.SendVersionAck(&Message{})) // Now regular messaging can proceed. require.NoError(t, tcpS.EnqueueP2PMessage(&Message{})) require.NoError(t, tcpC.EnqueueP2PMessage(&Message{})) }