Interworking: Move BSS ANQP information into separate struct
[mech_eap.git] / wpa_supplicant / interworking.c
1 /*
2  * Interworking (IEEE 802.11u)
3  * Copyright (c) 2011-2012, 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 "common/ieee802_11_defs.h"
13 #include "common/gas.h"
14 #include "common/wpa_ctrl.h"
15 #include "utils/pcsc_funcs.h"
16 #include "drivers/driver.h"
17 #include "eap_common/eap_defs.h"
18 #include "eap_peer/eap.h"
19 #include "eap_peer/eap_methods.h"
20 #include "wpa_supplicant_i.h"
21 #include "config.h"
22 #include "config_ssid.h"
23 #include "bss.h"
24 #include "scan.h"
25 #include "notify.h"
26 #include "gas_query.h"
27 #include "hs20_supplicant.h"
28 #include "interworking.h"
29
30
31 #if defined(EAP_SIM) | defined(EAP_SIM_DYNAMIC)
32 #define INTERWORKING_3GPP
33 #else
34 #if defined(EAP_AKA) | defined(EAP_AKA_DYNAMIC)
35 #define INTERWORKING_3GPP
36 #else
37 #if defined(EAP_AKA_PRIME) | defined(EAP_AKA_PRIME_DYNAMIC)
38 #define INTERWORKING_3GPP
39 #endif
40 #endif
41 #endif
42
43 static void interworking_next_anqp_fetch(struct wpa_supplicant *wpa_s);
44
45
46 static void interworking_reconnect(struct wpa_supplicant *wpa_s)
47 {
48         if (wpa_s->wpa_state >= WPA_AUTHENTICATING) {
49                 wpa_supplicant_cancel_sched_scan(wpa_s);
50                 wpa_supplicant_deauthenticate(wpa_s,
51                                               WLAN_REASON_DEAUTH_LEAVING);
52         }
53         wpa_s->disconnected = 0;
54         wpa_s->reassociate = 1;
55
56         if (wpa_s->last_scan_res_used > 0) {
57                 struct os_time now;
58                 os_get_time(&now);
59                 if (now.sec - wpa_s->last_scan.sec <= 5) {
60                         wpa_printf(MSG_DEBUG, "Interworking: Old scan results "
61                                    "are fresh - connect without new scan");
62                         if (wpas_select_network_from_last_scan(wpa_s) == 0)
63                                 return;
64                 }
65         }
66
67         wpa_supplicant_req_scan(wpa_s, 0, 0);
68 }
69
70
71 static struct wpabuf * anqp_build_req(u16 info_ids[], size_t num_ids,
72                                       struct wpabuf *extra)
73 {
74         struct wpabuf *buf;
75         size_t i;
76         u8 *len_pos;
77
78         buf = gas_anqp_build_initial_req(0, 4 + num_ids * 2 +
79                                          (extra ? wpabuf_len(extra) : 0));
80         if (buf == NULL)
81                 return NULL;
82
83         len_pos = gas_anqp_add_element(buf, ANQP_QUERY_LIST);
84         for (i = 0; i < num_ids; i++)
85                 wpabuf_put_le16(buf, info_ids[i]);
86         gas_anqp_set_element_len(buf, len_pos);
87         if (extra)
88                 wpabuf_put_buf(buf, extra);
89
90         gas_anqp_set_len(buf);
91
92         return buf;
93 }
94
95
96 static void interworking_anqp_resp_cb(void *ctx, const u8 *dst,
97                                       u8 dialog_token,
98                                       enum gas_query_result result,
99                                       const struct wpabuf *adv_proto,
100                                       const struct wpabuf *resp,
101                                       u16 status_code)
102 {
103         struct wpa_supplicant *wpa_s = ctx;
104
105         anqp_resp_cb(wpa_s, dst, dialog_token, result, adv_proto, resp,
106                      status_code);
107         interworking_next_anqp_fetch(wpa_s);
108 }
109
110
111 static int cred_with_roaming_consortium(struct wpa_supplicant *wpa_s)
112 {
113         struct wpa_cred *cred;
114
115         for (cred = wpa_s->conf->cred; cred; cred = cred->next) {
116                 if (cred->roaming_consortium_len)
117                         return 1;
118         }
119         return 0;
120 }
121
122
123 static int cred_with_3gpp(struct wpa_supplicant *wpa_s)
124 {
125         struct wpa_cred *cred;
126
127         for (cred = wpa_s->conf->cred; cred; cred = cred->next) {
128                 if (cred->pcsc || cred->imsi)
129                         return 1;
130         }
131         return 0;
132 }
133
134
135 static int cred_with_nai_realm(struct wpa_supplicant *wpa_s)
136 {
137         struct wpa_cred *cred;
138
139         for (cred = wpa_s->conf->cred; cred; cred = cred->next) {
140                 if (cred->pcsc || cred->imsi)
141                         continue;
142                 if (!cred->eap_method)
143                         return 1;
144                 if (cred->realm && cred->roaming_consortium_len == 0)
145                         return 1;
146         }
147         return 0;
148 }
149
150
151 static int cred_with_domain(struct wpa_supplicant *wpa_s)
152 {
153         struct wpa_cred *cred;
154
155         for (cred = wpa_s->conf->cred; cred; cred = cred->next) {
156                 if (cred->domain || cred->pcsc || cred->imsi)
157                         return 1;
158         }
159         return 0;
160 }
161
162
163 static int additional_roaming_consortiums(struct wpa_bss *bss)
164 {
165         const u8 *ie;
166         ie = wpa_bss_get_ie(bss, WLAN_EID_ROAMING_CONSORTIUM);
167         if (ie == NULL || ie[1] == 0)
168                 return 0;
169         return ie[2]; /* Number of ANQP OIs */
170 }
171
172
173 static int interworking_anqp_send_req(struct wpa_supplicant *wpa_s,
174                                       struct wpa_bss *bss)
175 {
176         struct wpabuf *buf;
177         int ret = 0;
178         int res;
179         u16 info_ids[8];
180         size_t num_info_ids = 0;
181         struct wpabuf *extra = NULL;
182         int all = wpa_s->fetch_all_anqp;
183
184         wpa_printf(MSG_DEBUG, "Interworking: ANQP Query Request to " MACSTR,
185                    MAC2STR(bss->bssid));
186
187         info_ids[num_info_ids++] = ANQP_CAPABILITY_LIST;
188         if (all) {
189                 info_ids[num_info_ids++] = ANQP_VENUE_NAME;
190                 info_ids[num_info_ids++] = ANQP_NETWORK_AUTH_TYPE;
191         }
192         if (all || (cred_with_roaming_consortium(wpa_s) &&
193                     additional_roaming_consortiums(bss)))
194                 info_ids[num_info_ids++] = ANQP_ROAMING_CONSORTIUM;
195         if (all)
196                 info_ids[num_info_ids++] = ANQP_IP_ADDR_TYPE_AVAILABILITY;
197         if (all || cred_with_nai_realm(wpa_s))
198                 info_ids[num_info_ids++] = ANQP_NAI_REALM;
199         if (all || cred_with_3gpp(wpa_s))
200                 info_ids[num_info_ids++] = ANQP_3GPP_CELLULAR_NETWORK;
201         if (all || cred_with_domain(wpa_s))
202                 info_ids[num_info_ids++] = ANQP_DOMAIN_NAME;
203         wpa_hexdump(MSG_DEBUG, "Interworking: ANQP Query info",
204                     (u8 *) info_ids, num_info_ids * 2);
205
206 #ifdef CONFIG_HS20
207         if (wpa_bss_get_vendor_ie(bss, HS20_IE_VENDOR_TYPE)) {
208                 u8 *len_pos;
209
210                 extra = wpabuf_alloc(100);
211                 if (!extra)
212                         return -1;
213
214                 len_pos = gas_anqp_add_element(extra, ANQP_VENDOR_SPECIFIC);
215                 wpabuf_put_be24(extra, OUI_WFA);
216                 wpabuf_put_u8(extra, HS20_ANQP_OUI_TYPE);
217                 wpabuf_put_u8(extra, HS20_STYPE_QUERY_LIST);
218                 wpabuf_put_u8(extra, 0); /* Reserved */
219                 wpabuf_put_u8(extra, HS20_STYPE_CAPABILITY_LIST);
220                 if (all) {
221                         wpabuf_put_u8(extra,
222                                       HS20_STYPE_OPERATOR_FRIENDLY_NAME);
223                         wpabuf_put_u8(extra, HS20_STYPE_WAN_METRICS);
224                         wpabuf_put_u8(extra, HS20_STYPE_CONNECTION_CAPABILITY);
225                         wpabuf_put_u8(extra, HS20_STYPE_OPERATING_CLASS);
226                 }
227                 gas_anqp_set_element_len(extra, len_pos);
228         }
229 #endif /* CONFIG_HS20 */
230
231         buf = anqp_build_req(info_ids, num_info_ids, extra);
232         wpabuf_free(extra);
233         if (buf == NULL)
234                 return -1;
235
236         res = gas_query_req(wpa_s->gas, bss->bssid, bss->freq, buf,
237                             interworking_anqp_resp_cb, wpa_s);
238         if (res < 0) {
239                 wpa_printf(MSG_DEBUG, "ANQP: Failed to send Query Request");
240                 ret = -1;
241         } else
242                 wpa_printf(MSG_DEBUG, "ANQP: Query started with dialog token "
243                            "%u", res);
244
245         wpabuf_free(buf);
246         return ret;
247 }
248
249
250 struct nai_realm_eap {
251         u8 method;
252         u8 inner_method;
253         enum nai_realm_eap_auth_inner_non_eap inner_non_eap;
254         u8 cred_type;
255         u8 tunneled_cred_type;
256 };
257
258 struct nai_realm {
259         u8 encoding;
260         char *realm;
261         u8 eap_count;
262         struct nai_realm_eap *eap;
263 };
264
265
266 static void nai_realm_free(struct nai_realm *realms, u16 count)
267 {
268         u16 i;
269
270         if (realms == NULL)
271                 return;
272         for (i = 0; i < count; i++) {
273                 os_free(realms[i].eap);
274                 os_free(realms[i].realm);
275         }
276         os_free(realms);
277 }
278
279
280 static const u8 * nai_realm_parse_eap(struct nai_realm_eap *e, const u8 *pos,
281                                       const u8 *end)
282 {
283         u8 elen, auth_count, a;
284         const u8 *e_end;
285
286         if (pos + 3 > end) {
287                 wpa_printf(MSG_DEBUG, "No room for EAP Method fixed fields");
288                 return NULL;
289         }
290
291         elen = *pos++;
292         if (pos + elen > end || elen < 2) {
293                 wpa_printf(MSG_DEBUG, "No room for EAP Method subfield");
294                 return NULL;
295         }
296         e_end = pos + elen;
297         e->method = *pos++;
298         auth_count = *pos++;
299         wpa_printf(MSG_DEBUG, "EAP Method: len=%u method=%u auth_count=%u",
300                    elen, e->method, auth_count);
301
302         for (a = 0; a < auth_count; a++) {
303                 u8 id, len;
304
305                 if (pos + 2 > end || pos + 2 + pos[1] > end) {
306                         wpa_printf(MSG_DEBUG, "No room for Authentication "
307                                    "Parameter subfield");
308                         return NULL;
309                 }
310
311                 id = *pos++;
312                 len = *pos++;
313
314                 switch (id) {
315                 case NAI_REALM_EAP_AUTH_NON_EAP_INNER_AUTH:
316                         if (len < 1)
317                                 break;
318                         e->inner_non_eap = *pos;
319                         if (e->method != EAP_TYPE_TTLS)
320                                 break;
321                         switch (*pos) {
322                         case NAI_REALM_INNER_NON_EAP_PAP:
323                                 wpa_printf(MSG_DEBUG, "EAP-TTLS/PAP");
324                                 break;
325                         case NAI_REALM_INNER_NON_EAP_CHAP:
326                                 wpa_printf(MSG_DEBUG, "EAP-TTLS/CHAP");
327                                 break;
328                         case NAI_REALM_INNER_NON_EAP_MSCHAP:
329                                 wpa_printf(MSG_DEBUG, "EAP-TTLS/MSCHAP");
330                                 break;
331                         case NAI_REALM_INNER_NON_EAP_MSCHAPV2:
332                                 wpa_printf(MSG_DEBUG, "EAP-TTLS/MSCHAPV2");
333                                 break;
334                         }
335                         break;
336                 case NAI_REALM_EAP_AUTH_INNER_AUTH_EAP_METHOD:
337                         if (len < 1)
338                                 break;
339                         e->inner_method = *pos;
340                         wpa_printf(MSG_DEBUG, "Inner EAP method: %u",
341                                    e->inner_method);
342                         break;
343                 case NAI_REALM_EAP_AUTH_CRED_TYPE:
344                         if (len < 1)
345                                 break;
346                         e->cred_type = *pos;
347                         wpa_printf(MSG_DEBUG, "Credential Type: %u",
348                                    e->cred_type);
349                         break;
350                 case NAI_REALM_EAP_AUTH_TUNNELED_CRED_TYPE:
351                         if (len < 1)
352                                 break;
353                         e->tunneled_cred_type = *pos;
354                         wpa_printf(MSG_DEBUG, "Tunneled EAP Method Credential "
355                                    "Type: %u", e->tunneled_cred_type);
356                         break;
357                 default:
358                         wpa_printf(MSG_DEBUG, "Unsupported Authentication "
359                                    "Parameter: id=%u len=%u", id, len);
360                         wpa_hexdump(MSG_DEBUG, "Authentication Parameter "
361                                     "Value", pos, len);
362                         break;
363                 }
364
365                 pos += len;
366         }
367
368         return e_end;
369 }
370
371
372 static const u8 * nai_realm_parse_realm(struct nai_realm *r, const u8 *pos,
373                                         const u8 *end)
374 {
375         u16 len;
376         const u8 *f_end;
377         u8 realm_len, e;
378
379         if (end - pos < 4) {
380                 wpa_printf(MSG_DEBUG, "No room for NAI Realm Data "
381                            "fixed fields");
382                 return NULL;
383         }
384
385         len = WPA_GET_LE16(pos); /* NAI Realm Data field Length */
386         pos += 2;
387         if (pos + len > end || len < 3) {
388                 wpa_printf(MSG_DEBUG, "No room for NAI Realm Data "
389                            "(len=%u; left=%u)",
390                            len, (unsigned int) (end - pos));
391                 return NULL;
392         }
393         f_end = pos + len;
394
395         r->encoding = *pos++;
396         realm_len = *pos++;
397         if (pos + realm_len > f_end) {
398                 wpa_printf(MSG_DEBUG, "No room for NAI Realm "
399                            "(len=%u; left=%u)",
400                            realm_len, (unsigned int) (f_end - pos));
401                 return NULL;
402         }
403         wpa_hexdump_ascii(MSG_DEBUG, "NAI Realm", pos, realm_len);
404         r->realm = os_malloc(realm_len + 1);
405         if (r->realm == NULL)
406                 return NULL;
407         os_memcpy(r->realm, pos, realm_len);
408         r->realm[realm_len] = '\0';
409         pos += realm_len;
410
411         if (pos + 1 > f_end) {
412                 wpa_printf(MSG_DEBUG, "No room for EAP Method Count");
413                 return NULL;
414         }
415         r->eap_count = *pos++;
416         wpa_printf(MSG_DEBUG, "EAP Count: %u", r->eap_count);
417         if (pos + r->eap_count * 3 > f_end) {
418                 wpa_printf(MSG_DEBUG, "No room for EAP Methods");
419                 return NULL;
420         }
421         r->eap = os_calloc(r->eap_count, sizeof(struct nai_realm_eap));
422         if (r->eap == NULL)
423                 return NULL;
424
425         for (e = 0; e < r->eap_count; e++) {
426                 pos = nai_realm_parse_eap(&r->eap[e], pos, f_end);
427                 if (pos == NULL)
428                         return NULL;
429         }
430
431         return f_end;
432 }
433
434
435 static struct nai_realm * nai_realm_parse(struct wpabuf *anqp, u16 *count)
436 {
437         struct nai_realm *realm;
438         const u8 *pos, *end;
439         u16 i, num;
440
441         if (anqp == NULL || wpabuf_len(anqp) < 2)
442                 return NULL;
443
444         pos = wpabuf_head_u8(anqp);
445         end = pos + wpabuf_len(anqp);
446         num = WPA_GET_LE16(pos);
447         wpa_printf(MSG_DEBUG, "NAI Realm Count: %u", num);
448         pos += 2;
449
450         if (num * 5 > end - pos) {
451                 wpa_printf(MSG_DEBUG, "Invalid NAI Realm Count %u - not "
452                            "enough data (%u octets) for that many realms",
453                            num, (unsigned int) (end - pos));
454                 return NULL;
455         }
456
457         realm = os_calloc(num, sizeof(struct nai_realm));
458         if (realm == NULL)
459                 return NULL;
460
461         for (i = 0; i < num; i++) {
462                 pos = nai_realm_parse_realm(&realm[i], pos, end);
463                 if (pos == NULL) {
464                         nai_realm_free(realm, num);
465                         return NULL;
466                 }
467         }
468
469         *count = num;
470         return realm;
471 }
472
473
474 static int nai_realm_match(struct nai_realm *realm, const char *home_realm)
475 {
476         char *tmp, *pos, *end;
477         int match = 0;
478
479         if (realm->realm == NULL || home_realm == NULL)
480                 return 0;
481
482         if (os_strchr(realm->realm, ';') == NULL)
483                 return os_strcasecmp(realm->realm, home_realm) == 0;
484
485         tmp = os_strdup(realm->realm);
486         if (tmp == NULL)
487                 return 0;
488
489         pos = tmp;
490         while (*pos) {
491                 end = os_strchr(pos, ';');
492                 if (end)
493                         *end = '\0';
494                 if (os_strcasecmp(pos, home_realm) == 0) {
495                         match = 1;
496                         break;
497                 }
498                 if (end == NULL)
499                         break;
500                 pos = end + 1;
501         }
502
503         os_free(tmp);
504
505         return match;
506 }
507
508
509 static int nai_realm_cred_username(struct nai_realm_eap *eap)
510 {
511         if (eap_get_name(EAP_VENDOR_IETF, eap->method) == NULL)
512                 return 0; /* method not supported */
513
514         if (eap->method != EAP_TYPE_TTLS && eap->method != EAP_TYPE_PEAP) {
515                 /* Only tunneled methods with username/password supported */
516                 return 0;
517         }
518
519         if (eap->method == EAP_TYPE_PEAP &&
520             eap_get_name(EAP_VENDOR_IETF, eap->inner_method) == NULL)
521                 return 0;
522
523         if (eap->method == EAP_TYPE_TTLS) {
524                 if (eap->inner_method == 0 && eap->inner_non_eap == 0)
525                         return 0;
526                 if (eap->inner_method &&
527                     eap_get_name(EAP_VENDOR_IETF, eap->inner_method) == NULL)
528                         return 0;
529                 if (eap->inner_non_eap &&
530                     eap->inner_non_eap != NAI_REALM_INNER_NON_EAP_PAP &&
531                     eap->inner_non_eap != NAI_REALM_INNER_NON_EAP_CHAP &&
532                     eap->inner_non_eap != NAI_REALM_INNER_NON_EAP_MSCHAP &&
533                     eap->inner_non_eap != NAI_REALM_INNER_NON_EAP_MSCHAPV2)
534                         return 0;
535         }
536
537         if (eap->inner_method &&
538             eap->inner_method != EAP_TYPE_GTC &&
539             eap->inner_method != EAP_TYPE_MSCHAPV2)
540                 return 0;
541
542         return 1;
543 }
544
545
546 static int nai_realm_cred_cert(struct nai_realm_eap *eap)
547 {
548         if (eap_get_name(EAP_VENDOR_IETF, eap->method) == NULL)
549                 return 0; /* method not supported */
550
551         if (eap->method != EAP_TYPE_TLS) {
552                 /* Only EAP-TLS supported for credential authentication */
553                 return 0;
554         }
555
556         return 1;
557 }
558
559
560 static struct nai_realm_eap * nai_realm_find_eap(struct wpa_cred *cred,
561                                                  struct nai_realm *realm)
562 {
563         u8 e;
564
565         if (cred == NULL ||
566             cred->username == NULL ||
567             cred->username[0] == '\0' ||
568             ((cred->password == NULL ||
569               cred->password[0] == '\0') &&
570              (cred->private_key == NULL ||
571               cred->private_key[0] == '\0')))
572                 return NULL;
573
574         for (e = 0; e < realm->eap_count; e++) {
575                 struct nai_realm_eap *eap = &realm->eap[e];
576                 if (cred->password && cred->password[0] &&
577                     nai_realm_cred_username(eap))
578                         return eap;
579                 if (cred->private_key && cred->private_key[0] &&
580                     nai_realm_cred_cert(eap))
581                         return eap;
582         }
583
584         return NULL;
585 }
586
587
588 #ifdef INTERWORKING_3GPP
589
590 static int plmn_id_match(struct wpabuf *anqp, const char *imsi, int mnc_len)
591 {
592         u8 plmn[3];
593         const u8 *pos, *end;
594         u8 udhl;
595
596         /* See Annex A of 3GPP TS 24.234 v8.1.0 for description */
597         plmn[0] = (imsi[0] - '0') | ((imsi[1] - '0') << 4);
598         plmn[1] = imsi[2] - '0';
599         /* default to MNC length 3 if unknown */
600         if (mnc_len != 2)
601                 plmn[1] |= (imsi[5] - '0') << 4;
602         else
603                 plmn[1] |= 0xf0;
604         plmn[2] = (imsi[3] - '0') | ((imsi[4] - '0') << 4);
605
606         if (anqp == NULL)
607                 return 0;
608         pos = wpabuf_head_u8(anqp);
609         end = pos + wpabuf_len(anqp);
610         if (pos + 2 > end)
611                 return 0;
612         if (*pos != 0) {
613                 wpa_printf(MSG_DEBUG, "Unsupported GUD version 0x%x", *pos);
614                 return 0;
615         }
616         pos++;
617         udhl = *pos++;
618         if (pos + udhl > end) {
619                 wpa_printf(MSG_DEBUG, "Invalid UDHL");
620                 return 0;
621         }
622         end = pos + udhl;
623
624         while (pos + 2 <= end) {
625                 u8 iei, len;
626                 const u8 *l_end;
627                 iei = *pos++;
628                 len = *pos++ & 0x7f;
629                 if (pos + len > end)
630                         break;
631                 l_end = pos + len;
632
633                 if (iei == 0 && len > 0) {
634                         /* PLMN List */
635                         u8 num, i;
636                         num = *pos++;
637                         for (i = 0; i < num; i++) {
638                                 if (pos + 3 > end)
639                                         break;
640                                 if (os_memcmp(pos, plmn, 3) == 0)
641                                         return 1; /* Found matching PLMN */
642                                 pos += 3;
643                         }
644                 }
645
646                 pos = l_end;
647         }
648
649         return 0;
650 }
651
652
653 static int build_root_nai(char *nai, size_t nai_len, const char *imsi,
654                           size_t mnc_len, char prefix)
655 {
656         const char *sep, *msin;
657         char *end, *pos;
658         size_t msin_len, plmn_len;
659
660         /*
661          * TS 23.003, Clause 14 (3GPP to WLAN Interworking)
662          * Root NAI:
663          * <aka:0|sim:1><IMSI>@wlan.mnc<MNC>.mcc<MCC>.3gppnetwork.org
664          * <MNC> is zero-padded to three digits in case two-digit MNC is used
665          */
666
667         if (imsi == NULL || os_strlen(imsi) > 16) {
668                 wpa_printf(MSG_DEBUG, "No valid IMSI available");
669                 return -1;
670         }
671         sep = os_strchr(imsi, '-');
672         if (sep) {
673                 plmn_len = sep - imsi;
674                 msin = sep + 1;
675         } else if (mnc_len && os_strlen(imsi) >= 3 + mnc_len) {
676                 plmn_len = 3 + mnc_len;
677                 msin = imsi + plmn_len;
678         } else
679                 return -1;
680         if (plmn_len != 5 && plmn_len != 6)
681                 return -1;
682         msin_len = os_strlen(msin);
683
684         pos = nai;
685         end = nai + nai_len;
686         if (prefix)
687                 *pos++ = prefix;
688         os_memcpy(pos, imsi, plmn_len);
689         pos += plmn_len;
690         os_memcpy(pos, msin, msin_len);
691         pos += msin_len;
692         pos += os_snprintf(pos, end - pos, "@wlan.mnc");
693         if (plmn_len == 5) {
694                 *pos++ = '0';
695                 *pos++ = imsi[3];
696                 *pos++ = imsi[4];
697         } else {
698                 *pos++ = imsi[3];
699                 *pos++ = imsi[4];
700                 *pos++ = imsi[5];
701         }
702         pos += os_snprintf(pos, end - pos, ".mcc%c%c%c.3gppnetwork.org",
703                            imsi[0], imsi[1], imsi[2]);
704
705         return 0;
706 }
707
708
709 static int set_root_nai(struct wpa_ssid *ssid, const char *imsi, char prefix)
710 {
711         char nai[100];
712         if (build_root_nai(nai, sizeof(nai), imsi, 0, prefix) < 0)
713                 return -1;
714         return wpa_config_set_quoted(ssid, "identity", nai);
715 }
716
717 #endif /* INTERWORKING_3GPP */
718
719
720 static int interworking_set_hs20_params(struct wpa_ssid *ssid)
721 {
722         if (wpa_config_set(ssid, "key_mgmt", "WPA-EAP", 0) < 0)
723                 return -1;
724         if (wpa_config_set(ssid, "proto", "RSN", 0) < 0)
725                 return -1;
726         if (wpa_config_set(ssid, "pairwise", "CCMP", 0) < 0)
727                 return -1;
728         return 0;
729 }
730
731
732 static int interworking_connect_3gpp(struct wpa_supplicant *wpa_s,
733                                      struct wpa_bss *bss)
734 {
735 #ifdef INTERWORKING_3GPP
736         struct wpa_cred *cred;
737         struct wpa_ssid *ssid;
738         const u8 *ie;
739
740         if (bss->anqp == NULL || bss->anqp->anqp_3gpp == NULL)
741                 return -1;
742
743         for (cred = wpa_s->conf->cred; cred; cred = cred->next) {
744                 char *sep;
745                 const char *imsi;
746                 int mnc_len;
747
748 #ifdef PCSC_FUNCS
749                 if (cred->pcsc && wpa_s->conf->pcsc_reader && wpa_s->scard &&
750                     wpa_s->imsi[0]) {
751                         imsi = wpa_s->imsi;
752                         mnc_len = wpa_s->mnc_len;
753                         goto compare;
754                 }
755 #endif /* PCSC_FUNCS */
756
757                 if (cred->imsi == NULL || !cred->imsi[0] ||
758                     cred->milenage == NULL || !cred->milenage[0])
759                         continue;
760
761                 sep = os_strchr(cred->imsi, '-');
762                 if (sep == NULL ||
763                     (sep - cred->imsi != 5 && sep - cred->imsi != 6))
764                         continue;
765                 mnc_len = sep - cred->imsi - 3;
766                 imsi = cred->imsi;
767
768 #ifdef PCSC_FUNCS
769         compare:
770 #endif /* PCSC_FUNCS */
771                 if (plmn_id_match(bss->anqp->anqp_3gpp, imsi, mnc_len))
772                         break;
773         }
774         if (cred == NULL)
775                 return -1;
776
777         ie = wpa_bss_get_ie(bss, WLAN_EID_SSID);
778         if (ie == NULL)
779                 return -1;
780         wpa_printf(MSG_DEBUG, "Interworking: Connect with " MACSTR " (3GPP)",
781                    MAC2STR(bss->bssid));
782
783         ssid = wpa_config_add_network(wpa_s->conf);
784         if (ssid == NULL)
785                 return -1;
786
787         wpas_notify_network_added(wpa_s, ssid);
788         wpa_config_set_network_defaults(ssid);
789         ssid->priority = cred->priority;
790         ssid->temporary = 1;
791         ssid->ssid = os_zalloc(ie[1] + 1);
792         if (ssid->ssid == NULL)
793                 goto fail;
794         os_memcpy(ssid->ssid, ie + 2, ie[1]);
795         ssid->ssid_len = ie[1];
796
797         if (interworking_set_hs20_params(ssid) < 0)
798                 goto fail;
799
800         /* TODO: figure out whether to use EAP-SIM, EAP-AKA, or EAP-AKA' */
801         if (wpa_config_set(ssid, "eap", "SIM", 0) < 0) {
802                 wpa_printf(MSG_DEBUG, "EAP-SIM not supported");
803                 goto fail;
804         }
805         if (cred->pcsc && wpa_s->scard && scard_supports_umts(wpa_s->scard))
806                 wpa_config_set(ssid, "eap", "AKA", 0);
807         if (!cred->pcsc && set_root_nai(ssid, cred->imsi, '1') < 0) {
808                 wpa_printf(MSG_DEBUG, "Failed to set Root NAI");
809                 goto fail;
810         }
811
812         if (cred->milenage && cred->milenage[0]) {
813                 if (wpa_config_set_quoted(ssid, "password",
814                                           cred->milenage) < 0)
815                         goto fail;
816         } else if (cred->pcsc) {
817                 if (wpa_config_set_quoted(ssid, "pcsc", "") < 0)
818                         goto fail;
819                 if (wpa_s->conf->pcsc_pin &&
820                     wpa_config_set_quoted(ssid, "pin", wpa_s->conf->pcsc_pin)
821                     < 0)
822                         goto fail;
823         }
824
825         if (cred->password && cred->password[0] &&
826             wpa_config_set_quoted(ssid, "password", cred->password) < 0)
827                 goto fail;
828
829         wpa_config_update_prio_list(wpa_s->conf);
830         interworking_reconnect(wpa_s);
831
832         return 0;
833
834 fail:
835         wpas_notify_network_removed(wpa_s, ssid);
836         wpa_config_remove_network(wpa_s->conf, ssid->id);
837 #endif /* INTERWORKING_3GPP */
838         return -1;
839 }
840
841
842 static int roaming_consortium_element_match(const u8 *ie, const u8 *rc_id,
843                                             size_t rc_len)
844 {
845         const u8 *pos, *end;
846         u8 lens;
847
848         if (ie == NULL)
849                 return 0;
850
851         pos = ie + 2;
852         end = ie + 2 + ie[1];
853
854         /* Roaming Consortium element:
855          * Number of ANQP OIs
856          * OI #1 and #2 lengths
857          * OI #1, [OI #2], [OI #3]
858          */
859
860         if (pos + 2 > end)
861                 return 0;
862
863         pos++; /* skip Number of ANQP OIs */
864         lens = *pos++;
865         if (pos + (lens & 0x0f) + (lens >> 4) > end)
866                 return 0;
867
868         if ((lens & 0x0f) == rc_len && os_memcmp(pos, rc_id, rc_len) == 0)
869                 return 1;
870         pos += lens & 0x0f;
871
872         if ((lens >> 4) == rc_len && os_memcmp(pos, rc_id, rc_len) == 0)
873                 return 1;
874         pos += lens >> 4;
875
876         if (pos < end && (size_t) (end - pos) == rc_len &&
877             os_memcmp(pos, rc_id, rc_len) == 0)
878                 return 1;
879
880         return 0;
881 }
882
883
884 static int roaming_consortium_anqp_match(const struct wpabuf *anqp,
885                                          const u8 *rc_id, size_t rc_len)
886 {
887         const u8 *pos, *end;
888         u8 len;
889
890         if (anqp == NULL)
891                 return 0;
892
893         pos = wpabuf_head(anqp);
894         end = pos + wpabuf_len(anqp);
895
896         /* Set of <OI Length, OI> duples */
897         while (pos < end) {
898                 len = *pos++;
899                 if (pos + len > end)
900                         break;
901                 if (len == rc_len && os_memcmp(pos, rc_id, rc_len) == 0)
902                         return 1;
903                 pos += len;
904         }
905
906         return 0;
907 }
908
909
910 static int roaming_consortium_match(const u8 *ie, const struct wpabuf *anqp,
911                                     const u8 *rc_id, size_t rc_len)
912 {
913         return roaming_consortium_element_match(ie, rc_id, rc_len) ||
914                 roaming_consortium_anqp_match(anqp, rc_id, rc_len);
915 }
916
917
918 static struct wpa_cred * interworking_credentials_available_roaming_consortium(
919         struct wpa_supplicant *wpa_s, struct wpa_bss *bss)
920 {
921         struct wpa_cred *cred, *selected = NULL;
922         const u8 *ie;
923
924         ie = wpa_bss_get_ie(bss, WLAN_EID_ROAMING_CONSORTIUM);
925
926         if (ie == NULL &&
927             (bss->anqp == NULL || bss->anqp->roaming_consortium == NULL))
928                 return NULL;
929
930         if (wpa_s->conf->cred == NULL)
931                 return NULL;
932
933         for (cred = wpa_s->conf->cred; cred; cred = cred->next) {
934                 if (cred->roaming_consortium_len == 0)
935                         continue;
936
937                 if (!roaming_consortium_match(ie,
938                                               bss->anqp ?
939                                               bss->anqp->roaming_consortium :
940                                               NULL,
941                                               cred->roaming_consortium,
942                                               cred->roaming_consortium_len))
943                         continue;
944
945                 if (selected == NULL ||
946                     selected->priority < cred->priority)
947                         selected = cred;
948         }
949
950         return selected;
951 }
952
953
954 static int interworking_set_eap_params(struct wpa_ssid *ssid,
955                                        struct wpa_cred *cred, int ttls)
956 {
957         if (cred->eap_method) {
958                 ttls = cred->eap_method->vendor == EAP_VENDOR_IETF &&
959                         cred->eap_method->method == EAP_TYPE_TTLS;
960
961                 os_free(ssid->eap.eap_methods);
962                 ssid->eap.eap_methods =
963                         os_malloc(sizeof(struct eap_method_type) * 2);
964                 if (ssid->eap.eap_methods == NULL)
965                         return -1;
966                 os_memcpy(ssid->eap.eap_methods, cred->eap_method,
967                           sizeof(*cred->eap_method));
968                 ssid->eap.eap_methods[1].vendor = EAP_VENDOR_IETF;
969                 ssid->eap.eap_methods[1].method = EAP_TYPE_NONE;
970         }
971
972         if (ttls && cred->username && cred->username[0]) {
973                 const char *pos;
974                 char *anon;
975                 /* Use anonymous NAI in Phase 1 */
976                 pos = os_strchr(cred->username, '@');
977                 if (pos) {
978                         size_t buflen = 9 + os_strlen(pos) + 1;
979                         anon = os_malloc(buflen);
980                         if (anon == NULL)
981                                 return -1;
982                         os_snprintf(anon, buflen, "anonymous%s", pos);
983                 } else if (cred->realm) {
984                         size_t buflen = 10 + os_strlen(cred->realm) + 1;
985                         anon = os_malloc(buflen);
986                         if (anon == NULL)
987                                 return -1;
988                         os_snprintf(anon, buflen, "anonymous@%s", cred->realm);
989                 } else {
990                         anon = os_strdup("anonymous");
991                         if (anon == NULL)
992                                 return -1;
993                 }
994                 if (wpa_config_set_quoted(ssid, "anonymous_identity", anon) <
995                     0) {
996                         os_free(anon);
997                         return -1;
998                 }
999                 os_free(anon);
1000         }
1001
1002         if (cred->username && cred->username[0] &&
1003             wpa_config_set_quoted(ssid, "identity", cred->username) < 0)
1004                 return -1;
1005
1006         if (cred->password && cred->password[0]) {
1007                 if (cred->ext_password &&
1008                     wpa_config_set(ssid, "password", cred->password, 0) < 0)
1009                         return -1;
1010                 if (!cred->ext_password &&
1011                     wpa_config_set_quoted(ssid, "password", cred->password) <
1012                     0)
1013                         return -1;
1014         }
1015
1016         if (cred->client_cert && cred->client_cert[0] &&
1017             wpa_config_set_quoted(ssid, "client_cert", cred->client_cert) < 0)
1018                 return -1;
1019
1020         if (cred->private_key && cred->private_key[0] &&
1021             wpa_config_set_quoted(ssid, "private_key", cred->private_key) < 0)
1022                 return -1;
1023
1024         if (cred->private_key_passwd && cred->private_key_passwd[0] &&
1025             wpa_config_set_quoted(ssid, "private_key_passwd",
1026                                   cred->private_key_passwd) < 0)
1027                 return -1;
1028
1029         if (cred->phase1) {
1030                 os_free(ssid->eap.phase1);
1031                 ssid->eap.phase1 = os_strdup(cred->phase1);
1032         }
1033         if (cred->phase2) {
1034                 os_free(ssid->eap.phase2);
1035                 ssid->eap.phase2 = os_strdup(cred->phase2);
1036         }
1037
1038         if (cred->ca_cert && cred->ca_cert[0] &&
1039             wpa_config_set_quoted(ssid, "ca_cert", cred->ca_cert) < 0)
1040                 return -1;
1041
1042         return 0;
1043 }
1044
1045
1046 static int interworking_connect_roaming_consortium(
1047         struct wpa_supplicant *wpa_s, struct wpa_cred *cred,
1048         struct wpa_bss *bss, const u8 *ssid_ie)
1049 {
1050         struct wpa_ssid *ssid;
1051
1052         wpa_printf(MSG_DEBUG, "Interworking: Connect with " MACSTR " based on "
1053                    "roaming consortium match", MAC2STR(bss->bssid));
1054
1055         ssid = wpa_config_add_network(wpa_s->conf);
1056         if (ssid == NULL)
1057                 return -1;
1058         wpas_notify_network_added(wpa_s, ssid);
1059         wpa_config_set_network_defaults(ssid);
1060         ssid->priority = cred->priority;
1061         ssid->temporary = 1;
1062         ssid->ssid = os_zalloc(ssid_ie[1] + 1);
1063         if (ssid->ssid == NULL)
1064                 goto fail;
1065         os_memcpy(ssid->ssid, ssid_ie + 2, ssid_ie[1]);
1066         ssid->ssid_len = ssid_ie[1];
1067
1068         if (interworking_set_hs20_params(ssid) < 0)
1069                 goto fail;
1070
1071         if (cred->eap_method == NULL) {
1072                 wpa_printf(MSG_DEBUG, "Interworking: No EAP method set for "
1073                            "credential using roaming consortium");
1074                 goto fail;
1075         }
1076
1077         if (interworking_set_eap_params(
1078                     ssid, cred,
1079                     cred->eap_method->vendor == EAP_VENDOR_IETF &&
1080                     cred->eap_method->method == EAP_TYPE_TTLS) < 0)
1081                 goto fail;
1082
1083         wpa_config_update_prio_list(wpa_s->conf);
1084         interworking_reconnect(wpa_s);
1085
1086         return 0;
1087
1088 fail:
1089         wpas_notify_network_removed(wpa_s, ssid);
1090         wpa_config_remove_network(wpa_s->conf, ssid->id);
1091         return -1;
1092 }
1093
1094
1095 int interworking_connect(struct wpa_supplicant *wpa_s, struct wpa_bss *bss)
1096 {
1097         struct wpa_cred *cred;
1098         struct wpa_ssid *ssid;
1099         struct nai_realm *realm;
1100         struct nai_realm_eap *eap = NULL;
1101         u16 count, i;
1102         char buf[100];
1103         const u8 *ie;
1104
1105         if (wpa_s->conf->cred == NULL || bss == NULL)
1106                 return -1;
1107         ie = wpa_bss_get_ie(bss, WLAN_EID_SSID);
1108         if (ie == NULL || ie[1] == 0) {
1109                 wpa_printf(MSG_DEBUG, "Interworking: No SSID known for "
1110                            MACSTR, MAC2STR(bss->bssid));
1111                 return -1;
1112         }
1113
1114         if (!wpa_bss_get_ie(bss, WLAN_EID_RSN)) {
1115                 /*
1116                  * We currently support only HS 2.0 networks and those are
1117                  * required to use WPA2-Enterprise.
1118                  */
1119                 wpa_printf(MSG_DEBUG, "Interworking: Network does not use "
1120                            "RSN");
1121                 return -1;
1122         }
1123
1124         cred = interworking_credentials_available_roaming_consortium(wpa_s,
1125                                                                      bss);
1126         if (cred)
1127                 return interworking_connect_roaming_consortium(wpa_s, cred,
1128                                                                bss, ie);
1129
1130         realm = nai_realm_parse(bss->anqp ? bss->anqp->nai_realm : NULL,
1131                                 &count);
1132         if (realm == NULL) {
1133                 wpa_printf(MSG_DEBUG, "Interworking: Could not parse NAI "
1134                            "Realm list from " MACSTR, MAC2STR(bss->bssid));
1135                 count = 0;
1136         }
1137
1138         for (cred = wpa_s->conf->cred; cred; cred = cred->next) {
1139                 for (i = 0; i < count; i++) {
1140                         if (!nai_realm_match(&realm[i], cred->realm))
1141                                 continue;
1142                         eap = nai_realm_find_eap(cred, &realm[i]);
1143                         if (eap)
1144                                 break;
1145                 }
1146                 if (eap)
1147                         break;
1148         }
1149
1150         if (!eap) {
1151                 if (interworking_connect_3gpp(wpa_s, bss) == 0) {
1152                         if (realm)
1153                                 nai_realm_free(realm, count);
1154                         return 0;
1155                 }
1156
1157                 wpa_printf(MSG_DEBUG, "Interworking: No matching credentials "
1158                            "and EAP method found for " MACSTR,
1159                            MAC2STR(bss->bssid));
1160                 nai_realm_free(realm, count);
1161                 return -1;
1162         }
1163
1164         wpa_printf(MSG_DEBUG, "Interworking: Connect with " MACSTR,
1165                    MAC2STR(bss->bssid));
1166
1167         ssid = wpa_config_add_network(wpa_s->conf);
1168         if (ssid == NULL) {
1169                 nai_realm_free(realm, count);
1170                 return -1;
1171         }
1172         wpas_notify_network_added(wpa_s, ssid);
1173         wpa_config_set_network_defaults(ssid);
1174         ssid->priority = cred->priority;
1175         ssid->temporary = 1;
1176         ssid->ssid = os_zalloc(ie[1] + 1);
1177         if (ssid->ssid == NULL)
1178                 goto fail;
1179         os_memcpy(ssid->ssid, ie + 2, ie[1]);
1180         ssid->ssid_len = ie[1];
1181
1182         if (interworking_set_hs20_params(ssid) < 0)
1183                 goto fail;
1184
1185         if (wpa_config_set(ssid, "eap", eap_get_name(EAP_VENDOR_IETF,
1186                                                      eap->method), 0) < 0)
1187                 goto fail;
1188
1189         switch (eap->method) {
1190         case EAP_TYPE_TTLS:
1191                 if (eap->inner_method) {
1192                         os_snprintf(buf, sizeof(buf), "\"autheap=%s\"",
1193                                     eap_get_name(EAP_VENDOR_IETF,
1194                                                  eap->inner_method));
1195                         if (wpa_config_set(ssid, "phase2", buf, 0) < 0)
1196                                 goto fail;
1197                         break;
1198                 }
1199                 switch (eap->inner_non_eap) {
1200                 case NAI_REALM_INNER_NON_EAP_PAP:
1201                         if (wpa_config_set(ssid, "phase2", "\"auth=PAP\"", 0) <
1202                             0)
1203                                 goto fail;
1204                         break;
1205                 case NAI_REALM_INNER_NON_EAP_CHAP:
1206                         if (wpa_config_set(ssid, "phase2", "\"auth=CHAP\"", 0)
1207                             < 0)
1208                                 goto fail;
1209                         break;
1210                 case NAI_REALM_INNER_NON_EAP_MSCHAP:
1211                         if (wpa_config_set(ssid, "phase2", "\"auth=MSCHAP\"",
1212                                            0) < 0)
1213                                 goto fail;
1214                         break;
1215                 case NAI_REALM_INNER_NON_EAP_MSCHAPV2:
1216                         if (wpa_config_set(ssid, "phase2", "\"auth=MSCHAPV2\"",
1217                                            0) < 0)
1218                                 goto fail;
1219                         break;
1220                 }
1221                 break;
1222         case EAP_TYPE_PEAP:
1223                 os_snprintf(buf, sizeof(buf), "\"auth=%s\"",
1224                             eap_get_name(EAP_VENDOR_IETF, eap->inner_method));
1225                 if (wpa_config_set(ssid, "phase2", buf, 0) < 0)
1226                         goto fail;
1227                 break;
1228         case EAP_TYPE_TLS:
1229                 break;
1230         }
1231
1232         if (interworking_set_eap_params(ssid, cred,
1233                                         eap->method == EAP_TYPE_TTLS) < 0)
1234                 goto fail;
1235
1236         nai_realm_free(realm, count);
1237
1238         wpa_config_update_prio_list(wpa_s->conf);
1239         interworking_reconnect(wpa_s);
1240
1241         return 0;
1242
1243 fail:
1244         wpas_notify_network_removed(wpa_s, ssid);
1245         wpa_config_remove_network(wpa_s->conf, ssid->id);
1246         nai_realm_free(realm, count);
1247         return -1;
1248 }
1249
1250
1251 static struct wpa_cred * interworking_credentials_available_3gpp(
1252         struct wpa_supplicant *wpa_s, struct wpa_bss *bss)
1253 {
1254         struct wpa_cred *cred, *selected = NULL;
1255         int ret;
1256
1257 #ifdef INTERWORKING_3GPP
1258         if (bss->anqp == NULL || bss->anqp->anqp_3gpp == NULL)
1259                 return NULL;
1260
1261         for (cred = wpa_s->conf->cred; cred; cred = cred->next) {
1262                 char *sep;
1263                 const char *imsi;
1264                 int mnc_len;
1265
1266 #ifdef PCSC_FUNCS
1267                 if (cred->pcsc && wpa_s->conf->pcsc_reader && wpa_s->scard &&
1268                     wpa_s->imsi[0]) {
1269                         imsi = wpa_s->imsi;
1270                         mnc_len = wpa_s->mnc_len;
1271                         goto compare;
1272                 }
1273 #endif /* PCSC_FUNCS */
1274
1275                 if (cred->imsi == NULL || !cred->imsi[0] ||
1276                     cred->milenage == NULL || !cred->milenage[0])
1277                         continue;
1278
1279                 sep = os_strchr(cred->imsi, '-');
1280                 if (sep == NULL ||
1281                     (sep - cred->imsi != 5 && sep - cred->imsi != 6))
1282                         continue;
1283                 mnc_len = sep - cred->imsi - 3;
1284                 imsi = cred->imsi;
1285
1286 #ifdef PCSC_FUNCS
1287         compare:
1288 #endif /* PCSC_FUNCS */
1289                 wpa_printf(MSG_DEBUG, "Interworking: Parsing 3GPP info from "
1290                            MACSTR, MAC2STR(bss->bssid));
1291                 ret = plmn_id_match(bss->anqp->anqp_3gpp, imsi, mnc_len);
1292                 wpa_printf(MSG_DEBUG, "PLMN match %sfound", ret ? "" : "not ");
1293                 if (ret) {
1294                         if (selected == NULL ||
1295                             selected->priority < cred->priority)
1296                                 selected = cred;
1297                 }
1298         }
1299 #endif /* INTERWORKING_3GPP */
1300         return selected;
1301 }
1302
1303
1304 static struct wpa_cred * interworking_credentials_available_realm(
1305         struct wpa_supplicant *wpa_s, struct wpa_bss *bss)
1306 {
1307         struct wpa_cred *cred, *selected = NULL;
1308         struct nai_realm *realm;
1309         u16 count, i;
1310
1311         if (bss->anqp == NULL || bss->anqp->nai_realm == NULL)
1312                 return NULL;
1313
1314         if (wpa_s->conf->cred == NULL)
1315                 return NULL;
1316
1317         wpa_printf(MSG_DEBUG, "Interworking: Parsing NAI Realm list from "
1318                    MACSTR, MAC2STR(bss->bssid));
1319         realm = nai_realm_parse(bss->anqp->nai_realm, &count);
1320         if (realm == NULL) {
1321                 wpa_printf(MSG_DEBUG, "Interworking: Could not parse NAI "
1322                            "Realm list from " MACSTR, MAC2STR(bss->bssid));
1323                 return NULL;
1324         }
1325
1326         for (cred = wpa_s->conf->cred; cred; cred = cred->next) {
1327                 if (cred->realm == NULL)
1328                         continue;
1329
1330                 for (i = 0; i < count; i++) {
1331                         if (!nai_realm_match(&realm[i], cred->realm))
1332                                 continue;
1333                         if (nai_realm_find_eap(cred, &realm[i])) {
1334                                 if (selected == NULL ||
1335                                     selected->priority < cred->priority)
1336                                         selected = cred;
1337                                 break;
1338                         }
1339                 }
1340         }
1341
1342         nai_realm_free(realm, count);
1343
1344         return selected;
1345 }
1346
1347
1348 static struct wpa_cred * interworking_credentials_available(
1349         struct wpa_supplicant *wpa_s, struct wpa_bss *bss)
1350 {
1351         struct wpa_cred *cred, *cred2;
1352
1353         cred = interworking_credentials_available_realm(wpa_s, bss);
1354         cred2 = interworking_credentials_available_3gpp(wpa_s, bss);
1355         if (cred && cred2 && cred2->priority >= cred->priority)
1356                 cred = cred2;
1357         if (!cred)
1358                 cred = cred2;
1359
1360         cred2 = interworking_credentials_available_roaming_consortium(wpa_s,
1361                                                                       bss);
1362         if (cred && cred2 && cred2->priority >= cred->priority)
1363                 cred = cred2;
1364         if (!cred)
1365                 cred = cred2;
1366
1367         return cred;
1368 }
1369
1370
1371 static int domain_name_list_contains(struct wpabuf *domain_names,
1372                                      const char *domain)
1373 {
1374         const u8 *pos, *end;
1375         size_t len;
1376
1377         len = os_strlen(domain);
1378         pos = wpabuf_head(domain_names);
1379         end = pos + wpabuf_len(domain_names);
1380
1381         while (pos + 1 < end) {
1382                 if (pos + 1 + pos[0] > end)
1383                         break;
1384
1385                 wpa_hexdump_ascii(MSG_DEBUG, "Interworking: AP domain name",
1386                                   pos + 1, pos[0]);
1387                 if (pos[0] == len &&
1388                     os_strncasecmp(domain, (const char *) (pos + 1), len) == 0)
1389                         return 1;
1390
1391                 pos += 1 + pos[0];
1392         }
1393
1394         return 0;
1395 }
1396
1397
1398 static int interworking_home_sp(struct wpa_supplicant *wpa_s,
1399                                 struct wpabuf *domain_names)
1400 {
1401         struct wpa_cred *cred;
1402 #ifdef INTERWORKING_3GPP
1403         char nai[100], *realm;
1404 #endif /* INTERWORKING_3GPP */
1405
1406         if (domain_names == NULL || wpa_s->conf->cred == NULL)
1407                 return -1;
1408
1409         for (cred = wpa_s->conf->cred; cred; cred = cred->next) {
1410 #ifdef INTERWORKING_3GPP
1411                 char *imsi = NULL;
1412                 int mnc_len = 0;
1413                 if (cred->imsi)
1414                         imsi = cred->imsi;
1415 #ifdef CONFIG_PCSC
1416                 else if (cred->pcsc && wpa_s->conf->pcsc_reader &&
1417                          wpa_s->scard && wpa_s->imsi[0]) {
1418                         imsi = wpa_s->imsi;
1419                         mnc_len = wpa_s->mnc_len;
1420                 }
1421 #endif /* CONFIG_PCSC */
1422                 if (imsi && build_root_nai(nai, sizeof(nai), imsi, mnc_len, 0)
1423                     == 0) {
1424                         realm = os_strchr(nai, '@');
1425                         if (realm)
1426                                 realm++;
1427                         wpa_printf(MSG_DEBUG, "Interworking: Search for match "
1428                                    "with SIM/USIM domain %s", realm);
1429                         if (realm &&
1430                             domain_name_list_contains(domain_names, realm))
1431                                 return 1;
1432                 }
1433 #endif /* INTERWORKING_3GPP */
1434
1435                 if (cred->domain == NULL)
1436                         continue;
1437
1438                 wpa_printf(MSG_DEBUG, "Interworking: Search for match with "
1439                            "home SP FQDN %s", cred->domain);
1440                 if (domain_name_list_contains(domain_names, cred->domain))
1441                         return 1;
1442         }
1443
1444         return 0;
1445 }
1446
1447
1448 static int interworking_find_network_match(struct wpa_supplicant *wpa_s)
1449 {
1450         struct wpa_bss *bss;
1451         struct wpa_ssid *ssid;
1452
1453         dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
1454                 for (ssid = wpa_s->conf->ssid; ssid; ssid = ssid->next) {
1455                         if (wpas_network_disabled(wpa_s, ssid) ||
1456                             ssid->mode != WPAS_MODE_INFRA)
1457                                 continue;
1458                         if (ssid->ssid_len != bss->ssid_len ||
1459                             os_memcmp(ssid->ssid, bss->ssid, ssid->ssid_len) !=
1460                             0)
1461                                 continue;
1462                         /*
1463                          * TODO: Consider more accurate matching of security
1464                          * configuration similarly to what is done in events.c
1465                          */
1466                         return 1;
1467                 }
1468         }
1469
1470         return 0;
1471 }
1472
1473
1474 static void interworking_select_network(struct wpa_supplicant *wpa_s)
1475 {
1476         struct wpa_bss *bss, *selected = NULL, *selected_home = NULL;
1477         int selected_prio = -999999, selected_home_prio = -999999;
1478         unsigned int count = 0;
1479         const char *type;
1480         int res;
1481         struct wpa_cred *cred;
1482
1483         wpa_s->network_select = 0;
1484
1485         dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
1486                 cred = interworking_credentials_available(wpa_s, bss);
1487                 if (!cred)
1488                         continue;
1489                 if (!wpa_bss_get_ie(bss, WLAN_EID_RSN)) {
1490                         /*
1491                          * We currently support only HS 2.0 networks and those
1492                          * are required to use WPA2-Enterprise.
1493                          */
1494                         wpa_printf(MSG_DEBUG, "Interworking: Credential match "
1495                                    "with " MACSTR " but network does not use "
1496                                    "RSN", MAC2STR(bss->bssid));
1497                         continue;
1498                 }
1499                 count++;
1500                 res = interworking_home_sp(wpa_s, bss->anqp ?
1501                                            bss->anqp->domain_name : NULL);
1502                 if (res > 0)
1503                         type = "home";
1504                 else if (res == 0)
1505                         type = "roaming";
1506                 else
1507                         type = "unknown";
1508                 wpa_msg(wpa_s, MSG_INFO, INTERWORKING_AP MACSTR " type=%s",
1509                         MAC2STR(bss->bssid), type);
1510                 if (wpa_s->auto_select ||
1511                     (wpa_s->conf->auto_interworking &&
1512                      wpa_s->auto_network_select)) {
1513                         if (selected == NULL ||
1514                             cred->priority > selected_prio) {
1515                                 selected = bss;
1516                                 selected_prio = cred->priority;
1517                         }
1518                         if (res > 0 &&
1519                             (selected_home == NULL ||
1520                              cred->priority > selected_home_prio)) {
1521                                 selected_home = bss;
1522                                 selected_home_prio = cred->priority;
1523                         }
1524                 }
1525         }
1526
1527         if (selected_home && selected_home != selected &&
1528             selected_home_prio >= selected_prio) {
1529                 /* Prefer network operated by the Home SP */
1530                 selected = selected_home;
1531         }
1532
1533         if (count == 0) {
1534                 /*
1535                  * No matching network was found based on configured
1536                  * credentials. Check whether any of the enabled network blocks
1537                  * have matching APs.
1538                  */
1539                 if (interworking_find_network_match(wpa_s)) {
1540                         wpa_printf(MSG_DEBUG, "Interworking: Possible BSS "
1541                                    "match for enabled network configurations");
1542                         if (wpa_s->auto_select)
1543                                 interworking_reconnect(wpa_s);
1544                         return;
1545                 }
1546
1547                 if (wpa_s->auto_network_select) {
1548                         wpa_printf(MSG_DEBUG, "Interworking: Continue "
1549                                    "scanning after ANQP fetch");
1550                         wpa_supplicant_req_scan(wpa_s, wpa_s->scan_interval,
1551                                                 0);
1552                         return;
1553                 }
1554
1555                 wpa_msg(wpa_s, MSG_INFO, INTERWORKING_NO_MATCH "No network "
1556                         "with matching credentials found");
1557         }
1558
1559         if (selected)
1560                 interworking_connect(wpa_s, selected);
1561 }
1562
1563
1564 static void interworking_next_anqp_fetch(struct wpa_supplicant *wpa_s)
1565 {
1566         struct wpa_bss *bss;
1567         int found = 0;
1568         const u8 *ie;
1569
1570         if (!wpa_s->fetch_anqp_in_progress)
1571                 return;
1572
1573         dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
1574                 if (!(bss->caps & IEEE80211_CAP_ESS))
1575                         continue;
1576                 ie = wpa_bss_get_ie(bss, WLAN_EID_EXT_CAPAB);
1577                 if (ie == NULL || ie[1] < 4 || !(ie[5] & 0x80))
1578                         continue; /* AP does not support Interworking */
1579
1580                 if (!(bss->flags & WPA_BSS_ANQP_FETCH_TRIED)) {
1581                         if (bss->anqp == NULL) {
1582                                 bss->anqp = wpa_bss_anqp_alloc();
1583                                 if (bss->anqp == NULL)
1584                                         break;
1585                         }
1586                         found++;
1587                         bss->flags |= WPA_BSS_ANQP_FETCH_TRIED;
1588                         wpa_msg(wpa_s, MSG_INFO, "Starting ANQP fetch for "
1589                                 MACSTR, MAC2STR(bss->bssid));
1590                         interworking_anqp_send_req(wpa_s, bss);
1591                         break;
1592                 }
1593         }
1594
1595         if (found == 0) {
1596                 wpa_msg(wpa_s, MSG_INFO, "ANQP fetch completed");
1597                 wpa_s->fetch_anqp_in_progress = 0;
1598                 if (wpa_s->network_select)
1599                         interworking_select_network(wpa_s);
1600         }
1601 }
1602
1603
1604 void interworking_start_fetch_anqp(struct wpa_supplicant *wpa_s)
1605 {
1606         struct wpa_bss *bss;
1607
1608         dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list)
1609                 bss->flags &= ~WPA_BSS_ANQP_FETCH_TRIED;
1610
1611         wpa_s->fetch_anqp_in_progress = 1;
1612         interworking_next_anqp_fetch(wpa_s);
1613 }
1614
1615
1616 int interworking_fetch_anqp(struct wpa_supplicant *wpa_s)
1617 {
1618         if (wpa_s->fetch_anqp_in_progress || wpa_s->network_select)
1619                 return 0;
1620
1621         wpa_s->network_select = 0;
1622         wpa_s->fetch_all_anqp = 1;
1623
1624         interworking_start_fetch_anqp(wpa_s);
1625
1626         return 0;
1627 }
1628
1629
1630 void interworking_stop_fetch_anqp(struct wpa_supplicant *wpa_s)
1631 {
1632         if (!wpa_s->fetch_anqp_in_progress)
1633                 return;
1634
1635         wpa_s->fetch_anqp_in_progress = 0;
1636 }
1637
1638
1639 int anqp_send_req(struct wpa_supplicant *wpa_s, const u8 *dst,
1640                   u16 info_ids[], size_t num_ids)
1641 {
1642         struct wpabuf *buf;
1643         int ret = 0;
1644         int freq;
1645         struct wpa_bss *bss;
1646         int res;
1647
1648         freq = wpa_s->assoc_freq;
1649         bss = wpa_bss_get_bssid(wpa_s, dst);
1650         if (bss)
1651                 freq = bss->freq;
1652         if (freq <= 0)
1653                 return -1;
1654
1655         wpa_printf(MSG_DEBUG, "ANQP: Query Request to " MACSTR " for %u id(s)",
1656                    MAC2STR(dst), (unsigned int) num_ids);
1657
1658         buf = anqp_build_req(info_ids, num_ids, NULL);
1659         if (buf == NULL)
1660                 return -1;
1661
1662         res = gas_query_req(wpa_s->gas, dst, freq, buf, anqp_resp_cb, wpa_s);
1663         if (res < 0) {
1664                 wpa_printf(MSG_DEBUG, "ANQP: Failed to send Query Request");
1665                 ret = -1;
1666         } else
1667                 wpa_printf(MSG_DEBUG, "ANQP: Query started with dialog token "
1668                            "%u", res);
1669
1670         wpabuf_free(buf);
1671         return ret;
1672 }
1673
1674
1675 static void interworking_parse_rx_anqp_resp(struct wpa_supplicant *wpa_s,
1676                                             const u8 *sa, u16 info_id,
1677                                             const u8 *data, size_t slen)
1678 {
1679         const u8 *pos = data;
1680         struct wpa_bss *bss = wpa_bss_get_bssid(wpa_s, sa);
1681         struct wpa_bss_anqp *anqp = NULL;
1682 #ifdef CONFIG_HS20
1683         u8 type;
1684 #endif /* CONFIG_HS20 */
1685
1686         if (bss)
1687                 anqp = bss->anqp;
1688
1689         switch (info_id) {
1690         case ANQP_CAPABILITY_LIST:
1691                 wpa_msg(wpa_s, MSG_INFO, "RX-ANQP " MACSTR
1692                         " ANQP Capability list", MAC2STR(sa));
1693                 break;
1694         case ANQP_VENUE_NAME:
1695                 wpa_msg(wpa_s, MSG_INFO, "RX-ANQP " MACSTR
1696                         " Venue Name", MAC2STR(sa));
1697                 wpa_hexdump_ascii(MSG_DEBUG, "ANQP: Venue Name", pos, slen);
1698                 if (anqp) {
1699                         wpabuf_free(anqp->venue_name);
1700                         anqp->venue_name = wpabuf_alloc_copy(pos, slen);
1701                 }
1702                 break;
1703         case ANQP_NETWORK_AUTH_TYPE:
1704                 wpa_msg(wpa_s, MSG_INFO, "RX-ANQP " MACSTR
1705                         " Network Authentication Type information",
1706                         MAC2STR(sa));
1707                 wpa_hexdump_ascii(MSG_DEBUG, "ANQP: Network Authentication "
1708                                   "Type", pos, slen);
1709                 if (anqp) {
1710                         wpabuf_free(anqp->network_auth_type);
1711                         anqp->network_auth_type = wpabuf_alloc_copy(pos, slen);
1712                 }
1713                 break;
1714         case ANQP_ROAMING_CONSORTIUM:
1715                 wpa_msg(wpa_s, MSG_INFO, "RX-ANQP " MACSTR
1716                         " Roaming Consortium list", MAC2STR(sa));
1717                 wpa_hexdump_ascii(MSG_DEBUG, "ANQP: Roaming Consortium",
1718                                   pos, slen);
1719                 if (anqp) {
1720                         wpabuf_free(anqp->roaming_consortium);
1721                         anqp->roaming_consortium = wpabuf_alloc_copy(pos, slen);
1722                 }
1723                 break;
1724         case ANQP_IP_ADDR_TYPE_AVAILABILITY:
1725                 wpa_msg(wpa_s, MSG_INFO, "RX-ANQP " MACSTR
1726                         " IP Address Type Availability information",
1727                         MAC2STR(sa));
1728                 wpa_hexdump(MSG_MSGDUMP, "ANQP: IP Address Availability",
1729                             pos, slen);
1730                 if (anqp) {
1731                         wpabuf_free(anqp->ip_addr_type_availability);
1732                         anqp->ip_addr_type_availability =
1733                                 wpabuf_alloc_copy(pos, slen);
1734                 }
1735                 break;
1736         case ANQP_NAI_REALM:
1737                 wpa_msg(wpa_s, MSG_INFO, "RX-ANQP " MACSTR
1738                         " NAI Realm list", MAC2STR(sa));
1739                 wpa_hexdump_ascii(MSG_DEBUG, "ANQP: NAI Realm", pos, slen);
1740                 if (anqp) {
1741                         wpabuf_free(anqp->nai_realm);
1742                         anqp->nai_realm = wpabuf_alloc_copy(pos, slen);
1743                 }
1744                 break;
1745         case ANQP_3GPP_CELLULAR_NETWORK:
1746                 wpa_msg(wpa_s, MSG_INFO, "RX-ANQP " MACSTR
1747                         " 3GPP Cellular Network information", MAC2STR(sa));
1748                 wpa_hexdump_ascii(MSG_DEBUG, "ANQP: 3GPP Cellular Network",
1749                                   pos, slen);
1750                 if (anqp) {
1751                         wpabuf_free(anqp->anqp_3gpp);
1752                         anqp->anqp_3gpp = wpabuf_alloc_copy(pos, slen);
1753                 }
1754                 break;
1755         case ANQP_DOMAIN_NAME:
1756                 wpa_msg(wpa_s, MSG_INFO, "RX-ANQP " MACSTR
1757                         " Domain Name list", MAC2STR(sa));
1758                 wpa_hexdump_ascii(MSG_MSGDUMP, "ANQP: Domain Name", pos, slen);
1759                 if (anqp) {
1760                         wpabuf_free(anqp->domain_name);
1761                         anqp->domain_name = wpabuf_alloc_copy(pos, slen);
1762                 }
1763                 break;
1764         case ANQP_VENDOR_SPECIFIC:
1765                 if (slen < 3)
1766                         return;
1767
1768                 switch (WPA_GET_BE24(pos)) {
1769 #ifdef CONFIG_HS20
1770                 case OUI_WFA:
1771                         pos += 3;
1772                         slen -= 3;
1773
1774                         if (slen < 1)
1775                                 return;
1776                         type = *pos++;
1777                         slen--;
1778
1779                         switch (type) {
1780                         case HS20_ANQP_OUI_TYPE:
1781                                 hs20_parse_rx_hs20_anqp_resp(wpa_s, sa, pos,
1782                                                              slen);
1783                                 break;
1784                         default:
1785                                 wpa_printf(MSG_DEBUG, "HS20: Unsupported ANQP "
1786                                            "vendor type %u", type);
1787                                 break;
1788                         }
1789                         break;
1790 #endif /* CONFIG_HS20 */
1791                 default:
1792                         wpa_printf(MSG_DEBUG, "Interworking: Unsupported "
1793                                    "vendor-specific ANQP OUI %06x",
1794                                    WPA_GET_BE24(pos));
1795                         return;
1796                 }
1797                 break;
1798         default:
1799                 wpa_printf(MSG_DEBUG, "Interworking: Unsupported ANQP Info ID "
1800                            "%u", info_id);
1801                 break;
1802         }
1803 }
1804
1805
1806 void anqp_resp_cb(void *ctx, const u8 *dst, u8 dialog_token,
1807                   enum gas_query_result result,
1808                   const struct wpabuf *adv_proto,
1809                   const struct wpabuf *resp, u16 status_code)
1810 {
1811         struct wpa_supplicant *wpa_s = ctx;
1812         const u8 *pos;
1813         const u8 *end;
1814         u16 info_id;
1815         u16 slen;
1816
1817         if (result != GAS_QUERY_SUCCESS)
1818                 return;
1819
1820         pos = wpabuf_head(adv_proto);
1821         if (wpabuf_len(adv_proto) < 4 || pos[0] != WLAN_EID_ADV_PROTO ||
1822             pos[1] < 2 || pos[3] != ACCESS_NETWORK_QUERY_PROTOCOL) {
1823                 wpa_printf(MSG_DEBUG, "ANQP: Unexpected Advertisement "
1824                            "Protocol in response");
1825                 return;
1826         }
1827
1828         pos = wpabuf_head(resp);
1829         end = pos + wpabuf_len(resp);
1830
1831         while (pos < end) {
1832                 if (pos + 4 > end) {
1833                         wpa_printf(MSG_DEBUG, "ANQP: Invalid element");
1834                         break;
1835                 }
1836                 info_id = WPA_GET_LE16(pos);
1837                 pos += 2;
1838                 slen = WPA_GET_LE16(pos);
1839                 pos += 2;
1840                 if (pos + slen > end) {
1841                         wpa_printf(MSG_DEBUG, "ANQP: Invalid element length "
1842                                    "for Info ID %u", info_id);
1843                         break;
1844                 }
1845                 interworking_parse_rx_anqp_resp(wpa_s, dst, info_id, pos,
1846                                                 slen);
1847                 pos += slen;
1848         }
1849 }
1850
1851
1852 static void interworking_scan_res_handler(struct wpa_supplicant *wpa_s,
1853                                           struct wpa_scan_results *scan_res)
1854 {
1855         wpa_printf(MSG_DEBUG, "Interworking: Scan results available - start "
1856                    "ANQP fetch");
1857         interworking_start_fetch_anqp(wpa_s);
1858 }
1859
1860
1861 int interworking_select(struct wpa_supplicant *wpa_s, int auto_select)
1862 {
1863         interworking_stop_fetch_anqp(wpa_s);
1864         wpa_s->network_select = 1;
1865         wpa_s->auto_network_select = 0;
1866         wpa_s->auto_select = !!auto_select;
1867         wpa_s->fetch_all_anqp = 0;
1868         wpa_printf(MSG_DEBUG, "Interworking: Start scan for network "
1869                    "selection");
1870         wpa_s->scan_res_handler = interworking_scan_res_handler;
1871         wpa_s->scan_req = 2;
1872         wpa_supplicant_req_scan(wpa_s, 0, 0);
1873
1874         return 0;
1875 }
1876
1877
1878 static void gas_resp_cb(void *ctx, const u8 *addr, u8 dialog_token,
1879                         enum gas_query_result result,
1880                         const struct wpabuf *adv_proto,
1881                         const struct wpabuf *resp, u16 status_code)
1882 {
1883         struct wpa_supplicant *wpa_s = ctx;
1884
1885         wpa_msg(wpa_s, MSG_INFO, GAS_RESPONSE_INFO "addr=" MACSTR
1886                 " dialog_token=%d status_code=%d resp_len=%d",
1887                 MAC2STR(addr), dialog_token, status_code,
1888                 resp ? (int) wpabuf_len(resp) : -1);
1889         if (!resp)
1890                 return;
1891
1892         wpabuf_free(wpa_s->last_gas_resp);
1893         wpa_s->last_gas_resp = wpabuf_dup(resp);
1894         if (wpa_s->last_gas_resp == NULL)
1895                 return;
1896         os_memcpy(wpa_s->last_gas_addr, addr, ETH_ALEN);
1897         wpa_s->last_gas_dialog_token = dialog_token;
1898 }
1899
1900
1901 int gas_send_request(struct wpa_supplicant *wpa_s, const u8 *dst,
1902                      const struct wpabuf *adv_proto,
1903                      const struct wpabuf *query)
1904 {
1905         struct wpabuf *buf;
1906         int ret = 0;
1907         int freq;
1908         struct wpa_bss *bss;
1909         int res;
1910         size_t len;
1911         u8 query_resp_len_limit = 0, pame_bi = 0;
1912
1913         freq = wpa_s->assoc_freq;
1914         bss = wpa_bss_get_bssid(wpa_s, dst);
1915         if (bss)
1916                 freq = bss->freq;
1917         if (freq <= 0)
1918                 return -1;
1919
1920         wpa_printf(MSG_DEBUG, "GAS request to " MACSTR " (freq %d MHz)",
1921                    MAC2STR(dst), freq);
1922         wpa_hexdump_buf(MSG_DEBUG, "Advertisement Protocol ID", adv_proto);
1923         wpa_hexdump_buf(MSG_DEBUG, "GAS Query", query);
1924
1925         len = 3 + wpabuf_len(adv_proto) + 2;
1926         if (query)
1927                 len += wpabuf_len(query);
1928         buf = gas_build_initial_req(0, len);
1929         if (buf == NULL)
1930                 return -1;
1931
1932         /* Advertisement Protocol IE */
1933         wpabuf_put_u8(buf, WLAN_EID_ADV_PROTO);
1934         wpabuf_put_u8(buf, 1 + wpabuf_len(adv_proto)); /* Length */
1935         wpabuf_put_u8(buf, (query_resp_len_limit & 0x7f) |
1936                       (pame_bi ? 0x80 : 0));
1937         wpabuf_put_buf(buf, adv_proto);
1938
1939         /* GAS Query */
1940         if (query) {
1941                 wpabuf_put_le16(buf, wpabuf_len(query));
1942                 wpabuf_put_buf(buf, query);
1943         } else
1944                 wpabuf_put_le16(buf, 0);
1945
1946         res = gas_query_req(wpa_s->gas, dst, freq, buf, gas_resp_cb, wpa_s);
1947         if (res < 0) {
1948                 wpa_printf(MSG_DEBUG, "GAS: Failed to send Query Request");
1949                 ret = -1;
1950         } else
1951                 wpa_printf(MSG_DEBUG, "GAS: Query started with dialog token "
1952                            "%u", res);
1953
1954         wpabuf_free(buf);
1955         return ret;
1956 }