Skip to content

Commit 1d51424

Browse files
committed
Update old all-zero counter declarations to the new counter from pycryptodome
1 parent c2c38da commit 1d51424

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/potr/crypt.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
import logging
2222
import struct
2323

24-
24+
from Crypto.Util import Counter
2525
from potr.compatcrypto import SHA256, SHA1, SHA1HMAC, SHA256HMAC, \
2626
AESCTR, PK, getrandbits, randrange
2727
from potr.utils import bytes_to_long, long_to_bytes, pack_mpi, read_mpi
@@ -69,8 +69,8 @@ def __init__(self, sendenc, sendmac, rcvenc, rcvmac):
6969
self.sendmac = sendmac
7070
self.rcvenc = rcvenc
7171
self.rcvmac = rcvmac
72-
self.sendctr = Counter(0)
73-
self.rcvctr = Counter(0)
72+
self.sendctr = Counter.new(nbits=64, prefix=long_to_bytes(0, 8), initial_value=0)
73+
self.rcvctr = Counter.new(nbits=64, prefix=long_to_bytes(0, 8), initial_value=0)
7474
self.sendmacused = False
7575
self.rcvmacused = False
7676

0 commit comments

Comments
 (0)