fix: fix int to bytes

2023-11-22 16:52:13 +08:00 · 2023-11-22 16:52:13 +08:00 · 487743be6a
commit 487743be6a
parent 329bf9805d
1 changed files with 2 additions and 1 deletions
--- a/src/client.py
+++ b/src/client.py
@ -210,7 +210,8 @@ def check_merge(ct: int, ip: str):
        temp_cfrag_cts = []
        for i in cfrag_cts:
            capsule = pickle.loads(i[0])
-            temp_cfrag_cts.append((capsule, int(i[1]).to_bytes(32)))
+            byte_length = (ct.bit_length() + 7) // 8
+            temp_cfrag_cts.append((capsule, int(i[1]).to_bytes(byte_length)))

        cfrags = mergecfrag(temp_cfrag_cts)