common_module_tests.c revision 717574375e969e8272c6d1a26137286eac158abb
1/*
2 * common module tests
3 * Copyright (c) 2014, Jouni Malinen <j@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
16struct ieee802_11_parse_test_data {
17	u8 *data;
18	size_t len;
19	ParseRes result;
20	int count;
21};
22
23static 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
52static 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
84struct rsn_ie_parse_test_data {
85	u8 *data;
86	size_t len;
87	int result;
88};
89
90static 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
139static 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
161int 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