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