wlantest/wlantest.h

   1 /*
   2  * wlantest - IEEE 802.11 protocol monitoring and testing tool
   3  * Copyright (c) 2010-2019, 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[PMK_LEN_MAX];
  39         size_t pmk_len;
  40 };
  41
  42 struct wlantest_ptk {
  43         struct dl_list list;
  44         struct wpa_ptk ptk;
  45         size_t ptk_len;
  46 };
  47
  48 struct wlantest_wep {
  49         struct dl_list list;
  50         size_t key_len;
  51         u8 key[13];
  52 };
  53
  54 struct wlantest_sta {
  55         struct dl_list list;
  56         struct wlantest_bss *bss;
  57         u8 addr[ETH_ALEN];
  58         enum {
  59                 STATE1 /* not authenticated */,
  60                 STATE2 /* authenticated */,
  61                 STATE3 /* associated */
  62         } state;
  63         u16 auth_alg;
  64         u16 aid;
  65         u8 rsnie[257]; /* WPA/RSN IE */
  66         u8 osenie[257]; /* OSEN IE */
  67         int proto;
  68         int pairwise_cipher;
  69         int group_cipher;
  70         int key_mgmt;
  71         int rsn_capab;
  72         u8 anonce[32]; /* ANonce from the previous EAPOL-Key msg 1/4 or 3/4 */
  73         u8 snonce[32]; /* SNonce from the previous EAPOL-Key msg 2/4 */
  74         u8 pmk_r0[PMK_LEN];
  75         u8 pmk_r0_name[WPA_PMK_NAME_LEN];
  76         struct wpa_ptk ptk; /* Derived PTK */
  77         int ptk_set;
  78         struct wpa_ptk tptk; /* Derived PTK during rekeying */
  79         int tptk_set;
  80         u8 rsc_tods[16 + 1][6];
  81         u8 rsc_fromds[16 + 1][6];
  82         u8 ap_sa_query_tr[2];
  83         u8 sta_sa_query_tr[2];
  84         u32 counters[NUM_WLANTEST_STA_COUNTER];
  85         u16 assocreq_capab_info;
  86         u16 assocreq_listen_int;
  87         u8 *assocreq_ies;
  88         size_t assocreq_ies_len;
  89
  90         /* Last ICMP Echo request information */
  91         u32 icmp_echo_req_src;
  92         u32 icmp_echo_req_dst;
  93         u16 icmp_echo_req_id;
  94         u16 icmp_echo_req_seq;
  95
  96         le16 seq_ctrl_to_sta[17];
  97         le16 seq_ctrl_to_ap[17];
  98         int allow_duplicate;
  99
 100         int pwrmgt;
 101         int pspoll;
 102
 103         u8 gtk[32];
 104         size_t gtk_len;
 105         int gtk_idx;
 106
 107         u32 tx_tid[16 + 1];
 108         u32 rx_tid[16 + 1];
 109 };
 110
 111 struct wlantest_tdls {
 112         struct dl_list list;
 113         struct wlantest_sta *init;
 114         struct wlantest_sta *resp;
 115         struct tpk {
 116                 u8 kck[16];
 117                 u8 tk[16];
 118         } tpk;
 119         int link_up;
 120         u8 dialog_token;
 121         u8 rsc_init[16 + 1][6];
 122         u8 rsc_resp[16 + 1][6];
 123         u32 counters[NUM_WLANTEST_TDLS_COUNTER];
 124         u8 inonce[32];
 125         u8 rnonce[32];
 126 };
 127
 128 struct wlantest_bss {
 129         struct dl_list list;
 130         u8 bssid[ETH_ALEN];
 131         u16 capab_info;
 132         u16 prev_capab_info;
 133         u8 ssid[32];
 134         size_t ssid_len;
 135         int beacon_seen;
 136         int proberesp_seen;
 137         int parse_error_reported;
 138         u8 wpaie[257];
 139         u8 rsnie[257];
 140         u8 osenie[257];
 141         int proto;
 142         int pairwise_cipher;
 143         int group_cipher;
 144         int mgmt_group_cipher;
 145         int key_mgmt;
 146         int rsn_capab;
 147         struct dl_list sta; /* struct wlantest_sta */
 148         struct dl_list pmk; /* struct wlantest_pmk */
 149         u8 gtk[4][32];
 150         size_t gtk_len[4];
 151         int gtk_idx;
 152         u8 rsc[4][6];
 153         u8 igtk[8][32];
 154         size_t igtk_len[8];
 155         int igtk_idx;
 156         u8 ipn[8][6];
 157         int bigtk_idx;
 158         u32 counters[NUM_WLANTEST_BSS_COUNTER];
 159         struct dl_list tdls; /* struct wlantest_tdls */
 160         u8 mdid[2];
 161         u8 r0kh_id[FT_R0KH_ID_MAX_LEN];
 162         size_t r0kh_id_len;
 163         u8 r1kh_id[FT_R1KH_ID_LEN];
 164 };
 165
 166 struct wlantest_radius {
 167         struct dl_list list;
 168         u32 srv;
 169         u32 cli;
 170         struct radius_msg *last_req;
 171 };
 172
 173
 174 #define MAX_CTRL_CONNECTIONS 10
 175 #define MAX_NOTES 10
 176
 177 struct wlantest {
 178         int monitor_sock;
 179         int monitor_wired;
 180
 181         int ctrl_sock;
 182         int ctrl_socks[MAX_CTRL_CONNECTIONS];
 183
 184         struct dl_list passphrase; /* struct wlantest_passphrase */
 185         struct dl_list bss; /* struct wlantest_bss */
 186         struct dl_list secret; /* struct wlantest_radius_secret */
 187         struct dl_list radius; /* struct wlantest_radius */
 188         struct dl_list pmk; /* struct wlantest_pmk */
 189         struct dl_list ptk; /* struct wlantest_ptk */
 190         struct dl_list wep; /* struct wlantest_wep */
 191
 192         unsigned int rx_mgmt;
 193         unsigned int rx_ctrl;
 194         unsigned int rx_data;
 195         unsigned int fcs_error;
 196
 197         void *write_pcap; /* pcap_t* */
 198         void *write_pcap_dumper; /* pcpa_dumper_t */
 199         struct timeval write_pcap_time;
 200         u8 *decrypted;
 201         size_t decrypted_len;
 202         FILE *pcapng;
 203         u32 write_pcapng_time_high;
 204         u32 write_pcapng_time_low;
 205
 206         u8 last_hdr[30];
 207         size_t last_len;
 208         int last_mgmt_valid;
 209
 210         unsigned int assume_fcs:1;
 211         unsigned int pcap_no_buffer:1;
 212         unsigned int ethernet:1;
 213
 214         char *notes[MAX_NOTES];
 215         size_t num_notes;
 216
 217         const char *write_file;
 218         const char *pcapng_file;
 219 };
 220
 221 void add_note(struct wlantest *wt, int level, const char *fmt, ...)
 222 PRINTF_FORMAT(3, 4);
 223 void clear_notes(struct wlantest *wt);
 224 size_t notes_len(struct wlantest *wt, size_t hdrlen);
 225
 226 int add_wep(struct wlantest *wt, const char *key);
 227 int read_cap_file(struct wlantest *wt, const char *fname);
 228 int read_wired_cap_file(struct wlantest *wt, const char *fname);
 229
 230 int write_pcap_init(struct wlantest *wt, const char *fname);
 231 void write_pcap_deinit(struct wlantest *wt);
 232 void write_pcap_captured(struct wlantest *wt, const u8 *buf, size_t len);
 233 void write_pcap_decrypted(struct wlantest *wt, const u8 *buf1, size_t len1,
 234                           const u8 *buf2, size_t len2);
 235
 236 int write_pcapng_init(struct wlantest *wt, const char *fname);
 237 void write_pcapng_deinit(struct wlantest *wt);
 238 struct pcap_pkthdr;
 239 void write_pcapng_write_read(struct wlantest *wt, int dlt,
 240                              struct pcap_pkthdr *hdr, const u8 *data);
 241 void write_pcapng_captured(struct wlantest *wt, const u8 *buf, size_t len);
 242
 243 void wlantest_process(struct wlantest *wt, const u8 *data, size_t len);
 244 void wlantest_process_prism(struct wlantest *wt, const u8 *data, size_t len);
 245 void wlantest_process_80211(struct wlantest *wt, const u8 *data, size_t len);
 246 void wlantest_process_wired(struct wlantest *wt, const u8 *data, size_t len);
 247 int monitor_init(struct wlantest *wt, const char *ifname);
 248 int monitor_init_wired(struct wlantest *wt, const char *ifname);
 249 void monitor_deinit(struct wlantest *wt);
 250 void rx_mgmt(struct wlantest *wt, const u8 *data, size_t len);
 251 void rx_mgmt_ack(struct wlantest *wt, const struct ieee80211_hdr *hdr);
 252 void rx_data(struct wlantest *wt, const u8 *data, size_t len);
 253 void rx_data_eapol(struct wlantest *wt, const u8 *bssid, const u8 *sta_addr,
 254                    const u8 *dst, const u8 *src,
 255                    const u8 *data, size_t len, int prot);
 256 void rx_data_ip(struct wlantest *wt, const u8 *bssid, const u8 *sta_addr,
 257                 const u8 *dst, const u8 *src, const u8 *data, size_t len,
 258                 const u8 *peer_addr);
 259 void rx_data_80211_encap(struct wlantest *wt, const u8 *bssid,
 260                          const u8 *sta_addr, const u8 *dst, const u8 *src,
 261                          const u8 *data, size_t len);
 262
 263 struct wlantest_bss * bss_find(struct wlantest *wt, const u8 *bssid);
 264 struct wlantest_bss * bss_get(struct wlantest *wt, const u8 *bssid);
 265 void bss_deinit(struct wlantest_bss *bss);
 266 void bss_update(struct wlantest *wt, struct wlantest_bss *bss,
 267                 struct ieee802_11_elems *elems);
 268 void bss_flush(struct wlantest *wt);
 269 int bss_add_pmk_from_passphrase(struct wlantest_bss *bss,
 270                                 const char *passphrase);
 271 void pmk_deinit(struct wlantest_pmk *pmk);
 272 void tdls_deinit(struct wlantest_tdls *tdls);
 273
 274 struct wlantest_sta * sta_find(struct wlantest_bss *bss, const u8 *addr);
 275 struct wlantest_sta * sta_get(struct wlantest_bss *bss, const u8 *addr);
 276 void sta_deinit(struct wlantest_sta *sta);
 277 void sta_update_assoc(struct wlantest_sta *sta,
 278                       struct ieee802_11_elems *elems);
 279
 280 u8 * ccmp_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
 281                   const u8 *data, size_t data_len, size_t *decrypted_len);
 282 u8 * ccmp_encrypt(const u8 *tk, u8 *frame, size_t len, size_t hdrlen, u8 *qos,
 283                   u8 *pn, int keyid, size_t *encrypted_len);
 284 u8 * ccmp_encrypt_pv1(const u8 *tk, const u8 *a1, const u8 *a2, const u8 *a3,
 285                       const u8 *frame, size_t len,
 286                       size_t hdrlen, const u8 *pn, int keyid,
 287                       size_t *encrypted_len);
 288 void ccmp_get_pn(u8 *pn, const u8 *data);
 289 u8 * ccmp_256_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
 290                       const u8 *data, size_t data_len, size_t *decrypted_len);
 291 u8 * ccmp_256_encrypt(const u8 *tk, u8 *frame, size_t len, size_t hdrlen,
 292                       u8 *qos, u8 *pn, int keyid, size_t *encrypted_len);
 293
 294 u8 * tkip_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
 295                   const u8 *data, size_t data_len, size_t *decrypted_len);
 296 u8 * tkip_encrypt(const u8 *tk, u8 *frame, size_t len, size_t hdrlen, u8 *qos,
 297                   u8 *pn, int keyid, size_t *encrypted_len);
 298 void tkip_get_pn(u8 *pn, const u8 *data);
 299
 300 u8 * wep_decrypt(struct wlantest *wt, const struct ieee80211_hdr *hdr,
 301                  const u8 *data, size_t data_len, size_t *decrypted_len);
 302
 303 u8 * bip_protect(const u8 *igtk, size_t igtk_len, u8 *frame, size_t len,
 304                  u8 *ipn, int keyid, size_t *prot_len);
 305 u8 * bip_gmac_protect(const u8 *igtk, size_t igtk_len, u8 *frame, size_t len,
 306                       u8 *ipn, int keyid, size_t *prot_len);
 307
 308 u8 * gcmp_decrypt(const u8 *tk, size_t tk_len, const struct ieee80211_hdr *hdr,
 309                   const u8 *data, size_t data_len, size_t *decrypted_len);
 310 u8 * gcmp_encrypt(const u8 *tk, size_t tk_len, const u8 *frame, size_t len,
 311                   size_t hdrlen, const u8 *qos,
 312                   const u8 *pn, int keyid, size_t *encrypted_len);
 313
 314 int ctrl_init(struct wlantest *wt);
 315 void ctrl_deinit(struct wlantest *wt);
 316
 317 int wlantest_inject(struct wlantest *wt, struct wlantest_bss *bss,
 318                     struct wlantest_sta *sta, u8 *frame, size_t len,
 319                     enum wlantest_inject_protection prot);
 320
 321 int wlantest_relog(struct wlantest *wt);
 322
 323 #endif /* WLANTEST_H */