contrib/wpa/src/common/common_module_tests.c

   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
  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 }