Home | History | Annotate | Download | only in common
      1 /*
      2  * common module tests
      3  * Copyright (c) 2014, Jouni Malinen <j (at) w1.fi>
      4  *
      5  * This software may be distributed under the terms of the BSD license.
      6  * See README for more details.
      7  */
      8 
      9 #include "utils/includes.h"
     10 
     11 #include "utils/common.h"
     12 #include "ieee802_11_common.h"
     13 #include "wpa_common.h"
     14 
     15 
     16 struct ieee802_11_parse_test_data {
     17 	u8 *data;
     18 	size_t len;
     19 	ParseRes result;
     20 	int count;
     21 };
     22 
     23 static const struct ieee802_11_parse_test_data parse_tests[] = {
     24 	{ (u8 *) "", 0, ParseOK, 0 },
     25 	{ (u8 *) " ", 1, ParseFailed, 0 },
     26 	{ (u8 *) "\xff\x00", 2, ParseUnknown, 1 },
     27 	{ (u8 *) "\xff\x01", 2, ParseFailed, 0 },
     28 	{ (u8 *) "\xdd\x03\x01\x02\x03", 5, ParseUnknown, 1 },
     29 	{ (u8 *) "\xdd\x04\x01\x02\x03\x04", 6, ParseUnknown, 1 },
     30 	{ (u8 *) "\xdd\x04\x00\x50\xf2\x02", 6, ParseUnknown, 1 },
     31 	{ (u8 *) "\xdd\x05\x00\x50\xf2\x02\x02", 7, ParseOK, 1 },
     32 	{ (u8 *) "\xdd\x05\x00\x50\xf2\x02\xff", 7, ParseUnknown, 1 },
     33 	{ (u8 *) "\xdd\x04\x00\x50\xf2\xff", 6, ParseUnknown, 1 },
     34 	{ (u8 *) "\xdd\x04\x50\x6f\x9a\xff", 6, ParseUnknown, 1 },
     35 	{ (u8 *) "\xdd\x04\x00\x90\x4c\x33", 6, ParseOK, 1 },
     36 	{ (u8 *) "\xdd\x04\x00\x90\x4c\xff\xdd\x04\x00\x90\x4c\x33", 12,
     37 	  ParseUnknown, 2 },
     38 	{ (u8 *) "\x10\x01\x00\x21\x00", 5, ParseOK, 2 },
     39 	{ (u8 *) "\x24\x00", 2, ParseOK, 1 },
     40 	{ (u8 *) "\x38\x00", 2, ParseOK, 1 },
     41 	{ (u8 *) "\x54\x00", 2, ParseOK, 1 },
     42 	{ (u8 *) "\x5a\x00", 2, ParseOK, 1 },
     43 	{ (u8 *) "\x65\x00", 2, ParseOK, 1 },
     44 	{ (u8 *) "\x65\x12\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f\x10\x11",
     45 	  20, ParseOK, 1 },
     46 	{ (u8 *) "\x6e\x00", 2, ParseOK, 1 },
     47 	{ (u8 *) "\xc7\x00", 2, ParseOK, 1 },
     48 	{ (u8 *) "\xc7\x01\x00", 3, ParseOK, 1 },
     49 	{ NULL, 0, ParseOK, 0 }
     50 };
     51 
     52 static int ieee802_11_parse_tests(void)
     53 {
     54 	int i, ret = 0;
     55 
     56 	wpa_printf(MSG_INFO, "ieee802_11_parse tests");
     57 
     58 	for (i = 0; parse_tests[i].data; i++) {
     59 		const struct ieee802_11_parse_test_data *test;
     60 		struct ieee802_11_elems elems;
     61 		ParseRes res;
     62 
     63 		test = &parse_tests[i];
     64 		res = ieee802_11_parse_elems(test->data, test->len, &elems, 1);
     65 		if (res != test->result ||
     66 		    ieee802_11_ie_count(test->data, test->len) != test->count) {
     67 			wpa_printf(MSG_ERROR, "ieee802_11_parse test %d failed",
     68 				   i);
     69 			ret = -1;
     70 		}
     71 	}
     72 
     73 	if (ieee802_11_vendor_ie_concat((const u8 *) "\x00\x01", 2, 0) != NULL)
     74 	{
     75 		wpa_printf(MSG_ERROR,
     76 			   "ieee802_11_vendor_ie_concat test failed");
     77 		ret = -1;
     78 	}
     79 
     80 	return ret;
     81 }
     82 
     83 
     84 struct rsn_ie_parse_test_data {
     85 	u8 *data;
     86 	size_t len;
     87 	int result;
     88 };
     89 
     90 static const struct rsn_ie_parse_test_data rsn_parse_tests[] = {
     91 	{ (u8 *) "", 0, -1 },
     92 	{ (u8 *) "\x30\x00", 2, -1 },
     93 	{ (u8 *) "\x30\x02\x01\x00", 4, 0 },
     94 	{ (u8 *) "\x30\x02\x00\x00", 4, -2 },
     95 	{ (u8 *) "\x30\x02\x02\x00", 4, -2 },
     96 	{ (u8 *) "\x30\x02\x00\x01", 4, -2 },
     97 	{ (u8 *) "\x30\x02\x00\x00\x00", 5, -2 },
     98 	{ (u8 *) "\x30\x03\x01\x00\x00", 5, -3 },
     99 	{ (u8 *) "\x30\x06\x01\x00\x00\x00\x00\x00", 8, -1 },
    100 	{ (u8 *) "\x30\x06\x01\x00\x00\x0f\xac\x04", 8, 0 },
    101 	{ (u8 *) "\x30\x07\x01\x00\x00\x0f\xac\x04\x00", 9, -5 },
    102 	{ (u8 *) "\x30\x08\x01\x00\x00\x0f\xac\x04\x00\x00", 10, -4 },
    103 	{ (u8 *) "\x30\x08\x01\x00\x00\x0f\xac\x04\x00\x01", 10, -4 },
    104 	{ (u8 *) "\x30\x0c\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04",
    105 	  14, 0 },
    106 	{ (u8 *) "\x30\x0c\x01\x00\x00\x0f\xac\x04\x00\x01\x00\x0f\xac\x04",
    107 	  14, -4 },
    108 	{ (u8 *) "\x30\x0c\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x06",
    109 	  14, -1 },
    110 	{ (u8 *) "\x30\x10\x01\x00\x00\x0f\xac\x04\x02\x00\x00\x0f\xac\x04\x00\x0f\xac\x08",
    111 	  18, 0 },
    112 	{ (u8 *) "\x30\x0d\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x00",
    113 	  15, -7 },
    114 	{ (u8 *) "\x30\x0e\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x00\x00",
    115 	  16, -6 },
    116 	{ (u8 *) "\x30\x0e\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x00\x01",
    117 	  16, -6 },
    118 	{ (u8 *) "\x30\x12\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x01",
    119 	  20, 0 },
    120 	{ (u8 *) "\x30\x16\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x02\x00\x00\x0f\xac\x01\x00\x0f\xac\x02",
    121 	  24, 0 },
    122 	{ (u8 *) "\x30\x13\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x01\x00",
    123 	  21, 0 },
    124 	{ (u8 *) "\x30\x14\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x01\x00\x00",
    125 	  22, 0 },
    126 	{ (u8 *) "\x30\x16\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x01\x00\x00\x00\x00",
    127 	  24, 0 },
    128 	{ (u8 *) "\x30\x16\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x01\x00\x00\x00\x01",
    129 	  24, -9 },
    130 	{ (u8 *) "\x30\x1a\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x01\x00\x00\x00\x00\x00\x00\x00\x00",
    131 	  28, -10 },
    132 	{ (u8 *) "\x30\x1a\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x01\x00\x00\x00\x00\x00\x0f\xac\x06",
    133 	  28, 0 },
    134 	{ (u8 *) "\x30\x1c\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x01\x00\x00\x00\x00\x00\x0f\xac\x06\x01\x02",
    135 	  30, 0 },
    136 	{ NULL, 0, 0 }
    137 };
    138 
    139 static int rsn_ie_parse_tests(void)
    140 {
    141 	int i, ret = 0;
    142 
    143 	wpa_printf(MSG_INFO, "rsn_ie_parse tests");
    144 
    145 	for (i = 0; rsn_parse_tests[i].data; i++) {
    146 		const struct rsn_ie_parse_test_data *test;
    147 		struct wpa_ie_data data;
    148 
    149 		test = &rsn_parse_tests[i];
    150 		if (wpa_parse_wpa_ie_rsn(test->data, test->len, &data) !=
    151 		    test->result) {
    152 			wpa_printf(MSG_ERROR, "rsn_ie_parse test %d failed", i);
    153 			ret = -1;
    154 		}
    155 	}
    156 
    157 	return ret;
    158 }
    159 
    160 
    161 int common_module_tests(void)
    162 {
    163 	int ret = 0;
    164 
    165 	wpa_printf(MSG_INFO, "common module tests");
    166 
    167 	if (ieee802_11_parse_tests() < 0 ||
    168 	    rsn_ie_parse_tests() < 0)
    169 		ret = -1;
    170 
    171 	return ret;
    172 }
    173