[thirdparty/hostap.git] / wlantest / bss.c

/*
 * BSS list
 * Copyright (c) 2010, Jouni Malinen <j@w1.fi>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * Alternatively, this software may be distributed under the terms of BSD
 * license.
 *
 * See README and COPYING for more details.
 */

#include "utils/includes.h"

#include "utils/common.h"
#include "common/defs.h"
#include "common/ieee802_11_common.h"
#include "crypto/sha1.h"
#include "wlantest.h"


struct wlantest_bss * bss_get(struct wlantest *wt, const u8 *bssid)
{
	struct wlantest_bss *bss;

	if (bssid[0] & 0x01)
		return NULL; /* Skip group addressed frames */

	dl_list_for_each(bss, &wt->bss, struct wlantest_bss, list) {
		if (os_memcmp(bss->bssid, bssid, ETH_ALEN) == 0)
			return bss;
	}

	bss = os_zalloc(sizeof(*bss));
	if (bss == NULL)
		return NULL;
	dl_list_init(&bss->sta);
	dl_list_init(&bss->pmk);
	os_memcpy(bss->bssid, bssid, ETH_ALEN);
	dl_list_add(&wt->bss, &bss->list);
	wpa_printf(MSG_DEBUG, "Discovered new BSS - " MACSTR,
		   MAC2STR(bss->bssid));
	return bss;
}


void pmk_deinit(struct wlantest_pmk *pmk)
{
	dl_list_del(&pmk->list);
	os_free(pmk);
}


void bss_deinit(struct wlantest_bss *bss)
{
	struct wlantest_sta *sta, *n;
	struct wlantest_pmk *pmk, *np;
	dl_list_for_each_safe(sta, n, &bss->sta, struct wlantest_sta, list)
		sta_deinit(sta);
	dl_list_for_each_safe(pmk, np, &bss->pmk, struct wlantest_pmk, list)
		pmk_deinit(pmk);
	dl_list_del(&bss->list);
	os_free(bss);
}


static void bss_add_pmk(struct wlantest *wt, struct wlantest_bss *bss)
{
	struct wlantest_passphrase *p;
	struct wlantest_pmk *pmk;

	dl_list_for_each(p, &wt->passphrase, struct wlantest_passphrase, list)
	{
		if (!is_zero_ether_addr(p->bssid) &&
		    os_memcmp(p->bssid, bss->bssid, ETH_ALEN) != 0)
			continue;
		if (p->ssid_len &&
		    (p->ssid_len != bss->ssid_len ||
		     os_memcmp(p->ssid, bss->ssid, p->ssid_len) != 0))
			continue;

		pmk = os_zalloc(sizeof(*pmk));
		if (pmk == NULL)
			break;
		if (pbkdf2_sha1(p->passphrase, (char *) bss->ssid,
				bss->ssid_len, 4096,
				pmk->pmk, sizeof(pmk->pmk)) < 0) {
			os_free(pmk);
			continue;
		}

		wpa_printf(MSG_INFO, "Add possible PMK for BSSID " MACSTR
			   " based on passphrase '%s'",
			   MAC2STR(bss->bssid), p->passphrase);
		wpa_hexdump(MSG_DEBUG, "Possible PMK",
			    pmk->pmk, sizeof(pmk->pmk));
		dl_list_add(&bss->pmk, &pmk->list);
	}
}


void bss_update(struct wlantest *wt, struct wlantest_bss *bss,
		struct ieee802_11_elems *elems)
{
	struct wpa_ie_data data;
	int update = 0;

	if (elems->ssid == NULL || elems->ssid_len > 32) {
		wpa_printf(MSG_INFO, "Invalid or missing SSID in a Beacon "
			   "frame for " MACSTR, MAC2STR(bss->bssid));
		bss->parse_error_reported = 1;
		return;
	}

	if (bss->ssid_len != elems->ssid_len ||
	    os_memcmp(bss->ssid, elems->ssid, bss->ssid_len) != 0) {
		wpa_printf(MSG_DEBUG, "Store SSID '%s' for BSSID " MACSTR,
			   wpa_ssid_txt(elems->ssid, elems->ssid_len),
			   MAC2STR(bss->bssid));
		os_memcpy(bss->ssid, elems->ssid, elems->ssid_len);
		bss->ssid_len = elems->ssid_len;
		bss_add_pmk(wt, bss);
	}


	if (elems->rsn_ie == NULL) {
		if (bss->rsnie[0]) {
			wpa_printf(MSG_INFO, "BSS " MACSTR " - RSN IE removed",
				   MAC2STR(bss->bssid));
			bss->rsnie[0] = 0;
			update = 1;
		}
	} else {
		if (bss->rsnie[0] == 0 ||
		    os_memcmp(bss->rsnie, elems->rsn_ie - 2,
			      elems->rsn_ie_len + 2) != 0) {
			wpa_printf(MSG_INFO, "BSS " MACSTR " - RSN IE "
				   "stored", MAC2STR(bss->bssid));
			wpa_hexdump(MSG_DEBUG, "RSN IE", elems->rsn_ie - 2,
				    elems->rsn_ie_len + 2);
			update = 1;
		}
		os_memcpy(bss->rsnie, elems->rsn_ie - 2,
			  elems->rsn_ie_len + 2);
	}

	if (elems->wpa_ie == NULL) {
		if (bss->wpaie[0]) {
			wpa_printf(MSG_INFO, "BSS " MACSTR " - WPA IE removed",
				   MAC2STR(bss->bssid));
			bss->wpaie[0] = 0;
			update = 1;
		}
	} else {
		if (bss->wpaie[0] == 0 ||
		    os_memcmp(bss->wpaie, elems->wpa_ie - 2,
			      elems->wpa_ie_len + 2) != 0) {
			wpa_printf(MSG_INFO, "BSS " MACSTR " - WPA IE "
				   "stored", MAC2STR(bss->bssid));
			wpa_hexdump(MSG_DEBUG, "WPA IE", elems->wpa_ie - 2,
				    elems->wpa_ie_len + 2);
			update = 1;
		}
		os_memcpy(bss->wpaie, elems->wpa_ie - 2,
			  elems->wpa_ie_len + 2);
	}

	if (!update)
		return;

	bss->proto = 0;
	bss->pairwise_cipher = 0;
	bss->group_cipher = 0;
	bss->key_mgmt = 0;
	bss->rsn_capab = 0;
	bss->mgmt_group_cipher = 0;

	if (bss->wpaie[0]) {
		if (wpa_parse_wpa_ie_wpa(bss->wpaie, 2 + bss->wpaie[1], &data)
		    < 0) {
			wpa_printf(MSG_INFO, "Failed to parse WPA IE from "
				   MACSTR, MAC2STR(bss->bssid));
		} else {
			bss->proto |= data.proto;
			bss->pairwise_cipher |= data.pairwise_cipher;
			bss->group_cipher |= data.group_cipher;
			bss->key_mgmt |= data.key_mgmt;
			bss->rsn_capab = data.capabilities;
			bss->mgmt_group_cipher |= data.mgmt_group_cipher;
		}
	}

	if (bss->rsnie[0]) {
		if (wpa_parse_wpa_ie_rsn(bss->rsnie, 2 + bss->rsnie[1], &data)
		    < 0) {
			wpa_printf(MSG_INFO, "Failed to parse RSN IE from "
				   MACSTR, MAC2STR(bss->bssid));
		} else {
			bss->proto |= data.proto;
			bss->pairwise_cipher |= data.pairwise_cipher;
			bss->group_cipher |= data.group_cipher;
			bss->key_mgmt |= data.key_mgmt;
			bss->rsn_capab = data.capabilities;
			bss->mgmt_group_cipher |= data.mgmt_group_cipher;
		}
	}

	if (!(bss->proto & WPA_PROTO_RSN) ||
	    !(bss->rsn_capab & WPA_CAPABILITY_MFPC))
		bss->mgmt_group_cipher = 0;

	wpa_printf(MSG_INFO, "BSS " MACSTR
		   " proto=%s%s%s"
		   "pairwise=%s%s%s%s"
		   "group=%s%s%s%s%s%s"
		   "mgmt_group_cipher=%s"
		   "key_mgmt=%s%s%s%s%s%s%s%s"
		   "rsn_capab=%s%s%s%s%s",
		   MAC2STR(bss->bssid),
		   bss->proto == 0 ? "OPEN " : "",
		   bss->proto & WPA_PROTO_WPA ? "WPA " : "",
		   bss->proto & WPA_PROTO_RSN ? "WPA2 " : "",
		   bss->pairwise_cipher == 0 ? "N/A " : "",
		   bss->pairwise_cipher & WPA_CIPHER_NONE ? "NONE " : "",
		   bss->pairwise_cipher & WPA_CIPHER_TKIP ? "TKIP " : "",
		   bss->pairwise_cipher & WPA_CIPHER_CCMP ? "CCMP " : "",
		   bss->group_cipher == 0 ? "N/A " : "",
		   bss->group_cipher & WPA_CIPHER_NONE ? "NONE " : "",
		   bss->group_cipher & WPA_CIPHER_WEP40 ? "WEP40 " : "",
		   bss->group_cipher & WPA_CIPHER_WEP104 ? "WEP104 " : "",
		   bss->group_cipher & WPA_CIPHER_TKIP ? "TKIP " : "",
		   bss->group_cipher & WPA_CIPHER_CCMP ? "CCMP " : "",
		   bss->mgmt_group_cipher & WPA_CIPHER_AES_128_CMAC ? "BIP " :
		   "N/A ",
		   bss->key_mgmt == 0 ? "N/A " : "",
		   bss->key_mgmt & WPA_KEY_MGMT_IEEE8021X ? "EAP " : "",
		   bss->key_mgmt & WPA_KEY_MGMT_PSK ? "PSK " : "",
		   bss->key_mgmt & WPA_KEY_MGMT_WPA_NONE ? "WPA-NONE " : "",
		   bss->key_mgmt & WPA_KEY_MGMT_FT_IEEE8021X ? "FT-EAP " : "",
		   bss->key_mgmt & WPA_KEY_MGMT_FT_PSK ? "FT-PSK " : "",
		   bss->key_mgmt & WPA_KEY_MGMT_IEEE8021X_SHA256 ?
		   "EAP-SHA256 " : "",
		   bss->key_mgmt & WPA_KEY_MGMT_PSK_SHA256 ?
		   "PSK-SHA256 " : "",
		   bss->rsn_capab & WPA_CAPABILITY_PREAUTH ? "PREAUTH " : "",
		   bss->rsn_capab & WPA_CAPABILITY_NO_PAIRWISE ?
		   "NO_PAIRWISE " : "",
		   bss->rsn_capab & WPA_CAPABILITY_MFPR ? "MFPR " : "",
		   bss->rsn_capab & WPA_CAPABILITY_MFPC ? "MFPC " : "",
		   bss->rsn_capab & WPA_CAPABILITY_PEERKEY_ENABLED ?
		   "PEERKEY " : "");
}


void bss_flush(struct wlantest *wt)
{
	struct wlantest_bss *bss, *n;
	dl_list_for_each_safe(bss, n, &wt->bss, struct wlantest_bss, list)
		bss_deinit(bss);
}
Commit	Line	Data
d84d3893 JM	1	/*
	2	* BSS list
	3	* Copyright (c) 2010, 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"
327f7160	18	#include "common/defs.h"
b50111fb	19	#include "common/ieee802_11_common.h"
53650bca	20	#include "crypto/sha1.h"
d84d3893 JM	21	#include "wlantest.h"
	22
	23
	24	struct wlantest_bss * bss_get(struct wlantest wt, const u8 bssid)
	25	{
	26	struct wlantest_bss *bss;
	27
	28	if (bssid[0] & 0x01)
	29	return NULL; /* Skip group addressed frames */
	30
	31	dl_list_for_each(bss, &wt->bss, struct wlantest_bss, list) {
	32	if (os_memcmp(bss->bssid, bssid, ETH_ALEN) == 0)
	33	return bss;
	34	}
	35
	36	bss = os_zalloc(sizeof(*bss));
	37	if (bss == NULL)
	38	return NULL;
422ef7d2	39	dl_list_init(&bss->sta);
53650bca	40	dl_list_init(&bss->pmk);
d84d3893 JM	41	os_memcpy(bss->bssid, bssid, ETH_ALEN);
	42	dl_list_add(&wt->bss, &bss->list);
	43	wpa_printf(MSG_DEBUG, "Discovered new BSS - " MACSTR,
	44	MAC2STR(bss->bssid));
	45	return bss;
	46	}
	47
	48
d06df64d	49	void pmk_deinit(struct wlantest_pmk *pmk)
53650bca JM	50	{
	51	dl_list_del(&pmk->list);
	52	os_free(pmk);
	53	}
	54
	55
d84d3893 JM	56	void bss_deinit(struct wlantest_bss *bss)
d84d3893 JM	57	{
422ef7d2	58	struct wlantest_sta sta, n;
53650bca	59	struct wlantest_pmk pmk, np;
422ef7d2 JM	60	dl_list_for_each_safe(sta, n, &bss->sta, struct wlantest_sta, list)
422ef7d2 JM	61	sta_deinit(sta);
53650bca JM	62	dl_list_for_each_safe(pmk, np, &bss->pmk, struct wlantest_pmk, list)
53650bca JM	63	pmk_deinit(pmk);
d84d3893 JM	64	dl_list_del(&bss->list);
	65	os_free(bss);
	66	}
b50111fb JM	67
b50111fb JM	68
53650bca JM	69	static void bss_add_pmk(struct wlantest wt, struct wlantest_bss bss)
	70	{
	71	struct wlantest_passphrase *p;
	72	struct wlantest_pmk *pmk;
	73
	74	dl_list_for_each(p, &wt->passphrase, struct wlantest_passphrase, list)
	75	{
	76	if (!is_zero_ether_addr(p->bssid) &&
	77	os_memcmp(p->bssid, bss->bssid, ETH_ALEN) != 0)
	78	continue;
	79	if (p->ssid_len &&
	80	(p->ssid_len != bss->ssid_len \|\|
	81	os_memcmp(p->ssid, bss->ssid, p->ssid_len) != 0))
	82	continue;
	83
	84	pmk = os_zalloc(sizeof(*pmk));
	85	if (pmk == NULL)
	86	break;
	87	if (pbkdf2_sha1(p->passphrase, (char *) bss->ssid,
	88	bss->ssid_len, 4096,
	89	pmk->pmk, sizeof(pmk->pmk)) < 0) {
	90	os_free(pmk);
	91	continue;
	92	}
	93
	94	wpa_printf(MSG_INFO, "Add possible PMK for BSSID " MACSTR
	95	" based on passphrase '%s'",
	96	MAC2STR(bss->bssid), p->passphrase);
	97	wpa_hexdump(MSG_DEBUG, "Possible PMK",
	98	pmk->pmk, sizeof(pmk->pmk));
	99	dl_list_add(&bss->pmk, &pmk->list);
	100	}
	101	}
	102
	103
	104	void bss_update(struct wlantest wt, struct wlantest_bss bss,
	105	struct ieee802_11_elems *elems)
b50111fb	106	{
327f7160 JM	107	struct wpa_ie_data data;
	108	int update = 0;
	109
b50111fb JM	110	if (elems->ssid == NULL \|\| elems->ssid_len > 32) {
	111	wpa_printf(MSG_INFO, "Invalid or missing SSID in a Beacon "
	112	"frame for " MACSTR, MAC2STR(bss->bssid));
	113	bss->parse_error_reported = 1;
	114	return;
	115	}
	116
53650bca JM	117	if (bss->ssid_len != elems->ssid_len \|\|
	118	os_memcmp(bss->ssid, elems->ssid, bss->ssid_len) != 0) {
	119	wpa_printf(MSG_DEBUG, "Store SSID '%s' for BSSID " MACSTR,
	120	wpa_ssid_txt(elems->ssid, elems->ssid_len),
	121	MAC2STR(bss->bssid));
	122	os_memcpy(bss->ssid, elems->ssid, elems->ssid_len);
	123	bss->ssid_len = elems->ssid_len;
	124	bss_add_pmk(wt, bss);
	125	}
	126
b50111fb JM	127
	128	if (elems->rsn_ie == NULL) {
	129	if (bss->rsnie[0]) {
	130	wpa_printf(MSG_INFO, "BSS " MACSTR " - RSN IE removed",
	131	MAC2STR(bss->bssid));
	132	bss->rsnie[0] = 0;
327f7160	133	update = 1;
b50111fb JM	134	}
	135	} else {
	136	if (bss->rsnie[0] == 0 \|\|
	137	os_memcmp(bss->rsnie, elems->rsn_ie - 2,
	138	elems->rsn_ie_len + 2) != 0) {
	139	wpa_printf(MSG_INFO, "BSS " MACSTR " - RSN IE "
	140	"stored", MAC2STR(bss->bssid));
	141	wpa_hexdump(MSG_DEBUG, "RSN IE", elems->rsn_ie - 2,
	142	elems->rsn_ie_len + 2);
327f7160	143	update = 1;
b50111fb JM	144	}
	145	os_memcpy(bss->rsnie, elems->rsn_ie - 2,
	146	elems->rsn_ie_len + 2);
	147	}
	148
	149	if (elems->wpa_ie == NULL) {
	150	if (bss->wpaie[0]) {
	151	wpa_printf(MSG_INFO, "BSS " MACSTR " - WPA IE removed",
	152	MAC2STR(bss->bssid));
	153	bss->wpaie[0] = 0;
327f7160	154	update = 1;
b50111fb JM	155	}
	156	} else {
	157	if (bss->wpaie[0] == 0 \|\|
	158	os_memcmp(bss->wpaie, elems->wpa_ie - 2,
	159	elems->wpa_ie_len + 2) != 0) {
	160	wpa_printf(MSG_INFO, "BSS " MACSTR " - WPA IE "
	161	"stored", MAC2STR(bss->bssid));
	162	wpa_hexdump(MSG_DEBUG, "WPA IE", elems->wpa_ie - 2,
	163	elems->wpa_ie_len + 2);
327f7160	164	update = 1;
b50111fb JM	165	}
	166	os_memcpy(bss->wpaie, elems->wpa_ie - 2,
	167	elems->wpa_ie_len + 2);
	168	}
327f7160 JM	169
	170	if (!update)
	171	return;
	172
	173	bss->proto = 0;
	174	bss->pairwise_cipher = 0;
	175	bss->group_cipher = 0;
	176	bss->key_mgmt = 0;
	177	bss->rsn_capab = 0;
	178	bss->mgmt_group_cipher = 0;
	179
	180	if (bss->wpaie[0]) {
	181	if (wpa_parse_wpa_ie_wpa(bss->wpaie, 2 + bss->wpaie[1], &data)
	182	< 0) {
	183	wpa_printf(MSG_INFO, "Failed to parse WPA IE from "
	184	MACSTR, MAC2STR(bss->bssid));
	185	} else {
	186	bss->proto \|= data.proto;
	187	bss->pairwise_cipher \|= data.pairwise_cipher;
	188	bss->group_cipher \|= data.group_cipher;
	189	bss->key_mgmt \|= data.key_mgmt;
	190	bss->rsn_capab = data.capabilities;
	191	bss->mgmt_group_cipher \|= data.mgmt_group_cipher;
	192	}
	193	}
	194
	195	if (bss->rsnie[0]) {
	196	if (wpa_parse_wpa_ie_rsn(bss->rsnie, 2 + bss->rsnie[1], &data)
	197	< 0) {
	198	wpa_printf(MSG_INFO, "Failed to parse RSN IE from "
	199	MACSTR, MAC2STR(bss->bssid));
	200	} else {
	201	bss->proto \|= data.proto;
	202	bss->pairwise_cipher \|= data.pairwise_cipher;
	203	bss->group_cipher \|= data.group_cipher;
	204	bss->key_mgmt \|= data.key_mgmt;
	205	bss->rsn_capab = data.capabilities;
	206	bss->mgmt_group_cipher \|= data.mgmt_group_cipher;
	207	}
	208	}
	209
	210	if (!(bss->proto & WPA_PROTO_RSN) \|\|
	211	!(bss->rsn_capab & WPA_CAPABILITY_MFPC))
	212	bss->mgmt_group_cipher = 0;
	213
	214	wpa_printf(MSG_INFO, "BSS " MACSTR
	215	" proto=%s%s%s"
	216	"pairwise=%s%s%s%s"
	217	"group=%s%s%s%s%s%s"
	218	"mgmt_group_cipher=%s"
	219	"key_mgmt=%s%s%s%s%s%s%s%s"
	220	"rsn_capab=%s%s%s%s%s",
	221	MAC2STR(bss->bssid),
	222	bss->proto == 0 ? "OPEN " : "",
	223	bss->proto & WPA_PROTO_WPA ? "WPA " : "",
	224	bss->proto & WPA_PROTO_RSN ? "WPA2 " : "",
	225	bss->pairwise_cipher == 0 ? "N/A " : "",
	226	bss->pairwise_cipher & WPA_CIPHER_NONE ? "NONE " : "",
	227	bss->pairwise_cipher & WPA_CIPHER_TKIP ? "TKIP " : "",
	228	bss->pairwise_cipher & WPA_CIPHER_CCMP ? "CCMP " : "",
	229	bss->group_cipher == 0 ? "N/A " : "",
	230	bss->group_cipher & WPA_CIPHER_NONE ? "NONE " : "",
	231	bss->group_cipher & WPA_CIPHER_WEP40 ? "WEP40 " : "",
	232	bss->group_cipher & WPA_CIPHER_WEP104 ? "WEP104 " : "",
233	bss->group_cipher & WPA_CIPHER_TKIP ? "TKIP " : "",
234	bss->group_cipher & WPA_CIPHER_CCMP ? "CCMP " : "",
235	bss->mgmt_group_cipher & WPA_CIPHER_AES_128_CMAC ? "BIP " :
236	"N/A ",
237	bss->key_mgmt == 0 ? "N/A " : "",
238	bss->key_mgmt & WPA_KEY_MGMT_IEEE8021X ? "EAP " : "",
239	bss->key_mgmt & WPA_KEY_MGMT_PSK ? "PSK " : "",
240	bss->key_mgmt & WPA_KEY_MGMT_WPA_NONE ? "WPA-NONE " : "",
241	bss->key_mgmt & WPA_KEY_MGMT_FT_IEEE8021X ? "FT-EAP " : "",
242	bss->key_mgmt & WPA_KEY_MGMT_FT_PSK ? "FT-PSK " : "",
243	bss->key_mgmt & WPA_KEY_MGMT_IEEE8021X_SHA256 ?
244	"EAP-SHA256 " : "",
245	bss->key_mgmt & WPA_KEY_MGMT_PSK_SHA256 ?
246	"PSK-SHA256 " : "",
247	bss->rsn_capab & WPA_CAPABILITY_PREAUTH ? "PREAUTH " : "",
248	bss->rsn_capab & WPA_CAPABILITY_NO_PAIRWISE ?
249	"NO_PAIRWISE " : "",
250	bss->rsn_capab & WPA_CAPABILITY_MFPR ? "MFPR " : "",
251	bss->rsn_capab & WPA_CAPABILITY_MFPC ? "MFPC " : "",
252	bss->rsn_capab & WPA_CAPABILITY_PEERKEY_ENABLED ?
253	"PEERKEY " : "");
b50111fb	254	}
d356bd63 JM	255
	256
	257	void bss_flush(struct wlantest *wt)
	258	{
	259	struct wlantest_bss bss, n;
	260	dl_list_for_each_safe(bss, n, &wt->bss, struct wlantest_bss, list)
	261	bss_deinit(bss);
	262	}