test_ap_ft.py 41 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999
  1. # Fast BSS Transition tests
  2. # Copyright (c) 2013-2015, Jouni Malinen <j@w1.fi>
  3. #
  4. # This software may be distributed under the terms of the BSD license.
  5. # See README for more details.
  6. import binascii
  7. import os
  8. import time
  9. import subprocess
  10. import logging
  11. logger = logging.getLogger()
  12. import hwsim_utils
  13. import hostapd
  14. from utils import HwsimSkip, alloc_fail, fail_test, skip_with_fips
  15. from wlantest import Wlantest
  16. from test_ap_psk import check_mib, find_wpas_process, read_process_memory, verify_not_present, get_key_locations
  17. def ft_base_rsn():
  18. params = { "wpa": "2",
  19. "wpa_key_mgmt": "FT-PSK",
  20. "rsn_pairwise": "CCMP" }
  21. return params
  22. def ft_base_mixed():
  23. params = { "wpa": "3",
  24. "wpa_key_mgmt": "WPA-PSK FT-PSK",
  25. "wpa_pairwise": "TKIP",
  26. "rsn_pairwise": "CCMP" }
  27. return params
  28. def ft_params(rsn=True, ssid=None, passphrase=None):
  29. if rsn:
  30. params = ft_base_rsn()
  31. else:
  32. params = ft_base_mixed()
  33. if ssid:
  34. params["ssid"] = ssid
  35. if passphrase:
  36. params["wpa_passphrase"] = passphrase
  37. params["mobility_domain"] = "a1b2"
  38. params["r0_key_lifetime"] = "10000"
  39. params["pmk_r1_push"] = "1"
  40. params["reassociation_deadline"] = "1000"
  41. return params
  42. def ft_params1(rsn=True, ssid=None, passphrase=None):
  43. params = ft_params(rsn, ssid, passphrase)
  44. params['nas_identifier'] = "nas1.w1.fi"
  45. params['r1_key_holder'] = "000102030405"
  46. params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 100102030405060708090a0b0c0d0e0f",
  47. "02:00:00:00:04:00 nas2.w1.fi 300102030405060708090a0b0c0d0e0f" ]
  48. params['r1kh'] = "02:00:00:00:04:00 00:01:02:03:04:06 200102030405060708090a0b0c0d0e0f"
  49. return params
  50. def ft_params2(rsn=True, ssid=None, passphrase=None):
  51. params = ft_params(rsn, ssid, passphrase)
  52. params['nas_identifier'] = "nas2.w1.fi"
  53. params['r1_key_holder'] = "000102030406"
  54. params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 200102030405060708090a0b0c0d0e0f",
  55. "02:00:00:00:04:00 nas2.w1.fi 000102030405060708090a0b0c0d0e0f" ]
  56. params['r1kh'] = "02:00:00:00:03:00 00:01:02:03:04:05 300102030405060708090a0b0c0d0e0f"
  57. return params
  58. def ft_params1_r0kh_mismatch(rsn=True, ssid=None, passphrase=None):
  59. params = ft_params(rsn, ssid, passphrase)
  60. params['nas_identifier'] = "nas1.w1.fi"
  61. params['r1_key_holder'] = "000102030405"
  62. params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 100102030405060708090a0b0c0d0e0f",
  63. "12:00:00:00:04:00 nas2.w1.fi 300102030405060708090a0b0c0d0e0f" ]
  64. params['r1kh'] = "12:00:00:00:04:00 10:01:02:03:04:06 200102030405060708090a0b0c0d0e0f"
  65. return params
  66. def ft_params2_incorrect_rrb_key(rsn=True, ssid=None, passphrase=None):
  67. params = ft_params(rsn, ssid, passphrase)
  68. params['nas_identifier'] = "nas2.w1.fi"
  69. params['r1_key_holder'] = "000102030406"
  70. params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 200102030405060708090a0b0c0d0ef1",
  71. "02:00:00:00:04:00 nas2.w1.fi 000102030405060708090a0b0c0d0ef2" ]
  72. params['r1kh'] = "02:00:00:00:03:00 00:01:02:03:04:05 300102030405060708090a0b0c0d0ef3"
  73. return params
  74. def ft_params2_r0kh_mismatch(rsn=True, ssid=None, passphrase=None):
  75. params = ft_params(rsn, ssid, passphrase)
  76. params['nas_identifier'] = "nas2.w1.fi"
  77. params['r1_key_holder'] = "000102030406"
  78. params['r0kh'] = [ "12:00:00:00:03:00 nas1.w1.fi 200102030405060708090a0b0c0d0e0f",
  79. "02:00:00:00:04:00 nas2.w1.fi 000102030405060708090a0b0c0d0e0f" ]
  80. params['r1kh'] = "12:00:00:00:03:00 10:01:02:03:04:05 300102030405060708090a0b0c0d0e0f"
  81. return params
  82. def run_roams(dev, apdev, hapd0, hapd1, ssid, passphrase, over_ds=False,
  83. sae=False, eap=False, fail_test=False, roams=1,
  84. pairwise_cipher="CCMP", group_cipher="TKIP CCMP"):
  85. logger.info("Connect to first AP")
  86. if eap:
  87. dev.connect(ssid, key_mgmt="FT-EAP", proto="WPA2", ieee80211w="1",
  88. eap="GPSK", identity="gpsk user",
  89. password="abcdefghijklmnop0123456789abcdef",
  90. scan_freq="2412",
  91. pairwise=pairwise_cipher, group=group_cipher)
  92. else:
  93. if sae:
  94. key_mgmt="FT-SAE"
  95. else:
  96. key_mgmt="FT-PSK"
  97. dev.connect(ssid, psk=passphrase, key_mgmt=key_mgmt, proto="WPA2",
  98. ieee80211w="1", scan_freq="2412",
  99. pairwise=pairwise_cipher, group=group_cipher)
  100. if dev.get_status_field('bssid') == apdev[0]['bssid']:
  101. ap1 = apdev[0]
  102. ap2 = apdev[1]
  103. hapd1ap = hapd0
  104. hapd2ap = hapd1
  105. else:
  106. ap1 = apdev[1]
  107. ap2 = apdev[0]
  108. hapd1ap = hapd1
  109. hapd2ap = hapd0
  110. hwsim_utils.test_connectivity(dev, hapd1ap)
  111. dev.scan_for_bss(ap2['bssid'], freq="2412")
  112. for i in range(0, roams):
  113. logger.info("Roam to the second AP")
  114. if over_ds:
  115. dev.roam_over_ds(ap2['bssid'], fail_test=fail_test)
  116. else:
  117. dev.roam(ap2['bssid'], fail_test=fail_test)
  118. if fail_test:
  119. return
  120. if dev.get_status_field('bssid') != ap2['bssid']:
  121. raise Exception("Did not connect to correct AP")
  122. if i == 0 or i == roams - 1:
  123. hwsim_utils.test_connectivity(dev, hapd2ap)
  124. logger.info("Roam back to the first AP")
  125. if over_ds:
  126. dev.roam_over_ds(ap1['bssid'])
  127. else:
  128. dev.roam(ap1['bssid'])
  129. if dev.get_status_field('bssid') != ap1['bssid']:
  130. raise Exception("Did not connect to correct AP")
  131. if i == 0 or i == roams - 1:
  132. hwsim_utils.test_connectivity(dev, hapd1ap)
  133. def test_ap_ft(dev, apdev):
  134. """WPA2-PSK-FT AP"""
  135. ssid = "test-ft"
  136. passphrase="12345678"
  137. params = ft_params1(ssid=ssid, passphrase=passphrase)
  138. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  139. params = ft_params2(ssid=ssid, passphrase=passphrase)
  140. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  141. run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase)
  142. if "[WPA2-FT/PSK-CCMP]" not in dev[0].request("SCAN_RESULTS"):
  143. raise Exception("Scan results missing RSN element info")
  144. def test_ap_ft_many(dev, apdev):
  145. """WPA2-PSK-FT AP multiple times"""
  146. ssid = "test-ft"
  147. passphrase="12345678"
  148. params = ft_params1(ssid=ssid, passphrase=passphrase)
  149. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  150. params = ft_params2(ssid=ssid, passphrase=passphrase)
  151. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  152. run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, roams=50)
  153. def test_ap_ft_mixed(dev, apdev):
  154. """WPA2-PSK-FT mixed-mode AP"""
  155. ssid = "test-ft-mixed"
  156. passphrase="12345678"
  157. params = ft_params1(rsn=False, ssid=ssid, passphrase=passphrase)
  158. hapd = hostapd.add_ap(apdev[0]['ifname'], params)
  159. key_mgmt = hapd.get_config()['key_mgmt']
  160. vals = key_mgmt.split(' ')
  161. if vals[0] != "WPA-PSK" or vals[1] != "FT-PSK":
  162. raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
  163. params = ft_params2(rsn=False, ssid=ssid, passphrase=passphrase)
  164. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  165. run_roams(dev[0], apdev, hapd, hapd1, ssid, passphrase)
  166. def test_ap_ft_pmf(dev, apdev):
  167. """WPA2-PSK-FT AP with PMF"""
  168. ssid = "test-ft"
  169. passphrase="12345678"
  170. params = ft_params1(ssid=ssid, passphrase=passphrase)
  171. params["ieee80211w"] = "2";
  172. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  173. params = ft_params2(ssid=ssid, passphrase=passphrase)
  174. params["ieee80211w"] = "2";
  175. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  176. run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase)
  177. def test_ap_ft_over_ds(dev, apdev):
  178. """WPA2-PSK-FT AP over DS"""
  179. ssid = "test-ft"
  180. passphrase="12345678"
  181. params = ft_params1(ssid=ssid, passphrase=passphrase)
  182. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  183. params = ft_params2(ssid=ssid, passphrase=passphrase)
  184. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  185. run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True)
  186. check_mib(dev[0], [ ("dot11RSNAAuthenticationSuiteRequested", "00-0f-ac-4"),
  187. ("dot11RSNAAuthenticationSuiteSelected", "00-0f-ac-4") ])
  188. def test_ap_ft_over_ds_many(dev, apdev):
  189. """WPA2-PSK-FT AP over DS multiple times"""
  190. ssid = "test-ft"
  191. passphrase="12345678"
  192. params = ft_params1(ssid=ssid, passphrase=passphrase)
  193. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  194. params = ft_params2(ssid=ssid, passphrase=passphrase)
  195. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  196. run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
  197. roams=50)
  198. def test_ap_ft_over_ds_unknown_target(dev, apdev):
  199. """WPA2-PSK-FT AP"""
  200. ssid = "test-ft"
  201. passphrase="12345678"
  202. params = ft_params1(ssid=ssid, passphrase=passphrase)
  203. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  204. dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
  205. scan_freq="2412")
  206. dev[0].roam_over_ds("02:11:22:33:44:55", fail_test=True)
  207. def test_ap_ft_over_ds_unexpected(dev, apdev):
  208. """WPA2-PSK-FT AP over DS and unexpected response"""
  209. ssid = "test-ft"
  210. passphrase="12345678"
  211. params = ft_params1(ssid=ssid, passphrase=passphrase)
  212. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  213. params = ft_params2(ssid=ssid, passphrase=passphrase)
  214. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  215. dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
  216. scan_freq="2412")
  217. if dev[0].get_status_field('bssid') == apdev[0]['bssid']:
  218. ap1 = apdev[0]
  219. ap2 = apdev[1]
  220. hapd1ap = hapd0
  221. hapd2ap = hapd1
  222. else:
  223. ap1 = apdev[1]
  224. ap2 = apdev[0]
  225. hapd1ap = hapd1
  226. hapd2ap = hapd0
  227. addr = dev[0].own_addr()
  228. hapd1ap.set("ext_mgmt_frame_handling", "1")
  229. logger.info("Foreign STA address")
  230. msg = {}
  231. msg['fc'] = 13 << 4
  232. msg['da'] = addr
  233. msg['sa'] = ap1['bssid']
  234. msg['bssid'] = ap1['bssid']
  235. msg['payload'] = binascii.unhexlify("06021122334455660102030405060000")
  236. hapd1ap.mgmt_tx(msg)
  237. logger.info("No over-the-DS in progress")
  238. msg['payload'] = binascii.unhexlify("0602" + addr.replace(':', '') + "0102030405060000")
  239. hapd1ap.mgmt_tx(msg)
  240. logger.info("Non-zero status code")
  241. msg['payload'] = binascii.unhexlify("0602" + addr.replace(':', '') + "0102030405060100")
  242. hapd1ap.mgmt_tx(msg)
  243. hapd1ap.dump_monitor()
  244. dev[0].scan_for_bss(ap2['bssid'], freq="2412")
  245. if "OK" not in dev[0].request("FT_DS " + ap2['bssid']):
  246. raise Exception("FT_DS failed")
  247. req = hapd1ap.mgmt_rx()
  248. logger.info("Foreign Target AP")
  249. msg['payload'] = binascii.unhexlify("0602" + addr.replace(':', '') + "0102030405060000")
  250. hapd1ap.mgmt_tx(msg)
  251. addrs = addr.replace(':', '') + ap2['bssid'].replace(':', '')
  252. logger.info("No IEs")
  253. msg['payload'] = binascii.unhexlify("0602" + addrs + "0000")
  254. hapd1ap.mgmt_tx(msg)
  255. logger.info("Invalid IEs (trigger parsing failure)")
  256. msg['payload'] = binascii.unhexlify("0602" + addrs + "00003700")
  257. hapd1ap.mgmt_tx(msg)
  258. logger.info("Too short MDIE")
  259. msg['payload'] = binascii.unhexlify("0602" + addrs + "000036021122")
  260. hapd1ap.mgmt_tx(msg)
  261. logger.info("Mobility domain mismatch")
  262. msg['payload'] = binascii.unhexlify("0602" + addrs + "00003603112201")
  263. hapd1ap.mgmt_tx(msg)
  264. logger.info("No FTIE")
  265. msg['payload'] = binascii.unhexlify("0602" + addrs + "00003603a1b201")
  266. hapd1ap.mgmt_tx(msg)
  267. logger.info("FTIE SNonce mismatch")
  268. msg['payload'] = binascii.unhexlify("0602" + addrs + "00003603a1b201375e0000" + "00000000000000000000000000000000" + "0000000000000000000000000000000000000000000000000000000000000000" + "1000000000000000000000000000000000000000000000000000000000000001" + "030a6e6173322e77312e6669")
  269. hapd1ap.mgmt_tx(msg)
  270. logger.info("No R0KH-ID subelem in FTIE")
  271. snonce = binascii.hexlify(req['payload'][111:111+32])
  272. msg['payload'] = binascii.unhexlify("0602" + addrs + "00003603a1b20137520000" + "00000000000000000000000000000000" + "0000000000000000000000000000000000000000000000000000000000000000" + snonce)
  273. hapd1ap.mgmt_tx(msg)
  274. logger.info("No R0KH-ID subelem mismatch in FTIE")
  275. snonce = binascii.hexlify(req['payload'][111:111+32])
  276. msg['payload'] = binascii.unhexlify("0602" + addrs + "00003603a1b201375e0000" + "00000000000000000000000000000000" + "0000000000000000000000000000000000000000000000000000000000000000" + snonce + "030a11223344556677889900")
  277. hapd1ap.mgmt_tx(msg)
  278. logger.info("No R1KH-ID subelem in FTIE")
  279. r0khid = binascii.hexlify(req['payload'][145:145+10])
  280. msg['payload'] = binascii.unhexlify("0602" + addrs + "00003603a1b201375e0000" + "00000000000000000000000000000000" + "0000000000000000000000000000000000000000000000000000000000000000" + snonce + "030a" + r0khid)
  281. hapd1ap.mgmt_tx(msg)
  282. logger.info("No RSNE")
  283. r0khid = binascii.hexlify(req['payload'][145:145+10])
  284. msg['payload'] = binascii.unhexlify("0602" + addrs + "00003603a1b20137660000" + "00000000000000000000000000000000" + "0000000000000000000000000000000000000000000000000000000000000000" + snonce + "030a" + r0khid + "0106000102030405")
  285. hapd1ap.mgmt_tx(msg)
  286. def test_ap_ft_pmf_over_ds(dev, apdev):
  287. """WPA2-PSK-FT AP over DS with PMF"""
  288. ssid = "test-ft"
  289. passphrase="12345678"
  290. params = ft_params1(ssid=ssid, passphrase=passphrase)
  291. params["ieee80211w"] = "2";
  292. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  293. params = ft_params2(ssid=ssid, passphrase=passphrase)
  294. params["ieee80211w"] = "2";
  295. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  296. run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True)
  297. def test_ap_ft_over_ds_pull(dev, apdev):
  298. """WPA2-PSK-FT AP over DS (pull PMK)"""
  299. ssid = "test-ft"
  300. passphrase="12345678"
  301. params = ft_params1(ssid=ssid, passphrase=passphrase)
  302. params["pmk_r1_push"] = "0"
  303. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  304. params = ft_params2(ssid=ssid, passphrase=passphrase)
  305. params["pmk_r1_push"] = "0"
  306. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  307. run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True)
  308. def test_ap_ft_sae(dev, apdev):
  309. """WPA2-PSK-FT-SAE AP"""
  310. if "SAE" not in dev[0].get_capability("auth_alg"):
  311. raise HwsimSkip("SAE not supported")
  312. ssid = "test-ft"
  313. passphrase="12345678"
  314. params = ft_params1(ssid=ssid, passphrase=passphrase)
  315. params['wpa_key_mgmt'] = "FT-SAE"
  316. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  317. params = ft_params2(ssid=ssid, passphrase=passphrase)
  318. params['wpa_key_mgmt'] = "FT-SAE"
  319. hapd = hostapd.add_ap(apdev[1]['ifname'], params)
  320. key_mgmt = hapd.get_config()['key_mgmt']
  321. if key_mgmt.split(' ')[0] != "FT-SAE":
  322. raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
  323. dev[0].request("SET sae_groups ")
  324. run_roams(dev[0], apdev, hapd0, hapd, ssid, passphrase, sae=True)
  325. def test_ap_ft_sae_over_ds(dev, apdev):
  326. """WPA2-PSK-FT-SAE AP over DS"""
  327. if "SAE" not in dev[0].get_capability("auth_alg"):
  328. raise HwsimSkip("SAE not supported")
  329. ssid = "test-ft"
  330. passphrase="12345678"
  331. params = ft_params1(ssid=ssid, passphrase=passphrase)
  332. params['wpa_key_mgmt'] = "FT-SAE"
  333. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  334. params = ft_params2(ssid=ssid, passphrase=passphrase)
  335. params['wpa_key_mgmt'] = "FT-SAE"
  336. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  337. dev[0].request("SET sae_groups ")
  338. run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, sae=True,
  339. over_ds=True)
  340. def test_ap_ft_eap(dev, apdev):
  341. """WPA2-EAP-FT AP"""
  342. ssid = "test-ft"
  343. passphrase="12345678"
  344. radius = hostapd.radius_params()
  345. params = ft_params1(ssid=ssid, passphrase=passphrase)
  346. params['wpa_key_mgmt'] = "FT-EAP"
  347. params["ieee8021x"] = "1"
  348. params = dict(radius.items() + params.items())
  349. hapd = hostapd.add_ap(apdev[0]['ifname'], params)
  350. key_mgmt = hapd.get_config()['key_mgmt']
  351. if key_mgmt.split(' ')[0] != "FT-EAP":
  352. raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
  353. params = ft_params2(ssid=ssid, passphrase=passphrase)
  354. params['wpa_key_mgmt'] = "FT-EAP"
  355. params["ieee8021x"] = "1"
  356. params = dict(radius.items() + params.items())
  357. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  358. run_roams(dev[0], apdev, hapd, hapd1, ssid, passphrase, eap=True)
  359. if "[WPA2-FT/EAP-CCMP]" not in dev[0].request("SCAN_RESULTS"):
  360. raise Exception("Scan results missing RSN element info")
  361. check_mib(dev[0], [ ("dot11RSNAAuthenticationSuiteRequested", "00-0f-ac-3"),
  362. ("dot11RSNAAuthenticationSuiteSelected", "00-0f-ac-3") ])
  363. # Verify EAPOL reauthentication after FT protocol
  364. if dev[0].get_status_field('bssid') == apdev[0]['bssid']:
  365. ap = hapd
  366. else:
  367. ap = hapd1
  368. ap.request("EAPOL_REAUTH " + dev[0].own_addr())
  369. ev = dev[0].wait_event(["CTRL-EVENT-EAP-STARTED"], timeout=5)
  370. if ev is None:
  371. raise Exception("EAP authentication did not start")
  372. ev = dev[0].wait_event(["CTRL-EVENT-EAP-SUCCESS"], timeout=5)
  373. if ev is None:
  374. raise Exception("EAP authentication did not succeed")
  375. time.sleep(0.1)
  376. hwsim_utils.test_connectivity(dev[0], ap)
  377. def test_ap_ft_eap_pull(dev, apdev):
  378. """WPA2-EAP-FT AP (pull PMK)"""
  379. ssid = "test-ft"
  380. passphrase="12345678"
  381. radius = hostapd.radius_params()
  382. params = ft_params1(ssid=ssid, passphrase=passphrase)
  383. params['wpa_key_mgmt'] = "FT-EAP"
  384. params["ieee8021x"] = "1"
  385. params["pmk_r1_push"] = "0"
  386. params = dict(radius.items() + params.items())
  387. hapd = hostapd.add_ap(apdev[0]['ifname'], params)
  388. key_mgmt = hapd.get_config()['key_mgmt']
  389. if key_mgmt.split(' ')[0] != "FT-EAP":
  390. raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
  391. params = ft_params2(ssid=ssid, passphrase=passphrase)
  392. params['wpa_key_mgmt'] = "FT-EAP"
  393. params["ieee8021x"] = "1"
  394. params["pmk_r1_push"] = "0"
  395. params = dict(radius.items() + params.items())
  396. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  397. run_roams(dev[0], apdev, hapd, hapd1, ssid, passphrase, eap=True)
  398. def test_ap_ft_mismatching_rrb_key_push(dev, apdev):
  399. """WPA2-PSK-FT AP over DS with mismatching RRB key (push)"""
  400. ssid = "test-ft"
  401. passphrase="12345678"
  402. params = ft_params1(ssid=ssid, passphrase=passphrase)
  403. params["ieee80211w"] = "2";
  404. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  405. params = ft_params2_incorrect_rrb_key(ssid=ssid, passphrase=passphrase)
  406. params["ieee80211w"] = "2";
  407. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  408. run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
  409. fail_test=True)
  410. def test_ap_ft_mismatching_rrb_key_pull(dev, apdev):
  411. """WPA2-PSK-FT AP over DS with mismatching RRB key (pull)"""
  412. ssid = "test-ft"
  413. passphrase="12345678"
  414. params = ft_params1(ssid=ssid, passphrase=passphrase)
  415. params["pmk_r1_push"] = "0"
  416. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  417. params = ft_params2_incorrect_rrb_key(ssid=ssid, passphrase=passphrase)
  418. params["pmk_r1_push"] = "0"
  419. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  420. run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
  421. fail_test=True)
  422. def test_ap_ft_mismatching_r0kh_id_pull(dev, apdev):
  423. """WPA2-PSK-FT AP over DS with mismatching R0KH-ID (pull)"""
  424. ssid = "test-ft"
  425. passphrase="12345678"
  426. params = ft_params1(ssid=ssid, passphrase=passphrase)
  427. params["pmk_r1_push"] = "0"
  428. params["nas_identifier"] = "nas0.w1.fi"
  429. hostapd.add_ap(apdev[0]['ifname'], params)
  430. dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
  431. scan_freq="2412")
  432. params = ft_params2(ssid=ssid, passphrase=passphrase)
  433. params["pmk_r1_push"] = "0"
  434. hostapd.add_ap(apdev[1]['ifname'], params)
  435. dev[0].scan_for_bss(apdev[1]['bssid'], freq="2412")
  436. dev[0].roam_over_ds(apdev[1]['bssid'], fail_test=True)
  437. def test_ap_ft_mismatching_rrb_r0kh_push(dev, apdev):
  438. """WPA2-PSK-FT AP over DS with mismatching R0KH key (push)"""
  439. ssid = "test-ft"
  440. passphrase="12345678"
  441. params = ft_params1(ssid=ssid, passphrase=passphrase)
  442. params["ieee80211w"] = "2";
  443. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  444. params = ft_params2_r0kh_mismatch(ssid=ssid, passphrase=passphrase)
  445. params["ieee80211w"] = "2";
  446. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  447. run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
  448. fail_test=True)
  449. def test_ap_ft_mismatching_rrb_r0kh_pull(dev, apdev):
  450. """WPA2-PSK-FT AP over DS with mismatching R0KH key (pull)"""
  451. ssid = "test-ft"
  452. passphrase="12345678"
  453. params = ft_params1_r0kh_mismatch(ssid=ssid, passphrase=passphrase)
  454. params["pmk_r1_push"] = "0"
  455. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  456. params = ft_params2(ssid=ssid, passphrase=passphrase)
  457. params["pmk_r1_push"] = "0"
  458. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  459. run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
  460. fail_test=True)
  461. def test_ap_ft_gtk_rekey(dev, apdev):
  462. """WPA2-PSK-FT AP and GTK rekey"""
  463. ssid = "test-ft"
  464. passphrase="12345678"
  465. params = ft_params1(ssid=ssid, passphrase=passphrase)
  466. params['wpa_group_rekey'] = '1'
  467. hapd = hostapd.add_ap(apdev[0]['ifname'], params)
  468. dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
  469. ieee80211w="1", scan_freq="2412")
  470. ev = dev[0].wait_event(["WPA: Group rekeying completed"], timeout=2)
  471. if ev is None:
  472. raise Exception("GTK rekey timed out after initial association")
  473. hwsim_utils.test_connectivity(dev[0], hapd)
  474. params = ft_params2(ssid=ssid, passphrase=passphrase)
  475. params['wpa_group_rekey'] = '1'
  476. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  477. dev[0].scan_for_bss(apdev[1]['bssid'], freq="2412")
  478. dev[0].roam(apdev[1]['bssid'])
  479. if dev[0].get_status_field('bssid') != apdev[1]['bssid']:
  480. raise Exception("Did not connect to correct AP")
  481. hwsim_utils.test_connectivity(dev[0], hapd1)
  482. ev = dev[0].wait_event(["WPA: Group rekeying completed"], timeout=2)
  483. if ev is None:
  484. raise Exception("GTK rekey timed out after FT protocol")
  485. hwsim_utils.test_connectivity(dev[0], hapd1)
  486. def test_ft_psk_key_lifetime_in_memory(dev, apdev, params):
  487. """WPA2-PSK-FT and key lifetime in memory"""
  488. ssid = "test-ft"
  489. passphrase="04c2726b4b8d5f1b4db9c07aa4d9e9d8f765cb5d25ec817e6cc4fcdd5255db0"
  490. psk = '93c90846ff67af9037ed83fb72b63dbeddaa81d47f926c20909b5886f1d9358d'
  491. pmk = binascii.unhexlify(psk)
  492. p = ft_params1(ssid=ssid, passphrase=passphrase)
  493. hapd0 = hostapd.add_ap(apdev[0]['ifname'], p)
  494. p = ft_params2(ssid=ssid, passphrase=passphrase)
  495. hapd1 = hostapd.add_ap(apdev[1]['ifname'], p)
  496. pid = find_wpas_process(dev[0])
  497. dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
  498. scan_freq="2412")
  499. time.sleep(1)
  500. buf = read_process_memory(pid, pmk)
  501. dev[0].request("DISCONNECT")
  502. dev[0].wait_disconnected()
  503. dev[0].relog()
  504. pmkr0 = None
  505. pmkr1 = None
  506. ptk = None
  507. gtk = None
  508. with open(os.path.join(params['logdir'], 'log0'), 'r') as f:
  509. for l in f.readlines():
  510. if "FT: PMK-R0 - hexdump" in l:
  511. val = l.strip().split(':')[3].replace(' ', '')
  512. pmkr0 = binascii.unhexlify(val)
  513. if "FT: PMK-R1 - hexdump" in l:
  514. val = l.strip().split(':')[3].replace(' ', '')
  515. pmkr1 = binascii.unhexlify(val)
  516. if "FT: KCK - hexdump" in l:
  517. val = l.strip().split(':')[3].replace(' ', '')
  518. kck = binascii.unhexlify(val)
  519. if "FT: KEK - hexdump" in l:
  520. val = l.strip().split(':')[3].replace(' ', '')
  521. kek = binascii.unhexlify(val)
  522. if "FT: TK - hexdump" in l:
  523. val = l.strip().split(':')[3].replace(' ', '')
  524. tk = binascii.unhexlify(val)
  525. if "WPA: Group Key - hexdump" in l:
  526. val = l.strip().split(':')[3].replace(' ', '')
  527. gtk = binascii.unhexlify(val)
  528. if not pmkr0 or not pmkr1 or not kck or not kek or not tk or not gtk:
  529. raise Exception("Could not find keys from debug log")
  530. if len(gtk) != 16:
  531. raise Exception("Unexpected GTK length")
  532. logger.info("Checking keys in memory while associated")
  533. get_key_locations(buf, pmk, "PMK")
  534. get_key_locations(buf, pmkr0, "PMK-R0")
  535. get_key_locations(buf, pmkr1, "PMK-R1")
  536. if pmk not in buf:
  537. raise HwsimSkip("PMK not found while associated")
  538. if pmkr0 not in buf:
  539. raise HwsimSkip("PMK-R0 not found while associated")
  540. if pmkr1 not in buf:
  541. raise HwsimSkip("PMK-R1 not found while associated")
  542. if kck not in buf:
  543. raise Exception("KCK not found while associated")
  544. if kek not in buf:
  545. raise Exception("KEK not found while associated")
  546. if tk in buf:
  547. raise Exception("TK found from memory")
  548. if gtk in buf:
  549. get_key_locations(buf, gtk, "GTK")
  550. raise Exception("GTK found from memory")
  551. logger.info("Checking keys in memory after disassociation")
  552. buf = read_process_memory(pid, pmk)
  553. get_key_locations(buf, pmk, "PMK")
  554. get_key_locations(buf, pmkr0, "PMK-R0")
  555. get_key_locations(buf, pmkr1, "PMK-R1")
  556. # Note: PMK/PSK is still present in network configuration
  557. fname = os.path.join(params['logdir'],
  558. 'ft_psk_key_lifetime_in_memory.memctx-')
  559. verify_not_present(buf, pmkr0, fname, "PMK-R0")
  560. verify_not_present(buf, pmkr1, fname, "PMK-R1")
  561. verify_not_present(buf, kck, fname, "KCK")
  562. verify_not_present(buf, kek, fname, "KEK")
  563. verify_not_present(buf, tk, fname, "TK")
  564. verify_not_present(buf, gtk, fname, "GTK")
  565. dev[0].request("REMOVE_NETWORK all")
  566. logger.info("Checking keys in memory after network profile removal")
  567. buf = read_process_memory(pid, pmk)
  568. get_key_locations(buf, pmk, "PMK")
  569. get_key_locations(buf, pmkr0, "PMK-R0")
  570. get_key_locations(buf, pmkr1, "PMK-R1")
  571. verify_not_present(buf, pmk, fname, "PMK")
  572. verify_not_present(buf, pmkr0, fname, "PMK-R0")
  573. verify_not_present(buf, pmkr1, fname, "PMK-R1")
  574. verify_not_present(buf, kck, fname, "KCK")
  575. verify_not_present(buf, kek, fname, "KEK")
  576. verify_not_present(buf, tk, fname, "TK")
  577. verify_not_present(buf, gtk, fname, "GTK")
  578. def test_ap_ft_invalid_resp(dev, apdev):
  579. """WPA2-PSK-FT AP and invalid response IEs"""
  580. ssid = "test-ft"
  581. passphrase="12345678"
  582. params = ft_params1(ssid=ssid, passphrase=passphrase)
  583. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  584. dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
  585. scan_freq="2412")
  586. params = ft_params2(ssid=ssid, passphrase=passphrase)
  587. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  588. tests = [
  589. # Various IEs for test coverage. The last one is FTIE with invalid
  590. # R1KH-ID subelement.
  591. "020002000000" + "3800" + "38051122334455" + "3754000102030405060708090001020304050607080900010203040506070809000102030405060708090001020304050607080900010203040506070809000102030405060708090001020304050607080900010100",
  592. # FTIE with invalid R0KH-ID subelement (len=0).
  593. "020002000000" + "3754000102030405060708090001020304050607080900010203040506070809000102030405060708090001020304050607080900010203040506070809000102030405060708090001020304050607080900010300",
  594. # FTIE with invalid R0KH-ID subelement (len=49).
  595. "020002000000" + "378500010203040506070809000102030405060708090001020304050607080900010203040506070809000102030405060708090001020304050607080900010203040506070809000102030405060708090001033101020304050607080910111213141516171819202122232425262728293031323334353637383940414243444546474849",
  596. # Invalid RSNE.
  597. "020002000000" + "3000",
  598. # Required IEs missing from protected IE count.
  599. "020002000000" + "3603a1b201" + "375200010203040506070809000102030405060708090001020304050607080900010203040506070809000102030405060708090001020304050607080900010203040506070809000102030405060708090001" + "3900",
  600. # RIC missing from protected IE count.
  601. "020002000000" + "3603a1b201" + "375200020203040506070809000102030405060708090001020304050607080900010203040506070809000102030405060708090001020304050607080900010203040506070809000102030405060708090001" + "3900",
  602. # Protected IE missing.
  603. "020002000000" + "3603a1b201" + "375200ff0203040506070809000102030405060708090001020304050607080900010203040506070809000102030405060708090001020304050607080900010203040506070809000102030405060708090001" + "3900" + "0000" ]
  604. for t in tests:
  605. dev[0].scan_for_bss(apdev[1]['bssid'], freq="2412")
  606. hapd1.set("ext_mgmt_frame_handling", "1")
  607. hapd1.dump_monitor()
  608. if "OK" not in dev[0].request("ROAM " + apdev[1]['bssid']):
  609. raise Exception("ROAM failed")
  610. auth = None
  611. for i in range(20):
  612. msg = hapd1.mgmt_rx()
  613. if msg['subtype'] == 11:
  614. auth = msg
  615. break
  616. if not auth:
  617. raise Exception("Authentication frame not seen")
  618. resp = {}
  619. resp['fc'] = auth['fc']
  620. resp['da'] = auth['sa']
  621. resp['sa'] = auth['da']
  622. resp['bssid'] = auth['bssid']
  623. resp['payload'] = binascii.unhexlify(t)
  624. hapd1.mgmt_tx(resp)
  625. hapd1.set("ext_mgmt_frame_handling", "0")
  626. dev[0].wait_disconnected()
  627. dev[0].request("RECONNECT")
  628. dev[0].wait_connected()
  629. def test_ap_ft_gcmp_256(dev, apdev):
  630. """WPA2-PSK-FT AP with GCMP-256 cipher"""
  631. if "GCMP-256" not in dev[0].get_capability("pairwise"):
  632. raise HwsimSkip("Cipher GCMP-256 not supported")
  633. ssid = "test-ft"
  634. passphrase="12345678"
  635. params = ft_params1(ssid=ssid, passphrase=passphrase)
  636. params['rsn_pairwise'] = "GCMP-256"
  637. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  638. params = ft_params2(ssid=ssid, passphrase=passphrase)
  639. params['rsn_pairwise'] = "GCMP-256"
  640. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  641. run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase,
  642. pairwise_cipher="GCMP-256", group_cipher="GCMP-256")
  643. def test_ap_ft_oom(dev, apdev):
  644. """WPA2-PSK-FT and OOM"""
  645. skip_with_fips(dev[0])
  646. ssid = "test-ft"
  647. passphrase="12345678"
  648. params = ft_params1(ssid=ssid, passphrase=passphrase)
  649. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  650. params = ft_params2(ssid=ssid, passphrase=passphrase)
  651. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  652. dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
  653. scan_freq="2412")
  654. if dev[0].get_status_field('bssid') == apdev[0]['bssid']:
  655. dst = apdev[1]['bssid']
  656. else:
  657. dst = apdev[0]['bssid']
  658. dev[0].scan_for_bss(dst, freq="2412")
  659. with alloc_fail(dev[0], 1, "wpa_ft_gen_req_ies"):
  660. dev[0].roam(dst)
  661. with alloc_fail(dev[0], 1, "wpa_ft_mic"):
  662. dev[0].roam(dst, fail_test=True)
  663. with fail_test(dev[0], 1, "os_get_random;wpa_ft_prepare_auth_request"):
  664. dev[0].roam(dst, fail_test=True)
  665. def test_ap_ft_over_ds_proto(dev, apdev):
  666. """WPA2-PSK-FT AP over DS protocol testing"""
  667. ssid = "test-ft"
  668. passphrase="12345678"
  669. params = ft_params1(ssid=ssid, passphrase=passphrase)
  670. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  671. dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
  672. scan_freq="2412")
  673. # FT Action Response while no FT-over-DS in progress
  674. msg = {}
  675. msg['fc'] = 13 << 4
  676. msg['da'] = dev[0].own_addr()
  677. msg['sa'] = apdev[0]['bssid']
  678. msg['bssid'] = apdev[0]['bssid']
  679. msg['payload'] = binascii.unhexlify("06020200000000000200000004000000")
  680. hapd0.mgmt_tx(msg)
  681. params = ft_params2(ssid=ssid, passphrase=passphrase)
  682. hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
  683. dev[0].scan_for_bss(apdev[1]['bssid'], freq="2412")
  684. hapd0.set("ext_mgmt_frame_handling", "1")
  685. hapd0.dump_monitor()
  686. dev[0].request("FT_DS " + apdev[1]['bssid'])
  687. for i in range(0, 10):
  688. req = hapd0.mgmt_rx()
  689. if req is None:
  690. raise Exception("MGMT RX wait timed out")
  691. if req['subtype'] == 13:
  692. break
  693. req = None
  694. if not req:
  695. raise Exception("FT Action frame not received")
  696. # FT Action Response for unexpected Target AP
  697. msg['payload'] = binascii.unhexlify("0602020000000000" + "f20000000400" + "0000")
  698. hapd0.mgmt_tx(msg)
  699. # FT Action Response without MDIE
  700. msg['payload'] = binascii.unhexlify("0602020000000000" + "020000000400" + "0000")
  701. hapd0.mgmt_tx(msg)
  702. # FT Action Response without FTIE
  703. msg['payload'] = binascii.unhexlify("0602020000000000" + "020000000400" + "0000" + "3603a1b201")
  704. hapd0.mgmt_tx(msg)
  705. # FT Action Response with FTIE SNonce mismatch
  706. msg['payload'] = binascii.unhexlify("0602020000000000" + "020000000400" + "0000" + "3603a1b201" + "3766000000000000000000000000000000000000c4e67ac1999bebd00ff4ae4d5dcaf87896bb060b469f7c78d49623fb395c3455ffffff6b693fe6f8d8c5dfac0a22344750775bd09437f98b238c9f87b97f790c0106000102030406030a6e6173312e77312e6669")
  707. hapd0.mgmt_tx(msg)
  708. def test_ap_ft_rrb(dev, apdev):
  709. """WPA2-PSK-FT RRB protocol testing"""
  710. ssid = "test-ft"
  711. passphrase="12345678"
  712. params = ft_params1(ssid=ssid, passphrase=passphrase)
  713. hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
  714. dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
  715. scan_freq="2412")
  716. _dst_ll = binascii.unhexlify(apdev[0]['bssid'].replace(':',''))
  717. _src_ll = binascii.unhexlify(dev[0].own_addr().replace(':',''))
  718. proto = '\x89\x0d'
  719. ehdr = _dst_ll + _src_ll + proto
  720. # Too short RRB frame
  721. pkt = ehdr + '\x01'
  722. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  723. raise Exception("DATA_TEST_FRAME failed")
  724. # RRB discarded frame wikth unrecognized type
  725. pkt = ehdr + '\x02' + '\x02' + '\x01\x00' + _src_ll
  726. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  727. raise Exception("DATA_TEST_FRAME failed")
  728. # RRB frame too short for action frame
  729. pkt = ehdr + '\x01' + '\x02' + '\x01\x00' + _src_ll
  730. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  731. raise Exception("DATA_TEST_FRAME failed")
  732. # Too short RRB frame (not enough room for Action Frame body)
  733. pkt = ehdr + '\x01' + '\x02' + '\x00\x00' + _src_ll
  734. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  735. raise Exception("DATA_TEST_FRAME failed")
  736. # Unexpected Action frame category
  737. pkt = ehdr + '\x01' + '\x02' + '\x0e\x00' + _src_ll + '\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
  738. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  739. raise Exception("DATA_TEST_FRAME failed")
  740. # Unexpected Action in RRB Request
  741. pkt = ehdr + '\x01' + '\x00' + '\x0e\x00' + _src_ll + '\x06\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
  742. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  743. raise Exception("DATA_TEST_FRAME failed")
  744. # Target AP address in RRB Request does not match with own address
  745. pkt = ehdr + '\x01' + '\x00' + '\x0e\x00' + _src_ll + '\x06\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
  746. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  747. raise Exception("DATA_TEST_FRAME failed")
  748. # Not enough room for status code in RRB Response
  749. pkt = ehdr + '\x01' + '\x01' + '\x0e\x00' + _src_ll + '\x06\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
  750. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  751. raise Exception("DATA_TEST_FRAME failed")
  752. # RRB discarded frame with unknown packet_type
  753. pkt = ehdr + '\x01' + '\x02' + '\x0e\x00' + _src_ll + '\x06\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
  754. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  755. raise Exception("DATA_TEST_FRAME failed")
  756. # RRB Response with non-zero status code; no STA match
  757. pkt = ehdr + '\x01' + '\x01' + '\x10\x00' + _src_ll + '\x06\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00' + '\xff\xff'
  758. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  759. raise Exception("DATA_TEST_FRAME failed")
  760. # RRB Response with zero status code and extra data; STA match
  761. pkt = ehdr + '\x01' + '\x01' + '\x11\x00' + _src_ll + '\x06\x01' + _src_ll + '\x00\x00\x00\x00\x00\x00' + '\x00\x00' + '\x00'
  762. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  763. raise Exception("DATA_TEST_FRAME failed")
  764. # Too short PMK-R1 pull
  765. pkt = ehdr + '\x01' + '\xc8' + '\x0e\x00' + _src_ll + '\x06\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
  766. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  767. raise Exception("DATA_TEST_FRAME failed")
  768. # Too short PMK-R1 resp
  769. pkt = ehdr + '\x01' + '\xc9' + '\x0e\x00' + _src_ll + '\x06\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
  770. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  771. raise Exception("DATA_TEST_FRAME failed")
  772. # Too short PMK-R1 push
  773. pkt = ehdr + '\x01' + '\xca' + '\x0e\x00' + _src_ll + '\x06\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
  774. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  775. raise Exception("DATA_TEST_FRAME failed")
  776. # No matching R0KH address found for PMK-R0 pull response
  777. pkt = ehdr + '\x01' + '\xc9' + '\x5a\x00' + _src_ll + '\x06\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00' + 76*'\00'
  778. if "OK" not in dev[0].request("DATA_TEST_FRAME " + binascii.hexlify(pkt)):
  779. raise Exception("DATA_TEST_FRAME failed")
  780. def test_rsn_ie_proto_ft_psk_sta(dev, apdev):
  781. """RSN element protocol testing for FT-PSK + PMF cases on STA side"""
  782. bssid = apdev[0]['bssid']
  783. ssid = "test-ft"
  784. passphrase="12345678"
  785. params = ft_params1(ssid=ssid, passphrase=passphrase)
  786. params["ieee80211w"] = "1";
  787. # This is the RSN element used normally by hostapd
  788. params['own_ie_override'] = '30140100000fac040100000fac040100000fac048c00' + '3603a1b201'
  789. hapd = hostapd.add_ap(apdev[0]['ifname'], params)
  790. id = dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
  791. ieee80211w="1", scan_freq="2412",
  792. pairwise="CCMP", group="CCMP")
  793. tests = [ ('PMKIDCount field included',
  794. '30160100000fac040100000fac040100000fac048c000000' + '3603a1b201'),
  795. ('Extra IE before RSNE',
  796. 'dd0400000000' + '30140100000fac040100000fac040100000fac048c00' + '3603a1b201'),
  797. ('PMKIDCount and Group Management Cipher suite fields included',
  798. '301a0100000fac040100000fac040100000fac048c000000000fac06' + '3603a1b201'),
  799. ('Extra octet after defined fields (future extensibility)',
  800. '301b0100000fac040100000fac040100000fac048c000000000fac0600' + '3603a1b201'),
  801. ('No RSN Capabilities field (PMF disabled in practice)',
  802. '30120100000fac040100000fac040100000fac04' + '3603a1b201') ]
  803. for txt,ie in tests:
  804. dev[0].request("DISCONNECT")
  805. dev[0].wait_disconnected()
  806. logger.info(txt)
  807. hapd.disable()
  808. hapd.set('own_ie_override', ie)
  809. hapd.enable()
  810. dev[0].request("BSS_FLUSH 0")
  811. dev[0].scan_for_bss(bssid, 2412, force_scan=True, only_new=True)
  812. dev[0].select_network(id, freq=2412)
  813. dev[0].wait_connected()
  814. dev[0].request("DISCONNECT")
  815. dev[0].wait_disconnected()
  816. logger.info('Invalid RSNE causing internal hostapd error')
  817. hapd.disable()
  818. hapd.set('own_ie_override', '30130100000fac040100000fac040100000fac048c' + '3603a1b201')
  819. hapd.enable()
  820. dev[0].request("BSS_FLUSH 0")
  821. dev[0].scan_for_bss(bssid, 2412, force_scan=True, only_new=True)
  822. dev[0].select_network(id, freq=2412)
  823. # hostapd fails to generate EAPOL-Key msg 3/4, so this connection cannot
  824. # complete.
  825. ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED"], timeout=1)
  826. if ev is not None:
  827. raise Exception("Unexpected connection")
  828. dev[0].request("DISCONNECT")
  829. logger.info('Unexpected PMKID causing internal hostapd error')
  830. hapd.disable()
  831. hapd.set('own_ie_override', '30260100000fac040100000fac040100000fac048c000100ffffffffffffffffffffffffffffffff' + '3603a1b201')
  832. hapd.enable()
  833. dev[0].request("BSS_FLUSH 0")
  834. dev[0].scan_for_bss(bssid, 2412, force_scan=True, only_new=True)
  835. dev[0].select_network(id, freq=2412)
  836. # hostapd fails to generate EAPOL-Key msg 3/4, so this connection cannot
  837. # complete.
  838. ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED"], timeout=1)
  839. if ev is not None:
  840. raise Exception("Unexpected connection")
  841. dev[0].request("DISCONNECT")