[thirdparty/hostap.git] / wlantest / wlantest.h

/*
 * wlantest - IEEE 802.11 protocol monitoring and testing tool
 * 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.
 */

#ifndef WLANTEST_H
#define WLANTEST_H

#include "utils/list.h"
#include "common/wpa_common.h"
#include "wlantest_ctrl.h"

struct ieee802_11_elems;
struct radius_msg;
struct ieee80211_hdr;
struct wlantest_bss;

#define MAX_RADIUS_SECRET_LEN 128

struct wlantest_radius_secret {
	struct dl_list list;
	char secret[MAX_RADIUS_SECRET_LEN];
};

struct wlantest_passphrase {
	struct dl_list list;
	char passphrase[64];
	u8 ssid[32];
	size_t ssid_len;
	u8 bssid[ETH_ALEN];
};

struct wlantest_pmk {
	struct dl_list list;
	u8 pmk[32];
};

struct wlantest_wep {
	struct dl_list list;
	size_t key_len;
	u8 key[13];
};

struct wlantest_sta {
	struct dl_list list;
	struct wlantest_bss *bss;
	u8 addr[ETH_ALEN];
	enum {
		STATE1 /* not authenticated */,
		STATE2 /* authenticated */,
		STATE3 /* associated */
	} state;
	u16 aid;
	u8 rsnie[257]; /* WPA/RSN IE */
	int proto;
	int pairwise_cipher;
	int group_cipher;
	int key_mgmt;
	int rsn_capab;
	u8 anonce[32]; /* ANonce from the previous EAPOL-Key msg 1/4 or 3/4 */
	u8 snonce[32]; /* SNonce from the previous EAPOL-Key msg 2/4 */
	struct wpa_ptk ptk; /* Derived PTK */
	int ptk_set;
	struct wpa_ptk tptk; /* Derived PTK during rekeying */
	int tptk_set;
	u8 rsc_tods[16 + 1][6];
	u8 rsc_fromds[16 + 1][6];
	u8 ap_sa_query_tr[2];
	u8 sta_sa_query_tr[2];
	u32 counters[NUM_WLANTEST_STA_COUNTER];
	u16 assocreq_capab_info;
	u16 assocreq_listen_int;
	u8 *assocreq_ies;
	size_t assocreq_ies_len;

	/* Last ICMP Echo request information */
	u32 icmp_echo_req_src;
	u32 icmp_echo_req_dst;
	u16 icmp_echo_req_id;
	u16 icmp_echo_req_seq;

	le16 seq_ctrl_to_sta[17];
	le16 seq_ctrl_to_ap[17];

	int pwrmgt;
	int pspoll;

	u8 gtk[32];
	size_t gtk_len;
	int gtk_idx;
};

struct wlantest_tdls {
	struct dl_list list;
	struct wlantest_sta *init;
	struct wlantest_sta *resp;
	struct tpk {
		u8 kck[16];
		u8 tk[16];
	} tpk;
	int link_up;
	u8 dialog_token;
	u8 rsc_init[16 + 1][6];
	u8 rsc_resp[16 + 1][6];
	u32 counters[NUM_WLANTEST_TDLS_COUNTER];
};

struct wlantest_bss {
	struct dl_list list;
	u8 bssid[ETH_ALEN];
	u16 capab_info;
	u16 prev_capab_info;
	u8 ssid[32];
	size_t ssid_len;
	int proberesp_seen;
	int parse_error_reported;
	u8 wpaie[257];
	u8 rsnie[257];
	int proto;
	int pairwise_cipher;
	int group_cipher;
	int mgmt_group_cipher;
	int key_mgmt;
	int rsn_capab;
	struct dl_list sta; /* struct wlantest_sta */
	struct dl_list pmk; /* struct wlantest_pmk */
	u8 gtk[4][32];
	size_t gtk_len[4];
	int gtk_idx;
	u8 rsc[4][6];
	u8 igtk[6][16];
	int igtk_set[6];
	int igtk_idx;
	u8 ipn[6][6];
	u32 counters[NUM_WLANTEST_BSS_COUNTER];
	struct dl_list tdls; /* struct wlantest_tdls */
};

struct wlantest_radius {
	struct dl_list list;
	u32 srv;
	u32 cli;
	struct radius_msg *last_req;
};


#define MAX_CTRL_CONNECTIONS 10

struct wlantest {
	int monitor_sock;
	int monitor_wired;

	int ctrl_sock;
	int ctrl_socks[MAX_CTRL_CONNECTIONS];

	struct dl_list passphrase; /* struct wlantest_passphrase */
	struct dl_list bss; /* struct wlantest_bss */
	struct dl_list secret; /* struct wlantest_radius_secret */
	struct dl_list radius; /* struct wlantest_radius */
	struct dl_list pmk; /* struct wlantest_pmk */
	struct dl_list wep; /* struct wlantest_wep */

	unsigned int rx_mgmt;
	unsigned int rx_ctrl;
	unsigned int rx_data;
	unsigned int fcs_error;

	void *write_pcap; /* pcap_t* */
	void *write_pcap_dumper; /* pcpa_dumper_t */
	struct timeval write_pcap_time;

	u8 last_hdr[30];
	size_t last_len;
	int last_mgmt_valid;
};

int add_wep(struct wlantest *wt, const char *key);
int read_cap_file(struct wlantest *wt, const char *fname);
int read_wired_cap_file(struct wlantest *wt, const char *fname);
int write_pcap_init(struct wlantest *wt, const char *fname);
void write_pcap_deinit(struct wlantest *wt);
void write_pcap_captured(struct wlantest *wt, const u8 *buf, size_t len);
void write_pcap_decrypted(struct wlantest *wt, const u8 *buf1, size_t len1,
			  const u8 *buf2, size_t len2);
void wlantest_process(struct wlantest *wt, const u8 *data, size_t len);
void wlantest_process_prism(struct wlantest *wt, const u8 *data, size_t len);
void wlantest_process_80211(struct wlantest *wt, const u8 *data, size_t len);
void wlantest_process_wired(struct wlantest *wt, const u8 *data, size_t len);
u32 crc32(const u8 *frame, size_t frame_len);
int monitor_init(struct wlantest *wt, const char *ifname);
int monitor_init_wired(struct wlantest *wt, const char *ifname);
void monitor_deinit(struct wlantest *wt);
void rx_mgmt(struct wlantest *wt, const u8 *data, size_t len);
void rx_mgmt_ack(struct wlantest *wt, const struct ieee80211_hdr *hdr);
void rx_data(struct wlantest *wt, const u8 *data, size_t len);
void rx_data_eapol(struct wlantest *wt, const u8 *dst, const u8 *src,
		   const u8 *data, size_t len, int prot);
void rx_data_ip(struct wlantest *wt, const u8 *bssid, const u8 *sta_addr,
		const u8 *dst, const u8 *src, const u8 *data, size_t len,
		const u8 *peer_addr);
void rx_data_80211_encap(struct wlantest *wt, const u8 *bssid,
			 const u8 *sta_addr, const u8 *dst, const u8 *src,
			 const u8 *data, size_t len);

struct wlantest_bss * bss_find(struct wlantest *wt, const u8 *bssid);
struct wlantest_bss * bss_get(struct wlantest *wt, const u8 *bssid);
void bss_deinit(struct wlantest_bss *bss);
void bss_update(struct wlantest *wt, struct wlantest_bss *bss,
		struct ieee802_11_elems *elems);
void bss_flush(struct wlantest *wt);
int bss_add_pmk_from_passphrase(struct wlantest_bss *bss,
				const char *passphrase);
void pmk_deinit(struct wlantest_pmk *pmk);
void tdls_deinit(struct wlantest_tdls *tdls);

struct wlantest_sta * sta_find(struct wlantest_bss *bss, const u8 *addr);
struct wlantest_sta * sta_get(struct wlantest_bss *bss, const u8 *addr);
void sta_deinit(struct wlantest_sta *sta);
void sta_update_assoc(struct wlantest_sta *sta,
		      struct ieee802_11_elems *elems);

u8 * ccmp_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
		  const u8 *data, size_t data_len, size_t *decrypted_len);
u8 * ccmp_encrypt(const u8 *tk, u8 *frame, size_t len, size_t hdrlen, u8 *qos,
		  u8 *pn, int keyid, size_t *encrypted_len);
void ccmp_get_pn(u8 *pn, const u8 *data);
u8 * ccmp_256_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
		      const u8 *data, size_t data_len, size_t *decrypted_len);
u8 * ccmp_256_encrypt(const u8 *tk, u8 *frame, size_t len, size_t hdrlen,
		      u8 *qos, u8 *pn, int keyid, size_t *encrypted_len);

u8 * tkip_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
		  const u8 *data, size_t data_len, size_t *decrypted_len);
u8 * tkip_encrypt(const u8 *tk, u8 *frame, size_t len, size_t hdrlen, u8 *qos,
		  u8 *pn, int keyid, size_t *encrypted_len);
void tkip_get_pn(u8 *pn, const u8 *data);

u8 * wep_decrypt(struct wlantest *wt, const struct ieee80211_hdr *hdr,
		 const u8 *data, size_t data_len, size_t *decrypted_len);

u8 * bip_protect(const u8 *igtk, u8 *frame, size_t len, u8 *ipn, int keyid,
		 size_t *prot_len);
u8 * bip_gmac_protect(const u8 *igtk, size_t igtk_len, u8 *frame, size_t len,
		      u8 *ipn, int keyid, size_t *prot_len);

u8 * gcmp_decrypt(const u8 *tk, size_t tk_len, const struct ieee80211_hdr *hdr,
		  const u8 *data, size_t data_len, size_t *decrypted_len);
u8 * gcmp_encrypt(const u8 *tk, size_t tk_len, u8 *frame, size_t len,
		  size_t hdrlen, u8 *qos,
		  u8 *pn, int keyid, size_t *encrypted_len);

int ctrl_init(struct wlantest *wt);
void ctrl_deinit(struct wlantest *wt);

int wlantest_inject(struct wlantest *wt, struct wlantest_bss *bss,
		    struct wlantest_sta *sta, u8 *frame, size_t len,
		    enum wlantest_inject_protection prot);

#endif /* WLANTEST_H */
Commit	Line	Data
a149fcc7 JM	1	/*
	2	* wlantest - IEEE 802.11 protocol monitoring and testing tool
	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.
a149fcc7 JM	7	*/
	8
	9	#ifndef WLANTEST_H
	10	#define WLANTEST_H
	11
d84d3893	12	#include "utils/list.h"
53650bca	13	#include "common/wpa_common.h"
6d5ce9fc	14	#include "wlantest_ctrl.h"
d84d3893	15
b50111fb	16	struct ieee802_11_elems;
d06df64d	17	struct radius_msg;
d318c534	18	struct ieee80211_hdr;
327f7160	19	struct wlantest_bss;
b50111fb	20
d06df64d JM	21	#define MAX_RADIUS_SECRET_LEN 128
	22
	23	struct wlantest_radius_secret {
	24	struct dl_list list;
	25	char secret[MAX_RADIUS_SECRET_LEN];
	26	};
d84d3893	27
53650bca JM	28	struct wlantest_passphrase {
	29	struct dl_list list;
	30	char passphrase[64];
	31	u8 ssid[32];
	32	size_t ssid_len;
	33	u8 bssid[ETH_ALEN];
	34	};
	35
	36	struct wlantest_pmk {
	37	struct dl_list list;
	38	u8 pmk[32];
	39	};
	40
2e479416 JM	41	struct wlantest_wep {
	42	struct dl_list list;
	43	size_t key_len;
	44	u8 key[13];
	45	};
	46
422ef7d2 JM	47	struct wlantest_sta {
422ef7d2 JM	48	struct dl_list list;
327f7160	49	struct wlantest_bss *bss;
422ef7d2	50	u8 addr[ETH_ALEN];
d30edf54 JM	51	enum {
	52	STATE1 /* not authenticated */,
	53	STATE2 /* authenticated */,
	54	STATE3 /* associated */
	55	} state;
	56	u16 aid;
021a6fe4	57	u8 rsnie[257]; /* WPA/RSN IE */
327f7160 JM	58	int proto;
	59	int pairwise_cipher;
	60	int group_cipher;
	61	int key_mgmt;
	62	int rsn_capab;
53650bca JM	63	u8 anonce[32]; /* ANonce from the previous EAPOL-Key msg 1/4 or 3/4 */
	64	u8 snonce[32]; /* SNonce from the previous EAPOL-Key msg 2/4 */
	65	struct wpa_ptk ptk; /* Derived PTK */
	66	int ptk_set;
d0b251d2 JM	67	struct wpa_ptk tptk; /* Derived PTK during rekeying */
d0b251d2 JM	68	int tptk_set;
47fe6880 JM	69	u8 rsc_tods[16 + 1][6];
47fe6880 JM	70	u8 rsc_fromds[16 + 1][6];
0819b65b JM	71	u8 ap_sa_query_tr[2];
0819b65b JM	72	u8 sta_sa_query_tr[2];
6d5ce9fc	73	u32 counters[NUM_WLANTEST_STA_COUNTER];
990153b4 JM	74	u16 assocreq_capab_info;
	75	u16 assocreq_listen_int;
	76	u8 *assocreq_ies;
	77	size_t assocreq_ies_len;
ee3b84be JM	78
	79	/* Last ICMP Echo request information */
	80	u32 icmp_echo_req_src;
	81	u32 icmp_echo_req_dst;
	82	u16 icmp_echo_req_id;
	83	u16 icmp_echo_req_seq;
a912dd16 JM	84
	85	le16 seq_ctrl_to_sta[17];
	86	le16 seq_ctrl_to_ap[17];
7e7a57ae JM	87
	88	int pwrmgt;
	89	int pspoll;
fd848ab9 JM	90
	91	u8 gtk[32];
	92	size_t gtk_len;
	93	int gtk_idx;
422ef7d2 JM	94	};
422ef7d2 JM	95
b39f5834 JM	96	struct wlantest_tdls {
	97	struct dl_list list;
	98	struct wlantest_sta *init;
	99	struct wlantest_sta *resp;
	100	struct tpk {
	101	u8 kck[16];
	102	u8 tk[16];
	103	} tpk;
	104	int link_up;
29ec7457	105	u8 dialog_token;
b39f5834 JM	106	u8 rsc_init[16 + 1][6];
b39f5834 JM	107	u8 rsc_resp[16 + 1][6];
0e42fff3	108	u32 counters[NUM_WLANTEST_TDLS_COUNTER];
b39f5834 JM	109	};
b39f5834 JM	110
d84d3893 JM	111	struct wlantest_bss {
	112	struct dl_list list;
	113	u8 bssid[ETH_ALEN];
	114	u16 capab_info;
2e479416	115	u16 prev_capab_info;
d84d3893 JM	116	u8 ssid[32];
	117	size_t ssid_len;
	118	int proberesp_seen;
	119	int parse_error_reported;
	120	u8 wpaie[257];
	121	u8 rsnie[257];
327f7160 JM	122	int proto;
	123	int pairwise_cipher;
	124	int group_cipher;
	125	int mgmt_group_cipher;
	126	int key_mgmt;
	127	int rsn_capab;
422ef7d2	128	struct dl_list sta; /* struct wlantest_sta */
53650bca	129	struct dl_list pmk; /* struct wlantest_pmk */
4d9f9ee7 JM	130	u8 gtk[4][32];
4d9f9ee7 JM	131	size_t gtk_len[4];
2e4c3469	132	int gtk_idx;
2edd5c23	133	u8 rsc[4][6];
4d9f9ee7 JM	134	u8 igtk[6][16];
4d9f9ee7 JM	135	int igtk_set[6];
2e4c3469	136	int igtk_idx;
bacc3128	137	u8 ipn[6][6];
6d5ce9fc	138	u32 counters[NUM_WLANTEST_BSS_COUNTER];
b39f5834	139	struct dl_list tdls; /* struct wlantest_tdls */
d84d3893 JM	140	};
d84d3893 JM	141
d06df64d JM	142	struct wlantest_radius {
	143	struct dl_list list;
	144	u32 srv;
	145	u32 cli;
	146	struct radius_msg *last_req;
	147	};
	148
644fb8c8 JM	149
	150	#define MAX_CTRL_CONNECTIONS 10
	151
a149fcc7 JM	152	struct wlantest {
a149fcc7 JM	153	int monitor_sock;
3215df77	154	int monitor_wired;
a149fcc7	155
644fb8c8 JM	156	int ctrl_sock;
	157	int ctrl_socks[MAX_CTRL_CONNECTIONS];
	158
53650bca	159	struct dl_list passphrase; /* struct wlantest_passphrase */
d84d3893	160	struct dl_list bss; /* struct wlantest_bss */
d06df64d JM	161	struct dl_list secret; /* struct wlantest_radius_secret */
	162	struct dl_list radius; /* struct wlantest_radius */
	163	struct dl_list pmk; /* struct wlantest_pmk */
2e479416	164	struct dl_list wep; /* struct wlantest_wep */
d84d3893	165
a149fcc7 JM	166	unsigned int rx_mgmt;
	167	unsigned int rx_ctrl;
	168	unsigned int rx_data;
	169	unsigned int fcs_error;
64f45d07 JM	170
	171	void write_pcap; / pcap_t* */
	172	void write_pcap_dumper; / pcpa_dumper_t */
	173	struct timeval write_pcap_time;
fb8f5fc6 JM	174
	175	u8 last_hdr[30];
	176	size_t last_len;
	177	int last_mgmt_valid;
a149fcc7 JM	178	};
a149fcc7 JM	179
9a994178	180	int add_wep(struct wlantest wt, const char key);
a149fcc7	181	int read_cap_file(struct wlantest wt, const char fname);
3215df77	182	int read_wired_cap_file(struct wlantest wt, const char fname);
64f45d07 JM	183	int write_pcap_init(struct wlantest wt, const char fname);
	184	void write_pcap_deinit(struct wlantest *wt);
	185	void write_pcap_captured(struct wlantest wt, const u8 buf, size_t len);
	186	void write_pcap_decrypted(struct wlantest wt, const u8 buf1, size_t len1,
	187	const u8 *buf2, size_t len2);
a149fcc7	188	void wlantest_process(struct wlantest wt, const u8 data, size_t len);
350132be	189	void wlantest_process_prism(struct wlantest wt, const u8 data, size_t len);
0f3e4f2a	190	void wlantest_process_80211(struct wlantest wt, const u8 data, size_t len);
3215df77	191	void wlantest_process_wired(struct wlantest wt, const u8 data, size_t len);
a149fcc7 JM	192	u32 crc32(const u8 *frame, size_t frame_len);
a149fcc7 JM	193	int monitor_init(struct wlantest wt, const char ifname);
3215df77	194	int monitor_init_wired(struct wlantest wt, const char ifname);
a149fcc7	195	void monitor_deinit(struct wlantest *wt);
2d73f0a8	196	void rx_mgmt(struct wlantest wt, const u8 data, size_t len);
fb8f5fc6	197	void rx_mgmt_ack(struct wlantest wt, const struct ieee80211_hdr hdr);
2d73f0a8	198	void rx_data(struct wlantest wt, const u8 data, size_t len);
161d0339 JM	199	void rx_data_eapol(struct wlantest wt, const u8 dst, const u8 *src,
161d0339 JM	200	const u8 *data, size_t len, int prot);
ee3b84be	201	void rx_data_ip(struct wlantest wt, const u8 bssid, const u8 *sta_addr,
244c9303 JM	202	const u8 dst, const u8 src, const u8 *data, size_t len,
244c9303 JM	203	const u8 *peer_addr);
b39f5834 JM	204	void rx_data_80211_encap(struct wlantest wt, const u8 bssid,
	205	const u8 sta_addr, const u8 dst, const u8 *src,
	206	const u8 *data, size_t len);
a149fcc7	207
57f7d03f	208	struct wlantest_bss * bss_find(struct wlantest wt, const u8 bssid);
d84d3893 JM	209	struct wlantest_bss * bss_get(struct wlantest wt, const u8 bssid);
d84d3893 JM	210	void bss_deinit(struct wlantest_bss *bss);
53650bca JM	211	void bss_update(struct wlantest wt, struct wlantest_bss bss,
53650bca JM	212	struct ieee802_11_elems *elems);
d356bd63	213	void bss_flush(struct wlantest *wt);
221519de JM	214	int bss_add_pmk_from_passphrase(struct wlantest_bss *bss,
221519de JM	215	const char *passphrase);
d06df64d	216	void pmk_deinit(struct wlantest_pmk *pmk);
0d2e395d	217	void tdls_deinit(struct wlantest_tdls *tdls);
d84d3893	218
57f7d03f	219	struct wlantest_sta * sta_find(struct wlantest_bss bss, const u8 addr);
422ef7d2 JM	220	struct wlantest_sta * sta_get(struct wlantest_bss bss, const u8 addr);
422ef7d2 JM	221	void sta_deinit(struct wlantest_sta *sta);
021a6fe4 JM	222	void sta_update_assoc(struct wlantest_sta *sta,
021a6fe4 JM	223	struct ieee802_11_elems *elems);
422ef7d2	224
d318c534 JM	225	u8 * ccmp_decrypt(const u8 tk, const struct ieee80211_hdr hdr,
d318c534 JM	226	const u8 data, size_t data_len, size_t decrypted_len);
571ab37b JM	227	u8 * ccmp_encrypt(const u8 tk, u8 frame, size_t len, size_t hdrlen, u8 *qos,
571ab37b JM	228	u8 pn, int keyid, size_t encrypted_len);
2edd5c23	229	void ccmp_get_pn(u8 pn, const u8 data);
7d19d3e9 JM	230	u8 * ccmp_256_decrypt(const u8 tk, const struct ieee80211_hdr hdr,
	231	const u8 data, size_t data_len, size_t decrypted_len);
	232	u8 * ccmp_256_encrypt(const u8 tk, u8 frame, size_t len, size_t hdrlen,
	233	u8 qos, u8 pn, int keyid, size_t *encrypted_len);
d318c534	234
2924b0eb JM	235	u8 * tkip_decrypt(const u8 tk, const struct ieee80211_hdr hdr,
2924b0eb JM	236	const u8 data, size_t data_len, size_t decrypted_len);
571ab37b JM	237	u8 * tkip_encrypt(const u8 tk, u8 frame, size_t len, size_t hdrlen, u8 *qos,
571ab37b JM	238	u8 pn, int keyid, size_t encrypted_len);
4dac8453	239	void tkip_get_pn(u8 pn, const u8 data);
2924b0eb	240
2e479416 JM	241	u8 * wep_decrypt(struct wlantest wt, const struct ieee80211_hdr hdr,
	242	const u8 data, size_t data_len, size_t decrypted_len);
	243
a9eae7ef JM	244	u8 * bip_protect(const u8 igtk, u8 frame, size_t len, u8 *ipn, int keyid,
a9eae7ef JM	245	size_t *prot_len);
e88f0901 JM	246	u8 * bip_gmac_protect(const u8 igtk, size_t igtk_len, u8 frame, size_t len,
e88f0901 JM	247	u8 ipn, int keyid, size_t prot_len);
a9eae7ef	248
f1732956	249	u8 * gcmp_decrypt(const u8 tk, size_t tk_len, const struct ieee80211_hdr hdr,
455bcc0f	250	const u8 data, size_t data_len, size_t decrypted_len);
f1732956 JM	251	u8 * gcmp_encrypt(const u8 tk, size_t tk_len, u8 frame, size_t len,
f1732956 JM	252	size_t hdrlen, u8 *qos,
455bcc0f JM	253	u8 pn, int keyid, size_t encrypted_len);
455bcc0f JM	254
644fb8c8 JM	255	int ctrl_init(struct wlantest *wt);
	256	void ctrl_deinit(struct wlantest *wt);
	257
7d23e971 JM	258	int wlantest_inject(struct wlantest wt, struct wlantest_bss bss,
	259	struct wlantest_sta sta, u8 frame, size_t len,
	260	enum wlantest_inject_protection prot);
	261
a149fcc7	262	#endif /* WLANTEST_H */