mirror of
https://github.com/status-im/nim-eth.git
synced 2025-02-18 00:46:32 +00:00
1467b145ae
* disconnect peers that send non-hello messages during initial hello step * fix devp2p protocol version - 4 because we don't implement snappy (yet) - this is cosmetic since this particular version field is not actually being used * fix ack message length checking * move RLPx transport code to separate module, annotate with asyncraises * increase max RLPx message size to 16mb, per EIP-706 * make sure both accept/connect timeout after 10s * aim to log every connection attempt once at debug level * make capability-id/context-id check more accurate * disallow random messages before hello
61 lines
1.5 KiB
Nim
61 lines
1.5 KiB
Nim
{.used.}
|
|
|
|
import
|
|
unittest2,
|
|
chronos/unittest2/asynctests,
|
|
../../eth/common/keys,
|
|
../../eth/p2p/rlpxtransport
|
|
|
|
suite "RLPx transport":
|
|
setup:
|
|
let
|
|
rng = newRng()
|
|
keys1 = KeyPair.random(rng[])
|
|
keys2 = KeyPair.random(rng[])
|
|
server = createStreamServer(initTAddress("127.0.0.1:0"), {ReuseAddr})
|
|
|
|
teardown:
|
|
waitFor server.closeWait()
|
|
|
|
asyncTest "Connect/accept":
|
|
const msg = @[byte 0, 1, 2, 3]
|
|
proc serveClient(server: StreamServer) {.async.} =
|
|
let transp = await server.accept()
|
|
let a = await RlpxTransport.accept(rng, keys1, transp)
|
|
await a.sendMsg(msg)
|
|
await a.closeWait()
|
|
|
|
let serverFut = server.serveClient()
|
|
defer:
|
|
await serverFut.wait(1.seconds)
|
|
|
|
let client =
|
|
await RlpxTransport.connect(rng, keys2, server.localAddress(), keys1.pubkey)
|
|
|
|
defer:
|
|
await client.closeWait()
|
|
let rmsg = await client.recvMsg().wait(1.seconds)
|
|
|
|
check:
|
|
msg == rmsg
|
|
|
|
await serverFut
|
|
|
|
asyncTest "Detect invalid pubkey":
|
|
proc serveClient(server: StreamServer) {.async.} =
|
|
let transp = await server.accept()
|
|
discard await RlpxTransport.accept(rng, keys1, transp)
|
|
raiseAssert "should fail to accept due to pubkey error"
|
|
|
|
let serverFut = server.serveClient()
|
|
defer:
|
|
expect(RlpxTransportError):
|
|
await serverFut.wait(1.seconds)
|
|
|
|
let keys3 = KeyPair.random(rng[])
|
|
|
|
# accept side should close connections
|
|
expect(TransportError):
|
|
discard
|
|
await RlpxTransport.connect(rng, keys2, server.localAddress(), keys3.pubkey)
|