test_ibss.py 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165
  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()
  10. import time
  11. import re
  12. import hwsim_utils
  13. def connect_ibss_cmd(dev, id):
  14. dev.dump_monitor()
  15. dev.select_network(id)
  16. def wait_ibss_connection(dev):
  17. logger.info(dev.ifname + " waiting for IBSS start/join to complete")
  18. ev = dev.wait_event(["CTRL-EVENT-CONNECTED"], timeout=20)
  19. if ev is None:
  20. raise Exception("Connection to the IBSS timed out")
  21. exp = r'<.>(CTRL-EVENT-CONNECTED) - Connection to ([0-9a-f:]*) completed.*'
  22. s = re.split(exp, ev)
  23. if len(s) < 3:
  24. return None
  25. return s[2]
  26. def wait_4way_handshake(dev1, dev2):
  27. logger.info(dev1.ifname + " waiting for 4-way handshake completion with " + dev2.ifname + " " + dev2.p2p_interface_addr())
  28. ev = dev1.wait_event(["IBSS-RSN-COMPLETED " + dev2.p2p_interface_addr()],
  29. timeout=20)
  30. if ev is None:
  31. raise Exception("4-way handshake in IBSS timed out")
  32. def wait_4way_handshake2(dev1, dev2, dev3):
  33. logger.info(dev1.ifname + " waiting for 4-way handshake completion with " + dev2.ifname + " " + dev2.p2p_interface_addr() + " and " + dev3.p2p_interface_addr())
  34. ev = dev1.wait_event(["IBSS-RSN-COMPLETED " + dev2.p2p_interface_addr(),
  35. "IBSS-RSN-COMPLETED " + dev3.p2p_interface_addr()],
  36. timeout=20)
  37. if ev is None:
  38. raise Exception("4-way handshake in IBSS timed out")
  39. ev = dev1.wait_event(["IBSS-RSN-COMPLETED " + dev2.p2p_interface_addr(),
  40. "IBSS-RSN-COMPLETED " + dev3.p2p_interface_addr()],
  41. timeout=20)
  42. if ev is None:
  43. raise Exception("4-way handshake in IBSS timed out")
  44. def add_ibss(dev, ssid, psk=None, proto=None, key_mgmt=None, pairwise=None, group=None):
  45. id = dev.add_network()
  46. dev.set_network(id, "mode", "1")
  47. dev.set_network(id, "frequency", "2412")
  48. dev.set_network_quoted(id, "ssid", ssid)
  49. if psk:
  50. dev.set_network_quoted(id, "psk", psk)
  51. if proto:
  52. dev.set_network(id, "proto", proto)
  53. if key_mgmt:
  54. dev.set_network(id, "key_mgmt", key_mgmt)
  55. if pairwise:
  56. dev.set_network(id, "pairwise", pairwise)
  57. if group:
  58. dev.set_network(id, "group", group)
  59. return id
  60. def add_ibss_rsn(dev, ssid):
  61. return add_ibss(dev, ssid, "12345678", "RSN", "WPA-PSK", "CCMP", "CCMP")
  62. def add_ibss_wpa_none(dev, ssid):
  63. return add_ibss(dev, ssid, "12345678", "WPA", "WPA-NONE", "TKIP", "TKIP")
  64. def test_ibss_rsn(dev):
  65. """IBSS RSN"""
  66. ssid="ibss-rsn"
  67. logger.info("Start IBSS on the first STA")
  68. id = add_ibss_rsn(dev[0], ssid)
  69. connect_ibss_cmd(dev[0], id)
  70. bssid0 = wait_ibss_connection(dev[0])
  71. logger.info("Join two STAs to the IBSS")
  72. id = add_ibss_rsn(dev[1], ssid)
  73. connect_ibss_cmd(dev[1], id)
  74. bssid1 = wait_ibss_connection(dev[1])
  75. if bssid0 != bssid1:
  76. logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
  77. # try to merge with a scan
  78. dev[1].scan()
  79. wait_4way_handshake(dev[0], dev[1])
  80. wait_4way_handshake(dev[1], dev[0])
  81. id = add_ibss_rsn(dev[2], ssid)
  82. connect_ibss_cmd(dev[2], id)
  83. bssid2 = wait_ibss_connection(dev[2])
  84. if bssid0 != bssid2:
  85. logger.info("STA0 BSSID " + bssid0 + " differs from STA2 BSSID " + bssid2)
  86. # try to merge with a scan
  87. dev[2].scan()
  88. wait_4way_handshake(dev[0], dev[2])
  89. wait_4way_handshake2(dev[2], dev[0], dev[1])
  90. # Allow some time for all peers to complete key setup
  91. time.sleep(3)
  92. hwsim_utils.test_connectivity(dev[0].ifname, dev[1].ifname)
  93. hwsim_utils.test_connectivity(dev[0].ifname, dev[2].ifname)
  94. hwsim_utils.test_connectivity(dev[1].ifname, dev[2].ifname)
  95. def test_ibss_wpa_none(dev):
  96. """IBSS WPA-None"""
  97. ssid="ibss-wpa-none"
  98. logger.info("Start IBSS on the first STA")
  99. id = add_ibss_wpa_none(dev[0], ssid)
  100. connect_ibss_cmd(dev[0], id)
  101. bssid0 = wait_ibss_connection(dev[0])
  102. logger.info("Join two STAs to the IBSS")
  103. id = add_ibss_wpa_none(dev[1], ssid)
  104. connect_ibss_cmd(dev[1], id)
  105. id = add_ibss_wpa_none(dev[2], ssid)
  106. connect_ibss_cmd(dev[2], id)
  107. # This is a bit ugly, but no one really cares about WPA-None, so there may
  108. # not be enough justification to clean this up.. For now, wpa_supplicant
  109. # will show two connection events with mac80211_hwsim where the first one
  110. # comes with all zeros address.
  111. if bssid0 == "00:00:00:00:00:00":
  112. logger.info("Waiting for real BSSID on the first STA")
  113. bssid0 = wait_ibss_connection(dev[0])
  114. bssid1 = wait_ibss_connection(dev[1])
  115. if bssid0 != bssid1:
  116. logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
  117. bssid1 = wait_ibss_connection(dev[1])
  118. bssid2 = wait_ibss_connection(dev[2])
  119. if bssid0 != bssid2:
  120. logger.info("STA0 BSSID " + bssid0 + " differs from STA2 BSSID " + bssid2)
  121. bssid2 = wait_ibss_connection(dev[2])
  122. print bssid0
  123. print bssid1
  124. print bssid2
  125. # Allow some time for all peers to complete key setup
  126. time.sleep(1)
  127. # This is supposed to work, but looks like WPA-None does not work with
  128. # mac80211 currently..
  129. try:
  130. hwsim_utils.test_connectivity(dev[0].ifname, dev[1].ifname)
  131. except Exception, e:
  132. logger.info("Ignoring known connectivity failure: " + str(e))
  133. try:
  134. hwsim_utils.test_connectivity(dev[0].ifname, dev[2].ifname)
  135. except Exception, e:
  136. logger.info("Ignoring known connectivity failure: " + str(e))
  137. try:
  138. hwsim_utils.test_connectivity(dev[1].ifname, dev[2].ifname)
  139. except Exception, e:
  140. logger.info("Ignoring known connectivity failure: " + str(e))