import shutil
def setup_dir(dir, pcap):
+ """Setup a socket wrapper directory.
+
+ :param dir: Socket wrapper directory (None if socket wrapper should be
+ disabled)
+ :param pcap: Whether to generate pcap files
+ :return: The socket wrapper directory
+ """
pcap_dir = None
if dir is not None:
shutil.rmtree(dir)
os.mkdir(dir, 0777)
- if pcap is not None:
+ if pcap:
pcap_dir = os.path.join(dir, "pcap")
if os.path.isdir(pcap_dir):
os.environ["SOCKET_WRAPPER_PCAP_FILE"] = pcap_file
def set_default_iface(i):
- os.environ["SOCKET_WRAPPER_DEFAULT_IFACE"] = i
+ os.environ["SOCKET_WRAPPER_DEFAULT_IFACE"] = str(i)