codekingpro/portable-devtools

1from mitmproxy import dns2from mitmproxy import http3 4 5def tdnsreq(**kwargs) -> dns.DNSMessage:6    default = dict(7        timestamp=946681200,8        id=42,9        query=True,10        op_code=dns.op_codes.QUERY,11        authoritative_answer=False,12        truncation=False,13        recursion_desired=True,14        recursion_available=False,15        reserved=0,16        response_code=dns.response_codes.NOERROR,17        questions=[dns.Question("dns.google", dns.types.A, dns.classes.IN)],18        answers=[],19        authorities=[],20        additionals=[],21    )22    default.update(kwargs)23    return dns.DNSMessage(**default)  # type: ignore24 25 26def tdnsresp(**kwargs) -> dns.DNSMessage:27    default = dict(28        timestamp=946681201,29        id=42,30        query=False,31        op_code=dns.op_codes.QUERY,32        authoritative_answer=False,33        truncation=False,34        recursion_desired=True,35        recursion_available=True,36        reserved=0,37        response_code=dns.response_codes.NOERROR,38        questions=[dns.Question("dns.google", dns.types.A, dns.classes.IN)],39        answers=[40            dns.ResourceRecord(41                "dns.google", dns.types.A, dns.classes.IN, 32, b"\x08\x08\x08\x08"42            ),43            dns.ResourceRecord(44                "dns.google", dns.types.A, dns.classes.IN, 32, b"\x08\x08\x04\x04"45            ),46        ],47        authorities=[],48        additionals=[],49    )50    default.update(kwargs)51    return dns.DNSMessage(**default)  # type: ignore52 53 54def treq(**kwargs) -> http.Request:55    """56    Returns:57        mitmproxy.net.http.Request58    """59    default = dict(60        host="address",61        port=22,62        method=b"GET",63        scheme=b"http",64        authority=b"",65        path=b"/path",66        http_version=b"HTTP/1.1",67        headers=http.Headers(((b"header", b"qvalue"), (b"content-length", b"7"))),68        content=b"content",69        trailers=None,70        timestamp_start=946681200,71        timestamp_end=946681201,72    )73    default.update(kwargs)74    return http.Request(**default)  # type: ignore75 76 77def tresp(**kwargs) -> http.Response:78    """79    Returns:80        mitmproxy.net.http.Response81    """82    default = dict(83        http_version=b"HTTP/1.1",84        status_code=200,85        reason=b"OK",86        headers=http.Headers(87            ((b"header-response", b"svalue"), (b"content-length", b"7"))88        ),89        content=b"message",90        trailers=None,91        timestamp_start=946681202,92        timestamp_end=946681203,93    )94    default.update(kwargs)95    return http.Response(**default)  # type: ignore96