hostapd: Fix segfault after ACS when flushing STAs
[mech_eap.git] / src / ap / sta_info.c
1 /*
2  * hostapd / Station table
3  * Copyright (c) 2002-2013, 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 "utils/eloop.h"
13 #include "common/ieee802_11_defs.h"
14 #include "common/wpa_ctrl.h"
15 #include "common/sae.h"
16 #include "radius/radius.h"
17 #include "radius/radius_client.h"
18 #include "drivers/driver.h"
19 #include "p2p/p2p.h"
20 #include "hostapd.h"
21 #include "accounting.h"
22 #include "ieee802_1x.h"
23 #include "ieee802_11.h"
24 #include "ieee802_11_auth.h"
25 #include "wpa_auth.h"
26 #include "preauth_auth.h"
27 #include "ap_config.h"
28 #include "beacon.h"
29 #include "ap_mlme.h"
30 #include "vlan_init.h"
31 #include "p2p_hostapd.h"
32 #include "ap_drv_ops.h"
33 #include "gas_serv.h"
34 #include "sta_info.h"
35
36 static void ap_sta_remove_in_other_bss(struct hostapd_data *hapd,
37                                        struct sta_info *sta);
38 static void ap_handle_session_timer(void *eloop_ctx, void *timeout_ctx);
39 static void ap_sta_deauth_cb_timeout(void *eloop_ctx, void *timeout_ctx);
40 static void ap_sta_disassoc_cb_timeout(void *eloop_ctx, void *timeout_ctx);
41 #ifdef CONFIG_IEEE80211W
42 static void ap_sa_query_timer(void *eloop_ctx, void *timeout_ctx);
43 #endif /* CONFIG_IEEE80211W */
44 static int ap_sta_remove(struct hostapd_data *hapd, struct sta_info *sta);
45
46 int ap_for_each_sta(struct hostapd_data *hapd,
47                     int (*cb)(struct hostapd_data *hapd, struct sta_info *sta,
48                               void *ctx),
49                     void *ctx)
50 {
51         struct sta_info *sta;
52
53         for (sta = hapd->sta_list; sta; sta = sta->next) {
54                 if (cb(hapd, sta, ctx))
55                         return 1;
56         }
57
58         return 0;
59 }
60
61
62 struct sta_info * ap_get_sta(struct hostapd_data *hapd, const u8 *sta)
63 {
64         struct sta_info *s;
65
66         s = hapd->sta_hash[STA_HASH(sta)];
67         while (s != NULL && os_memcmp(s->addr, sta, 6) != 0)
68                 s = s->hnext;
69         return s;
70 }
71
72
73 #ifdef CONFIG_P2P
74 struct sta_info * ap_get_sta_p2p(struct hostapd_data *hapd, const u8 *addr)
75 {
76         struct sta_info *sta;
77
78         for (sta = hapd->sta_list; sta; sta = sta->next) {
79                 const u8 *p2p_dev_addr;
80
81                 if (sta->p2p_ie == NULL)
82                         continue;
83
84                 p2p_dev_addr = p2p_get_go_dev_addr(sta->p2p_ie);
85                 if (p2p_dev_addr == NULL)
86                         continue;
87
88                 if (os_memcmp(p2p_dev_addr, addr, ETH_ALEN) == 0)
89                         return sta;
90         }
91
92         return NULL;
93 }
94 #endif /* CONFIG_P2P */
95
96
97 static void ap_sta_list_del(struct hostapd_data *hapd, struct sta_info *sta)
98 {
99         struct sta_info *tmp;
100
101         if (hapd->sta_list == sta) {
102                 hapd->sta_list = sta->next;
103                 return;
104         }
105
106         tmp = hapd->sta_list;
107         while (tmp != NULL && tmp->next != sta)
108                 tmp = tmp->next;
109         if (tmp == NULL) {
110                 wpa_printf(MSG_DEBUG, "Could not remove STA " MACSTR " from "
111                            "list.", MAC2STR(sta->addr));
112         } else
113                 tmp->next = sta->next;
114 }
115
116
117 void ap_sta_hash_add(struct hostapd_data *hapd, struct sta_info *sta)
118 {
119         sta->hnext = hapd->sta_hash[STA_HASH(sta->addr)];
120         hapd->sta_hash[STA_HASH(sta->addr)] = sta;
121 }
122
123
124 static void ap_sta_hash_del(struct hostapd_data *hapd, struct sta_info *sta)
125 {
126         struct sta_info *s;
127
128         s = hapd->sta_hash[STA_HASH(sta->addr)];
129         if (s == NULL) return;
130         if (os_memcmp(s->addr, sta->addr, 6) == 0) {
131                 hapd->sta_hash[STA_HASH(sta->addr)] = s->hnext;
132                 return;
133         }
134
135         while (s->hnext != NULL &&
136                os_memcmp(s->hnext->addr, sta->addr, ETH_ALEN) != 0)
137                 s = s->hnext;
138         if (s->hnext != NULL)
139                 s->hnext = s->hnext->hnext;
140         else
141                 wpa_printf(MSG_DEBUG, "AP: could not remove STA " MACSTR
142                            " from hash table", MAC2STR(sta->addr));
143 }
144
145
146 void ap_free_sta(struct hostapd_data *hapd, struct sta_info *sta)
147 {
148         int set_beacon = 0;
149
150         accounting_sta_stop(hapd, sta);
151
152         /* just in case */
153         ap_sta_set_authorized(hapd, sta, 0);
154
155         if (sta->flags & WLAN_STA_WDS)
156                 hostapd_set_wds_sta(hapd, NULL, sta->addr, sta->aid, 0);
157
158         if (!(sta->flags & WLAN_STA_PREAUTH))
159                 hostapd_drv_sta_remove(hapd, sta->addr);
160
161         ap_sta_hash_del(hapd, sta);
162         ap_sta_list_del(hapd, sta);
163
164         if (sta->aid > 0)
165                 hapd->sta_aid[(sta->aid - 1) / 32] &=
166                         ~BIT((sta->aid - 1) % 32);
167
168         hapd->num_sta--;
169         if (sta->nonerp_set) {
170                 sta->nonerp_set = 0;
171                 hapd->iface->num_sta_non_erp--;
172                 if (hapd->iface->num_sta_non_erp == 0)
173                         set_beacon++;
174         }
175
176         if (sta->no_short_slot_time_set) {
177                 sta->no_short_slot_time_set = 0;
178                 hapd->iface->num_sta_no_short_slot_time--;
179                 if (hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G
180                     && hapd->iface->num_sta_no_short_slot_time == 0)
181                         set_beacon++;
182         }
183
184         if (sta->no_short_preamble_set) {
185                 sta->no_short_preamble_set = 0;
186                 hapd->iface->num_sta_no_short_preamble--;
187                 if (hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G
188                     && hapd->iface->num_sta_no_short_preamble == 0)
189                         set_beacon++;
190         }
191
192         if (sta->no_ht_gf_set) {
193                 sta->no_ht_gf_set = 0;
194                 hapd->iface->num_sta_ht_no_gf--;
195         }
196
197         if (sta->no_ht_set) {
198                 sta->no_ht_set = 0;
199                 hapd->iface->num_sta_no_ht--;
200         }
201
202         if (sta->ht_20mhz_set) {
203                 sta->ht_20mhz_set = 0;
204                 hapd->iface->num_sta_ht_20mhz--;
205         }
206
207 #ifdef CONFIG_P2P
208         if (sta->no_p2p_set) {
209                 sta->no_p2p_set = 0;
210                 hapd->num_sta_no_p2p--;
211                 if (hapd->num_sta_no_p2p == 0)
212                         hostapd_p2p_non_p2p_sta_disconnected(hapd);
213         }
214 #endif /* CONFIG_P2P */
215
216 #if defined(NEED_AP_MLME) && defined(CONFIG_IEEE80211N)
217         if (hostapd_ht_operation_update(hapd->iface) > 0)
218                 set_beacon++;
219 #endif /* NEED_AP_MLME && CONFIG_IEEE80211N */
220
221         if (set_beacon)
222                 ieee802_11_set_beacons(hapd->iface);
223
224         wpa_printf(MSG_DEBUG, "%s: cancel ap_handle_timer for " MACSTR,
225                    __func__, MAC2STR(sta->addr));
226         eloop_cancel_timeout(ap_handle_timer, hapd, sta);
227         eloop_cancel_timeout(ap_handle_session_timer, hapd, sta);
228         eloop_cancel_timeout(ap_sta_deauth_cb_timeout, hapd, sta);
229         eloop_cancel_timeout(ap_sta_disassoc_cb_timeout, hapd, sta);
230
231         ieee802_1x_free_station(sta);
232         wpa_auth_sta_deinit(sta->wpa_sm);
233         rsn_preauth_free_station(hapd, sta);
234 #ifndef CONFIG_NO_RADIUS
235         if (hapd->radius)
236                 radius_client_flush_auth(hapd->radius, sta->addr);
237 #endif /* CONFIG_NO_RADIUS */
238
239         os_free(sta->last_assoc_req);
240         os_free(sta->challenge);
241
242 #ifdef CONFIG_IEEE80211W
243         os_free(sta->sa_query_trans_id);
244         eloop_cancel_timeout(ap_sa_query_timer, hapd, sta);
245 #endif /* CONFIG_IEEE80211W */
246
247 #ifdef CONFIG_P2P
248         p2p_group_notif_disassoc(hapd->p2p_group, sta->addr);
249 #endif /* CONFIG_P2P */
250
251 #ifdef CONFIG_INTERWORKING
252         if (sta->gas_dialog) {
253                 int i;
254                 for (i = 0; i < GAS_DIALOG_MAX; i++)
255                         gas_serv_dialog_clear(&sta->gas_dialog[i]);
256                 os_free(sta->gas_dialog);
257         }
258 #endif /* CONFIG_INTERWORKING */
259
260         wpabuf_free(sta->wps_ie);
261         wpabuf_free(sta->p2p_ie);
262         wpabuf_free(sta->hs20_ie);
263
264         os_free(sta->ht_capabilities);
265         hostapd_free_psk_list(sta->psk);
266         os_free(sta->identity);
267         os_free(sta->radius_cui);
268
269 #ifdef CONFIG_SAE
270         sae_clear_data(sta->sae);
271         os_free(sta->sae);
272 #endif /* CONFIG_SAE */
273
274         os_free(sta);
275 }
276
277
278 void hostapd_free_stas(struct hostapd_data *hapd)
279 {
280         struct sta_info *sta, *prev;
281
282         sta = hapd->sta_list;
283
284         while (sta) {
285                 prev = sta;
286                 if (sta->flags & WLAN_STA_AUTH) {
287                         mlme_deauthenticate_indication(
288                                 hapd, sta, WLAN_REASON_UNSPECIFIED);
289                 }
290                 sta = sta->next;
291                 wpa_printf(MSG_DEBUG, "Removing station " MACSTR,
292                            MAC2STR(prev->addr));
293                 ap_free_sta(hapd, prev);
294         }
295 }
296
297
298 /**
299  * ap_handle_timer - Per STA timer handler
300  * @eloop_ctx: struct hostapd_data *
301  * @timeout_ctx: struct sta_info *
302  *
303  * This function is called to check station activity and to remove inactive
304  * stations.
305  */
306 void ap_handle_timer(void *eloop_ctx, void *timeout_ctx)
307 {
308         struct hostapd_data *hapd = eloop_ctx;
309         struct sta_info *sta = timeout_ctx;
310         unsigned long next_time = 0;
311         int reason;
312
313         wpa_printf(MSG_DEBUG, "%s: " MACSTR " flags=0x%x timeout_next=%d",
314                    __func__, MAC2STR(sta->addr), sta->flags,
315                    sta->timeout_next);
316         if (sta->timeout_next == STA_REMOVE) {
317                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
318                                HOSTAPD_LEVEL_INFO, "deauthenticated due to "
319                                "local deauth request");
320                 ap_free_sta(hapd, sta);
321                 return;
322         }
323
324         if ((sta->flags & WLAN_STA_ASSOC) &&
325             (sta->timeout_next == STA_NULLFUNC ||
326              sta->timeout_next == STA_DISASSOC)) {
327                 int inactive_sec;
328                 /*
329                  * Add random value to timeout so that we don't end up bouncing
330                  * all stations at the same time if we have lots of associated
331                  * stations that are idle (but keep re-associating).
332                  */
333                 int fuzz = os_random() % 20;
334                 inactive_sec = hostapd_drv_get_inact_sec(hapd, sta->addr);
335                 if (inactive_sec == -1) {
336                         wpa_msg(hapd->msg_ctx, MSG_DEBUG,
337                                 "Check inactivity: Could not "
338                                 "get station info from kernel driver for "
339                                 MACSTR, MAC2STR(sta->addr));
340                         /*
341                          * The driver may not support this functionality.
342                          * Anyway, try again after the next inactivity timeout,
343                          * but do not disconnect the station now.
344                          */
345                         next_time = hapd->conf->ap_max_inactivity + fuzz;
346                 } else if (inactive_sec < hapd->conf->ap_max_inactivity &&
347                            sta->flags & WLAN_STA_ASSOC) {
348                         /* station activity detected; reset timeout state */
349                         wpa_msg(hapd->msg_ctx, MSG_DEBUG,
350                                 "Station " MACSTR " has been active %is ago",
351                                 MAC2STR(sta->addr), inactive_sec);
352                         sta->timeout_next = STA_NULLFUNC;
353                         next_time = hapd->conf->ap_max_inactivity + fuzz -
354                                 inactive_sec;
355                 } else {
356                         wpa_msg(hapd->msg_ctx, MSG_DEBUG,
357                                 "Station " MACSTR " has been "
358                                 "inactive too long: %d sec, max allowed: %d",
359                                 MAC2STR(sta->addr), inactive_sec,
360                                 hapd->conf->ap_max_inactivity);
361
362                         if (hapd->conf->skip_inactivity_poll)
363                                 sta->timeout_next = STA_DISASSOC;
364                 }
365         }
366
367         if ((sta->flags & WLAN_STA_ASSOC) &&
368             sta->timeout_next == STA_DISASSOC &&
369             !(sta->flags & WLAN_STA_PENDING_POLL) &&
370             !hapd->conf->skip_inactivity_poll) {
371                 wpa_msg(hapd->msg_ctx, MSG_DEBUG, "Station " MACSTR
372                         " has ACKed data poll", MAC2STR(sta->addr));
373                 /* data nullfunc frame poll did not produce TX errors; assume
374                  * station ACKed it */
375                 sta->timeout_next = STA_NULLFUNC;
376                 next_time = hapd->conf->ap_max_inactivity;
377         }
378
379         if (next_time) {
380                 wpa_printf(MSG_DEBUG, "%s: register ap_handle_timer timeout "
381                            "for " MACSTR " (%lu seconds)",
382                            __func__, MAC2STR(sta->addr), next_time);
383                 eloop_register_timeout(next_time, 0, ap_handle_timer, hapd,
384                                        sta);
385                 return;
386         }
387
388         if (sta->timeout_next == STA_NULLFUNC &&
389             (sta->flags & WLAN_STA_ASSOC)) {
390                 wpa_printf(MSG_DEBUG, "  Polling STA");
391                 sta->flags |= WLAN_STA_PENDING_POLL;
392                 hostapd_drv_poll_client(hapd, hapd->own_addr, sta->addr,
393                                         sta->flags & WLAN_STA_WMM);
394         } else if (sta->timeout_next != STA_REMOVE) {
395                 int deauth = sta->timeout_next == STA_DEAUTH;
396
397                 wpa_dbg(hapd->msg_ctx, MSG_DEBUG,
398                         "Timeout, sending %s info to STA " MACSTR,
399                         deauth ? "deauthentication" : "disassociation",
400                         MAC2STR(sta->addr));
401
402                 if (deauth) {
403                         hostapd_drv_sta_deauth(
404                                 hapd, sta->addr,
405                                 WLAN_REASON_PREV_AUTH_NOT_VALID);
406                 } else {
407                         reason = (sta->timeout_next == STA_DISASSOC) ?
408                                 WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY :
409                                 WLAN_REASON_PREV_AUTH_NOT_VALID;
410
411                         hostapd_drv_sta_disassoc(hapd, sta->addr, reason);
412                 }
413         }
414
415         switch (sta->timeout_next) {
416         case STA_NULLFUNC:
417                 sta->timeout_next = STA_DISASSOC;
418                 wpa_printf(MSG_DEBUG, "%s: register ap_handle_timer timeout "
419                            "for " MACSTR " (%d seconds - AP_DISASSOC_DELAY)",
420                            __func__, MAC2STR(sta->addr), AP_DISASSOC_DELAY);
421                 eloop_register_timeout(AP_DISASSOC_DELAY, 0, ap_handle_timer,
422                                        hapd, sta);
423                 break;
424         case STA_DISASSOC:
425         case STA_DISASSOC_FROM_CLI:
426                 ap_sta_set_authorized(hapd, sta, 0);
427                 sta->flags &= ~WLAN_STA_ASSOC;
428                 ieee802_1x_notify_port_enabled(sta->eapol_sm, 0);
429                 if (!sta->acct_terminate_cause)
430                         sta->acct_terminate_cause =
431                                 RADIUS_ACCT_TERMINATE_CAUSE_IDLE_TIMEOUT;
432                 accounting_sta_stop(hapd, sta);
433                 ieee802_1x_free_station(sta);
434                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
435                                HOSTAPD_LEVEL_INFO, "disassociated due to "
436                                "inactivity");
437                 reason = (sta->timeout_next == STA_DISASSOC) ?
438                         WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY :
439                         WLAN_REASON_PREV_AUTH_NOT_VALID;
440                 sta->timeout_next = STA_DEAUTH;
441                 wpa_printf(MSG_DEBUG, "%s: register ap_handle_timer timeout "
442                            "for " MACSTR " (%d seconds - AP_DEAUTH_DELAY)",
443                            __func__, MAC2STR(sta->addr), AP_DEAUTH_DELAY);
444                 eloop_register_timeout(AP_DEAUTH_DELAY, 0, ap_handle_timer,
445                                        hapd, sta);
446                 mlme_disassociate_indication(hapd, sta, reason);
447                 break;
448         case STA_DEAUTH:
449         case STA_REMOVE:
450                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
451                                HOSTAPD_LEVEL_INFO, "deauthenticated due to "
452                                "inactivity (timer DEAUTH/REMOVE)");
453                 if (!sta->acct_terminate_cause)
454                         sta->acct_terminate_cause =
455                                 RADIUS_ACCT_TERMINATE_CAUSE_IDLE_TIMEOUT;
456                 mlme_deauthenticate_indication(
457                         hapd, sta,
458                         WLAN_REASON_PREV_AUTH_NOT_VALID);
459                 ap_free_sta(hapd, sta);
460                 break;
461         }
462 }
463
464
465 static void ap_handle_session_timer(void *eloop_ctx, void *timeout_ctx)
466 {
467         struct hostapd_data *hapd = eloop_ctx;
468         struct sta_info *sta = timeout_ctx;
469         u8 addr[ETH_ALEN];
470
471         if (!(sta->flags & WLAN_STA_AUTH)) {
472                 if (sta->flags & WLAN_STA_GAS) {
473                         wpa_printf(MSG_DEBUG, "GAS: Remove temporary STA "
474                                    "entry " MACSTR, MAC2STR(sta->addr));
475                         ap_free_sta(hapd, sta);
476                 }
477                 return;
478         }
479
480         mlme_deauthenticate_indication(hapd, sta,
481                                        WLAN_REASON_PREV_AUTH_NOT_VALID);
482         hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
483                        HOSTAPD_LEVEL_INFO, "deauthenticated due to "
484                        "session timeout");
485         sta->acct_terminate_cause =
486                 RADIUS_ACCT_TERMINATE_CAUSE_SESSION_TIMEOUT;
487         os_memcpy(addr, sta->addr, ETH_ALEN);
488         ap_free_sta(hapd, sta);
489         hostapd_drv_sta_deauth(hapd, addr, WLAN_REASON_PREV_AUTH_NOT_VALID);
490 }
491
492
493 void ap_sta_session_timeout(struct hostapd_data *hapd, struct sta_info *sta,
494                             u32 session_timeout)
495 {
496         hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
497                        HOSTAPD_LEVEL_DEBUG, "setting session timeout to %d "
498                        "seconds", session_timeout);
499         eloop_cancel_timeout(ap_handle_session_timer, hapd, sta);
500         eloop_register_timeout(session_timeout, 0, ap_handle_session_timer,
501                                hapd, sta);
502 }
503
504
505 void ap_sta_no_session_timeout(struct hostapd_data *hapd, struct sta_info *sta)
506 {
507         eloop_cancel_timeout(ap_handle_session_timer, hapd, sta);
508 }
509
510
511 struct sta_info * ap_sta_add(struct hostapd_data *hapd, const u8 *addr)
512 {
513         struct sta_info *sta;
514
515         sta = ap_get_sta(hapd, addr);
516         if (sta)
517                 return sta;
518
519         wpa_printf(MSG_DEBUG, "  New STA");
520         if (hapd->num_sta >= hapd->conf->max_num_sta) {
521                 /* FIX: might try to remove some old STAs first? */
522                 wpa_printf(MSG_DEBUG, "no more room for new STAs (%d/%d)",
523                            hapd->num_sta, hapd->conf->max_num_sta);
524                 return NULL;
525         }
526
527         sta = os_zalloc(sizeof(struct sta_info));
528         if (sta == NULL) {
529                 wpa_printf(MSG_ERROR, "malloc failed");
530                 return NULL;
531         }
532         sta->acct_interim_interval = hapd->conf->acct_interim_interval;
533         accounting_sta_get_id(hapd, sta);
534
535         /* initialize STA info data */
536         wpa_printf(MSG_DEBUG, "%s: register ap_handle_timer timeout "
537                    "for " MACSTR " (%d seconds - ap_max_inactivity)",
538                    __func__, MAC2STR(addr),
539                    hapd->conf->ap_max_inactivity);
540         eloop_register_timeout(hapd->conf->ap_max_inactivity, 0,
541                                ap_handle_timer, hapd, sta);
542         os_memcpy(sta->addr, addr, ETH_ALEN);
543         sta->next = hapd->sta_list;
544         hapd->sta_list = sta;
545         hapd->num_sta++;
546         ap_sta_hash_add(hapd, sta);
547         sta->ssid = &hapd->conf->ssid;
548         ap_sta_remove_in_other_bss(hapd, sta);
549
550         return sta;
551 }
552
553
554 static int ap_sta_remove(struct hostapd_data *hapd, struct sta_info *sta)
555 {
556         ieee802_1x_notify_port_enabled(sta->eapol_sm, 0);
557
558         wpa_printf(MSG_DEBUG, "Removing STA " MACSTR " from kernel driver",
559                    MAC2STR(sta->addr));
560         if (hostapd_drv_sta_remove(hapd, sta->addr) &&
561             sta->flags & WLAN_STA_ASSOC) {
562                 wpa_printf(MSG_DEBUG, "Could not remove station " MACSTR
563                            " from kernel driver.", MAC2STR(sta->addr));
564                 return -1;
565         }
566         return 0;
567 }
568
569
570 static void ap_sta_remove_in_other_bss(struct hostapd_data *hapd,
571                                        struct sta_info *sta)
572 {
573         struct hostapd_iface *iface = hapd->iface;
574         size_t i;
575
576         for (i = 0; i < iface->num_bss; i++) {
577                 struct hostapd_data *bss = iface->bss[i];
578                 struct sta_info *sta2;
579                 /* bss should always be set during operation, but it may be
580                  * NULL during reconfiguration. Assume the STA is not
581                  * associated to another BSS in that case to avoid NULL pointer
582                  * dereferences. */
583                 if (bss == hapd || bss == NULL)
584                         continue;
585                 sta2 = ap_get_sta(bss, sta->addr);
586                 if (!sta2)
587                         continue;
588
589                 ap_sta_disconnect(bss, sta2, sta2->addr,
590                                   WLAN_REASON_PREV_AUTH_NOT_VALID);
591         }
592 }
593
594
595 static void ap_sta_disassoc_cb_timeout(void *eloop_ctx, void *timeout_ctx)
596 {
597         struct hostapd_data *hapd = eloop_ctx;
598         struct sta_info *sta = timeout_ctx;
599
600         ap_sta_remove(hapd, sta);
601         mlme_disassociate_indication(hapd, sta, sta->disassoc_reason);
602 }
603
604
605 void ap_sta_disassociate(struct hostapd_data *hapd, struct sta_info *sta,
606                          u16 reason)
607 {
608         wpa_printf(MSG_DEBUG, "%s: disassociate STA " MACSTR,
609                    hapd->conf->iface, MAC2STR(sta->addr));
610         sta->flags &= ~(WLAN_STA_ASSOC | WLAN_STA_ASSOC_REQ_OK);
611         ap_sta_set_authorized(hapd, sta, 0);
612         sta->timeout_next = STA_DEAUTH;
613         wpa_printf(MSG_DEBUG, "%s: reschedule ap_handle_timer timeout "
614                    "for " MACSTR " (%d seconds - "
615                    "AP_MAX_INACTIVITY_AFTER_DISASSOC)",
616                    __func__, MAC2STR(sta->addr),
617                    AP_MAX_INACTIVITY_AFTER_DISASSOC);
618         eloop_cancel_timeout(ap_handle_timer, hapd, sta);
619         eloop_register_timeout(AP_MAX_INACTIVITY_AFTER_DISASSOC, 0,
620                                ap_handle_timer, hapd, sta);
621         accounting_sta_stop(hapd, sta);
622         ieee802_1x_free_station(sta);
623
624         sta->disassoc_reason = reason;
625         sta->flags |= WLAN_STA_PENDING_DISASSOC_CB;
626         eloop_cancel_timeout(ap_sta_disassoc_cb_timeout, hapd, sta);
627         eloop_register_timeout(hapd->iface->drv_flags &
628                                WPA_DRIVER_FLAGS_DEAUTH_TX_STATUS ? 2 : 0, 0,
629                                ap_sta_disassoc_cb_timeout, hapd, sta);
630 }
631
632
633 static void ap_sta_deauth_cb_timeout(void *eloop_ctx, void *timeout_ctx)
634 {
635         struct hostapd_data *hapd = eloop_ctx;
636         struct sta_info *sta = timeout_ctx;
637
638         ap_sta_remove(hapd, sta);
639         mlme_deauthenticate_indication(hapd, sta, sta->deauth_reason);
640 }
641
642
643 void ap_sta_deauthenticate(struct hostapd_data *hapd, struct sta_info *sta,
644                            u16 reason)
645 {
646         wpa_printf(MSG_DEBUG, "%s: deauthenticate STA " MACSTR,
647                    hapd->conf->iface, MAC2STR(sta->addr));
648         sta->flags &= ~(WLAN_STA_AUTH | WLAN_STA_ASSOC);
649         ap_sta_set_authorized(hapd, sta, 0);
650         sta->timeout_next = STA_REMOVE;
651         wpa_printf(MSG_DEBUG, "%s: reschedule ap_handle_timer timeout "
652                    "for " MACSTR " (%d seconds - "
653                    "AP_MAX_INACTIVITY_AFTER_DEAUTH)",
654                    __func__, MAC2STR(sta->addr),
655                    AP_MAX_INACTIVITY_AFTER_DEAUTH);
656         eloop_cancel_timeout(ap_handle_timer, hapd, sta);
657         eloop_register_timeout(AP_MAX_INACTIVITY_AFTER_DEAUTH, 0,
658                                ap_handle_timer, hapd, sta);
659         accounting_sta_stop(hapd, sta);
660         ieee802_1x_free_station(sta);
661
662         sta->deauth_reason = reason;
663         sta->flags |= WLAN_STA_PENDING_DEAUTH_CB;
664         eloop_cancel_timeout(ap_sta_deauth_cb_timeout, hapd, sta);
665         eloop_register_timeout(hapd->iface->drv_flags &
666                                WPA_DRIVER_FLAGS_DEAUTH_TX_STATUS ? 2 : 0, 0,
667                                ap_sta_deauth_cb_timeout, hapd, sta);
668 }
669
670
671 #ifdef CONFIG_WPS
672 int ap_sta_wps_cancel(struct hostapd_data *hapd,
673                       struct sta_info *sta, void *ctx)
674 {
675         if (sta && (sta->flags & WLAN_STA_WPS)) {
676                 ap_sta_deauthenticate(hapd, sta,
677                                       WLAN_REASON_PREV_AUTH_NOT_VALID);
678                 wpa_printf(MSG_DEBUG, "WPS: %s: Deauth sta=" MACSTR,
679                            __func__, MAC2STR(sta->addr));
680                 return 1;
681         }
682
683         return 0;
684 }
685 #endif /* CONFIG_WPS */
686
687
688 int ap_sta_bind_vlan(struct hostapd_data *hapd, struct sta_info *sta,
689                      int old_vlanid)
690 {
691 #ifndef CONFIG_NO_VLAN
692         const char *iface;
693         struct hostapd_vlan *vlan = NULL;
694         int ret;
695
696         /*
697          * Do not proceed furthur if the vlan id remains same. We do not want
698          * duplicate dynamic vlan entries.
699          */
700         if (sta->vlan_id == old_vlanid)
701                 return 0;
702
703         /*
704          * During 1x reauth, if the vlan id changes, then remove the old id and
705          * proceed furthur to add the new one.
706          */
707         if (old_vlanid > 0)
708                 vlan_remove_dynamic(hapd, old_vlanid);
709
710         iface = hapd->conf->iface;
711         if (sta->ssid->vlan[0])
712                 iface = sta->ssid->vlan;
713
714         if (sta->ssid->dynamic_vlan == DYNAMIC_VLAN_DISABLED)
715                 sta->vlan_id = 0;
716         else if (sta->vlan_id > 0) {
717                 struct hostapd_vlan *wildcard_vlan = NULL;
718                 vlan = hapd->conf->vlan;
719                 while (vlan) {
720                         if (vlan->vlan_id == sta->vlan_id)
721                                 break;
722                         if (vlan->vlan_id == VLAN_ID_WILDCARD)
723                                 wildcard_vlan = vlan;
724                         vlan = vlan->next;
725                 }
726                 if (!vlan)
727                         vlan = wildcard_vlan;
728                 if (vlan)
729                         iface = vlan->ifname;
730         }
731
732         if (sta->vlan_id > 0 && vlan == NULL) {
733                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
734                                HOSTAPD_LEVEL_DEBUG, "could not find VLAN for "
735                                "binding station to (vlan_id=%d)",
736                                sta->vlan_id);
737                 return -1;
738         } else if (sta->vlan_id > 0 && vlan->vlan_id == VLAN_ID_WILDCARD) {
739                 vlan = vlan_add_dynamic(hapd, vlan, sta->vlan_id);
740                 if (vlan == NULL) {
741                         hostapd_logger(hapd, sta->addr,
742                                        HOSTAPD_MODULE_IEEE80211,
743                                        HOSTAPD_LEVEL_DEBUG, "could not add "
744                                        "dynamic VLAN interface for vlan_id=%d",
745                                        sta->vlan_id);
746                         return -1;
747                 }
748
749                 iface = vlan->ifname;
750                 if (vlan_setup_encryption_dyn(hapd, sta->ssid, iface) != 0) {
751                         hostapd_logger(hapd, sta->addr,
752                                        HOSTAPD_MODULE_IEEE80211,
753                                        HOSTAPD_LEVEL_DEBUG, "could not "
754                                        "configure encryption for dynamic VLAN "
755                                        "interface for vlan_id=%d",
756                                        sta->vlan_id);
757                 }
758
759                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
760                                HOSTAPD_LEVEL_DEBUG, "added new dynamic VLAN "
761                                "interface '%s'", iface);
762         } else if (vlan && vlan->vlan_id == sta->vlan_id) {
763                 if (sta->vlan_id > 0) {
764                         vlan->dynamic_vlan++;
765                         hostapd_logger(hapd, sta->addr,
766                                        HOSTAPD_MODULE_IEEE80211,
767                                        HOSTAPD_LEVEL_DEBUG, "updated existing "
768                                        "dynamic VLAN interface '%s'", iface);
769                 }
770
771                 /*
772                  * Update encryption configuration for statically generated
773                  * VLAN interface. This is only used for static WEP
774                  * configuration for the case where hostapd did not yet know
775                  * which keys are to be used when the interface was added.
776                  */
777                 if (vlan_setup_encryption_dyn(hapd, sta->ssid, iface) != 0) {
778                         hostapd_logger(hapd, sta->addr,
779                                        HOSTAPD_MODULE_IEEE80211,
780                                        HOSTAPD_LEVEL_DEBUG, "could not "
781                                        "configure encryption for VLAN "
782                                        "interface for vlan_id=%d",
783                                        sta->vlan_id);
784                 }
785         }
786
787         hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
788                        HOSTAPD_LEVEL_DEBUG, "binding station to interface "
789                        "'%s'", iface);
790
791         if (wpa_auth_sta_set_vlan(sta->wpa_sm, sta->vlan_id) < 0)
792                 wpa_printf(MSG_INFO, "Failed to update VLAN-ID for WPA");
793
794         ret = hostapd_drv_set_sta_vlan(iface, hapd, sta->addr, sta->vlan_id);
795         if (ret < 0) {
796                 hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
797                                HOSTAPD_LEVEL_DEBUG, "could not bind the STA "
798                                "entry to vlan_id=%d", sta->vlan_id);
799         }
800         return ret;
801 #else /* CONFIG_NO_VLAN */
802         return 0;
803 #endif /* CONFIG_NO_VLAN */
804 }
805
806
807 #ifdef CONFIG_IEEE80211W
808
809 int ap_check_sa_query_timeout(struct hostapd_data *hapd, struct sta_info *sta)
810 {
811         u32 tu;
812         struct os_time now, passed;
813         os_get_time(&now);
814         os_time_sub(&now, &sta->sa_query_start, &passed);
815         tu = (passed.sec * 1000000 + passed.usec) / 1024;
816         if (hapd->conf->assoc_sa_query_max_timeout < tu) {
817                 hostapd_logger(hapd, sta->addr,
818                                HOSTAPD_MODULE_IEEE80211,
819                                HOSTAPD_LEVEL_DEBUG,
820                                "association SA Query timed out");
821                 sta->sa_query_timed_out = 1;
822                 os_free(sta->sa_query_trans_id);
823                 sta->sa_query_trans_id = NULL;
824                 sta->sa_query_count = 0;
825                 eloop_cancel_timeout(ap_sa_query_timer, hapd, sta);
826                 return 1;
827         }
828
829         return 0;
830 }
831
832
833 static void ap_sa_query_timer(void *eloop_ctx, void *timeout_ctx)
834 {
835         struct hostapd_data *hapd = eloop_ctx;
836         struct sta_info *sta = timeout_ctx;
837         unsigned int timeout, sec, usec;
838         u8 *trans_id, *nbuf;
839
840         if (sta->sa_query_count > 0 &&
841             ap_check_sa_query_timeout(hapd, sta))
842                 return;
843
844         nbuf = os_realloc_array(sta->sa_query_trans_id,
845                                 sta->sa_query_count + 1,
846                                 WLAN_SA_QUERY_TR_ID_LEN);
847         if (nbuf == NULL)
848                 return;
849         if (sta->sa_query_count == 0) {
850                 /* Starting a new SA Query procedure */
851                 os_get_time(&sta->sa_query_start);
852         }
853         trans_id = nbuf + sta->sa_query_count * WLAN_SA_QUERY_TR_ID_LEN;
854         sta->sa_query_trans_id = nbuf;
855         sta->sa_query_count++;
856
857         os_get_random(trans_id, WLAN_SA_QUERY_TR_ID_LEN);
858
859         timeout = hapd->conf->assoc_sa_query_retry_timeout;
860         sec = ((timeout / 1000) * 1024) / 1000;
861         usec = (timeout % 1000) * 1024;
862         eloop_register_timeout(sec, usec, ap_sa_query_timer, hapd, sta);
863
864         hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
865                        HOSTAPD_LEVEL_DEBUG,
866                        "association SA Query attempt %d", sta->sa_query_count);
867
868         ieee802_11_send_sa_query_req(hapd, sta->addr, trans_id);
869 }
870
871
872 void ap_sta_start_sa_query(struct hostapd_data *hapd, struct sta_info *sta)
873 {
874         ap_sa_query_timer(hapd, sta);
875 }
876
877
878 void ap_sta_stop_sa_query(struct hostapd_data *hapd, struct sta_info *sta)
879 {
880         eloop_cancel_timeout(ap_sa_query_timer, hapd, sta);
881         os_free(sta->sa_query_trans_id);
882         sta->sa_query_trans_id = NULL;
883         sta->sa_query_count = 0;
884 }
885
886 #endif /* CONFIG_IEEE80211W */
887
888
889 void ap_sta_set_authorized(struct hostapd_data *hapd, struct sta_info *sta,
890                            int authorized)
891 {
892         const u8 *dev_addr = NULL;
893         char buf[100];
894 #ifdef CONFIG_P2P
895         u8 addr[ETH_ALEN];
896 #endif /* CONFIG_P2P */
897
898         if (!!authorized == !!(sta->flags & WLAN_STA_AUTHORIZED))
899                 return;
900
901 #ifdef CONFIG_P2P
902         if (hapd->p2p_group == NULL) {
903                 if (sta->p2p_ie != NULL &&
904                     p2p_parse_dev_addr_in_p2p_ie(sta->p2p_ie, addr) == 0)
905                         dev_addr = addr;
906         } else
907                 dev_addr = p2p_group_get_dev_addr(hapd->p2p_group, sta->addr);
908 #endif /* CONFIG_P2P */
909
910         if (dev_addr)
911                 os_snprintf(buf, sizeof(buf), MACSTR " p2p_dev_addr=" MACSTR,
912                             MAC2STR(sta->addr), MAC2STR(dev_addr));
913         else
914                 os_snprintf(buf, sizeof(buf), MACSTR, MAC2STR(sta->addr));
915
916         if (authorized) {
917                 wpa_msg(hapd->msg_ctx, MSG_INFO, AP_STA_CONNECTED "%s", buf);
918
919                 if (hapd->msg_ctx_parent &&
920                     hapd->msg_ctx_parent != hapd->msg_ctx)
921                         wpa_msg_no_global(hapd->msg_ctx_parent, MSG_INFO,
922                                           AP_STA_CONNECTED "%s", buf);
923
924                 sta->flags |= WLAN_STA_AUTHORIZED;
925         } else {
926                 wpa_msg(hapd->msg_ctx, MSG_INFO, AP_STA_DISCONNECTED "%s", buf);
927
928                 if (hapd->msg_ctx_parent &&
929                     hapd->msg_ctx_parent != hapd->msg_ctx)
930                         wpa_msg_no_global(hapd->msg_ctx_parent, MSG_INFO,
931                                           AP_STA_DISCONNECTED "%s", buf);
932
933                 sta->flags &= ~WLAN_STA_AUTHORIZED;
934         }
935
936         if (hapd->sta_authorized_cb)
937                 hapd->sta_authorized_cb(hapd->sta_authorized_cb_ctx,
938                                         sta->addr, authorized, dev_addr);
939 }
940
941
942 void ap_sta_disconnect(struct hostapd_data *hapd, struct sta_info *sta,
943                        const u8 *addr, u16 reason)
944 {
945
946         if (sta == NULL && addr)
947                 sta = ap_get_sta(hapd, addr);
948
949         if (addr)
950                 hostapd_drv_sta_deauth(hapd, addr, reason);
951
952         if (sta == NULL)
953                 return;
954         ap_sta_set_authorized(hapd, sta, 0);
955         wpa_auth_sm_event(sta->wpa_sm, WPA_DEAUTH);
956         ieee802_1x_notify_port_enabled(sta->eapol_sm, 0);
957         sta->flags &= ~(WLAN_STA_AUTH | WLAN_STA_ASSOC);
958         wpa_printf(MSG_DEBUG, "%s: reschedule ap_handle_timer timeout "
959                    "for " MACSTR " (%d seconds - "
960                    "AP_MAX_INACTIVITY_AFTER_DEAUTH)",
961                    __func__, MAC2STR(sta->addr),
962                    AP_MAX_INACTIVITY_AFTER_DEAUTH);
963         eloop_cancel_timeout(ap_handle_timer, hapd, sta);
964         eloop_register_timeout(AP_MAX_INACTIVITY_AFTER_DEAUTH, 0,
965                                ap_handle_timer, hapd, sta);
966         sta->timeout_next = STA_REMOVE;
967
968         sta->deauth_reason = reason;
969         sta->flags |= WLAN_STA_PENDING_DEAUTH_CB;
970         eloop_cancel_timeout(ap_sta_deauth_cb_timeout, hapd, sta);
971         eloop_register_timeout(hapd->iface->drv_flags &
972                                WPA_DRIVER_FLAGS_DEAUTH_TX_STATUS ? 2 : 0, 0,
973                                ap_sta_deauth_cb_timeout, hapd, sta);
974 }
975
976
977 void ap_sta_deauth_cb(struct hostapd_data *hapd, struct sta_info *sta)
978 {
979         if (!(sta->flags & WLAN_STA_PENDING_DEAUTH_CB)) {
980                 wpa_printf(MSG_DEBUG, "Ignore deauth cb for test frame");
981                 return;
982         }
983         sta->flags &= ~WLAN_STA_PENDING_DEAUTH_CB;
984         eloop_cancel_timeout(ap_sta_deauth_cb_timeout, hapd, sta);
985         ap_sta_deauth_cb_timeout(hapd, sta);
986 }
987
988
989 void ap_sta_disassoc_cb(struct hostapd_data *hapd, struct sta_info *sta)
990 {
991         if (!(sta->flags & WLAN_STA_PENDING_DISASSOC_CB)) {
992                 wpa_printf(MSG_DEBUG, "Ignore disassoc cb for test frame");
993                 return;
994         }
995         sta->flags &= ~WLAN_STA_PENDING_DISASSOC_CB;
996         eloop_cancel_timeout(ap_sta_disassoc_cb_timeout, hapd, sta);
997         ap_sta_disassoc_cb_timeout(hapd, sta);
998 }