tests: FTM capability indication
[mech_eap.git] / wlantest / rx_mgmt.c
index 7630dcb..6242bc6 100644 (file)
@@ -1,20 +1,15 @@
 /*
  * Received Management frame processing
- * Copyright (c) 2010, Jouni Malinen <j@w1.fi>
+ * Copyright (c) 2010-2015, Jouni Malinen <j@w1.fi>
  *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License version 2 as
- * published by the Free Software Foundation.
- *
- * Alternatively, this software may be distributed under the terms of BSD
- * license.
- *
- * See README and COPYING for more details.
+ * This software may be distributed under the terms of the BSD license.
+ * See README for more details.
  */
 
 #include "utils/includes.h"
 
 #include "utils/common.h"
+#include "common/defs.h"
 #include "common/ieee802_11_defs.h"
 #include "common/ieee802_11_common.h"
 #include "crypto/aes_wrap.h"
@@ -58,21 +53,24 @@ static void rx_mgmt_beacon(struct wlantest *wt, const u8 *data, size_t len)
        const struct ieee80211_mgmt *mgmt;
        struct wlantest_bss *bss;
        struct ieee802_11_elems elems;
+       size_t offset;
 
        mgmt = (const struct ieee80211_mgmt *) data;
+       offset = mgmt->u.beacon.variable - data;
+       if (len < offset)
+               return;
        bss = bss_get(wt, mgmt->bssid);
        if (bss == NULL)
                return;
        if (bss->proberesp_seen)
                return; /* do not override with Beacon data */
        bss->capab_info = le_to_host16(mgmt->u.beacon.capab_info);
-       if (ieee802_11_parse_elems(mgmt->u.beacon.variable,
-                                  len - (mgmt->u.beacon.variable - data),
+       if (ieee802_11_parse_elems(mgmt->u.beacon.variable, len - offset,
                                   &elems, 0) == ParseFailed) {
                if (bss->parse_error_reported)
                        return;
-               wpa_printf(MSG_INFO, "Invalid IEs in a Beacon frame from "
-                          MACSTR, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Invalid IEs in a Beacon frame from "
+                        MACSTR, MAC2STR(mgmt->sa));
                bss->parse_error_reported = 1;
                return;
        }
@@ -86,20 +84,24 @@ static void rx_mgmt_probe_resp(struct wlantest *wt, const u8 *data, size_t len)
        const struct ieee80211_mgmt *mgmt;
        struct wlantest_bss *bss;
        struct ieee802_11_elems elems;
+       size_t offset;
 
        mgmt = (const struct ieee80211_mgmt *) data;
+       offset = mgmt->u.probe_resp.variable - data;
+       if (len < offset)
+               return;
        bss = bss_get(wt, mgmt->bssid);
        if (bss == NULL)
                return;
 
+       bss->counters[WLANTEST_BSS_COUNTER_PROBE_RESPONSE]++;
        bss->capab_info = le_to_host16(mgmt->u.probe_resp.capab_info);
-       if (ieee802_11_parse_elems(mgmt->u.probe_resp.variable,
-                                  len - (mgmt->u.probe_resp.variable - data),
+       if (ieee802_11_parse_elems(mgmt->u.probe_resp.variable, len - offset,
                                   &elems, 0) == ParseFailed) {
                if (bss->parse_error_reported)
                        return;
-               wpa_printf(MSG_INFO, "Invalid IEs in a Probe Response frame "
-                          "from " MACSTR, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Invalid IEs in a Probe Response frame "
+                        "from " MACSTR, MAC2STR(mgmt->sa));
                bss->parse_error_reported = 1;
                return;
        }
@@ -127,8 +129,8 @@ static void rx_mgmt_auth(struct wlantest *wt, const u8 *data, size_t len)
                return;
 
        if (len < 24 + 6) {
-               wpa_printf(MSG_INFO, "Too short Authentication frame from "
-                          MACSTR, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Too short Authentication frame from "
+                        MACSTR, MAC2STR(mgmt->sa));
                return;
        }
 
@@ -142,9 +144,9 @@ static void rx_mgmt_auth(struct wlantest *wt, const u8 *data, size_t len)
 
        if (alg == 0 && trans == 2 && status == 0) {
                if (sta->state == STATE1) {
-                       wpa_printf(MSG_DEBUG, "STA " MACSTR
-                                  " moved to State 2 with " MACSTR,
-                                  MAC2STR(sta->addr), MAC2STR(bss->bssid));
+                       add_note(wt, MSG_DEBUG, "STA " MACSTR
+                                " moved to State 2 with " MACSTR,
+                                MAC2STR(sta->addr), MAC2STR(bss->bssid));
                        sta->state = STATE2;
                }
        }
@@ -156,26 +158,42 @@ static void rx_mgmt_auth(struct wlantest *wt, const u8 *data, size_t len)
 }
 
 
-static void deauth_all_stas(struct wlantest_bss *bss)
+static void deauth_all_stas(struct wlantest *wt, struct wlantest_bss *bss)
 {
        struct wlantest_sta *sta;
        dl_list_for_each(sta, &bss->sta, struct wlantest_sta, list) {
                if (sta->state == STATE1)
                        continue;
-               wpa_printf(MSG_DEBUG, "STA " MACSTR
-                          " moved to State 1 with " MACSTR,
-                          MAC2STR(sta->addr), MAC2STR(bss->bssid));
+               add_note(wt, MSG_DEBUG, "STA " MACSTR
+                        " moved to State 1 with " MACSTR,
+                        MAC2STR(sta->addr), MAC2STR(bss->bssid));
                sta->state = STATE1;
        }
 }
 
 
+static void tdls_link_down(struct wlantest *wt, struct wlantest_bss *bss,
+                          struct wlantest_sta *sta)
+{
+       struct wlantest_tdls *tdls;
+       dl_list_for_each(tdls, &bss->tdls, struct wlantest_tdls, list) {
+               if ((tdls->init == sta || tdls->resp == sta) && tdls->link_up)
+               {
+                       add_note(wt, MSG_DEBUG, "TDLS: Set link down based on "
+                                "STA deauth/disassoc");
+                       tdls->link_up = 0;
+               }
+       }
+}
+
+
 static void rx_mgmt_deauth(struct wlantest *wt, const u8 *data, size_t len,
                           int valid)
 {
        const struct ieee80211_mgmt *mgmt;
        struct wlantest_bss *bss;
        struct wlantest_sta *sta;
+       u16 fc, reason;
 
        mgmt = (const struct ieee80211_mgmt *) data;
        bss = bss_get(wt, mgmt->bssid);
@@ -187,43 +205,55 @@ static void rx_mgmt_deauth(struct wlantest *wt, const u8 *data, size_t len,
                sta = sta_get(bss, mgmt->sa);
 
        if (len < 24 + 2) {
-               wpa_printf(MSG_INFO, "Too short Deauthentication frame from "
-                          MACSTR, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Too short Deauthentication frame from "
+                        MACSTR, MAC2STR(mgmt->sa));
                return;
        }
 
+       reason = le_to_host16(mgmt->u.deauth.reason_code);
        wpa_printf(MSG_DEBUG, "DEAUTH " MACSTR " -> " MACSTR
                   " (reason=%u) (valid=%d)",
                   MAC2STR(mgmt->sa), MAC2STR(mgmt->da),
-                  le_to_host16(mgmt->u.deauth.reason_code), valid);
+                  reason, valid);
        wpa_hexdump(MSG_MSGDUMP, "DEAUTH payload", data + 24, len - 24);
 
        if (sta == NULL) {
                if (valid && mgmt->da[0] == 0xff)
-                       deauth_all_stas(bss);
+                       deauth_all_stas(wt, bss);
                return;
        }
 
-       if (os_memcmp(mgmt->sa, mgmt->bssid, ETH_ALEN) == 0)
+       if (os_memcmp(mgmt->sa, mgmt->bssid, ETH_ALEN) == 0) {
                sta->counters[valid ? WLANTEST_STA_COUNTER_VALID_DEAUTH_RX :
                              WLANTEST_STA_COUNTER_INVALID_DEAUTH_RX]++;
-       else
+               if (sta->pwrmgt && !sta->pspoll)
+                       sta->counters[WLANTEST_STA_COUNTER_DEAUTH_RX_ASLEEP]++;
+               else
+                       sta->counters[WLANTEST_STA_COUNTER_DEAUTH_RX_AWAKE]++;
+
+               fc = le_to_host16(mgmt->frame_control);
+               if (!(fc & WLAN_FC_ISWEP) && reason == 6)
+                       sta->counters[WLANTEST_STA_COUNTER_DEAUTH_RX_RC6]++;
+               else if (!(fc & WLAN_FC_ISWEP) && reason == 7)
+                       sta->counters[WLANTEST_STA_COUNTER_DEAUTH_RX_RC7]++;
+       } else
                sta->counters[valid ? WLANTEST_STA_COUNTER_VALID_DEAUTH_TX :
                              WLANTEST_STA_COUNTER_INVALID_DEAUTH_TX]++;
 
        if (!valid) {
-               wpa_printf(MSG_INFO, "Do not change STA " MACSTR " State "
-                          "since Disassociation frame was not protected "
-                          "correctly", MAC2STR(sta->addr));
+               add_note(wt, MSG_INFO, "Do not change STA " MACSTR " State "
+                        "since Disassociation frame was not protected "
+                        "correctly", MAC2STR(sta->addr));
                return;
        }
 
        if (sta->state != STATE1) {
-               wpa_printf(MSG_DEBUG, "STA " MACSTR
-                          " moved to State 1 with " MACSTR,
-                          MAC2STR(sta->addr), MAC2STR(bss->bssid));
+               add_note(wt, MSG_DEBUG, "STA " MACSTR
+                        " moved to State 1 with " MACSTR,
+                        MAC2STR(sta->addr), MAC2STR(bss->bssid));
                sta->state = STATE1;
        }
+       tdls_link_down(wt, bss, sta);
 }
 
 
@@ -243,8 +273,8 @@ static void rx_mgmt_assoc_req(struct wlantest *wt, const u8 *data, size_t len)
                return;
 
        if (len < 24 + 4) {
-               wpa_printf(MSG_INFO, "Too short Association Request frame "
-                          "from " MACSTR, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Too short Association Request frame "
+                        "from " MACSTR, MAC2STR(mgmt->sa));
                return;
        }
 
@@ -259,8 +289,8 @@ static void rx_mgmt_assoc_req(struct wlantest *wt, const u8 *data, size_t len)
        if (ieee802_11_parse_elems(mgmt->u.assoc_req.variable,
                                   len - (mgmt->u.assoc_req.variable - data),
                                   &elems, 0) == ParseFailed) {
-               wpa_printf(MSG_INFO, "Invalid IEs in Association Request "
-                          "frame from " MACSTR, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Invalid IEs in Association Request "
+                        "frame from " MACSTR, MAC2STR(mgmt->sa));
                return;
        }
 
@@ -284,6 +314,9 @@ static void rx_mgmt_assoc_resp(struct wlantest *wt, const u8 *data, size_t len)
        struct wlantest_bss *bss;
        struct wlantest_sta *sta;
        u16 capab, status, aid;
+       const u8 *ies;
+       size_t ies_len;
+       struct wpa_ft_ies parse;
 
        mgmt = (const struct ieee80211_mgmt *) data;
        bss = bss_get(wt, mgmt->bssid);
@@ -294,11 +327,14 @@ static void rx_mgmt_assoc_resp(struct wlantest *wt, const u8 *data, size_t len)
                return;
 
        if (len < 24 + 6) {
-               wpa_printf(MSG_INFO, "Too short Association Response frame "
-                          "from " MACSTR, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Too short Association Response frame "
+                        "from " MACSTR, MAC2STR(mgmt->sa));
                return;
        }
 
+       ies = mgmt->u.assoc_resp.variable;
+       ies_len = len - (mgmt->u.assoc_resp.variable - data);
+
        capab = le_to_host16(mgmt->u.assoc_resp.capab_info);
        status = le_to_host16(mgmt->u.assoc_resp.status_code);
        aid = le_to_host16(mgmt->u.assoc_resp.aid);
@@ -310,18 +346,18 @@ static void rx_mgmt_assoc_resp(struct wlantest *wt, const u8 *data, size_t len)
 
        if (status == WLAN_STATUS_ASSOC_REJECTED_TEMPORARILY) {
                struct ieee802_11_elems elems;
-               const u8 *ies = mgmt->u.assoc_resp.variable;
-               size_t ies_len = len - (mgmt->u.assoc_resp.variable - data);
                if (ieee802_11_parse_elems(ies, ies_len, &elems, 0) ==
                    ParseFailed) {
-                       wpa_printf(MSG_INFO, "Failed to parse IEs in "
-                                  "AssocResp from " MACSTR,
-                                  MAC2STR(mgmt->sa));
-               } else if (elems.timeout_int == 0 ||
-                          elems.timeout_int_len != 5) {
-                       wpa_printf(MSG_INFO, "No valid Timeout Interval IE in "
-                                  "AssocResp (status=30) from " MACSTR,
-                                  MAC2STR(mgmt->sa));
+                       add_note(wt, MSG_INFO, "Failed to parse IEs in "
+                                "AssocResp from " MACSTR,
+                                MAC2STR(mgmt->sa));
+               } else if (elems.timeout_int == NULL ||
+                          elems.timeout_int[0] !=
+                          WLAN_TIMEOUT_ASSOC_COMEBACK) {
+                       add_note(wt, MSG_INFO, "No valid Timeout Interval IE "
+                                "with Assoc Comeback time in AssocResp "
+                                "(status=30) from " MACSTR,
+                                MAC2STR(mgmt->sa));
                } else {
                        sta->counters[
                                WLANTEST_STA_COUNTER_ASSOCRESP_COMEBACK]++;
@@ -332,23 +368,34 @@ static void rx_mgmt_assoc_resp(struct wlantest *wt, const u8 *data, size_t len)
                return;
 
        if ((aid & 0xc000) != 0xc000) {
-               wpa_printf(MSG_DEBUG, "Two MSBs of the AID were not set to 1 "
-                          "in Association Response from " MACSTR,
-                          MAC2STR(mgmt->sa));
+               add_note(wt, MSG_DEBUG, "Two MSBs of the AID were not set to 1 "
+                        "in Association Response from " MACSTR,
+                        MAC2STR(mgmt->sa));
        }
        sta->aid = aid & 0xc000;
 
        if (sta->state < STATE2) {
-               wpa_printf(MSG_DEBUG, "STA " MACSTR " was not in State 2 when "
-                          "getting associated", MAC2STR(sta->addr));
+               add_note(wt, MSG_DEBUG,
+                        "STA " MACSTR " was not in State 2 when "
+                        "getting associated", MAC2STR(sta->addr));
        }
 
        if (sta->state < STATE3) {
-               wpa_printf(MSG_DEBUG, "STA " MACSTR
-                          " moved to State 3 with " MACSTR,
-                          MAC2STR(sta->addr), MAC2STR(bss->bssid));
+               add_note(wt, MSG_DEBUG, "STA " MACSTR
+                        " moved to State 3 with " MACSTR,
+                        MAC2STR(sta->addr), MAC2STR(bss->bssid));
                sta->state = STATE3;
        }
+
+       if (wpa_ft_parse_ies(ies, ies_len, &parse) == 0) {
+               if (parse.r0kh_id) {
+                       os_memcpy(bss->r0kh_id, parse.r0kh_id,
+                                 parse.r0kh_id_len);
+                       bss->r0kh_id_len = parse.r0kh_id_len;
+               }
+               if (parse.r1kh_id)
+                       os_memcpy(bss->r1kh_id, parse.r1kh_id, FT_R1KH_ID_LEN);
+       }
 }
 
 
@@ -369,8 +416,8 @@ static void rx_mgmt_reassoc_req(struct wlantest *wt, const u8 *data,
                return;
 
        if (len < 24 + 4 + ETH_ALEN) {
-               wpa_printf(MSG_INFO, "Too short Reassociation Request frame "
-                          "from " MACSTR, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Too short Reassociation Request frame "
+                        "from " MACSTR, MAC2STR(mgmt->sa));
                return;
        }
 
@@ -386,8 +433,8 @@ static void rx_mgmt_reassoc_req(struct wlantest *wt, const u8 *data,
        if (ieee802_11_parse_elems(mgmt->u.reassoc_req.variable,
                                   len - (mgmt->u.reassoc_req.variable - data),
                                   &elems, 0) == ParseFailed) {
-               wpa_printf(MSG_INFO, "Invalid IEs in Reassociation Request "
-                          "frame from " MACSTR, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Invalid IEs in Reassociation Request "
+                        "frame from " MACSTR, MAC2STR(mgmt->sa));
                return;
        }
 
@@ -423,8 +470,8 @@ static void rx_mgmt_reassoc_resp(struct wlantest *wt, const u8 *data,
                return;
 
        if (len < 24 + 6) {
-               wpa_printf(MSG_INFO, "Too short Reassociation Response frame "
-                          "from " MACSTR, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Too short Reassociation Response frame "
+                        "from " MACSTR, MAC2STR(mgmt->sa));
                return;
        }
 
@@ -443,14 +490,16 @@ static void rx_mgmt_reassoc_resp(struct wlantest *wt, const u8 *data,
                size_t ies_len = len - (mgmt->u.reassoc_resp.variable - data);
                if (ieee802_11_parse_elems(ies, ies_len, &elems, 0) ==
                    ParseFailed) {
-                       wpa_printf(MSG_INFO, "Failed to parse IEs in "
-                                  "ReassocResp from " MACSTR,
-                                  MAC2STR(mgmt->sa));
-               } else if (elems.timeout_int == 0 ||
-                          elems.timeout_int_len != 5) {
-                       wpa_printf(MSG_INFO, "No valid Timeout Interval IE in "
-                                  "ReassocResp (status=30) from " MACSTR,
-                                  MAC2STR(mgmt->sa));
+                       add_note(wt, MSG_INFO, "Failed to parse IEs in "
+                                "ReassocResp from " MACSTR,
+                                MAC2STR(mgmt->sa));
+               } else if (elems.timeout_int == NULL ||
+                          elems.timeout_int[0] !=
+                          WLAN_TIMEOUT_ASSOC_COMEBACK) {
+                       add_note(wt, MSG_INFO, "No valid Timeout Interval IE "
+                                "with Assoc Comeback time in ReassocResp "
+                                "(status=30) from " MACSTR,
+                                MAC2STR(mgmt->sa));
                } else {
                        sta->counters[
                                WLANTEST_STA_COUNTER_REASSOCRESP_COMEBACK]++;
@@ -461,35 +510,36 @@ static void rx_mgmt_reassoc_resp(struct wlantest *wt, const u8 *data,
                return;
 
        if ((aid & 0xc000) != 0xc000) {
-               wpa_printf(MSG_DEBUG, "Two MSBs of the AID were not set to 1 "
-                          "in Reassociation Response from " MACSTR,
-                          MAC2STR(mgmt->sa));
+               add_note(wt, MSG_DEBUG, "Two MSBs of the AID were not set to 1 "
+                        "in Reassociation Response from " MACSTR,
+                        MAC2STR(mgmt->sa));
        }
        sta->aid = aid & 0xc000;
 
        if (sta->state < STATE2) {
-               wpa_printf(MSG_DEBUG, "STA " MACSTR " was not in State 2 when "
-                          "getting associated", MAC2STR(sta->addr));
+               add_note(wt, MSG_DEBUG,
+                        "STA " MACSTR " was not in State 2 when "
+                        "getting associated", MAC2STR(sta->addr));
        }
 
        if (sta->state < STATE3) {
-               wpa_printf(MSG_DEBUG, "STA " MACSTR
-                          " moved to State 3 with " MACSTR,
-                          MAC2STR(sta->addr), MAC2STR(bss->bssid));
+               add_note(wt, MSG_DEBUG, "STA " MACSTR
+                        " moved to State 3 with " MACSTR,
+                        MAC2STR(sta->addr), MAC2STR(bss->bssid));
                sta->state = STATE3;
        }
 }
 
 
-static void disassoc_all_stas(struct wlantest_bss *bss)
+static void disassoc_all_stas(struct wlantest *wt, struct wlantest_bss *bss)
 {
        struct wlantest_sta *sta;
        dl_list_for_each(sta, &bss->sta, struct wlantest_sta, list) {
                if (sta->state <= STATE2)
                        continue;
-               wpa_printf(MSG_DEBUG, "STA " MACSTR
-                          " moved to State 2 with " MACSTR,
-                          MAC2STR(sta->addr), MAC2STR(bss->bssid));
+               add_note(wt, MSG_DEBUG, "STA " MACSTR
+                        " moved to State 2 with " MACSTR,
+                        MAC2STR(sta->addr), MAC2STR(bss->bssid));
                sta->state = STATE2;
        }
 }
@@ -501,6 +551,7 @@ static void rx_mgmt_disassoc(struct wlantest *wt, const u8 *data, size_t len,
        const struct ieee80211_mgmt *mgmt;
        struct wlantest_bss *bss;
        struct wlantest_sta *sta;
+       u16 fc, reason;
 
        mgmt = (const struct ieee80211_mgmt *) data;
        bss = bss_get(wt, mgmt->bssid);
@@ -512,48 +563,63 @@ static void rx_mgmt_disassoc(struct wlantest *wt, const u8 *data, size_t len,
                sta = sta_get(bss, mgmt->sa);
 
        if (len < 24 + 2) {
-               wpa_printf(MSG_INFO, "Too short Disassociation frame from "
-                          MACSTR, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Too short Disassociation frame from "
+                        MACSTR, MAC2STR(mgmt->sa));
                return;
        }
 
+       reason = le_to_host16(mgmt->u.disassoc.reason_code);
        wpa_printf(MSG_DEBUG, "DISASSOC " MACSTR " -> " MACSTR
                   " (reason=%u) (valid=%d)",
                   MAC2STR(mgmt->sa), MAC2STR(mgmt->da),
-                  le_to_host16(mgmt->u.disassoc.reason_code), valid);
+                  reason, valid);
        wpa_hexdump(MSG_MSGDUMP, "DISASSOC payload", data + 24, len - 24);
 
        if (sta == NULL) {
                if (valid && mgmt->da[0] == 0xff)
-                       disassoc_all_stas(bss);
+                       disassoc_all_stas(wt, bss);
                return;
        }
 
-       if (os_memcmp(mgmt->sa, mgmt->bssid, ETH_ALEN) == 0)
+       if (os_memcmp(mgmt->sa, mgmt->bssid, ETH_ALEN) == 0) {
                sta->counters[valid ? WLANTEST_STA_COUNTER_VALID_DISASSOC_RX :
                              WLANTEST_STA_COUNTER_INVALID_DISASSOC_RX]++;
-       else
+               if (sta->pwrmgt && !sta->pspoll)
+                       sta->counters[
+                               WLANTEST_STA_COUNTER_DISASSOC_RX_ASLEEP]++;
+               else
+                       sta->counters[
+                               WLANTEST_STA_COUNTER_DISASSOC_RX_AWAKE]++;
+
+               fc = le_to_host16(mgmt->frame_control);
+               if (!(fc & WLAN_FC_ISWEP) && reason == 6)
+                       sta->counters[WLANTEST_STA_COUNTER_DISASSOC_RX_RC6]++;
+               else if (!(fc & WLAN_FC_ISWEP) && reason == 7)
+                       sta->counters[WLANTEST_STA_COUNTER_DISASSOC_RX_RC7]++;
+       } else
                sta->counters[valid ? WLANTEST_STA_COUNTER_VALID_DISASSOC_TX :
                              WLANTEST_STA_COUNTER_INVALID_DISASSOC_TX]++;
 
        if (!valid) {
-               wpa_printf(MSG_INFO, "Do not change STA " MACSTR " State "
-                          "since Disassociation frame was not protected "
-                          "correctly", MAC2STR(sta->addr));
+               add_note(wt, MSG_INFO, "Do not change STA " MACSTR " State "
+                        "since Disassociation frame was not protected "
+                        "correctly", MAC2STR(sta->addr));
                return;
        }
 
        if (sta->state < STATE2) {
-               wpa_printf(MSG_DEBUG, "STA " MACSTR " was not in State 2 or 3 "
-                          "when getting disassociated", MAC2STR(sta->addr));
+               add_note(wt, MSG_DEBUG,
+                        "STA " MACSTR " was not in State 2 or 3 "
+                        "when getting disassociated", MAC2STR(sta->addr));
        }
 
        if (sta->state > STATE2) {
-               wpa_printf(MSG_DEBUG, "STA " MACSTR
-                          " moved to State 2 with " MACSTR,
-                          MAC2STR(sta->addr), MAC2STR(bss->bssid));
+               add_note(wt, MSG_DEBUG, "STA " MACSTR
+                        " moved to State 2 with " MACSTR,
+                        MAC2STR(sta->addr), MAC2STR(bss->bssid));
                sta->state = STATE2;
        }
+       tdls_link_down(wt, bss, sta);
 }
 
 
@@ -570,10 +636,10 @@ static void rx_mgmt_action_sa_query_req(struct wlantest *wt,
                id = sta->ap_sa_query_tr;
        else
                id = sta->sta_sa_query_tr;
-       wpa_printf(MSG_INFO, "SA Query Request " MACSTR " -> " MACSTR
-                  " (trans_id=%02x%02x)%s",
-                  MAC2STR(mgmt->sa), MAC2STR(mgmt->da), rx_id[0], rx_id[1],
-                  valid ? "" : " (invalid protection)");
+       add_note(wt, MSG_INFO, "SA Query Request " MACSTR " -> " MACSTR
+                " (trans_id=%02x%02x)%s",
+                MAC2STR(mgmt->sa), MAC2STR(mgmt->da), rx_id[0], rx_id[1],
+                valid ? "" : " (invalid protection)");
        os_memcpy(id, mgmt->u.action.u.sa_query_req.trans_id, 2);
        if (os_memcmp(mgmt->sa, sta->addr, ETH_ALEN) == 0)
                sta->counters[valid ?
@@ -601,11 +667,11 @@ static void rx_mgmt_action_sa_query_resp(struct wlantest *wt,
        else
                id = sta->ap_sa_query_tr;
        match = os_memcmp(rx_id, id, 2) == 0;
-       wpa_printf(MSG_INFO, "SA Query Response " MACSTR " -> " MACSTR
-                  " (trans_id=%02x%02x; %s)%s",
-                  MAC2STR(mgmt->sa), MAC2STR(mgmt->da), rx_id[0], rx_id[1],
-                  match ? "match" : "mismatch",
-                  valid ? "" : " (invalid protection)");
+       add_note(wt, MSG_INFO, "SA Query Response " MACSTR " -> " MACSTR
+                " (trans_id=%02x%02x; %s)%s",
+                MAC2STR(mgmt->sa), MAC2STR(mgmt->da), rx_id[0], rx_id[1],
+                match ? "match" : "mismatch",
+                valid ? "" : " (invalid protection)");
        if (os_memcmp(mgmt->sa, sta->addr, ETH_ALEN) == 0)
                sta->counters[(valid && match) ?
                              WLANTEST_STA_COUNTER_VALID_SAQUERYRESP_TX :
@@ -623,16 +689,16 @@ static void rx_mgmt_action_sa_query(struct wlantest *wt,
                                    size_t len, int valid)
 {
        if (len < 24 + 2 + WLAN_SA_QUERY_TR_ID_LEN) {
-               wpa_printf(MSG_INFO, "Too short SA Query frame from " MACSTR,
-                          MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Too short SA Query frame from " MACSTR,
+                        MAC2STR(mgmt->sa));
                return;
        }
 
        if (len > 24 + 2 + WLAN_SA_QUERY_TR_ID_LEN) {
                size_t elen = len - (24 + 2 + WLAN_SA_QUERY_TR_ID_LEN);
-               wpa_printf(MSG_INFO, "Unexpected %u octets of extra data at "
-                          "the end of SA Query frame from " MACSTR,
-                          (unsigned) elen, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Unexpected %u octets of extra data at "
+                        "the end of SA Query frame from " MACSTR,
+                        (unsigned) elen, MAC2STR(mgmt->sa));
                wpa_hexdump(MSG_INFO, "SA Query extra data",
                            ((const u8 *) mgmt) + len - elen, elen);
        }
@@ -645,10 +711,10 @@ static void rx_mgmt_action_sa_query(struct wlantest *wt,
                rx_mgmt_action_sa_query_resp(wt, sta, mgmt, len, valid);
                break;
        default:
-               wpa_printf(MSG_INFO, "Unexpected SA Query action value %u "
-                          "from " MACSTR,
-                          mgmt->u.action.u.sa_query_req.action,
-                          MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Unexpected SA Query action value %u "
+                        "from " MACSTR,
+                        mgmt->u.action.u.sa_query_req.action,
+                        MAC2STR(mgmt->sa));
        }
 }
 
@@ -662,11 +728,11 @@ static void rx_mgmt_action(struct wlantest *wt, const u8 *data, size_t len,
 
        mgmt = (const struct ieee80211_mgmt *) data;
        if (mgmt->da[0] & 0x01) {
-               wpa_printf(MSG_DEBUG, "Group addressed Action frame: DA="
-                          MACSTR " SA=" MACSTR " BSSID=" MACSTR
-                          " category=%u",
-                          MAC2STR(mgmt->da), MAC2STR(mgmt->sa),
-                          MAC2STR(mgmt->bssid), mgmt->u.action.category);
+               add_note(wt, MSG_DEBUG, "Group addressed Action frame: DA="
+                        MACSTR " SA=" MACSTR " BSSID=" MACSTR
+                        " category=%u",
+                        MAC2STR(mgmt->da), MAC2STR(mgmt->sa),
+                        MAC2STR(mgmt->bssid), mgmt->u.action.category);
                return; /* Ignore group addressed Action frames for now */
        }
        bss = bss_get(wt, mgmt->bssid);
@@ -680,8 +746,8 @@ static void rx_mgmt_action(struct wlantest *wt, const u8 *data, size_t len,
                return;
 
        if (len < 24 + 1) {
-               wpa_printf(MSG_INFO, "Too short Action frame from "
-                          MACSTR, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Too short Action frame from " MACSTR,
+                        MAC2STR(mgmt->sa));
                return;
        }
 
@@ -693,9 +759,9 @@ static void rx_mgmt_action(struct wlantest *wt, const u8 *data, size_t len,
 
        if (mgmt->u.action.category != WLAN_ACTION_PUBLIC &&
            sta->state < STATE3) {
-               wpa_printf(MSG_INFO, "Action frame sent when STA is not in "
-                          "State 3 (SA=" MACSTR " DATA=" MACSTR ")",
-                          MAC2STR(mgmt->sa), MAC2STR(mgmt->da));
+               add_note(wt, MSG_INFO, "Action frame sent when STA is not in "
+                        "State 3 (SA=" MACSTR " DATA=" MACSTR ")",
+                        MAC2STR(mgmt->sa), MAC2STR(mgmt->da));
        }
 
        switch (mgmt->u.action.category) {
@@ -706,12 +772,22 @@ static void rx_mgmt_action(struct wlantest *wt, const u8 *data, size_t len,
 }
 
 
-static int check_mmie_mic(const u8 *igtk, const u8 *data, size_t len)
+static int check_mmie_mic(unsigned int mgmt_group_cipher,
+                         const u8 *igtk, size_t igtk_len,
+                         const u8 *data, size_t len)
 {
        u8 *buf;
        u8 mic[16];
        u16 fc;
        const struct ieee80211_hdr *hdr;
+       int ret, mic_len;
+
+       if (!mgmt_group_cipher || igtk_len < 16)
+               return -1;
+       mic_len = mgmt_group_cipher == WPA_CIPHER_AES_128_CMAC ? 8 : 16;
+
+       if (len < 24 || len - 24 < mic_len)
+               return -1;
 
        buf = os_malloc(len + 20 - 24);
        if (buf == NULL)
@@ -725,19 +801,45 @@ static int check_mmie_mic(const u8 *igtk, const u8 *data, size_t len)
        os_memcpy(buf + 2, hdr->addr1, 3 * ETH_ALEN);
 
        /* Frame body with MMIE MIC masked to zero */
-       os_memcpy(buf + 20, data + 24, len - 24 - 8);
-       os_memset(buf + 20 + len - 24 - 8, 0, 8);
+       os_memcpy(buf + 20, data + 24, len - 24 - mic_len);
+       os_memset(buf + 20 + len - 24 - mic_len, 0, mic_len);
 
        wpa_hexdump(MSG_MSGDUMP, "BIP: AAD|Body(masked)", buf, len + 20 - 24);
        /* MIC = L(AES-128-CMAC(AAD || Frame Body(masked)), 0, 64) */
-       if (omac1_aes_128(igtk, buf, len + 20 - 24, mic) < 0) {
+       if (mgmt_group_cipher == WPA_CIPHER_AES_128_CMAC) {
+               ret = omac1_aes_128(igtk, buf, len + 20 - 24, mic);
+       } else if (mgmt_group_cipher == WPA_CIPHER_BIP_CMAC_256) {
+               ret = omac1_aes_256(igtk, buf, len + 20 - 24, mic);
+       } else if (mgmt_group_cipher == WPA_CIPHER_BIP_GMAC_128 ||
+                mgmt_group_cipher == WPA_CIPHER_BIP_GMAC_256) {
+               u8 nonce[12], *npos;
+               const u8 *ipn;
+
+               ipn = data + len - mic_len - 6;
+
+               /* Nonce: A2 | IPN */
+               os_memcpy(nonce, hdr->addr2, ETH_ALEN);
+               npos = nonce + ETH_ALEN;
+               *npos++ = ipn[5];
+               *npos++ = ipn[4];
+               *npos++ = ipn[3];
+               *npos++ = ipn[2];
+               *npos++ = ipn[1];
+               *npos++ = ipn[0];
+
+               ret = aes_gmac(igtk, igtk_len, nonce, sizeof(nonce),
+                              buf, len + 20 - 24, mic);
+       } else {
+               ret = -1;
+       }
+       if (ret < 0) {
                os_free(buf);
                return -1;
        }
 
        os_free(buf);
 
-       if (os_memcmp(data + len - 8, mic, 8) != 0)
+       if (os_memcmp(data + len - mic_len, mic, mic_len) != 0)
                return -1;
 
        return 0;
@@ -751,6 +853,7 @@ static int check_bip(struct wlantest *wt, const u8 *data, size_t len)
        const u8 *mmie;
        u16 keyid;
        struct wlantest_bss *bss;
+       size_t mic_len;
 
        mgmt = (const struct ieee80211_mgmt *) data;
        fc = le_to_host16(mgmt->frame_control);
@@ -767,56 +870,60 @@ static int check_bip(struct wlantest *wt, const u8 *data, size_t len)
        if (bss == NULL)
                return 0; /* No key known yet */
 
-       if (len < 24 + 18 || data[len - 18] != WLAN_EID_MMIE ||
-           data[len - 17] != 16) {
+       mic_len = bss->mgmt_group_cipher == WPA_CIPHER_AES_128_CMAC ? 8 : 16;
+
+       if (len < 24 + 10 + mic_len ||
+           data[len - (10 + mic_len)] != WLAN_EID_MMIE ||
+           data[len - (10 + mic_len - 1)] != 8 + mic_len) {
                /* No MMIE */
                if (bss->rsn_capab & WPA_CAPABILITY_MFPC) {
-                       wpa_printf(MSG_INFO, "Robust group-addressed "
-                                  "management frame sent without BIP by "
-                                  MACSTR, MAC2STR(mgmt->sa));
+                       add_note(wt, MSG_INFO, "Robust group-addressed "
+                                "management frame sent without BIP by "
+                                MACSTR, MAC2STR(mgmt->sa));
                        bss->counters[WLANTEST_BSS_COUNTER_MISSING_BIP_MMIE]++;
                        return -1;
                }
                return 0;
        }
 
-       mmie = data + len - 16;
+       mmie = data + len - (8 + mic_len);
        keyid = WPA_GET_LE16(mmie);
        if (keyid & 0xf000) {
-               wpa_printf(MSG_INFO, "MMIE KeyID reserved bits not zero "
-                          "(%04x) from " MACSTR, keyid, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "MMIE KeyID reserved bits not zero "
+                        "(%04x) from " MACSTR, keyid, MAC2STR(mgmt->sa));
                keyid &= 0x0fff;
        }
        if (keyid < 4 || keyid > 5) {
-               wpa_printf(MSG_INFO, "Unexpected MMIE KeyID %u from " MACSTR,
-                          keyid, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Unexpected MMIE KeyID %u from " MACSTR,
+                        keyid, MAC2STR(mgmt->sa));
                bss->counters[WLANTEST_BSS_COUNTER_INVALID_BIP_MMIE]++;
                return 0;
        }
        wpa_printf(MSG_DEBUG, "MMIE KeyID %u", keyid);
        wpa_hexdump(MSG_MSGDUMP, "MMIE IPN", mmie + 2, 6);
-       wpa_hexdump(MSG_MSGDUMP, "MMIE MIC", mmie + 8, 8);
+       wpa_hexdump(MSG_MSGDUMP, "MMIE MIC", mmie + 8, mic_len);
 
-       if (!bss->igtk_set[keyid]) {
-               wpa_printf(MSG_DEBUG, "No IGTK known to validate BIP frame");
+       if (!bss->igtk_len[keyid]) {
+               add_note(wt, MSG_DEBUG, "No IGTK known to validate BIP frame");
                return 0;
        }
 
        if (os_memcmp(mmie + 2, bss->ipn[keyid], 6) <= 0) {
-               wpa_printf(MSG_INFO, "BIP replay detected: SA=" MACSTR,
-                          MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "BIP replay detected: SA=" MACSTR,
+                        MAC2STR(mgmt->sa));
                wpa_hexdump(MSG_INFO, "RX IPN", mmie + 2, 6);
                wpa_hexdump(MSG_INFO, "Last RX IPN", bss->ipn[keyid], 6);
        }
 
-       if (check_mmie_mic(bss->igtk[keyid], data, len) < 0) {
-               wpa_printf(MSG_INFO, "Invalid MMIE MIC in a frame from "
-                          MACSTR, MAC2STR(mgmt->sa));
+       if (check_mmie_mic(bss->mgmt_group_cipher, bss->igtk[keyid],
+                          bss->igtk_len[keyid], data, len) < 0) {
+               add_note(wt, MSG_INFO, "Invalid MMIE MIC in a frame from "
+                        MACSTR, MAC2STR(mgmt->sa));
                bss->counters[WLANTEST_BSS_COUNTER_INVALID_BIP_MMIE]++;
                return -1;
        }
 
-       wpa_printf(MSG_DEBUG, "Valid MMIE MIC");
+       add_note(wt, MSG_DEBUG, "Valid MMIE MIC");
        os_memcpy(bss->ipn[keyid], mmie + 2, 6);
        bss->counters[WLANTEST_BSS_COUNTER_VALID_BIP_MMIE]++;
 
@@ -848,7 +955,7 @@ static u8 * mgmt_ccmp_decrypt(struct wlantest *wt, const u8 *data, size_t len,
        else
                sta = sta_get(bss, hdr->addr1);
        if (sta == NULL || !sta->ptk_set) {
-               wpa_printf(MSG_MSGDUMP, "No PTK known to decrypt the frame");
+               add_note(wt, MSG_MSGDUMP, "No PTK known to decrypt the frame");
                return NULL;
        }
 
@@ -856,22 +963,22 @@ static u8 * mgmt_ccmp_decrypt(struct wlantest *wt, const u8 *data, size_t len,
                return NULL;
 
        if (!(data[24 + 3] & 0x20)) {
-               wpa_printf(MSG_INFO, "Expected CCMP frame from " MACSTR
-                          " did not have ExtIV bit set to 1",
-                          MAC2STR(hdr->addr2));
+               add_note(wt, MSG_INFO, "Expected CCMP frame from " MACSTR
+                        " did not have ExtIV bit set to 1",
+                        MAC2STR(hdr->addr2));
                return NULL;
        }
 
        if (data[24 + 2] != 0 || (data[24 + 3] & 0x1f) != 0) {
-               wpa_printf(MSG_INFO, "CCMP mgmt frame from " MACSTR " used "
-                          "non-zero reserved bit", MAC2STR(hdr->addr2));
+               add_note(wt, MSG_INFO, "CCMP mgmt frame from " MACSTR " used "
+                        "non-zero reserved bit", MAC2STR(hdr->addr2));
        }
 
        keyid = data[24 + 3] >> 6;
        if (keyid != 0) {
-               wpa_printf(MSG_INFO, "Unexpected non-zero KeyID %d in "
-                          "individually addressed Management frame from "
-                          MACSTR, keyid, MAC2STR(hdr->addr2));
+               add_note(wt, MSG_INFO, "Unexpected non-zero KeyID %d in "
+                        "individually addressed Management frame from "
+                        MACSTR, keyid, MAC2STR(hdr->addr2));
        }
 
        if (os_memcmp(hdr->addr1, hdr->addr3, ETH_ALEN) == 0)
@@ -881,13 +988,20 @@ static u8 * mgmt_ccmp_decrypt(struct wlantest *wt, const u8 *data, size_t len,
 
        ccmp_get_pn(pn, data + 24);
        if (os_memcmp(pn, rsc, 6) <= 0) {
-               wpa_printf(MSG_INFO, "CCMP/TKIP replay detected: SA=" MACSTR,
-                          MAC2STR(hdr->addr2));
+               u16 seq_ctrl = le_to_host16(hdr->seq_ctrl);
+               add_note(wt, MSG_INFO, "CCMP/TKIP replay detected: A1=" MACSTR
+                        " A2=" MACSTR " A3=" MACSTR " seq=%u frag=%u%s",
+                        MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
+                        MAC2STR(hdr->addr3),
+                        WLAN_GET_SEQ_SEQ(seq_ctrl),
+                        WLAN_GET_SEQ_FRAG(seq_ctrl),
+                        (le_to_host16(hdr->frame_control) & WLAN_FC_RETRY) ?
+                        " Retry" : "");
                wpa_hexdump(MSG_INFO, "RX PN", pn, 6);
                wpa_hexdump(MSG_INFO, "RSC", rsc, 6);
        }
 
-       decrypted = ccmp_decrypt(sta->ptk.tk1, hdr, data + 24, len - 24, dlen);
+       decrypted = ccmp_decrypt(sta->ptk.tk, hdr, data + 24, len - 24, dlen);
        if (decrypted) {
                os_memcpy(rsc, pn, 6);
                frame = os_malloc(24 + *dlen);
@@ -933,9 +1047,9 @@ static int check_mgmt_ccmp(struct wlantest *wt, const u8 *data, size_t len)
        if ((sta->rsn_capab & WPA_CAPABILITY_MFPC) &&
            (sta->state == STATE3 ||
             WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_ACTION)) {
-               wpa_printf(MSG_INFO, "Robust individually-addressed "
-                          "management frame sent without CCMP by "
-                          MACSTR, MAC2STR(mgmt->sa));
+               add_note(wt, MSG_INFO, "Robust individually-addressed "
+                        "management frame sent without CCMP by "
+                        MACSTR, MAC2STR(mgmt->sa));
                return -1;
        }
 
@@ -1035,4 +1149,88 @@ void rx_mgmt(struct wlantest *wt, const u8 *data, size_t len)
        }
 
        os_free(decrypted);
+
+       wt->last_mgmt_valid = valid;
+}
+
+
+static void rx_mgmt_deauth_ack(struct wlantest *wt,
+                              const struct ieee80211_hdr *hdr)
+{
+       const struct ieee80211_mgmt *mgmt;
+       struct wlantest_bss *bss;
+       struct wlantest_sta *sta;
+
+       mgmt = (const struct ieee80211_mgmt *) hdr;
+       bss = bss_get(wt, mgmt->bssid);
+       if (bss == NULL)
+               return;
+       if (os_memcmp(mgmt->sa, mgmt->bssid, ETH_ALEN) == 0)
+               sta = sta_get(bss, mgmt->da);
+       else
+               sta = sta_get(bss, mgmt->sa);
+       if (sta == NULL)
+               return;
+
+       add_note(wt, MSG_DEBUG, "DEAUTH from " MACSTR " acknowledged by "
+                MACSTR, MAC2STR(mgmt->sa), MAC2STR(mgmt->da));
+       if (os_memcmp(mgmt->sa, mgmt->bssid, ETH_ALEN) == 0) {
+               int c;
+               c = wt->last_mgmt_valid ?
+                       WLANTEST_STA_COUNTER_VALID_DEAUTH_RX_ACK :
+                       WLANTEST_STA_COUNTER_INVALID_DEAUTH_RX_ACK;
+               sta->counters[c]++;
+       }
+}
+
+
+static void rx_mgmt_disassoc_ack(struct wlantest *wt,
+                                const struct ieee80211_hdr *hdr)
+{
+       const struct ieee80211_mgmt *mgmt;
+       struct wlantest_bss *bss;
+       struct wlantest_sta *sta;
+
+       mgmt = (const struct ieee80211_mgmt *) hdr;
+       bss = bss_get(wt, mgmt->bssid);
+       if (bss == NULL)
+               return;
+       if (os_memcmp(mgmt->sa, mgmt->bssid, ETH_ALEN) == 0)
+               sta = sta_get(bss, mgmt->da);
+       else
+               sta = sta_get(bss, mgmt->sa);
+       if (sta == NULL)
+               return;
+
+       add_note(wt, MSG_DEBUG, "DISASSOC from " MACSTR " acknowledged by "
+                MACSTR, MAC2STR(mgmt->sa), MAC2STR(mgmt->da));
+       if (os_memcmp(mgmt->sa, mgmt->bssid, ETH_ALEN) == 0) {
+               int c;
+               c = wt->last_mgmt_valid ?
+                       WLANTEST_STA_COUNTER_VALID_DISASSOC_RX_ACK :
+                       WLANTEST_STA_COUNTER_INVALID_DISASSOC_RX_ACK;
+               sta->counters[c]++;
+       }
+}
+
+
+void rx_mgmt_ack(struct wlantest *wt, const struct ieee80211_hdr *hdr)
+{
+       u16 fc, stype;
+       fc = le_to_host16(hdr->frame_control);
+       stype = WLAN_FC_GET_STYPE(fc);
+
+       wpa_printf(MSG_MSGDUMP, "MGMT ACK: stype=%u a1=" MACSTR " a2=" MACSTR
+                  " a3=" MACSTR,
+                  stype, MAC2STR(hdr->addr1), MAC2STR(hdr->addr2),
+                  MAC2STR(hdr->addr3));
+
+       switch (stype) {
+       case WLAN_FC_STYPE_DEAUTH:
+               rx_mgmt_deauth_ack(wt, hdr);
+               break;
+       case WLAN_FC_STYPE_DISASSOC:
+               rx_mgmt_disassoc_ack(wt, hdr);
+               break;
+       }
 }