wlantest: Decrypt (Re)Association Request frame when EPPKE is used

This extends wlantest support for additional Management frame encryption
cases: association frames with EPPKE (IEEE 802.11bi). When a valid TK is
available, (Re)Association Request/Response frames are now decrypted
before processing.

Signed-off-by: Jouni Malinen <jouni.malinen@oss.qualcomm.com>

diff --git a/wlantest/rx_mgmt.c b/wlantest/rx_mgmt.c
index c79b06c1837e4afcfab78ca29b7cc06d7b320963..e2e038eea5409dc10ee8bbb1fced226ebd34672c 100644 (file)
--- a/wlantest/rx_mgmt.c
+++ b/wlantest/rx_mgmt.c
@@ -3961,6 +3961,10 @@ void rx_mgmt(struct wlantest *wt, const u8 *data, size_t len)
            !(hdr->addr1[0] & 0x01) &&
            (stype == WLAN_FC_STYPE_DEAUTH ||
             stype == WLAN_FC_STYPE_DISASSOC ||
+            stype == WLAN_FC_STYPE_ASSOC_REQ ||
+            stype == WLAN_FC_STYPE_ASSOC_RESP ||
+            stype == WLAN_FC_STYPE_REASSOC_REQ ||
+            stype == WLAN_FC_STYPE_REASSOC_RESP ||
             stype == WLAN_FC_STYPE_ACTION ||
             stype == WLAN_FC_STYPE_ACTION_NO_ACK)) {
                decrypted = mgmt_decrypt(wt, data, len, &dlen);
@@ -3976,6 +3980,10 @@ void rx_mgmt(struct wlantest *wt, const u8 *data, size_t len)
            !(hdr->addr1[0] & 0x01) &&
            (stype == WLAN_FC_STYPE_DEAUTH ||
             stype == WLAN_FC_STYPE_DISASSOC ||
+            stype == WLAN_FC_STYPE_ASSOC_REQ ||
+            stype == WLAN_FC_STYPE_ASSOC_RESP ||
+            stype == WLAN_FC_STYPE_REASSOC_REQ ||
+            stype == WLAN_FC_STYPE_REASSOC_RESP ||
             stype == WLAN_FC_STYPE_ACTION ||
             stype == WLAN_FC_STYPE_ACTION_NO_ACK)) {
                if (check_mgmt_ccmp_gcmp(wt, data, len) < 0)