Make a code path easier for static analyzers to understand
[mech_eap.git] / src / ap / beacon.c
1 /*
2  * hostapd / IEEE 802.11 Management: Beacon and Probe Request/Response
3  * Copyright (c) 2002-2004, Instant802 Networks, Inc.
4  * Copyright (c) 2005-2006, Devicescape Software, Inc.
5  * Copyright (c) 2008-2012, Jouni Malinen <j@w1.fi>
6  *
7  * This software may be distributed under the terms of the BSD license.
8  * See README for more details.
9  */
10
11 #include "utils/includes.h"
12
13 #ifndef CONFIG_NATIVE_WINDOWS
14
15 #include "utils/common.h"
16 #include "common/ieee802_11_defs.h"
17 #include "common/ieee802_11_common.h"
18 #include "wps/wps_defs.h"
19 #include "p2p/p2p.h"
20 #include "hostapd.h"
21 #include "ieee802_11.h"
22 #include "wpa_auth.h"
23 #include "wmm.h"
24 #include "ap_config.h"
25 #include "sta_info.h"
26 #include "p2p_hostapd.h"
27 #include "ap_drv_ops.h"
28 #include "beacon.h"
29 #include "hs20.h"
30 #include "dfs.h"
31
32
33 #ifdef NEED_AP_MLME
34
35 static u8 * hostapd_eid_bss_load(struct hostapd_data *hapd, u8 *eid, size_t len)
36 {
37 #ifdef CONFIG_TESTING_OPTIONS
38         if (hapd->conf->bss_load_test_set) {
39                 if (2 + 5 > len)
40                         return eid;
41                 *eid++ = WLAN_EID_BSS_LOAD;
42                 *eid++ = 5;
43                 os_memcpy(eid, hapd->conf->bss_load_test, 5);
44                 eid += 5;
45         }
46 #endif /* CONFIG_TESTING_OPTIONS */
47         return eid;
48 }
49
50
51 static u8 ieee802_11_erp_info(struct hostapd_data *hapd)
52 {
53         u8 erp = 0;
54
55         if (hapd->iface->current_mode == NULL ||
56             hapd->iface->current_mode->mode != HOSTAPD_MODE_IEEE80211G)
57                 return 0;
58
59         if (hapd->iface->olbc)
60                 erp |= ERP_INFO_USE_PROTECTION;
61         if (hapd->iface->num_sta_non_erp > 0) {
62                 erp |= ERP_INFO_NON_ERP_PRESENT |
63                         ERP_INFO_USE_PROTECTION;
64         }
65         if (hapd->iface->num_sta_no_short_preamble > 0 ||
66             hapd->iconf->preamble == LONG_PREAMBLE)
67                 erp |= ERP_INFO_BARKER_PREAMBLE_MODE;
68
69         return erp;
70 }
71
72
73 static u8 * hostapd_eid_ds_params(struct hostapd_data *hapd, u8 *eid)
74 {
75         *eid++ = WLAN_EID_DS_PARAMS;
76         *eid++ = 1;
77         *eid++ = hapd->iconf->channel;
78         return eid;
79 }
80
81
82 static u8 * hostapd_eid_erp_info(struct hostapd_data *hapd, u8 *eid)
83 {
84         if (hapd->iface->current_mode == NULL ||
85             hapd->iface->current_mode->mode != HOSTAPD_MODE_IEEE80211G)
86                 return eid;
87
88         /* Set NonERP_present and use_protection bits if there
89          * are any associated NonERP stations. */
90         /* TODO: use_protection bit can be set to zero even if
91          * there are NonERP stations present. This optimization
92          * might be useful if NonERP stations are "quiet".
93          * See 802.11g/D6 E-1 for recommended practice.
94          * In addition, Non ERP present might be set, if AP detects Non ERP
95          * operation on other APs. */
96
97         /* Add ERP Information element */
98         *eid++ = WLAN_EID_ERP_INFO;
99         *eid++ = 1;
100         *eid++ = ieee802_11_erp_info(hapd);
101
102         return eid;
103 }
104
105
106 static u8 * hostapd_eid_pwr_constraint(struct hostapd_data *hapd, u8 *eid)
107 {
108         u8 *pos = eid;
109         u8 local_pwr_constraint = 0;
110         int dfs;
111
112         if (hapd->iface->current_mode == NULL ||
113             hapd->iface->current_mode->mode != HOSTAPD_MODE_IEEE80211A)
114                 return eid;
115
116         /*
117          * There is no DFS support and power constraint was not directly
118          * requested by config option.
119          */
120         if (!hapd->iconf->ieee80211h &&
121             hapd->iconf->local_pwr_constraint == -1)
122                 return eid;
123
124         /* Check if DFS is required by regulatory. */
125         dfs = hostapd_is_dfs_required(hapd->iface);
126         if (dfs < 0) {
127                 wpa_printf(MSG_WARNING, "Failed to check if DFS is required; ret=%d",
128                            dfs);
129                 dfs = 0;
130         }
131
132         if (dfs == 0 && hapd->iconf->local_pwr_constraint == -1)
133                 return eid;
134
135         /*
136          * ieee80211h (DFS) is enabled so Power Constraint element shall
137          * be added when running on DFS channel whenever local_pwr_constraint
138          * is configured or not. In order to meet regulations when TPC is not
139          * implemented using a transmit power that is below the legal maximum
140          * (including any mitigation factor) should help. In this case,
141          * indicate 3 dB below maximum allowed transmit power.
142          */
143         if (hapd->iconf->local_pwr_constraint == -1)
144                 local_pwr_constraint = 3;
145
146         /*
147          * A STA that is not an AP shall use a transmit power less than or
148          * equal to the local maximum transmit power level for the channel.
149          * The local maximum transmit power can be calculated from the formula:
150          * local max TX pwr = max TX pwr - local pwr constraint
151          * Where max TX pwr is maximum transmit power level specified for
152          * channel in Country element and local pwr constraint is specified
153          * for channel in this Power Constraint element.
154          */
155
156         /* Element ID */
157         *pos++ = WLAN_EID_PWR_CONSTRAINT;
158         /* Length */
159         *pos++ = 1;
160         /* Local Power Constraint */
161         if (local_pwr_constraint)
162                 *pos++ = local_pwr_constraint;
163         else
164                 *pos++ = hapd->iconf->local_pwr_constraint;
165
166         return pos;
167 }
168
169
170 static u8 * hostapd_eid_country_add(u8 *pos, u8 *end, int chan_spacing,
171                                     struct hostapd_channel_data *start,
172                                     struct hostapd_channel_data *prev)
173 {
174         if (end - pos < 3)
175                 return pos;
176
177         /* first channel number */
178         *pos++ = start->chan;
179         /* number of channels */
180         *pos++ = (prev->chan - start->chan) / chan_spacing + 1;
181         /* maximum transmit power level */
182         *pos++ = start->max_tx_power;
183
184         return pos;
185 }
186
187
188 static u8 * hostapd_eid_country(struct hostapd_data *hapd, u8 *eid,
189                                 int max_len)
190 {
191         u8 *pos = eid;
192         u8 *end = eid + max_len;
193         int i;
194         struct hostapd_hw_modes *mode;
195         struct hostapd_channel_data *start, *prev;
196         int chan_spacing = 1;
197
198         if (!hapd->iconf->ieee80211d || max_len < 6 ||
199             hapd->iface->current_mode == NULL)
200                 return eid;
201
202         *pos++ = WLAN_EID_COUNTRY;
203         pos++; /* length will be set later */
204         os_memcpy(pos, hapd->iconf->country, 3); /* e.g., 'US ' */
205         pos += 3;
206
207         mode = hapd->iface->current_mode;
208         if (mode->mode == HOSTAPD_MODE_IEEE80211A)
209                 chan_spacing = 4;
210
211         start = prev = NULL;
212         for (i = 0; i < mode->num_channels; i++) {
213                 struct hostapd_channel_data *chan = &mode->channels[i];
214                 if (chan->flag & HOSTAPD_CHAN_DISABLED)
215                         continue;
216                 if (start && prev &&
217                     prev->chan + chan_spacing == chan->chan &&
218                     start->max_tx_power == chan->max_tx_power) {
219                         prev = chan;
220                         continue; /* can use same entry */
221                 }
222
223                 if (start && prev) {
224                         pos = hostapd_eid_country_add(pos, end, chan_spacing,
225                                                       start, prev);
226                         start = NULL;
227                 }
228
229                 /* Start new group */
230                 start = prev = chan;
231         }
232
233         if (start) {
234                 pos = hostapd_eid_country_add(pos, end, chan_spacing,
235                                               start, prev);
236         }
237
238         if ((pos - eid) & 1) {
239                 if (end - pos < 1)
240                         return eid;
241                 *pos++ = 0; /* pad for 16-bit alignment */
242         }
243
244         eid[1] = (pos - eid) - 2;
245
246         return pos;
247 }
248
249
250 static u8 * hostapd_eid_wpa(struct hostapd_data *hapd, u8 *eid, size_t len)
251 {
252         const u8 *ie;
253         size_t ielen;
254
255         ie = wpa_auth_get_wpa_ie(hapd->wpa_auth, &ielen);
256         if (ie == NULL || ielen > len)
257                 return eid;
258
259         os_memcpy(eid, ie, ielen);
260         return eid + ielen;
261 }
262
263
264 static u8 * hostapd_eid_csa(struct hostapd_data *hapd, u8 *eid)
265 {
266         u8 chan;
267
268         if (!hapd->iface->cs_freq_params.freq)
269                 return eid;
270
271         if (ieee80211_freq_to_chan(hapd->iface->cs_freq_params.freq, &chan) ==
272             NUM_HOSTAPD_MODES)
273                 return eid;
274
275         *eid++ = WLAN_EID_CHANNEL_SWITCH;
276         *eid++ = 3;
277         *eid++ = hapd->iface->cs_block_tx;
278         *eid++ = chan;
279         *eid++ = hapd->iface->cs_count;
280
281         return eid;
282 }
283
284
285 static u8 * hostapd_eid_secondary_channel(struct hostapd_data *hapd, u8 *eid)
286 {
287         u8 sec_ch;
288
289         if (!hapd->iface->cs_freq_params.sec_channel_offset)
290                 return eid;
291
292         if (hapd->iface->cs_freq_params.sec_channel_offset == -1)
293                 sec_ch = HT_INFO_HT_PARAM_SECONDARY_CHNL_BELOW;
294         else if (hapd->iface->cs_freq_params.sec_channel_offset == 1)
295                 sec_ch = HT_INFO_HT_PARAM_SECONDARY_CHNL_ABOVE;
296         else
297                 return eid;
298
299         *eid++ = WLAN_EID_SECONDARY_CHANNEL_OFFSET;
300         *eid++ = 1;
301         *eid++ = sec_ch;
302
303         return eid;
304 }
305
306
307 static u8 * hostapd_add_csa_elems(struct hostapd_data *hapd, u8 *pos,
308                                   u8 *start, unsigned int *csa_counter_off)
309 {
310         u8 *old_pos = pos;
311
312         if (!csa_counter_off)
313                 return pos;
314
315         *csa_counter_off = 0;
316         pos = hostapd_eid_csa(hapd, pos);
317
318         if (pos != old_pos) {
319                 /* save an offset to the counter - should be last byte */
320                 *csa_counter_off = pos - start - 1;
321                 pos = hostapd_eid_secondary_channel(hapd, pos);
322         }
323
324         return pos;
325 }
326
327
328 static u8 * hostapd_gen_probe_resp(struct hostapd_data *hapd,
329                                    struct sta_info *sta,
330                                    const struct ieee80211_mgmt *req,
331                                    int is_p2p, size_t *resp_len)
332 {
333         struct ieee80211_mgmt *resp;
334         u8 *pos, *epos;
335         size_t buflen;
336
337 #define MAX_PROBERESP_LEN 768
338         buflen = MAX_PROBERESP_LEN;
339 #ifdef CONFIG_WPS
340         if (hapd->wps_probe_resp_ie)
341                 buflen += wpabuf_len(hapd->wps_probe_resp_ie);
342 #endif /* CONFIG_WPS */
343 #ifdef CONFIG_P2P
344         if (hapd->p2p_probe_resp_ie)
345                 buflen += wpabuf_len(hapd->p2p_probe_resp_ie);
346 #endif /* CONFIG_P2P */
347         if (hapd->conf->vendor_elements)
348                 buflen += wpabuf_len(hapd->conf->vendor_elements);
349         resp = os_zalloc(buflen);
350         if (resp == NULL)
351                 return NULL;
352
353         epos = ((u8 *) resp) + MAX_PROBERESP_LEN;
354
355         resp->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
356                                            WLAN_FC_STYPE_PROBE_RESP);
357         if (req)
358                 os_memcpy(resp->da, req->sa, ETH_ALEN);
359         os_memcpy(resp->sa, hapd->own_addr, ETH_ALEN);
360
361         os_memcpy(resp->bssid, hapd->own_addr, ETH_ALEN);
362         resp->u.probe_resp.beacon_int =
363                 host_to_le16(hapd->iconf->beacon_int);
364
365         /* hardware or low-level driver will setup seq_ctrl and timestamp */
366         resp->u.probe_resp.capab_info =
367                 host_to_le16(hostapd_own_capab_info(hapd, sta, 1));
368
369         pos = resp->u.probe_resp.variable;
370         *pos++ = WLAN_EID_SSID;
371         *pos++ = hapd->conf->ssid.ssid_len;
372         os_memcpy(pos, hapd->conf->ssid.ssid, hapd->conf->ssid.ssid_len);
373         pos += hapd->conf->ssid.ssid_len;
374
375         /* Supported rates */
376         pos = hostapd_eid_supp_rates(hapd, pos);
377
378         /* DS Params */
379         pos = hostapd_eid_ds_params(hapd, pos);
380
381         pos = hostapd_eid_country(hapd, pos, epos - pos);
382
383         /* Power Constraint element */
384         pos = hostapd_eid_pwr_constraint(hapd, pos);
385
386         /* ERP Information element */
387         pos = hostapd_eid_erp_info(hapd, pos);
388
389         /* Extended supported rates */
390         pos = hostapd_eid_ext_supp_rates(hapd, pos);
391
392         /* RSN, MDIE, WPA */
393         pos = hostapd_eid_wpa(hapd, pos, epos - pos);
394
395         pos = hostapd_eid_bss_load(hapd, pos, epos - pos);
396
397 #ifdef CONFIG_IEEE80211N
398         pos = hostapd_eid_ht_capabilities(hapd, pos);
399         pos = hostapd_eid_ht_operation(hapd, pos);
400 #endif /* CONFIG_IEEE80211N */
401
402         pos = hostapd_eid_ext_capab(hapd, pos);
403
404         pos = hostapd_eid_time_adv(hapd, pos);
405         pos = hostapd_eid_time_zone(hapd, pos);
406
407         pos = hostapd_eid_interworking(hapd, pos);
408         pos = hostapd_eid_adv_proto(hapd, pos);
409         pos = hostapd_eid_roaming_consortium(hapd, pos);
410
411         pos = hostapd_add_csa_elems(hapd, pos, (u8 *)resp,
412                                     &hapd->iface->cs_c_off_proberesp);
413 #ifdef CONFIG_IEEE80211AC
414         pos = hostapd_eid_vht_capabilities(hapd, pos);
415         pos = hostapd_eid_vht_operation(hapd, pos);
416 #endif /* CONFIG_IEEE80211AC */
417
418         /* Wi-Fi Alliance WMM */
419         pos = hostapd_eid_wmm(hapd, pos);
420
421 #ifdef CONFIG_WPS
422         if (hapd->conf->wps_state && hapd->wps_probe_resp_ie) {
423                 os_memcpy(pos, wpabuf_head(hapd->wps_probe_resp_ie),
424                           wpabuf_len(hapd->wps_probe_resp_ie));
425                 pos += wpabuf_len(hapd->wps_probe_resp_ie);
426         }
427 #endif /* CONFIG_WPS */
428
429 #ifdef CONFIG_P2P
430         if ((hapd->conf->p2p & P2P_ENABLED) && is_p2p &&
431             hapd->p2p_probe_resp_ie) {
432                 os_memcpy(pos, wpabuf_head(hapd->p2p_probe_resp_ie),
433                           wpabuf_len(hapd->p2p_probe_resp_ie));
434                 pos += wpabuf_len(hapd->p2p_probe_resp_ie);
435         }
436 #endif /* CONFIG_P2P */
437 #ifdef CONFIG_P2P_MANAGER
438         if ((hapd->conf->p2p & (P2P_MANAGE | P2P_ENABLED | P2P_GROUP_OWNER)) ==
439             P2P_MANAGE)
440                 pos = hostapd_eid_p2p_manage(hapd, pos);
441 #endif /* CONFIG_P2P_MANAGER */
442
443 #ifdef CONFIG_HS20
444         pos = hostapd_eid_hs20_indication(hapd, pos);
445         pos = hostapd_eid_osen(hapd, pos);
446 #endif /* CONFIG_HS20 */
447
448         if (hapd->conf->vendor_elements) {
449                 os_memcpy(pos, wpabuf_head(hapd->conf->vendor_elements),
450                           wpabuf_len(hapd->conf->vendor_elements));
451                 pos += wpabuf_len(hapd->conf->vendor_elements);
452         }
453
454         *resp_len = pos - (u8 *) resp;
455         return (u8 *) resp;
456 }
457
458
459 enum ssid_match_result {
460         NO_SSID_MATCH,
461         EXACT_SSID_MATCH,
462         WILDCARD_SSID_MATCH
463 };
464
465 static enum ssid_match_result ssid_match(struct hostapd_data *hapd,
466                                          const u8 *ssid, size_t ssid_len,
467                                          const u8 *ssid_list,
468                                          size_t ssid_list_len)
469 {
470         const u8 *pos, *end;
471         int wildcard = 0;
472
473         if (ssid_len == 0)
474                 wildcard = 1;
475         if (ssid_len == hapd->conf->ssid.ssid_len &&
476             os_memcmp(ssid, hapd->conf->ssid.ssid, ssid_len) == 0)
477                 return EXACT_SSID_MATCH;
478
479         if (ssid_list == NULL)
480                 return wildcard ? WILDCARD_SSID_MATCH : NO_SSID_MATCH;
481
482         pos = ssid_list;
483         end = ssid_list + ssid_list_len;
484         while (pos + 1 <= end) {
485                 if (pos + 2 + pos[1] > end)
486                         break;
487                 if (pos[1] == 0)
488                         wildcard = 1;
489                 if (pos[1] == hapd->conf->ssid.ssid_len &&
490                     os_memcmp(pos + 2, hapd->conf->ssid.ssid, pos[1]) == 0)
491                         return EXACT_SSID_MATCH;
492                 pos += 2 + pos[1];
493         }
494
495         return wildcard ? WILDCARD_SSID_MATCH : NO_SSID_MATCH;
496 }
497
498
499 void handle_probe_req(struct hostapd_data *hapd,
500                       const struct ieee80211_mgmt *mgmt, size_t len,
501                       int ssi_signal)
502 {
503         u8 *resp;
504         struct ieee802_11_elems elems;
505         const u8 *ie;
506         size_t ie_len;
507         struct sta_info *sta = NULL;
508         size_t i, resp_len;
509         int noack;
510         enum ssid_match_result res;
511
512         ie = mgmt->u.probe_req.variable;
513         if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.probe_req))
514                 return;
515         ie_len = len - (IEEE80211_HDRLEN + sizeof(mgmt->u.probe_req));
516
517         for (i = 0; hapd->probereq_cb && i < hapd->num_probereq_cb; i++)
518                 if (hapd->probereq_cb[i].cb(hapd->probereq_cb[i].ctx,
519                                             mgmt->sa, mgmt->da, mgmt->bssid,
520                                             ie, ie_len, ssi_signal) > 0)
521                         return;
522
523         if (!hapd->iconf->send_probe_response)
524                 return;
525
526         if (ieee802_11_parse_elems(ie, ie_len, &elems, 0) == ParseFailed) {
527                 wpa_printf(MSG_DEBUG, "Could not parse ProbeReq from " MACSTR,
528                            MAC2STR(mgmt->sa));
529                 return;
530         }
531
532         if ((!elems.ssid || !elems.supp_rates)) {
533                 wpa_printf(MSG_DEBUG, "STA " MACSTR " sent probe request "
534                            "without SSID or supported rates element",
535                            MAC2STR(mgmt->sa));
536                 return;
537         }
538
539 #ifdef CONFIG_P2P
540         if (hapd->p2p && elems.wps_ie) {
541                 struct wpabuf *wps;
542                 wps = ieee802_11_vendor_ie_concat(ie, ie_len, WPS_DEV_OUI_WFA);
543                 if (wps && !p2p_group_match_dev_type(hapd->p2p_group, wps)) {
544                         wpa_printf(MSG_MSGDUMP, "P2P: Ignore Probe Request "
545                                    "due to mismatch with Requested Device "
546                                    "Type");
547                         wpabuf_free(wps);
548                         return;
549                 }
550                 wpabuf_free(wps);
551         }
552
553         if (hapd->p2p && elems.p2p) {
554                 struct wpabuf *p2p;
555                 p2p = ieee802_11_vendor_ie_concat(ie, ie_len, P2P_IE_VENDOR_TYPE);
556                 if (p2p && !p2p_group_match_dev_id(hapd->p2p_group, p2p)) {
557                         wpa_printf(MSG_MSGDUMP, "P2P: Ignore Probe Request "
558                                    "due to mismatch with Device ID");
559                         wpabuf_free(p2p);
560                         return;
561                 }
562                 wpabuf_free(p2p);
563         }
564 #endif /* CONFIG_P2P */
565
566         if (hapd->conf->ignore_broadcast_ssid && elems.ssid_len == 0 &&
567             elems.ssid_list_len == 0) {
568                 wpa_printf(MSG_MSGDUMP, "Probe Request from " MACSTR " for "
569                            "broadcast SSID ignored", MAC2STR(mgmt->sa));
570                 return;
571         }
572
573         sta = ap_get_sta(hapd, mgmt->sa);
574
575 #ifdef CONFIG_P2P
576         if ((hapd->conf->p2p & P2P_GROUP_OWNER) &&
577             elems.ssid_len == P2P_WILDCARD_SSID_LEN &&
578             os_memcmp(elems.ssid, P2P_WILDCARD_SSID,
579                       P2P_WILDCARD_SSID_LEN) == 0) {
580                 /* Process P2P Wildcard SSID like Wildcard SSID */
581                 elems.ssid_len = 0;
582         }
583 #endif /* CONFIG_P2P */
584
585         res = ssid_match(hapd, elems.ssid, elems.ssid_len,
586                          elems.ssid_list, elems.ssid_list_len);
587         if (res != NO_SSID_MATCH) {
588                 if (sta)
589                         sta->ssid_probe = &hapd->conf->ssid;
590         } else {
591                 if (!(mgmt->da[0] & 0x01)) {
592                         wpa_printf(MSG_MSGDUMP, "Probe Request from " MACSTR
593                                    " for foreign SSID '%s' (DA " MACSTR ")%s",
594                                    MAC2STR(mgmt->sa),
595                                    wpa_ssid_txt(elems.ssid, elems.ssid_len),
596                                    MAC2STR(mgmt->da),
597                                    elems.ssid_list ? " (SSID list)" : "");
598                 }
599                 return;
600         }
601
602 #ifdef CONFIG_INTERWORKING
603         if (hapd->conf->interworking &&
604             elems.interworking && elems.interworking_len >= 1) {
605                 u8 ant = elems.interworking[0] & 0x0f;
606                 if (ant != INTERWORKING_ANT_WILDCARD &&
607                     ant != hapd->conf->access_network_type) {
608                         wpa_printf(MSG_MSGDUMP, "Probe Request from " MACSTR
609                                    " for mismatching ANT %u ignored",
610                                    MAC2STR(mgmt->sa), ant);
611                         return;
612                 }
613         }
614
615         if (hapd->conf->interworking && elems.interworking &&
616             (elems.interworking_len == 7 || elems.interworking_len == 9)) {
617                 const u8 *hessid;
618                 if (elems.interworking_len == 7)
619                         hessid = elems.interworking + 1;
620                 else
621                         hessid = elems.interworking + 1 + 2;
622                 if (!is_broadcast_ether_addr(hessid) &&
623                     os_memcmp(hessid, hapd->conf->hessid, ETH_ALEN) != 0) {
624                         wpa_printf(MSG_MSGDUMP, "Probe Request from " MACSTR
625                                    " for mismatching HESSID " MACSTR
626                                    " ignored",
627                                    MAC2STR(mgmt->sa), MAC2STR(hessid));
628                         return;
629                 }
630         }
631 #endif /* CONFIG_INTERWORKING */
632
633 #ifdef CONFIG_P2P
634         if ((hapd->conf->p2p & P2P_GROUP_OWNER) &&
635             supp_rates_11b_only(&elems)) {
636                 /* Indicates support for 11b rates only */
637                 wpa_printf(MSG_EXCESSIVE, "P2P: Ignore Probe Request from "
638                            MACSTR " with only 802.11b rates",
639                            MAC2STR(mgmt->sa));
640                 return;
641         }
642 #endif /* CONFIG_P2P */
643
644         /* TODO: verify that supp_rates contains at least one matching rate
645          * with AP configuration */
646
647 #ifdef CONFIG_TESTING_OPTIONS
648         if (hapd->iconf->ignore_probe_probability > 0.0 &&
649             drand48() < hapd->iconf->ignore_probe_probability) {
650                 wpa_printf(MSG_INFO,
651                            "TESTING: ignoring probe request from " MACSTR,
652                            MAC2STR(mgmt->sa));
653                 return;
654         }
655 #endif /* CONFIG_TESTING_OPTIONS */
656
657         resp = hostapd_gen_probe_resp(hapd, sta, mgmt, elems.p2p != NULL,
658                                       &resp_len);
659         if (resp == NULL)
660                 return;
661
662         /*
663          * If this is a broadcast probe request, apply no ack policy to avoid
664          * excessive retries.
665          */
666         noack = !!(res == WILDCARD_SSID_MATCH &&
667                    is_broadcast_ether_addr(mgmt->da));
668
669         if (hostapd_drv_send_mlme(hapd, resp, resp_len, noack) < 0)
670                 wpa_printf(MSG_INFO, "handle_probe_req: send failed");
671
672         os_free(resp);
673
674         wpa_printf(MSG_EXCESSIVE, "STA " MACSTR " sent probe request for %s "
675                    "SSID", MAC2STR(mgmt->sa),
676                    elems.ssid_len == 0 ? "broadcast" : "our");
677 }
678
679
680 static u8 * hostapd_probe_resp_offloads(struct hostapd_data *hapd,
681                                         size_t *resp_len)
682 {
683         /* check probe response offloading caps and print warnings */
684         if (!(hapd->iface->drv_flags & WPA_DRIVER_FLAGS_PROBE_RESP_OFFLOAD))
685                 return NULL;
686
687 #ifdef CONFIG_WPS
688         if (hapd->conf->wps_state && hapd->wps_probe_resp_ie &&
689             (!(hapd->iface->probe_resp_offloads &
690                (WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS |
691                 WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS2))))
692                 wpa_printf(MSG_WARNING, "Device is trying to offload WPS "
693                            "Probe Response while not supporting this");
694 #endif /* CONFIG_WPS */
695
696 #ifdef CONFIG_P2P
697         if ((hapd->conf->p2p & P2P_ENABLED) && hapd->p2p_probe_resp_ie &&
698             !(hapd->iface->probe_resp_offloads &
699               WPA_DRIVER_PROBE_RESP_OFFLOAD_P2P))
700                 wpa_printf(MSG_WARNING, "Device is trying to offload P2P "
701                            "Probe Response while not supporting this");
702 #endif  /* CONFIG_P2P */
703
704         if (hapd->conf->interworking &&
705             !(hapd->iface->probe_resp_offloads &
706               WPA_DRIVER_PROBE_RESP_OFFLOAD_INTERWORKING))
707                 wpa_printf(MSG_WARNING, "Device is trying to offload "
708                            "Interworking Probe Response while not supporting "
709                            "this");
710
711         /* Generate a Probe Response template for the non-P2P case */
712         return hostapd_gen_probe_resp(hapd, NULL, NULL, 0, resp_len);
713 }
714
715 #endif /* NEED_AP_MLME */
716
717
718 int ieee802_11_build_ap_params(struct hostapd_data *hapd,
719                                struct wpa_driver_ap_params *params)
720 {
721         struct ieee80211_mgmt *head = NULL;
722         u8 *tail = NULL;
723         size_t head_len = 0, tail_len = 0;
724         u8 *resp = NULL;
725         size_t resp_len = 0;
726 #ifdef NEED_AP_MLME
727         u16 capab_info;
728         u8 *pos, *tailpos;
729
730 #define BEACON_HEAD_BUF_SIZE 256
731 #define BEACON_TAIL_BUF_SIZE 512
732         head = os_zalloc(BEACON_HEAD_BUF_SIZE);
733         tail_len = BEACON_TAIL_BUF_SIZE;
734 #ifdef CONFIG_WPS
735         if (hapd->conf->wps_state && hapd->wps_beacon_ie)
736                 tail_len += wpabuf_len(hapd->wps_beacon_ie);
737 #endif /* CONFIG_WPS */
738 #ifdef CONFIG_P2P
739         if (hapd->p2p_beacon_ie)
740                 tail_len += wpabuf_len(hapd->p2p_beacon_ie);
741 #endif /* CONFIG_P2P */
742         if (hapd->conf->vendor_elements)
743                 tail_len += wpabuf_len(hapd->conf->vendor_elements);
744         tailpos = tail = os_malloc(tail_len);
745         if (head == NULL || tail == NULL) {
746                 wpa_printf(MSG_ERROR, "Failed to set beacon data");
747                 os_free(head);
748                 os_free(tail);
749                 return -1;
750         }
751
752         head->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
753                                            WLAN_FC_STYPE_BEACON);
754         head->duration = host_to_le16(0);
755         os_memset(head->da, 0xff, ETH_ALEN);
756
757         os_memcpy(head->sa, hapd->own_addr, ETH_ALEN);
758         os_memcpy(head->bssid, hapd->own_addr, ETH_ALEN);
759         head->u.beacon.beacon_int =
760                 host_to_le16(hapd->iconf->beacon_int);
761
762         /* hardware or low-level driver will setup seq_ctrl and timestamp */
763         capab_info = hostapd_own_capab_info(hapd, NULL, 0);
764         head->u.beacon.capab_info = host_to_le16(capab_info);
765         pos = &head->u.beacon.variable[0];
766
767         /* SSID */
768         *pos++ = WLAN_EID_SSID;
769         if (hapd->conf->ignore_broadcast_ssid == 2) {
770                 /* clear the data, but keep the correct length of the SSID */
771                 *pos++ = hapd->conf->ssid.ssid_len;
772                 os_memset(pos, 0, hapd->conf->ssid.ssid_len);
773                 pos += hapd->conf->ssid.ssid_len;
774         } else if (hapd->conf->ignore_broadcast_ssid) {
775                 *pos++ = 0; /* empty SSID */
776         } else {
777                 *pos++ = hapd->conf->ssid.ssid_len;
778                 os_memcpy(pos, hapd->conf->ssid.ssid,
779                           hapd->conf->ssid.ssid_len);
780                 pos += hapd->conf->ssid.ssid_len;
781         }
782
783         /* Supported rates */
784         pos = hostapd_eid_supp_rates(hapd, pos);
785
786         /* DS Params */
787         pos = hostapd_eid_ds_params(hapd, pos);
788
789         head_len = pos - (u8 *) head;
790
791         tailpos = hostapd_eid_country(hapd, tailpos,
792                                       tail + BEACON_TAIL_BUF_SIZE - tailpos);
793
794         /* Power Constraint element */
795         tailpos = hostapd_eid_pwr_constraint(hapd, tailpos);
796
797         /* ERP Information element */
798         tailpos = hostapd_eid_erp_info(hapd, tailpos);
799
800         /* Extended supported rates */
801         tailpos = hostapd_eid_ext_supp_rates(hapd, tailpos);
802
803         /* RSN, MDIE, WPA */
804         tailpos = hostapd_eid_wpa(hapd, tailpos, tail + BEACON_TAIL_BUF_SIZE -
805                                   tailpos);
806
807         tailpos = hostapd_eid_bss_load(hapd, tailpos,
808                                        tail + BEACON_TAIL_BUF_SIZE - tailpos);
809
810 #ifdef CONFIG_IEEE80211N
811         tailpos = hostapd_eid_ht_capabilities(hapd, tailpos);
812         tailpos = hostapd_eid_ht_operation(hapd, tailpos);
813 #endif /* CONFIG_IEEE80211N */
814
815         tailpos = hostapd_eid_ext_capab(hapd, tailpos);
816
817         /*
818          * TODO: Time Advertisement element should only be included in some
819          * DTIM Beacon frames.
820          */
821         tailpos = hostapd_eid_time_adv(hapd, tailpos);
822
823         tailpos = hostapd_eid_interworking(hapd, tailpos);
824         tailpos = hostapd_eid_adv_proto(hapd, tailpos);
825         tailpos = hostapd_eid_roaming_consortium(hapd, tailpos);
826         tailpos = hostapd_add_csa_elems(hapd, tailpos, tail,
827                                         &hapd->iface->cs_c_off_beacon);
828 #ifdef CONFIG_IEEE80211AC
829         tailpos = hostapd_eid_vht_capabilities(hapd, tailpos);
830         tailpos = hostapd_eid_vht_operation(hapd, tailpos);
831 #endif /* CONFIG_IEEE80211AC */
832
833         /* Wi-Fi Alliance WMM */
834         tailpos = hostapd_eid_wmm(hapd, tailpos);
835
836 #ifdef CONFIG_WPS
837         if (hapd->conf->wps_state && hapd->wps_beacon_ie) {
838                 os_memcpy(tailpos, wpabuf_head(hapd->wps_beacon_ie),
839                           wpabuf_len(hapd->wps_beacon_ie));
840                 tailpos += wpabuf_len(hapd->wps_beacon_ie);
841         }
842 #endif /* CONFIG_WPS */
843
844 #ifdef CONFIG_P2P
845         if ((hapd->conf->p2p & P2P_ENABLED) && hapd->p2p_beacon_ie) {
846                 os_memcpy(tailpos, wpabuf_head(hapd->p2p_beacon_ie),
847                           wpabuf_len(hapd->p2p_beacon_ie));
848                 tailpos += wpabuf_len(hapd->p2p_beacon_ie);
849         }
850 #endif /* CONFIG_P2P */
851 #ifdef CONFIG_P2P_MANAGER
852         if ((hapd->conf->p2p & (P2P_MANAGE | P2P_ENABLED | P2P_GROUP_OWNER)) ==
853             P2P_MANAGE)
854                 tailpos = hostapd_eid_p2p_manage(hapd, tailpos);
855 #endif /* CONFIG_P2P_MANAGER */
856
857 #ifdef CONFIG_HS20
858         tailpos = hostapd_eid_hs20_indication(hapd, tailpos);
859         tailpos = hostapd_eid_osen(hapd, tailpos);
860 #endif /* CONFIG_HS20 */
861
862         if (hapd->conf->vendor_elements) {
863                 os_memcpy(tailpos, wpabuf_head(hapd->conf->vendor_elements),
864                           wpabuf_len(hapd->conf->vendor_elements));
865                 tailpos += wpabuf_len(hapd->conf->vendor_elements);
866         }
867
868         tail_len = tailpos > tail ? tailpos - tail : 0;
869
870         resp = hostapd_probe_resp_offloads(hapd, &resp_len);
871 #endif /* NEED_AP_MLME */
872
873         os_memset(params, 0, sizeof(*params));
874         params->head = (u8 *) head;
875         params->head_len = head_len;
876         params->tail = tail;
877         params->tail_len = tail_len;
878         params->proberesp = resp;
879         params->proberesp_len = resp_len;
880         params->dtim_period = hapd->conf->dtim_period;
881         params->beacon_int = hapd->iconf->beacon_int;
882         params->basic_rates = hapd->iface->basic_rates;
883         params->ssid = hapd->conf->ssid.ssid;
884         params->ssid_len = hapd->conf->ssid.ssid_len;
885         params->pairwise_ciphers = hapd->conf->wpa_pairwise |
886                 hapd->conf->rsn_pairwise;
887         params->group_cipher = hapd->conf->wpa_group;
888         params->key_mgmt_suites = hapd->conf->wpa_key_mgmt;
889         params->auth_algs = hapd->conf->auth_algs;
890         params->wpa_version = hapd->conf->wpa;
891         params->privacy = hapd->conf->ssid.wep.keys_set || hapd->conf->wpa ||
892                 (hapd->conf->ieee802_1x &&
893                  (hapd->conf->default_wep_key_len ||
894                   hapd->conf->individual_wep_key_len));
895         switch (hapd->conf->ignore_broadcast_ssid) {
896         case 0:
897                 params->hide_ssid = NO_SSID_HIDING;
898                 break;
899         case 1:
900                 params->hide_ssid = HIDDEN_SSID_ZERO_LEN;
901                 break;
902         case 2:
903                 params->hide_ssid = HIDDEN_SSID_ZERO_CONTENTS;
904                 break;
905         }
906         params->isolate = hapd->conf->isolate;
907 #ifdef NEED_AP_MLME
908         params->cts_protect = !!(ieee802_11_erp_info(hapd) &
909                                 ERP_INFO_USE_PROTECTION);
910         params->preamble = hapd->iface->num_sta_no_short_preamble == 0 &&
911                 hapd->iconf->preamble == SHORT_PREAMBLE;
912         if (hapd->iface->current_mode &&
913             hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G)
914                 params->short_slot_time =
915                         hapd->iface->num_sta_no_short_slot_time > 0 ? 0 : 1;
916         else
917                 params->short_slot_time = -1;
918         if (!hapd->iconf->ieee80211n || hapd->conf->disable_11n)
919                 params->ht_opmode = -1;
920         else
921                 params->ht_opmode = hapd->iface->ht_op_mode;
922 #endif /* NEED_AP_MLME */
923         params->interworking = hapd->conf->interworking;
924         if (hapd->conf->interworking &&
925             !is_zero_ether_addr(hapd->conf->hessid))
926                 params->hessid = hapd->conf->hessid;
927         params->access_network_type = hapd->conf->access_network_type;
928         params->ap_max_inactivity = hapd->conf->ap_max_inactivity;
929 #ifdef CONFIG_HS20
930         params->disable_dgaf = hapd->conf->disable_dgaf;
931         if (hapd->conf->osen) {
932                 params->privacy = 1;
933                 params->osen = 1;
934         }
935 #endif /* CONFIG_HS20 */
936         return 0;
937 }
938
939
940 void ieee802_11_free_ap_params(struct wpa_driver_ap_params *params)
941 {
942         os_free(params->tail);
943         params->tail = NULL;
944         os_free(params->head);
945         params->head = NULL;
946         os_free(params->proberesp);
947         params->proberesp = NULL;
948 }
949
950
951 int ieee802_11_set_beacon(struct hostapd_data *hapd)
952 {
953         struct wpa_driver_ap_params params;
954         struct hostapd_freq_params freq;
955         struct hostapd_iface *iface = hapd->iface;
956         struct hostapd_config *iconf = iface->conf;
957         struct wpabuf *beacon, *proberesp, *assocresp;
958         int res, ret = -1;
959
960         if (hapd->iface->csa_in_progress) {
961                 wpa_printf(MSG_ERROR, "Cannot set beacons during CSA period");
962                 return -1;
963         }
964
965         hapd->beacon_set_done = 1;
966
967         if (ieee802_11_build_ap_params(hapd, &params) < 0)
968                 return -1;
969
970         if (hostapd_build_ap_extra_ies(hapd, &beacon, &proberesp, &assocresp) <
971             0)
972                 goto fail;
973
974         params.beacon_ies = beacon;
975         params.proberesp_ies = proberesp;
976         params.assocresp_ies = assocresp;
977
978         if (iface->current_mode &&
979             hostapd_set_freq_params(&freq, iconf->hw_mode, iface->freq,
980                                     iconf->channel, iconf->ieee80211n,
981                                     iconf->ieee80211ac,
982                                     iconf->secondary_channel,
983                                     iconf->vht_oper_chwidth,
984                                     iconf->vht_oper_centr_freq_seg0_idx,
985                                     iconf->vht_oper_centr_freq_seg1_idx,
986                                     iface->current_mode->vht_capab) == 0)
987                 params.freq = &freq;
988
989         res = hostapd_drv_set_ap(hapd, &params);
990         hostapd_free_ap_extra_ies(hapd, beacon, proberesp, assocresp);
991         if (res)
992                 wpa_printf(MSG_ERROR, "Failed to set beacon parameters");
993         else
994                 ret = 0;
995 fail:
996         ieee802_11_free_ap_params(&params);
997         return ret;
998 }
999
1000
1001 int ieee802_11_set_beacons(struct hostapd_iface *iface)
1002 {
1003         size_t i;
1004         int ret = 0;
1005
1006         for (i = 0; i < iface->num_bss; i++) {
1007                 if (iface->bss[i]->started &&
1008                     ieee802_11_set_beacon(iface->bss[i]) < 0)
1009                         ret = -1;
1010         }
1011
1012         return ret;
1013 }
1014
1015
1016 /* only update beacons if started */
1017 int ieee802_11_update_beacons(struct hostapd_iface *iface)
1018 {
1019         size_t i;
1020         int ret = 0;
1021
1022         for (i = 0; i < iface->num_bss; i++) {
1023                 if (iface->bss[i]->beacon_set_done && iface->bss[i]->started &&
1024                     ieee802_11_set_beacon(iface->bss[i]) < 0)
1025                         ret = -1;
1026         }
1027
1028         return ret;
1029 }
1030
1031 #endif /* CONFIG_NATIVE_WINDOWS */