wlantest/wlantest.h - third_party/hostap - Git at Google

 /*
  * wlantest - IEEE 802.11 protocol monitoring and testing tool
  * Copyright (c) 2010-2013, 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_ptk {
 	struct dl_list list;
 	struct wpa_ptk ptk;
 	size_t ptk_len;
 };

 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 */
 	u8 osenie[257]; /* OSEN 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 */
 	size_t tk_len;
 	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;

 	u32 tx_tid[16 + 1];
 	u32 rx_tid[16 + 1];
 };

 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];
 	u8 inonce[32];
 	u8 rnonce[32];
 };

 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];
 	u8 osenie[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][32];
 	size_t igtk_len[6];
 	int igtk_idx;
 	u8 ipn[6][6];
 	u32 counters[NUM_WLANTEST_BSS_COUNTER];
 	struct dl_list tdls; /* struct wlantest_tdls */
 	u8 mdid[2];
 	u8 r0kh_id[FT_R0KH_ID_MAX_LEN];
 	size_t r0kh_id_len;
 	u8 r1kh_id[FT_R1KH_ID_LEN];
 };

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


 #define MAX_CTRL_CONNECTIONS 10
 #define MAX_NOTES 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 ptk; /* struct wlantest_ptk */
 	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 *decrypted;
 	size_t decrypted_len;
 	FILE *pcapng;
 	u32 write_pcapng_time_high;
 	u32 write_pcapng_time_low;

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

 	unsigned int assume_fcs:1;

 	char *notes[MAX_NOTES];
 	size_t num_notes;

 	const char *write_file;
 	const char *pcapng_file;
 };

 void add_note(struct wlantest *wt, int level, const char *fmt, ...)
 PRINTF_FORMAT(3, 4);
 void clear_notes(struct wlantest *wt);
 size_t notes_len(struct wlantest *wt, size_t hdrlen);

 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);

 int write_pcapng_init(struct wlantest *wt, const char *fname);
 void write_pcapng_deinit(struct wlantest *wt);
 struct pcap_pkthdr;
 void write_pcapng_write_read(struct wlantest *wt, int dlt,
 			     struct pcap_pkthdr *hdr, const u8 *data);
 void write_pcapng_captured(struct wlantest *wt, const u8 *buf, size_t len);

 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, size_t igtk_len, 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, const u8 *frame, size_t len,
 		  size_t hdrlen, const u8 *qos,
 		  const 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);

 int wlantest_relog(struct wlantest *wt);

 #endif /* WLANTEST_H */
	/*
	* wlantest - IEEE 802.11 protocol monitoring and testing tool
	* Copyright (c) 2010-2013, 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_ptk {
	struct dl_list list;
	struct wpa_ptk ptk;
	size_t ptk_len;
	};

	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 */
	u8 osenie[257]; /* OSEN 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 */
	size_t tk_len;
	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;

	u32 tx_tid[16 + 1];
	u32 rx_tid[16 + 1];
	};

	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];
	u8 inonce[32];
	u8 rnonce[32];
	};

	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];
	u8 osenie[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][32];
	size_t igtk_len[6];
	int igtk_idx;
	u8 ipn[6][6];
	u32 counters[NUM_WLANTEST_BSS_COUNTER];
	struct dl_list tdls; /* struct wlantest_tdls */
	u8 mdid[2];
	u8 r0kh_id[FT_R0KH_ID_MAX_LEN];
	size_t r0kh_id_len;
	u8 r1kh_id[FT_R1KH_ID_LEN];
	};

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


	#define MAX_CTRL_CONNECTIONS 10
	#define MAX_NOTES 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 ptk; /* struct wlantest_ptk */
	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 *decrypted;
	size_t decrypted_len;
	FILE *pcapng;
	u32 write_pcapng_time_high;
	u32 write_pcapng_time_low;

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

	unsigned int assume_fcs:1;

	char *notes[MAX_NOTES];
	size_t num_notes;

	const char *write_file;
	const char *pcapng_file;
	};

	void add_note(struct wlantest wt, int level, const char fmt, ...)
	PRINTF_FORMAT(3, 4);
	void clear_notes(struct wlantest *wt);
	size_t notes_len(struct wlantest *wt, size_t hdrlen);

	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);

	int write_pcapng_init(struct wlantest wt, const char fname);
	void write_pcapng_deinit(struct wlantest *wt);
	struct pcap_pkthdr;
	void write_pcapng_write_read(struct wlantest *wt, int dlt,
	struct pcap_pkthdr hdr, const u8 data);
	void write_pcapng_captured(struct wlantest wt, const u8 buf, size_t len);

	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, size_t igtk_len, 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, const u8 frame, size_t len,
	size_t hdrlen, const u8 *qos,
	const 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);

	int wlantest_relog(struct wlantest *wt);

	#endif /* WLANTEST_H */