| # Wi-Fi Display test cases |
| # Copyright (c) 2013, Jouni Malinen <j@w1.fi> |
| # |
| # This software may be distributed under the terms of the BSD license. |
| # See README for more details. |
| |
| import logging |
| logger = logging.getLogger() |
| import time |
| import threading |
| import Queue |
| |
| import hwsim_utils |
| import utils |
| from test_p2p_autogo import connect_cli |
| from test_p2p_persistent import form, invite, invite_from_cli, invite_from_go |
| |
| def test_wifi_display(dev): |
| """Wi-Fi Display extensions to P2P""" |
| wfd_devinfo = "00411c440028" |
| dev[0].request("SET wifi_display 1") |
| dev[0].request("WFD_SUBELEM_SET 0 0006" + wfd_devinfo) |
| if wfd_devinfo not in dev[0].request("WFD_SUBELEM_GET 0"): |
| raise Exception("Could not fetch back configured subelement") |
| |
| # Associated BSSID |
| dev[0].request("WFD_SUBELEM_SET 1 0006020304050607") |
| # Coupled Sink |
| dev[0].request("WFD_SUBELEM_SET 6 000700000000000000") |
| # Session Info |
| dev[0].request("WFD_SUBELEM_SET 9 0000") |
| # WFD Extended Capability |
| dev[0].request("WFD_SUBELEM_SET 7 00020000") |
| # WFD Content Protection |
| prot = "0001" + "00" |
| dev[0].request("WFD_SUBELEM_SET 5 " + prot) |
| # WFD Video Formats |
| video = "0015" + "010203040506070809101112131415161718192021" |
| dev[0].request("WFD_SUBELEM_SET 3 " + video) |
| # WFD 3D Video Formats |
| video_3d = "0011" + "0102030405060708091011121314151617" |
| dev[0].request("WFD_SUBELEM_SET 4 " + video_3d) |
| # WFD Audio Formats |
| audio = "000f" + "010203040506070809101112131415" |
| dev[0].request("WFD_SUBELEM_SET 2 " + audio) |
| |
| elems = dev[0].request("WFD_SUBELEM_GET all") |
| if wfd_devinfo not in elems: |
| raise Exception("Could not fetch back configured subelements") |
| |
| wfd_devinfo2 = "00001c440028" |
| dev[1].request("SET wifi_display 1") |
| dev[1].request("WFD_SUBELEM_SET 0 0006" + wfd_devinfo2) |
| if wfd_devinfo2 not in dev[1].request("WFD_SUBELEM_GET 0"): |
| raise Exception("Could not fetch back configured subelement") |
| |
| dev[0].p2p_listen() |
| if "FAIL" in dev[1].request("P2P_SERV_DISC_REQ " + dev[0].p2p_dev_addr() + " wifi-display [source][pri-sink] 2,3,4,5"): |
| raise Exception("Setting SD request failed") |
| dev[1].p2p_find(social=True) |
| ev = dev[0].wait_global_event(["P2P-SERV-DISC-REQ"], timeout=10) |
| if ev is None: |
| raise Exception("Device discovery request not reported") |
| ev = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=5) |
| if ev is None: |
| raise Exception("Device discovery timed out") |
| if "wfd_dev_info=0x" + wfd_devinfo not in ev: |
| raise Exception("Wi-Fi Display Info not in P2P-DEVICE-FOUND event") |
| if "new=1" not in ev: |
| raise Exception("new=1 flag missing from P2P-DEVICE-FOUND event") |
| ev = dev[1].wait_global_event(["P2P-SERV-DISC-RESP"], timeout=5) |
| if ev is None: |
| raise Exception("Service discovery timed out") |
| if prot not in ev: |
| raise Exception("WFD Content Protection missing from WSD response") |
| if video not in ev: |
| raise Exception("WFD Video Formats missing from WSD response") |
| if video_3d not in ev: |
| raise Exception("WFD 3D Video Formats missing from WSD response") |
| if audio not in ev: |
| raise Exception("WFD Audio Formats missing from WSD response") |
| |
| dev[1].dump_monitor() |
| dev[0].request("WFD_SUBELEM_SET 0 0006" + wfd_devinfo2) |
| ev = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=15) |
| if ev is None: |
| raise Exception("Peer info update timed out") |
| if "new=0" not in ev: |
| raise Exception("new=0 flag missing from P2P-DEVICE-FOUND event") |
| if "wfd_dev_info=0x" + wfd_devinfo2 not in ev: |
| raise Exception("Wi-Fi Display Info not in P2P-DEVICE-FOUND event") |
| dev[1].dump_monitor() |
| dev[0].request("WFD_SUBELEM_SET 0 0006" + wfd_devinfo) |
| ev = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=15) |
| if ev is None: |
| raise Exception("Peer info update timed out") |
| if "new=0" not in ev: |
| raise Exception("new=0 flag missing from P2P-DEVICE-FOUND event") |
| if "wfd_dev_info=0x" + wfd_devinfo not in ev: |
| raise Exception("Wi-Fi Display Info not in P2P-DEVICE-FOUND event") |
| |
| pin = dev[0].wps_read_pin() |
| dev[0].p2p_go_neg_auth(dev[1].p2p_dev_addr(), pin, 'display') |
| res1 = dev[1].p2p_go_neg_init(dev[0].p2p_dev_addr(), pin, 'enter', timeout=20, go_intent=15) |
| res2 = dev[0].p2p_go_neg_auth_result() |
| |
| bss = dev[0].get_bss("p2p_dev_addr=" + dev[1].p2p_dev_addr()) |
| if bss['bssid'] != dev[1].p2p_interface_addr(): |
| raise Exception("Unexpected BSSID in the BSS entry for the GO") |
| if wfd_devinfo2 not in bss['wfd_subelems']: |
| raise Exception("Could not see wfd_subelems in GO's BSS entry") |
| peer = dev[0].get_peer(dev[1].p2p_dev_addr()) |
| if wfd_devinfo2 not in peer['wfd_subelems']: |
| raise Exception("Could not see wfd_subelems in GO's peer entry") |
| peer = dev[1].get_peer(dev[0].p2p_dev_addr()) |
| if wfd_devinfo not in peer['wfd_subelems']: |
| raise Exception("Could not see wfd_subelems in client's peer entry") |
| |
| wfd_devinfo3 = "00001c440028" |
| dev[2].request("SET wifi_display 1") |
| dev[2].request("WFD_SUBELEM_SET 0 0006" + wfd_devinfo3) |
| dev[2].p2p_find(social=True) |
| ev = dev[2].wait_event(["P2P-DEVICE-FOUND"], timeout=5) |
| if ev is None: |
| raise Exception("Device discovery timed out") |
| if dev[1].p2p_dev_addr() not in ev: |
| ev = dev[2].wait_event(["P2P-DEVICE-FOUND"], timeout=5) |
| if ev is None: |
| raise Exception("Device discovery timed out") |
| if dev[1].p2p_dev_addr() not in ev: |
| raise Exception("Could not discover GO") |
| if "wfd_dev_info=0x" + wfd_devinfo2 not in ev: |
| raise Exception("Wi-Fi Display Info not in P2P-DEVICE-FOUND event") |
| bss = dev[2].get_bss("p2p_dev_addr=" + dev[1].p2p_dev_addr()) |
| if bss['bssid'] != dev[1].p2p_interface_addr(): |
| raise Exception("Unexpected BSSID in the BSS entry for the GO") |
| if wfd_devinfo2 not in bss['wfd_subelems']: |
| raise Exception("Could not see wfd_subelems in GO's BSS entry") |
| peer = dev[2].get_peer(dev[1].p2p_dev_addr()) |
| if wfd_devinfo2 not in peer['wfd_subelems']: |
| raise Exception("Could not see wfd_subelems in GO's peer entry") |
| dev[2].p2p_stop_find() |
| |
| if dev[0].request("WFD_SUBELEM_GET 2") != audio: |
| raise Exception("Unexpected WFD_SUBELEM_GET 2 value") |
| if dev[0].request("WFD_SUBELEM_GET 3") != video: |
| raise Exception("Unexpected WFD_SUBELEM_GET 3 value") |
| if dev[0].request("WFD_SUBELEM_GET 4") != video_3d: |
| raise Exception("Unexpected WFD_SUBELEM_GET 42 value") |
| if dev[0].request("WFD_SUBELEM_GET 5") != prot: |
| raise Exception("Unexpected WFD_SUBELEM_GET 5 value") |
| if "FAIL" not in dev[0].request("WFD_SUBELEM_SET "): |
| raise Exception("Unexpected WFD_SUBELEM_SET success") |
| if "FAIL" not in dev[0].request("WFD_SUBELEM_SET 6"): |
| raise Exception("Unexpected WFD_SUBELEM_SET success") |
| if "OK" not in dev[0].request("WFD_SUBELEM_SET 6 "): |
| raise Exception("Unexpected WFD_SUBELEM_SET failure") |
| if "FAIL" not in dev[0].request("WFD_SUBELEM_SET 6 0"): |
| raise Exception("Unexpected WFD_SUBELEM_SET success") |
| if "FAIL" not in dev[0].request("WFD_SUBELEM_SET 6 0q"): |
| raise Exception("Unexpected WFD_SUBELEM_SET success") |
| if dev[0].request("WFD_SUBELEM_GET 6") != "": |
| raise Exception("Unexpected WFD_SUBELEM_GET 6 response") |
| if dev[0].request("WFD_SUBELEM_GET 8") != "": |
| raise Exception("Unexpected WFD_SUBELEM_GET 8 response") |
| |
| if dev[0].global_request("WFD_SUBELEM_GET 2") != audio: |
| raise Exception("Unexpected WFD_SUBELEM_GET 2 value from global interface") |
| if "OK" not in dev[0].global_request("WFD_SUBELEM_SET 1 0006020304050608"): |
| raise Exception("WFD_SUBELEM_SET failed on global interface") |
| if dev[0].request("WFD_SUBELEM_GET 1") != "0006020304050608": |
| raise Exception("Unexpected WFD_SUBELEM_GET 1 value (per-interface)") |
| |
| elems = dev[0].request("WFD_SUBELEM_GET all") |
| if "OK" not in dev[0].request("WFD_SUBELEM_SET all " + elems): |
| raise Exception("WFD_SUBELEM_SET all failed") |
| if dev[0].request("WFD_SUBELEM_GET all") != elems: |
| raise Exception("Mismatch in WFS_SUBELEM_SET/GET all") |
| test = "00000600411c440028" |
| if "OK" not in dev[0].request("WFD_SUBELEM_SET all " + test): |
| raise Exception("WFD_SUBELEM_SET all failed") |
| if dev[0].request("WFD_SUBELEM_GET all") != test: |
| raise Exception("Mismatch in WFS_SUBELEM_SET/GET all") |
| |
| if "FAIL" not in dev[0].request("WFD_SUBELEM_SET all qwerty"): |
| raise Exception("Invalid WFD_SUBELEM_SET all succeeded") |
| if "FAIL" not in dev[0].request("WFD_SUBELEM_SET all 11"): |
| raise Exception("Invalid WFD_SUBELEM_SET all succeeded") |
| dev[0].request("WFD_SUBELEM_SET all 112233445566") |
| dev[0].request("WFD_SUBELEM_SET all ff0000fe0000fd00") |
| |
| if "FAIL" not in dev[0].request("WFD_SUBELEM_SET 300 112233"): |
| raise Exception("Invalid WFD_SUBELEM_SET 300 succeeded") |
| if "FAIL" not in dev[0].request("WFD_SUBELEM_SET -1 112233"): |
| raise Exception("Invalid WFD_SUBELEM_SET -1 succeeded") |
| if "FAIL" not in dev[0].request("WFD_SUBELEM_GET 300"): |
| raise Exception("Invalid WFD_SUBELEM_GET 300 succeeded") |
| if "FAIL" not in dev[0].request("WFD_SUBELEM_GET -1"): |
| raise Exception("Invalid WFD_SUBELEM_GET -1 succeeded") |
| |
| dev[0].request("SET wifi_display 0") |
| dev[1].request("SET wifi_display 0") |
| dev[2].request("SET wifi_display 0") |
| |
| def enable_wifi_display(dev): |
| dev.request("SET wifi_display 1") |
| dev.request("WFD_SUBELEM_SET 0 000600411c440028") |
| |
| def test_wifi_display_go_invite(dev): |
| """P2P GO with Wi-Fi Display inviting a client to join""" |
| addr0 = dev[0].p2p_dev_addr() |
| addr1 = dev[1].p2p_dev_addr() |
| |
| try: |
| enable_wifi_display(dev[0]) |
| enable_wifi_display(dev[1]) |
| enable_wifi_display(dev[2]) |
| |
| dev[1].p2p_listen() |
| if not dev[0].discover_peer(addr1, social=True): |
| raise Exception("Peer " + addr1 + " not found") |
| dev[0].p2p_listen() |
| if not dev[1].discover_peer(addr0, social=True): |
| raise Exception("Peer " + addr0 + " not found") |
| dev[1].p2p_listen() |
| |
| logger.info("Authorize invitation") |
| pin = dev[1].wps_read_pin() |
| dev[1].global_request("P2P_CONNECT " + addr0 + " " + pin + " join auth") |
| |
| dev[0].p2p_start_go(freq=2412) |
| |
| # Add test client to the group |
| connect_cli(dev[0], dev[2], social=True, freq=2412) |
| |
| logger.info("Invite peer to join the group") |
| dev[0].p2p_go_authorize_client(pin) |
| dev[0].global_request("P2P_INVITE group=" + dev[0].group_ifname + " peer=" + addr1) |
| ev = dev[1].wait_global_event(["P2P-INVITATION-RECEIVED", |
| "P2P-GROUP-STARTED"], timeout=20) |
| if ev is None: |
| raise Exception("Timeout on invitation on peer") |
| if "P2P-INVITATION-RECEIVED" in ev: |
| raise Exception("Unexpected request to accept pre-authorized invitation") |
| |
| dev[0].remove_group() |
| dev[1].wait_go_ending_session() |
| dev[2].wait_go_ending_session() |
| |
| finally: |
| dev[0].request("SET wifi_display 0") |
| dev[1].request("SET wifi_display 0") |
| dev[2].request("SET wifi_display 0") |
| |
| def test_wifi_display_persistent_group(dev): |
| """P2P persistent group formation and re-invocation with Wi-Fi Display enabled""" |
| try: |
| enable_wifi_display(dev[0]) |
| enable_wifi_display(dev[1]) |
| enable_wifi_display(dev[2]) |
| |
| form(dev[0], dev[1]) |
| peer = dev[1].get_peer(dev[0].p2p_dev_addr()) |
| listen_freq = peer['listen_freq'] |
| invite_from_cli(dev[0], dev[1]) |
| invite_from_go(dev[0], dev[1]) |
| |
| dev[0].dump_monitor() |
| dev[1].dump_monitor() |
| networks = dev[0].list_networks() |
| if len(networks) != 1: |
| raise Exception("Unexpected number of networks") |
| if "[P2P-PERSISTENT]" not in networks[0]['flags']: |
| raise Exception("Not the persistent group data") |
| if "OK" not in dev[0].global_request("P2P_GROUP_ADD persistent=" + networks[0]['id'] + " freq=" + listen_freq): |
| raise Exception("Could not start GO") |
| connect_cli(dev[0], dev[2], social=True, freq=listen_freq) |
| dev[0].dump_monitor() |
| dev[1].dump_monitor() |
| invite(dev[1], dev[0]) |
| ev = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=30) |
| if ev is None: |
| raise Exception("Timeout on group re-invocation (on client)") |
| ev = dev[0].wait_global_event(["P2P-GROUP-STARTED"], timeout=0.1) |
| if ev is not None: |
| raise Exception("Unexpected P2P-GROUP-START on GO") |
| hwsim_utils.test_connectivity_p2p(dev[0], dev[1]) |
| |
| finally: |
| dev[0].request("SET wifi_display 0") |
| dev[1].request("SET wifi_display 0") |
| dev[2].request("SET wifi_display 0") |
| |
| def test_wifi_display_invalid_subelem(dev): |
| """Wi-Fi Display and invalid subelement parsing""" |
| addr1 = dev[1].p2p_dev_addr() |
| |
| try: |
| enable_wifi_display(dev[0]) |
| enable_wifi_display(dev[1]) |
| dev[1].request("WFD_SUBELEM_SET 0 ffff00411c440028") |
| |
| dev[1].p2p_listen() |
| dev[0].p2p_find(social=True) |
| ev = dev[0].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10) |
| if ev is None: |
| raise Exception("Device discovery timed out") |
| if "wfd_dev_info=" in ev: |
| raise Exception("Invalid WFD subelement was shown") |
| |
| finally: |
| dev[0].request("SET wifi_display 0") |
| dev[1].request("SET wifi_display 0") |
