import threading import inspect import os import utils import logging import time from datetime import datetime from threading import Timer test_meta_config = utils.parse_config("config/test_meta_config.json") log_level = utils.parse_logging_level( test_meta_config["general_config"]["logging_level"]) logging.basicConfig(level=log_level) logger = logging.getLogger(__name__) current_script = os.path.basename(__file__) def Threading_tcpdump_capture_cycle(task_time, pcap_result_subpath_variant, server_iperf_port, selected_network, selected_direction, selected_variant): logger.debug("{}_{}--> Thread {} Start".format( current_script, inspect.currentframe().f_lineno, inspect.stack()[0][3])) doki_timer = utils.DokiTimer(expired_time=task_time, repeat=True) output_pcap = os.path.join( pcap_result_subpath_variant, "{}.pcap".format( datetime.fromtimestamp(time.time()).strftime("%Y_%m_%d_%H_%M"))) if selected_variant == "udp": logger.debug("{}_{}--> Output pcap to {}".format( current_script,
import time import socket import select import logging import utils import os meta_config = utils.parse_config("config/test_meta_config.json") logging.basicConfig(level=utils.parse_logging_level( meta_config["general_config"]["logging_level"])) logger = logging.getLogger(__name__) current_script = os.path.basename(__file__) def doki_wait_receive_message(my_socket, timeout=120): message = "" my_socket.setblocking(0) try: while True: ready = select.select([my_socket], [], [], timeout) if ready[0]: data = my_socket.recv(1024).decode("utf-8") message = message + data if "##DOKI##" in data: break message = message.replace("##DOKI##", "") return message except Exception as e: logger.warning("{}--> Exception happen when receive message {}".format( current_script, e)) logger.debug("{}--> message is: {}".format(current_script, message))