From 6a4477236ebe1ebc808415c8bbd1f8a3ab0f6777 Mon Sep 17 00:00:00 2001
From: Jouni Malinen <jouni.malinen@atheros.com>
Date: Wed, 21 Jul 2010 19:13:31 -0700
Subject: [PATCH] WPS 2.0: Only reject Probe Request frames from WPS 2.0
 devices

---
 src/wps/wps_validate.c | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/wps/wps_validate.c b/src/wps/wps_validate.c
index 755f1d4f4..55d3677e9 100644
--- a/src/wps/wps_validate.c
+++ b/src/wps/wps_validate.c
@@ -1167,7 +1167,12 @@ int wps_validate_beacon_probe_resp(const struct wpabuf *wps_ie, int probe)
 					 attr.authorized_macs_len, 0)) {
 		wpa_printf(MSG_INFO, "WPS-STRICT: Invalid %sProbe Response "
 			   "frame", probe ? "" : "Beacon/");
+#ifdef WPS_STRICT_WPS2
+		if (wps2)
+			return -1;
+#else /* WPS_STRICT_WPS2 */
 		return -1;
+#endif /* WPS_STRICT_WPS2 */
 	}
 
 	return 0;