/* * * Wireless daemon for Linux * * Copyright (C) 2015 Intel Corporation. All rights reserved. * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA * */ #ifdef HAVE_CONFIG_H #include #endif #include #include #include #include #include #include "src/wsc.h" static const unsigned char wsc_attrs1[] = { 0x10, 0x4a, 0x00, 0x01, 0x10, 0x10, 0x44, 0x00, 0x01, 0x02, 0x10, 0x41, 0x00, 0x01, 0x01, 0x10, 0x12, 0x00, 0x02, 0x00, 0x04, 0x10, 0x53, 0x00, 0x02, 0x26, 0x88, 0x10, 0x3b, 0x00, 0x01, 0x03, 0x10, 0x47, 0x00, 0x10, 0xc4, 0x4a, 0xad, 0x8d, 0x25, 0x2f, 0x52, 0xc6, 0xf9, 0x6b, 0x38, 0x5d, 0xcb, 0x23, 0x31, 0xae, 0x10, 0x21, 0x00, 0x15, 0x41, 0x53, 0x55, 0x53, 0x54, 0x65, 0x4b, 0x20, 0x43, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x65, 0x72, 0x20, 0x49, 0x6e, 0x63, 0x2e, 0x10, 0x23, 0x00, 0x1c, 0x57, 0x69, 0x2d, 0x46, 0x69, 0x20, 0x50, 0x72, 0x6f, 0x74, 0x65, 0x63, 0x74, 0x65, 0x64, 0x20, 0x53, 0x65, 0x74, 0x75, 0x70, 0x20, 0x52, 0x6f, 0x75, 0x74, 0x65, 0x72, 0x10, 0x24, 0x00, 0x08, 0x52, 0x54, 0x2d, 0x41, 0x43, 0x36, 0x38, 0x55, 0x10, 0x42, 0x00, 0x11, 0x31, 0x30, 0x3a, 0x63, 0x33, 0x3a, 0x37, 0x62, 0x3a, 0x35, 0x34, 0x3a, 0x37, 0x34, 0x3a, 0x64, 0x30, 0x10, 0x54, 0x00, 0x08, 0x00, 0x06, 0x00, 0x50, 0xf2, 0x04, 0x00, 0x01, 0x10, 0x11, 0x00, 0x08, 0x52, 0x54, 0x2d, 0x41, 0x43, 0x36, 0x38, 0x55, 0x10, 0x08, 0x00, 0x02, 0x20, 0x08, 0x10, 0x3c, 0x00, 0x01, 0x03, 0x10, 0x49, 0x00, 0x0e, 0x00, 0x37, 0x2a, 0x00, 0x01, 0x20, 0x01, 0x06, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, }; static void wsc_test_iter_sanity_check(const void *data) { struct wsc_attr_iter iter; struct wsc_wfa_ext_iter wfa_iter; wsc_attr_iter_init(&iter, wsc_attrs1, sizeof(wsc_attrs1)); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_VERSION); assert(wsc_attr_iter_get_length(&iter) == 1); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_WSC_STATE); assert(wsc_attr_iter_get_length(&iter) == 1); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_SELECTED_REGISTRAR); assert(wsc_attr_iter_get_length(&iter) == 1); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_DEVICE_PASSWORD_ID); assert(wsc_attr_iter_get_length(&iter) == 2); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_SELECTED_REGISTRAR_CONFIGURATION_METHODS); assert(wsc_attr_iter_get_length(&iter) == 2); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_RESPONSE_TYPE); assert(wsc_attr_iter_get_length(&iter) == 1); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_UUID_E); assert(wsc_attr_iter_get_length(&iter) == 16); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_MANUFACTURER); assert(wsc_attr_iter_get_length(&iter) == 21); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_MODEL_NAME); assert(wsc_attr_iter_get_length(&iter) == 28); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_MODEL_NUMBER); assert(wsc_attr_iter_get_length(&iter) == 8); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_SERIAL_NUMBER); assert(wsc_attr_iter_get_length(&iter) == 17); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_PRIMARY_DEVICE_TYPE); assert(wsc_attr_iter_get_length(&iter) == 8); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_DEVICE_NAME); assert(wsc_attr_iter_get_length(&iter) == 8); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_CONFIGURATION_METHODS); assert(wsc_attr_iter_get_length(&iter) == 2); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_RF_BANDS); assert(wsc_attr_iter_get_length(&iter) == 1); assert(wsc_attr_iter_next(&iter)); assert(wsc_attr_iter_get_type(&iter) == WSC_ATTR_VENDOR_EXTENSION); assert(wsc_attr_iter_get_length(&iter) == 14); assert(wsc_attr_iter_recurse_wfa_ext(&iter, &wfa_iter)); assert(wsc_wfa_ext_iter_next(&wfa_iter)); assert(wsc_wfa_ext_iter_get_type(&wfa_iter) == WSC_WFA_EXTENSION_VERSION2); assert(wsc_wfa_ext_iter_get_length(&wfa_iter) == 1); assert(wsc_wfa_ext_iter_next(&wfa_iter)); assert(wsc_wfa_ext_iter_get_type(&wfa_iter) == WSC_WFA_EXTENSION_AUTHORIZED_MACS); assert(wsc_wfa_ext_iter_get_length(&wfa_iter) == 6); assert(!wsc_attr_iter_next(&iter)); } static const unsigned char beacon1[] = { 0x10, 0x4a, 0x00, 0x01, 0x10, 0x10, 0x44, 0x00, 0x01, 0x02, 0x10, 0x49, 0x00, 0x06, 0x00, 0x37, 0x2a, 0x00, 0x01, 0x20 }; struct beacon_data { struct wsc_beacon expected; const void *pdu; unsigned int len; }; static const struct beacon_data beacon_data_1 = { .pdu = beacon1, .len = sizeof(beacon1), .expected = { .version2 = true, .config_state = WSC_CONFIG_STATE_CONFIGURED, .ap_setup_locked = false, .selected_registrar = false, }, }; static void wsc_test_parse_beacon(const void *data) { const struct beacon_data *test = data; struct wsc_beacon beacon; const struct wsc_beacon *expected = &test->expected; int r; r = wsc_parse_beacon(test->pdu, test->len, &beacon); assert(r == 0); assert(expected->version2 == beacon.version2); assert(expected->config_state == beacon.config_state); assert(expected->ap_setup_locked == beacon.ap_setup_locked); assert(expected->selected_registrar == beacon.selected_registrar); assert(expected->device_password_id == beacon.device_password_id); assert(expected->selected_reg_config_methods == beacon.selected_reg_config_methods); assert(!memcmp(expected->uuid_e, beacon.uuid_e, 16)); assert(expected->rf_bands == beacon.rf_bands); assert(!memcmp(expected->authorized_macs, beacon.authorized_macs, sizeof(beacon.authorized_macs))); assert(expected->reg_config_methods == beacon.reg_config_methods); } struct probe_response_data { struct wsc_probe_response expected; const void *pdu; unsigned int len; }; static const struct probe_response_data probe_response_data_1 = { .pdu = wsc_attrs1, .len = sizeof(wsc_attrs1), .expected = { .version2 = true, .config_state = WSC_CONFIG_STATE_CONFIGURED, .ap_setup_locked = false, .selected_registrar = true, .device_password_id = WSC_DEVICE_PASSWORD_ID_PUSH_BUTTON, .selected_reg_config_methods = WSC_CONFIGURATION_METHOD_VIRTUAL_DISPLAY_PIN | WSC_CONFIGURATION_METHOD_VIRTUAL_PUSH_BUTTON | WSC_CONFIGURATION_METHOD_PHYSICAL_PUSH_BUTTON, .response_type = WSC_RESPONSE_TYPE_AP, .uuid_e = { 0xc4, 0x4a, 0xad, 0x8d, 0x25, 0x2f, 0x52, 0xc6, 0xf9, 0x6b, 0x38, 0x5d, 0xcb, 0x23, 0x31, 0xae }, .manufacturer = "ASUSTeK Computer Inc.", .model_name = "Wi-Fi Protected Setup Router", .model_number = "RT-AC68U", .serial_number = "10:c3:7b:54:74:d0", .primary_device_type = { .category = 6, .oui = { 0x00, 0x50, 0xf2 }, .oui_type = 0x04, .subcategory = 1, }, .device_name = "RT-AC68U", .config_methods = WSC_CONFIGURATION_METHOD_VIRTUAL_DISPLAY_PIN, .rf_bands = WSC_RF_BAND_2_4_GHZ | WSC_RF_BAND_5_0_GHZ, .authorized_macs = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, }, }, }; static void wsc_test_parse_probe_response(const void *data) { const struct probe_response_data *test = data; struct wsc_probe_response probe_response; const struct wsc_probe_response *expected = &test->expected; int r; r = wsc_parse_probe_response(test->pdu, test->len, &probe_response); assert(r == 0); assert(expected->version2 == probe_response.version2); assert(expected->config_state == probe_response.config_state); assert(expected->ap_setup_locked == probe_response.ap_setup_locked); assert(expected->selected_registrar == probe_response.selected_registrar); assert(expected->device_password_id == probe_response.device_password_id); assert(expected->selected_reg_config_methods == probe_response.selected_reg_config_methods); assert(expected->response_type == probe_response.response_type); assert(!memcmp(expected->uuid_e, probe_response.uuid_e, 16)); assert(!strcmp(expected->manufacturer, probe_response.manufacturer)); assert(!strcmp(expected->model_name, probe_response.model_name)); assert(!strcmp(expected->model_number, probe_response.model_number)); assert(!strcmp(expected->serial_number, probe_response.serial_number)); assert(expected->primary_device_type.category == probe_response.primary_device_type.category); assert(!memcmp(expected->primary_device_type.oui, probe_response.primary_device_type.oui, 3)); assert(expected->primary_device_type.oui_type == probe_response.primary_device_type.oui_type); assert(expected->primary_device_type.subcategory == probe_response.primary_device_type.subcategory); assert(!strcmp(expected->device_name, probe_response.device_name)); assert(expected->config_methods == probe_response.config_methods); assert(expected->rf_bands == probe_response.rf_bands); assert(!memcmp(expected->authorized_macs, probe_response.authorized_macs, sizeof(probe_response.authorized_macs))); assert(expected->reg_config_methods == probe_response.reg_config_methods); } int main(int argc, char *argv[]) { l_test_init(&argc, &argv); l_test_add("/wsc/iter/sanity-check", wsc_test_iter_sanity_check, NULL); l_test_add("/wsc/parse/beacon 1", wsc_test_parse_beacon, &beacon_data_1); l_test_add("/wsc/parse/probe response 1", wsc_test_parse_probe_response, &probe_response_data_1); return l_test_run(); }