import pathlib
import re
import signal
-import struct
import sys
import dns.exception
if not wire_length_bytes:
return None
- (wire_length,) = struct.unpack("!H", wire_length_bytes)
-
- return wire_length
+ return int.from_bytes(wire_length_bytes, byteorder="big")
async def _read_tcp_query_wire(
self, reader: asyncio.StreamReader, peer: Peer, wire_length: int
if protocol == DnsProtocol.UDP:
yield response
else:
- response_length = struct.pack("!H", len(response))
- yield response_length + response
+ yield len(payload).to_bytes(2, byteorder="big") + payload
def _parse_message(self, wire: bytes) -> dns.message.Message:
try: