logos-blockchain/logos-blockchain-specs
1
0
Fork 0
mirror of https://github.com/logos-blockchain/logos-blockchain-specs.git synced 2026-01-04 22:23:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

pad final msg

Browse Source
This commit is contained in:
Youngjoon Lee 2024-05-13 14:12:10 +09:00
parent 2d79ff8672
commit ca2fb46c86
No known key found for this signature in database
GPG Key ID: 09B750B5BD6F08A2
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
mixnet/v2/sim/node.py
View File

@ -13,6 +13,7 @@ class Node:
INCENTIVE_TX_SIZE = 512
REAL_PAYLOAD = b"BLOCK"
COVER_PAYLOAD = b"COVER"
PADDING_SEPARATOR = b'\x01'
def __init__(self, id: int, env: simpy.Environment, p2p: P2p):
self.id = id
@ -60,7 +61,12 @@ class Node:
self.log("Receiving SphinxPacket. It's mine!")
if msg.is_all_unwrapped():
if msg.payload == self.REAL_PAYLOAD:
self.env.process(self.p2p.broadcast(msg.payload))
# Pad the final msg to the same size as a SphinxPacket,
# assuming that the final msg is going to be sent via secure channels (TLS, Noise, etc.)
final_padded_msg = (msg.payload
+ self.PADDING_SEPARATOR
+ bytes(len(msg) - len(msg.payload) - len(self.PADDING_SEPARATOR)))
self.env.process(self.p2p.broadcast(final_padded_msg))
else:
# TODO: use Poisson delay
yield self.env.timeout(random.randint(0, 5))
@ -68,7 +74,8 @@ class Node:
else:
self.log("Receiving SphinxPacket, but not mine")
else:
self.log("Received original message: %s" % msg)
final_msg = msg[:msg.rfind(self.PADDING_SEPARATOR)]
self.log("Received final message: %s" % final_msg)
# TODO: This is a dummy logic
@classmethod