rx_data.c 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662
  1. /*
  2. * Received Data frame processing
  3. * Copyright (c) 2010-2015, 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. #include "utils/includes.h"
  9. #include <linux/if_ether.h>
  10. #include "utils/common.h"
  11. #include "common/defs.h"
  12. #include "common/ieee802_11_defs.h"
  13. #include "wlantest.h"
  14. static const char * data_stype(u16 stype)
  15. {
  16. switch (stype) {
  17. case WLAN_FC_STYPE_DATA:
  18. return "DATA";
  19. case WLAN_FC_STYPE_DATA_CFACK:
  20. return "DATA-CFACK";
  21. case WLAN_FC_STYPE_DATA_CFPOLL:
  22. return "DATA-CFPOLL";
  23. case WLAN_FC_STYPE_DATA_CFACKPOLL:
  24. return "DATA-CFACKPOLL";
  25. case WLAN_FC_STYPE_NULLFUNC:
  26. return "NULLFUNC";
  27. case WLAN_FC_STYPE_CFACK:
  28. return "CFACK";
  29. case WLAN_FC_STYPE_CFPOLL:
  30. return "CFPOLL";
  31. case WLAN_FC_STYPE_CFACKPOLL:
  32. return "CFACKPOLL";
  33. case WLAN_FC_STYPE_QOS_DATA:
  34. return "QOSDATA";
  35. case WLAN_FC_STYPE_QOS_DATA_CFACK:
  36. return "QOSDATA-CFACK";
  37. case WLAN_FC_STYPE_QOS_DATA_CFPOLL:
  38. return "QOSDATA-CFPOLL";
  39. case WLAN_FC_STYPE_QOS_DATA_CFACKPOLL:
  40. return "QOSDATA-CFACKPOLL";
  41. case WLAN_FC_STYPE_QOS_NULL:
  42. return "QOS-NULL";
  43. case WLAN_FC_STYPE_QOS_CFPOLL:
  44. return "QOS-CFPOLL";
  45. case WLAN_FC_STYPE_QOS_CFACKPOLL:
  46. return "QOS-CFACKPOLL";
  47. }
  48. return "??";
  49. }
  50. static void rx_data_eth(struct wlantest *wt, const u8 *bssid,
  51. const u8 *sta_addr, const u8 *dst, const u8 *src,
  52. u16 ethertype, const u8 *data, size_t len, int prot,
  53. const u8 *peer_addr)
  54. {
  55. switch (ethertype) {
  56. case ETH_P_PAE:
  57. rx_data_eapol(wt, dst, src, data, len, prot);
  58. break;
  59. case ETH_P_IP:
  60. rx_data_ip(wt, bssid, sta_addr, dst, src, data, len,
  61. peer_addr);
  62. break;
  63. case 0x890d:
  64. rx_data_80211_encap(wt, bssid, sta_addr, dst, src, data, len);
  65. break;
  66. }
  67. }
  68. static void rx_data_process(struct wlantest *wt, const u8 *bssid,
  69. const u8 *sta_addr,
  70. const u8 *dst, const u8 *src,
  71. const u8 *data, size_t len, int prot,
  72. const u8 *peer_addr)
  73. {
  74. if (len == 0)
  75. return;
  76. if (len >= 8 && os_memcmp(data, "\xaa\xaa\x03\x00\x00\x00", 6) == 0) {
  77. rx_data_eth(wt, bssid, sta_addr, dst, src,
  78. WPA_GET_BE16(data + 6), data + 8, len - 8, prot,
  79. peer_addr);
  80. return;
  81. }
  82. wpa_hexdump(MSG_DEBUG, "Unrecognized LLC", data, len > 8 ? 8 : len);
  83. }
  84. static u8 * try_all_ptk(struct wlantest *wt, int pairwise_cipher,
  85. const struct ieee80211_hdr *hdr,
  86. const u8 *data, size_t data_len, size_t *decrypted_len)
  87. {
  88. struct wlantest_ptk *ptk;
  89. u8 *decrypted;
  90. int prev_level = wpa_debug_level;
  91. wpa_debug_level = MSG_WARNING;
  92. dl_list_for_each(ptk, &wt->ptk, struct wlantest_ptk, list) {
  93. unsigned int tk_len = ptk->ptk_len - 32;
  94. decrypted = NULL;
  95. if ((pairwise_cipher == WPA_CIPHER_CCMP ||
  96. pairwise_cipher == 0) && tk_len == 16) {
  97. decrypted = ccmp_decrypt(ptk->ptk.tk, hdr, data,
  98. data_len, decrypted_len);
  99. } else if ((pairwise_cipher == WPA_CIPHER_CCMP_256 ||
  100. pairwise_cipher == 0) && tk_len == 32) {
  101. decrypted = ccmp_256_decrypt(ptk->ptk.tk, hdr, data,
  102. data_len, decrypted_len);
  103. } else if ((pairwise_cipher == WPA_CIPHER_GCMP ||
  104. pairwise_cipher == WPA_CIPHER_GCMP_256 ||
  105. pairwise_cipher == 0) &&
  106. (tk_len == 16 || tk_len == 32)) {
  107. decrypted = gcmp_decrypt(ptk->ptk.tk, tk_len, hdr,
  108. data, data_len, decrypted_len);
  109. } else if ((pairwise_cipher == WPA_CIPHER_TKIP ||
  110. pairwise_cipher == 0) && tk_len == 32) {
  111. decrypted = tkip_decrypt(ptk->ptk.tk, hdr, data,
  112. data_len, decrypted_len);
  113. }
  114. if (decrypted) {
  115. wpa_debug_level = prev_level;
  116. add_note(wt, MSG_DEBUG, "Found PTK match from list of all known PTKs");
  117. return decrypted;
  118. }
  119. }
  120. wpa_debug_level = prev_level;
  121. return NULL;
  122. }
  123. static void rx_data_bss_prot_group(struct wlantest *wt,
  124. const struct ieee80211_hdr *hdr,
  125. const u8 *qos, const u8 *dst, const u8 *src,
  126. const u8 *data, size_t len)
  127. {
  128. struct wlantest_bss *bss;
  129. int keyid;
  130. u8 *decrypted = NULL;
  131. size_t dlen;
  132. u8 pn[6];
  133. bss = bss_get(wt, hdr->addr2);
  134. if (bss == NULL)
  135. return;
  136. if (len < 4) {
  137. add_note(wt, MSG_INFO, "Too short group addressed data frame");
  138. return;
  139. }
  140. if (bss->group_cipher & (WPA_CIPHER_TKIP | WPA_CIPHER_CCMP) &&
  141. !(data[3] & 0x20)) {
  142. add_note(wt, MSG_INFO, "Expected TKIP/CCMP frame from "
  143. MACSTR " did not have ExtIV bit set to 1",
  144. MAC2STR(bss->bssid));
  145. return;
  146. }
  147. if (bss->group_cipher == WPA_CIPHER_TKIP) {
  148. if (data[3] & 0x1f) {
  149. add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
  150. " used non-zero reserved bit",
  151. MAC2STR(bss->bssid));
  152. }
  153. if (data[1] != ((data[0] | 0x20) & 0x7f)) {
  154. add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
  155. " used incorrect WEPSeed[1] (was 0x%x, "
  156. "expected 0x%x)",
  157. MAC2STR(bss->bssid), data[1],
  158. (data[0] | 0x20) & 0x7f);
  159. }
  160. } else if (bss->group_cipher == WPA_CIPHER_CCMP) {
  161. if (data[2] != 0 || (data[3] & 0x1f) != 0) {
  162. add_note(wt, MSG_INFO, "CCMP frame from " MACSTR
  163. " used non-zero reserved bit",
  164. MAC2STR(bss->bssid));
  165. }
  166. }
  167. keyid = data[3] >> 6;
  168. if (bss->gtk_len[keyid] == 0 && bss->group_cipher != WPA_CIPHER_WEP40)
  169. {
  170. add_note(wt, MSG_MSGDUMP, "No GTK known to decrypt the frame "
  171. "(A2=" MACSTR " KeyID=%d)",
  172. MAC2STR(hdr->addr2), keyid);
  173. return;
  174. }
  175. if (bss->group_cipher == WPA_CIPHER_TKIP)
  176. tkip_get_pn(pn, data);
  177. else if (bss->group_cipher == WPA_CIPHER_WEP40)
  178. goto skip_replay_det;
  179. else
  180. ccmp_get_pn(pn, data);
  181. if (os_memcmp(pn, bss->rsc[keyid], 6) <= 0) {
  182. u16 seq_ctrl = le_to_host16(hdr->seq_ctrl);
  183. add_note(wt, MSG_INFO, "CCMP/TKIP replay detected: A1=" MACSTR
  184. " A2=" MACSTR " A3=" MACSTR " seq=%u frag=%u%s",
  185. MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
  186. MAC2STR(hdr->addr3),
  187. WLAN_GET_SEQ_SEQ(seq_ctrl),
  188. WLAN_GET_SEQ_FRAG(seq_ctrl),
  189. (le_to_host16(hdr->frame_control) & WLAN_FC_RETRY) ?
  190. " Retry" : "");
  191. wpa_hexdump(MSG_INFO, "RX PN", pn, 6);
  192. wpa_hexdump(MSG_INFO, "RSC", bss->rsc[keyid], 6);
  193. }
  194. skip_replay_det:
  195. if (bss->group_cipher == WPA_CIPHER_TKIP)
  196. decrypted = tkip_decrypt(bss->gtk[keyid], hdr, data, len,
  197. &dlen);
  198. else if (bss->group_cipher == WPA_CIPHER_WEP40)
  199. decrypted = wep_decrypt(wt, hdr, data, len, &dlen);
  200. else if (bss->group_cipher == WPA_CIPHER_CCMP)
  201. decrypted = ccmp_decrypt(bss->gtk[keyid], hdr, data, len,
  202. &dlen);
  203. else if (bss->group_cipher == WPA_CIPHER_CCMP_256)
  204. decrypted = ccmp_256_decrypt(bss->gtk[keyid], hdr, data, len,
  205. &dlen);
  206. else if (bss->group_cipher == WPA_CIPHER_GCMP ||
  207. bss->group_cipher == WPA_CIPHER_GCMP_256)
  208. decrypted = gcmp_decrypt(bss->gtk[keyid], bss->gtk_len[keyid],
  209. hdr, data, len, &dlen);
  210. if (decrypted) {
  211. rx_data_process(wt, bss->bssid, NULL, dst, src, decrypted,
  212. dlen, 1, NULL);
  213. os_memcpy(bss->rsc[keyid], pn, 6);
  214. write_pcap_decrypted(wt, (const u8 *) hdr, 24 + (qos ? 2 : 0),
  215. decrypted, dlen);
  216. } else
  217. add_note(wt, MSG_DEBUG, "Failed to decrypt frame");
  218. os_free(decrypted);
  219. }
  220. static void rx_data_bss_prot(struct wlantest *wt,
  221. const struct ieee80211_hdr *hdr, const u8 *qos,
  222. const u8 *dst, const u8 *src, const u8 *data,
  223. size_t len)
  224. {
  225. struct wlantest_bss *bss;
  226. struct wlantest_sta *sta, *sta2;
  227. int keyid;
  228. u16 fc = le_to_host16(hdr->frame_control);
  229. u8 *decrypted;
  230. size_t dlen;
  231. int tid;
  232. u8 pn[6], *rsc;
  233. struct wlantest_tdls *tdls = NULL, *found;
  234. const u8 *tk = NULL;
  235. int ptk_iter_done = 0;
  236. int try_ptk_iter = 0;
  237. if (hdr->addr1[0] & 0x01) {
  238. rx_data_bss_prot_group(wt, hdr, qos, dst, src, data, len);
  239. return;
  240. }
  241. if (fc & WLAN_FC_TODS) {
  242. bss = bss_get(wt, hdr->addr1);
  243. if (bss == NULL)
  244. return;
  245. sta = sta_get(bss, hdr->addr2);
  246. if (sta)
  247. sta->counters[WLANTEST_STA_COUNTER_PROT_DATA_TX]++;
  248. } else if (fc & WLAN_FC_FROMDS) {
  249. bss = bss_get(wt, hdr->addr2);
  250. if (bss == NULL)
  251. return;
  252. sta = sta_get(bss, hdr->addr1);
  253. } else {
  254. bss = bss_get(wt, hdr->addr3);
  255. if (bss == NULL)
  256. return;
  257. sta = sta_find(bss, hdr->addr2);
  258. sta2 = sta_find(bss, hdr->addr1);
  259. if (sta == NULL || sta2 == NULL)
  260. return;
  261. found = NULL;
  262. dl_list_for_each(tdls, &bss->tdls, struct wlantest_tdls, list)
  263. {
  264. if ((tdls->init == sta && tdls->resp == sta2) ||
  265. (tdls->init == sta2 && tdls->resp == sta)) {
  266. found = tdls;
  267. if (tdls->link_up)
  268. break;
  269. }
  270. }
  271. if (found) {
  272. if (!found->link_up)
  273. add_note(wt, MSG_DEBUG,
  274. "TDLS: Link not up, but Data "
  275. "frame seen");
  276. tk = found->tpk.tk;
  277. tdls = found;
  278. }
  279. }
  280. if ((sta == NULL ||
  281. (!sta->ptk_set && sta->pairwise_cipher != WPA_CIPHER_WEP40)) &&
  282. tk == NULL) {
  283. add_note(wt, MSG_MSGDUMP, "No PTK known to decrypt the frame");
  284. if (dl_list_empty(&wt->ptk))
  285. return;
  286. try_ptk_iter = 1;
  287. }
  288. if (len < 4) {
  289. add_note(wt, MSG_INFO, "Too short encrypted data frame");
  290. return;
  291. }
  292. if (sta == NULL)
  293. return;
  294. if (sta->pairwise_cipher & (WPA_CIPHER_TKIP | WPA_CIPHER_CCMP) &&
  295. !(data[3] & 0x20)) {
  296. add_note(wt, MSG_INFO, "Expected TKIP/CCMP frame from "
  297. MACSTR " did not have ExtIV bit set to 1",
  298. MAC2STR(src));
  299. return;
  300. }
  301. if (tk == NULL && sta->pairwise_cipher == WPA_CIPHER_TKIP) {
  302. if (data[3] & 0x1f) {
  303. add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
  304. " used non-zero reserved bit",
  305. MAC2STR(hdr->addr2));
  306. }
  307. if (data[1] != ((data[0] | 0x20) & 0x7f)) {
  308. add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
  309. " used incorrect WEPSeed[1] (was 0x%x, "
  310. "expected 0x%x)",
  311. MAC2STR(hdr->addr2), data[1],
  312. (data[0] | 0x20) & 0x7f);
  313. }
  314. } else if (tk || sta->pairwise_cipher == WPA_CIPHER_CCMP) {
  315. if (data[2] != 0 || (data[3] & 0x1f) != 0) {
  316. add_note(wt, MSG_INFO, "CCMP frame from " MACSTR
  317. " used non-zero reserved bit",
  318. MAC2STR(hdr->addr2));
  319. }
  320. }
  321. keyid = data[3] >> 6;
  322. if (keyid != 0) {
  323. add_note(wt, MSG_INFO, "Unexpected non-zero KeyID %d in "
  324. "individually addressed Data frame from " MACSTR,
  325. keyid, MAC2STR(hdr->addr2));
  326. }
  327. if (qos) {
  328. tid = qos[0] & 0x0f;
  329. if (fc & WLAN_FC_TODS)
  330. sta->tx_tid[tid]++;
  331. else
  332. sta->rx_tid[tid]++;
  333. } else {
  334. tid = 0;
  335. if (fc & WLAN_FC_TODS)
  336. sta->tx_tid[16]++;
  337. else
  338. sta->rx_tid[16]++;
  339. }
  340. if (tk) {
  341. if (os_memcmp(hdr->addr2, tdls->init->addr, ETH_ALEN) == 0)
  342. rsc = tdls->rsc_init[tid];
  343. else
  344. rsc = tdls->rsc_resp[tid];
  345. } else if (fc & WLAN_FC_TODS)
  346. rsc = sta->rsc_tods[tid];
  347. else
  348. rsc = sta->rsc_fromds[tid];
  349. if (tk == NULL && sta->pairwise_cipher == WPA_CIPHER_TKIP)
  350. tkip_get_pn(pn, data);
  351. else if (sta->pairwise_cipher == WPA_CIPHER_WEP40)
  352. goto skip_replay_det;
  353. else
  354. ccmp_get_pn(pn, data);
  355. if (os_memcmp(pn, rsc, 6) <= 0) {
  356. u16 seq_ctrl = le_to_host16(hdr->seq_ctrl);
  357. add_note(wt, MSG_INFO, "CCMP/TKIP replay detected: A1=" MACSTR
  358. " A2=" MACSTR " A3=" MACSTR " seq=%u frag=%u%s",
  359. MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
  360. MAC2STR(hdr->addr3),
  361. WLAN_GET_SEQ_SEQ(seq_ctrl),
  362. WLAN_GET_SEQ_FRAG(seq_ctrl),
  363. (le_to_host16(hdr->frame_control) & WLAN_FC_RETRY) ?
  364. " Retry" : "");
  365. wpa_hexdump(MSG_INFO, "RX PN", pn, 6);
  366. wpa_hexdump(MSG_INFO, "RSC", rsc, 6);
  367. }
  368. skip_replay_det:
  369. if (tk) {
  370. if (sta->pairwise_cipher == WPA_CIPHER_CCMP_256)
  371. decrypted = ccmp_256_decrypt(tk, hdr, data, len, &dlen);
  372. else if (sta->pairwise_cipher == WPA_CIPHER_GCMP ||
  373. sta->pairwise_cipher == WPA_CIPHER_GCMP_256)
  374. decrypted = gcmp_decrypt(tk, sta->tk_len, hdr, data,
  375. len, &dlen);
  376. else
  377. decrypted = ccmp_decrypt(tk, hdr, data, len, &dlen);
  378. } else if (sta->pairwise_cipher == WPA_CIPHER_TKIP) {
  379. decrypted = tkip_decrypt(sta->ptk.tk, hdr, data, len, &dlen);
  380. } else if (sta->pairwise_cipher == WPA_CIPHER_WEP40) {
  381. decrypted = wep_decrypt(wt, hdr, data, len, &dlen);
  382. } else if (sta->ptk_set) {
  383. if (sta->pairwise_cipher == WPA_CIPHER_CCMP_256)
  384. decrypted = ccmp_256_decrypt(sta->ptk.tk, hdr, data,
  385. len, &dlen);
  386. else if (sta->pairwise_cipher == WPA_CIPHER_GCMP ||
  387. sta->pairwise_cipher == WPA_CIPHER_GCMP_256)
  388. decrypted = gcmp_decrypt(sta->ptk.tk, sta->tk_len,
  389. hdr, data, len, &dlen);
  390. else
  391. decrypted = ccmp_decrypt(sta->ptk.tk, hdr, data, len,
  392. &dlen);
  393. } else {
  394. decrypted = try_all_ptk(wt, sta->pairwise_cipher, hdr, data,
  395. len, &dlen);
  396. ptk_iter_done = 1;
  397. }
  398. if (!decrypted && !ptk_iter_done) {
  399. decrypted = try_all_ptk(wt, sta->pairwise_cipher, hdr, data,
  400. len, &dlen);
  401. if (decrypted) {
  402. add_note(wt, MSG_DEBUG, "Current PTK did not work, but found a match from all known PTKs");
  403. }
  404. }
  405. if (decrypted) {
  406. u16 fc = le_to_host16(hdr->frame_control);
  407. const u8 *peer_addr = NULL;
  408. if (!(fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)))
  409. peer_addr = hdr->addr1;
  410. os_memcpy(rsc, pn, 6);
  411. rx_data_process(wt, bss->bssid, sta->addr, dst, src, decrypted,
  412. dlen, 1, peer_addr);
  413. write_pcap_decrypted(wt, (const u8 *) hdr, 24 + (qos ? 2 : 0),
  414. decrypted, dlen);
  415. } else if (!try_ptk_iter)
  416. add_note(wt, MSG_DEBUG, "Failed to decrypt frame");
  417. os_free(decrypted);
  418. }
  419. static void rx_data_bss(struct wlantest *wt, const struct ieee80211_hdr *hdr,
  420. const u8 *qos, const u8 *dst, const u8 *src,
  421. const u8 *data, size_t len)
  422. {
  423. u16 fc = le_to_host16(hdr->frame_control);
  424. int prot = !!(fc & WLAN_FC_ISWEP);
  425. if (qos) {
  426. u8 ack = (qos[0] & 0x60) >> 5;
  427. wpa_printf(MSG_MSGDUMP, "BSS DATA: " MACSTR " -> " MACSTR
  428. " len=%u%s tid=%u%s%s",
  429. MAC2STR(src), MAC2STR(dst), (unsigned int) len,
  430. prot ? " Prot" : "", qos[0] & 0x0f,
  431. (qos[0] & 0x10) ? " EOSP" : "",
  432. ack == 0 ? "" :
  433. (ack == 1 ? " NoAck" :
  434. (ack == 2 ? " NoExpAck" : " BA")));
  435. } else {
  436. wpa_printf(MSG_MSGDUMP, "BSS DATA: " MACSTR " -> " MACSTR
  437. " len=%u%s",
  438. MAC2STR(src), MAC2STR(dst), (unsigned int) len,
  439. prot ? " Prot" : "");
  440. }
  441. if (prot)
  442. rx_data_bss_prot(wt, hdr, qos, dst, src, data, len);
  443. else {
  444. const u8 *bssid, *sta_addr, *peer_addr;
  445. struct wlantest_bss *bss;
  446. if (fc & WLAN_FC_TODS) {
  447. bssid = hdr->addr1;
  448. sta_addr = hdr->addr2;
  449. peer_addr = NULL;
  450. } else if (fc & WLAN_FC_FROMDS) {
  451. bssid = hdr->addr2;
  452. sta_addr = hdr->addr1;
  453. peer_addr = NULL;
  454. } else {
  455. bssid = hdr->addr3;
  456. sta_addr = hdr->addr2;
  457. peer_addr = hdr->addr1;
  458. }
  459. bss = bss_get(wt, bssid);
  460. if (bss) {
  461. struct wlantest_sta *sta = sta_get(bss, sta_addr);
  462. if (sta) {
  463. if (qos) {
  464. int tid = qos[0] & 0x0f;
  465. if (fc & WLAN_FC_TODS)
  466. sta->tx_tid[tid]++;
  467. else
  468. sta->rx_tid[tid]++;
  469. } else {
  470. if (fc & WLAN_FC_TODS)
  471. sta->tx_tid[16]++;
  472. else
  473. sta->rx_tid[16]++;
  474. }
  475. }
  476. }
  477. rx_data_process(wt, bssid, sta_addr, dst, src, data, len, 0,
  478. peer_addr);
  479. }
  480. }
  481. static struct wlantest_tdls * get_tdls(struct wlantest *wt, const u8 *bssid,
  482. const u8 *sta1_addr,
  483. const u8 *sta2_addr)
  484. {
  485. struct wlantest_bss *bss;
  486. struct wlantest_sta *sta1, *sta2;
  487. struct wlantest_tdls *tdls, *found = NULL;
  488. bss = bss_find(wt, bssid);
  489. if (bss == NULL)
  490. return NULL;
  491. sta1 = sta_find(bss, sta1_addr);
  492. if (sta1 == NULL)
  493. return NULL;
  494. sta2 = sta_find(bss, sta2_addr);
  495. if (sta2 == NULL)
  496. return NULL;
  497. dl_list_for_each(tdls, &bss->tdls, struct wlantest_tdls, list) {
  498. if ((tdls->init == sta1 && tdls->resp == sta2) ||
  499. (tdls->init == sta2 && tdls->resp == sta1)) {
  500. found = tdls;
  501. if (tdls->link_up)
  502. break;
  503. }
  504. }
  505. return found;
  506. }
  507. static void add_direct_link(struct wlantest *wt, const u8 *bssid,
  508. const u8 *sta1_addr, const u8 *sta2_addr)
  509. {
  510. struct wlantest_tdls *tdls;
  511. tdls = get_tdls(wt, bssid, sta1_addr, sta2_addr);
  512. if (tdls == NULL)
  513. return;
  514. if (tdls->link_up)
  515. tdls->counters[WLANTEST_TDLS_COUNTER_VALID_DIRECT_LINK]++;
  516. else
  517. tdls->counters[WLANTEST_TDLS_COUNTER_INVALID_DIRECT_LINK]++;
  518. }
  519. static void add_ap_path(struct wlantest *wt, const u8 *bssid,
  520. const u8 *sta1_addr, const u8 *sta2_addr)
  521. {
  522. struct wlantest_tdls *tdls;
  523. tdls = get_tdls(wt, bssid, sta1_addr, sta2_addr);
  524. if (tdls == NULL)
  525. return;
  526. if (tdls->link_up)
  527. tdls->counters[WLANTEST_TDLS_COUNTER_INVALID_AP_PATH]++;
  528. else
  529. tdls->counters[WLANTEST_TDLS_COUNTER_VALID_AP_PATH]++;
  530. }
  531. void rx_data(struct wlantest *wt, const u8 *data, size_t len)
  532. {
  533. const struct ieee80211_hdr *hdr;
  534. u16 fc, stype;
  535. size_t hdrlen;
  536. const u8 *qos = NULL;
  537. if (len < 24)
  538. return;
  539. hdr = (const struct ieee80211_hdr *) data;
  540. fc = le_to_host16(hdr->frame_control);
  541. stype = WLAN_FC_GET_STYPE(fc);
  542. hdrlen = 24;
  543. if ((fc & (WLAN_FC_TODS | WLAN_FC_FROMDS)) ==
  544. (WLAN_FC_TODS | WLAN_FC_FROMDS))
  545. hdrlen += ETH_ALEN;
  546. if (stype & 0x08) {
  547. qos = data + hdrlen;
  548. hdrlen += 2;
  549. }
  550. if (len < hdrlen)
  551. return;
  552. wt->rx_data++;
  553. switch (fc & (WLAN_FC_TODS | WLAN_FC_FROMDS)) {
  554. case 0:
  555. wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s IBSS DA=" MACSTR " SA="
  556. MACSTR " BSSID=" MACSTR,
  557. data_stype(WLAN_FC_GET_STYPE(fc)),
  558. fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
  559. fc & WLAN_FC_ISWEP ? " Prot" : "",
  560. MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
  561. MAC2STR(hdr->addr3));
  562. add_direct_link(wt, hdr->addr3, hdr->addr1, hdr->addr2);
  563. rx_data_bss(wt, hdr, qos, hdr->addr1, hdr->addr2,
  564. data + hdrlen, len - hdrlen);
  565. break;
  566. case WLAN_FC_FROMDS:
  567. wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s FromDS DA=" MACSTR
  568. " BSSID=" MACSTR " SA=" MACSTR,
  569. data_stype(WLAN_FC_GET_STYPE(fc)),
  570. fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
  571. fc & WLAN_FC_ISWEP ? " Prot" : "",
  572. MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
  573. MAC2STR(hdr->addr3));
  574. add_ap_path(wt, hdr->addr2, hdr->addr1, hdr->addr3);
  575. rx_data_bss(wt, hdr, qos, hdr->addr1, hdr->addr3,
  576. data + hdrlen, len - hdrlen);
  577. break;
  578. case WLAN_FC_TODS:
  579. wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s ToDS BSSID=" MACSTR
  580. " SA=" MACSTR " DA=" MACSTR,
  581. data_stype(WLAN_FC_GET_STYPE(fc)),
  582. fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
  583. fc & WLAN_FC_ISWEP ? " Prot" : "",
  584. MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
  585. MAC2STR(hdr->addr3));
  586. add_ap_path(wt, hdr->addr1, hdr->addr3, hdr->addr2);
  587. rx_data_bss(wt, hdr, qos, hdr->addr3, hdr->addr2,
  588. data + hdrlen, len - hdrlen);
  589. break;
  590. case WLAN_FC_TODS | WLAN_FC_FROMDS:
  591. wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s WDS RA=" MACSTR " TA="
  592. MACSTR " DA=" MACSTR " SA=" MACSTR,
  593. data_stype(WLAN_FC_GET_STYPE(fc)),
  594. fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
  595. fc & WLAN_FC_ISWEP ? " Prot" : "",
  596. MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
  597. MAC2STR(hdr->addr3),
  598. MAC2STR((const u8 *) (hdr + 1)));
  599. break;
  600. }
  601. }