fragattacks/tests/hwsim/hwsim_utils.py
Jouni Malinen 5be9dcbb86 tests: Remove unnecessary interpreter line from most python files
Only run-tests.py is actually executed, so there is no need to specify
the interpreter in all the helper files and test script files.

Signed-off-by: Jouni Malinen <j@w1.fi>
2014-02-21 20:21:53 +02:00

49 lines
1.5 KiB
Python

# hwsim testing utilities
# Copyright (c) 2013, Jouni Malinen <j@w1.fi>
#
# This software may be distributed under the terms of the BSD license.
# See README for more details.
import os
import subprocess
import logging
logger = logging.getLogger()
def test_connectivity(ifname1, ifname2, dscp=None, tos=None):
if os.path.isfile("../../mac80211_hwsim/tools/hwsim_test"):
hwsim_test = "../../mac80211_hwsim/tools/hwsim_test"
else:
hwsim_test = "hwsim_test"
cmd = ["sudo",
hwsim_test,
ifname1,
ifname2]
if dscp:
cmd.append('-D')
cmd.append(str(dscp))
elif tos:
cmd.append('-t')
cmd.append(str(tos))
try:
s = subprocess.check_output(cmd)
logger.debug(s)
except subprocess.CalledProcessError, e:
logger.info("hwsim failed: " + str(e.returncode))
logger.info(e.output)
raise
def test_connectivity_p2p(dev1, dev2, dscp=None, tos=None):
ifname1 = dev1.group_ifname if dev1.group_ifname else dev1.ifname
ifname2 = dev2.group_ifname if dev2.group_ifname else dev2.ifname
test_connectivity(ifname1, ifname2, dscp, tos)
def test_connectivity_p2p_sta(dev1, dev2, dscp=None, tos=None):
ifname1 = dev1.group_ifname if dev1.group_ifname else dev1.ifname
ifname2 = dev2.ifname
test_connectivity(ifname1, ifname2, dscp, tos)
def test_connectivity_sta(dev1, dev2, dscp=None, tos=None):
ifname1 = dev1.ifname
ifname2 = dev2.ifname
test_connectivity(ifname1, ifname2, dscp, tos)