WPS: Fix association when both WPS and WPA/RSN IE are included
[libeap.git] / src / ap / ieee802_11.c
1 /*
2  * hostapd / IEEE 802.11 Management
3  * Copyright (c) 2002-2010, Jouni Malinen <j@w1.fi>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation.
8  *
9  * Alternatively, this software may be distributed under the terms of BSD
10  * license.
11  *
12  * See README and COPYING for more details.
13  */
14
15 #include "utils/includes.h"
16
17 #ifndef CONFIG_NATIVE_WINDOWS
18
19 #include "utils/common.h"
20 #include "utils/eloop.h"
21 #include "crypto/crypto.h"
22 #include "drivers/driver.h"
23 #include "common/ieee802_11_defs.h"
24 #include "common/ieee802_11_common.h"
25 #include "common/wpa_ctrl.h"
26 #include "radius/radius.h"
27 #include "radius/radius_client.h"
28 #include "hostapd.h"
29 #include "beacon.h"
30 #include "ieee802_11_auth.h"
31 #include "sta_info.h"
32 #include "ieee802_1x.h"
33 #include "wpa_auth.h"
34 #include "wmm.h"
35 #include "ap_list.h"
36 #include "accounting.h"
37 #include "ap_config.h"
38 #include "ap_mlme.h"
39 #include "ieee802_11.h"
40
41
42 u8 * hostapd_eid_supp_rates(struct hostapd_data *hapd, u8 *eid)
43 {
44         u8 *pos = eid;
45         int i, num, count;
46
47         if (hapd->iface->current_rates == NULL)
48                 return eid;
49
50         *pos++ = WLAN_EID_SUPP_RATES;
51         num = hapd->iface->num_rates;
52         if (num > 8) {
53                 /* rest of the rates are encoded in Extended supported
54                  * rates element */
55                 num = 8;
56         }
57
58         *pos++ = num;
59         count = 0;
60         for (i = 0, count = 0; i < hapd->iface->num_rates && count < num;
61              i++) {
62                 count++;
63                 *pos = hapd->iface->current_rates[i].rate / 5;
64                 if (hapd->iface->current_rates[i].flags & HOSTAPD_RATE_BASIC)
65                         *pos |= 0x80;
66                 pos++;
67         }
68
69         return pos;
70 }
71
72
73 u8 * hostapd_eid_ext_supp_rates(struct hostapd_data *hapd, u8 *eid)
74 {
75         u8 *pos = eid;
76         int i, num, count;
77
78         if (hapd->iface->current_rates == NULL)
79                 return eid;
80
81         num = hapd->iface->num_rates;
82         if (num <= 8)
83                 return eid;
84         num -= 8;
85
86         *pos++ = WLAN_EID_EXT_SUPP_RATES;
87         *pos++ = num;
88         count = 0;
89         for (i = 0, count = 0; i < hapd->iface->num_rates && count < num + 8;
90              i++) {
91                 count++;
92                 if (count <= 8)
93                         continue; /* already in SuppRates IE */
94                 *pos = hapd->iface->current_rates[i].rate / 5;
95                 if (hapd->iface->current_rates[i].flags & HOSTAPD_RATE_BASIC)
96                         *pos |= 0x80;
97                 pos++;
98         }
99
100         return pos;
101 }
102
103
104 u16 hostapd_own_capab_info(struct hostapd_data *hapd, struct sta_info *sta,
105                            int probe)
106 {
107         int capab = WLAN_CAPABILITY_ESS;
108         int privacy;
109
110         if (hapd->iface->num_sta_no_short_preamble == 0 &&
111             hapd->iconf->preamble == SHORT_PREAMBLE)
112                 capab |= WLAN_CAPABILITY_SHORT_PREAMBLE;
113
114         privacy = hapd->conf->ssid.wep.keys_set;
115
116         if (hapd->conf->ieee802_1x &&
117             (hapd->conf->default_wep_key_len ||
118              hapd->conf->individual_wep_key_len))
119                 privacy = 1;
120
121         if (hapd->conf->wpa)
122                 privacy = 1;
123
124         if (sta) {
125                 int policy, def_klen;
126                 if (probe && sta->ssid_probe) {
127                         policy = sta->ssid_probe->security_policy;
128                         def_klen = sta->ssid_probe->wep.default_len;
129                 } else {
130                         policy = sta->ssid->security_policy;
131                         def_klen = sta->ssid->wep.default_len;
132                 }
133                 privacy = policy != SECURITY_PLAINTEXT;
134                 if (policy == SECURITY_IEEE_802_1X && def_klen == 0)
135                         privacy = 0;
136         }
137
138         if (privacy)
139                 capab |= WLAN_CAPABILITY_PRIVACY;
140
141         if (hapd->iface->current_mode &&
142             hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G &&
143             hapd->iface->num_sta_no_short_slot_time == 0)
144                 capab |= WLAN_CAPABILITY_SHORT_SLOT_TIME;
145
146         return capab;
147 }
148
149
150 #ifdef CONFIG_IEEE80211W
151 static u8 * hostapd_eid_assoc_comeback_time(struct hostapd_data *hapd,
152                                             struct sta_info *sta, u8 *eid)
153 {
154         u8 *pos = eid;
155         u32 timeout, tu;
156         struct os_time now, passed;
157
158         *pos++ = WLAN_EID_TIMEOUT_INTERVAL;
159         *pos++ = 5;
160         *pos++ = WLAN_TIMEOUT_ASSOC_COMEBACK;
161         os_get_time(&now);
162         os_time_sub(&now, &sta->sa_query_start, &passed);
163         tu = (passed.sec * 1000000 + passed.usec) / 1024;
164         if (hapd->conf->assoc_sa_query_max_timeout > tu)
165                 timeout = hapd->conf->assoc_sa_query_max_timeout - tu;
166         else
167                 timeout = 0;
168         if (timeout < hapd->conf->assoc_sa_query_max_timeout)
169                 timeout++; /* add some extra time for local timers */
170         WPA_PUT_LE32(pos, timeout);
171         pos += 4;
172
173         return pos;
174 }
175 #endif /* CONFIG_IEEE80211W */
176
177
178 void ieee802_11_print_ssid(char *buf, const u8 *ssid, u8 len)
179 {
180         int i;
181         if (len > HOSTAPD_MAX_SSID_LEN)
182                 len = HOSTAPD_MAX_SSID_LEN;
183         for (i = 0; i < len; i++) {
184                 if (ssid[i] >= 32 && ssid[i] < 127)
185                         buf[i] = ssid[i];
186                 else
187                         buf[i] = '.';
188         }
189         buf[len] = '\0';
190 }
191
192
193 /**
194  * ieee802_11_send_deauth - Send Deauthentication frame
195  * @hapd: hostapd BSS data
196  * @addr: Address of the destination STA
197  * @reason: Reason code for Deauthentication
198  */
199 void ieee802_11_send_deauth(struct hostapd_data *hapd, const u8 *addr,
200                             u16 reason)
201 {
202         struct ieee80211_mgmt mgmt;
203
204         hostapd_logger(hapd, addr, HOSTAPD_MODULE_IEEE80211,
205                        HOSTAPD_LEVEL_DEBUG,
206                        "deauthenticate - reason %d", reason);
207         os_memset(&mgmt, 0, sizeof(mgmt));
208         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
209                                           WLAN_FC_STYPE_DEAUTH);
210         os_memcpy(mgmt.da, addr, ETH_ALEN);
211         os_memcpy(mgmt.sa, hapd->own_addr, ETH_ALEN);
212         os_memcpy(mgmt.bssid, hapd->own_addr, ETH_ALEN);
213         mgmt.u.deauth.reason_code = host_to_le16(reason);
214         if (hapd->drv.send_mgmt_frame(hapd, &mgmt, IEEE80211_HDRLEN +
215                                        sizeof(mgmt.u.deauth)) < 0)
216                 perror("ieee802_11_send_deauth: send");
217 }
218
219
220 static u16 auth_shared_key(struct hostapd_data *hapd, struct sta_info *sta,
221                            u16 auth_transaction, const u8 *challenge,
222                            int iswep)
223 {
224         hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
225                        HOSTAPD_LEVEL_DEBUG,
226                        "authentication (shared key, transaction %d)",
227                        auth_transaction);
228
229         if (auth_transaction == 1) {
230                 if (!sta->challenge) {
231                         /* Generate a pseudo-random challenge */
232                         u8 key[8];
233                         time_t now;
234                         int r;
235                         sta->challenge = os_zalloc(WLAN_AUTH_CHALLENGE_LEN);
236                         if (sta->challenge == NULL)
237                                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
238
239                         now = time(NULL);
240                         r = random();
241                         os_memcpy(key, &now, 4);
242                         os_memcpy(key + 4, &r, 4);
243                         rc4_skip(key, sizeof(key), 0,
244                                  sta->challenge, WLAN_AUTH_CHALLENGE_LEN);
245                 }
246                 return 0;
247         }
248
249         if (auth_transaction != 3)
250                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
251
252         /* Transaction 3 */
253         if (!iswep || !sta->challenge || !challenge ||
254             os_memcmp(sta->challenge, challenge, WLAN_AUTH_CHALLENGE_LEN)) {
255                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
256                                HOSTAPD_LEVEL_INFO,
257                                "shared key authentication - invalid "
258                                "challenge-response");
259                 return WLAN_STATUS_CHALLENGE_FAIL;
260         }
261
262         hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
263                        HOSTAPD_LEVEL_DEBUG,
264                        "authentication OK (shared key)");
265 #ifdef IEEE80211_REQUIRE_AUTH_ACK
266         /* Station will be marked authenticated if it ACKs the
267          * authentication reply. */
268 #else
269         sta->flags |= WLAN_STA_AUTH;
270         wpa_auth_sm_event(sta->wpa_sm, WPA_AUTH);
271 #endif
272         os_free(sta->challenge);
273         sta->challenge = NULL;
274
275         return 0;
276 }
277
278
279 static void send_auth_reply(struct hostapd_data *hapd,
280                             const u8 *dst, const u8 *bssid,
281                             u16 auth_alg, u16 auth_transaction, u16 resp,
282                             const u8 *ies, size_t ies_len)
283 {
284         struct ieee80211_mgmt *reply;
285         u8 *buf;
286         size_t rlen;
287
288         rlen = IEEE80211_HDRLEN + sizeof(reply->u.auth) + ies_len;
289         buf = os_zalloc(rlen);
290         if (buf == NULL)
291                 return;
292
293         reply = (struct ieee80211_mgmt *) buf;
294         reply->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
295                                             WLAN_FC_STYPE_AUTH);
296         os_memcpy(reply->da, dst, ETH_ALEN);
297         os_memcpy(reply->sa, hapd->own_addr, ETH_ALEN);
298         os_memcpy(reply->bssid, bssid, ETH_ALEN);
299
300         reply->u.auth.auth_alg = host_to_le16(auth_alg);
301         reply->u.auth.auth_transaction = host_to_le16(auth_transaction);
302         reply->u.auth.status_code = host_to_le16(resp);
303
304         if (ies && ies_len)
305                 os_memcpy(reply->u.auth.variable, ies, ies_len);
306
307         wpa_printf(MSG_DEBUG, "authentication reply: STA=" MACSTR
308                    " auth_alg=%d auth_transaction=%d resp=%d (IE len=%lu)",
309                    MAC2STR(dst), auth_alg, auth_transaction,
310                    resp, (unsigned long) ies_len);
311         if (hapd->drv.send_mgmt_frame(hapd, reply, rlen) < 0)
312                 perror("send_auth_reply: send");
313
314         os_free(buf);
315 }
316
317
318 #ifdef CONFIG_IEEE80211R
319 static void handle_auth_ft_finish(void *ctx, const u8 *dst, const u8 *bssid,
320                                   u16 auth_transaction, u16 status,
321                                   const u8 *ies, size_t ies_len)
322 {
323         struct hostapd_data *hapd = ctx;
324         struct sta_info *sta;
325
326         send_auth_reply(hapd, dst, bssid, WLAN_AUTH_FT, auth_transaction,
327                         status, ies, ies_len);
328
329         if (status != WLAN_STATUS_SUCCESS)
330                 return;
331
332         sta = ap_get_sta(hapd, dst);
333         if (sta == NULL)
334                 return;
335
336         hostapd_logger(hapd, dst, HOSTAPD_MODULE_IEEE80211,
337                        HOSTAPD_LEVEL_DEBUG, "authentication OK (FT)");
338         sta->flags |= WLAN_STA_AUTH;
339         mlme_authenticate_indication(hapd, sta);
340 }
341 #endif /* CONFIG_IEEE80211R */
342
343
344 static void handle_auth(struct hostapd_data *hapd,
345                         const struct ieee80211_mgmt *mgmt, size_t len)
346 {
347         u16 auth_alg, auth_transaction, status_code;
348         u16 resp = WLAN_STATUS_SUCCESS;
349         struct sta_info *sta = NULL;
350         int res;
351         u16 fc;
352         const u8 *challenge = NULL;
353         u32 session_timeout, acct_interim_interval;
354         int vlan_id = 0;
355         u8 resp_ies[2 + WLAN_AUTH_CHALLENGE_LEN];
356         size_t resp_ies_len = 0;
357
358         if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.auth)) {
359                 printf("handle_auth - too short payload (len=%lu)\n",
360                        (unsigned long) len);
361                 return;
362         }
363
364         auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
365         auth_transaction = le_to_host16(mgmt->u.auth.auth_transaction);
366         status_code = le_to_host16(mgmt->u.auth.status_code);
367         fc = le_to_host16(mgmt->frame_control);
368
369         if (len >= IEEE80211_HDRLEN + sizeof(mgmt->u.auth) +
370             2 + WLAN_AUTH_CHALLENGE_LEN &&
371             mgmt->u.auth.variable[0] == WLAN_EID_CHALLENGE &&
372             mgmt->u.auth.variable[1] == WLAN_AUTH_CHALLENGE_LEN)
373                 challenge = &mgmt->u.auth.variable[2];
374
375         wpa_printf(MSG_DEBUG, "authentication: STA=" MACSTR " auth_alg=%d "
376                    "auth_transaction=%d status_code=%d wep=%d%s",
377                    MAC2STR(mgmt->sa), auth_alg, auth_transaction,
378                    status_code, !!(fc & WLAN_FC_ISWEP),
379                    challenge ? " challenge" : "");
380
381         if (hapd->tkip_countermeasures) {
382                 resp = WLAN_REASON_MICHAEL_MIC_FAILURE;
383                 goto fail;
384         }
385
386         if (!(((hapd->conf->auth_algs & WPA_AUTH_ALG_OPEN) &&
387                auth_alg == WLAN_AUTH_OPEN) ||
388 #ifdef CONFIG_IEEE80211R
389               (hapd->conf->wpa &&
390                (hapd->conf->wpa_key_mgmt &
391                 (WPA_KEY_MGMT_FT_IEEE8021X | WPA_KEY_MGMT_FT_PSK)) &&
392                auth_alg == WLAN_AUTH_FT) ||
393 #endif /* CONFIG_IEEE80211R */
394               ((hapd->conf->auth_algs & WPA_AUTH_ALG_SHARED) &&
395                auth_alg == WLAN_AUTH_SHARED_KEY))) {
396                 printf("Unsupported authentication algorithm (%d)\n",
397                        auth_alg);
398                 resp = WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG;
399                 goto fail;
400         }
401
402         if (!(auth_transaction == 1 ||
403               (auth_alg == WLAN_AUTH_SHARED_KEY && auth_transaction == 3))) {
404                 printf("Unknown authentication transaction number (%d)\n",
405                        auth_transaction);
406                 resp = WLAN_STATUS_UNKNOWN_AUTH_TRANSACTION;
407                 goto fail;
408         }
409
410         if (os_memcmp(mgmt->sa, hapd->own_addr, ETH_ALEN) == 0) {
411                 printf("Station " MACSTR " not allowed to authenticate.\n",
412                        MAC2STR(mgmt->sa));
413                 resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
414                 goto fail;
415         }
416
417         res = hostapd_allowed_address(hapd, mgmt->sa, (u8 *) mgmt, len,
418                                       &session_timeout,
419                                       &acct_interim_interval, &vlan_id);
420         if (res == HOSTAPD_ACL_REJECT) {
421                 printf("Station " MACSTR " not allowed to authenticate.\n",
422                        MAC2STR(mgmt->sa));
423                 resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
424                 goto fail;
425         }
426         if (res == HOSTAPD_ACL_PENDING) {
427                 wpa_printf(MSG_DEBUG, "Authentication frame from " MACSTR
428                            " waiting for an external authentication",
429                            MAC2STR(mgmt->sa));
430                 /* Authentication code will re-send the authentication frame
431                  * after it has received (and cached) information from the
432                  * external source. */
433                 return;
434         }
435
436         sta = ap_sta_add(hapd, mgmt->sa);
437         if (!sta) {
438                 resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
439                 goto fail;
440         }
441
442         if (vlan_id > 0) {
443                 if (hostapd_get_vlan_id_ifname(hapd->conf->vlan,
444                                                vlan_id) == NULL) {
445                         hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_RADIUS,
446                                        HOSTAPD_LEVEL_INFO, "Invalid VLAN ID "
447                                        "%d received from RADIUS server",
448                                        vlan_id);
449                         resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
450                         goto fail;
451                 }
452                 sta->vlan_id = vlan_id;
453                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_RADIUS,
454                                HOSTAPD_LEVEL_INFO, "VLAN ID %d", sta->vlan_id);
455         }
456
457         sta->flags &= ~WLAN_STA_PREAUTH;
458         ieee802_1x_notify_pre_auth(sta->eapol_sm, 0);
459
460         if (hapd->conf->acct_interim_interval == 0 && acct_interim_interval)
461                 sta->acct_interim_interval = acct_interim_interval;
462         if (res == HOSTAPD_ACL_ACCEPT_TIMEOUT)
463                 ap_sta_session_timeout(hapd, sta, session_timeout);
464         else
465                 ap_sta_no_session_timeout(hapd, sta);
466
467         switch (auth_alg) {
468         case WLAN_AUTH_OPEN:
469                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
470                                HOSTAPD_LEVEL_DEBUG,
471                                "authentication OK (open system)");
472 #ifdef IEEE80211_REQUIRE_AUTH_ACK
473                 /* Station will be marked authenticated if it ACKs the
474                  * authentication reply. */
475 #else
476                 sta->flags |= WLAN_STA_AUTH;
477                 wpa_auth_sm_event(sta->wpa_sm, WPA_AUTH);
478                 sta->auth_alg = WLAN_AUTH_OPEN;
479                 mlme_authenticate_indication(hapd, sta);
480 #endif
481                 break;
482         case WLAN_AUTH_SHARED_KEY:
483                 resp = auth_shared_key(hapd, sta, auth_transaction, challenge,
484                                        fc & WLAN_FC_ISWEP);
485                 sta->auth_alg = WLAN_AUTH_SHARED_KEY;
486                 mlme_authenticate_indication(hapd, sta);
487                 if (sta->challenge && auth_transaction == 1) {
488                         resp_ies[0] = WLAN_EID_CHALLENGE;
489                         resp_ies[1] = WLAN_AUTH_CHALLENGE_LEN;
490                         os_memcpy(resp_ies + 2, sta->challenge,
491                                   WLAN_AUTH_CHALLENGE_LEN);
492                         resp_ies_len = 2 + WLAN_AUTH_CHALLENGE_LEN;
493                 }
494                 break;
495 #ifdef CONFIG_IEEE80211R
496         case WLAN_AUTH_FT:
497                 sta->auth_alg = WLAN_AUTH_FT;
498                 if (sta->wpa_sm == NULL)
499                         sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,
500                                                         sta->addr);
501                 if (sta->wpa_sm == NULL) {
502                         wpa_printf(MSG_DEBUG, "FT: Failed to initialize WPA "
503                                    "state machine");
504                         resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
505                         goto fail;
506                 }
507                 wpa_ft_process_auth(sta->wpa_sm, mgmt->bssid,
508                                     auth_transaction, mgmt->u.auth.variable,
509                                     len - IEEE80211_HDRLEN -
510                                     sizeof(mgmt->u.auth),
511                                     handle_auth_ft_finish, hapd);
512                 /* handle_auth_ft_finish() callback will complete auth. */
513                 return;
514 #endif /* CONFIG_IEEE80211R */
515         }
516
517  fail:
518         send_auth_reply(hapd, mgmt->sa, mgmt->bssid, auth_alg,
519                         auth_transaction + 1, resp, resp_ies, resp_ies_len);
520 }
521
522
523 static int hostapd_get_aid(struct hostapd_data *hapd, struct sta_info *sta)
524 {
525         int i, j = 32, aid;
526
527         /* get a unique AID */
528         if (sta->aid > 0) {
529                 wpa_printf(MSG_DEBUG, "  old AID %d", sta->aid);
530                 return 0;
531         }
532
533         for (i = 0; i < AID_WORDS; i++) {
534                 if (hapd->sta_aid[i] == (u32) -1)
535                         continue;
536                 for (j = 0; j < 32; j++) {
537                         if (!(hapd->sta_aid[i] & BIT(j)))
538                                 break;
539                 }
540                 if (j < 32)
541                         break;
542         }
543         if (j == 32)
544                 return -1;
545         aid = i * 32 + j + 1;
546         if (aid > 2007)
547                 return -1;
548
549         sta->aid = aid;
550         hapd->sta_aid[i] |= BIT(j);
551         wpa_printf(MSG_DEBUG, "  new AID %d", sta->aid);
552         return 0;
553 }
554
555
556 static u16 check_ssid(struct hostapd_data *hapd, struct sta_info *sta,
557                       const u8 *ssid_ie, size_t ssid_ie_len)
558 {
559         if (ssid_ie == NULL)
560                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
561
562         if (ssid_ie_len != hapd->conf->ssid.ssid_len ||
563             os_memcmp(ssid_ie, hapd->conf->ssid.ssid, ssid_ie_len) != 0) {
564                 char ssid_txt[33];
565                 ieee802_11_print_ssid(ssid_txt, ssid_ie, ssid_ie_len);
566                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
567                                HOSTAPD_LEVEL_INFO,
568                                "Station tried to associate with unknown SSID "
569                                "'%s'", ssid_txt);
570                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
571         }
572
573         return WLAN_STATUS_SUCCESS;
574 }
575
576
577 static u16 check_wmm(struct hostapd_data *hapd, struct sta_info *sta,
578                      const u8 *wmm_ie, size_t wmm_ie_len)
579 {
580         sta->flags &= ~WLAN_STA_WMM;
581         if (wmm_ie && hapd->conf->wmm_enabled) {
582                 if (hostapd_eid_wmm_valid(hapd, wmm_ie, wmm_ie_len))
583                         hostapd_logger(hapd, sta->addr,
584                                        HOSTAPD_MODULE_WPA,
585                                        HOSTAPD_LEVEL_DEBUG,
586                                        "invalid WMM element in association "
587                                        "request");
588                 else
589                         sta->flags |= WLAN_STA_WMM;
590         }
591         return WLAN_STATUS_SUCCESS;
592 }
593
594
595 static u16 copy_supp_rates(struct hostapd_data *hapd, struct sta_info *sta,
596                            struct ieee802_11_elems *elems)
597 {
598         if (!elems->supp_rates) {
599                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
600                                HOSTAPD_LEVEL_DEBUG,
601                                "No supported rates element in AssocReq");
602                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
603         }
604
605         if (elems->supp_rates_len > sizeof(sta->supported_rates)) {
606                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
607                                HOSTAPD_LEVEL_DEBUG,
608                                "Invalid supported rates element length %d",
609                                elems->supp_rates_len);
610                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
611         }
612
613         os_memset(sta->supported_rates, 0, sizeof(sta->supported_rates));
614         os_memcpy(sta->supported_rates, elems->supp_rates,
615                   elems->supp_rates_len);
616         sta->supported_rates_len = elems->supp_rates_len;
617
618         if (elems->ext_supp_rates) {
619                 if (elems->supp_rates_len + elems->ext_supp_rates_len >
620                     sizeof(sta->supported_rates)) {
621                         hostapd_logger(hapd, sta->addr,
622                                        HOSTAPD_MODULE_IEEE80211,
623                                        HOSTAPD_LEVEL_DEBUG,
624                                        "Invalid supported rates element length"
625                                        " %d+%d", elems->supp_rates_len,
626                                        elems->ext_supp_rates_len);
627                         return WLAN_STATUS_UNSPECIFIED_FAILURE;
628                 }
629
630                 os_memcpy(sta->supported_rates + elems->supp_rates_len,
631                           elems->ext_supp_rates, elems->ext_supp_rates_len);
632                 sta->supported_rates_len += elems->ext_supp_rates_len;
633         }
634
635         return WLAN_STATUS_SUCCESS;
636 }
637
638
639 static u16 check_assoc_ies(struct hostapd_data *hapd, struct sta_info *sta,
640                            const u8 *ies, size_t ies_len, int reassoc)
641 {
642         struct ieee802_11_elems elems;
643         u16 resp;
644         const u8 *wpa_ie;
645         size_t wpa_ie_len;
646
647         if (ieee802_11_parse_elems(ies, ies_len, &elems, 1) == ParseFailed) {
648                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
649                                HOSTAPD_LEVEL_INFO, "Station sent an invalid "
650                                "association request");
651                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
652         }
653
654         resp = check_ssid(hapd, sta, elems.ssid, elems.ssid_len);
655         if (resp != WLAN_STATUS_SUCCESS)
656                 return resp;
657         resp = check_wmm(hapd, sta, elems.wmm, elems.wmm_len);
658         if (resp != WLAN_STATUS_SUCCESS)
659                 return resp;
660         resp = copy_supp_rates(hapd, sta, &elems);
661         if (resp != WLAN_STATUS_SUCCESS)
662                 return resp;
663 #ifdef CONFIG_IEEE80211N
664         resp = copy_sta_ht_capab(sta, elems.ht_capabilities,
665                                  elems.ht_capabilities_len);
666         if (resp != WLAN_STATUS_SUCCESS)
667                 return resp;
668 #endif /* CONFIG_IEEE80211N */
669
670         if ((hapd->conf->wpa & WPA_PROTO_RSN) && elems.rsn_ie) {
671                 wpa_ie = elems.rsn_ie;
672                 wpa_ie_len = elems.rsn_ie_len;
673         } else if ((hapd->conf->wpa & WPA_PROTO_WPA) &&
674                    elems.wpa_ie) {
675                 wpa_ie = elems.wpa_ie;
676                 wpa_ie_len = elems.wpa_ie_len;
677         } else {
678                 wpa_ie = NULL;
679                 wpa_ie_len = 0;
680         }
681
682 #ifdef CONFIG_WPS
683         sta->flags &= ~(WLAN_STA_WPS | WLAN_STA_MAYBE_WPS);
684         if (hapd->conf->wps_state && elems.wps_ie) {
685                 wpa_printf(MSG_DEBUG, "STA included WPS IE in (Re)Association "
686                            "Request - assume WPS is used");
687                 sta->flags |= WLAN_STA_WPS;
688                 wpabuf_free(sta->wps_ie);
689                 sta->wps_ie = wpabuf_alloc_copy(elems.wps_ie + 4,
690                                                 elems.wps_ie_len - 4);
691                 wpa_ie = NULL;
692                 wpa_ie_len = 0;
693         } else if (hapd->conf->wps_state && wpa_ie == NULL) {
694                 wpa_printf(MSG_DEBUG, "STA did not include WPA/RSN IE in "
695                            "(Re)Association Request - possible WPS use");
696                 sta->flags |= WLAN_STA_MAYBE_WPS;
697         } else
698 #endif /* CONFIG_WPS */
699         if (hapd->conf->wpa && wpa_ie == NULL) {
700                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
701                                HOSTAPD_LEVEL_INFO,
702                                "No WPA/RSN IE in association request");
703                 return WLAN_STATUS_INVALID_IE;
704         }
705
706         if (hapd->conf->wpa && wpa_ie) {
707                 int res;
708                 wpa_ie -= 2;
709                 wpa_ie_len += 2;
710                 if (sta->wpa_sm == NULL)
711                         sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,
712                                                         sta->addr);
713                 if (sta->wpa_sm == NULL) {
714                         wpa_printf(MSG_WARNING, "Failed to initialize WPA "
715                                    "state machine");
716                         return WLAN_STATUS_UNSPECIFIED_FAILURE;
717                 }
718                 res = wpa_validate_wpa_ie(hapd->wpa_auth, sta->wpa_sm,
719                                           wpa_ie, wpa_ie_len,
720                                           elems.mdie, elems.mdie_len);
721                 if (res == WPA_INVALID_GROUP)
722                         resp = WLAN_STATUS_GROUP_CIPHER_NOT_VALID;
723                 else if (res == WPA_INVALID_PAIRWISE)
724                         resp = WLAN_STATUS_PAIRWISE_CIPHER_NOT_VALID;
725                 else if (res == WPA_INVALID_AKMP)
726                         resp = WLAN_STATUS_AKMP_NOT_VALID;
727                 else if (res == WPA_ALLOC_FAIL)
728                         resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
729 #ifdef CONFIG_IEEE80211W
730                 else if (res == WPA_MGMT_FRAME_PROTECTION_VIOLATION)
731                         resp = WLAN_STATUS_ROBUST_MGMT_FRAME_POLICY_VIOLATION;
732                 else if (res == WPA_INVALID_MGMT_GROUP_CIPHER)
733                         resp = WLAN_STATUS_ROBUST_MGMT_FRAME_POLICY_VIOLATION;
734 #endif /* CONFIG_IEEE80211W */
735                 else if (res == WPA_INVALID_MDIE)
736                         resp = WLAN_STATUS_INVALID_MDIE;
737                 else if (res != WPA_IE_OK)
738                         resp = WLAN_STATUS_INVALID_IE;
739                 if (resp != WLAN_STATUS_SUCCESS)
740                         return resp;
741 #ifdef CONFIG_IEEE80211W
742                 if ((sta->flags & WLAN_STA_MFP) && !sta->sa_query_timed_out &&
743                     sta->sa_query_count > 0)
744                         ap_check_sa_query_timeout(hapd, sta);
745                 if ((sta->flags & WLAN_STA_MFP) && !sta->sa_query_timed_out &&
746                     (!reassoc || sta->auth_alg != WLAN_AUTH_FT)) {
747                         /*
748                          * STA has already been associated with MFP and SA
749                          * Query timeout has not been reached. Reject the
750                          * association attempt temporarily and start SA Query,
751                          * if one is not pending.
752                          */
753
754                         if (sta->sa_query_count == 0)
755                                 ap_sta_start_sa_query(hapd, sta);
756
757                         return WLAN_STATUS_ASSOC_REJECTED_TEMPORARILY;
758                 }
759
760                 if (wpa_auth_uses_mfp(sta->wpa_sm))
761                         sta->flags |= WLAN_STA_MFP;
762                 else
763                         sta->flags &= ~WLAN_STA_MFP;
764 #endif /* CONFIG_IEEE80211W */
765
766 #ifdef CONFIG_IEEE80211R
767                 if (sta->auth_alg == WLAN_AUTH_FT) {
768                         if (!reassoc) {
769                                 wpa_printf(MSG_DEBUG, "FT: " MACSTR " tried "
770                                            "to use association (not "
771                                            "re-association) with FT auth_alg",
772                                            MAC2STR(sta->addr));
773                                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
774                         }
775
776                         resp = wpa_ft_validate_reassoc(sta->wpa_sm, ies,
777                                                        ies_len);
778                         if (resp != WLAN_STATUS_SUCCESS)
779                                 return resp;
780                 }
781 #endif /* CONFIG_IEEE80211R */
782
783 #ifdef CONFIG_IEEE80211N
784                 if ((sta->flags & WLAN_STA_HT) &&
785                     wpa_auth_get_pairwise(sta->wpa_sm) == WPA_CIPHER_TKIP) {
786                         hostapd_logger(hapd, sta->addr,
787                                        HOSTAPD_MODULE_IEEE80211,
788                                        HOSTAPD_LEVEL_INFO,
789                                        "Station tried to use TKIP with HT "
790                                        "association");
791                         return WLAN_STATUS_CIPHER_REJECTED_PER_POLICY;
792                 }
793 #endif /* CONFIG_IEEE80211N */
794         } else
795                 wpa_auth_sta_no_wpa(sta->wpa_sm);
796
797         return WLAN_STATUS_SUCCESS;
798 }
799
800
801 static void send_deauth(struct hostapd_data *hapd, const u8 *addr,
802                         u16 reason_code)
803 {
804         int send_len;
805         struct ieee80211_mgmt reply;
806
807         os_memset(&reply, 0, sizeof(reply));
808         reply.frame_control =
809                 IEEE80211_FC(WLAN_FC_TYPE_MGMT, WLAN_FC_STYPE_DEAUTH);
810         os_memcpy(reply.da, addr, ETH_ALEN);
811         os_memcpy(reply.sa, hapd->own_addr, ETH_ALEN);
812         os_memcpy(reply.bssid, hapd->own_addr, ETH_ALEN);
813
814         send_len = IEEE80211_HDRLEN + sizeof(reply.u.deauth);
815         reply.u.deauth.reason_code = host_to_le16(reason_code);
816
817         if (hapd->drv.send_mgmt_frame(hapd, &reply, send_len) < 0)
818                 wpa_printf(MSG_INFO, "Failed to send deauth: %s",
819                            strerror(errno));
820 }
821
822
823 static void send_assoc_resp(struct hostapd_data *hapd, struct sta_info *sta,
824                             u16 status_code, int reassoc, const u8 *ies,
825                             size_t ies_len)
826 {
827         int send_len;
828         u8 buf[sizeof(struct ieee80211_mgmt) + 1024];
829         struct ieee80211_mgmt *reply;
830         u8 *p;
831
832         os_memset(buf, 0, sizeof(buf));
833         reply = (struct ieee80211_mgmt *) buf;
834         reply->frame_control =
835                 IEEE80211_FC(WLAN_FC_TYPE_MGMT,
836                              (reassoc ? WLAN_FC_STYPE_REASSOC_RESP :
837                               WLAN_FC_STYPE_ASSOC_RESP));
838         os_memcpy(reply->da, sta->addr, ETH_ALEN);
839         os_memcpy(reply->sa, hapd->own_addr, ETH_ALEN);
840         os_memcpy(reply->bssid, hapd->own_addr, ETH_ALEN);
841
842         send_len = IEEE80211_HDRLEN;
843         send_len += sizeof(reply->u.assoc_resp);
844         reply->u.assoc_resp.capab_info =
845                 host_to_le16(hostapd_own_capab_info(hapd, sta, 0));
846         reply->u.assoc_resp.status_code = host_to_le16(status_code);
847         reply->u.assoc_resp.aid = host_to_le16((sta ? sta->aid : 0)
848                                                | BIT(14) | BIT(15));
849         /* Supported rates */
850         p = hostapd_eid_supp_rates(hapd, reply->u.assoc_resp.variable);
851         /* Extended supported rates */
852         p = hostapd_eid_ext_supp_rates(hapd, p);
853         if (sta->flags & WLAN_STA_WMM)
854                 p = hostapd_eid_wmm(hapd, p);
855
856 #ifdef CONFIG_IEEE80211N
857         p = hostapd_eid_ht_capabilities(hapd, p);
858         p = hostapd_eid_ht_operation(hapd, p);
859 #endif /* CONFIG_IEEE80211N */
860
861 #ifdef CONFIG_IEEE80211R
862         if (status_code == WLAN_STATUS_SUCCESS) {
863                 /* IEEE 802.11r: Mobility Domain Information, Fast BSS
864                  * Transition Information, RSN, [RIC Response] */
865                 p = wpa_sm_write_assoc_resp_ies(sta->wpa_sm, p,
866                                                 buf + sizeof(buf) - p,
867                                                 sta->auth_alg, ies, ies_len);
868         }
869 #endif /* CONFIG_IEEE80211R */
870
871 #ifdef CONFIG_IEEE80211W
872         if (status_code == WLAN_STATUS_ASSOC_REJECTED_TEMPORARILY)
873                 p = hostapd_eid_assoc_comeback_time(hapd, sta, p);
874 #endif /* CONFIG_IEEE80211W */
875
876         send_len += p - reply->u.assoc_resp.variable;
877
878         if (hapd->drv.send_mgmt_frame(hapd, reply, send_len) < 0)
879                 wpa_printf(MSG_INFO, "Failed to send assoc resp: %s",
880                            strerror(errno));
881 }
882
883
884 static void handle_assoc(struct hostapd_data *hapd,
885                          const struct ieee80211_mgmt *mgmt, size_t len,
886                          int reassoc)
887 {
888         u16 capab_info, listen_interval;
889         u16 resp = WLAN_STATUS_SUCCESS;
890         const u8 *pos;
891         int left, i;
892         struct sta_info *sta;
893
894         if (len < IEEE80211_HDRLEN + (reassoc ? sizeof(mgmt->u.reassoc_req) :
895                                       sizeof(mgmt->u.assoc_req))) {
896                 printf("handle_assoc(reassoc=%d) - too short payload (len=%lu)"
897                        "\n", reassoc, (unsigned long) len);
898                 return;
899         }
900
901         if (reassoc) {
902                 capab_info = le_to_host16(mgmt->u.reassoc_req.capab_info);
903                 listen_interval = le_to_host16(
904                         mgmt->u.reassoc_req.listen_interval);
905                 wpa_printf(MSG_DEBUG, "reassociation request: STA=" MACSTR
906                            " capab_info=0x%02x listen_interval=%d current_ap="
907                            MACSTR,
908                            MAC2STR(mgmt->sa), capab_info, listen_interval,
909                            MAC2STR(mgmt->u.reassoc_req.current_ap));
910                 left = len - (IEEE80211_HDRLEN + sizeof(mgmt->u.reassoc_req));
911                 pos = mgmt->u.reassoc_req.variable;
912         } else {
913                 capab_info = le_to_host16(mgmt->u.assoc_req.capab_info);
914                 listen_interval = le_to_host16(
915                         mgmt->u.assoc_req.listen_interval);
916                 wpa_printf(MSG_DEBUG, "association request: STA=" MACSTR
917                            " capab_info=0x%02x listen_interval=%d",
918                            MAC2STR(mgmt->sa), capab_info, listen_interval);
919                 left = len - (IEEE80211_HDRLEN + sizeof(mgmt->u.assoc_req));
920                 pos = mgmt->u.assoc_req.variable;
921         }
922
923         sta = ap_get_sta(hapd, mgmt->sa);
924 #ifdef CONFIG_IEEE80211R
925         if (sta && sta->auth_alg == WLAN_AUTH_FT &&
926             (sta->flags & WLAN_STA_AUTH) == 0) {
927                 wpa_printf(MSG_DEBUG, "FT: Allow STA " MACSTR " to associate "
928                            "prior to authentication since it is using "
929                            "over-the-DS FT", MAC2STR(mgmt->sa));
930         } else
931 #endif /* CONFIG_IEEE80211R */
932         if (sta == NULL || (sta->flags & WLAN_STA_AUTH) == 0) {
933                 hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
934                                HOSTAPD_LEVEL_INFO, "Station tried to "
935                                "associate before authentication "
936                                "(aid=%d flags=0x%x)",
937                                sta ? sta->aid : -1,
938                                sta ? sta->flags : 0);
939                 send_deauth(hapd, mgmt->sa,
940                             WLAN_REASON_CLASS2_FRAME_FROM_NONAUTH_STA);
941                 return;
942         }
943
944         if (hapd->tkip_countermeasures) {
945                 resp = WLAN_REASON_MICHAEL_MIC_FAILURE;
946                 goto fail;
947         }
948
949         if (listen_interval > hapd->conf->max_listen_interval) {
950                 hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
951                                HOSTAPD_LEVEL_DEBUG,
952                                "Too large Listen Interval (%d)",
953                                listen_interval);
954                 resp = WLAN_STATUS_ASSOC_DENIED_LISTEN_INT_TOO_LARGE;
955                 goto fail;
956         }
957
958         /* followed by SSID and Supported rates; and HT capabilities if 802.11n
959          * is used */
960         resp = check_assoc_ies(hapd, sta, pos, left, reassoc);
961         if (resp != WLAN_STATUS_SUCCESS)
962                 goto fail;
963
964         if (hostapd_get_aid(hapd, sta) < 0) {
965                 hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
966                                HOSTAPD_LEVEL_INFO, "No room for more AIDs");
967                 resp = WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA;
968                 goto fail;
969         }
970
971         sta->capability = capab_info;
972         sta->listen_interval = listen_interval;
973
974         if (hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G)
975                 sta->flags |= WLAN_STA_NONERP;
976         for (i = 0; i < sta->supported_rates_len; i++) {
977                 if ((sta->supported_rates[i] & 0x7f) > 22) {
978                         sta->flags &= ~WLAN_STA_NONERP;
979                         break;
980                 }
981         }
982         if (sta->flags & WLAN_STA_NONERP && !sta->nonerp_set) {
983                 sta->nonerp_set = 1;
984                 hapd->iface->num_sta_non_erp++;
985                 if (hapd->iface->num_sta_non_erp == 1)
986                         ieee802_11_set_beacons(hapd->iface);
987         }
988
989         if (!(sta->capability & WLAN_CAPABILITY_SHORT_SLOT_TIME) &&
990             !sta->no_short_slot_time_set) {
991                 sta->no_short_slot_time_set = 1;
992                 hapd->iface->num_sta_no_short_slot_time++;
993                 if (hapd->iface->current_mode->mode ==
994                     HOSTAPD_MODE_IEEE80211G &&
995                     hapd->iface->num_sta_no_short_slot_time == 1)
996                         ieee802_11_set_beacons(hapd->iface);
997         }
998
999         if (sta->capability & WLAN_CAPABILITY_SHORT_PREAMBLE)
1000                 sta->flags |= WLAN_STA_SHORT_PREAMBLE;
1001         else
1002                 sta->flags &= ~WLAN_STA_SHORT_PREAMBLE;
1003
1004         if (!(sta->capability & WLAN_CAPABILITY_SHORT_PREAMBLE) &&
1005             !sta->no_short_preamble_set) {
1006                 sta->no_short_preamble_set = 1;
1007                 hapd->iface->num_sta_no_short_preamble++;
1008                 if (hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G
1009                     && hapd->iface->num_sta_no_short_preamble == 1)
1010                         ieee802_11_set_beacons(hapd->iface);
1011         }
1012
1013 #ifdef CONFIG_IEEE80211N
1014         update_ht_state(hapd, sta);
1015 #endif /* CONFIG_IEEE80211N */
1016
1017         hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
1018                        HOSTAPD_LEVEL_DEBUG,
1019                        "association OK (aid %d)", sta->aid);
1020         /* Station will be marked associated, after it acknowledges AssocResp
1021          */
1022
1023 #ifdef CONFIG_IEEE80211W
1024         if ((sta->flags & WLAN_STA_MFP) && sta->sa_query_timed_out) {
1025                 wpa_printf(MSG_DEBUG, "Allowing %sassociation after timed out "
1026                            "SA Query procedure", reassoc ? "re" : "");
1027                 /* TODO: Send a protected Disassociate frame to the STA using
1028                  * the old key and Reason Code "Previous Authentication no
1029                  * longer valid". Make sure this is only sent protected since
1030                  * unprotected frame would be received by the STA that is now
1031                  * trying to associate.
1032                  */
1033         }
1034 #endif /* CONFIG_IEEE80211W */
1035
1036         if (reassoc) {
1037                 os_memcpy(sta->previous_ap, mgmt->u.reassoc_req.current_ap,
1038                           ETH_ALEN);
1039         }
1040
1041         if (sta->last_assoc_req)
1042                 os_free(sta->last_assoc_req);
1043         sta->last_assoc_req = os_malloc(len);
1044         if (sta->last_assoc_req)
1045                 os_memcpy(sta->last_assoc_req, mgmt, len);
1046
1047         /* Make sure that the previously registered inactivity timer will not
1048          * remove the STA immediately. */
1049         sta->timeout_next = STA_NULLFUNC;
1050
1051  fail:
1052         send_assoc_resp(hapd, sta, resp, reassoc, pos, left);
1053 }
1054
1055
1056 static void handle_disassoc(struct hostapd_data *hapd,
1057                             const struct ieee80211_mgmt *mgmt, size_t len)
1058 {
1059         struct sta_info *sta;
1060
1061         if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.disassoc)) {
1062                 printf("handle_disassoc - too short payload (len=%lu)\n",
1063                        (unsigned long) len);
1064                 return;
1065         }
1066
1067         wpa_printf(MSG_DEBUG, "disassocation: STA=" MACSTR " reason_code=%d",
1068                    MAC2STR(mgmt->sa),
1069                    le_to_host16(mgmt->u.disassoc.reason_code));
1070
1071         sta = ap_get_sta(hapd, mgmt->sa);
1072         if (sta == NULL) {
1073                 printf("Station " MACSTR " trying to disassociate, but it "
1074                        "is not associated.\n", MAC2STR(mgmt->sa));
1075                 return;
1076         }
1077
1078         sta->flags &= ~WLAN_STA_ASSOC;
1079         wpa_msg(hapd->msg_ctx, MSG_INFO, AP_STA_DISCONNECTED MACSTR,
1080                 MAC2STR(sta->addr));
1081         wpa_auth_sm_event(sta->wpa_sm, WPA_DISASSOC);
1082         hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
1083                        HOSTAPD_LEVEL_INFO, "disassociated");
1084         sta->acct_terminate_cause = RADIUS_ACCT_TERMINATE_CAUSE_USER_REQUEST;
1085         ieee802_1x_notify_port_enabled(sta->eapol_sm, 0);
1086         /* Stop Accounting and IEEE 802.1X sessions, but leave the STA
1087          * authenticated. */
1088         accounting_sta_stop(hapd, sta);
1089         ieee802_1x_free_station(sta);
1090         hapd->drv.sta_remove(hapd, sta->addr);
1091
1092         if (sta->timeout_next == STA_NULLFUNC ||
1093             sta->timeout_next == STA_DISASSOC) {
1094                 sta->timeout_next = STA_DEAUTH;
1095                 eloop_cancel_timeout(ap_handle_timer, hapd, sta);
1096                 eloop_register_timeout(AP_DEAUTH_DELAY, 0, ap_handle_timer,
1097                                        hapd, sta);
1098         }
1099
1100         mlme_disassociate_indication(
1101                 hapd, sta, le_to_host16(mgmt->u.disassoc.reason_code));
1102 }
1103
1104
1105 static void handle_deauth(struct hostapd_data *hapd,
1106                           const struct ieee80211_mgmt *mgmt, size_t len)
1107 {
1108         struct sta_info *sta;
1109
1110         if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.deauth)) {
1111                 printf("handle_deauth - too short payload (len=%lu)\n",
1112                        (unsigned long) len);
1113                 return;
1114         }
1115
1116         wpa_printf(MSG_DEBUG, "deauthentication: STA=" MACSTR
1117                    " reason_code=%d",
1118                    MAC2STR(mgmt->sa),
1119                    le_to_host16(mgmt->u.deauth.reason_code));
1120
1121         sta = ap_get_sta(hapd, mgmt->sa);
1122         if (sta == NULL) {
1123                 printf("Station " MACSTR " trying to deauthenticate, but it "
1124                        "is not authenticated.\n", MAC2STR(mgmt->sa));
1125                 return;
1126         }
1127
1128         sta->flags &= ~(WLAN_STA_AUTH | WLAN_STA_ASSOC);
1129         wpa_msg(hapd->msg_ctx, MSG_INFO, AP_STA_DISCONNECTED MACSTR,
1130                 MAC2STR(sta->addr));
1131         wpa_auth_sm_event(sta->wpa_sm, WPA_DEAUTH);
1132         hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
1133                        HOSTAPD_LEVEL_DEBUG, "deauthenticated");
1134         mlme_deauthenticate_indication(
1135                 hapd, sta, le_to_host16(mgmt->u.deauth.reason_code));
1136         sta->acct_terminate_cause = RADIUS_ACCT_TERMINATE_CAUSE_USER_REQUEST;
1137         ieee802_1x_notify_port_enabled(sta->eapol_sm, 0);
1138         ap_free_sta(hapd, sta);
1139 }
1140
1141
1142 static void handle_beacon(struct hostapd_data *hapd,
1143                           const struct ieee80211_mgmt *mgmt, size_t len,
1144                           struct hostapd_frame_info *fi)
1145 {
1146         struct ieee802_11_elems elems;
1147
1148         if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.beacon)) {
1149                 printf("handle_beacon - too short payload (len=%lu)\n",
1150                        (unsigned long) len);
1151                 return;
1152         }
1153
1154         (void) ieee802_11_parse_elems(mgmt->u.beacon.variable,
1155                                       len - (IEEE80211_HDRLEN +
1156                                              sizeof(mgmt->u.beacon)), &elems,
1157                                       0);
1158
1159         ap_list_process_beacon(hapd->iface, mgmt, &elems, fi);
1160 }
1161
1162
1163 #ifdef CONFIG_IEEE80211W
1164
1165 /* MLME-SAQuery.request */
1166 void ieee802_11_send_sa_query_req(struct hostapd_data *hapd,
1167                                   const u8 *addr, const u8 *trans_id)
1168 {
1169         struct ieee80211_mgmt mgmt;
1170         u8 *end;
1171
1172         wpa_printf(MSG_DEBUG, "IEEE 802.11: Sending SA Query Request to "
1173                    MACSTR, MAC2STR(addr));
1174         wpa_hexdump(MSG_DEBUG, "IEEE 802.11: SA Query Transaction ID",
1175                     trans_id, WLAN_SA_QUERY_TR_ID_LEN);
1176
1177         os_memset(&mgmt, 0, sizeof(mgmt));
1178         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
1179                                           WLAN_FC_STYPE_ACTION);
1180         os_memcpy(mgmt.da, addr, ETH_ALEN);
1181         os_memcpy(mgmt.sa, hapd->own_addr, ETH_ALEN);
1182         os_memcpy(mgmt.bssid, hapd->own_addr, ETH_ALEN);
1183         mgmt.u.action.category = WLAN_ACTION_SA_QUERY;
1184         mgmt.u.action.u.sa_query_req.action = WLAN_SA_QUERY_REQUEST;
1185         os_memcpy(mgmt.u.action.u.sa_query_req.trans_id, trans_id,
1186                   WLAN_SA_QUERY_TR_ID_LEN);
1187         end = mgmt.u.action.u.sa_query_req.trans_id + WLAN_SA_QUERY_TR_ID_LEN;
1188         if (hapd->drv.send_mgmt_frame(hapd, &mgmt, end - (u8 *) &mgmt) < 0)
1189                 perror("ieee802_11_send_sa_query_req: send");
1190 }
1191
1192
1193 static void hostapd_sa_query_request(struct hostapd_data *hapd,
1194                                      const struct ieee80211_mgmt *mgmt)
1195 {
1196         struct sta_info *sta;
1197         struct ieee80211_mgmt resp;
1198         u8 *end;
1199
1200         wpa_printf(MSG_DEBUG, "IEEE 802.11: Received SA Query Request from "
1201                    MACSTR, MAC2STR(mgmt->sa));
1202         wpa_hexdump(MSG_DEBUG, "IEEE 802.11: SA Query Transaction ID",
1203                     mgmt->u.action.u.sa_query_resp.trans_id,
1204                     WLAN_SA_QUERY_TR_ID_LEN);
1205
1206         sta = ap_get_sta(hapd, mgmt->sa);
1207         if (sta == NULL || !(sta->flags & WLAN_STA_ASSOC)) {
1208                 wpa_printf(MSG_DEBUG, "IEEE 802.11: Ignore SA Query Request "
1209                            "from unassociated STA " MACSTR, MAC2STR(mgmt->sa));
1210                 return;
1211         }
1212
1213         wpa_printf(MSG_DEBUG, "IEEE 802.11: Sending SA Query Response to "
1214                    MACSTR, MAC2STR(mgmt->sa));
1215
1216         os_memset(&resp, 0, sizeof(resp));
1217         resp.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
1218                                           WLAN_FC_STYPE_ACTION);
1219         os_memcpy(resp.da, mgmt->sa, ETH_ALEN);
1220         os_memcpy(resp.sa, hapd->own_addr, ETH_ALEN);
1221         os_memcpy(resp.bssid, hapd->own_addr, ETH_ALEN);
1222         resp.u.action.category = WLAN_ACTION_SA_QUERY;
1223         resp.u.action.u.sa_query_req.action = WLAN_SA_QUERY_RESPONSE;
1224         os_memcpy(resp.u.action.u.sa_query_req.trans_id,
1225                   mgmt->u.action.u.sa_query_req.trans_id,
1226                   WLAN_SA_QUERY_TR_ID_LEN);
1227         end = resp.u.action.u.sa_query_req.trans_id + WLAN_SA_QUERY_TR_ID_LEN;
1228         if (hapd->drv.send_mgmt_frame(hapd, &resp, end - (u8 *) &resp) < 0)
1229                 perror("hostapd_sa_query_request: send");
1230 }
1231
1232
1233 static void hostapd_sa_query_action(struct hostapd_data *hapd,
1234                                     const struct ieee80211_mgmt *mgmt,
1235                                     size_t len)
1236 {
1237         struct sta_info *sta;
1238         const u8 *end;
1239         int i;
1240
1241         end = mgmt->u.action.u.sa_query_resp.trans_id +
1242                 WLAN_SA_QUERY_TR_ID_LEN;
1243         if (((u8 *) mgmt) + len < end) {
1244                 wpa_printf(MSG_DEBUG, "IEEE 802.11: Too short SA Query Action "
1245                            "frame (len=%lu)", (unsigned long) len);
1246                 return;
1247         }
1248
1249         if (mgmt->u.action.u.sa_query_resp.action == WLAN_SA_QUERY_REQUEST) {
1250                 hostapd_sa_query_request(hapd, mgmt);
1251                 return;
1252         }
1253
1254         if (mgmt->u.action.u.sa_query_resp.action != WLAN_SA_QUERY_RESPONSE) {
1255                 wpa_printf(MSG_DEBUG, "IEEE 802.11: Unexpected SA Query "
1256                            "Action %d", mgmt->u.action.u.sa_query_resp.action);
1257                 return;
1258         }
1259
1260         wpa_printf(MSG_DEBUG, "IEEE 802.11: Received SA Query Response from "
1261                    MACSTR, MAC2STR(mgmt->sa));
1262         wpa_hexdump(MSG_DEBUG, "IEEE 802.11: SA Query Transaction ID",
1263                     mgmt->u.action.u.sa_query_resp.trans_id,
1264                     WLAN_SA_QUERY_TR_ID_LEN);
1265
1266         /* MLME-SAQuery.confirm */
1267
1268         sta = ap_get_sta(hapd, mgmt->sa);
1269         if (sta == NULL || sta->sa_query_trans_id == NULL) {
1270                 wpa_printf(MSG_DEBUG, "IEEE 802.11: No matching STA with "
1271                            "pending SA Query request found");
1272                 return;
1273         }
1274
1275         for (i = 0; i < sta->sa_query_count; i++) {
1276                 if (os_memcmp(sta->sa_query_trans_id +
1277                               i * WLAN_SA_QUERY_TR_ID_LEN,
1278                               mgmt->u.action.u.sa_query_resp.trans_id,
1279                               WLAN_SA_QUERY_TR_ID_LEN) == 0)
1280                         break;
1281         }
1282
1283         if (i >= sta->sa_query_count) {
1284                 wpa_printf(MSG_DEBUG, "IEEE 802.11: No matching SA Query "
1285                            "transaction identifier found");
1286                 return;
1287         }
1288
1289         hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
1290                        HOSTAPD_LEVEL_DEBUG,
1291                        "Reply to pending SA Query received");
1292         ap_sta_stop_sa_query(hapd, sta);
1293 }
1294
1295
1296 static int robust_action_frame(u8 category)
1297 {
1298         return category != WLAN_ACTION_PUBLIC &&
1299                 category != WLAN_ACTION_HT;
1300 }
1301 #endif /* CONFIG_IEEE80211W */
1302
1303
1304 static void handle_action(struct hostapd_data *hapd,
1305                           const struct ieee80211_mgmt *mgmt, size_t len)
1306 {
1307         struct sta_info *sta;
1308
1309         if (len < IEEE80211_HDRLEN + 1) {
1310                 hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
1311                                HOSTAPD_LEVEL_DEBUG,
1312                                "handle_action - too short payload (len=%lu)",
1313                                (unsigned long) len);
1314                 return;
1315         }
1316
1317         sta = ap_get_sta(hapd, mgmt->sa);
1318 #ifdef CONFIG_IEEE80211W
1319         if (sta && (sta->flags & WLAN_STA_MFP) &&
1320             !(mgmt->frame_control & host_to_le16(WLAN_FC_ISWEP) &&
1321               robust_action_frame(mgmt->u.action.category))) {
1322                 hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
1323                                HOSTAPD_LEVEL_DEBUG,
1324                                "Dropped unprotected Robust Action frame from "
1325                                "an MFP STA");
1326                 return;
1327         }
1328 #endif /* CONFIG_IEEE80211W */
1329
1330         switch (mgmt->u.action.category) {
1331 #ifdef CONFIG_IEEE80211R
1332         case WLAN_ACTION_FT:
1333         {
1334                 if (sta == NULL || !(sta->flags & WLAN_STA_ASSOC)) {
1335                         wpa_printf(MSG_DEBUG, "IEEE 802.11: Ignored FT Action "
1336                                    "frame from unassociated STA " MACSTR,
1337                                    MAC2STR(mgmt->sa));
1338                         return;
1339                 }
1340
1341                 if (wpa_ft_action_rx(sta->wpa_sm, (u8 *) &mgmt->u.action,
1342                                      len - IEEE80211_HDRLEN))
1343                         break;
1344
1345                 return;
1346         }
1347 #endif /* CONFIG_IEEE80211R */
1348         case WLAN_ACTION_WMM:
1349                 hostapd_wmm_action(hapd, mgmt, len);
1350                 return;
1351 #ifdef CONFIG_IEEE80211W
1352         case WLAN_ACTION_SA_QUERY:
1353                 hostapd_sa_query_action(hapd, mgmt, len);
1354                 return;
1355 #endif /* CONFIG_IEEE80211W */
1356         case WLAN_ACTION_PUBLIC:
1357                 if (hapd->public_action_cb) {
1358                         hapd->public_action_cb(hapd->public_action_cb_ctx,
1359                                                (u8 *) mgmt, len,
1360                                                hapd->iface->freq);
1361                         return;
1362                 }
1363                 break;
1364         }
1365
1366         hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
1367                        HOSTAPD_LEVEL_DEBUG,
1368                        "handle_action - unknown action category %d or invalid "
1369                        "frame",
1370                        mgmt->u.action.category);
1371         if (!(mgmt->da[0] & 0x01) && !(mgmt->u.action.category & 0x80) &&
1372             !(mgmt->sa[0] & 0x01)) {
1373                 struct ieee80211_mgmt *resp;
1374
1375                 /*
1376                  * IEEE 802.11-REVma/D9.0 - 7.3.1.11
1377                  * Return the Action frame to the source without change
1378                  * except that MSB of the Category set to 1.
1379                  */
1380                 wpa_printf(MSG_DEBUG, "IEEE 802.11: Return unknown Action "
1381                            "frame back to sender");
1382                 resp = os_malloc(len);
1383                 if (resp == NULL)
1384                         return;
1385                 os_memcpy(resp, mgmt, len);
1386                 os_memcpy(resp->da, resp->sa, ETH_ALEN);
1387                 os_memcpy(resp->sa, hapd->own_addr, ETH_ALEN);
1388                 os_memcpy(resp->bssid, hapd->own_addr, ETH_ALEN);
1389                 resp->u.action.category |= 0x80;
1390
1391                 hapd->drv.send_mgmt_frame(hapd, resp, len);
1392                 os_free(resp);
1393         }
1394 }
1395
1396
1397 /**
1398  * ieee802_11_mgmt - process incoming IEEE 802.11 management frames
1399  * @hapd: hostapd BSS data structure (the BSS to which the management frame was
1400  * sent to)
1401  * @buf: management frame data (starting from IEEE 802.11 header)
1402  * @len: length of frame data in octets
1403  * @fi: meta data about received frame (signal level, etc.)
1404  *
1405  * Process all incoming IEEE 802.11 management frames. This will be called for
1406  * each frame received from the kernel driver through wlan#ap interface. In
1407  * addition, it can be called to re-inserted pending frames (e.g., when using
1408  * external RADIUS server as an MAC ACL).
1409  */
1410 void ieee802_11_mgmt(struct hostapd_data *hapd, const u8 *buf, size_t len,
1411                      struct hostapd_frame_info *fi)
1412 {
1413         struct ieee80211_mgmt *mgmt;
1414         int broadcast;
1415         u16 fc, stype;
1416
1417         mgmt = (struct ieee80211_mgmt *) buf;
1418         fc = le_to_host16(mgmt->frame_control);
1419         stype = WLAN_FC_GET_STYPE(fc);
1420
1421         if (stype == WLAN_FC_STYPE_BEACON) {
1422                 handle_beacon(hapd, mgmt, len, fi);
1423                 return;
1424         }
1425
1426         broadcast = mgmt->bssid[0] == 0xff && mgmt->bssid[1] == 0xff &&
1427                 mgmt->bssid[2] == 0xff && mgmt->bssid[3] == 0xff &&
1428                 mgmt->bssid[4] == 0xff && mgmt->bssid[5] == 0xff;
1429
1430         if (!broadcast &&
1431             os_memcmp(mgmt->bssid, hapd->own_addr, ETH_ALEN) != 0) {
1432                 printf("MGMT: BSSID=" MACSTR " not our address\n",
1433                        MAC2STR(mgmt->bssid));
1434                 return;
1435         }
1436
1437
1438         if (stype == WLAN_FC_STYPE_PROBE_REQ) {
1439                 handle_probe_req(hapd, mgmt, len);
1440                 return;
1441         }
1442
1443         if (os_memcmp(mgmt->da, hapd->own_addr, ETH_ALEN) != 0) {
1444                 hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
1445                                HOSTAPD_LEVEL_DEBUG,
1446                                "MGMT: DA=" MACSTR " not our address",
1447                                MAC2STR(mgmt->da));
1448                 return;
1449         }
1450
1451         switch (stype) {
1452         case WLAN_FC_STYPE_AUTH:
1453                 wpa_printf(MSG_DEBUG, "mgmt::auth");
1454                 handle_auth(hapd, mgmt, len);
1455                 break;
1456         case WLAN_FC_STYPE_ASSOC_REQ:
1457                 wpa_printf(MSG_DEBUG, "mgmt::assoc_req");
1458                 handle_assoc(hapd, mgmt, len, 0);
1459                 break;
1460         case WLAN_FC_STYPE_REASSOC_REQ:
1461                 wpa_printf(MSG_DEBUG, "mgmt::reassoc_req");
1462                 handle_assoc(hapd, mgmt, len, 1);
1463                 break;
1464         case WLAN_FC_STYPE_DISASSOC:
1465                 wpa_printf(MSG_DEBUG, "mgmt::disassoc");
1466                 handle_disassoc(hapd, mgmt, len);
1467                 break;
1468         case WLAN_FC_STYPE_DEAUTH:
1469                 wpa_printf(MSG_DEBUG, "mgmt::deauth");
1470                 handle_deauth(hapd, mgmt, len);
1471                 break;
1472         case WLAN_FC_STYPE_ACTION:
1473                 wpa_printf(MSG_DEBUG, "mgmt::action");
1474                 handle_action(hapd, mgmt, len);
1475                 break;
1476         default:
1477                 hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
1478                                HOSTAPD_LEVEL_DEBUG,
1479                                "unknown mgmt frame subtype %d", stype);
1480                 break;
1481         }
1482 }
1483
1484
1485 static void handle_auth_cb(struct hostapd_data *hapd,
1486                            const struct ieee80211_mgmt *mgmt,
1487                            size_t len, int ok)
1488 {
1489         u16 auth_alg, auth_transaction, status_code;
1490         struct sta_info *sta;
1491
1492         if (!ok) {
1493                 hostapd_logger(hapd, mgmt->da, HOSTAPD_MODULE_IEEE80211,
1494                                HOSTAPD_LEVEL_NOTICE,
1495                                "did not acknowledge authentication response");
1496                 return;
1497         }
1498
1499         if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.auth)) {
1500                 printf("handle_auth_cb - too short payload (len=%lu)\n",
1501                        (unsigned long) len);
1502                 return;
1503         }
1504
1505         auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
1506         auth_transaction = le_to_host16(mgmt->u.auth.auth_transaction);
1507         status_code = le_to_host16(mgmt->u.auth.status_code);
1508
1509         sta = ap_get_sta(hapd, mgmt->da);
1510         if (!sta) {
1511                 printf("handle_auth_cb: STA " MACSTR " not found\n",
1512                        MAC2STR(mgmt->da));
1513                 return;
1514         }
1515
1516         if (status_code == WLAN_STATUS_SUCCESS &&
1517             ((auth_alg == WLAN_AUTH_OPEN && auth_transaction == 2) ||
1518              (auth_alg == WLAN_AUTH_SHARED_KEY && auth_transaction == 4))) {
1519                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
1520                                HOSTAPD_LEVEL_INFO, "authenticated");
1521                 sta->flags |= WLAN_STA_AUTH;
1522         }
1523 }
1524
1525
1526 static void handle_assoc_cb(struct hostapd_data *hapd,
1527                             const struct ieee80211_mgmt *mgmt,
1528                             size_t len, int reassoc, int ok)
1529 {
1530         u16 status;
1531         struct sta_info *sta;
1532         int new_assoc = 1;
1533         struct ieee80211_ht_capabilities ht_cap;
1534
1535         if (!ok) {
1536                 hostapd_logger(hapd, mgmt->da, HOSTAPD_MODULE_IEEE80211,
1537                                HOSTAPD_LEVEL_DEBUG,
1538                                "did not acknowledge association response");
1539                 return;
1540         }
1541
1542         if (len < IEEE80211_HDRLEN + (reassoc ? sizeof(mgmt->u.reassoc_resp) :
1543                                       sizeof(mgmt->u.assoc_resp))) {
1544                 printf("handle_assoc_cb(reassoc=%d) - too short payload "
1545                        "(len=%lu)\n", reassoc, (unsigned long) len);
1546                 return;
1547         }
1548
1549         if (reassoc)
1550                 status = le_to_host16(mgmt->u.reassoc_resp.status_code);
1551         else
1552                 status = le_to_host16(mgmt->u.assoc_resp.status_code);
1553
1554         sta = ap_get_sta(hapd, mgmt->da);
1555         if (!sta) {
1556                 printf("handle_assoc_cb: STA " MACSTR " not found\n",
1557                        MAC2STR(mgmt->da));
1558                 return;
1559         }
1560
1561         if (status != WLAN_STATUS_SUCCESS)
1562                 goto fail;
1563
1564         /* Stop previous accounting session, if one is started, and allocate
1565          * new session id for the new session. */
1566         accounting_sta_stop(hapd, sta);
1567
1568         hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
1569                        HOSTAPD_LEVEL_INFO,
1570                        "associated (aid %d)",
1571                        sta->aid);
1572
1573         if (sta->flags & WLAN_STA_ASSOC)
1574                 new_assoc = 0;
1575         sta->flags |= WLAN_STA_ASSOC;
1576         if ((!hapd->conf->ieee802_1x && !hapd->conf->wpa) ||
1577             sta->auth_alg == WLAN_AUTH_FT) {
1578                 /*
1579                  * Open, static WEP, or FT protocol; no separate authorization
1580                  * step.
1581                  */
1582                 sta->flags |= WLAN_STA_AUTHORIZED;
1583                 wpa_msg(hapd->msg_ctx, MSG_INFO,
1584                         AP_STA_CONNECTED MACSTR, MAC2STR(sta->addr));
1585         }
1586
1587         if (reassoc)
1588                 mlme_reassociate_indication(hapd, sta);
1589         else
1590                 mlme_associate_indication(hapd, sta);
1591
1592 #ifdef CONFIG_IEEE80211W
1593         sta->sa_query_timed_out = 0;
1594 #endif /* CONFIG_IEEE80211W */
1595
1596         /*
1597          * Remove the STA entry in order to make sure the STA PS state gets
1598          * cleared and configuration gets updated in case of reassociation back
1599          * to the same AP.
1600          */
1601         hapd->drv.sta_remove(hapd, sta->addr);
1602
1603 #ifdef CONFIG_IEEE80211N
1604         if (sta->flags & WLAN_STA_HT)
1605                 hostapd_get_ht_capab(hapd, sta->ht_capabilities, &ht_cap);
1606 #endif /* CONFIG_IEEE80211N */
1607
1608         if (hapd->drv.sta_add(hapd, sta->addr, sta->aid, sta->capability,
1609                               sta->supported_rates, sta->supported_rates_len,
1610                               sta->listen_interval,
1611                               sta->flags & WLAN_STA_HT ? &ht_cap : NULL)) {
1612                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
1613                                HOSTAPD_LEVEL_NOTICE,
1614                                "Could not add STA to kernel driver");
1615         }
1616
1617         if (sta->eapol_sm == NULL) {
1618                 /*
1619                  * This STA does not use RADIUS server for EAP authentication,
1620                  * so bind it to the selected VLAN interface now, since the
1621                  * interface selection is not going to change anymore.
1622                  */
1623                 ap_sta_bind_vlan(hapd, sta, 0);
1624         } else if (sta->vlan_id) {
1625                 /* VLAN ID already set (e.g., by PMKSA caching), so bind STA */
1626                 ap_sta_bind_vlan(hapd, sta, 0);
1627         }
1628
1629         hapd->drv.set_sta_flags(hapd, sta);
1630
1631         if (sta->auth_alg == WLAN_AUTH_FT)
1632                 wpa_auth_sm_event(sta->wpa_sm, WPA_ASSOC_FT);
1633         else
1634                 wpa_auth_sm_event(sta->wpa_sm, WPA_ASSOC);
1635         hapd->new_assoc_sta_cb(hapd, sta, !new_assoc);
1636
1637         ieee802_1x_notify_port_enabled(sta->eapol_sm, 1);
1638
1639  fail:
1640         /* Copy of the association request is not needed anymore */
1641         if (sta->last_assoc_req) {
1642                 os_free(sta->last_assoc_req);
1643                 sta->last_assoc_req = NULL;
1644         }
1645 }
1646
1647
1648 /**
1649  * ieee802_11_mgmt_cb - Process management frame TX status callback
1650  * @hapd: hostapd BSS data structure (the BSS from which the management frame
1651  * was sent from)
1652  * @buf: management frame data (starting from IEEE 802.11 header)
1653  * @len: length of frame data in octets
1654  * @stype: management frame subtype from frame control field
1655  * @ok: Whether the frame was ACK'ed
1656  */
1657 void ieee802_11_mgmt_cb(struct hostapd_data *hapd, const u8 *buf, size_t len,
1658                         u16 stype, int ok)
1659 {
1660         const struct ieee80211_mgmt *mgmt;
1661         mgmt = (const struct ieee80211_mgmt *) buf;
1662
1663         switch (stype) {
1664         case WLAN_FC_STYPE_AUTH:
1665                 wpa_printf(MSG_DEBUG, "mgmt::auth cb");
1666                 handle_auth_cb(hapd, mgmt, len, ok);
1667                 break;
1668         case WLAN_FC_STYPE_ASSOC_RESP:
1669                 wpa_printf(MSG_DEBUG, "mgmt::assoc_resp cb");
1670                 handle_assoc_cb(hapd, mgmt, len, 0, ok);
1671                 break;
1672         case WLAN_FC_STYPE_REASSOC_RESP:
1673                 wpa_printf(MSG_DEBUG, "mgmt::reassoc_resp cb");
1674                 handle_assoc_cb(hapd, mgmt, len, 1, ok);
1675                 break;
1676         case WLAN_FC_STYPE_PROBE_RESP:
1677                 wpa_printf(MSG_DEBUG, "mgmt::proberesp cb");
1678                 break;
1679         case WLAN_FC_STYPE_DEAUTH:
1680                 /* ignore */
1681                 break;
1682         case WLAN_FC_STYPE_ACTION:
1683                 wpa_printf(MSG_DEBUG, "mgmt::action cb");
1684                 break;
1685         default:
1686                 printf("unknown mgmt cb frame subtype %d\n", stype);
1687                 break;
1688         }
1689 }
1690
1691
1692 int ieee802_11_get_mib(struct hostapd_data *hapd, char *buf, size_t buflen)
1693 {
1694         /* TODO */
1695         return 0;
1696 }
1697
1698
1699 int ieee802_11_get_mib_sta(struct hostapd_data *hapd, struct sta_info *sta,
1700                            char *buf, size_t buflen)
1701 {
1702         /* TODO */
1703         return 0;
1704 }
1705
1706
1707 void hostapd_tx_status(struct hostapd_data *hapd, const u8 *addr,
1708                        const u8 *buf, size_t len, int ack)
1709 {
1710         struct sta_info *sta;
1711         struct hostapd_iface *iface = hapd->iface;
1712
1713         sta = ap_get_sta(hapd, addr);
1714         if (sta == NULL && iface->num_bss > 1) {
1715                 size_t j;
1716                 for (j = 0; j < iface->num_bss; j++) {
1717                         hapd = iface->bss[j];
1718                         sta = ap_get_sta(hapd, addr);
1719                         if (sta)
1720                                 break;
1721                 }
1722         }
1723         if (sta == NULL)
1724                 return;
1725         if (sta->flags & WLAN_STA_PENDING_POLL) {
1726                 wpa_printf(MSG_DEBUG, "STA " MACSTR " %s pending "
1727                            "activity poll", MAC2STR(sta->addr),
1728                            ack ? "ACKed" : "did not ACK");
1729                 if (ack)
1730                         sta->flags &= ~WLAN_STA_PENDING_POLL;
1731         }
1732
1733         ieee802_1x_tx_status(hapd, sta, buf, len, ack);
1734 }
1735
1736
1737 void ieee802_11_rx_from_unknown(struct hostapd_data *hapd, const u8 *src,
1738                                 int wds)
1739 {
1740         struct sta_info *sta;
1741
1742         sta = ap_get_sta(hapd, src);
1743         if (sta && (sta->flags & WLAN_STA_ASSOC)) {
1744                 if (wds && !(sta->flags & WLAN_STA_WDS)) {
1745                         wpa_printf(MSG_DEBUG, "Enable 4-address WDS mode for "
1746                                    "STA " MACSTR " (aid %u)",
1747                                    MAC2STR(sta->addr), sta->aid);
1748                         sta->flags |= WLAN_STA_WDS;
1749                         hapd->drv.set_wds_sta(hapd, sta->addr, sta->aid, 1);
1750                 }
1751                 return;
1752         }
1753
1754         wpa_printf(MSG_DEBUG, "Data/PS-poll frame from not associated STA "
1755                    MACSTR, MAC2STR(src));
1756         if (sta && (sta->flags & WLAN_STA_AUTH))
1757                 hapd->drv.sta_disassoc(
1758                         hapd, src,
1759                         WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
1760         else
1761                 hapd->drv.sta_deauth(
1762                         hapd, src,
1763                         WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
1764 }
1765
1766
1767 #endif /* CONFIG_NATIVE_WINDOWS */