wlantest.h 7.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247
  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. u8 rsc_tods[16 + 1][6];
  65. u8 rsc_fromds[16 + 1][6];
  66. u8 ap_sa_query_tr[2];
  67. u8 sta_sa_query_tr[2];
  68. u32 counters[NUM_WLANTEST_STA_COUNTER];
  69. u16 assocreq_capab_info;
  70. u16 assocreq_listen_int;
  71. u8 *assocreq_ies;
  72. size_t assocreq_ies_len;
  73. /* Last ICMP Echo request information */
  74. u32 icmp_echo_req_src;
  75. u32 icmp_echo_req_dst;
  76. u16 icmp_echo_req_id;
  77. u16 icmp_echo_req_seq;
  78. le16 seq_ctrl_to_sta[17];
  79. le16 seq_ctrl_to_ap[17];
  80. int pwrmgt;
  81. int pspoll;
  82. };
  83. struct wlantest_tdls {
  84. struct dl_list list;
  85. struct wlantest_sta *init;
  86. struct wlantest_sta *resp;
  87. struct tpk {
  88. u8 kck[16];
  89. u8 tk[16];
  90. } tpk;
  91. int link_up;
  92. u8 dialog_token;
  93. u8 rsc_init[16 + 1][6];
  94. u8 rsc_resp[16 + 1][6];
  95. u32 counters[NUM_WLANTEST_TDLS_COUNTER];
  96. };
  97. struct wlantest_bss {
  98. struct dl_list list;
  99. u8 bssid[ETH_ALEN];
  100. u16 capab_info;
  101. u16 prev_capab_info;
  102. u8 ssid[32];
  103. size_t ssid_len;
  104. int proberesp_seen;
  105. int parse_error_reported;
  106. u8 wpaie[257];
  107. u8 rsnie[257];
  108. int proto;
  109. int pairwise_cipher;
  110. int group_cipher;
  111. int mgmt_group_cipher;
  112. int key_mgmt;
  113. int rsn_capab;
  114. struct dl_list sta; /* struct wlantest_sta */
  115. struct dl_list pmk; /* struct wlantest_pmk */
  116. u8 gtk[4][32];
  117. size_t gtk_len[4];
  118. int gtk_idx;
  119. u8 rsc[4][6];
  120. u8 igtk[6][16];
  121. int igtk_set[6];
  122. int igtk_idx;
  123. u8 ipn[6][6];
  124. u32 counters[NUM_WLANTEST_BSS_COUNTER];
  125. struct dl_list tdls; /* struct wlantest_tdls */
  126. };
  127. struct wlantest_radius {
  128. struct dl_list list;
  129. u32 srv;
  130. u32 cli;
  131. struct radius_msg *last_req;
  132. };
  133. #define MAX_CTRL_CONNECTIONS 10
  134. struct wlantest {
  135. int monitor_sock;
  136. int monitor_wired;
  137. int ctrl_sock;
  138. int ctrl_socks[MAX_CTRL_CONNECTIONS];
  139. struct dl_list passphrase; /* struct wlantest_passphrase */
  140. struct dl_list bss; /* struct wlantest_bss */
  141. struct dl_list secret; /* struct wlantest_radius_secret */
  142. struct dl_list radius; /* struct wlantest_radius */
  143. struct dl_list pmk; /* struct wlantest_pmk */
  144. struct dl_list wep; /* struct wlantest_wep */
  145. unsigned int rx_mgmt;
  146. unsigned int rx_ctrl;
  147. unsigned int rx_data;
  148. unsigned int fcs_error;
  149. void *write_pcap; /* pcap_t* */
  150. void *write_pcap_dumper; /* pcpa_dumper_t */
  151. struct timeval write_pcap_time;
  152. u8 last_hdr[30];
  153. size_t last_len;
  154. int last_mgmt_valid;
  155. };
  156. int add_wep(struct wlantest *wt, const char *key);
  157. int read_cap_file(struct wlantest *wt, const char *fname);
  158. int read_wired_cap_file(struct wlantest *wt, const char *fname);
  159. int write_pcap_init(struct wlantest *wt, const char *fname);
  160. void write_pcap_deinit(struct wlantest *wt);
  161. void write_pcap_captured(struct wlantest *wt, const u8 *buf, size_t len);
  162. void write_pcap_decrypted(struct wlantest *wt, const u8 *buf1, size_t len1,
  163. const u8 *buf2, size_t len2);
  164. void wlantest_process(struct wlantest *wt, const u8 *data, size_t len);
  165. void wlantest_process_prism(struct wlantest *wt, const u8 *data, size_t len);
  166. void wlantest_process_80211(struct wlantest *wt, const u8 *data, size_t len);
  167. void wlantest_process_wired(struct wlantest *wt, const u8 *data, size_t len);
  168. u32 crc32(const u8 *frame, size_t frame_len);
  169. int monitor_init(struct wlantest *wt, const char *ifname);
  170. int monitor_init_wired(struct wlantest *wt, const char *ifname);
  171. void monitor_deinit(struct wlantest *wt);
  172. void rx_mgmt(struct wlantest *wt, const u8 *data, size_t len);
  173. void rx_mgmt_ack(struct wlantest *wt, const struct ieee80211_hdr *hdr);
  174. void rx_data(struct wlantest *wt, const u8 *data, size_t len);
  175. void rx_data_eapol(struct wlantest *wt, const u8 *dst, const u8 *src,
  176. const u8 *data, size_t len, int prot);
  177. void rx_data_ip(struct wlantest *wt, const u8 *bssid, const u8 *sta_addr,
  178. const u8 *dst, const u8 *src, const u8 *data, size_t len,
  179. const u8 *peer_addr);
  180. void rx_data_80211_encap(struct wlantest *wt, const u8 *bssid,
  181. const u8 *sta_addr, const u8 *dst, const u8 *src,
  182. const u8 *data, size_t len);
  183. struct wlantest_bss * bss_find(struct wlantest *wt, const u8 *bssid);
  184. struct wlantest_bss * bss_get(struct wlantest *wt, const u8 *bssid);
  185. void bss_deinit(struct wlantest_bss *bss);
  186. void bss_update(struct wlantest *wt, struct wlantest_bss *bss,
  187. struct ieee802_11_elems *elems);
  188. void bss_flush(struct wlantest *wt);
  189. int bss_add_pmk_from_passphrase(struct wlantest_bss *bss,
  190. const char *passphrase);
  191. void pmk_deinit(struct wlantest_pmk *pmk);
  192. void tdls_deinit(struct wlantest_tdls *tdls);
  193. struct wlantest_sta * sta_find(struct wlantest_bss *bss, const u8 *addr);
  194. struct wlantest_sta * sta_get(struct wlantest_bss *bss, const u8 *addr);
  195. void sta_deinit(struct wlantest_sta *sta);
  196. void sta_update_assoc(struct wlantest_sta *sta,
  197. struct ieee802_11_elems *elems);
  198. u8 * ccmp_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
  199. const u8 *data, size_t data_len, size_t *decrypted_len);
  200. u8 * ccmp_encrypt(const u8 *tk, u8 *frame, size_t len, size_t hdrlen, u8 *qos,
  201. u8 *pn, int keyid, size_t *encrypted_len);
  202. void ccmp_get_pn(u8 *pn, const u8 *data);
  203. u8 * tkip_decrypt(const u8 *tk, const struct ieee80211_hdr *hdr,
  204. const u8 *data, size_t data_len, size_t *decrypted_len);
  205. u8 * tkip_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 tkip_get_pn(u8 *pn, const u8 *data);
  208. u8 * wep_decrypt(struct wlantest *wt, const struct ieee80211_hdr *hdr,
  209. const u8 *data, size_t data_len, size_t *decrypted_len);
  210. int ctrl_init(struct wlantest *wt);
  211. void ctrl_deinit(struct wlantest *wt);
  212. int wlantest_inject(struct wlantest *wt, struct wlantest_bss *bss,
  213. struct wlantest_sta *sta, u8 *frame, size_t len,
  214. enum wlantest_inject_protection prot);
  215. #endif /* WLANTEST_H */