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

/*
 * Received Data frame processing
 * Copyright (c) 2010, Jouni Malinen <j@w1.fi>
 *
 * This software may be distributed under the terms of the BSD license.
 * See README for more details.
 */

#include "utils/includes.h"
#include <linux/if_ether.h>

#include "utils/common.h"
#include "common/defs.h"
#include "common/ieee802_11_defs.h"
#include "wlantest.h"


static const char * data_stype(u16 stype)
{
	switch (stype) {
	case WLAN_FC_STYPE_DATA:
		return "DATA";
	case WLAN_FC_STYPE_DATA_CFACK:
		return "DATA-CFACK";
	case WLAN_FC_STYPE_DATA_CFPOLL:
		return "DATA-CFPOLL";
	case WLAN_FC_STYPE_DATA_CFACKPOLL:
		return "DATA-CFACKPOLL";
	case WLAN_FC_STYPE_NULLFUNC:
		return "NULLFUNC";
	case WLAN_FC_STYPE_CFACK:
		return "CFACK";
	case WLAN_FC_STYPE_CFPOLL:
		return "CFPOLL";
	case WLAN_FC_STYPE_CFACKPOLL:
		return "CFACKPOLL";
	case WLAN_FC_STYPE_QOS_DATA:
		return "QOSDATA";
	case WLAN_FC_STYPE_QOS_DATA_CFACK:
		return "QOSDATA-CFACK";
	case WLAN_FC_STYPE_QOS_DATA_CFPOLL:
		return "QOSDATA-CFPOLL";
	case WLAN_FC_STYPE_QOS_DATA_CFACKPOLL:
		return "QOSDATA-CFACKPOLL";
	case WLAN_FC_STYPE_QOS_NULL:
		return "QOS-NULL";
	case WLAN_FC_STYPE_QOS_CFPOLL:
		return "QOS-CFPOLL";
	case WLAN_FC_STYPE_QOS_CFACKPOLL:
		return "QOS-CFACKPOLL";
	}
	return "??";
}


static void rx_data_eth(struct wlantest *wt, const u8 *bssid,
			const u8 *sta_addr, const u8 *dst, const u8 *src,
			u16 ethertype, const u8 *data, size_t len, int prot,
			const u8 *peer_addr)
{
	switch (ethertype) {
	case ETH_P_PAE:
		rx_data_eapol(wt, dst, src, data, len, prot);
		break;
	case ETH_P_IP:
		rx_data_ip(wt, bssid, sta_addr, dst, src, data, len,
			   peer_addr);
		break;
	case 0x890d:
		rx_data_80211_encap(wt, bssid, sta_addr, dst, src, data, len);
		break;
	}
}


static void rx_data_process(struct wlantest *wt, const u8 *bssid,
			    const u8 *sta_addr,
			    const u8 *dst, const u8 *src,
			    const u8 *data, size_t len, int prot,
			    const u8 *peer_addr)
{
	if (len == 0)
		return;

	if (len >= 8 && os_memcmp(data, "\xaa\xaa\x03\x00\x00\x00", 6) == 0) {
		rx_data_eth(wt, bssid, sta_addr, dst, src,
			    WPA_GET_BE16(data + 6), data + 8, len - 8, prot,
			    peer_addr);
		return;
	}

	wpa_hexdump(MSG_DEBUG, "Unrecognized LLC", data, len > 8 ? 8 : len);
}


static u8 * try_all_ptk(struct wlantest *wt, int pairwise_cipher,
			const struct ieee80211_hdr *hdr,
			const u8 *data, size_t data_len, size_t *decrypted_len)
{
	struct wlantest_ptk *ptk;
	u8 *decrypted;
	int prev_level = wpa_debug_level;

	wpa_debug_level = MSG_WARNING;
	dl_list_for_each(ptk, &wt->ptk, struct wlantest_ptk, list) {
		decrypted = NULL;
		if ((pairwise_cipher == WPA_CIPHER_CCMP ||
		     pairwise_cipher == 0) && ptk->ptk_len == 48) {
			decrypted = ccmp_decrypt(ptk->ptk.tk1, hdr, data,
						 data_len, decrypted_len);
		}
		if ((pairwise_cipher == WPA_CIPHER_TKIP ||
		     pairwise_cipher == 0) && ptk->ptk_len == 64) {
			decrypted = tkip_decrypt(ptk->ptk.tk1, hdr, data,
						 data_len, decrypted_len);
		}
		if (decrypted) {
			wpa_debug_level = prev_level;
			add_note(wt, MSG_DEBUG, "Found PTK match from list of all known PTKs");
			return decrypted;
		}
	}
	wpa_debug_level = prev_level;

	return NULL;
}


static void rx_data_bss_prot_group(struct wlantest *wt,
				   const struct ieee80211_hdr *hdr,
				   const u8 *qos, const u8 *dst, const u8 *src,
				   const u8 *data, size_t len)
{
	struct wlantest_bss *bss;
	int keyid;
	u8 *decrypted;
	size_t dlen;
	u8 pn[6];

	bss = bss_get(wt, hdr->addr2);
	if (bss == NULL)
		return;
	if (len < 4) {
		add_note(wt, MSG_INFO, "Too short group addressed data frame");
		return;
	}

	if (bss->group_cipher & (WPA_CIPHER_TKIP | WPA_CIPHER_CCMP) &&
	    !(data[3] & 0x20)) {
		add_note(wt, MSG_INFO, "Expected TKIP/CCMP frame from "
			 MACSTR " did not have ExtIV bit set to 1",
			 MAC2STR(bss->bssid));
		return;
	}

	if (bss->group_cipher == WPA_CIPHER_TKIP) {
		if (data[3] & 0x1f) {
			add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
				 " used non-zero reserved bit",
				 MAC2STR(bss->bssid));
		}
		if (data[1] != ((data[0] | 0x20) & 0x7f)) {
			add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
				 " used incorrect WEPSeed[1] (was 0x%x, "
				 "expected 0x%x)",
				 MAC2STR(bss->bssid), data[1],
				 (data[0] | 0x20) & 0x7f);
		}
	} else if (bss->group_cipher == WPA_CIPHER_CCMP) {
		if (data[2] != 0 || (data[3] & 0x1f) != 0) {
			add_note(wt, MSG_INFO, "CCMP frame from " MACSTR
				 " used non-zero reserved bit",
				 MAC2STR(bss->bssid));
		}
	}

	keyid = data[3] >> 6;
	if (bss->gtk_len[keyid] == 0 && bss->group_cipher != WPA_CIPHER_WEP40)
	{
		add_note(wt, MSG_MSGDUMP, "No GTK known to decrypt the frame "
			 "(A2=" MACSTR " KeyID=%d)",
			 MAC2STR(hdr->addr2), keyid);
		return;
	}

	if (bss->group_cipher == WPA_CIPHER_TKIP)
		tkip_get_pn(pn, data);
	else if (bss->group_cipher == WPA_CIPHER_WEP40)
		goto skip_replay_det;
	else
		ccmp_get_pn(pn, data);
	if (os_memcmp(pn, bss->rsc[keyid], 6) <= 0) {
		u16 seq_ctrl = le_to_host16(hdr->seq_ctrl);
		add_note(wt, MSG_INFO, "CCMP/TKIP replay detected: A1=" MACSTR
			 " A2=" MACSTR " A3=" MACSTR " seq=%u frag=%u",
			 MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
			 MAC2STR(hdr->addr3),
			 WLAN_GET_SEQ_SEQ(seq_ctrl),
			 WLAN_GET_SEQ_FRAG(seq_ctrl));
		wpa_hexdump(MSG_INFO, "RX PN", pn, 6);
		wpa_hexdump(MSG_INFO, "RSC", bss->rsc[keyid], 6);
	}

skip_replay_det:
	if (bss->group_cipher == WPA_CIPHER_TKIP)
		decrypted = tkip_decrypt(bss->gtk[keyid], hdr, data, len,
					 &dlen);
	else if (bss->group_cipher == WPA_CIPHER_WEP40)
		decrypted = wep_decrypt(wt, hdr, data, len, &dlen);
	else
		decrypted = ccmp_decrypt(bss->gtk[keyid], hdr, data, len,
					 &dlen);
	if (decrypted) {
		rx_data_process(wt, bss->bssid, NULL, dst, src, decrypted,
				dlen, 1, NULL);
		os_memcpy(bss->rsc[keyid], pn, 6);
		write_pcap_decrypted(wt, (const u8 *) hdr, 24 + (qos ? 2 : 0),
				     decrypted, dlen);
	} else
		add_note(wt, MSG_DEBUG, "Failed to decrypt frame");
	os_free(decrypted);
}


static void rx_data_bss_prot(struct wlantest *wt,
			     const struct ieee80211_hdr *hdr, const u8 *qos,
			     const u8 *dst, const u8 *src, const u8 *data,
			     size_t len)
{
	struct wlantest_bss *bss;
	struct wlantest_sta *sta, *sta2;
	int keyid;
	u16 fc = le_to_host16(hdr->frame_control);
	u8 *decrypted;
	size_t dlen;
	int tid;
	u8 pn[6], *rsc;
	struct wlantest_tdls *tdls = NULL, *found;
	const u8 *tk = NULL;
	int ptk_iter_done = 0;
	int try_ptk_iter = 0;

	if (hdr->addr1[0] & 0x01) {
		rx_data_bss_prot_group(wt, hdr, qos, dst, src, data, len);
		return;
	}

	if (fc & WLAN_FC_TODS) {
		bss = bss_get(wt, hdr->addr1);
		if (bss == NULL)
			return;
		sta = sta_get(bss, hdr->addr2);
		if (sta)
			sta->counters[WLANTEST_STA_COUNTER_PROT_DATA_TX]++;
	} else if (fc & WLAN_FC_FROMDS) {
		bss = bss_get(wt, hdr->addr2);
		if (bss == NULL)
			return;
		sta = sta_get(bss, hdr->addr1);
	} else {
		bss = bss_get(wt, hdr->addr3);
		if (bss == NULL)
			return;
		sta = sta_find(bss, hdr->addr2);
		sta2 = sta_find(bss, hdr->addr1);
		if (sta == NULL || sta2 == NULL)
			return;
		found = NULL;
		dl_list_for_each(tdls, &bss->tdls, struct wlantest_tdls, list)
		{
			if ((tdls->init == sta && tdls->resp == sta2) ||
			    (tdls->init == sta2 && tdls->resp == sta)) {
				found = tdls;
				if (tdls->link_up)
					break;
			}
		}
		if (found) {
			if (!found->link_up)
				add_note(wt, MSG_DEBUG,
					 "TDLS: Link not up, but Data "
					 "frame seen");
			tk = found->tpk.tk;
			tdls = found;
		}
	}
	if ((sta == NULL ||
	     (!sta->ptk_set && sta->pairwise_cipher != WPA_CIPHER_WEP40)) &&
	    tk == NULL) {
		add_note(wt, MSG_MSGDUMP, "No PTK known to decrypt the frame");
		if (dl_list_empty(&wt->ptk))
			return;
		try_ptk_iter = 1;
	}

	if (len < 4) {
		add_note(wt, MSG_INFO, "Too short encrypted data frame");
		return;
	}

	if (sta == NULL)
		return;
	if (sta->pairwise_cipher & (WPA_CIPHER_TKIP | WPA_CIPHER_CCMP) &&
	    !(data[3] & 0x20)) {
		add_note(wt, MSG_INFO, "Expected TKIP/CCMP frame from "
			 MACSTR " did not have ExtIV bit set to 1",
			 MAC2STR(src));
		return;
	}

	if (tk == NULL && sta->pairwise_cipher == WPA_CIPHER_TKIP) {
		if (data[3] & 0x1f) {
			add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
				 " used non-zero reserved bit",
				 MAC2STR(hdr->addr2));
		}
		if (data[1] != ((data[0] | 0x20) & 0x7f)) {
			add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
				 " used incorrect WEPSeed[1] (was 0x%x, "
				 "expected 0x%x)",
				 MAC2STR(hdr->addr2), data[1],
				 (data[0] | 0x20) & 0x7f);
		}
	} else if (tk || sta->pairwise_cipher == WPA_CIPHER_CCMP) {
		if (data[2] != 0 || (data[3] & 0x1f) != 0) {
			add_note(wt, MSG_INFO, "CCMP frame from " MACSTR
				 " used non-zero reserved bit",
				 MAC2STR(hdr->addr2));
		}
	}

	keyid = data[3] >> 6;
	if (keyid != 0) {
		add_note(wt, MSG_INFO, "Unexpected non-zero KeyID %d in "
			 "individually addressed Data frame from " MACSTR,
			 keyid, MAC2STR(hdr->addr2));
	}

	if (qos) {
		tid = qos[0] & 0x0f;
		if (fc & WLAN_FC_TODS)
			sta->tx_tid[tid]++;
		else
			sta->rx_tid[tid]++;
	} else {
		tid = 0;
		if (fc & WLAN_FC_TODS)
			sta->tx_tid[16]++;
		else
			sta->rx_tid[16]++;
	}
	if (tk) {
		if (os_memcmp(hdr->addr2, tdls->init->addr, ETH_ALEN) == 0)
			rsc = tdls->rsc_init[tid];
		else
			rsc = tdls->rsc_resp[tid];
	} else if (fc & WLAN_FC_TODS)
		rsc = sta->rsc_tods[tid];
	else
		rsc = sta->rsc_fromds[tid];


	if (tk == NULL && sta->pairwise_cipher == WPA_CIPHER_TKIP)
		tkip_get_pn(pn, data);
	else if (sta->pairwise_cipher == WPA_CIPHER_WEP40)
		goto skip_replay_det;
	else
		ccmp_get_pn(pn, data);
	if (os_memcmp(pn, rsc, 6) <= 0) {
		u16 seq_ctrl = le_to_host16(hdr->seq_ctrl);
		add_note(wt, MSG_INFO, "CCMP/TKIP replay detected: A1=" MACSTR
			 " A2=" MACSTR " A3=" MACSTR " seq=%u frag=%u",
			 MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
			 MAC2STR(hdr->addr3),
			 WLAN_GET_SEQ_SEQ(seq_ctrl),
			 WLAN_GET_SEQ_FRAG(seq_ctrl));
		wpa_hexdump(MSG_INFO, "RX PN", pn, 6);
		wpa_hexdump(MSG_INFO, "RSC", rsc, 6);
	}

skip_replay_det:
	if (tk)
		decrypted = ccmp_decrypt(tk, hdr, data, len, &dlen);
	else if (sta->pairwise_cipher == WPA_CIPHER_TKIP)
		decrypted = tkip_decrypt(sta->ptk.tk1, hdr, data, len, &dlen);
	else if (sta->pairwise_cipher == WPA_CIPHER_WEP40)
		decrypted = wep_decrypt(wt, hdr, data, len, &dlen);
	else if (sta->ptk_set)
		decrypted = ccmp_decrypt(sta->ptk.tk1, hdr, data, len, &dlen);
	else {
		decrypted = try_all_ptk(wt, sta->pairwise_cipher, hdr, data,
					len, &dlen);
		ptk_iter_done = 1;
	}
	if (!decrypted && !ptk_iter_done) {
		decrypted = try_all_ptk(wt, sta->pairwise_cipher, hdr, data,
					len, &dlen);
		if (decrypted) {
			add_note(wt, MSG_DEBUG, "Current PTK did not work, but found a match from all known PTKs");
		}
	}
	if (decrypted) {
		u16 fc = le_to_host16(hdr->frame_control);
		const u8 *peer_addr = NULL;
		if (!(fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)))
			peer_addr = hdr->addr1;
		os_memcpy(rsc, pn, 6);
		rx_data_process(wt, bss->bssid, sta->addr, dst, src, decrypted,
				dlen, 1, peer_addr);
		write_pcap_decrypted(wt, (const u8 *) hdr, 24 + (qos ? 2 : 0),
				     decrypted, dlen);
	} else if (!try_ptk_iter)
		add_note(wt, MSG_DEBUG, "Failed to decrypt frame");
	os_free(decrypted);
}


static void rx_data_bss(struct wlantest *wt, const struct ieee80211_hdr *hdr,
			const u8 *qos, const u8 *dst, const u8 *src,
			const u8 *data, size_t len)
{
	u16 fc = le_to_host16(hdr->frame_control);
	int prot = !!(fc & WLAN_FC_ISWEP);

	if (qos) {
		u8 ack = (qos[0] & 0x60) >> 5;
		wpa_printf(MSG_MSGDUMP, "BSS DATA: " MACSTR " -> " MACSTR
			   " len=%u%s tid=%u%s%s",
			   MAC2STR(src), MAC2STR(dst), (unsigned int) len,
			   prot ? " Prot" : "", qos[0] & 0x0f,
			   (qos[0] & 0x10) ? " EOSP" : "",
			   ack == 0 ? "" :
			   (ack == 1 ? " NoAck" :
			    (ack == 2 ? " NoExpAck" : " BA")));
	} else {
		wpa_printf(MSG_MSGDUMP, "BSS DATA: " MACSTR " -> " MACSTR
			   " len=%u%s",
			   MAC2STR(src), MAC2STR(dst), (unsigned int) len,
			   prot ? " Prot" : "");
	}

	if (prot)
		rx_data_bss_prot(wt, hdr, qos, dst, src, data, len);
	else {
		const u8 *bssid, *sta_addr, *peer_addr;
		struct wlantest_bss *bss;

		if (fc & WLAN_FC_TODS) {
			bssid = hdr->addr1;
			sta_addr = hdr->addr2;
			peer_addr = NULL;
		} else if (fc & WLAN_FC_FROMDS) {
			bssid = hdr->addr2;
			sta_addr = hdr->addr1;
			peer_addr = NULL;
		} else {
			bssid = hdr->addr3;
			sta_addr = hdr->addr2;
			peer_addr = hdr->addr1;
		}

		bss = bss_get(wt, bssid);
		if (bss) {
			struct wlantest_sta *sta = sta_get(bss, sta_addr);

			if (sta) {
				if (qos) {
					int tid = qos[0] & 0x0f;
					if (fc & WLAN_FC_TODS)
						sta->tx_tid[tid]++;
					else
						sta->rx_tid[tid]++;
				} else {
					if (fc & WLAN_FC_TODS)
						sta->tx_tid[16]++;
					else
						sta->rx_tid[16]++;
				}
			}
		}

		rx_data_process(wt, bssid, sta_addr, dst, src, data, len, 0,
				peer_addr);
	}
}


static struct wlantest_tdls * get_tdls(struct wlantest *wt, const u8 *bssid,
				       const u8 *sta1_addr,
				       const u8 *sta2_addr)
{
	struct wlantest_bss *bss;
	struct wlantest_sta *sta1, *sta2;
	struct wlantest_tdls *tdls, *found = NULL;

	bss = bss_find(wt, bssid);
	if (bss == NULL)
		return NULL;
	sta1 = sta_find(bss, sta1_addr);
	if (sta1 == NULL)
		return NULL;
	sta2 = sta_find(bss, sta2_addr);
	if (sta2 == NULL)
		return NULL;

	dl_list_for_each(tdls, &bss->tdls, struct wlantest_tdls, list) {
		if ((tdls->init == sta1 && tdls->resp == sta2) ||
		    (tdls->init == sta2 && tdls->resp == sta1)) {
			found = tdls;
			if (tdls->link_up)
				break;
		}
	}

	return found;
}


static void add_direct_link(struct wlantest *wt, const u8 *bssid,
			    const u8 *sta1_addr, const u8 *sta2_addr)
{
	struct wlantest_tdls *tdls;

	tdls = get_tdls(wt, bssid, sta1_addr, sta2_addr);
	if (tdls == NULL)
		return;

	if (tdls->link_up)
		tdls->counters[WLANTEST_TDLS_COUNTER_VALID_DIRECT_LINK]++;
	else
		tdls->counters[WLANTEST_TDLS_COUNTER_INVALID_DIRECT_LINK]++;
}


static void add_ap_path(struct wlantest *wt, const u8 *bssid,
			const u8 *sta1_addr, const u8 *sta2_addr)
{
	struct wlantest_tdls *tdls;

	tdls = get_tdls(wt, bssid, sta1_addr, sta2_addr);
	if (tdls == NULL)
		return;

	if (tdls->link_up)
		tdls->counters[WLANTEST_TDLS_COUNTER_INVALID_AP_PATH]++;
	else
		tdls->counters[WLANTEST_TDLS_COUNTER_VALID_AP_PATH]++;
}


void rx_data(struct wlantest *wt, const u8 *data, size_t len)
{
	const struct ieee80211_hdr *hdr;
	u16 fc, stype;
	size_t hdrlen;
	const u8 *qos = NULL;

	if (len < 24)
		return;

	hdr = (const struct ieee80211_hdr *) data;
	fc = le_to_host16(hdr->frame_control);
	stype = WLAN_FC_GET_STYPE(fc);
	hdrlen = 24;
	if ((fc & (WLAN_FC_TODS | WLAN_FC_FROMDS)) ==
	    (WLAN_FC_TODS | WLAN_FC_FROMDS))
		hdrlen += ETH_ALEN;
	if (stype & 0x08) {
		qos = data + hdrlen;
		hdrlen += 2;
	}
	if (len < hdrlen)
		return;
	wt->rx_data++;

	switch (fc & (WLAN_FC_TODS | WLAN_FC_FROMDS)) {
	case 0:
		wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s IBSS DA=" MACSTR " SA="
			   MACSTR " BSSID=" MACSTR,
			   data_stype(WLAN_FC_GET_STYPE(fc)),
			   fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
			   fc & WLAN_FC_ISWEP ? " Prot" : "",
			   MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
			   MAC2STR(hdr->addr3));
		add_direct_link(wt, hdr->addr3, hdr->addr1, hdr->addr2);
		rx_data_bss(wt, hdr, qos, hdr->addr1, hdr->addr2,
			    data + hdrlen, len - hdrlen);
		break;
	case WLAN_FC_FROMDS:
		wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s FromDS DA=" MACSTR
			   " BSSID=" MACSTR " SA=" MACSTR,
			   data_stype(WLAN_FC_GET_STYPE(fc)),
			   fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
			   fc & WLAN_FC_ISWEP ? " Prot" : "",
			   MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
			   MAC2STR(hdr->addr3));
		add_ap_path(wt, hdr->addr2, hdr->addr1, hdr->addr3);
		rx_data_bss(wt, hdr, qos, hdr->addr1, hdr->addr3,
			    data + hdrlen, len - hdrlen);
		break;
	case WLAN_FC_TODS:
		wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s ToDS BSSID=" MACSTR
			   " SA=" MACSTR " DA=" MACSTR,
			   data_stype(WLAN_FC_GET_STYPE(fc)),
			   fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
			   fc & WLAN_FC_ISWEP ? " Prot" : "",
			   MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
			   MAC2STR(hdr->addr3));
		add_ap_path(wt, hdr->addr1, hdr->addr3, hdr->addr2);
		rx_data_bss(wt, hdr, qos, hdr->addr3, hdr->addr2,
			    data + hdrlen, len - hdrlen);
		break;
	case WLAN_FC_TODS | WLAN_FC_FROMDS:
		wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s WDS RA=" MACSTR " TA="
			   MACSTR " DA=" MACSTR " SA=" MACSTR,
			   data_stype(WLAN_FC_GET_STYPE(fc)),
			   fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
			   fc & WLAN_FC_ISWEP ? " Prot" : "",
			   MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
			   MAC2STR(hdr->addr3),
			   MAC2STR((const u8 *) (hdr + 1)));
		break;
	}
}
Commit	Line	Data
2d73f0a8 JM	1	/*
	2	* Received Data frame processing
	3	* Copyright (c) 2010, Jouni Malinen <j@w1.fi>
	4	*
0f3d578e JM	5	* This software may be distributed under the terms of the BSD license.
0f3d578e JM	6	* See README for more details.
2d73f0a8 JM	7	*/
	8
	9	#include "utils/includes.h"
ee3b84be	10	#include <linux/if_ether.h>
2d73f0a8 JM	11
2d73f0a8 JM	12	#include "utils/common.h"
a9dd29d3	13	#include "common/defs.h"
2d73f0a8 JM	14	#include "common/ieee802_11_defs.h"
2d73f0a8 JM	15	#include "wlantest.h"
a0530dff	16
2d73f0a8 JM	17
	18	static const char * data_stype(u16 stype)
	19	{
	20	switch (stype) {
	21	case WLAN_FC_STYPE_DATA:
	22	return "DATA";
	23	case WLAN_FC_STYPE_DATA_CFACK:
	24	return "DATA-CFACK";
	25	case WLAN_FC_STYPE_DATA_CFPOLL:
	26	return "DATA-CFPOLL";
	27	case WLAN_FC_STYPE_DATA_CFACKPOLL:
	28	return "DATA-CFACKPOLL";
	29	case WLAN_FC_STYPE_NULLFUNC:
	30	return "NULLFUNC";
	31	case WLAN_FC_STYPE_CFACK:
	32	return "CFACK";
	33	case WLAN_FC_STYPE_CFPOLL:
	34	return "CFPOLL";
	35	case WLAN_FC_STYPE_CFACKPOLL:
	36	return "CFACKPOLL";
	37	case WLAN_FC_STYPE_QOS_DATA:
	38	return "QOSDATA";
	39	case WLAN_FC_STYPE_QOS_DATA_CFACK:
	40	return "QOSDATA-CFACK";
	41	case WLAN_FC_STYPE_QOS_DATA_CFPOLL:
	42	return "QOSDATA-CFPOLL";
	43	case WLAN_FC_STYPE_QOS_DATA_CFACKPOLL:
	44	return "QOSDATA-CFACKPOLL";
	45	case WLAN_FC_STYPE_QOS_NULL:
	46	return "QOS-NULL";
	47	case WLAN_FC_STYPE_QOS_CFPOLL:
	48	return "QOS-CFPOLL";
	49	case WLAN_FC_STYPE_QOS_CFACKPOLL:
	50	return "QOS-CFACKPOLL";
	51	}
	52	return "??";
	53	}
	54
	55
ee3b84be JM	56	static void rx_data_eth(struct wlantest wt, const u8 bssid,
ee3b84be JM	57	const u8 sta_addr, const u8 dst, const u8 *src,
244c9303 JM	58	u16 ethertype, const u8 *data, size_t len, int prot,
244c9303 JM	59	const u8 *peer_addr)
32234bba	60	{
ee3b84be JM	61	switch (ethertype) {
ee3b84be JM	62	case ETH_P_PAE:
32234bba	63	rx_data_eapol(wt, dst, src, data, len, prot);
ee3b84be JM	64	break;
ee3b84be JM	65	case ETH_P_IP:
244c9303 JM	66	rx_data_ip(wt, bssid, sta_addr, dst, src, data, len,
244c9303 JM	67	peer_addr);
ee3b84be	68	break;
b39f5834 JM	69	case 0x890d:
	70	rx_data_80211_encap(wt, bssid, sta_addr, dst, src, data, len);
	71	break;
ee3b84be	72	}
32234bba JM	73	}
	74
	75
ee3b84be JM	76	static void rx_data_process(struct wlantest wt, const u8 bssid,
	77	const u8 *sta_addr,
	78	const u8 dst, const u8 src,
244c9303 JM	79	const u8 *data, size_t len, int prot,
244c9303 JM	80	const u8 *peer_addr)
32234bba	81	{
4bc82fc7 JM	82	if (len == 0)
	83	return;
	84
32234bba	85	if (len >= 8 && os_memcmp(data, "\xaa\xaa\x03\x00\x00\x00", 6) == 0) {
ee3b84be	86	rx_data_eth(wt, bssid, sta_addr, dst, src,
244c9303 JM	87	WPA_GET_BE16(data + 6), data + 8, len - 8, prot,
244c9303 JM	88	peer_addr);
32234bba JM	89	return;
	90	}
	91
4bc82fc7	92	wpa_hexdump(MSG_DEBUG, "Unrecognized LLC", data, len > 8 ? 8 : len);
32234bba JM	93	}
	94
	95
a0530dff JM	96	static u8 * try_all_ptk(struct wlantest *wt, int pairwise_cipher,
	97	const struct ieee80211_hdr *hdr,
	98	const u8 data, size_t data_len, size_t decrypted_len)
	99	{
	100	struct wlantest_ptk *ptk;
	101	u8 *decrypted;
	102	int prev_level = wpa_debug_level;
	103
	104	wpa_debug_level = MSG_WARNING;
	105	dl_list_for_each(ptk, &wt->ptk, struct wlantest_ptk, list) {
	106	decrypted = NULL;
	107	if ((pairwise_cipher == WPA_CIPHER_CCMP \|\|
	108	pairwise_cipher == 0) && ptk->ptk_len == 48) {
	109	decrypted = ccmp_decrypt(ptk->ptk.tk1, hdr, data,
	110	data_len, decrypted_len);
	111	}
	112	if ((pairwise_cipher == WPA_CIPHER_TKIP \|\|
	113	pairwise_cipher == 0) && ptk->ptk_len == 64) {
	114	decrypted = tkip_decrypt(ptk->ptk.tk1, hdr, data,
	115	data_len, decrypted_len);
	116	}
	117	if (decrypted) {
	118	wpa_debug_level = prev_level;
	119	add_note(wt, MSG_DEBUG, "Found PTK match from list of all known PTKs");
	120	return decrypted;
	121	}
	122	}
	123	wpa_debug_level = prev_level;
	124
	125	return NULL;
	126	}
	127
	128
d318c534 JM	129	static void rx_data_bss_prot_group(struct wlantest *wt,
	130	const struct ieee80211_hdr *hdr,
	131	const u8 qos, const u8 dst, const u8 *src,
	132	const u8 *data, size_t len)
	133	{
	134	struct wlantest_bss *bss;
	135	int keyid;
07d0a5be JM	136	u8 *decrypted;
07d0a5be JM	137	size_t dlen;
2edd5c23	138	u8 pn[6];
d318c534 JM	139
	140	bss = bss_get(wt, hdr->addr2);
	141	if (bss == NULL)
	142	return;
	143	if (len < 4) {
e4d99217	144	add_note(wt, MSG_INFO, "Too short group addressed data frame");
d318c534 JM	145	return;
	146	}
	147
20062114 JM	148	if (bss->group_cipher & (WPA_CIPHER_TKIP \| WPA_CIPHER_CCMP) &&
20062114 JM	149	!(data[3] & 0x20)) {
e4d99217 JM	150	add_note(wt, MSG_INFO, "Expected TKIP/CCMP frame from "
	151	MACSTR " did not have ExtIV bit set to 1",
	152	MAC2STR(bss->bssid));
	153	return;
20062114 JM	154	}
20062114 JM	155
16b8b6ea JM	156	if (bss->group_cipher == WPA_CIPHER_TKIP) {
16b8b6ea JM	157	if (data[3] & 0x1f) {
e4d99217 JM	158	add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
	159	" used non-zero reserved bit",
	160	MAC2STR(bss->bssid));
16b8b6ea JM	161	}
16b8b6ea JM	162	if (data[1] != ((data[0] \| 0x20) & 0x7f)) {
e4d99217 JM	163	add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
	164	" used incorrect WEPSeed[1] (was 0x%x, "
	165	"expected 0x%x)",
	166	MAC2STR(bss->bssid), data[1],
	167	(data[0] \| 0x20) & 0x7f);
16b8b6ea JM	168	}
	169	} else if (bss->group_cipher == WPA_CIPHER_CCMP) {
	170	if (data[2] != 0 \|\| (data[3] & 0x1f) != 0) {
e4d99217 JM	171	add_note(wt, MSG_INFO, "CCMP frame from " MACSTR
	172	" used non-zero reserved bit",
	173	MAC2STR(bss->bssid));
16b8b6ea JM	174	}
	175	}
	176
d318c534	177	keyid = data[3] >> 6;
2e479416 JM	178	if (bss->gtk_len[keyid] == 0 && bss->group_cipher != WPA_CIPHER_WEP40)
2e479416 JM	179	{
e4d99217 JM	180	add_note(wt, MSG_MSGDUMP, "No GTK known to decrypt the frame "
	181	"(A2=" MACSTR " KeyID=%d)",
	182	MAC2STR(hdr->addr2), keyid);
d318c534 JM	183	return;
	184	}
	185
4dac8453 JM	186	if (bss->group_cipher == WPA_CIPHER_TKIP)
4dac8453 JM	187	tkip_get_pn(pn, data);
2e479416 JM	188	else if (bss->group_cipher == WPA_CIPHER_WEP40)
2e479416 JM	189	goto skip_replay_det;
4dac8453 JM	190	else
4dac8453 JM	191	ccmp_get_pn(pn, data);
2edd5c23	192	if (os_memcmp(pn, bss->rsc[keyid], 6) <= 0) {
01b397dd	193	u16 seq_ctrl = le_to_host16(hdr->seq_ctrl);
e4d99217 JM	194	add_note(wt, MSG_INFO, "CCMP/TKIP replay detected: A1=" MACSTR
	195	" A2=" MACSTR " A3=" MACSTR " seq=%u frag=%u",
	196	MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
	197	MAC2STR(hdr->addr3),
	198	WLAN_GET_SEQ_SEQ(seq_ctrl),
	199	WLAN_GET_SEQ_FRAG(seq_ctrl));
2edd5c23 JM	200	wpa_hexdump(MSG_INFO, "RX PN", pn, 6);
	201	wpa_hexdump(MSG_INFO, "RSC", bss->rsc[keyid], 6);
	202	}
	203
2e479416	204	skip_replay_det:
2924b0eb JM	205	if (bss->group_cipher == WPA_CIPHER_TKIP)
	206	decrypted = tkip_decrypt(bss->gtk[keyid], hdr, data, len,
	207	&dlen);
2e479416 JM	208	else if (bss->group_cipher == WPA_CIPHER_WEP40)
2e479416 JM	209	decrypted = wep_decrypt(wt, hdr, data, len, &dlen);
2924b0eb JM	210	else
	211	decrypted = ccmp_decrypt(bss->gtk[keyid], hdr, data, len,
	212	&dlen);
2edd5c23	213	if (decrypted) {
ee3b84be	214	rx_data_process(wt, bss->bssid, NULL, dst, src, decrypted,
244c9303	215	dlen, 1, NULL);
2edd5c23	216	os_memcpy(bss->rsc[keyid], pn, 6);
64f45d07 JM	217	write_pcap_decrypted(wt, (const u8 *) hdr, 24 + (qos ? 2 : 0),
64f45d07 JM	218	decrypted, dlen);
e4d99217 JM	219	} else
e4d99217 JM	220	add_note(wt, MSG_DEBUG, "Failed to decrypt frame");
07d0a5be	221	os_free(decrypted);
d318c534 JM	222	}
	223
	224
32234bba JM	225	static void rx_data_bss_prot(struct wlantest *wt,
	226	const struct ieee80211_hdr hdr, const u8 qos,
	227	const u8 dst, const u8 src, const u8 *data,
	228	size_t len)
	229	{
d318c534	230	struct wlantest_bss *bss;
b39f5834	231	struct wlantest_sta sta, sta2;
d318c534 JM	232	int keyid;
	233	u16 fc = le_to_host16(hdr->frame_control);
	234	u8 *decrypted;
	235	size_t dlen;
2edd5c23 JM	236	int tid;
2edd5c23 JM	237	u8 pn[6], *rsc;
ace4e460	238	struct wlantest_tdls tdls = NULL, found;
b39f5834	239	const u8 *tk = NULL;
a0530dff JM	240	int ptk_iter_done = 0;
a0530dff JM	241	int try_ptk_iter = 0;
d318c534 JM	242
	243	if (hdr->addr1[0] & 0x01) {
	244	rx_data_bss_prot_group(wt, hdr, qos, dst, src, data, len);
	245	return;
	246	}
	247
	248	if (fc & WLAN_FC_TODS) {
	249	bss = bss_get(wt, hdr->addr1);
	250	if (bss == NULL)
	251	return;
	252	sta = sta_get(bss, hdr->addr2);
0a9ddd92 JM	253	if (sta)
0a9ddd92 JM	254	sta->counters[WLANTEST_STA_COUNTER_PROT_DATA_TX]++;
b39f5834	255	} else if (fc & WLAN_FC_FROMDS) {
d318c534 JM	256	bss = bss_get(wt, hdr->addr2);
	257	if (bss == NULL)
	258	return;
	259	sta = sta_get(bss, hdr->addr1);
b39f5834 JM	260	} else {
	261	bss = bss_get(wt, hdr->addr3);
	262	if (bss == NULL)
	263	return;
	264	sta = sta_find(bss, hdr->addr2);
	265	sta2 = sta_find(bss, hdr->addr1);
	266	if (sta == NULL \|\| sta2 == NULL)
	267	return;
ace4e460	268	found = NULL;
b39f5834 JM	269	dl_list_for_each(tdls, &bss->tdls, struct wlantest_tdls, list)
	270	{
	271	if ((tdls->init == sta && tdls->resp == sta2) \|\|
	272	(tdls->init == sta2 && tdls->resp == sta)) {
ace4e460 JM	273	found = tdls;
	274	if (tdls->link_up)
	275	break;
b39f5834 JM	276	}
b39f5834 JM	277	}
ace4e460 JM	278	if (found) {
	279	if (!found->link_up)
	280	add_note(wt, MSG_DEBUG,
	281	"TDLS: Link not up, but Data "
	282	"frame seen");
	283	tk = found->tpk.tk;
	284	tdls = found;
	285	}
d318c534	286	}
2e479416 JM	287	if ((sta == NULL \|\|
	288	(!sta->ptk_set && sta->pairwise_cipher != WPA_CIPHER_WEP40)) &&
	289	tk == NULL) {
e4d99217	290	add_note(wt, MSG_MSGDUMP, "No PTK known to decrypt the frame");
a0530dff JM	291	if (dl_list_empty(&wt->ptk))
	292	return;
	293	try_ptk_iter = 1;
d318c534 JM	294	}
	295
	296	if (len < 4) {
e4d99217	297	add_note(wt, MSG_INFO, "Too short encrypted data frame");
d318c534 JM	298	return;
	299	}
	300
99d7c1de JM	301	if (sta == NULL)
99d7c1de JM	302	return;
20062114 JM	303	if (sta->pairwise_cipher & (WPA_CIPHER_TKIP \| WPA_CIPHER_CCMP) &&
20062114 JM	304	!(data[3] & 0x20)) {
e4d99217 JM	305	add_note(wt, MSG_INFO, "Expected TKIP/CCMP frame from "
	306	MACSTR " did not have ExtIV bit set to 1",
	307	MAC2STR(src));
	308	return;
20062114 JM	309	}
20062114 JM	310
b39f5834	311	if (tk == NULL && sta->pairwise_cipher == WPA_CIPHER_TKIP) {
16b8b6ea	312	if (data[3] & 0x1f) {
e4d99217 JM	313	add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
	314	" used non-zero reserved bit",
	315	MAC2STR(hdr->addr2));
16b8b6ea JM	316	}
16b8b6ea JM	317	if (data[1] != ((data[0] \| 0x20) & 0x7f)) {
e4d99217 JM	318	add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
	319	" used incorrect WEPSeed[1] (was 0x%x, "
	320	"expected 0x%x)",
	321	MAC2STR(hdr->addr2), data[1],
	322	(data[0] \| 0x20) & 0x7f);
16b8b6ea	323	}
b39f5834	324	} else if (tk \|\| sta->pairwise_cipher == WPA_CIPHER_CCMP) {
16b8b6ea	325	if (data[2] != 0 \|\| (data[3] & 0x1f) != 0) {
e4d99217 JM	326	add_note(wt, MSG_INFO, "CCMP frame from " MACSTR
	327	" used non-zero reserved bit",
	328	MAC2STR(hdr->addr2));
16b8b6ea JM	329	}
	330	}
	331
d318c534 JM	332	keyid = data[3] >> 6;
d318c534 JM	333	if (keyid != 0) {
e4d99217 JM	334	add_note(wt, MSG_INFO, "Unexpected non-zero KeyID %d in "
	335	"individually addressed Data frame from " MACSTR,
	336	keyid, MAC2STR(hdr->addr2));
d318c534 JM	337	}
d318c534 JM	338
99d7c1de	339	if (qos) {
2edd5c23	340	tid = qos[0] & 0x0f;
99d7c1de JM	341	if (fc & WLAN_FC_TODS)
	342	sta->tx_tid[tid]++;
	343	else
	344	sta->rx_tid[tid]++;
	345	} else {
2edd5c23	346	tid = 0;
99d7c1de JM	347	if (fc & WLAN_FC_TODS)
	348	sta->tx_tid[16]++;
	349	else
	350	sta->rx_tid[16]++;
	351	}
b39f5834 JM	352	if (tk) {
	353	if (os_memcmp(hdr->addr2, tdls->init->addr, ETH_ALEN) == 0)
	354	rsc = tdls->rsc_init[tid];
	355	else
	356	rsc = tdls->rsc_resp[tid];
	357	} else if (fc & WLAN_FC_TODS)
2edd5c23 JM	358	rsc = sta->rsc_tods[tid];
	359	else
	360	rsc = sta->rsc_fromds[tid];
	361
	362
b39f5834	363	if (tk == NULL && sta->pairwise_cipher == WPA_CIPHER_TKIP)
4dac8453	364	tkip_get_pn(pn, data);
2e479416 JM	365	else if (sta->pairwise_cipher == WPA_CIPHER_WEP40)
2e479416 JM	366	goto skip_replay_det;
4dac8453 JM	367	else
4dac8453 JM	368	ccmp_get_pn(pn, data);
2edd5c23	369	if (os_memcmp(pn, rsc, 6) <= 0) {
01b397dd	370	u16 seq_ctrl = le_to_host16(hdr->seq_ctrl);
e4d99217 JM	371	add_note(wt, MSG_INFO, "CCMP/TKIP replay detected: A1=" MACSTR
	372	" A2=" MACSTR " A3=" MACSTR " seq=%u frag=%u",
	373	MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
	374	MAC2STR(hdr->addr3),
	375	WLAN_GET_SEQ_SEQ(seq_ctrl),
	376	WLAN_GET_SEQ_FRAG(seq_ctrl));
2edd5c23 JM	377	wpa_hexdump(MSG_INFO, "RX PN", pn, 6);
	378	wpa_hexdump(MSG_INFO, "RSC", rsc, 6);
	379	}
	380
2e479416	381	skip_replay_det:
b39f5834 JM	382	if (tk)
	383	decrypted = ccmp_decrypt(tk, hdr, data, len, &dlen);
	384	else if (sta->pairwise_cipher == WPA_CIPHER_TKIP)
2924b0eb	385	decrypted = tkip_decrypt(sta->ptk.tk1, hdr, data, len, &dlen);
2e479416 JM	386	else if (sta->pairwise_cipher == WPA_CIPHER_WEP40)
2e479416 JM	387	decrypted = wep_decrypt(wt, hdr, data, len, &dlen);
a0530dff	388	else if (sta->ptk_set)
2924b0eb	389	decrypted = ccmp_decrypt(sta->ptk.tk1, hdr, data, len, &dlen);
a0530dff JM	390	else {
	391	decrypted = try_all_ptk(wt, sta->pairwise_cipher, hdr, data,
	392	len, &dlen);
	393	ptk_iter_done = 1;
	394	}
	395	if (!decrypted && !ptk_iter_done) {
	396	decrypted = try_all_ptk(wt, sta->pairwise_cipher, hdr, data,
	397	len, &dlen);
	398	if (decrypted) {
	399	add_note(wt, MSG_DEBUG, "Current PTK did not work, but found a match from all known PTKs");
	400	}
	401	}
2edd5c23	402	if (decrypted) {
244c9303	403	u16 fc = le_to_host16(hdr->frame_control);
9559a7f8	404	const u8 *peer_addr = NULL;
244c9303 JM	405	if (!(fc & (WLAN_FC_FROMDS \| WLAN_FC_TODS)))
244c9303 JM	406	peer_addr = hdr->addr1;
d0b251d2	407	os_memcpy(rsc, pn, 6);
ee3b84be	408	rx_data_process(wt, bss->bssid, sta->addr, dst, src, decrypted,
244c9303	409	dlen, 1, peer_addr);
64f45d07 JM	410	write_pcap_decrypted(wt, (const u8 *) hdr, 24 + (qos ? 2 : 0),
64f45d07 JM	411	decrypted, dlen);
a0530dff	412	} else if (!try_ptk_iter)
e4d99217	413	add_note(wt, MSG_DEBUG, "Failed to decrypt frame");
d318c534	414	os_free(decrypted);
32234bba JM	415	}
	416
	417
	418	static void rx_data_bss(struct wlantest wt, const struct ieee80211_hdr hdr,
	419	const u8 qos, const u8 dst, const u8 *src,
	420	const u8 *data, size_t len)
	421	{
	422	u16 fc = le_to_host16(hdr->frame_control);
	423	int prot = !!(fc & WLAN_FC_ISWEP);
	424
	425	if (qos) {
	426	u8 ack = (qos[0] & 0x60) >> 5;
	427	wpa_printf(MSG_MSGDUMP, "BSS DATA: " MACSTR " -> " MACSTR
	428	" len=%u%s tid=%u%s%s",
	429	MAC2STR(src), MAC2STR(dst), (unsigned int) len,
	430	prot ? " Prot" : "", qos[0] & 0x0f,
	431	(qos[0] & 0x10) ? " EOSP" : "",
	432	ack == 0 ? "" :
	433	(ack == 1 ? " NoAck" :
	434	(ack == 2 ? " NoExpAck" : " BA")));
	435	} else {
	436	wpa_printf(MSG_MSGDUMP, "BSS DATA: " MACSTR " -> " MACSTR
	437	" len=%u%s",
	438	MAC2STR(src), MAC2STR(dst), (unsigned int) len,
	439	prot ? " Prot" : "");
	440	}
	441
	442	if (prot)
	443	rx_data_bss_prot(wt, hdr, qos, dst, src, data, len);
ee3b84be	444	else {
244c9303	445	const u8 bssid, sta_addr, *peer_addr;
99d7c1de JM	446	struct wlantest_bss *bss;
99d7c1de JM	447
ee3b84be JM	448	if (fc & WLAN_FC_TODS) {
	449	bssid = hdr->addr1;
	450	sta_addr = hdr->addr2;
244c9303 JM	451	peer_addr = NULL;
244c9303 JM	452	} else if (fc & WLAN_FC_FROMDS) {
ee3b84be JM	453	bssid = hdr->addr2;
ee3b84be JM	454	sta_addr = hdr->addr1;
244c9303 JM	455	peer_addr = NULL;
	456	} else {
	457	bssid = hdr->addr3;
	458	sta_addr = hdr->addr2;
	459	peer_addr = hdr->addr1;
ee3b84be	460	}
99d7c1de JM	461
	462	bss = bss_get(wt, bssid);
	463	if (bss) {
	464	struct wlantest_sta *sta = sta_get(bss, sta_addr);
	465
	466	if (sta) {
	467	if (qos) {
	468	int tid = qos[0] & 0x0f;
	469	if (fc & WLAN_FC_TODS)
	470	sta->tx_tid[tid]++;
	471	else
	472	sta->rx_tid[tid]++;
	473	} else {
	474	if (fc & WLAN_FC_TODS)
	475	sta->tx_tid[16]++;
	476	else
	477	sta->rx_tid[16]++;
	478	}
	479	}
	480	}
	481
244c9303 JM	482	rx_data_process(wt, bssid, sta_addr, dst, src, data, len, 0,
244c9303 JM	483	peer_addr);
ee3b84be	484	}
32234bba JM	485	}
	486
	487
0e42fff3 JM	488	static struct wlantest_tdls * get_tdls(struct wlantest wt, const u8 bssid,
	489	const u8 *sta1_addr,
	490	const u8 *sta2_addr)
	491	{
	492	struct wlantest_bss *bss;
	493	struct wlantest_sta sta1, sta2;
ace4e460	494	struct wlantest_tdls tdls, found = NULL;
0e42fff3 JM	495
	496	bss = bss_find(wt, bssid);
	497	if (bss == NULL)
	498	return NULL;
	499	sta1 = sta_find(bss, sta1_addr);
	500	if (sta1 == NULL)
	501	return NULL;
	502	sta2 = sta_find(bss, sta2_addr);
	503	if (sta2 == NULL)
	504	return NULL;
	505
	506	dl_list_for_each(tdls, &bss->tdls, struct wlantest_tdls, list) {
	507	if ((tdls->init == sta1 && tdls->resp == sta2) \|\|
ace4e460 JM	508	(tdls->init == sta2 && tdls->resp == sta1)) {
	509	found = tdls;
	510	if (tdls->link_up)
	511	break;
	512	}
0e42fff3 JM	513	}
0e42fff3 JM	514
ace4e460	515	return found;
0e42fff3 JM	516	}
	517
	518
	519	static void add_direct_link(struct wlantest wt, const u8 bssid,
	520	const u8 sta1_addr, const u8 sta2_addr)
	521	{
	522	struct wlantest_tdls *tdls;
	523
	524	tdls = get_tdls(wt, bssid, sta1_addr, sta2_addr);
	525	if (tdls == NULL)
	526	return;
	527
	528	if (tdls->link_up)
	529	tdls->counters[WLANTEST_TDLS_COUNTER_VALID_DIRECT_LINK]++;
	530	else
	531	tdls->counters[WLANTEST_TDLS_COUNTER_INVALID_DIRECT_LINK]++;
	532	}
	533
	534
	535	static void add_ap_path(struct wlantest wt, const u8 bssid,
	536	const u8 sta1_addr, const u8 sta2_addr)
	537	{
	538	struct wlantest_tdls *tdls;
	539
	540	tdls = get_tdls(wt, bssid, sta1_addr, sta2_addr);
	541	if (tdls == NULL)
	542	return;
	543
	544	if (tdls->link_up)
	545	tdls->counters[WLANTEST_TDLS_COUNTER_INVALID_AP_PATH]++;
	546	else
	547	tdls->counters[WLANTEST_TDLS_COUNTER_VALID_AP_PATH]++;
	548	}
	549
	550
2d73f0a8 JM	551	void rx_data(struct wlantest wt, const u8 data, size_t len)
	552	{
	553	const struct ieee80211_hdr *hdr;
32234bba JM	554	u16 fc, stype;
	555	size_t hdrlen;
	556	const u8 *qos = NULL;
2d73f0a8 JM	557
	558	if (len < 24)
	559	return;
	560
	561	hdr = (const struct ieee80211_hdr *) data;
	562	fc = le_to_host16(hdr->frame_control);
32234bba JM	563	stype = WLAN_FC_GET_STYPE(fc);
	564	hdrlen = 24;
	565	if ((fc & (WLAN_FC_TODS \| WLAN_FC_FROMDS)) ==
	566	(WLAN_FC_TODS \| WLAN_FC_FROMDS))
	567	hdrlen += ETH_ALEN;
	568	if (stype & 0x08) {
	569	qos = data + hdrlen;
	570	hdrlen += 2;
	571	}
	572	if (len < hdrlen)
	573	return;
2d73f0a8 JM	574	wt->rx_data++;
	575
	576	switch (fc & (WLAN_FC_TODS \| WLAN_FC_FROMDS)) {
	577	case 0:
	578	wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s IBSS DA=" MACSTR " SA="
	579	MACSTR " BSSID=" MACSTR,
	580	data_stype(WLAN_FC_GET_STYPE(fc)),
	581	fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
	582	fc & WLAN_FC_ISWEP ? " Prot" : "",
	583	MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
	584	MAC2STR(hdr->addr3));
0e42fff3	585	add_direct_link(wt, hdr->addr3, hdr->addr1, hdr->addr2);
b39f5834 JM	586	rx_data_bss(wt, hdr, qos, hdr->addr1, hdr->addr2,
b39f5834 JM	587	data + hdrlen, len - hdrlen);
2d73f0a8 JM	588	break;
	589	case WLAN_FC_FROMDS:
	590	wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s FromDS DA=" MACSTR
	591	" BSSID=" MACSTR " SA=" MACSTR,
	592	data_stype(WLAN_FC_GET_STYPE(fc)),
	593	fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
	594	fc & WLAN_FC_ISWEP ? " Prot" : "",
	595	MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
	596	MAC2STR(hdr->addr3));
0e42fff3	597	add_ap_path(wt, hdr->addr2, hdr->addr1, hdr->addr3);
39c14726	598	rx_data_bss(wt, hdr, qos, hdr->addr1, hdr->addr3,
32234bba	599	data + hdrlen, len - hdrlen);
2d73f0a8 JM	600	break;
	601	case WLAN_FC_TODS:
	602	wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s ToDS BSSID=" MACSTR
	603	" SA=" MACSTR " DA=" MACSTR,
	604	data_stype(WLAN_FC_GET_STYPE(fc)),
	605	fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
	606	fc & WLAN_FC_ISWEP ? " Prot" : "",
	607	MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
	608	MAC2STR(hdr->addr3));
0e42fff3	609	add_ap_path(wt, hdr->addr1, hdr->addr3, hdr->addr2);
32234bba JM	610	rx_data_bss(wt, hdr, qos, hdr->addr3, hdr->addr2,
32234bba JM	611	data + hdrlen, len - hdrlen);
2d73f0a8 JM	612	break;
	613	case WLAN_FC_TODS \| WLAN_FC_FROMDS:
	614	wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s WDS RA=" MACSTR " TA="
	615	MACSTR " DA=" MACSTR " SA=" MACSTR,
	616	data_stype(WLAN_FC_GET_STYPE(fc)),
	617	fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
	618	fc & WLAN_FC_ISWEP ? " Prot" : "",
	619	MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
	620	MAC2STR(hdr->addr3),
	621	MAC2STR((const u8 *) (hdr + 1)));
	622	break;
	623	}
	624	}