wlantest/rx_data.c

   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
   9 #include "utils/includes.h"
  10
  11 #include "utils/common.h"
  12 #include "common/defs.h"
  13 #include "common/ieee802_11_defs.h"
  14 #include "wlantest.h"
  15
  16
  17 static const char * data_stype(u16 stype)
  18 {
  19         switch (stype) {
  20         case WLAN_FC_STYPE_DATA:
  21                 return "DATA";
  22         case WLAN_FC_STYPE_DATA_CFACK:
  23                 return "DATA-CFACK";
  24         case WLAN_FC_STYPE_DATA_CFPOLL:
  25                 return "DATA-CFPOLL";
  26         case WLAN_FC_STYPE_DATA_CFACKPOLL:
  27                 return "DATA-CFACKPOLL";
  28         case WLAN_FC_STYPE_NULLFUNC:
  29                 return "NULLFUNC";
  30         case WLAN_FC_STYPE_CFACK:
  31                 return "CFACK";
  32         case WLAN_FC_STYPE_CFPOLL:
  33                 return "CFPOLL";
  34         case WLAN_FC_STYPE_CFACKPOLL:
  35                 return "CFACKPOLL";
  36         case WLAN_FC_STYPE_QOS_DATA:
  37                 return "QOSDATA";
  38         case WLAN_FC_STYPE_QOS_DATA_CFACK:
  39                 return "QOSDATA-CFACK";
  40         case WLAN_FC_STYPE_QOS_DATA_CFPOLL:
  41                 return "QOSDATA-CFPOLL";
  42         case WLAN_FC_STYPE_QOS_DATA_CFACKPOLL:
  43                 return "QOSDATA-CFACKPOLL";
  44         case WLAN_FC_STYPE_QOS_NULL:
  45                 return "QOS-NULL";
  46         case WLAN_FC_STYPE_QOS_CFPOLL:
  47                 return "QOS-CFPOLL";
  48         case WLAN_FC_STYPE_QOS_CFACKPOLL:
  49                 return "QOS-CFACKPOLL";
  50         }
  51         return "??";
  52 }
  53
  54
  55 static void rx_data_eth(struct wlantest *wt, const u8 *bssid,
  56                         const u8 *sta_addr, const u8 *dst, const u8 *src,
  57                         u16 ethertype, const u8 *data, size_t len, int prot,
  58                         const u8 *peer_addr);
  59
  60 static void rx_data_vlan(struct wlantest *wt, const u8 *bssid,
  61                          const u8 *sta_addr, const u8 *dst, const u8 *src,
  62                          const u8 *data, size_t len, int prot,
  63                          const u8 *peer_addr)
  64 {
  65         u16 tag;
  66
  67         if (len < 4)
  68                 return;
  69         tag = WPA_GET_BE16(data);
  70         wpa_printf(MSG_MSGDUMP, "VLAN tag: Priority=%u ID=%u",
  71                    tag >> 12, tag & 0x0ffff);
  72         /* ignore VLAN information and process the original frame */
  73         rx_data_eth(wt, bssid, sta_addr, dst, src, WPA_GET_BE16(data + 2),
  74                     data + 4, len - 4, prot, peer_addr);
  75 }
  76
  77
  78 static void rx_data_eth(struct wlantest *wt, const u8 *bssid,
  79                         const u8 *sta_addr, const u8 *dst, const u8 *src,
  80                         u16 ethertype, const u8 *data, size_t len, int prot,
  81                         const u8 *peer_addr)
  82 {
  83         switch (ethertype) {
  84         case ETH_P_PAE:
  85                 rx_data_eapol(wt, bssid, sta_addr, dst, src, data, len, prot);
  86                 break;
  87         case ETH_P_IP:
  88                 rx_data_ip(wt, bssid, sta_addr, dst, src, data, len,
  89                            peer_addr);
  90                 break;
  91         case 0x890d:
  92                 rx_data_80211_encap(wt, bssid, sta_addr, dst, src, data, len);
  93                 break;
  94         case ETH_P_8021Q:
  95                 rx_data_vlan(wt, bssid, sta_addr, dst, src, data, len, prot,
  96                              peer_addr);
  97                 break;
  98         }
  99 }
 100
 101
 102 static void rx_data_process(struct wlantest *wt, const u8 *bssid,
 103                             const u8 *sta_addr,
 104                             const u8 *dst, const u8 *src,
 105                             const u8 *data, size_t len, int prot,
 106                             const u8 *peer_addr)
 107 {
 108         if (len == 0)
 109                 return;
 110
 111         if (len >= 8 && os_memcmp(data, "\xaa\xaa\x03\x00\x00\x00", 6) == 0) {
 112                 rx_data_eth(wt, bssid, sta_addr, dst, src,
 113                             WPA_GET_BE16(data + 6), data + 8, len - 8, prot,
 114                             peer_addr);
 115                 return;
 116         }
 117
 118         wpa_hexdump(MSG_DEBUG, "Unrecognized LLC", data, len > 8 ? 8 : len);
 119 }
 120
 121
 122 static void write_decrypted_note(struct wlantest *wt, const u8 *decrypted,
 123                                  const u8 *tk, size_t tk_len, int keyid)
 124 {
 125         char tk_hex[65];
 126
 127         if (!decrypted)
 128                 return;
 129
 130         wpa_snprintf_hex(tk_hex, sizeof(tk_hex), tk, tk_len);
 131         add_note(wt, MSG_EXCESSIVE, "TK[%d] %s", keyid, tk_hex);
 132 }
 133
 134
 135 static u8 * try_ptk(int pairwise_cipher, struct wpa_ptk *ptk,
 136                     const struct ieee80211_hdr *hdr,
 137                     const u8 *data, size_t data_len, size_t *decrypted_len)
 138 {
 139         u8 *decrypted;
 140         unsigned int tk_len = ptk->tk_len;
 141
 142         decrypted = NULL;
 143         if ((pairwise_cipher == WPA_CIPHER_CCMP ||
 144              pairwise_cipher == 0) && tk_len == 16) {
 145                 decrypted = ccmp_decrypt(ptk->tk, hdr, data,
 146                                          data_len, decrypted_len);
 147         } else if ((pairwise_cipher == WPA_CIPHER_CCMP_256 ||
 148                     pairwise_cipher == 0) && tk_len == 32) {
 149                 decrypted = ccmp_256_decrypt(ptk->tk, hdr, data,
 150                                              data_len, decrypted_len);
 151         } else if ((pairwise_cipher == WPA_CIPHER_GCMP ||
 152                     pairwise_cipher == WPA_CIPHER_GCMP_256 ||
 153                     pairwise_cipher == 0) &&
 154                    (tk_len == 16 || tk_len == 32)) {
 155                 decrypted = gcmp_decrypt(ptk->tk, tk_len, hdr,
 156                                          data, data_len, decrypted_len);
 157         } else if ((pairwise_cipher == WPA_CIPHER_TKIP ||
 158                     pairwise_cipher == 0) && tk_len == 32) {
 159                 decrypted = tkip_decrypt(ptk->tk, hdr, data, data_len,
 160                                          decrypted_len);
 161         }
 162
 163         return decrypted;
 164 }
 165
 166
 167 static u8 * try_all_ptk(struct wlantest *wt, int pairwise_cipher,
 168                         const struct ieee80211_hdr *hdr, int keyid,
 169                         const u8 *data, size_t data_len, size_t *decrypted_len)
 170 {
 171         struct wlantest_ptk *ptk;
 172         u8 *decrypted;
 173         int prev_level = wpa_debug_level;
 174
 175         wpa_debug_level = MSG_WARNING;
 176         dl_list_for_each(ptk, &wt->ptk, struct wlantest_ptk, list) {
 177                 decrypted = try_ptk(pairwise_cipher, &ptk->ptk, hdr,
 178                                     data, data_len, decrypted_len);
 179                 if (decrypted) {
 180                         wpa_debug_level = prev_level;
 181                         add_note(wt, MSG_DEBUG,
 182                                  "Found PTK match from list of all known PTKs");
 183                         write_decrypted_note(wt, decrypted, ptk->ptk.tk,
 184                                              ptk->ptk.tk_len, keyid);
 185                         return decrypted;
 186                 }
 187         }
 188         wpa_debug_level = prev_level;
 189
 190         return NULL;
 191 }
 192
 193
 194 static void check_plaintext_prot(struct wlantest *wt,
 195                                  const struct ieee80211_hdr *hdr,
 196                                  const u8 *data, size_t len)
 197 {
 198         if (len < 8 + 3 || data[8] != 0xaa || data[9] != 0xaa ||
 199             data[10] != 0x03)
 200                 return;
 201
 202         add_note(wt, MSG_DEBUG,
 203                  "Plaintext payload in protected frame");
 204         wpa_printf(MSG_INFO, "Plaintext payload in protected frame #%u: A2="
 205                    MACSTR " seq=%u",
 206                    wt->frame_num, MAC2STR(hdr->addr2),
 207                    WLAN_GET_SEQ_SEQ(le_to_host16(hdr->seq_ctrl)));
 208 }
 209
 210
 211 static void rx_data_bss_prot_group(struct wlantest *wt,
 212                                    const struct ieee80211_hdr *hdr,
 213                                    size_t hdrlen,
 214                                    const u8 *qos, const u8 *dst, const u8 *src,
 215                                    const u8 *data, size_t len)
 216 {
 217         struct wlantest_bss *bss;
 218         int keyid;
 219         u8 *decrypted = NULL;
 220         size_t dlen;
 221         u8 pn[6];
 222         int replay = 0;
 223
 224         bss = bss_get(wt, hdr->addr2);
 225         if (bss == NULL)
 226                 return;
 227         if (len < 4) {
 228                 add_note(wt, MSG_INFO, "Too short group addressed data frame");
 229                 return;
 230         }
 231
 232         if (bss->group_cipher & (WPA_CIPHER_TKIP | WPA_CIPHER_CCMP) &&
 233             !(data[3] & 0x20)) {
 234                 add_note(wt, MSG_INFO, "Expected TKIP/CCMP frame from "
 235                          MACSTR " did not have ExtIV bit set to 1",
 236                          MAC2STR(bss->bssid));
 237                 return;
 238         }
 239
 240         if (bss->group_cipher == WPA_CIPHER_TKIP) {
 241                 if (data[3] & 0x1f) {
 242                         add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
 243                                  " used non-zero reserved bit",
 244                                  MAC2STR(bss->bssid));
 245                 }
 246                 if (data[1] != ((data[0] | 0x20) & 0x7f)) {
 247                         add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
 248                                  " used incorrect WEPSeed[1] (was 0x%x, "
 249                                  "expected 0x%x)",
 250                                  MAC2STR(bss->bssid), data[1],
 251                                  (data[0] | 0x20) & 0x7f);
 252                 }
 253         } else if (bss->group_cipher == WPA_CIPHER_CCMP) {
 254                 if (data[2] != 0 || (data[3] & 0x1f) != 0) {
 255                         add_note(wt, MSG_INFO, "CCMP frame from " MACSTR
 256                                  " used non-zero reserved bit",
 257                                  MAC2STR(bss->bssid));
 258                 }
 259         }
 260
 261         check_plaintext_prot(wt, hdr, data, len);
 262         keyid = data[3] >> 6;
 263         if (bss->gtk_len[keyid] == 0 && bss->group_cipher != WPA_CIPHER_WEP40)
 264         {
 265                 add_note(wt, MSG_MSGDUMP, "No GTK known to decrypt the frame "
 266                          "(A2=" MACSTR " KeyID=%d)",
 267                          MAC2STR(hdr->addr2), keyid);
 268                 return;
 269         }
 270
 271         if (bss->group_cipher == WPA_CIPHER_TKIP)
 272                 tkip_get_pn(pn, data);
 273         else if (bss->group_cipher == WPA_CIPHER_WEP40)
 274                 goto skip_replay_det;
 275         else
 276                 ccmp_get_pn(pn, data);
 277         if (os_memcmp(pn, bss->rsc[keyid], 6) <= 0) {
 278                 u16 seq_ctrl = le_to_host16(hdr->seq_ctrl);
 279                 char pn_hex[6 * 2 + 1], rsc_hex[6 * 2 + 1];
 280
 281                 wpa_snprintf_hex(pn_hex, sizeof(pn_hex), pn, 6);
 282                 wpa_snprintf_hex(rsc_hex, sizeof(rsc_hex), bss->rsc[keyid], 6);
 283                 add_note(wt, MSG_INFO, "replay detected: A1=" MACSTR
 284                          " A2=" MACSTR " A3=" MACSTR
 285                          " seq=%u frag=%u%s keyid=%d %s<=%s",
 286                          MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
 287                          MAC2STR(hdr->addr3),
 288                          WLAN_GET_SEQ_SEQ(seq_ctrl),
 289                          WLAN_GET_SEQ_FRAG(seq_ctrl),
 290                          (le_to_host16(hdr->frame_control) & WLAN_FC_RETRY) ?
 291                          " Retry" : "",
 292                          keyid, pn_hex, rsc_hex);
 293                 wpa_hexdump(MSG_INFO, "RX PN", pn, 6);
 294                 wpa_hexdump(MSG_INFO, "RSC", bss->rsc[keyid], 6);
 295                 replay = 1;
 296         }
 297
 298 skip_replay_det:
 299         if (bss->group_cipher == WPA_CIPHER_TKIP)
 300                 decrypted = tkip_decrypt(bss->gtk[keyid], hdr, data, len,
 301                                          &dlen);
 302         else if (bss->group_cipher == WPA_CIPHER_WEP40)
 303                 decrypted = wep_decrypt(wt, hdr, data, len, &dlen);
 304         else if (bss->group_cipher == WPA_CIPHER_CCMP)
 305                 decrypted = ccmp_decrypt(bss->gtk[keyid], hdr, data, len,
 306                                          &dlen);
 307         else if (bss->group_cipher == WPA_CIPHER_CCMP_256)
 308                 decrypted = ccmp_256_decrypt(bss->gtk[keyid], hdr, data, len,
 309                                              &dlen);
 310         else if (bss->group_cipher == WPA_CIPHER_GCMP ||
 311                  bss->group_cipher == WPA_CIPHER_GCMP_256)
 312                 decrypted = gcmp_decrypt(bss->gtk[keyid], bss->gtk_len[keyid],
 313                                          hdr, data, len, &dlen);
 314
 315         if (decrypted) {
 316                 char gtk[65];
 317
 318                 wpa_snprintf_hex(gtk, sizeof(gtk), bss->gtk[keyid],
 319                                  bss->gtk_len[keyid]);
 320                 add_note(wt, MSG_EXCESSIVE, "GTK[%d] %s", keyid, gtk);
 321                 rx_data_process(wt, bss->bssid, NULL, dst, src, decrypted,
 322                                 dlen, 1, NULL);
 323                 if (!replay)
 324                         os_memcpy(bss->rsc[keyid], pn, 6);
 325                 write_pcap_decrypted(wt, (const u8 *) hdr, hdrlen,
 326                                      decrypted, dlen);
 327         } else
 328                 add_note(wt, MSG_DEBUG, "Failed to decrypt frame");
 329         os_free(decrypted);
 330 }
 331
 332
 333 static void rx_data_bss_prot(struct wlantest *wt,
 334                              const struct ieee80211_hdr *hdr, size_t hdrlen,
 335                              const u8 *qos, const u8 *dst, const u8 *src,
 336                              const u8 *data, size_t len)
 337 {
 338         struct wlantest_bss *bss, *bss2;
 339         struct wlantest_sta *sta, *sta2;
 340         int keyid;
 341         u16 fc = le_to_host16(hdr->frame_control);
 342         u8 *decrypted = NULL;
 343         size_t dlen;
 344         int tid;
 345         u8 pn[6], *rsc = NULL;
 346         struct wlantest_tdls *tdls = NULL, *found;
 347         const u8 *tk = NULL;
 348         int ptk_iter_done = 0;
 349         int try_ptk_iter = 0;
 350         int replay = 0;
 351
 352         if (hdr->addr1[0] & 0x01) {
 353                 rx_data_bss_prot_group(wt, hdr, hdrlen, qos, dst, src,
 354                                        data, len);
 355                 return;
 356         }
 357
 358         if ((fc & (WLAN_FC_TODS | WLAN_FC_FROMDS)) ==
 359             (WLAN_FC_TODS | WLAN_FC_FROMDS)) {
 360                 bss = bss_find(wt, hdr->addr1);
 361                 if (bss) {
 362                         sta = sta_find(bss, hdr->addr2);
 363                         if (sta) {
 364                                 sta->counters[
 365                                         WLANTEST_STA_COUNTER_PROT_DATA_TX]++;
 366                         }
 367                         if (!sta || !sta->ptk_set) {
 368                                 bss2 = bss_find(wt, hdr->addr2);
 369                                 if (bss2) {
 370                                         sta2 = sta_find(bss2, hdr->addr1);
 371                                         if (sta2 && (!sta || sta2->ptk_set)) {
 372                                                 bss = bss2;
 373                                                 sta = sta2;
 374                                         }
 375                                 }
 376                         }
 377                 } else {
 378                         bss = bss_find(wt, hdr->addr2);
 379                         if (!bss)
 380                                 return;
 381                         sta = sta_find(bss, hdr->addr1);
 382                 }
 383         } else if (fc & WLAN_FC_TODS) {
 384                 bss = bss_get(wt, hdr->addr1);
 385                 if (bss == NULL)
 386                         return;
 387                 sta = sta_get(bss, hdr->addr2);
 388                 if (sta)
 389                         sta->counters[WLANTEST_STA_COUNTER_PROT_DATA_TX]++;
 390         } else if (fc & WLAN_FC_FROMDS) {
 391                 bss = bss_get(wt, hdr->addr2);
 392                 if (bss == NULL)
 393                         return;
 394                 sta = sta_get(bss, hdr->addr1);
 395         } else {
 396                 bss = bss_get(wt, hdr->addr3);
 397                 if (bss == NULL)
 398                         return;
 399                 sta = sta_find(bss, hdr->addr2);
 400                 sta2 = sta_find(bss, hdr->addr1);
 401                 if (sta == NULL || sta2 == NULL)
 402                         return;
 403                 found = NULL;
 404                 dl_list_for_each(tdls, &bss->tdls, struct wlantest_tdls, list)
 405                 {
 406                         if ((tdls->init == sta && tdls->resp == sta2) ||
 407                             (tdls->init == sta2 && tdls->resp == sta)) {
 408                                 found = tdls;
 409                                 if (tdls->link_up)
 410                                         break;
 411                         }
 412                 }
 413                 if (found) {
 414                         if (!found->link_up)
 415                                 add_note(wt, MSG_DEBUG,
 416                                          "TDLS: Link not up, but Data "
 417                                          "frame seen");
 418                         tk = found->tpk.tk;
 419                         tdls = found;
 420                 }
 421         }
 422         check_plaintext_prot(wt, hdr, data, len);
 423         if ((sta == NULL ||
 424              (!sta->ptk_set && sta->pairwise_cipher != WPA_CIPHER_WEP40)) &&
 425             tk == NULL) {
 426                 add_note(wt, MSG_MSGDUMP, "No PTK known to decrypt the frame");
 427                 if (dl_list_empty(&wt->ptk)) {
 428                         if (len >= 4 && sta) {
 429                                 keyid = data[3] >> 6;
 430                                 goto check_zero_tk;
 431                         }
 432                         return;
 433                 }
 434
 435                 try_ptk_iter = 1;
 436         }
 437
 438         if (len < 4) {
 439                 add_note(wt, MSG_INFO, "Too short encrypted data frame");
 440                 return;
 441         }
 442
 443         if (sta == NULL)
 444                 return;
 445         if (sta->pairwise_cipher & (WPA_CIPHER_TKIP | WPA_CIPHER_CCMP) &&
 446             !(data[3] & 0x20)) {
 447                 add_note(wt, MSG_INFO, "Expected TKIP/CCMP frame from "
 448                          MACSTR " did not have ExtIV bit set to 1",
 449                          MAC2STR(src));
 450                 return;
 451         }
 452
 453         if (tk == NULL && sta->pairwise_cipher == WPA_CIPHER_TKIP) {
 454                 if (data[3] & 0x1f) {
 455                         add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
 456                                  " used non-zero reserved bit",
 457                                  MAC2STR(hdr->addr2));
 458                 }
 459                 if (data[1] != ((data[0] | 0x20) & 0x7f)) {
 460                         add_note(wt, MSG_INFO, "TKIP frame from " MACSTR
 461                                  " used incorrect WEPSeed[1] (was 0x%x, "
 462                                  "expected 0x%x)",
 463                                  MAC2STR(hdr->addr2), data[1],
 464                                  (data[0] | 0x20) & 0x7f);
 465                 }
 466         } else if (tk || sta->pairwise_cipher == WPA_CIPHER_CCMP) {
 467                 if (data[2] != 0 || (data[3] & 0x1f) != 0) {
 468                         add_note(wt, MSG_INFO, "CCMP frame from " MACSTR
 469                                  " used non-zero reserved bit",
 470                                  MAC2STR(hdr->addr2));
 471                 }
 472         }
 473
 474         keyid = data[3] >> 6;
 475         if (keyid != 0 &&
 476             (!(sta->rsn_capab & WPA_CAPABILITY_EXT_KEY_ID_FOR_UNICAST) ||
 477              !(bss->rsn_capab & WPA_CAPABILITY_EXT_KEY_ID_FOR_UNICAST) ||
 478              keyid != 1)) {
 479                 add_note(wt, MSG_INFO,
 480                          "Unexpected KeyID %d in individually addressed Data frame from "
 481                          MACSTR,
 482                          keyid, MAC2STR(hdr->addr2));
 483         }
 484
 485         if (qos) {
 486                 tid = qos[0] & 0x0f;
 487                 if (fc & WLAN_FC_TODS)
 488                         sta->tx_tid[tid]++;
 489                 else
 490                         sta->rx_tid[tid]++;
 491         } else {
 492                 tid = 0;
 493                 if (fc & WLAN_FC_TODS)
 494                         sta->tx_tid[16]++;
 495                 else
 496                         sta->rx_tid[16]++;
 497         }
 498         if (tk) {
 499                 if (os_memcmp(hdr->addr2, tdls->init->addr, ETH_ALEN) == 0)
 500                         rsc = tdls->rsc_init[tid];
 501                 else
 502                         rsc = tdls->rsc_resp[tid];
 503         } else if ((fc & (WLAN_FC_TODS | WLAN_FC_FROMDS)) ==
 504                    (WLAN_FC_TODS | WLAN_FC_FROMDS)) {
 505                 if (os_memcmp(sta->addr, hdr->addr2, ETH_ALEN) == 0)
 506                         rsc = sta->rsc_tods[tid];
 507                 else
 508                         rsc = sta->rsc_fromds[tid];
 509         } else if (fc & WLAN_FC_TODS)
 510                 rsc = sta->rsc_tods[tid];
 511         else
 512                 rsc = sta->rsc_fromds[tid];
 513
 514
 515         if (tk == NULL && sta->pairwise_cipher == WPA_CIPHER_TKIP)
 516                 tkip_get_pn(pn, data);
 517         else if (sta->pairwise_cipher == WPA_CIPHER_WEP40)
 518                 goto skip_replay_det;
 519         else
 520                 ccmp_get_pn(pn, data);
 521         if (os_memcmp(pn, rsc, 6) <= 0) {
 522                 u16 seq_ctrl = le_to_host16(hdr->seq_ctrl);
 523                 char pn_hex[6 * 2 + 1], rsc_hex[6 * 2 + 1];
 524
 525                 wpa_snprintf_hex(pn_hex, sizeof(pn_hex), pn, 6);
 526                 wpa_snprintf_hex(rsc_hex, sizeof(rsc_hex), rsc, 6);
 527                 add_note(wt, MSG_INFO, "replay detected: A1=" MACSTR
 528                          " A2=" MACSTR " A3=" MACSTR
 529                          " seq=%u frag=%u%s keyid=%d tid=%d %s<=%s",
 530                          MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
 531                          MAC2STR(hdr->addr3),
 532                          WLAN_GET_SEQ_SEQ(seq_ctrl),
 533                          WLAN_GET_SEQ_FRAG(seq_ctrl),
 534                          (le_to_host16(hdr->frame_control) &  WLAN_FC_RETRY) ?
 535                          " Retry" : "",
 536                          keyid, tid, pn_hex, rsc_hex);
 537                 wpa_hexdump(MSG_INFO, "RX PN", pn, 6);
 538                 wpa_hexdump(MSG_INFO, "RSC", rsc, 6);
 539                 replay = 1;
 540         }
 541
 542 skip_replay_det:
 543         if (tk) {
 544                 if (sta->pairwise_cipher == WPA_CIPHER_CCMP_256) {
 545                         decrypted = ccmp_256_decrypt(tk, hdr, data, len, &dlen);
 546                         write_decrypted_note(wt, decrypted, tk, 32, keyid);
 547                 } else if (sta->pairwise_cipher == WPA_CIPHER_GCMP ||
 548                            sta->pairwise_cipher == WPA_CIPHER_GCMP_256) {
 549                         decrypted = gcmp_decrypt(tk, sta->ptk.tk_len, hdr, data,
 550                                                  len, &dlen);
 551                         write_decrypted_note(wt, decrypted, tk, sta->ptk.tk_len,
 552                                              keyid);
 553                 } else {
 554                         decrypted = ccmp_decrypt(tk, hdr, data, len, &dlen);
 555                         write_decrypted_note(wt, decrypted, tk, 16, keyid);
 556                 }
 557         } else if (sta->pairwise_cipher == WPA_CIPHER_TKIP) {
 558                 decrypted = tkip_decrypt(sta->ptk.tk, hdr, data, len, &dlen);
 559                 write_decrypted_note(wt, decrypted, sta->ptk.tk, 32, keyid);
 560         } else if (sta->pairwise_cipher == WPA_CIPHER_WEP40) {
 561                 decrypted = wep_decrypt(wt, hdr, data, len, &dlen);
 562         } else if (sta->ptk_set) {
 563                 if (sta->pairwise_cipher == WPA_CIPHER_CCMP_256)
 564                         decrypted = ccmp_256_decrypt(sta->ptk.tk, hdr, data,
 565                                                      len, &dlen);
 566                 else if (sta->pairwise_cipher == WPA_CIPHER_GCMP ||
 567                          sta->pairwise_cipher == WPA_CIPHER_GCMP_256)
 568                         decrypted = gcmp_decrypt(sta->ptk.tk, sta->ptk.tk_len,
 569                                                  hdr, data, len, &dlen);
 570                 else
 571                         decrypted = ccmp_decrypt(sta->ptk.tk, hdr, data, len,
 572                                                  &dlen);
 573                 write_decrypted_note(wt, decrypted, sta->ptk.tk,
 574                                      sta->ptk.tk_len, keyid);
 575         } else {
 576                 decrypted = try_all_ptk(wt, sta->pairwise_cipher, hdr, keyid,
 577                                         data, len, &dlen);
 578                 ptk_iter_done = 1;
 579         }
 580         if (!decrypted && !ptk_iter_done) {
 581                 decrypted = try_all_ptk(wt, sta->pairwise_cipher, hdr, keyid,
 582                                         data, len, &dlen);
 583                 if (decrypted) {
 584                         add_note(wt, MSG_DEBUG, "Current PTK did not work, but found a match from all known PTKs");
 585                 }
 586         }
 587 check_zero_tk:
 588         if (!decrypted) {
 589                 struct wpa_ptk zero_ptk;
 590                 int old_debug_level = wpa_debug_level;
 591
 592                 os_memset(&zero_ptk, 0, sizeof(zero_ptk));
 593                 zero_ptk.tk_len = wpa_cipher_key_len(sta->pairwise_cipher);
 594                 wpa_debug_level = MSG_ERROR;
 595                 decrypted = try_ptk(sta->pairwise_cipher, &zero_ptk, hdr,
 596                                     data, len, &dlen);
 597                 wpa_debug_level = old_debug_level;
 598                 if (decrypted) {
 599                         add_note(wt, MSG_DEBUG,
 600                                  "Frame was encrypted with zero TK");
 601                         wpa_printf(MSG_INFO, "Zero TK used in frame #%u: A2="
 602                                    MACSTR " seq=%u",
 603                                    wt->frame_num, MAC2STR(hdr->addr2),
 604                                    WLAN_GET_SEQ_SEQ(
 605                                            le_to_host16(hdr->seq_ctrl)));
 606                         write_decrypted_note(wt, decrypted, zero_ptk.tk,
 607                                              zero_ptk.tk_len, keyid);
 608                 }
 609         }
 610         if (decrypted) {
 611                 u16 fc = le_to_host16(hdr->frame_control);
 612                 const u8 *peer_addr = NULL;
 613                 if (!(fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)))
 614                         peer_addr = hdr->addr1;
 615                 if (!replay && rsc)
 616                         os_memcpy(rsc, pn, 6);
 617                 rx_data_process(wt, bss->bssid, sta->addr, dst, src, decrypted,
 618                                 dlen, 1, peer_addr);
 619                 write_pcap_decrypted(wt, (const u8 *) hdr, hdrlen,
 620                                      decrypted, dlen);
 621         } else {
 622                 if (!try_ptk_iter)
 623                         add_note(wt, MSG_DEBUG, "Failed to decrypt frame");
 624
 625                 /* Assume the frame was corrupted and there was no FCS to check.
 626                  * Allow retry of this particular frame to be processed so that
 627                  * it could end up getting decrypted if it was received without
 628                  * corruption. */
 629                 sta->allow_duplicate = 1;
 630         }
 631         os_free(decrypted);
 632 }
 633
 634
 635 static void rx_data_bss(struct wlantest *wt, const struct ieee80211_hdr *hdr,
 636                         size_t hdrlen, const u8 *qos, const u8 *dst,
 637                         const u8 *src, const u8 *data, size_t len)
 638 {
 639         u16 fc = le_to_host16(hdr->frame_control);
 640         int prot = !!(fc & WLAN_FC_ISWEP);
 641
 642         if (qos) {
 643                 u8 ack = (qos[0] & 0x60) >> 5;
 644                 wpa_printf(MSG_MSGDUMP, "BSS DATA: " MACSTR " -> " MACSTR
 645                            " len=%u%s tid=%u%s%s",
 646                            MAC2STR(src), MAC2STR(dst), (unsigned int) len,
 647                            prot ? " Prot" : "", qos[0] & 0x0f,
 648                            (qos[0] & 0x10) ? " EOSP" : "",
 649                            ack == 0 ? "" :
 650                            (ack == 1 ? " NoAck" :
 651                             (ack == 2 ? " NoExpAck" : " BA")));
 652         } else {
 653                 wpa_printf(MSG_MSGDUMP, "BSS DATA: " MACSTR " -> " MACSTR
 654                            " len=%u%s",
 655                            MAC2STR(src), MAC2STR(dst), (unsigned int) len,
 656                            prot ? " Prot" : "");
 657         }
 658
 659         if (prot)
 660                 rx_data_bss_prot(wt, hdr, hdrlen, qos, dst, src, data, len);
 661         else {
 662                 const u8 *bssid, *sta_addr, *peer_addr;
 663                 struct wlantest_bss *bss;
 664
 665                 if (fc & WLAN_FC_TODS) {
 666                         bssid = hdr->addr1;
 667                         sta_addr = hdr->addr2;
 668                         peer_addr = NULL;
 669                 } else if (fc & WLAN_FC_FROMDS) {
 670                         bssid = hdr->addr2;
 671                         sta_addr = hdr->addr1;
 672                         peer_addr = NULL;
 673                 } else {
 674                         bssid = hdr->addr3;
 675                         sta_addr = hdr->addr2;
 676                         peer_addr = hdr->addr1;
 677                 }
 678
 679                 bss = bss_get(wt, bssid);
 680                 if (bss) {
 681                         struct wlantest_sta *sta = sta_get(bss, sta_addr);
 682
 683                         if (sta) {
 684                                 if (qos) {
 685                                         int tid = qos[0] & 0x0f;
 686                                         if (fc & WLAN_FC_TODS)
 687                                                 sta->tx_tid[tid]++;
 688                                         else
 689                                                 sta->rx_tid[tid]++;
 690                                 } else {
 691                                         if (fc & WLAN_FC_TODS)
 692                                                 sta->tx_tid[16]++;
 693                                         else
 694                                                 sta->rx_tid[16]++;
 695                                 }
 696                         }
 697                 }
 698
 699                 rx_data_process(wt, bssid, sta_addr, dst, src, data, len, 0,
 700                                 peer_addr);
 701         }
 702 }
 703
 704
 705 static struct wlantest_tdls * get_tdls(struct wlantest *wt, const u8 *bssid,
 706                                        const u8 *sta1_addr,
 707                                        const u8 *sta2_addr)
 708 {
 709         struct wlantest_bss *bss;
 710         struct wlantest_sta *sta1, *sta2;
 711         struct wlantest_tdls *tdls, *found = NULL;
 712
 713         bss = bss_find(wt, bssid);
 714         if (bss == NULL)
 715                 return NULL;
 716         sta1 = sta_find(bss, sta1_addr);
 717         if (sta1 == NULL)
 718                 return NULL;
 719         sta2 = sta_find(bss, sta2_addr);
 720         if (sta2 == NULL)
 721                 return NULL;
 722
 723         dl_list_for_each(tdls, &bss->tdls, struct wlantest_tdls, list) {
 724                 if ((tdls->init == sta1 && tdls->resp == sta2) ||
 725                     (tdls->init == sta2 && tdls->resp == sta1)) {
 726                         found = tdls;
 727                         if (tdls->link_up)
 728                                 break;
 729                 }
 730         }
 731
 732         return found;
 733 }
 734
 735
 736 static void add_direct_link(struct wlantest *wt, const u8 *bssid,
 737                             const u8 *sta1_addr, const u8 *sta2_addr)
 738 {
 739         struct wlantest_tdls *tdls;
 740
 741         tdls = get_tdls(wt, bssid, sta1_addr, sta2_addr);
 742         if (tdls == NULL)
 743                 return;
 744
 745         if (tdls->link_up)
 746                 tdls->counters[WLANTEST_TDLS_COUNTER_VALID_DIRECT_LINK]++;
 747         else
 748                 tdls->counters[WLANTEST_TDLS_COUNTER_INVALID_DIRECT_LINK]++;
 749 }
 750
 751
 752 static void add_ap_path(struct wlantest *wt, const u8 *bssid,
 753                         const u8 *sta1_addr, const u8 *sta2_addr)
 754 {
 755         struct wlantest_tdls *tdls;
 756
 757         tdls = get_tdls(wt, bssid, sta1_addr, sta2_addr);
 758         if (tdls == NULL)
 759                 return;
 760
 761         if (tdls->link_up)
 762                 tdls->counters[WLANTEST_TDLS_COUNTER_INVALID_AP_PATH]++;
 763         else
 764                 tdls->counters[WLANTEST_TDLS_COUNTER_VALID_AP_PATH]++;
 765 }
 766
 767
 768 void rx_data(struct wlantest *wt, const u8 *data, size_t len)
 769 {
 770         const struct ieee80211_hdr *hdr;
 771         u16 fc, stype;
 772         size_t hdrlen;
 773         const u8 *qos = NULL;
 774
 775         if (len < 24)
 776                 return;
 777
 778         hdr = (const struct ieee80211_hdr *) data;
 779         fc = le_to_host16(hdr->frame_control);
 780         stype = WLAN_FC_GET_STYPE(fc);
 781         hdrlen = 24;
 782         if ((fc & (WLAN_FC_TODS | WLAN_FC_FROMDS)) ==
 783             (WLAN_FC_TODS | WLAN_FC_FROMDS))
 784                 hdrlen += ETH_ALEN;
 785         if (stype & 0x08) {
 786                 qos = data + hdrlen;
 787                 hdrlen += 2;
 788         }
 789         if (len < hdrlen)
 790                 return;
 791         wt->rx_data++;
 792
 793         switch (fc & (WLAN_FC_TODS | WLAN_FC_FROMDS)) {
 794         case 0:
 795                 wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s IBSS DA=" MACSTR " SA="
 796                            MACSTR " BSSID=" MACSTR,
 797                            data_stype(WLAN_FC_GET_STYPE(fc)),
 798                            fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
 799                            fc & WLAN_FC_ISWEP ? " Prot" : "",
 800                            MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
 801                            MAC2STR(hdr->addr3));
 802                 add_direct_link(wt, hdr->addr3, hdr->addr1, hdr->addr2);
 803                 rx_data_bss(wt, hdr, hdrlen, qos, hdr->addr1, hdr->addr2,
 804                             data + hdrlen, len - hdrlen);
 805                 break;
 806         case WLAN_FC_FROMDS:
 807                 wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s FromDS DA=" MACSTR
 808                            " BSSID=" MACSTR " SA=" MACSTR,
 809                            data_stype(WLAN_FC_GET_STYPE(fc)),
 810                            fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
 811                            fc & WLAN_FC_ISWEP ? " Prot" : "",
 812                            MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
 813                            MAC2STR(hdr->addr3));
 814                 add_ap_path(wt, hdr->addr2, hdr->addr1, hdr->addr3);
 815                 rx_data_bss(wt, hdr, hdrlen, qos, hdr->addr1, hdr->addr3,
 816                             data + hdrlen, len - hdrlen);
 817                 break;
 818         case WLAN_FC_TODS:
 819                 wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s ToDS BSSID=" MACSTR
 820                            " SA=" MACSTR " DA=" MACSTR,
 821                            data_stype(WLAN_FC_GET_STYPE(fc)),
 822                            fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
 823                            fc & WLAN_FC_ISWEP ? " Prot" : "",
 824                            MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
 825                            MAC2STR(hdr->addr3));
 826                 add_ap_path(wt, hdr->addr1, hdr->addr3, hdr->addr2);
 827                 rx_data_bss(wt, hdr, hdrlen, qos, hdr->addr3, hdr->addr2,
 828                             data + hdrlen, len - hdrlen);
 829                 break;
 830         case WLAN_FC_TODS | WLAN_FC_FROMDS:
 831                 wpa_printf(MSG_EXCESSIVE, "DATA %s%s%s WDS RA=" MACSTR " TA="
 832                            MACSTR " DA=" MACSTR " SA=" MACSTR,
 833                            data_stype(WLAN_FC_GET_STYPE(fc)),
 834                            fc & WLAN_FC_PWRMGT ? " PwrMgt" : "",
 835                            fc & WLAN_FC_ISWEP ? " Prot" : "",
 836                            MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
 837                            MAC2STR(hdr->addr3),
 838                            MAC2STR((const u8 *) (hdr + 1)));
 839                 rx_data_bss(wt, hdr, hdrlen, qos, hdr->addr1, hdr->addr2,
 840                             data + hdrlen, len - hdrlen);
 841                 break;
 842         }
 843 }