wlantest.h 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253
  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. #ifndef WLANTEST_H
  15. #define WLANTEST_H
  16. #include "utils/list.h"
  17. #include "common/wpa_common.h"
  18. #include "wlantest_ctrl.h"
  19. struct ieee802_11_elems;
  20. struct radius_msg;
  21. struct ieee80211_hdr;
  22. struct wlantest_bss;
  23. #define MAX_RADIUS_SECRET_LEN 128
  24. struct wlantest_radius_secret {
  25. struct dl_list list;
  26. char secret[MAX_RADIUS_SECRET_LEN];
  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. struct wlantest_pmk {
  36. struct dl_list list;
  37. u8 pmk[32];
  38. };
  39. struct wlantest_wep {
  40. struct dl_list list;
  41. size_t key_len;
  42. u8 key[13];
  43. };
  44. struct wlantest_sta {
  45. struct dl_list list;
  46. struct wlantest_bss *bss;
  47. u8 addr[ETH_ALEN];
  48. enum {
  49. STATE1 /* not authenticated */,
  50. STATE2 /* authenticated */,
  51. STATE3 /* associated */
  52. } state;
  53. u16 aid;
  54. u8 rsnie[257]; /* WPA/RSN IE */
  55. int proto;
  56. int pairwise_cipher;
  57. int group_cipher;
  58. int key_mgmt;
  59. int rsn_capab;
  60. u8 anonce[32]; /* ANonce from the previous EAPOL-Key msg 1/4 or 3/4 */
  61. u8 snonce[32]; /* SNonce from the previous EAPOL-Key msg 2/4 */
  62. struct wpa_ptk ptk; /* Derived PTK */
  63. int ptk_set;
  64. struct wpa_ptk tptk; /* Derived PTK during rekeying */
  65. int tptk_set;
  66. u8 rsc_tods[16 + 1][6];
  67. u8 rsc_fromds[16 + 1][6];
  68. u8 ap_sa_query_tr[2];
  69. u8 sta_sa_query_tr[2];
  70. u32 counters[NUM_WLANTEST_STA_COUNTER];
  71. u16 assocreq_capab_info;
  72. u16 assocreq_listen_int;
  73. u8 *assocreq_ies;
  74. size_t assocreq_ies_len;
  75. /* Last ICMP Echo request information */
  76. u32 icmp_echo_req_src;
  77. u32 icmp_echo_req_dst;
  78. u16 icmp_echo_req_id;
  79. u16 icmp_echo_req_seq;
  80. le16 seq_ctrl_to_sta[17];
  81. le16 seq_ctrl_to_ap[17];
  82. int pwrmgt;
  83. int pspoll;
  84. u8 gtk[32];
  85. size_t gtk_len;
  86. int gtk_idx;
  87. };
  88. struct wlantest_tdls {
  89. struct dl_list list;
  90. struct wlantest_sta *init;
  91. struct wlantest_sta *resp;
  92. struct tpk {
  93. u8 kck[16];
  94. u8 tk[16];
  95. } tpk;
  96. int link_up;
  97. u8 dialog_token;
  98. u8 rsc_init[16 + 1][6];
  99. u8 rsc_resp[16 + 1][6];
  100. u32 counters[NUM_WLANTEST_TDLS_COUNTER];
  101. };
  102. struct wlantest_bss {
  103. struct dl_list list;
  104. u8 bssid[ETH_ALEN];
  105. u16 capab_info;
  106. u16 prev_capab_info;
  107. u8 ssid[32];
  108. size_t ssid_len;
  109. int proberesp_seen;
  110. int parse_error_reported;
  111. u8 wpaie[257];
  112. u8 rsnie[257];
  113. int proto;
  114. int pairwise_cipher;
  115. int group_cipher;
  116. int mgmt_group_cipher;
  117. int key_mgmt;
  118. int rsn_capab;
  119. struct dl_list sta; /* struct wlantest_sta */
  120. struct dl_list pmk; /* struct wlantest_pmk */
  121. u8 gtk[4][32];
  122. size_t gtk_len[4];
  123. int gtk_idx;
  124. u8 rsc[4][6];
  125. u8 igtk[6][16];
  126. int igtk_set[6];
  127. int igtk_idx;
  128. u8 ipn[6][6];
  129. u32 counters[NUM_WLANTEST_BSS_COUNTER];
  130. struct dl_list tdls; /* struct wlantest_tdls */
  131. };
  132. struct wlantest_radius {
  133. struct dl_list list;
  134. u32 srv;
  135. u32 cli;
  136. struct radius_msg *last_req;
  137. };
  138. #define MAX_CTRL_CONNECTIONS 10
  139. struct wlantest {
  140. int monitor_sock;
  141. int monitor_wired;
  142. int ctrl_sock;
  143. int ctrl_socks[MAX_CTRL_CONNECTIONS];
  144. struct dl_list passphrase; /* struct wlantest_passphrase */
  145. struct dl_list bss; /* struct wlantest_bss */
  146. struct dl_list secret; /* struct wlantest_radius_secret */
  147. struct dl_list radius; /* struct wlantest_radius */
  148. struct dl_list pmk; /* struct wlantest_pmk */
  149. struct dl_list wep; /* struct wlantest_wep */
  150. unsigned int rx_mgmt;
  151. unsigned int rx_ctrl;
  152. unsigned int rx_data;
  153. unsigned int fcs_error;
  154. void *write_pcap; /* pcap_t* */
  155. void *write_pcap_dumper; /* pcpa_dumper_t */
  156. struct timeval write_pcap_time;
  157. u8 last_hdr[30];
  158. size_t last_len;
  159. int last_mgmt_valid;
  160. };
  161. int add_wep(struct wlantest *wt, const char *key);
  162. int read_cap_file(struct wlantest *wt, const char *fname);
  163. int read_wired_cap_file(struct wlantest *wt, const char *fname);
  164. int write_pcap_init(struct wlantest *wt, const char *fname);
  165. void write_pcap_deinit(struct wlantest *wt);
  166. void write_pcap_captured(struct wlantest *wt, const u8 *buf, size_t len);
  167. void write_pcap_decrypted(struct wlantest *wt, const u8 *buf1, size_t len1,
  168. const u8 *buf2, size_t len2);
  169. void wlantest_process(struct wlantest *wt, const u8 *data, size_t len);
  170. void wlantest_process_prism(struct wlantest *wt, const u8 *data, size_t len);
  171. void wlantest_process_80211(struct wlantest *wt, const u8 *data, size_t len);
  172. void wlantest_process_wired(struct wlantest *wt, const u8 *data, size_t len);
  173. u32 crc32(const u8 *frame, size_t frame_len);
  174. int monitor_init(struct wlantest *wt, const char *ifname);
  175. int monitor_init_wired(struct wlantest *wt, const char *ifname);
  176. void monitor_deinit(struct wlantest *wt);
  177. void rx_mgmt(struct wlantest *wt, const u8 *data, size_t len);
  178. void rx_mgmt_ack(struct wlantest *wt, const struct ieee80211_hdr *hdr);
  179. void rx_data(struct wlantest *wt, const u8 *data, size_t len);
  180. void rx_data_eapol(struct wlantest *wt, const u8 *dst, const u8 *src,
  181. const u8 *data, size_t len, int prot);
  182. void rx_data_ip(struct wlantest *wt, const u8 *bssid, const u8 *sta_addr,
  183. const u8 *dst, const u8 *src, const u8 *data, size_t len,
  184. const u8 *peer_addr);
  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. struct wlantest_bss * bss_find(struct wlantest *wt, const u8 *bssid);
  189. struct wlantest_bss * bss_get(struct wlantest *wt, const u8 *bssid);
  190. void bss_deinit(struct wlantest_bss *bss);
  191. void bss_update(struct wlantest *wt, struct wlantest_bss *bss,
  192. struct ieee802_11_elems *elems);
  193. void bss_flush(struct wlantest *wt);
  194. int bss_add_pmk_from_passphrase(struct wlantest_bss *bss,
  195. const char *passphrase);
  196. void pmk_deinit(struct wlantest_pmk *pmk);
  197. void tdls_deinit(struct wlantest_tdls *tdls);
  198. struct wlantest_sta * sta_find(struct wlantest_bss *bss, const u8 *addr);
  199. struct wlantest_sta * sta_get(struct wlantest_bss *bss, const u8 *addr);
  200. void sta_deinit(struct wlantest_sta *sta);
  201. void sta_update_assoc(struct wlantest_sta *sta,
  202. struct ieee802_11_elems *elems);
  203. u8 * ccmp_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
  204. const u8 *data, size_t data_len, size_t *decrypted_len);
  205. u8 * ccmp_encrypt(const u8 *tk, u8 *frame, size_t len, size_t hdrlen, u8 *qos,
  206. u8 *pn, int keyid, size_t *encrypted_len);
  207. void ccmp_get_pn(u8 *pn, const u8 *data);
  208. u8 * tkip_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
  209. const u8 *data, size_t data_len, size_t *decrypted_len);
  210. u8 * tkip_encrypt(const u8 *tk, u8 *frame, size_t len, size_t hdrlen, u8 *qos,
  211. u8 *pn, int keyid, size_t *encrypted_len);
  212. void tkip_get_pn(u8 *pn, const u8 *data);
  213. u8 * wep_decrypt(struct wlantest *wt, const struct ieee80211_hdr *hdr,
  214. const u8 *data, size_t data_len, size_t *decrypted_len);
  215. int ctrl_init(struct wlantest *wt);
  216. void ctrl_deinit(struct wlantest *wt);
  217. int wlantest_inject(struct wlantest *wt, struct wlantest_bss *bss,
  218. struct wlantest_sta *sta, u8 *frame, size_t len,
  219. enum wlantest_inject_protection prot);
  220. #endif /* WLANTEST_H */