wpa_supplicant/wps_supplicant.c

   1 /*
   2  * wpa_supplicant / WPS integration
   3  * Copyright (c) 2008-2012, 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 "includes.h"
  10
  11 #include "common.h"
  12 #include "eloop.h"
  13 #include "uuid.h"
  14 #include "crypto/random.h"
  15 #include "crypto/dh_group5.h"
  16 #include "common/ieee802_11_defs.h"
  17 #include "common/ieee802_11_common.h"
  18 #include "common/wpa_common.h"
  19 #include "common/wpa_ctrl.h"
  20 #include "eap_common/eap_wsc_common.h"
  21 #include "eap_peer/eap.h"
  22 #include "eapol_supp/eapol_supp_sm.h"
  23 #include "rsn_supp/wpa.h"
  24 #include "wps/wps_attr_parse.h"
  25 #include "config.h"
  26 #include "wpa_supplicant_i.h"
  27 #include "driver_i.h"
  28 #include "notify.h"
  29 #include "blacklist.h"
  30 #include "bss.h"
  31 #include "scan.h"
  32 #include "ap.h"
  33 #include "p2p/p2p.h"
  34 #include "p2p_supplicant.h"
  35 #include "wps_supplicant.h"
  36
  37
  38 #ifndef WPS_PIN_SCAN_IGNORE_SEL_REG
  39 #define WPS_PIN_SCAN_IGNORE_SEL_REG 3
  40 #endif /* WPS_PIN_SCAN_IGNORE_SEL_REG */
  41
  42 static void wpas_wps_timeout(void *eloop_ctx, void *timeout_ctx);
  43 static void wpas_clear_wps(struct wpa_supplicant *wpa_s);
  44
  45
  46 static void wpas_wps_clear_ap_info(struct wpa_supplicant *wpa_s)
  47 {
  48         os_free(wpa_s->wps_ap);
  49         wpa_s->wps_ap = NULL;
  50         wpa_s->num_wps_ap = 0;
  51         wpa_s->wps_ap_iter = 0;
  52 }
  53
  54
  55 int wpas_wps_eapol_cb(struct wpa_supplicant *wpa_s)
  56 {
  57         if (!wpa_s->wps_success &&
  58             wpa_s->current_ssid &&
  59             eap_is_wps_pin_enrollee(&wpa_s->current_ssid->eap)) {
  60                 const u8 *bssid = wpa_s->bssid;
  61                 if (is_zero_ether_addr(bssid))
  62                         bssid = wpa_s->pending_bssid;
  63
  64                 wpa_printf(MSG_DEBUG, "WPS: PIN registration with " MACSTR
  65                            " did not succeed - continue trying to find "
  66                            "suitable AP", MAC2STR(bssid));
  67                 wpa_blacklist_add(wpa_s, bssid);
  68
  69                 wpa_supplicant_deauthenticate(wpa_s,
  70                                               WLAN_REASON_DEAUTH_LEAVING);
  71                 wpa_s->reassociate = 1;
  72                 wpa_supplicant_req_scan(wpa_s,
  73                                         wpa_s->blacklist_cleared ? 5 : 0, 0);
  74                 wpa_s->blacklist_cleared = 0;
  75                 return 1;
  76         }
  77
  78         wpas_wps_clear_ap_info(wpa_s);
  79         eloop_cancel_timeout(wpas_wps_timeout, wpa_s, NULL);
  80         if (wpa_s->key_mgmt == WPA_KEY_MGMT_WPS && !wpa_s->wps_success)
  81                 wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_FAIL);
  82
  83         if (wpa_s->key_mgmt == WPA_KEY_MGMT_WPS && wpa_s->current_ssid &&
  84             !(wpa_s->current_ssid->key_mgmt & WPA_KEY_MGMT_WPS)) {
  85                 int disabled = wpa_s->current_ssid->disabled;
  86                 unsigned int freq = wpa_s->assoc_freq;
  87                 wpa_printf(MSG_DEBUG, "WPS: Network configuration replaced - "
  88                            "try to associate with the received credential "
  89                            "(freq=%u)", freq);
  90                 wpa_supplicant_deauthenticate(wpa_s,
  91                                               WLAN_REASON_DEAUTH_LEAVING);
  92                 if (disabled) {
  93                         wpa_printf(MSG_DEBUG, "WPS: Current network is "
  94                                    "disabled - wait for user to enable");
  95                         return 1;
  96                 }
  97                 wpa_s->after_wps = 5;
  98                 wpa_s->wps_freq = freq;
  99                 wpa_s->normal_scans = 0;
 100                 wpa_s->reassociate = 1;
 101                 wpa_supplicant_req_scan(wpa_s, 0, 0);
 102                 return 1;
 103         }
 104
 105         if (wpa_s->key_mgmt == WPA_KEY_MGMT_WPS && wpa_s->current_ssid) {
 106                 wpa_printf(MSG_DEBUG, "WPS: Registration completed - waiting "
 107                            "for external credential processing");
 108                 wpas_clear_wps(wpa_s);
 109                 wpa_supplicant_deauthenticate(wpa_s,
 110                                               WLAN_REASON_DEAUTH_LEAVING);
 111                 return 1;
 112         }
 113
 114         return 0;
 115 }
 116
 117
 118 static void wpas_wps_security_workaround(struct wpa_supplicant *wpa_s,
 119                                          struct wpa_ssid *ssid,
 120                                          const struct wps_credential *cred)
 121 {
 122         struct wpa_driver_capa capa;
 123         struct wpa_bss *bss;
 124         const u8 *ie;
 125         struct wpa_ie_data adv;
 126         int wpa2 = 0, ccmp = 0;
 127
 128         /*
 129          * Many existing WPS APs do not know how to negotiate WPA2 or CCMP in
 130          * case they are configured for mixed mode operation (WPA+WPA2 and
 131          * TKIP+CCMP). Try to use scan results to figure out whether the AP
 132          * actually supports stronger security and select that if the client
 133          * has support for it, too.
 134          */
 135
 136         if (wpa_drv_get_capa(wpa_s, &capa))
 137                 return; /* Unknown what driver supports */
 138
 139         if (ssid->ssid == NULL)
 140                 return;
 141         bss = wpa_bss_get(wpa_s, cred->mac_addr, ssid->ssid, ssid->ssid_len);
 142         if (bss == NULL) {
 143                 wpa_printf(MSG_DEBUG, "WPS: The AP was not found from BSS "
 144                            "table - use credential as-is");
 145                 return;
 146         }
 147
 148         wpa_printf(MSG_DEBUG, "WPS: AP found from BSS table");
 149
 150         ie = wpa_bss_get_ie(bss, WLAN_EID_RSN);
 151         if (ie && wpa_parse_wpa_ie(ie, 2 + ie[1], &adv) == 0) {
 152                 wpa2 = 1;
 153                 if (adv.pairwise_cipher & WPA_CIPHER_CCMP)
 154                         ccmp = 1;
 155         } else {
 156                 ie = wpa_bss_get_vendor_ie(bss, WPA_IE_VENDOR_TYPE);
 157                 if (ie && wpa_parse_wpa_ie(ie, 2 + ie[1], &adv) == 0 &&
 158                     adv.pairwise_cipher & WPA_CIPHER_CCMP)
 159                         ccmp = 1;
 160         }
 161
 162         if (ie == NULL && (ssid->proto & WPA_PROTO_WPA) &&
 163             (ssid->pairwise_cipher & WPA_CIPHER_TKIP)) {
 164                 /*
 165                  * TODO: This could be the initial AP configuration and the
 166                  * Beacon contents could change shortly. Should request a new
 167                  * scan and delay addition of the network until the updated
 168                  * scan results are available.
 169                  */
 170                 wpa_printf(MSG_DEBUG, "WPS: The AP did not yet advertise WPA "
 171                            "support - use credential as-is");
 172                 return;
 173         }
 174
 175         if (ccmp && !(ssid->pairwise_cipher & WPA_CIPHER_CCMP) &&
 176             (ssid->pairwise_cipher & WPA_CIPHER_TKIP) &&
 177             (capa.key_mgmt & WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK)) {
 178                 wpa_printf(MSG_DEBUG, "WPS: Add CCMP into the credential "
 179                            "based on scan results");
 180                 if (wpa_s->conf->ap_scan == 1)
 181                         ssid->pairwise_cipher |= WPA_CIPHER_CCMP;
 182                 else
 183                         ssid->pairwise_cipher = WPA_CIPHER_CCMP;
 184         }
 185
 186         if (wpa2 && !(ssid->proto & WPA_PROTO_RSN) &&
 187             (ssid->proto & WPA_PROTO_WPA) &&
 188             (capa.enc & WPA_DRIVER_CAPA_ENC_CCMP)) {
 189                 wpa_printf(MSG_DEBUG, "WPS: Add WPA2 into the credential "
 190                            "based on scan results");
 191                 if (wpa_s->conf->ap_scan == 1)
 192                         ssid->proto |= WPA_PROTO_RSN;
 193                 else
 194                         ssid->proto = WPA_PROTO_RSN;
 195         }
 196 }
 197
 198
 199 static int wpa_supplicant_wps_cred(void *ctx,
 200                                    const struct wps_credential *cred)
 201 {
 202         struct wpa_supplicant *wpa_s = ctx;
 203         struct wpa_ssid *ssid = wpa_s->current_ssid;
 204         u8 key_idx = 0;
 205         u16 auth_type;
 206 #ifdef CONFIG_WPS_REG_DISABLE_OPEN
 207         int registrar = 0;
 208 #endif /* CONFIG_WPS_REG_DISABLE_OPEN */
 209
 210         if ((wpa_s->conf->wps_cred_processing == 1 ||
 211              wpa_s->conf->wps_cred_processing == 2) && cred->cred_attr) {
 212                 size_t blen = cred->cred_attr_len * 2 + 1;
 213                 char *buf = os_malloc(blen);
 214                 if (buf) {
 215                         wpa_snprintf_hex(buf, blen,
 216                                          cred->cred_attr, cred->cred_attr_len);
 217                         wpa_msg(wpa_s, MSG_INFO, "%s%s",
 218                                 WPS_EVENT_CRED_RECEIVED, buf);
 219                         os_free(buf);
 220                 }
 221
 222                 wpas_notify_wps_credential(wpa_s, cred);
 223         } else
 224                 wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_CRED_RECEIVED);
 225
 226         wpa_hexdump_key(MSG_DEBUG, "WPS: Received Credential attribute",
 227                         cred->cred_attr, cred->cred_attr_len);
 228
 229         if (wpa_s->conf->wps_cred_processing == 1)
 230                 return 0;
 231
 232         wpa_hexdump_ascii(MSG_DEBUG, "WPS: SSID", cred->ssid, cred->ssid_len);
 233         wpa_printf(MSG_DEBUG, "WPS: Authentication Type 0x%x",
 234                    cred->auth_type);
 235         wpa_printf(MSG_DEBUG, "WPS: Encryption Type 0x%x", cred->encr_type);
 236         wpa_printf(MSG_DEBUG, "WPS: Network Key Index %d", cred->key_idx);
 237         wpa_hexdump_key(MSG_DEBUG, "WPS: Network Key",
 238                         cred->key, cred->key_len);
 239         wpa_printf(MSG_DEBUG, "WPS: MAC Address " MACSTR,
 240                    MAC2STR(cred->mac_addr));
 241
 242         auth_type = cred->auth_type;
 243         if (auth_type == (WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK)) {
 244                 wpa_printf(MSG_DEBUG, "WPS: Workaround - convert mixed-mode "
 245                            "auth_type into WPA2PSK");
 246                 auth_type = WPS_AUTH_WPA2PSK;
 247         }
 248
 249         if (auth_type != WPS_AUTH_OPEN &&
 250             auth_type != WPS_AUTH_SHARED &&
 251             auth_type != WPS_AUTH_WPAPSK &&
 252             auth_type != WPS_AUTH_WPA2PSK) {
 253                 wpa_printf(MSG_DEBUG, "WPS: Ignored credentials for "
 254                            "unsupported authentication type 0x%x",
 255                            auth_type);
 256                 return 0;
 257         }
 258
 259         if (auth_type == WPS_AUTH_WPAPSK || auth_type == WPS_AUTH_WPA2PSK) {
 260                 if (cred->key_len < 8 || cred->key_len > 2 * PMK_LEN) {
 261                         wpa_printf(MSG_ERROR, "WPS: Reject PSK credential with "
 262                                    "invalid Network Key length %lu",
 263                                    (unsigned long) cred->key_len);
 264                         return -1;
 265                 }
 266         }
 267
 268         if (ssid && (ssid->key_mgmt & WPA_KEY_MGMT_WPS)) {
 269                 wpa_printf(MSG_DEBUG, "WPS: Replace WPS network block based "
 270                            "on the received credential");
 271 #ifdef CONFIG_WPS_REG_DISABLE_OPEN
 272                 if (ssid->eap.identity &&
 273                     ssid->eap.identity_len == WSC_ID_REGISTRAR_LEN &&
 274                     os_memcmp(ssid->eap.identity, WSC_ID_REGISTRAR,
 275                               WSC_ID_REGISTRAR_LEN) == 0)
 276                         registrar = 1;
 277 #endif /* CONFIG_WPS_REG_DISABLE_OPEN */
 278                 os_free(ssid->eap.identity);
 279                 ssid->eap.identity = NULL;
 280                 ssid->eap.identity_len = 0;
 281                 os_free(ssid->eap.phase1);
 282                 ssid->eap.phase1 = NULL;
 283                 os_free(ssid->eap.eap_methods);
 284                 ssid->eap.eap_methods = NULL;
 285                 if (!ssid->p2p_group) {
 286                         ssid->temporary = 0;
 287                         ssid->bssid_set = 0;
 288                 }
 289                 ssid->disabled_until.sec = 0;
 290                 ssid->disabled_until.usec = 0;
 291                 ssid->auth_failures = 0;
 292         } else {
 293                 wpa_printf(MSG_DEBUG, "WPS: Create a new network based on the "
 294                            "received credential");
 295                 ssid = wpa_config_add_network(wpa_s->conf);
 296                 if (ssid == NULL)
 297                         return -1;
 298                 wpas_notify_network_added(wpa_s, ssid);
 299         }
 300
 301         wpa_config_set_network_defaults(ssid);
 302
 303         os_free(ssid->ssid);
 304         ssid->ssid = os_malloc(cred->ssid_len);
 305         if (ssid->ssid) {
 306                 os_memcpy(ssid->ssid, cred->ssid, cred->ssid_len);
 307                 ssid->ssid_len = cred->ssid_len;
 308         }
 309
 310         switch (cred->encr_type) {
 311         case WPS_ENCR_NONE:
 312                 break;
 313         case WPS_ENCR_WEP:
 314                 if (cred->key_len <= 0)
 315                         break;
 316                 if (cred->key_len != 5 && cred->key_len != 13 &&
 317                     cred->key_len != 10 && cred->key_len != 26) {
 318                         wpa_printf(MSG_ERROR, "WPS: Invalid WEP Key length "
 319                                    "%lu", (unsigned long) cred->key_len);
 320                         return -1;
 321                 }
 322                 if (cred->key_idx > NUM_WEP_KEYS) {
 323                         wpa_printf(MSG_ERROR, "WPS: Invalid WEP Key index %d",
 324                                    cred->key_idx);
 325                         return -1;
 326                 }
 327                 if (cred->key_idx)
 328                         key_idx = cred->key_idx - 1;
 329                 if (cred->key_len == 10 || cred->key_len == 26) {
 330                         if (hexstr2bin((char *) cred->key,
 331                                        ssid->wep_key[key_idx],
 332                                        cred->key_len / 2) < 0) {
 333                                 wpa_printf(MSG_ERROR, "WPS: Invalid WEP Key "
 334                                            "%d", key_idx);
 335                                 return -1;
 336                         }
 337                         ssid->wep_key_len[key_idx] = cred->key_len / 2;
 338                 } else {
 339                         os_memcpy(ssid->wep_key[key_idx], cred->key,
 340                                   cred->key_len);
 341                         ssid->wep_key_len[key_idx] = cred->key_len;
 342                 }
 343                 ssid->wep_tx_keyidx = key_idx;
 344                 break;
 345         case WPS_ENCR_TKIP:
 346                 ssid->pairwise_cipher = WPA_CIPHER_TKIP;
 347                 break;
 348         case WPS_ENCR_AES:
 349                 ssid->pairwise_cipher = WPA_CIPHER_CCMP;
 350                 break;
 351         }
 352
 353         switch (auth_type) {
 354         case WPS_AUTH_OPEN:
 355                 ssid->auth_alg = WPA_AUTH_ALG_OPEN;
 356                 ssid->key_mgmt = WPA_KEY_MGMT_NONE;
 357                 ssid->proto = 0;
 358 #ifdef CONFIG_WPS_REG_DISABLE_OPEN
 359                 if (registrar) {
 360                         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_OPEN_NETWORK
 361                                 "id=%d - Credentials for an open "
 362                                 "network disabled by default - use "
 363                                 "'select_network %d' to enable",
 364                                 ssid->id, ssid->id);
 365                         ssid->disabled = 1;
 366                 }
 367 #endif /* CONFIG_WPS_REG_DISABLE_OPEN */
 368                 break;
 369         case WPS_AUTH_SHARED:
 370                 ssid->auth_alg = WPA_AUTH_ALG_SHARED;
 371                 ssid->key_mgmt = WPA_KEY_MGMT_NONE;
 372                 ssid->proto = 0;
 373                 break;
 374         case WPS_AUTH_WPAPSK:
 375                 ssid->auth_alg = WPA_AUTH_ALG_OPEN;
 376                 ssid->key_mgmt = WPA_KEY_MGMT_PSK;
 377                 ssid->proto = WPA_PROTO_WPA;
 378                 break;
 379         case WPS_AUTH_WPA2PSK:
 380                 ssid->auth_alg = WPA_AUTH_ALG_OPEN;
 381                 ssid->key_mgmt = WPA_KEY_MGMT_PSK;
 382                 ssid->proto = WPA_PROTO_RSN;
 383                 break;
 384         }
 385
 386         if (ssid->key_mgmt == WPA_KEY_MGMT_PSK) {
 387                 if (cred->key_len == 2 * PMK_LEN) {
 388                         if (hexstr2bin((const char *) cred->key, ssid->psk,
 389                                        PMK_LEN)) {
 390                                 wpa_printf(MSG_ERROR, "WPS: Invalid Network "
 391                                            "Key");
 392                                 return -1;
 393                         }
 394                         ssid->psk_set = 1;
 395                         ssid->export_keys = 1;
 396                 } else if (cred->key_len >= 8 && cred->key_len < 2 * PMK_LEN) {
 397                         os_free(ssid->passphrase);
 398                         ssid->passphrase = os_malloc(cred->key_len + 1);
 399                         if (ssid->passphrase == NULL)
 400                                 return -1;
 401                         os_memcpy(ssid->passphrase, cred->key, cred->key_len);
 402                         ssid->passphrase[cred->key_len] = '\0';
 403                         wpa_config_update_psk(ssid);
 404                         ssid->export_keys = 1;
 405                 } else {
 406                         wpa_printf(MSG_ERROR, "WPS: Invalid Network Key "
 407                                    "length %lu",
 408                                    (unsigned long) cred->key_len);
 409                         return -1;
 410                 }
 411         }
 412
 413         wpas_wps_security_workaround(wpa_s, ssid, cred);
 414
 415         if (cred->ap_channel)
 416                 wpa_s->wps_ap_channel = cred->ap_channel;
 417
 418 #ifndef CONFIG_NO_CONFIG_WRITE
 419         if (wpa_s->conf->update_config &&
 420             wpa_config_write(wpa_s->confname, wpa_s->conf)) {
 421                 wpa_printf(MSG_DEBUG, "WPS: Failed to update configuration");
 422                 return -1;
 423         }
 424 #endif /* CONFIG_NO_CONFIG_WRITE */
 425
 426         return 0;
 427 }
 428
 429
 430 #ifdef CONFIG_P2P
 431 static void wpas_wps_pbc_overlap_cb(void *eloop_ctx, void *timeout_ctx)
 432 {
 433         struct wpa_supplicant *wpa_s = eloop_ctx;
 434         wpas_p2p_notif_pbc_overlap(wpa_s);
 435 }
 436 #endif /* CONFIG_P2P */
 437
 438
 439 static void wpa_supplicant_wps_event_m2d(struct wpa_supplicant *wpa_s,
 440                                          struct wps_event_m2d *m2d)
 441 {
 442         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_M2D
 443                 "dev_password_id=%d config_error=%d",
 444                 m2d->dev_password_id, m2d->config_error);
 445         wpas_notify_wps_event_m2d(wpa_s, m2d);
 446 #ifdef CONFIG_P2P
 447         if (wpa_s->parent && wpa_s->parent != wpa_s) {
 448                 wpa_msg(wpa_s->parent, MSG_INFO, WPS_EVENT_M2D
 449                         "dev_password_id=%d config_error=%d",
 450                         m2d->dev_password_id, m2d->config_error);
 451         }
 452         if (m2d->config_error == WPS_CFG_MULTIPLE_PBC_DETECTED) {
 453                 /*
 454                  * Notify P2P from eloop timeout to avoid issues with the
 455                  * interface getting removed while processing a message.
 456                  */
 457                 eloop_register_timeout(0, 0, wpas_wps_pbc_overlap_cb, wpa_s,
 458                                        NULL);
 459         }
 460 #endif /* CONFIG_P2P */
 461 }
 462
 463
 464 static const char * wps_event_fail_reason[NUM_WPS_EI_VALUES] = {
 465         "No Error", /* WPS_EI_NO_ERROR */
 466         "TKIP Only Prohibited", /* WPS_EI_SECURITY_TKIP_ONLY_PROHIBITED */
 467         "WEP Prohibited" /* WPS_EI_SECURITY_WEP_PROHIBITED */
 468 };
 469
 470 static void wpa_supplicant_wps_event_fail(struct wpa_supplicant *wpa_s,
 471                                           struct wps_event_fail *fail)
 472 {
 473         if (fail->error_indication > 0 &&
 474             fail->error_indication < NUM_WPS_EI_VALUES) {
 475                 wpa_msg(wpa_s, MSG_INFO,
 476                         WPS_EVENT_FAIL "msg=%d config_error=%d reason=%d (%s)",
 477                         fail->msg, fail->config_error, fail->error_indication,
 478                         wps_event_fail_reason[fail->error_indication]);
 479                 if (wpa_s->parent && wpa_s->parent != wpa_s)
 480                         wpa_msg(wpa_s->parent, MSG_INFO, WPS_EVENT_FAIL
 481                                 "msg=%d config_error=%d reason=%d (%s)",
 482                                 fail->msg, fail->config_error,
 483                                 fail->error_indication,
 484                                 wps_event_fail_reason[fail->error_indication]);
 485         } else {
 486                 wpa_msg(wpa_s, MSG_INFO,
 487                         WPS_EVENT_FAIL "msg=%d config_error=%d",
 488                         fail->msg, fail->config_error);
 489                 if (wpa_s->parent && wpa_s->parent != wpa_s)
 490                         wpa_msg(wpa_s->parent, MSG_INFO, WPS_EVENT_FAIL
 491                                 "msg=%d config_error=%d",
 492                                 fail->msg, fail->config_error);
 493         }
 494         wpas_clear_wps(wpa_s);
 495         wpas_notify_wps_event_fail(wpa_s, fail);
 496 #ifdef CONFIG_P2P
 497         wpas_p2p_wps_failed(wpa_s, fail);
 498 #endif /* CONFIG_P2P */
 499 }
 500
 501
 502 static void wpas_wps_reenable_networks_cb(void *eloop_ctx, void *timeout_ctx);
 503
 504 static void wpas_wps_reenable_networks(struct wpa_supplicant *wpa_s)
 505 {
 506         struct wpa_ssid *ssid;
 507
 508         eloop_cancel_timeout(wpas_wps_reenable_networks_cb, wpa_s, NULL);
 509
 510         for (ssid = wpa_s->conf->ssid; ssid; ssid = ssid->next) {
 511                 if (ssid->disabled_for_connect && ssid->disabled) {
 512                         ssid->disabled_for_connect = 0;
 513                         ssid->disabled = 0;
 514                         wpas_notify_network_enabled_changed(wpa_s, ssid);
 515                 }
 516         }
 517 }
 518
 519
 520 static void wpas_wps_reenable_networks_cb(void *eloop_ctx, void *timeout_ctx)
 521 {
 522         struct wpa_supplicant *wpa_s = eloop_ctx;
 523         /* Enable the networks disabled during wpas_wps_reassoc */
 524         wpas_wps_reenable_networks(wpa_s);
 525 }
 526
 527
 528 static void wpa_supplicant_wps_event_success(struct wpa_supplicant *wpa_s)
 529 {
 530         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_SUCCESS);
 531         wpa_s->wps_success = 1;
 532         wpas_notify_wps_event_success(wpa_s);
 533
 534         /*
 535          * Enable the networks disabled during wpas_wps_reassoc after 10
 536          * seconds. The 10 seconds timer is to allow the data connection to be
 537          * formed before allowing other networks to be selected.
 538          */
 539         eloop_register_timeout(10, 0, wpas_wps_reenable_networks_cb, wpa_s,
 540                                NULL);
 541
 542 #ifdef CONFIG_P2P
 543         wpas_p2p_wps_success(wpa_s, wpa_s->bssid, 0);
 544 #endif /* CONFIG_P2P */
 545 }
 546
 547
 548 static void wpa_supplicant_wps_event_er_ap_add(struct wpa_supplicant *wpa_s,
 549                                                struct wps_event_er_ap *ap)
 550 {
 551         char uuid_str[100];
 552         char dev_type[WPS_DEV_TYPE_BUFSIZE];
 553
 554         uuid_bin2str(ap->uuid, uuid_str, sizeof(uuid_str));
 555         if (ap->pri_dev_type)
 556                 wps_dev_type_bin2str(ap->pri_dev_type, dev_type,
 557                                      sizeof(dev_type));
 558         else
 559                 dev_type[0] = '\0';
 560
 561         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_AP_ADD "%s " MACSTR
 562                 " pri_dev_type=%s wps_state=%d |%s|%s|%s|%s|%s|%s|",
 563                 uuid_str, MAC2STR(ap->mac_addr), dev_type, ap->wps_state,
 564                 ap->friendly_name ? ap->friendly_name : "",
 565                 ap->manufacturer ? ap->manufacturer : "",
 566                 ap->model_description ? ap->model_description : "",
 567                 ap->model_name ? ap->model_name : "",
 568                 ap->manufacturer_url ? ap->manufacturer_url : "",
 569                 ap->model_url ? ap->model_url : "");
 570 }
 571
 572
 573 static void wpa_supplicant_wps_event_er_ap_remove(struct wpa_supplicant *wpa_s,
 574                                                   struct wps_event_er_ap *ap)
 575 {
 576         char uuid_str[100];
 577         uuid_bin2str(ap->uuid, uuid_str, sizeof(uuid_str));
 578         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_AP_REMOVE "%s", uuid_str);
 579 }
 580
 581
 582 static void wpa_supplicant_wps_event_er_enrollee_add(
 583         struct wpa_supplicant *wpa_s, struct wps_event_er_enrollee *enrollee)
 584 {
 585         char uuid_str[100];
 586         char dev_type[WPS_DEV_TYPE_BUFSIZE];
 587
 588         uuid_bin2str(enrollee->uuid, uuid_str, sizeof(uuid_str));
 589         if (enrollee->pri_dev_type)
 590                 wps_dev_type_bin2str(enrollee->pri_dev_type, dev_type,
 591                                      sizeof(dev_type));
 592         else
 593                 dev_type[0] = '\0';
 594
 595         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_ENROLLEE_ADD "%s " MACSTR
 596                 " M1=%d config_methods=0x%x dev_passwd_id=%d pri_dev_type=%s "
 597                 "|%s|%s|%s|%s|%s|",
 598                 uuid_str, MAC2STR(enrollee->mac_addr), enrollee->m1_received,
 599                 enrollee->config_methods, enrollee->dev_passwd_id, dev_type,
 600                 enrollee->dev_name ? enrollee->dev_name : "",
 601                 enrollee->manufacturer ? enrollee->manufacturer : "",
 602                 enrollee->model_name ? enrollee->model_name : "",
 603                 enrollee->model_number ? enrollee->model_number : "",
 604                 enrollee->serial_number ? enrollee->serial_number : "");
 605 }
 606
 607
 608 static void wpa_supplicant_wps_event_er_enrollee_remove(
 609         struct wpa_supplicant *wpa_s, struct wps_event_er_enrollee *enrollee)
 610 {
 611         char uuid_str[100];
 612         uuid_bin2str(enrollee->uuid, uuid_str, sizeof(uuid_str));
 613         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_ENROLLEE_REMOVE "%s " MACSTR,
 614                 uuid_str, MAC2STR(enrollee->mac_addr));
 615 }
 616
 617
 618 static void wpa_supplicant_wps_event_er_ap_settings(
 619         struct wpa_supplicant *wpa_s,
 620         struct wps_event_er_ap_settings *ap_settings)
 621 {
 622         char uuid_str[100];
 623         char key_str[65];
 624         const struct wps_credential *cred = ap_settings->cred;
 625
 626         key_str[0] = '\0';
 627         if (cred->auth_type & (WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK)) {
 628                 if (cred->key_len >= 8 && cred->key_len <= 64) {
 629                         os_memcpy(key_str, cred->key, cred->key_len);
 630                         key_str[cred->key_len] = '\0';
 631                 }
 632         }
 633
 634         uuid_bin2str(ap_settings->uuid, uuid_str, sizeof(uuid_str));
 635         /* Use wpa_msg_ctrl to avoid showing the key in debug log */
 636         wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_ER_AP_SETTINGS
 637                      "uuid=%s ssid=%s auth_type=0x%04x encr_type=0x%04x "
 638                      "key=%s",
 639                      uuid_str, wpa_ssid_txt(cred->ssid, cred->ssid_len),
 640                      cred->auth_type, cred->encr_type, key_str);
 641 }
 642
 643
 644 static void wpa_supplicant_wps_event_er_set_sel_reg(
 645         struct wpa_supplicant *wpa_s,
 646         struct wps_event_er_set_selected_registrar *ev)
 647 {
 648         char uuid_str[100];
 649
 650         uuid_bin2str(ev->uuid, uuid_str, sizeof(uuid_str));
 651         switch (ev->state) {
 652         case WPS_ER_SET_SEL_REG_START:
 653                 wpa_msg(wpa_s, MSG_DEBUG, WPS_EVENT_ER_SET_SEL_REG
 654                         "uuid=%s state=START sel_reg=%d dev_passwd_id=%u "
 655                         "sel_reg_config_methods=0x%x",
 656                         uuid_str, ev->sel_reg, ev->dev_passwd_id,
 657                         ev->sel_reg_config_methods);
 658                 break;
 659         case WPS_ER_SET_SEL_REG_DONE:
 660                 wpa_msg(wpa_s, MSG_DEBUG, WPS_EVENT_ER_SET_SEL_REG
 661                         "uuid=%s state=DONE", uuid_str);
 662                 break;
 663         case WPS_ER_SET_SEL_REG_FAILED:
 664                 wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_SET_SEL_REG
 665                         "uuid=%s state=FAILED", uuid_str);
 666                 break;
 667         }
 668 }
 669
 670
 671 static void wpa_supplicant_wps_event(void *ctx, enum wps_event event,
 672                                      union wps_event_data *data)
 673 {
 674         struct wpa_supplicant *wpa_s = ctx;
 675         switch (event) {
 676         case WPS_EV_M2D:
 677                 wpa_supplicant_wps_event_m2d(wpa_s, &data->m2d);
 678                 break;
 679         case WPS_EV_FAIL:
 680                 wpa_supplicant_wps_event_fail(wpa_s, &data->fail);
 681                 break;
 682         case WPS_EV_SUCCESS:
 683                 wpa_supplicant_wps_event_success(wpa_s);
 684                 break;
 685         case WPS_EV_PWD_AUTH_FAIL:
 686 #ifdef CONFIG_AP
 687                 if (wpa_s->ap_iface && data->pwd_auth_fail.enrollee)
 688                         wpa_supplicant_ap_pwd_auth_fail(wpa_s);
 689 #endif /* CONFIG_AP */
 690                 break;
 691         case WPS_EV_PBC_OVERLAP:
 692                 break;
 693         case WPS_EV_PBC_TIMEOUT:
 694                 break;
 695         case WPS_EV_ER_AP_ADD:
 696                 wpa_supplicant_wps_event_er_ap_add(wpa_s, &data->ap);
 697                 break;
 698         case WPS_EV_ER_AP_REMOVE:
 699                 wpa_supplicant_wps_event_er_ap_remove(wpa_s, &data->ap);
 700                 break;
 701         case WPS_EV_ER_ENROLLEE_ADD:
 702                 wpa_supplicant_wps_event_er_enrollee_add(wpa_s,
 703                                                          &data->enrollee);
 704                 break;
 705         case WPS_EV_ER_ENROLLEE_REMOVE:
 706                 wpa_supplicant_wps_event_er_enrollee_remove(wpa_s,
 707                                                             &data->enrollee);
 708                 break;
 709         case WPS_EV_ER_AP_SETTINGS:
 710                 wpa_supplicant_wps_event_er_ap_settings(wpa_s,
 711                                                         &data->ap_settings);
 712                 break;
 713         case WPS_EV_ER_SET_SELECTED_REGISTRAR:
 714                 wpa_supplicant_wps_event_er_set_sel_reg(wpa_s,
 715                                                         &data->set_sel_reg);
 716                 break;
 717         case WPS_EV_AP_PIN_SUCCESS:
 718                 break;
 719         }
 720 }
 721
 722
 723 enum wps_request_type wpas_wps_get_req_type(struct wpa_ssid *ssid)
 724 {
 725         if (eap_is_wps_pbc_enrollee(&ssid->eap) ||
 726             eap_is_wps_pin_enrollee(&ssid->eap))
 727                 return WPS_REQ_ENROLLEE;
 728         else
 729                 return WPS_REQ_REGISTRAR;
 730 }
 731
 732
 733 static void wpas_clear_wps(struct wpa_supplicant *wpa_s)
 734 {
 735         int id;
 736         struct wpa_ssid *ssid, *remove_ssid = NULL, *prev_current;
 737
 738         prev_current = wpa_s->current_ssid;
 739
 740         /* Enable the networks disabled during wpas_wps_reassoc */
 741         wpas_wps_reenable_networks(wpa_s);
 742
 743         eloop_cancel_timeout(wpas_wps_timeout, wpa_s, NULL);
 744
 745         /* Remove any existing WPS network from configuration */
 746         ssid = wpa_s->conf->ssid;
 747         while (ssid) {
 748                 if (ssid->key_mgmt & WPA_KEY_MGMT_WPS) {
 749                         if (ssid == wpa_s->current_ssid) {
 750                                 wpa_s->current_ssid = NULL;
 751                                 if (ssid != NULL)
 752                                         wpas_notify_network_changed(wpa_s);
 753                         }
 754                         id = ssid->id;
 755                         remove_ssid = ssid;
 756                 } else
 757                         id = -1;
 758                 ssid = ssid->next;
 759                 if (id >= 0) {
 760                         if (prev_current == remove_ssid) {
 761                                 wpa_sm_set_config(wpa_s->wpa, NULL);
 762                                 eapol_sm_notify_config(wpa_s->eapol, NULL,
 763                                                        NULL);
 764                         }
 765                         wpas_notify_network_removed(wpa_s, remove_ssid);
 766                         wpa_config_remove_network(wpa_s->conf, id);
 767                 }
 768         }
 769
 770         wpas_wps_clear_ap_info(wpa_s);
 771 }
 772
 773
 774 static void wpas_wps_timeout(void *eloop_ctx, void *timeout_ctx)
 775 {
 776         struct wpa_supplicant *wpa_s = eloop_ctx;
 777         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_TIMEOUT "Requested operation timed "
 778                 "out");
 779         wpas_clear_wps(wpa_s);
 780 }
 781
 782
 783 static struct wpa_ssid * wpas_wps_add_network(struct wpa_supplicant *wpa_s,
 784                                               int registrar, const u8 *bssid)
 785 {
 786         struct wpa_ssid *ssid;
 787
 788         ssid = wpa_config_add_network(wpa_s->conf);
 789         if (ssid == NULL)
 790                 return NULL;
 791         wpas_notify_network_added(wpa_s, ssid);
 792         wpa_config_set_network_defaults(ssid);
 793         ssid->temporary = 1;
 794         if (wpa_config_set(ssid, "key_mgmt", "WPS", 0) < 0 ||
 795             wpa_config_set(ssid, "eap", "WSC", 0) < 0 ||
 796             wpa_config_set(ssid, "identity", registrar ?
 797                            "\"" WSC_ID_REGISTRAR "\"" :
 798                            "\"" WSC_ID_ENROLLEE "\"", 0) < 0) {
 799                 wpas_notify_network_removed(wpa_s, ssid);
 800                 wpa_config_remove_network(wpa_s->conf, ssid->id);
 801                 return NULL;
 802         }
 803
 804         if (bssid) {
 805 #ifndef CONFIG_P2P
 806                 struct wpa_bss *bss;
 807                 int count = 0;
 808 #endif /* CONFIG_P2P */
 809
 810                 os_memcpy(ssid->bssid, bssid, ETH_ALEN);
 811                 ssid->bssid_set = 1;
 812
 813                 /*
 814                  * Note: With P2P, the SSID may change at the time the WPS
 815                  * provisioning is started, so better not filter the AP based
 816                  * on the current SSID in the scan results.
 817                  */
 818 #ifndef CONFIG_P2P
 819                 dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
 820                         if (os_memcmp(bssid, bss->bssid, ETH_ALEN) != 0)
 821                                 continue;
 822
 823                         os_free(ssid->ssid);
 824                         ssid->ssid = os_malloc(bss->ssid_len);
 825                         if (ssid->ssid == NULL)
 826                                 break;
 827                         os_memcpy(ssid->ssid, bss->ssid, bss->ssid_len);
 828                         ssid->ssid_len = bss->ssid_len;
 829                         wpa_hexdump_ascii(MSG_DEBUG, "WPS: Picked SSID from "
 830                                           "scan results",
 831                                           ssid->ssid, ssid->ssid_len);
 832                         count++;
 833                 }
 834
 835                 if (count > 1) {
 836                         wpa_printf(MSG_DEBUG, "WPS: More than one SSID found "
 837                                    "for the AP; use wildcard");
 838                         os_free(ssid->ssid);
 839                         ssid->ssid = NULL;
 840                         ssid->ssid_len = 0;
 841                 }
 842 #endif /* CONFIG_P2P */
 843         }
 844
 845         return ssid;
 846 }
 847
 848
 849 static void wpas_wps_reassoc(struct wpa_supplicant *wpa_s,
 850                              struct wpa_ssid *selected, const u8 *bssid)
 851 {
 852         struct wpa_ssid *ssid;
 853         struct wpa_bss *bss;
 854
 855         wpa_s->known_wps_freq = 0;
 856         if (bssid) {
 857                 bss = wpa_bss_get_bssid(wpa_s, bssid);
 858                 if (bss && bss->freq > 0) {
 859                         wpa_s->known_wps_freq = 1;
 860                         wpa_s->wps_freq = bss->freq;
 861                 }
 862         }
 863
 864         if (wpa_s->current_ssid)
 865                 wpa_supplicant_deauthenticate(
 866                         wpa_s, WLAN_REASON_DEAUTH_LEAVING);
 867
 868         /* Mark all other networks disabled and trigger reassociation */
 869         ssid = wpa_s->conf->ssid;
 870         while (ssid) {
 871                 int was_disabled = ssid->disabled;
 872                 ssid->disabled_for_connect = 0;
 873                 /*
 874                  * In case the network object corresponds to a persistent group
 875                  * then do not send out network disabled signal. In addition,
 876                  * do not change disabled status of persistent network objects
 877                  * from 2 to 1 should we connect to another network.
 878                  */
 879                 if (was_disabled != 2) {
 880                         ssid->disabled = ssid != selected;
 881                         if (was_disabled != ssid->disabled) {
 882                                 if (ssid->disabled)
 883                                         ssid->disabled_for_connect = 1;
 884                                 wpas_notify_network_enabled_changed(wpa_s,
 885                                                                     ssid);
 886                         }
 887                 }
 888                 ssid = ssid->next;
 889         }
 890         wpa_s->disconnected = 0;
 891         wpa_s->reassociate = 1;
 892         wpa_s->scan_runs = 0;
 893         wpa_s->normal_scans = 0;
 894         wpa_s->wps_success = 0;
 895         wpa_s->blacklist_cleared = 0;
 896         wpa_supplicant_req_scan(wpa_s, 0, 0);
 897 }
 898
 899
 900 int wpas_wps_start_pbc(struct wpa_supplicant *wpa_s, const u8 *bssid,
 901                        int p2p_group)
 902 {
 903         struct wpa_ssid *ssid;
 904         wpas_clear_wps(wpa_s);
 905         ssid = wpas_wps_add_network(wpa_s, 0, bssid);
 906         if (ssid == NULL)
 907                 return -1;
 908         ssid->temporary = 1;
 909         ssid->p2p_group = p2p_group;
 910 #ifdef CONFIG_P2P
 911         if (p2p_group && wpa_s->go_params && wpa_s->go_params->ssid_len) {
 912                 ssid->ssid = os_zalloc(wpa_s->go_params->ssid_len + 1);
 913                 if (ssid->ssid) {
 914                         ssid->ssid_len = wpa_s->go_params->ssid_len;
 915                         os_memcpy(ssid->ssid, wpa_s->go_params->ssid,
 916                                   ssid->ssid_len);
 917                         wpa_hexdump_ascii(MSG_DEBUG, "WPS: Use specific AP "
 918                                           "SSID", ssid->ssid, ssid->ssid_len);
 919                 }
 920         }
 921 #endif /* CONFIG_P2P */
 922         wpa_config_set(ssid, "phase1", "\"pbc=1\"", 0);
 923         if (wpa_s->wps_fragment_size)
 924                 ssid->eap.fragment_size = wpa_s->wps_fragment_size;
 925         eloop_register_timeout(WPS_PBC_WALK_TIME, 0, wpas_wps_timeout,
 926                                wpa_s, NULL);
 927         wpas_wps_reassoc(wpa_s, ssid, bssid);
 928         return 0;
 929 }
 930
 931
 932 int wpas_wps_start_pin(struct wpa_supplicant *wpa_s, const u8 *bssid,
 933                        const char *pin, int p2p_group, u16 dev_pw_id)
 934 {
 935         struct wpa_ssid *ssid;
 936         char val[128];
 937         unsigned int rpin = 0;
 938
 939         wpas_clear_wps(wpa_s);
 940         ssid = wpas_wps_add_network(wpa_s, 0, bssid);
 941         if (ssid == NULL)
 942                 return -1;
 943         ssid->temporary = 1;
 944         ssid->p2p_group = p2p_group;
 945 #ifdef CONFIG_P2P
 946         if (p2p_group && wpa_s->go_params && wpa_s->go_params->ssid_len) {
 947                 ssid->ssid = os_zalloc(wpa_s->go_params->ssid_len + 1);
 948                 if (ssid->ssid) {
 949                         ssid->ssid_len = wpa_s->go_params->ssid_len;
 950                         os_memcpy(ssid->ssid, wpa_s->go_params->ssid,
 951                                   ssid->ssid_len);
 952                         wpa_hexdump_ascii(MSG_DEBUG, "WPS: Use specific AP "
 953                                           "SSID", ssid->ssid, ssid->ssid_len);
 954                 }
 955         }
 956 #endif /* CONFIG_P2P */
 957         if (pin)
 958                 os_snprintf(val, sizeof(val), "\"pin=%s dev_pw_id=%u\"",
 959                             pin, dev_pw_id);
 960         else {
 961                 rpin = wps_generate_pin();
 962                 os_snprintf(val, sizeof(val), "\"pin=%08d dev_pw_id=%u\"",
 963                             rpin, dev_pw_id);
 964         }
 965         wpa_config_set(ssid, "phase1", val, 0);
 966         if (wpa_s->wps_fragment_size)
 967                 ssid->eap.fragment_size = wpa_s->wps_fragment_size;
 968         eloop_register_timeout(WPS_PBC_WALK_TIME, 0, wpas_wps_timeout,
 969                                wpa_s, NULL);
 970         wpa_s->wps_ap_iter = 1;
 971         wpas_wps_reassoc(wpa_s, ssid, bssid);
 972         return rpin;
 973 }
 974
 975
 976 /* Cancel the wps pbc/pin requests */
 977 int wpas_wps_cancel(struct wpa_supplicant *wpa_s)
 978 {
 979 #ifdef CONFIG_AP
 980         if (wpa_s->ap_iface) {
 981                 wpa_printf(MSG_DEBUG, "WPS: Cancelling in AP mode");
 982                 return wpa_supplicant_ap_wps_cancel(wpa_s);
 983         }
 984 #endif /* CONFIG_AP */
 985
 986         if (wpa_s->wpa_state == WPA_SCANNING ||
 987             wpa_s->wpa_state == WPA_DISCONNECTED) {
 988                 wpa_printf(MSG_DEBUG, "WPS: Cancel operation - cancel scan");
 989                 wpa_supplicant_cancel_scan(wpa_s);
 990                 wpas_clear_wps(wpa_s);
 991         } else if (wpa_s->wpa_state >= WPA_ASSOCIATED) {
 992                 wpa_printf(MSG_DEBUG, "WPS: Cancel operation - "
 993                            "deauthenticate");
 994                 wpa_supplicant_deauthenticate(wpa_s,
 995                                               WLAN_REASON_DEAUTH_LEAVING);
 996                 wpas_clear_wps(wpa_s);
 997         } else {
 998                 wpas_wps_reenable_networks(wpa_s);
 999                 wpas_wps_clear_ap_info(wpa_s);
1000         }
1001
1002         return 0;
1003 }
1004
1005
1006 #ifdef CONFIG_WPS_OOB
1007 int wpas_wps_start_oob(struct wpa_supplicant *wpa_s, char *device_type,
1008                        char *path, char *method)
1009 {
1010         struct wps_context *wps = wpa_s->wps;
1011         struct oob_device_data *oob_dev;
1012
1013         oob_dev = wps_get_oob_device(device_type);
1014         if (oob_dev == NULL)
1015                 return -1;
1016         oob_dev->device_path = path;
1017         wps->oob_conf.oob_method = wps_get_oob_method(method);
1018
1019         if (wps->oob_conf.oob_method == OOB_METHOD_DEV_PWD_E) {
1020                 /*
1021                  * Use pre-configured DH keys in order to be able to write the
1022                  * key hash into the OOB file.
1023                  */
1024                 wpabuf_free(wps->dh_pubkey);
1025                 wpabuf_free(wps->dh_privkey);
1026                 wps->dh_privkey = NULL;
1027                 wps->dh_pubkey = NULL;
1028                 dh5_free(wps->dh_ctx);
1029                 wps->dh_ctx = dh5_init(&wps->dh_privkey, &wps->dh_pubkey);
1030                 wps->dh_pubkey = wpabuf_zeropad(wps->dh_pubkey, 192);
1031                 if (wps->dh_ctx == NULL || wps->dh_pubkey == NULL) {
1032                         wpa_printf(MSG_ERROR, "WPS: Failed to initialize "
1033                                    "Diffie-Hellman handshake");
1034                         return -1;
1035                 }
1036         }
1037
1038         if (wps->oob_conf.oob_method == OOB_METHOD_CRED)
1039                 wpas_clear_wps(wpa_s);
1040
1041         if (wps_process_oob(wps, oob_dev, 0) < 0)
1042                 return -1;
1043
1044         if ((wps->oob_conf.oob_method == OOB_METHOD_DEV_PWD_E ||
1045              wps->oob_conf.oob_method == OOB_METHOD_DEV_PWD_R) &&
1046             wpas_wps_start_pin(wpa_s, NULL,
1047                                wpabuf_head(wps->oob_conf.dev_password), 0,
1048                                DEV_PW_DEFAULT) < 0)
1049                         return -1;
1050
1051         return 0;
1052 }
1053 #endif /* CONFIG_WPS_OOB */
1054
1055
1056 int wpas_wps_start_reg(struct wpa_supplicant *wpa_s, const u8 *bssid,
1057                        const char *pin, struct wps_new_ap_settings *settings)
1058 {
1059         struct wpa_ssid *ssid;
1060         char val[200];
1061         char *pos, *end;
1062         int res;
1063
1064         if (!pin)
1065                 return -1;
1066         wpas_clear_wps(wpa_s);
1067         ssid = wpas_wps_add_network(wpa_s, 1, bssid);
1068         if (ssid == NULL)
1069                 return -1;
1070         ssid->temporary = 1;
1071         pos = val;
1072         end = pos + sizeof(val);
1073         res = os_snprintf(pos, end - pos, "\"pin=%s", pin);
1074         if (res < 0 || res >= end - pos)
1075                 return -1;
1076         pos += res;
1077         if (settings) {
1078                 res = os_snprintf(pos, end - pos, " new_ssid=%s new_auth=%s "
1079                                   "new_encr=%s new_key=%s",
1080                                   settings->ssid_hex, settings->auth,
1081                                   settings->encr, settings->key_hex);
1082                 if (res < 0 || res >= end - pos)
1083                         return -1;
1084                 pos += res;
1085         }
1086         res = os_snprintf(pos, end - pos, "\"");
1087         if (res < 0 || res >= end - pos)
1088                 return -1;
1089         wpa_config_set(ssid, "phase1", val, 0);
1090         if (wpa_s->wps_fragment_size)
1091                 ssid->eap.fragment_size = wpa_s->wps_fragment_size;
1092         eloop_register_timeout(WPS_PBC_WALK_TIME, 0, wpas_wps_timeout,
1093                                wpa_s, NULL);
1094         wpas_wps_reassoc(wpa_s, ssid, bssid);
1095         return 0;
1096 }
1097
1098
1099 static int wpas_wps_new_psk_cb(void *ctx, const u8 *mac_addr, const u8 *psk,
1100                                size_t psk_len)
1101 {
1102         wpa_printf(MSG_DEBUG, "WPS: Received new WPA/WPA2-PSK from WPS for "
1103                    "STA " MACSTR, MAC2STR(mac_addr));
1104         wpa_hexdump_key(MSG_DEBUG, "Per-device PSK", psk, psk_len);
1105
1106         /* TODO */
1107
1108         return 0;
1109 }
1110
1111
1112 static void wpas_wps_pin_needed_cb(void *ctx, const u8 *uuid_e,
1113                                    const struct wps_device_data *dev)
1114 {
1115         char uuid[40], txt[400];
1116         int len;
1117         char devtype[WPS_DEV_TYPE_BUFSIZE];
1118         if (uuid_bin2str(uuid_e, uuid, sizeof(uuid)))
1119                 return;
1120         wpa_printf(MSG_DEBUG, "WPS: PIN needed for UUID-E %s", uuid);
1121         len = os_snprintf(txt, sizeof(txt), "WPS-EVENT-PIN-NEEDED %s " MACSTR
1122                           " [%s|%s|%s|%s|%s|%s]",
1123                           uuid, MAC2STR(dev->mac_addr), dev->device_name,
1124                           dev->manufacturer, dev->model_name,
1125                           dev->model_number, dev->serial_number,
1126                           wps_dev_type_bin2str(dev->pri_dev_type, devtype,
1127                                                sizeof(devtype)));
1128         if (len > 0 && len < (int) sizeof(txt))
1129                 wpa_printf(MSG_INFO, "%s", txt);
1130 }
1131
1132
1133 static void wpas_wps_set_sel_reg_cb(void *ctx, int sel_reg, u16 dev_passwd_id,
1134                                     u16 sel_reg_config_methods)
1135 {
1136 #ifdef CONFIG_WPS_ER
1137         struct wpa_supplicant *wpa_s = ctx;
1138
1139         if (wpa_s->wps_er == NULL)
1140                 return;
1141         wpa_printf(MSG_DEBUG, "WPS ER: SetSelectedRegistrar - sel_reg=%d "
1142                    "dev_password_id=%u sel_reg_config_methods=0x%x",
1143                    sel_reg, dev_passwd_id, sel_reg_config_methods);
1144         wps_er_set_sel_reg(wpa_s->wps_er, sel_reg, dev_passwd_id,
1145                            sel_reg_config_methods);
1146 #endif /* CONFIG_WPS_ER */
1147 }
1148
1149
1150 static u16 wps_fix_config_methods(u16 config_methods)
1151 {
1152 #ifdef CONFIG_WPS2
1153         if ((config_methods &
1154              (WPS_CONFIG_DISPLAY | WPS_CONFIG_VIRT_DISPLAY |
1155               WPS_CONFIG_PHY_DISPLAY)) == WPS_CONFIG_DISPLAY) {
1156                 wpa_printf(MSG_INFO, "WPS: Converting display to "
1157                            "virtual_display for WPS 2.0 compliance");
1158                 config_methods |= WPS_CONFIG_VIRT_DISPLAY;
1159         }
1160         if ((config_methods &
1161              (WPS_CONFIG_PUSHBUTTON | WPS_CONFIG_VIRT_PUSHBUTTON |
1162               WPS_CONFIG_PHY_PUSHBUTTON)) == WPS_CONFIG_PUSHBUTTON) {
1163                 wpa_printf(MSG_INFO, "WPS: Converting push_button to "
1164                            "virtual_push_button for WPS 2.0 compliance");
1165                 config_methods |= WPS_CONFIG_VIRT_PUSHBUTTON;
1166         }
1167 #endif /* CONFIG_WPS2 */
1168
1169         return config_methods;
1170 }
1171
1172
1173 static void wpas_wps_set_uuid(struct wpa_supplicant *wpa_s,
1174                               struct wps_context *wps)
1175 {
1176         wpa_printf(MSG_DEBUG, "WPS: Set UUID for interface %s", wpa_s->ifname);
1177         if (is_nil_uuid(wpa_s->conf->uuid)) {
1178                 struct wpa_supplicant *first;
1179                 first = wpa_s->global->ifaces;
1180                 while (first && first->next)
1181                         first = first->next;
1182                 if (first && first != wpa_s) {
1183                         if (wps != wpa_s->global->ifaces->wps)
1184                                 os_memcpy(wps->uuid,
1185                                           wpa_s->global->ifaces->wps->uuid,
1186                                           WPS_UUID_LEN);
1187                         wpa_hexdump(MSG_DEBUG, "WPS: UUID from the first "
1188                                     "interface", wps->uuid, WPS_UUID_LEN);
1189                 } else {
1190                         uuid_gen_mac_addr(wpa_s->own_addr, wps->uuid);
1191                         wpa_hexdump(MSG_DEBUG, "WPS: UUID based on MAC "
1192                                     "address", wps->uuid, WPS_UUID_LEN);
1193                 }
1194         } else {
1195                 os_memcpy(wps->uuid, wpa_s->conf->uuid, WPS_UUID_LEN);
1196                 wpa_hexdump(MSG_DEBUG, "WPS: UUID based on configuration",
1197                             wps->uuid, WPS_UUID_LEN);
1198         }
1199 }
1200
1201
1202 static void wpas_wps_set_vendor_ext_m1(struct wpa_supplicant *wpa_s,
1203                                        struct wps_context *wps)
1204 {
1205         wpabuf_free(wps->dev.vendor_ext_m1);
1206         wps->dev.vendor_ext_m1 = NULL;
1207
1208         if (wpa_s->conf->wps_vendor_ext_m1) {
1209                 wps->dev.vendor_ext_m1 =
1210                         wpabuf_dup(wpa_s->conf->wps_vendor_ext_m1);
1211                 if (!wps->dev.vendor_ext_m1) {
1212                         wpa_printf(MSG_ERROR, "WPS: Cannot "
1213                                    "allocate memory for vendor_ext_m1");
1214                 }
1215         }
1216 }
1217
1218
1219 int wpas_wps_init(struct wpa_supplicant *wpa_s)
1220 {
1221         struct wps_context *wps;
1222         struct wps_registrar_config rcfg;
1223         struct hostapd_hw_modes *modes;
1224         u16 m;
1225
1226         wps = os_zalloc(sizeof(*wps));
1227         if (wps == NULL)
1228                 return -1;
1229
1230         wps->cred_cb = wpa_supplicant_wps_cred;
1231         wps->event_cb = wpa_supplicant_wps_event;
1232         wps->cb_ctx = wpa_s;
1233
1234         wps->dev.device_name = wpa_s->conf->device_name;
1235         wps->dev.manufacturer = wpa_s->conf->manufacturer;
1236         wps->dev.model_name = wpa_s->conf->model_name;
1237         wps->dev.model_number = wpa_s->conf->model_number;
1238         wps->dev.serial_number = wpa_s->conf->serial_number;
1239         wps->config_methods =
1240                 wps_config_methods_str2bin(wpa_s->conf->config_methods);
1241         if ((wps->config_methods & (WPS_CONFIG_DISPLAY | WPS_CONFIG_LABEL)) ==
1242             (WPS_CONFIG_DISPLAY | WPS_CONFIG_LABEL)) {
1243                 wpa_printf(MSG_ERROR, "WPS: Both Label and Display config "
1244                            "methods are not allowed at the same time");
1245                 os_free(wps);
1246                 return -1;
1247         }
1248         wps->config_methods = wps_fix_config_methods(wps->config_methods);
1249         wps->dev.config_methods = wps->config_methods;
1250         os_memcpy(wps->dev.pri_dev_type, wpa_s->conf->device_type,
1251                   WPS_DEV_TYPE_LEN);
1252
1253         wps->dev.num_sec_dev_types = wpa_s->conf->num_sec_device_types;
1254         os_memcpy(wps->dev.sec_dev_type, wpa_s->conf->sec_device_type,
1255                   WPS_DEV_TYPE_LEN * wps->dev.num_sec_dev_types);
1256
1257         wpas_wps_set_vendor_ext_m1(wpa_s, wps);
1258
1259         wps->dev.os_version = WPA_GET_BE32(wpa_s->conf->os_version);
1260         modes = wpa_s->hw.modes;
1261         if (modes) {
1262                 for (m = 0; m < wpa_s->hw.num_modes; m++) {
1263                         if (modes[m].mode == HOSTAPD_MODE_IEEE80211B ||
1264                             modes[m].mode == HOSTAPD_MODE_IEEE80211G)
1265                                 wps->dev.rf_bands |= WPS_RF_24GHZ;
1266                         else if (modes[m].mode == HOSTAPD_MODE_IEEE80211A)
1267                                 wps->dev.rf_bands |= WPS_RF_50GHZ;
1268                 }
1269         }
1270         if (wps->dev.rf_bands == 0) {
1271                 /*
1272                  * Default to claiming support for both bands if the driver
1273                  * does not provide support for fetching supported bands.
1274                  */
1275                 wps->dev.rf_bands = WPS_RF_24GHZ | WPS_RF_50GHZ;
1276         }
1277         os_memcpy(wps->dev.mac_addr, wpa_s->own_addr, ETH_ALEN);
1278         wpas_wps_set_uuid(wpa_s, wps);
1279
1280         wps->auth_types = WPS_AUTH_WPA2PSK | WPS_AUTH_WPAPSK;
1281         wps->encr_types = WPS_ENCR_AES | WPS_ENCR_TKIP;
1282
1283         os_memset(&rcfg, 0, sizeof(rcfg));
1284         rcfg.new_psk_cb = wpas_wps_new_psk_cb;
1285         rcfg.pin_needed_cb = wpas_wps_pin_needed_cb;
1286         rcfg.set_sel_reg_cb = wpas_wps_set_sel_reg_cb;
1287         rcfg.cb_ctx = wpa_s;
1288
1289         wps->registrar = wps_registrar_init(wps, &rcfg);
1290         if (wps->registrar == NULL) {
1291                 wpa_printf(MSG_DEBUG, "Failed to initialize WPS Registrar");
1292                 os_free(wps);
1293                 return -1;
1294         }
1295
1296         wpa_s->wps = wps;
1297
1298         return 0;
1299 }
1300
1301
1302 void wpas_wps_deinit(struct wpa_supplicant *wpa_s)
1303 {
1304         eloop_cancel_timeout(wpas_wps_timeout, wpa_s, NULL);
1305         eloop_cancel_timeout(wpas_wps_reenable_networks_cb, wpa_s, NULL);
1306         wpas_wps_clear_ap_info(wpa_s);
1307
1308         if (wpa_s->wps == NULL)
1309                 return;
1310
1311 #ifdef CONFIG_WPS_ER
1312         wps_er_deinit(wpa_s->wps_er, NULL, NULL);
1313         wpa_s->wps_er = NULL;
1314 #endif /* CONFIG_WPS_ER */
1315
1316         wps_registrar_deinit(wpa_s->wps->registrar);
1317         wpabuf_free(wpa_s->wps->dh_pubkey);
1318         wpabuf_free(wpa_s->wps->dh_privkey);
1319         wpabuf_free(wpa_s->wps->oob_conf.pubkey_hash);
1320         wpabuf_free(wpa_s->wps->oob_conf.dev_password);
1321         wpabuf_free(wpa_s->wps->dev.vendor_ext_m1);
1322         os_free(wpa_s->wps->network_key);
1323         os_free(wpa_s->wps);
1324         wpa_s->wps = NULL;
1325 }
1326
1327
1328 int wpas_wps_ssid_bss_match(struct wpa_supplicant *wpa_s,
1329                             struct wpa_ssid *ssid, struct wpa_bss *bss)
1330 {
1331         struct wpabuf *wps_ie;
1332
1333         if (!(ssid->key_mgmt & WPA_KEY_MGMT_WPS))
1334                 return -1;
1335
1336         wps_ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1337         if (eap_is_wps_pbc_enrollee(&ssid->eap)) {
1338                 if (!wps_ie) {
1339                         wpa_printf(MSG_DEBUG, "   skip - non-WPS AP");
1340                         return 0;
1341                 }
1342
1343                 if (!wps_is_selected_pbc_registrar(wps_ie)) {
1344                         wpa_printf(MSG_DEBUG, "   skip - WPS AP "
1345                                    "without active PBC Registrar");
1346                         wpabuf_free(wps_ie);
1347                         return 0;
1348                 }
1349
1350                 /* TODO: overlap detection */
1351                 wpa_printf(MSG_DEBUG, "   selected based on WPS IE "
1352                            "(Active PBC)");
1353                 wpabuf_free(wps_ie);
1354                 return 1;
1355         }
1356
1357         if (eap_is_wps_pin_enrollee(&ssid->eap)) {
1358                 if (!wps_ie) {
1359                         wpa_printf(MSG_DEBUG, "   skip - non-WPS AP");
1360                         return 0;
1361                 }
1362
1363                 /*
1364                  * Start with WPS APs that advertise our address as an
1365                  * authorized MAC (v2.0) or active PIN Registrar (v1.0) and
1366                  * allow any WPS AP after couple of scans since some APs do not
1367                  * set Selected Registrar attribute properly when using
1368                  * external Registrar.
1369                  */
1370                 if (!wps_is_addr_authorized(wps_ie, wpa_s->own_addr, 1)) {
1371                         if (wpa_s->scan_runs < WPS_PIN_SCAN_IGNORE_SEL_REG) {
1372                                 wpa_printf(MSG_DEBUG, "   skip - WPS AP "
1373                                            "without active PIN Registrar");
1374                                 wpabuf_free(wps_ie);
1375                                 return 0;
1376                         }
1377                         wpa_printf(MSG_DEBUG, "   selected based on WPS IE");
1378                 } else {
1379                         wpa_printf(MSG_DEBUG, "   selected based on WPS IE "
1380                                    "(Authorized MAC or Active PIN)");
1381                 }
1382                 wpabuf_free(wps_ie);
1383                 return 1;
1384         }
1385
1386         if (wps_ie) {
1387                 wpa_printf(MSG_DEBUG, "   selected based on WPS IE");
1388                 wpabuf_free(wps_ie);
1389                 return 1;
1390         }
1391
1392         return -1;
1393 }
1394
1395
1396 int wpas_wps_ssid_wildcard_ok(struct wpa_supplicant *wpa_s,
1397                               struct wpa_ssid *ssid,
1398                               struct wpa_bss *bss)
1399 {
1400         struct wpabuf *wps_ie = NULL;
1401         int ret = 0;
1402
1403         if (eap_is_wps_pbc_enrollee(&ssid->eap)) {
1404                 wps_ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1405                 if (wps_ie && wps_is_selected_pbc_registrar(wps_ie)) {
1406                         /* allow wildcard SSID for WPS PBC */
1407                         ret = 1;
1408                 }
1409         } else if (eap_is_wps_pin_enrollee(&ssid->eap)) {
1410                 wps_ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1411                 if (wps_ie &&
1412                     (wps_is_addr_authorized(wps_ie, wpa_s->own_addr, 1) ||
1413                      wpa_s->scan_runs >= WPS_PIN_SCAN_IGNORE_SEL_REG)) {
1414                         /* allow wildcard SSID for WPS PIN */
1415                         ret = 1;
1416                 }
1417         }
1418
1419         if (!ret && ssid->bssid_set &&
1420             os_memcmp(ssid->bssid, bss->bssid, ETH_ALEN) == 0) {
1421                 /* allow wildcard SSID due to hardcoded BSSID match */
1422                 ret = 1;
1423         }
1424
1425 #ifdef CONFIG_WPS_STRICT
1426         if (wps_ie) {
1427                 if (wps_validate_beacon_probe_resp(wps_ie, bss->beacon_ie_len >
1428                                                    0, bss->bssid) < 0)
1429                         ret = 0;
1430                 if (bss->beacon_ie_len) {
1431                         struct wpabuf *bcn_wps;
1432                         bcn_wps = wpa_bss_get_vendor_ie_multi_beacon(
1433                                 bss, WPS_IE_VENDOR_TYPE);
1434                         if (bcn_wps == NULL) {
1435                                 wpa_printf(MSG_DEBUG, "WPS: Mandatory WPS IE "
1436                                            "missing from AP Beacon");
1437                                 ret = 0;
1438                         } else {
1439                                 if (wps_validate_beacon(wps_ie) < 0)
1440                                         ret = 0;
1441                                 wpabuf_free(bcn_wps);
1442                         }
1443                 }
1444         }
1445 #endif /* CONFIG_WPS_STRICT */
1446
1447         wpabuf_free(wps_ie);
1448
1449         return ret;
1450 }
1451
1452
1453 int wpas_wps_scan_pbc_overlap(struct wpa_supplicant *wpa_s,
1454                               struct wpa_bss *selected, struct wpa_ssid *ssid)
1455 {
1456         const u8 *sel_uuid, *uuid;
1457         struct wpabuf *wps_ie;
1458         int ret = 0;
1459         struct wpa_bss *bss;
1460
1461         if (!eap_is_wps_pbc_enrollee(&ssid->eap))
1462                 return 0;
1463
1464         wpa_printf(MSG_DEBUG, "WPS: Check whether PBC session overlap is "
1465                    "present in scan results; selected BSSID " MACSTR,
1466                    MAC2STR(selected->bssid));
1467
1468         /* Make sure that only one AP is in active PBC mode */
1469         wps_ie = wpa_bss_get_vendor_ie_multi(selected, WPS_IE_VENDOR_TYPE);
1470         if (wps_ie) {
1471                 sel_uuid = wps_get_uuid_e(wps_ie);
1472                 wpa_hexdump(MSG_DEBUG, "WPS: UUID of the selected BSS",
1473                             sel_uuid, UUID_LEN);
1474         } else {
1475                 wpa_printf(MSG_DEBUG, "WPS: Selected BSS does not include "
1476                            "WPS IE?!");
1477                 sel_uuid = NULL;
1478         }
1479
1480         dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
1481                 struct wpabuf *ie;
1482                 if (bss == selected)
1483                         continue;
1484                 ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1485                 if (!ie)
1486                         continue;
1487                 if (!wps_is_selected_pbc_registrar(ie)) {
1488                         wpabuf_free(ie);
1489                         continue;
1490                 }
1491                 wpa_printf(MSG_DEBUG, "WPS: Another BSS in active PBC mode: "
1492                            MACSTR, MAC2STR(bss->bssid));
1493                 uuid = wps_get_uuid_e(ie);
1494                 wpa_hexdump(MSG_DEBUG, "WPS: UUID of the other BSS",
1495                             uuid, UUID_LEN);
1496                 if (sel_uuid == NULL || uuid == NULL ||
1497                     os_memcmp(sel_uuid, uuid, UUID_LEN) != 0) {
1498                         ret = 1; /* PBC overlap */
1499                         wpa_msg(wpa_s, MSG_INFO, "WPS: PBC overlap detected: "
1500                                 MACSTR " and " MACSTR,
1501                                 MAC2STR(selected->bssid),
1502                                 MAC2STR(bss->bssid));
1503                         wpabuf_free(ie);
1504                         break;
1505                 }
1506
1507                 /* TODO: verify that this is reasonable dual-band situation */
1508
1509                 wpabuf_free(ie);
1510         }
1511
1512         wpabuf_free(wps_ie);
1513
1514         return ret;
1515 }
1516
1517
1518 void wpas_wps_notify_scan_results(struct wpa_supplicant *wpa_s)
1519 {
1520         struct wpa_bss *bss;
1521         unsigned int pbc = 0, auth = 0, pin = 0, wps = 0;
1522
1523         if (wpa_s->disconnected || wpa_s->wpa_state >= WPA_ASSOCIATED)
1524                 return;
1525
1526         dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
1527                 struct wpabuf *ie;
1528                 ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1529                 if (!ie)
1530                         continue;
1531                 if (wps_is_selected_pbc_registrar(ie))
1532                         pbc++;
1533                 else if (wps_is_addr_authorized(ie, wpa_s->own_addr, 0))
1534                         auth++;
1535                 else if (wps_is_selected_pin_registrar(ie))
1536                         pin++;
1537                 else
1538                         wps++;
1539                 wpabuf_free(ie);
1540         }
1541
1542         if (pbc)
1543                 wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_AP_AVAILABLE_PBC);
1544         else if (auth)
1545                 wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_AP_AVAILABLE_AUTH);
1546         else if (pin)
1547                 wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_AP_AVAILABLE_PIN);
1548         else if (wps)
1549                 wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_AP_AVAILABLE);
1550 }
1551
1552
1553 int wpas_wps_searching(struct wpa_supplicant *wpa_s)
1554 {
1555         struct wpa_ssid *ssid;
1556
1557         for (ssid = wpa_s->conf->ssid; ssid; ssid = ssid->next) {
1558                 if ((ssid->key_mgmt & WPA_KEY_MGMT_WPS) && !ssid->disabled)
1559                         return 1;
1560         }
1561
1562         return 0;
1563 }
1564
1565
1566 int wpas_wps_scan_result_text(const u8 *ies, size_t ies_len, char *buf,
1567                               char *end)
1568 {
1569         struct wpabuf *wps_ie;
1570         int ret;
1571
1572         wps_ie = ieee802_11_vendor_ie_concat(ies, ies_len, WPS_DEV_OUI_WFA);
1573         if (wps_ie == NULL)
1574                 return 0;
1575
1576         ret = wps_attr_text(wps_ie, buf, end);
1577         wpabuf_free(wps_ie);
1578         return ret;
1579 }
1580
1581
1582 int wpas_wps_er_start(struct wpa_supplicant *wpa_s, const char *filter)
1583 {
1584 #ifdef CONFIG_WPS_ER
1585         if (wpa_s->wps_er) {
1586                 wps_er_refresh(wpa_s->wps_er);
1587                 return 0;
1588         }
1589         wpa_s->wps_er = wps_er_init(wpa_s->wps, wpa_s->ifname, filter);
1590         if (wpa_s->wps_er == NULL)
1591                 return -1;
1592         return 0;
1593 #else /* CONFIG_WPS_ER */
1594         return 0;
1595 #endif /* CONFIG_WPS_ER */
1596 }
1597
1598
1599 int wpas_wps_er_stop(struct wpa_supplicant *wpa_s)
1600 {
1601 #ifdef CONFIG_WPS_ER
1602         wps_er_deinit(wpa_s->wps_er, NULL, NULL);
1603         wpa_s->wps_er = NULL;
1604 #endif /* CONFIG_WPS_ER */
1605         return 0;
1606 }
1607
1608
1609 #ifdef CONFIG_WPS_ER
1610 int wpas_wps_er_add_pin(struct wpa_supplicant *wpa_s, const u8 *addr,
1611                         const char *uuid, const char *pin)
1612 {
1613         u8 u[UUID_LEN];
1614         int any = 0;
1615
1616         if (os_strcmp(uuid, "any") == 0)
1617                 any = 1;
1618         else if (uuid_str2bin(uuid, u))
1619                 return -1;
1620         return wps_registrar_add_pin(wpa_s->wps->registrar, addr,
1621                                      any ? NULL : u,
1622                                      (const u8 *) pin, os_strlen(pin), 300);
1623 }
1624
1625
1626 int wpas_wps_er_pbc(struct wpa_supplicant *wpa_s, const char *uuid)
1627 {
1628         u8 u[UUID_LEN];
1629
1630         if (uuid_str2bin(uuid, u))
1631                 return -1;
1632         return wps_er_pbc(wpa_s->wps_er, u);
1633 }
1634
1635
1636 int wpas_wps_er_learn(struct wpa_supplicant *wpa_s, const char *uuid,
1637                       const char *pin)
1638 {
1639         u8 u[UUID_LEN];
1640
1641         if (uuid_str2bin(uuid, u))
1642                 return -1;
1643         return wps_er_learn(wpa_s->wps_er, u, (const u8 *) pin,
1644                             os_strlen(pin));
1645 }
1646
1647
1648 int wpas_wps_er_set_config(struct wpa_supplicant *wpa_s, const char *uuid,
1649                            int id)
1650 {
1651         u8 u[UUID_LEN];
1652         struct wpa_ssid *ssid;
1653         struct wps_credential cred;
1654
1655         if (uuid_str2bin(uuid, u))
1656                 return -1;
1657         ssid = wpa_config_get_network(wpa_s->conf, id);
1658         if (ssid == NULL || ssid->ssid == NULL)
1659                 return -1;
1660
1661         os_memset(&cred, 0, sizeof(cred));
1662         if (ssid->ssid_len > 32)
1663                 return -1;
1664         os_memcpy(cred.ssid, ssid->ssid, ssid->ssid_len);
1665         cred.ssid_len = ssid->ssid_len;
1666         if (ssid->key_mgmt & WPA_KEY_MGMT_PSK) {
1667                 cred.auth_type = (ssid->proto & WPA_PROTO_RSN) ?
1668                         WPS_AUTH_WPA2PSK : WPS_AUTH_WPAPSK;
1669                 if (ssid->pairwise_cipher & WPA_CIPHER_CCMP)
1670                         cred.encr_type = WPS_ENCR_AES;
1671                 else
1672                         cred.encr_type = WPS_ENCR_TKIP;
1673                 if (ssid->passphrase) {
1674                         cred.key_len = os_strlen(ssid->passphrase);
1675                         if (cred.key_len >= 64)
1676                                 return -1;
1677                         os_memcpy(cred.key, ssid->passphrase, cred.key_len);
1678                 } else if (ssid->psk_set) {
1679                         cred.key_len = 32;
1680                         os_memcpy(cred.key, ssid->psk, 32);
1681                 } else
1682                         return -1;
1683         } else {
1684                 cred.auth_type = WPS_AUTH_OPEN;
1685                 cred.encr_type = WPS_ENCR_NONE;
1686         }
1687         return wps_er_set_config(wpa_s->wps_er, u, &cred);
1688 }
1689
1690
1691 int wpas_wps_er_config(struct wpa_supplicant *wpa_s, const char *uuid,
1692                        const char *pin, struct wps_new_ap_settings *settings)
1693 {
1694         u8 u[UUID_LEN];
1695         struct wps_credential cred;
1696         size_t len;
1697
1698         if (uuid_str2bin(uuid, u))
1699                 return -1;
1700         if (settings->ssid_hex == NULL || settings->auth == NULL ||
1701             settings->encr == NULL || settings->key_hex == NULL)
1702                 return -1;
1703
1704         os_memset(&cred, 0, sizeof(cred));
1705         len = os_strlen(settings->ssid_hex);
1706         if ((len & 1) || len > 2 * sizeof(cred.ssid) ||
1707             hexstr2bin(settings->ssid_hex, cred.ssid, len / 2))
1708                 return -1;
1709         cred.ssid_len = len / 2;
1710
1711         len = os_strlen(settings->key_hex);
1712         if ((len & 1) || len > 2 * sizeof(cred.key) ||
1713             hexstr2bin(settings->key_hex, cred.key, len / 2))
1714                 return -1;
1715         cred.key_len = len / 2;
1716
1717         if (os_strcmp(settings->auth, "OPEN") == 0)
1718                 cred.auth_type = WPS_AUTH_OPEN;
1719         else if (os_strcmp(settings->auth, "WPAPSK") == 0)
1720                 cred.auth_type = WPS_AUTH_WPAPSK;
1721         else if (os_strcmp(settings->auth, "WPA2PSK") == 0)
1722                 cred.auth_type = WPS_AUTH_WPA2PSK;
1723         else
1724                 return -1;
1725
1726         if (os_strcmp(settings->encr, "NONE") == 0)
1727                 cred.encr_type = WPS_ENCR_NONE;
1728         else if (os_strcmp(settings->encr, "WEP") == 0)
1729                 cred.encr_type = WPS_ENCR_WEP;
1730         else if (os_strcmp(settings->encr, "TKIP") == 0)
1731                 cred.encr_type = WPS_ENCR_TKIP;
1732         else if (os_strcmp(settings->encr, "CCMP") == 0)
1733                 cred.encr_type = WPS_ENCR_AES;
1734         else
1735                 return -1;
1736
1737         return wps_er_config(wpa_s->wps_er, u, (const u8 *) pin,
1738                              os_strlen(pin), &cred);
1739 }
1740
1741
1742 #ifdef CONFIG_WPS_NFC
1743 struct wpabuf * wpas_wps_er_nfc_config_token(struct wpa_supplicant *wpa_s,
1744                                              int ndef, const char *uuid)
1745 {
1746         struct wpabuf *ret;
1747         u8 u[UUID_LEN];
1748
1749         if (!wpa_s->wps_er)
1750                 return NULL;
1751
1752         if (uuid_str2bin(uuid, u))
1753                 return NULL;
1754
1755         ret = wps_er_nfc_config_token(wpa_s->wps_er, u);
1756         if (ndef && ret) {
1757                 struct wpabuf *tmp;
1758                 tmp = ndef_build_wifi(ret);
1759                 wpabuf_free(ret);
1760                 if (tmp == NULL)
1761                         return NULL;
1762                 ret = tmp;
1763         }
1764
1765         return ret;
1766 }
1767 #endif /* CONFIG_WPS_NFC */
1768
1769
1770 static int callbacks_pending = 0;
1771
1772 static void wpas_wps_terminate_cb(void *ctx)
1773 {
1774         wpa_printf(MSG_DEBUG, "WPS ER: Terminated");
1775         if (--callbacks_pending <= 0)
1776                 eloop_terminate();
1777 }
1778 #endif /* CONFIG_WPS_ER */
1779
1780
1781 int wpas_wps_terminate_pending(struct wpa_supplicant *wpa_s)
1782 {
1783 #ifdef CONFIG_WPS_ER
1784         if (wpa_s->wps_er) {
1785                 callbacks_pending++;
1786                 wps_er_deinit(wpa_s->wps_er, wpas_wps_terminate_cb, wpa_s);
1787                 wpa_s->wps_er = NULL;
1788                 return 1;
1789         }
1790 #endif /* CONFIG_WPS_ER */
1791         return 0;
1792 }
1793
1794
1795 int wpas_wps_in_progress(struct wpa_supplicant *wpa_s)
1796 {
1797         struct wpa_ssid *ssid;
1798
1799         for (ssid = wpa_s->conf->ssid; ssid; ssid = ssid->next) {
1800                 if (!ssid->disabled && ssid->key_mgmt == WPA_KEY_MGMT_WPS)
1801                         return 1;
1802         }
1803
1804         return 0;
1805 }
1806
1807
1808 void wpas_wps_update_config(struct wpa_supplicant *wpa_s)
1809 {
1810         struct wps_context *wps = wpa_s->wps;
1811
1812         if (wps == NULL)
1813                 return;
1814
1815         if (wpa_s->conf->changed_parameters & CFG_CHANGED_CONFIG_METHODS) {
1816                 wps->config_methods = wps_config_methods_str2bin(
1817                         wpa_s->conf->config_methods);
1818                 if ((wps->config_methods &
1819                      (WPS_CONFIG_DISPLAY | WPS_CONFIG_LABEL)) ==
1820                     (WPS_CONFIG_DISPLAY | WPS_CONFIG_LABEL)) {
1821                         wpa_printf(MSG_ERROR, "WPS: Both Label and Display "
1822                                    "config methods are not allowed at the "
1823                                    "same time");
1824                         wps->config_methods &= ~WPS_CONFIG_LABEL;
1825                 }
1826         }
1827         wps->config_methods = wps_fix_config_methods(wps->config_methods);
1828         wps->dev.config_methods = wps->config_methods;
1829
1830         if (wpa_s->conf->changed_parameters & CFG_CHANGED_DEVICE_TYPE)
1831                 os_memcpy(wps->dev.pri_dev_type, wpa_s->conf->device_type,
1832                           WPS_DEV_TYPE_LEN);
1833
1834         if (wpa_s->conf->changed_parameters & CFG_CHANGED_SEC_DEVICE_TYPE) {
1835                 wps->dev.num_sec_dev_types = wpa_s->conf->num_sec_device_types;
1836                 os_memcpy(wps->dev.sec_dev_type, wpa_s->conf->sec_device_type,
1837                           wps->dev.num_sec_dev_types * WPS_DEV_TYPE_LEN);
1838         }
1839
1840         if (wpa_s->conf->changed_parameters & CFG_CHANGED_VENDOR_EXTENSION)
1841                 wpas_wps_set_vendor_ext_m1(wpa_s, wps);
1842
1843         if (wpa_s->conf->changed_parameters & CFG_CHANGED_OS_VERSION)
1844                 wps->dev.os_version = WPA_GET_BE32(wpa_s->conf->os_version);
1845
1846         if (wpa_s->conf->changed_parameters & CFG_CHANGED_UUID)
1847                 wpas_wps_set_uuid(wpa_s, wps);
1848
1849         if (wpa_s->conf->changed_parameters &
1850             (CFG_CHANGED_DEVICE_NAME | CFG_CHANGED_WPS_STRING)) {
1851                 /* Update pointers to make sure they refer current values */
1852                 wps->dev.device_name = wpa_s->conf->device_name;
1853                 wps->dev.manufacturer = wpa_s->conf->manufacturer;
1854                 wps->dev.model_name = wpa_s->conf->model_name;
1855                 wps->dev.model_number = wpa_s->conf->model_number;
1856                 wps->dev.serial_number = wpa_s->conf->serial_number;
1857         }
1858 }
1859
1860
1861 #ifdef CONFIG_WPS_NFC
1862
1863 struct wpabuf * wpas_wps_nfc_token(struct wpa_supplicant *wpa_s, int ndef)
1864 {
1865         return wps_nfc_token_gen(ndef, &wpa_s->conf->wps_nfc_dev_pw_id,
1866                                  &wpa_s->conf->wps_nfc_dh_pubkey,
1867                                  &wpa_s->conf->wps_nfc_dh_privkey,
1868                                  &wpa_s->conf->wps_nfc_dev_pw);
1869 }
1870
1871
1872 int wpas_wps_start_nfc(struct wpa_supplicant *wpa_s, const u8 *bssid)
1873 {
1874         struct wps_context *wps = wpa_s->wps;
1875         char pw[32 * 2 + 1];
1876
1877         if (wpa_s->conf->wps_nfc_dh_pubkey == NULL ||
1878             wpa_s->conf->wps_nfc_dh_privkey == NULL ||
1879             wpa_s->conf->wps_nfc_dev_pw == NULL)
1880                 return -1;
1881
1882         dh5_free(wps->dh_ctx);
1883         wpabuf_free(wps->dh_pubkey);
1884         wpabuf_free(wps->dh_privkey);
1885         wps->dh_privkey = wpabuf_dup(wpa_s->conf->wps_nfc_dh_privkey);
1886         wps->dh_pubkey = wpabuf_dup(wpa_s->conf->wps_nfc_dh_pubkey);
1887         if (wps->dh_privkey == NULL || wps->dh_pubkey == NULL) {
1888                 wps->dh_ctx = NULL;
1889                 wpabuf_free(wps->dh_pubkey);
1890                 wps->dh_pubkey = NULL;
1891                 wpabuf_free(wps->dh_privkey);
1892                 wps->dh_privkey = NULL;
1893                 return -1;
1894         }
1895         wps->dh_ctx = dh5_init_fixed(wps->dh_privkey, wps->dh_pubkey);
1896         if (wps->dh_ctx == NULL)
1897                 return -1;
1898
1899         wpa_snprintf_hex_uppercase(pw, sizeof(pw),
1900                                    wpabuf_head(wpa_s->conf->wps_nfc_dev_pw),
1901                                    wpabuf_len(wpa_s->conf->wps_nfc_dev_pw));
1902         return wpas_wps_start_pin(wpa_s, bssid, pw, 0,
1903                                   wpa_s->conf->wps_nfc_dev_pw_id);
1904 }
1905
1906
1907 static int wpas_wps_use_cred(struct wpa_supplicant *wpa_s,
1908                              struct wps_parse_attr *attr)
1909 {
1910         wpa_s->wps_ap_channel = 0;
1911
1912         if (wps_oob_use_cred(wpa_s->wps, attr) < 0)
1913                 return -1;
1914
1915         if (wpa_s->wpa_state == WPA_INTERFACE_DISABLED)
1916                 return 0;
1917
1918         wpa_printf(MSG_DEBUG, "WPS: Request reconnection with new network "
1919                    "based on the received credential added");
1920         wpa_s->normal_scans = 0;
1921         wpa_supplicant_reinit_autoscan(wpa_s);
1922         if (wpa_s->wps_ap_channel) {
1923                 u16 chan = wpa_s->wps_ap_channel;
1924                 int freq = 0;
1925
1926                 if (chan >= 1 && chan <= 13)
1927                         freq = 2407 + 5 * chan;
1928                 else if (chan == 14)
1929                         freq = 2484;
1930                 else if (chan >= 30)
1931                         freq = 5000 + 5 * chan;
1932
1933                 if (freq) {
1934                         wpa_printf(MSG_DEBUG, "WPS: Credential indicated "
1935                                    "AP channel %u -> %u MHz", chan, freq);
1936                         wpa_s->after_wps = 5;
1937                         wpa_s->wps_freq = freq;
1938                 }
1939         }
1940         wpa_s->disconnected = 0;
1941         wpa_s->reassociate = 1;
1942         wpa_supplicant_req_scan(wpa_s, 0, 0);
1943
1944         return 0;
1945 }
1946
1947
1948 #ifdef CONFIG_WPS_ER
1949 static int wpas_wps_add_nfc_password_token(struct wpa_supplicant *wpa_s,
1950                                            struct wps_parse_attr *attr)
1951 {
1952         return wps_registrar_add_nfc_password_token(
1953                 wpa_s->wps->registrar, attr->oob_dev_password,
1954                 attr->oob_dev_password_len);
1955 }
1956 #endif /* CONFIG_WPS_ER */
1957
1958
1959 static int wpas_wps_nfc_tag_process(struct wpa_supplicant *wpa_s,
1960                                     const struct wpabuf *wps)
1961 {
1962         struct wps_parse_attr attr;
1963
1964         wpa_hexdump_buf(MSG_DEBUG, "WPS: Received NFC tag payload", wps);
1965
1966         if (wps_parse_msg(wps, &attr)) {
1967                 wpa_printf(MSG_DEBUG, "WPS: Ignore invalid data from NFC tag");
1968                 return -1;
1969         }
1970
1971         if (attr.num_cred)
1972                 return wpas_wps_use_cred(wpa_s, &attr);
1973
1974 #ifdef CONFIG_WPS_ER
1975         if (attr.oob_dev_password)
1976                 return wpas_wps_add_nfc_password_token(wpa_s, &attr);
1977 #endif /* CONFIG_WPS_ER */
1978
1979         wpa_printf(MSG_DEBUG, "WPS: Ignore unrecognized NFC tag");
1980         return -1;
1981 }
1982
1983
1984 int wpas_wps_nfc_tag_read(struct wpa_supplicant *wpa_s,
1985                           const struct wpabuf *data)
1986 {
1987         const struct wpabuf *wps = data;
1988         struct wpabuf *tmp = NULL;
1989         int ret;
1990
1991         if (wpabuf_len(data) < 4)
1992                 return -1;
1993
1994         if (*wpabuf_head_u8(data) != 0x10) {
1995                 /* Assume this contains full NDEF record */
1996                 tmp = ndef_parse_wifi(data);
1997                 if (tmp == NULL) {
1998                         wpa_printf(MSG_DEBUG, "WPS: Could not parse NDEF");
1999                         return -1;
2000                 }
2001                 wps = tmp;
2002         }
2003
2004         ret = wpas_wps_nfc_tag_process(wpa_s, wps);
2005         wpabuf_free(tmp);
2006         return ret;
2007 }
2008
2009
2010 struct wpabuf * wpas_wps_nfc_handover_req(struct wpa_supplicant *wpa_s)
2011 {
2012         return ndef_build_wifi_hr();
2013 }
2014
2015
2016 struct wpabuf * wpas_wps_nfc_handover_sel(struct wpa_supplicant *wpa_s)
2017 {
2018         return NULL;
2019 }
2020
2021
2022 int wpas_wps_nfc_rx_handover_req(struct wpa_supplicant *wpa_s,
2023                                  const struct wpabuf *data)
2024 {
2025         /* TODO */
2026         return -1;
2027 }
2028
2029
2030 int wpas_wps_nfc_rx_handover_sel(struct wpa_supplicant *wpa_s,
2031                                  const struct wpabuf *data)
2032 {
2033         struct wpabuf *wps;
2034         int ret;
2035
2036         wps = ndef_parse_wifi(data);
2037         if (wps == NULL)
2038                 return -1;
2039         wpa_printf(MSG_DEBUG, "WPS: Received application/vnd.wfa.wsc "
2040                    "payload from NFC connection handover");
2041         wpa_hexdump_buf_key(MSG_DEBUG, "WPS: NFC payload", wps);
2042         ret = wpas_wps_nfc_tag_process(wpa_s, wps);
2043         wpabuf_free(wps);
2044
2045         return ret;
2046 }
2047
2048 #endif /* CONFIG_WPS_NFC */
2049
2050
2051 extern int wpa_debug_level;
2052
2053 static void wpas_wps_dump_ap_info(struct wpa_supplicant *wpa_s)
2054 {
2055         size_t i;
2056         struct os_time now;
2057
2058         if (wpa_debug_level > MSG_DEBUG)
2059                 return;
2060
2061         if (wpa_s->wps_ap == NULL)
2062                 return;
2063
2064         os_get_time(&now);
2065
2066         for (i = 0; i < wpa_s->num_wps_ap; i++) {
2067                 struct wps_ap_info *ap = &wpa_s->wps_ap[i];
2068                 struct wpa_blacklist *e = wpa_blacklist_get(wpa_s, ap->bssid);
2069
2070                 wpa_printf(MSG_DEBUG, "WPS: AP[%d] " MACSTR " type=%d "
2071                            "tries=%d last_attempt=%d sec ago blacklist=%d",
2072                            (int) i, MAC2STR(ap->bssid), ap->type, ap->tries,
2073                            ap->last_attempt.sec > 0 ?
2074                            (int) now.sec - (int) ap->last_attempt.sec : -1,
2075                            e ? e->count : 0);
2076         }
2077 }
2078
2079
2080 static struct wps_ap_info * wpas_wps_get_ap_info(struct wpa_supplicant *wpa_s,
2081                                                  const u8 *bssid)
2082 {
2083         size_t i;
2084
2085         if (wpa_s->wps_ap == NULL)
2086                 return NULL;
2087
2088         for (i = 0; i < wpa_s->num_wps_ap; i++) {
2089                 struct wps_ap_info *ap = &wpa_s->wps_ap[i];
2090                 if (os_memcmp(ap->bssid, bssid, ETH_ALEN) == 0)
2091                         return ap;
2092         }
2093
2094         return NULL;
2095 }
2096
2097
2098 static void wpas_wps_update_ap_info_bss(struct wpa_supplicant *wpa_s,
2099                                         struct wpa_scan_res *res)
2100 {
2101         struct wpabuf *wps;
2102         enum wps_ap_info_type type;
2103         struct wps_ap_info *ap;
2104         int r;
2105
2106         if (wpa_scan_get_vendor_ie(res, WPS_IE_VENDOR_TYPE) == NULL)
2107                 return;
2108
2109         wps = wpa_scan_get_vendor_ie_multi(res, WPS_IE_VENDOR_TYPE);
2110         if (wps == NULL)
2111                 return;
2112
2113         r = wps_is_addr_authorized(wps, wpa_s->own_addr, 1);
2114         if (r == 2)
2115                 type = WPS_AP_SEL_REG_OUR;
2116         else if (r == 1)
2117                 type = WPS_AP_SEL_REG;
2118         else
2119                 type = WPS_AP_NOT_SEL_REG;
2120
2121         wpabuf_free(wps);
2122
2123         ap = wpas_wps_get_ap_info(wpa_s, res->bssid);
2124         if (ap) {
2125                 if (ap->type != type) {
2126                         wpa_printf(MSG_DEBUG, "WPS: AP " MACSTR
2127                                    " changed type %d -> %d",
2128                                    MAC2STR(res->bssid), ap->type, type);
2129                         ap->type = type;
2130                         if (type != WPS_AP_NOT_SEL_REG)
2131                                 wpa_blacklist_del(wpa_s, ap->bssid);
2132                 }
2133                 return;
2134         }
2135
2136         ap = os_realloc_array(wpa_s->wps_ap, wpa_s->num_wps_ap + 1,
2137                               sizeof(struct wps_ap_info));
2138         if (ap == NULL)
2139                 return;
2140
2141         wpa_s->wps_ap = ap;
2142         ap = &wpa_s->wps_ap[wpa_s->num_wps_ap];
2143         wpa_s->num_wps_ap++;
2144
2145         os_memset(ap, 0, sizeof(*ap));
2146         os_memcpy(ap->bssid, res->bssid, ETH_ALEN);
2147         ap->type = type;
2148         wpa_printf(MSG_DEBUG, "WPS: AP " MACSTR " type %d added",
2149                    MAC2STR(ap->bssid), ap->type);
2150 }
2151
2152
2153 void wpas_wps_update_ap_info(struct wpa_supplicant *wpa_s,
2154                              struct wpa_scan_results *scan_res)
2155 {
2156         size_t i;
2157
2158         for (i = 0; i < scan_res->num; i++)
2159                 wpas_wps_update_ap_info_bss(wpa_s, scan_res->res[i]);
2160
2161         wpas_wps_dump_ap_info(wpa_s);
2162 }
2163
2164
2165 void wpas_wps_notify_assoc(struct wpa_supplicant *wpa_s, const u8 *bssid)
2166 {
2167         struct wps_ap_info *ap;
2168         if (!wpa_s->wps_ap_iter)
2169                 return;
2170         ap = wpas_wps_get_ap_info(wpa_s, bssid);
2171         if (ap == NULL)
2172                 return;
2173         ap->tries++;
2174         os_get_time(&ap->last_attempt);
2175 }