Skip to content

Commit 3844d40

Browse files
committed
rename .next method of Counter
1 parent ce9fee9 commit 3844d40

File tree

5 files changed

+7
-7
lines changed

5 files changed

+7
-7
lines changed

Diff for: Counter.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ class Counter:
66
def __init__(self, start):
77
self.current_value = start
88

9-
def next(self):
9+
def count(self):
1010
self.current_value += 1
1111

1212
def __bytes__(self):

Diff for: LinkEncryption.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ def __init__(self, link_key):
1010
self.counter = Counter(LINK_CTR_START)
1111

1212
def encrypt(self, plain_text):
13-
self.counter.next()
13+
self.counter.count()
1414

1515
msg_type, chan_id, ctr_prefix, payload = cut(
1616
plain_text, MSG_TYPE_FLAG_LEN, CHAN_ID_SIZE, CTR_PREFIX_LEN)

Diff for: UDPChannel.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ def get_completed_responses(self):
104104
return packets
105105

106106
def _get_init_message(self):
107-
self.request_counter.next()
107+
self.request_counter.count()
108108

109109
ip, port = self.dest_addr
110110

@@ -180,7 +180,7 @@ def _receive_response_fragment(self, response):
180180
return
181181

182182
def _encrypt_fragment(self, fragment):
183-
self.request_counter.next()
183+
self.request_counter.count()
184184

185185
for key in reversed(self.sym_keys):
186186
counter = self.request_counter
@@ -288,7 +288,7 @@ def forward_response(self, response):
288288
if msg_ctr != bytes(CTR_PREFIX_LEN):
289289
self.response_replay_detector.check_replay_window(b2i(msg_ctr))
290290

291-
self.response_counter.next()
291+
self.response_counter.count()
292292

293293
cipher = ctr_cipher(self.key, int(self.response_counter))
294294

Diff for: stubs/Counter.pyi

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ class Counter:
22
current_value: int
33

44
def __init__(self, start: int) -> None: ...
5-
def next(self) -> int: ...
5+
def count(self) -> int: ...
66
def __bytes__(self) -> bytes: ...
77
def __int__(self) -> int: ...
88
def __str__(self) -> str: ...

Diff for: tests/Counter_test.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -40,4 +40,4 @@ def test_next():
4040
for i in range(0, 10000):
4141
assert i == int(counter)
4242

43-
counter.next()
43+
counter.count()

0 commit comments

Comments
 (0)