P2P: Save a copy of P2P IE(s) data from (Re)Association Request
[mech_eap.git] / src / ap / drv_callbacks.c
1 /*
2  * hostapd / Callback functions for driver wrappers
3  * Copyright (c) 2002-2009, Jouni Malinen <j@w1.fi>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation.
8  *
9  * Alternatively, this software may be distributed under the terms of BSD
10  * license.
11  *
12  * See README and COPYING for more details.
13  */
14
15 #include "utils/includes.h"
16
17 #include "utils/common.h"
18 #include "radius/radius.h"
19 #include "drivers/driver.h"
20 #include "common/ieee802_11_defs.h"
21 #include "common/ieee802_11_common.h"
22 #include "common/wpa_ctrl.h"
23 #include "wps/wps.h"
24 #include "hostapd.h"
25 #include "ieee802_11.h"
26 #include "sta_info.h"
27 #include "accounting.h"
28 #include "tkip_countermeasures.h"
29 #include "iapp.h"
30 #include "ieee802_1x.h"
31 #include "wpa_auth.h"
32 #include "wmm.h"
33 #include "wps_hostapd.h"
34 #include "ap_config.h"
35
36
37 int hostapd_notif_assoc(struct hostapd_data *hapd, const u8 *addr,
38                         const u8 *ie, size_t ielen)
39 {
40         struct sta_info *sta;
41         int new_assoc, res;
42         struct ieee802_11_elems elems;
43 #ifdef CONFIG_P2P
44         const u8 *all_ies = ie;
45         size_t all_ies_len = ielen;
46 #endif /* CONFIG_P2P */
47
48         if (addr == NULL) {
49                 /*
50                  * This could potentially happen with unexpected event from the
51                  * driver wrapper. This was seen at least in one case where the
52                  * driver ended up being set to station mode while hostapd was
53                  * running, so better make sure we stop processing such an
54                  * event here.
55                  */
56                 wpa_printf(MSG_DEBUG, "hostapd_notif_assoc: Skip event with "
57                            "no address");
58                 return -1;
59         }
60
61         hostapd_logger(hapd, addr, HOSTAPD_MODULE_IEEE80211,
62                        HOSTAPD_LEVEL_INFO, "associated");
63
64         ieee802_11_parse_elems(ie, ielen, &elems, 0);
65         if (elems.wps_ie) {
66                 ie = elems.wps_ie - 2;
67                 ielen = elems.wps_ie_len + 2;
68                 wpa_printf(MSG_DEBUG, "STA included WPS IE in (Re)AssocReq");
69         } else if (elems.rsn_ie) {
70                 ie = elems.rsn_ie - 2;
71                 ielen = elems.rsn_ie_len + 2;
72                 wpa_printf(MSG_DEBUG, "STA included RSN IE in (Re)AssocReq");
73         } else if (elems.wpa_ie) {
74                 ie = elems.wpa_ie - 2;
75                 ielen = elems.wpa_ie_len + 2;
76                 wpa_printf(MSG_DEBUG, "STA included WPA IE in (Re)AssocReq");
77         } else {
78                 ie = NULL;
79                 ielen = 0;
80                 wpa_printf(MSG_DEBUG, "STA did not include WPS/RSN/WPA IE in "
81                            "(Re)AssocReq");
82         }
83
84         sta = ap_get_sta(hapd, addr);
85         if (sta) {
86                 accounting_sta_stop(hapd, sta);
87         } else {
88                 sta = ap_sta_add(hapd, addr);
89                 if (sta == NULL)
90                         return -1;
91         }
92         sta->flags &= ~(WLAN_STA_WPS | WLAN_STA_MAYBE_WPS);
93
94 #ifdef CONFIG_P2P
95         if (elems.p2p) {
96                 wpabuf_free(sta->p2p_ie);
97                 sta->p2p_ie = ieee802_11_vendor_ie_concat(all_ies, all_ies_len,
98                                                           P2P_IE_VENDOR_TYPE);
99         }
100 #endif /* CONFIG_P2P */
101
102         if (hapd->conf->wpa) {
103                 if (ie == NULL || ielen == 0) {
104                         if (hapd->conf->wps_state) {
105                                 wpa_printf(MSG_DEBUG, "STA did not include "
106                                            "WPA/RSN IE in (Re)Association "
107                                            "Request - possible WPS use");
108                                 sta->flags |= WLAN_STA_MAYBE_WPS;
109                                 goto skip_wpa_check;
110                         }
111
112                         wpa_printf(MSG_DEBUG, "No WPA/RSN IE from STA");
113                         return -1;
114                 }
115                 if (hapd->conf->wps_state && ie[0] == 0xdd && ie[1] >= 4 &&
116                     os_memcmp(ie + 2, "\x00\x50\xf2\x04", 4) == 0) {
117                         sta->flags |= WLAN_STA_WPS;
118                         goto skip_wpa_check;
119                 }
120
121                 if (sta->wpa_sm == NULL)
122                         sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,
123                                                         sta->addr);
124                 if (sta->wpa_sm == NULL) {
125                         wpa_printf(MSG_ERROR, "Failed to initialize WPA state "
126                                    "machine");
127                         return -1;
128                 }
129                 res = wpa_validate_wpa_ie(hapd->wpa_auth, sta->wpa_sm,
130                                           ie, ielen, NULL, 0);
131                 if (res != WPA_IE_OK) {
132                         int resp;
133                         wpa_printf(MSG_DEBUG, "WPA/RSN information element "
134                                    "rejected? (res %u)", res);
135                         wpa_hexdump(MSG_DEBUG, "IE", ie, ielen);
136                         if (res == WPA_INVALID_GROUP)
137                                 resp = WLAN_REASON_GROUP_CIPHER_NOT_VALID;
138                         else if (res == WPA_INVALID_PAIRWISE)
139                                 resp = WLAN_REASON_PAIRWISE_CIPHER_NOT_VALID;
140                         else if (res == WPA_INVALID_AKMP)
141                                 resp = WLAN_REASON_AKMP_NOT_VALID;
142 #ifdef CONFIG_IEEE80211W
143                         else if (res == WPA_MGMT_FRAME_PROTECTION_VIOLATION)
144                                 resp = WLAN_REASON_INVALID_IE;
145                         else if (res == WPA_INVALID_MGMT_GROUP_CIPHER)
146                                 resp = WLAN_REASON_GROUP_CIPHER_NOT_VALID;
147 #endif /* CONFIG_IEEE80211W */
148                         else
149                                 resp = WLAN_REASON_INVALID_IE;
150                         hapd->drv.sta_disassoc(hapd, sta->addr, resp);
151                         ap_free_sta(hapd, sta);
152                         return -1;
153                 }
154         } else if (hapd->conf->wps_state) {
155 #ifdef CONFIG_WPS_STRICT
156                 struct wpabuf *wps;
157                 wps = ieee802_11_vendor_ie_concat(ie, ielen,
158                                                   WPS_IE_VENDOR_TYPE);
159                 if (wps && wps_validate_assoc_req(wps) < 0) {
160                         hapd->drv.sta_disassoc(hapd, sta->addr,
161                                                WLAN_REASON_INVALID_IE);
162                         ap_free_sta(hapd, sta);
163                         wpabuf_free(wps);
164                         return -1;
165                 }
166                 wpabuf_free(wps);
167 #endif /* CONFIG_WPS_STRICT */
168                 if (ie && ielen > 4 && ie[0] == 0xdd && ie[1] >= 4 &&
169                     os_memcmp(ie + 2, "\x00\x50\xf2\x04", 4) == 0) {
170                         sta->flags |= WLAN_STA_WPS;
171                 } else
172                         sta->flags |= WLAN_STA_MAYBE_WPS;
173         }
174 skip_wpa_check:
175
176         new_assoc = (sta->flags & WLAN_STA_ASSOC) == 0;
177         sta->flags |= WLAN_STA_AUTH | WLAN_STA_ASSOC;
178         wpa_auth_sm_event(sta->wpa_sm, WPA_ASSOC);
179
180         hostapd_new_assoc_sta(hapd, sta, !new_assoc);
181
182         ieee802_1x_notify_port_enabled(sta->eapol_sm, 1);
183
184         return 0;
185 }
186
187
188 void hostapd_notif_disassoc(struct hostapd_data *hapd, const u8 *addr)
189 {
190         struct sta_info *sta;
191
192         hostapd_logger(hapd, addr, HOSTAPD_MODULE_IEEE80211,
193                        HOSTAPD_LEVEL_INFO, "disassociated");
194
195         sta = ap_get_sta(hapd, addr);
196         if (sta == NULL) {
197                 wpa_printf(MSG_DEBUG, "Disassociation notification for "
198                            "unknown STA " MACSTR, MAC2STR(addr));
199                 return;
200         }
201
202         sta->flags &= ~(WLAN_STA_AUTH | WLAN_STA_ASSOC);
203         wpa_msg(hapd->msg_ctx, MSG_INFO, AP_STA_DISCONNECTED MACSTR,
204                 MAC2STR(sta->addr));
205         wpa_auth_sm_event(sta->wpa_sm, WPA_DISASSOC);
206         sta->acct_terminate_cause = RADIUS_ACCT_TERMINATE_CAUSE_USER_REQUEST;
207         ieee802_1x_notify_port_enabled(sta->eapol_sm, 0);
208         ap_free_sta(hapd, sta);
209 }
210
211
212 #ifdef HOSTAPD
213
214 #ifdef NEED_AP_MLME
215
216 static const u8 * get_hdr_bssid(const struct ieee80211_hdr *hdr, size_t len)
217 {
218         u16 fc, type, stype;
219
220         /*
221          * PS-Poll frames are 16 bytes. All other frames are
222          * 24 bytes or longer.
223          */
224         if (len < 16)
225                 return NULL;
226
227         fc = le_to_host16(hdr->frame_control);
228         type = WLAN_FC_GET_TYPE(fc);
229         stype = WLAN_FC_GET_STYPE(fc);
230
231         switch (type) {
232         case WLAN_FC_TYPE_DATA:
233                 if (len < 24)
234                         return NULL;
235                 switch (fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)) {
236                 case WLAN_FC_FROMDS | WLAN_FC_TODS:
237                 case WLAN_FC_TODS:
238                         return hdr->addr1;
239                 case WLAN_FC_FROMDS:
240                         return hdr->addr2;
241                 default:
242                         return NULL;
243                 }
244         case WLAN_FC_TYPE_CTRL:
245                 if (stype != WLAN_FC_STYPE_PSPOLL)
246                         return NULL;
247                 return hdr->addr1;
248         case WLAN_FC_TYPE_MGMT:
249                 return hdr->addr3;
250         default:
251                 return NULL;
252         }
253 }
254
255
256 #define HAPD_BROADCAST ((struct hostapd_data *) -1)
257
258 static struct hostapd_data * get_hapd_bssid(struct hostapd_iface *iface,
259                                             const u8 *bssid)
260 {
261         size_t i;
262
263         if (bssid == NULL)
264                 return NULL;
265         if (bssid[0] == 0xff && bssid[1] == 0xff && bssid[2] == 0xff &&
266             bssid[3] == 0xff && bssid[4] == 0xff && bssid[5] == 0xff)
267                 return HAPD_BROADCAST;
268
269         for (i = 0; i < iface->num_bss; i++) {
270                 if (os_memcmp(bssid, iface->bss[i]->own_addr, ETH_ALEN) == 0)
271                         return iface->bss[i];
272         }
273
274         return NULL;
275 }
276
277
278 static void hostapd_rx_from_unknown_sta(struct hostapd_data *hapd,
279                                         const u8 *frame, size_t len)
280 {
281         const struct ieee80211_hdr *hdr = (const struct ieee80211_hdr *) frame;
282         u16 fc = le_to_host16(hdr->frame_control);
283         hapd = get_hapd_bssid(hapd->iface, get_hdr_bssid(hdr, len));
284         if (hapd == NULL || hapd == HAPD_BROADCAST)
285                 return;
286
287         ieee802_11_rx_from_unknown(hapd, hdr->addr2,
288                                    (fc & (WLAN_FC_TODS | WLAN_FC_FROMDS)) ==
289                                    (WLAN_FC_TODS | WLAN_FC_FROMDS));
290 }
291
292
293 static void hostapd_mgmt_rx(struct hostapd_data *hapd, struct rx_mgmt *rx_mgmt)
294 {
295         struct hostapd_iface *iface = hapd->iface;
296         const struct ieee80211_hdr *hdr;
297         const u8 *bssid;
298         struct hostapd_frame_info fi;
299
300         hdr = (const struct ieee80211_hdr *) rx_mgmt->frame;
301         bssid = get_hdr_bssid(hdr, rx_mgmt->frame_len);
302         if (bssid == NULL)
303                 return;
304
305         hapd = get_hapd_bssid(iface, bssid);
306         if (hapd == NULL) {
307                 u16 fc;
308                 fc = le_to_host16(hdr->frame_control);
309
310                 /*
311                  * Drop frames to unknown BSSIDs except for Beacon frames which
312                  * could be used to update neighbor information.
313                  */
314                 if (WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_MGMT &&
315                     WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_BEACON)
316                         hapd = iface->bss[0];
317                 else
318                         return;
319         }
320
321         os_memset(&fi, 0, sizeof(fi));
322         fi.datarate = rx_mgmt->datarate;
323         fi.ssi_signal = rx_mgmt->ssi_signal;
324
325         if (hapd == HAPD_BROADCAST) {
326                 size_t i;
327                 for (i = 0; i < iface->num_bss; i++)
328                         ieee802_11_mgmt(iface->bss[i], rx_mgmt->frame,
329                                         rx_mgmt->frame_len, &fi);
330         } else
331                 ieee802_11_mgmt(hapd, rx_mgmt->frame, rx_mgmt->frame_len, &fi);
332 }
333
334
335 static void hostapd_mgmt_tx_cb(struct hostapd_data *hapd, const u8 *buf,
336                                size_t len, u16 stype, int ok)
337 {
338         struct ieee80211_hdr *hdr;
339         hdr = (struct ieee80211_hdr *) buf;
340         hapd = get_hapd_bssid(hapd->iface, get_hdr_bssid(hdr, len));
341         if (hapd == NULL || hapd == HAPD_BROADCAST)
342                 return;
343         ieee802_11_mgmt_cb(hapd, buf, len, stype, ok);
344 }
345
346 #endif /* NEED_AP_MLME */
347
348
349 static int hostapd_probe_req_rx(struct hostapd_data *hapd, const u8 *sa,
350                                 const u8 *ie, size_t ie_len)
351 {
352         size_t i;
353         int ret = 0;
354
355         for (i = 0; hapd->probereq_cb && i < hapd->num_probereq_cb; i++) {
356                 if (hapd->probereq_cb[i].cb(hapd->probereq_cb[i].ctx,
357                                             sa, ie, ie_len) > 0) {
358                         ret = 1;
359                         break;
360                 }
361         }
362         return ret;
363 }
364
365
366 static int hostapd_event_new_sta(struct hostapd_data *hapd, const u8 *addr)
367 {
368         struct sta_info *sta = ap_get_sta(hapd, addr);
369         if (sta)
370                 return 0;
371
372         wpa_printf(MSG_DEBUG, "Data frame from unknown STA " MACSTR
373                    " - adding a new STA", MAC2STR(addr));
374         sta = ap_sta_add(hapd, addr);
375         if (sta) {
376                 hostapd_new_assoc_sta(hapd, sta, 0);
377         } else {
378                 wpa_printf(MSG_DEBUG, "Failed to add STA entry for " MACSTR,
379                            MAC2STR(addr));
380                 return -1;
381         }
382
383         return 0;
384 }
385
386
387 static void hostapd_event_eapol_rx(struct hostapd_data *hapd, const u8 *src,
388                                    const u8 *data, size_t data_len)
389 {
390         struct hostapd_iface *iface = hapd->iface;
391         size_t j;
392
393         for (j = 0; j < iface->num_bss; j++) {
394                 if (ap_get_sta(iface->bss[j], src)) {
395                         hapd = iface->bss[j];
396                         break;
397                 }
398         }
399
400         ieee802_1x_receive(hapd, src, data, data_len);
401 }
402
403
404 void wpa_supplicant_event(void *ctx, enum wpa_event_type event,
405                           union wpa_event_data *data)
406 {
407         struct hostapd_data *hapd = ctx;
408
409         switch (event) {
410         case EVENT_MICHAEL_MIC_FAILURE:
411                 michael_mic_failure(hapd, data->michael_mic_failure.src, 1);
412                 break;
413         case EVENT_SCAN_RESULTS:
414                 if (hapd->iface->scan_cb)
415                         hapd->iface->scan_cb(hapd->iface);
416                 break;
417 #ifdef CONFIG_IEEE80211R
418         case EVENT_FT_RRB_RX:
419                 wpa_ft_rrb_rx(hapd->wpa_auth, data->ft_rrb_rx.src,
420                               data->ft_rrb_rx.data, data->ft_rrb_rx.data_len);
421                 break;
422 #endif /* CONFIG_IEEE80211R */
423         case EVENT_WPS_BUTTON_PUSHED:
424                 hostapd_wps_button_pushed(hapd);
425                 break;
426 #ifdef NEED_AP_MLME
427         case EVENT_TX_STATUS:
428                 switch (data->tx_status.type) {
429                 case WLAN_FC_TYPE_MGMT:
430                         hostapd_mgmt_tx_cb(hapd, data->tx_status.data,
431                                            data->tx_status.data_len,
432                                            data->tx_status.stype,
433                                            data->tx_status.ack);
434                         break;
435                 case WLAN_FC_TYPE_DATA:
436                         hostapd_tx_status(hapd, data->tx_status.dst,
437                                           data->tx_status.data,
438                                           data->tx_status.data_len,
439                                           data->tx_status.ack);
440                         break;
441                 }
442                 break;
443         case EVENT_RX_FROM_UNKNOWN:
444                 hostapd_rx_from_unknown_sta(hapd, data->rx_from_unknown.frame,
445                                             data->rx_from_unknown.len);
446                 break;
447         case EVENT_RX_MGMT:
448                 hostapd_mgmt_rx(hapd, &data->rx_mgmt);
449                 break;
450 #endif /* NEED_AP_MLME */
451         case EVENT_RX_PROBE_REQ:
452                 hostapd_probe_req_rx(hapd, data->rx_probe_req.sa,
453                                      data->rx_probe_req.ie,
454                                      data->rx_probe_req.ie_len);
455                 break;
456         case EVENT_NEW_STA:
457                 hostapd_event_new_sta(hapd, data->new_sta.addr);
458                 break;
459         case EVENT_EAPOL_RX:
460                 hostapd_event_eapol_rx(hapd, data->eapol_rx.src,
461                                        data->eapol_rx.data,
462                                        data->eapol_rx.data_len);
463                 break;
464         case EVENT_ASSOC:
465                 hostapd_notif_assoc(hapd, data->assoc_info.addr,
466                                     data->assoc_info.req_ies,
467                                     data->assoc_info.req_ies_len);
468                 break;
469         case EVENT_DISASSOC:
470                 if (data)
471                         hostapd_notif_disassoc(hapd, data->disassoc_info.addr);
472                 break;
473         case EVENT_DEAUTH:
474                 if (data)
475                         hostapd_notif_disassoc(hapd, data->deauth_info.addr);
476                 break;
477         default:
478                 wpa_printf(MSG_DEBUG, "Unknown event %d", event);
479                 break;
480         }
481 }
482
483 #endif /* HOSTAPD */