|
3 | 3 | import socket |
4 | 4 | import socks |
5 | 5 |
|
| 6 | + |
| 7 | +SERVER_IP = '127.0.0.1' |
| 8 | +SERVER_PORT = 1081 |
| 9 | + |
| 10 | + |
6 | 11 | if __name__ == '__main__': |
7 | 12 | # Test 1: same source port IPv4 |
8 | 13 | sock_out = socks.socksocket(socket.AF_INET, socket.SOCK_DGRAM, |
9 | 14 | socket.SOL_UDP) |
10 | | - sock_out.set_proxy(socks.SOCKS5, '127.0.0.1', 1081) |
| 15 | + sock_out.set_proxy(socks.SOCKS5, SERVER_IP, SERVER_PORT) |
11 | 16 | sock_out.bind(('127.0.0.1', 9000)) |
12 | 17 |
|
13 | 18 | sock_in1 = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, |
|
35 | 40 | # try again from the same port but IPv6 |
36 | 41 | sock_out = socks.socksocket(socket.AF_INET, socket.SOCK_DGRAM, |
37 | 42 | socket.SOL_UDP) |
38 | | - sock_out.set_proxy(socks.SOCKS5, '127.0.0.1', 1081) |
| 43 | + sock_out.set_proxy(socks.SOCKS5, SERVER_IP, SERVER_PORT) |
39 | 44 | sock_out.bind(('127.0.0.1', 9000)) |
40 | 45 |
|
41 | 46 | sock_in1 = socket.socket(socket.AF_INET6, socket.SOCK_DGRAM, |
|
62 | 67 | # Test 3: different source ports IPv6 |
63 | 68 | sock_out = socks.socksocket(socket.AF_INET, socket.SOCK_DGRAM, |
64 | 69 | socket.SOL_UDP) |
65 | | - sock_out.set_proxy(socks.SOCKS5, '127.0.0.1', 1081) |
| 70 | + sock_out.set_proxy(socks.SOCKS5, SERVER_IP, SERVER_PORT) |
66 | 71 | sock_out.bind(('127.0.0.1', 9003)) |
67 | 72 |
|
68 | 73 | sock_in1 = socket.socket(socket.AF_INET6, socket.SOCK_DGRAM, |
|
0 commit comments