mirror of
https://github.com/status-im/nimbus-eth1.git
synced 2025-01-14 22:34:23 +00:00
10a3946194
* Rename FindNode to FindNodes as per spec * Use consistently lower case starting camelCase for consts Style guide nep1 allows for both CamelCase and camelCase for consts, but we seem to use more often camelCase. Using this now consistently. * Change some procs to func
223 lines
8.3 KiB
Nim
223 lines
8.3 KiB
Nim
# Nimbus - Portal Network
|
|
# Copyright (c) 2021 Status Research & Development GmbH
|
|
# Licensed and distributed under either of
|
|
# * MIT license (license terms in the root directory or at https://opensource.org/licenses/MIT).
|
|
# * Apache v2 license (license terms in the root directory or at https://www.apache.org/licenses/LICENSE-2.0).
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
{.used.}
|
|
|
|
import
|
|
unittest2, stint, stew/[byteutils, results], eth/p2p/discoveryv5/enr,
|
|
../network/wire/messages
|
|
|
|
suite "Portal Wire Protocol Message Encodings":
|
|
test "Ping Request":
|
|
let
|
|
dataRadius = UInt256.high() - 1 # Full radius - 1
|
|
enrSeq = 1'u64
|
|
# Can be any custom payload, testing with just dataRadius here.
|
|
customPayload = ByteList(SSZ.encode(CustomPayload(dataRadius: dataRadius)))
|
|
p = PingMessage(enrSeq: enrSeq, customPayload: customPayload)
|
|
|
|
let encoded = encodeMessage(p)
|
|
check encoded.toHex ==
|
|
"0001000000000000000c000000feffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff"
|
|
let decoded = decodeMessage(encoded)
|
|
check decoded.isOk()
|
|
|
|
let message = decoded.get()
|
|
check:
|
|
message.kind == ping
|
|
message.ping.enrSeq == enrSeq
|
|
message.ping.customPayload == customPayload
|
|
|
|
test "Pong Response":
|
|
let
|
|
dataRadius = UInt256.high() div 2.stuint(256) # Radius of half the UInt256
|
|
enrSeq = 1'u64
|
|
# Can be any custom payload, testing with just dataRadius here.
|
|
customPayload = ByteList(SSZ.encode(CustomPayload(dataRadius: dataRadius)))
|
|
p = PongMessage(enrSeq: enrSeq, customPayload: customPayload)
|
|
|
|
let encoded = encodeMessage(p)
|
|
check encoded.toHex ==
|
|
"0101000000000000000c000000ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff7f"
|
|
let decoded = decodeMessage(encoded)
|
|
check decoded.isOk()
|
|
|
|
let message = decoded.get()
|
|
check:
|
|
message.kind == pong
|
|
message.pong.enrSeq == enrSeq
|
|
message.pong.customPayload == customPayload
|
|
|
|
test "FindNodes Request":
|
|
let
|
|
distances = List[uint16, 256](@[0x0100'u16, 0x00ff'u16])
|
|
fn = FindNodesMessage(distances: distances)
|
|
|
|
let encoded = encodeMessage(fn)
|
|
check encoded.toHex == "02040000000001ff00"
|
|
|
|
let decoded = decodeMessage(encoded)
|
|
check decoded.isOk()
|
|
|
|
let message = decoded.get()
|
|
check:
|
|
message.kind == findnodes
|
|
message.findnodes.distances == distances
|
|
|
|
test "Nodes Response - empty enr list":
|
|
let
|
|
total = 0x1'u8
|
|
n = NodesMessage(total: total)
|
|
|
|
let encoded = encodeMessage(n)
|
|
check encoded.toHex == "030105000000"
|
|
|
|
let decoded = decodeMessage(encoded)
|
|
check decoded.isOk()
|
|
|
|
let message = decoded.get()
|
|
check:
|
|
message.kind == nodes
|
|
message.nodes.total == total
|
|
message.nodes.enrs.len() == 0
|
|
|
|
test "Nodes Response - enrs":
|
|
var e1, e2: Record
|
|
check:
|
|
e1.fromURI("enr:-HW4QBzimRxkmT18hMKaAL3IcZF1UcfTMPyi3Q1pxwZZbcZVRI8DC5infUAB_UauARLOJtYTxaagKoGmIjzQxO2qUygBgmlkgnY0iXNlY3AyNTZrMaEDymNMrg1JrLQB2KTGtv6MVbcNEVv0AHacwUAPMljNMTg")
|
|
e2.fromURI("enr:-HW4QNfxw543Ypf4HXKXdYxkyzfcxcO-6p9X986WldfVpnVTQX1xlTnWrktEWUbeTZnmgOuAY_KUhbVV1Ft98WoYUBMBgmlkgnY0iXNlY3AyNTZrMaEDDiy3QkHAxPyOgWbxp5oF1bDdlYE6dLCUUp8xfVw50jU")
|
|
|
|
let
|
|
total = 0x1'u8
|
|
n = NodesMessage(total: total, enrs: List[ByteList, 32](@[ByteList(e1.raw), ByteList(e2.raw)]))
|
|
|
|
let encoded = encodeMessage(n)
|
|
check encoded.toHex == "030105000000080000007f000000f875b8401ce2991c64993d7c84c29a00bdc871917551c7d330fca2dd0d69c706596dc655448f030b98a77d4001fd46ae0112ce26d613c5a6a02a81a6223cd0c4edaa53280182696482763489736563703235366b31a103ca634cae0d49acb401d8a4c6b6fe8c55b70d115bf400769cc1400f3258cd3138f875b840d7f1c39e376297f81d7297758c64cb37dcc5c3beea9f57f7ce9695d7d5a67553417d719539d6ae4b445946de4d99e680eb8063f29485b555d45b7df16a1850130182696482763489736563703235366b31a1030e2cb74241c0c4fc8e8166f1a79a05d5b0dd95813a74b094529f317d5c39d235"
|
|
|
|
let decoded = decodeMessage(encoded)
|
|
check decoded.isOk()
|
|
|
|
let message = decoded.get()
|
|
check:
|
|
message.kind == nodes
|
|
message.nodes.total == total
|
|
message.nodes.enrs.len() == 2
|
|
message.nodes.enrs[0] == ByteList(e1.raw)
|
|
message.nodes.enrs[1] == ByteList(e2.raw)
|
|
|
|
test "FindContent Request":
|
|
const contentKeyString = "0x706f7274616c"
|
|
let
|
|
contentKey = ByteList.init(hexToSeqByte(contentKeyString))
|
|
fc = FindContentMessage(contentKey: contentKey)
|
|
|
|
let encoded = encodeMessage(fc)
|
|
check encoded.toHex == "0404000000706f7274616c"
|
|
|
|
let decoded = decodeMessage(encoded)
|
|
check decoded.isOk()
|
|
|
|
let message = decoded.get()
|
|
check:
|
|
message.kind == findcontent
|
|
message.findcontent.contentKey == contentKey
|
|
|
|
test "Content Response - connection id":
|
|
let
|
|
connectionId = Bytes2([byte 0x01, 0x02])
|
|
c = ContentMessage(
|
|
contentMessageType: connectionIdType, connectionId: connectionId)
|
|
|
|
let encoded = encodeMessage(c)
|
|
check encoded.toHex == "05000102"
|
|
|
|
let decoded = decodeMessage(encoded)
|
|
check decoded.isOk()
|
|
|
|
let message = decoded.get()
|
|
check:
|
|
message.kind == MessageKind.content
|
|
message.content.contentMessageType == connectionIdType
|
|
message.content.connectionId == connectionId
|
|
|
|
test "Content Response - content payload":
|
|
const contentString = "0x7468652063616b652069732061206c6965"
|
|
let
|
|
content = ByteList(hexToSeqByte(contentString))
|
|
c = ContentMessage(contentMessageType: contentType, content: content)
|
|
|
|
let encoded = encodeMessage(c)
|
|
check encoded.toHex == "05017468652063616b652069732061206c6965"
|
|
|
|
let decoded = decodeMessage(encoded)
|
|
check decoded.isOk()
|
|
|
|
let message = decoded.get()
|
|
check:
|
|
message.kind == MessageKind.content
|
|
message.content.contentMessageType == contentType
|
|
message.content.content == content
|
|
|
|
test "Content Response - enrs":
|
|
var e1, e2: Record
|
|
check:
|
|
e1.fromURI("enr:-HW4QBzimRxkmT18hMKaAL3IcZF1UcfTMPyi3Q1pxwZZbcZVRI8DC5infUAB_UauARLOJtYTxaagKoGmIjzQxO2qUygBgmlkgnY0iXNlY3AyNTZrMaEDymNMrg1JrLQB2KTGtv6MVbcNEVv0AHacwUAPMljNMTg")
|
|
e2.fromURI("enr:-HW4QNfxw543Ypf4HXKXdYxkyzfcxcO-6p9X986WldfVpnVTQX1xlTnWrktEWUbeTZnmgOuAY_KUhbVV1Ft98WoYUBMBgmlkgnY0iXNlY3AyNTZrMaEDDiy3QkHAxPyOgWbxp5oF1bDdlYE6dLCUUp8xfVw50jU")
|
|
|
|
let
|
|
enrs = List[ByteList, 32](@[ByteList(e1.raw), ByteList(e2.raw)])
|
|
c = ContentMessage(contentMessageType: enrsType, enrs: enrs)
|
|
|
|
let encoded = encodeMessage(c)
|
|
check encoded.toHex == "0502080000007f000000f875b8401ce2991c64993d7c84c29a00bdc871917551c7d330fca2dd0d69c706596dc655448f030b98a77d4001fd46ae0112ce26d613c5a6a02a81a6223cd0c4edaa53280182696482763489736563703235366b31a103ca634cae0d49acb401d8a4c6b6fe8c55b70d115bf400769cc1400f3258cd3138f875b840d7f1c39e376297f81d7297758c64cb37dcc5c3beea9f57f7ce9695d7d5a67553417d719539d6ae4b445946de4d99e680eb8063f29485b555d45b7df16a1850130182696482763489736563703235366b31a1030e2cb74241c0c4fc8e8166f1a79a05d5b0dd95813a74b094529f317d5c39d235"
|
|
|
|
let decoded = decodeMessage(encoded)
|
|
check decoded.isOk()
|
|
|
|
let message = decoded.get()
|
|
check:
|
|
message.kind == MessageKind.content
|
|
message.content.contentMessageType == enrsType
|
|
message.content.enrs.len() == 2
|
|
message.content.enrs[0] == ByteList(e1.raw)
|
|
message.content.enrs[1] == ByteList(e2.raw)
|
|
|
|
test "Offer Request":
|
|
let
|
|
contentKeys = ContentKeysList(List(@[ByteList(@[byte 0x01, 0x02, 0x03])]))
|
|
o = OfferMessage(contentKeys: contentKeys)
|
|
|
|
let encoded = encodeMessage(o)
|
|
check encoded.toHex == "060400000004000000010203"
|
|
|
|
let decoded = decodeMessage(encoded)
|
|
check decoded.isOk()
|
|
|
|
let message = decoded.get()
|
|
check:
|
|
message.kind == offer
|
|
message.offer.contentKeys == contentKeys
|
|
|
|
test "Accept Response":
|
|
var contentKeys = ContentKeysBitList.init(8)
|
|
contentKeys.setBit(0)
|
|
let
|
|
connectionId = Bytes2([byte 0x01, 0x02])
|
|
a = AcceptMessage(connectionId: connectionId, contentKeys: contentKeys)
|
|
|
|
let encoded = encodeMessage(a)
|
|
check encoded.toHex == "070102060000000101"
|
|
|
|
let decoded = decodeMessage(encoded)
|
|
check decoded.isOk()
|
|
|
|
let message = decoded.get()
|
|
check:
|
|
message.kind == MessageKind.accept
|
|
message.accept.connectionId == connectionId
|
|
message.accept.contentKeys == contentKeys
|