1 /*
2  * BSS list
3  * Copyright (c) 2010, Jouni Malinen <j@w1.fi>
4  *
5  * This software may be distributed under the terms of the BSD license.
6  * See README for more details.
7  */
8 
9 #include "utils/includes.h"
10 
11 #include "utils/common.h"
12 #include "common/defs.h"
13 #include "common/ieee802_11_defs.h"
14 #include "common/ieee802_11_common.h"
15 #include "crypto/sha1.h"
16 #include "wlantest.h"
17 
18 
bss_find(struct wlantest * wt,const u8 * bssid)19 struct wlantest_bss * bss_find(struct wlantest *wt, const u8 *bssid)
20 {
21 	struct wlantest_bss *bss;
22 
23 	dl_list_for_each(bss, &wt->bss, struct wlantest_bss, list) {
24 		if (ether_addr_equal(bss->bssid, bssid))
25 			return bss;
26 	}
27 
28 	return NULL;
29 }
30 
31 
bss_find_mld(struct wlantest * wt,const u8 * mld_mac_addr,int link_id)32 struct wlantest_bss * bss_find_mld(struct wlantest *wt, const u8 *mld_mac_addr,
33 				   int link_id)
34 {
35 	struct wlantest_bss *bss;
36 
37 	dl_list_for_each(bss, &wt->bss, struct wlantest_bss, list) {
38 		if (ether_addr_equal(bss->mld_mac_addr, mld_mac_addr) &&
39 		    (link_id < 0 ||
40 		     (bss->link_id_set && bss->link_id == link_id)))
41 			return bss;
42 	}
43 
44 	return NULL;
45 }
46 
47 
bss_get(struct wlantest * wt,const u8 * bssid)48 struct wlantest_bss * bss_get(struct wlantest *wt, const u8 *bssid)
49 {
50 	struct wlantest_bss *bss;
51 
52 	if (bssid[0] & 0x01)
53 		return NULL; /* Skip group addressed frames */
54 
55 	bss = bss_find(wt, bssid);
56 	if (bss)
57 		return bss;
58 
59 	bss = os_zalloc(sizeof(*bss));
60 	if (bss == NULL)
61 		return NULL;
62 	dl_list_init(&bss->sta);
63 	dl_list_init(&bss->pmk);
64 	dl_list_init(&bss->tdls);
65 	os_memcpy(bss->bssid, bssid, ETH_ALEN);
66 	dl_list_add(&wt->bss, &bss->list);
67 	wpa_printf(MSG_DEBUG, "Discovered new BSS - " MACSTR,
68 		   MAC2STR(bss->bssid));
69 	return bss;
70 }
71 
72 
pmk_deinit(struct wlantest_pmk * pmk)73 void pmk_deinit(struct wlantest_pmk *pmk)
74 {
75 	dl_list_del(&pmk->list);
76 	os_free(pmk);
77 }
78 
79 
tdls_deinit(struct wlantest_tdls * tdls)80 void tdls_deinit(struct wlantest_tdls *tdls)
81 {
82 	dl_list_del(&tdls->list);
83 	os_free(tdls);
84 }
85 
86 
bss_deinit(struct wlantest_bss * bss)87 void bss_deinit(struct wlantest_bss *bss)
88 {
89 	struct wlantest_sta *sta, *n;
90 	struct wlantest_pmk *pmk, *np;
91 	struct wlantest_tdls *tdls, *nt;
92 	dl_list_for_each_safe(sta, n, &bss->sta, struct wlantest_sta, list)
93 		sta_deinit(sta);
94 	dl_list_for_each_safe(pmk, np, &bss->pmk, struct wlantest_pmk, list)
95 		pmk_deinit(pmk);
96 	dl_list_for_each_safe(tdls, nt, &bss->tdls, struct wlantest_tdls, list)
97 		tdls_deinit(tdls);
98 	dl_list_del(&bss->list);
99 	os_free(bss);
100 }
101 
102 
bss_add_pmk_from_passphrase(struct wlantest_bss * bss,const char * passphrase)103 int bss_add_pmk_from_passphrase(struct wlantest_bss *bss,
104 				const char *passphrase)
105 {
106 	struct wlantest_pmk *pmk;
107 
108 	pmk = os_zalloc(sizeof(*pmk));
109 	if (pmk == NULL)
110 		return -1;
111 	if (pbkdf2_sha1(passphrase, bss->ssid, bss->ssid_len, 4096,
112 			pmk->pmk, PMK_LEN) < 0) {
113 		os_free(pmk);
114 		return -1;
115 	}
116 
117 	wpa_printf(MSG_INFO, "Add possible PMK for BSSID " MACSTR
118 		   " based on passphrase '%s'",
119 		   MAC2STR(bss->bssid), passphrase);
120 	wpa_hexdump(MSG_DEBUG, "Possible PMK", pmk->pmk, PMK_LEN);
121 	pmk->pmk_len = PMK_LEN;
122 	dl_list_add(&bss->pmk, &pmk->list);
123 
124 	return 0;
125 }
126 
127 
bss_add_pmk(struct wlantest * wt,struct wlantest_bss * bss)128 static void bss_add_pmk(struct wlantest *wt, struct wlantest_bss *bss)
129 {
130 	struct wlantest_passphrase *p;
131 
132 	dl_list_for_each(p, &wt->passphrase, struct wlantest_passphrase, list)
133 	{
134 		if (!is_zero_ether_addr(p->bssid) &&
135 		    !ether_addr_equal(p->bssid, bss->bssid))
136 			continue;
137 		if (p->ssid_len &&
138 		    (p->ssid_len != bss->ssid_len ||
139 		     os_memcmp(p->ssid, bss->ssid, p->ssid_len) != 0))
140 			continue;
141 
142 		if (bss_add_pmk_from_passphrase(bss, p->passphrase) < 0)
143 			break;
144 	}
145 }
146 
147 
bss_update(struct wlantest * wt,struct wlantest_bss * bss,struct ieee802_11_elems * elems,int beacon)148 void bss_update(struct wlantest *wt, struct wlantest_bss *bss,
149 		struct ieee802_11_elems *elems, int beacon)
150 {
151 	struct wpa_ie_data data;
152 	int update = 0;
153 
154 	if (bss->capab_info != bss->prev_capab_info)
155 		update = 1;
156 
157 	if (beacon && (!elems->ssid || elems->ssid_len > 32)) {
158 		wpa_printf(MSG_INFO,
159 			   "Invalid or missing SSID in a %s frame for " MACSTR,
160 			   beacon == 1 ? "Beacon" : "Probe Response",
161 			   MAC2STR(bss->bssid));
162 		bss->parse_error_reported = 1;
163 		return;
164 	}
165 
166 	if (beacon &&
167 	    (bss->ssid_len != elems->ssid_len ||
168 	     os_memcmp(bss->ssid, elems->ssid, bss->ssid_len) != 0)) {
169 		wpa_printf(MSG_DEBUG, "Store SSID '%s' for BSSID " MACSTR,
170 			   wpa_ssid_txt(elems->ssid, elems->ssid_len),
171 			   MAC2STR(bss->bssid));
172 		os_memcpy(bss->ssid, elems->ssid, elems->ssid_len);
173 		bss->ssid_len = elems->ssid_len;
174 		bss_add_pmk(wt, bss);
175 	}
176 
177 	if (elems->osen == NULL) {
178 		if (bss->osenie[0]) {
179 			add_note(wt, MSG_INFO, "BSS " MACSTR
180 				 " - OSEN IE removed", MAC2STR(bss->bssid));
181 			bss->rsnie[0] = 0;
182 			update = 1;
183 		}
184 	} else {
185 		if (bss->osenie[0] == 0 ||
186 		    os_memcmp(bss->osenie, elems->osen - 2,
187 			      elems->osen_len + 2) != 0) {
188 			wpa_printf(MSG_INFO, "BSS " MACSTR " - OSEN IE "
189 				   "stored", MAC2STR(bss->bssid));
190 			wpa_hexdump(MSG_DEBUG, "OSEN IE", elems->osen - 2,
191 				    elems->osen_len + 2);
192 			update = 1;
193 		}
194 		os_memcpy(bss->osenie, elems->osen - 2,
195 			  elems->osen_len + 2);
196 	}
197 
198 	/* S1G does not include RSNE in beacon, so only clear it from
199 	 * Probe Response frames. Note this assumes short beacons were dropped
200 	 * due to missing SSID above.
201 	 */
202 	if (!elems->rsn_ie && (!elems->s1g_capab || beacon != 1)) {
203 		if (bss->rsnie[0]) {
204 			add_note(wt, MSG_INFO, "BSS " MACSTR
205 				 " - RSN IE removed", MAC2STR(bss->bssid));
206 			bss->rsnie[0] = 0;
207 			update = 1;
208 		}
209 	} else if (elems->rsn_ie) {
210 		if (bss->rsnie[0] == 0 ||
211 		    os_memcmp(bss->rsnie, elems->rsn_ie - 2,
212 			      elems->rsn_ie_len + 2) != 0) {
213 			wpa_printf(MSG_INFO, "BSS " MACSTR " - RSN IE "
214 				   "stored", MAC2STR(bss->bssid));
215 			wpa_hexdump(MSG_DEBUG, "RSN IE", elems->rsn_ie - 2,
216 				    elems->rsn_ie_len + 2);
217 			update = 1;
218 		}
219 		os_memcpy(bss->rsnie, elems->rsn_ie - 2,
220 			  elems->rsn_ie_len + 2);
221 	}
222 
223 	if (elems->wpa_ie == NULL) {
224 		if (bss->wpaie[0]) {
225 			add_note(wt, MSG_INFO, "BSS " MACSTR
226 				 " - WPA IE removed", MAC2STR(bss->bssid));
227 			bss->wpaie[0] = 0;
228 			update = 1;
229 		}
230 	} else {
231 		if (bss->wpaie[0] == 0 ||
232 		    os_memcmp(bss->wpaie, elems->wpa_ie - 2,
233 			      elems->wpa_ie_len + 2) != 0) {
234 			wpa_printf(MSG_INFO, "BSS " MACSTR " - WPA IE "
235 				   "stored", MAC2STR(bss->bssid));
236 			wpa_hexdump(MSG_DEBUG, "WPA IE", elems->wpa_ie - 2,
237 				    elems->wpa_ie_len + 2);
238 			update = 1;
239 		}
240 		os_memcpy(bss->wpaie, elems->wpa_ie - 2,
241 			  elems->wpa_ie_len + 2);
242 	}
243 
244 	if (elems->mdie)
245 		os_memcpy(bss->mdid, elems->mdie, 2);
246 
247 	bss->mesh = elems->mesh_id != NULL;
248 
249 	if (is_zero_ether_addr(bss->mld_mac_addr) &&
250 	    elems->basic_mle && elems->basic_mle_len >= 2 + 1 + ETH_ALEN &&
251 	    elems->basic_mle[2] >= 1 + ETH_ALEN) {
252 		os_memcpy(bss->mld_mac_addr, &elems->basic_mle[2 + 1],
253 			  ETH_ALEN);
254 		wpa_printf(MSG_DEBUG,
255 			   "Learned AP MLD MAC Address from Beacon/Probe Response frame: "
256 			   MACSTR " (BSSID " MACSTR ")",
257 			   MAC2STR(bss->mld_mac_addr), MAC2STR(bss->bssid));
258 	}
259 
260 	if (!bss->link_id_set &&
261 	    elems->basic_mle && elems->basic_mle_len >= 2 + 1 + ETH_ALEN + 1 &&
262 	    elems->basic_mle[2] >= 1 + ETH_ALEN + 1 &&
263 	    (WPA_GET_LE16(elems->basic_mle) &
264 	     BASIC_MULTI_LINK_CTRL_PRES_LINK_ID)) {
265 		    bss->link_id = elems->basic_mle[2 + 1 + ETH_ALEN] & 0x0f;
266 		    wpa_printf(MSG_DEBUG,
267 			       "Learned AP MLD Link ID %u for this affiliated link",
268 			       bss->link_id);
269 		    bss->link_id_set = true;
270 	}
271 
272 	if (!update)
273 		return;
274 
275 	if (beacon == 1)
276 		bss->beacon_seen = 1;
277 	else if (beacon == 2)
278 		bss->proberesp_seen = 1;
279 	bss->ies_set = 1;
280 	bss->prev_capab_info = bss->capab_info;
281 	bss->proto = 0;
282 	bss->pairwise_cipher = 0;
283 	bss->group_cipher = 0;
284 	bss->key_mgmt = 0;
285 	bss->rsn_capab = 0;
286 	bss->mgmt_group_cipher = 0;
287 
288 	if (bss->wpaie[0]) {
289 		if (wpa_parse_wpa_ie_wpa(bss->wpaie, 2 + bss->wpaie[1], &data)
290 		    < 0) {
291 			add_note(wt, MSG_INFO, "Failed to parse WPA IE from "
292 				 MACSTR, MAC2STR(bss->bssid));
293 		} else {
294 			bss->proto |= data.proto;
295 			bss->pairwise_cipher |= data.pairwise_cipher;
296 			bss->group_cipher |= data.group_cipher;
297 			bss->key_mgmt |= data.key_mgmt;
298 			bss->rsn_capab = data.capabilities;
299 			bss->mgmt_group_cipher |= data.mgmt_group_cipher;
300 		}
301 	}
302 
303 	if (bss->rsnie[0]) {
304 		if (wpa_parse_wpa_ie_rsn(bss->rsnie, 2 + bss->rsnie[1], &data)
305 		    < 0) {
306 			add_note(wt, MSG_INFO, "Failed to parse RSN IE from "
307 				 MACSTR, MAC2STR(bss->bssid));
308 		} else {
309 			bss->proto |= data.proto;
310 			bss->pairwise_cipher |= data.pairwise_cipher;
311 			bss->group_cipher |= data.group_cipher;
312 			bss->key_mgmt |= data.key_mgmt;
313 			bss->rsn_capab = data.capabilities;
314 			bss->mgmt_group_cipher |= data.mgmt_group_cipher;
315 		}
316 	}
317 
318 	if (bss->osenie[0]) {
319 		bss->proto |= WPA_PROTO_OSEN;
320 		bss->pairwise_cipher |= WPA_CIPHER_CCMP;
321 		bss->group_cipher |= WPA_CIPHER_CCMP;
322 		bss->key_mgmt |= WPA_KEY_MGMT_OSEN;
323 	}
324 
325 	if (!(bss->proto & WPA_PROTO_RSN) ||
326 	    !(bss->rsn_capab & WPA_CAPABILITY_MFPC))
327 		bss->mgmt_group_cipher = 0;
328 
329 	if (!bss->wpaie[0] && !bss->rsnie[0] && !bss->osenie[0] &&
330 	    (bss->capab_info & WLAN_CAPABILITY_PRIVACY))
331 		bss->group_cipher = WPA_CIPHER_WEP40;
332 
333 	wpa_printf(MSG_INFO, "BSS " MACSTR
334 		   " proto=%s%s%s%s"
335 		   "pairwise=%s%s%s%s%s%s%s"
336 		   "group=%s%s%s%s%s%s%s%s%s"
337 		   "mgmt_group_cipher=%s%s%s%s%s"
338 		   "key_mgmt=%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s"
339 		   "rsn_capab=%s%s%s%s%s%s%s%s%s%s",
340 		   MAC2STR(bss->bssid),
341 		   bss->proto == 0 ? "OPEN " : "",
342 		   bss->proto & WPA_PROTO_WPA ? "WPA " : "",
343 		   bss->proto & WPA_PROTO_RSN ? "WPA2 " : "",
344 		   bss->proto & WPA_PROTO_OSEN ? "OSEN " : "",
345 		   bss->pairwise_cipher == 0 ? "N/A " : "",
346 		   bss->pairwise_cipher & WPA_CIPHER_NONE ? "NONE " : "",
347 		   bss->pairwise_cipher & WPA_CIPHER_TKIP ? "TKIP " : "",
348 		   bss->pairwise_cipher & WPA_CIPHER_CCMP ? "CCMP " : "",
349 		   bss->pairwise_cipher & WPA_CIPHER_CCMP_256 ? "CCMP-256 " :
350 		   "",
351 		   bss->pairwise_cipher & WPA_CIPHER_GCMP ? "GCMP " : "",
352 		   bss->pairwise_cipher & WPA_CIPHER_GCMP_256 ? "GCMP-256 " :
353 		   "",
354 		   bss->group_cipher == 0 ? "N/A " : "",
355 		   bss->group_cipher & WPA_CIPHER_NONE ? "NONE " : "",
356 		   bss->group_cipher & WPA_CIPHER_WEP40 ? "WEP40 " : "",
357 		   bss->group_cipher & WPA_CIPHER_WEP104 ? "WEP104 " : "",
358 		   bss->group_cipher & WPA_CIPHER_TKIP ? "TKIP " : "",
359 		   bss->group_cipher & WPA_CIPHER_CCMP ? "CCMP " : "",
360 		   bss->group_cipher & WPA_CIPHER_CCMP_256 ? "CCMP-256 " : "",
361 		   bss->group_cipher & WPA_CIPHER_GCMP ? "GCMP " : "",
362 		   bss->group_cipher & WPA_CIPHER_GCMP_256 ? "GCMP-256 " : "",
363 		   bss->mgmt_group_cipher == 0 ? "N/A " : "",
364 		   bss->mgmt_group_cipher & WPA_CIPHER_AES_128_CMAC ?
365 		   "BIP " : "",
366 		   bss->mgmt_group_cipher & WPA_CIPHER_BIP_GMAC_128 ?
367 		   "BIP-GMAC-128 " : "",
368 		   bss->mgmt_group_cipher & WPA_CIPHER_BIP_GMAC_256 ?
369 		   "BIP-GMAC-256 " : "",
370 		   bss->mgmt_group_cipher & WPA_CIPHER_BIP_CMAC_256 ?
371 		   "BIP-CMAC-256 " : "",
372 		   bss->key_mgmt == 0 ? "N/A " : "",
373 		   bss->key_mgmt & WPA_KEY_MGMT_IEEE8021X ? "EAP " : "",
374 		   bss->key_mgmt & WPA_KEY_MGMT_PSK ? "PSK " : "",
375 		   bss->key_mgmt & WPA_KEY_MGMT_WPA_NONE ? "WPA-NONE " : "",
376 		   bss->key_mgmt & WPA_KEY_MGMT_FT_IEEE8021X ? "FT-EAP " : "",
377 		   bss->key_mgmt & WPA_KEY_MGMT_FT_PSK ? "FT-PSK " : "",
378 		   bss->key_mgmt & WPA_KEY_MGMT_IEEE8021X_SHA256 ?
379 		   "EAP-SHA256 " : "",
380 		   bss->key_mgmt & WPA_KEY_MGMT_PSK_SHA256 ?
381 		   "PSK-SHA256 " : "",
382 		   bss->key_mgmt & WPA_KEY_MGMT_OWE ? "OWE " : "",
383 		   bss->key_mgmt & WPA_KEY_MGMT_PASN ? "PASN " : "",
384 		   bss->key_mgmt & WPA_KEY_MGMT_OSEN ? "OSEN " : "",
385 		   bss->key_mgmt & WPA_KEY_MGMT_DPP ? "DPP " : "",
386 		   bss->key_mgmt & WPA_KEY_MGMT_IEEE8021X_SUITE_B ?
387 		   "EAP-SUITE-B " : "",
388 		   bss->key_mgmt & WPA_KEY_MGMT_IEEE8021X_SUITE_B_192 ?
389 		   "EAP-SUITE-B-192 " : "",
390 		   bss->key_mgmt & WPA_KEY_MGMT_IEEE8021X_SHA384 ?
391 		   "EAP-SHA384 " : "",
392 		   bss->rsn_capab & WPA_CAPABILITY_PREAUTH ? "PREAUTH " : "",
393 		   bss->rsn_capab & WPA_CAPABILITY_NO_PAIRWISE ?
394 		   "NO_PAIRWISE " : "",
395 		   bss->rsn_capab & WPA_CAPABILITY_MFPR ? "MFPR " : "",
396 		   bss->rsn_capab & WPA_CAPABILITY_MFPC ? "MFPC " : "",
397 		   bss->rsn_capab & WPA_CAPABILITY_PEERKEY_ENABLED ?
398 		   "PEERKEY " : "",
399 		   bss->rsn_capab & WPA_CAPABILITY_SPP_A_MSDU_CAPABLE ?
400 		   "SPP-A-MSDU-CAPAB " : "",
401 		   bss->rsn_capab & WPA_CAPABILITY_SPP_A_MSDU_REQUIRED ?
402 		   "SPP-A-MSDU-REQUIRED " : "",
403 		   bss->rsn_capab & WPA_CAPABILITY_PBAC ? "PBAC " : "",
404 		   bss->rsn_capab & WPA_CAPABILITY_OCVC ? "OCVC " : "",
405 		   bss->rsn_capab & WPA_CAPABILITY_EXT_KEY_ID_FOR_UNICAST ?
406 		   "ExtKeyID " : "");
407 }
408 
409 
bss_flush(struct wlantest * wt)410 void bss_flush(struct wlantest *wt)
411 {
412 	struct wlantest_bss *bss, *n;
413 	dl_list_for_each_safe(bss, n, &wt->bss, struct wlantest_bss, list)
414 		bss_deinit(bss);
415 }
416