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 import time
  10
  11 import hwsim_utils
  12 from wpasupplicant import WpaSupplicant
  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_pd_retries(dev):
  79     """P2P device discovery and provision discovery retries"""
  80     addr0 = dev[0].p2p_dev_addr()
  81     addr1 = dev[1].p2p_dev_addr()
  82     dev[1].p2p_listen()
  83     if not dev[0].discover_peer(addr1):
  84         raise Exception("Device discovery timed out")
  85     dev[1].p2p_stop_find()
  86     dev[0].p2p_stop_find()
  87     dev[0].global_request("P2P_PROV_DISC " + addr1 + " display")
  88     ev = dev[0].wait_event(["P2P-PROV-DISC-FAILURE"], timeout=60)
  89     if ev is None:
  90         raise Exception("No PD failure reported")
  91
  92 def test_discovery_group_client(dev):
  93     """P2P device discovery for a client in a group"""
  94     logger.info("Start autonomous GO " + dev[0].ifname)
  95     res = dev[0].p2p_start_go(freq="2422")
  96     logger.debug("res: " + str(res))
  97     logger.info("Connect a client to the GO")
  98     pin = dev[1].wps_read_pin()
  99     dev[0].p2p_go_authorize_client(pin)
 100     dev[1].p2p_connect_group(dev[0].p2p_dev_addr(), pin, timeout=60)
 101     logger.info("Client connected")
 102     hwsim_utils.test_connectivity_p2p(dev[0], dev[1])
 103     logger.info("Try to discover a P2P client in a group")
 104     if not dev[2].discover_peer(dev[1].p2p_dev_addr(), social=False, timeout=10):
 105         if not dev[2].discover_peer(dev[1].p2p_dev_addr(), social=False, timeout=10):
 106             if not dev[2].discover_peer(dev[1].p2p_dev_addr(), social=False, timeout=10):
 107                 raise Exception("Could not discover group client")
 108
 109     # This is not really perfect, but something to get a bit more testing
 110     # coverage.. For proper discoverability mechanism validation, the P2P
 111     # client would need to go to sleep to avoid acknowledging the GO Negotiation
 112     # Request frame. Offchannel Listen mode operation on the P2P Client with
 113     # mac80211_hwsim is apparently not enough to avoid the acknowledgement on
 114     # the operating channel, so need to disconnect from the group which removes
 115     # the GO-to-P2P Client part of the discoverability exchange in practice.
 116
 117     pin = dev[2].wps_read_pin()
 118     # make group client non-responsive on operating channel
 119     dev[1].dump_monitor()
 120     dev[1].group_request("DISCONNECT")
 121     ev = dev[1].wait_event(["CTRL-EVENT-DISCONNECTED"])
 122     if ev is None:
 123         raise Exception("Timeout on waiting disconnection")
 124     dev[2].request("P2P_CONNECT {} {} display".format(dev[1].p2p_dev_addr(),
 125                                                       pin))
 126     ev = dev[1].wait_event(["P2P-GO-NEG-REQUEST"], timeout=2)
 127     if ev:
 128         raise Exception("Unexpected frame RX on P2P client")
 129     # make group client available on operating channe
 130     dev[1].request("REASSOCIATE")
 131     ev = dev[1].wait_event(["CTRL-EVENT-CONNECTED", "P2P-GO-NEG-REQUEST"])
 132     if ev is None:
 133         raise Exception("Timeout on reconnection to group")
 134     if "P2P-GO-NEG-REQUEST" not in ev:
 135         ev = dev[1].wait_event(["P2P-GO-NEG-REQUEST"])
 136         if ev is None:
 137             raise Exception("Timeout on waiting for GO Negotiation Request")
 138
 139 def test_discovery_dev_type(dev):
 140     """P2P device discovery with Device Type filter"""
 141     dev[1].request("SET sec_device_type 1-0050F204-2")
 142     dev[1].p2p_listen()
 143     dev[0].p2p_find(social=True, dev_type="5-0050F204-1")
 144     ev = dev[0].wait_event(['P2P-DEVICE-FOUND'], timeout=1)
 145     if ev:
 146         raise Exception("Unexpected P2P device found")
 147     dev[0].p2p_find(social=True, dev_type="1-0050F204-2")
 148     ev = dev[0].wait_event(['P2P-DEVICE-FOUND'], timeout=2)
 149     if ev is None:
 150         raise Exception("P2P device not found")
 151     peer = dev[0].get_peer(dev[1].p2p_dev_addr())
 152     if "1-0050F204-2" not in peer['sec_dev_type']:
 153         raise Exception("sec_device_type not reported properly")
 154
 155 def test_discovery_dev_type_go(dev):
 156     """P2P device discovery with Device Type filter on GO"""
 157     addr1 = dev[1].p2p_dev_addr()
 158     dev[1].request("SET sec_device_type 1-0050F204-2")
 159     res = dev[0].p2p_start_go(freq="2412")
 160     pin = dev[1].wps_read_pin()
 161     dev[0].p2p_go_authorize_client(pin)
 162     dev[1].p2p_connect_group(dev[0].p2p_dev_addr(), pin, timeout=60)
 163
 164     dev[2].p2p_find(social=True, dev_type="5-0050F204-1")
 165     ev = dev[2].wait_event(['P2P-DEVICE-FOUND'], timeout=1)
 166     if ev:
 167         raise Exception("Unexpected P2P device found")
 168     dev[2].p2p_find(social=True, dev_type="1-0050F204-2")
 169     ev = dev[2].wait_event(['P2P-DEVICE-FOUND ' + addr1], timeout=2)
 170     if ev is None:
 171         raise Exception("P2P device not found")
 172
 173 def test_discovery_dev_id(dev):
 174     """P2P device discovery with Device ID filter"""
 175     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
 176     wpas.interface_add("wlan5")
 177     wpas.request("P2P_LISTEN 1")
 178     status = wpas.global_request("STATUS")
 179     if "p2p_state=LISTEN_ONLY" not in status:
 180         raise Exception("Unexpected status: " + status)
 181     addr1 = dev[1].p2p_dev_addr()
 182     dev[1].p2p_listen()
 183     dev[0].p2p_find(social=True, dev_id="02:03:04:05:06:07")
 184     ev = dev[0].wait_event(['P2P-DEVICE-FOUND'], timeout=1)
 185     if ev:
 186         raise Exception("Unexpected P2P device found")
 187     dev[0].p2p_find(social=True, dev_id=addr1)
 188     ev = dev[0].wait_event(['P2P-DEVICE-FOUND'], timeout=2)
 189     if ev is None:
 190         raise Exception("P2P device not found")
 191     if addr1 not in ev:
 192         raise Exception("Unexpected P2P peer found")
 193     status = wpas.global_request("STATUS")
 194     for i in range(0, 2):
 195         if "p2p_state=IDLE" in status:
 196             break
 197         time.sleep(0.5)
 198         status = wpas.global_request("STATUS")
 199     if "p2p_state=IDLE" not in status:
 200         raise Exception("Unexpected status: " + status)
 201
 202 def test_discovery_dev_id_go(dev):
 203     """P2P device discovery with Device ID filter on GO"""
 204     addr1 = dev[1].p2p_dev_addr()
 205     res = dev[0].p2p_start_go(freq="2412")
 206     pin = dev[1].wps_read_pin()
 207     dev[0].p2p_go_authorize_client(pin)
 208     dev[1].p2p_connect_group(dev[0].p2p_dev_addr(), pin, timeout=60)
 209
 210     dev[2].p2p_find(social=True, dev_id="02:03:04:05:06:07")
 211     ev = dev[2].wait_event(['P2P-DEVICE-FOUND'], timeout=1)
 212     if ev:
 213         raise Exception("Unexpected P2P device found")
 214     dev[2].p2p_find(social=True, dev_id=addr1)
 215     ev = dev[2].wait_event(['P2P-DEVICE-FOUND ' + addr1], timeout=2)
 216     if ev is None:
 217         raise Exception("P2P device not found")
 218
 219 def test_discovery_social_plus_one(dev):
 220     """P2P device discovery with social-plus-one"""
 221     logger.info("Start autonomous GO " + dev[0].ifname)
 222     dev[1].p2p_find(social=True)
 223     dev[0].p2p_find(progressive=True)
 224     logger.info("Wait for initial progressive find phases")
 225     dev[0].wait_event(["CTRL-EVENT-SCAN-STARTED"])
 226     dev[0].wait_event(["CTRL-EVENT-SCAN-STARTED"])
 227     go = dev[2].p2p_dev_addr()
 228     dev[2].p2p_start_go(freq="2422")
 229     logger.info("Verify whether the GO on non-social channel can be found")
 230     ev = dev[0].wait_global_event(["P2P-DEVICE-FOUND"], timeout=15)
 231     if ev is None:
 232         raise Exception("Peer not found")
 233     if go not in ev:
 234         ev = dev[0].wait_global_event(["P2P-DEVICE-FOUND"], timeout=15)
 235         if ev is None:
 236             raise Exception("Peer not found")
 237     ev = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=15)
 238     if ev is None:
 239         raise Exception("Peer not found")
 240     dev[0].p2p_stop_find()
 241     dev[1].p2p_stop_find()
 242     if not dev[0].peer_known(go):
 243         raise Exception("GO not found in progressive scan")
 244     if dev[1].peer_known(go):
 245         raise Exception("GO found in social-only scan")
 246
 247 def test_discovery_and_interface_disabled(dev):
 248     """P2P device discovery with interface getting didabled"""
 249     try:
 250         if "OK" not in dev[0].p2p_find():
 251             raise Exception("Failed to start P2P find")
 252         ev = dev[0].wait_event(["CTRL-EVENT-SCAN-STARTED"])
 253         if ev is None:
 254             raise Exception("Scan did not start")
 255         dev[0].request("DRIVER_EVENT INTERFACE_DISABLED")
 256         time.sleep(1)
 257
 258         # verify that P2P_FIND is rejected
 259         if "FAIL" not in dev[0].p2p_find():
 260             raise Exception("New P2P_FIND request was accepted unexpectedly")
 261
 262         dev[0].request("DRIVER_EVENT INTERFACE_ENABLED")
 263         time.sleep(3)
 264         dev[0].scan(freq="2412")
 265         if "OK" not in dev[0].p2p_find():
 266             raise Exception("Failed to start P2P find")
 267         dev[0].dump_monitor()
 268         dev[1].p2p_listen()
 269         ev = dev[0].wait_global_event(["P2P-DEVICE-FOUND"], timeout=15)
 270         if ev is None:
 271             raise Exception("Peer not found")
 272     finally:
 273         dev[0].request("DRIVER_EVENT INTERFACE_ENABLED")
 274
 275 def test_discovery_auto(dev):
 276     """P2P device discovery and provision discovery with auto GO/dev selection"""
 277     addr0 = dev[0].p2p_dev_addr()
 278     addr1 = dev[1].p2p_dev_addr()
 279     addr2 = dev[2].p2p_dev_addr()
 280     dev[2].p2p_start_go(freq="2412")
 281     logger.info("Start device discovery")
 282     dev[0].p2p_listen()
 283     if not dev[1].discover_peer(addr0):
 284         raise Exception("Device discovery timed out")
 285     dev[1].p2p_listen()
 286     if not dev[0].discover_peer(addr1):
 287         raise Exception("Device discovery timed out")
 288     if not dev[0].discover_peer(addr2):
 289         raise Exception("Device discovery timed out")
 290
 291     logger.info("Test provision discovery for display (device)")
 292     dev[0].global_request("P2P_PROV_DISC " + addr1 + " display auto")
 293     ev1 = dev[1].wait_global_event(["P2P-PROV-DISC-SHOW-PIN"], timeout=15)
 294     if ev1 is None:
 295         raise Exception("Provision discovery timed out (display/dev1)")
 296     if addr0 not in ev1:
 297         raise Exception("Dev0 not in provision discovery event")
 298     if " group=" in ev1:
 299         raise Exception("Unexpected group parameter from non-GO")
 300     ev0 = dev[0].wait_global_event(["P2P-PROV-DISC-ENTER-PIN",
 301                                     "P2P-PROV-DISC-FAILURE"], timeout=15)
 302     if ev0 is None:
 303         raise Exception("Provision discovery timed out (display/dev0)")
 304     if "P2P-PROV-DISC-FAILURE" in ev0:
 305         raise Exception("Provision discovery failed (display/dev0)")
 306     if addr1 not in ev0:
 307         raise Exception("Dev1 not in provision discovery event")
 308     if "peer_go=0" not in ev0:
 309         raise Exception("peer_go incorrect in PD response from non-GO")
 310
 311     logger.info("Test provision discovery for display (GO)")
 312     dev[0].global_request("P2P_PROV_DISC " + addr2 + " display auto")
 313     ev2 = dev[2].wait_global_event(["P2P-PROV-DISC-SHOW-PIN"], timeout=15)
 314     if ev2 is None:
 315         raise Exception("Provision discovery timed out (display/dev2)")
 316     if addr0 not in ev2:
 317         raise Exception("Dev0 not in provision discovery event")
 318     if " group=" not in ev2:
 319         raise Exception("Group parameter missing from GO")
 320     ev0 = dev[0].wait_global_event(["P2P-PROV-DISC-ENTER-PIN",
 321                                     "P2P-PROV-DISC-FAILURE"], timeout=15)
 322     if ev0 is None:
 323         raise Exception("Provision discovery timed out (display/dev0)")
 324     if "P2P-PROV-DISC-FAILURE" in ev0:
 325         raise Exception("Provision discovery failed (display/dev0)")
 326     if addr2 not in ev0:
 327         raise Exception("Dev1 not in provision discovery event")
 328     if "peer_go=1" not in ev0:
 329         raise Exception("peer_go incorrect in PD response from GO")