0xb00b5/firegex-traffic-viewer
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added tests for Porthijack

Browse Source
This commit is contained in:
nik012003
2022-08-12 16:30:43 +02:00
parent 25d2cfd562
commit 50b4871dfb
5 changed files with 179 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/utils/udpserver.py
View File

@@ -2,7 +2,7 @@ from multiprocessing import Process
import socket
class UdpServer:
def __init__(self,port,ipv6):
def __init__(self,port,ipv6, proxy_port = None):
def _startServer(port):
sock = socket.socket(socket.AF_INET6 if ipv6 else socket.AF_INET, socket.SOCK_DGRAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
@@ -15,6 +15,7 @@ class UdpServer:
self.ipv6 = ipv6
self.port = port
self.proxy_port = proxy_port
self.server = Process(target=_startServer,args=[port])
def start(self):
@@ -26,7 +27,7 @@ class UdpServer:
def sendCheckData(self,data):
s = socket.socket(socket.AF_INET6 if self.ipv6 else socket.AF_INET, socket.SOCK_DGRAM)
s.settimeout(2)
s.sendto(data, ('::1' if self.ipv6 else '127.0.0.1', self.port))
s.sendto(data, ('::1' if self.ipv6 else '127.0.0.1', self.proxy_port if self.proxy_port else self.port))
try:
received_data = s.recvfrom(432)
except Exception: