Home Explore Help
Register Sign In
wareck
/
krackattacks
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Tree: d18df681b0
Branches Tags
krackattacks
/
tests
/
hwsim
/
test_ap_roam.py

test_ap_roam.py 2.6 KB
History Raw

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667 
  1. # Roaming tests
    2. 

  2. # Copyright (c) 2013, Jouni Malinen <j@w1.fi>
    3. 

  3. #
    4. 

  4. # This software may be distributed under the terms of the BSD license.
    5. 

  5. # See README for more details.
    6. 

  6. 

  7. import time
    8. 

  8. import subprocess
    9. 

  9. import logging
    10. 

  10. logger = logging.getLogger()
    11. 

  11. 

  12. import hwsim_utils
    13. 

  13. import hostapd
    14. 

  14. 

  15. def test_ap_roam_open(dev, apdev):
    16. 

  16.     """Roam between two open APs"""
    17. 

  17.     hapd0 = hostapd.add_ap(apdev[0]['ifname'], { "ssid": "test-open" })
    18. 

  18.     dev[0].connect("test-open", key_mgmt="NONE")
    19. 

  19.     hwsim_utils.test_connectivity(dev[0], hapd0)
    20. 

  20.     hapd1 = hostapd.add_ap(apdev[1]['ifname'], { "ssid": "test-open" })
    21. 

  21.     dev[0].scan(type="ONLY")
    22. 

  22.     dev[0].roam(apdev[1]['bssid'])
    23. 

  23.     hwsim_utils.test_connectivity(dev[0], hapd1)
    24. 

  24.     dev[0].roam(apdev[0]['bssid'])
    25. 

  25.     hwsim_utils.test_connectivity(dev[0], hapd0)
    26. 

  26. 

  27. def test_ap_roam_wpa2_psk(dev, apdev):
    28. 

  28.     """Roam between two WPA2-PSK APs"""
    29. 

  29.     params = hostapd.wpa2_params(ssid="test-wpa2-psk", passphrase="12345678")
    30. 

  30.     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
    31. 

  31.     dev[0].connect("test-wpa2-psk", psk="12345678")
    32. 

  32.     hwsim_utils.test_connectivity(dev[0], hapd0)
    33. 

  33.     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
    34. 

  34.     dev[0].scan(type="ONLY")
    35. 

  35.     dev[0].roam(apdev[1]['bssid'])
    36. 

  36.     hwsim_utils.test_connectivity(dev[0], hapd1)
    37. 

  37.     dev[0].roam(apdev[0]['bssid'])
    38. 

  38.     hwsim_utils.test_connectivity(dev[0], hapd0)
    39. 

  39. 

  40. def test_ap_reassociation_to_same_bss(dev, apdev):
    41. 

  41.     """Reassociate to the same BSS"""
    42. 

  42.     hapd = hostapd.add_ap(apdev[0]['ifname'], { "ssid": "test-open" })
    43. 

  43.     dev[0].connect("test-open", key_mgmt="NONE")
    44. 

  44. 

  45.     dev[0].request("REASSOCIATE")
    46. 

  46.     dev[0].wait_connected(timeout=10, error="Reassociation timed out")
    47. 

  47.     hwsim_utils.test_connectivity(dev[0], hapd)
    48. 

  48. 

  49.     dev[0].request("REATTACH")
    50. 

  50.     dev[0].wait_connected(timeout=10, error="Reattach timed out")
    51. 

  51.     hwsim_utils.test_connectivity(dev[0], hapd)
    52. 

  52. 

  53. def test_ap_roam_set_bssid(dev, apdev):
    54. 

  54.     """Roam control"""
    55. 

  55.     hostapd.add_ap(apdev[0]['ifname'], { "ssid": "test-open" })
    56. 

  56.     hostapd.add_ap(apdev[1]['ifname'], { "ssid": "test-open" })
    57. 

  57.     id = dev[0].connect("test-open", key_mgmt="NONE", bssid=apdev[1]['bssid'],
    58. 

  58.                         scan_freq="2412")
    59. 

  59.     if dev[0].get_status_field('bssid') != apdev[1]['bssid']:
    60. 

  60.         raise Exception("Unexpected BSS")
    61. 

  61.     # for now, these are just verifying that the code path to indicate
    62. 

  62.     # within-ESS roaming changes can be executed; the actual results of those
    63. 

  63.     # operations are not currently verified (that would require a test driver
    64. 

  64.     # that does BSS selection)
    65. 

  65.     dev[0].set_network(id, "bssid", "")
    66. 

  66.     dev[0].set_network(id, "bssid", apdev[0]['bssid'])
    67. 

  67.     dev[0].set_network(id, "bssid", apdev[1]['bssid'])
    68.