tests/hwsim/test_p2p_discovery.py

   1 #!/usr/bin/python
   2 #
   3 # P2P device discovery test cases
   4 # Copyright (c) 2013, Jouni Malinen <j@w1.fi>
   5 #
   6 # This software may be distributed under the terms of the BSD license.
   7 # See README for more details.
   8
   9 import logging
  10 logger = logging.getLogger(__name__)
  11
  12 import hwsim_utils
  13
  14 def test_discovery(dev):
  15     """P2P device discovery and provision discovery"""
  16     addr0 = dev[0].p2p_dev_addr()
  17     addr1 = dev[1].p2p_dev_addr()
  18     logger.info("Start device discovery")
  19     dev[0].p2p_find(social=True)
  20     if not dev[1].discover_peer(addr0):
  21         raise Exception("Device discovery timed out")
  22     if not dev[0].discover_peer(addr1):
  23         raise Exception("Device discovery timed out")
  24
  25     logger.info("Test provision discovery for display")
  26     dev[0].global_request("P2P_PROV_DISC " + addr1 + " display")
  27     ev1 = dev[1].wait_global_event(["P2P-PROV-DISC-SHOW-PIN"], timeout=15)
  28     if ev1 is None:
  29         raise Exception("Provision discovery timed out (display/dev1)")
  30     if addr0 not in ev1:
  31         raise Exception("Dev0 not in provision discovery event")
  32     ev0 = dev[0].wait_global_event(["P2P-PROV-DISC-ENTER-PIN",
  33                                     "P2P-PROV-DISC-FAILURE"], timeout=15)
  34     if ev0 is None:
  35         raise Exception("Provision discovery timed out (display/dev0)")
  36     if "P2P-PROV-DISC-FAILURE" in ev0:
  37         raise Exception("Provision discovery failed (display/dev0)")
  38     if addr1 not in ev0:
  39         raise Exception("Dev1 not in provision discovery event")
  40
  41     logger.info("Test provision discovery for keypad")
  42     dev[0].global_request("P2P_PROV_DISC " + addr1 + " keypad")
  43     ev1 = dev[1].wait_global_event(["P2P-PROV-DISC-ENTER-PIN"], timeout=15)
  44     if ev1 is None:
  45         raise Exception("Provision discovery timed out (keypad/dev1)")
  46     if addr0 not in ev1:
  47         raise Exception("Dev0 not in provision discovery event")
  48     ev0 = dev[0].wait_global_event(["P2P-PROV-DISC-SHOW-PIN",
  49                                     "P2P-PROV-DISC-FAILURE"],
  50                                    timeout=15)
  51     if ev0 is None:
  52         raise Exception("Provision discovery timed out (keypad/dev0)")
  53     if "P2P-PROV-DISC-FAILURE" in ev0:
  54         raise Exception("Provision discovery failed (keypad/dev0)")
  55     if addr1 not in ev0:
  56         raise Exception("Dev1 not in provision discovery event")
  57
  58     logger.info("Test provision discovery for push button")
  59     dev[0].global_request("P2P_PROV_DISC " + addr1 + " pbc")
  60     ev1 = dev[1].wait_global_event(["P2P-PROV-DISC-PBC-REQ"], timeout=15)
  61     if ev1 is None:
  62         raise Exception("Provision discovery timed out (pbc/dev1)")
  63     if addr0 not in ev1:
  64         raise Exception("Dev0 not in provision discovery event")
  65     ev0 = dev[0].wait_global_event(["P2P-PROV-DISC-PBC-RESP",
  66                                     "P2P-PROV-DISC-FAILURE"],
  67                                    timeout=15)
  68     if ev0 is None:
  69         raise Exception("Provision discovery timed out (pbc/dev0)")
  70     if "P2P-PROV-DISC-FAILURE" in ev0:
  71         raise Exception("Provision discovery failed (pbc/dev0)")
  72     if addr1 not in ev0:
  73         raise Exception("Dev1 not in provision discovery event")
  74
  75     dev[0].p2p_stop_find
  76     dev[1].p2p_stop_find
  77
  78 def test_discovery_group_client(dev):
  79     """P2P device discovery for a client in a group"""
  80     logger.info("Start autonomous GO " + dev[0].ifname)
  81     res = dev[0].p2p_start_go(freq="2422")
  82     logger.debug("res: " + str(res))
  83     logger.info("Connect a client to the GO")
  84     pin = dev[1].wps_read_pin()
  85     dev[0].p2p_go_authorize_client(pin)
  86     dev[1].p2p_connect_group(dev[0].p2p_dev_addr(), pin, timeout=60)
  87     logger.info("Client connected")
  88     hwsim_utils.test_connectivity_p2p(dev[0], dev[1])
  89     logger.info("Try to discover a P2P client in a group")
  90     if not dev[2].discover_peer(dev[1].p2p_dev_addr(), social=False):
  91         raise Exception("Could not discover group client")