From 293a01f3b316a38a7c23f1f13b11c7bd6bc772bf Mon Sep 17 00:00:00 2001
From: Jouni Malinen <jouni@codeaurora.org>
Date: Sat, 7 Sep 2019 15:24:24 +0300
Subject: [PATCH] SAE: Handle BSS membership selector indication for H2E-only
 in STA mode

Signed-off-by: Jouni Malinen <jouni@codeaurora.org>
---
 wpa_supplicant/events.c | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/wpa_supplicant/events.c b/wpa_supplicant/events.c
index 1a2fcd26c..523b7bd46 100644
--- a/wpa_supplicant/events.c
+++ b/wpa_supplicant/events.c
@@ -850,6 +850,19 @@ static int rate_match(struct wpa_supplicant *wpa_s, struct wpa_bss *bss,
 				continue;
 			}
 
+#ifdef CONFIG_SAE
+			if (flagged && ((rate_ie[j] & 0x7f) ==
+					BSS_MEMBERSHIP_SELECTOR_SAE_H2E_ONLY)) {
+				if (wpa_s->conf->sae_pwe == 0) {
+					if (debug_print)
+						wpa_dbg(wpa_s, MSG_DEBUG,
+							"   SAE H2E disabled");
+					return 0;
+				}
+				continue;
+			}
+#endif /* CONFIG_SAE */
+
 			if (!flagged)
 				continue;