Skip to content

Commit

Permalink
refatcor!: Make a function name more proper
Browse files Browse the repository at this point in the history
  • Loading branch information
Szilárd Pfeiffer committed Nov 22, 2021
1 parent a0b3d85 commit c8a785f
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions dheater/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -269,7 +269,7 @@ def _prepare_packets(self):
return message_bytes

@classmethod
def _receive_and_flush_record(cls, client):
def _skip_record(cls, client):
received_byte_count = client.l4_transfer.receive(4)
received_byte_count += client.l4_transfer.receive(struct.unpack('!I', client.l4_transfer.buffer)[0])
client.l4_transfer.flush_buffer()
Expand All @@ -284,11 +284,11 @@ def _send_packets(self, client):
client.l4_transfer.flush_buffer()

# Receive key exchange init message
received_byte_count += self._receive_and_flush_record(client)
received_byte_count += self._skip_record(client)

if self.group_exchange:
# Wait for DH group exchange group message
received_byte_count += self._receive_and_flush_record(client)
received_byte_count += self._skip_record(client)

# Wait for DH group/key exchange reply by receiving record length
received_byte_count += client.l4_transfer.receive(4)
Expand Down

0 comments on commit c8a785f

Please sign in to comment.