rx_data.c 19 KB

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