unit: Add checking of SSID security settings

Makefile.am

@@ -82,7 +82,7 @@ unit_tests = unit/test-cmac-aes \
 		unit/test-hmac-md5 unit/test-hmac-sha1 unit/test-hmac-sha256 \
 		unit/test-pbkdf2-sha1 unit/test-prf-sha1 \
 		unit/test-crypto unit/test-eapol unit/test-mpdu \
-		unit/test-ie unit/test-ssid-to-utf8
+		unit/test-ie unit/test-ssid-to-utf8 unit/test-ssid-security
 
 if MAINTAINER_MODE
 noinst_PROGRAMS += $(unit_tests)
@@ -135,6 +135,10 @@ unit_test_ssid_to_utf8_SOURCES = src/util.h src/util.c \
 				unit/test-ssid-to-utf8.c
 unit_test_ssid_to_utf8_LDADD = ell/libell-internal.la
 
+unit_test_ssid_security_SOURCES = unit/test-ssid-security.c src/ie.h src/ie.c \
+				src/scan.h src/scan.c
+unit_test_ssid_security_LDADD = ell/libell-internal.la
+
 TESTS = $(unit_tests)
 
 manual_pages = doc/iwmon.1

unit/test-ssid-security.c

@@ -0,0 +1,148 @@
+/*
+ *
+ *  Wireless daemon for Linux
+ *
+ *  Copyright (C) 2015  Intel Corporation. All rights reserved.
+ *
+ *  This library is free software; you can redistribute it and/or
+ *  modify it under the terms of the GNU Lesser General Public
+ *  License as published by the Free Software Foundation; either
+ *  version 2.1 of the License, or (at your option) any later version.
+ *
+ *  This library is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ *  Lesser General Public License for more details.
+ *
+ *  You should have received a copy of the GNU Lesser General Public
+ *  License along with this library; if not, write to the Free Software
+ *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+ *
+ */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <stdio.h>
+#include <string.h>
+#include <assert.h>
+#include <ell/ell.h>
+
+#include "src/ie.h"
+#include "src/scan.h"
+
+struct test_data {
+	unsigned int len;
+	unsigned char *buf;
+	enum ie_bss_capability capability;
+};
+
+
+static unsigned char ssid_security_wpa_data_1[] = {
+	0x30, 0x14, 0x01, 0x00, 0x00, 0x0f, 0xac, 0x04,
+	0x01, 0x00, 0x00, 0x0f, 0xac, 0x04, 0x01, 0x00,
+	0x00, 0x0f, 0xac, 0x02, 0x0c, 0x00,
+};
+
+static struct test_data ssid_security_wpa_test_1 = {
+	.len = sizeof(ssid_security_wpa_data_1),
+	.buf = ssid_security_wpa_data_1,
+	.capability = IE_BSS_CAP_ESS,
+};
+
+static unsigned char ssid_security_wpa2_data_1[] = {
+	0x30, 0x14, 0x01, 0x00, 0x00, 0x0f, 0xac, 0x04,
+	0x01, 0x00, 0x00, 0x0f, 0xac, 0x04, 0x01, 0x00,
+	0x00, 0x0f, 0xac, 0x02, 0x00, 0x00,
+};
+
+static struct test_data ssid_security_wpa_test_2 = {
+	.len = sizeof(ssid_security_wpa2_data_1),
+	.buf = ssid_security_wpa2_data_1,
+	.capability = IE_BSS_CAP_ESS,
+};
+
+static unsigned char ssid_security_8021x_data_1[] = {
+	0x30, 0x14, 0x01, 0x00, 0x00, 0x0f, 0xac, 0x04,
+	0x01, 0x00, 0x00, 0x0f, 0xac, 0x04, 0x01, 0x00,
+	0x00, 0x0f, 0xac, 0x01, 0x28, 0x00,
+};
+
+static struct test_data ssid_security_8021x_test_1 = {
+	.len = sizeof(ssid_security_8021x_data_1),
+	.buf = ssid_security_8021x_data_1,
+	.capability = IE_BSS_CAP_ESS,
+};
+
+static struct test_data ssid_security_wep_test_1 = {
+	.len = 0,
+	.buf = NULL,
+	.capability = IE_BSS_CAP_ESS | IE_BSS_CAP_PRIVACY,
+};
+
+static struct test_data ssid_security_open_test_1 = {
+	.len = 0,
+	.buf = NULL,
+	.capability = IE_BSS_CAP_ESS,
+};
+
+static void ssid_security_open_test(const void *data)
+{
+	const struct test_data *test_data = data;
+
+	assert(scan_get_ssid_security(test_data->capability,
+		(struct ie_rsn_info *)test_data->buf) ==
+						SCAN_SSID_SECURITY_NONE);
+}
+
+static void ssid_security_wep_test(const void *data)
+{
+	const struct test_data *test_data = data;
+
+	assert(scan_get_ssid_security(test_data->capability,
+		(struct ie_rsn_info *)test_data->buf) ==
+						SCAN_SSID_SECURITY_WEP);
+}
+
+static void ssid_security_psk_test(const void *data)
+{
+	const struct test_data *test_data = data;
+	struct ie_rsn_info info;
+	int ret;
+
+	ret = ie_parse_rsne_from_data(test_data->buf, test_data->len, &info);
+	assert(ret == 0);
+	assert(scan_get_ssid_security(test_data->capability,
+					&info) == SCAN_SSID_SECURITY_PSK);
+}
+
+static void ssid_security_8021x_test(const void *data)
+{
+	const struct test_data *test_data = data;
+	struct ie_rsn_info info;
+	int ret;
+
+	ret = ie_parse_rsne_from_data(test_data->buf, test_data->len, &info);
+	assert(ret == 0);
+	assert(scan_get_ssid_security(test_data->capability,
+					&info) == SCAN_SSID_SECURITY_8021X);
+}
+
+int main(int argc, char *argv[])
+{
+	l_test_init(&argc, &argv);
+
+	l_test_add("/SSID Security/Open",
+			ssid_security_open_test, &ssid_security_open_test_1);
+	l_test_add("/SSID Security/WPA",
+			ssid_security_psk_test, &ssid_security_wpa_test_1);
+	l_test_add("/SSID Security/WPA2",
+			ssid_security_psk_test, &ssid_security_wpa_test_2);
+	l_test_add("/SSID Security/8021x",
+			ssid_security_8021x_test, &ssid_security_8021x_test_1);
+	l_test_add("/SSID Security/WEP",
+			ssid_security_wep_test, &ssid_security_wep_test_1);
+
+	return l_test_run();
+}