test_ap_wpa2.py 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141
  1. #!/usr/bin/python
  2. #
  3. # Tests with a WPA2-PSK AP
  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. import time
  9. import subprocess
  10. import logging
  11. logger = logging.getLogger(__name__)
  12. import hwsim_utils
  13. def connect_sta(sta):
  14. logger.info("Connect STA " + sta.ifname + " to AP")
  15. id = sta.add_network()
  16. sta.set_network_quoted(id, "ssid", "test-wpa2-psk")
  17. sta.set_network_quoted(id, "psk", "12345678")
  18. sta.connect_network(id)
  19. def connect_2sta(dev):
  20. connect_sta(dev[0])
  21. connect_sta(dev[1])
  22. hwsim_utils.test_connectivity_sta(dev[0], dev[1])
  23. hwsim_utils.test_connectivity(dev[0].ifname, "wlan2")
  24. hwsim_utils.test_connectivity(dev[1].ifname, "wlan2")
  25. def test_ap_wpa2_psk_2sta(dev):
  26. """WPA2-PSK AP and two stations"""
  27. connect_2sta(dev)
  28. def wlantest_tdls(field, bssid, addr1, addr2):
  29. res = subprocess.check_output(["../../wlantest/wlantest_cli",
  30. "get_tdls_counter", field, bssid, addr1,
  31. addr2]);
  32. if "FAIL" in res:
  33. raise Exception("wlantest_cli command failed")
  34. return int(res)
  35. def wlantest_tdls_clear(bssid, addr1, addr2):
  36. subprocess.call(["../../wlantest/wlantest_cli",
  37. "clear_tdls_counters", bssid, addr1, addr2]);
  38. def wlantest_setup():
  39. subprocess.call(["../../wlantest/wlantest_cli", "flush"]);
  40. subprocess.call(["../../wlantest/wlantest_cli", "add_passphrase",
  41. "12345678"]);
  42. def wlantest_tdls_packet_counters(bssid, addr0, addr1):
  43. dl = wlantest_tdls("valid_direct_link", bssid, addr0, addr1);
  44. inv_dl = wlantest_tdls("invalid_direct_link", bssid, addr0, addr1);
  45. ap = wlantest_tdls("valid_ap_path", bssid, addr0, addr1);
  46. inv_ap = wlantest_tdls("invalid_ap_path", bssid, addr0, addr1);
  47. return [dl,inv_dl,ap,inv_ap]
  48. def tdls_check_dl(sta0, sta1, bssid, addr0, addr1):
  49. wlantest_tdls_clear(bssid, addr0, addr1);
  50. hwsim_utils.test_connectivity_sta(sta0, sta1)
  51. [dl,inv_dl,ap,inv_ap] = wlantest_tdls_packet_counters(bssid, addr0, addr1)
  52. if dl == 0:
  53. raise Exception("No valid frames through direct link")
  54. if inv_dl > 0:
  55. raise Exception("Invalid frames through direct link")
  56. if ap > 0:
  57. raise Exception("Unexpected frames through AP path")
  58. if inv_ap > 0:
  59. raise Exception("Invalid frames through AP path")
  60. def tdls_check_ap(sta0, sta1, bssid, addr0, addr1):
  61. wlantest_tdls_clear(bssid, addr0, addr1);
  62. hwsim_utils.test_connectivity_sta(sta0, sta1)
  63. [dl,inv_dl,ap,inv_ap] = wlantest_tdls_packet_counters(bssid, addr0, addr1)
  64. if dl > 0:
  65. raise Exception("Unexpected frames through direct link")
  66. if inv_dl > 0:
  67. raise Exception("Invalid frames through direct link")
  68. if ap == 0:
  69. raise Exception("No valid frames through AP path")
  70. if inv_ap > 0:
  71. raise Exception("Invalid frames through AP path")
  72. def setup_tdls(sta0, sta1, bssid, reverse=False):
  73. logger.info("Setup TDLS")
  74. addr0 = sta0.p2p_interface_addr()
  75. addr1 = sta1.p2p_interface_addr()
  76. wlantest_tdls_clear(bssid, addr0, addr1);
  77. wlantest_tdls_clear(bssid, addr1, addr0);
  78. sta0.tdls_setup(addr1)
  79. time.sleep(1)
  80. if reverse:
  81. addr1 = sta0.p2p_interface_addr()
  82. addr0 = sta1.p2p_interface_addr()
  83. conf = wlantest_tdls("setup_conf_ok", bssid, addr0, addr1);
  84. if conf == 0:
  85. raise Exception("No TDLS Setup Confirm (success) seen")
  86. tdls_check_dl(sta0, sta1, bssid, addr0, addr1)
  87. def teardown_tdls(sta0, sta1, bssid):
  88. logger.info("Teardown TDLS")
  89. addr0 = sta0.p2p_interface_addr()
  90. addr1 = sta1.p2p_interface_addr()
  91. sta0.tdls_teardown(addr1)
  92. time.sleep(1)
  93. teardown = wlantest_tdls("teardown", bssid, addr0, addr1);
  94. if teardown == 0:
  95. raise Exception("No TDLS Setup Teardown seen")
  96. tdls_check_ap(sta0, sta1, bssid, addr0, addr1)
  97. def test_ap_wpa2_tdls(dev):
  98. """WPA2-PSK AP and two stations using TDLS"""
  99. bssid = "02:00:00:00:02:00"
  100. wlantest_setup()
  101. connect_2sta(dev)
  102. setup_tdls(dev[0], dev[1], bssid)
  103. teardown_tdls(dev[0], dev[1], bssid)
  104. setup_tdls(dev[1], dev[0], bssid)
  105. #teardown_tdls(dev[0], dev[1], bssid)
  106. def test_ap_wpa2_tdls_concurrent_init(dev):
  107. """Concurrent TDLS setup initiation"""
  108. bssid = "02:00:00:00:02:00"
  109. wlantest_setup()
  110. connect_2sta(dev)
  111. dev[0].request("SET tdls_testing 0x80")
  112. setup_tdls(dev[1], dev[0], bssid, reverse=True)
  113. def test_ap_wpa2_tdls_concurrent_init2(dev):
  114. """Concurrent TDLS setup initiation (reverse)"""
  115. bssid = "02:00:00:00:02:00"
  116. wlantest_setup()
  117. connect_2sta(dev)
  118. dev[1].request("SET tdls_testing 0x80")
  119. setup_tdls(dev[0], dev[1], bssid)
  120. def add_tests(tests):
  121. tests.append(test_ap_wpa2_psk_2sta)
  122. tests.append(test_ap_wpa2_tdls)
  123. tests.append(test_ap_wpa2_tdls_concurrent_init)
  124. tests.append(test_ap_wpa2_tdls_concurrent_init2)