src/wps/wps.c

   1 /*
   2  * Wi-Fi Protected Setup
   3  * Copyright (c) 2007-2009, Jouni Malinen <j@w1.fi>
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License version 2 as
   7  * published by the Free Software Foundation.
   8  *
   9  * Alternatively, this software may be distributed under the terms of BSD
  10  * license.
  11  *
  12  * See README and COPYING for more details.
  13  */
  14
  15 #include "includes.h"
  16
  17 #include "common.h"
  18 #include "crypto/dh_group5.h"
  19 #include "common/ieee802_11_defs.h"
  20 #include "wps_i.h"
  21 #include "wps_dev_attr.h"
  22
  23
  24 /**
  25  * wps_init - Initialize WPS Registration protocol data
  26  * @cfg: WPS configuration
  27  * Returns: Pointer to allocated data or %NULL on failure
  28  *
  29  * This function is used to initialize WPS data for a registration protocol
  30  * instance (i.e., each run of registration protocol as a Registrar of
  31  * Enrollee. The caller is responsible for freeing this data after the
  32  * registration run has been completed by calling wps_deinit().
  33  */
  34 struct wps_data * wps_init(const struct wps_config *cfg)
  35 {
  36         struct wps_data *data = os_zalloc(sizeof(*data));
  37         if (data == NULL)
  38                 return NULL;
  39         data->wps = cfg->wps;
  40         data->registrar = cfg->registrar;
  41         if (cfg->registrar) {
  42                 os_memcpy(data->uuid_r, cfg->wps->uuid, WPS_UUID_LEN);
  43         } else {
  44                 os_memcpy(data->mac_addr_e, cfg->wps->dev.mac_addr, ETH_ALEN);
  45                 os_memcpy(data->uuid_e, cfg->wps->uuid, WPS_UUID_LEN);
  46         }
  47         if (cfg->pin) {
  48                 data->dev_pw_id = data->wps->oob_dev_pw_id == 0 ?
  49                         cfg->dev_pw_id : data->wps->oob_dev_pw_id;
  50                 data->dev_password = os_malloc(cfg->pin_len);
  51                 if (data->dev_password == NULL) {
  52                         os_free(data);
  53                         return NULL;
  54                 }
  55                 os_memcpy(data->dev_password, cfg->pin, cfg->pin_len);
  56                 data->dev_password_len = cfg->pin_len;
  57         }
  58
  59         data->pbc = cfg->pbc;
  60         if (cfg->pbc) {
  61                 /* Use special PIN '00000000' for PBC */
  62                 data->dev_pw_id = DEV_PW_PUSHBUTTON;
  63                 os_free(data->dev_password);
  64                 data->dev_password = os_malloc(8);
  65                 if (data->dev_password == NULL) {
  66                         os_free(data);
  67                         return NULL;
  68                 }
  69                 os_memset(data->dev_password, '0', 8);
  70                 data->dev_password_len = 8;
  71         }
  72
  73         data->state = data->registrar ? RECV_M1 : SEND_M1;
  74
  75         if (cfg->assoc_wps_ie) {
  76                 struct wps_parse_attr attr;
  77                 wpa_hexdump_buf(MSG_DEBUG, "WPS: WPS IE from (Re)AssocReq",
  78                                 cfg->assoc_wps_ie);
  79                 if (wps_parse_msg(cfg->assoc_wps_ie, &attr) < 0) {
  80                         wpa_printf(MSG_DEBUG, "WPS: Failed to parse WPS IE "
  81                                    "from (Re)AssocReq");
  82                 } else if (attr.request_type == NULL) {
  83                         wpa_printf(MSG_DEBUG, "WPS: No Request Type attribute "
  84                                    "in (Re)AssocReq WPS IE");
  85                 } else {
  86                         wpa_printf(MSG_DEBUG, "WPS: Request Type (from WPS IE "
  87                                    "in (Re)AssocReq WPS IE): %d",
  88                                    *attr.request_type);
  89                         data->request_type = *attr.request_type;
  90                 }
  91         }
  92
  93         if (cfg->new_ap_settings) {
  94                 data->new_ap_settings =
  95                         os_malloc(sizeof(*data->new_ap_settings));
  96                 if (data->new_ap_settings == NULL) {
  97                         os_free(data);
  98                         return NULL;
  99                 }
 100                 os_memcpy(data->new_ap_settings, cfg->new_ap_settings,
 101                           sizeof(*data->new_ap_settings));
 102         }
 103
 104         if (cfg->peer_addr)
 105                 os_memcpy(data->peer_dev.mac_addr, cfg->peer_addr, ETH_ALEN);
 106
 107         data->use_psk_key = cfg->use_psk_key;
 108
 109         return data;
 110 }
 111
 112
 113 /**
 114  * wps_deinit - Deinitialize WPS Registration protocol data
 115  * @data: WPS Registration protocol data from wps_init()
 116  */
 117 void wps_deinit(struct wps_data *data)
 118 {
 119         if (data->wps_pin_revealed) {
 120                 wpa_printf(MSG_DEBUG, "WPS: Full PIN information revealed and "
 121                            "negotiation failed");
 122                 if (data->registrar)
 123                         wps_registrar_invalidate_pin(data->wps->registrar,
 124                                                      data->uuid_e);
 125         } else if (data->registrar)
 126                 wps_registrar_unlock_pin(data->wps->registrar, data->uuid_e);
 127
 128         wpabuf_free(data->dh_privkey);
 129         wpabuf_free(data->dh_pubkey_e);
 130         wpabuf_free(data->dh_pubkey_r);
 131         wpabuf_free(data->last_msg);
 132         os_free(data->dev_password);
 133         os_free(data->new_psk);
 134         wps_device_data_free(&data->peer_dev);
 135         os_free(data->new_ap_settings);
 136         dh5_free(data->dh_ctx);
 137         os_free(data);
 138 }
 139
 140
 141 /**
 142  * wps_process_msg - Process a WPS message
 143  * @wps: WPS Registration protocol data from wps_init()
 144  * @op_code: Message OP Code
 145  * @msg: Message data
 146  * Returns: Processing result
 147  *
 148  * This function is used to process WPS messages with OP Codes WSC_ACK,
 149  * WSC_NACK, WSC_MSG, and WSC_Done. The caller (e.g., EAP server/peer) is
 150  * responsible for reassembling the messages before calling this function.
 151  * Response to this message is built by calling wps_get_msg().
 152  */
 153 enum wps_process_res wps_process_msg(struct wps_data *wps,
 154                                      enum wsc_op_code op_code,
 155                                      const struct wpabuf *msg)
 156 {
 157         if (wps->registrar)
 158                 return wps_registrar_process_msg(wps, op_code, msg);
 159         else
 160                 return wps_enrollee_process_msg(wps, op_code, msg);
 161 }
 162
 163
 164 /**
 165  * wps_get_msg - Build a WPS message
 166  * @wps: WPS Registration protocol data from wps_init()
 167  * @op_code: Buffer for returning message OP Code
 168  * Returns: The generated WPS message or %NULL on failure
 169  *
 170  * This function is used to build a response to a message processed by calling
 171  * wps_process_msg(). The caller is responsible for freeing the buffer.
 172  */
 173 struct wpabuf * wps_get_msg(struct wps_data *wps, enum wsc_op_code *op_code)
 174 {
 175         if (wps->registrar)
 176                 return wps_registrar_get_msg(wps, op_code);
 177         else
 178                 return wps_enrollee_get_msg(wps, op_code);
 179 }
 180
 181
 182 /**
 183  * wps_is_selected_pbc_registrar - Check whether WPS IE indicates active PBC
 184  * @msg: WPS IE contents from Beacon or Probe Response frame
 185  * Returns: 1 if PBC Registrar is active, 0 if not
 186  */
 187 int wps_is_selected_pbc_registrar(const struct wpabuf *msg)
 188 {
 189         struct wps_parse_attr attr;
 190
 191         /*
 192          * In theory, this could also verify that attr.sel_reg_config_methods
 193          * includes WPS_CONFIG_PUSHBUTTON, but some deployed AP implementations
 194          * do not set Selected Registrar Config Methods attribute properly, so
 195          * it is safer to just use Device Password ID here.
 196          */
 197
 198         if (wps_parse_msg(msg, &attr) < 0 ||
 199             !attr.selected_registrar || *attr.selected_registrar == 0 ||
 200             !attr.dev_password_id ||
 201             WPA_GET_BE16(attr.dev_password_id) != DEV_PW_PUSHBUTTON)
 202                 return 0;
 203
 204 #ifdef CONFIG_WPS_STRICT
 205         if (!attr.sel_reg_config_methods ||
 206             !(WPA_GET_BE16(attr.sel_reg_config_methods) &
 207               WPS_CONFIG_PUSHBUTTON))
 208                 return 0;
 209 #endif /* CONFIG_WPS_STRICT */
 210
 211         return 1;
 212 }
 213
 214
 215 static int is_selected_pin_registrar(struct wps_parse_attr *attr)
 216 {
 217         /*
 218          * In theory, this could also verify that attr.sel_reg_config_methods
 219          * includes WPS_CONFIG_LABEL, WPS_CONFIG_DISPLAY, or WPS_CONFIG_KEYPAD,
 220          * but some deployed AP implementations do not set Selected Registrar
 221          * Config Methods attribute properly, so it is safer to just use
 222          * Device Password ID here.
 223          */
 224
 225         if (!attr->selected_registrar || *attr->selected_registrar == 0)
 226                 return 0;
 227
 228         if (attr->dev_password_id != NULL &&
 229             WPA_GET_BE16(attr->dev_password_id) == DEV_PW_PUSHBUTTON)
 230                 return 0;
 231
 232 #ifdef CONFIG_WPS_STRICT
 233         if (!attr->sel_reg_config_methods ||
 234             !(WPA_GET_BE16(attr->sel_reg_config_methods) &
 235               (WPS_CONFIG_LABEL | WPS_CONFIG_DISPLAY | WPS_CONFIG_KEYPAD)))
 236                 return 0;
 237 #endif /* CONFIG_WPS_STRICT */
 238
 239         return 1;
 240 }
 241
 242
 243 /**
 244  * wps_is_selected_pin_registrar - Check whether WPS IE indicates active PIN
 245  * @msg: WPS IE contents from Beacon or Probe Response frame
 246  * Returns: 1 if PIN Registrar is active, 0 if not
 247  */
 248 int wps_is_selected_pin_registrar(const struct wpabuf *msg)
 249 {
 250         struct wps_parse_attr attr;
 251
 252         if (wps_parse_msg(msg, &attr) < 0)
 253                 return 0;
 254
 255         return is_selected_pin_registrar(&attr);
 256 }
 257
 258
 259 /**
 260  * wps_is_addr_authorized - Check whether WPS IE authorizes MAC address
 261  * @msg: WPS IE contents from Beacon or Probe Response frame
 262  * @addr: MAC address to search for
 263  * @ver1_compat: Whether to use version 1 compatibility mode
 264  * Returns: 1 if address is authorized, 0 if not
 265  */
 266 int wps_is_addr_authorized(const struct wpabuf *msg, const u8 *addr,
 267                            int ver1_compat)
 268 {
 269         struct wps_parse_attr attr;
 270         unsigned int i;
 271         const u8 *pos;
 272         const u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff };
 273
 274         if (wps_parse_msg(msg, &attr) < 0)
 275                 return 0;
 276
 277         if (!attr.version2 && ver1_compat) {
 278                 /*
 279                  * Version 1.0 AP - AuthorizedMACs not used, so revert back to
 280                  * old mechanism of using SelectedRegistrar.
 281                  */
 282                 return is_selected_pin_registrar(&attr);
 283         }
 284
 285         if (!attr.authorized_macs)
 286                 return 0;
 287
 288         pos = attr.authorized_macs;
 289         for (i = 0; i < attr.authorized_macs_len / ETH_ALEN; i++) {
 290                 if (os_memcmp(pos, addr, ETH_ALEN) == 0 ||
 291                     os_memcmp(pos, bcast, ETH_ALEN) == 0)
 292                         return 1;
 293                 pos += ETH_ALEN;
 294         }
 295
 296         return 0;
 297 }
 298
 299
 300 /**
 301  * wps_ap_priority_compar - Prioritize WPS IE from two APs
 302  * @wps_a: WPS IE contents from Beacon or Probe Response frame
 303  * @wps_b: WPS IE contents from Beacon or Probe Response frame
 304  * Returns: 1 if wps_b is considered more likely selection for WPS
 305  * provisioning, -1 if wps_a is considered more like, or 0 if no preference
 306  */
 307 int wps_ap_priority_compar(const struct wpabuf *wps_a,
 308                            const struct wpabuf *wps_b)
 309 {
 310         struct wps_parse_attr attr_a, attr_b;
 311         int sel_a, sel_b;
 312
 313         if (wps_a == NULL || wps_parse_msg(wps_a, &attr_a) < 0)
 314                 return 1;
 315         if (wps_b == NULL || wps_parse_msg(wps_b, &attr_b) < 0)
 316                 return -1;
 317
 318         sel_a = attr_a.selected_registrar && *attr_a.selected_registrar != 0;
 319         sel_b = attr_b.selected_registrar && *attr_b.selected_registrar != 0;
 320
 321         if (sel_a && !sel_b)
 322                 return -1;
 323         if (!sel_a && sel_b)
 324                 return 1;
 325
 326         return 0;
 327 }
 328
 329
 330 /**
 331  * wps_get_uuid_e - Get UUID-E from WPS IE
 332  * @msg: WPS IE contents from Beacon or Probe Response frame
 333  * Returns: Pointer to UUID-E or %NULL if not included
 334  *
 335  * The returned pointer is to the msg contents and it remains valid only as
 336  * long as the msg buffer is valid.
 337  */
 338 const u8 * wps_get_uuid_e(const struct wpabuf *msg)
 339 {
 340         struct wps_parse_attr attr;
 341
 342         if (wps_parse_msg(msg, &attr) < 0)
 343                 return NULL;
 344         return attr.uuid_e;
 345 }
 346
 347
 348 /**
 349  * wps_build_assoc_req_ie - Build WPS IE for (Re)Association Request
 350  * @req_type: Value for Request Type attribute
 351  * Returns: WPS IE or %NULL on failure
 352  *
 353  * The caller is responsible for freeing the buffer.
 354  */
 355 struct wpabuf * wps_build_assoc_req_ie(enum wps_request_type req_type)
 356 {
 357         struct wpabuf *ie;
 358         u8 *len;
 359
 360         wpa_printf(MSG_DEBUG, "WPS: Building WPS IE for (Re)Association "
 361                    "Request");
 362         ie = wpabuf_alloc(100);
 363         if (ie == NULL)
 364                 return NULL;
 365
 366         wpabuf_put_u8(ie, WLAN_EID_VENDOR_SPECIFIC);
 367         len = wpabuf_put(ie, 1);
 368         wpabuf_put_be32(ie, WPS_DEV_OUI_WFA);
 369
 370         if (wps_build_version(ie) ||
 371             wps_build_req_type(ie, req_type) ||
 372             wps_build_wfa_ext(ie, 0, NULL, 0)) {
 373                 wpabuf_free(ie);
 374                 return NULL;
 375         }
 376
 377         *len = wpabuf_len(ie) - 2;
 378
 379         return ie;
 380 }
 381
 382
 383 /**
 384  * wps_build_assoc_resp_ie - Build WPS IE for (Re)Association Response
 385  * Returns: WPS IE or %NULL on failure
 386  *
 387  * The caller is responsible for freeing the buffer.
 388  */
 389 struct wpabuf * wps_build_assoc_resp_ie(void)
 390 {
 391         struct wpabuf *ie;
 392         u8 *len;
 393
 394         wpa_printf(MSG_DEBUG, "WPS: Building WPS IE for (Re)Association "
 395                    "Response");
 396         ie = wpabuf_alloc(100);
 397         if (ie == NULL)
 398                 return NULL;
 399
 400         wpabuf_put_u8(ie, WLAN_EID_VENDOR_SPECIFIC);
 401         len = wpabuf_put(ie, 1);
 402         wpabuf_put_be32(ie, WPS_DEV_OUI_WFA);
 403
 404         if (wps_build_version(ie) ||
 405             wps_build_resp_type(ie, WPS_RESP_AP) ||
 406             wps_build_wfa_ext(ie, 0, NULL, 0)) {
 407                 wpabuf_free(ie);
 408                 return NULL;
 409         }
 410
 411         *len = wpabuf_len(ie) - 2;
 412
 413         return ie;
 414 }
 415
 416
 417 /**
 418  * wps_build_probe_req_ie - Build WPS IE for Probe Request
 419  * @pbc: Whether searching for PBC mode APs
 420  * @dev: Device attributes
 421  * @uuid: Own UUID
 422  * @req_type: Value for Request Type attribute
 423  * Returns: WPS IE or %NULL on failure
 424  *
 425  * The caller is responsible for freeing the buffer.
 426  */
 427 struct wpabuf * wps_build_probe_req_ie(int pbc, struct wps_device_data *dev,
 428                                        const u8 *uuid,
 429                                        enum wps_request_type req_type)
 430 {
 431         struct wpabuf *ie;
 432         u16 methods;
 433
 434         wpa_printf(MSG_DEBUG, "WPS: Building WPS IE for Probe Request");
 435
 436         ie = wpabuf_alloc(500);
 437         if (ie == NULL)
 438                 return NULL;
 439
 440         if (pbc) {
 441                 methods = WPS_CONFIG_PUSHBUTTON;
 442 #ifdef CONFIG_WPS2
 443                 /*
 444                  * TODO: At least in theory, should figure out whether this
 445                  * Probe Request was triggered with physical or virtual
 446                  * pushbutton.
 447                  */
 448                 methods |= WPS_CONFIG_VIRT_PUSHBUTTON;
 449 #endif /* CONFIG_WPS2 */
 450         } else {
 451                 /*
 452                  * TODO: At least in theory, should figure out whether this
 453                  * Probe Request was triggered using physical or virtual
 454                  * display.
 455                  */
 456                 methods = WPS_CONFIG_LABEL | WPS_CONFIG_DISPLAY |
 457 #ifdef CONFIG_WPS2
 458                         WPS_CONFIG_VIRT_DISPLAY |
 459 #endif /* CONFIG_WPS2 */
 460                         WPS_CONFIG_KEYPAD;
 461 #ifdef CONFIG_WPS_UFD
 462                 methods |= WPS_CONFIG_USBA;
 463 #endif /* CONFIG_WPS_UFD */
 464 #ifdef CONFIG_WPS_NFC
 465                 methods |= WPS_CONFIG_NFC_INTERFACE;
 466 #endif /* CONFIG_WPS_NFC */
 467         }
 468
 469         if (wps_build_version(ie) ||
 470             wps_build_req_type(ie, req_type) ||
 471             wps_build_config_methods(ie, methods) ||
 472             wps_build_uuid_e(ie, uuid) ||
 473             wps_build_primary_dev_type(dev, ie) ||
 474             wps_build_rf_bands(dev, ie) ||
 475             wps_build_assoc_state(NULL, ie) ||
 476             wps_build_config_error(ie, WPS_CFG_NO_ERROR) ||
 477             wps_build_dev_password_id(ie, pbc ? DEV_PW_PUSHBUTTON :
 478                                       DEV_PW_DEFAULT) ||
 479 #ifdef CONFIG_WPS2
 480             wps_build_manufacturer(dev, ie) ||
 481             wps_build_model_name(dev, ie) ||
 482             wps_build_model_number(dev, ie) ||
 483             wps_build_dev_name(dev, ie) ||
 484             wps_build_wfa_ext(ie, req_type == WPS_REQ_ENROLLEE, NULL, 0)
 485 #else /* CONFIG_WPS2 */
 486             0
 487 #endif /* CONFIG_WPS2 */
 488                 ) {
 489                 wpabuf_free(ie);
 490                 return NULL;
 491         }
 492
 493 #ifndef CONFIG_WPS2
 494         if (dev->p2p && wps_build_dev_name(dev, ie)) {
 495                 wpabuf_free(ie);
 496                 return NULL;
 497         }
 498 #endif /* CONFIG_WPS2 */
 499
 500         return wps_ie_encapsulate(ie);
 501 }
 502
 503
 504 void wps_free_pending_msgs(struct upnp_pending_message *msgs)
 505 {
 506         struct upnp_pending_message *p, *prev;
 507         p = msgs;
 508         while (p) {
 509                 prev = p;
 510                 p = p->next;
 511                 wpabuf_free(prev->msg);
 512                 os_free(prev);
 513         }
 514 }
 515
 516
 517 int wps_attr_text(struct wpabuf *data, char *buf, char *end)
 518 {
 519         struct wps_parse_attr attr;
 520         char *pos = buf;
 521         int ret;
 522
 523         if (wps_parse_msg(data, &attr) < 0)
 524                 return -1;
 525
 526         if (attr.wps_state) {
 527                 if (*attr.wps_state == WPS_STATE_NOT_CONFIGURED)
 528                         ret = os_snprintf(pos, end - pos,
 529                                           "wps_state=unconfigured\n");
 530                 else if (*attr.wps_state == WPS_STATE_CONFIGURED)
 531                         ret = os_snprintf(pos, end - pos,
 532                                           "wps_state=configured\n");
 533                 else
 534                         ret = 0;
 535                 if (ret < 0 || ret >= end - pos)
 536                         return pos - buf;
 537                 pos += ret;
 538         }
 539
 540         if (attr.ap_setup_locked && *attr.ap_setup_locked) {
 541                 ret = os_snprintf(pos, end - pos,
 542                                   "wps_ap_setup_locked=1\n");
 543                 if (ret < 0 || ret >= end - pos)
 544                         return pos - buf;
 545                 pos += ret;
 546         }
 547
 548         if (attr.selected_registrar && *attr.selected_registrar) {
 549                 ret = os_snprintf(pos, end - pos,
 550                                   "wps_selected_registrar=1\n");
 551                 if (ret < 0 || ret >= end - pos)
 552                         return pos - buf;
 553                 pos += ret;
 554         }
 555
 556         if (attr.dev_password_id) {
 557                 ret = os_snprintf(pos, end - pos,
 558                                   "wps_device_password_id=%u\n",
 559                                   WPA_GET_BE16(attr.dev_password_id));
 560                 if (ret < 0 || ret >= end - pos)
 561                         return pos - buf;
 562                 pos += ret;
 563         }
 564
 565         if (attr.sel_reg_config_methods) {
 566                 ret = os_snprintf(pos, end - pos,
 567                                   "wps_selected_registrar_config_methods="
 568                                   "0x%04x\n",
 569                                   WPA_GET_BE16(attr.sel_reg_config_methods));
 570                 if (ret < 0 || ret >= end - pos)
 571                         return pos - buf;
 572                 pos += ret;
 573         }
 574
 575         if (attr.primary_dev_type) {
 576                 char devtype[WPS_DEV_TYPE_BUFSIZE];
 577                 ret = os_snprintf(pos, end - pos,
 578                                   "wps_primary_device_type=%s\n",
 579                                   wps_dev_type_bin2str(attr.primary_dev_type,
 580                                                        devtype,
 581                                                        sizeof(devtype)));
 582                 if (ret < 0 || ret >= end - pos)
 583                         return pos - buf;
 584                 pos += ret;
 585         }
 586
 587         if (attr.dev_name) {
 588                 char *str = os_malloc(attr.dev_name_len + 1);
 589                 size_t i;
 590                 if (str == NULL)
 591                         return pos - buf;
 592                 for (i = 0; i < attr.dev_name_len; i++) {
 593                         if (attr.dev_name[i] < 32)
 594                                 str[i] = '_';
 595                         else
 596                                 str[i] = attr.dev_name[i];
 597                 }
 598                 str[i] = '\0';
 599                 ret = os_snprintf(pos, end - pos, "wps_device_name=%s\n", str);
 600                 os_free(str);
 601                 if (ret < 0 || ret >= end - pos)
 602                         return pos - buf;
 603                 pos += ret;
 604         }
 605
 606         if (attr.config_methods) {
 607                 ret = os_snprintf(pos, end - pos,
 608                                   "wps_config_methods=0x%04x\n",
 609                                   WPA_GET_BE16(attr.config_methods));
 610                 if (ret < 0 || ret >= end - pos)
 611                         return pos - buf;
 612                 pos += ret;
 613         }
 614
 615         return pos - buf;
 616 }