+++ /dev/null
-/*
- * Wi-Fi Protected Setup
- * Copyright (c) 2007-2009, Jouni Malinen <j@w1.fi>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License version 2 as
- * published by the Free Software Foundation.
- *
- * Alternatively, this software may be distributed under the terms of BSD
- * license.
- *
- * See README and COPYING for more details.
- */
-
-#include "includes.h"
-
-#include "common.h"
-#include "crypto/dh_group5.h"
-#include "common/ieee802_11_defs.h"
-#include "wps_i.h"
-#include "wps_dev_attr.h"
-
-
-#ifdef CONFIG_WPS_TESTING
-int wps_version_number = 0x20;
-int wps_testing_dummy_cred = 0;
-#endif /* CONFIG_WPS_TESTING */
-
-
-/**
- * wps_init - Initialize WPS Registration protocol data
- * @cfg: WPS configuration
- * Returns: Pointer to allocated data or %NULL on failure
- *
- * This function is used to initialize WPS data for a registration protocol
- * instance (i.e., each run of registration protocol as a Registrar of
- * Enrollee. The caller is responsible for freeing this data after the
- * registration run has been completed by calling wps_deinit().
- */
-struct wps_data * wps_init(const struct wps_config *cfg)
-{
- struct wps_data *data = os_zalloc(sizeof(*data));
- if (data == NULL)
- return NULL;
- data->wps = cfg->wps;
- data->registrar = cfg->registrar;
- if (cfg->registrar) {
- os_memcpy(data->uuid_r, cfg->wps->uuid, WPS_UUID_LEN);
- } else {
- os_memcpy(data->mac_addr_e, cfg->wps->dev.mac_addr, ETH_ALEN);
- os_memcpy(data->uuid_e, cfg->wps->uuid, WPS_UUID_LEN);
- }
- if (cfg->pin) {
- data->dev_pw_id = data->wps->oob_dev_pw_id == 0 ?
- cfg->dev_pw_id : data->wps->oob_dev_pw_id;
- data->dev_password = os_malloc(cfg->pin_len);
- if (data->dev_password == NULL) {
- os_free(data);
- return NULL;
- }
- os_memcpy(data->dev_password, cfg->pin, cfg->pin_len);
- data->dev_password_len = cfg->pin_len;
- }
-
- data->pbc = cfg->pbc;
- if (cfg->pbc) {
- /* Use special PIN '00000000' for PBC */
- data->dev_pw_id = DEV_PW_PUSHBUTTON;
- os_free(data->dev_password);
- data->dev_password = os_malloc(8);
- if (data->dev_password == NULL) {
- os_free(data);
- return NULL;
- }
- os_memset(data->dev_password, '0', 8);
- data->dev_password_len = 8;
- }
-
- data->state = data->registrar ? RECV_M1 : SEND_M1;
-
- if (cfg->assoc_wps_ie) {
- struct wps_parse_attr attr;
- wpa_hexdump_buf(MSG_DEBUG, "WPS: WPS IE from (Re)AssocReq",
- cfg->assoc_wps_ie);
- if (wps_parse_msg(cfg->assoc_wps_ie, &attr) < 0) {
- wpa_printf(MSG_DEBUG, "WPS: Failed to parse WPS IE "
- "from (Re)AssocReq");
- } else if (attr.request_type == NULL) {
- wpa_printf(MSG_DEBUG, "WPS: No Request Type attribute "
- "in (Re)AssocReq WPS IE");
- } else {
- wpa_printf(MSG_DEBUG, "WPS: Request Type (from WPS IE "
- "in (Re)AssocReq WPS IE): %d",
- *attr.request_type);
- data->request_type = *attr.request_type;
- }
- }
-
- if (cfg->new_ap_settings) {
- data->new_ap_settings =
- os_malloc(sizeof(*data->new_ap_settings));
- if (data->new_ap_settings == NULL) {
- os_free(data);
- return NULL;
- }
- os_memcpy(data->new_ap_settings, cfg->new_ap_settings,
- sizeof(*data->new_ap_settings));
- }
-
- if (cfg->peer_addr)
- os_memcpy(data->peer_dev.mac_addr, cfg->peer_addr, ETH_ALEN);
-
- data->use_psk_key = cfg->use_psk_key;
-
- return data;
-}
-
-
-/**
- * wps_deinit - Deinitialize WPS Registration protocol data
- * @data: WPS Registration protocol data from wps_init()
- */
-void wps_deinit(struct wps_data *data)
-{
- if (data->wps_pin_revealed) {
- wpa_printf(MSG_DEBUG, "WPS: Full PIN information revealed and "
- "negotiation failed");
- if (data->registrar)
- wps_registrar_invalidate_pin(data->wps->registrar,
- data->uuid_e);
- } else if (data->registrar)
- wps_registrar_unlock_pin(data->wps->registrar, data->uuid_e);
-
- wpabuf_free(data->dh_privkey);
- wpabuf_free(data->dh_pubkey_e);
- wpabuf_free(data->dh_pubkey_r);
- wpabuf_free(data->last_msg);
- os_free(data->dev_password);
- os_free(data->new_psk);
- wps_device_data_free(&data->peer_dev);
- os_free(data->new_ap_settings);
- dh5_free(data->dh_ctx);
- os_free(data);
-}
-
-
-/**
- * wps_process_msg - Process a WPS message
- * @wps: WPS Registration protocol data from wps_init()
- * @op_code: Message OP Code
- * @msg: Message data
- * Returns: Processing result
- *
- * This function is used to process WPS messages with OP Codes WSC_ACK,
- * WSC_NACK, WSC_MSG, and WSC_Done. The caller (e.g., EAP server/peer) is
- * responsible for reassembling the messages before calling this function.
- * Response to this message is built by calling wps_get_msg().
- */
-enum wps_process_res wps_process_msg(struct wps_data *wps,
- enum wsc_op_code op_code,
- const struct wpabuf *msg)
-{
- if (wps->registrar)
- return wps_registrar_process_msg(wps, op_code, msg);
- else
- return wps_enrollee_process_msg(wps, op_code, msg);
-}
-
-
-/**
- * wps_get_msg - Build a WPS message
- * @wps: WPS Registration protocol data from wps_init()
- * @op_code: Buffer for returning message OP Code
- * Returns: The generated WPS message or %NULL on failure
- *
- * This function is used to build a response to a message processed by calling
- * wps_process_msg(). The caller is responsible for freeing the buffer.
- */
-struct wpabuf * wps_get_msg(struct wps_data *wps, enum wsc_op_code *op_code)
-{
- if (wps->registrar)
- return wps_registrar_get_msg(wps, op_code);
- else
- return wps_enrollee_get_msg(wps, op_code);
-}
-
-
-/**
- * wps_is_selected_pbc_registrar - Check whether WPS IE indicates active PBC
- * @msg: WPS IE contents from Beacon or Probe Response frame
- * Returns: 1 if PBC Registrar is active, 0 if not
- */
-int wps_is_selected_pbc_registrar(const struct wpabuf *msg)
-{
- struct wps_parse_attr attr;
-
- /*
- * In theory, this could also verify that attr.sel_reg_config_methods
- * includes WPS_CONFIG_PUSHBUTTON, but some deployed AP implementations
- * do not set Selected Registrar Config Methods attribute properly, so
- * it is safer to just use Device Password ID here.
- */
-
- if (wps_parse_msg(msg, &attr) < 0 ||
- !attr.selected_registrar || *attr.selected_registrar == 0 ||
- !attr.dev_password_id ||
- WPA_GET_BE16(attr.dev_password_id) != DEV_PW_PUSHBUTTON)
- return 0;
-
-#ifdef CONFIG_WPS_STRICT
- if (!attr.sel_reg_config_methods ||
- !(WPA_GET_BE16(attr.sel_reg_config_methods) &
- WPS_CONFIG_PUSHBUTTON))
- return 0;
-#endif /* CONFIG_WPS_STRICT */
-
- return 1;
-}
-
-
-static int is_selected_pin_registrar(struct wps_parse_attr *attr)
-{
- /*
- * In theory, this could also verify that attr.sel_reg_config_methods
- * includes WPS_CONFIG_LABEL, WPS_CONFIG_DISPLAY, or WPS_CONFIG_KEYPAD,
- * but some deployed AP implementations do not set Selected Registrar
- * Config Methods attribute properly, so it is safer to just use
- * Device Password ID here.
- */
-
- if (!attr->selected_registrar || *attr->selected_registrar == 0)
- return 0;
-
- if (attr->dev_password_id != NULL &&
- WPA_GET_BE16(attr->dev_password_id) == DEV_PW_PUSHBUTTON)
- return 0;
-
-#ifdef CONFIG_WPS_STRICT
- if (!attr->sel_reg_config_methods ||
- !(WPA_GET_BE16(attr->sel_reg_config_methods) &
- (WPS_CONFIG_LABEL | WPS_CONFIG_DISPLAY | WPS_CONFIG_KEYPAD)))
- return 0;
-#endif /* CONFIG_WPS_STRICT */
-
- return 1;
-}
-
-
-/**
- * wps_is_selected_pin_registrar - Check whether WPS IE indicates active PIN
- * @msg: WPS IE contents from Beacon or Probe Response frame
- * Returns: 1 if PIN Registrar is active, 0 if not
- */
-int wps_is_selected_pin_registrar(const struct wpabuf *msg)
-{
- struct wps_parse_attr attr;
-
- if (wps_parse_msg(msg, &attr) < 0)
- return 0;
-
- return is_selected_pin_registrar(&attr);
-}
-
-
-/**
- * wps_is_addr_authorized - Check whether WPS IE authorizes MAC address
- * @msg: WPS IE contents from Beacon or Probe Response frame
- * @addr: MAC address to search for
- * @ver1_compat: Whether to use version 1 compatibility mode
- * Returns: 1 if address is authorized, 0 if not
- */
-int wps_is_addr_authorized(const struct wpabuf *msg, const u8 *addr,
- int ver1_compat)
-{
- struct wps_parse_attr attr;
- unsigned int i;
- const u8 *pos;
- const u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff };
-
- if (wps_parse_msg(msg, &attr) < 0)
- return 0;
-
- if (!attr.version2 && ver1_compat) {
- /*
- * Version 1.0 AP - AuthorizedMACs not used, so revert back to
- * old mechanism of using SelectedRegistrar.
- */
- return is_selected_pin_registrar(&attr);
- }
-
- if (!attr.authorized_macs)
- return 0;
-
- pos = attr.authorized_macs;
- for (i = 0; i < attr.authorized_macs_len / ETH_ALEN; i++) {
- if (os_memcmp(pos, addr, ETH_ALEN) == 0 ||
- os_memcmp(pos, bcast, ETH_ALEN) == 0)
- return 1;
- pos += ETH_ALEN;
- }
-
- return 0;
-}
-
-
-/**
- * wps_ap_priority_compar - Prioritize WPS IE from two APs
- * @wps_a: WPS IE contents from Beacon or Probe Response frame
- * @wps_b: WPS IE contents from Beacon or Probe Response frame
- * Returns: 1 if wps_b is considered more likely selection for WPS
- * provisioning, -1 if wps_a is considered more like, or 0 if no preference
- */
-int wps_ap_priority_compar(const struct wpabuf *wps_a,
- const struct wpabuf *wps_b)
-{
- struct wps_parse_attr attr_a, attr_b;
- int sel_a, sel_b;
-
- if (wps_a == NULL || wps_parse_msg(wps_a, &attr_a) < 0)
- return 1;
- if (wps_b == NULL || wps_parse_msg(wps_b, &attr_b) < 0)
- return -1;
-
- sel_a = attr_a.selected_registrar && *attr_a.selected_registrar != 0;
- sel_b = attr_b.selected_registrar && *attr_b.selected_registrar != 0;
-
- if (sel_a && !sel_b)
- return -1;
- if (!sel_a && sel_b)
- return 1;
-
- return 0;
-}
-
-
-/**
- * wps_get_uuid_e - Get UUID-E from WPS IE
- * @msg: WPS IE contents from Beacon or Probe Response frame
- * Returns: Pointer to UUID-E or %NULL if not included
- *
- * The returned pointer is to the msg contents and it remains valid only as
- * long as the msg buffer is valid.
- */
-const u8 * wps_get_uuid_e(const struct wpabuf *msg)
-{
- struct wps_parse_attr attr;
-
- if (wps_parse_msg(msg, &attr) < 0)
- return NULL;
- return attr.uuid_e;
-}
-
-
-/**
- * wps_build_assoc_req_ie - Build WPS IE for (Re)Association Request
- * @req_type: Value for Request Type attribute
- * Returns: WPS IE or %NULL on failure
- *
- * The caller is responsible for freeing the buffer.
- */
-struct wpabuf * wps_build_assoc_req_ie(enum wps_request_type req_type)
-{
- struct wpabuf *ie;
- u8 *len;
-
- wpa_printf(MSG_DEBUG, "WPS: Building WPS IE for (Re)Association "
- "Request");
- ie = wpabuf_alloc(100);
- if (ie == NULL)
- return NULL;
-
- wpabuf_put_u8(ie, WLAN_EID_VENDOR_SPECIFIC);
- len = wpabuf_put(ie, 1);
- wpabuf_put_be32(ie, WPS_DEV_OUI_WFA);
-
- if (wps_build_version(ie) ||
- wps_build_req_type(ie, req_type) ||
- wps_build_wfa_ext(ie, 0, NULL, 0)) {
- wpabuf_free(ie);
- return NULL;
- }
-
- *len = wpabuf_len(ie) - 2;
-
- return ie;
-}
-
-
-/**
- * wps_build_assoc_resp_ie - Build WPS IE for (Re)Association Response
- * Returns: WPS IE or %NULL on failure
- *
- * The caller is responsible for freeing the buffer.
- */
-struct wpabuf * wps_build_assoc_resp_ie(void)
-{
- struct wpabuf *ie;
- u8 *len;
-
- wpa_printf(MSG_DEBUG, "WPS: Building WPS IE for (Re)Association "
- "Response");
- ie = wpabuf_alloc(100);
- if (ie == NULL)
- return NULL;
-
- wpabuf_put_u8(ie, WLAN_EID_VENDOR_SPECIFIC);
- len = wpabuf_put(ie, 1);
- wpabuf_put_be32(ie, WPS_DEV_OUI_WFA);
-
- if (wps_build_version(ie) ||
- wps_build_resp_type(ie, WPS_RESP_AP) ||
- wps_build_wfa_ext(ie, 0, NULL, 0)) {
- wpabuf_free(ie);
- return NULL;
- }
-
- *len = wpabuf_len(ie) - 2;
-
- return ie;
-}
-
-
-/**
- * wps_build_probe_req_ie - Build WPS IE for Probe Request
- * @pbc: Whether searching for PBC mode APs
- * @dev: Device attributes
- * @uuid: Own UUID
- * @req_type: Value for Request Type attribute
- * Returns: WPS IE or %NULL on failure
- *
- * The caller is responsible for freeing the buffer.
- */
-struct wpabuf * wps_build_probe_req_ie(int pbc, struct wps_device_data *dev,
- const u8 *uuid,
- enum wps_request_type req_type)
-{
- struct wpabuf *ie;
- u16 methods;
-
- wpa_printf(MSG_DEBUG, "WPS: Building WPS IE for Probe Request");
-
- ie = wpabuf_alloc(500);
- if (ie == NULL)
- return NULL;
-
- if (pbc) {
- methods = WPS_CONFIG_PUSHBUTTON;
-#ifdef CONFIG_WPS2
- /*
- * TODO: At least in theory, should figure out whether this
- * Probe Request was triggered with physical or virtual
- * pushbutton.
- */
- methods |= WPS_CONFIG_VIRT_PUSHBUTTON;
-#endif /* CONFIG_WPS2 */
- } else {
- /*
- * TODO: At least in theory, should figure out whether this
- * Probe Request was triggered using physical or virtual
- * display.
- */
- methods = WPS_CONFIG_LABEL | WPS_CONFIG_DISPLAY |
-#ifdef CONFIG_WPS2
- WPS_CONFIG_VIRT_DISPLAY |
-#endif /* CONFIG_WPS2 */
- WPS_CONFIG_KEYPAD;
-#ifdef CONFIG_WPS_UFD
- methods |= WPS_CONFIG_USBA;
-#endif /* CONFIG_WPS_UFD */
-#ifdef CONFIG_WPS_NFC
- methods |= WPS_CONFIG_NFC_INTERFACE;
-#endif /* CONFIG_WPS_NFC */
- }
-
- if (wps_build_version(ie) ||
- wps_build_req_type(ie, req_type) ||
- wps_build_config_methods(ie, methods) ||
- wps_build_uuid_e(ie, uuid) ||
- wps_build_primary_dev_type(dev, ie) ||
- wps_build_rf_bands(dev, ie) ||
- wps_build_assoc_state(NULL, ie) ||
- wps_build_config_error(ie, WPS_CFG_NO_ERROR) ||
- wps_build_dev_password_id(ie, pbc ? DEV_PW_PUSHBUTTON :
- DEV_PW_DEFAULT) ||
-#ifdef CONFIG_WPS2
- wps_build_manufacturer(dev, ie) ||
- wps_build_model_name(dev, ie) ||
- wps_build_model_number(dev, ie) ||
- wps_build_dev_name(dev, ie) ||
- wps_build_wfa_ext(ie, req_type == WPS_REQ_ENROLLEE, NULL, 0)
-#else /* CONFIG_WPS2 */
- 0
-#endif /* CONFIG_WPS2 */
- ) {
- wpabuf_free(ie);
- return NULL;
- }
-
-#ifndef CONFIG_WPS2
- if (dev->p2p && wps_build_dev_name(dev, ie)) {
- wpabuf_free(ie);
- return NULL;
- }
-#endif /* CONFIG_WPS2 */
-
- return wps_ie_encapsulate(ie);
-}
-
-
-void wps_free_pending_msgs(struct upnp_pending_message *msgs)
-{
- struct upnp_pending_message *p, *prev;
- p = msgs;
- while (p) {
- prev = p;
- p = p->next;
- wpabuf_free(prev->msg);
- os_free(prev);
- }
-}
-
-
-int wps_attr_text(struct wpabuf *data, char *buf, char *end)
-{
- struct wps_parse_attr attr;
- char *pos = buf;
- int ret;
-
- if (wps_parse_msg(data, &attr) < 0)
- return -1;
-
- if (attr.wps_state) {
- if (*attr.wps_state == WPS_STATE_NOT_CONFIGURED)
- ret = os_snprintf(pos, end - pos,
- "wps_state=unconfigured\n");
- else if (*attr.wps_state == WPS_STATE_CONFIGURED)
- ret = os_snprintf(pos, end - pos,
- "wps_state=configured\n");
- else
- ret = 0;
- if (ret < 0 || ret >= end - pos)
- return pos - buf;
- pos += ret;
- }
-
- if (attr.ap_setup_locked && *attr.ap_setup_locked) {
- ret = os_snprintf(pos, end - pos,
- "wps_ap_setup_locked=1\n");
- if (ret < 0 || ret >= end - pos)
- return pos - buf;
- pos += ret;
- }
-
- if (attr.selected_registrar && *attr.selected_registrar) {
- ret = os_snprintf(pos, end - pos,
- "wps_selected_registrar=1\n");
- if (ret < 0 || ret >= end - pos)
- return pos - buf;
- pos += ret;
- }
-
- if (attr.dev_password_id) {
- ret = os_snprintf(pos, end - pos,
- "wps_device_password_id=%u\n",
- WPA_GET_BE16(attr.dev_password_id));
- if (ret < 0 || ret >= end - pos)
- return pos - buf;
- pos += ret;
- }
-
- if (attr.sel_reg_config_methods) {
- ret = os_snprintf(pos, end - pos,
- "wps_selected_registrar_config_methods="
- "0x%04x\n",
- WPA_GET_BE16(attr.sel_reg_config_methods));
- if (ret < 0 || ret >= end - pos)
- return pos - buf;
- pos += ret;
- }
-
- if (attr.primary_dev_type) {
- char devtype[WPS_DEV_TYPE_BUFSIZE];
- ret = os_snprintf(pos, end - pos,
- "wps_primary_device_type=%s\n",
- wps_dev_type_bin2str(attr.primary_dev_type,
- devtype,
- sizeof(devtype)));
- if (ret < 0 || ret >= end - pos)
- return pos - buf;
- pos += ret;
- }
-
- if (attr.dev_name) {
- char *str = os_malloc(attr.dev_name_len + 1);
- size_t i;
- if (str == NULL)
- return pos - buf;
- for (i = 0; i < attr.dev_name_len; i++) {
- if (attr.dev_name[i] < 32)
- str[i] = '_';
- else
- str[i] = attr.dev_name[i];
- }
- str[i] = '\0';
- ret = os_snprintf(pos, end - pos, "wps_device_name=%s\n", str);
- os_free(str);
- if (ret < 0 || ret >= end - pos)
- return pos - buf;
- pos += ret;
- }
-
- if (attr.config_methods) {
- ret = os_snprintf(pos, end - pos,
- "wps_config_methods=0x%04x\n",
- WPA_GET_BE16(attr.config_methods));
- if (ret < 0 || ret >= end - pos)
- return pos - buf;
- pos += ret;
- }
-
- return pos - buf;
-}