tests/hwsim/hwsim_utils.py

   1 # hwsim testing utilities
   2 # Copyright (c) 2013, Jouni Malinen <j@w1.fi>
   3 #
   4 # This software may be distributed under the terms of the BSD license.
   5 # See README for more details.
   6
   7 import os
   8 import subprocess
   9 import logging
  10 logger = logging.getLogger()
  11
  12 def test_connectivity(ifname1, ifname2, dscp=None, tos=None):
  13     if os.path.isfile("../../mac80211_hwsim/tools/hwsim_test"):
  14         hwsim_test = "../../mac80211_hwsim/tools/hwsim_test"
  15     else:
  16         hwsim_test = "hwsim_test"
  17     cmd = ["sudo",
  18            hwsim_test,
  19            ifname1,
  20            ifname2]
  21     if dscp:
  22         cmd.append('-D')
  23         cmd.append(str(dscp))
  24     elif tos:
  25         cmd.append('-t')
  26         cmd.append(str(tos))
  27     try:
  28         s = subprocess.check_output(cmd)
  29         logger.debug(s)
  30     except subprocess.CalledProcessError, e:
  31         logger.info("hwsim failed: " + str(e.returncode))
  32         logger.info(e.output)
  33         raise
  34
  35 def test_connectivity_p2p(dev1, dev2, dscp=None, tos=None):
  36     ifname1 = dev1.group_ifname if dev1.group_ifname else dev1.ifname
  37     ifname2 = dev2.group_ifname if dev2.group_ifname else dev2.ifname
  38     test_connectivity(ifname1, ifname2, dscp, tos)
  39
  40 def test_connectivity_p2p_sta(dev1, dev2, dscp=None, tos=None):
  41     ifname1 = dev1.group_ifname if dev1.group_ifname else dev1.ifname
  42     ifname2 = dev2.ifname
  43     test_connectivity(ifname1, ifname2, dscp, tos)
  44
  45 def test_connectivity_sta(dev1, dev2, dscp=None, tos=None):
  46     ifname1 = dev1.ifname
  47     ifname2 = dev2.ifname
  48     test_connectivity(ifname1, ifname2, dscp, tos)