tests/hwsim/test_p2p_persistent.py

   1 #!/usr/bin/python
   2 #
   3 # P2P persistent group 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()
  11
  12 import hwsim_utils
  13 from test_p2p_autogo import connect_cli
  14
  15 def go_neg_pin_authorized_persistent(i_dev, r_dev, i_intent=None, r_intent=None, i_method='enter', r_method='display', test_data=True):
  16     r_dev.p2p_listen()
  17     i_dev.p2p_listen()
  18     pin = r_dev.wps_read_pin()
  19     logger.info("Start GO negotiation " + i_dev.ifname + " -> " + r_dev.ifname)
  20     r_dev.p2p_go_neg_auth(i_dev.p2p_dev_addr(), pin, r_method,
  21                           go_intent=r_intent, persistent=True)
  22     i_res = i_dev.p2p_go_neg_init(r_dev.p2p_dev_addr(), pin, i_method,
  23                                   timeout=20, go_intent=i_intent,
  24                                   persistent=True)
  25     r_res = r_dev.p2p_go_neg_auth_result()
  26     logger.debug("i_res: " + str(i_res))
  27     logger.debug("r_res: " + str(r_res))
  28     r_dev.dump_monitor()
  29     i_dev.dump_monitor()
  30     logger.info("Group formed")
  31     if test_data:
  32         hwsim_utils.test_connectivity_p2p(r_dev, i_dev)
  33     return [i_res, r_res]
  34
  35 def terminate_group(go, cli):
  36     logger.info("Terminate persistent group")
  37     go.remove_group()
  38     cli.wait_go_ending_session()
  39
  40 def invite(inv, resp, extra=None):
  41     addr = resp.p2p_dev_addr()
  42     resp.request("SET persistent_reconnect 1")
  43     resp.p2p_listen()
  44     if not inv.discover_peer(addr, social=True):
  45         raise Exception("Peer " + addr + " not found")
  46     inv.dump_monitor()
  47     peer = inv.get_peer(addr)
  48     cmd = "P2P_INVITE persistent=" + peer['persistent'] + " peer=" + addr
  49     if extra:
  50         cmd = cmd + " " + extra;
  51     inv.global_request(cmd)
  52
  53 def check_result(go, cli):
  54     ev = go.wait_global_event(["P2P-GROUP-STARTED"], timeout=30)
  55     if ev is None:
  56         raise Exception("Timeout on group re-invocation (on GO)")
  57     if "[PERSISTENT]" not in ev:
  58         raise Exception("Re-invoked group not marked persistent")
  59     go_res = go.group_form_result(ev)
  60     if go_res['role'] != 'GO':
  61         raise Exception("Persistent group GO did not become GO")
  62     if not go_res['persistent']:
  63         raise Exception("Persistent group not re-invoked as persistent (GO)")
  64     ev = cli.wait_global_event(["P2P-GROUP-STARTED"], timeout=30)
  65     if ev is None:
  66         raise Exception("Timeout on group re-invocation (on client)")
  67     if "[PERSISTENT]" not in ev:
  68         raise Exception("Re-invoked group not marked persistent")
  69     cli_res = cli.group_form_result(ev)
  70     if cli_res['role'] != 'client':
  71         raise Exception("Persistent group client did not become client")
  72     if not cli_res['persistent']:
  73         raise Exception("Persistent group not re-invoked as persistent (cli)")
  74     return [go_res, cli_res]
  75
  76 def form(go, cli, test_data=True):
  77     logger.info("Form a persistent group")
  78     [i_res, r_res] = go_neg_pin_authorized_persistent(i_dev=go, i_intent=15,
  79                                                       r_dev=cli, r_intent=0,
  80                                                       test_data=test_data)
  81     if not i_res['persistent'] or not r_res['persistent']:
  82         raise Exception("Formed group was not persistent")
  83     terminate_group(go, cli)
  84
  85 def invite_from_cli(go, cli):
  86     logger.info("Re-invoke persistent group from client")
  87     invite(cli, go)
  88     check_result(go, cli)
  89     hwsim_utils.test_connectivity_p2p(go, cli)
  90     terminate_group(go, cli)
  91
  92 def invite_from_go(go, cli):
  93     logger.info("Re-invoke persistent group from GO")
  94     invite(go, cli)
  95     check_result(go, cli)
  96     hwsim_utils.test_connectivity_p2p(go, cli)
  97     terminate_group(go, cli)
  98
  99 def test_persistent_group(dev):
 100     """P2P persistent group formation and re-invocation"""
 101     form(dev[0], dev[1])
 102     invite_from_cli(dev[0], dev[1])
 103     invite_from_go(dev[0], dev[1])
 104
 105 def test_persistent_group_per_sta_psk(dev):
 106     """P2P persistent group formation and re-invocation using per-client PSK"""
 107     addr0 = dev[0].p2p_dev_addr()
 108     addr1 = dev[1].p2p_dev_addr()
 109     addr2 = dev[2].p2p_dev_addr()
 110     dev[0].request("P2P_SET per_sta_psk 1")
 111     logger.info("Form a persistent group")
 112     [i_res, r_res] = go_neg_pin_authorized_persistent(i_dev=dev[0], i_intent=15,
 113                                                       r_dev=dev[1], r_intent=0)
 114     if not i_res['persistent'] or not r_res['persistent']:
 115         raise Exception("Formed group was not persistent")
 116
 117     logger.info("Join another client to the group")
 118     pin = dev[2].wps_read_pin()
 119     dev[0].p2p_go_authorize_client(pin)
 120     c_res = dev[2].p2p_connect_group(addr0, pin, timeout=60)
 121     if not c_res['persistent']:
 122         raise Exception("Joining client did not recognize persistent group")
 123     if r_res['psk'] == c_res['psk']:
 124         raise Exception("Same PSK assigned for both clients")
 125     hwsim_utils.test_connectivity_p2p_sta(dev[1], dev[2])
 126
 127     logger.info("Leave persistent group and rejoin it")
 128     dev[2].remove_group()
 129     ev = dev[2].wait_event(["P2P-GROUP-REMOVED"], timeout=3)
 130     if ev is None:
 131         raise Exception("Group removal event timed out")
 132     if not dev[2].discover_peer(addr0, social=True):
 133         raise Exception("Peer " + peer + " not found")
 134     dev[2].dump_monitor()
 135     peer = dev[2].get_peer(addr0)
 136     dev[2].global_request("P2P_GROUP_ADD persistent=" + peer['persistent'])
 137     ev = dev[2].wait_global_event(["P2P-GROUP-STARTED"], timeout=30)
 138     if ev is None:
 139         raise Exception("Timeout on group restart (on client)")
 140     cli_res = dev[2].group_form_result(ev)
 141     if not cli_res['persistent']:
 142         raise Exception("Persistent group not restarted as persistent (cli)")
 143     hwsim_utils.test_connectivity_p2p(dev[1], dev[2])
 144
 145     logger.info("Remove one of the clients from the group")
 146     dev[0].global_request("P2P_REMOVE_CLIENT " + addr2)
 147     dev[2].wait_go_ending_session()
 148
 149     logger.info("Try to reconnect after having been removed from group")
 150     if not dev[2].discover_peer(addr0, social=True):
 151         raise Exception("Peer " + peer + " not found")
 152     dev[2].dump_monitor()
 153     peer = dev[2].get_peer(addr0)
 154     dev[2].global_request("P2P_GROUP_ADD persistent=" + peer['persistent'])
 155     ev = dev[2].wait_global_event(["P2P-GROUP-STARTED","WPA: 4-Way Handshake failed"], timeout=30)
 156     if ev is None:
 157         raise Exception("Timeout on group restart (on client)")
 158     if "P2P-GROUP-STARTED" in ev:
 159         raise Exception("Client managed to connect after being removed")
 160
 161     logger.info("Remove the remaining client from the group")
 162     dev[0].global_request("P2P_REMOVE_CLIENT " + addr1)
 163     dev[1].wait_go_ending_session()
 164
 165     logger.info("Terminate persistent group")
 166     dev[0].remove_group()
 167     dev[0].dump_monitor()
 168
 169     logger.info("Try to re-invoke persistent group from client")
 170     dev[0].request("SET persistent_reconnect 1")
 171     dev[0].p2p_listen()
 172     if not dev[1].discover_peer(addr0, social=True):
 173         raise Exception("Peer " + peer + " not found")
 174     dev[1].dump_monitor()
 175     peer = dev[1].get_peer(addr0)
 176     dev[1].global_request("P2P_INVITE persistent=" + peer['persistent'] + " peer=" + addr0)
 177     ev = dev[1].wait_global_event(["P2P-GROUP-STARTED","WPA: 4-Way Handshake failed"], timeout=30)
 178     if ev is None:
 179         raise Exception("Timeout on group restart (on client)")
 180     if "P2P-GROUP-STARTED" in ev:
 181         raise Exception("Client managed to re-invoke after being removed")
 182     dev[0].dump_monitor()
 183
 184     logger.info("Terminate persistent group")
 185     dev[0].remove_group()
 186     dev[0].dump_monitor()
 187
 188 def test_persistent_group_invite_removed_client(dev):
 189     """P2P persistent group client removal and re-invitation"""
 190     addr0 = dev[0].p2p_dev_addr()
 191     addr1 = dev[1].p2p_dev_addr()
 192     dev[0].request("P2P_SET per_sta_psk 1")
 193     logger.info("Form a persistent group")
 194     [i_res, r_res] = go_neg_pin_authorized_persistent(i_dev=dev[0], i_intent=15,
 195                                                       r_dev=dev[1], r_intent=0)
 196     if not i_res['persistent'] or not r_res['persistent']:
 197         raise Exception("Formed group was not persistent")
 198
 199     logger.info("Remove client from the group")
 200     dev[0].global_request("P2P_REMOVE_CLIENT " + addr1)
 201     dev[1].wait_go_ending_session()
 202
 203     logger.info("Re-invite the removed client to join the group")
 204     dev[1].p2p_listen()
 205     if not dev[0].discover_peer(addr1, social=True):
 206         raise Exception("Peer " + peer + " not found")
 207     dev[0].global_request("P2P_INVITE group=" + dev[0].group_ifname + " peer=" + addr1)
 208     ev = dev[1].wait_global_event(["P2P-INVITATION-RECEIVED"], timeout=10)
 209     if ev is None:
 210         raise Exception("Timeout on invitation")
 211     if "sa=" + addr0 + " persistent=" not in ev:
 212         raise Exception("Unexpected invitation event")
 213     [event,addr,persistent] = ev.split(' ', 2)
 214     dev[1].global_request("P2P_GROUP_ADD " + persistent)
 215     ev = dev[1].wait_global_event(["P2P-PERSISTENT-PSK-FAIL"], timeout=30)
 216     if ev is None:
 217         raise Exception("Did not receive PSK failure report")
 218     [tmp,id] = ev.split('=', 1)
 219     ev = dev[1].wait_global_event(["P2P-GROUP-REMOVED"], timeout=10)
 220     if ev is None:
 221         raise Exception("Group removal event timed out")
 222     if "reason=PSK_FAILURE" not in ev:
 223         raise Exception("Unexpected group removal reason")
 224     dev[1].request("REMOVE_NETWORK " + id)
 225
 226     logger.info("Re-invite after client removed persistent group info")
 227     dev[1].p2p_listen()
 228     if not dev[0].discover_peer(addr1, social=True):
 229         raise Exception("Peer " + peer + " not found")
 230     dev[0].global_request("P2P_INVITE group=" + dev[0].group_ifname + " peer=" + addr1)
 231     ev = dev[1].wait_global_event(["P2P-INVITATION-RECEIVED"], timeout=10)
 232     if ev is None:
 233         raise Exception("Timeout on invitation")
 234     if " persistent=" in ev:
 235         raise Exception("Unexpected invitation event")
 236     pin = dev[1].wps_read_pin()
 237     dev[0].p2p_go_authorize_client(pin)
 238     c_res = dev[1].p2p_connect_group(addr0, pin, timeout=60)
 239     if not c_res['persistent']:
 240         raise Exception("Joining client did not recognize persistent group")
 241     if r_res['psk'] == c_res['psk']:
 242         raise Exception("Same PSK assigned on both times")
 243     hwsim_utils.test_connectivity_p2p(dev[0], dev[1])
 244
 245     terminate_group(dev[0], dev[1])
 246
 247 def test_persistent_group_channel(dev):
 248     """P2P persistent group re-invocation with channel selection"""
 249     form(dev[0], dev[1], test_data=False)
 250
 251     logger.info("Re-invoke persistent group from client with forced channel")
 252     invite(dev[1], dev[0], "freq=2427")
 253     [go_res, cli_res] = check_result(dev[0], dev[1])
 254     if go_res['freq'] != "2427":
 255         raise Exception("Persistent group client forced channel not followed")
 256     terminate_group(dev[0], dev[1])
 257
 258     logger.info("Re-invoke persistent group from GO with forced channel")
 259     invite(dev[0], dev[1], "freq=2432")
 260     [go_res, cli_res] = check_result(dev[0], dev[1])
 261     if go_res['freq'] != "2432":
 262         raise Exception("Persistent group GO channel preference not followed")
 263     terminate_group(dev[0], dev[1])
 264
 265     logger.info("Re-invoke persistent group from client with channel preference")
 266     invite(dev[1], dev[0], "pref=2417")
 267     [go_res, cli_res] = check_result(dev[0], dev[1])
 268     if go_res['freq'] != "2417":
 269         raise Exception("Persistent group client channel preference not followed")
 270     terminate_group(dev[0], dev[1])
 271
 272 def test_persistent_group_and_role_change(dev):
 273     """P2P persistent group, auto GO in another role, and re-invocation"""
 274     form(dev[0], dev[1])
 275
 276     logger.info("Start and stop autonomous GO on previous P2P client device")
 277     dev[1].p2p_start_go()
 278     dev[1].remove_group()
 279     dev[1].dump_monitor()
 280
 281     logger.info("Re-invoke the persistent group")
 282     invite_from_go(dev[0], dev[1])
 283
 284 def test_persistent_go_client_list(dev):
 285     """P2P GO and list of clients in persistent group"""
 286     addr0 = dev[0].p2p_dev_addr()
 287     addr1 = dev[1].p2p_dev_addr()
 288     addr2 = dev[2].p2p_dev_addr()
 289
 290     res = dev[0].p2p_start_go(persistent=True)
 291     id = None
 292     for n in dev[0].list_networks():
 293         if "[P2P-PERSISTENT]" in n['flags']:
 294             id = n['id']
 295             break
 296     if id is None:
 297         raise Exception("Could not find persistent group entry")
 298
 299     connect_cli(dev[0], dev[1])
 300     clients = dev[0].request("GET_NETWORK " + id + " p2p_client_list").rstrip()
 301     if clients != addr1:
 302         raise Exception("Unexpected p2p_client_list entry(2): " + clients)
 303     connect_cli(dev[0], dev[2])
 304     clients = dev[0].request("GET_NETWORK " + id + " p2p_client_list").rstrip()
 305     if clients != addr2 + " " + addr1:
 306         raise Exception("Unexpected p2p_client_list entry(3): " + clients)
 307
 308     peer = dev[1].get_peer(res['go_dev_addr'])
 309     dev[1].remove_group()
 310     dev[1].request("P2P_GROUP_ADD persistent=" + peer['persistent'])
 311     ev = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=30)
 312     if ev is None:
 313         raise Exception("Timeout on group restart (on client)")
 314     clients = dev[0].request("GET_NETWORK " + id + " p2p_client_list").rstrip()
 315     if clients != addr1 + " " + addr2:
 316         raise Exception("Unexpected p2p_client_list entry(4): " + clients)
 317
 318     dev[2].remove_group()
 319     dev[1].remove_group()
 320     dev[0].remove_group()
 321
 322     clients = dev[0].request("GET_NETWORK " + id + " p2p_client_list").rstrip()
 323     if clients != addr1 + " " + addr2:
 324         raise Exception("Unexpected p2p_client_list entry(5): " + clients)
 325
 326     dev[1].p2p_listen()
 327     dev[2].p2p_listen()
 328     dev[0].request("P2P_FLUSH")
 329     dev[0].discover_peer(addr1, social=True)
 330     peer = dev[0].get_peer(addr1)
 331     if 'persistent' not in peer or peer['persistent'] != id:
 332         raise Exception("Persistent group client not recognized(1)")
 333
 334     dev[0].discover_peer(addr2, social=True)
 335     peer = dev[0].get_peer(addr2)
 336     if 'persistent' not in peer or peer['persistent'] != id:
 337         raise Exception("Persistent group client not recognized(2)")