From 00e4fbdcc5fdb37f4a08e6495c78787cf19601d4 Mon Sep 17 00:00:00 2001
From: Jouni Malinen <jouni@codeaurora.org>
Date: Sat, 30 May 2020 23:30:42 +0300
Subject: [PATCH] tests: Module test for SAE-PK

Signed-off-by: Jouni Malinen <jouni@codeaurora.org>
---
 src/common/common_module_tests.c | 107 +++++++++++++++++++++++++++++++
 1 file changed, 107 insertions(+)

diff --git a/src/common/common_module_tests.c b/src/common/common_module_tests.c
index a58bf666a..77d6001c8 100644
--- a/src/common/common_module_tests.c
+++ b/src/common/common_module_tests.c
@@ -548,6 +548,112 @@ fail:
 }
 
 
+static int sae_pk_tests(void)
+{
+#ifdef CONFIG_SAE_PK
+	const char *invalid[] = { "a2bc-de3f-ghi4-", "a2bcde3fghi4", "", NULL };
+	struct {
+		const char *pw;
+		const u8 *val;
+	} valid[] = {
+		{ "a2bc-de3f-ghi4", (u8 *) "\x06\x82\x21\x93\x65\x31\xd1\xc0" },
+		{ "ci2f-m6e2", (u8 *) "\x12\x34\x56\x78\x9a" },
+		{ "aaaa-aaaa-aaaa-a",
+		  (u8 *) "\x00\x00\x00\x00\x00\x00\x00\x00\x00" },
+		{ "aaaa-aaaa-aaaa", (u8 *) "\x00\x00\x00\x00\x00\x00\x00\x00" },
+		{ "aaaa-aaaa-aa", (u8 *) "\x00\x00\x00\x00\x00\x00\x00" },
+		{ "aaaa-aaaa", (u8 *) "\x00\x00\x00\x00\x00" },
+		{ "aaaa-aaa", (u8 *) "\x00\x00\x00\x00\x00" },
+		{ "aaaa-a", (u8 *) "\x00\x00\x00\x00" },
+		{ "aeaa-a", (u8 *) "\x01\x00\x00\x00" },
+		{ "7777-7", (u8 *) "\xff\xff\xff\x80" },
+		{ "7777-77", (u8 *) "\xff\xff\xff\xfc" },
+		{ "7777-777", (u8 *) "\xff\xff\xff\xff\xe0" },
+		{ "7777-7777", (u8 *) "\xff\xff\xff\xff\xff" },
+		{ "7777-7777-7", (u8 *) "\xff\xff\xff\xff\xff\xf8" },
+		{ "7777-7777-77", (u8 *) "\xff\xff\xff\xff\xff\xff\xc0" },
+		{ "7777-7777-777", (u8 *) "\xff\xff\xff\xff\xff\xff\xfe" },
+		{ "7777-7777-7777", (u8 *) "\xff\xff\xff\xff\xff\xff\xff\xf0" },
+		{ "7777-7777-7777-7",
+		  (u8 *) "\xff\xff\xff\xff\xff\xff\xff\xff\x80" },
+		{ NULL, NULL }
+	};
+	int i;
+	bool failed;
+
+	for (i = 0; invalid[i]; i++) {
+		if (sae_pk_valid_password(invalid[i])) {
+			wpa_printf(MSG_ERROR,
+				   "SAE-PK: Invalid password '%s' not recognized",
+				   invalid[i]);
+			return -1;
+		}
+	}
+
+	failed = false;
+	for (i = 0; valid[i].pw; i++) {
+		u8 *res;
+		size_t res_len;
+		char *b32;
+		const char *pw = valid[i].pw;
+		const u8 *val = valid[i].val;
+		size_t pw_len = os_strlen(pw);
+		size_t bits = (pw_len - pw_len / 5) * 5;
+		size_t bytes = (bits + 7) / 8;
+
+		if (!sae_pk_valid_password(pw)) {
+			wpa_printf(MSG_ERROR,
+				   "SAE-PK: Valid password '%s' not recognized",
+				   pw);
+			failed = true;
+			continue;
+		}
+
+		res = sae_pk_base32_decode(pw, pw_len, &res_len);
+		if (!res) {
+			wpa_printf(MSG_ERROR,
+				   "SAE-PK: Failed to decode password '%s'",
+				   valid[i].pw);
+			failed = true;
+			continue;
+		}
+		if (res_len != bytes || os_memcmp(val, res, res_len) != 0) {
+			wpa_printf(MSG_ERROR,
+				   "SAE-PK: Mismatch for decoded password '%s'",
+				   valid[i].pw);
+			wpa_hexdump(MSG_INFO, "SAE-PK: Decoded value",
+				    res, res_len);
+			wpa_hexdump(MSG_INFO, "SAE-PK: Expected value",
+				    val, bytes);
+			failed = true;
+		}
+		os_free(res);
+
+		b32 = sae_pk_base32_encode(val, bits);
+		if (!b32) {
+			wpa_printf(MSG_ERROR,
+				   "SAE-PK: Failed to encode password '%s'",
+				   pw);
+			failed = true;
+			continue;
+		}
+		if (os_strcmp(b32, pw) != 0) {
+			wpa_printf(MSG_ERROR,
+				   "SAE-PK: Mismatch for password '%s'", pw);
+			wpa_printf(MSG_INFO, "SAE-PK: Encoded value: '%s'",
+				   b32);
+			failed = true;
+		}
+		os_free(b32);
+	}
+
+	return failed ? -1 : 0;
+#else /* CONFIG_SAE_PK */
+	return 0;
+#endif /* CONFIG_SAE_PK */
+}
+
+
 int common_module_tests(void)
 {
 	int ret = 0;
@@ -557,6 +663,7 @@ int common_module_tests(void)
 	if (ieee802_11_parse_tests() < 0 ||
 	    gas_tests() < 0 ||
 	    sae_tests() < 0 ||
+	    sae_pk_tests() < 0 ||
 	    rsn_ie_parse_tests() < 0)
 		ret = -1;