wlantest/wlantest.h

   1 /*
   2  * wlantest - IEEE 802.11 protocol monitoring and testing tool
   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 #ifndef WLANTEST_H
  10 #define WLANTEST_H
  11
  12 #include "utils/list.h"
  13 #include "common/wpa_common.h"
  14 #include "wlantest_ctrl.h"
  15
  16 struct ieee802_11_elems;
  17 struct radius_msg;
  18 struct ieee80211_hdr;
  19 struct wlantest_bss;
  20
  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 };
  27
  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
  41 struct wlantest_wep {
  42         struct dl_list list;
  43         size_t key_len;
  44         u8 key[13];
  45 };
  46
  47 struct wlantest_sta {
  48         struct dl_list list;
  49         struct wlantest_bss *bss;
  50         u8 addr[ETH_ALEN];
  51         enum {
  52                 STATE1 /* not authenticated */,
  53                 STATE2 /* authenticated */,
  54                 STATE3 /* associated */
  55         } state;
  56         u16 aid;
  57         u8 rsnie[257]; /* WPA/RSN IE */
  58         int proto;
  59         int pairwise_cipher;
  60         int group_cipher;
  61         int key_mgmt;
  62         int rsn_capab;
  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;
  67         struct wpa_ptk tptk; /* Derived PTK during rekeying */
  68         int tptk_set;
  69         u8 rsc_tods[16 + 1][6];
  70         u8 rsc_fromds[16 + 1][6];
  71         u8 ap_sa_query_tr[2];
  72         u8 sta_sa_query_tr[2];
  73         u32 counters[NUM_WLANTEST_STA_COUNTER];
  74         u16 assocreq_capab_info;
  75         u16 assocreq_listen_int;
  76         u8 *assocreq_ies;
  77         size_t assocreq_ies_len;
  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;
  84
  85         le16 seq_ctrl_to_sta[17];
  86         le16 seq_ctrl_to_ap[17];
  87
  88         int pwrmgt;
  89         int pspoll;
  90
  91         u8 gtk[32];
  92         size_t gtk_len;
  93         int gtk_idx;
  94 };
  95
  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;
 105         u8 dialog_token;
 106         u8 rsc_init[16 + 1][6];
 107         u8 rsc_resp[16 + 1][6];
 108         u32 counters[NUM_WLANTEST_TDLS_COUNTER];
 109 };
 110
 111 struct wlantest_bss {
 112         struct dl_list list;
 113         u8 bssid[ETH_ALEN];
 114         u16 capab_info;
 115         u16 prev_capab_info;
 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];
 122         int proto;
 123         int pairwise_cipher;
 124         int group_cipher;
 125         int mgmt_group_cipher;
 126         int key_mgmt;
 127         int rsn_capab;
 128         struct dl_list sta; /* struct wlantest_sta */
 129         struct dl_list pmk; /* struct wlantest_pmk */
 130         u8 gtk[4][32];
 131         size_t gtk_len[4];
 132         int gtk_idx;
 133         u8 rsc[4][6];
 134         u8 igtk[6][16];
 135         int igtk_set[6];
 136         int igtk_idx;
 137         u8 ipn[6][6];
 138         u32 counters[NUM_WLANTEST_BSS_COUNTER];
 139         struct dl_list tdls; /* struct wlantest_tdls */
 140 };
 141
 142 struct wlantest_radius {
 143         struct dl_list list;
 144         u32 srv;
 145         u32 cli;
 146         struct radius_msg *last_req;
 147 };
 148
 149
 150 #define MAX_CTRL_CONNECTIONS 10
 151
 152 struct wlantest {
 153         int monitor_sock;
 154         int monitor_wired;
 155
 156         int ctrl_sock;
 157         int ctrl_socks[MAX_CTRL_CONNECTIONS];
 158
 159         struct dl_list passphrase; /* struct wlantest_passphrase */
 160         struct dl_list bss; /* struct wlantest_bss */
 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 */
 164         struct dl_list wep; /* struct wlantest_wep */
 165
 166         unsigned int rx_mgmt;
 167         unsigned int rx_ctrl;
 168         unsigned int rx_data;
 169         unsigned int fcs_error;
 170
 171         void *write_pcap; /* pcap_t* */
 172         void *write_pcap_dumper; /* pcpa_dumper_t */
 173         struct timeval write_pcap_time;
 174
 175         u8 last_hdr[30];
 176         size_t last_len;
 177         int last_mgmt_valid;
 178 };
 179
 180 int add_wep(struct wlantest *wt, const char *key);
 181 int read_cap_file(struct wlantest *wt, const char *fname);
 182 int read_wired_cap_file(struct wlantest *wt, const char *fname);
 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);
 188 void wlantest_process(struct wlantest *wt, const u8 *data, size_t len);
 189 void wlantest_process_prism(struct wlantest *wt, const u8 *data, size_t len);
 190 void wlantest_process_80211(struct wlantest *wt, const u8 *data, size_t len);
 191 void wlantest_process_wired(struct wlantest *wt, const u8 *data, size_t len);
 192 u32 crc32(const u8 *frame, size_t frame_len);
 193 int monitor_init(struct wlantest *wt, const char *ifname);
 194 int monitor_init_wired(struct wlantest *wt, const char *ifname);
 195 void monitor_deinit(struct wlantest *wt);
 196 void rx_mgmt(struct wlantest *wt, const u8 *data, size_t len);
 197 void rx_mgmt_ack(struct wlantest *wt, const struct ieee80211_hdr *hdr);
 198 void rx_data(struct wlantest *wt, const u8 *data, size_t len);
 199 void rx_data_eapol(struct wlantest *wt, const u8 *dst, const u8 *src,
 200                    const u8 *data, size_t len, int prot);
 201 void rx_data_ip(struct wlantest *wt, const u8 *bssid, const u8 *sta_addr,
 202                 const u8 *dst, const u8 *src, const u8 *data, size_t len,
 203                 const u8 *peer_addr);
 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);
 207
 208 struct wlantest_bss * bss_find(struct wlantest *wt, const u8 *bssid);
 209 struct wlantest_bss * bss_get(struct wlantest *wt, const u8 *bssid);
 210 void bss_deinit(struct wlantest_bss *bss);
 211 void bss_update(struct wlantest *wt, struct wlantest_bss *bss,
 212                 struct ieee802_11_elems *elems);
 213 void bss_flush(struct wlantest *wt);
 214 int bss_add_pmk_from_passphrase(struct wlantest_bss *bss,
 215                                 const char *passphrase);
 216 void pmk_deinit(struct wlantest_pmk *pmk);
 217 void tdls_deinit(struct wlantest_tdls *tdls);
 218
 219 struct wlantest_sta * sta_find(struct wlantest_bss *bss, const u8 *addr);
 220 struct wlantest_sta * sta_get(struct wlantest_bss *bss, const u8 *addr);
 221 void sta_deinit(struct wlantest_sta *sta);
 222 void sta_update_assoc(struct wlantest_sta *sta,
 223                       struct ieee802_11_elems *elems);
 224
 225 u8 * ccmp_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
 226                   const u8 *data, size_t data_len, size_t *decrypted_len);
 227 u8 * ccmp_encrypt(const u8 *tk, u8 *frame, size_t len, size_t hdrlen, u8 *qos,
 228                   u8 *pn, int keyid, size_t *encrypted_len);
 229 void ccmp_get_pn(u8 *pn, const u8 *data);
 230
 231 u8 * tkip_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
 232                   const u8 *data, size_t data_len, size_t *decrypted_len);
 233 u8 * tkip_encrypt(const u8 *tk, u8 *frame, size_t len, size_t hdrlen, u8 *qos,
 234                   u8 *pn, int keyid, size_t *encrypted_len);
 235 void tkip_get_pn(u8 *pn, const u8 *data);
 236
 237 u8 * wep_decrypt(struct wlantest *wt, const struct ieee80211_hdr *hdr,
 238                  const u8 *data, size_t data_len, size_t *decrypted_len);
 239
 240 u8 * bip_protect(const u8 *igtk, u8 *frame, size_t len, u8 *ipn, int keyid,
 241                  size_t *prot_len);
 242
 243 int ctrl_init(struct wlantest *wt);
 244 void ctrl_deinit(struct wlantest *wt);
 245
 246 int wlantest_inject(struct wlantest *wt, struct wlantest_bss *bss,
 247                     struct wlantest_sta *sta, u8 *frame, size_t len,
 248                     enum wlantest_inject_protection prot);
 249
 250 #endif /* WLANTEST_H */