phantom/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-01-27 03:33:27 -03:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: Check msg type in msg capture is followed by zeros

Browse source
This commit is contained in:
MacroFake 2022-05-12 16:17:54 +02:00
parent dd9f61a184
commit faa5a7a573
No known key found for this signature in database
GPG key ID: CE2B75697E69A548
1 changed files with 1 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
test/functional/p2p_message_capture.py
View file

@ -43,12 +43,8 @@ def mini_parser(dat_file):
break break
tmp_header = BytesIO(tmp_header_raw) tmp_header = BytesIO(tmp_header_raw)
tmp_header.read(TIME_SIZE) # skip the timestamp field tmp_header.read(TIME_SIZE) # skip the timestamp field
raw_msgtype = tmp_header.read(MSGTYPE_SIZE) msgtype = tmp_header.read(MSGTYPE_SIZE).rstrip(b'\x00')
msgtype: bytes = raw_msgtype.split(b'\x00', 1)[0]
remainder = raw_msgtype.split(b'\x00', 1)[1]
assert(len(msgtype) > 0)
assert(msgtype in MESSAGEMAP) assert(msgtype in MESSAGEMAP)
assert(len(remainder) == 0 or not remainder.decode().isprintable())
length: int = int.from_bytes(tmp_header.read(LENGTH_SIZE), "little") length: int = int.from_bytes(tmp_header.read(LENGTH_SIZE), "little")
data = f_in.read(length) data = f_in.read(length)
assert_equal(len(data), length) assert_equal(len(data), length)