use payload_id to build a payload content unique

This commit is contained in:
Youngjoon Lee 2024-05-23 15:49:38 +09:00
parent 7820e298c8
commit 39b0a6a7ce
No known key found for this signature in database
GPG Key ID: 09B750B5BD6F08A2

View File

@ -24,6 +24,7 @@ class Node:
self.private_key = X25519PrivateKey.generate()
self.public_key = self.private_key.public_key()
self.config = config
self.payload_id = 0
self.action = self.env.process(self.send_message())
def send_message(self):
@ -100,7 +101,9 @@ class Node:
self.log("Received final message: %s" % final_msg)
def build_payload(self) -> bytes:
return b"P" + bytes(self.config.mixnet.payload_size - len(b"P"))
payload = bytes(f"{self.id}-{self.payload_id}", "utf-8")
self.payload_id += 1
return payload + bytes(self.config.mixnet.payload_size - len(payload))
def pad_payload(self, payload: bytes, target_size: int) -> bytes:
"""