FT: Verify that MDIE and FTIE matches between AssocResp and EAPOL-Key 3/4
[libeap.git] / wpa_supplicant / mlme.c
1 /*
2  * WPA Supplicant - Client mode MLME
3  * Copyright (c) 2003-2008, Jouni Malinen <j@w1.fi>
4  * Copyright (c) 2004, Instant802 Networks, Inc.
5  * Copyright (c) 2005-2006, Devicescape Software, Inc.
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License version 2 as
9  * published by the Free Software Foundation.
10  *
11  * Alternatively, this software may be distributed under the terms of BSD
12  * license.
13  *
14  * See README and COPYING for more details.
15  */
16
17 #include "includes.h"
18
19 #include "common.h"
20 #include "eloop.h"
21 #include "config_ssid.h"
22 #include "wpa_supplicant_i.h"
23 #include "notify.h"
24 #include "driver_i.h"
25 #include "rsn_supp/wpa.h"
26 #include "common/ieee802_11_defs.h"
27 #include "common/ieee802_11_common.h"
28 #include "mlme.h"
29
30
31 /* Timeouts and intervals in milliseconds */
32 #define IEEE80211_AUTH_TIMEOUT (200)
33 #define IEEE80211_AUTH_MAX_TRIES 3
34 #define IEEE80211_ASSOC_TIMEOUT (200)
35 #define IEEE80211_ASSOC_MAX_TRIES 3
36 #define IEEE80211_MONITORING_INTERVAL (2000)
37 #define IEEE80211_PROBE_INTERVAL (60000)
38 #define IEEE80211_RETRY_AUTH_INTERVAL (1000)
39 #define IEEE80211_SCAN_INTERVAL (2000)
40 #define IEEE80211_SCAN_INTERVAL_SLOW (15000)
41 #define IEEE80211_IBSS_JOIN_TIMEOUT (20000)
42
43 #define IEEE80211_PROBE_DELAY (33)
44 #define IEEE80211_CHANNEL_TIME (33)
45 #define IEEE80211_PASSIVE_CHANNEL_TIME (200)
46 #define IEEE80211_SCAN_RESULT_EXPIRE (10000)
47 #define IEEE80211_IBSS_MERGE_INTERVAL (30000)
48 #define IEEE80211_IBSS_INACTIVITY_LIMIT (60000)
49
50 #define IEEE80211_IBSS_MAX_STA_ENTRIES 128
51
52
53 #define IEEE80211_FC(type, stype) host_to_le16((type << 2) | (stype << 4))
54
55
56 struct ieee80211_sta_bss {
57         struct ieee80211_sta_bss *next;
58         struct ieee80211_sta_bss *hnext;
59
60         u8 bssid[ETH_ALEN];
61         u8 ssid[MAX_SSID_LEN];
62         size_t ssid_len;
63         u16 capability; /* host byte order */
64         int hw_mode;
65         int channel;
66         int freq;
67         int rssi;
68         u8 *ie;
69         size_t ie_len;
70         u8 *wpa_ie;
71         size_t wpa_ie_len;
72         u8 *rsn_ie;
73         size_t rsn_ie_len;
74         u8 *wmm_ie;
75         size_t wmm_ie_len;
76         u8 *mdie;
77         size_t mdie_len;
78 #define IEEE80211_MAX_SUPP_RATES 32
79         u8 supp_rates[IEEE80211_MAX_SUPP_RATES];
80         size_t supp_rates_len;
81         int beacon_int;
82         u64 timestamp;
83
84         int probe_resp;
85         struct os_time last_update;
86 };
87
88
89 static void ieee80211_send_probe_req(struct wpa_supplicant *wpa_s,
90                                      const u8 *dst,
91                                      const u8 *ssid, size_t ssid_len);
92 static struct ieee80211_sta_bss *
93 ieee80211_bss_get(struct wpa_supplicant *wpa_s, const u8 *bssid);
94 static int ieee80211_sta_find_ibss(struct wpa_supplicant *wpa_s);
95 static int ieee80211_sta_wep_configured(struct wpa_supplicant *wpa_s);
96 static void ieee80211_sta_timer(void *eloop_ctx, void *timeout_ctx);
97 static void ieee80211_sta_scan_timer(void *eloop_ctx, void *timeout_ctx);
98 static void ieee80211_build_tspec(struct wpabuf *buf);
99 static int ieee80211_sta_set_probe_req_ie(struct wpa_supplicant *wpa_s,
100                                           const u8 *ies, size_t ies_len);
101
102
103 static int ieee80211_sta_set_channel(struct wpa_supplicant *wpa_s,
104                                      enum hostapd_hw_mode phymode, int chan,
105                                      int freq)
106 {
107         size_t i;
108         struct hostapd_hw_modes *mode;
109
110         for (i = 0; i < wpa_s->mlme.num_modes; i++) {
111                 mode = &wpa_s->mlme.modes[i];
112                 if (mode->mode == phymode) {
113                         wpa_s->mlme.curr_rates = mode->rates;
114                         wpa_s->mlme.num_curr_rates = mode->num_rates;
115                         break;
116                 }
117         }
118
119         return wpa_drv_set_channel(wpa_s, phymode, chan, freq);
120 }
121
122
123 static int ecw2cw(int ecw)
124 {
125         int cw = 1;
126         while (ecw > 0) {
127                 cw <<= 1;
128                 ecw--;
129         }
130         return cw - 1;
131 }
132
133
134 static void ieee80211_sta_wmm_params(struct wpa_supplicant *wpa_s,
135                                      const u8 *wmm_param, size_t wmm_param_len)
136 {
137         size_t left;
138         int count;
139         const u8 *pos;
140         u8 wmm_acm;
141
142         if (wmm_param_len < 8 || wmm_param[5] /* version */ != 1)
143                 return;
144         count = wmm_param[6] & 0x0f;
145         if (count == wpa_s->mlme.wmm_last_param_set)
146                 return;
147         wpa_s->mlme.wmm_last_param_set = count;
148
149         pos = wmm_param + 8;
150         left = wmm_param_len - 8;
151
152         wmm_acm = 0;
153         for (; left >= 4; left -= 4, pos += 4) {
154                 int aci = (pos[0] >> 5) & 0x03;
155                 int acm = (pos[0] >> 4) & 0x01;
156                 int aifs, cw_max, cw_min, burst_time;
157
158                 switch (aci) {
159                 case 1: /* AC_BK */
160                         if (acm)
161                                 wmm_acm |= BIT(1) | BIT(2); /* BK/- */
162                         break;
163                 case 2: /* AC_VI */
164                         if (acm)
165                                 wmm_acm |= BIT(4) | BIT(5); /* CL/VI */
166                         break;
167                 case 3: /* AC_VO */
168                         if (acm)
169                                 wmm_acm |= BIT(6) | BIT(7); /* VO/NC */
170                         break;
171                 case 0: /* AC_BE */
172                 default:
173                         if (acm)
174                                 wmm_acm |= BIT(0) | BIT(3); /* BE/EE */
175                         break;
176                 }
177
178                 aifs = pos[0] & 0x0f;
179                 cw_max = ecw2cw((pos[1] & 0xf0) >> 4);
180                 cw_min = ecw2cw(pos[1] & 0x0f);
181                 /* TXOP is in units of 32 usec; burst_time in 0.1 ms */
182                 burst_time = (pos[2] | (pos[3] << 8)) * 32 / 100;
183                 wpa_printf(MSG_DEBUG, "MLME: WMM aci=%d acm=%d aifs=%d "
184                            "cWmin=%d cWmax=%d burst=%d",
185                            aci, acm, aifs, cw_min, cw_max, burst_time);
186                 /* TODO: driver configuration */
187         }
188 }
189
190
191 static void ieee80211_set_associated(struct wpa_supplicant *wpa_s, int assoc)
192 {
193         if (wpa_s->mlme.associated == assoc && !assoc)
194                 return;
195
196         wpa_s->mlme.associated = assoc;
197
198         if (assoc) {
199                 union wpa_event_data data;
200                 os_memset(&data, 0, sizeof(data));
201                 wpa_s->mlme.prev_bssid_set = 1;
202                 os_memcpy(wpa_s->mlme.prev_bssid, wpa_s->bssid, ETH_ALEN);
203                 data.assoc_info.req_ies = wpa_s->mlme.assocreq_ies;
204                 data.assoc_info.req_ies_len = wpa_s->mlme.assocreq_ies_len;
205                 data.assoc_info.resp_ies = wpa_s->mlme.assocresp_ies;
206                 data.assoc_info.resp_ies_len = wpa_s->mlme.assocresp_ies_len;
207                 data.assoc_info.freq = wpa_s->mlme.freq;
208                 wpa_supplicant_event(wpa_s, EVENT_ASSOC, &data);
209         } else {
210                 wpa_supplicant_event(wpa_s, EVENT_DISASSOC, NULL);
211         }
212         os_get_time(&wpa_s->mlme.last_probe);
213 }
214
215
216 static int ieee80211_sta_tx(struct wpa_supplicant *wpa_s, const u8 *buf,
217                             size_t len)
218 {
219         return wpa_drv_send_mlme(wpa_s, buf, len);
220 }
221
222
223 static void ieee80211_send_auth(struct wpa_supplicant *wpa_s,
224                                 int transaction, const u8 *extra,
225                                 size_t extra_len, int encrypt)
226 {
227         u8 *buf;
228         size_t len;
229         struct ieee80211_mgmt *mgmt;
230
231         buf = os_malloc(sizeof(*mgmt) + 6 + extra_len);
232         if (buf == NULL) {
233                 wpa_printf(MSG_DEBUG, "MLME: failed to allocate buffer for "
234                            "auth frame");
235                 return;
236         }
237
238         mgmt = (struct ieee80211_mgmt *) buf;
239         len = 24 + 6;
240         os_memset(mgmt, 0, 24 + 6);
241         mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
242                                            WLAN_FC_STYPE_AUTH);
243         if (encrypt)
244                 mgmt->frame_control |= host_to_le16(WLAN_FC_ISWEP);
245         os_memcpy(mgmt->da, wpa_s->bssid, ETH_ALEN);
246         os_memcpy(mgmt->sa, wpa_s->own_addr, ETH_ALEN);
247         os_memcpy(mgmt->bssid, wpa_s->bssid, ETH_ALEN);
248         mgmt->u.auth.auth_alg = host_to_le16(wpa_s->mlme.auth_alg);
249         mgmt->u.auth.auth_transaction = host_to_le16(transaction);
250         wpa_s->mlme.auth_transaction = transaction + 1;
251         mgmt->u.auth.status_code = host_to_le16(0);
252         if (extra) {
253                 os_memcpy(buf + len, extra, extra_len);
254                 len += extra_len;
255         }
256
257         ieee80211_sta_tx(wpa_s, buf, len);
258         os_free(buf);
259 }
260
261
262 static void ieee80211_reschedule_timer(struct wpa_supplicant *wpa_s, int ms)
263 {
264         eloop_cancel_timeout(ieee80211_sta_timer, wpa_s, NULL);
265         eloop_register_timeout(ms / 1000, 1000 * (ms % 1000),
266                                ieee80211_sta_timer, wpa_s, NULL);
267 }
268
269
270 static void ieee80211_authenticate(struct wpa_supplicant *wpa_s)
271 {
272         u8 *extra;
273         size_t extra_len;
274
275         wpa_s->mlme.auth_tries++;
276         if (wpa_s->mlme.auth_tries > IEEE80211_AUTH_MAX_TRIES) {
277                 wpa_printf(MSG_DEBUG, "MLME: authentication with AP " MACSTR
278                            " timed out", MAC2STR(wpa_s->bssid));
279                 return;
280         }
281
282         wpa_s->mlme.state = IEEE80211_AUTHENTICATE;
283         wpa_printf(MSG_DEBUG, "MLME: authenticate with AP " MACSTR,
284                    MAC2STR(wpa_s->bssid));
285
286         extra = NULL;
287         extra_len = 0;
288
289 #ifdef CONFIG_IEEE80211R
290         if ((wpa_s->mlme.key_mgmt == KEY_MGMT_FT_802_1X ||
291              wpa_s->mlme.key_mgmt == KEY_MGMT_FT_PSK) &&
292             wpa_s->mlme.ft_ies) {
293                 struct ieee80211_sta_bss *bss;
294                 struct rsn_mdie *mdie = NULL;
295                 bss = ieee80211_bss_get(wpa_s, wpa_s->bssid);
296                 if (bss && bss->mdie_len >= 2 + sizeof(*mdie))
297                         mdie = (struct rsn_mdie *) (bss->mdie + 2);
298                 if (mdie &&
299                     os_memcmp(mdie->mobility_domain, wpa_s->mlme.current_md,
300                               MOBILITY_DOMAIN_ID_LEN) == 0) {
301                         wpa_printf(MSG_DEBUG, "MLME: Trying to use FT "
302                                    "over-the-air");
303                         wpa_s->mlme.auth_alg = WLAN_AUTH_FT;
304                         extra = wpa_s->mlme.ft_ies;
305                         extra_len = wpa_s->mlme.ft_ies_len;
306                 }
307         }
308 #endif /* CONFIG_IEEE80211R */
309
310         ieee80211_send_auth(wpa_s, 1, extra, extra_len, 0);
311
312         ieee80211_reschedule_timer(wpa_s, IEEE80211_AUTH_TIMEOUT);
313 }
314
315
316 static void ieee80211_send_assoc(struct wpa_supplicant *wpa_s)
317 {
318         struct ieee80211_mgmt *mgmt;
319         u8 *pos, *ies, *buf;
320         int i, len;
321         u16 capab;
322         struct ieee80211_sta_bss *bss;
323         int wmm = 0;
324         size_t blen, buflen;
325
326         if (wpa_s->mlme.curr_rates == NULL) {
327                 wpa_printf(MSG_DEBUG, "MLME: curr_rates not set for assoc");
328                 return;
329         }
330
331         buflen = sizeof(*mgmt) + 200 + wpa_s->mlme.extra_ie_len +
332                 wpa_s->mlme.ssid_len;
333 #ifdef CONFIG_IEEE80211R
334         if (wpa_s->mlme.ft_ies)
335                 buflen += wpa_s->mlme.ft_ies_len;
336 #endif /* CONFIG_IEEE80211R */
337         buf = os_malloc(buflen);
338         if (buf == NULL) {
339                 wpa_printf(MSG_DEBUG, "MLME: failed to allocate buffer for "
340                            "assoc frame");
341                 return;
342         }
343         blen = 0;
344
345         capab = wpa_s->mlme.capab;
346         if (wpa_s->mlme.phymode == HOSTAPD_MODE_IEEE80211G) {
347                 capab |= WLAN_CAPABILITY_SHORT_SLOT_TIME |
348                         WLAN_CAPABILITY_SHORT_PREAMBLE;
349         }
350         bss = ieee80211_bss_get(wpa_s, wpa_s->bssid);
351         if (bss) {
352                 if (bss->capability & WLAN_CAPABILITY_PRIVACY)
353                         capab |= WLAN_CAPABILITY_PRIVACY;
354                 if (bss->wmm_ie) {
355                         wmm = 1;
356                 }
357         }
358
359         mgmt = (struct ieee80211_mgmt *) buf;
360         blen += 24;
361         os_memset(mgmt, 0, 24);
362         os_memcpy(mgmt->da, wpa_s->bssid, ETH_ALEN);
363         os_memcpy(mgmt->sa, wpa_s->own_addr, ETH_ALEN);
364         os_memcpy(mgmt->bssid, wpa_s->bssid, ETH_ALEN);
365
366         if (wpa_s->mlme.prev_bssid_set) {
367                 blen += 10;
368                 mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
369                                                    WLAN_FC_STYPE_REASSOC_REQ);
370                 mgmt->u.reassoc_req.capab_info = host_to_le16(capab);
371                 mgmt->u.reassoc_req.listen_interval = host_to_le16(1);
372                 os_memcpy(mgmt->u.reassoc_req.current_ap,
373                           wpa_s->mlme.prev_bssid,
374                           ETH_ALEN);
375         } else {
376                 blen += 4;
377                 mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
378                                                    WLAN_FC_STYPE_ASSOC_REQ);
379                 mgmt->u.assoc_req.capab_info = host_to_le16(capab);
380                 mgmt->u.assoc_req.listen_interval = host_to_le16(1);
381         }
382
383         /* SSID */
384         ies = pos = buf + blen;
385         blen += 2 + wpa_s->mlme.ssid_len;
386         *pos++ = WLAN_EID_SSID;
387         *pos++ = wpa_s->mlme.ssid_len;
388         os_memcpy(pos, wpa_s->mlme.ssid, wpa_s->mlme.ssid_len);
389
390         len = wpa_s->mlme.num_curr_rates;
391         if (len > 8)
392                 len = 8;
393         pos = buf + blen;
394         blen += len + 2;
395         *pos++ = WLAN_EID_SUPP_RATES;
396         *pos++ = len;
397         for (i = 0; i < len; i++)
398                 *pos++ = (u8) (wpa_s->mlme.curr_rates[i] / 5);
399
400         if (wpa_s->mlme.num_curr_rates > len) {
401                 pos = buf + blen;
402                 blen += wpa_s->mlme.num_curr_rates - len + 2;
403                 *pos++ = WLAN_EID_EXT_SUPP_RATES;
404                 *pos++ = wpa_s->mlme.num_curr_rates - len;
405                 for (i = len; i < wpa_s->mlme.num_curr_rates; i++)
406                         *pos++ = (u8) (wpa_s->mlme.curr_rates[i] / 5);
407         }
408
409         if (wpa_s->mlme.extra_ie && wpa_s->mlme.auth_alg != WLAN_AUTH_FT) {
410                 pos = buf + blen;
411                 blen += wpa_s->mlme.extra_ie_len;
412                 os_memcpy(pos, wpa_s->mlme.extra_ie, wpa_s->mlme.extra_ie_len);
413         }
414
415 #ifdef CONFIG_IEEE80211R
416         if ((wpa_s->mlme.key_mgmt == KEY_MGMT_FT_802_1X ||
417              wpa_s->mlme.key_mgmt == KEY_MGMT_FT_PSK) &&
418             wpa_s->mlme.auth_alg != WLAN_AUTH_FT &&
419             bss && bss->mdie &&
420             bss->mdie_len >= 2 + sizeof(struct rsn_mdie) &&
421             bss->mdie[1] >= sizeof(struct rsn_mdie)) {
422                 pos = buf + blen;
423                 blen += 2 + sizeof(struct rsn_mdie);
424                 *pos++ = WLAN_EID_MOBILITY_DOMAIN;
425                 *pos++ = sizeof(struct rsn_mdie);
426                 os_memcpy(pos, bss->mdie + 2, MOBILITY_DOMAIN_ID_LEN);
427                 pos += MOBILITY_DOMAIN_ID_LEN;
428                 *pos++ = 0; /* FIX: copy from the target AP's MDIE */
429         }
430
431         if ((wpa_s->mlme.key_mgmt == KEY_MGMT_FT_802_1X ||
432              wpa_s->mlme.key_mgmt == KEY_MGMT_FT_PSK) &&
433             wpa_s->mlme.auth_alg == WLAN_AUTH_FT && wpa_s->mlme.ft_ies) {
434                 pos = buf + blen;
435                 os_memcpy(pos, wpa_s->mlme.ft_ies, wpa_s->mlme.ft_ies_len);
436                 pos += wpa_s->mlme.ft_ies_len;
437                 blen += wpa_s->mlme.ft_ies_len;
438         }
439 #endif /* CONFIG_IEEE80211R */
440
441         if (wmm && wpa_s->mlme.wmm_enabled) {
442                 pos = buf + blen;
443                 blen += 9;
444                 *pos++ = WLAN_EID_VENDOR_SPECIFIC;
445                 *pos++ = 7; /* len */
446                 *pos++ = 0x00; /* Microsoft OUI 00:50:F2 */
447                 *pos++ = 0x50;
448                 *pos++ = 0xf2;
449                 *pos++ = 2; /* WMM */
450                 *pos++ = 0; /* WMM info */
451                 *pos++ = 1; /* WMM ver */
452                 *pos++ = 0;
453         }
454
455         os_free(wpa_s->mlme.assocreq_ies);
456         wpa_s->mlme.assocreq_ies_len = (buf + blen) - ies;
457         wpa_s->mlme.assocreq_ies = os_malloc(wpa_s->mlme.assocreq_ies_len);
458         if (wpa_s->mlme.assocreq_ies) {
459                 os_memcpy(wpa_s->mlme.assocreq_ies, ies,
460                           wpa_s->mlme.assocreq_ies_len);
461         }
462
463         ieee80211_sta_tx(wpa_s, buf, blen);
464         os_free(buf);
465 }
466
467
468 static void ieee80211_send_deauth(struct wpa_supplicant *wpa_s, u16 reason)
469 {
470         u8 *buf;
471         size_t len;
472         struct ieee80211_mgmt *mgmt;
473
474         buf = os_zalloc(sizeof(*mgmt));
475         if (buf == NULL) {
476                 wpa_printf(MSG_DEBUG, "MLME: failed to allocate buffer for "
477                            "deauth frame");
478                 return;
479         }
480
481         mgmt = (struct ieee80211_mgmt *) buf;
482         len = 24;
483         os_memcpy(mgmt->da, wpa_s->bssid, ETH_ALEN);
484         os_memcpy(mgmt->sa, wpa_s->own_addr, ETH_ALEN);
485         os_memcpy(mgmt->bssid, wpa_s->bssid, ETH_ALEN);
486         mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
487                                            WLAN_FC_STYPE_DEAUTH);
488         len += 2;
489         mgmt->u.deauth.reason_code = host_to_le16(reason);
490
491         ieee80211_sta_tx(wpa_s, buf, len);
492         os_free(buf);
493 }
494
495
496 static void ieee80211_send_disassoc(struct wpa_supplicant *wpa_s, u16 reason)
497 {
498         u8 *buf;
499         size_t len;
500         struct ieee80211_mgmt *mgmt;
501
502         buf = os_zalloc(sizeof(*mgmt));
503         if (buf == NULL) {
504                 wpa_printf(MSG_DEBUG, "MLME: failed to allocate buffer for "
505                            "disassoc frame");
506                 return;
507         }
508
509         mgmt = (struct ieee80211_mgmt *) buf;
510         len = 24;
511         os_memcpy(mgmt->da, wpa_s->bssid, ETH_ALEN);
512         os_memcpy(mgmt->sa, wpa_s->own_addr, ETH_ALEN);
513         os_memcpy(mgmt->bssid, wpa_s->bssid, ETH_ALEN);
514         mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
515                                            WLAN_FC_STYPE_DISASSOC);
516         len += 2;
517         mgmt->u.disassoc.reason_code = host_to_le16(reason);
518
519         ieee80211_sta_tx(wpa_s, buf, len);
520         os_free(buf);
521 }
522
523
524 static int ieee80211_privacy_mismatch(struct wpa_supplicant *wpa_s)
525 {
526         struct ieee80211_sta_bss *bss;
527         int res = 0;
528
529         if (wpa_s->mlme.mixed_cell ||
530             wpa_s->mlme.key_mgmt != KEY_MGMT_NONE)
531                 return 0;
532
533         bss = ieee80211_bss_get(wpa_s, wpa_s->bssid);
534         if (bss == NULL)
535                 return 0;
536
537         if (ieee80211_sta_wep_configured(wpa_s) !=
538             !!(bss->capability & WLAN_CAPABILITY_PRIVACY))
539                 res = 1;
540
541         return res;
542 }
543
544
545 static void ieee80211_associate(struct wpa_supplicant *wpa_s)
546 {
547         wpa_s->mlme.assoc_tries++;
548         if (wpa_s->mlme.assoc_tries > IEEE80211_ASSOC_MAX_TRIES) {
549                 wpa_printf(MSG_DEBUG, "MLME: association with AP " MACSTR
550                            " timed out", MAC2STR(wpa_s->bssid));
551                 return;
552         }
553
554         wpa_s->mlme.state = IEEE80211_ASSOCIATE;
555         wpa_printf(MSG_DEBUG, "MLME: associate with AP " MACSTR,
556                    MAC2STR(wpa_s->bssid));
557         if (ieee80211_privacy_mismatch(wpa_s)) {
558                 wpa_printf(MSG_DEBUG, "MLME: mismatch in privacy "
559                            "configuration and mixed-cell disabled - abort "
560                            "association");
561                 return;
562         }
563
564         ieee80211_send_assoc(wpa_s);
565
566         ieee80211_reschedule_timer(wpa_s, IEEE80211_ASSOC_TIMEOUT);
567 }
568
569
570 static void ieee80211_associated(struct wpa_supplicant *wpa_s)
571 {
572         int disassoc;
573
574         /* TODO: start monitoring current AP signal quality and number of
575          * missed beacons. Scan other channels every now and then and search
576          * for better APs. */
577         /* TODO: remove expired BSSes */
578
579         wpa_s->mlme.state = IEEE80211_ASSOCIATED;
580
581 #if 0 /* FIX */
582         sta = sta_info_get(local, wpa_s->bssid);
583         if (sta == NULL) {
584                 wpa_printf(MSG_DEBUG "MLME: No STA entry for own AP " MACSTR,
585                            MAC2STR(wpa_s->bssid));
586                 disassoc = 1;
587         } else {
588                 disassoc = 0;
589                 if (time_after(jiffies,
590                                sta->last_rx + IEEE80211_MONITORING_INTERVAL)) {
591                         if (wpa_s->mlme.probereq_poll) {
592                                 wpa_printf(MSG_DEBUG "MLME: No ProbeResp from "
593                                            "current AP " MACSTR " - assume "
594                                            "out of range",
595                                            MAC2STR(wpa_s->bssid));
596                                 disassoc = 1;
597                         } else {
598                                 ieee80211_send_probe_req(
599                                         wpa_s->bssid,
600                                         wpa_s->mlme.scan_ssid,
601                                         wpa_s->mlme.scan_ssid_len);
602                                 wpa_s->mlme.probereq_poll = 1;
603                         }
604                 } else {
605                         wpa_s->mlme.probereq_poll = 0;
606                         if (time_after(jiffies, wpa_s->mlme.last_probe +
607                                        IEEE80211_PROBE_INTERVAL)) {
608                                 wpa_s->mlme.last_probe = jiffies;
609                                 ieee80211_send_probe_req(wpa_s->bssid,
610                                                          wpa_s->mlme.ssid,
611                                                          wpa_s->mlme.ssid_len);
612                         }
613                 }
614                 sta_info_release(local, sta);
615         }
616 #else
617         disassoc = 0;
618 #endif
619         if (disassoc) {
620                 wpa_supplicant_event(wpa_s, EVENT_DISASSOC, NULL);
621                 ieee80211_reschedule_timer(wpa_s,
622                                            IEEE80211_MONITORING_INTERVAL +
623                                            30000);
624         } else {
625                 ieee80211_reschedule_timer(wpa_s,
626                                            IEEE80211_MONITORING_INTERVAL);
627         }
628 }
629
630
631 static void ieee80211_send_probe_req(struct wpa_supplicant *wpa_s,
632                                      const u8 *dst,
633                                      const u8 *ssid, size_t ssid_len)
634 {
635         u8 *buf;
636         size_t len;
637         struct ieee80211_mgmt *mgmt;
638         u8 *pos, *supp_rates;
639         u8 *esupp_rates = NULL;
640         int i;
641
642         buf = os_malloc(sizeof(*mgmt) + 200 + wpa_s->mlme.extra_probe_ie_len);
643         if (buf == NULL) {
644                 wpa_printf(MSG_DEBUG, "MLME: failed to allocate buffer for "
645                            "probe request");
646                 return;
647         }
648
649         mgmt = (struct ieee80211_mgmt *) buf;
650         len = 24;
651         os_memset(mgmt, 0, 24);
652         mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
653                                            WLAN_FC_STYPE_PROBE_REQ);
654         os_memcpy(mgmt->sa, wpa_s->own_addr, ETH_ALEN);
655         if (dst) {
656                 os_memcpy(mgmt->da, dst, ETH_ALEN);
657                 os_memcpy(mgmt->bssid, dst, ETH_ALEN);
658         } else {
659                 os_memset(mgmt->da, 0xff, ETH_ALEN);
660                 os_memset(mgmt->bssid, 0xff, ETH_ALEN);
661         }
662         pos = buf + len;
663         len += 2 + ssid_len;
664         *pos++ = WLAN_EID_SSID;
665         *pos++ = ssid_len;
666         os_memcpy(pos, ssid, ssid_len);
667
668         supp_rates = buf + len;
669         len += 2;
670         supp_rates[0] = WLAN_EID_SUPP_RATES;
671         supp_rates[1] = 0;
672         for (i = 0; i < wpa_s->mlme.num_curr_rates; i++) {
673                 if (esupp_rates) {
674                         pos = buf + len;
675                         len++;
676                         esupp_rates[1]++;
677                 } else if (supp_rates[1] == 8) {
678                         esupp_rates = pos;
679                         esupp_rates[0] = WLAN_EID_EXT_SUPP_RATES;
680                         esupp_rates[1] = 1;
681                         pos = &esupp_rates[2];
682                         len += 3;
683                 } else {
684                         pos = buf + len;
685                         len++;
686                         supp_rates[1]++;
687                 }
688                 *pos++ = wpa_s->mlme.curr_rates[i] / 5;
689         }
690
691         if (wpa_s->mlme.extra_probe_ie) {
692                 os_memcpy(pos, wpa_s->mlme.extra_probe_ie,
693                           wpa_s->mlme.extra_probe_ie_len);
694                 len += wpa_s->mlme.extra_probe_ie_len;
695         }
696
697         ieee80211_sta_tx(wpa_s, buf, len);
698         os_free(buf);
699 }
700
701
702 static int ieee80211_sta_wep_configured(struct wpa_supplicant *wpa_s)
703 {
704 #if 0 /* FIX */
705         if (sdata == NULL || sdata->default_key == NULL ||
706             sdata->default_key->alg != ALG_WEP)
707                 return 0;
708         return 1;
709 #else
710         return 0;
711 #endif
712 }
713
714
715 static void ieee80211_auth_completed(struct wpa_supplicant *wpa_s)
716 {
717         wpa_printf(MSG_DEBUG, "MLME: authenticated");
718         wpa_s->mlme.authenticated = 1;
719         ieee80211_associate(wpa_s);
720 }
721
722
723 static void ieee80211_auth_challenge(struct wpa_supplicant *wpa_s,
724                                      struct ieee80211_mgmt *mgmt,
725                                      size_t len,
726                                      struct ieee80211_rx_status *rx_status)
727 {
728         u8 *pos;
729         struct ieee802_11_elems elems;
730
731         wpa_printf(MSG_DEBUG, "MLME: replying to auth challenge");
732         pos = mgmt->u.auth.variable;
733         if (ieee802_11_parse_elems(pos, len - (pos - (u8 *) mgmt), &elems, 0)
734             == ParseFailed) {
735                 wpa_printf(MSG_DEBUG, "MLME: failed to parse Auth(challenge)");
736                 return;
737         }
738         if (elems.challenge == NULL) {
739                 wpa_printf(MSG_DEBUG, "MLME: no challenge IE in shared key "
740                            "auth frame");
741                 return;
742         }
743         ieee80211_send_auth(wpa_s, 3, elems.challenge - 2,
744                             elems.challenge_len + 2, 1);
745 }
746
747
748 static void ieee80211_rx_mgmt_auth(struct wpa_supplicant *wpa_s,
749                                    struct ieee80211_mgmt *mgmt,
750                                    size_t len,
751                                    struct ieee80211_rx_status *rx_status)
752 {
753         struct wpa_ssid *ssid = wpa_s->current_ssid;
754         u16 auth_alg, auth_transaction, status_code;
755         int adhoc;
756
757         adhoc = ssid && ssid->mode == WPAS_MODE_IBSS;
758
759         if (wpa_s->mlme.state != IEEE80211_AUTHENTICATE && !adhoc) {
760                 wpa_printf(MSG_DEBUG, "MLME: authentication frame received "
761                            "from " MACSTR ", but not in authenticate state - "
762                            "ignored", MAC2STR(mgmt->sa));
763                 return;
764         }
765
766         if (len < 24 + 6) {
767                 wpa_printf(MSG_DEBUG, "MLME: too short (%lu) authentication "
768                            "frame received from " MACSTR " - ignored",
769                            (unsigned long) len, MAC2STR(mgmt->sa));
770                 return;
771         }
772
773         if (!adhoc && os_memcmp(wpa_s->bssid, mgmt->sa, ETH_ALEN) != 0) {
774                 wpa_printf(MSG_DEBUG, "MLME: authentication frame received "
775                            "from unknown AP (SA=" MACSTR " BSSID=" MACSTR
776                            ") - ignored",
777                            MAC2STR(mgmt->sa), MAC2STR(mgmt->bssid));
778                 return;
779         }
780
781         if (adhoc && os_memcmp(wpa_s->bssid, mgmt->bssid, ETH_ALEN) != 0) {
782                 wpa_printf(MSG_DEBUG, "MLME: authentication frame received "
783                            "from unknown BSSID (SA=" MACSTR " BSSID=" MACSTR
784                            ") - ignored",
785                            MAC2STR(mgmt->sa), MAC2STR(mgmt->bssid));
786                 return;
787         }
788
789         auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
790         auth_transaction = le_to_host16(mgmt->u.auth.auth_transaction);
791         status_code = le_to_host16(mgmt->u.auth.status_code);
792
793         wpa_printf(MSG_DEBUG, "MLME: RX authentication from " MACSTR
794                    " (alg=%d transaction=%d status=%d)",
795                    MAC2STR(mgmt->sa), auth_alg, auth_transaction, status_code);
796
797         if (adhoc) {
798                 /* IEEE 802.11 standard does not require authentication in IBSS
799                  * networks and most implementations do not seem to use it.
800                  * However, try to reply to authentication attempts if someone
801                  * has actually implemented this.
802                  * TODO: Could implement shared key authentication. */
803                 if (auth_alg != WLAN_AUTH_OPEN || auth_transaction != 1) {
804                         wpa_printf(MSG_DEBUG, "MLME: unexpected IBSS "
805                                    "authentication frame (alg=%d "
806                                    "transaction=%d)",
807                                    auth_alg, auth_transaction);
808                         return;
809                 }
810                 ieee80211_send_auth(wpa_s, 2, NULL, 0, 0);
811         }
812
813         if (auth_alg != wpa_s->mlme.auth_alg ||
814             auth_transaction != wpa_s->mlme.auth_transaction) {
815                 wpa_printf(MSG_DEBUG, "MLME: unexpected authentication frame "
816                            "(alg=%d transaction=%d)",
817                            auth_alg, auth_transaction);
818                 return;
819         }
820
821         if (status_code != WLAN_STATUS_SUCCESS) {
822                 wpa_printf(MSG_DEBUG, "MLME: AP denied authentication "
823                            "(auth_alg=%d code=%d)", wpa_s->mlme.auth_alg,
824                            status_code);
825                 if (status_code == WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG) {
826                         const int num_algs = 3;
827                         u8 algs[num_algs];
828                         int i, pos;
829                         algs[0] = algs[1] = algs[2] = 0xff;
830                         if (wpa_s->mlme.auth_algs & WPA_AUTH_ALG_OPEN)
831                                 algs[0] = WLAN_AUTH_OPEN;
832                         if (wpa_s->mlme.auth_algs & WPA_AUTH_ALG_SHARED)
833                                 algs[1] = WLAN_AUTH_SHARED_KEY;
834                         if (wpa_s->mlme.auth_algs & WPA_AUTH_ALG_LEAP)
835                                 algs[2] = WLAN_AUTH_LEAP;
836                         if (wpa_s->mlme.auth_alg == WLAN_AUTH_OPEN)
837                                 pos = 0;
838                         else if (wpa_s->mlme.auth_alg == WLAN_AUTH_SHARED_KEY)
839                                 pos = 1;
840                         else
841                                 pos = 2;
842                         for (i = 0; i < num_algs; i++) {
843                                 pos++;
844                                 if (pos >= num_algs)
845                                         pos = 0;
846                                 if (algs[pos] == wpa_s->mlme.auth_alg ||
847                                     algs[pos] == 0xff)
848                                         continue;
849                                 if (algs[pos] == WLAN_AUTH_SHARED_KEY &&
850                                     !ieee80211_sta_wep_configured(wpa_s))
851                                         continue;
852                                 wpa_s->mlme.auth_alg = algs[pos];
853                                 wpa_printf(MSG_DEBUG, "MLME: set auth_alg=%d "
854                                            "for next try",
855                                            wpa_s->mlme.auth_alg);
856                                 break;
857                         }
858                 }
859                 return;
860         }
861
862         switch (wpa_s->mlme.auth_alg) {
863         case WLAN_AUTH_OPEN:
864         case WLAN_AUTH_LEAP:
865                 ieee80211_auth_completed(wpa_s);
866                 break;
867         case WLAN_AUTH_SHARED_KEY:
868                 if (wpa_s->mlme.auth_transaction == 4)
869                         ieee80211_auth_completed(wpa_s);
870                 else
871                         ieee80211_auth_challenge(wpa_s, mgmt, len,
872                                                  rx_status);
873                 break;
874 #ifdef CONFIG_IEEE80211R
875         case WLAN_AUTH_FT:
876         {
877                 union wpa_event_data data;
878                 struct wpabuf *ric = NULL;
879                 os_memset(&data, 0, sizeof(data));
880                 data.ft_ies.ies = mgmt->u.auth.variable;
881                 data.ft_ies.ies_len = len -
882                         (mgmt->u.auth.variable - (u8 *) mgmt);
883                 os_memcpy(data.ft_ies.target_ap, wpa_s->bssid, ETH_ALEN);
884                 if (os_strcmp(wpa_s->driver->name, "test") == 0 &&
885                     wpa_s->mlme.wmm_enabled) {
886                         ric = wpabuf_alloc(200);
887                         if (ric) {
888                                 /* Build simple RIC-Request: RDIE | TSPEC */
889
890                                 /* RIC Data (RDIE) */
891                                 wpabuf_put_u8(ric, WLAN_EID_RIC_DATA);
892                                 wpabuf_put_u8(ric, 4);
893                                 wpabuf_put_u8(ric, 0); /* RDIE Identifier */
894                                 wpabuf_put_u8(ric, 1); /* Resource Descriptor
895                                                         * Count */
896                                 wpabuf_put_le16(ric, 0); /* Status Code */
897
898                                 /* WMM TSPEC */
899                                 ieee80211_build_tspec(ric);
900
901                                 data.ft_ies.ric_ies = wpabuf_head(ric);
902                                 data.ft_ies.ric_ies_len = wpabuf_len(ric);
903                         }
904                 }
905
906                 wpa_supplicant_event(wpa_s, EVENT_FT_RESPONSE, &data);
907                 wpabuf_free(ric);
908                 ieee80211_auth_completed(wpa_s);
909                 break;
910         }
911 #endif /* CONFIG_IEEE80211R */
912         }
913 }
914
915
916 static void ieee80211_rx_mgmt_deauth(struct wpa_supplicant *wpa_s,
917                                      struct ieee80211_mgmt *mgmt,
918                                      size_t len,
919                                      struct ieee80211_rx_status *rx_status)
920 {
921         u16 reason_code;
922
923         if (len < 24 + 2) {
924                 wpa_printf(MSG_DEBUG, "MLME: too short (%lu) deauthentication "
925                            "frame received from " MACSTR " - ignored",
926                            (unsigned long) len, MAC2STR(mgmt->sa));
927                 return;
928         }
929
930         if (os_memcmp(wpa_s->bssid, mgmt->sa, ETH_ALEN) != 0) {
931                 wpa_printf(MSG_DEBUG, "MLME: deauthentication frame received "
932                            "from unknown AP (SA=" MACSTR " BSSID=" MACSTR
933                            ") - ignored",
934                            MAC2STR(mgmt->sa), MAC2STR(mgmt->bssid));
935                 return;
936         }
937
938         reason_code = le_to_host16(mgmt->u.deauth.reason_code);
939
940         wpa_printf(MSG_DEBUG, "MLME: RX deauthentication from " MACSTR
941                    " (reason=%d)", MAC2STR(mgmt->sa), reason_code);
942
943         if (wpa_s->mlme.authenticated)
944                 wpa_printf(MSG_DEBUG, "MLME: deauthenticated");
945
946         if (wpa_s->mlme.state == IEEE80211_AUTHENTICATE ||
947             wpa_s->mlme.state == IEEE80211_ASSOCIATE ||
948             wpa_s->mlme.state == IEEE80211_ASSOCIATED) {
949                 wpa_s->mlme.state = IEEE80211_AUTHENTICATE;
950                 ieee80211_reschedule_timer(wpa_s,
951                                            IEEE80211_RETRY_AUTH_INTERVAL);
952         }
953
954         ieee80211_set_associated(wpa_s, 0);
955         wpa_s->mlme.authenticated = 0;
956 }
957
958
959 static void ieee80211_rx_mgmt_disassoc(struct wpa_supplicant *wpa_s,
960                                        struct ieee80211_mgmt *mgmt,
961                                        size_t len,
962                                        struct ieee80211_rx_status *rx_status)
963 {
964         u16 reason_code;
965
966         if (len < 24 + 2) {
967                 wpa_printf(MSG_DEBUG, "MLME: too short (%lu) disassociation "
968                            "frame received from " MACSTR " - ignored",
969                            (unsigned long) len, MAC2STR(mgmt->sa));
970                 return;
971         }
972
973         if (os_memcmp(wpa_s->bssid, mgmt->sa, ETH_ALEN) != 0) {
974                 wpa_printf(MSG_DEBUG, "MLME: disassociation frame received "
975                            "from unknown AP (SA=" MACSTR " BSSID=" MACSTR
976                            ") - ignored",
977                            MAC2STR(mgmt->sa), MAC2STR(mgmt->bssid));
978                 return;
979         }
980
981         reason_code = le_to_host16(mgmt->u.disassoc.reason_code);
982
983         wpa_printf(MSG_DEBUG, "MLME: RX disassociation from " MACSTR
984                    " (reason=%d)", MAC2STR(mgmt->sa), reason_code);
985
986         if (wpa_s->mlme.associated)
987                 wpa_printf(MSG_DEBUG, "MLME: disassociated");
988
989         if (wpa_s->mlme.state == IEEE80211_ASSOCIATED) {
990                 wpa_s->mlme.state = IEEE80211_ASSOCIATE;
991                 ieee80211_reschedule_timer(wpa_s,
992                                            IEEE80211_RETRY_AUTH_INTERVAL);
993         }
994
995         ieee80211_set_associated(wpa_s, 0);
996 }
997
998
999 static void ieee80211_build_tspec(struct wpabuf *buf)
1000 {
1001         struct wmm_tspec_element *tspec;
1002         int tid, up;
1003
1004         tspec = wpabuf_put(buf, sizeof(*tspec));
1005         tspec->eid = WLAN_EID_VENDOR_SPECIFIC;
1006         tspec->length = sizeof(*tspec) - 2;
1007         tspec->oui[0] = 0x00;
1008         tspec->oui[1] = 0x50;
1009         tspec->oui[2] = 0xf2;
1010         tspec->oui_type = 2;
1011         tspec->oui_subtype = 2;
1012         tspec->version = 1;
1013
1014         tid = 1;
1015         up = 6; /* Voice */
1016         tspec->ts_info[0] = (tid << 1) |
1017                 (WMM_TSPEC_DIRECTION_BI_DIRECTIONAL << 5) |
1018                 BIT(7);
1019         tspec->ts_info[1] = up << 3;
1020         tspec->nominal_msdu_size = host_to_le16(1530);
1021         tspec->mean_data_rate = host_to_le32(128000); /* bits per second */
1022         tspec->minimum_phy_rate = host_to_le32(6000000);
1023         tspec->surplus_bandwidth_allowance = host_to_le16(0x3000); /* 150% */
1024 }
1025
1026
1027 static void ieee80211_tx_addts(struct wpa_supplicant *wpa_s)
1028 {
1029         struct wpabuf *buf;
1030         struct ieee80211_mgmt *mgmt;
1031         size_t alen;
1032
1033         wpa_printf(MSG_DEBUG, "MLME: Send ADDTS Request for Voice TSPEC");
1034         mgmt = NULL;
1035         alen = mgmt->u.action.u.wmm_action.variable - (u8 *) mgmt;
1036
1037         buf = wpabuf_alloc(alen + sizeof(struct wmm_tspec_element));
1038         if (buf == NULL)
1039                 return;
1040
1041         mgmt = wpabuf_put(buf, alen);
1042         os_memcpy(mgmt->da, wpa_s->bssid, ETH_ALEN);
1043         os_memcpy(mgmt->sa, wpa_s->own_addr, ETH_ALEN);
1044         os_memcpy(mgmt->bssid, wpa_s->bssid, ETH_ALEN);
1045         mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
1046                                            WLAN_FC_STYPE_ACTION);
1047         mgmt->u.action.category = WLAN_ACTION_WMM;
1048         mgmt->u.action.u.wmm_action.action_code = WMM_ACTION_CODE_ADDTS_REQ;
1049         mgmt->u.action.u.wmm_action.dialog_token = 1;
1050         mgmt->u.action.u.wmm_action.status_code = 0;
1051
1052         ieee80211_build_tspec(buf);
1053
1054         ieee80211_sta_tx(wpa_s, wpabuf_head(buf), wpabuf_len(buf));
1055         wpabuf_free(buf);
1056 }
1057
1058
1059 static void ieee80211_rx_mgmt_assoc_resp(struct wpa_supplicant *wpa_s,
1060                                          struct ieee80211_mgmt *mgmt,
1061                                          size_t len,
1062                                          struct ieee80211_rx_status *rx_status,
1063                                          int reassoc)
1064 {
1065         u8 rates[32];
1066         size_t rates_len;
1067         u16 capab_info, status_code, aid;
1068         struct ieee802_11_elems elems;
1069         u8 *pos;
1070
1071         /* AssocResp and ReassocResp have identical structure, so process both
1072          * of them in this function. */
1073
1074         if (wpa_s->mlme.state != IEEE80211_ASSOCIATE) {
1075                 wpa_printf(MSG_DEBUG, "MLME: association frame received from "
1076                            MACSTR ", but not in associate state - ignored",
1077                            MAC2STR(mgmt->sa));
1078                 return;
1079         }
1080
1081         if (len < 24 + 6) {
1082                 wpa_printf(MSG_DEBUG, "MLME: too short (%lu) association "
1083                            "frame received from " MACSTR " - ignored",
1084                            (unsigned long) len, MAC2STR(mgmt->sa));
1085                 return;
1086         }
1087
1088         if (os_memcmp(wpa_s->bssid, mgmt->sa, ETH_ALEN) != 0) {
1089                 wpa_printf(MSG_DEBUG, "MLME: association frame received from "
1090                            "unknown AP (SA=" MACSTR " BSSID=" MACSTR ") - "
1091                            "ignored", MAC2STR(mgmt->sa), MAC2STR(mgmt->bssid));
1092                 return;
1093         }
1094
1095         capab_info = le_to_host16(mgmt->u.assoc_resp.capab_info);
1096         status_code = le_to_host16(mgmt->u.assoc_resp.status_code);
1097         aid = le_to_host16(mgmt->u.assoc_resp.aid);
1098         if ((aid & (BIT(15) | BIT(14))) != (BIT(15) | BIT(14)))
1099                 wpa_printf(MSG_DEBUG, "MLME: invalid aid value %d; bits 15:14 "
1100                            "not set", aid);
1101         aid &= ~(BIT(15) | BIT(14));
1102
1103         wpa_printf(MSG_DEBUG, "MLME: RX %sssocResp from " MACSTR
1104                    " (capab=0x%x status=%d aid=%d)",
1105                    reassoc ? "Rea" : "A", MAC2STR(mgmt->sa),
1106                    capab_info, status_code, aid);
1107
1108         pos = mgmt->u.assoc_resp.variable;
1109         if (ieee802_11_parse_elems(pos, len - (pos - (u8 *) mgmt), &elems, 0)
1110             == ParseFailed) {
1111                 wpa_printf(MSG_DEBUG, "MLME: failed to parse AssocResp");
1112                 return;
1113         }
1114
1115         if (status_code != WLAN_STATUS_SUCCESS) {
1116                 wpa_printf(MSG_DEBUG, "MLME: AP denied association (code=%d)",
1117                            status_code);
1118 #ifdef CONFIG_IEEE80211W
1119                 if (status_code == WLAN_STATUS_ASSOC_REJECTED_TEMPORARILY &&
1120                     elems.timeout_int && elems.timeout_int_len == 5 &&
1121                     elems.timeout_int[0] == WLAN_TIMEOUT_ASSOC_COMEBACK) {
1122                         u32 tu, ms;
1123                         tu = WPA_GET_LE32(elems.timeout_int + 1);
1124                         ms = tu * 1024 / 1000;
1125                         wpa_printf(MSG_DEBUG, "MLME: AP rejected association "
1126                                    "temporarily; comeback duration %u TU "
1127                                    "(%u ms)", tu, ms);
1128                         if (ms > IEEE80211_ASSOC_TIMEOUT) {
1129                                 wpa_printf(MSG_DEBUG, "MLME: Update timer "
1130                                            "based on comeback duration");
1131                                 ieee80211_reschedule_timer(wpa_s, ms);
1132                         }
1133                 }
1134 #endif /* CONFIG_IEEE80211W */
1135                 return;
1136         }
1137
1138         if (elems.supp_rates == NULL) {
1139                 wpa_printf(MSG_DEBUG, "MLME: no SuppRates element in "
1140                            "AssocResp");
1141                 return;
1142         }
1143
1144         if (wpa_s->mlme.auth_alg == WLAN_AUTH_FT) {
1145                 if (!reassoc) {
1146                         wpa_printf(MSG_DEBUG, "MLME: AP tried to use "
1147                                    "association, not reassociation, response "
1148                                    "with FT");
1149                         return;
1150                 }
1151                 if (wpa_ft_validate_reassoc_resp(
1152                             wpa_s->wpa, pos, len - (pos - (u8 *) mgmt),
1153                             mgmt->sa) < 0) {
1154                         wpa_printf(MSG_DEBUG, "MLME: FT validation of Reassoc"
1155                                    "Resp failed");
1156                         return;
1157                 }
1158         } else if (wpa_sm_set_ft_params(wpa_s->wpa, pos,
1159                                         len - (pos - (u8 *) mgmt)) < 0)
1160                 return;
1161
1162         wpa_printf(MSG_DEBUG, "MLME: associated");
1163         wpa_s->mlme.aid = aid;
1164         wpa_s->mlme.ap_capab = capab_info;
1165
1166         os_free(wpa_s->mlme.assocresp_ies);
1167         wpa_s->mlme.assocresp_ies_len = len - (pos - (u8 *) mgmt);
1168         wpa_s->mlme.assocresp_ies = os_malloc(wpa_s->mlme.assocresp_ies_len);
1169         if (wpa_s->mlme.assocresp_ies) {
1170                 os_memcpy(wpa_s->mlme.assocresp_ies, pos,
1171                           wpa_s->mlme.assocresp_ies_len);
1172         }
1173
1174         ieee80211_set_associated(wpa_s, 1);
1175
1176         rates_len = elems.supp_rates_len;
1177         if (rates_len > sizeof(rates))
1178                 rates_len = sizeof(rates);
1179         os_memcpy(rates, elems.supp_rates, rates_len);
1180         if (elems.ext_supp_rates) {
1181                 size_t _len = elems.ext_supp_rates_len;
1182                 if (_len > sizeof(rates) - rates_len)
1183                         _len = sizeof(rates) - rates_len;
1184                 os_memcpy(rates + rates_len, elems.ext_supp_rates, _len);
1185                 rates_len += _len;
1186         }
1187
1188         if (wpa_drv_set_bssid(wpa_s, wpa_s->bssid) < 0) {
1189                 wpa_printf(MSG_DEBUG, "MLME: failed to set BSSID for the "
1190                            "netstack");
1191         }
1192         if (wpa_drv_set_ssid(wpa_s, wpa_s->mlme.ssid, wpa_s->mlme.ssid_len) <
1193             0) {
1194                 wpa_printf(MSG_DEBUG, "MLME: failed to set SSID for the "
1195                            "netstack");
1196         }
1197
1198         /* Remove STA entry before adding a new one just in case to avoid
1199          * problems with existing configuration (e.g., keys). */
1200         wpa_drv_mlme_remove_sta(wpa_s, wpa_s->bssid);
1201         if (wpa_drv_mlme_add_sta(wpa_s, wpa_s->bssid, rates, rates_len) < 0) {
1202                 wpa_printf(MSG_DEBUG, "MLME: failed to add STA entry to the "
1203                            "netstack");
1204         }
1205
1206         if (elems.wmm && wpa_s->mlme.wmm_enabled)
1207                 ieee80211_sta_wmm_params(wpa_s, elems.wmm, elems.wmm_len);
1208
1209         ieee80211_associated(wpa_s);
1210
1211         if (wpa_s->mlme.auth_alg != WLAN_AUTH_FT &&
1212             os_strcmp(wpa_s->driver->name, "test") == 0 &&
1213             elems.wmm && wpa_s->mlme.wmm_enabled) {
1214                 /* Test WMM-AC - send ADDTS for WMM TSPEC */
1215                 ieee80211_tx_addts(wpa_s);
1216         }
1217 }
1218
1219
1220 /* Caller must hold local->sta_bss_lock */
1221 static void __ieee80211_bss_hash_add(struct wpa_supplicant *wpa_s,
1222                                      struct ieee80211_sta_bss *bss)
1223 {
1224         bss->hnext = wpa_s->mlme.sta_bss_hash[STA_HASH(bss->bssid)];
1225         wpa_s->mlme.sta_bss_hash[STA_HASH(bss->bssid)] = bss;
1226 }
1227
1228
1229 /* Caller must hold local->sta_bss_lock */
1230 static void __ieee80211_bss_hash_del(struct wpa_supplicant *wpa_s,
1231                                      struct ieee80211_sta_bss *bss)
1232 {
1233         struct ieee80211_sta_bss *b, *prev = NULL;
1234         b = wpa_s->mlme.sta_bss_hash[STA_HASH(bss->bssid)];
1235         while (b) {
1236                 if (b == bss) {
1237                         if (prev == NULL) {
1238                                 wpa_s->mlme.sta_bss_hash[STA_HASH(bss->bssid)]
1239                                         = bss->hnext;
1240                         } else {
1241                                 prev->hnext = bss->hnext;
1242                         }
1243                         break;
1244                 }
1245                 prev = b;
1246                 b = b->hnext;
1247         }
1248 }
1249
1250
1251 static struct ieee80211_sta_bss *
1252 ieee80211_bss_add(struct wpa_supplicant *wpa_s, const u8 *bssid)
1253 {
1254         struct ieee80211_sta_bss *bss;
1255
1256         bss = os_zalloc(sizeof(*bss));
1257         if (bss == NULL)
1258                 return NULL;
1259         os_memcpy(bss->bssid, bssid, ETH_ALEN);
1260
1261         /* TODO: order by RSSI? */
1262         bss->next = wpa_s->mlme.sta_bss_list;
1263         wpa_s->mlme.sta_bss_list = bss;
1264         __ieee80211_bss_hash_add(wpa_s, bss);
1265         return bss;
1266 }
1267
1268
1269 static struct ieee80211_sta_bss *
1270 ieee80211_bss_get(struct wpa_supplicant *wpa_s, const u8 *bssid)
1271 {
1272         struct ieee80211_sta_bss *bss;
1273
1274         bss = wpa_s->mlme.sta_bss_hash[STA_HASH(bssid)];
1275         while (bss) {
1276                 if (os_memcmp(bss->bssid, bssid, ETH_ALEN) == 0)
1277                         break;
1278                 bss = bss->hnext;
1279         }
1280         return bss;
1281 }
1282
1283
1284 static void ieee80211_bss_free(struct wpa_supplicant *wpa_s,
1285                                struct ieee80211_sta_bss *bss)
1286 {
1287         __ieee80211_bss_hash_del(wpa_s, bss);
1288         os_free(bss->ie);
1289         os_free(bss->wpa_ie);
1290         os_free(bss->rsn_ie);
1291         os_free(bss->wmm_ie);
1292         os_free(bss->mdie);
1293         os_free(bss);
1294 }
1295
1296
1297 static void ieee80211_bss_list_deinit(struct wpa_supplicant *wpa_s)
1298 {
1299         struct ieee80211_sta_bss *bss, *prev;
1300
1301         bss = wpa_s->mlme.sta_bss_list;
1302         wpa_s->mlme.sta_bss_list = NULL;
1303         while (bss) {
1304                 prev = bss;
1305                 bss = bss->next;
1306                 ieee80211_bss_free(wpa_s, prev);
1307         }
1308 }
1309
1310
1311 static void ieee80211_bss_info(struct wpa_supplicant *wpa_s,
1312                                struct ieee80211_mgmt *mgmt,
1313                                size_t len,
1314                                struct ieee80211_rx_status *rx_status,
1315                                int beacon)
1316 {
1317         struct ieee802_11_elems elems;
1318         size_t baselen;
1319         int channel, invalid = 0, clen;
1320         struct ieee80211_sta_bss *bss;
1321         u64 timestamp;
1322         u8 *pos, *ie_pos;
1323         size_t ie_len;
1324
1325         if (!beacon && os_memcmp(mgmt->da, wpa_s->own_addr, ETH_ALEN))
1326                 return; /* ignore ProbeResp to foreign address */
1327
1328 #if 0
1329         wpa_printf(MSG_MSGDUMP, "MLME: RX %s from " MACSTR " to " MACSTR,
1330                    beacon ? "Beacon" : "Probe Response",
1331                    MAC2STR(mgmt->sa), MAC2STR(mgmt->da));
1332 #endif
1333
1334         baselen = (u8 *) mgmt->u.beacon.variable - (u8 *) mgmt;
1335         if (baselen > len)
1336                 return;
1337
1338         pos = mgmt->u.beacon.timestamp;
1339         timestamp = WPA_GET_LE64(pos);
1340
1341 #if 0 /* FIX */
1342         if (local->conf.mode == IW_MODE_ADHOC && beacon &&
1343             os_memcmp(mgmt->bssid, local->bssid, ETH_ALEN) == 0) {
1344 #ifdef IEEE80211_IBSS_DEBUG
1345                 static unsigned long last_tsf_debug = 0;
1346                 u64 tsf;
1347                 if (local->hw->get_tsf)
1348                         tsf = local->hw->get_tsf(local->mdev);
1349                 else
1350                         tsf = -1LLU;
1351                 if (time_after(jiffies, last_tsf_debug + 5 * HZ)) {
1352                         wpa_printf(MSG_DEBUG, "RX beacon SA=" MACSTR " BSSID="
1353                                    MACSTR " TSF=0x%llx BCN=0x%llx diff=%lld "
1354                                    "@%ld",
1355                                    MAC2STR(mgmt->sa), MAC2STR(mgmt->bssid),
1356                                    tsf, timestamp, tsf - timestamp, jiffies);
1357                         last_tsf_debug = jiffies;
1358                 }
1359 #endif /* IEEE80211_IBSS_DEBUG */
1360         }
1361 #endif
1362
1363         ie_pos = mgmt->u.beacon.variable;
1364         ie_len = len - baselen;
1365         if (ieee802_11_parse_elems(ie_pos, ie_len, &elems, 0) == ParseFailed)
1366                 invalid = 1;
1367
1368 #if 0 /* FIX */
1369         if (local->conf.mode == IW_MODE_ADHOC && elems.supp_rates &&
1370             os_memcmp(mgmt->bssid, local->bssid, ETH_ALEN) == 0 &&
1371             (sta = sta_info_get(local, mgmt->sa))) {
1372                 struct ieee80211_rate *rates;
1373                 size_t num_rates;
1374                 u32 supp_rates, prev_rates;
1375                 int i, j, oper_mode;
1376
1377                 rates = local->curr_rates;
1378                 num_rates = local->num_curr_rates;
1379                 oper_mode = wpa_s->mlme.sta_scanning ?
1380                         local->scan_oper_phymode : local->conf.phymode;
1381                 for (i = 0; i < local->hw->num_modes; i++) {
1382                         struct ieee80211_hw_modes *mode = &local->hw->modes[i];
1383                         if (oper_mode == mode->mode) {
1384                                 rates = mode->rates;
1385                                 num_rates = mode->num_rates;
1386                                 break;
1387                         }
1388                 }
1389
1390                 supp_rates = 0;
1391                 for (i = 0; i < elems.supp_rates_len +
1392                              elems.ext_supp_rates_len; i++) {
1393                         u8 rate = 0;
1394                         int own_rate;
1395                         if (i < elems.supp_rates_len)
1396                                 rate = elems.supp_rates[i];
1397                         else if (elems.ext_supp_rates)
1398                                 rate = elems.ext_supp_rates
1399                                         [i - elems.supp_rates_len];
1400                         own_rate = 5 * (rate & 0x7f);
1401                         if (oper_mode == MODE_ATHEROS_TURBO)
1402                                 own_rate *= 2;
1403                         for (j = 0; j < num_rates; j++)
1404                                 if (rates[j].rate == own_rate)
1405                                         supp_rates |= BIT(j);
1406                 }
1407
1408                 prev_rates = sta->supp_rates;
1409                 sta->supp_rates &= supp_rates;
1410                 if (sta->supp_rates == 0) {
1411                         /* No matching rates - this should not really happen.
1412                          * Make sure that at least one rate is marked
1413                          * supported to avoid issues with TX rate ctrl. */
1414                         sta->supp_rates = wpa_s->mlme.supp_rates_bits;
1415                 }
1416                 if (sta->supp_rates != prev_rates) {
1417                         wpa_printf(MSG_DEBUG, "MLME: updated supp_rates set "
1418                                    "for " MACSTR " based on beacon info "
1419                                    "(0x%x & 0x%x -> 0x%x)",
1420                                    MAC2STR(sta->addr), prev_rates,
1421                                    supp_rates, sta->supp_rates);
1422                 }
1423                 sta_info_release(local, sta);
1424         }
1425 #endif
1426
1427         if (elems.ssid == NULL)
1428                 return;
1429
1430         if (elems.ds_params && elems.ds_params_len == 1)
1431                 channel = elems.ds_params[0];
1432         else
1433                 channel = rx_status->channel;
1434
1435         bss = ieee80211_bss_get(wpa_s, mgmt->bssid);
1436         if (bss == NULL) {
1437                 bss = ieee80211_bss_add(wpa_s, mgmt->bssid);
1438                 if (bss == NULL)
1439                         return;
1440         } else {
1441 #if 0
1442                 /* TODO: order by RSSI? */
1443                 spin_lock_bh(&local->sta_bss_lock);
1444                 list_move_tail(&bss->list, &local->sta_bss_list);
1445                 spin_unlock_bh(&local->sta_bss_lock);
1446 #endif
1447         }
1448
1449         if (bss->probe_resp && beacon) {
1450                 /* Do not allow beacon to override data from Probe Response. */
1451                 return;
1452         }
1453
1454         bss->beacon_int = le_to_host16(mgmt->u.beacon.beacon_int);
1455         bss->capability = le_to_host16(mgmt->u.beacon.capab_info);
1456
1457         if (bss->ie == NULL || bss->ie_len < ie_len) {
1458                 os_free(bss->ie);
1459                 bss->ie = os_malloc(ie_len);
1460         }
1461         if (bss->ie) {
1462                 os_memcpy(bss->ie, ie_pos, ie_len);
1463                 bss->ie_len = ie_len;
1464         }
1465
1466         if (elems.ssid && elems.ssid_len <= MAX_SSID_LEN) {
1467                 os_memcpy(bss->ssid, elems.ssid, elems.ssid_len);
1468                 bss->ssid_len = elems.ssid_len;
1469         }
1470
1471         bss->supp_rates_len = 0;
1472         if (elems.supp_rates) {
1473                 clen = IEEE80211_MAX_SUPP_RATES - bss->supp_rates_len;
1474                 if (clen > elems.supp_rates_len)
1475                         clen = elems.supp_rates_len;
1476                 os_memcpy(&bss->supp_rates[bss->supp_rates_len],
1477                           elems.supp_rates, clen);
1478                 bss->supp_rates_len += clen;
1479         }
1480         if (elems.ext_supp_rates) {
1481                 clen = IEEE80211_MAX_SUPP_RATES - bss->supp_rates_len;
1482                 if (clen > elems.ext_supp_rates_len)
1483                         clen = elems.ext_supp_rates_len;
1484                 os_memcpy(&bss->supp_rates[bss->supp_rates_len],
1485                           elems.ext_supp_rates, clen);
1486                 bss->supp_rates_len += clen;
1487         }
1488
1489         if (elems.wpa_ie &&
1490             (bss->wpa_ie == NULL || bss->wpa_ie_len != elems.wpa_ie_len ||
1491              os_memcmp(bss->wpa_ie, elems.wpa_ie, elems.wpa_ie_len))) {
1492                 os_free(bss->wpa_ie);
1493                 bss->wpa_ie = os_malloc(elems.wpa_ie_len + 2);
1494                 if (bss->wpa_ie) {
1495                         os_memcpy(bss->wpa_ie, elems.wpa_ie - 2,
1496                                   elems.wpa_ie_len + 2);
1497                         bss->wpa_ie_len = elems.wpa_ie_len + 2;
1498                 } else
1499                         bss->wpa_ie_len = 0;
1500         } else if (!elems.wpa_ie && bss->wpa_ie) {
1501                 os_free(bss->wpa_ie);
1502                 bss->wpa_ie = NULL;
1503                 bss->wpa_ie_len = 0;
1504         }
1505
1506         if (elems.rsn_ie &&
1507             (bss->rsn_ie == NULL || bss->rsn_ie_len != elems.rsn_ie_len ||
1508              os_memcmp(bss->rsn_ie, elems.rsn_ie, elems.rsn_ie_len))) {
1509                 os_free(bss->rsn_ie);
1510                 bss->rsn_ie = os_malloc(elems.rsn_ie_len + 2);
1511                 if (bss->rsn_ie) {
1512                         os_memcpy(bss->rsn_ie, elems.rsn_ie - 2,
1513                                   elems.rsn_ie_len + 2);
1514                         bss->rsn_ie_len = elems.rsn_ie_len + 2;
1515                 } else
1516                         bss->rsn_ie_len = 0;
1517         } else if (!elems.rsn_ie && bss->rsn_ie) {
1518                 os_free(bss->rsn_ie);
1519                 bss->rsn_ie = NULL;
1520                 bss->rsn_ie_len = 0;
1521         }
1522
1523         if (elems.wmm &&
1524             (bss->wmm_ie == NULL || bss->wmm_ie_len != elems.wmm_len ||
1525              os_memcmp(bss->wmm_ie, elems.wmm, elems.wmm_len))) {
1526                 os_free(bss->wmm_ie);
1527                 bss->wmm_ie = os_malloc(elems.wmm_len + 2);
1528                 if (bss->wmm_ie) {
1529                         os_memcpy(bss->wmm_ie, elems.wmm - 2,
1530                                   elems.wmm_len + 2);
1531                         bss->wmm_ie_len = elems.wmm_len + 2;
1532                 } else
1533                         bss->wmm_ie_len = 0;
1534         } else if (!elems.wmm && bss->wmm_ie) {
1535                 os_free(bss->wmm_ie);
1536                 bss->wmm_ie = NULL;
1537                 bss->wmm_ie_len = 0;
1538         }
1539
1540 #ifdef CONFIG_IEEE80211R
1541         if (elems.mdie &&
1542             (bss->mdie == NULL || bss->mdie_len != elems.mdie_len ||
1543              os_memcmp(bss->mdie, elems.mdie, elems.mdie_len))) {
1544                 os_free(bss->mdie);
1545                 bss->mdie = os_malloc(elems.mdie_len + 2);
1546                 if (bss->mdie) {
1547                         os_memcpy(bss->mdie, elems.mdie - 2,
1548                                   elems.mdie_len + 2);
1549                         bss->mdie_len = elems.mdie_len + 2;
1550                 } else
1551                         bss->mdie_len = 0;
1552         } else if (!elems.mdie && bss->mdie) {
1553                 os_free(bss->mdie);
1554                 bss->mdie = NULL;
1555                 bss->mdie_len = 0;
1556         }
1557 #endif /* CONFIG_IEEE80211R */
1558
1559         bss->hw_mode = wpa_s->mlme.phymode;
1560         bss->channel = channel;
1561         bss->freq = wpa_s->mlme.freq;
1562         if (channel != wpa_s->mlme.channel &&
1563             (wpa_s->mlme.phymode == HOSTAPD_MODE_IEEE80211G ||
1564              wpa_s->mlme.phymode == HOSTAPD_MODE_IEEE80211B) &&
1565             channel >= 1 && channel <= 14) {
1566                 static const int freq_list[] = {
1567                         2412, 2417, 2422, 2427, 2432, 2437, 2442,
1568                         2447, 2452, 2457, 2462, 2467, 2472, 2484
1569                 };
1570                 /* IEEE 802.11g/b mode can receive packets from neighboring
1571                  * channels, so map the channel into frequency. */
1572                 bss->freq = freq_list[channel - 1];
1573         }
1574         bss->timestamp = timestamp;
1575         os_get_time(&bss->last_update);
1576         bss->rssi = rx_status->ssi;
1577         if (!beacon)
1578                 bss->probe_resp++;
1579 }
1580
1581
1582 static void ieee80211_rx_mgmt_probe_resp(struct wpa_supplicant *wpa_s,
1583                                          struct ieee80211_mgmt *mgmt,
1584                                          size_t len,
1585                                          struct ieee80211_rx_status *rx_status)
1586 {
1587         ieee80211_bss_info(wpa_s, mgmt, len, rx_status, 0);
1588 }
1589
1590
1591 static void ieee80211_rx_mgmt_beacon(struct wpa_supplicant *wpa_s,
1592                                      struct ieee80211_mgmt *mgmt,
1593                                      size_t len,
1594                                      struct ieee80211_rx_status *rx_status)
1595 {
1596         int use_protection;
1597         size_t baselen;
1598         struct ieee802_11_elems elems;
1599
1600         ieee80211_bss_info(wpa_s, mgmt, len, rx_status, 1);
1601
1602         if (!wpa_s->mlme.associated ||
1603             os_memcmp(wpa_s->bssid, mgmt->bssid, ETH_ALEN) != 0)
1604                 return;
1605
1606         /* Process beacon from the current BSS */
1607         baselen = (u8 *) mgmt->u.beacon.variable - (u8 *) mgmt;
1608         if (baselen > len)
1609                 return;
1610
1611         if (ieee802_11_parse_elems(mgmt->u.beacon.variable, len - baselen,
1612                                    &elems, 0) == ParseFailed)
1613                 return;
1614
1615         use_protection = 0;
1616         if (elems.erp_info && elems.erp_info_len >= 1) {
1617                 use_protection =
1618                         (elems.erp_info[0] & ERP_INFO_USE_PROTECTION) != 0;
1619         }
1620
1621         if (use_protection != !!wpa_s->mlme.use_protection) {
1622                 wpa_printf(MSG_DEBUG, "MLME: CTS protection %s (BSSID=" MACSTR
1623                            ")",
1624                            use_protection ? "enabled" : "disabled",
1625                            MAC2STR(wpa_s->bssid));
1626                 wpa_s->mlme.use_protection = use_protection ? 1 : 0;
1627                 wpa_s->mlme.cts_protect_erp_frames = use_protection;
1628         }
1629
1630         if (elems.wmm && wpa_s->mlme.wmm_enabled) {
1631                 ieee80211_sta_wmm_params(wpa_s, elems.wmm,
1632                                          elems.wmm_len);
1633         }
1634 }
1635
1636
1637 static void ieee80211_rx_mgmt_probe_req(struct wpa_supplicant *wpa_s,
1638                                         struct ieee80211_mgmt *mgmt,
1639                                         size_t len,
1640                                         struct ieee80211_rx_status *rx_status)
1641 {
1642         int tx_last_beacon, adhoc;
1643 #if 0 /* FIX */
1644         struct ieee80211_mgmt *resp;
1645 #endif
1646         u8 *pos, *end;
1647         struct wpa_ssid *ssid = wpa_s->current_ssid;
1648
1649         adhoc = ssid && ssid->mode == WPAS_MODE_IBSS;
1650
1651         if (!adhoc || wpa_s->mlme.state != IEEE80211_IBSS_JOINED ||
1652             len < 24 + 2 || wpa_s->mlme.probe_resp == NULL)
1653                 return;
1654
1655 #if 0 /* FIX */
1656         if (local->hw->tx_last_beacon)
1657                 tx_last_beacon = local->hw->tx_last_beacon(local->mdev);
1658         else
1659 #endif
1660                 tx_last_beacon = 1;
1661
1662 #ifdef IEEE80211_IBSS_DEBUG
1663         wpa_printf(MSG_DEBUG, "MLME: RX ProbeReq SA=" MACSTR " DA=" MACSTR
1664                    " BSSID=" MACSTR " (tx_last_beacon=%d)",
1665                    MAC2STR(mgmt->sa), MAC2STR(mgmt->da),
1666                    MAC2STR(mgmt->bssid), tx_last_beacon);
1667 #endif /* IEEE80211_IBSS_DEBUG */
1668
1669         if (!tx_last_beacon)
1670                 return;
1671
1672         if (os_memcmp(mgmt->bssid, wpa_s->bssid, ETH_ALEN) != 0 &&
1673             os_memcmp(mgmt->bssid, "\xff\xff\xff\xff\xff\xff", ETH_ALEN) != 0)
1674                 return;
1675
1676         end = ((u8 *) mgmt) + len;
1677         pos = mgmt->u.probe_req.variable;
1678         if (pos[0] != WLAN_EID_SSID ||
1679             pos + 2 + pos[1] > end) {
1680                 wpa_printf(MSG_DEBUG, "MLME: Invalid SSID IE in ProbeReq from "
1681                            MACSTR, MAC2STR(mgmt->sa));
1682                 return;
1683         }
1684         if (pos[1] != 0 &&
1685             (pos[1] != wpa_s->mlme.ssid_len ||
1686              os_memcmp(pos + 2, wpa_s->mlme.ssid, wpa_s->mlme.ssid_len) != 0))
1687         {
1688                 /* Ignore ProbeReq for foreign SSID */
1689                 return;
1690         }
1691
1692 #if 0 /* FIX */
1693         /* Reply with ProbeResp */
1694         skb = skb_copy(wpa_s->mlme.probe_resp, GFP_ATOMIC);
1695         if (skb == NULL)
1696                 return;
1697
1698         resp = (struct ieee80211_mgmt *) skb->data;
1699         os_memcpy(resp->da, mgmt->sa, ETH_ALEN);
1700 #ifdef IEEE80211_IBSS_DEBUG
1701         wpa_printf(MSG_DEBUG, "MLME: Sending ProbeResp to " MACSTR,
1702                    MAC2STR(resp->da));
1703 #endif /* IEEE80211_IBSS_DEBUG */
1704         ieee80211_sta_tx(wpa_s, skb, 0, 1);
1705 #endif
1706 }
1707
1708
1709 #ifdef CONFIG_IEEE80211R
1710 static void ieee80211_rx_mgmt_ft_action(struct wpa_supplicant *wpa_s,
1711                                         struct ieee80211_mgmt *mgmt,
1712                                         size_t len,
1713                                         struct ieee80211_rx_status *rx_status)
1714 {
1715         union wpa_event_data data;
1716         u16 status;
1717         u8 *sta_addr, *target_ap_addr;
1718
1719         if (len < 24 + 1 + sizeof(mgmt->u.action.u.ft_action_resp)) {
1720                 wpa_printf(MSG_DEBUG, "MLME: Too short FT Action frame");
1721                 return;
1722         }
1723
1724         /*
1725          * Only FT Action Response is needed for now since reservation
1726          * protocol is not supported.
1727          */
1728         if (mgmt->u.action.u.ft_action_resp.action != 2) {
1729                 wpa_printf(MSG_DEBUG, "MLME: Unexpected FT Action %d",
1730                            mgmt->u.action.u.ft_action_resp.action);
1731                 return;
1732         }
1733
1734         status = le_to_host16(mgmt->u.action.u.ft_action_resp.status_code);
1735         sta_addr = mgmt->u.action.u.ft_action_resp.sta_addr;
1736         target_ap_addr = mgmt->u.action.u.ft_action_resp.target_ap_addr;
1737         wpa_printf(MSG_DEBUG, "MLME: Received FT Action Response: STA " MACSTR
1738                    " TargetAP " MACSTR " Status Code %d",
1739                    MAC2STR(sta_addr), MAC2STR(target_ap_addr), status);
1740         if (os_memcmp(sta_addr, wpa_s->own_addr, ETH_ALEN) != 0) {
1741                 wpa_printf(MSG_DEBUG, "MLME: Foreign STA Address " MACSTR
1742                            " in FT Action Response", MAC2STR(sta_addr));
1743                 return;
1744         }
1745
1746         if (status) {
1747                 wpa_printf(MSG_DEBUG, "MLME: FT Action Response indicates "
1748                            "failure (status code %d)", status);
1749                 /* TODO: report error to FT code(?) */
1750                 return;
1751         }
1752
1753         os_memset(&data, 0, sizeof(data));
1754         data.ft_ies.ies = mgmt->u.action.u.ft_action_resp.variable;
1755         data.ft_ies.ies_len = len - (mgmt->u.action.u.ft_action_resp.variable -
1756                                      (u8 *) mgmt);
1757         data.ft_ies.ft_action = 1;
1758         os_memcpy(data.ft_ies.target_ap, target_ap_addr, ETH_ALEN);
1759         wpa_supplicant_event(wpa_s, EVENT_FT_RESPONSE, &data);
1760         /* TODO: should only re-associate, if EVENT_FT_RESPONSE was processed
1761          * successfully */
1762         wpa_s->mlme.prev_bssid_set = 1;
1763         wpa_s->mlme.auth_alg = WLAN_AUTH_FT;
1764         os_memcpy(wpa_s->mlme.prev_bssid, wpa_s->bssid, ETH_ALEN);
1765         os_memcpy(wpa_s->bssid, target_ap_addr, ETH_ALEN);
1766         ieee80211_associate(wpa_s);
1767 }
1768 #endif /* CONFIG_IEEE80211R */
1769
1770
1771 #ifdef CONFIG_IEEE80211W
1772
1773 /* MLME-SAQuery.response */
1774 static int ieee80211_sta_send_sa_query_resp(struct wpa_supplicant *wpa_s,
1775                                             const u8 *addr, const u8 *trans_id)
1776 {
1777         struct ieee80211_mgmt *mgmt;
1778         int res;
1779         size_t len;
1780
1781         mgmt = os_zalloc(sizeof(*mgmt));
1782         if (mgmt == NULL) {
1783                 wpa_printf(MSG_DEBUG, "MLME: Failed to allocate buffer for "
1784                            "SA Query action frame");
1785                 return -1;
1786         }
1787
1788         len = 24;
1789         os_memcpy(mgmt->da, addr, ETH_ALEN);
1790         os_memcpy(mgmt->sa, wpa_s->own_addr, ETH_ALEN);
1791         os_memcpy(mgmt->bssid, wpa_s->bssid, ETH_ALEN);
1792         mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
1793                                            WLAN_FC_STYPE_ACTION);
1794         mgmt->u.action.category = WLAN_ACTION_SA_QUERY;
1795         mgmt->u.action.u.sa_query_resp.action = WLAN_SA_QUERY_RESPONSE;
1796         os_memcpy(mgmt->u.action.u.sa_query_resp.trans_id, trans_id,
1797                   WLAN_SA_QUERY_TR_ID_LEN);
1798         len += 1 + sizeof(mgmt->u.action.u.sa_query_resp);
1799
1800         res = ieee80211_sta_tx(wpa_s, (u8 *) mgmt, len);
1801         os_free(mgmt);
1802
1803         return res;
1804 }
1805
1806
1807 static void ieee80211_rx_mgmt_sa_query_action(
1808         struct wpa_supplicant *wpa_s, struct ieee80211_mgmt *mgmt, size_t len,
1809         struct ieee80211_rx_status *rx_status)
1810 {
1811         if (len < 24 + 1 + sizeof(mgmt->u.action.u.sa_query_req)) {
1812                 wpa_printf(MSG_DEBUG, "MLME: Too short SA Query Action frame");
1813                 return;
1814         }
1815
1816         if (mgmt->u.action.u.sa_query_req.action != WLAN_SA_QUERY_REQUEST) {
1817                 wpa_printf(MSG_DEBUG, "MLME: Unexpected SA Query Action %d",
1818                            mgmt->u.action.u.sa_query_req.action);
1819                 return;
1820         }
1821
1822         if (os_memcmp(mgmt->sa, wpa_s->bssid, ETH_ALEN) != 0) {
1823                 wpa_printf(MSG_DEBUG, "MLME: Ignore SA Query from unknown "
1824                            "source " MACSTR, MAC2STR(mgmt->sa));
1825                 return;
1826         }
1827
1828         if (wpa_s->mlme.state == IEEE80211_ASSOCIATE) {
1829                 wpa_printf(MSG_DEBUG, "MLME: Ignore SA query request during "
1830                            "association process");
1831                 return;
1832         }
1833
1834         wpa_printf(MSG_DEBUG, "MLME: Replying to SA Query request");
1835         ieee80211_sta_send_sa_query_resp(wpa_s, mgmt->sa, mgmt->u.action.u.
1836                                          sa_query_req.trans_id);
1837 }
1838
1839 #endif /* CONFIG_IEEE80211W */
1840
1841
1842 static void dump_tspec(struct wmm_tspec_element *tspec)
1843 {
1844         int up, psb, dir, tid;
1845         u16 val;
1846
1847         up = (tspec->ts_info[1] >> 3) & 0x07;
1848         psb = (tspec->ts_info[1] >> 2) & 0x01;
1849         dir = (tspec->ts_info[0] >> 5) & 0x03;
1850         tid = (tspec->ts_info[0] >> 1) & 0x0f;
1851         wpa_printf(MSG_DEBUG, "WMM: TS Info: UP=%d PSB=%d Direction=%d TID=%d",
1852                    up, psb, dir, tid);
1853         val = le_to_host16(tspec->nominal_msdu_size);
1854         wpa_printf(MSG_DEBUG, "WMM: Nominal MSDU Size: %d%s",
1855                    val & 0x7fff, val & 0x8000 ? " (fixed)" : "");
1856         wpa_printf(MSG_DEBUG, "WMM: Mean Data Rate: %u bps",
1857                    le_to_host32(tspec->mean_data_rate));
1858         wpa_printf(MSG_DEBUG, "WMM: Minimum PHY Rate: %u bps",
1859                    le_to_host32(tspec->minimum_phy_rate));
1860         val = le_to_host16(tspec->surplus_bandwidth_allowance);
1861         wpa_printf(MSG_DEBUG, "WMM: Surplus Bandwidth Allowance: %u.%04u",
1862                    val >> 13, 10000 * (val & 0x1fff) / 0x2000);
1863         val = le_to_host16(tspec->medium_time);
1864         wpa_printf(MSG_DEBUG, "WMM: Medium Time: %u (= %u usec/sec)",
1865                    val, 32 * val);
1866 }
1867
1868
1869 static int is_wmm_tspec(const u8 *ie, size_t len)
1870 {
1871         const struct wmm_tspec_element *tspec;
1872
1873         if (len < sizeof(*tspec))
1874                 return 0;
1875
1876         tspec = (const struct wmm_tspec_element *) ie;
1877         if (tspec->eid != WLAN_EID_VENDOR_SPECIFIC ||
1878             tspec->length < sizeof(*tspec) - 2 ||
1879             tspec->oui[0] != 0x00 || tspec->oui[1] != 0x50 ||
1880             tspec->oui[2] != 0xf2 || tspec->oui_type != 2 ||
1881             tspec->oui_subtype != 2 || tspec->version != 1)
1882                 return 0;
1883
1884         return 1;
1885 }
1886
1887
1888 static void ieee80211_rx_addts_resp(
1889         struct wpa_supplicant *wpa_s, struct ieee80211_mgmt *mgmt, size_t len,
1890         size_t var_len)
1891 {
1892         struct wmm_tspec_element *tspec;
1893
1894         wpa_printf(MSG_DEBUG, "WMM: Received ADDTS Response");
1895         wpa_hexdump(MSG_MSGDUMP, "WMM: ADDTS Response IE(s)",
1896                     mgmt->u.action.u.wmm_action.variable, var_len);
1897         if (!is_wmm_tspec(mgmt->u.action.u.wmm_action.variable, var_len))
1898                 return;
1899         tspec = (struct wmm_tspec_element *)
1900                 mgmt->u.action.u.wmm_action.variable;
1901         dump_tspec(tspec);
1902 }
1903
1904
1905 static void ieee80211_rx_delts(
1906         struct wpa_supplicant *wpa_s, struct ieee80211_mgmt *mgmt, size_t len,
1907         size_t var_len)
1908 {
1909         struct wmm_tspec_element *tspec;
1910
1911         wpa_printf(MSG_DEBUG, "WMM: Received DELTS");
1912         wpa_hexdump(MSG_MSGDUMP, "WMM: DELTS IE(s)",
1913                     mgmt->u.action.u.wmm_action.variable, var_len);
1914         if (!is_wmm_tspec(mgmt->u.action.u.wmm_action.variable, var_len))
1915                 return;
1916         tspec = (struct wmm_tspec_element *)
1917                 mgmt->u.action.u.wmm_action.variable;
1918         dump_tspec(tspec);
1919 }
1920
1921
1922 static void ieee80211_rx_mgmt_wmm_action(
1923         struct wpa_supplicant *wpa_s, struct ieee80211_mgmt *mgmt, size_t len,
1924         struct ieee80211_rx_status *rx_status)
1925 {
1926         size_t alen;
1927
1928         alen = mgmt->u.action.u.wmm_action.variable - (u8 *) mgmt;
1929         if (len < alen) {
1930                 wpa_printf(MSG_DEBUG, "WMM: Received Action frame too short");
1931                 return;
1932         }
1933
1934         wpa_printf(MSG_DEBUG, "WMM: Received Action frame: Action Code %d, "
1935                    "Dialog Token %d, Status Code %d",
1936                    mgmt->u.action.u.wmm_action.action_code,
1937                    mgmt->u.action.u.wmm_action.dialog_token,
1938                    mgmt->u.action.u.wmm_action.status_code);
1939
1940         switch (mgmt->u.action.u.wmm_action.action_code) {
1941         case WMM_ACTION_CODE_ADDTS_RESP:
1942                 ieee80211_rx_addts_resp(wpa_s, mgmt, len, len - alen);
1943                 break;
1944         case WMM_ACTION_CODE_DELTS:
1945                 ieee80211_rx_delts(wpa_s, mgmt, len, len - alen);
1946                 break;
1947         default:
1948                 wpa_printf(MSG_DEBUG, "WMM: Unsupported Action Code %d",
1949                            mgmt->u.action.u.wmm_action.action_code);
1950                 break;
1951         }
1952 }
1953
1954
1955 static void ieee80211_rx_mgmt_action(struct wpa_supplicant *wpa_s,
1956                                      struct ieee80211_mgmt *mgmt,
1957                                      size_t len,
1958                                      struct ieee80211_rx_status *rx_status)
1959 {
1960         wpa_printf(MSG_DEBUG, "MLME: received Action frame");
1961
1962         if (len < 25)
1963                 return;
1964
1965         switch (mgmt->u.action.category) {
1966 #ifdef CONFIG_IEEE80211R
1967         case WLAN_ACTION_FT:
1968                 ieee80211_rx_mgmt_ft_action(wpa_s, mgmt, len, rx_status);
1969                 break;
1970 #endif /* CONFIG_IEEE80211R */
1971 #ifdef CONFIG_IEEE80211W
1972         case WLAN_ACTION_SA_QUERY:
1973                 ieee80211_rx_mgmt_sa_query_action(wpa_s, mgmt, len, rx_status);
1974                 break;
1975 #endif /* CONFIG_IEEE80211W */
1976         case WLAN_ACTION_WMM:
1977                 ieee80211_rx_mgmt_wmm_action(wpa_s, mgmt, len, rx_status);
1978                 break;
1979         default:
1980                 wpa_printf(MSG_DEBUG, "MLME: unknown Action Category %d",
1981                            mgmt->u.action.category);
1982                 break;
1983         }
1984 }
1985
1986
1987 static void ieee80211_sta_rx_mgmt(struct wpa_supplicant *wpa_s,
1988                                   const u8 *buf, size_t len,
1989                                   struct ieee80211_rx_status *rx_status)
1990 {
1991         struct ieee80211_mgmt *mgmt;
1992         u16 fc;
1993
1994         if (len < 24)
1995                 return;
1996
1997         mgmt = (struct ieee80211_mgmt *) buf;
1998         fc = le_to_host16(mgmt->frame_control);
1999
2000         switch (WLAN_FC_GET_STYPE(fc)) {
2001         case WLAN_FC_STYPE_PROBE_REQ:
2002                 ieee80211_rx_mgmt_probe_req(wpa_s, mgmt, len, rx_status);
2003                 break;
2004         case WLAN_FC_STYPE_PROBE_RESP:
2005                 ieee80211_rx_mgmt_probe_resp(wpa_s, mgmt, len, rx_status);
2006                 break;
2007         case WLAN_FC_STYPE_BEACON:
2008                 ieee80211_rx_mgmt_beacon(wpa_s, mgmt, len, rx_status);
2009                 break;
2010         case WLAN_FC_STYPE_AUTH:
2011                 ieee80211_rx_mgmt_auth(wpa_s, mgmt, len, rx_status);
2012                 break;
2013         case WLAN_FC_STYPE_ASSOC_RESP:
2014                 ieee80211_rx_mgmt_assoc_resp(wpa_s, mgmt, len, rx_status, 0);
2015                 break;
2016         case WLAN_FC_STYPE_REASSOC_RESP:
2017                 ieee80211_rx_mgmt_assoc_resp(wpa_s, mgmt, len, rx_status, 1);
2018                 break;
2019         case WLAN_FC_STYPE_DEAUTH:
2020                 ieee80211_rx_mgmt_deauth(wpa_s, mgmt, len, rx_status);
2021                 break;
2022         case WLAN_FC_STYPE_DISASSOC:
2023                 ieee80211_rx_mgmt_disassoc(wpa_s, mgmt, len, rx_status);
2024                 break;
2025         case WLAN_FC_STYPE_ACTION:
2026                 ieee80211_rx_mgmt_action(wpa_s, mgmt, len, rx_status);
2027                 break;
2028         default:
2029                 wpa_printf(MSG_DEBUG, "MLME: received unknown management "
2030                            "frame - stype=%d", WLAN_FC_GET_STYPE(fc));
2031                 break;
2032         }
2033 }
2034
2035
2036 static void ieee80211_sta_rx_scan(struct wpa_supplicant *wpa_s,
2037                                   const u8 *buf, size_t len,
2038                                   struct ieee80211_rx_status *rx_status)
2039 {
2040         struct ieee80211_mgmt *mgmt;
2041         u16 fc;
2042
2043         if (len < 24)
2044                 return;
2045
2046         mgmt = (struct ieee80211_mgmt *) buf;
2047         fc = le_to_host16(mgmt->frame_control);
2048
2049         if (WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_MGMT) {
2050                 if (WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_PROBE_RESP) {
2051                         ieee80211_rx_mgmt_probe_resp(wpa_s, mgmt,
2052                                                      len, rx_status);
2053                 } else if (WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_BEACON) {
2054                         ieee80211_rx_mgmt_beacon(wpa_s, mgmt, len, rx_status);
2055                 }
2056         }
2057 }
2058
2059
2060 static int ieee80211_sta_active_ibss(struct wpa_supplicant *wpa_s)
2061 {
2062         int active = 0;
2063
2064 #if 0 /* FIX */
2065         list_for_each(ptr, &local->sta_list) {
2066                 sta = list_entry(ptr, struct sta_info, list);
2067                 if (sta->dev == dev &&
2068                     time_after(sta->last_rx + IEEE80211_IBSS_MERGE_INTERVAL,
2069                                jiffies)) {
2070                         active++;
2071                         break;
2072                 }
2073         }
2074 #endif
2075
2076         return active;
2077 }
2078
2079
2080 static void ieee80211_sta_expire(struct wpa_supplicant *wpa_s)
2081 {
2082 #if 0 /* FIX */
2083         list_for_each_safe(ptr, n, &local->sta_list) {
2084                 sta = list_entry(ptr, struct sta_info, list);
2085                 if (time_after(jiffies, sta->last_rx +
2086                                IEEE80211_IBSS_INACTIVITY_LIMIT)) {
2087                         wpa_printf(MSG_DEBUG, "MLME: expiring inactive STA "
2088                                    MACSTR, MAC2STR(sta->addr));
2089                         sta_info_free(local, sta, 1);
2090                 }
2091         }
2092 #endif
2093 }
2094
2095
2096 static void ieee80211_sta_merge_ibss(struct wpa_supplicant *wpa_s)
2097 {
2098         struct wpa_driver_scan_params params;
2099
2100         ieee80211_reschedule_timer(wpa_s, IEEE80211_IBSS_MERGE_INTERVAL);
2101
2102         ieee80211_sta_expire(wpa_s);
2103         if (ieee80211_sta_active_ibss(wpa_s))
2104                 return;
2105
2106         wpa_printf(MSG_DEBUG, "MLME: No active IBSS STAs - trying to scan for "
2107                    "other IBSS networks with same SSID (merge)");
2108         os_memset(&params, 0, sizeof(params));
2109         params.ssids[0].ssid = wpa_s->mlme.ssid;
2110         params.ssids[0].ssid_len = wpa_s->mlme.ssid_len;
2111         params.num_ssids = wpa_s->mlme.ssid_len ? 1 : 0;
2112         ieee80211_sta_req_scan(wpa_s, &params);
2113 }
2114
2115
2116 static void ieee80211_sta_timer(void *eloop_ctx, void *timeout_ctx)
2117 {
2118         struct wpa_supplicant *wpa_s = eloop_ctx;
2119
2120         switch (wpa_s->mlme.state) {
2121         case IEEE80211_DISABLED:
2122                 break;
2123         case IEEE80211_AUTHENTICATE:
2124                 ieee80211_authenticate(wpa_s);
2125                 break;
2126         case IEEE80211_ASSOCIATE:
2127                 ieee80211_associate(wpa_s);
2128                 break;
2129         case IEEE80211_ASSOCIATED:
2130                 ieee80211_associated(wpa_s);
2131                 break;
2132         case IEEE80211_IBSS_SEARCH:
2133                 ieee80211_sta_find_ibss(wpa_s);
2134                 break;
2135         case IEEE80211_IBSS_JOINED:
2136                 ieee80211_sta_merge_ibss(wpa_s);
2137                 break;
2138         default:
2139                 wpa_printf(MSG_DEBUG, "ieee80211_sta_timer: Unknown state %d",
2140                            wpa_s->mlme.state);
2141                 break;
2142         }
2143
2144         if (ieee80211_privacy_mismatch(wpa_s)) {
2145                 wpa_printf(MSG_DEBUG, "MLME: privacy configuration mismatch "
2146                            "and mixed-cell disabled - disassociate");
2147
2148                 ieee80211_send_disassoc(wpa_s, WLAN_REASON_UNSPECIFIED);
2149                 ieee80211_set_associated(wpa_s, 0);
2150         }
2151 }
2152
2153
2154 static void ieee80211_sta_new_auth(struct wpa_supplicant *wpa_s)
2155 {
2156         struct wpa_ssid *ssid = wpa_s->current_ssid;
2157         if (ssid && ssid->mode != WPAS_MODE_INFRA)
2158                 return;
2159
2160 #if 0 /* FIX */
2161         if (local->hw->reset_tsf) {
2162                 /* Reset own TSF to allow time synchronization work. */
2163                 local->hw->reset_tsf(local->mdev);
2164         }
2165 #endif
2166
2167         wpa_s->mlme.wmm_last_param_set = -1; /* allow any WMM update */
2168
2169
2170         if (wpa_s->mlme.auth_algs & WPA_AUTH_ALG_OPEN)
2171                 wpa_s->mlme.auth_alg = WLAN_AUTH_OPEN;
2172         else if (wpa_s->mlme.auth_algs & WPA_AUTH_ALG_SHARED)
2173                 wpa_s->mlme.auth_alg = WLAN_AUTH_SHARED_KEY;
2174         else if (wpa_s->mlme.auth_algs & WPA_AUTH_ALG_LEAP)
2175                 wpa_s->mlme.auth_alg = WLAN_AUTH_LEAP;
2176         else
2177                 wpa_s->mlme.auth_alg = WLAN_AUTH_OPEN;
2178         wpa_printf(MSG_DEBUG, "MLME: Initial auth_alg=%d",
2179                    wpa_s->mlme.auth_alg);
2180         wpa_s->mlme.auth_transaction = -1;
2181         wpa_s->mlme.auth_tries = wpa_s->mlme.assoc_tries = 0;
2182         ieee80211_authenticate(wpa_s);
2183 }
2184
2185
2186 static int ieee80211_ibss_allowed(struct wpa_supplicant *wpa_s)
2187 {
2188 #if 0 /* FIX */
2189         int m, c;
2190
2191         for (m = 0; m < local->hw->num_modes; m++) {
2192                 struct ieee80211_hw_modes *mode = &local->hw->modes[m];
2193                 if (mode->mode != local->conf.phymode)
2194                         continue;
2195                 for (c = 0; c < mode->num_channels; c++) {
2196                         struct ieee80211_channel *chan = &mode->channels[c];
2197                         if (chan->flag & IEEE80211_CHAN_W_SCAN &&
2198                             chan->chan == local->conf.channel) {
2199                                 if (chan->flag & IEEE80211_CHAN_W_IBSS)
2200                                         return 1;
2201                                 break;
2202                         }
2203                 }
2204         }
2205 #endif
2206
2207         return 0;
2208 }
2209
2210
2211 static int ieee80211_sta_join_ibss(struct wpa_supplicant *wpa_s,
2212                                    struct ieee80211_sta_bss *bss)
2213 {
2214         int res = 0, rates, done = 0, bssid_changed;
2215         struct ieee80211_mgmt *mgmt;
2216 #if 0 /* FIX */
2217         struct ieee80211_tx_control control;
2218         struct ieee80211_rate *rate;
2219         struct rate_control_extra extra;
2220 #endif
2221         u8 *pos, *buf;
2222         size_t len;
2223
2224         /* Remove possible STA entries from other IBSS networks. */
2225 #if 0 /* FIX */
2226         sta_info_flush(local, NULL);
2227
2228         if (local->hw->reset_tsf) {
2229                 /* Reset own TSF to allow time synchronization work. */
2230                 local->hw->reset_tsf(local->mdev);
2231         }
2232 #endif
2233         bssid_changed = os_memcmp(wpa_s->bssid, bss->bssid, ETH_ALEN);
2234         os_memcpy(wpa_s->bssid, bss->bssid, ETH_ALEN);
2235         if (bssid_changed)
2236                 wpas_notify_bssid_changed(wpa_s);
2237
2238 #if 0 /* FIX */
2239         local->conf.beacon_int = bss->beacon_int >= 10 ? bss->beacon_int : 10;
2240
2241         sdata->drop_unencrypted = bss->capability &
2242                 host_to_le16(WLAN_CAPABILITY_PRIVACY) ? 1 : 0;
2243 #endif
2244
2245 #if 0 /* FIX */
2246         os_memset(&rq, 0, sizeof(rq));
2247         rq.m = bss->freq * 100000;
2248         rq.e = 1;
2249         res = ieee80211_ioctl_siwfreq(wpa_s, NULL, &rq, NULL);
2250 #endif
2251
2252         if (!ieee80211_ibss_allowed(wpa_s)) {
2253 #if 0 /* FIX */
2254                 wpa_printf(MSG_DEBUG, "MLME: IBSS not allowed on channel %d "
2255                            "(%d MHz)", local->conf.channel,
2256                            local->conf.freq);
2257 #endif
2258                 return -1;
2259         }
2260
2261         /* Set beacon template based on scan results */
2262         buf = os_malloc(400);
2263         len = 0;
2264         do {
2265                 if (buf == NULL)
2266                         break;
2267
2268                 mgmt = (struct ieee80211_mgmt *) buf;
2269                 len += 24 + sizeof(mgmt->u.beacon);
2270                 os_memset(mgmt, 0, 24 + sizeof(mgmt->u.beacon));
2271                 mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
2272                                                    WLAN_FC_STYPE_BEACON);
2273                 os_memset(mgmt->da, 0xff, ETH_ALEN);
2274                 os_memcpy(mgmt->sa, wpa_s->own_addr, ETH_ALEN);
2275                 os_memcpy(mgmt->bssid, wpa_s->bssid, ETH_ALEN);
2276 #if 0 /* FIX */
2277                 mgmt->u.beacon.beacon_int =
2278                         host_to_le16(local->conf.beacon_int);
2279 #endif
2280                 mgmt->u.beacon.capab_info = host_to_le16(bss->capability);
2281
2282                 pos = buf + len;
2283                 len += 2 + wpa_s->mlme.ssid_len;
2284                 *pos++ = WLAN_EID_SSID;
2285                 *pos++ = wpa_s->mlme.ssid_len;
2286                 os_memcpy(pos, wpa_s->mlme.ssid, wpa_s->mlme.ssid_len);
2287
2288                 rates = bss->supp_rates_len;
2289                 if (rates > 8)
2290                         rates = 8;
2291                 pos = buf + len;
2292                 len += 2 + rates;
2293                 *pos++ = WLAN_EID_SUPP_RATES;
2294                 *pos++ = rates;
2295                 os_memcpy(pos, bss->supp_rates, rates);
2296
2297                 pos = buf + len;
2298                 len += 2 + 1;
2299                 *pos++ = WLAN_EID_DS_PARAMS;
2300                 *pos++ = 1;
2301                 *pos++ = bss->channel;
2302
2303                 pos = buf + len;
2304                 len += 2 + 2;
2305                 *pos++ = WLAN_EID_IBSS_PARAMS;
2306                 *pos++ = 2;
2307                 /* FIX: set ATIM window based on scan results */
2308                 *pos++ = 0;
2309                 *pos++ = 0;
2310
2311                 if (bss->supp_rates_len > 8) {
2312                         rates = bss->supp_rates_len - 8;
2313                         pos = buf + len;
2314                         len += 2 + rates;
2315                         *pos++ = WLAN_EID_EXT_SUPP_RATES;
2316                         *pos++ = rates;
2317                         os_memcpy(pos, &bss->supp_rates[8], rates);
2318                 }
2319
2320 #if 0 /* FIX */
2321                 os_memset(&control, 0, sizeof(control));
2322                 control.pkt_type = PKT_PROBE_RESP;
2323                 os_memset(&extra, 0, sizeof(extra));
2324                 extra.endidx = local->num_curr_rates;
2325                 rate = rate_control_get_rate(wpa_s, skb, &extra);
2326                 if (rate == NULL) {
2327                         wpa_printf(MSG_DEBUG, "MLME: Failed to determine TX "
2328                                    "rate for IBSS beacon");
2329                         break;
2330                 }
2331                 control.tx_rate = (wpa_s->mlme.short_preamble &&
2332                                    (rate->flags & IEEE80211_RATE_PREAMBLE2)) ?
2333                         rate->val2 : rate->val;
2334                 control.antenna_sel = local->conf.antenna_sel;
2335                 control.power_level = local->conf.power_level;
2336                 control.no_ack = 1;
2337                 control.retry_limit = 1;
2338                 control.rts_cts_duration = 0;
2339 #endif
2340
2341 #if 0 /* FIX */
2342                 wpa_s->mlme.probe_resp = skb_copy(skb, GFP_ATOMIC);
2343                 if (wpa_s->mlme.probe_resp) {
2344                         mgmt = (struct ieee80211_mgmt *)
2345                                 wpa_s->mlme.probe_resp->data;
2346                         mgmt->frame_control =
2347                                 IEEE80211_FC(WLAN_FC_TYPE_MGMT,
2348                                              WLAN_FC_STYPE_PROBE_RESP);
2349                 } else {
2350                         wpa_printf(MSG_DEBUG, "MLME: Could not allocate "
2351                                    "ProbeResp template for IBSS");
2352                 }
2353
2354                 if (local->hw->beacon_update &&
2355                     local->hw->beacon_update(wpa_s, skb, &control) == 0) {
2356                         wpa_printf(MSG_DEBUG, "MLME: Configured IBSS beacon "
2357                                    "template based on scan results");
2358                         skb = NULL;
2359                 }
2360
2361                 rates = 0;
2362                 for (i = 0; i < bss->supp_rates_len; i++) {
2363                         int rate = (bss->supp_rates[i] & 0x7f) * 5;
2364                         if (local->conf.phymode == MODE_ATHEROS_TURBO)
2365                                 rate *= 2;
2366                         for (j = 0; j < local->num_curr_rates; j++)
2367                                 if (local->curr_rates[j] == rate)
2368                                         rates |= BIT(j);
2369                 }
2370                 wpa_s->mlme.supp_rates_bits = rates;
2371 #endif
2372                 done = 1;
2373         } while (0);
2374
2375         os_free(buf);
2376         if (!done) {
2377                 wpa_printf(MSG_DEBUG, "MLME: Failed to configure IBSS beacon "
2378                            "template");
2379         }
2380
2381         wpa_s->mlme.state = IEEE80211_IBSS_JOINED;
2382         ieee80211_reschedule_timer(wpa_s, IEEE80211_IBSS_MERGE_INTERVAL);
2383
2384         return res;
2385 }
2386
2387
2388 #if 0 /* FIX */
2389 static int ieee80211_sta_create_ibss(struct wpa_supplicant *wpa_s)
2390 {
2391         struct ieee80211_sta_bss *bss;
2392         u8 bssid[ETH_ALEN], *pos;
2393         int i;
2394
2395 #if 0
2396         /* Easier testing, use fixed BSSID. */
2397         os_memset(bssid, 0xfe, ETH_ALEN);
2398 #else
2399         /* Generate random, not broadcast, locally administered BSSID. Mix in
2400          * own MAC address to make sure that devices that do not have proper
2401          * random number generator get different BSSID. */
2402         os_get_random(bssid, ETH_ALEN);
2403         for (i = 0; i < ETH_ALEN; i++)
2404                 bssid[i] ^= wpa_s->own_addr[i];
2405         bssid[0] &= ~0x01;
2406         bssid[0] |= 0x02;
2407 #endif
2408
2409         wpa_printf(MSG_DEBUG, "MLME: Creating new IBSS network, BSSID "
2410                    MACSTR "", MAC2STR(bssid));
2411
2412         bss = ieee80211_bss_add(wpa_s, bssid);
2413         if (bss == NULL)
2414                 return -ENOMEM;
2415
2416 #if 0 /* FIX */
2417         if (local->conf.beacon_int == 0)
2418                 local->conf.beacon_int = 100;
2419         bss->beacon_int = local->conf.beacon_int;
2420         bss->hw_mode = local->conf.phymode;
2421         bss->channel = local->conf.channel;
2422         bss->freq = local->conf.freq;
2423 #endif
2424         os_get_time(&bss->last_update);
2425         bss->capability = host_to_le16(WLAN_CAPABILITY_IBSS);
2426 #if 0 /* FIX */
2427         if (sdata->default_key) {
2428                 bss->capability |= host_to_le16(WLAN_CAPABILITY_PRIVACY);
2429         } else
2430                 sdata->drop_unencrypted = 0;
2431         bss->supp_rates_len = local->num_curr_rates;
2432 #endif
2433         pos = bss->supp_rates;
2434 #if 0 /* FIX */
2435         for (i = 0; i < local->num_curr_rates; i++) {
2436                 int rate = local->curr_rates[i];
2437                 if (local->conf.phymode == MODE_ATHEROS_TURBO)
2438                         rate /= 2;
2439                 *pos++ = (u8) (rate / 5);
2440         }
2441 #endif
2442
2443         return ieee80211_sta_join_ibss(wpa_s, bss);
2444 }
2445 #endif
2446
2447
2448 static int ieee80211_sta_find_ibss(struct wpa_supplicant *wpa_s)
2449 {
2450         struct ieee80211_sta_bss *bss;
2451         int found = 0;
2452         u8 bssid[ETH_ALEN];
2453         int active_ibss;
2454         struct os_time now;
2455
2456         if (wpa_s->mlme.ssid_len == 0)
2457                 return -EINVAL;
2458
2459         active_ibss = ieee80211_sta_active_ibss(wpa_s);
2460 #ifdef IEEE80211_IBSS_DEBUG
2461         wpa_printf(MSG_DEBUG, "MLME: sta_find_ibss (active_ibss=%d)",
2462                    active_ibss);
2463 #endif /* IEEE80211_IBSS_DEBUG */
2464         for (bss = wpa_s->mlme.sta_bss_list; bss; bss = bss->next) {
2465                 if (wpa_s->mlme.ssid_len != bss->ssid_len ||
2466                     os_memcmp(wpa_s->mlme.ssid, bss->ssid, bss->ssid_len) != 0
2467                     || !(bss->capability & WLAN_CAPABILITY_IBSS))
2468                         continue;
2469 #ifdef IEEE80211_IBSS_DEBUG
2470                 wpa_printf(MSG_DEBUG, "   bssid=" MACSTR " found",
2471                            MAC2STR(bss->bssid));
2472 #endif /* IEEE80211_IBSS_DEBUG */
2473                 os_memcpy(bssid, bss->bssid, ETH_ALEN);
2474                 found = 1;
2475                 if (active_ibss ||
2476                     os_memcmp(bssid, wpa_s->bssid, ETH_ALEN) != 0)
2477                         break;
2478         }
2479
2480 #ifdef IEEE80211_IBSS_DEBUG
2481         wpa_printf(MSG_DEBUG, "   sta_find_ibss: selected " MACSTR " current "
2482                    MACSTR, MAC2STR(bssid), MAC2STR(wpa_s->bssid));
2483 #endif /* IEEE80211_IBSS_DEBUG */
2484         if (found && os_memcmp(wpa_s->bssid, bssid, ETH_ALEN) != 0 &&
2485             (bss = ieee80211_bss_get(wpa_s, bssid))) {
2486                 wpa_printf(MSG_DEBUG, "MLME: Selected IBSS BSSID " MACSTR
2487                            " based on configured SSID",
2488                            MAC2STR(bssid));
2489                 return ieee80211_sta_join_ibss(wpa_s, bss);
2490         }
2491 #ifdef IEEE80211_IBSS_DEBUG
2492         wpa_printf(MSG_DEBUG, "   did not try to join ibss");
2493 #endif /* IEEE80211_IBSS_DEBUG */
2494
2495         /* Selected IBSS not found in current scan results - try to scan */
2496         os_get_time(&now);
2497 #if 0 /* FIX */
2498         if (wpa_s->mlme.state == IEEE80211_IBSS_JOINED &&
2499             !ieee80211_sta_active_ibss(wpa_s)) {
2500                 ieee80211_reschedule_timer(wpa_s,
2501                                            IEEE80211_IBSS_MERGE_INTERVAL);
2502         } else if (time_after(jiffies, wpa_s->mlme.last_scan_completed +
2503                               IEEE80211_SCAN_INTERVAL)) {
2504                 wpa_printf(MSG_DEBUG, "MLME: Trigger new scan to find an IBSS "
2505                            "to join");
2506                 return ieee80211_sta_req_scan(wpa_s->mlme.ssid,
2507                                               wpa_s->mlme.ssid_len);
2508         } else if (wpa_s->mlme.state != IEEE80211_IBSS_JOINED) {
2509                 int interval = IEEE80211_SCAN_INTERVAL;
2510
2511                 if (time_after(jiffies, wpa_s->mlme.ibss_join_req +
2512                                IEEE80211_IBSS_JOIN_TIMEOUT)) {
2513                         if (wpa_s->mlme.create_ibss &&
2514                             ieee80211_ibss_allowed(wpa_s))
2515                                 return ieee80211_sta_create_ibss(wpa_s);
2516                         if (wpa_s->mlme.create_ibss) {
2517                                 wpa_printf(MSG_DEBUG, "MLME: IBSS not allowed "
2518                                            "on the configured channel %d "
2519                                            "(%d MHz)",
2520                                            local->conf.channel,
2521                                            local->conf.freq);
2522                         }
2523
2524                         /* No IBSS found - decrease scan interval and continue
2525                          * scanning. */
2526                         interval = IEEE80211_SCAN_INTERVAL_SLOW;
2527                 }
2528
2529                 wpa_s->mlme.state = IEEE80211_IBSS_SEARCH;
2530                 ieee80211_reschedule_timer(wpa_s, interval);
2531                 return 0;
2532         }
2533 #endif
2534
2535         return 0;
2536 }
2537
2538
2539 int ieee80211_sta_get_ssid(struct wpa_supplicant *wpa_s, u8 *ssid,
2540                            size_t *len)
2541 {
2542         os_memcpy(ssid, wpa_s->mlme.ssid, wpa_s->mlme.ssid_len);
2543         *len = wpa_s->mlme.ssid_len;
2544         return 0;
2545 }
2546
2547
2548 int ieee80211_sta_associate(struct wpa_supplicant *wpa_s,
2549                             struct wpa_driver_associate_params *params)
2550 {
2551         struct ieee80211_sta_bss *bss;
2552         int bssid_changed;
2553
2554         wpa_s->mlme.bssid_set = 0;
2555         wpa_s->mlme.freq = params->freq;
2556         if (params->bssid) {
2557                 bssid_changed = os_memcmp(wpa_s->bssid, params->bssid,
2558                                           ETH_ALEN);
2559                 os_memcpy(wpa_s->bssid, params->bssid, ETH_ALEN);
2560                 if (bssid_changed)
2561                         wpas_notify_bssid_changed(wpa_s);
2562
2563                 if (!is_zero_ether_addr(params->bssid))
2564                         wpa_s->mlme.bssid_set = 1;
2565                 bss = ieee80211_bss_get(wpa_s, wpa_s->bssid);
2566                 if (bss) {
2567                         wpa_s->mlme.phymode = bss->hw_mode;
2568                         wpa_s->mlme.channel = bss->channel;
2569                         wpa_s->mlme.freq = bss->freq;
2570                 }
2571         }
2572
2573 #if 0 /* FIX */
2574         /* TODO: This should always be done for IBSS, even if IEEE80211_QOS is
2575          * not defined. */
2576         if (local->hw->conf_tx) {
2577                 struct ieee80211_tx_queue_params qparam;
2578                 int i;
2579
2580                 os_memset(&qparam, 0, sizeof(qparam));
2581                 /* TODO: are these ok defaults for all hw_modes? */
2582                 qparam.aifs = 2;
2583                 qparam.cw_min =
2584                         local->conf.phymode == MODE_IEEE80211B ? 31 : 15;
2585                 qparam.cw_max = 1023;
2586                 qparam.burst_time = 0;
2587                 for (i = IEEE80211_TX_QUEUE_DATA0; i < NUM_TX_DATA_QUEUES; i++)
2588                 {
2589                         local->hw->conf_tx(wpa_s, i + IEEE80211_TX_QUEUE_DATA0,
2590                                            &qparam);
2591                 }
2592                 /* IBSS uses different parameters for Beacon sending */
2593                 qparam.cw_min++;
2594                 qparam.cw_min *= 2;
2595                 qparam.cw_min--;
2596                 local->hw->conf_tx(wpa_s, IEEE80211_TX_QUEUE_BEACON, &qparam);
2597         }
2598 #endif
2599
2600         if (wpa_s->mlme.ssid_len != params->ssid_len ||
2601             os_memcmp(wpa_s->mlme.ssid, params->ssid, params->ssid_len) != 0)
2602                 wpa_s->mlme.prev_bssid_set = 0;
2603         os_memcpy(wpa_s->mlme.ssid, params->ssid, params->ssid_len);
2604         os_memset(wpa_s->mlme.ssid + params->ssid_len, 0,
2605                   MAX_SSID_LEN - params->ssid_len);
2606         wpa_s->mlme.ssid_len = params->ssid_len;
2607         wpa_s->mlme.ssid_set = 1;
2608
2609         os_free(wpa_s->mlme.extra_ie);
2610         if (params->wpa_ie == NULL || params->wpa_ie_len == 0) {
2611                 wpa_s->mlme.extra_ie = NULL;
2612                 wpa_s->mlme.extra_ie_len = 0;
2613         } else {
2614                 wpa_s->mlme.extra_ie = os_malloc(params->wpa_ie_len);
2615                 if (wpa_s->mlme.extra_ie == NULL) {
2616                         wpa_s->mlme.extra_ie_len = 0;
2617                         return -1;
2618                 }
2619                 os_memcpy(wpa_s->mlme.extra_ie, params->wpa_ie,
2620                           params->wpa_ie_len);
2621                 wpa_s->mlme.extra_ie_len = params->wpa_ie_len;
2622         }
2623
2624         wpa_s->mlme.key_mgmt = params->key_mgmt_suite;
2625
2626         ieee80211_sta_set_channel(wpa_s, wpa_s->mlme.phymode,
2627                                   wpa_s->mlme.channel, wpa_s->mlme.freq);
2628
2629         if (params->mode == WPAS_MODE_IBSS && !wpa_s->mlme.bssid_set) {
2630                 os_get_time(&wpa_s->mlme.ibss_join_req);
2631                 wpa_s->mlme.state = IEEE80211_IBSS_SEARCH;
2632                 return ieee80211_sta_find_ibss(wpa_s);
2633         }
2634
2635         if (wpa_s->mlme.bssid_set)
2636                 ieee80211_sta_new_auth(wpa_s);
2637
2638         return 0;
2639 }
2640
2641
2642 static void ieee80211_sta_save_oper_chan(struct wpa_supplicant *wpa_s)
2643 {
2644         wpa_s->mlme.scan_oper_channel = wpa_s->mlme.channel;
2645         wpa_s->mlme.scan_oper_freq = wpa_s->mlme.freq;
2646         wpa_s->mlme.scan_oper_phymode = wpa_s->mlme.phymode;
2647 }
2648
2649
2650 static int ieee80211_sta_restore_oper_chan(struct wpa_supplicant *wpa_s)
2651 {
2652         wpa_s->mlme.channel = wpa_s->mlme.scan_oper_channel;
2653         wpa_s->mlme.freq = wpa_s->mlme.scan_oper_freq;
2654         wpa_s->mlme.phymode = wpa_s->mlme.scan_oper_phymode;
2655         if (wpa_s->mlme.freq == 0)
2656                 return 0;
2657         return ieee80211_sta_set_channel(wpa_s, wpa_s->mlme.phymode,
2658                                          wpa_s->mlme.channel,
2659                                          wpa_s->mlme.freq);
2660 }
2661
2662
2663 static int ieee80211_active_scan(struct wpa_supplicant *wpa_s)
2664 {
2665         size_t m;
2666         int c;
2667
2668         for (m = 0; m < wpa_s->mlme.num_modes; m++) {
2669                 struct hostapd_hw_modes *mode = &wpa_s->mlme.modes[m];
2670                 if ((int) mode->mode != (int) wpa_s->mlme.phymode)
2671                         continue;
2672                 for (c = 0; c < mode->num_channels; c++) {
2673                         struct hostapd_channel_data *chan = &mode->channels[c];
2674                         if (!(chan->flag & HOSTAPD_CHAN_DISABLED) &&
2675                             chan->chan == wpa_s->mlme.channel) {
2676                                 if (!(chan->flag & HOSTAPD_CHAN_PASSIVE_SCAN))
2677                                         return 1;
2678                                 break;
2679                         }
2680                 }
2681         }
2682
2683         return 0;
2684 }
2685
2686
2687 static void ieee80211_sta_scan_timer(void *eloop_ctx, void *timeout_ctx)
2688 {
2689         struct wpa_supplicant *wpa_s = eloop_ctx;
2690         struct hostapd_hw_modes *mode;
2691         struct hostapd_channel_data *chan;
2692         int skip = 0;
2693         int timeout = 0;
2694         struct wpa_ssid *ssid = wpa_s->current_ssid;
2695         int adhoc;
2696
2697         if (!wpa_s->mlme.sta_scanning || wpa_s->mlme.modes == NULL)
2698                 return;
2699
2700         adhoc = ssid && ssid->mode == 1;
2701
2702         switch (wpa_s->mlme.scan_state) {
2703         case SCAN_SET_CHANNEL:
2704                 mode = &wpa_s->mlme.modes[wpa_s->mlme.scan_hw_mode_idx];
2705                 if (wpa_s->mlme.scan_hw_mode_idx >=
2706                     (int) wpa_s->mlme.num_modes ||
2707                     (wpa_s->mlme.scan_hw_mode_idx + 1 ==
2708                      (int) wpa_s->mlme.num_modes
2709                      && wpa_s->mlme.scan_channel_idx >= mode->num_channels)) {
2710                         if (ieee80211_sta_restore_oper_chan(wpa_s)) {
2711                                 wpa_printf(MSG_DEBUG, "MLME: failed to "
2712                                            "restore operational channel after "
2713                                            "scan");
2714                         }
2715                         wpa_printf(MSG_DEBUG, "MLME: scan completed");
2716                         wpa_s->mlme.sta_scanning = 0;
2717                         os_get_time(&wpa_s->mlme.last_scan_completed);
2718                         wpa_supplicant_event(wpa_s, EVENT_SCAN_RESULTS, NULL);
2719                         if (adhoc) {
2720                                 if (!wpa_s->mlme.bssid_set ||
2721                                     (wpa_s->mlme.state ==
2722                                      IEEE80211_IBSS_JOINED &&
2723                                      !ieee80211_sta_active_ibss(wpa_s)))
2724                                         ieee80211_sta_find_ibss(wpa_s);
2725                         }
2726                         return;
2727                 }
2728                 skip = !(wpa_s->mlme.hw_modes & (1 << mode->mode));
2729                 chan = &mode->channels[wpa_s->mlme.scan_channel_idx];
2730                 if ((chan->flag & HOSTAPD_CHAN_DISABLED) ||
2731                     (adhoc && (chan->flag & HOSTAPD_CHAN_NO_IBSS)) ||
2732                     (wpa_s->mlme.hw_modes & (1 << HOSTAPD_MODE_IEEE80211G) &&
2733                      mode->mode == HOSTAPD_MODE_IEEE80211B &&
2734                      wpa_s->mlme.scan_skip_11b))
2735                         skip = 1;
2736                 if (!skip && wpa_s->mlme.scan_freqs) {
2737                         int i, found = 0;
2738                         for (i = 0; wpa_s->mlme.scan_freqs[i]; i++) {
2739                                 if (wpa_s->mlme.scan_freqs[i] == chan->freq) {
2740                                         found = 1;
2741                                         break;
2742                                 }
2743                         }
2744                         if (!found)
2745                                 skip = 1;
2746                 }
2747
2748                 if (!skip) {
2749                         wpa_printf(MSG_MSGDUMP,
2750                                    "MLME: scan channel %d (%d MHz)",
2751                                    chan->chan, chan->freq);
2752
2753                         wpa_s->mlme.channel = chan->chan;
2754                         wpa_s->mlme.freq = chan->freq;
2755                         wpa_s->mlme.phymode = mode->mode;
2756                         if (ieee80211_sta_set_channel(wpa_s, mode->mode,
2757                                                       chan->chan, chan->freq))
2758                         {
2759                                 wpa_printf(MSG_DEBUG, "MLME: failed to set "
2760                                            "channel %d (%d MHz) for scan",
2761                                            chan->chan, chan->freq);
2762                                 skip = 1;
2763                         }
2764                 }
2765
2766                 wpa_s->mlme.scan_channel_idx++;
2767                 if (wpa_s->mlme.scan_channel_idx >=
2768                     wpa_s->mlme.modes[wpa_s->mlme.scan_hw_mode_idx].
2769                     num_channels) {
2770                         wpa_s->mlme.scan_hw_mode_idx++;
2771                         wpa_s->mlme.scan_channel_idx = 0;
2772                 }
2773
2774                 if (skip) {
2775                         timeout = 0;
2776                         break;
2777                 }
2778
2779                 timeout = IEEE80211_PROBE_DELAY;
2780                 wpa_s->mlme.scan_state = SCAN_SEND_PROBE;
2781                 break;
2782         case SCAN_SEND_PROBE:
2783                 if (ieee80211_active_scan(wpa_s)) {
2784                         ieee80211_send_probe_req(wpa_s, NULL,
2785                                                  wpa_s->mlme.scan_ssid,
2786                                                  wpa_s->mlme.scan_ssid_len);
2787                         timeout = IEEE80211_CHANNEL_TIME;
2788                 } else {
2789                         timeout = IEEE80211_PASSIVE_CHANNEL_TIME;
2790                 }
2791                 wpa_s->mlme.scan_state = SCAN_SET_CHANNEL;
2792                 break;
2793         }
2794
2795         eloop_register_timeout(timeout / 1000, 1000 * (timeout % 1000),
2796                                ieee80211_sta_scan_timer, wpa_s, NULL);
2797 }
2798
2799
2800 int ieee80211_sta_req_scan(struct wpa_supplicant *wpa_s,
2801                            struct wpa_driver_scan_params *params)
2802 {
2803         const u8 *ssid = params->ssids[0].ssid;
2804         size_t ssid_len = params->ssids[0].ssid_len;
2805
2806         if (ssid_len > MAX_SSID_LEN)
2807                 return -1;
2808
2809         /* MLME-SCAN.request (page 118)  page 144 (11.1.3.1)
2810          * BSSType: INFRASTRUCTURE, INDEPENDENT, ANY_BSS
2811          * BSSID: MACAddress
2812          * SSID
2813          * ScanType: ACTIVE, PASSIVE
2814          * ProbeDelay: delay (in microseconds) to be used prior to transmitting
2815          *    a Probe frame during active scanning
2816          * ChannelList
2817          * MinChannelTime (>= ProbeDelay), in TU
2818          * MaxChannelTime: (>= MinChannelTime), in TU
2819          */
2820
2821          /* MLME-SCAN.confirm
2822           * BSSDescriptionSet
2823           * ResultCode: SUCCESS, INVALID_PARAMETERS
2824          */
2825
2826         /* TODO: if assoc, move to power save mode for the duration of the
2827          * scan */
2828
2829         if (wpa_s->mlme.sta_scanning)
2830                 return -1;
2831
2832         wpa_printf(MSG_DEBUG, "MLME: starting scan");
2833
2834         ieee80211_sta_set_probe_req_ie(wpa_s, params->extra_ies,
2835                                        params->extra_ies_len);
2836
2837         os_free(wpa_s->mlme.scan_freqs);
2838         if (params->freqs) {
2839                 int i;
2840                 for (i = 0; params->freqs[i]; i++)
2841                         ;
2842                 wpa_s->mlme.scan_freqs = os_malloc((i + 1) * sizeof(int));
2843                 if (wpa_s->mlme.scan_freqs)
2844                         os_memcpy(wpa_s->mlme.scan_freqs, params->freqs,
2845                                   (i + 1) * sizeof(int));
2846         } else
2847                 wpa_s->mlme.scan_freqs = NULL;
2848
2849         ieee80211_sta_save_oper_chan(wpa_s);
2850
2851         wpa_s->mlme.sta_scanning = 1;
2852         /* TODO: stop TX queue? */
2853
2854         if (ssid) {
2855                 wpa_s->mlme.scan_ssid_len = ssid_len;
2856                 os_memcpy(wpa_s->mlme.scan_ssid, ssid, ssid_len);
2857         } else
2858                 wpa_s->mlme.scan_ssid_len = 0;
2859         wpa_s->mlme.scan_skip_11b = 1; /* FIX: clear this is 11g is not
2860                                         * supported */
2861         wpa_s->mlme.scan_state = SCAN_SET_CHANNEL;
2862         wpa_s->mlme.scan_hw_mode_idx = 0;
2863         wpa_s->mlme.scan_channel_idx = 0;
2864         eloop_register_timeout(0, 1, ieee80211_sta_scan_timer, wpa_s, NULL);
2865
2866         return 0;
2867 }
2868
2869
2870 struct wpa_scan_results *
2871 ieee80211_sta_get_scan_results(struct wpa_supplicant *wpa_s)
2872 {
2873         size_t ap_num = 0;
2874         struct wpa_scan_results *res;
2875         struct wpa_scan_res *r;
2876         struct ieee80211_sta_bss *bss;
2877
2878         res = os_zalloc(sizeof(*res));
2879         for (bss = wpa_s->mlme.sta_bss_list; bss; bss = bss->next)
2880                 ap_num++;
2881         res->res = os_zalloc(ap_num * sizeof(struct wpa_scan_res *));
2882         if (res->res == NULL) {
2883                 os_free(res);
2884                 return NULL;
2885         }
2886
2887         for (bss = wpa_s->mlme.sta_bss_list; bss; bss = bss->next) {
2888                 r = os_zalloc(sizeof(*r) + bss->ie_len);
2889                 if (r == NULL)
2890                         break;
2891                 os_memcpy(r->bssid, bss->bssid, ETH_ALEN);
2892                 r->freq = bss->freq;
2893                 r->beacon_int = bss->beacon_int;
2894                 r->caps = bss->capability;
2895                 r->level = bss->rssi;
2896                 r->tsf = bss->timestamp;
2897                 if (bss->ie) {
2898                         r->ie_len = bss->ie_len;
2899                         os_memcpy(r + 1, bss->ie, bss->ie_len);
2900                 }
2901
2902                 res->res[res->num++] = r;
2903         }
2904
2905         return res;
2906 }
2907
2908
2909 #if 0 /* FIX */
2910 struct sta_info * ieee80211_ibss_add_sta(struct wpa_supplicant *wpa_s,
2911                                          struct sk_buff *skb, u8 *bssid,
2912                                          u8 *addr)
2913 {
2914         struct ieee80211_local *local = dev->priv;
2915         struct list_head *ptr;
2916         struct sta_info *sta;
2917         struct wpa_supplicant *sta_dev = NULL;
2918
2919         /* TODO: Could consider removing the least recently used entry and
2920          * allow new one to be added. */
2921         if (local->num_sta >= IEEE80211_IBSS_MAX_STA_ENTRIES) {
2922                 if (net_ratelimit()) {
2923                         wpa_printf(MSG_DEBUG, "MLME: No room for a new IBSS "
2924                                    "STA entry " MACSTR, MAC2STR(addr));
2925                 }
2926                 return NULL;
2927         }
2928
2929         spin_lock_bh(&local->sub_if_lock);
2930         list_for_each(ptr, &local->sub_if_list) {
2931                 sdata = list_entry(ptr, struct ieee80211_sub_if_data, list);
2932                 if (sdata->type == IEEE80211_SUB_IF_TYPE_STA &&
2933                     os_memcmp(bssid, sdata->u.sta.bssid, ETH_ALEN) == 0) {
2934                         sta_dev = sdata->dev;
2935                         break;
2936                 }
2937         }
2938         spin_unlock_bh(&local->sub_if_lock);
2939
2940         if (sta_dev == NULL)
2941                 return NULL;
2942
2943         wpa_printf(MSG_DEBUG, "MLME: Adding new IBSS station " MACSTR
2944                    " (dev=%s)", MAC2STR(addr), sta_dev->name);
2945
2946         sta = sta_info_add(wpa_s, addr);
2947         if (sta == NULL) {
2948                 return NULL;
2949         }
2950
2951         sta->dev = sta_dev;
2952         sta->supp_rates = wpa_s->mlme.supp_rates_bits;
2953
2954         rate_control_rate_init(local, sta);
2955
2956         return sta; /* caller will call sta_info_release() */
2957 }
2958 #endif
2959
2960
2961 int ieee80211_sta_deauthenticate(struct wpa_supplicant *wpa_s, u16 reason)
2962 {
2963         wpa_printf(MSG_DEBUG, "MLME: deauthenticate(reason=%d)", reason);
2964
2965         ieee80211_send_deauth(wpa_s, reason);
2966         ieee80211_set_associated(wpa_s, 0);
2967         return 0;
2968 }
2969
2970
2971 int ieee80211_sta_disassociate(struct wpa_supplicant *wpa_s, u16 reason)
2972 {
2973         wpa_printf(MSG_DEBUG, "MLME: disassociate(reason=%d)", reason);
2974
2975         if (!wpa_s->mlme.associated)
2976                 return -1;
2977
2978         ieee80211_send_disassoc(wpa_s, reason);
2979         ieee80211_set_associated(wpa_s, 0);
2980         return 0;
2981 }
2982
2983
2984 void ieee80211_sta_rx(struct wpa_supplicant *wpa_s, const u8 *buf, size_t len,
2985                       struct ieee80211_rx_status *rx_status)
2986 {
2987         struct ieee80211_mgmt *mgmt;
2988         u16 fc;
2989         const u8 *pos;
2990
2991         /* wpa_hexdump(MSG_MSGDUMP, "MLME: Received frame", buf, len); */
2992
2993         if (wpa_s->mlme.sta_scanning) {
2994                 ieee80211_sta_rx_scan(wpa_s, buf, len, rx_status);
2995                 return;
2996         }
2997
2998         if (len < 24)
2999                 return;
3000
3001         mgmt = (struct ieee80211_mgmt *) buf;
3002         fc = le_to_host16(mgmt->frame_control);
3003
3004         if (WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_MGMT)
3005                 ieee80211_sta_rx_mgmt(wpa_s, buf, len, rx_status);
3006         else if (WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_DATA) {
3007                 if ((fc & (WLAN_FC_TODS | WLAN_FC_FROMDS)) !=
3008                     WLAN_FC_FROMDS)
3009                         return;
3010                 /* mgmt->sa is actually BSSID for FromDS data frames */
3011                 if (os_memcmp(mgmt->sa, wpa_s->bssid, ETH_ALEN) != 0)
3012                         return;
3013                 /* Skip IEEE 802.11 and LLC headers */
3014                 pos = buf + 24 + 6;
3015                 if (WPA_GET_BE16(pos) != ETH_P_EAPOL)
3016                         return;
3017                 pos += 2;
3018                 /* mgmt->bssid is actually BSSID for SA data frames */
3019                 wpa_supplicant_rx_eapol(wpa_s, mgmt->bssid,
3020                                         pos, buf + len - pos);
3021         }
3022 }
3023
3024
3025 void ieee80211_sta_free_hw_features(struct hostapd_hw_modes *hw_features,
3026                                     size_t num_hw_features)
3027 {
3028         size_t i;
3029
3030         if (hw_features == NULL)
3031                 return;
3032
3033         for (i = 0; i < num_hw_features; i++) {
3034                 os_free(hw_features[i].channels);
3035                 os_free(hw_features[i].rates);
3036         }
3037
3038         os_free(hw_features);
3039 }
3040
3041
3042 int ieee80211_sta_init(struct wpa_supplicant *wpa_s)
3043 {
3044         u16 num_modes, flags;
3045
3046         wpa_s->mlme.modes = wpa_drv_get_hw_feature_data(wpa_s, &num_modes,
3047                                                         &flags);
3048         if (wpa_s->mlme.modes == NULL) {
3049                 wpa_printf(MSG_ERROR, "MLME: Failed to read supported "
3050                            "channels and rates from the driver");
3051                 return -1;
3052         }
3053
3054         wpa_s->mlme.num_modes = num_modes;
3055
3056         wpa_s->mlme.hw_modes = 1 << HOSTAPD_MODE_IEEE80211A;
3057         wpa_s->mlme.hw_modes |= 1 << HOSTAPD_MODE_IEEE80211B;
3058         wpa_s->mlme.hw_modes |= 1 << HOSTAPD_MODE_IEEE80211G;
3059
3060         wpa_s->mlme.wmm_enabled = 1;
3061
3062         return 0;
3063 }
3064
3065
3066 void ieee80211_sta_deinit(struct wpa_supplicant *wpa_s)
3067 {
3068         eloop_cancel_timeout(ieee80211_sta_timer, wpa_s, NULL);
3069         eloop_cancel_timeout(ieee80211_sta_scan_timer, wpa_s, NULL);
3070         os_free(wpa_s->mlme.extra_ie);
3071         wpa_s->mlme.extra_ie = NULL;
3072         os_free(wpa_s->mlme.extra_probe_ie);
3073         wpa_s->mlme.extra_probe_ie = NULL;
3074         os_free(wpa_s->mlme.assocreq_ies);
3075         wpa_s->mlme.assocreq_ies = NULL;
3076         os_free(wpa_s->mlme.assocresp_ies);
3077         wpa_s->mlme.assocresp_ies = NULL;
3078         ieee80211_bss_list_deinit(wpa_s);
3079         ieee80211_sta_free_hw_features(wpa_s->mlme.modes,
3080                                        wpa_s->mlme.num_modes);
3081 #ifdef CONFIG_IEEE80211R
3082         os_free(wpa_s->mlme.ft_ies);
3083         wpa_s->mlme.ft_ies = NULL;
3084         wpa_s->mlme.ft_ies_len = 0;
3085 #endif /* CONFIG_IEEE80211R */
3086
3087         os_free(wpa_s->mlme.scan_freqs);
3088         wpa_s->mlme.scan_freqs = NULL;
3089 }
3090
3091
3092 #ifdef CONFIG_IEEE80211R
3093
3094 int ieee80211_sta_update_ft_ies(struct wpa_supplicant *wpa_s, const u8 *md,
3095                                 const u8 *ies, size_t ies_len)
3096 {
3097         if (md == NULL) {
3098                 wpa_printf(MSG_DEBUG, "MLME: Clear FT mobility domain");
3099                 os_memset(wpa_s->mlme.current_md, 0, MOBILITY_DOMAIN_ID_LEN);
3100         } else {
3101                 wpa_printf(MSG_DEBUG, "MLME: Update FT IEs for MD " MACSTR,
3102                            MAC2STR(md));
3103                 os_memcpy(wpa_s->mlme.current_md, md, MOBILITY_DOMAIN_ID_LEN);
3104         }
3105
3106         wpa_hexdump(MSG_DEBUG, "MLME: FT IEs", ies, ies_len);
3107         os_free(wpa_s->mlme.ft_ies);
3108         wpa_s->mlme.ft_ies = os_malloc(ies_len);
3109         if (wpa_s->mlme.ft_ies == NULL)
3110                 return -1;
3111         os_memcpy(wpa_s->mlme.ft_ies, ies, ies_len);
3112         wpa_s->mlme.ft_ies_len = ies_len;
3113
3114         return 0;
3115 }
3116
3117
3118 int ieee80211_sta_send_ft_action(struct wpa_supplicant *wpa_s, u8 action,
3119                                  const u8 *target_ap,
3120                                  const u8 *ies, size_t ies_len)
3121 {
3122         u8 *buf;
3123         size_t len;
3124         struct ieee80211_mgmt *mgmt;
3125         int res;
3126
3127         /*
3128          * Action frame payload:
3129          * Category[1] = 6 (Fast BSS Transition)
3130          * Action[1] = 1 (Fast BSS Transition Request)
3131          * STA Address
3132          * Target AP Address
3133          * FT IEs
3134          */
3135
3136         buf = os_zalloc(sizeof(*mgmt) + ies_len);
3137         if (buf == NULL) {
3138                 wpa_printf(MSG_DEBUG, "MLME: Failed to allocate buffer for "
3139                            "FT action frame");
3140                 return -1;
3141         }
3142
3143         mgmt = (struct ieee80211_mgmt *) buf;
3144         len = 24;
3145         os_memcpy(mgmt->da, wpa_s->bssid, ETH_ALEN);
3146         os_memcpy(mgmt->sa, wpa_s->own_addr, ETH_ALEN);
3147         os_memcpy(mgmt->bssid, wpa_s->bssid, ETH_ALEN);
3148         mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
3149                                            WLAN_FC_STYPE_ACTION);
3150         mgmt->u.action.category = WLAN_ACTION_FT;
3151         mgmt->u.action.u.ft_action_req.action = action;
3152         os_memcpy(mgmt->u.action.u.ft_action_req.sta_addr, wpa_s->own_addr,
3153                   ETH_ALEN);
3154         os_memcpy(mgmt->u.action.u.ft_action_req.target_ap_addr, target_ap,
3155                   ETH_ALEN);
3156         os_memcpy(mgmt->u.action.u.ft_action_req.variable, ies, ies_len);
3157         len += 1 + sizeof(mgmt->u.action.u.ft_action_req) + ies_len;
3158
3159         wpa_printf(MSG_DEBUG, "MLME: Send FT Action Frame: Action=%d "
3160                    "Target AP=" MACSTR " body_len=%lu",
3161                    action, MAC2STR(target_ap), (unsigned long) ies_len);
3162
3163         res = ieee80211_sta_tx(wpa_s, buf, len);
3164         os_free(buf);
3165
3166         return res;
3167 }
3168
3169 #endif /* CONFIG_IEEE80211R */
3170
3171
3172 static int ieee80211_sta_set_probe_req_ie(struct wpa_supplicant *wpa_s,
3173                                           const u8 *ies, size_t ies_len)
3174 {
3175         os_free(wpa_s->mlme.extra_probe_ie);
3176         wpa_s->mlme.extra_probe_ie = NULL;
3177         wpa_s->mlme.extra_probe_ie_len = 0;
3178
3179         if (ies == NULL)
3180                 return 0;
3181
3182         wpa_s->mlme.extra_probe_ie = os_malloc(ies_len);
3183         if (wpa_s->mlme.extra_probe_ie == NULL)
3184                 return -1;
3185
3186         os_memcpy(wpa_s->mlme.extra_probe_ie, ies, ies_len);
3187         wpa_s->mlme.extra_probe_ie_len = ies_len;
3188
3189         return 0;
3190 }