libeap/src/common/common_module_tests.c

   1 /*
   2  * common module tests
   3  * Copyright (c) 2014-2015, 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 "ieee802_11_defs.h"
  14 #include "gas.h"
  15 #include "wpa_common.h"
  16
  17
  18 struct ieee802_11_parse_test_data {
  19         u8 *data;
  20         size_t len;
  21         ParseRes result;
  22         int count;
  23 };
  24
  25 static const struct ieee802_11_parse_test_data parse_tests[] = {
  26         { (u8 *) "", 0, ParseOK, 0 },
  27         { (u8 *) " ", 1, ParseFailed, 0 },
  28         { (u8 *) "\xff\x00", 2, ParseUnknown, 1 },
  29         { (u8 *) "\xff\x01", 2, ParseFailed, 0 },
  30         { (u8 *) "\xdd\x03\x01\x02\x03", 5, ParseUnknown, 1 },
  31         { (u8 *) "\xdd\x04\x01\x02\x03\x04", 6, ParseUnknown, 1 },
  32         { (u8 *) "\xdd\x04\x00\x50\xf2\x02", 6, ParseUnknown, 1 },
  33         { (u8 *) "\xdd\x05\x00\x50\xf2\x02\x02", 7, ParseOK, 1 },
  34         { (u8 *) "\xdd\x05\x00\x50\xf2\x02\xff", 7, ParseUnknown, 1 },
  35         { (u8 *) "\xdd\x04\x00\x50\xf2\xff", 6, ParseUnknown, 1 },
  36         { (u8 *) "\xdd\x04\x50\x6f\x9a\xff", 6, ParseUnknown, 1 },
  37         { (u8 *) "\xdd\x04\x00\x90\x4c\x33", 6, ParseOK, 1 },
  38         { (u8 *) "\xdd\x04\x00\x90\x4c\xff\xdd\x04\x00\x90\x4c\x33", 12,
  39           ParseUnknown, 2 },
  40         { (u8 *) "\x10\x01\x00\x21\x00", 5, ParseOK, 2 },
  41         { (u8 *) "\x24\x00", 2, ParseOK, 1 },
  42         { (u8 *) "\x38\x00", 2, ParseOK, 1 },
  43         { (u8 *) "\x54\x00", 2, ParseOK, 1 },
  44         { (u8 *) "\x5a\x00", 2, ParseOK, 1 },
  45         { (u8 *) "\x65\x00", 2, ParseOK, 1 },
  46         { (u8 *) "\x65\x12\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f\x10\x11",
  47           20, ParseOK, 1 },
  48         { (u8 *) "\x6e\x00", 2, ParseOK, 1 },
  49         { (u8 *) "\xc7\x00", 2, ParseOK, 1 },
  50         { (u8 *) "\xc7\x01\x00", 3, ParseOK, 1 },
  51         { (u8 *) "\x03\x00\x2a\x00\x36\x00\x37\x00\x38\x00\x2d\x00\x3d\x00\xbf\x00\xc0\x00",
  52           18, ParseOK, 9 },
  53         { (u8 *) "\x8b\x00", 2, ParseOK, 1 },
  54         { (u8 *) "\xdd\x04\x00\x90\x4c\x04", 6, ParseUnknown, 1 },
  55         { NULL, 0, ParseOK, 0 }
  56 };
  57
  58 static int ieee802_11_parse_tests(void)
  59 {
  60         int i, ret = 0;
  61
  62         wpa_printf(MSG_INFO, "ieee802_11_parse tests");
  63
  64         for (i = 0; parse_tests[i].data; i++) {
  65                 const struct ieee802_11_parse_test_data *test;
  66                 struct ieee802_11_elems elems;
  67                 ParseRes res;
  68
  69                 test = &parse_tests[i];
  70                 res = ieee802_11_parse_elems(test->data, test->len, &elems, 1);
  71                 if (res != test->result ||
  72                     ieee802_11_ie_count(test->data, test->len) != test->count) {
  73                         wpa_printf(MSG_ERROR, "ieee802_11_parse test %d failed",
  74                                    i);
  75                         ret = -1;
  76                 }
  77         }
  78
  79         if (ieee802_11_vendor_ie_concat((const u8 *) "\x00\x01", 2, 0) != NULL)
  80         {
  81                 wpa_printf(MSG_ERROR,
  82                            "ieee802_11_vendor_ie_concat test failed");
  83                 ret = -1;
  84         }
  85
  86         return ret;
  87 }
  88
  89
  90 struct rsn_ie_parse_test_data {
  91         u8 *data;
  92         size_t len;
  93         int result;
  94 };
  95
  96 static const struct rsn_ie_parse_test_data rsn_parse_tests[] = {
  97         { (u8 *) "", 0, -1 },
  98         { (u8 *) "\x30\x00", 2, -1 },
  99         { (u8 *) "\x30\x02\x01\x00", 4, 0 },
 100         { (u8 *) "\x30\x02\x00\x00", 4, -2 },
 101         { (u8 *) "\x30\x02\x02\x00", 4, -2 },
 102         { (u8 *) "\x30\x02\x00\x01", 4, -2 },
 103         { (u8 *) "\x30\x02\x00\x00\x00", 5, -2 },
 104         { (u8 *) "\x30\x03\x01\x00\x00", 5, -3 },
 105         { (u8 *) "\x30\x06\x01\x00\x00\x00\x00\x00", 8, -1 },
 106         { (u8 *) "\x30\x06\x01\x00\x00\x0f\xac\x04", 8, 0 },
 107         { (u8 *) "\x30\x07\x01\x00\x00\x0f\xac\x04\x00", 9, -5 },
 108         { (u8 *) "\x30\x08\x01\x00\x00\x0f\xac\x04\x00\x00", 10, -4 },
 109         { (u8 *) "\x30\x08\x01\x00\x00\x0f\xac\x04\x00\x01", 10, -4 },
 110         { (u8 *) "\x30\x0c\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04",
 111           14, 0 },
 112         { (u8 *) "\x30\x0c\x01\x00\x00\x0f\xac\x04\x00\x01\x00\x0f\xac\x04",
 113           14, -4 },
 114         { (u8 *) "\x30\x0c\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x06",
 115           14, -1 },
 116         { (u8 *) "\x30\x10\x01\x00\x00\x0f\xac\x04\x02\x00\x00\x0f\xac\x04\x00\x0f\xac\x08",
 117           18, 0 },
 118         { (u8 *) "\x30\x0d\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x00",
 119           15, -7 },
 120         { (u8 *) "\x30\x0e\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x00\x00",
 121           16, -6 },
 122         { (u8 *) "\x30\x0e\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x00\x01",
 123           16, -6 },
 124         { (u8 *) "\x30\x12\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x01",
 125           20, 0 },
 126         { (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",
 127           24, 0 },
 128         { (u8 *) "\x30\x13\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x01\x00",
 129           21, 0 },
 130         { (u8 *) "\x30\x14\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x04\x01\x00\x00\x0f\xac\x01\x00\x00",
 131           22, 0 },
 132         { (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",
 133           24, 0 },
 134         { (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",
 135           24, -9 },
 136         { (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",
 137           28, -10 },
 138         { (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",
 139           28, 0 },
 140         { (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",
 141           30, 0 },
 142         { NULL, 0, 0 }
 143 };
 144
 145 static int rsn_ie_parse_tests(void)
 146 {
 147         int i, ret = 0;
 148
 149         wpa_printf(MSG_INFO, "rsn_ie_parse tests");
 150
 151         for (i = 0; rsn_parse_tests[i].data; i++) {
 152                 const struct rsn_ie_parse_test_data *test;
 153                 struct wpa_ie_data data;
 154
 155                 test = &rsn_parse_tests[i];
 156                 if (wpa_parse_wpa_ie_rsn(test->data, test->len, &data) !=
 157                     test->result) {
 158                         wpa_printf(MSG_ERROR, "rsn_ie_parse test %d failed", i);
 159                         ret = -1;
 160                 }
 161         }
 162
 163         return ret;
 164 }
 165
 166
 167 static int gas_tests(void)
 168 {
 169         struct wpabuf *buf;
 170
 171         wpa_printf(MSG_INFO, "gas tests");
 172         gas_anqp_set_len(NULL);
 173
 174         buf = wpabuf_alloc(1);
 175         if (buf == NULL)
 176                 return -1;
 177         gas_anqp_set_len(buf);
 178         wpabuf_free(buf);
 179
 180         buf = wpabuf_alloc(20);
 181         if (buf == NULL)
 182                 return -1;
 183         wpabuf_put_u8(buf, WLAN_ACTION_PUBLIC);
 184         wpabuf_put_u8(buf, WLAN_PA_GAS_INITIAL_REQ);
 185         wpabuf_put_u8(buf, 0);
 186         wpabuf_put_be32(buf, 0);
 187         wpabuf_put_u8(buf, 0);
 188         gas_anqp_set_len(buf);
 189         wpabuf_free(buf);
 190
 191         return 0;
 192 }
 193
 194
 195 int common_module_tests(void)
 196 {
 197         int ret = 0;
 198
 199         wpa_printf(MSG_INFO, "common module tests");
 200
 201         if (ieee802_11_parse_tests() < 0 ||
 202             gas_tests() < 0 ||
 203             rsn_ie_parse_tests() < 0)
 204                 ret = -1;
 205
 206         return ret;
 207 }