From 5047c3af6d0054bbeafec5e1e3cde1d2c9c39982 Mon Sep 17 00:00:00 2001
From: Zhang Jun Hao <zhangjunhao@espressif.com>
Date: Fri, 21 May 2021 15:18:46 +0800
Subject: [PATCH] fix(wifi): fragment and forge vulnerability detection

---
 components/wpa_supplicant/src/common/wpa_common.h | 2 ++
 components/wpa_supplicant/src/rsn_supp/wpa_ie.c   | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/components/wpa_supplicant/src/common/wpa_common.h b/components/wpa_supplicant/src/common/wpa_common.h
index f88e8a6f..fd1d9995 100644
--- a/components/wpa_supplicant/src/common/wpa_common.h
+++ b/components/wpa_supplicant/src/common/wpa_common.h
@@ -110,6 +110,8 @@
 #define WPA_CAPABILITY_MFPR BIT(6)
 #define WPA_CAPABILITY_MFPC BIT(7)
 #define WPA_CAPABILITY_PEERKEY_ENABLED BIT(9)
+#define WPA_CAPABILITY_SPP_CAPABLE BIT(10)
+#define WPA_CAPABILITY_SPP_REQUIRED BIT(11)
 
 
 /* IEEE 802.11r */
diff --git a/components/wpa_supplicant/src/rsn_supp/wpa_ie.c b/components/wpa_supplicant/src/rsn_supp/wpa_ie.c
index d648a454..5ad24bb9 100644
--- a/components/wpa_supplicant/src/rsn_supp/wpa_ie.c
+++ b/components/wpa_supplicant/src/rsn_supp/wpa_ie.c
@@ -225,6 +225,9 @@ static int  wpa_gen_wpa_ie_rsn(u8 *rsn_ie, size_t rsn_ie_len,
         }
     }
 #endif /* CONFIG_IEEE80211W */
+
+    capab |= WPA_CAPABILITY_SPP_CAPABLE;
+
     WPA_PUT_LE16(pos, capab);
     pos += 2;