X-Git-Url: http://www.project-moonshot.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=src%2Fwps%2Fwps_registrar.c;h=6fed6a5ea5da19f32f77d0b05636a28769d1f5af;hb=3237bfb1a3cdd4d3efa34f6fa205478b38bf5b75;hp=0d8c72e68ef5930291d8f6b311fcf9183f67945b;hpb=70153d385cb025ce1faef7461db67bd256c9b76f;p=libeap.git

diff --git a/src/wps/wps_registrar.c b/src/wps/wps_registrar.c
index 0d8c72e..6fed6a5 100644
--- a/src/wps/wps_registrar.c
+++ b/src/wps/wps_registrar.c
@@ -2426,7 +2426,7 @@ static enum wps_process_res wps_process_m5(struct wps_data *wps,
 		return WPS_CONTINUE;
 	}
 
-	if (wps_validate_m5_encr(decrypted) < 0) {
+	if (wps_validate_m5_encr(decrypted, attr->version2 != NULL) < 0) {
 		wpabuf_free(decrypted);
 		wps->state = SEND_WSC_NACK;
 		return WPS_CONTINUE;
@@ -2555,7 +2555,8 @@ static enum wps_process_res wps_process_m7(struct wps_data *wps,
 		return WPS_CONTINUE;
 	}
 
-	if (wps_validate_m7_encr(decrypted, wps->wps->ap || wps->er) < 0) {
+	if (wps_validate_m7_encr(decrypted, wps->wps->ap || wps->er,
+				 attr->version2 != NULL) < 0) {
 		wpabuf_free(decrypted);
 		wps->state = SEND_WSC_NACK;
 		return WPS_CONTINUE;