Handle NULL return from os_zalloc() in sta_track_add()
[mech_eap.git] / tests / hwsim / test_gas.py
1 # GAS tests
2 # Copyright (c) 2013, Qualcomm Atheros, Inc.
3 # Copyright (c) 2013-2015, 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
8 from remotehost import remote_compatible
9 import time
10 import binascii
11 import logging
12 logger = logging.getLogger()
13 import os
14 import re
15 import struct
16
17 import hostapd
18 from wpasupplicant import WpaSupplicant
19 from tshark import run_tshark
20 from utils import alloc_fail, wait_fail_trigger, skip_with_fips
21 from hwsim import HWSimRadio
22
23 def hs20_ap_params():
24     params = hostapd.wpa2_params(ssid="test-gas")
25     params['wpa_key_mgmt'] = "WPA-EAP"
26     params['ieee80211w'] = "1"
27     params['ieee8021x'] = "1"
28     params['auth_server_addr'] = "127.0.0.1"
29     params['auth_server_port'] = "1812"
30     params['auth_server_shared_secret'] = "radius"
31     params['interworking'] = "1"
32     params['access_network_type'] = "14"
33     params['internet'] = "1"
34     params['asra'] = "0"
35     params['esr'] = "0"
36     params['uesa'] = "0"
37     params['venue_group'] = "7"
38     params['venue_type'] = "1"
39     params['venue_name'] = [ "eng:Example venue", "fin:Esimerkkipaikka" ]
40     params['roaming_consortium'] = [ "112233", "1020304050", "010203040506",
41                                      "fedcba" ]
42     params['domain_name'] = "example.com,another.example.com"
43     params['nai_realm'] = [ "0,example.com,13[5:6],21[2:4][5:7]",
44                             "0,another.example.com" ]
45     params['anqp_3gpp_cell_net'] = "244,91"
46     params['network_auth_type'] = "02http://www.example.com/redirect/me/here/"
47     params['ipaddr_type_availability'] = "14"
48     params['hs20'] = "1"
49     params['hs20_oper_friendly_name'] = [ "eng:Example operator", "fin:Esimerkkioperaattori" ]
50     params['hs20_wan_metrics'] = "01:8000:1000:80:240:3000"
51     params['hs20_conn_capab'] = [ "1:0:2", "6:22:1", "17:5060:0" ]
52     params['hs20_operating_class'] = "5173"
53     return params
54
55 def start_ap(ap):
56     params = hs20_ap_params()
57     params['hessid'] = ap['bssid']
58     return hostapd.add_ap(ap, params)
59
60 def get_gas_response(dev, bssid, info, allow_fetch_failure=False,
61                      extra_test=False):
62     exp = r'<.>(GAS-RESPONSE-INFO) addr=([0-9a-f:]*) dialog_token=([0-9]*) status_code=([0-9]*) resp_len=([\-0-9]*)'
63     res = re.split(exp, info)
64     if len(res) < 6:
65         raise Exception("Could not parse GAS-RESPONSE-INFO")
66     if res[2] != bssid:
67         raise Exception("Unexpected BSSID in response")
68     token = res[3]
69     status = res[4]
70     if status != "0":
71         raise Exception("GAS query failed")
72     resp_len = res[5]
73     if resp_len == "-1":
74         raise Exception("GAS query reported invalid response length")
75     if int(resp_len) > 2000:
76         raise Exception("Unexpected long GAS response")
77
78     if extra_test:
79         if "FAIL" not in dev.request("GAS_RESPONSE_GET " + bssid + " 123456"):
80             raise Exception("Invalid dialog token accepted")
81         if "FAIL-Invalid range" not in dev.request("GAS_RESPONSE_GET " + bssid + " " + token + " 10000,10001"):
82             raise Exception("Invalid range accepted")
83         if "FAIL-Invalid range" not in dev.request("GAS_RESPONSE_GET " + bssid + " " + token + " 0,10000"):
84             raise Exception("Invalid range accepted")
85         if "FAIL" not in dev.request("GAS_RESPONSE_GET " + bssid + " " + token + " 0"):
86             raise Exception("Invalid GAS_RESPONSE_GET accepted")
87
88         res1_2 = dev.request("GAS_RESPONSE_GET " + bssid + " " + token + " 1,2")
89         res5_3 = dev.request("GAS_RESPONSE_GET " + bssid + " " + token + " 5,3")
90
91     resp = dev.request("GAS_RESPONSE_GET " + bssid + " " + token)
92     if "FAIL" in resp:
93         if allow_fetch_failure:
94             logger.debug("GAS response was not available anymore")
95             return
96         raise Exception("Could not fetch GAS response")
97     if len(resp) != int(resp_len) * 2:
98         raise Exception("Unexpected GAS response length")
99     logger.debug("GAS response: " + resp)
100     if extra_test:
101         if resp[2:6] != res1_2:
102             raise Exception("Unexpected response substring res1_2: " + res1_2)
103         if resp[10:16] != res5_3:
104             raise Exception("Unexpected response substring res5_3: " + res5_3)
105
106 def test_gas_generic(dev, apdev):
107     """Generic GAS query"""
108     bssid = apdev[0]['bssid']
109     params = hs20_ap_params()
110     params['hessid'] = bssid
111     hostapd.add_ap(apdev[0], params)
112
113     cmds = [ "foo",
114              "00:11:22:33:44:55",
115              "00:11:22:33:44:55 ",
116              "00:11:22:33:44:55  ",
117              "00:11:22:33:44:55 1",
118              "00:11:22:33:44:55 1 1234",
119              "00:11:22:33:44:55 qq",
120              "00:11:22:33:44:55 qq 1234",
121              "00:11:22:33:44:55 00      1",
122              "00:11:22:33:44:55 00 123",
123              "00:11:22:33:44:55 00 ",
124              "00:11:22:33:44:55 00 qq" ]
125     for cmd in cmds:
126         if "FAIL" not in dev[0].request("GAS_REQUEST " + cmd):
127             raise Exception("Invalid GAS_REQUEST accepted: " + cmd)
128
129     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
130     req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000101")
131     if "FAIL" in req:
132         raise Exception("GAS query request rejected")
133     ev = dev[0].wait_event(["GAS-RESPONSE-INFO"], timeout=10)
134     if ev is None:
135         raise Exception("GAS query timed out")
136     get_gas_response(dev[0], bssid, ev, extra_test=True)
137
138     if "FAIL" not in dev[0].request("GAS_RESPONSE_GET ff"):
139         raise Exception("Invalid GAS_RESPONSE_GET accepted")
140
141 def test_gas_concurrent_scan(dev, apdev):
142     """Generic GAS queries with concurrent scan operation"""
143     bssid = apdev[0]['bssid']
144     params = hs20_ap_params()
145     params['hessid'] = bssid
146     hostapd.add_ap(apdev[0], params)
147
148     # get BSS entry available to allow GAS query
149     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
150
151     logger.info("Request concurrent operations")
152     req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000101")
153     if "FAIL" in req:
154         raise Exception("GAS query request rejected")
155     req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000801")
156     if "FAIL" in req:
157         raise Exception("GAS query request rejected")
158     dev[0].scan(no_wait=True)
159     req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000201")
160     if "FAIL" in req:
161         raise Exception("GAS query request rejected")
162     req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000501")
163     if "FAIL" in req:
164         raise Exception("GAS query request rejected")
165
166     responses = 0
167     for i in range(0, 5):
168         ev = dev[0].wait_event(["GAS-RESPONSE-INFO", "CTRL-EVENT-SCAN-RESULTS"],
169                                timeout=10)
170         if ev is None:
171             raise Exception("Operation timed out")
172         if "GAS-RESPONSE-INFO" in ev:
173             responses = responses + 1
174             get_gas_response(dev[0], bssid, ev, allow_fetch_failure=True)
175
176     if responses != 4:
177         raise Exception("Unexpected number of GAS responses")
178
179 def test_gas_concurrent_connect(dev, apdev):
180     """Generic GAS queries with concurrent connection operation"""
181     skip_with_fips(dev[0])
182     bssid = apdev[0]['bssid']
183     params = hs20_ap_params()
184     params['hessid'] = bssid
185     hostapd.add_ap(apdev[0], params)
186
187     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
188
189     logger.debug("Start concurrent connect and GAS request")
190     dev[0].connect("test-gas", key_mgmt="WPA-EAP", eap="TTLS",
191                    identity="DOMAIN\mschapv2 user", anonymous_identity="ttls",
192                    password="password", phase2="auth=MSCHAPV2",
193                    ca_cert="auth_serv/ca.pem", wait_connect=False,
194                    scan_freq="2412")
195     req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000101")
196     if "FAIL" in req:
197         raise Exception("GAS query request rejected")
198
199     ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED", "GAS-RESPONSE-INFO"],
200                            timeout=20)
201     if ev is None:
202         raise Exception("Operation timed out")
203     if "CTRL-EVENT-CONNECTED" not in ev:
204         raise Exception("Unexpected operation order")
205
206     ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED", "GAS-RESPONSE-INFO"],
207                            timeout=20)
208     if ev is None:
209         raise Exception("Operation timed out")
210     if "GAS-RESPONSE-INFO" not in ev:
211         raise Exception("Unexpected operation order")
212     get_gas_response(dev[0], bssid, ev)
213
214     dev[0].request("DISCONNECT")
215     dev[0].wait_disconnected(timeout=5)
216
217     logger.debug("Wait six seconds for expiration of connect-without-scan")
218     time.sleep(6)
219     dev[0].dump_monitor()
220
221     logger.debug("Start concurrent GAS request and connect")
222     req = dev[0].request("GAS_REQUEST " + bssid + " 00 000102000101")
223     if "FAIL" in req:
224         raise Exception("GAS query request rejected")
225     dev[0].request("RECONNECT")
226
227     ev = dev[0].wait_event(["GAS-RESPONSE-INFO"], timeout=10)
228     if ev is None:
229         raise Exception("Operation timed out")
230     get_gas_response(dev[0], bssid, ev)
231
232     ev = dev[0].wait_event(["CTRL-EVENT-SCAN-RESULTS"], timeout=20)
233     if ev is None:
234         raise Exception("No new scan results reported")
235
236     ev = dev[0].wait_connected(timeout=20, error="Operation tiemd out")
237     if "CTRL-EVENT-CONNECTED" not in ev:
238         raise Exception("Unexpected operation order")
239
240 def gas_fragment_and_comeback(dev, apdev, frag_limit=0, comeback_delay=0):
241     hapd = start_ap(apdev)
242     if frag_limit:
243         hapd.set("gas_frag_limit", str(frag_limit))
244     if comeback_delay:
245         hapd.set("gas_comeback_delay", str(comeback_delay))
246
247     dev.scan_for_bss(apdev['bssid'], freq="2412", force_scan=True)
248     dev.request("FETCH_ANQP")
249     ev = dev.wait_event(["GAS-QUERY-DONE"], timeout=5)
250     if ev is None:
251         raise Exception("No GAS-QUERY-DONE event")
252     if "result=SUCCESS" not in ev:
253         raise Exception("Unexpected GAS result: " + ev)
254     for i in range(0, 13):
255         ev = dev.wait_event(["RX-ANQP", "RX-HS20-ANQP"], timeout=5)
256         if ev is None:
257             raise Exception("Operation timed out")
258     ev = dev.wait_event(["ANQP-QUERY-DONE"], timeout=1)
259     if ev is None:
260         raise Exception("No ANQP-QUERY-DONE event")
261     if "result=SUCCESS" not in ev:
262         raise Exception("Unexpected ANQP result: " + ev)
263
264 def test_gas_fragment(dev, apdev):
265     """GAS fragmentation"""
266     gas_fragment_and_comeback(dev[0], apdev[0], frag_limit=50)
267
268 def test_gas_fragment_mcc(dev, apdev):
269     """GAS fragmentation with mac80211_hwsim MCC enabled"""
270     with HWSimRadio(n_channels=2) as (radio, iface):
271         wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
272         wpas.interface_add(iface)
273         gas_fragment_and_comeback(wpas, apdev[0], frag_limit=50)
274
275 def test_gas_fragment_with_comeback_delay(dev, apdev):
276     """GAS fragmentation and comeback delay"""
277     gas_fragment_and_comeback(dev[0], apdev[0], frag_limit=50,
278                               comeback_delay=500)
279
280 def test_gas_fragment_with_comeback_delay_mcc(dev, apdev):
281     """GAS fragmentation and comeback delay with mac80211_hwsim MCC enabled"""
282     with HWSimRadio(n_channels=2) as (radio, iface):
283         wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
284         wpas.interface_add(iface)
285         gas_fragment_and_comeback(wpas, apdev[0], frag_limit=50,
286                                   comeback_delay=500)
287
288 def test_gas_comeback_delay(dev, apdev):
289     """GAS comeback delay"""
290     hapd = start_ap(apdev[0])
291     hapd.set("gas_comeback_delay", "500")
292
293     dev[0].scan_for_bss(apdev[0]['bssid'], freq="2412", force_scan=True)
294     dev[0].request("FETCH_ANQP")
295     if "FAIL-BUSY" not in dev[0].request("SCAN"):
296         raise Exception("SCAN accepted during FETCH_ANQP")
297     for i in range(0, 6):
298         ev = dev[0].wait_event(["RX-ANQP"], timeout=5)
299         if ev is None:
300             raise Exception("Operation timed out")
301
302 @remote_compatible
303 def test_gas_stop_fetch_anqp(dev, apdev):
304     """Stop FETCH_ANQP operation"""
305     hapd = start_ap(apdev[0])
306
307     dev[0].scan_for_bss(apdev[0]['bssid'], freq="2412", force_scan=True)
308     hapd.set("ext_mgmt_frame_handling", "1")
309     dev[0].request("FETCH_ANQP")
310     dev[0].request("STOP_FETCH_ANQP")
311     hapd.set("ext_mgmt_frame_handling", "0")
312     ev = dev[0].wait_event(["RX-ANQP", "GAS-QUERY-DONE"], timeout=10)
313     if ev is None:
314         raise Exception("GAS-QUERY-DONE timed out")
315     if "RX-ANQP" in ev:
316         raise Exception("Unexpected ANQP response received")
317
318 def test_gas_anqp_get(dev, apdev):
319     """GAS/ANQP query for both IEEE 802.11 and Hotspot 2.0 elements"""
320     hapd = start_ap(apdev[0])
321     bssid = apdev[0]['bssid']
322
323     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
324     if "OK" not in dev[0].request("ANQP_GET " + bssid + " 258,268,hs20:3,hs20:4"):
325         raise Exception("ANQP_GET command failed")
326
327     ev = dev[0].wait_event(["GAS-QUERY-START"], timeout=5)
328     if ev is None:
329         raise Exception("GAS query start timed out")
330
331     ev = dev[0].wait_event(["GAS-QUERY-DONE"], timeout=10)
332     if ev is None:
333         raise Exception("GAS query timed out")
334
335     ev = dev[0].wait_event(["RX-ANQP"], timeout=1)
336     if ev is None or "Venue Name" not in ev:
337         raise Exception("Did not receive Venue Name")
338
339     ev = dev[0].wait_event(["RX-ANQP"], timeout=1)
340     if ev is None or "Domain Name list" not in ev:
341         raise Exception("Did not receive Domain Name list")
342
343     ev = dev[0].wait_event(["RX-HS20-ANQP"], timeout=1)
344     if ev is None or "Operator Friendly Name" not in ev:
345         raise Exception("Did not receive Operator Friendly Name")
346
347     ev = dev[0].wait_event(["RX-HS20-ANQP"], timeout=1)
348     if ev is None or "WAN Metrics" not in ev:
349         raise Exception("Did not receive WAN Metrics")
350
351     ev = dev[0].wait_event(["ANQP-QUERY-DONE"], timeout=10)
352     if ev is None:
353         raise Exception("ANQP-QUERY-DONE event not seen")
354     if "result=SUCCESS" not in ev:
355         raise Exception("Unexpected result: " + ev)
356
357     if "OK" not in dev[0].request("HS20_ANQP_GET " + bssid + " 3,4"):
358         raise Exception("ANQP_GET command failed")
359
360     ev = dev[0].wait_event(["RX-HS20-ANQP"], timeout=1)
361     if ev is None or "Operator Friendly Name" not in ev:
362         raise Exception("Did not receive Operator Friendly Name")
363
364     ev = dev[0].wait_event(["RX-HS20-ANQP"], timeout=1)
365     if ev is None or "WAN Metrics" not in ev:
366         raise Exception("Did not receive WAN Metrics")
367
368     cmds = [ "",
369              "foo",
370              "00:11:22:33:44:55 258,hs20:-1",
371              "00:11:22:33:44:55 258,hs20:0",
372              "00:11:22:33:44:55 258,hs20:32",
373              "00:11:22:33:44:55 hs20:-1",
374              "00:11:22:33:44:55 hs20:0",
375              "00:11:22:33:44:55 hs20:32",
376              "00:11:22:33:44:55",
377              "00:11:22:33:44:55 ",
378              "00:11:22:33:44:55 0",
379              "00:11:22:33:44:55 1" ]
380     for cmd in cmds:
381         if "FAIL" not in dev[0].request("ANQP_GET " + cmd):
382             raise Exception("Invalid ANQP_GET accepted")
383
384     cmds = [ "",
385              "foo",
386              "00:11:22:33:44:55 -1",
387              "00:11:22:33:44:55 0",
388              "00:11:22:33:44:55 32",
389              "00:11:22:33:44:55",
390              "00:11:22:33:44:55 ",
391              "00:11:22:33:44:55 0",
392              "00:11:22:33:44:55 1" ]
393     for cmd in cmds:
394         if "FAIL" not in dev[0].request("HS20_ANQP_GET " + cmd):
395             raise Exception("Invalid HS20_ANQP_GET accepted")
396
397 def test_gas_anqp_get_oom(dev, apdev):
398     """GAS/ANQP query OOM"""
399     hapd = start_ap(apdev[0])
400     bssid = apdev[0]['bssid']
401
402     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
403     with alloc_fail(dev[0], 1, "wpabuf_alloc;anqp_send_req"):
404         if "FAIL" not in dev[0].request("ANQP_GET " + bssid + " 258,268,hs20:3,hs20:4"):
405             raise Exception("ANQP_GET command accepted during OOM")
406     with alloc_fail(dev[0], 1, "hs20_build_anqp_req;hs20_anqp_send_req"):
407         if "FAIL" not in dev[0].request("HS20_ANQP_GET " + bssid + " 1"):
408             raise Exception("HS20_ANQP_GET command accepted during OOM")
409     with alloc_fail(dev[0], 1, "gas_query_req;hs20_anqp_send_req"):
410         if "FAIL" not in dev[0].request("HS20_ANQP_GET " + bssid + " 1"):
411             raise Exception("HS20_ANQP_GET command accepted during OOM")
412     with alloc_fail(dev[0], 1, "=hs20_anqp_send_req"):
413         if "FAIL" not in dev[0].request("REQ_HS20_ICON " + bssid + " w1fi_logo"):
414             raise Exception("REQ_HS20_ICON command accepted during OOM")
415     with alloc_fail(dev[0], 2, "=hs20_anqp_send_req"):
416         if "FAIL" not in dev[0].request("REQ_HS20_ICON " + bssid + " w1fi_logo"):
417             raise Exception("REQ_HS20_ICON command accepted during OOM")
418
419 def test_gas_anqp_icon_binary_proto(dev, apdev):
420     """GAS/ANQP and icon binary protocol testing"""
421     hapd = start_ap(apdev[0])
422     bssid = apdev[0]['bssid']
423
424     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
425     hapd.set("ext_mgmt_frame_handling", "1")
426
427     tests = [ '010000', '01000000', '00000000', '00030000', '00020000',
428               '00000100', '0001ff0100ee', '0001ff0200ee' ]
429     for test in tests:
430         dev[0].request("HS20_ICON_REQUEST " + bssid + " w1fi_logo")
431         query = gas_rx(hapd)
432         gas = parse_gas(query['payload'])
433         resp = action_response(query)
434         data = binascii.unhexlify(test)
435         data = binascii.unhexlify('506f9a110b00') + data
436         data = struct.pack('<HHH', len(data) + 4, 0xdddd, len(data)) + data
437         resp['payload'] = anqp_initial_resp(gas['dialog_token'], 0) + data
438         send_gas_resp(hapd, resp)
439         expect_gas_result(dev[0], "SUCCESS")
440
441 def test_gas_anqp_hs20_proto(dev, apdev):
442     """GAS/ANQP and Hotspot 2.0 element protocol testing"""
443     hapd = start_ap(apdev[0])
444     bssid = apdev[0]['bssid']
445
446     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
447     hapd.set("ext_mgmt_frame_handling", "1")
448
449     tests = [ '00', '0100', '0201', '0300', '0400', '0500', '0600', '0700',
450               '0800', '0900', '0a00', '0b0000000000' ]
451     for test in tests:
452         dev[0].request("HS20_ANQP_GET " + bssid + " 3,4")
453         query = gas_rx(hapd)
454         gas = parse_gas(query['payload'])
455         resp = action_response(query)
456         data = binascii.unhexlify(test)
457         data = binascii.unhexlify('506f9a11') + data
458         data = struct.pack('<HHH', len(data) + 4, 0xdddd, len(data)) + data
459         resp['payload'] = anqp_initial_resp(gas['dialog_token'], 0) + data
460         send_gas_resp(hapd, resp)
461         expect_gas_result(dev[0], "SUCCESS")
462
463 def expect_gas_result(dev, result, status=None):
464     ev = dev.wait_event(["GAS-QUERY-DONE"], timeout=10)
465     if ev is None:
466         raise Exception("GAS query timed out")
467     if "result=" + result not in ev:
468         raise Exception("Unexpected GAS query result")
469     if status and "status_code=" + str(status) + ' ' not in ev:
470         raise Exception("Unexpected GAS status code")
471
472 def anqp_get(dev, bssid, id):
473     if "OK" not in dev.request("ANQP_GET " + bssid + " " + str(id)):
474         raise Exception("ANQP_GET command failed")
475     ev = dev.wait_event(["GAS-QUERY-START"], timeout=5)
476     if ev is None:
477         raise Exception("GAS query start timed out")
478
479 def test_gas_timeout(dev, apdev):
480     """GAS timeout"""
481     hapd = start_ap(apdev[0])
482     bssid = apdev[0]['bssid']
483
484     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
485     hapd.set("ext_mgmt_frame_handling", "1")
486
487     anqp_get(dev[0], bssid, 263)
488
489     ev = hapd.wait_event(["MGMT-RX"], timeout=5)
490     if ev is None:
491         raise Exception("MGMT RX wait timed out")
492
493     expect_gas_result(dev[0], "TIMEOUT")
494
495 MGMT_SUBTYPE_ACTION = 13
496 ACTION_CATEG_PUBLIC = 4
497
498 GAS_INITIAL_REQUEST = 10
499 GAS_INITIAL_RESPONSE = 11
500 GAS_COMEBACK_REQUEST = 12
501 GAS_COMEBACK_RESPONSE = 13
502 GAS_ACTIONS = [ GAS_INITIAL_REQUEST, GAS_INITIAL_RESPONSE,
503                 GAS_COMEBACK_REQUEST, GAS_COMEBACK_RESPONSE ]
504
505 def anqp_adv_proto():
506     return struct.pack('BBBB', 108, 2, 127, 0)
507
508 def anqp_initial_resp(dialog_token, status_code, comeback_delay=0):
509     return struct.pack('<BBBHH', ACTION_CATEG_PUBLIC, GAS_INITIAL_RESPONSE,
510                        dialog_token, status_code, comeback_delay) + anqp_adv_proto()
511
512 def anqp_comeback_resp(dialog_token, status_code=0, id=0, more=False, comeback_delay=0, bogus_adv_proto=False):
513     if more:
514         id |= 0x80
515     if bogus_adv_proto:
516         adv = struct.pack('BBBB', 108, 2, 127, 1)
517     else:
518         adv = anqp_adv_proto()
519     return struct.pack('<BBBHBH', ACTION_CATEG_PUBLIC, GAS_COMEBACK_RESPONSE,
520                        dialog_token, status_code, id, comeback_delay) + adv
521
522 def gas_rx(hapd):
523     count = 0
524     while count < 30:
525         count = count + 1
526         query = hapd.mgmt_rx()
527         if query is None:
528             raise Exception("Action frame not received")
529         if query['subtype'] != MGMT_SUBTYPE_ACTION:
530             continue
531         payload = query['payload']
532         if len(payload) < 2:
533             continue
534         (category, action) = struct.unpack('BB', payload[0:2])
535         if category != ACTION_CATEG_PUBLIC or action not in GAS_ACTIONS:
536             continue
537         return query
538     raise Exception("No Action frame received")
539
540 def parse_gas(payload):
541     pos = payload
542     (category, action, dialog_token) = struct.unpack('BBB', pos[0:3])
543     if category != ACTION_CATEG_PUBLIC:
544         return None
545     if action not in GAS_ACTIONS:
546         return None
547     gas = {}
548     gas['action'] = action
549     pos = pos[3:]
550
551     if len(pos) < 1 and action != GAS_COMEBACK_REQUEST:
552         return None
553
554     gas['dialog_token'] = dialog_token
555
556     if action == GAS_INITIAL_RESPONSE:
557         if len(pos) < 4:
558             return None
559         (status_code, comeback_delay) = struct.unpack('<HH', pos[0:4])
560         gas['status_code'] = status_code
561         gas['comeback_delay'] = comeback_delay
562
563     if action == GAS_COMEBACK_RESPONSE:
564         if len(pos) < 5:
565             return None
566         (status_code, frag, comeback_delay) = struct.unpack('<HBH', pos[0:5])
567         gas['status_code'] = status_code
568         gas['frag'] = frag
569         gas['comeback_delay'] = comeback_delay
570
571     return gas
572
573 def action_response(req):
574     resp = {}
575     resp['fc'] = req['fc']
576     resp['da'] = req['sa']
577     resp['sa'] = req['da']
578     resp['bssid'] = req['bssid']
579     return resp
580
581 def send_gas_resp(hapd, resp):
582     hapd.mgmt_tx(resp)
583     ev = hapd.wait_event(["MGMT-TX-STATUS"], timeout=5)
584     if ev is None:
585         raise Exception("Missing TX status for GAS response")
586     if "ok=1" not in ev:
587         raise Exception("GAS response not acknowledged")
588
589 def test_gas_invalid_response_type(dev, apdev):
590     """GAS invalid response type"""
591     hapd = start_ap(apdev[0])
592     bssid = apdev[0]['bssid']
593
594     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
595     hapd.set("ext_mgmt_frame_handling", "1")
596
597     anqp_get(dev[0], bssid, 263)
598
599     query = gas_rx(hapd)
600     gas = parse_gas(query['payload'])
601
602     resp = action_response(query)
603     # GAS Comeback Response instead of GAS Initial Response
604     resp['payload'] = anqp_comeback_resp(gas['dialog_token']) + struct.pack('<H', 0)
605     send_gas_resp(hapd, resp)
606
607     # station drops the invalid frame, so this needs to result in GAS timeout
608     expect_gas_result(dev[0], "TIMEOUT")
609
610 def test_gas_failure_status_code(dev, apdev):
611     """GAS failure status code"""
612     hapd = start_ap(apdev[0])
613     bssid = apdev[0]['bssid']
614
615     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
616     hapd.set("ext_mgmt_frame_handling", "1")
617
618     anqp_get(dev[0], bssid, 263)
619
620     query = gas_rx(hapd)
621     gas = parse_gas(query['payload'])
622
623     resp = action_response(query)
624     resp['payload'] = anqp_initial_resp(gas['dialog_token'], 61) + struct.pack('<H', 0)
625     send_gas_resp(hapd, resp)
626
627     expect_gas_result(dev[0], "FAILURE")
628
629 def test_gas_malformed(dev, apdev):
630     """GAS malformed response frames"""
631     hapd = start_ap(apdev[0])
632     bssid = apdev[0]['bssid']
633
634     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
635     hapd.set("ext_mgmt_frame_handling", "1")
636
637     anqp_get(dev[0], bssid, 263)
638
639     query = gas_rx(hapd)
640     gas = parse_gas(query['payload'])
641
642     resp = action_response(query)
643
644     resp['payload'] = struct.pack('<BBBH', ACTION_CATEG_PUBLIC,
645                                   GAS_COMEBACK_RESPONSE,
646                                   gas['dialog_token'], 0)
647     hapd.mgmt_tx(resp)
648
649     resp['payload'] = struct.pack('<BBBHB', ACTION_CATEG_PUBLIC,
650                                   GAS_COMEBACK_RESPONSE,
651                                   gas['dialog_token'], 0, 0)
652     hapd.mgmt_tx(resp)
653
654     hdr = struct.pack('<BBBHH', ACTION_CATEG_PUBLIC, GAS_INITIAL_RESPONSE,
655                       gas['dialog_token'], 0, 0)
656     resp['payload'] = hdr + struct.pack('B', 108)
657     hapd.mgmt_tx(resp)
658     resp['payload'] = hdr + struct.pack('BB', 108, 0)
659     hapd.mgmt_tx(resp)
660     resp['payload'] = hdr + struct.pack('BB', 108, 1)
661     hapd.mgmt_tx(resp)
662     resp['payload'] = hdr + struct.pack('BB', 108, 255)
663     hapd.mgmt_tx(resp)
664     resp['payload'] = hdr + struct.pack('BBB', 108, 1, 127)
665     hapd.mgmt_tx(resp)
666     resp['payload'] = hdr + struct.pack('BBB', 108, 2, 127)
667     hapd.mgmt_tx(resp)
668     resp['payload'] = hdr + struct.pack('BBBB', 0, 2, 127, 0)
669     hapd.mgmt_tx(resp)
670
671     resp['payload'] = anqp_initial_resp(gas['dialog_token'], 0) + struct.pack('<H', 1)
672     hapd.mgmt_tx(resp)
673
674     resp['payload'] = anqp_initial_resp(gas['dialog_token'], 0) + struct.pack('<HB', 2, 0)
675     hapd.mgmt_tx(resp)
676
677     resp['payload'] = anqp_initial_resp(gas['dialog_token'], 0) + struct.pack('<H', 65535)
678     hapd.mgmt_tx(resp)
679
680     resp['payload'] = anqp_initial_resp(gas['dialog_token'], 0) + struct.pack('<HBB', 1, 0, 0)
681     hapd.mgmt_tx(resp)
682
683     # Station drops invalid frames, but the last of the responses is valid from
684     # GAS view point even though it has an extra octet in the end and the ANQP
685     # part of the response is not valid. This is reported as successfully
686     # completed GAS exchange.
687     expect_gas_result(dev[0], "SUCCESS")
688
689     ev = dev[0].wait_event(["ANQP-QUERY-DONE"], timeout=5)
690     if ev is None:
691         raise Exception("ANQP-QUERY-DONE not reported")
692     if "result=INVALID_FRAME" not in ev:
693         raise Exception("Unexpected result: " + ev)
694
695 def init_gas(hapd, bssid, dev):
696     anqp_get(dev, bssid, 263)
697     query = gas_rx(hapd)
698     gas = parse_gas(query['payload'])
699     dialog_token = gas['dialog_token']
700
701     resp = action_response(query)
702     resp['payload'] = anqp_initial_resp(dialog_token, 0, comeback_delay=1) + struct.pack('<H', 0)
703     send_gas_resp(hapd, resp)
704
705     query = gas_rx(hapd)
706     gas = parse_gas(query['payload'])
707     if gas['action'] != GAS_COMEBACK_REQUEST:
708         raise Exception("Unexpected request action")
709     if gas['dialog_token'] != dialog_token:
710         raise Exception("Unexpected dialog token change")
711     return query, dialog_token
712
713 def allow_gas_initial_req(hapd, dialog_token):
714     msg = hapd.mgmt_rx(timeout=1)
715     if msg is not None:
716         gas = parse_gas(msg['payload'])
717         if gas['action'] != GAS_INITIAL_REQUEST or dialog_token == gas['dialog_token']:
718             raise Exception("Unexpected management frame")
719
720 def test_gas_malformed_comeback_resp(dev, apdev):
721     """GAS malformed comeback response frames"""
722     hapd = start_ap(apdev[0])
723     bssid = apdev[0]['bssid']
724
725     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
726     hapd.set("ext_mgmt_frame_handling", "1")
727
728     logger.debug("Non-zero status code in comeback response")
729     query, dialog_token = init_gas(hapd, bssid, dev[0])
730     resp = action_response(query)
731     resp['payload'] = anqp_comeback_resp(dialog_token, status_code=2) + struct.pack('<H', 0)
732     send_gas_resp(hapd, resp)
733     expect_gas_result(dev[0], "FAILURE", status=2)
734
735     logger.debug("Different advertisement protocol in comeback response")
736     query, dialog_token = init_gas(hapd, bssid, dev[0])
737     resp = action_response(query)
738     resp['payload'] = anqp_comeback_resp(dialog_token, bogus_adv_proto=True) + struct.pack('<H', 0)
739     send_gas_resp(hapd, resp)
740     expect_gas_result(dev[0], "PEER_ERROR")
741
742     logger.debug("Non-zero frag id and comeback delay in comeback response")
743     query, dialog_token = init_gas(hapd, bssid, dev[0])
744     resp = action_response(query)
745     resp['payload'] = anqp_comeback_resp(dialog_token, id=1, comeback_delay=1) + struct.pack('<H', 0)
746     send_gas_resp(hapd, resp)
747     expect_gas_result(dev[0], "PEER_ERROR")
748
749     logger.debug("Unexpected frag id in comeback response")
750     query, dialog_token = init_gas(hapd, bssid, dev[0])
751     resp = action_response(query)
752     resp['payload'] = anqp_comeback_resp(dialog_token, id=1) + struct.pack('<H', 0)
753     send_gas_resp(hapd, resp)
754     expect_gas_result(dev[0], "PEER_ERROR")
755
756     logger.debug("Empty fragment and replay in comeback response")
757     query, dialog_token = init_gas(hapd, bssid, dev[0])
758     resp = action_response(query)
759     resp['payload'] = anqp_comeback_resp(dialog_token, more=True) + struct.pack('<H', 0)
760     send_gas_resp(hapd, resp)
761     query = gas_rx(hapd)
762     gas = parse_gas(query['payload'])
763     if gas['action'] != GAS_COMEBACK_REQUEST:
764         raise Exception("Unexpected request action")
765     if gas['dialog_token'] != dialog_token:
766         raise Exception("Unexpected dialog token change")
767     resp = action_response(query)
768     resp['payload'] = anqp_comeback_resp(dialog_token) + struct.pack('<H', 0)
769     send_gas_resp(hapd, resp)
770     resp['payload'] = anqp_comeback_resp(dialog_token, id=1) + struct.pack('<H', 0)
771     send_gas_resp(hapd, resp)
772     expect_gas_result(dev[0], "SUCCESS")
773
774     logger.debug("Unexpected initial response when waiting for comeback response")
775     query, dialog_token = init_gas(hapd, bssid, dev[0])
776     resp = action_response(query)
777     resp['payload'] = anqp_initial_resp(dialog_token, 0) + struct.pack('<H', 0)
778     send_gas_resp(hapd, resp)
779     allow_gas_initial_req(hapd, dialog_token)
780     expect_gas_result(dev[0], "TIMEOUT")
781
782     logger.debug("Too short comeback response")
783     query, dialog_token = init_gas(hapd, bssid, dev[0])
784     resp = action_response(query)
785     resp['payload'] = struct.pack('<BBBH', ACTION_CATEG_PUBLIC,
786                                   GAS_COMEBACK_RESPONSE, dialog_token, 0)
787     send_gas_resp(hapd, resp)
788     allow_gas_initial_req(hapd, dialog_token)
789     expect_gas_result(dev[0], "TIMEOUT")
790
791     logger.debug("Too short comeback response(2)")
792     query, dialog_token = init_gas(hapd, bssid, dev[0])
793     resp = action_response(query)
794     resp['payload'] = struct.pack('<BBBHBB', ACTION_CATEG_PUBLIC,
795                                   GAS_COMEBACK_RESPONSE, dialog_token, 0, 0x80,
796                                   0)
797     send_gas_resp(hapd, resp)
798     allow_gas_initial_req(hapd, dialog_token)
799     expect_gas_result(dev[0], "TIMEOUT")
800
801     logger.debug("Maximum comeback response fragment claiming more fragments")
802     query, dialog_token = init_gas(hapd, bssid, dev[0])
803     resp = action_response(query)
804     resp['payload'] = anqp_comeback_resp(dialog_token, more=True) + struct.pack('<H', 0)
805     send_gas_resp(hapd, resp)
806     for i in range(1, 129):
807         query = gas_rx(hapd)
808         gas = parse_gas(query['payload'])
809         if gas['action'] != GAS_COMEBACK_REQUEST:
810             raise Exception("Unexpected request action")
811         if gas['dialog_token'] != dialog_token:
812             raise Exception("Unexpected dialog token change")
813         resp = action_response(query)
814         resp['payload'] = anqp_comeback_resp(dialog_token, id=i, more=True) + struct.pack('<H', 0)
815         send_gas_resp(hapd, resp)
816     expect_gas_result(dev[0], "PEER_ERROR")
817
818 def test_gas_comeback_resp_additional_delay(dev, apdev):
819     """GAS comeback response requesting additional delay"""
820     hapd = start_ap(apdev[0])
821     bssid = apdev[0]['bssid']
822
823     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
824     hapd.set("ext_mgmt_frame_handling", "1")
825
826     query, dialog_token = init_gas(hapd, bssid, dev[0])
827     for i in range(0, 2):
828         resp = action_response(query)
829         resp['payload'] = anqp_comeback_resp(dialog_token, status_code=95, comeback_delay=50) + struct.pack('<H', 0)
830         send_gas_resp(hapd, resp)
831         query = gas_rx(hapd)
832         gas = parse_gas(query['payload'])
833         if gas['action'] != GAS_COMEBACK_REQUEST:
834             raise Exception("Unexpected request action")
835         if gas['dialog_token'] != dialog_token:
836             raise Exception("Unexpected dialog token change")
837     resp = action_response(query)
838     resp['payload'] = anqp_comeback_resp(dialog_token, status_code=0) + struct.pack('<H', 0)
839     send_gas_resp(hapd, resp)
840     expect_gas_result(dev[0], "SUCCESS")
841
842 def test_gas_unknown_adv_proto(dev, apdev):
843     """Unknown advertisement protocol id"""
844     bssid = apdev[0]['bssid']
845     params = hs20_ap_params()
846     params['hessid'] = bssid
847     hostapd.add_ap(apdev[0], params)
848
849     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
850     req = dev[0].request("GAS_REQUEST " + bssid + " 42 000102000101")
851     if "FAIL" in req:
852         raise Exception("GAS query request rejected")
853     expect_gas_result(dev[0], "FAILURE", "59")
854     ev = dev[0].wait_event(["GAS-RESPONSE-INFO"], timeout=10)
855     if ev is None:
856         raise Exception("GAS query timed out")
857     exp = r'<.>(GAS-RESPONSE-INFO) addr=([0-9a-f:]*) dialog_token=([0-9]*) status_code=([0-9]*) resp_len=([\-0-9]*)'
858     res = re.split(exp, ev)
859     if len(res) < 6:
860         raise Exception("Could not parse GAS-RESPONSE-INFO")
861     if res[2] != bssid:
862         raise Exception("Unexpected BSSID in response")
863     status = res[4]
864     if status != "59":
865         raise Exception("Unexpected GAS-RESPONSE-INFO status")
866
867 def test_gas_request_oom(dev, apdev):
868     """GAS_REQUEST OOM"""
869     bssid = apdev[0]['bssid']
870     params = hs20_ap_params()
871     params['hessid'] = bssid
872     hostapd.add_ap(apdev[0], params)
873
874     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
875
876     with alloc_fail(dev[0], 1, "gas_build_req;gas_send_request"):
877         if "FAIL" not in dev[0].request("GAS_REQUEST " + bssid + " 42"):
878             raise Exception("GAS query request rejected")
879
880     with alloc_fail(dev[0], 1, "gas_query_req;gas_send_request"):
881         if "FAIL" not in dev[0].request("GAS_REQUEST " + bssid + " 42"):
882             raise Exception("GAS query request rejected")
883
884     with alloc_fail(dev[0], 1, "wpabuf_dup;gas_resp_cb"):
885         if "OK" not in dev[0].request("GAS_REQUEST " + bssid + " 00 000102000101"):
886             raise Exception("GAS query request rejected")
887         ev = dev[0].wait_event(["GAS-RESPONSE-INFO"], timeout=10)
888         if ev is None:
889             raise Exception("No GAS response")
890         if "status_code=0" not in ev:
891             raise Exception("GAS response indicated a failure")
892
893 def test_gas_max_pending(dev, apdev):
894     """GAS and maximum pending query limit"""
895     hapd = start_ap(apdev[0])
896     hapd.set("gas_frag_limit", "50")
897     bssid = apdev[0]['bssid']
898
899     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
900     wpas.interface_add("wlan5")
901     if "OK" not in wpas.request("P2P_SET listen_channel 1"):
902         raise Exception("Failed to set listen channel")
903     if "OK" not in wpas.p2p_listen():
904         raise Exception("Failed to start listen state")
905     if "FAIL" in wpas.request("SET ext_mgmt_frame_handling 1"):
906         raise Exception("Failed to enable external management frame handling")
907
908     anqp_query = struct.pack('<HHHHHHHHHH', 256, 16, 257, 258, 260, 261, 262, 263, 264, 268)
909     gas = struct.pack('<H', len(anqp_query)) + anqp_query
910
911     for dialog_token in range(1, 10):
912         msg = struct.pack('<BBB', ACTION_CATEG_PUBLIC, GAS_INITIAL_REQUEST,
913                           dialog_token) + anqp_adv_proto() + gas
914         req = "MGMT_TX {} {} freq=2412 wait_time=10 action={}".format(bssid, bssid, binascii.hexlify(msg))
915         if "OK" not in wpas.request(req):
916             raise Exception("Could not send management frame")
917         resp = wpas.mgmt_rx()
918         if resp is None:
919             raise Exception("MGMT-RX timeout")
920         if 'payload' not in resp:
921             raise Exception("Missing payload")
922         gresp = parse_gas(resp['payload'])
923         if gresp['dialog_token'] != dialog_token:
924             raise Exception("Dialog token mismatch")
925         status_code = gresp['status_code']
926         if dialog_token < 9 and status_code != 0:
927             raise Exception("Unexpected failure status code {} for dialog token {}".format(status_code, dialog_token))
928         if dialog_token > 8 and status_code == 0:
929             raise Exception("Unexpected success status code {} for dialog token {}".format(status_code, dialog_token))
930
931 def test_gas_no_pending(dev, apdev):
932     """GAS and no pending query for comeback request"""
933     hapd = start_ap(apdev[0])
934     bssid = apdev[0]['bssid']
935
936     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
937     wpas.interface_add("wlan5")
938     if "OK" not in wpas.request("P2P_SET listen_channel 1"):
939         raise Exception("Failed to set listen channel")
940     if "OK" not in wpas.p2p_listen():
941         raise Exception("Failed to start listen state")
942     if "FAIL" in wpas.request("SET ext_mgmt_frame_handling 1"):
943         raise Exception("Failed to enable external management frame handling")
944
945     msg = struct.pack('<BBB', ACTION_CATEG_PUBLIC, GAS_COMEBACK_REQUEST, 1)
946     req = "MGMT_TX {} {} freq=2412 wait_time=10 action={}".format(bssid, bssid, binascii.hexlify(msg))
947     if "OK" not in wpas.request(req):
948         raise Exception("Could not send management frame")
949     resp = wpas.mgmt_rx()
950     if resp is None:
951         raise Exception("MGMT-RX timeout")
952     if 'payload' not in resp:
953         raise Exception("Missing payload")
954     gresp = parse_gas(resp['payload'])
955     status_code = gresp['status_code']
956     if status_code != 60:
957         raise Exception("Unexpected status code {} (expected 60)".format(status_code))
958
959 def test_gas_delete_at_deinit(dev, apdev):
960     """GAS query deleted at deinit"""
961     hapd = start_ap(apdev[0])
962     hapd.set("gas_comeback_delay", "1000")
963     bssid = apdev[0]['bssid']
964
965     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
966     wpas.interface_add("wlan5")
967     wpas.scan_for_bss(apdev[0]['bssid'], freq="2412", force_scan=True)
968     wpas.request("ANQP_GET " + bssid + " 258")
969
970     wpas.global_request("INTERFACE_REMOVE " + wpas.ifname)
971     ev = wpas.wait_event(["GAS-QUERY-DONE"], timeout=2)
972     del wpas
973     if ev is None:
974         raise Exception("GAS-QUERY-DONE not seen")
975     if "result=DELETED_AT_DEINIT" not in ev:
976         raise Exception("Unexpected result code: " + ev)
977
978 def test_gas_missing_payload(dev, apdev):
979     """No action code in the query frame"""
980     bssid = apdev[0]['bssid']
981     params = hs20_ap_params()
982     params['hessid'] = bssid
983     hostapd.add_ap(apdev[0], params)
984
985     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
986
987     cmd = "MGMT_TX {} {} freq=2412 action=040A".format(bssid, bssid)
988     if "FAIL" in dev[0].request(cmd):
989         raise Exception("Could not send test Action frame")
990     ev = dev[0].wait_event(["MGMT-TX-STATUS"], timeout=10)
991     if ev is None:
992         raise Exception("Timeout on MGMT-TX-STATUS")
993     if "result=SUCCESS" not in ev:
994         raise Exception("AP did not ack Action frame")
995
996     cmd = "MGMT_TX {} {} freq=2412 action=04".format(bssid, bssid)
997     if "FAIL" in dev[0].request(cmd):
998         raise Exception("Could not send test Action frame")
999     ev = dev[0].wait_event(["MGMT-TX-STATUS"], timeout=10)
1000     if ev is None:
1001         raise Exception("Timeout on MGMT-TX-STATUS")
1002     if "result=SUCCESS" not in ev:
1003         raise Exception("AP did not ack Action frame")
1004
1005 def test_gas_query_deinit(dev, apdev):
1006     """Pending GAS/ANQP query during deinit"""
1007     hapd = start_ap(apdev[0])
1008     bssid = apdev[0]['bssid']
1009
1010     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
1011     wpas.interface_add("wlan5")
1012
1013     wpas.scan_for_bss(bssid, freq="2412", force_scan=True)
1014     id = wpas.request("RADIO_WORK add block-work")
1015     if "OK" not in wpas.request("ANQP_GET " + bssid + " 258"):
1016         raise Exception("ANQP_GET command failed")
1017
1018     ev = wpas.wait_event(["GAS-QUERY-START", "EXT-RADIO-WORK-START"], timeout=5)
1019     if ev is None:
1020         raise Exception("Timeout while waiting radio work to start")
1021     ev = wpas.wait_event(["GAS-QUERY-START", "EXT-RADIO-WORK-START"], timeout=5)
1022     if ev is None:
1023         raise Exception("Timeout while waiting radio work to start (2)")
1024
1025     # Remove the interface while the gas-query radio work is still pending and
1026     # GAS query has not yet been started.
1027     wpas.interface_remove("wlan5")
1028
1029 @remote_compatible
1030 def test_gas_anqp_oom_wpas(dev, apdev):
1031     """GAS/ANQP query and OOM in wpa_supplicant"""
1032     hapd = start_ap(apdev[0])
1033     bssid = apdev[0]['bssid']
1034
1035     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
1036
1037     with alloc_fail(dev[0], 1, "wpa_bss_anqp_alloc"):
1038         if "OK" not in dev[0].request("ANQP_GET " + bssid + " 258"):
1039             raise Exception("ANQP_GET command failed")
1040         ev = dev[0].wait_event(["ANQP-QUERY-DONE"], timeout=5)
1041         if ev is None:
1042             raise Exception("ANQP query did not complete")
1043
1044     with alloc_fail(dev[0], 1, "gas_build_req"):
1045         if "FAIL" not in dev[0].request("ANQP_GET " + bssid + " 258"):
1046             raise Exception("Unexpected ANQP_GET command success (OOM)")
1047
1048 def test_gas_anqp_oom_hapd(dev, apdev):
1049     """GAS/ANQP query and OOM in hostapd"""
1050     hapd = start_ap(apdev[0])
1051     bssid = apdev[0]['bssid']
1052
1053     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
1054
1055     with alloc_fail(hapd, 1, "gas_build_resp"):
1056         # This query will time out due to the AP not sending a response (OOM).
1057         if "OK" not in dev[0].request("ANQP_GET " + bssid + " 258"):
1058             raise Exception("ANQP_GET command failed")
1059
1060         ev = dev[0].wait_event(["GAS-QUERY-START"], timeout=5)
1061         if ev is None:
1062             raise Exception("GAS query start timed out")
1063
1064         ev = dev[0].wait_event(["GAS-QUERY-DONE"], timeout=10)
1065         if ev is None:
1066             raise Exception("GAS query timed out")
1067         if "result=TIMEOUT" not in ev:
1068             raise Exception("Unexpected result: " + ev)
1069
1070         ev = dev[0].wait_event(["ANQP-QUERY-DONE"], timeout=10)
1071         if ev is None:
1072             raise Exception("ANQP-QUERY-DONE event not seen")
1073         if "result=FAILURE" not in ev:
1074             raise Exception("Unexpected result: " + ev)
1075
1076     with alloc_fail(hapd, 1, "gas_anqp_build_comeback_resp"):
1077         hapd.set("gas_frag_limit", "50")
1078
1079         # The first attempt of this query will time out due to the AP not
1080         # sending a response (OOM), but the retry succeeds.
1081         dev[0].request("FETCH_ANQP")
1082         ev = dev[0].wait_event(["GAS-QUERY-START"], timeout=5)
1083         if ev is None:
1084             raise Exception("GAS query start timed out")
1085
1086         ev = dev[0].wait_event(["GAS-QUERY-DONE"], timeout=10)
1087         if ev is None:
1088             raise Exception("GAS query timed out")
1089         if "result=SUCCESS" not in ev:
1090             raise Exception("Unexpected result: " + ev)
1091
1092         ev = dev[0].wait_event(["ANQP-QUERY-DONE"], timeout=10)
1093         if ev is None:
1094             raise Exception("ANQP-QUERY-DONE event not seen")
1095         if "result=SUCCESS" not in ev:
1096             raise Exception("Unexpected result: " + ev)
1097
1098 def test_gas_anqp_extra_elements(dev, apdev):
1099     """GAS/ANQP and extra ANQP elements"""
1100     geo_loc = "001052834d12efd2b08b9b4bf1cc2c00004104050000000000060100"
1101     civic_loc = "0000f9555302f50102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f202122232425262728292a2b2c2d2e2f303132333435363738393a3b3c3d3e3f404142434445464748494a4b4c4d4e4f505152535455565758595a5b5c5d5e5f606162636465666768696a6b6c6d6e6f707172737475767778797a7b7c7d7e7f808182838485868788898a8b8c8d8e8f909192939495969798999a9b9c9d9e9fa0a1a2a3a4a5a6a7a8a9aaabacadaeafb0b1b2b3b4b5b6b7b8b9babbbcbdbebfc0c1c2c3c4c5c6c7c8c9cacbcccdcecfd0d1d2d3d4d5d6d7d8d9dadbdcdddedfe0e1e2e3e4e5e6e7e8e9eaebecedeeeff0f1f2f3f4f5"
1102     held_uri = "https://held.example.com/location"
1103     held = struct.pack('BBB', 0, 1 + len(held_uri), 1) + held_uri
1104     supl_fqdn = "supl.example.com"
1105     supl = struct.pack('BBB', 0, 1 + len(supl_fqdn), 1) + supl_fqdn
1106     public_id = binascii.hexlify(held + supl)
1107     params = { "ssid": "gas/anqp",
1108                "interworking": "1",
1109                "anqp_elem": [ "265:" + geo_loc,
1110                               "266:" + civic_loc,
1111                               "262:1122334455",
1112                               "267:" + public_id,
1113                               "275:01020304",
1114                               "60000:01",
1115                               "299:0102" ] }
1116     hapd = hostapd.add_ap(apdev[0], params)
1117     bssid = apdev[0]['bssid']
1118
1119     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
1120     if "OK" not in dev[0].request("ANQP_GET " + bssid + " 265,266"):
1121         raise Exception("ANQP_GET command failed")
1122
1123     ev = dev[0].wait_event(["GAS-QUERY-DONE"], timeout=10)
1124     if ev is None:
1125         raise Exception("GAS query timed out")
1126
1127     bss = dev[0].get_bss(bssid)
1128
1129     if 'anqp[265]' not in bss:
1130         raise Exception("AP Geospatial Location ANQP-element not seen")
1131     if bss['anqp[265]'] != geo_loc:
1132         raise Exception("Unexpected AP Geospatial Location ANQP-element value: " + bss['anqp[265]'])
1133
1134     if 'anqp[266]' not in bss:
1135         raise Exception("AP Civic Location ANQP-element not seen")
1136     if bss['anqp[266]'] != civic_loc:
1137         raise Exception("Unexpected AP Civic Location ANQP-element value: " + bss['anqp[266]'])
1138
1139     dev[1].scan_for_bss(bssid, freq="2412", force_scan=True)
1140     if "OK" not in dev[1].request("ANQP_GET " + bssid + " 257,258,259,260,261,262,263,264,265,267,268,269,270,271,272,273,274,275,276,277,278,279,280,281,282,283,284,285,286,287,288,289,290,291,292,293,294,295,296,297,298,299"):
1141         raise Exception("ANQP_GET command failed")
1142
1143     ev = dev[1].wait_event(["GAS-QUERY-DONE"], timeout=10)
1144     if ev is None:
1145         raise Exception("GAS query timed out")
1146
1147     bss = dev[1].get_bss(bssid)
1148
1149     if 'anqp[265]' not in bss:
1150         raise Exception("AP Geospatial Location ANQP-element not seen")
1151     if bss['anqp[265]'] != geo_loc:
1152         raise Exception("Unexpected AP Geospatial Location ANQP-element value: " + bss['anqp[265]'])
1153
1154     if 'anqp[266]' in bss:
1155         raise Exception("AP Civic Location ANQP-element unexpectedly seen")
1156
1157     if 'anqp[267]' not in bss:
1158         raise Exception("AP Location Public Identifier ANQP-element not seen")
1159     if bss['anqp[267]'] != public_id:
1160         raise Exception("Unexpected AP Location Public Identifier ANQP-element value: " + bss['anqp[267]'])
1161
1162     if 'anqp[275]' not in bss:
1163         raise Exception("ANQP-element Info ID 275 not seen")
1164     if bss['anqp[275]'] != "01020304":
1165         raise Exception("Unexpected AP ANQP-element Info ID 299 value: " + bss['anqp[299]'])
1166
1167     if 'anqp[299]' not in bss:
1168         raise Exception("ANQP-element Info ID 299 not seen")
1169     if bss['anqp[299]'] != "0102":
1170         raise Exception("Unexpected AP ANQP-element Info ID 299 value: " + bss['anqp[299]'])
1171
1172     if 'anqp_ip_addr_type_availability' not in bss:
1173         raise Exception("ANQP-element Info ID 292 not seen")
1174     if bss['anqp_ip_addr_type_availability'] != "1122334455":
1175         raise Exception("Unexpected AP ANQP-element Info ID 262 value: " + bss['anqp_ip_addr_type_availability'])
1176
1177 def test_gas_anqp_address3_not_assoc(dev, apdev, params):
1178     """GAS/ANQP query using IEEE 802.11 compliant Address 3 value when not associated"""
1179     try:
1180         _test_gas_anqp_address3_not_assoc(dev, apdev, params)
1181     finally:
1182         dev[0].request("SET gas_address3 0")
1183
1184 def _test_gas_anqp_address3_not_assoc(dev, apdev, params):
1185     hapd = start_ap(apdev[0])
1186     bssid = apdev[0]['bssid']
1187
1188     if "OK" not in dev[0].request("SET gas_address3 1"):
1189         raise Exception("Failed to set gas_address3")
1190
1191     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
1192     if "OK" not in dev[0].request("ANQP_GET " + bssid + " 258"):
1193         raise Exception("ANQP_GET command failed")
1194
1195     ev = dev[0].wait_event(["GAS-QUERY-START"], timeout=5)
1196     if ev is None:
1197         raise Exception("GAS query start timed out")
1198
1199     ev = dev[0].wait_event(["GAS-QUERY-DONE"], timeout=10)
1200     if ev is None:
1201         raise Exception("GAS query timed out")
1202
1203     ev = dev[0].wait_event(["RX-ANQP"], timeout=1)
1204     if ev is None or "Venue Name" not in ev:
1205         raise Exception("Did not receive Venue Name")
1206
1207     ev = dev[0].wait_event(["ANQP-QUERY-DONE"], timeout=10)
1208     if ev is None:
1209         raise Exception("ANQP-QUERY-DONE event not seen")
1210     if "result=SUCCESS" not in ev:
1211         raise Exception("Unexpected result: " + ev)
1212
1213     out = run_tshark(os.path.join(params['logdir'], "hwsim0.pcapng"),
1214                      "wlan_mgt.fixed.category_code == 4 && (wlan_mgt.fixed.publicact == 0x0a || wlan_mgt.fixed.publicact == 0x0b)",
1215                      display=["wlan.bssid"])
1216     res = out.splitlines()
1217     if len(res) != 2:
1218         raise Exception("Unexpected number of GAS frames")
1219     if res[0] != 'ff:ff:ff:ff:ff:ff':
1220         raise Exception("GAS request used unexpected Address3 field value: " + res[0])
1221     if res[1] != 'ff:ff:ff:ff:ff:ff':
1222         raise Exception("GAS response used unexpected Address3 field value: " + res[1])
1223
1224 def test_gas_anqp_address3_assoc(dev, apdev, params):
1225     """GAS/ANQP query using IEEE 802.11 compliant Address 3 value when associated"""
1226     try:
1227         _test_gas_anqp_address3_assoc(dev, apdev, params)
1228     finally:
1229         dev[0].request("SET gas_address3 0")
1230
1231 def _test_gas_anqp_address3_assoc(dev, apdev, params):
1232     hapd = start_ap(apdev[0])
1233     bssid = apdev[0]['bssid']
1234
1235     if "OK" not in dev[0].request("SET gas_address3 1"):
1236         raise Exception("Failed to set gas_address3")
1237
1238     dev[0].scan_for_bss(bssid, freq="2412")
1239     dev[0].connect("test-gas", key_mgmt="WPA-EAP", eap="TTLS",
1240                    identity="DOMAIN\mschapv2 user", anonymous_identity="ttls",
1241                    password="password", phase2="auth=MSCHAPV2",
1242                    ca_cert="auth_serv/ca.pem", scan_freq="2412")
1243
1244     if "OK" not in dev[0].request("ANQP_GET " + bssid + " 258"):
1245         raise Exception("ANQP_GET command failed")
1246
1247     ev = dev[0].wait_event(["GAS-QUERY-START"], timeout=5)
1248     if ev is None:
1249         raise Exception("GAS query start timed out")
1250
1251     ev = dev[0].wait_event(["GAS-QUERY-DONE"], timeout=10)
1252     if ev is None:
1253         raise Exception("GAS query timed out")
1254
1255     ev = dev[0].wait_event(["RX-ANQP"], timeout=1)
1256     if ev is None or "Venue Name" not in ev:
1257         raise Exception("Did not receive Venue Name")
1258
1259     ev = dev[0].wait_event(["ANQP-QUERY-DONE"], timeout=10)
1260     if ev is None:
1261         raise Exception("ANQP-QUERY-DONE event not seen")
1262     if "result=SUCCESS" not in ev:
1263         raise Exception("Unexpected result: " + ev)
1264
1265     out = run_tshark(os.path.join(params['logdir'], "hwsim0.pcapng"),
1266                      "wlan_mgt.fixed.category_code == 4 && (wlan_mgt.fixed.publicact == 0x0a || wlan_mgt.fixed.publicact == 0x0b)",
1267                      display=["wlan.bssid"])
1268     res = out.splitlines()
1269     if len(res) != 2:
1270         raise Exception("Unexpected number of GAS frames")
1271     if res[0] != bssid:
1272         raise Exception("GAS request used unexpected Address3 field value: " + res[0])
1273     if res[1] != bssid:
1274         raise Exception("GAS response used unexpected Address3 field value: " + res[1])
1275
1276 def test_gas_anqp_address3_ap_forced(dev, apdev, params):
1277     """GAS/ANQP query using IEEE 802.11 compliant Address 3 value on AP"""
1278     hapd = start_ap(apdev[0])
1279     bssid = apdev[0]['bssid']
1280     hapd.set("gas_address3", "1")
1281
1282     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
1283     if "OK" not in dev[0].request("ANQP_GET " + bssid + " 258"):
1284         raise Exception("ANQP_GET command failed")
1285
1286     ev = dev[0].wait_event(["GAS-QUERY-START"], timeout=5)
1287     if ev is None:
1288         raise Exception("GAS query start timed out")
1289
1290     ev = dev[0].wait_event(["GAS-QUERY-DONE"], timeout=10)
1291     if ev is None:
1292         raise Exception("GAS query timed out")
1293
1294     ev = dev[0].wait_event(["RX-ANQP"], timeout=1)
1295     if ev is None or "Venue Name" not in ev:
1296         raise Exception("Did not receive Venue Name")
1297
1298     ev = dev[0].wait_event(["ANQP-QUERY-DONE"], timeout=10)
1299     if ev is None:
1300         raise Exception("ANQP-QUERY-DONE event not seen")
1301     if "result=SUCCESS" not in ev:
1302         raise Exception("Unexpected result: " + ev)
1303
1304     out = run_tshark(os.path.join(params['logdir'], "hwsim0.pcapng"),
1305                      "wlan_mgt.fixed.category_code == 4 && (wlan_mgt.fixed.publicact == 0x0a || wlan_mgt.fixed.publicact == 0x0b)",
1306                      display=["wlan.bssid"])
1307     res = out.splitlines()
1308     if len(res) != 2:
1309         raise Exception("Unexpected number of GAS frames")
1310     if res[0] != bssid:
1311         raise Exception("GAS request used unexpected Address3 field value: " + res[0])
1312     if res[1] != 'ff:ff:ff:ff:ff:ff':
1313         raise Exception("GAS response used unexpected Address3 field value: " + res[1])
1314
1315 def test_gas_anqp_address3_ap_non_compliant(dev, apdev, params):
1316     """GAS/ANQP query using IEEE 802.11 non-compliant Address 3 (AP)"""
1317     try:
1318         _test_gas_anqp_address3_ap_non_compliant(dev, apdev, params)
1319     finally:
1320         dev[0].request("SET gas_address3 0")
1321
1322 def _test_gas_anqp_address3_ap_non_compliant(dev, apdev, params):
1323     hapd = start_ap(apdev[0])
1324     bssid = apdev[0]['bssid']
1325     hapd.set("gas_address3", "2")
1326
1327     if "OK" not in dev[0].request("SET gas_address3 1"):
1328         raise Exception("Failed to set gas_address3")
1329
1330     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
1331     if "OK" not in dev[0].request("ANQP_GET " + bssid + " 258"):
1332         raise Exception("ANQP_GET command failed")
1333
1334     ev = dev[0].wait_event(["GAS-QUERY-START"], timeout=5)
1335     if ev is None:
1336         raise Exception("GAS query start timed out")
1337
1338     ev = dev[0].wait_event(["GAS-QUERY-DONE"], timeout=10)
1339     if ev is None:
1340         raise Exception("GAS query timed out")
1341
1342     ev = dev[0].wait_event(["RX-ANQP"], timeout=1)
1343     if ev is None or "Venue Name" not in ev:
1344         raise Exception("Did not receive Venue Name")
1345
1346     ev = dev[0].wait_event(["ANQP-QUERY-DONE"], timeout=10)
1347     if ev is None:
1348         raise Exception("ANQP-QUERY-DONE event not seen")
1349     if "result=SUCCESS" not in ev:
1350         raise Exception("Unexpected result: " + ev)
1351
1352     out = run_tshark(os.path.join(params['logdir'], "hwsim0.pcapng"),
1353                      "wlan_mgt.fixed.category_code == 4 && (wlan_mgt.fixed.publicact == 0x0a || wlan_mgt.fixed.publicact == 0x0b)",
1354                      display=["wlan.bssid"])
1355     res = out.splitlines()
1356     if len(res) != 2:
1357         raise Exception("Unexpected number of GAS frames")
1358     if res[0] != 'ff:ff:ff:ff:ff:ff':
1359         raise Exception("GAS request used unexpected Address3 field value: " + res[0])
1360     if res[1] != bssid:
1361         raise Exception("GAS response used unexpected Address3 field value: " + res[1])
1362
1363 def test_gas_prot_vs_not_prot(dev, apdev, params):
1364     """GAS/ANQP query protected vs. not protected"""
1365     hapd = start_ap(apdev[0])
1366     bssid = apdev[0]['bssid']
1367
1368     dev[0].scan_for_bss(bssid, freq="2412")
1369     dev[0].connect("test-gas", key_mgmt="WPA-EAP", eap="TTLS",
1370                    identity="DOMAIN\mschapv2 user", anonymous_identity="ttls",
1371                    password="password", phase2="auth=MSCHAPV2",
1372                    ca_cert="auth_serv/ca.pem", scan_freq="2412",
1373                    ieee80211w="2")
1374
1375     if "OK" not in dev[0].request("ANQP_GET " + bssid + " 258"):
1376         raise Exception("ANQP_GET command failed")
1377
1378     ev = dev[0].wait_event(["GAS-QUERY-DONE"], timeout=5)
1379     if ev is None:
1380         raise Exception("No GAS-QUERY-DONE event")
1381     if "result=SUCCESS" not in ev:
1382         raise Exception("Unexpected GAS result: " + ev)
1383
1384     # GAS: Drop unexpected unprotected GAS frame when PMF is enabled
1385     dev[0].request("SET ext_mgmt_frame_handling 1")
1386     res = dev[0].request("MGMT_RX_PROCESS freq=2412 datarate=0 ssi_signal=-30 frame=d0003a010200000000000200000003000200000003001000040b00000005006c027f000000")
1387     dev[0].request("SET ext_mgmt_frame_handling 0")
1388     if "OK" not in res:
1389         raise Exception("MGMT_RX_PROCESS failed")
1390
1391     dev[0].request("DISCONNECT")
1392     dev[0].wait_disconnected()
1393
1394     # GAS: No pending query found for 02:00:00:00:03:00 dialog token 0
1395     dev[0].request("SET ext_mgmt_frame_handling 1")
1396     res = dev[0].request("MGMT_RX_PROCESS freq=2412 datarate=0 ssi_signal=-30 frame=d0003a010200000000000200000003000200000003001000040b00000005006c027f000000")
1397     dev[0].request("SET ext_mgmt_frame_handling 0")
1398     if "OK" not in res:
1399         raise Exception("MGMT_RX_PROCESS failed")
1400
1401     # GAS: Drop unexpected protected GAS frame when PMF is disabled
1402     dev[0].request("SET ext_mgmt_frame_handling 1")
1403     res = dev[0].request("MGMT_RX_PROCESS freq=2412 datarate=0 ssi_signal=-30 frame=d0003a010200000000000200000003000200000003001000090b00000005006c027f000000")
1404     dev[0].request("SET ext_mgmt_frame_handling 0")
1405     if "OK" not in res:
1406         raise Exception("MGMT_RX_PROCESS failed")
1407
1408 def test_gas_failures(dev, apdev):
1409     """GAS failure cases"""
1410     hapd = start_ap(apdev[0])
1411     hapd.set("gas_comeback_delay", "5")
1412     bssid = apdev[0]['bssid']
1413
1414     hapd2 = start_ap(apdev[1])
1415     bssid2 = apdev[1]['bssid']
1416
1417     dev[0].scan_for_bss(bssid, freq="2412", force_scan=True)
1418     dev[0].scan_for_bss(bssid2, freq="2412")
1419
1420     tests = [ (bssid, "gas_build_req;gas_query_tx_comeback_req"),
1421               (bssid, "gas_query_tx;gas_query_tx_comeback_req"),
1422               (bssid, "gas_query_append;gas_query_rx_comeback"),
1423               (bssid2, "gas_query_append;gas_query_rx_initial"),
1424               (bssid2, "wpabuf_alloc_copy;gas_query_rx_initial"),
1425               (bssid, "gas_query_tx;gas_query_tx_initial_req") ]
1426     for addr,func in tests:
1427         with alloc_fail(dev[0], 1, func):
1428             dev[0].request("ANQP_GET " + addr + " 258")
1429             ev = dev[0].wait_event(["GAS-QUERY-DONE"], timeout=5)
1430             if ev is None:
1431                 raise Exception("No GAS-QUERY-DONE seen")
1432             if "result=INTERNAL_ERROR" not in ev:
1433                 raise Exception("Unexpected result code: " + ev)
1434         dev[0].dump_monitor()
1435
1436     tests = [ "=gas_query_req", "radio_add_work;gas_query_req" ]
1437     for func in tests:
1438         with alloc_fail(dev[0], 1, func):
1439             if "FAIL" not in dev[0].request("ANQP_GET " + bssid + " 258"):
1440                 raise Exception("ANQP_GET succeeded unexpectedly during OOM")
1441         dev[0].dump_monitor()
1442
1443     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
1444     wpas.interface_add("wlan5")
1445     wpas.scan_for_bss(bssid2, freq="2412")
1446     wpas.request("SET preassoc_mac_addr 1111")
1447     wpas.request("ANQP_GET " + bssid2 + " 258")
1448     ev = wpas.wait_event(["Failed to assign random MAC address for GAS"],
1449                          timeout=5)
1450     wpas.request("SET preassoc_mac_addr 0")
1451     if ev is None:
1452         raise Exception("No random MAC address error seen")