Function pack_message

mitmproxy/proxy/layers/dns.py:46–53

(
    message: dns.DNSMessage, transport_protocol: Literal["tcp", "udp"]
)

def pack_message(
    message: dns.DNSMessage, transport_protocol: Literal["tcp", "udp"]
) -> bytes:
    packed = message.packed
    if transport_protocol == "tcp":
        return struct.pack("!H", len(packed)) + packed
    else:
        return packed


class DNSLayer(layer.Layer):