test_ibss.py 2.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. #!/usr/bin/python
  2. #
  3. # IBSS 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. import logging
  9. logger = logging.getLogger(__name__)
  10. import time
  11. import hwsim_utils
  12. def connect_ibss_cmd(dev, id):
  13. dev.dump_monitor()
  14. dev.select_network(id)
  15. def wait_ibss_connection(dev):
  16. logger.info(dev.ifname + " waiting for IBSS start/join to complete")
  17. ev = dev.wait_event(["CTRL-EVENT-CONNECTED"], timeout=20)
  18. if ev is None:
  19. raise Exception("Connection to the IBSS timed out")
  20. def wait_4way_handshake(dev1, dev2):
  21. logger.info(dev1.ifname + " waiting for 4-way handshake completion with " + dev2.ifname + " " + dev2.p2p_interface_addr())
  22. ev = dev1.wait_event(["IBSS-RSN-COMPLETED " + dev2.p2p_interface_addr()],
  23. timeout=20)
  24. if ev is None:
  25. raise Exception("4-way handshake in IBSS timed out")
  26. def add_ibss(dev, ssid, psk=None, proto=None, key_mgmt=None, pairwise=None, group=None):
  27. id = dev.add_network()
  28. dev.set_network(id, "mode", "1")
  29. dev.set_network(id, "frequency", "2412")
  30. dev.set_network_quoted(id, "ssid", ssid)
  31. if psk:
  32. dev.set_network_quoted(id, "psk", psk)
  33. if proto:
  34. dev.set_network(id, "proto", proto)
  35. if key_mgmt:
  36. dev.set_network(id, "key_mgmt", key_mgmt)
  37. if pairwise:
  38. dev.set_network(id, "pairwise", pairwise)
  39. if group:
  40. dev.set_network(id, "group", group)
  41. return id
  42. def add_ibss_rsn(dev, ssid):
  43. return add_ibss(dev, ssid, "12345678", "RSN", "WPA-PSK", "CCMP", "CCMP")
  44. def test_ibss_rsn(dev):
  45. """IBSS RSN"""
  46. ssid="ibss-rsn"
  47. logger.info("Start IBSS on the first STA")
  48. id = add_ibss_rsn(dev[0], ssid)
  49. connect_ibss_cmd(dev[0], id)
  50. wait_ibss_connection(dev[0])
  51. logger.info("Join two STAs to the IBSS")
  52. id = add_ibss_rsn(dev[1], ssid)
  53. connect_ibss_cmd(dev[1], id)
  54. wait_ibss_connection(dev[1])
  55. wait_4way_handshake(dev[0], dev[1])
  56. wait_4way_handshake(dev[1], dev[0])
  57. id = add_ibss_rsn(dev[2], ssid)
  58. connect_ibss_cmd(dev[2], id)
  59. wait_ibss_connection(dev[2])
  60. wait_4way_handshake(dev[0], dev[2])
  61. wait_4way_handshake(dev[2], dev[0])
  62. # Allow some time for all peers to complete key setup
  63. time.sleep(3)
  64. hwsim_utils.test_connectivity(dev[0].ifname, dev[1].ifname)
  65. hwsim_utils.test_connectivity(dev[0].ifname, dev[2].ifname)
  66. time.sleep(2)
  67. hwsim_utils.test_connectivity(dev[1].ifname, dev[2].ifname)