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 program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License version 2 as
   7  * published by the Free Software Foundation.
   8  *
   9  * Alternatively, this software may be distributed under the terms of BSD
  10  * license.
  11  *
  12  * See README and COPYING for more details.
  13  */
  14
  15 #ifndef WLANTEST_H
  16 #define WLANTEST_H
  17
  18 #include "utils/list.h"
  19 #include "common/wpa_common.h"
  20 #include "wlantest_ctrl.h"
  21
  22 struct ieee802_11_elems;
  23 struct radius_msg;
  24 struct ieee80211_hdr;
  25 struct wlantest_bss;
  26
  27 #define MAX_RADIUS_SECRET_LEN 128
  28
  29 struct wlantest_radius_secret {
  30         struct dl_list list;
  31         char secret[MAX_RADIUS_SECRET_LEN];
  32 };
  33
  34 struct wlantest_passphrase {
  35         struct dl_list list;
  36         char passphrase[64];
  37         u8 ssid[32];
  38         size_t ssid_len;
  39         u8 bssid[ETH_ALEN];
  40 };
  41
  42 struct wlantest_pmk {
  43         struct dl_list list;
  44         u8 pmk[32];
  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         u8 rsc_tods[16 + 1][6];
  68         u8 rsc_fromds[16 + 1][6];
  69         u8 ap_sa_query_tr[2];
  70         u8 sta_sa_query_tr[2];
  71         u32 counters[NUM_WLANTEST_STA_COUNTER];
  72         u16 assocreq_capab_info;
  73         u16 assocreq_listen_int;
  74         u8 *assocreq_ies;
  75         size_t assocreq_ies_len;
  76
  77         /* Last ICMP Echo request information */
  78         u32 icmp_echo_req_src;
  79         u32 icmp_echo_req_dst;
  80         u16 icmp_echo_req_id;
  81         u16 icmp_echo_req_seq;
  82
  83         le16 seq_ctrl_to_sta[17];
  84         le16 seq_ctrl_to_ap[17];
  85 };
  86
  87 struct wlantest_tdls {
  88         struct dl_list list;
  89         struct wlantest_sta *init;
  90         struct wlantest_sta *resp;
  91         struct tpk {
  92                 u8 kck[16];
  93                 u8 tk[16];
  94         } tpk;
  95         int link_up;
  96         u8 dialog_token;
  97         u8 rsc_init[16 + 1][6];
  98         u8 rsc_resp[16 + 1][6];
  99 };
 100
 101 struct wlantest_bss {
 102         struct dl_list list;
 103         u8 bssid[ETH_ALEN];
 104         u16 capab_info;
 105         u8 ssid[32];
 106         size_t ssid_len;
 107         int proberesp_seen;
 108         int parse_error_reported;
 109         u8 wpaie[257];
 110         u8 rsnie[257];
 111         int proto;
 112         int pairwise_cipher;
 113         int group_cipher;
 114         int mgmt_group_cipher;
 115         int key_mgmt;
 116         int rsn_capab;
 117         struct dl_list sta; /* struct wlantest_sta */
 118         struct dl_list pmk; /* struct wlantest_pmk */
 119         u8 gtk[4][32];
 120         size_t gtk_len[4];
 121         int gtk_idx;
 122         u8 rsc[4][6];
 123         u8 igtk[6][16];
 124         int igtk_set[6];
 125         int igtk_idx;
 126         u8 ipn[6][6];
 127         u32 counters[NUM_WLANTEST_BSS_COUNTER];
 128         struct dl_list tdls; /* struct wlantest_tdls */
 129 };
 130
 131 struct wlantest_radius {
 132         struct dl_list list;
 133         u32 srv;
 134         u32 cli;
 135         struct radius_msg *last_req;
 136 };
 137
 138
 139 #define MAX_CTRL_CONNECTIONS 10
 140
 141 struct wlantest {
 142         int monitor_sock;
 143         int monitor_wired;
 144
 145         int ctrl_sock;
 146         int ctrl_socks[MAX_CTRL_CONNECTIONS];
 147
 148         struct dl_list passphrase; /* struct wlantest_passphrase */
 149         struct dl_list bss; /* struct wlantest_bss */
 150         struct dl_list secret; /* struct wlantest_radius_secret */
 151         struct dl_list radius; /* struct wlantest_radius */
 152         struct dl_list pmk; /* struct wlantest_pmk */
 153
 154         unsigned int rx_mgmt;
 155         unsigned int rx_ctrl;
 156         unsigned int rx_data;
 157         unsigned int fcs_error;
 158
 159         void *write_pcap; /* pcap_t* */
 160         void *write_pcap_dumper; /* pcpa_dumper_t */
 161         struct timeval write_pcap_time;
 162 };
 163
 164 int read_cap_file(struct wlantest *wt, const char *fname);
 165 int read_wired_cap_file(struct wlantest *wt, const char *fname);
 166 int write_pcap_init(struct wlantest *wt, const char *fname);
 167 void write_pcap_deinit(struct wlantest *wt);
 168 void write_pcap_captured(struct wlantest *wt, const u8 *buf, size_t len);
 169 void write_pcap_decrypted(struct wlantest *wt, const u8 *buf1, size_t len1,
 170                           const u8 *buf2, size_t len2);
 171 void wlantest_process(struct wlantest *wt, const u8 *data, size_t len);
 172 void wlantest_process_prism(struct wlantest *wt, const u8 *data, size_t len);
 173 void wlantest_process_80211(struct wlantest *wt, const u8 *data, size_t len);
 174 void wlantest_process_wired(struct wlantest *wt, const u8 *data, size_t len);
 175 u32 crc32(const u8 *frame, size_t frame_len);
 176 int monitor_init(struct wlantest *wt, const char *ifname);
 177 int monitor_init_wired(struct wlantest *wt, const char *ifname);
 178 void monitor_deinit(struct wlantest *wt);
 179 void rx_mgmt(struct wlantest *wt, const u8 *data, size_t len);
 180 void rx_data(struct wlantest *wt, const u8 *data, size_t len);
 181 void rx_data_eapol(struct wlantest *wt, const u8 *dst, const u8 *src,
 182                    const u8 *data, size_t len, int prot);
 183 void rx_data_ip(struct wlantest *wt, const u8 *bssid, const u8 *sta_addr,
 184                 const u8 *dst, const u8 *src, const u8 *data, size_t len);
 185 void rx_data_80211_encap(struct wlantest *wt, const u8 *bssid,
 186                          const u8 *sta_addr, const u8 *dst, const u8 *src,
 187                          const u8 *data, size_t len);
 188
 189 struct wlantest_bss * bss_find(struct wlantest *wt, const u8 *bssid);
 190 struct wlantest_bss * bss_get(struct wlantest *wt, const u8 *bssid);
 191 void bss_deinit(struct wlantest_bss *bss);
 192 void bss_update(struct wlantest *wt, struct wlantest_bss *bss,
 193                 struct ieee802_11_elems *elems);
 194 void bss_flush(struct wlantest *wt);
 195 int bss_add_pmk_from_passphrase(struct wlantest_bss *bss,
 196                                 const char *passphrase);
 197 void pmk_deinit(struct wlantest_pmk *pmk);
 198
 199 struct wlantest_sta * sta_find(struct wlantest_bss *bss, const u8 *addr);
 200 struct wlantest_sta * sta_get(struct wlantest_bss *bss, const u8 *addr);
 201 void sta_deinit(struct wlantest_sta *sta);
 202 void sta_update_assoc(struct wlantest_sta *sta,
 203                       struct ieee802_11_elems *elems);
 204
 205 u8 * ccmp_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
 206                   const u8 *data, size_t data_len, size_t *decrypted_len);
 207 u8 * ccmp_encrypt(const u8 *tk, u8 *frame, size_t len, size_t hdrlen, u8 *qos,
 208                   u8 *pn, int keyid, size_t *encrypted_len);
 209 void ccmp_get_pn(u8 *pn, const u8 *data);
 210
 211 u8 * tkip_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
 212                   const u8 *data, size_t data_len, size_t *decrypted_len);
 213 u8 * tkip_encrypt(const u8 *tk, u8 *frame, size_t len, size_t hdrlen, u8 *qos,
 214                   u8 *pn, int keyid, size_t *encrypted_len);
 215 void tkip_get_pn(u8 *pn, const u8 *data);
 216
 217 int ctrl_init(struct wlantest *wt);
 218 void ctrl_deinit(struct wlantest *wt);
 219
 220 int wlantest_inject(struct wlantest *wt, struct wlantest_bss *bss,
 221                     struct wlantest_sta *sta, u8 *frame, size_t len,
 222                     enum wlantest_inject_protection prot);
 223
 224 #endif /* WLANTEST_H */