tests/hwsim/test_p2p_discovery.py

   1 # P2P device discovery test cases
   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 logging
   8 logger = logging.getLogger()
   9
  10 import hwsim_utils
  11
  12 def test_discovery(dev):
  13     """P2P device discovery and provision discovery"""
  14     addr0 = dev[0].p2p_dev_addr()
  15     addr1 = dev[1].p2p_dev_addr()
  16     logger.info("Start device discovery")
  17     dev[0].p2p_find(social=True)
  18     if not dev[1].discover_peer(addr0):
  19         raise Exception("Device discovery timed out")
  20     if not dev[0].discover_peer(addr1):
  21         raise Exception("Device discovery timed out")
  22
  23     logger.info("Test provision discovery for display")
  24     dev[0].global_request("P2P_PROV_DISC " + addr1 + " display")
  25     ev1 = dev[1].wait_global_event(["P2P-PROV-DISC-SHOW-PIN"], timeout=15)
  26     if ev1 is None:
  27         raise Exception("Provision discovery timed out (display/dev1)")
  28     if addr0 not in ev1:
  29         raise Exception("Dev0 not in provision discovery event")
  30     ev0 = dev[0].wait_global_event(["P2P-PROV-DISC-ENTER-PIN",
  31                                     "P2P-PROV-DISC-FAILURE"], timeout=15)
  32     if ev0 is None:
  33         raise Exception("Provision discovery timed out (display/dev0)")
  34     if "P2P-PROV-DISC-FAILURE" in ev0:
  35         raise Exception("Provision discovery failed (display/dev0)")
  36     if addr1 not in ev0:
  37         raise Exception("Dev1 not in provision discovery event")
  38
  39     logger.info("Test provision discovery for keypad")
  40     dev[0].global_request("P2P_PROV_DISC " + addr1 + " keypad")
  41     ev1 = dev[1].wait_global_event(["P2P-PROV-DISC-ENTER-PIN"], timeout=15)
  42     if ev1 is None:
  43         raise Exception("Provision discovery timed out (keypad/dev1)")
  44     if addr0 not in ev1:
  45         raise Exception("Dev0 not in provision discovery event")
  46     ev0 = dev[0].wait_global_event(["P2P-PROV-DISC-SHOW-PIN",
  47                                     "P2P-PROV-DISC-FAILURE"],
  48                                    timeout=15)
  49     if ev0 is None:
  50         raise Exception("Provision discovery timed out (keypad/dev0)")
  51     if "P2P-PROV-DISC-FAILURE" in ev0:
  52         raise Exception("Provision discovery failed (keypad/dev0)")
  53     if addr1 not in ev0:
  54         raise Exception("Dev1 not in provision discovery event")
  55
  56     logger.info("Test provision discovery for push button")
  57     dev[0].global_request("P2P_PROV_DISC " + addr1 + " pbc")
  58     ev1 = dev[1].wait_global_event(["P2P-PROV-DISC-PBC-REQ"], timeout=15)
  59     if ev1 is None:
  60         raise Exception("Provision discovery timed out (pbc/dev1)")
  61     if addr0 not in ev1:
  62         raise Exception("Dev0 not in provision discovery event")
  63     ev0 = dev[0].wait_global_event(["P2P-PROV-DISC-PBC-RESP",
  64                                     "P2P-PROV-DISC-FAILURE"],
  65                                    timeout=15)
  66     if ev0 is None:
  67         raise Exception("Provision discovery timed out (pbc/dev0)")
  68     if "P2P-PROV-DISC-FAILURE" in ev0:
  69         raise Exception("Provision discovery failed (pbc/dev0)")
  70     if addr1 not in ev0:
  71         raise Exception("Dev1 not in provision discovery event")
  72
  73     dev[0].p2p_stop_find
  74     dev[1].p2p_stop_find
  75
  76 def test_discovery_pd_retries(dev):
  77     """P2P device discovery and provision discovery retries"""
  78     addr0 = dev[0].p2p_dev_addr()
  79     addr1 = dev[1].p2p_dev_addr()
  80     dev[1].p2p_listen()
  81     if not dev[0].discover_peer(addr1):
  82         raise Exception("Device discovery timed out")
  83     dev[1].p2p_stop_find()
  84     dev[0].p2p_stop_find()
  85     dev[0].global_request("P2P_PROV_DISC " + addr1 + " display")
  86     ev = dev[0].wait_event(["P2P-PROV-DISC-FAILURE"], timeout=60)
  87     if ev is None:
  88         raise Exception("No PD failure reported")
  89
  90 def test_discovery_group_client(dev):
  91     """P2P device discovery for a client in a group"""
  92     logger.info("Start autonomous GO " + dev[0].ifname)
  93     res = dev[0].p2p_start_go(freq="2422")
  94     logger.debug("res: " + str(res))
  95     logger.info("Connect a client to the GO")
  96     pin = dev[1].wps_read_pin()
  97     dev[0].p2p_go_authorize_client(pin)
  98     dev[1].p2p_connect_group(dev[0].p2p_dev_addr(), pin, timeout=60)
  99     logger.info("Client connected")
 100     hwsim_utils.test_connectivity_p2p(dev[0], dev[1])
 101     logger.info("Try to discover a P2P client in a group")
 102     if not dev[2].discover_peer(dev[1].p2p_dev_addr(), social=False):
 103         raise Exception("Could not discover group client")
 104
 105     # This is not really perfect, but something to get a bit more testing
 106     # coverage.. For proper discoverability mechanism validation, the P2P
 107     # client would need to go to sleep to avoid acknowledging the GO Negotiation
 108     # Request frame. Offchannel Listen mode operation on the P2P Client with
 109     # mac80211_hwsim is apparently not enough to avoid the acknowledgement on
 110     # the operating channel, so need to disconnect from the group which removes
 111     # the GO-to-P2P Client part of the discoverability exchange in practice.
 112
 113     pin = dev[2].wps_read_pin()
 114     # make group client non-responsive on operating channel
 115     dev[1].dump_monitor()
 116     dev[1].group_request("DISCONNECT")
 117     ev = dev[1].wait_event(["CTRL-EVENT-DISCONNECTED"])
 118     if ev is None:
 119         raise Exception("Timeout on waiting disconnection")
 120     dev[2].request("P2P_CONNECT {} {} display".format(dev[1].p2p_dev_addr(),
 121                                                       pin))
 122     ev = dev[1].wait_event(["P2P-GO-NEG-REQUEST"], timeout=2)
 123     if ev:
 124         raise Exception("Unexpected frame RX on P2P client")
 125     # make group client available on operating channe
 126     dev[1].request("REASSOCIATE")
 127     ev = dev[1].wait_event(["CTRL-EVENT-CONNECTED", "P2P-GO-NEG-REQUEST"])
 128     if ev is None:
 129         raise Exception("Timeout on reconnection to group")
 130     if "P2P-GO-NEG-REQUEST" not in ev:
 131         ev = dev[1].wait_event(["P2P-GO-NEG-REQUEST"])
 132         if ev is None:
 133             raise Exception("Timeout on waiting for GO Negotiation Request")
 134
 135 def test_discovery_dev_type(dev):
 136     """P2P device discovery with Device Type filter"""
 137     dev[1].request("SET sec_device_type 1-0050F204-2")
 138     dev[1].p2p_listen()
 139     dev[0].p2p_find(social=True, dev_type="5-0050F204-1")
 140     ev = dev[0].wait_event(['P2P-DEVICE-FOUND'], timeout=1)
 141     if ev:
 142         raise Exception("Unexpected P2P device found")
 143     dev[0].p2p_find(social=True, dev_type="1-0050F204-2")
 144     ev = dev[0].wait_event(['P2P-DEVICE-FOUND'], timeout=2)
 145     if ev is None:
 146         raise Exception("P2P device not found")
 147
 148 def test_discovery_dev_type_go(dev):
 149     """P2P device discovery with Device Type filter on GO"""
 150     addr1 = dev[1].p2p_dev_addr()
 151     dev[1].request("SET sec_device_type 1-0050F204-2")
 152     res = dev[0].p2p_start_go(freq="2412")
 153     pin = dev[1].wps_read_pin()
 154     dev[0].p2p_go_authorize_client(pin)
 155     dev[1].p2p_connect_group(dev[0].p2p_dev_addr(), pin, timeout=60)
 156
 157     dev[2].p2p_find(social=True, dev_type="5-0050F204-1")
 158     ev = dev[2].wait_event(['P2P-DEVICE-FOUND'], timeout=1)
 159     if ev:
 160         raise Exception("Unexpected P2P device found")
 161     dev[2].p2p_find(social=True, dev_type="1-0050F204-2")
 162     ev = dev[2].wait_event(['P2P-DEVICE-FOUND ' + addr1], timeout=2)
 163     if ev is None:
 164         raise Exception("P2P device not found")
 165
 166 def test_discovery_dev_id(dev):
 167     """P2P device discovery with Device ID filter"""
 168     addr1 = dev[1].p2p_dev_addr()
 169     dev[1].p2p_listen()
 170     dev[0].p2p_find(social=True, dev_id="02:03:04:05:06:07")
 171     ev = dev[0].wait_event(['P2P-DEVICE-FOUND'], timeout=1)
 172     if ev:
 173         raise Exception("Unexpected P2P device found")
 174     dev[0].p2p_find(social=True, dev_id=addr1)
 175     ev = dev[0].wait_event(['P2P-DEVICE-FOUND'], timeout=2)
 176     if ev is None:
 177         raise Exception("P2P device not found")
 178
 179 def test_discovery_dev_id_go(dev):
 180     """P2P device discovery with Device ID filter on GO"""
 181     addr1 = dev[1].p2p_dev_addr()
 182     res = dev[0].p2p_start_go(freq="2412")
 183     pin = dev[1].wps_read_pin()
 184     dev[0].p2p_go_authorize_client(pin)
 185     dev[1].p2p_connect_group(dev[0].p2p_dev_addr(), pin, timeout=60)
 186
 187     dev[2].p2p_find(social=True, dev_id="02:03:04:05:06:07")
 188     ev = dev[2].wait_event(['P2P-DEVICE-FOUND'], timeout=1)
 189     if ev:
 190         raise Exception("Unexpected P2P device found")
 191     dev[2].p2p_find(social=True, dev_id=addr1)
 192     ev = dev[2].wait_event(['P2P-DEVICE-FOUND ' + addr1], timeout=2)
 193     if ev is None:
 194         raise Exception("P2P device not found")