From f7efdeb5e2159566c77f28a84a3d9cec4b45dbcf Mon Sep 17 00:00:00 2001 From: Jukka Rissanen Date: Wed, 28 Jan 2015 14:14:56 +0200 Subject: [PATCH] unit: Add checking of SSID security settings --- Makefile.am | 6 +- unit/test-ssid-security.c | 148 ++++++++++++++++++++++++++++++++++++++ 2 files changed, 153 insertions(+), 1 deletion(-) create mode 100644 unit/test-ssid-security.c diff --git a/Makefile.am b/Makefile.am index 3661238b..45b15da2 100644 --- a/Makefile.am +++ b/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 diff --git a/unit/test-ssid-security.c b/unit/test-ssid-security.c new file mode 100644 index 00000000..23f9fe69 --- /dev/null +++ b/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 +#endif + +#include +#include +#include +#include + +#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(); +}