def getTrace(visit_file, direction=None): with open(visit_file, 'r') as f: sample = json.load(f) Config.hostname.update(sample[u'ip_to_name']) webId, traceId = sample['visit_log'][u'current_url'], sample['visit_log']['visit_id'] trace = Trace(traceId, webId) for tcp_conn in sample['tcp_connections']: connection_id = tcp_conn['connection_id'] TCP = TCPConnection(connection_id, webId, hostip=sample['visit_log'][u'host_ip']) for pkt in tcp_conn['packets']: pkt_time, pkt_size = pkt[0], abs(pkt[1]) if pkt_size == 0: continue pkt_dir = Packet.Outgoing if pkt[1] < 0 else Packet.Incoming if direction is None or direction == pkt_dir: TCP.addPacket(Packet(pkt_time, pkt_size, pkt_dir)) TCP._packets.sort(key=lambda x: x.getTime()) trace.addTcpCon(TCP) return trace
def setUp(self): self.tcp = TCPConnection()
from UDPConnection import UDPConnection from TCPConnection import TCPConnection from AudioServer import AudioSocketServer HOST = socket.gethostbyname(socket.gethostname()) PORT = 50007 LISTEN_ALL_HOST = '0.0.0.0' app = QtWidgets.QApplication(sys.argv) application = MainWindow() application.show() UDP_socket = UDPConnection(HOST, PORT) UDP_socket.setsockopt_reuseaddr() UDP_socket.bind(LISTEN_ALL_HOST, PORT) UDP_socket.send_address_to_sender() UDP_socket.start_UDP_receiving() TCP_socket = TCPConnection(application.signal.new_mes) TCP_socket.setsockopt_reuseaddr() TCP_socket.bind(HOST, PORT) TCP_socket.listen(10) TCP_socket.set_server_socket() application.set_socket(TCP_socket) application.start_threading() audio_socket = AudioSocketServer(HOST, PORT + 1) sys.exit(app.exec())