test_gas.py 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602
  1. # GAS tests
  2. # Copyright (c) 2013, Qualcomm Atheros, Inc.
  3. # Copyright (c) 2013-2014, Jouni Malinen <j@w1.fi>
  4. #
  5. # This software may be distributed under the terms of the BSD license.
  6. # See README for more details.
  7. import time
  8. import binascii
  9. import logging
  10. logger = logging.getLogger()
  11. import re
  12. import struct
  13. import hostapd
  14. def hs20_ap_params():
  15. params = hostapd.wpa2_params(ssid="test-gas")
  16. params['wpa_key_mgmt'] = "WPA-EAP"
  17. params['ieee80211w'] = "1"
  18. params['ieee8021x'] = "1"
  19. params['auth_server_addr'] = "127.0.0.1"
  20. params['auth_server_port'] = "1812"
  21. params['auth_server_shared_secret'] = "radius"
  22. params['interworking'] = "1"
  23. params['access_network_type'] = "14"
  24. params['internet'] = "1"
  25. params['asra'] = "0"
  26. params['esr'] = "0"
  27. params['uesa'] = "0"
  28. params['venue_group'] = "7"
  29. params['venue_type'] = "1"
  30. params['venue_name'] = [ "eng:Example venue", "fin:Esimerkkipaikka" ]
  31. params['roaming_consortium'] = [ "112233", "1020304050", "010203040506",
  32. "fedcba" ]
  33. params['domain_name'] = "example.com,another.example.com"
  34. params['nai_realm'] = [ "0,example.com,13[5:6],21[2:4][5:7]",
  35. "0,another.example.com" ]
  36. params['anqp_3gpp_cell_net'] = "244,91"
  37. params['network_auth_type'] = "02http://www.example.com/redirect/me/here/"
  38. params['ipaddr_type_availability'] = "14"
  39. params['hs20'] = "1"
  40. params['hs20_oper_friendly_name'] = [ "eng:Example operator", "fin:Esimerkkioperaattori" ]
  41. params['hs20_wan_metrics'] = "01:8000:1000:80:240:3000"
  42. params['hs20_conn_capab'] = [ "1:0:2", "6:22:1", "17:5060:0" ]
  43. params['hs20_operating_class'] = "5173"
  44. return params
  45. def start_ap(ap):
  46. params = hs20_ap_params()
  47. params['hessid'] = ap['bssid']
  48. hostapd.add_ap(ap['ifname'], params)
  49. return hostapd.Hostapd(ap['ifname'])
  50. def get_gas_response(dev, bssid, info, allow_fetch_failure=False):
  51. exp = r'<.>(GAS-RESPONSE-INFO) addr=([0-9a-f:]*) dialog_token=([0-9]*) status_code=([0-9]*) resp_len=([\-0-9]*)'
  52. res = re.split(exp, info)
  53. if len(res) < 6:
  54. raise Exception("Could not parse GAS-RESPONSE-INFO")
  55. if res[2] != bssid:
  56. raise Exception("Unexpected BSSID in response")
  57. token = res[3]
  58. status = res[4]
  59. if status != "0":
  60. raise Exception("GAS query failed")
  61. resp_len = res[5]
  62. if resp_len == "-1":
  63. raise Exception("GAS query reported invalid response length")
  64. if int(resp_len) > 2000:
  65. raise Exception("Unexpected long GAS response")
  66. resp = dev.request("GAS_RESPONSE_GET " + bssid + " " + token)
  67. if "FAIL" in resp:
  68. if allow_fetch_failure:
  69. logger.debug("GAS response was not available anymore")
  70. return
  71. raise Exception("Could not fetch GAS response")
  72. if len(resp) != int(resp_len) * 2:
  73. raise Exception("Unexpected GAS response length")
  74. logger.debug("GAS response: " + resp)
  75. def test_gas_generic(dev, apdev):
  76. """Generic GAS query"""
  77. bssid = apdev[0]['bssid']
  78. params = hs20_ap_params()
  79. params['hessid'] = bssid
  80. hostapd.add_ap(apdev[0]['ifname'], params)
  81. dev[0].scan(freq="2412")
  82. req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000101")
  83. if "FAIL" in req:
  84. raise Exception("GAS query request rejected")
  85. ev = dev[0].wait_event(["GAS-RESPONSE-INFO"], timeout=10)
  86. if ev is None:
  87. raise Exception("GAS query timed out")
  88. get_gas_response(dev[0], bssid, ev)
  89. def test_gas_concurrent_scan(dev, apdev):
  90. """Generic GAS queries with concurrent scan operation"""
  91. bssid = apdev[0]['bssid']
  92. params = hs20_ap_params()
  93. params['hessid'] = bssid
  94. hostapd.add_ap(apdev[0]['ifname'], params)
  95. # get BSS entry available to allow GAS query
  96. dev[0].scan(freq="2412")
  97. logger.info("Request concurrent operations")
  98. req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000101")
  99. if "FAIL" in req:
  100. raise Exception("GAS query request rejected")
  101. req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000801")
  102. if "FAIL" in req:
  103. raise Exception("GAS query request rejected")
  104. dev[0].scan(no_wait=True)
  105. req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000201")
  106. if "FAIL" in req:
  107. raise Exception("GAS query request rejected")
  108. req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000501")
  109. if "FAIL" in req:
  110. raise Exception("GAS query request rejected")
  111. responses = 0
  112. for i in range(0, 5):
  113. ev = dev[0].wait_event(["GAS-RESPONSE-INFO", "CTRL-EVENT-SCAN-RESULTS"],
  114. timeout=10)
  115. if ev is None:
  116. raise Exception("Operation timed out")
  117. if "GAS-RESPONSE-INFO" in ev:
  118. responses = responses + 1
  119. get_gas_response(dev[0], bssid, ev, allow_fetch_failure=True)
  120. if responses != 4:
  121. raise Exception("Unexpected number of GAS responses")
  122. def test_gas_concurrent_connect(dev, apdev):
  123. """Generic GAS queries with concurrent connection operation"""
  124. bssid = apdev[0]['bssid']
  125. params = hs20_ap_params()
  126. params['hessid'] = bssid
  127. hostapd.add_ap(apdev[0]['ifname'], params)
  128. dev[0].scan(freq="2412")
  129. logger.debug("Start concurrent connect and GAS request")
  130. dev[0].connect("test-gas", key_mgmt="WPA-EAP", eap="TTLS",
  131. identity="DOMAIN\mschapv2 user", anonymous_identity="ttls",
  132. password="password", phase2="auth=MSCHAPV2",
  133. ca_cert="auth_serv/ca.pem", wait_connect=False,
  134. scan_freq="2412")
  135. req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000101")
  136. if "FAIL" in req:
  137. raise Exception("GAS query request rejected")
  138. ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED", "GAS-RESPONSE-INFO"],
  139. timeout=20)
  140. if ev is None:
  141. raise Exception("Operation timed out")
  142. if "CTRL-EVENT-CONNECTED" not in ev:
  143. raise Exception("Unexpected operation order")
  144. ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED", "GAS-RESPONSE-INFO"],
  145. timeout=20)
  146. if ev is None:
  147. raise Exception("Operation timed out")
  148. if "GAS-RESPONSE-INFO" not in ev:
  149. raise Exception("Unexpected operation order")
  150. get_gas_response(dev[0], bssid, ev)
  151. dev[0].request("DISCONNECT")
  152. ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=5)
  153. if ev is None:
  154. raise Exception("Disconnection timed out")
  155. logger.debug("Wait six seconds for expiration of connect-without-scan")
  156. time.sleep(6)
  157. dev[0].dump_monitor()
  158. logger.debug("Start concurrent GAS request and connect")
  159. req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000101")
  160. if "FAIL" in req:
  161. raise Exception("GAS query request rejected")
  162. dev[0].request("RECONNECT")
  163. ev = dev[0].wait_event(["GAS-RESPONSE-INFO"], timeout=10)
  164. if ev is None:
  165. raise Exception("Operation timed out")
  166. get_gas_response(dev[0], bssid, ev)
  167. ev = dev[0].wait_event(["CTRL-EVENT-SCAN-RESULTS"], timeout=20)
  168. if ev is None:
  169. raise Exception("No new scan results reported")
  170. ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED"], timeout=20)
  171. if ev is None:
  172. raise Exception("Operation timed out")
  173. if "CTRL-EVENT-CONNECTED" not in ev:
  174. raise Exception("Unexpected operation order")
  175. def test_gas_fragment(dev, apdev):
  176. """GAS fragmentation"""
  177. hapd = start_ap(apdev[0])
  178. hapd.set("gas_frag_limit", "50")
  179. dev[0].scan(freq="2412")
  180. dev[0].request("FETCH_ANQP")
  181. for i in range(0, 13):
  182. ev = dev[0].wait_event(["RX-ANQP", "RX-HS20-ANQP"], timeout=5)
  183. if ev is None:
  184. raise Exception("Operation timed out")
  185. def test_gas_comeback_delay(dev, apdev):
  186. """GAS fragmentation"""
  187. hapd = start_ap(apdev[0])
  188. hapd.set("gas_comeback_delay", "500")
  189. dev[0].scan(freq="2412")
  190. dev[0].request("FETCH_ANQP")
  191. for i in range(0, 6):
  192. ev = dev[0].wait_event(["RX-ANQP"], timeout=5)
  193. if ev is None:
  194. raise Exception("Operation timed out")
  195. def expect_gas_result(dev, result, status=None):
  196. ev = dev.wait_event(["GAS-QUERY-DONE"], timeout=10)
  197. if ev is None:
  198. raise Exception("GAS query timed out")
  199. if "result=" + result not in ev:
  200. raise Exception("Unexpected GAS query result")
  201. if status and "status_code=" + str(status) + ' ' not in ev:
  202. raise Exception("Unexpected GAS status code")
  203. def anqp_get(dev, bssid, id):
  204. dev.request("ANQP_GET " + bssid + " " + str(id))
  205. ev = dev.wait_event(["GAS-QUERY-START"], timeout=5)
  206. if ev is None:
  207. raise Exception("GAS query start timed out")
  208. def test_gas_timeout(dev, apdev):
  209. """GAS timeout"""
  210. hapd = start_ap(apdev[0])
  211. bssid = apdev[0]['bssid']
  212. dev[0].scan(freq="2412")
  213. hapd.set("ext_mgmt_frame_handling", "1")
  214. anqp_get(dev[0], bssid, 263)
  215. ev = hapd.wait_event(["MGMT-RX"], timeout=5)
  216. if ev is None:
  217. raise Exception("MGMT RX wait timed out")
  218. expect_gas_result(dev[0], "TIMEOUT")
  219. MGMT_SUBTYPE_ACTION = 13
  220. ACTION_CATEG_PUBLIC = 4
  221. GAS_INITIAL_REQUEST = 10
  222. GAS_INITIAL_RESPONSE = 11
  223. GAS_COMEBACK_REQUEST = 12
  224. GAS_COMEBACK_RESPONSE = 13
  225. GAS_ACTIONS = [ GAS_INITIAL_REQUEST, GAS_INITIAL_RESPONSE,
  226. GAS_COMEBACK_REQUEST, GAS_COMEBACK_RESPONSE ]
  227. def anqp_adv_proto():
  228. return struct.pack('BBBB', 108, 2, 127, 0)
  229. def anqp_initial_resp(dialog_token, status_code, comeback_delay=0):
  230. return struct.pack('<BBBHH', ACTION_CATEG_PUBLIC, GAS_INITIAL_RESPONSE,
  231. dialog_token, status_code, comeback_delay) + anqp_adv_proto()
  232. def anqp_comeback_resp(dialog_token, status_code=0, id=0, more=False, comeback_delay=0, bogus_adv_proto=False):
  233. if more:
  234. id |= 0x80
  235. if bogus_adv_proto:
  236. adv = struct.pack('BBBB', 108, 2, 127, 1)
  237. else:
  238. adv = anqp_adv_proto()
  239. return struct.pack('<BBBHBH', ACTION_CATEG_PUBLIC, GAS_COMEBACK_RESPONSE,
  240. dialog_token, status_code, id, comeback_delay) + adv
  241. def gas_rx(hapd):
  242. count = 0
  243. while count < 30:
  244. count = count + 1
  245. query = hapd.mgmt_rx()
  246. if query is None:
  247. raise Exception("Action frame not received")
  248. if query['subtype'] != MGMT_SUBTYPE_ACTION:
  249. continue
  250. payload = query['payload']
  251. if len(payload) < 2:
  252. continue
  253. (category, action) = struct.unpack('BB', payload[0:2])
  254. if category != ACTION_CATEG_PUBLIC or action not in GAS_ACTIONS:
  255. continue
  256. return query
  257. raise Exception("No Action frame received")
  258. def parse_gas(payload):
  259. pos = payload
  260. (category, action, dialog_token) = struct.unpack('BBB', pos[0:3])
  261. if category != ACTION_CATEG_PUBLIC:
  262. return None
  263. if action not in GAS_ACTIONS:
  264. return None
  265. gas = {}
  266. gas['action'] = action
  267. pos = pos[3:]
  268. if len(pos) < 1 and action != GAS_COMEBACK_REQUEST:
  269. return None
  270. gas['dialog_token'] = dialog_token
  271. return gas
  272. def action_response(req):
  273. resp = {}
  274. resp['fc'] = req['fc']
  275. resp['da'] = req['sa']
  276. resp['sa'] = req['da']
  277. resp['bssid'] = req['bssid']
  278. return resp
  279. def send_gas_resp(hapd, resp):
  280. hapd.mgmt_tx(resp)
  281. ev = hapd.wait_event(["MGMT-TX-STATUS"], timeout=5)
  282. if ev is None:
  283. raise Exception("Missing TX status for GAS response")
  284. if "ok=1" not in ev:
  285. raise Exception("GAS response not acknowledged")
  286. def test_gas_invalid_response_type(dev, apdev):
  287. """GAS invalid response type"""
  288. hapd = start_ap(apdev[0])
  289. bssid = apdev[0]['bssid']
  290. dev[0].scan(freq="2412")
  291. hapd.set("ext_mgmt_frame_handling", "1")
  292. anqp_get(dev[0], bssid, 263)
  293. query = gas_rx(hapd)
  294. gas = parse_gas(query['payload'])
  295. resp = action_response(query)
  296. # GAS Comeback Response instead of GAS Initial Response
  297. resp['payload'] = anqp_comeback_resp(gas['dialog_token']) + struct.pack('<H', 0)
  298. send_gas_resp(hapd, resp)
  299. # station drops the invalid frame, so this needs to result in GAS timeout
  300. expect_gas_result(dev[0], "TIMEOUT")
  301. def test_gas_failure_status_code(dev, apdev):
  302. """GAS failure status code"""
  303. hapd = start_ap(apdev[0])
  304. bssid = apdev[0]['bssid']
  305. dev[0].scan(freq="2412")
  306. hapd.set("ext_mgmt_frame_handling", "1")
  307. anqp_get(dev[0], bssid, 263)
  308. query = gas_rx(hapd)
  309. gas = parse_gas(query['payload'])
  310. resp = action_response(query)
  311. resp['payload'] = anqp_initial_resp(gas['dialog_token'], 61) + struct.pack('<H', 0)
  312. send_gas_resp(hapd, resp)
  313. expect_gas_result(dev[0], "FAILURE")
  314. def test_gas_malformed(dev, apdev):
  315. """GAS malformed response frames"""
  316. hapd = start_ap(apdev[0])
  317. bssid = apdev[0]['bssid']
  318. dev[0].scan(freq="2412")
  319. hapd.set("ext_mgmt_frame_handling", "1")
  320. anqp_get(dev[0], bssid, 263)
  321. query = gas_rx(hapd)
  322. gas = parse_gas(query['payload'])
  323. resp = action_response(query)
  324. resp['payload'] = struct.pack('<BBBH', ACTION_CATEG_PUBLIC,
  325. GAS_COMEBACK_RESPONSE,
  326. gas['dialog_token'], 0)
  327. hapd.mgmt_tx(resp)
  328. resp['payload'] = struct.pack('<BBBHB', ACTION_CATEG_PUBLIC,
  329. GAS_COMEBACK_RESPONSE,
  330. gas['dialog_token'], 0, 0)
  331. hapd.mgmt_tx(resp)
  332. hdr = struct.pack('<BBBHH', ACTION_CATEG_PUBLIC, GAS_INITIAL_RESPONSE,
  333. gas['dialog_token'], 0, 0)
  334. resp['payload'] = hdr + struct.pack('B', 108)
  335. hapd.mgmt_tx(resp)
  336. resp['payload'] = hdr + struct.pack('BB', 108, 0)
  337. hapd.mgmt_tx(resp)
  338. resp['payload'] = hdr + struct.pack('BB', 108, 1)
  339. hapd.mgmt_tx(resp)
  340. resp['payload'] = hdr + struct.pack('BB', 108, 255)
  341. hapd.mgmt_tx(resp)
  342. resp['payload'] = hdr + struct.pack('BBB', 108, 1, 127)
  343. hapd.mgmt_tx(resp)
  344. resp['payload'] = hdr + struct.pack('BBB', 108, 2, 127)
  345. hapd.mgmt_tx(resp)
  346. resp['payload'] = hdr + struct.pack('BBBB', 0, 2, 127, 0)
  347. hapd.mgmt_tx(resp)
  348. resp['payload'] = anqp_initial_resp(gas['dialog_token'], 0) + struct.pack('<H', 1)
  349. hapd.mgmt_tx(resp)
  350. resp['payload'] = anqp_initial_resp(gas['dialog_token'], 0) + struct.pack('<HB', 2, 0)
  351. hapd.mgmt_tx(resp)
  352. resp['payload'] = anqp_initial_resp(gas['dialog_token'], 0) + struct.pack('<H', 65535)
  353. hapd.mgmt_tx(resp)
  354. resp['payload'] = anqp_initial_resp(gas['dialog_token'], 0) + struct.pack('<HBB', 1, 0, 0)
  355. hapd.mgmt_tx(resp)
  356. # Station drops invalid frames, but the last of the responses is valid from
  357. # GAS view point even though it has an extra octet in the end and the ANQP
  358. # part of the response is not valid. This is reported as successfulyl
  359. # completed GAS exchange.
  360. expect_gas_result(dev[0], "SUCCESS")
  361. def init_gas(hapd, bssid, dev):
  362. anqp_get(dev, bssid, 263)
  363. query = gas_rx(hapd)
  364. gas = parse_gas(query['payload'])
  365. dialog_token = gas['dialog_token']
  366. resp = action_response(query)
  367. resp['payload'] = anqp_initial_resp(dialog_token, 0, comeback_delay=1) + struct.pack('<H', 0)
  368. send_gas_resp(hapd, resp)
  369. query = gas_rx(hapd)
  370. gas = parse_gas(query['payload'])
  371. if gas['action'] != GAS_COMEBACK_REQUEST:
  372. raise Exception("Unexpected request action")
  373. if gas['dialog_token'] != dialog_token:
  374. raise Exception("Unexpected dialog token change")
  375. return query, dialog_token
  376. def test_gas_malformed_comeback_resp(dev, apdev):
  377. """GAS malformed comeback response frames"""
  378. hapd = start_ap(apdev[0])
  379. bssid = apdev[0]['bssid']
  380. dev[0].scan(freq="2412")
  381. hapd.set("ext_mgmt_frame_handling", "1")
  382. logger.debug("Non-zero status code in comeback response")
  383. query, dialog_token = init_gas(hapd, bssid, dev[0])
  384. resp = action_response(query)
  385. resp['payload'] = anqp_comeback_resp(dialog_token, status_code=2) + struct.pack('<H', 0)
  386. send_gas_resp(hapd, resp)
  387. expect_gas_result(dev[0], "FAILURE", status=2)
  388. logger.debug("Different advertisement protocol in comeback response")
  389. query, dialog_token = init_gas(hapd, bssid, dev[0])
  390. resp = action_response(query)
  391. resp['payload'] = anqp_comeback_resp(dialog_token, bogus_adv_proto=True) + struct.pack('<H', 0)
  392. send_gas_resp(hapd, resp)
  393. expect_gas_result(dev[0], "PEER_ERROR")
  394. logger.debug("Non-zero frag id and comeback delay in comeback response")
  395. query, dialog_token = init_gas(hapd, bssid, dev[0])
  396. resp = action_response(query)
  397. resp['payload'] = anqp_comeback_resp(dialog_token, id=1, comeback_delay=1) + struct.pack('<H', 0)
  398. send_gas_resp(hapd, resp)
  399. expect_gas_result(dev[0], "PEER_ERROR")
  400. logger.debug("Unexpected frag id in comeback response")
  401. query, dialog_token = init_gas(hapd, bssid, dev[0])
  402. resp = action_response(query)
  403. resp['payload'] = anqp_comeback_resp(dialog_token, id=1) + struct.pack('<H', 0)
  404. send_gas_resp(hapd, resp)
  405. expect_gas_result(dev[0], "PEER_ERROR")
  406. logger.debug("Empty fragment and replay in comeback response")
  407. query, dialog_token = init_gas(hapd, bssid, dev[0])
  408. resp = action_response(query)
  409. resp['payload'] = anqp_comeback_resp(dialog_token, more=True) + struct.pack('<H', 0)
  410. send_gas_resp(hapd, resp)
  411. query = gas_rx(hapd)
  412. gas = parse_gas(query['payload'])
  413. if gas['action'] != GAS_COMEBACK_REQUEST:
  414. raise Exception("Unexpected request action")
  415. if gas['dialog_token'] != dialog_token:
  416. raise Exception("Unexpected dialog token change")
  417. resp = action_response(query)
  418. resp['payload'] = anqp_comeback_resp(dialog_token) + struct.pack('<H', 0)
  419. send_gas_resp(hapd, resp)
  420. resp['payload'] = anqp_comeback_resp(dialog_token, id=1) + struct.pack('<H', 0)
  421. send_gas_resp(hapd, resp)
  422. expect_gas_result(dev[0], "SUCCESS")
  423. logger.debug("Unexpected initial response when waiting for comeback response")
  424. query, dialog_token = init_gas(hapd, bssid, dev[0])
  425. resp = action_response(query)
  426. resp['payload'] = anqp_initial_resp(dialog_token, 0) + struct.pack('<H', 0)
  427. send_gas_resp(hapd, resp)
  428. ev = hapd.wait_event(["MGMT-RX"], timeout=1)
  429. if ev is not None:
  430. raise Exception("Unexpected management frame")
  431. expect_gas_result(dev[0], "TIMEOUT")
  432. logger.debug("Too short comeback response")
  433. query, dialog_token = init_gas(hapd, bssid, dev[0])
  434. resp = action_response(query)
  435. resp['payload'] = struct.pack('<BBBH', ACTION_CATEG_PUBLIC,
  436. GAS_COMEBACK_RESPONSE, dialog_token, 0)
  437. send_gas_resp(hapd, resp)
  438. ev = hapd.wait_event(["MGMT-RX"], timeout=1)
  439. if ev is not None:
  440. raise Exception("Unexpected management frame")
  441. expect_gas_result(dev[0], "TIMEOUT")
  442. logger.debug("Too short comeback response(2)")
  443. query, dialog_token = init_gas(hapd, bssid, dev[0])
  444. resp = action_response(query)
  445. resp['payload'] = struct.pack('<BBBHBB', ACTION_CATEG_PUBLIC,
  446. GAS_COMEBACK_RESPONSE, dialog_token, 0, 0x80,
  447. 0)
  448. send_gas_resp(hapd, resp)
  449. ev = hapd.wait_event(["MGMT-RX"], timeout=1)
  450. if ev is not None:
  451. raise Exception("Unexpected management frame")
  452. expect_gas_result(dev[0], "TIMEOUT")
  453. logger.debug("Maximum comeback response fragment claiming more fragments")
  454. query, dialog_token = init_gas(hapd, bssid, dev[0])
  455. resp = action_response(query)
  456. resp['payload'] = anqp_comeback_resp(dialog_token, more=True) + struct.pack('<H', 0)
  457. send_gas_resp(hapd, resp)
  458. for i in range(1, 129):
  459. query = gas_rx(hapd)
  460. gas = parse_gas(query['payload'])
  461. if gas['action'] != GAS_COMEBACK_REQUEST:
  462. raise Exception("Unexpected request action")
  463. if gas['dialog_token'] != dialog_token:
  464. raise Exception("Unexpected dialog token change")
  465. resp = action_response(query)
  466. resp['payload'] = anqp_comeback_resp(dialog_token, id=i, more=True) + struct.pack('<H', 0)
  467. send_gas_resp(hapd, resp)
  468. expect_gas_result(dev[0], "PEER_ERROR")
  469. def test_gas_comeback_resp_additional_delay(dev, apdev):
  470. """GAS comeback response requesting additional delay"""
  471. hapd = start_ap(apdev[0])
  472. bssid = apdev[0]['bssid']
  473. dev[0].scan(freq="2412")
  474. hapd.set("ext_mgmt_frame_handling", "1")
  475. query, dialog_token = init_gas(hapd, bssid, dev[0])
  476. for i in range(0, 2):
  477. resp = action_response(query)
  478. resp['payload'] = anqp_comeback_resp(dialog_token, status_code=95, comeback_delay=50) + struct.pack('<H', 0)
  479. send_gas_resp(hapd, resp)
  480. query = gas_rx(hapd)
  481. gas = parse_gas(query['payload'])
  482. if gas['action'] != GAS_COMEBACK_REQUEST:
  483. raise Exception("Unexpected request action")
  484. if gas['dialog_token'] != dialog_token:
  485. raise Exception("Unexpected dialog token change")
  486. resp = action_response(query)
  487. resp['payload'] = anqp_comeback_resp(dialog_token, status_code=0) + struct.pack('<H', 0)
  488. send_gas_resp(hapd, resp)
  489. expect_gas_result(dev[0], "SUCCESS")
  490. def test_gas_unknown_adv_proto(dev, apdev):
  491. """Unknown advertisement protocol id"""
  492. bssid = apdev[0]['bssid']
  493. params = hs20_ap_params()
  494. params['hessid'] = bssid
  495. hostapd.add_ap(apdev[0]['ifname'], params)
  496. dev[0].scan(freq="2412")
  497. req = dev[0].request("GAS_REQUEST " + bssid + " 42 000102000101")
  498. if "FAIL" in req:
  499. raise Exception("GAS query request rejected")
  500. expect_gas_result(dev[0], "FAILURE", "59")
  501. ev = dev[0].wait_event(["GAS-RESPONSE-INFO"], timeout=10)
  502. if ev is None:
  503. raise Exception("GAS query timed out")
  504. exp = r'<.>(GAS-RESPONSE-INFO) addr=([0-9a-f:]*) dialog_token=([0-9]*) status_code=([0-9]*) resp_len=([\-0-9]*)'
  505. res = re.split(exp, ev)
  506. if len(res) < 6:
  507. raise Exception("Could not parse GAS-RESPONSE-INFO")
  508. if res[2] != bssid:
  509. raise Exception("Unexpected BSSID in response")
  510. status = res[4]
  511. if status != "59":
  512. raise Exception("Unexpected GAS-RESPONSE-INFO status")