コード例 #1
0
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,
コード例 #2
0
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))