libeap/src/wps/wps_module_tests.c

   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
  14 struct wps_attr_parse_test {
  15         const char *data;
  16         int result;
  17         int extra;
  18 };
  19
  20 const struct 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
 263 static 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                 const 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
 327 int 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 }