Updated through tag hostap_2_5 from git://w1.fi/hostap.git
[mech_eap.git] / libeap / src / ap / wpa_auth_ie.c
1 /*
2  * hostapd - WPA/RSN IE and KDE definitions
3  * Copyright (c) 2004-2015, Jouni Malinen <j@w1.fi>
4  *
5  * This software may be distributed under the terms of the BSD license.
6  * See README for more details.
7  */
8
9 #include "utils/includes.h"
10
11 #include "utils/common.h"
12 #include "common/ieee802_11_defs.h"
13 #include "eapol_auth/eapol_auth_sm.h"
14 #include "ap_config.h"
15 #include "ieee802_11.h"
16 #include "wpa_auth.h"
17 #include "pmksa_cache_auth.h"
18 #include "wpa_auth_ie.h"
19 #include "wpa_auth_i.h"
20
21
22 #ifdef CONFIG_RSN_TESTING
23 int rsn_testing = 0;
24 #endif /* CONFIG_RSN_TESTING */
25
26
27 static int wpa_write_wpa_ie(struct wpa_auth_config *conf, u8 *buf, size_t len)
28 {
29         struct wpa_ie_hdr *hdr;
30         int num_suites;
31         u8 *pos, *count;
32         u32 suite;
33
34         hdr = (struct wpa_ie_hdr *) buf;
35         hdr->elem_id = WLAN_EID_VENDOR_SPECIFIC;
36         RSN_SELECTOR_PUT(hdr->oui, WPA_OUI_TYPE);
37         WPA_PUT_LE16(hdr->version, WPA_VERSION);
38         pos = (u8 *) (hdr + 1);
39
40         suite = wpa_cipher_to_suite(WPA_PROTO_WPA, conf->wpa_group);
41         if (suite == 0) {
42                 wpa_printf(MSG_DEBUG, "Invalid group cipher (%d).",
43                            conf->wpa_group);
44                 return -1;
45         }
46         RSN_SELECTOR_PUT(pos, suite);
47         pos += WPA_SELECTOR_LEN;
48
49         count = pos;
50         pos += 2;
51
52         num_suites = wpa_cipher_put_suites(pos, conf->wpa_pairwise);
53         if (num_suites == 0) {
54                 wpa_printf(MSG_DEBUG, "Invalid pairwise cipher (%d).",
55                            conf->wpa_pairwise);
56                 return -1;
57         }
58         pos += num_suites * WPA_SELECTOR_LEN;
59         WPA_PUT_LE16(count, num_suites);
60
61         num_suites = 0;
62         count = pos;
63         pos += 2;
64
65         if (conf->wpa_key_mgmt & WPA_KEY_MGMT_IEEE8021X) {
66                 RSN_SELECTOR_PUT(pos, WPA_AUTH_KEY_MGMT_UNSPEC_802_1X);
67                 pos += WPA_SELECTOR_LEN;
68                 num_suites++;
69         }
70         if (conf->wpa_key_mgmt & WPA_KEY_MGMT_PSK) {
71                 RSN_SELECTOR_PUT(pos, WPA_AUTH_KEY_MGMT_PSK_OVER_802_1X);
72                 pos += WPA_SELECTOR_LEN;
73                 num_suites++;
74         }
75
76         if (num_suites == 0) {
77                 wpa_printf(MSG_DEBUG, "Invalid key management type (%d).",
78                            conf->wpa_key_mgmt);
79                 return -1;
80         }
81         WPA_PUT_LE16(count, num_suites);
82
83         /* WPA Capabilities; use defaults, so no need to include it */
84
85         hdr->len = (pos - buf) - 2;
86
87         return pos - buf;
88 }
89
90
91 int wpa_write_rsn_ie(struct wpa_auth_config *conf, u8 *buf, size_t len,
92                      const u8 *pmkid)
93 {
94         struct rsn_ie_hdr *hdr;
95         int num_suites, res;
96         u8 *pos, *count;
97         u16 capab;
98         u32 suite;
99
100         hdr = (struct rsn_ie_hdr *) buf;
101         hdr->elem_id = WLAN_EID_RSN;
102         WPA_PUT_LE16(hdr->version, RSN_VERSION);
103         pos = (u8 *) (hdr + 1);
104
105         suite = wpa_cipher_to_suite(WPA_PROTO_RSN, conf->wpa_group);
106         if (suite == 0) {
107                 wpa_printf(MSG_DEBUG, "Invalid group cipher (%d).",
108                            conf->wpa_group);
109                 return -1;
110         }
111         RSN_SELECTOR_PUT(pos, suite);
112         pos += RSN_SELECTOR_LEN;
113
114         num_suites = 0;
115         count = pos;
116         pos += 2;
117
118 #ifdef CONFIG_RSN_TESTING
119         if (rsn_testing) {
120                 RSN_SELECTOR_PUT(pos, RSN_SELECTOR(0x12, 0x34, 0x56, 1));
121                 pos += RSN_SELECTOR_LEN;
122                 num_suites++;
123         }
124 #endif /* CONFIG_RSN_TESTING */
125
126         res = rsn_cipher_put_suites(pos, conf->rsn_pairwise);
127         num_suites += res;
128         pos += res * RSN_SELECTOR_LEN;
129
130 #ifdef CONFIG_RSN_TESTING
131         if (rsn_testing) {
132                 RSN_SELECTOR_PUT(pos, RSN_SELECTOR(0x12, 0x34, 0x56, 2));
133                 pos += RSN_SELECTOR_LEN;
134                 num_suites++;
135         }
136 #endif /* CONFIG_RSN_TESTING */
137
138         if (num_suites == 0) {
139                 wpa_printf(MSG_DEBUG, "Invalid pairwise cipher (%d).",
140                            conf->rsn_pairwise);
141                 return -1;
142         }
143         WPA_PUT_LE16(count, num_suites);
144
145         num_suites = 0;
146         count = pos;
147         pos += 2;
148
149 #ifdef CONFIG_RSN_TESTING
150         if (rsn_testing) {
151                 RSN_SELECTOR_PUT(pos, RSN_SELECTOR(0x12, 0x34, 0x56, 1));
152                 pos += RSN_SELECTOR_LEN;
153                 num_suites++;
154         }
155 #endif /* CONFIG_RSN_TESTING */
156
157         if (conf->wpa_key_mgmt & WPA_KEY_MGMT_IEEE8021X) {
158                 RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_UNSPEC_802_1X);
159                 pos += RSN_SELECTOR_LEN;
160                 num_suites++;
161         }
162         if (conf->wpa_key_mgmt & WPA_KEY_MGMT_PSK) {
163                 RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_PSK_OVER_802_1X);
164                 pos += RSN_SELECTOR_LEN;
165                 num_suites++;
166         }
167 #ifdef CONFIG_IEEE80211R
168         if (conf->wpa_key_mgmt & WPA_KEY_MGMT_FT_IEEE8021X) {
169                 RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_FT_802_1X);
170                 pos += RSN_SELECTOR_LEN;
171                 num_suites++;
172         }
173         if (conf->wpa_key_mgmt & WPA_KEY_MGMT_FT_PSK) {
174                 RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_FT_PSK);
175                 pos += RSN_SELECTOR_LEN;
176                 num_suites++;
177         }
178 #endif /* CONFIG_IEEE80211R */
179 #ifdef CONFIG_IEEE80211W
180         if (conf->wpa_key_mgmt & WPA_KEY_MGMT_IEEE8021X_SHA256) {
181                 RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_802_1X_SHA256);
182                 pos += RSN_SELECTOR_LEN;
183                 num_suites++;
184         }
185         if (conf->wpa_key_mgmt & WPA_KEY_MGMT_PSK_SHA256) {
186                 RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_PSK_SHA256);
187                 pos += RSN_SELECTOR_LEN;
188                 num_suites++;
189         }
190 #endif /* CONFIG_IEEE80211W */
191 #ifdef CONFIG_SAE
192         if (conf->wpa_key_mgmt & WPA_KEY_MGMT_SAE) {
193                 RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_SAE);
194                 pos += RSN_SELECTOR_LEN;
195                 num_suites++;
196         }
197         if (conf->wpa_key_mgmt & WPA_KEY_MGMT_FT_SAE) {
198                 RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_FT_SAE);
199                 pos += RSN_SELECTOR_LEN;
200                 num_suites++;
201         }
202 #endif /* CONFIG_SAE */
203         if (conf->wpa_key_mgmt & WPA_KEY_MGMT_IEEE8021X_SUITE_B) {
204                 RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_802_1X_SUITE_B);
205                 pos += RSN_SELECTOR_LEN;
206                 num_suites++;
207         }
208         if (conf->wpa_key_mgmt & WPA_KEY_MGMT_IEEE8021X_SUITE_B_192) {
209                 RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_802_1X_SUITE_B_192);
210                 pos += RSN_SELECTOR_LEN;
211                 num_suites++;
212         }
213
214 #ifdef CONFIG_RSN_TESTING
215         if (rsn_testing) {
216                 RSN_SELECTOR_PUT(pos, RSN_SELECTOR(0x12, 0x34, 0x56, 2));
217                 pos += RSN_SELECTOR_LEN;
218                 num_suites++;
219         }
220 #endif /* CONFIG_RSN_TESTING */
221
222         if (num_suites == 0) {
223                 wpa_printf(MSG_DEBUG, "Invalid key management type (%d).",
224                            conf->wpa_key_mgmt);
225                 return -1;
226         }
227         WPA_PUT_LE16(count, num_suites);
228
229         /* RSN Capabilities */
230         capab = 0;
231         if (conf->rsn_preauth)
232                 capab |= WPA_CAPABILITY_PREAUTH;
233         if (conf->peerkey)
234                 capab |= WPA_CAPABILITY_PEERKEY_ENABLED;
235         if (conf->wmm_enabled) {
236                 /* 4 PTKSA replay counters when using WMM */
237                 capab |= (RSN_NUM_REPLAY_COUNTERS_16 << 2);
238         }
239 #ifdef CONFIG_IEEE80211W
240         if (conf->ieee80211w != NO_MGMT_FRAME_PROTECTION) {
241                 capab |= WPA_CAPABILITY_MFPC;
242                 if (conf->ieee80211w == MGMT_FRAME_PROTECTION_REQUIRED)
243                         capab |= WPA_CAPABILITY_MFPR;
244         }
245 #endif /* CONFIG_IEEE80211W */
246 #ifdef CONFIG_RSN_TESTING
247         if (rsn_testing)
248                 capab |= BIT(8) | BIT(14) | BIT(15);
249 #endif /* CONFIG_RSN_TESTING */
250         WPA_PUT_LE16(pos, capab);
251         pos += 2;
252
253         if (pmkid) {
254                 if (pos + 2 + PMKID_LEN > buf + len)
255                         return -1;
256                 /* PMKID Count */
257                 WPA_PUT_LE16(pos, 1);
258                 pos += 2;
259                 os_memcpy(pos, pmkid, PMKID_LEN);
260                 pos += PMKID_LEN;
261         }
262
263 #ifdef CONFIG_IEEE80211W
264         if (conf->ieee80211w != NO_MGMT_FRAME_PROTECTION &&
265             conf->group_mgmt_cipher != WPA_CIPHER_AES_128_CMAC) {
266                 if (pos + 2 + 4 > buf + len)
267                         return -1;
268                 if (pmkid == NULL) {
269                         /* PMKID Count */
270                         WPA_PUT_LE16(pos, 0);
271                         pos += 2;
272                 }
273
274                 /* Management Group Cipher Suite */
275                 switch (conf->group_mgmt_cipher) {
276                 case WPA_CIPHER_AES_128_CMAC:
277                         RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_AES_128_CMAC);
278                         break;
279                 case WPA_CIPHER_BIP_GMAC_128:
280                         RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_BIP_GMAC_128);
281                         break;
282                 case WPA_CIPHER_BIP_GMAC_256:
283                         RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_BIP_GMAC_256);
284                         break;
285                 case WPA_CIPHER_BIP_CMAC_256:
286                         RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_BIP_CMAC_256);
287                         break;
288                 default:
289                         wpa_printf(MSG_DEBUG,
290                                    "Invalid group management cipher (0x%x)",
291                                    conf->group_mgmt_cipher);
292                         return -1;
293                 }
294                 pos += RSN_SELECTOR_LEN;
295         }
296 #endif /* CONFIG_IEEE80211W */
297
298 #ifdef CONFIG_RSN_TESTING
299         if (rsn_testing) {
300                 /*
301                  * Fill in any defined fields and add extra data to the end of
302                  * the element.
303                  */
304                 int pmkid_count_set = pmkid != NULL;
305                 if (conf->ieee80211w != NO_MGMT_FRAME_PROTECTION)
306                         pmkid_count_set = 1;
307                 /* PMKID Count */
308                 WPA_PUT_LE16(pos, 0);
309                 pos += 2;
310                 if (conf->ieee80211w == NO_MGMT_FRAME_PROTECTION) {
311                         /* Management Group Cipher Suite */
312                         RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_AES_128_CMAC);
313                         pos += RSN_SELECTOR_LEN;
314                 }
315
316                 os_memset(pos, 0x12, 17);
317                 pos += 17;
318         }
319 #endif /* CONFIG_RSN_TESTING */
320
321         hdr->len = (pos - buf) - 2;
322
323         return pos - buf;
324 }
325
326
327 static u8 * wpa_write_osen(struct wpa_auth_config *conf, u8 *eid)
328 {
329         u8 *len;
330         u16 capab;
331
332         *eid++ = WLAN_EID_VENDOR_SPECIFIC;
333         len = eid++; /* to be filled */
334         WPA_PUT_BE24(eid, OUI_WFA);
335         eid += 3;
336         *eid++ = HS20_OSEN_OUI_TYPE;
337
338         /* Group Data Cipher Suite */
339         RSN_SELECTOR_PUT(eid, RSN_CIPHER_SUITE_NO_GROUP_ADDRESSED);
340         eid += RSN_SELECTOR_LEN;
341
342         /* Pairwise Cipher Suite Count and List */
343         WPA_PUT_LE16(eid, 1);
344         eid += 2;
345         RSN_SELECTOR_PUT(eid, RSN_CIPHER_SUITE_CCMP);
346         eid += RSN_SELECTOR_LEN;
347
348         /* AKM Suite Count and List */
349         WPA_PUT_LE16(eid, 1);
350         eid += 2;
351         RSN_SELECTOR_PUT(eid, RSN_AUTH_KEY_MGMT_OSEN);
352         eid += RSN_SELECTOR_LEN;
353
354         /* RSN Capabilities */
355         capab = 0;
356         if (conf->wmm_enabled) {
357                 /* 4 PTKSA replay counters when using WMM */
358                 capab |= (RSN_NUM_REPLAY_COUNTERS_16 << 2);
359         }
360 #ifdef CONFIG_IEEE80211W
361         if (conf->ieee80211w != NO_MGMT_FRAME_PROTECTION) {
362                 capab |= WPA_CAPABILITY_MFPC;
363                 if (conf->ieee80211w == MGMT_FRAME_PROTECTION_REQUIRED)
364                         capab |= WPA_CAPABILITY_MFPR;
365         }
366 #endif /* CONFIG_IEEE80211W */
367         WPA_PUT_LE16(eid, capab);
368         eid += 2;
369
370         *len = eid - len - 1;
371
372         return eid;
373 }
374
375
376 int wpa_auth_gen_wpa_ie(struct wpa_authenticator *wpa_auth)
377 {
378         u8 *pos, buf[128];
379         int res;
380
381 #ifdef CONFIG_TESTING_OPTIONS
382         if (wpa_auth->conf.own_ie_override_len) {
383                 wpa_hexdump(MSG_DEBUG, "WPA: Forced own IE(s) for testing",
384                             wpa_auth->conf.own_ie_override,
385                             wpa_auth->conf.own_ie_override_len);
386                 os_free(wpa_auth->wpa_ie);
387                 wpa_auth->wpa_ie =
388                         os_malloc(wpa_auth->conf.own_ie_override_len);
389                 if (wpa_auth->wpa_ie == NULL)
390                         return -1;
391                 os_memcpy(wpa_auth->wpa_ie, wpa_auth->conf.own_ie_override,
392                           wpa_auth->conf.own_ie_override_len);
393                 wpa_auth->wpa_ie_len = wpa_auth->conf.own_ie_override_len;
394                 return 0;
395         }
396 #endif /* CONFIG_TESTING_OPTIONS */
397
398         pos = buf;
399
400         if (wpa_auth->conf.wpa == WPA_PROTO_OSEN) {
401                 pos = wpa_write_osen(&wpa_auth->conf, pos);
402         }
403         if (wpa_auth->conf.wpa & WPA_PROTO_RSN) {
404                 res = wpa_write_rsn_ie(&wpa_auth->conf,
405                                        pos, buf + sizeof(buf) - pos, NULL);
406                 if (res < 0)
407                         return res;
408                 pos += res;
409         }
410 #ifdef CONFIG_IEEE80211R
411         if (wpa_key_mgmt_ft(wpa_auth->conf.wpa_key_mgmt)) {
412                 res = wpa_write_mdie(&wpa_auth->conf, pos,
413                                      buf + sizeof(buf) - pos);
414                 if (res < 0)
415                         return res;
416                 pos += res;
417         }
418 #endif /* CONFIG_IEEE80211R */
419         if (wpa_auth->conf.wpa & WPA_PROTO_WPA) {
420                 res = wpa_write_wpa_ie(&wpa_auth->conf,
421                                        pos, buf + sizeof(buf) - pos);
422                 if (res < 0)
423                         return res;
424                 pos += res;
425         }
426
427         os_free(wpa_auth->wpa_ie);
428         wpa_auth->wpa_ie = os_malloc(pos - buf);
429         if (wpa_auth->wpa_ie == NULL)
430                 return -1;
431         os_memcpy(wpa_auth->wpa_ie, buf, pos - buf);
432         wpa_auth->wpa_ie_len = pos - buf;
433
434         return 0;
435 }
436
437
438 u8 * wpa_add_kde(u8 *pos, u32 kde, const u8 *data, size_t data_len,
439                  const u8 *data2, size_t data2_len)
440 {
441         *pos++ = WLAN_EID_VENDOR_SPECIFIC;
442         *pos++ = RSN_SELECTOR_LEN + data_len + data2_len;
443         RSN_SELECTOR_PUT(pos, kde);
444         pos += RSN_SELECTOR_LEN;
445         os_memcpy(pos, data, data_len);
446         pos += data_len;
447         if (data2) {
448                 os_memcpy(pos, data2, data2_len);
449                 pos += data2_len;
450         }
451         return pos;
452 }
453
454
455 struct wpa_auth_okc_iter_data {
456         struct rsn_pmksa_cache_entry *pmksa;
457         const u8 *aa;
458         const u8 *spa;
459         const u8 *pmkid;
460 };
461
462
463 static int wpa_auth_okc_iter(struct wpa_authenticator *a, void *ctx)
464 {
465         struct wpa_auth_okc_iter_data *data = ctx;
466         data->pmksa = pmksa_cache_get_okc(a->pmksa, data->aa, data->spa,
467                                           data->pmkid);
468         if (data->pmksa)
469                 return 1;
470         return 0;
471 }
472
473
474 int wpa_validate_wpa_ie(struct wpa_authenticator *wpa_auth,
475                         struct wpa_state_machine *sm,
476                         const u8 *wpa_ie, size_t wpa_ie_len,
477                         const u8 *mdie, size_t mdie_len)
478 {
479         struct wpa_ie_data data;
480         int ciphers, key_mgmt, res, version;
481         u32 selector;
482         size_t i;
483         const u8 *pmkid = NULL;
484
485         if (wpa_auth == NULL || sm == NULL)
486                 return WPA_NOT_ENABLED;
487
488         if (wpa_ie == NULL || wpa_ie_len < 1)
489                 return WPA_INVALID_IE;
490
491         if (wpa_ie[0] == WLAN_EID_RSN)
492                 version = WPA_PROTO_RSN;
493         else
494                 version = WPA_PROTO_WPA;
495
496         if (!(wpa_auth->conf.wpa & version)) {
497                 wpa_printf(MSG_DEBUG, "Invalid WPA proto (%d) from " MACSTR,
498                            version, MAC2STR(sm->addr));
499                 return WPA_INVALID_PROTO;
500         }
501
502         if (version == WPA_PROTO_RSN) {
503                 res = wpa_parse_wpa_ie_rsn(wpa_ie, wpa_ie_len, &data);
504
505                 selector = RSN_AUTH_KEY_MGMT_UNSPEC_802_1X;
506                 if (0) {
507                 }
508                 else if (data.key_mgmt & WPA_KEY_MGMT_IEEE8021X_SUITE_B_192)
509                         selector = RSN_AUTH_KEY_MGMT_802_1X_SUITE_B_192;
510                 else if (data.key_mgmt & WPA_KEY_MGMT_IEEE8021X_SUITE_B)
511                         selector = RSN_AUTH_KEY_MGMT_802_1X_SUITE_B;
512 #ifdef CONFIG_IEEE80211R
513                 else if (data.key_mgmt & WPA_KEY_MGMT_FT_IEEE8021X)
514                         selector = RSN_AUTH_KEY_MGMT_FT_802_1X;
515                 else if (data.key_mgmt & WPA_KEY_MGMT_FT_PSK)
516                         selector = RSN_AUTH_KEY_MGMT_FT_PSK;
517 #endif /* CONFIG_IEEE80211R */
518 #ifdef CONFIG_IEEE80211W
519                 else if (data.key_mgmt & WPA_KEY_MGMT_IEEE8021X_SHA256)
520                         selector = RSN_AUTH_KEY_MGMT_802_1X_SHA256;
521                 else if (data.key_mgmt & WPA_KEY_MGMT_PSK_SHA256)
522                         selector = RSN_AUTH_KEY_MGMT_PSK_SHA256;
523 #endif /* CONFIG_IEEE80211W */
524 #ifdef CONFIG_SAE
525                 else if (data.key_mgmt & WPA_KEY_MGMT_SAE)
526                         selector = RSN_AUTH_KEY_MGMT_SAE;
527                 else if (data.key_mgmt & WPA_KEY_MGMT_FT_SAE)
528                         selector = RSN_AUTH_KEY_MGMT_FT_SAE;
529 #endif /* CONFIG_SAE */
530                 else if (data.key_mgmt & WPA_KEY_MGMT_IEEE8021X)
531                         selector = RSN_AUTH_KEY_MGMT_UNSPEC_802_1X;
532                 else if (data.key_mgmt & WPA_KEY_MGMT_PSK)
533                         selector = RSN_AUTH_KEY_MGMT_PSK_OVER_802_1X;
534                 wpa_auth->dot11RSNAAuthenticationSuiteSelected = selector;
535
536                 selector = wpa_cipher_to_suite(WPA_PROTO_RSN,
537                                                data.pairwise_cipher);
538                 if (!selector)
539                         selector = RSN_CIPHER_SUITE_CCMP;
540                 wpa_auth->dot11RSNAPairwiseCipherSelected = selector;
541
542                 selector = wpa_cipher_to_suite(WPA_PROTO_RSN,
543                                                data.group_cipher);
544                 if (!selector)
545                         selector = RSN_CIPHER_SUITE_CCMP;
546                 wpa_auth->dot11RSNAGroupCipherSelected = selector;
547         } else {
548                 res = wpa_parse_wpa_ie_wpa(wpa_ie, wpa_ie_len, &data);
549
550                 selector = WPA_AUTH_KEY_MGMT_UNSPEC_802_1X;
551                 if (data.key_mgmt & WPA_KEY_MGMT_IEEE8021X)
552                         selector = WPA_AUTH_KEY_MGMT_UNSPEC_802_1X;
553                 else if (data.key_mgmt & WPA_KEY_MGMT_PSK)
554                         selector = WPA_AUTH_KEY_MGMT_PSK_OVER_802_1X;
555                 wpa_auth->dot11RSNAAuthenticationSuiteSelected = selector;
556
557                 selector = wpa_cipher_to_suite(WPA_PROTO_WPA,
558                                                data.pairwise_cipher);
559                 if (!selector)
560                         selector = RSN_CIPHER_SUITE_TKIP;
561                 wpa_auth->dot11RSNAPairwiseCipherSelected = selector;
562
563                 selector = wpa_cipher_to_suite(WPA_PROTO_WPA,
564                                                data.group_cipher);
565                 if (!selector)
566                         selector = WPA_CIPHER_SUITE_TKIP;
567                 wpa_auth->dot11RSNAGroupCipherSelected = selector;
568         }
569         if (res) {
570                 wpa_printf(MSG_DEBUG, "Failed to parse WPA/RSN IE from "
571                            MACSTR " (res=%d)", MAC2STR(sm->addr), res);
572                 wpa_hexdump(MSG_DEBUG, "WPA/RSN IE", wpa_ie, wpa_ie_len);
573                 return WPA_INVALID_IE;
574         }
575
576         if (data.group_cipher != wpa_auth->conf.wpa_group) {
577                 wpa_printf(MSG_DEBUG, "Invalid WPA group cipher (0x%x) from "
578                            MACSTR, data.group_cipher, MAC2STR(sm->addr));
579                 return WPA_INVALID_GROUP;
580         }
581
582         key_mgmt = data.key_mgmt & wpa_auth->conf.wpa_key_mgmt;
583         if (!key_mgmt) {
584                 wpa_printf(MSG_DEBUG, "Invalid WPA key mgmt (0x%x) from "
585                            MACSTR, data.key_mgmt, MAC2STR(sm->addr));
586                 return WPA_INVALID_AKMP;
587         }
588         if (0) {
589         }
590         else if (key_mgmt & WPA_KEY_MGMT_IEEE8021X_SUITE_B_192)
591                 sm->wpa_key_mgmt = WPA_KEY_MGMT_IEEE8021X_SUITE_B_192;
592         else if (key_mgmt & WPA_KEY_MGMT_IEEE8021X_SUITE_B)
593                 sm->wpa_key_mgmt = WPA_KEY_MGMT_IEEE8021X_SUITE_B;
594 #ifdef CONFIG_IEEE80211R
595         else if (key_mgmt & WPA_KEY_MGMT_FT_IEEE8021X)
596                 sm->wpa_key_mgmt = WPA_KEY_MGMT_FT_IEEE8021X;
597         else if (key_mgmt & WPA_KEY_MGMT_FT_PSK)
598                 sm->wpa_key_mgmt = WPA_KEY_MGMT_FT_PSK;
599 #endif /* CONFIG_IEEE80211R */
600 #ifdef CONFIG_IEEE80211W
601         else if (key_mgmt & WPA_KEY_MGMT_IEEE8021X_SHA256)
602                 sm->wpa_key_mgmt = WPA_KEY_MGMT_IEEE8021X_SHA256;
603         else if (key_mgmt & WPA_KEY_MGMT_PSK_SHA256)
604                 sm->wpa_key_mgmt = WPA_KEY_MGMT_PSK_SHA256;
605 #endif /* CONFIG_IEEE80211W */
606 #ifdef CONFIG_SAE
607         else if (key_mgmt & WPA_KEY_MGMT_SAE)
608                 sm->wpa_key_mgmt = WPA_KEY_MGMT_SAE;
609         else if (key_mgmt & WPA_KEY_MGMT_FT_SAE)
610                 sm->wpa_key_mgmt = WPA_KEY_MGMT_FT_SAE;
611 #endif /* CONFIG_SAE */
612         else if (key_mgmt & WPA_KEY_MGMT_IEEE8021X)
613                 sm->wpa_key_mgmt = WPA_KEY_MGMT_IEEE8021X;
614         else
615                 sm->wpa_key_mgmt = WPA_KEY_MGMT_PSK;
616
617         if (version == WPA_PROTO_RSN)
618                 ciphers = data.pairwise_cipher & wpa_auth->conf.rsn_pairwise;
619         else
620                 ciphers = data.pairwise_cipher & wpa_auth->conf.wpa_pairwise;
621         if (!ciphers) {
622                 wpa_printf(MSG_DEBUG, "Invalid %s pairwise cipher (0x%x) "
623                            "from " MACSTR,
624                            version == WPA_PROTO_RSN ? "RSN" : "WPA",
625                            data.pairwise_cipher, MAC2STR(sm->addr));
626                 return WPA_INVALID_PAIRWISE;
627         }
628
629 #ifdef CONFIG_IEEE80211W
630         if (wpa_auth->conf.ieee80211w == MGMT_FRAME_PROTECTION_REQUIRED) {
631                 if (!(data.capabilities & WPA_CAPABILITY_MFPC)) {
632                         wpa_printf(MSG_DEBUG, "Management frame protection "
633                                    "required, but client did not enable it");
634                         return WPA_MGMT_FRAME_PROTECTION_VIOLATION;
635                 }
636
637                 if (ciphers & WPA_CIPHER_TKIP) {
638                         wpa_printf(MSG_DEBUG, "Management frame protection "
639                                    "cannot use TKIP");
640                         return WPA_MGMT_FRAME_PROTECTION_VIOLATION;
641                 }
642
643                 if (data.mgmt_group_cipher != wpa_auth->conf.group_mgmt_cipher)
644                 {
645                         wpa_printf(MSG_DEBUG, "Unsupported management group "
646                                    "cipher %d", data.mgmt_group_cipher);
647                         return WPA_INVALID_MGMT_GROUP_CIPHER;
648                 }
649         }
650
651         if (wpa_auth->conf.ieee80211w == NO_MGMT_FRAME_PROTECTION ||
652             !(data.capabilities & WPA_CAPABILITY_MFPC))
653                 sm->mgmt_frame_prot = 0;
654         else
655                 sm->mgmt_frame_prot = 1;
656 #endif /* CONFIG_IEEE80211W */
657
658 #ifdef CONFIG_IEEE80211R
659         if (wpa_key_mgmt_ft(sm->wpa_key_mgmt)) {
660                 if (mdie == NULL || mdie_len < MOBILITY_DOMAIN_ID_LEN + 1) {
661                         wpa_printf(MSG_DEBUG, "RSN: Trying to use FT, but "
662                                    "MDIE not included");
663                         return WPA_INVALID_MDIE;
664                 }
665                 if (os_memcmp(mdie, wpa_auth->conf.mobility_domain,
666                               MOBILITY_DOMAIN_ID_LEN) != 0) {
667                         wpa_hexdump(MSG_DEBUG, "RSN: Attempted to use unknown "
668                                     "MDIE", mdie, MOBILITY_DOMAIN_ID_LEN);
669                         return WPA_INVALID_MDIE;
670                 }
671         }
672 #endif /* CONFIG_IEEE80211R */
673
674         sm->pairwise = wpa_pick_pairwise_cipher(ciphers, 0);
675         if (sm->pairwise < 0)
676                 return WPA_INVALID_PAIRWISE;
677
678         /* TODO: clear WPA/WPA2 state if STA changes from one to another */
679         if (wpa_ie[0] == WLAN_EID_RSN)
680                 sm->wpa = WPA_VERSION_WPA2;
681         else
682                 sm->wpa = WPA_VERSION_WPA;
683
684         sm->pmksa = NULL;
685         for (i = 0; i < data.num_pmkid; i++) {
686                 wpa_hexdump(MSG_DEBUG, "RSN IE: STA PMKID",
687                             &data.pmkid[i * PMKID_LEN], PMKID_LEN);
688                 sm->pmksa = pmksa_cache_auth_get(wpa_auth->pmksa, sm->addr,
689                                                  &data.pmkid[i * PMKID_LEN]);
690                 if (sm->pmksa) {
691                         pmkid = sm->pmksa->pmkid;
692                         break;
693                 }
694         }
695         for (i = 0; sm->pmksa == NULL && wpa_auth->conf.okc &&
696                      i < data.num_pmkid; i++) {
697                 struct wpa_auth_okc_iter_data idata;
698                 idata.pmksa = NULL;
699                 idata.aa = wpa_auth->addr;
700                 idata.spa = sm->addr;
701                 idata.pmkid = &data.pmkid[i * PMKID_LEN];
702                 wpa_auth_for_each_auth(wpa_auth, wpa_auth_okc_iter, &idata);
703                 if (idata.pmksa) {
704                         wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_DEBUG,
705                                          "OKC match for PMKID");
706                         sm->pmksa = pmksa_cache_add_okc(wpa_auth->pmksa,
707                                                         idata.pmksa,
708                                                         wpa_auth->addr,
709                                                         idata.pmkid);
710                         pmkid = idata.pmkid;
711                         break;
712                 }
713         }
714         if (sm->pmksa && pmkid) {
715                 wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_DEBUG,
716                                  "PMKID found from PMKSA cache "
717                                  "eap_type=%d vlan_id=%d",
718                                  sm->pmksa->eap_type_authsrv,
719                                  sm->pmksa->vlan_id);
720                 os_memcpy(wpa_auth->dot11RSNAPMKIDUsed, pmkid, PMKID_LEN);
721         }
722
723         if (sm->wpa_ie == NULL || sm->wpa_ie_len < wpa_ie_len) {
724                 os_free(sm->wpa_ie);
725                 sm->wpa_ie = os_malloc(wpa_ie_len);
726                 if (sm->wpa_ie == NULL)
727                         return WPA_ALLOC_FAIL;
728         }
729         os_memcpy(sm->wpa_ie, wpa_ie, wpa_ie_len);
730         sm->wpa_ie_len = wpa_ie_len;
731
732         return WPA_IE_OK;
733 }
734
735
736 #ifdef CONFIG_HS20
737 int wpa_validate_osen(struct wpa_authenticator *wpa_auth,
738                       struct wpa_state_machine *sm,
739                       const u8 *osen_ie, size_t osen_ie_len)
740 {
741         if (wpa_auth == NULL || sm == NULL)
742                 return -1;
743
744         /* TODO: parse OSEN element */
745         sm->wpa_key_mgmt = WPA_KEY_MGMT_OSEN;
746         sm->mgmt_frame_prot = 1;
747         sm->pairwise = WPA_CIPHER_CCMP;
748         sm->wpa = WPA_VERSION_WPA2;
749
750         if (sm->wpa_ie == NULL || sm->wpa_ie_len < osen_ie_len) {
751                 os_free(sm->wpa_ie);
752                 sm->wpa_ie = os_malloc(osen_ie_len);
753                 if (sm->wpa_ie == NULL)
754                         return -1;
755         }
756
757         os_memcpy(sm->wpa_ie, osen_ie, osen_ie_len);
758         sm->wpa_ie_len = osen_ie_len;
759
760         return 0;
761 }
762
763 #endif /* CONFIG_HS20 */
764
765
766 /**
767  * wpa_parse_generic - Parse EAPOL-Key Key Data Generic IEs
768  * @pos: Pointer to the IE header
769  * @end: Pointer to the end of the Key Data buffer
770  * @ie: Pointer to parsed IE data
771  * Returns: 0 on success, 1 if end mark is found, -1 on failure
772  */
773 static int wpa_parse_generic(const u8 *pos, const u8 *end,
774                              struct wpa_eapol_ie_parse *ie)
775 {
776         if (pos[1] == 0)
777                 return 1;
778
779         if (pos[1] >= 6 &&
780             RSN_SELECTOR_GET(pos + 2) == WPA_OUI_TYPE &&
781             pos[2 + WPA_SELECTOR_LEN] == 1 &&
782             pos[2 + WPA_SELECTOR_LEN + 1] == 0) {
783                 ie->wpa_ie = pos;
784                 ie->wpa_ie_len = pos[1] + 2;
785                 return 0;
786         }
787
788         if (pos[1] >= 4 && WPA_GET_BE32(pos + 2) == OSEN_IE_VENDOR_TYPE) {
789                 ie->osen = pos;
790                 ie->osen_len = pos[1] + 2;
791                 return 0;
792         }
793
794         if (pos + 1 + RSN_SELECTOR_LEN < end &&
795             pos[1] >= RSN_SELECTOR_LEN + PMKID_LEN &&
796             RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_PMKID) {
797                 ie->pmkid = pos + 2 + RSN_SELECTOR_LEN;
798                 return 0;
799         }
800
801         if (pos[1] > RSN_SELECTOR_LEN + 2 &&
802             RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_GROUPKEY) {
803                 ie->gtk = pos + 2 + RSN_SELECTOR_LEN;
804                 ie->gtk_len = pos[1] - RSN_SELECTOR_LEN;
805                 return 0;
806         }
807
808         if (pos[1] > RSN_SELECTOR_LEN + 2 &&
809             RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_MAC_ADDR) {
810                 ie->mac_addr = pos + 2 + RSN_SELECTOR_LEN;
811                 ie->mac_addr_len = pos[1] - RSN_SELECTOR_LEN;
812                 return 0;
813         }
814
815 #ifdef CONFIG_PEERKEY
816         if (pos[1] > RSN_SELECTOR_LEN + 2 &&
817             RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_SMK) {
818                 ie->smk = pos + 2 + RSN_SELECTOR_LEN;
819                 ie->smk_len = pos[1] - RSN_SELECTOR_LEN;
820                 return 0;
821         }
822
823         if (pos[1] > RSN_SELECTOR_LEN + 2 &&
824             RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_NONCE) {
825                 ie->nonce = pos + 2 + RSN_SELECTOR_LEN;
826                 ie->nonce_len = pos[1] - RSN_SELECTOR_LEN;
827                 return 0;
828         }
829
830         if (pos[1] > RSN_SELECTOR_LEN + 2 &&
831             RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_LIFETIME) {
832                 ie->lifetime = pos + 2 + RSN_SELECTOR_LEN;
833                 ie->lifetime_len = pos[1] - RSN_SELECTOR_LEN;
834                 return 0;
835         }
836
837         if (pos[1] > RSN_SELECTOR_LEN + 2 &&
838             RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_ERROR) {
839                 ie->error = pos + 2 + RSN_SELECTOR_LEN;
840                 ie->error_len = pos[1] - RSN_SELECTOR_LEN;
841                 return 0;
842         }
843 #endif /* CONFIG_PEERKEY */
844
845 #ifdef CONFIG_IEEE80211W
846         if (pos[1] > RSN_SELECTOR_LEN + 2 &&
847             RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_IGTK) {
848                 ie->igtk = pos + 2 + RSN_SELECTOR_LEN;
849                 ie->igtk_len = pos[1] - RSN_SELECTOR_LEN;
850                 return 0;
851         }
852 #endif /* CONFIG_IEEE80211W */
853
854 #ifdef CONFIG_P2P
855         if (pos[1] >= RSN_SELECTOR_LEN + 1 &&
856             RSN_SELECTOR_GET(pos + 2) == WFA_KEY_DATA_IP_ADDR_REQ) {
857                 ie->ip_addr_req = pos + 2 + RSN_SELECTOR_LEN;
858                 wpa_hexdump(MSG_DEBUG, "WPA: IP Address Request in EAPOL-Key",
859                             ie->ip_addr_req, pos[1] - RSN_SELECTOR_LEN);
860                 return 0;
861         }
862
863         if (pos[1] >= RSN_SELECTOR_LEN + 3 * 4 &&
864             RSN_SELECTOR_GET(pos + 2) == WFA_KEY_DATA_IP_ADDR_ALLOC) {
865                 ie->ip_addr_alloc = pos + 2 + RSN_SELECTOR_LEN;
866                 wpa_hexdump(MSG_DEBUG,
867                             "WPA: IP Address Allocation in EAPOL-Key",
868                             ie->ip_addr_alloc, pos[1] - RSN_SELECTOR_LEN);
869                 return 0;
870         }
871 #endif /* CONFIG_P2P */
872
873         return 0;
874 }
875
876
877 /**
878  * wpa_parse_kde_ies - Parse EAPOL-Key Key Data IEs
879  * @buf: Pointer to the Key Data buffer
880  * @len: Key Data Length
881  * @ie: Pointer to parsed IE data
882  * Returns: 0 on success, -1 on failure
883  */
884 int wpa_parse_kde_ies(const u8 *buf, size_t len, struct wpa_eapol_ie_parse *ie)
885 {
886         const u8 *pos, *end;
887         int ret = 0;
888
889         os_memset(ie, 0, sizeof(*ie));
890         for (pos = buf, end = pos + len; pos + 1 < end; pos += 2 + pos[1]) {
891                 if (pos[0] == 0xdd &&
892                     ((pos == buf + len - 1) || pos[1] == 0)) {
893                         /* Ignore padding */
894                         break;
895                 }
896                 if (pos + 2 + pos[1] > end) {
897                         wpa_printf(MSG_DEBUG, "WPA: EAPOL-Key Key Data "
898                                    "underflow (ie=%d len=%d pos=%d)",
899                                    pos[0], pos[1], (int) (pos - buf));
900                         wpa_hexdump_key(MSG_DEBUG, "WPA: Key Data",
901                                         buf, len);
902                         ret = -1;
903                         break;
904                 }
905                 if (*pos == WLAN_EID_RSN) {
906                         ie->rsn_ie = pos;
907                         ie->rsn_ie_len = pos[1] + 2;
908 #ifdef CONFIG_IEEE80211R
909                 } else if (*pos == WLAN_EID_MOBILITY_DOMAIN) {
910                         ie->mdie = pos;
911                         ie->mdie_len = pos[1] + 2;
912                 } else if (*pos == WLAN_EID_FAST_BSS_TRANSITION) {
913                         ie->ftie = pos;
914                         ie->ftie_len = pos[1] + 2;
915 #endif /* CONFIG_IEEE80211R */
916                 } else if (*pos == WLAN_EID_VENDOR_SPECIFIC) {
917                         ret = wpa_parse_generic(pos, end, ie);
918                         if (ret < 0)
919                                 break;
920                         if (ret > 0) {
921                                 ret = 0;
922                                 break;
923                         }
924                 } else {
925                         wpa_hexdump(MSG_DEBUG, "WPA: Unrecognized EAPOL-Key "
926                                     "Key Data IE", pos, 2 + pos[1]);
927                 }
928         }
929
930         return ret;
931 }
932
933
934 int wpa_auth_uses_mfp(struct wpa_state_machine *sm)
935 {
936         return sm ? sm->mgmt_frame_prot : 0;
937 }