wps_module_tests.c revision 7f93d6fa14a343127754c33a03cf2f641b6ff6e4 
1/*
2 * WPS 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 "wps_attr_parse.h"
13
14struct wps_attr_parse_test {
15	const char *data;
16	int result;
17	int extra;
18};
19
20struct wps_attr_parse_test wps_attr_parse_test_cases[] = {
21	/* Empty message */
22	{ "", 0, 0 },
23	/* Truncated attribute header */
24	{ "10", -1, 0 },
25	{ "1010", -1, 0 },
26	{ "101000", -1, 0 },
27	/* Attribute overflow */
28	{ "10100001", -1, 0 },
29#ifdef CONFIG_WPS_STRICT
30	{ "10270000001057000101", -1, 0 },
31	{ "1027000010570001010000000000", -1, 0 },
32#else /* CONFIG_WPS_STRICT */
33	/* Network Key workaround */
34	{ "10270000001057000101", 0, 1 },
35	{ "10230000001057000101", -1, 0 },
36	{ "10270000101057000101", -1, 0 },
37	/* Mac OS X 10.6 padding workaround */
38	{ "1027000010570001010000000000", 0, 1 },
39	{ "1027000010570001010000000000000001000000", -1, 0 },
40#endif /* CONFIG_WPS_STRICT */
41	/* Version */
42	{ "104a000110", 0, 0 },
43	{ "104a0000", -1, 0 },
44	/* Message Type */
45	{ "1022000101", 0, 0 },
46	{ "10220000", -1, 0 },
47	/* Enrollee Nonce */
48	{ "101a001000112233445566778899aabbccddeeff", 0, 0 },
49	{ "101a00111122334455667788990011223344556677", -1, 0 },
50	/* Registrar Nonce */
51	{ "1039001000112233445566778899aabbccddeeff", 0, 0 },
52	{ "103900111122334455667788990011223344556677", -1, 0 },
53	/* UUID-E */
54	{ "1047001000112233445566778899aabbccddeeff", 0, 0 },
55	{ "10470000", -1, 0 },
56	{ "104700111122334455667788990011223344556677", -1, 0 },
57	/* UUID-R */
58	{ "1048001000112233445566778899aabbccddeeff", 0, 0 },
59	{ "10480000", -1, 0 },
60	{ "104800111122334455667788990011223344556677", -1, 0 },
61	/* Auth Type Flags */
62	{ "100400021122", 0, 0 },
63	{ "10040001ff", -1, 0 },
64	/* Encr Type Flags */
65	{ "101000021122", 0, 0 },
66	{ "10100001ff", -1, 0 },
67	/* Connection Type Flags */
68	{ "100d0001ff", 0, 0 },
69	{ "100d0002ffff", -1, 0 },
70	/* Config Methods */
71	{ "10080002ffff", 0, 0 },
72	{ "10080001ff", -1, 0 },
73	/* Selected Registrar Config Methods */
74	{ "10530002ffff", 0, 0 },
75	{ "10530001ff", -1, 0 },
76	/* Primary Device Type */
77	{ "105400081122334455667788", 0, 0 },
78	{ "105400111122334455667788990011223344556677", -1, 0 },
79	/* RF Bands */
80	{ "103c0001ff", 0, 0 },
81	{ "103c0002ffff", -1, 0 },
82	/* Association State */
83	{ "10020002ffff", 0, 0 },
84	{ "10020001ff", -1, 0 },
85	/* Config Error */
86	{ "100900020001", 0, 0 },
87	{ "10090001ff", -1, 0 },
88	/* Device Password ID */
89	{ "101200020004", 0, 0 },
90	{ "10120001ff", -1, 0 },
91	/* OOB Device Password */
92	{ "102c001611223344556677889900112233445566778899000007", 0, 0 },
93	{ "102c0036112233445566778899001122334455667788990011223344556677889900112233445566778899001122334455667788990011223344", 0, 0 },
94	{ "102c0001ff", -1, 0 },
95	{ "102c003711223344556677889900112233445566778899001122334455667788990011223344556677889900112233445566778899001122334455", -1, 0 },
96	{ "102c002511223344556677889900112233445566778899001122334455667788990011223344556677", -1, 0 },
97	/* OS Version */
98	{ "102d000411223344", 0, 0 },
99	{ "102d00111122334455667788990011223344556677", -1, 0 },
100	/* WPS State */
101	{ "1044000101", 0, 0 },
102	{ "10440002ffff", -1, 0 },
103	/* Authenticator */
104	{ "100500081122334455667788", 0, 0 },
105	{ "10050000", -1, 0 },
106	{ "100500111122334455667788990011223344556677", -1, 0 },
107	/* R-Hash1 */
108	{ "103d00201122334455667788990011223344556677889900112233445566778899001122", 0, 0 },
109	{ "103d0000", -1, 0 },
110	{ "103d0021112233445566778899001122334455667788990011223344556677889900112233", -1, 0 },
111	/* R-Hash2 */
112	{ "103e00201122334455667788990011223344556677889900112233445566778899001122", 0, 0 },
113	{ "103e0000", -1, 0 },
114	{ "103e0021112233445566778899001122334455667788990011223344556677889900112233", -1, 0 },
115	/* E-Hash1 */
116	{ "101400201122334455667788990011223344556677889900112233445566778899001122", 0, 0 },
117	{ "10140000", -1, 0 },
118	{ "10140021112233445566778899001122334455667788990011223344556677889900112233", -1, 0 },
119	/* E-Hash2 */
120	{ "101500201122334455667788990011223344556677889900112233445566778899001122", 0, 0 },
121	{ "10150000", -1, 0 },
122	{ "10150021112233445566778899001122334455667788990011223344556677889900112233", -1, 0 },
123	/* R-SNonce1 */
124	{ "103f001011223344556677889900112233445566", 0, 0 },
125	{ "103f0000", -1, 0 },
126	{ "103f00111122334455667788990011223344556677", -1, 0 },
127	/* R-SNonce2 */
128	{ "1040001011223344556677889900112233445566", 0, 0 },
129	{ "10400000", -1, 0 },
130	{ "104000111122334455667788990011223344556677", -1, 0 },
131	/* E-SNonce1 */
132	{ "1016001011223344556677889900112233445566", 0, 0 },
133	{ "10160000", -1, 0 },
134	{ "101600111122334455667788990011223344556677", -1, 0 },
135	/* E-SNonce2 */
136	{ "1017001011223344556677889900112233445566", 0, 0 },
137	{ "10170000", -1, 0 },
138	{ "101700111122334455667788990011223344556677", -1, 0 },
139	/* Key Wrap Authenticator */
140	{ "101e00081122334455667788", 0, 0 },
141	{ "101e0000", -1, 0 },
142	{ "101e0009112233445566778899", -1, 0 },
143	/* Authentication Type */
144	{ "100300020001", 0, 0 },
145	{ "10030001ff", -1, 0 },
146	/* Encryption Type */
147	{ "100f00020001", 0, 0 },
148	{ "100f0001ff", -1, 0 },
149	/* Network Index */
150	{ "1026000101", 0, 0 },
151	{ "10260002ffff", -1, 0 },
152	/* Network Key Index */
153	{ "1028000101", 0, 3 },
154	{ "10280002ffff", -1, 0 },
155	/* MAC Address */
156	{ "10200006112233445566", 0, 0 },
157	{ "10200000", -1, 0 },
158	{ "1020000711223344556677", -1, 0 },
159	/* Selected Registrar */
160	{ "1041000101", 0, 0 },
161	{ "10410002ffff", -1, 0 },
162	/* Request Type */
163	{ "103a000101", 0, 0 },
164	{ "103a0002ffff", -1, 0 },
165	/* Response Type */
166	{ "103b000101", 0, 0 },
167	{ "103b0002ffff", -1, 0 },
168	/* Manufacturer */
169	{ "10210000", 0, 0 },
170	/* Model Name */
171	{ "10230000", 0, 0 },
172	/* Model Number */
173	{ "10240000", 0, 0 },
174	/* Serial Number */
175	{ "10420000", 0, 0 },
176	/* Device Name */
177	{ "10110000", 0, 0 },
178	/* Public Key */
179	{ "10320000", 0, 0 },
180	/* Enc Settings */
181	{ "10180000", 0, 0 },
182	/* SSID */
183	{ "10450000", 0, 0 },
184	/* AP Setup Locked */
185	{ "1057000101", 0, 0 },
186	{ "10570002ffff", -1, 0 },
187	/* Requested Device Type */
188	{ "106a00081122334455667788", 0, 0 },
189	{ "106a0000", -1, 0 },
190	{ "106a0009112233445566778899", -1, 0 },
191	/* More than maximum Requested Device Type attributes */
192	{ "106a00081122334455667788106a00081122334455667788106a00081122334455667788106a00081122334455667788106a00081122334455667788106a00081122334455667788106a00081122334455667788106a00081122334455667788106a00081122334455667788106a00081122334455667788106a00081122334455667788106a00081122334455667788", 0, 4 },
193	/* Secondary Device Type List */
194	{ "105500081122334455667788", 0, 0 },
195	{ "1055000711223344556677", -1, 0 },
196	{ "1055008811223344556677889900112233445566778899001122334455667788990011223344556677889900112233445566778899001122334455667788990011223344556677889900112233445566778899001122334455667788990011223344556677889900112233445566778899001122334455667788990011223344556677889900112233445566", -1, 0 },
197	/* AP Channel */
198	{ "100100020001", 0, 0 },
199	{ "1001000101", -1, 0 },
200	/* Skip invalid Vendor Extension */
201	{ "10490000", 0, 0 },
202	{ "1049000100", 0, 0 },
203	{ "104900020000", 0, 0 },
204	/* Too long unknown vendor extension */
205	{ "10490401"
206	  "112233445566778899001122334455667788990011223344556677889900"
207	  "112233445566778899001122334455667788990011223344556677889900"
208	  "112233445566778899001122334455667788990011223344556677889900"
209	  "112233445566778899001122334455667788990011223344556677889900"
210	  "112233445566778899001122334455667788990011223344556677889900"
211	  "112233445566778899001122334455667788990011223344556677889900"
212	  "112233445566778899001122334455667788990011223344556677889900"
213	  "112233445566778899001122334455667788990011223344556677889900"
214	  "112233445566778899001122334455667788990011223344556677889900"
215	  "112233445566778899001122334455667788990011223344556677889900"
216	  "112233445566778899001122334455667788990011223344556677889900"
217	  "112233445566778899001122334455667788990011223344556677889900"
218	  "112233445566778899001122334455667788990011223344556677889900"
219	  "112233445566778899001122334455667788990011223344556677889900"
220	  "112233445566778899001122334455667788990011223344556677889900"
221	  "112233445566778899001122334455667788990011223344556677889900"
222	  "112233445566778899001122334455667788990011223344556677889900"
223	  "112233445566778899001122334455667788990011223344556677889900"
224	  "112233445566778899001122334455667788990011223344556677889900"
225	  "112233445566778899001122334455667788990011223344556677889900"
226	  "112233445566778899001122334455667788990011223344556677889900"
227	  "112233445566778899001122334455667788990011223344556677889900"
228	  "112233445566778899001122334455667788990011223344556677889900"
229	  "112233445566778899001122334455667788990011223344556677889900"
230	  "112233445566778899001122334455667788990011223344556677889900"
231	  "112233445566778899001122334455667788990011223344556677889900"
232	  "112233445566778899001122334455667788990011223344556677889900"
233	  "112233445566778899001122334455667788990011223344556677889900"
234	  "112233445566778899001122334455667788990011223344556677889900"
235	  "112233445566778899001122334455667788990011223344556677889900"
236	  "112233445566778899001122334455667788990011223344556677889900"
237	  "112233445566778899001122334455667788990011223344556677889900"
238	  "112233445566778899001122334455667788990011223344556677889900"
239	  "112233445566778899001122334455667788990011223344556677889900"
240	  "1122334455", -1, 0 },
241	/* Maximum unknown vendor extensions */
242	{ "10490003111111104900032222221049000333333310490003444444104900035555551049000366666610490003777777104900038888881049000399999910490003AAAAAA", 0, 5 },
243	/* More than maximum unknown vendor extensions */
244	{ "10490003111111104900032222221049000333333310490003444444104900035555551049000366666610490003777777104900038888881049000399999910490003AAAAAA10490003BBBBBB", -1, 0 },
245	/* WFA vendor extensions */
246	{ "1049000300372a", 0, 0 },
247	{ "1049000400372a00", 0, 0 },
248	{ "1049000500372a0001", 0, 0 },
249	{ "1049001600372a0001ff0100020101030101040101ff00fe0101", 0, 6 },
250	/* Invalid Version2 length */
251	{ "1049000500372a0000", -1, 0 },
252	/* Invalid Network Key Shareable length */
253	{ "1049000500372a0200", -1, 0 },
254	/* Invalid Requedt To Enroll length */
255	{ "1049000500372a0300", -1, 0 },
256	/* Invalid Settings Delay Time length */
257	{ "1049000500372a0400", -1, 0 },
258	/* More than maximum Credential attributes */
259	{ "100e0000100e0000100e0000100e0000100e0000100e0000100e0000100e0000100e0000100e0000100e0000100e0000", 0, 2 },
260};
261
262
263static int wps_attr_parse_tests(void)
264{
265	struct wps_parse_attr attr;
266	unsigned int i;
267	int ret = 0;
268
269	wpa_printf(MSG_INFO, "WPS attribute parsing tests");
270
271	for (i = 0; i < ARRAY_SIZE(wps_attr_parse_test_cases); i++) {
272		struct wpabuf *buf;
273		size_t len;
274		struct wps_attr_parse_test *test =
275			&wps_attr_parse_test_cases[i];
276
277		len = os_strlen(test->data) / 2;
278		buf = wpabuf_alloc(len);
279		if (buf == NULL)
280			return -1;
281		if (hexstr2bin(test->data, wpabuf_put(buf, len), len) < 0) {
282			wpabuf_free(buf);
283			return -1;
284		}
285		if (wps_parse_msg(buf, &attr) != test->result) {
286			wpa_printf(MSG_ERROR, "WPS attribute parsing test %u failed: %s",
287				   i, test->data);
288			ret = -1;
289		}
290		switch (test->extra) {
291		case 1:
292			if (!attr.network_key || !attr.ap_setup_locked)
293				ret = -1;
294			break;
295		case 2:
296			if (attr.num_cred != MAX_CRED_COUNT)
297				ret = -1;
298			break;
299		case 3:
300			if (!attr.network_key_idx)
301				ret = -1;
302			break;
303		case 4:
304			if (attr.num_req_dev_type != MAX_REQ_DEV_TYPE_COUNT)
305				ret = -1;
306			break;
307		case 5:
308			if (attr.num_vendor_ext != MAX_WPS_PARSE_VENDOR_EXT)
309				ret = -1;
310			break;
311		case 6:
312			if (!attr.version2 ||
313			    !attr.authorized_macs ||
314			    !attr.network_key_shareable ||
315			    !attr.request_to_enroll ||
316			    !attr.settings_delay_time)
317				ret = -1;
318			break;
319		}
320		wpabuf_free(buf);
321	}
322
323	return ret;
324}
325
326
327int wps_module_tests(void)
328{
329	int ret = 0;
330
331	wpa_printf(MSG_INFO, "WPS module tests");
332
333	if (wps_attr_parse_tests() < 0)
334		ret = -1;
335
336	return ret;
337}
338