HS 2.0R2: Try to scan multiple times for OSU providers
[mech_eap.git] / wpa_supplicant / hs20_supplicant.c
1 /*
2  * Copyright (c) 2009, Atheros Communications, Inc.
3  * Copyright (c) 2011-2013, Qualcomm Atheros, Inc.
4  *
5  * This software may be distributed under the terms of the BSD license.
6  * See README for more details.
7  */
8
9 #include "includes.h"
10
11 #include "common.h"
12 #include "eloop.h"
13 #include "common/ieee802_11_common.h"
14 #include "common/ieee802_11_defs.h"
15 #include "common/gas.h"
16 #include "common/wpa_ctrl.h"
17 #include "rsn_supp/wpa.h"
18 #include "wpa_supplicant_i.h"
19 #include "driver_i.h"
20 #include "config.h"
21 #include "scan.h"
22 #include "bss.h"
23 #include "blacklist.h"
24 #include "gas_query.h"
25 #include "interworking.h"
26 #include "hs20_supplicant.h"
27
28
29 #define OSU_MAX_ITEMS 10
30
31 struct osu_lang_string {
32         char lang[4];
33         char text[253];
34 };
35
36 struct osu_icon {
37         u16 width;
38         u16 height;
39         char lang[4];
40         char icon_type[256];
41         char filename[256];
42         unsigned int id;
43         unsigned int failed:1;
44 };
45
46 struct osu_provider {
47         u8 bssid[ETH_ALEN];
48         u8 osu_ssid[32];
49         u8 osu_ssid_len;
50         char server_uri[256];
51         u32 osu_methods; /* bit 0 = OMA-DM, bit 1 = SOAP-XML SPP */
52         char osu_nai[256];
53         struct osu_lang_string friendly_name[OSU_MAX_ITEMS];
54         size_t friendly_name_count;
55         struct osu_lang_string serv_desc[OSU_MAX_ITEMS];
56         size_t serv_desc_count;
57         struct osu_icon icon[OSU_MAX_ITEMS];
58         size_t icon_count;
59 };
60
61
62 void wpas_hs20_add_indication(struct wpabuf *buf, int pps_mo_id)
63 {
64         u8 conf;
65
66         wpabuf_put_u8(buf, WLAN_EID_VENDOR_SPECIFIC);
67         wpabuf_put_u8(buf, pps_mo_id >= 0 ? 7 : 5);
68         wpabuf_put_be24(buf, OUI_WFA);
69         wpabuf_put_u8(buf, HS20_INDICATION_OUI_TYPE);
70         conf = HS20_VERSION;
71         if (pps_mo_id >= 0)
72                 conf |= HS20_PPS_MO_ID_PRESENT;
73         wpabuf_put_u8(buf, conf);
74         if (pps_mo_id >= 0)
75                 wpabuf_put_le16(buf, pps_mo_id);
76 }
77
78
79 int is_hs20_network(struct wpa_supplicant *wpa_s, struct wpa_ssid *ssid,
80                     struct wpa_bss *bss)
81 {
82         if (!wpa_s->conf->hs20 || !ssid)
83                 return 0;
84
85         if (ssid->parent_cred)
86                 return 1;
87
88         if (bss && !wpa_bss_get_vendor_ie(bss, HS20_IE_VENDOR_TYPE))
89                 return 0;
90
91         /*
92          * This may catch some non-Hotspot 2.0 cases, but it is safer to do that
93          * than cause Hotspot 2.0 connections without indication element getting
94          * added. Non-Hotspot 2.0 APs should ignore the unknown vendor element.
95          */
96
97         if (!(ssid->key_mgmt & WPA_KEY_MGMT_IEEE8021X))
98                 return 0;
99         if (!(ssid->pairwise_cipher & WPA_CIPHER_CCMP))
100                 return 0;
101         if (ssid->proto != WPA_PROTO_RSN)
102                 return 0;
103
104         return 1;
105 }
106
107
108 int hs20_get_pps_mo_id(struct wpa_supplicant *wpa_s, struct wpa_ssid *ssid)
109 {
110         struct wpa_cred *cred;
111
112         if (ssid == NULL || ssid->parent_cred == NULL)
113                 return 0;
114
115         for (cred = wpa_s->conf->cred; cred; cred = cred->next) {
116                 if (ssid->parent_cred == cred)
117                         return cred->update_identifier;
118         }
119
120         return 0;
121 }
122
123
124 struct wpabuf * hs20_build_anqp_req(u32 stypes, const u8 *payload,
125                                     size_t payload_len)
126 {
127         struct wpabuf *buf;
128         u8 *len_pos;
129
130         buf = gas_anqp_build_initial_req(0, 100 + payload_len);
131         if (buf == NULL)
132                 return NULL;
133
134         len_pos = gas_anqp_add_element(buf, ANQP_VENDOR_SPECIFIC);
135         wpabuf_put_be24(buf, OUI_WFA);
136         wpabuf_put_u8(buf, HS20_ANQP_OUI_TYPE);
137         if (stypes == BIT(HS20_STYPE_NAI_HOME_REALM_QUERY)) {
138                 wpabuf_put_u8(buf, HS20_STYPE_NAI_HOME_REALM_QUERY);
139                 wpabuf_put_u8(buf, 0); /* Reserved */
140                 if (payload)
141                         wpabuf_put_data(buf, payload, payload_len);
142         } else if (stypes == BIT(HS20_STYPE_ICON_REQUEST)) {
143                 wpabuf_put_u8(buf, HS20_STYPE_ICON_REQUEST);
144                 wpabuf_put_u8(buf, 0); /* Reserved */
145                 if (payload)
146                         wpabuf_put_data(buf, payload, payload_len);
147         } else {
148                 u8 i;
149                 wpabuf_put_u8(buf, HS20_STYPE_QUERY_LIST);
150                 wpabuf_put_u8(buf, 0); /* Reserved */
151                 for (i = 0; i < 32; i++) {
152                         if (stypes & BIT(i))
153                                 wpabuf_put_u8(buf, i);
154                 }
155         }
156         gas_anqp_set_element_len(buf, len_pos);
157
158         gas_anqp_set_len(buf);
159
160         return buf;
161 }
162
163
164 int hs20_anqp_send_req(struct wpa_supplicant *wpa_s, const u8 *dst, u32 stypes,
165                        const u8 *payload, size_t payload_len)
166 {
167         struct wpabuf *buf;
168         int ret = 0;
169         int freq;
170         struct wpa_bss *bss;
171         int res;
172
173         freq = wpa_s->assoc_freq;
174         bss = wpa_bss_get_bssid(wpa_s, dst);
175         if (bss) {
176                 wpa_bss_anqp_unshare_alloc(bss);
177                 freq = bss->freq;
178         }
179         if (freq <= 0)
180                 return -1;
181
182         wpa_printf(MSG_DEBUG, "HS20: ANQP Query Request to " MACSTR " for "
183                    "subtypes 0x%x", MAC2STR(dst), stypes);
184
185         buf = hs20_build_anqp_req(stypes, payload, payload_len);
186         if (buf == NULL)
187                 return -1;
188
189         res = gas_query_req(wpa_s->gas, dst, freq, buf, anqp_resp_cb, wpa_s);
190         if (res < 0) {
191                 wpa_printf(MSG_DEBUG, "ANQP: Failed to send Query Request");
192                 wpabuf_free(buf);
193                 ret = -1;
194         } else
195                 wpa_printf(MSG_DEBUG, "ANQP: Query started with dialog token "
196                            "%u", res);
197
198         return ret;
199 }
200
201
202 static int hs20_process_icon_binary_file(struct wpa_supplicant *wpa_s,
203                                          const u8 *sa, const u8 *pos,
204                                          size_t slen)
205 {
206         char fname[256];
207         int png;
208         FILE *f;
209         u16 data_len;
210
211         wpa_msg(wpa_s, MSG_INFO, "RX-HS20-ANQP " MACSTR " Icon Binary File",
212                 MAC2STR(sa));
213
214         if (slen < 4) {
215                 wpa_dbg(wpa_s, MSG_DEBUG, "HS 2.0: Too short Icon Binary File "
216                         "value from " MACSTR, MAC2STR(sa));
217                 return -1;
218         }
219
220         wpa_printf(MSG_DEBUG, "HS 2.0: Download Status Code %u", *pos);
221         if (*pos != 0)
222                 return -1;
223         pos++;
224         slen--;
225
226         if ((size_t) 1 + pos[0] > slen) {
227                 wpa_dbg(wpa_s, MSG_DEBUG, "HS 2.0: Too short Icon Binary File "
228                         "value from " MACSTR, MAC2STR(sa));
229                 return -1;
230         }
231         wpa_hexdump_ascii(MSG_DEBUG, "Icon Type", pos + 1, pos[0]);
232         png = os_strncasecmp((char *) pos + 1, "image/png", 9) == 0;
233         slen -= 1 + pos[0];
234         pos += 1 + pos[0];
235
236         if (slen < 2) {
237                 wpa_dbg(wpa_s, MSG_DEBUG, "HS 2.0: Too short Icon Binary File "
238                         "value from " MACSTR, MAC2STR(sa));
239                 return -1;
240         }
241         data_len = WPA_GET_LE16(pos);
242         pos += 2;
243         slen -= 2;
244
245         if (data_len > slen) {
246                 wpa_dbg(wpa_s, MSG_DEBUG, "HS 2.0: Too short Icon Binary File "
247                         "value from " MACSTR, MAC2STR(sa));
248                 return -1;
249         }
250
251         wpa_printf(MSG_DEBUG, "Icon Binary Data: %u bytes", data_len);
252         if (wpa_s->conf->osu_dir == NULL)
253                 return -1;
254
255         wpa_s->osu_icon_id++;
256         if (wpa_s->osu_icon_id == 0)
257                 wpa_s->osu_icon_id++;
258         snprintf(fname, sizeof(fname), "%s/osu-icon-%u.%s",
259                  wpa_s->conf->osu_dir, wpa_s->osu_icon_id,
260                  png ? "png" : "icon");
261         f = fopen(fname, "wb");
262         if (f == NULL)
263                 return -1;
264         if (fwrite(pos, slen, 1, f) != 1) {
265                 fclose(f);
266                 unlink(fname);
267                 return -1;
268         }
269         fclose(f);
270
271         wpa_msg(wpa_s, MSG_INFO, "RX-HS20-ANQP-ICON %s", fname);
272         return 0;
273 }
274
275
276 static void hs20_continue_icon_fetch(void *eloop_ctx, void *sock_ctx)
277 {
278         struct wpa_supplicant *wpa_s = eloop_ctx;
279         if (wpa_s->fetch_osu_icon_in_progress)
280                 hs20_next_osu_icon(wpa_s);
281 }
282
283
284 static void hs20_osu_icon_fetch_result(struct wpa_supplicant *wpa_s, int res)
285 {
286         size_t i, j;
287         struct os_reltime now, tmp;
288         int dur;
289
290         os_get_reltime(&now);
291         os_reltime_sub(&now, &wpa_s->osu_icon_fetch_start, &tmp);
292         dur = tmp.sec * 1000 + tmp.usec / 1000;
293         wpa_printf(MSG_DEBUG, "HS 2.0: Icon fetch dur=%d ms res=%d",
294                    dur, res);
295
296         for (i = 0; i < wpa_s->osu_prov_count; i++) {
297                 struct osu_provider *osu = &wpa_s->osu_prov[i];
298                 for (j = 0; j < osu->icon_count; j++) {
299                         struct osu_icon *icon = &osu->icon[j];
300                         if (icon->id || icon->failed)
301                                 continue;
302                         if (res < 0)
303                                 icon->failed = 1;
304                         else
305                                 icon->id = wpa_s->osu_icon_id;
306                         return;
307                 }
308         }
309 }
310
311
312 void hs20_parse_rx_hs20_anqp_resp(struct wpa_supplicant *wpa_s,
313                                   const u8 *sa, const u8 *data, size_t slen)
314 {
315         const u8 *pos = data;
316         u8 subtype;
317         struct wpa_bss *bss = wpa_bss_get_bssid(wpa_s, sa);
318         struct wpa_bss_anqp *anqp = NULL;
319         int ret;
320
321         if (slen < 2)
322                 return;
323
324         if (bss)
325                 anqp = bss->anqp;
326
327         subtype = *pos++;
328         slen--;
329
330         pos++; /* Reserved */
331         slen--;
332
333         switch (subtype) {
334         case HS20_STYPE_CAPABILITY_LIST:
335                 wpa_msg(wpa_s, MSG_INFO, "RX-HS20-ANQP " MACSTR
336                         " HS Capability List", MAC2STR(sa));
337                 wpa_hexdump_ascii(MSG_DEBUG, "HS Capability List", pos, slen);
338                 break;
339         case HS20_STYPE_OPERATOR_FRIENDLY_NAME:
340                 wpa_msg(wpa_s, MSG_INFO, "RX-HS20-ANQP " MACSTR
341                         " Operator Friendly Name", MAC2STR(sa));
342                 wpa_hexdump_ascii(MSG_DEBUG, "oper friendly name", pos, slen);
343                 if (anqp) {
344                         wpabuf_free(anqp->hs20_operator_friendly_name);
345                         anqp->hs20_operator_friendly_name =
346                                 wpabuf_alloc_copy(pos, slen);
347                 }
348                 break;
349         case HS20_STYPE_WAN_METRICS:
350                 wpa_hexdump(MSG_DEBUG, "WAN Metrics", pos, slen);
351                 if (slen < 13) {
352                         wpa_dbg(wpa_s, MSG_DEBUG, "HS 2.0: Too short WAN "
353                                 "Metrics value from " MACSTR, MAC2STR(sa));
354                         break;
355                 }
356                 wpa_msg(wpa_s, MSG_INFO, "RX-HS20-ANQP " MACSTR
357                         " WAN Metrics %02x:%u:%u:%u:%u:%u", MAC2STR(sa),
358                         pos[0], WPA_GET_LE32(pos + 1), WPA_GET_LE32(pos + 5),
359                         pos[9], pos[10], WPA_GET_LE16(pos + 11));
360                 if (anqp) {
361                         wpabuf_free(anqp->hs20_wan_metrics);
362                         anqp->hs20_wan_metrics = wpabuf_alloc_copy(pos, slen);
363                 }
364                 break;
365         case HS20_STYPE_CONNECTION_CAPABILITY:
366                 wpa_msg(wpa_s, MSG_INFO, "RX-HS20-ANQP " MACSTR
367                         " Connection Capability", MAC2STR(sa));
368                 wpa_hexdump_ascii(MSG_DEBUG, "conn capability", pos, slen);
369                 if (anqp) {
370                         wpabuf_free(anqp->hs20_connection_capability);
371                         anqp->hs20_connection_capability =
372                                 wpabuf_alloc_copy(pos, slen);
373                 }
374                 break;
375         case HS20_STYPE_OPERATING_CLASS:
376                 wpa_msg(wpa_s, MSG_INFO, "RX-HS20-ANQP " MACSTR
377                         " Operating Class", MAC2STR(sa));
378                 wpa_hexdump_ascii(MSG_DEBUG, "Operating Class", pos, slen);
379                 if (anqp) {
380                         wpabuf_free(anqp->hs20_operating_class);
381                         anqp->hs20_operating_class =
382                                 wpabuf_alloc_copy(pos, slen);
383                 }
384                 break;
385         case HS20_STYPE_OSU_PROVIDERS_LIST:
386                 wpa_msg(wpa_s, MSG_INFO, "RX-HS20-ANQP " MACSTR
387                         " OSU Providers list", MAC2STR(sa));
388                 wpa_s->num_prov_found++;
389                 if (anqp) {
390                         wpabuf_free(anqp->hs20_osu_providers_list);
391                         anqp->hs20_osu_providers_list =
392                                 wpabuf_alloc_copy(pos, slen);
393                 }
394                 break;
395         case HS20_STYPE_ICON_BINARY_FILE:
396                 ret = hs20_process_icon_binary_file(wpa_s, sa, pos, slen);
397                 if (wpa_s->fetch_osu_icon_in_progress) {
398                         hs20_osu_icon_fetch_result(wpa_s, ret);
399                         eloop_cancel_timeout(hs20_continue_icon_fetch,
400                                              wpa_s, NULL);
401                         eloop_register_timeout(0, 0, hs20_continue_icon_fetch,
402                                                wpa_s, NULL);
403                 }
404                 break;
405         default:
406                 wpa_printf(MSG_DEBUG, "HS20: Unsupported subtype %u", subtype);
407                 break;
408         }
409 }
410
411
412 void hs20_notify_parse_done(struct wpa_supplicant *wpa_s)
413 {
414         if (!wpa_s->fetch_osu_icon_in_progress)
415                 return;
416         if (eloop_is_timeout_registered(hs20_continue_icon_fetch, wpa_s, NULL))
417                 return;
418         /*
419          * We are going through icon fetch, but no icon response was received.
420          * Assume this means the current AP could not provide an answer to avoid
421          * getting stuck in fetch iteration.
422          */
423         hs20_icon_fetch_failed(wpa_s);
424 }
425
426
427 static void hs20_free_osu_prov_entry(struct osu_provider *prov)
428 {
429 }
430
431
432 void hs20_free_osu_prov(struct wpa_supplicant *wpa_s)
433 {
434         size_t i;
435         for (i = 0; i < wpa_s->osu_prov_count; i++)
436                 hs20_free_osu_prov_entry(&wpa_s->osu_prov[i]);
437         os_free(wpa_s->osu_prov);
438         wpa_s->osu_prov = NULL;
439         wpa_s->osu_prov_count = 0;
440 }
441
442
443 static void hs20_osu_fetch_done(struct wpa_supplicant *wpa_s)
444 {
445         char fname[256];
446         FILE *f;
447         size_t i, j;
448
449         wpa_s->fetch_osu_info = 0;
450         wpa_s->fetch_osu_icon_in_progress = 0;
451
452         if (wpa_s->conf->osu_dir == NULL) {
453                 hs20_free_osu_prov(wpa_s);
454                 wpa_s->fetch_anqp_in_progress = 0;
455                 return;
456         }
457
458         snprintf(fname, sizeof(fname), "%s/osu-providers.txt",
459                  wpa_s->conf->osu_dir);
460         f = fopen(fname, "w");
461         if (f == NULL) {
462                 hs20_free_osu_prov(wpa_s);
463                 return;
464         }
465         for (i = 0; i < wpa_s->osu_prov_count; i++) {
466                 struct osu_provider *osu = &wpa_s->osu_prov[i];
467                 if (i > 0)
468                         fprintf(f, "\n");
469                 fprintf(f, "OSU-PROVIDER " MACSTR "\n"
470                         "uri=%s\n"
471                         "methods=%08x\n",
472                         MAC2STR(osu->bssid), osu->server_uri, osu->osu_methods);
473                 if (osu->osu_ssid_len) {
474                         fprintf(f, "osu_ssid=%s\n",
475                                 wpa_ssid_txt(osu->osu_ssid,
476                                              osu->osu_ssid_len));
477                 }
478                 if (osu->osu_nai[0])
479                         fprintf(f, "osu_nai=%s\n", osu->osu_nai);
480                 for (j = 0; j < osu->friendly_name_count; j++) {
481                         fprintf(f, "friendly_name=%s:%s\n",
482                                 osu->friendly_name[j].lang,
483                                 osu->friendly_name[j].text);
484                 }
485                 for (j = 0; j < osu->serv_desc_count; j++) {
486                         fprintf(f, "desc=%s:%s\n",
487                                 osu->serv_desc[j].lang,
488                                 osu->serv_desc[j].text);
489                 }
490                 for (j = 0; j < osu->icon_count; j++) {
491                         struct osu_icon *icon = &osu->icon[j];
492                         if (icon->failed)
493                                 continue; /* could not fetch icon */
494                         fprintf(f, "icon=%u:%u:%u:%s:%s:%s\n",
495                                 icon->id, icon->width, icon->height, icon->lang,
496                                 icon->icon_type, icon->filename);
497                 }
498         }
499         fclose(f);
500         hs20_free_osu_prov(wpa_s);
501
502         wpa_msg(wpa_s, MSG_INFO, "OSU provider fetch completed");
503         wpa_s->fetch_anqp_in_progress = 0;
504 }
505
506
507 void hs20_next_osu_icon(struct wpa_supplicant *wpa_s)
508 {
509         size_t i, j;
510
511         wpa_printf(MSG_DEBUG, "HS 2.0: Ready to fetch next icon");
512
513         for (i = 0; i < wpa_s->osu_prov_count; i++) {
514                 struct osu_provider *osu = &wpa_s->osu_prov[i];
515                 for (j = 0; j < osu->icon_count; j++) {
516                         struct osu_icon *icon = &osu->icon[j];
517                         if (icon->id || icon->failed)
518                                 continue;
519
520                         wpa_printf(MSG_DEBUG, "HS 2.0: Try to fetch icon '%s' "
521                                    "from " MACSTR, icon->filename,
522                                    MAC2STR(osu->bssid));
523                         os_get_reltime(&wpa_s->osu_icon_fetch_start);
524                         if (hs20_anqp_send_req(wpa_s, osu->bssid,
525                                                BIT(HS20_STYPE_ICON_REQUEST),
526                                                (u8 *) icon->filename,
527                                                os_strlen(icon->filename)) < 0) {
528                                 icon->failed = 1;
529                                 continue;
530                         }
531                         return;
532                 }
533         }
534
535         wpa_printf(MSG_DEBUG, "HS 2.0: No more icons to fetch");
536         hs20_osu_fetch_done(wpa_s);
537 }
538
539
540 static void hs20_osu_add_prov(struct wpa_supplicant *wpa_s, struct wpa_bss *bss,
541                               const u8 *osu_ssid, u8 osu_ssid_len,
542                               const u8 *pos, size_t len)
543 {
544         struct osu_provider *prov;
545         const u8 *end = pos + len;
546         u16 len2;
547         const u8 *pos2;
548
549         wpa_hexdump(MSG_DEBUG, "HS 2.0: Parsing OSU Provider", pos, len);
550         prov = os_realloc_array(wpa_s->osu_prov,
551                                 wpa_s->osu_prov_count + 1,
552                                 sizeof(*prov));
553         if (prov == NULL)
554                 return;
555         wpa_s->osu_prov = prov;
556         prov = &prov[wpa_s->osu_prov_count];
557         os_memset(prov, 0, sizeof(*prov));
558
559         os_memcpy(prov->bssid, bss->bssid, ETH_ALEN);
560         os_memcpy(prov->osu_ssid, osu_ssid, osu_ssid_len);
561         prov->osu_ssid_len = osu_ssid_len;
562
563         /* OSU Friendly Name Length */
564         if (pos + 2 > end) {
565                 wpa_printf(MSG_DEBUG, "HS 2.0: Not enough room for OSU "
566                            "Friendly Name Length");
567                 return;
568         }
569         len2 = WPA_GET_LE16(pos);
570         pos += 2;
571         if (pos + len2 > end) {
572                 wpa_printf(MSG_DEBUG, "HS 2.0: Not enough room for OSU "
573                            "Friendly Name Duples");
574                 return;
575         }
576         pos2 = pos;
577         pos += len2;
578
579         /* OSU Friendly Name Duples */
580         while (pos2 + 4 <= pos && prov->friendly_name_count < OSU_MAX_ITEMS) {
581                 struct osu_lang_string *f;
582                 if (pos2 + 1 + pos2[0] > pos || pos2[0] < 3) {
583                         wpa_printf(MSG_DEBUG, "Invalid OSU Friendly Name");
584                         break;
585                 }
586                 f = &prov->friendly_name[prov->friendly_name_count++];
587                 os_memcpy(f->lang, pos2 + 1, 3);
588                 os_memcpy(f->text, pos2 + 1 + 3, pos2[0] - 3);
589                 pos2 += 1 + pos2[0];
590         }
591
592         /* OSU Server URI */
593         if (pos + 1 > end || pos + 1 + pos[0] > end) {
594                 wpa_printf(MSG_DEBUG, "HS 2.0: Not enough room for OSU Server "
595                            "URI");
596                 return;
597         }
598         os_memcpy(prov->server_uri, pos + 1, pos[0]);
599         pos += 1 + pos[0];
600
601         /* OSU Method list */
602         if (pos + 1 > end || pos + 1 + pos[0] > end) {
603                 wpa_printf(MSG_DEBUG, "HS 2.0: Not enough room for OSU Method "
604                            "list");
605                 return;
606         }
607         pos2 = pos + 1;
608         pos += 1 + pos[0];
609         while (pos2 < pos) {
610                 if (*pos2 < 32)
611                         prov->osu_methods |= BIT(*pos2);
612                 pos2++;
613         }
614
615         /* Icons Available Length */
616         if (pos + 2 > end) {
617                 wpa_printf(MSG_DEBUG, "HS 2.0: Not enough room for Icons "
618                            "Available Length");
619                 return;
620         }
621         len2 = WPA_GET_LE16(pos);
622         pos += 2;
623         if (pos + len2 > end) {
624                 wpa_printf(MSG_DEBUG, "HS 2.0: Not enough room for Icons "
625                            "Available");
626                 return;
627         }
628         pos2 = pos;
629         pos += len2;
630
631         /* Icons Available */
632         while (pos2 < pos) {
633                 struct osu_icon *icon = &prov->icon[prov->icon_count];
634                 if (pos2 + 2 + 2 + 3 + 1 + 1 > pos) {
635                         wpa_printf(MSG_DEBUG, "HS 2.0: Invalid Icon Metadata");
636                         break;
637                 }
638
639                 icon->width = WPA_GET_LE16(pos2);
640                 pos2 += 2;
641                 icon->height = WPA_GET_LE16(pos2);
642                 pos2 += 2;
643                 os_memcpy(icon->lang, pos2, 3);
644                 pos2 += 3;
645
646                 if (pos2 + 1 + pos2[0] > pos) {
647                         wpa_printf(MSG_DEBUG, "HS 2.0: Not room for Icon Type");
648                         break;
649                 }
650                 os_memcpy(icon->icon_type, pos2 + 1, pos2[0]);
651                 pos2 += 1 + pos2[0];
652
653                 if (pos2 + 1 + pos2[0] > pos) {
654                         wpa_printf(MSG_DEBUG, "HS 2.0: Not room for Icon "
655                                    "Filename");
656                         break;
657                 }
658                 os_memcpy(icon->filename, pos2 + 1, pos2[0]);
659                 pos2 += 1 + pos2[0];
660
661                 prov->icon_count++;
662         }
663
664         /* OSU_NAI */
665         if (pos + 1 > end || pos + 1 + pos[0] > end) {
666                 wpa_printf(MSG_DEBUG, "HS 2.0: Not enough room for OSU_NAI");
667                 return;
668         }
669         os_memcpy(prov->osu_nai, pos + 1, pos[0]);
670         pos += 1 + pos[0];
671
672         /* OSU Service Description Length */
673         if (pos + 2 > end) {
674                 wpa_printf(MSG_DEBUG, "HS 2.0: Not enough room for OSU "
675                            "Service Description Length");
676                 return;
677         }
678         len2 = WPA_GET_LE16(pos);
679         pos += 2;
680         if (pos + len2 > end) {
681                 wpa_printf(MSG_DEBUG, "HS 2.0: Not enough room for OSU "
682                            "Service Description Duples");
683                 return;
684         }
685         pos2 = pos;
686         pos += len2;
687
688         /* OSU Service Description Duples */
689         while (pos2 + 4 <= pos && prov->serv_desc_count < OSU_MAX_ITEMS) {
690                 struct osu_lang_string *f;
691                 if (pos2 + 1 + pos2[0] > pos || pos2[0] < 3) {
692                         wpa_printf(MSG_DEBUG, "Invalid OSU Service "
693                                    "Description");
694                         break;
695                 }
696                 f = &prov->serv_desc[prov->serv_desc_count++];
697                 os_memcpy(f->lang, pos2 + 1, 3);
698                 os_memcpy(f->text, pos2 + 1 + 3, pos2[0] - 3);
699                 pos2 += 1 + pos2[0];
700         }
701
702         wpa_printf(MSG_DEBUG, "HS 2.0: Added OSU Provider through " MACSTR,
703                    MAC2STR(bss->bssid));
704         wpa_s->osu_prov_count++;
705 }
706
707
708 void hs20_osu_icon_fetch(struct wpa_supplicant *wpa_s)
709 {
710         struct wpa_bss *bss;
711         struct wpabuf *prov_anqp;
712         const u8 *pos, *end;
713         u16 len;
714         const u8 *osu_ssid;
715         u8 osu_ssid_len;
716         u8 num_providers;
717
718         hs20_free_osu_prov(wpa_s);
719
720         dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
721                 if (bss->anqp == NULL)
722                         continue;
723                 prov_anqp = bss->anqp->hs20_osu_providers_list;
724                 if (prov_anqp == NULL)
725                         continue;
726                 wpa_printf(MSG_DEBUG, "HS 2.0: Parsing OSU Providers list from "
727                            MACSTR, MAC2STR(bss->bssid));
728                 wpa_hexdump_buf(MSG_DEBUG, "HS 2.0: OSU Providers list",
729                                 prov_anqp);
730                 pos = wpabuf_head(prov_anqp);
731                 end = pos + wpabuf_len(prov_anqp);
732
733                 /* OSU SSID */
734                 if (pos + 1 > end)
735                         continue;
736                 if (pos + 1 + pos[0] > end) {
737                         wpa_printf(MSG_DEBUG, "HS 2.0: Not enough room for "
738                                    "OSU SSID");
739                         continue;
740                 }
741                 osu_ssid_len = *pos++;
742                 if (osu_ssid_len > 32) {
743                         wpa_printf(MSG_DEBUG, "HS 2.0: Invalid OSU SSID "
744                                    "Length %u", osu_ssid_len);
745                         continue;
746                 }
747                 osu_ssid = pos;
748                 pos += osu_ssid_len;
749
750                 if (pos + 1 > end) {
751                         wpa_printf(MSG_DEBUG, "HS 2.0: Not enough room for "
752                                    "Number of OSU Providers");
753                         continue;
754                 }
755                 num_providers = *pos++;
756                 wpa_printf(MSG_DEBUG, "HS 2.0: Number of OSU Providers: %u",
757                            num_providers);
758
759                 /* OSU Providers */
760                 while (pos + 2 < end && num_providers > 0) {
761                         num_providers--;
762                         len = WPA_GET_LE16(pos);
763                         pos += 2;
764                         if (pos + len > end)
765                                 break;
766                         hs20_osu_add_prov(wpa_s, bss, osu_ssid,
767                                           osu_ssid_len, pos, len);
768                         pos += len;
769                 }
770
771                 if (pos != end) {
772                         wpa_printf(MSG_DEBUG, "HS 2.0: Ignored %d bytes of "
773                                    "extra data after OSU Providers",
774                                    (int) (end - pos));
775                 }
776         }
777
778         wpa_s->fetch_osu_icon_in_progress = 1;
779         hs20_next_osu_icon(wpa_s);
780 }
781
782
783 static void hs20_osu_scan_res_handler(struct wpa_supplicant *wpa_s,
784                                       struct wpa_scan_results *scan_res)
785 {
786         wpa_printf(MSG_DEBUG, "OSU provisioning fetch scan completed");
787         wpa_s->network_select = 0;
788         wpa_s->fetch_all_anqp = 1;
789         wpa_s->fetch_osu_info = 1;
790         wpa_s->fetch_osu_icon_in_progress = 0;
791
792         interworking_start_fetch_anqp(wpa_s);
793 }
794
795
796 int hs20_fetch_osu(struct wpa_supplicant *wpa_s)
797 {
798         if (wpa_s->wpa_state == WPA_INTERFACE_DISABLED) {
799                 wpa_printf(MSG_DEBUG, "HS 2.0: Cannot start fetch_osu - "
800                            "interface disabled");
801                 return -1;
802         }
803
804         if (wpa_s->scanning) {
805                 wpa_printf(MSG_DEBUG, "HS 2.0: Cannot start fetch_osu - "
806                            "scanning");
807                 return -1;
808         }
809
810         if (wpa_s->conf->osu_dir == NULL) {
811                 wpa_printf(MSG_DEBUG, "HS 2.0: Cannot start fetch_osu - "
812                            "osu_dir not configured");
813                 return -1;
814         }
815
816         if (wpa_s->fetch_anqp_in_progress || wpa_s->network_select) {
817                 wpa_printf(MSG_DEBUG, "HS 2.0: Cannot start fetch_osu - "
818                            "fetch in progress (%d, %d)",
819                            wpa_s->fetch_anqp_in_progress,
820                            wpa_s->network_select);
821                 return -1;
822         }
823
824         wpa_msg(wpa_s, MSG_INFO, "Starting OSU provisioning information fetch");
825         wpa_s->num_osu_scans = 0;
826         wpa_s->num_prov_found = 0;
827         hs20_start_osu_scan(wpa_s);
828
829         return 0;
830 }
831
832
833 void hs20_start_osu_scan(struct wpa_supplicant *wpa_s)
834 {
835         wpa_s->num_osu_scans++;
836         wpa_s->scan_req = MANUAL_SCAN_REQ;
837         wpa_s->scan_res_handler = hs20_osu_scan_res_handler;
838         wpa_supplicant_req_scan(wpa_s, 0, 0);
839 }
840
841
842 void hs20_cancel_fetch_osu(struct wpa_supplicant *wpa_s)
843 {
844         wpa_printf(MSG_DEBUG, "Cancel OSU fetch");
845         interworking_stop_fetch_anqp(wpa_s);
846         wpa_s->network_select = 0;
847         wpa_s->fetch_osu_info = 0;
848         wpa_s->fetch_osu_icon_in_progress = 0;
849 }
850
851
852 void hs20_icon_fetch_failed(struct wpa_supplicant *wpa_s)
853 {
854         hs20_osu_icon_fetch_result(wpa_s, -1);
855         eloop_cancel_timeout(hs20_continue_icon_fetch, wpa_s, NULL);
856         eloop_register_timeout(0, 0, hs20_continue_icon_fetch, wpa_s, NULL);
857 }
858
859
860 void hs20_rx_subscription_remediation(struct wpa_supplicant *wpa_s,
861                                       const char *url, u8 osu_method)
862 {
863         if (url)
864                 wpa_msg(wpa_s, MSG_INFO, HS20_SUBSCRIPTION_REMEDIATION "%u %s",
865                         osu_method, url);
866         else
867                 wpa_msg(wpa_s, MSG_INFO, HS20_SUBSCRIPTION_REMEDIATION);
868 }
869
870
871 void hs20_rx_deauth_imminent_notice(struct wpa_supplicant *wpa_s, u8 code,
872                                     u16 reauth_delay, const char *url)
873 {
874         if (!wpa_sm_pmf_enabled(wpa_s->wpa)) {
875                 wpa_printf(MSG_DEBUG, "HS 2.0: Ignore deauthentication imminent notice since PMF was not enabled");
876                 return;
877         }
878
879         wpa_msg(wpa_s, MSG_INFO, HS20_DEAUTH_IMMINENT_NOTICE "%u %u %s",
880                 code, reauth_delay, url);
881
882         if (code == HS20_DEAUTH_REASON_CODE_BSS) {
883                 wpa_printf(MSG_DEBUG, "HS 2.0: Add BSS to blacklist");
884                 wpa_blacklist_add(wpa_s, wpa_s->bssid);
885                 /* TODO: For now, disable full ESS since some drivers may not
886                  * support disabling per BSS. */
887                 if (wpa_s->current_ssid) {
888                         struct os_time now;
889                         os_get_time(&now);
890                         if (now.sec + reauth_delay <=
891                             wpa_s->current_ssid->disabled_until.sec)
892                                 return;
893                         wpa_printf(MSG_DEBUG, "HS 2.0: Disable network for %u seconds (BSS)",
894                                    reauth_delay);
895                         wpa_s->current_ssid->disabled_until.sec =
896                                 now.sec + reauth_delay;
897                 }
898         }
899
900         if (code == HS20_DEAUTH_REASON_CODE_ESS && wpa_s->current_ssid) {
901                 struct os_time now;
902                 os_get_time(&now);
903                 if (now.sec + reauth_delay <=
904                     wpa_s->current_ssid->disabled_until.sec)
905                         return;
906                 wpa_printf(MSG_DEBUG, "HS 2.0: Disable network for %u seconds",
907                            reauth_delay);
908                 wpa_s->current_ssid->disabled_until.sec =
909                         now.sec + reauth_delay;
910         }
911 }