Files
firegex-traffic-viewer/tests/utils/tcpserver.py
2022-08-01 23:59:32 +02:00

31 lines
993 B
Python

from multiprocessing import Process
import socket
class TcpServer:
def __init__(self,port):
def _startServer(port):
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind(('127.0.0.1', port))
sock.listen(8)
while True:
connection,address = sock.accept()
buf = connection.recv(4096)
connection.send(buf)
connection.close()
self.port = port
self.server = Process(target=_startServer,args=[port])
def start(self):
self.server.start()
def stop(self):
self.server.terminate()
def sendCheckData(self,data):
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(('localhost', self.port))
s.sendall(data)
received_data = s.recv(4096)
s.close()
return received_data == data