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