2025-12-25 upload

2025-12-25 11:16:59 +08:00
commit 322ac74336
2241 changed files with 639966 additions and 0 deletions
--- a/venv/Lib/site-packages/pydivert/tests/test_packet.py
+++ b/venv/Lib/site-packages/pydivert/tests/test_packet.py
@@ -0,0 +1,517 @@
+# -*- coding: utf-8 -*-
+# Copyright (C) 2016  Fabio Falcinelli, Maximilian Hils
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+import socket
+
+import pydivert
+import pytest
+from hypothesis import given, example
+from hypothesis.strategies import binary
+from pydivert import util
+from pydivert.consts import Protocol, Direction
+
+
+def p(raw):
+    return pydivert.Packet(raw, (0, 0), Direction.OUTBOUND)
+
+
+ipv4_hdr = util.fromhex("45200028fa8d40002906368b345ad4f0c0a856a4")
+ipv6_hdr = util.fromhex("600d684a00280640fc000002000000020000000000000001fc000002000000010000000000000001")
+
+
+@given(raw=binary(0, 500, 1600))
+@example(raw=b'`\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')
+@example(raw=b'E\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')
+def test_fuzz(raw):
+    assert repr(p(raw))
+    assert repr(p(ipv4_hdr + raw))
+    assert repr(p(ipv6_hdr + raw))
+
+
+def test_ipv6_tcp():
+    raw = util.fromhex("600d684a007d0640fc000002000000020000000000000001fc000002000000010000000000000001a9a01f90021b638"
+                       "dba311e8e801800cfc92e00000101080a801da522801da522474554202f68656c6c6f2e74787420485454502f312e31"
+                       "0d0a557365722d4167656e743a206375726c2f372e33382e300d0a486f73743a205b666330303a323a303a313a3a315"
+                       "d3a383038300d0a4163636570743a202a2f2a0d0a0d0a")
+    x = p(raw)
+    assert x.address_family == socket.AF_INET6
+    assert x.protocol[0] == Protocol.TCP
+    assert x.src_addr == "fc00:2:0:2::1"
+    assert x.dst_addr == "fc00:2:0:1::1"
+    assert x.src_port == 43424
+    assert x.dst_port == 8080
+    assert not x.ipv4
+    assert x.ipv6
+    assert x.tcp
+    assert not x.udp
+    assert not x.icmp
+    assert x.payload == (
+        b"GET /hello.txt HTTP/1.1\r\n"
+        b"User-Agent: curl/7.38.0\r\n"
+        b"Host: [fc00:2:0:1::1]:8080\r\n"
+        b"Accept: */*\r\n\r\n"
+    )
+    assert x.ip.packet_len == 165
+    assert repr(x)
+
+
+def test_ipv4_udp():
+    raw = util.fromhex("4500004281bf000040112191c0a82b09c0a82b01c9dd0035002ef268528e01000001000000000000013801380138013"
+                       "807696e2d61646472046172706100000c0001")
+    x = p(raw)
+    assert x.address_family == socket.AF_INET
+    assert x.protocol[0] == Protocol.UDP
+    assert x.src_addr == "192.168.43.9"
+    assert x.dst_addr == "192.168.43.1"
+    assert x.src_port == 51677
+    assert x.dst_port == 53
+    assert x.ipv4
+    assert not x.ipv6
+    assert not x.tcp
+    assert x.udp
+    assert not x.icmp
+    assert x.payload == util.fromhex("528e01000001000000000000013801380138013807696e2d61646472046172706100000c0001")
+    assert x.udp.payload_len == 38
+    assert repr(x)
+
+
+def test_icmp_ping():
+    raw = util.fromhex("4500005426ef0000400157f9c0a82b09080808080800bbb3d73b000051a7d67d000451e408090a0b0c0d0e0f1011121"
+                       "31415161718191a1b1c1d1e1f202122232425262728292a2b2c2d2e2f3031323334353637")
+    x = p(raw)
+    assert x.address_family == socket.AF_INET
+    assert x.protocol[0] == Protocol.ICMP
+    assert x.src_addr == "192.168.43.9"
+    assert x.dst_addr == "8.8.8.8"
+    assert x.src_port is None
+    assert x.dst_port is None
+    assert x.icmp.type == 8
+    assert x.icmp.code == 0
+    assert x.ipv4
+    assert not x.ipv6
+    assert not x.tcp
+    assert not x.udp
+    assert x.icmpv4
+    assert not x.icmpv6
+    assert x.payload == util.fromhex("d73b000051a7d67d000451e408090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f202122232"
+                                     "425262728292a2b2c2d2e2f3031323334353637")
+    assert repr(x)
+
+
+def test_icmpv6_unreachable():
+    raw = util.fromhex("6000000000443a3d3ffe05010410000002c0dffffe47033e3ffe050700000001020086fffe0580da010413520000000"
+                       "060000000001411013ffe050700000001020086fffe0580da3ffe05010410000002c0dffffe47033ea07582a40014cf"
+                       "470a040000f9c8e7369d250b00")
+    x = p(raw)
+    assert x.address_family == socket.AF_INET6
+    assert x.protocol[0] == Protocol.ICMPV6
+    assert x.src_addr == "3ffe:501:410:0:2c0:dfff:fe47:33e"
+    assert x.dst_addr == "3ffe:507:0:1:200:86ff:fe05:80da"
+    assert x.src_port is None
+    assert x.dst_port is None
+    assert x.icmp.type == 1
+    assert x.icmp.code == 4
+    assert not x.ipv4
+    assert x.ipv6
+    assert not x.tcp
+    assert not x.udp
+    assert not x.icmpv4
+    assert x.icmpv6
+    assert x.payload == util.fromhex("0000000060000000001411013ffe050700000001020086fffe0580da3ffe05010410000002c0dffff"
+                                     "e47033ea07582a40014cf470a040000f9c8e7369d250b00")
+    assert repr(x)
+
+
+def test_ipv4_tcp_modify():
+    raw = util.fromhex("45000051476040008006f005c0a856a936f274fdd84201bb0876cfd0c19f9320501800ff8dba0000170303002400000"
+                       "00000000c2f53831a37ed3c3a632f47440594cab95283b558bf82cb7784344c3314")
+    x = p(raw)
+    assert x.protocol[0] == Protocol.TCP
+
+    # src_addr
+    x.src_addr = "1.2.3.4"
+    with pytest.raises(Exception):
+        x.src_addr = "::1"
+    with pytest.raises(Exception):
+        x.src_addr = 42
+    assert x.src_addr == "1.2.3.4"
+
+    # dst_addr
+    x.dst_addr = "4.3.2.1"
+    with pytest.raises(Exception):
+        x.dst_addr = "::1"
+    assert x.dst_addr == "4.3.2.1"
+
+    # src_port
+    x.src_port = 42
+    with pytest.raises(Exception):
+        x.src_port = "bogus"
+    assert x.src_port == 42
+
+    # dst_port
+    x.dst_port = 43
+    with pytest.raises(Exception):
+        x.dst_port = "bogus"
+    assert x.dst_port == 43
+
+    # tcp_ack (others follow trivially)
+    x.tcp.ack = False
+    assert x.tcp.ack is False
+    x.tcp.ack = True
+    assert x.tcp.ack is True
+
+    # payload
+    x.payload = b"test"
+    with pytest.raises(Exception):
+        x.payload = 42
+    assert x.payload == b"test"
+
+    # checksum
+    a = x.raw.tobytes()
+    assert x.recalculate_checksums(
+        pydivert.CalcChecksumsOption.NO_IP_CHECKSUM |
+        pydivert.CalcChecksumsOption.NO_TCP_CHECKSUM
+    ) == 0
+    assert x.raw.tobytes() == a
+
+    assert x.recalculate_checksums() == 2
+    assert x.raw.tobytes() != a
+
+    # test same length raw replace.
+    x.tcp.raw = x.tcp.raw.tobytes().replace(b"test", b"abcd")
+
+    # catch typo in headers
+    with pytest.raises(AttributeError):
+        x.tcp.typo = 42
+
+
+def test_ipv6_udp_modify():
+    raw = util.fromhex("60000000002711403ffe050700000001020086fffe0580da3ffe0501481900000000000000000042095d0035002746b"
+                       "700060100000100000000000003777777057961686f6f03636f6d00000f0001")
+    x = p(raw)
+    assert x.protocol[0] == Protocol.UDP
+
+    # src_addr
+    x.src_addr = "::1"
+    with pytest.raises(Exception):
+        x.src_addr = "127.0.0.1"
+    with pytest.raises(Exception):
+        x.src_addr = 42
+    assert x.src_addr == "::1"
+
+    # dst_addr
+    x.dst_addr = "::2"
+    with pytest.raises(Exception):
+        x.dst_addr = "bogus"
+    assert x.dst_addr == "::2"
+
+    # src_port
+    x.src_port = 42
+    with pytest.raises(Exception):
+        x.src_port = "bogus"
+    assert x.src_port == 42
+
+    # dst_port
+    x.dst_port = 43
+    with pytest.raises(Exception):
+        x.dst_port = "bogus"
+    assert x.dst_port == 43
+
+    # payload
+    x.payload = b"test"
+    with pytest.raises(Exception):
+        x.payload = 42
+    assert x.payload == b"test"
+
+    # checksum
+    a = x.raw.tobytes()
+    assert x.recalculate_checksums(
+        pydivert.CalcChecksumsOption.NO_IP_CHECKSUM |
+        pydivert.CalcChecksumsOption.NO_UDP_CHECKSUM
+    ) == 0
+    assert x.raw.tobytes() == a
+
+    assert x.recalculate_checksums() == 1
+    assert x.raw.tobytes() != a
+
+
+def test_icmp_modify():
+    raw = util.fromhex("4500005426ef0000400157f9c0a82b09080808080800bbb3d73b000051a7d67d000451e408090a0b0c0d0e0f1011121"
+                       "31415161718191a1b1c1d1e1f202122232425262728292a2b2c2d2e2f3031323334353637")
+    x = p(raw)
+    assert x.protocol[0] == Protocol.ICMP
+
+    # src_addr
+    x.src_addr = "1.2.3.4"
+    with pytest.raises(Exception):
+        x.src_addr = "::1"
+    with pytest.raises(Exception):
+        x.src_addr = 42
+    assert x.src_addr == "1.2.3.4"
+
+    # dst_addr
+    x.dst_addr = "4.3.2.1"
+    with pytest.raises(Exception):
+        x.dst_addr = "::1"
+    assert x.dst_addr == "4.3.2.1"
+
+    # payload
+    x.payload = b"test"
+    with pytest.raises(Exception):
+        x.payload = 42
+    assert x.payload == b"test"
+
+    # icmp
+    x.icmp.type = 42
+    with pytest.raises(Exception):
+        x.icmp.type = "bogus"
+    assert x.icmp.type == 42
+    x.icmp.code = 42
+    with pytest.raises(Exception):
+        x.icmp.code = "bogus"
+    assert x.icmp.code == 42
+
+    # checksum
+    a = x.raw.tobytes()
+    assert x.recalculate_checksums(
+        pydivert.CalcChecksumsOption.NO_IP_CHECKSUM |
+        pydivert.CalcChecksumsOption.NO_ICMP_CHECKSUM
+    ) == 0
+    assert x.raw.tobytes() == a
+
+    assert x.recalculate_checksums() == 2
+    assert x.raw.tobytes() != a
+
+
+def test_meta():
+    p = pydivert.Packet(b"", (1, 1), Direction.OUTBOUND)
+    assert p.is_outbound
+    assert not p.is_inbound
+    assert p.is_loopback
+
+    p2 = pydivert.Packet(b"", (2, 2), Direction.INBOUND)
+    assert not p2.is_outbound
+    assert p2.is_inbound
+    assert not p2.is_loopback
+
+
+def test_bogus():
+    x = p(b"")
+    with pytest.raises(Exception):
+        x.src_addr = "127.0.0.1"
+    with pytest.raises(Exception):
+        x.dst_addr = "127.0.0.1"
+    with pytest.raises(Exception):
+        x.src_port = 80
+    with pytest.raises(Exception):
+        x.dst_port = 80
+    with pytest.raises(Exception):
+        x.payload = b""
+    with pytest.raises(Exception):
+        x.icmp.code = 42
+    with pytest.raises(Exception):
+        x.tcp.ack = True
+    with pytest.raises(Exception):
+        x.tcp.unknown_attr = True
+    assert x.recalculate_checksums() == 0
+
+
+def test_ipv6_extension_headers():
+    # AH Header
+    raw = util.fromhex("6e000000003c3301fe800000000000000000000000000001ff020000000000000000000000000005590400000000010"
+                       "00000001321d3a95c5ffd4d184622b9f8030100240101010100000001fb8600000000000501000013000a0028000000"
+                       "0000000000")
+    assert p(raw).protocol[0] == 89
+
+    # Fragmented...
+    raw = util.fromhex("6000000005b02c80fe8000000000000002105afffeaa20a2fe800000000000000250dafffed8c1533a0000010000000"
+                       "580009e9d0000000d6162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70"
+                       "717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717"
+                       "273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273"
+                       "747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747"
+                       "576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576"
+                       "776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776"
+                       "162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776162"
+                       "636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636"
+                       "465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465"
+                       "666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666"
+                       "768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768"
+                       "696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696"
+                       "a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b"
+                       "6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6"
+                       "d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e"
+                       "6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f7"
+                       "0717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f7071"
+                       "7273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f7071727"
+                       "3747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f7071727374"
+                       "7576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f7071727374757"
+                       "6776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f7071727374757677"
+                       "6162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f7071727374757677616"
+                       "2636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f7071727374757677616263"
+                       "6465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f7071727374757677616263646"
+                       "5666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f7071727374757677616263646566"
+                       "6768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f7071727374757677616263646566676"
+                       "8696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f7071727374757677616263646566676869"
+                       "6a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6"
+                       "b6c6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c"
+                       "6d6e6f70717273747576776162636465666768696a6b6c6d6e6f70717273747576776162636465666768696a6b6c6d6"
+                       "e6f70717273747576776162636465666768696a6b6c6d6e")
+    assert p(raw).protocol[0] == Protocol.ICMPV6
+
+    # HOPOPTS
+    raw = util.fromhex("600000000020000100000000000000000000000000000000ff0200000000000000000000000000013a0005020000000"
+                       "082007ac103e8000000000000000000000000000000000000")
+    assert p(raw).protocol[0] == Protocol.ICMPV6
+
+
+def test_ipv4_fields():
+    raw = util.fromhex("4500005426ef0000400157f9c0a82b09080808080800bbb3d73b000051a7d67d000451e408090a0b0c0d0e0f1011121"
+                       "31415161718191a1b1c1d1e1f202122232425262728292a2b2c2d2e2f3031323334353637")
+    ip = p(raw).ipv4
+
+    assert not ip.df
+    ip.df = True
+    assert ip.df
+    assert ip.flags == 2
+    assert ip.frag_offset == 0
+    ip.flags = 3
+    assert ip.flags == 3
+    assert ip.mf
+    assert ip.df
+    assert ip.frag_offset == 0
+    ip.ecn = 3
+    assert ip.ecn == 3
+    ip.dscp = 18
+    assert ip.dscp == 18
+    assert ip.diff_serv == ip.dscp
+    assert ip.ecn == 3
+    assert ip.tos == 75
+    ip.tos = 1
+    assert ip.tos == 1
+    assert ip.ecn == 1
+    assert ip.dscp == 0
+    ip.flags = 1
+    assert ip.mf
+    ip.mf = False
+    assert not ip.mf
+    assert ip.flags == 0
+    ip.frag_offset = 65
+    assert ip.frag_offset == 65
+    assert ip.flags == 0
+    ip.flags = 7
+    assert ip.frag_offset == 65
+    assert ip.evil
+    assert ip.reserved == ip.evil
+    ip.evil = False
+    assert not ip.evil
+    assert ip.reserved == ip.evil
+    assert ip.flags == 3
+    ip.ident = 257
+    assert ip.ident == 257
+    assert ip.hdr_len == 5
+    ip.cksum = 514
+    assert ip.cksum == 514
+    ip.hdr_len = 6
+    assert ip.hdr_len == 6
+    assert ip.header_len == 6 * 4
+    ip.ttl = 4
+    assert ip.ttl == 4
+    ip.protocol = Protocol.FRAGMENT
+    assert ip.protocol == Protocol.FRAGMENT
+    with pytest.raises(ValueError):
+        ip.hdr_len = 4
+
+
+def test_ipv6_fields():
+    raw = util.fromhex("6e000000003c3301fe800000000000000000000000000001ff020000000000000000000000000005590400000000010"
+                       "00000001321d3a95c5ffd4d184622b9f8030100240101010100000001fb8600000000000501000013000a0028000000"
+                       "0000000000")
+    ip = p(raw).ipv6
+
+    ip.traffic_class = 3
+    assert ip.traffic_class == 3
+    assert ip.ecn == 3
+    ip.ecn = 0
+    assert ip.ecn == 0
+    assert ip.traffic_class == 0
+    ip.diff_serv = 8
+    assert ip.diff_serv == 8
+    assert ip.traffic_class == 32
+    ip.flow_label = 17
+    assert ip.flow_label == 17
+    assert ip.traffic_class == 32
+
+
+def test_icmp_fields():
+    raw = util.fromhex("4500005426ef0000400157f9c0a82b09080808080800bbb3d73b000051a7d67d000451e408090a0b0c0d0e0f1011121"
+                       "31415161718191a1b1c1d1e1f202122232425262728292a2b2c2d2e2f3031323334353637")
+    icmp = p(raw).icmp
+
+    icmp.cksum = 11
+    assert icmp.cksum == 11
+
+
+def test_tcp_fields():
+    raw = util.fromhex("45000051476040008006f005c0a856a936f274fdd84201bb0876cfd0c19f9320501800ff8dba0000170303002400000"
+                       "00000000c2f53831a37ed3c3a632f47440594cab95283b558bf82cb7784344c3314")
+    tcp = p(raw).tcp
+
+    assert tcp.reserved == 0
+    tcp.reserved = 7
+    assert tcp.reserved == 7
+    assert not tcp.ns
+    tcp.ns = True
+    assert tcp.ns
+    assert tcp.reserved == 0b111
+    assert tcp.header_len == tcp.data_offset * 4
+    tcp.data_offset = 5
+    assert tcp.data_offset == 5
+    with pytest.raises(ValueError):
+        tcp.data_offset = 4
+    with pytest.raises(ValueError):
+        tcp.data_offset = 16
+
+    tcp.cwr = True
+    assert tcp.cwr
+    tcp.ece = True
+    assert tcp.ece
+    tcp.syn = True
+    tcp.control_bits = 0x01F0
+    assert not tcp.fin
+    assert not tcp.syn
+    assert tcp.control_bits == 0x01F0
+    assert tcp.ece
+    assert tcp.ns
+    tcp.ns = False
+    assert tcp.control_bits == 0x00F0
+
+
+def test_udp_fields():
+    raw = util.fromhex("4500004281bf000040112191c0a82b09c0a82b01c9dd0035002ef268528e01000001000000000000013801380138013"
+                       "807696e2d61646472046172706100000c0001")
+    udp = p(raw).udp
+
+    udp.cksum = 0xAAAA
+    assert udp.cksum == 0xAAAA
+
+
+def test_filter_match():
+    raw = util.fromhex("4500004281bf000040112191c0a82b09c0a82b01c9dd0035002ef268528e01000001000000000000013801380138013"
+                       "807696e2d61646472046172706100000c0001")
+    p = pydivert.Packet(raw, (1, 1), Direction.OUTBOUND)
+
+    assert p.matches("true")
+    assert p.matches("udp and outbound")
+    assert not p.matches("tcp")