wpa_supplicant/wps_supplicant.c

   1 /*
   2  * wpa_supplicant / WPS integration
   3  * Copyright (c) 2008-2014, 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 #ifdef CONFIG_P2P
  58         if (wpas_p2p_wps_eapol_cb(wpa_s) > 0)
  59                 return 1;
  60 #endif /* CONFIG_P2P */
  61
  62         if (!wpa_s->wps_success &&
  63             wpa_s->current_ssid &&
  64             eap_is_wps_pin_enrollee(&wpa_s->current_ssid->eap)) {
  65                 const u8 *bssid = wpa_s->bssid;
  66                 if (is_zero_ether_addr(bssid))
  67                         bssid = wpa_s->pending_bssid;
  68
  69                 wpa_printf(MSG_DEBUG, "WPS: PIN registration with " MACSTR
  70                            " did not succeed - continue trying to find "
  71                            "suitable AP", MAC2STR(bssid));
  72                 wpa_blacklist_add(wpa_s, bssid);
  73
  74                 wpa_supplicant_deauthenticate(wpa_s,
  75                                               WLAN_REASON_DEAUTH_LEAVING);
  76                 wpa_s->reassociate = 1;
  77                 wpa_supplicant_req_scan(wpa_s,
  78                                         wpa_s->blacklist_cleared ? 5 : 0, 0);
  79                 wpa_s->blacklist_cleared = 0;
  80                 return 1;
  81         }
  82
  83         wpas_wps_clear_ap_info(wpa_s);
  84         eloop_cancel_timeout(wpas_wps_timeout, wpa_s, NULL);
  85         if (wpa_s->key_mgmt == WPA_KEY_MGMT_WPS && !wpa_s->wps_success)
  86                 wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_FAIL);
  87
  88         if (wpa_s->key_mgmt == WPA_KEY_MGMT_WPS && wpa_s->current_ssid &&
  89             !(wpa_s->current_ssid->key_mgmt & WPA_KEY_MGMT_WPS)) {
  90                 int disabled = wpa_s->current_ssid->disabled;
  91                 unsigned int freq = wpa_s->assoc_freq;
  92                 struct wpa_bss *bss;
  93                 struct wpa_ssid *ssid = NULL;
  94                 int use_fast_assoc = 0;
  95
  96                 wpa_printf(MSG_DEBUG, "WPS: Network configuration replaced - "
  97                            "try to associate with the received credential "
  98                            "(freq=%u)", freq);
  99                 wpa_supplicant_deauthenticate(wpa_s,
 100                                               WLAN_REASON_DEAUTH_LEAVING);
 101                 if (disabled) {
 102                         wpa_printf(MSG_DEBUG, "WPS: Current network is "
 103                                    "disabled - wait for user to enable");
 104                         return 1;
 105                 }
 106                 wpa_s->after_wps = 5;
 107                 wpa_s->wps_freq = freq;
 108                 wpa_s->normal_scans = 0;
 109                 wpa_s->reassociate = 1;
 110
 111                 wpa_printf(MSG_DEBUG, "WPS: Checking whether fast association "
 112                            "without a new scan can be used");
 113                 bss = wpa_supplicant_pick_network(wpa_s, &ssid);
 114                 if (bss) {
 115                         struct wpabuf *wps;
 116                         struct wps_parse_attr attr;
 117
 118                         wps = wpa_bss_get_vendor_ie_multi(bss,
 119                                                           WPS_IE_VENDOR_TYPE);
 120                         if (wps && wps_parse_msg(wps, &attr) == 0 &&
 121                             attr.wps_state &&
 122                             *attr.wps_state == WPS_STATE_CONFIGURED)
 123                                 use_fast_assoc = 1;
 124                         wpabuf_free(wps);
 125                 }
 126
 127                 if (!use_fast_assoc ||
 128                     wpa_supplicant_fast_associate(wpa_s) != 1)
 129                         wpa_supplicant_req_scan(wpa_s, 0, 0);
 130                 return 1;
 131         }
 132
 133         if (wpa_s->key_mgmt == WPA_KEY_MGMT_WPS && wpa_s->current_ssid) {
 134                 wpa_printf(MSG_DEBUG, "WPS: Registration completed - waiting "
 135                            "for external credential processing");
 136                 wpas_clear_wps(wpa_s);
 137                 wpa_supplicant_deauthenticate(wpa_s,
 138                                               WLAN_REASON_DEAUTH_LEAVING);
 139                 return 1;
 140         }
 141
 142         return 0;
 143 }
 144
 145
 146 static void wpas_wps_security_workaround(struct wpa_supplicant *wpa_s,
 147                                          struct wpa_ssid *ssid,
 148                                          const struct wps_credential *cred)
 149 {
 150         struct wpa_driver_capa capa;
 151         struct wpa_bss *bss;
 152         const u8 *ie;
 153         struct wpa_ie_data adv;
 154         int wpa2 = 0, ccmp = 0;
 155
 156         /*
 157          * Many existing WPS APs do not know how to negotiate WPA2 or CCMP in
 158          * case they are configured for mixed mode operation (WPA+WPA2 and
 159          * TKIP+CCMP). Try to use scan results to figure out whether the AP
 160          * actually supports stronger security and select that if the client
 161          * has support for it, too.
 162          */
 163
 164         if (wpa_drv_get_capa(wpa_s, &capa))
 165                 return; /* Unknown what driver supports */
 166
 167         if (ssid->ssid == NULL)
 168                 return;
 169         bss = wpa_bss_get(wpa_s, cred->mac_addr, ssid->ssid, ssid->ssid_len);
 170         if (bss == NULL) {
 171                 wpa_printf(MSG_DEBUG, "WPS: The AP was not found from BSS "
 172                            "table - use credential as-is");
 173                 return;
 174         }
 175
 176         wpa_printf(MSG_DEBUG, "WPS: AP found from BSS table");
 177
 178         ie = wpa_bss_get_ie(bss, WLAN_EID_RSN);
 179         if (ie && wpa_parse_wpa_ie(ie, 2 + ie[1], &adv) == 0) {
 180                 wpa2 = 1;
 181                 if (adv.pairwise_cipher & WPA_CIPHER_CCMP)
 182                         ccmp = 1;
 183         } else {
 184                 ie = wpa_bss_get_vendor_ie(bss, WPA_IE_VENDOR_TYPE);
 185                 if (ie && wpa_parse_wpa_ie(ie, 2 + ie[1], &adv) == 0 &&
 186                     adv.pairwise_cipher & WPA_CIPHER_CCMP)
 187                         ccmp = 1;
 188         }
 189
 190         if (ie == NULL && (ssid->proto & WPA_PROTO_WPA) &&
 191             (ssid->pairwise_cipher & WPA_CIPHER_TKIP)) {
 192                 /*
 193                  * TODO: This could be the initial AP configuration and the
 194                  * Beacon contents could change shortly. Should request a new
 195                  * scan and delay addition of the network until the updated
 196                  * scan results are available.
 197                  */
 198                 wpa_printf(MSG_DEBUG, "WPS: The AP did not yet advertise WPA "
 199                            "support - use credential as-is");
 200                 return;
 201         }
 202
 203         if (ccmp && !(ssid->pairwise_cipher & WPA_CIPHER_CCMP) &&
 204             (ssid->pairwise_cipher & WPA_CIPHER_TKIP) &&
 205             (capa.key_mgmt & WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK)) {
 206                 wpa_printf(MSG_DEBUG, "WPS: Add CCMP into the credential "
 207                            "based on scan results");
 208                 if (wpa_s->conf->ap_scan == 1)
 209                         ssid->pairwise_cipher |= WPA_CIPHER_CCMP;
 210                 else
 211                         ssid->pairwise_cipher = WPA_CIPHER_CCMP;
 212         }
 213
 214         if (wpa2 && !(ssid->proto & WPA_PROTO_RSN) &&
 215             (ssid->proto & WPA_PROTO_WPA) &&
 216             (capa.enc & WPA_DRIVER_CAPA_ENC_CCMP)) {
 217                 wpa_printf(MSG_DEBUG, "WPS: Add WPA2 into the credential "
 218                            "based on scan results");
 219                 if (wpa_s->conf->ap_scan == 1)
 220                         ssid->proto |= WPA_PROTO_RSN;
 221                 else
 222                         ssid->proto = WPA_PROTO_RSN;
 223         }
 224 }
 225
 226
 227 static void wpas_wps_remove_dup_network(struct wpa_supplicant *wpa_s,
 228                                         struct wpa_ssid *new_ssid)
 229 {
 230         struct wpa_ssid *ssid, *next;
 231
 232         for (ssid = wpa_s->conf->ssid, next = ssid ? ssid->next : NULL; ssid;
 233              ssid = next, next = ssid ? ssid->next : NULL) {
 234                 /*
 235                  * new_ssid has already been added to the list in
 236                  * wpas_wps_add_network(), so skip it.
 237                  */
 238                 if (ssid == new_ssid)
 239                         continue;
 240
 241                 if (ssid->bssid_set || new_ssid->bssid_set) {
 242                         if (ssid->bssid_set != new_ssid->bssid_set)
 243                                 continue;
 244                         if (os_memcmp(ssid->bssid, new_ssid->bssid, ETH_ALEN) !=
 245                             0)
 246                                 continue;
 247                 }
 248
 249                 /* compare SSID */
 250                 if (ssid->ssid_len == 0 || ssid->ssid_len != new_ssid->ssid_len)
 251                         continue;
 252
 253                 if (ssid->ssid && new_ssid->ssid) {
 254                         if (os_memcmp(ssid->ssid, new_ssid->ssid,
 255                                       ssid->ssid_len) != 0)
 256                                 continue;
 257                 } else if (ssid->ssid || new_ssid->ssid)
 258                         continue;
 259
 260                 /* compare security parameters */
 261                 if (ssid->auth_alg != new_ssid->auth_alg ||
 262                     ssid->key_mgmt != new_ssid->key_mgmt ||
 263                     ssid->proto != new_ssid->proto ||
 264                     ssid->pairwise_cipher != new_ssid->pairwise_cipher ||
 265                     ssid->group_cipher != new_ssid->group_cipher)
 266                         continue;
 267
 268                 /* Remove the duplicated older network entry. */
 269                 wpa_printf(MSG_DEBUG, "Remove duplicate network %d", ssid->id);
 270                 wpas_notify_network_removed(wpa_s, ssid);
 271                 wpa_config_remove_network(wpa_s->conf, ssid->id);
 272         }
 273 }
 274
 275
 276 static int wpa_supplicant_wps_cred(void *ctx,
 277                                    const struct wps_credential *cred)
 278 {
 279         struct wpa_supplicant *wpa_s = ctx;
 280         struct wpa_ssid *ssid = wpa_s->current_ssid;
 281         u8 key_idx = 0;
 282         u16 auth_type;
 283 #ifdef CONFIG_WPS_REG_DISABLE_OPEN
 284         int registrar = 0;
 285 #endif /* CONFIG_WPS_REG_DISABLE_OPEN */
 286
 287         if ((wpa_s->conf->wps_cred_processing == 1 ||
 288              wpa_s->conf->wps_cred_processing == 2) && cred->cred_attr) {
 289                 size_t blen = cred->cred_attr_len * 2 + 1;
 290                 char *buf = os_malloc(blen);
 291                 if (buf) {
 292                         wpa_snprintf_hex(buf, blen,
 293                                          cred->cred_attr, cred->cred_attr_len);
 294                         wpa_msg(wpa_s, MSG_INFO, "%s%s",
 295                                 WPS_EVENT_CRED_RECEIVED, buf);
 296                         os_free(buf);
 297                 }
 298
 299                 wpas_notify_wps_credential(wpa_s, cred);
 300         } else
 301                 wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_CRED_RECEIVED);
 302
 303         wpa_hexdump_key(MSG_DEBUG, "WPS: Received Credential attribute",
 304                         cred->cred_attr, cred->cred_attr_len);
 305
 306         if (wpa_s->conf->wps_cred_processing == 1)
 307                 return 0;
 308
 309         wpa_hexdump_ascii(MSG_DEBUG, "WPS: SSID", cred->ssid, cred->ssid_len);
 310         wpa_printf(MSG_DEBUG, "WPS: Authentication Type 0x%x",
 311                    cred->auth_type);
 312         wpa_printf(MSG_DEBUG, "WPS: Encryption Type 0x%x", cred->encr_type);
 313         wpa_printf(MSG_DEBUG, "WPS: Network Key Index %d", cred->key_idx);
 314         wpa_hexdump_key(MSG_DEBUG, "WPS: Network Key",
 315                         cred->key, cred->key_len);
 316         wpa_printf(MSG_DEBUG, "WPS: MAC Address " MACSTR,
 317                    MAC2STR(cred->mac_addr));
 318
 319         auth_type = cred->auth_type;
 320         if (auth_type == (WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK)) {
 321                 wpa_printf(MSG_DEBUG, "WPS: Workaround - convert mixed-mode "
 322                            "auth_type into WPA2PSK");
 323                 auth_type = WPS_AUTH_WPA2PSK;
 324         }
 325
 326         if (auth_type != WPS_AUTH_OPEN &&
 327             auth_type != WPS_AUTH_SHARED &&
 328             auth_type != WPS_AUTH_WPAPSK &&
 329             auth_type != WPS_AUTH_WPA2PSK) {
 330                 wpa_printf(MSG_DEBUG, "WPS: Ignored credentials for "
 331                            "unsupported authentication type 0x%x",
 332                            auth_type);
 333                 return 0;
 334         }
 335
 336         if (auth_type == WPS_AUTH_WPAPSK || auth_type == WPS_AUTH_WPA2PSK) {
 337                 if (cred->key_len < 8 || cred->key_len > 2 * PMK_LEN) {
 338                         wpa_printf(MSG_ERROR, "WPS: Reject PSK credential with "
 339                                    "invalid Network Key length %lu",
 340                                    (unsigned long) cred->key_len);
 341                         return -1;
 342                 }
 343         }
 344
 345         if (ssid && (ssid->key_mgmt & WPA_KEY_MGMT_WPS)) {
 346                 wpa_printf(MSG_DEBUG, "WPS: Replace WPS network block based "
 347                            "on the received credential");
 348 #ifdef CONFIG_WPS_REG_DISABLE_OPEN
 349                 if (ssid->eap.identity &&
 350                     ssid->eap.identity_len == WSC_ID_REGISTRAR_LEN &&
 351                     os_memcmp(ssid->eap.identity, WSC_ID_REGISTRAR,
 352                               WSC_ID_REGISTRAR_LEN) == 0)
 353                         registrar = 1;
 354 #endif /* CONFIG_WPS_REG_DISABLE_OPEN */
 355                 os_free(ssid->eap.identity);
 356                 ssid->eap.identity = NULL;
 357                 ssid->eap.identity_len = 0;
 358                 os_free(ssid->eap.phase1);
 359                 ssid->eap.phase1 = NULL;
 360                 os_free(ssid->eap.eap_methods);
 361                 ssid->eap.eap_methods = NULL;
 362                 if (!ssid->p2p_group) {
 363                         ssid->temporary = 0;
 364                         ssid->bssid_set = 0;
 365                 }
 366                 ssid->disabled_until.sec = 0;
 367                 ssid->disabled_until.usec = 0;
 368                 ssid->auth_failures = 0;
 369         } else {
 370                 wpa_printf(MSG_DEBUG, "WPS: Create a new network based on the "
 371                            "received credential");
 372                 ssid = wpa_config_add_network(wpa_s->conf);
 373                 if (ssid == NULL)
 374                         return -1;
 375                 wpas_notify_network_added(wpa_s, ssid);
 376         }
 377
 378         wpa_config_set_network_defaults(ssid);
 379
 380         os_free(ssid->ssid);
 381         ssid->ssid = os_malloc(cred->ssid_len);
 382         if (ssid->ssid) {
 383                 os_memcpy(ssid->ssid, cred->ssid, cred->ssid_len);
 384                 ssid->ssid_len = cred->ssid_len;
 385         }
 386
 387         switch (cred->encr_type) {
 388         case WPS_ENCR_NONE:
 389                 break;
 390         case WPS_ENCR_WEP:
 391                 if (cred->key_len <= 0)
 392                         break;
 393                 if (cred->key_len != 5 && cred->key_len != 13 &&
 394                     cred->key_len != 10 && cred->key_len != 26) {
 395                         wpa_printf(MSG_ERROR, "WPS: Invalid WEP Key length "
 396                                    "%lu", (unsigned long) cred->key_len);
 397                         return -1;
 398                 }
 399                 if (cred->key_idx > NUM_WEP_KEYS) {
 400                         wpa_printf(MSG_ERROR, "WPS: Invalid WEP Key index %d",
 401                                    cred->key_idx);
 402                         return -1;
 403                 }
 404                 if (cred->key_idx)
 405                         key_idx = cred->key_idx - 1;
 406                 if (cred->key_len == 10 || cred->key_len == 26) {
 407                         if (hexstr2bin((char *) cred->key,
 408                                        ssid->wep_key[key_idx],
 409                                        cred->key_len / 2) < 0) {
 410                                 wpa_printf(MSG_ERROR, "WPS: Invalid WEP Key "
 411                                            "%d", key_idx);
 412                                 return -1;
 413                         }
 414                         ssid->wep_key_len[key_idx] = cred->key_len / 2;
 415                 } else {
 416                         os_memcpy(ssid->wep_key[key_idx], cred->key,
 417                                   cred->key_len);
 418                         ssid->wep_key_len[key_idx] = cred->key_len;
 419                 }
 420                 ssid->wep_tx_keyidx = key_idx;
 421                 break;
 422         case WPS_ENCR_TKIP:
 423                 ssid->pairwise_cipher = WPA_CIPHER_TKIP;
 424                 break;
 425         case WPS_ENCR_AES:
 426                 ssid->pairwise_cipher = WPA_CIPHER_CCMP;
 427                 break;
 428         }
 429
 430         switch (auth_type) {
 431         case WPS_AUTH_OPEN:
 432                 ssid->auth_alg = WPA_AUTH_ALG_OPEN;
 433                 ssid->key_mgmt = WPA_KEY_MGMT_NONE;
 434                 ssid->proto = 0;
 435 #ifdef CONFIG_WPS_REG_DISABLE_OPEN
 436                 if (registrar) {
 437                         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_OPEN_NETWORK
 438                                 "id=%d - Credentials for an open "
 439                                 "network disabled by default - use "
 440                                 "'select_network %d' to enable",
 441                                 ssid->id, ssid->id);
 442                         ssid->disabled = 1;
 443                 }
 444 #endif /* CONFIG_WPS_REG_DISABLE_OPEN */
 445                 break;
 446         case WPS_AUTH_SHARED:
 447                 ssid->auth_alg = WPA_AUTH_ALG_SHARED;
 448                 ssid->key_mgmt = WPA_KEY_MGMT_NONE;
 449                 ssid->proto = 0;
 450                 break;
 451         case WPS_AUTH_WPAPSK:
 452                 ssid->auth_alg = WPA_AUTH_ALG_OPEN;
 453                 ssid->key_mgmt = WPA_KEY_MGMT_PSK;
 454                 ssid->proto = WPA_PROTO_WPA;
 455                 break;
 456         case WPS_AUTH_WPA2PSK:
 457                 ssid->auth_alg = WPA_AUTH_ALG_OPEN;
 458                 ssid->key_mgmt = WPA_KEY_MGMT_PSK;
 459                 ssid->proto = WPA_PROTO_RSN;
 460                 break;
 461         }
 462
 463         if (ssid->key_mgmt == WPA_KEY_MGMT_PSK) {
 464                 if (cred->key_len == 2 * PMK_LEN) {
 465                         if (hexstr2bin((const char *) cred->key, ssid->psk,
 466                                        PMK_LEN)) {
 467                                 wpa_printf(MSG_ERROR, "WPS: Invalid Network "
 468                                            "Key");
 469                                 return -1;
 470                         }
 471                         ssid->psk_set = 1;
 472                         ssid->export_keys = 1;
 473                 } else if (cred->key_len >= 8 && cred->key_len < 2 * PMK_LEN) {
 474                         os_free(ssid->passphrase);
 475                         ssid->passphrase = os_malloc(cred->key_len + 1);
 476                         if (ssid->passphrase == NULL)
 477                                 return -1;
 478                         os_memcpy(ssid->passphrase, cred->key, cred->key_len);
 479                         ssid->passphrase[cred->key_len] = '\0';
 480                         wpa_config_update_psk(ssid);
 481                         ssid->export_keys = 1;
 482                 } else {
 483                         wpa_printf(MSG_ERROR, "WPS: Invalid Network Key "
 484                                    "length %lu",
 485                                    (unsigned long) cred->key_len);
 486                         return -1;
 487                 }
 488         }
 489
 490         wpas_wps_security_workaround(wpa_s, ssid, cred);
 491
 492         if (cred->ap_channel)
 493                 wpa_s->wps_ap_channel = cred->ap_channel;
 494
 495         wpas_wps_remove_dup_network(wpa_s, ssid);
 496
 497 #ifndef CONFIG_NO_CONFIG_WRITE
 498         if (wpa_s->conf->update_config &&
 499             wpa_config_write(wpa_s->confname, wpa_s->conf)) {
 500                 wpa_printf(MSG_DEBUG, "WPS: Failed to update configuration");
 501                 return -1;
 502         }
 503 #endif /* CONFIG_NO_CONFIG_WRITE */
 504
 505         /*
 506          * Optimize the post-WPS scan based on the channel used during
 507          * the provisioning in case EAP-Failure is not received.
 508          */
 509         wpa_s->after_wps = 5;
 510         wpa_s->wps_freq = wpa_s->assoc_freq;
 511
 512         return 0;
 513 }
 514
 515
 516 #ifdef CONFIG_P2P
 517 static void wpas_wps_pbc_overlap_cb(void *eloop_ctx, void *timeout_ctx)
 518 {
 519         struct wpa_supplicant *wpa_s = eloop_ctx;
 520         wpas_p2p_notif_pbc_overlap(wpa_s);
 521 }
 522 #endif /* CONFIG_P2P */
 523
 524
 525 static void wpa_supplicant_wps_event_m2d(struct wpa_supplicant *wpa_s,
 526                                          struct wps_event_m2d *m2d)
 527 {
 528         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_M2D
 529                 "dev_password_id=%d config_error=%d",
 530                 m2d->dev_password_id, m2d->config_error);
 531         wpas_notify_wps_event_m2d(wpa_s, m2d);
 532 #ifdef CONFIG_P2P
 533         if (wpa_s->parent && wpa_s->parent != wpa_s) {
 534                 wpa_msg(wpa_s->parent, MSG_INFO, WPS_EVENT_M2D
 535                         "dev_password_id=%d config_error=%d",
 536                         m2d->dev_password_id, m2d->config_error);
 537         }
 538         if (m2d->config_error == WPS_CFG_MULTIPLE_PBC_DETECTED) {
 539                 /*
 540                  * Notify P2P from eloop timeout to avoid issues with the
 541                  * interface getting removed while processing a message.
 542                  */
 543                 eloop_register_timeout(0, 0, wpas_wps_pbc_overlap_cb, wpa_s,
 544                                        NULL);
 545         }
 546 #endif /* CONFIG_P2P */
 547 }
 548
 549
 550 static void wpas_wps_clear_timeout(void *eloop_ctx, void *timeout_ctx)
 551 {
 552         struct wpa_supplicant *wpa_s = eloop_ctx;
 553         wpa_printf(MSG_DEBUG, "WPS: Clear WPS network from timeout");
 554         wpas_clear_wps(wpa_s);
 555 }
 556
 557
 558 static void wpa_supplicant_wps_event_fail(struct wpa_supplicant *wpa_s,
 559                                           struct wps_event_fail *fail)
 560 {
 561         if (fail->error_indication > 0 &&
 562             fail->error_indication < NUM_WPS_EI_VALUES) {
 563                 wpa_msg(wpa_s, MSG_INFO,
 564                         WPS_EVENT_FAIL "msg=%d config_error=%d reason=%d (%s)",
 565                         fail->msg, fail->config_error, fail->error_indication,
 566                         wps_ei_str(fail->error_indication));
 567                 if (wpa_s->parent && wpa_s->parent != wpa_s)
 568                         wpa_msg(wpa_s->parent, MSG_INFO, WPS_EVENT_FAIL
 569                                 "msg=%d config_error=%d reason=%d (%s)",
 570                                 fail->msg, fail->config_error,
 571                                 fail->error_indication,
 572                                 wps_ei_str(fail->error_indication));
 573         } else {
 574                 wpa_msg(wpa_s, MSG_INFO,
 575                         WPS_EVENT_FAIL "msg=%d config_error=%d",
 576                         fail->msg, fail->config_error);
 577                 if (wpa_s->parent && wpa_s->parent != wpa_s)
 578                         wpa_msg(wpa_s->parent, MSG_INFO, WPS_EVENT_FAIL
 579                                 "msg=%d config_error=%d",
 580                                 fail->msg, fail->config_error);
 581         }
 582
 583         /*
 584          * Need to allow WPS processing to complete, e.g., by sending WSC_NACK.
 585          */
 586         wpa_printf(MSG_DEBUG, "WPS: Register timeout to clear WPS network");
 587         eloop_cancel_timeout(wpas_wps_clear_timeout, wpa_s, NULL);
 588         eloop_register_timeout(0, 100000, wpas_wps_clear_timeout, wpa_s, NULL);
 589
 590         wpas_notify_wps_event_fail(wpa_s, fail);
 591 #ifdef CONFIG_P2P
 592         wpas_p2p_wps_failed(wpa_s, fail);
 593 #endif /* CONFIG_P2P */
 594 }
 595
 596
 597 static void wpas_wps_reenable_networks_cb(void *eloop_ctx, void *timeout_ctx);
 598
 599 static void wpas_wps_reenable_networks(struct wpa_supplicant *wpa_s)
 600 {
 601         struct wpa_ssid *ssid;
 602         int changed = 0;
 603
 604         eloop_cancel_timeout(wpas_wps_reenable_networks_cb, wpa_s, NULL);
 605
 606         for (ssid = wpa_s->conf->ssid; ssid; ssid = ssid->next) {
 607                 if (ssid->disabled_for_connect && ssid->disabled) {
 608                         ssid->disabled_for_connect = 0;
 609                         ssid->disabled = 0;
 610                         wpas_notify_network_enabled_changed(wpa_s, ssid);
 611                         changed++;
 612                 }
 613         }
 614
 615         if (changed) {
 616 #ifndef CONFIG_NO_CONFIG_WRITE
 617                 if (wpa_s->conf->update_config &&
 618                     wpa_config_write(wpa_s->confname, wpa_s->conf)) {
 619                         wpa_printf(MSG_DEBUG, "WPS: Failed to update "
 620                                    "configuration");
 621                 }
 622 #endif /* CONFIG_NO_CONFIG_WRITE */
 623         }
 624 }
 625
 626
 627 static void wpas_wps_reenable_networks_cb(void *eloop_ctx, void *timeout_ctx)
 628 {
 629         struct wpa_supplicant *wpa_s = eloop_ctx;
 630         /* Enable the networks disabled during wpas_wps_reassoc */
 631         wpas_wps_reenable_networks(wpa_s);
 632 }
 633
 634
 635 static void wpa_supplicant_wps_event_success(struct wpa_supplicant *wpa_s)
 636 {
 637         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_SUCCESS);
 638         wpa_s->wps_success = 1;
 639         wpas_notify_wps_event_success(wpa_s);
 640         if (wpa_s->current_ssid)
 641                 wpas_clear_temp_disabled(wpa_s, wpa_s->current_ssid, 1);
 642         wpa_s->extra_blacklist_count = 0;
 643
 644         /*
 645          * Enable the networks disabled during wpas_wps_reassoc after 10
 646          * seconds. The 10 seconds timer is to allow the data connection to be
 647          * formed before allowing other networks to be selected.
 648          */
 649         eloop_register_timeout(10, 0, wpas_wps_reenable_networks_cb, wpa_s,
 650                                NULL);
 651
 652 #ifdef CONFIG_P2P
 653         wpas_p2p_wps_success(wpa_s, wpa_s->bssid, 0);
 654 #endif /* CONFIG_P2P */
 655 }
 656
 657
 658 static void wpa_supplicant_wps_event_er_ap_add(struct wpa_supplicant *wpa_s,
 659                                                struct wps_event_er_ap *ap)
 660 {
 661         char uuid_str[100];
 662         char dev_type[WPS_DEV_TYPE_BUFSIZE];
 663
 664         uuid_bin2str(ap->uuid, uuid_str, sizeof(uuid_str));
 665         if (ap->pri_dev_type)
 666                 wps_dev_type_bin2str(ap->pri_dev_type, dev_type,
 667                                      sizeof(dev_type));
 668         else
 669                 dev_type[0] = '\0';
 670
 671         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_AP_ADD "%s " MACSTR
 672                 " pri_dev_type=%s wps_state=%d |%s|%s|%s|%s|%s|%s|",
 673                 uuid_str, MAC2STR(ap->mac_addr), dev_type, ap->wps_state,
 674                 ap->friendly_name ? ap->friendly_name : "",
 675                 ap->manufacturer ? ap->manufacturer : "",
 676                 ap->model_description ? ap->model_description : "",
 677                 ap->model_name ? ap->model_name : "",
 678                 ap->manufacturer_url ? ap->manufacturer_url : "",
 679                 ap->model_url ? ap->model_url : "");
 680 }
 681
 682
 683 static void wpa_supplicant_wps_event_er_ap_remove(struct wpa_supplicant *wpa_s,
 684                                                   struct wps_event_er_ap *ap)
 685 {
 686         char uuid_str[100];
 687         uuid_bin2str(ap->uuid, uuid_str, sizeof(uuid_str));
 688         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_AP_REMOVE "%s", uuid_str);
 689 }
 690
 691
 692 static void wpa_supplicant_wps_event_er_enrollee_add(
 693         struct wpa_supplicant *wpa_s, struct wps_event_er_enrollee *enrollee)
 694 {
 695         char uuid_str[100];
 696         char dev_type[WPS_DEV_TYPE_BUFSIZE];
 697
 698         uuid_bin2str(enrollee->uuid, uuid_str, sizeof(uuid_str));
 699         if (enrollee->pri_dev_type)
 700                 wps_dev_type_bin2str(enrollee->pri_dev_type, dev_type,
 701                                      sizeof(dev_type));
 702         else
 703                 dev_type[0] = '\0';
 704
 705         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_ENROLLEE_ADD "%s " MACSTR
 706                 " M1=%d config_methods=0x%x dev_passwd_id=%d pri_dev_type=%s "
 707                 "|%s|%s|%s|%s|%s|",
 708                 uuid_str, MAC2STR(enrollee->mac_addr), enrollee->m1_received,
 709                 enrollee->config_methods, enrollee->dev_passwd_id, dev_type,
 710                 enrollee->dev_name ? enrollee->dev_name : "",
 711                 enrollee->manufacturer ? enrollee->manufacturer : "",
 712                 enrollee->model_name ? enrollee->model_name : "",
 713                 enrollee->model_number ? enrollee->model_number : "",
 714                 enrollee->serial_number ? enrollee->serial_number : "");
 715 }
 716
 717
 718 static void wpa_supplicant_wps_event_er_enrollee_remove(
 719         struct wpa_supplicant *wpa_s, struct wps_event_er_enrollee *enrollee)
 720 {
 721         char uuid_str[100];
 722         uuid_bin2str(enrollee->uuid, uuid_str, sizeof(uuid_str));
 723         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_ENROLLEE_REMOVE "%s " MACSTR,
 724                 uuid_str, MAC2STR(enrollee->mac_addr));
 725 }
 726
 727
 728 static void wpa_supplicant_wps_event_er_ap_settings(
 729         struct wpa_supplicant *wpa_s,
 730         struct wps_event_er_ap_settings *ap_settings)
 731 {
 732         char uuid_str[100];
 733         char key_str[65];
 734         const struct wps_credential *cred = ap_settings->cred;
 735
 736         key_str[0] = '\0';
 737         if (cred->auth_type & (WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK)) {
 738                 if (cred->key_len >= 8 && cred->key_len <= 64) {
 739                         os_memcpy(key_str, cred->key, cred->key_len);
 740                         key_str[cred->key_len] = '\0';
 741                 }
 742         }
 743
 744         uuid_bin2str(ap_settings->uuid, uuid_str, sizeof(uuid_str));
 745         /* Use wpa_msg_ctrl to avoid showing the key in debug log */
 746         wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_ER_AP_SETTINGS
 747                      "uuid=%s ssid=%s auth_type=0x%04x encr_type=0x%04x "
 748                      "key=%s",
 749                      uuid_str, wpa_ssid_txt(cred->ssid, cred->ssid_len),
 750                      cred->auth_type, cred->encr_type, key_str);
 751 }
 752
 753
 754 static void wpa_supplicant_wps_event_er_set_sel_reg(
 755         struct wpa_supplicant *wpa_s,
 756         struct wps_event_er_set_selected_registrar *ev)
 757 {
 758         char uuid_str[100];
 759
 760         uuid_bin2str(ev->uuid, uuid_str, sizeof(uuid_str));
 761         switch (ev->state) {
 762         case WPS_ER_SET_SEL_REG_START:
 763                 wpa_msg(wpa_s, MSG_DEBUG, WPS_EVENT_ER_SET_SEL_REG
 764                         "uuid=%s state=START sel_reg=%d dev_passwd_id=%u "
 765                         "sel_reg_config_methods=0x%x",
 766                         uuid_str, ev->sel_reg, ev->dev_passwd_id,
 767                         ev->sel_reg_config_methods);
 768                 break;
 769         case WPS_ER_SET_SEL_REG_DONE:
 770                 wpa_msg(wpa_s, MSG_DEBUG, WPS_EVENT_ER_SET_SEL_REG
 771                         "uuid=%s state=DONE", uuid_str);
 772                 break;
 773         case WPS_ER_SET_SEL_REG_FAILED:
 774                 wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_SET_SEL_REG
 775                         "uuid=%s state=FAILED", uuid_str);
 776                 break;
 777         }
 778 }
 779
 780
 781 static void wpa_supplicant_wps_event(void *ctx, enum wps_event event,
 782                                      union wps_event_data *data)
 783 {
 784         struct wpa_supplicant *wpa_s = ctx;
 785         switch (event) {
 786         case WPS_EV_M2D:
 787                 wpa_supplicant_wps_event_m2d(wpa_s, &data->m2d);
 788                 break;
 789         case WPS_EV_FAIL:
 790                 wpa_supplicant_wps_event_fail(wpa_s, &data->fail);
 791                 break;
 792         case WPS_EV_SUCCESS:
 793                 wpa_supplicant_wps_event_success(wpa_s);
 794                 break;
 795         case WPS_EV_PWD_AUTH_FAIL:
 796 #ifdef CONFIG_AP
 797                 if (wpa_s->ap_iface && data->pwd_auth_fail.enrollee)
 798                         wpa_supplicant_ap_pwd_auth_fail(wpa_s);
 799 #endif /* CONFIG_AP */
 800                 break;
 801         case WPS_EV_PBC_OVERLAP:
 802                 break;
 803         case WPS_EV_PBC_TIMEOUT:
 804                 break;
 805         case WPS_EV_PBC_ACTIVE:
 806                 wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ACTIVE);
 807                 break;
 808         case WPS_EV_PBC_DISABLE:
 809                 wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_DISABLE);
 810                 break;
 811         case WPS_EV_ER_AP_ADD:
 812                 wpa_supplicant_wps_event_er_ap_add(wpa_s, &data->ap);
 813                 break;
 814         case WPS_EV_ER_AP_REMOVE:
 815                 wpa_supplicant_wps_event_er_ap_remove(wpa_s, &data->ap);
 816                 break;
 817         case WPS_EV_ER_ENROLLEE_ADD:
 818                 wpa_supplicant_wps_event_er_enrollee_add(wpa_s,
 819                                                          &data->enrollee);
 820                 break;
 821         case WPS_EV_ER_ENROLLEE_REMOVE:
 822                 wpa_supplicant_wps_event_er_enrollee_remove(wpa_s,
 823                                                             &data->enrollee);
 824                 break;
 825         case WPS_EV_ER_AP_SETTINGS:
 826                 wpa_supplicant_wps_event_er_ap_settings(wpa_s,
 827                                                         &data->ap_settings);
 828                 break;
 829         case WPS_EV_ER_SET_SELECTED_REGISTRAR:
 830                 wpa_supplicant_wps_event_er_set_sel_reg(wpa_s,
 831                                                         &data->set_sel_reg);
 832                 break;
 833         case WPS_EV_AP_PIN_SUCCESS:
 834                 break;
 835         }
 836 }
 837
 838
 839 static int wpa_supplicant_wps_rf_band(void *ctx)
 840 {
 841         struct wpa_supplicant *wpa_s = ctx;
 842
 843         if (!wpa_s->current_ssid || !wpa_s->assoc_freq)
 844                 return 0;
 845
 846         return (wpa_s->assoc_freq > 2484) ? WPS_RF_50GHZ : WPS_RF_24GHZ;
 847 }
 848
 849
 850 enum wps_request_type wpas_wps_get_req_type(struct wpa_ssid *ssid)
 851 {
 852         if (eap_is_wps_pbc_enrollee(&ssid->eap) ||
 853             eap_is_wps_pin_enrollee(&ssid->eap))
 854                 return WPS_REQ_ENROLLEE;
 855         else
 856                 return WPS_REQ_REGISTRAR;
 857 }
 858
 859
 860 static void wpas_clear_wps(struct wpa_supplicant *wpa_s)
 861 {
 862         int id;
 863         struct wpa_ssid *ssid, *remove_ssid = NULL, *prev_current;
 864
 865         wpa_s->after_wps = 0;
 866         wpa_s->known_wps_freq = 0;
 867
 868         prev_current = wpa_s->current_ssid;
 869
 870         /* Enable the networks disabled during wpas_wps_reassoc */
 871         wpas_wps_reenable_networks(wpa_s);
 872
 873         eloop_cancel_timeout(wpas_wps_timeout, wpa_s, NULL);
 874         eloop_cancel_timeout(wpas_wps_clear_timeout, wpa_s, NULL);
 875
 876         /* Remove any existing WPS network from configuration */
 877         ssid = wpa_s->conf->ssid;
 878         while (ssid) {
 879                 if (ssid->key_mgmt & WPA_KEY_MGMT_WPS) {
 880                         if (ssid == wpa_s->current_ssid) {
 881                                 wpa_supplicant_deauthenticate(
 882                                         wpa_s, WLAN_REASON_DEAUTH_LEAVING);
 883                         }
 884                         id = ssid->id;
 885                         remove_ssid = ssid;
 886                 } else
 887                         id = -1;
 888                 ssid = ssid->next;
 889                 if (id >= 0) {
 890                         if (prev_current == remove_ssid) {
 891                                 wpa_sm_set_config(wpa_s->wpa, NULL);
 892                                 eapol_sm_notify_config(wpa_s->eapol, NULL,
 893                                                        NULL);
 894                         }
 895                         wpas_notify_network_removed(wpa_s, remove_ssid);
 896                         wpa_config_remove_network(wpa_s->conf, id);
 897                 }
 898         }
 899
 900         wpas_wps_clear_ap_info(wpa_s);
 901 }
 902
 903
 904 static void wpas_wps_timeout(void *eloop_ctx, void *timeout_ctx)
 905 {
 906         struct wpa_supplicant *wpa_s = eloop_ctx;
 907         wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_TIMEOUT "Requested operation timed "
 908                 "out");
 909         wpas_clear_wps(wpa_s);
 910 }
 911
 912
 913 static struct wpa_ssid * wpas_wps_add_network(struct wpa_supplicant *wpa_s,
 914                                               int registrar, const u8 *dev_addr,
 915                                               const u8 *bssid)
 916 {
 917         struct wpa_ssid *ssid;
 918
 919         ssid = wpa_config_add_network(wpa_s->conf);
 920         if (ssid == NULL)
 921                 return NULL;
 922         wpas_notify_network_added(wpa_s, ssid);
 923         wpa_config_set_network_defaults(ssid);
 924         ssid->temporary = 1;
 925         if (wpa_config_set(ssid, "key_mgmt", "WPS", 0) < 0 ||
 926             wpa_config_set(ssid, "eap", "WSC", 0) < 0 ||
 927             wpa_config_set(ssid, "identity", registrar ?
 928                            "\"" WSC_ID_REGISTRAR "\"" :
 929                            "\"" WSC_ID_ENROLLEE "\"", 0) < 0) {
 930                 wpas_notify_network_removed(wpa_s, ssid);
 931                 wpa_config_remove_network(wpa_s->conf, ssid->id);
 932                 return NULL;
 933         }
 934
 935 #ifdef CONFIG_P2P
 936         if (dev_addr)
 937                 os_memcpy(ssid->go_p2p_dev_addr, dev_addr, ETH_ALEN);
 938 #endif /* CONFIG_P2P */
 939
 940         if (bssid) {
 941 #ifndef CONFIG_P2P
 942                 struct wpa_bss *bss;
 943                 int count = 0;
 944 #endif /* CONFIG_P2P */
 945
 946                 os_memcpy(ssid->bssid, bssid, ETH_ALEN);
 947                 ssid->bssid_set = 1;
 948
 949                 /*
 950                  * Note: With P2P, the SSID may change at the time the WPS
 951                  * provisioning is started, so better not filter the AP based
 952                  * on the current SSID in the scan results.
 953                  */
 954 #ifndef CONFIG_P2P
 955                 dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
 956                         if (os_memcmp(bssid, bss->bssid, ETH_ALEN) != 0)
 957                                 continue;
 958
 959                         os_free(ssid->ssid);
 960                         ssid->ssid = os_malloc(bss->ssid_len);
 961                         if (ssid->ssid == NULL)
 962                                 break;
 963                         os_memcpy(ssid->ssid, bss->ssid, bss->ssid_len);
 964                         ssid->ssid_len = bss->ssid_len;
 965                         wpa_hexdump_ascii(MSG_DEBUG, "WPS: Picked SSID from "
 966                                           "scan results",
 967                                           ssid->ssid, ssid->ssid_len);
 968                         count++;
 969                 }
 970
 971                 if (count > 1) {
 972                         wpa_printf(MSG_DEBUG, "WPS: More than one SSID found "
 973                                    "for the AP; use wildcard");
 974                         os_free(ssid->ssid);
 975                         ssid->ssid = NULL;
 976                         ssid->ssid_len = 0;
 977                 }
 978 #endif /* CONFIG_P2P */
 979         }
 980
 981         return ssid;
 982 }
 983
 984
 985 static void wpas_wps_temp_disable(struct wpa_supplicant *wpa_s,
 986                                   struct wpa_ssid *selected)
 987 {
 988         struct wpa_ssid *ssid;
 989
 990         if (wpa_s->current_ssid)
 991                 wpa_supplicant_deauthenticate(
 992                         wpa_s, WLAN_REASON_DEAUTH_LEAVING);
 993
 994         /* Mark all other networks disabled and trigger reassociation */
 995         ssid = wpa_s->conf->ssid;
 996         while (ssid) {
 997                 int was_disabled = ssid->disabled;
 998                 ssid->disabled_for_connect = 0;
 999                 /*
1000                  * In case the network object corresponds to a persistent group
1001                  * then do not send out network disabled signal. In addition,
1002                  * do not change disabled status of persistent network objects
1003                  * from 2 to 1 should we connect to another network.
1004                  */
1005                 if (was_disabled != 2) {
1006                         ssid->disabled = ssid != selected;
1007                         if (was_disabled != ssid->disabled) {
1008                                 if (ssid->disabled)
1009                                         ssid->disabled_for_connect = 1;
1010                                 wpas_notify_network_enabled_changed(wpa_s,
1011                                                                     ssid);
1012                         }
1013                 }
1014                 ssid = ssid->next;
1015         }
1016 }
1017
1018
1019 static void wpas_wps_reassoc(struct wpa_supplicant *wpa_s,
1020                              struct wpa_ssid *selected, const u8 *bssid)
1021 {
1022         struct wpa_bss *bss;
1023
1024         wpa_s->after_wps = 0;
1025         wpa_s->known_wps_freq = 0;
1026         if (bssid) {
1027                 bss = wpa_bss_get_bssid_latest(wpa_s, bssid);
1028                 if (bss && bss->freq > 0) {
1029                         wpa_s->known_wps_freq = 1;
1030                         wpa_s->wps_freq = bss->freq;
1031                 }
1032         }
1033
1034         wpas_wps_temp_disable(wpa_s, selected);
1035
1036         wpa_s->disconnected = 0;
1037         wpa_s->reassociate = 1;
1038         wpa_s->scan_runs = 0;
1039         wpa_s->normal_scans = 0;
1040         wpa_s->wps_success = 0;
1041         wpa_s->blacklist_cleared = 0;
1042
1043         wpa_supplicant_cancel_sched_scan(wpa_s);
1044         wpa_supplicant_req_scan(wpa_s, 0, 0);
1045 }
1046
1047
1048 int wpas_wps_start_pbc(struct wpa_supplicant *wpa_s, const u8 *bssid,
1049                        int p2p_group)
1050 {
1051         struct wpa_ssid *ssid;
1052         wpas_clear_wps(wpa_s);
1053         ssid = wpas_wps_add_network(wpa_s, 0, NULL, bssid);
1054         if (ssid == NULL)
1055                 return -1;
1056         ssid->temporary = 1;
1057         ssid->p2p_group = p2p_group;
1058 #ifdef CONFIG_P2P
1059         if (p2p_group && wpa_s->go_params && wpa_s->go_params->ssid_len) {
1060                 ssid->ssid = os_zalloc(wpa_s->go_params->ssid_len + 1);
1061                 if (ssid->ssid) {
1062                         ssid->ssid_len = wpa_s->go_params->ssid_len;
1063                         os_memcpy(ssid->ssid, wpa_s->go_params->ssid,
1064                                   ssid->ssid_len);
1065                         wpa_hexdump_ascii(MSG_DEBUG, "WPS: Use specific AP "
1066                                           "SSID", ssid->ssid, ssid->ssid_len);
1067                 }
1068         }
1069 #endif /* CONFIG_P2P */
1070         if (wpa_config_set(ssid, "phase1", "\"pbc=1\"", 0) < 0)
1071                 return -1;
1072         if (wpa_s->wps_fragment_size)
1073                 ssid->eap.fragment_size = wpa_s->wps_fragment_size;
1074         eloop_register_timeout(WPS_PBC_WALK_TIME, 0, wpas_wps_timeout,
1075                                wpa_s, NULL);
1076         wpas_wps_reassoc(wpa_s, ssid, bssid);
1077         return 0;
1078 }
1079
1080
1081 static int wpas_wps_start_dev_pw(struct wpa_supplicant *wpa_s,
1082                                  const u8 *dev_addr, const u8 *bssid,
1083                                  const char *pin, int p2p_group, u16 dev_pw_id,
1084                                  const u8 *peer_pubkey_hash,
1085                                  const u8 *ssid_val, size_t ssid_len)
1086 {
1087         struct wpa_ssid *ssid;
1088         char val[128 + 2 * WPS_OOB_PUBKEY_HASH_LEN];
1089         unsigned int rpin = 0;
1090         char hash[2 * WPS_OOB_PUBKEY_HASH_LEN + 10];
1091
1092         wpas_clear_wps(wpa_s);
1093         if (bssid && is_zero_ether_addr(bssid))
1094                 bssid = NULL;
1095         ssid = wpas_wps_add_network(wpa_s, 0, dev_addr, bssid);
1096         if (ssid == NULL) {
1097                 wpa_printf(MSG_DEBUG, "WPS: Could not add network");
1098                 return -1;
1099         }
1100         ssid->temporary = 1;
1101         ssid->p2p_group = p2p_group;
1102         if (ssid_val) {
1103                 ssid->ssid = os_malloc(ssid_len);
1104                 if (ssid->ssid) {
1105                         os_memcpy(ssid->ssid, ssid_val, ssid_len);
1106                         ssid->ssid_len = ssid_len;
1107                 }
1108         }
1109         if (peer_pubkey_hash) {
1110                 os_memcpy(hash, " pkhash=", 8);
1111                 wpa_snprintf_hex_uppercase(hash + 8, sizeof(hash) - 8,
1112                                            peer_pubkey_hash,
1113                                            WPS_OOB_PUBKEY_HASH_LEN);
1114         } else {
1115                 hash[0] = '\0';
1116         }
1117 #ifdef CONFIG_P2P
1118         if (p2p_group && wpa_s->go_params && wpa_s->go_params->ssid_len) {
1119                 ssid->ssid = os_zalloc(wpa_s->go_params->ssid_len + 1);
1120                 if (ssid->ssid) {
1121                         ssid->ssid_len = wpa_s->go_params->ssid_len;
1122                         os_memcpy(ssid->ssid, wpa_s->go_params->ssid,
1123                                   ssid->ssid_len);
1124                         wpa_hexdump_ascii(MSG_DEBUG, "WPS: Use specific AP "
1125                                           "SSID", ssid->ssid, ssid->ssid_len);
1126                 }
1127         }
1128 #endif /* CONFIG_P2P */
1129         if (pin)
1130                 os_snprintf(val, sizeof(val), "\"pin=%s dev_pw_id=%u%s\"",
1131                             pin, dev_pw_id, hash);
1132         else if (pin == NULL && dev_pw_id == DEV_PW_NFC_CONNECTION_HANDOVER) {
1133                 os_snprintf(val, sizeof(val), "\"dev_pw_id=%u%s\"",
1134                             dev_pw_id, hash);
1135         } else {
1136                 rpin = wps_generate_pin();
1137                 os_snprintf(val, sizeof(val), "\"pin=%08d dev_pw_id=%u%s\"",
1138                             rpin, dev_pw_id, hash);
1139         }
1140         if (wpa_config_set(ssid, "phase1", val, 0) < 0) {
1141                 wpa_printf(MSG_DEBUG, "WPS: Failed to set phase1 '%s'", val);
1142                 return -1;
1143         }
1144         if (wpa_s->wps_fragment_size)
1145                 ssid->eap.fragment_size = wpa_s->wps_fragment_size;
1146         eloop_register_timeout(WPS_PBC_WALK_TIME, 0, wpas_wps_timeout,
1147                                wpa_s, NULL);
1148         wpa_s->wps_ap_iter = 1;
1149         wpas_wps_reassoc(wpa_s, ssid, bssid);
1150         return rpin;
1151 }
1152
1153
1154 int wpas_wps_start_pin(struct wpa_supplicant *wpa_s, const u8 *bssid,
1155                        const char *pin, int p2p_group, u16 dev_pw_id)
1156 {
1157         return wpas_wps_start_dev_pw(wpa_s, NULL, bssid, pin, p2p_group,
1158                                      dev_pw_id, NULL, NULL, 0);
1159 }
1160
1161
1162 /* Cancel the wps pbc/pin requests */
1163 int wpas_wps_cancel(struct wpa_supplicant *wpa_s)
1164 {
1165 #ifdef CONFIG_AP
1166         if (wpa_s->ap_iface) {
1167                 wpa_printf(MSG_DEBUG, "WPS: Cancelling in AP mode");
1168                 return wpa_supplicant_ap_wps_cancel(wpa_s);
1169         }
1170 #endif /* CONFIG_AP */
1171
1172         if (wpa_s->wpa_state == WPA_SCANNING ||
1173             wpa_s->wpa_state == WPA_DISCONNECTED) {
1174                 wpa_printf(MSG_DEBUG, "WPS: Cancel operation - cancel scan");
1175                 wpa_supplicant_cancel_scan(wpa_s);
1176                 wpas_clear_wps(wpa_s);
1177         } else if (wpa_s->wpa_state >= WPA_ASSOCIATED) {
1178                 wpa_printf(MSG_DEBUG, "WPS: Cancel operation - "
1179                            "deauthenticate");
1180                 wpa_supplicant_deauthenticate(wpa_s,
1181                                               WLAN_REASON_DEAUTH_LEAVING);
1182                 wpas_clear_wps(wpa_s);
1183         } else {
1184                 wpas_wps_reenable_networks(wpa_s);
1185                 wpas_wps_clear_ap_info(wpa_s);
1186                 if (eloop_cancel_timeout(wpas_wps_clear_timeout, wpa_s, NULL) >
1187                     0)
1188                         wpas_clear_wps(wpa_s);
1189         }
1190
1191         wpa_s->after_wps = 0;
1192
1193         return 0;
1194 }
1195
1196
1197 int wpas_wps_start_reg(struct wpa_supplicant *wpa_s, const u8 *bssid,
1198                        const char *pin, struct wps_new_ap_settings *settings)
1199 {
1200         struct wpa_ssid *ssid;
1201         char val[200];
1202         char *pos, *end;
1203         int res;
1204
1205         if (!pin)
1206                 return -1;
1207         wpas_clear_wps(wpa_s);
1208         ssid = wpas_wps_add_network(wpa_s, 1, NULL, bssid);
1209         if (ssid == NULL)
1210                 return -1;
1211         ssid->temporary = 1;
1212         pos = val;
1213         end = pos + sizeof(val);
1214         res = os_snprintf(pos, end - pos, "\"pin=%s", pin);
1215         if (res < 0 || res >= end - pos)
1216                 return -1;
1217         pos += res;
1218         if (settings) {
1219                 res = os_snprintf(pos, end - pos, " new_ssid=%s new_auth=%s "
1220                                   "new_encr=%s new_key=%s",
1221                                   settings->ssid_hex, settings->auth,
1222                                   settings->encr, settings->key_hex);
1223                 if (res < 0 || res >= end - pos)
1224                         return -1;
1225                 pos += res;
1226         }
1227         res = os_snprintf(pos, end - pos, "\"");
1228         if (res < 0 || res >= end - pos)
1229                 return -1;
1230         if (wpa_config_set(ssid, "phase1", val, 0) < 0)
1231                 return -1;
1232         if (wpa_s->wps_fragment_size)
1233                 ssid->eap.fragment_size = wpa_s->wps_fragment_size;
1234         eloop_register_timeout(WPS_PBC_WALK_TIME, 0, wpas_wps_timeout,
1235                                wpa_s, NULL);
1236         wpas_wps_reassoc(wpa_s, ssid, bssid);
1237         return 0;
1238 }
1239
1240
1241 static int wpas_wps_new_psk_cb(void *ctx, const u8 *mac_addr,
1242                                const u8 *p2p_dev_addr, const u8 *psk,
1243                                size_t psk_len)
1244 {
1245         if (is_zero_ether_addr(p2p_dev_addr)) {
1246                 wpa_printf(MSG_DEBUG,
1247                            "Received new WPA/WPA2-PSK from WPS for STA " MACSTR,
1248                            MAC2STR(mac_addr));
1249         } else {
1250                 wpa_printf(MSG_DEBUG,
1251                            "Received new WPA/WPA2-PSK from WPS for STA " MACSTR
1252                            " P2P Device Addr " MACSTR,
1253                            MAC2STR(mac_addr), MAC2STR(p2p_dev_addr));
1254         }
1255         wpa_hexdump_key(MSG_DEBUG, "Per-device PSK", psk, psk_len);
1256
1257         /* TODO */
1258
1259         return 0;
1260 }
1261
1262
1263 static void wpas_wps_pin_needed_cb(void *ctx, const u8 *uuid_e,
1264                                    const struct wps_device_data *dev)
1265 {
1266         char uuid[40], txt[400];
1267         int len;
1268         char devtype[WPS_DEV_TYPE_BUFSIZE];
1269         if (uuid_bin2str(uuid_e, uuid, sizeof(uuid)))
1270                 return;
1271         wpa_printf(MSG_DEBUG, "WPS: PIN needed for UUID-E %s", uuid);
1272         len = os_snprintf(txt, sizeof(txt), "WPS-EVENT-PIN-NEEDED %s " MACSTR
1273                           " [%s|%s|%s|%s|%s|%s]",
1274                           uuid, MAC2STR(dev->mac_addr), dev->device_name,
1275                           dev->manufacturer, dev->model_name,
1276                           dev->model_number, dev->serial_number,
1277                           wps_dev_type_bin2str(dev->pri_dev_type, devtype,
1278                                                sizeof(devtype)));
1279         if (len > 0 && len < (int) sizeof(txt))
1280                 wpa_printf(MSG_INFO, "%s", txt);
1281 }
1282
1283
1284 static void wpas_wps_set_sel_reg_cb(void *ctx, int sel_reg, u16 dev_passwd_id,
1285                                     u16 sel_reg_config_methods)
1286 {
1287 #ifdef CONFIG_WPS_ER
1288         struct wpa_supplicant *wpa_s = ctx;
1289
1290         if (wpa_s->wps_er == NULL)
1291                 return;
1292         wpa_printf(MSG_DEBUG, "WPS ER: SetSelectedRegistrar - sel_reg=%d "
1293                    "dev_password_id=%u sel_reg_config_methods=0x%x",
1294                    sel_reg, dev_passwd_id, sel_reg_config_methods);
1295         wps_er_set_sel_reg(wpa_s->wps_er, sel_reg, dev_passwd_id,
1296                            sel_reg_config_methods);
1297 #endif /* CONFIG_WPS_ER */
1298 }
1299
1300
1301 static u16 wps_fix_config_methods(u16 config_methods)
1302 {
1303 #ifdef CONFIG_WPS2
1304         if ((config_methods &
1305              (WPS_CONFIG_DISPLAY | WPS_CONFIG_VIRT_DISPLAY |
1306               WPS_CONFIG_PHY_DISPLAY)) == WPS_CONFIG_DISPLAY) {
1307                 wpa_printf(MSG_INFO, "WPS: Converting display to "
1308                            "virtual_display for WPS 2.0 compliance");
1309                 config_methods |= WPS_CONFIG_VIRT_DISPLAY;
1310         }
1311         if ((config_methods &
1312              (WPS_CONFIG_PUSHBUTTON | WPS_CONFIG_VIRT_PUSHBUTTON |
1313               WPS_CONFIG_PHY_PUSHBUTTON)) == WPS_CONFIG_PUSHBUTTON) {
1314                 wpa_printf(MSG_INFO, "WPS: Converting push_button to "
1315                            "virtual_push_button for WPS 2.0 compliance");
1316                 config_methods |= WPS_CONFIG_VIRT_PUSHBUTTON;
1317         }
1318 #endif /* CONFIG_WPS2 */
1319
1320         return config_methods;
1321 }
1322
1323
1324 static void wpas_wps_set_uuid(struct wpa_supplicant *wpa_s,
1325                               struct wps_context *wps)
1326 {
1327         char buf[50];
1328         const char *src;
1329
1330         if (is_nil_uuid(wpa_s->conf->uuid)) {
1331                 struct wpa_supplicant *first;
1332                 first = wpa_s->global->ifaces;
1333                 while (first && first->next)
1334                         first = first->next;
1335                 if (first && first != wpa_s) {
1336                         if (wps != wpa_s->global->ifaces->wps)
1337                                 os_memcpy(wps->uuid,
1338                                           wpa_s->global->ifaces->wps->uuid,
1339                                           WPS_UUID_LEN);
1340                         src = "from the first interface";
1341                 } else {
1342                         uuid_gen_mac_addr(wpa_s->own_addr, wps->uuid);
1343                         src = "based on MAC address";
1344                 }
1345         } else {
1346                 os_memcpy(wps->uuid, wpa_s->conf->uuid, WPS_UUID_LEN);
1347                 src = "based on configuration";
1348         }
1349
1350         uuid_bin2str(wps->uuid, buf, sizeof(buf));
1351         wpa_dbg(wpa_s, MSG_DEBUG, "WPS: UUID %s: %s", src, buf);
1352 }
1353
1354
1355 static void wpas_wps_set_vendor_ext_m1(struct wpa_supplicant *wpa_s,
1356                                        struct wps_context *wps)
1357 {
1358         wpabuf_free(wps->dev.vendor_ext_m1);
1359         wps->dev.vendor_ext_m1 = NULL;
1360
1361         if (wpa_s->conf->wps_vendor_ext_m1) {
1362                 wps->dev.vendor_ext_m1 =
1363                         wpabuf_dup(wpa_s->conf->wps_vendor_ext_m1);
1364                 if (!wps->dev.vendor_ext_m1) {
1365                         wpa_printf(MSG_ERROR, "WPS: Cannot "
1366                                    "allocate memory for vendor_ext_m1");
1367                 }
1368         }
1369 }
1370
1371
1372 int wpas_wps_init(struct wpa_supplicant *wpa_s)
1373 {
1374         struct wps_context *wps;
1375         struct wps_registrar_config rcfg;
1376         struct hostapd_hw_modes *modes;
1377         u16 m;
1378
1379         wps = os_zalloc(sizeof(*wps));
1380         if (wps == NULL)
1381                 return -1;
1382
1383         wps->cred_cb = wpa_supplicant_wps_cred;
1384         wps->event_cb = wpa_supplicant_wps_event;
1385         wps->rf_band_cb = wpa_supplicant_wps_rf_band;
1386         wps->cb_ctx = wpa_s;
1387
1388         wps->dev.device_name = wpa_s->conf->device_name;
1389         wps->dev.manufacturer = wpa_s->conf->manufacturer;
1390         wps->dev.model_name = wpa_s->conf->model_name;
1391         wps->dev.model_number = wpa_s->conf->model_number;
1392         wps->dev.serial_number = wpa_s->conf->serial_number;
1393         wps->config_methods =
1394                 wps_config_methods_str2bin(wpa_s->conf->config_methods);
1395         if ((wps->config_methods & (WPS_CONFIG_DISPLAY | WPS_CONFIG_LABEL)) ==
1396             (WPS_CONFIG_DISPLAY | WPS_CONFIG_LABEL)) {
1397                 wpa_printf(MSG_ERROR, "WPS: Both Label and Display config "
1398                            "methods are not allowed at the same time");
1399                 os_free(wps);
1400                 return -1;
1401         }
1402         wps->config_methods = wps_fix_config_methods(wps->config_methods);
1403         wps->dev.config_methods = wps->config_methods;
1404         os_memcpy(wps->dev.pri_dev_type, wpa_s->conf->device_type,
1405                   WPS_DEV_TYPE_LEN);
1406
1407         wps->dev.num_sec_dev_types = wpa_s->conf->num_sec_device_types;
1408         os_memcpy(wps->dev.sec_dev_type, wpa_s->conf->sec_device_type,
1409                   WPS_DEV_TYPE_LEN * wps->dev.num_sec_dev_types);
1410
1411         wpas_wps_set_vendor_ext_m1(wpa_s, wps);
1412
1413         wps->dev.os_version = WPA_GET_BE32(wpa_s->conf->os_version);
1414         modes = wpa_s->hw.modes;
1415         if (modes) {
1416                 for (m = 0; m < wpa_s->hw.num_modes; m++) {
1417                         if (modes[m].mode == HOSTAPD_MODE_IEEE80211B ||
1418                             modes[m].mode == HOSTAPD_MODE_IEEE80211G)
1419                                 wps->dev.rf_bands |= WPS_RF_24GHZ;
1420                         else if (modes[m].mode == HOSTAPD_MODE_IEEE80211A)
1421                                 wps->dev.rf_bands |= WPS_RF_50GHZ;
1422                 }
1423         }
1424         if (wps->dev.rf_bands == 0) {
1425                 /*
1426                  * Default to claiming support for both bands if the driver
1427                  * does not provide support for fetching supported bands.
1428                  */
1429                 wps->dev.rf_bands = WPS_RF_24GHZ | WPS_RF_50GHZ;
1430         }
1431         os_memcpy(wps->dev.mac_addr, wpa_s->own_addr, ETH_ALEN);
1432         wpas_wps_set_uuid(wpa_s, wps);
1433
1434         wps->auth_types = WPS_AUTH_WPA2PSK | WPS_AUTH_WPAPSK;
1435         wps->encr_types = WPS_ENCR_AES | WPS_ENCR_TKIP;
1436
1437         os_memset(&rcfg, 0, sizeof(rcfg));
1438         rcfg.new_psk_cb = wpas_wps_new_psk_cb;
1439         rcfg.pin_needed_cb = wpas_wps_pin_needed_cb;
1440         rcfg.set_sel_reg_cb = wpas_wps_set_sel_reg_cb;
1441         rcfg.cb_ctx = wpa_s;
1442
1443         wps->registrar = wps_registrar_init(wps, &rcfg);
1444         if (wps->registrar == NULL) {
1445                 wpa_printf(MSG_DEBUG, "Failed to initialize WPS Registrar");
1446                 os_free(wps);
1447                 return -1;
1448         }
1449
1450         wpa_s->wps = wps;
1451
1452         return 0;
1453 }
1454
1455
1456 #ifdef CONFIG_WPS_ER
1457 static void wpas_wps_nfc_clear(struct wps_context *wps)
1458 {
1459         wps->ap_nfc_dev_pw_id = 0;
1460         wpabuf_free(wps->ap_nfc_dh_pubkey);
1461         wps->ap_nfc_dh_pubkey = NULL;
1462         wpabuf_free(wps->ap_nfc_dh_privkey);
1463         wps->ap_nfc_dh_privkey = NULL;
1464         wpabuf_free(wps->ap_nfc_dev_pw);
1465         wps->ap_nfc_dev_pw = NULL;
1466 }
1467 #endif /* CONFIG_WPS_ER */
1468
1469
1470 void wpas_wps_deinit(struct wpa_supplicant *wpa_s)
1471 {
1472         eloop_cancel_timeout(wpas_wps_timeout, wpa_s, NULL);
1473         eloop_cancel_timeout(wpas_wps_clear_timeout, wpa_s, NULL);
1474         eloop_cancel_timeout(wpas_wps_reenable_networks_cb, wpa_s, NULL);
1475         wpas_wps_clear_ap_info(wpa_s);
1476
1477         if (wpa_s->wps == NULL)
1478                 return;
1479
1480 #ifdef CONFIG_WPS_ER
1481         wps_er_deinit(wpa_s->wps_er, NULL, NULL);
1482         wpa_s->wps_er = NULL;
1483         wpas_wps_nfc_clear(wpa_s->wps);
1484 #endif /* CONFIG_WPS_ER */
1485
1486         wps_registrar_deinit(wpa_s->wps->registrar);
1487         wpabuf_free(wpa_s->wps->dh_pubkey);
1488         wpabuf_free(wpa_s->wps->dh_privkey);
1489         wpabuf_free(wpa_s->wps->dev.vendor_ext_m1);
1490         os_free(wpa_s->wps->network_key);
1491         os_free(wpa_s->wps);
1492         wpa_s->wps = NULL;
1493 }
1494
1495
1496 int wpas_wps_ssid_bss_match(struct wpa_supplicant *wpa_s,
1497                             struct wpa_ssid *ssid, struct wpa_bss *bss)
1498 {
1499         struct wpabuf *wps_ie;
1500
1501         if (!(ssid->key_mgmt & WPA_KEY_MGMT_WPS))
1502                 return -1;
1503
1504         wps_ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1505         if (eap_is_wps_pbc_enrollee(&ssid->eap)) {
1506                 if (!wps_ie) {
1507                         wpa_printf(MSG_DEBUG, "   skip - non-WPS AP");
1508                         return 0;
1509                 }
1510
1511                 if (!wps_is_selected_pbc_registrar(wps_ie)) {
1512                         wpa_printf(MSG_DEBUG, "   skip - WPS AP "
1513                                    "without active PBC Registrar");
1514                         wpabuf_free(wps_ie);
1515                         return 0;
1516                 }
1517
1518                 /* TODO: overlap detection */
1519                 wpa_printf(MSG_DEBUG, "   selected based on WPS IE "
1520                            "(Active PBC)");
1521                 wpabuf_free(wps_ie);
1522                 return 1;
1523         }
1524
1525         if (eap_is_wps_pin_enrollee(&ssid->eap)) {
1526                 if (!wps_ie) {
1527                         wpa_printf(MSG_DEBUG, "   skip - non-WPS AP");
1528                         return 0;
1529                 }
1530
1531                 /*
1532                  * Start with WPS APs that advertise our address as an
1533                  * authorized MAC (v2.0) or active PIN Registrar (v1.0) and
1534                  * allow any WPS AP after couple of scans since some APs do not
1535                  * set Selected Registrar attribute properly when using
1536                  * external Registrar.
1537                  */
1538                 if (!wps_is_addr_authorized(wps_ie, wpa_s->own_addr, 1)) {
1539                         if (wpa_s->scan_runs < WPS_PIN_SCAN_IGNORE_SEL_REG) {
1540                                 wpa_printf(MSG_DEBUG, "   skip - WPS AP "
1541                                            "without active PIN Registrar");
1542                                 wpabuf_free(wps_ie);
1543                                 return 0;
1544                         }
1545                         wpa_printf(MSG_DEBUG, "   selected based on WPS IE");
1546                 } else {
1547                         wpa_printf(MSG_DEBUG, "   selected based on WPS IE "
1548                                    "(Authorized MAC or Active PIN)");
1549                 }
1550                 wpabuf_free(wps_ie);
1551                 return 1;
1552         }
1553
1554         if (wps_ie) {
1555                 wpa_printf(MSG_DEBUG, "   selected based on WPS IE");
1556                 wpabuf_free(wps_ie);
1557                 return 1;
1558         }
1559
1560         return -1;
1561 }
1562
1563
1564 int wpas_wps_ssid_wildcard_ok(struct wpa_supplicant *wpa_s,
1565                               struct wpa_ssid *ssid,
1566                               struct wpa_bss *bss)
1567 {
1568         struct wpabuf *wps_ie = NULL;
1569         int ret = 0;
1570
1571         if (eap_is_wps_pbc_enrollee(&ssid->eap)) {
1572                 wps_ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1573                 if (wps_ie && wps_is_selected_pbc_registrar(wps_ie)) {
1574                         /* allow wildcard SSID for WPS PBC */
1575                         ret = 1;
1576                 }
1577         } else if (eap_is_wps_pin_enrollee(&ssid->eap)) {
1578                 wps_ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1579                 if (wps_ie &&
1580                     (wps_is_addr_authorized(wps_ie, wpa_s->own_addr, 1) ||
1581                      wpa_s->scan_runs >= WPS_PIN_SCAN_IGNORE_SEL_REG)) {
1582                         /* allow wildcard SSID for WPS PIN */
1583                         ret = 1;
1584                 }
1585         }
1586
1587         if (!ret && ssid->bssid_set &&
1588             os_memcmp(ssid->bssid, bss->bssid, ETH_ALEN) == 0) {
1589                 /* allow wildcard SSID due to hardcoded BSSID match */
1590                 ret = 1;
1591         }
1592
1593 #ifdef CONFIG_WPS_STRICT
1594         if (wps_ie) {
1595                 if (wps_validate_beacon_probe_resp(wps_ie, bss->beacon_ie_len >
1596                                                    0, bss->bssid) < 0)
1597                         ret = 0;
1598                 if (bss->beacon_ie_len) {
1599                         struct wpabuf *bcn_wps;
1600                         bcn_wps = wpa_bss_get_vendor_ie_multi_beacon(
1601                                 bss, WPS_IE_VENDOR_TYPE);
1602                         if (bcn_wps == NULL) {
1603                                 wpa_printf(MSG_DEBUG, "WPS: Mandatory WPS IE "
1604                                            "missing from AP Beacon");
1605                                 ret = 0;
1606                         } else {
1607                                 if (wps_validate_beacon(wps_ie) < 0)
1608                                         ret = 0;
1609                                 wpabuf_free(bcn_wps);
1610                         }
1611                 }
1612         }
1613 #endif /* CONFIG_WPS_STRICT */
1614
1615         wpabuf_free(wps_ie);
1616
1617         return ret;
1618 }
1619
1620
1621 int wpas_wps_scan_pbc_overlap(struct wpa_supplicant *wpa_s,
1622                               struct wpa_bss *selected, struct wpa_ssid *ssid)
1623 {
1624         const u8 *sel_uuid, *uuid;
1625         struct wpabuf *wps_ie;
1626         int ret = 0;
1627         struct wpa_bss *bss;
1628
1629         if (!eap_is_wps_pbc_enrollee(&ssid->eap))
1630                 return 0;
1631
1632         wpa_printf(MSG_DEBUG, "WPS: Check whether PBC session overlap is "
1633                    "present in scan results; selected BSSID " MACSTR,
1634                    MAC2STR(selected->bssid));
1635
1636         /* Make sure that only one AP is in active PBC mode */
1637         wps_ie = wpa_bss_get_vendor_ie_multi(selected, WPS_IE_VENDOR_TYPE);
1638         if (wps_ie) {
1639                 sel_uuid = wps_get_uuid_e(wps_ie);
1640                 wpa_hexdump(MSG_DEBUG, "WPS: UUID of the selected BSS",
1641                             sel_uuid, UUID_LEN);
1642         } else {
1643                 wpa_printf(MSG_DEBUG, "WPS: Selected BSS does not include "
1644                            "WPS IE?!");
1645                 sel_uuid = NULL;
1646         }
1647
1648         dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
1649                 struct wpabuf *ie;
1650                 if (bss == selected)
1651                         continue;
1652                 ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1653                 if (!ie)
1654                         continue;
1655                 if (!wps_is_selected_pbc_registrar(ie)) {
1656                         wpabuf_free(ie);
1657                         continue;
1658                 }
1659                 wpa_printf(MSG_DEBUG, "WPS: Another BSS in active PBC mode: "
1660                            MACSTR, MAC2STR(bss->bssid));
1661                 uuid = wps_get_uuid_e(ie);
1662                 wpa_hexdump(MSG_DEBUG, "WPS: UUID of the other BSS",
1663                             uuid, UUID_LEN);
1664                 if (sel_uuid == NULL || uuid == NULL ||
1665                     os_memcmp(sel_uuid, uuid, UUID_LEN) != 0) {
1666                         ret = 1; /* PBC overlap */
1667                         wpa_msg(wpa_s, MSG_INFO, "WPS: PBC overlap detected: "
1668                                 MACSTR " and " MACSTR,
1669                                 MAC2STR(selected->bssid),
1670                                 MAC2STR(bss->bssid));
1671                         wpabuf_free(ie);
1672                         break;
1673                 }
1674
1675                 /* TODO: verify that this is reasonable dual-band situation */
1676
1677                 wpabuf_free(ie);
1678         }
1679
1680         wpabuf_free(wps_ie);
1681
1682         return ret;
1683 }
1684
1685
1686 void wpas_wps_notify_scan_results(struct wpa_supplicant *wpa_s)
1687 {
1688         struct wpa_bss *bss;
1689         unsigned int pbc = 0, auth = 0, pin = 0, wps = 0;
1690
1691         if (wpa_s->disconnected || wpa_s->wpa_state >= WPA_ASSOCIATED)
1692                 return;
1693
1694         dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
1695                 struct wpabuf *ie;
1696                 ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1697                 if (!ie)
1698                         continue;
1699                 if (wps_is_selected_pbc_registrar(ie))
1700                         pbc++;
1701                 else if (wps_is_addr_authorized(ie, wpa_s->own_addr, 0))
1702                         auth++;
1703                 else if (wps_is_selected_pin_registrar(ie))
1704                         pin++;
1705                 else
1706                         wps++;
1707                 wpabuf_free(ie);
1708         }
1709
1710         if (pbc)
1711                 wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_AP_AVAILABLE_PBC);
1712         else if (auth)
1713                 wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_AP_AVAILABLE_AUTH);
1714         else if (pin)
1715                 wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_AP_AVAILABLE_PIN);
1716         else if (wps)
1717                 wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_AP_AVAILABLE);
1718 }
1719
1720
1721 int wpas_wps_searching(struct wpa_supplicant *wpa_s)
1722 {
1723         struct wpa_ssid *ssid;
1724
1725         for (ssid = wpa_s->conf->ssid; ssid; ssid = ssid->next) {
1726                 if ((ssid->key_mgmt & WPA_KEY_MGMT_WPS) && !ssid->disabled)
1727                         return 1;
1728         }
1729
1730         return 0;
1731 }
1732
1733
1734 int wpas_wps_scan_result_text(const u8 *ies, size_t ies_len, char *buf,
1735                               char *end)
1736 {
1737         struct wpabuf *wps_ie;
1738         int ret;
1739
1740         wps_ie = ieee802_11_vendor_ie_concat(ies, ies_len, WPS_DEV_OUI_WFA);
1741         if (wps_ie == NULL)
1742                 return 0;
1743
1744         ret = wps_attr_text(wps_ie, buf, end);
1745         wpabuf_free(wps_ie);
1746         return ret;
1747 }
1748
1749
1750 int wpas_wps_er_start(struct wpa_supplicant *wpa_s, const char *filter)
1751 {
1752 #ifdef CONFIG_WPS_ER
1753         if (wpa_s->wps_er) {
1754                 wps_er_refresh(wpa_s->wps_er);
1755                 return 0;
1756         }
1757         wpa_s->wps_er = wps_er_init(wpa_s->wps, wpa_s->ifname, filter);
1758         if (wpa_s->wps_er == NULL)
1759                 return -1;
1760         return 0;
1761 #else /* CONFIG_WPS_ER */
1762         return 0;
1763 #endif /* CONFIG_WPS_ER */
1764 }
1765
1766
1767 int wpas_wps_er_stop(struct wpa_supplicant *wpa_s)
1768 {
1769 #ifdef CONFIG_WPS_ER
1770         wps_er_deinit(wpa_s->wps_er, NULL, NULL);
1771         wpa_s->wps_er = NULL;
1772 #endif /* CONFIG_WPS_ER */
1773         return 0;
1774 }
1775
1776
1777 #ifdef CONFIG_WPS_ER
1778 int wpas_wps_er_add_pin(struct wpa_supplicant *wpa_s, const u8 *addr,
1779                         const char *uuid, const char *pin)
1780 {
1781         u8 u[UUID_LEN];
1782         const u8 *use_uuid = NULL;
1783         u8 addr_buf[ETH_ALEN];
1784
1785         if (os_strcmp(uuid, "any") == 0) {
1786         } else if (uuid_str2bin(uuid, u) == 0) {
1787                 use_uuid = u;
1788         } else if (hwaddr_aton(uuid, addr_buf) == 0) {
1789                 use_uuid = wps_er_get_sta_uuid(wpa_s->wps_er, addr_buf);
1790                 if (use_uuid == NULL)
1791                         return -1;
1792         } else
1793                 return -1;
1794         return wps_registrar_add_pin(wpa_s->wps->registrar, addr,
1795                                      use_uuid,
1796                                      (const u8 *) pin, os_strlen(pin), 300);
1797 }
1798
1799
1800 int wpas_wps_er_pbc(struct wpa_supplicant *wpa_s, const char *uuid)
1801 {
1802         u8 u[UUID_LEN], *use_uuid = NULL;
1803         u8 addr[ETH_ALEN], *use_addr = NULL;
1804
1805         if (uuid_str2bin(uuid, u) == 0)
1806                 use_uuid = u;
1807         else if (hwaddr_aton(uuid, addr) == 0)
1808                 use_addr = addr;
1809         else
1810                 return -1;
1811         return wps_er_pbc(wpa_s->wps_er, use_uuid, use_addr);
1812 }
1813
1814
1815 int wpas_wps_er_learn(struct wpa_supplicant *wpa_s, const char *uuid,
1816                       const char *pin)
1817 {
1818         u8 u[UUID_LEN], *use_uuid = NULL;
1819         u8 addr[ETH_ALEN], *use_addr = NULL;
1820
1821         if (uuid_str2bin(uuid, u) == 0)
1822                 use_uuid = u;
1823         else if (hwaddr_aton(uuid, addr) == 0)
1824                 use_addr = addr;
1825         else
1826                 return -1;
1827
1828         return wps_er_learn(wpa_s->wps_er, use_uuid, use_addr, (const u8 *) pin,
1829                             os_strlen(pin));
1830 }
1831
1832
1833 static int wpas_wps_network_to_cred(struct wpa_ssid *ssid,
1834                                     struct wps_credential *cred)
1835 {
1836         os_memset(cred, 0, sizeof(*cred));
1837         if (ssid->ssid_len > 32)
1838                 return -1;
1839         os_memcpy(cred->ssid, ssid->ssid, ssid->ssid_len);
1840         cred->ssid_len = ssid->ssid_len;
1841         if (ssid->key_mgmt & WPA_KEY_MGMT_PSK) {
1842                 cred->auth_type = (ssid->proto & WPA_PROTO_RSN) ?
1843                         WPS_AUTH_WPA2PSK : WPS_AUTH_WPAPSK;
1844                 if (ssid->pairwise_cipher & WPA_CIPHER_CCMP)
1845                         cred->encr_type = WPS_ENCR_AES;
1846                 else
1847                         cred->encr_type = WPS_ENCR_TKIP;
1848                 if (ssid->passphrase) {
1849                         cred->key_len = os_strlen(ssid->passphrase);
1850                         if (cred->key_len >= 64)
1851                                 return -1;
1852                         os_memcpy(cred->key, ssid->passphrase, cred->key_len);
1853                 } else if (ssid->psk_set) {
1854                         cred->key_len = 32;
1855                         os_memcpy(cred->key, ssid->psk, 32);
1856                 } else
1857                         return -1;
1858         } else {
1859                 cred->auth_type = WPS_AUTH_OPEN;
1860                 cred->encr_type = WPS_ENCR_NONE;
1861         }
1862
1863         return 0;
1864 }
1865
1866
1867 int wpas_wps_er_set_config(struct wpa_supplicant *wpa_s, const char *uuid,
1868                            int id)
1869 {
1870         u8 u[UUID_LEN], *use_uuid = NULL;
1871         u8 addr[ETH_ALEN], *use_addr = NULL;
1872         struct wpa_ssid *ssid;
1873         struct wps_credential cred;
1874
1875         if (uuid_str2bin(uuid, u) == 0)
1876                 use_uuid = u;
1877         else if (hwaddr_aton(uuid, addr) == 0)
1878                 use_addr = addr;
1879         else
1880                 return -1;
1881         ssid = wpa_config_get_network(wpa_s->conf, id);
1882         if (ssid == NULL || ssid->ssid == NULL)
1883                 return -1;
1884
1885         if (wpas_wps_network_to_cred(ssid, &cred) < 0)
1886                 return -1;
1887         return wps_er_set_config(wpa_s->wps_er, use_uuid, use_addr, &cred);
1888 }
1889
1890
1891 int wpas_wps_er_config(struct wpa_supplicant *wpa_s, const char *uuid,
1892                        const char *pin, struct wps_new_ap_settings *settings)
1893 {
1894         u8 u[UUID_LEN], *use_uuid = NULL;
1895         u8 addr[ETH_ALEN], *use_addr = NULL;
1896         struct wps_credential cred;
1897         size_t len;
1898
1899         if (uuid_str2bin(uuid, u) == 0)
1900                 use_uuid = u;
1901         else if (hwaddr_aton(uuid, addr) == 0)
1902                 use_addr = addr;
1903         else
1904                 return -1;
1905         if (settings->ssid_hex == NULL || settings->auth == NULL ||
1906             settings->encr == NULL || settings->key_hex == NULL)
1907                 return -1;
1908
1909         os_memset(&cred, 0, sizeof(cred));
1910         len = os_strlen(settings->ssid_hex);
1911         if ((len & 1) || len > 2 * sizeof(cred.ssid) ||
1912             hexstr2bin(settings->ssid_hex, cred.ssid, len / 2))
1913                 return -1;
1914         cred.ssid_len = len / 2;
1915
1916         len = os_strlen(settings->key_hex);
1917         if ((len & 1) || len > 2 * sizeof(cred.key) ||
1918             hexstr2bin(settings->key_hex, cred.key, len / 2))
1919                 return -1;
1920         cred.key_len = len / 2;
1921
1922         if (os_strcmp(settings->auth, "OPEN") == 0)
1923                 cred.auth_type = WPS_AUTH_OPEN;
1924         else if (os_strcmp(settings->auth, "WPAPSK") == 0)
1925                 cred.auth_type = WPS_AUTH_WPAPSK;
1926         else if (os_strcmp(settings->auth, "WPA2PSK") == 0)
1927                 cred.auth_type = WPS_AUTH_WPA2PSK;
1928         else
1929                 return -1;
1930
1931         if (os_strcmp(settings->encr, "NONE") == 0)
1932                 cred.encr_type = WPS_ENCR_NONE;
1933         else if (os_strcmp(settings->encr, "WEP") == 0)
1934                 cred.encr_type = WPS_ENCR_WEP;
1935         else if (os_strcmp(settings->encr, "TKIP") == 0)
1936                 cred.encr_type = WPS_ENCR_TKIP;
1937         else if (os_strcmp(settings->encr, "CCMP") == 0)
1938                 cred.encr_type = WPS_ENCR_AES;
1939         else
1940                 return -1;
1941
1942         return wps_er_config(wpa_s->wps_er, use_uuid, use_addr,
1943                              (const u8 *) pin, os_strlen(pin), &cred);
1944 }
1945
1946
1947 #ifdef CONFIG_WPS_NFC
1948 struct wpabuf * wpas_wps_er_nfc_config_token(struct wpa_supplicant *wpa_s,
1949                                              int ndef, const char *uuid)
1950 {
1951         struct wpabuf *ret;
1952         u8 u[UUID_LEN], *use_uuid = NULL;
1953         u8 addr[ETH_ALEN], *use_addr = NULL;
1954
1955         if (!wpa_s->wps_er)
1956                 return NULL;
1957
1958         if (uuid_str2bin(uuid, u) == 0)
1959                 use_uuid = u;
1960         else if (hwaddr_aton(uuid, addr) == 0)
1961                 use_addr = addr;
1962         else
1963                 return NULL;
1964
1965         ret = wps_er_nfc_config_token(wpa_s->wps_er, use_uuid, use_addr);
1966         if (ndef && ret) {
1967                 struct wpabuf *tmp;
1968                 tmp = ndef_build_wifi(ret);
1969                 wpabuf_free(ret);
1970                 if (tmp == NULL)
1971                         return NULL;
1972                 ret = tmp;
1973         }
1974
1975         return ret;
1976 }
1977 #endif /* CONFIG_WPS_NFC */
1978
1979
1980 static int callbacks_pending = 0;
1981
1982 static void wpas_wps_terminate_cb(void *ctx)
1983 {
1984         wpa_printf(MSG_DEBUG, "WPS ER: Terminated");
1985         if (--callbacks_pending <= 0)
1986                 eloop_terminate();
1987 }
1988 #endif /* CONFIG_WPS_ER */
1989
1990
1991 int wpas_wps_terminate_pending(struct wpa_supplicant *wpa_s)
1992 {
1993 #ifdef CONFIG_WPS_ER
1994         if (wpa_s->wps_er) {
1995                 callbacks_pending++;
1996                 wps_er_deinit(wpa_s->wps_er, wpas_wps_terminate_cb, wpa_s);
1997                 wpa_s->wps_er = NULL;
1998                 return 1;
1999         }
2000 #endif /* CONFIG_WPS_ER */
2001         return 0;
2002 }
2003
2004
2005 void wpas_wps_update_config(struct wpa_supplicant *wpa_s)
2006 {
2007         struct wps_context *wps = wpa_s->wps;
2008
2009         if (wps == NULL)
2010                 return;
2011
2012         if (wpa_s->conf->changed_parameters & CFG_CHANGED_CONFIG_METHODS) {
2013                 wps->config_methods = wps_config_methods_str2bin(
2014                         wpa_s->conf->config_methods);
2015                 if ((wps->config_methods &
2016                      (WPS_CONFIG_DISPLAY | WPS_CONFIG_LABEL)) ==
2017                     (WPS_CONFIG_DISPLAY | WPS_CONFIG_LABEL)) {
2018                         wpa_printf(MSG_ERROR, "WPS: Both Label and Display "
2019                                    "config methods are not allowed at the "
2020                                    "same time");
2021                         wps->config_methods &= ~WPS_CONFIG_LABEL;
2022                 }
2023         }
2024         wps->config_methods = wps_fix_config_methods(wps->config_methods);
2025         wps->dev.config_methods = wps->config_methods;
2026
2027         if (wpa_s->conf->changed_parameters & CFG_CHANGED_DEVICE_TYPE)
2028                 os_memcpy(wps->dev.pri_dev_type, wpa_s->conf->device_type,
2029                           WPS_DEV_TYPE_LEN);
2030
2031         if (wpa_s->conf->changed_parameters & CFG_CHANGED_SEC_DEVICE_TYPE) {
2032                 wps->dev.num_sec_dev_types = wpa_s->conf->num_sec_device_types;
2033                 os_memcpy(wps->dev.sec_dev_type, wpa_s->conf->sec_device_type,
2034                           wps->dev.num_sec_dev_types * WPS_DEV_TYPE_LEN);
2035         }
2036
2037         if (wpa_s->conf->changed_parameters & CFG_CHANGED_VENDOR_EXTENSION)
2038                 wpas_wps_set_vendor_ext_m1(wpa_s, wps);
2039
2040         if (wpa_s->conf->changed_parameters & CFG_CHANGED_OS_VERSION)
2041                 wps->dev.os_version = WPA_GET_BE32(wpa_s->conf->os_version);
2042
2043         if (wpa_s->conf->changed_parameters & CFG_CHANGED_UUID)
2044                 wpas_wps_set_uuid(wpa_s, wps);
2045
2046         if (wpa_s->conf->changed_parameters &
2047             (CFG_CHANGED_DEVICE_NAME | CFG_CHANGED_WPS_STRING)) {
2048                 /* Update pointers to make sure they refer current values */
2049                 wps->dev.device_name = wpa_s->conf->device_name;
2050                 wps->dev.manufacturer = wpa_s->conf->manufacturer;
2051                 wps->dev.model_name = wpa_s->conf->model_name;
2052                 wps->dev.model_number = wpa_s->conf->model_number;
2053                 wps->dev.serial_number = wpa_s->conf->serial_number;
2054         }
2055 }
2056
2057
2058 #ifdef CONFIG_WPS_NFC
2059
2060 #ifdef CONFIG_WPS_ER
2061 static struct wpabuf *
2062 wpas_wps_network_config_token(struct wpa_supplicant *wpa_s, int ndef,
2063                               struct wpa_ssid *ssid)
2064 {
2065         struct wpabuf *ret;
2066         struct wps_credential cred;
2067
2068         if (wpas_wps_network_to_cred(ssid, &cred) < 0)
2069                 return NULL;
2070
2071         ret = wps_er_config_token_from_cred(wpa_s->wps, &cred);
2072
2073         if (ndef && ret) {
2074                 struct wpabuf *tmp;
2075                 tmp = ndef_build_wifi(ret);
2076                 wpabuf_free(ret);
2077                 if (tmp == NULL)
2078                         return NULL;
2079                 ret = tmp;
2080         }
2081
2082         return ret;
2083 }
2084 #endif /* CONFIG_WPS_ER */
2085
2086
2087 struct wpabuf * wpas_wps_nfc_config_token(struct wpa_supplicant *wpa_s,
2088                                           int ndef, const char *id_str)
2089 {
2090 #ifdef CONFIG_WPS_ER
2091         if (id_str) {
2092                 int id;
2093                 char *end = NULL;
2094                 struct wpa_ssid *ssid;
2095
2096                 id = strtol(id_str, &end, 10);
2097                 if (end && *end)
2098                         return NULL;
2099
2100                 ssid = wpa_config_get_network(wpa_s->conf, id);
2101                 if (ssid == NULL)
2102                         return NULL;
2103                 return wpas_wps_network_config_token(wpa_s, ndef, ssid);
2104         }
2105 #endif /* CONFIG_WPS_ER */
2106 #ifdef CONFIG_AP
2107         if (wpa_s->ap_iface)
2108                 return wpas_ap_wps_nfc_config_token(wpa_s, ndef);
2109 #endif /* CONFIG_AP */
2110         return NULL;
2111 }
2112
2113
2114 struct wpabuf * wpas_wps_nfc_token(struct wpa_supplicant *wpa_s, int ndef)
2115 {
2116         if (wpa_s->conf->wps_nfc_pw_from_config) {
2117                 return wps_nfc_token_build(ndef,
2118                                            wpa_s->conf->wps_nfc_dev_pw_id,
2119                                            wpa_s->conf->wps_nfc_dh_pubkey,
2120                                            wpa_s->conf->wps_nfc_dev_pw);
2121         }
2122
2123         return wps_nfc_token_gen(ndef, &wpa_s->conf->wps_nfc_dev_pw_id,
2124                                  &wpa_s->conf->wps_nfc_dh_pubkey,
2125                                  &wpa_s->conf->wps_nfc_dh_privkey,
2126                                  &wpa_s->conf->wps_nfc_dev_pw);
2127 }
2128
2129
2130 int wpas_wps_start_nfc(struct wpa_supplicant *wpa_s, const u8 *go_dev_addr,
2131                        const u8 *bssid,
2132                        const struct wpabuf *dev_pw, u16 dev_pw_id,
2133                        int p2p_group, const u8 *peer_pubkey_hash,
2134                        const u8 *ssid, size_t ssid_len)
2135 {
2136         struct wps_context *wps = wpa_s->wps;
2137         char pw[32 * 2 + 1];
2138
2139         if (dev_pw_id != DEV_PW_NFC_CONNECTION_HANDOVER && dev_pw == NULL) {
2140                 dev_pw = wpa_s->conf->wps_nfc_dev_pw;
2141                 dev_pw_id = wpa_s->conf->wps_nfc_dev_pw_id;
2142         }
2143
2144         if (wpa_s->conf->wps_nfc_dh_pubkey == NULL ||
2145             wpa_s->conf->wps_nfc_dh_privkey == NULL) {
2146                 wpa_printf(MSG_DEBUG, "WPS: Missing DH params - "
2147                            "cannot start NFC-triggered connection");
2148                 return -1;
2149         }
2150
2151         if (dev_pw_id != DEV_PW_NFC_CONNECTION_HANDOVER && dev_pw == NULL) {
2152                 wpa_printf(MSG_DEBUG, "WPS: Missing Device Password (id=%u) - "
2153                            "cannot start NFC-triggered connection", dev_pw_id);
2154                 return -1;
2155         }
2156
2157         dh5_free(wps->dh_ctx);
2158         wpabuf_free(wps->dh_pubkey);
2159         wpabuf_free(wps->dh_privkey);
2160         wps->dh_privkey = wpabuf_dup(wpa_s->conf->wps_nfc_dh_privkey);
2161         wps->dh_pubkey = wpabuf_dup(wpa_s->conf->wps_nfc_dh_pubkey);
2162         if (wps->dh_privkey == NULL || wps->dh_pubkey == NULL) {
2163                 wps->dh_ctx = NULL;
2164                 wpabuf_free(wps->dh_pubkey);
2165                 wps->dh_pubkey = NULL;
2166                 wpabuf_free(wps->dh_privkey);
2167                 wps->dh_privkey = NULL;
2168                 wpa_printf(MSG_DEBUG, "WPS: Failed to get DH priv/pub key");
2169                 return -1;
2170         }
2171         wps->dh_ctx = dh5_init_fixed(wps->dh_privkey, wps->dh_pubkey);
2172         if (wps->dh_ctx == NULL) {
2173                 wpabuf_free(wps->dh_pubkey);
2174                 wps->dh_pubkey = NULL;
2175                 wpabuf_free(wps->dh_privkey);
2176                 wps->dh_privkey = NULL;
2177                 wpa_printf(MSG_DEBUG, "WPS: Failed to initialize DH context");
2178                 return -1;
2179         }
2180
2181         if (dev_pw) {
2182                 wpa_snprintf_hex_uppercase(pw, sizeof(pw),
2183                                            wpabuf_head(dev_pw),
2184                                            wpabuf_len(dev_pw));
2185         }
2186         return wpas_wps_start_dev_pw(wpa_s, go_dev_addr, bssid,
2187                                      dev_pw ? pw : NULL,
2188                                      p2p_group, dev_pw_id, peer_pubkey_hash,
2189                                      ssid, ssid_len);
2190 }
2191
2192
2193 static int wpas_wps_use_cred(struct wpa_supplicant *wpa_s,
2194                              struct wps_parse_attr *attr)
2195 {
2196         wpa_s->wps_ap_channel = 0;
2197
2198         /*
2199          * Disable existing networks temporarily to allow the newly learned
2200          * credential to be preferred. Enable the temporarily disabled networks
2201          * after 10 seconds.
2202          */
2203         wpas_wps_temp_disable(wpa_s, NULL);
2204         eloop_register_timeout(10, 0, wpas_wps_reenable_networks_cb, wpa_s,
2205                                NULL);
2206
2207         if (wps_oob_use_cred(wpa_s->wps, attr) < 0)
2208                 return -1;
2209
2210         if (wpa_s->wpa_state == WPA_INTERFACE_DISABLED)
2211                 return 0;
2212
2213         if (!wpa_s->wps_ap_channel && attr->ap_channel) {
2214                 wpa_s->wps_ap_channel = WPA_GET_BE16(attr->ap_channel);
2215                 wpa_printf(MSG_DEBUG, "WPS: Credential container indicated AP Channel %d",
2216                            wpa_s->wps_ap_channel);
2217         }
2218
2219         wpa_printf(MSG_DEBUG, "WPS: Request reconnection with new network "
2220                    "based on the received credential added");
2221         wpa_s->normal_scans = 0;
2222         wpa_supplicant_reinit_autoscan(wpa_s);
2223         if (wpa_s->wps_ap_channel) {
2224                 u16 chan = wpa_s->wps_ap_channel;
2225                 int freq = 0;
2226
2227                 if (chan >= 1 && chan <= 13)
2228                         freq = 2407 + 5 * chan;
2229                 else if (chan == 14)
2230                         freq = 2484;
2231                 else if (chan >= 30)
2232                         freq = 5000 + 5 * chan;
2233
2234                 if (freq) {
2235                         wpa_printf(MSG_DEBUG, "WPS: Credential indicated "
2236                                    "AP channel %u -> %u MHz", chan, freq);
2237                         wpa_s->after_wps = 5;
2238                         wpa_s->wps_freq = freq;
2239                 }
2240         }
2241         wpa_s->disconnected = 0;
2242         wpa_s->reassociate = 1;
2243
2244         wpa_supplicant_cancel_sched_scan(wpa_s);
2245         wpa_supplicant_req_scan(wpa_s, 0, 0);
2246
2247         return 0;
2248 }
2249
2250
2251 #ifdef CONFIG_WPS_ER
2252 static int wpas_wps_add_nfc_password_token(struct wpa_supplicant *wpa_s,
2253                                            struct wps_parse_attr *attr)
2254 {
2255         return wps_registrar_add_nfc_password_token(
2256                 wpa_s->wps->registrar, attr->oob_dev_password,
2257                 attr->oob_dev_password_len);
2258 }
2259 #endif /* CONFIG_WPS_ER */
2260
2261
2262 static int wpas_wps_nfc_tag_process(struct wpa_supplicant *wpa_s,
2263                                     const struct wpabuf *wps)
2264 {
2265         struct wps_parse_attr attr;
2266
2267         wpa_hexdump_buf(MSG_DEBUG, "WPS: Received NFC tag payload", wps);
2268
2269         if (wps_parse_msg(wps, &attr)) {
2270                 wpa_printf(MSG_DEBUG, "WPS: Ignore invalid data from NFC tag");
2271                 return -1;
2272         }
2273
2274         if (attr.num_cred)
2275                 return wpas_wps_use_cred(wpa_s, &attr);
2276
2277 #ifdef CONFIG_WPS_ER
2278         if (attr.oob_dev_password)
2279                 return wpas_wps_add_nfc_password_token(wpa_s, &attr);
2280 #endif /* CONFIG_WPS_ER */
2281
2282         wpa_printf(MSG_DEBUG, "WPS: Ignore unrecognized NFC tag");
2283         return -1;
2284 }
2285
2286
2287 int wpas_wps_nfc_tag_read(struct wpa_supplicant *wpa_s,
2288                           const struct wpabuf *data)
2289 {
2290         const struct wpabuf *wps = data;
2291         struct wpabuf *tmp = NULL;
2292         int ret;
2293
2294         if (wpabuf_len(data) < 4)
2295                 return -1;
2296
2297         if (*wpabuf_head_u8(data) != 0x10) {
2298                 /* Assume this contains full NDEF record */
2299                 tmp = ndef_parse_wifi(data);
2300                 if (tmp == NULL) {
2301 #ifdef CONFIG_P2P
2302                         tmp = ndef_parse_p2p(data);
2303                         if (tmp) {
2304                                 ret = wpas_p2p_nfc_tag_process(wpa_s, tmp);
2305                                 wpabuf_free(tmp);
2306                                 return ret;
2307                         }
2308 #endif /* CONFIG_P2P */
2309                         wpa_printf(MSG_DEBUG, "WPS: Could not parse NDEF");
2310                         return -1;
2311                 }
2312                 wps = tmp;
2313         }
2314
2315         ret = wpas_wps_nfc_tag_process(wpa_s, wps);
2316         wpabuf_free(tmp);
2317         return ret;
2318 }
2319
2320
2321 struct wpabuf * wpas_wps_nfc_handover_req(struct wpa_supplicant *wpa_s,
2322                                           int ndef)
2323 {
2324         struct wpabuf *ret;
2325
2326         if (wpa_s->conf->wps_nfc_dh_pubkey == NULL &&
2327             wps_nfc_gen_dh(&wpa_s->conf->wps_nfc_dh_pubkey,
2328                            &wpa_s->conf->wps_nfc_dh_privkey) < 0)
2329                 return NULL;
2330
2331         ret = wps_build_nfc_handover_req(wpa_s->wps,
2332                                          wpa_s->conf->wps_nfc_dh_pubkey);
2333
2334         if (ndef && ret) {
2335                 struct wpabuf *tmp;
2336                 tmp = ndef_build_wifi(ret);
2337                 wpabuf_free(ret);
2338                 if (tmp == NULL)
2339                         return NULL;
2340                 ret = tmp;
2341         }
2342
2343         return ret;
2344 }
2345
2346
2347 #ifdef CONFIG_WPS_NFC
2348
2349 static struct wpabuf *
2350 wpas_wps_er_nfc_handover_sel(struct wpa_supplicant *wpa_s, int ndef,
2351                              const char *uuid)
2352 {
2353 #ifdef CONFIG_WPS_ER
2354         struct wpabuf *ret;
2355         u8 u[UUID_LEN], *use_uuid = NULL;
2356         u8 addr[ETH_ALEN], *use_addr = NULL;
2357         struct wps_context *wps = wpa_s->wps;
2358
2359         if (wps == NULL)
2360                 return NULL;
2361
2362         if (uuid == NULL)
2363                 return NULL;
2364         if (uuid_str2bin(uuid, u) == 0)
2365                 use_uuid = u;
2366         else if (hwaddr_aton(uuid, addr) == 0)
2367                 use_addr = addr;
2368         else
2369                 return NULL;
2370
2371         if (wpa_s->conf->wps_nfc_dh_pubkey == NULL) {
2372                 if (wps_nfc_gen_dh(&wpa_s->conf->wps_nfc_dh_pubkey,
2373                                    &wpa_s->conf->wps_nfc_dh_privkey) < 0)
2374                         return NULL;
2375         }
2376
2377         wpas_wps_nfc_clear(wps);
2378         wps->ap_nfc_dev_pw_id = DEV_PW_NFC_CONNECTION_HANDOVER;
2379         wps->ap_nfc_dh_pubkey = wpabuf_dup(wpa_s->conf->wps_nfc_dh_pubkey);
2380         wps->ap_nfc_dh_privkey = wpabuf_dup(wpa_s->conf->wps_nfc_dh_privkey);
2381         if (!wps->ap_nfc_dh_pubkey || !wps->ap_nfc_dh_privkey) {
2382                 wpas_wps_nfc_clear(wps);
2383                 return NULL;
2384         }
2385
2386         ret = wps_er_nfc_handover_sel(wpa_s->wps_er, wpa_s->wps, use_uuid,
2387                                       use_addr, wpa_s->conf->wps_nfc_dh_pubkey);
2388         if (ndef && ret) {
2389                 struct wpabuf *tmp;
2390                 tmp = ndef_build_wifi(ret);
2391                 wpabuf_free(ret);
2392                 if (tmp == NULL)
2393                         return NULL;
2394                 ret = tmp;
2395         }
2396
2397         return ret;
2398 #else /* CONFIG_WPS_ER */
2399         return NULL;
2400 #endif /* CONFIG_WPS_ER */
2401 }
2402 #endif /* CONFIG_WPS_NFC */
2403
2404
2405 struct wpabuf * wpas_wps_nfc_handover_sel(struct wpa_supplicant *wpa_s,
2406                                           int ndef, int cr, const char *uuid)
2407 {
2408         struct wpabuf *ret;
2409         if (!cr)
2410                 return NULL;
2411         ret = wpas_ap_wps_nfc_handover_sel(wpa_s, ndef);
2412         if (ret)
2413                 return ret;
2414         return wpas_wps_er_nfc_handover_sel(wpa_s, ndef, uuid);
2415 }
2416
2417
2418 int wpas_wps_nfc_rx_handover_req(struct wpa_supplicant *wpa_s,
2419                                  const struct wpabuf *data)
2420 {
2421         /* TODO */
2422         return -1;
2423 }
2424
2425
2426 int wpas_wps_nfc_rx_handover_sel(struct wpa_supplicant *wpa_s,
2427                                  const struct wpabuf *data)
2428 {
2429         struct wpabuf *wps;
2430         int ret = -1;
2431         u16 wsc_len;
2432         const u8 *pos;
2433         struct wpabuf msg;
2434         struct wps_parse_attr attr;
2435         u16 dev_pw_id;
2436
2437         wps = ndef_parse_wifi(data);
2438         if (wps == NULL)
2439                 return -1;
2440         wpa_printf(MSG_DEBUG, "WPS: Received application/vnd.wfa.wsc "
2441                    "payload from NFC connection handover");
2442         wpa_hexdump_buf(MSG_DEBUG, "WPS: NFC payload", wps);
2443         if (wpabuf_len(wps) < 2) {
2444                 wpa_printf(MSG_DEBUG, "WPS: Too short Wi-Fi Handover Select "
2445                            "Message");
2446                 goto out;
2447         }
2448         pos = wpabuf_head(wps);
2449         wsc_len = WPA_GET_BE16(pos);
2450         if (wsc_len > wpabuf_len(wps) - 2) {
2451                 wpa_printf(MSG_DEBUG, "WPS: Invalid WSC attribute length (%u) "
2452                            "in Wi-Fi Handover Select Message", wsc_len);
2453                 goto out;
2454         }
2455         pos += 2;
2456
2457         wpa_hexdump(MSG_DEBUG,
2458                     "WPS: WSC attributes in Wi-Fi Handover Select Message",
2459                     pos, wsc_len);
2460         if (wsc_len < wpabuf_len(wps) - 2) {
2461                 wpa_hexdump(MSG_DEBUG,
2462                             "WPS: Ignore extra data after WSC attributes",
2463                             pos + wsc_len, wpabuf_len(wps) - 2 - wsc_len);
2464         }
2465
2466         wpabuf_set(&msg, pos, wsc_len);
2467         ret = wps_parse_msg(&msg, &attr);
2468         if (ret < 0) {
2469                 wpa_printf(MSG_DEBUG, "WPS: Could not parse WSC attributes in "
2470                            "Wi-Fi Handover Select Message");
2471                 goto out;
2472         }
2473
2474         if (attr.oob_dev_password == NULL ||
2475             attr.oob_dev_password_len < WPS_OOB_PUBKEY_HASH_LEN + 2) {
2476                 wpa_printf(MSG_DEBUG, "WPS: No Out-of-Band Device Password "
2477                            "included in Wi-Fi Handover Select Message");
2478                 ret = -1;
2479                 goto out;
2480         }
2481
2482         if (attr.ssid == NULL) {
2483                 wpa_printf(MSG_DEBUG, "WPS: No SSID included in Wi-Fi Handover "
2484                            "Select Message");
2485                 ret = -1;
2486                 goto out;
2487         }
2488
2489         wpa_hexdump_ascii(MSG_DEBUG, "WPS: SSID", attr.ssid, attr.ssid_len);
2490
2491         wpa_hexdump(MSG_DEBUG, "WPS: Out-of-Band Device Password",
2492                     attr.oob_dev_password, attr.oob_dev_password_len);
2493         dev_pw_id = WPA_GET_BE16(attr.oob_dev_password +
2494                                  WPS_OOB_PUBKEY_HASH_LEN);
2495         if (dev_pw_id != DEV_PW_NFC_CONNECTION_HANDOVER) {
2496                 wpa_printf(MSG_DEBUG, "WPS: Unexpected OOB Device Password ID "
2497                            "%u in Wi-Fi Handover Select Message", dev_pw_id);
2498                 ret = -1;
2499                 goto out;
2500         }
2501         wpa_hexdump(MSG_DEBUG, "WPS: AP Public Key hash",
2502                     attr.oob_dev_password, WPS_OOB_PUBKEY_HASH_LEN);
2503
2504         ret = wpas_wps_start_nfc(wpa_s, NULL, NULL, NULL, dev_pw_id, 0,
2505                                  attr.oob_dev_password,
2506                                  attr.ssid, attr.ssid_len);
2507
2508 out:
2509         wpabuf_free(wps);
2510         return ret;
2511 }
2512
2513
2514 int wpas_wps_nfc_report_handover(struct wpa_supplicant *wpa_s,
2515                                  const struct wpabuf *req,
2516                                  const struct wpabuf *sel)
2517 {
2518         wpa_printf(MSG_DEBUG, "NFC: WPS connection handover reported");
2519         wpa_hexdump_buf_key(MSG_DEBUG, "WPS: Carrier record in request", req);
2520         wpa_hexdump_buf_key(MSG_DEBUG, "WPS: Carrier record in select", sel);
2521         return wpas_wps_nfc_rx_handover_sel(wpa_s, sel);
2522 }
2523
2524
2525 int wpas_er_wps_nfc_report_handover(struct wpa_supplicant *wpa_s,
2526                                     const struct wpabuf *req,
2527                                     const struct wpabuf *sel)
2528 {
2529         struct wpabuf *wps;
2530         int ret = -1;
2531         u16 wsc_len;
2532         const u8 *pos;
2533         struct wpabuf msg;
2534         struct wps_parse_attr attr;
2535         u16 dev_pw_id;
2536
2537         /*
2538          * Enrollee/station is always initiator of the NFC connection handover,
2539          * so use the request message here to find Enrollee public key hash.
2540          */
2541         wps = ndef_parse_wifi(req);
2542         if (wps == NULL)
2543                 return -1;
2544         wpa_printf(MSG_DEBUG, "WPS: Received application/vnd.wfa.wsc "
2545                    "payload from NFC connection handover");
2546         wpa_hexdump_buf(MSG_DEBUG, "WPS: NFC payload", wps);
2547         if (wpabuf_len(wps) < 2) {
2548                 wpa_printf(MSG_DEBUG, "WPS: Too short Wi-Fi Handover Request "
2549                            "Message");
2550                 goto out;
2551         }
2552         pos = wpabuf_head(wps);
2553         wsc_len = WPA_GET_BE16(pos);
2554         if (wsc_len > wpabuf_len(wps) - 2) {
2555                 wpa_printf(MSG_DEBUG, "WPS: Invalid WSC attribute length (%u) "
2556                            "in rt Wi-Fi Handover Request Message", wsc_len);
2557                 goto out;
2558         }
2559         pos += 2;
2560
2561         wpa_hexdump(MSG_DEBUG,
2562                     "WPS: WSC attributes in Wi-Fi Handover Request Message",
2563                     pos, wsc_len);
2564         if (wsc_len < wpabuf_len(wps) - 2) {
2565                 wpa_hexdump(MSG_DEBUG,
2566                             "WPS: Ignore extra data after WSC attributes",
2567                             pos + wsc_len, wpabuf_len(wps) - 2 - wsc_len);
2568         }
2569
2570         wpabuf_set(&msg, pos, wsc_len);
2571         ret = wps_parse_msg(&msg, &attr);
2572         if (ret < 0) {
2573                 wpa_printf(MSG_DEBUG, "WPS: Could not parse WSC attributes in "
2574                            "Wi-Fi Handover Request Message");
2575                 goto out;
2576         }
2577
2578         if (attr.oob_dev_password == NULL ||
2579             attr.oob_dev_password_len < WPS_OOB_PUBKEY_HASH_LEN + 2) {
2580                 wpa_printf(MSG_DEBUG, "WPS: No Out-of-Band Device Password "
2581                            "included in Wi-Fi Handover Request Message");
2582                 ret = -1;
2583                 goto out;
2584         }
2585
2586         if (attr.uuid_e == NULL) {
2587                 wpa_printf(MSG_DEBUG, "WPS: No UUID-E included in Wi-Fi "
2588                            "Handover Request Message");
2589                 ret = -1;
2590                 goto out;
2591         }
2592
2593         wpa_hexdump(MSG_DEBUG, "WPS: UUID-E", attr.uuid_e, WPS_UUID_LEN);
2594
2595         wpa_hexdump(MSG_DEBUG, "WPS: Out-of-Band Device Password",
2596                     attr.oob_dev_password, attr.oob_dev_password_len);
2597         dev_pw_id = WPA_GET_BE16(attr.oob_dev_password +
2598                                  WPS_OOB_PUBKEY_HASH_LEN);
2599         if (dev_pw_id != DEV_PW_NFC_CONNECTION_HANDOVER) {
2600                 wpa_printf(MSG_DEBUG, "WPS: Unexpected OOB Device Password ID "
2601                            "%u in Wi-Fi Handover Request Message", dev_pw_id);
2602                 ret = -1;
2603                 goto out;
2604         }
2605         wpa_hexdump(MSG_DEBUG, "WPS: Enrollee Public Key hash",
2606                     attr.oob_dev_password, WPS_OOB_PUBKEY_HASH_LEN);
2607
2608         ret = wps_registrar_add_nfc_pw_token(wpa_s->wps->registrar,
2609                                              attr.oob_dev_password,
2610                                              DEV_PW_NFC_CONNECTION_HANDOVER,
2611                                              NULL, 0, 1);
2612
2613 out:
2614         wpabuf_free(wps);
2615         return ret;
2616 }
2617
2618 #endif /* CONFIG_WPS_NFC */
2619
2620
2621 static void wpas_wps_dump_ap_info(struct wpa_supplicant *wpa_s)
2622 {
2623         size_t i;
2624         struct os_reltime now;
2625
2626         if (wpa_debug_level > MSG_DEBUG)
2627                 return;
2628
2629         if (wpa_s->wps_ap == NULL)
2630                 return;
2631
2632         os_get_reltime(&now);
2633
2634         for (i = 0; i < wpa_s->num_wps_ap; i++) {
2635                 struct wps_ap_info *ap = &wpa_s->wps_ap[i];
2636                 struct wpa_blacklist *e = wpa_blacklist_get(wpa_s, ap->bssid);
2637
2638                 wpa_printf(MSG_DEBUG, "WPS: AP[%d] " MACSTR " type=%d "
2639                            "tries=%d last_attempt=%d sec ago blacklist=%d",
2640                            (int) i, MAC2STR(ap->bssid), ap->type, ap->tries,
2641                            ap->last_attempt.sec > 0 ?
2642                            (int) now.sec - (int) ap->last_attempt.sec : -1,
2643                            e ? e->count : 0);
2644         }
2645 }
2646
2647
2648 static struct wps_ap_info * wpas_wps_get_ap_info(struct wpa_supplicant *wpa_s,
2649                                                  const u8 *bssid)
2650 {
2651         size_t i;
2652
2653         if (wpa_s->wps_ap == NULL)
2654                 return NULL;
2655
2656         for (i = 0; i < wpa_s->num_wps_ap; i++) {
2657                 struct wps_ap_info *ap = &wpa_s->wps_ap[i];
2658                 if (os_memcmp(ap->bssid, bssid, ETH_ALEN) == 0)
2659                         return ap;
2660         }
2661
2662         return NULL;
2663 }
2664
2665
2666 static void wpas_wps_update_ap_info_bss(struct wpa_supplicant *wpa_s,
2667                                         struct wpa_scan_res *res)
2668 {
2669         struct wpabuf *wps;
2670         enum wps_ap_info_type type;
2671         struct wps_ap_info *ap;
2672         int r;
2673
2674         if (wpa_scan_get_vendor_ie(res, WPS_IE_VENDOR_TYPE) == NULL)
2675                 return;
2676
2677         wps = wpa_scan_get_vendor_ie_multi(res, WPS_IE_VENDOR_TYPE);
2678         if (wps == NULL)
2679                 return;
2680
2681         r = wps_is_addr_authorized(wps, wpa_s->own_addr, 1);
2682         if (r == 2)
2683                 type = WPS_AP_SEL_REG_OUR;
2684         else if (r == 1)
2685                 type = WPS_AP_SEL_REG;
2686         else
2687                 type = WPS_AP_NOT_SEL_REG;
2688
2689         wpabuf_free(wps);
2690
2691         ap = wpas_wps_get_ap_info(wpa_s, res->bssid);
2692         if (ap) {
2693                 if (ap->type != type) {
2694                         wpa_printf(MSG_DEBUG, "WPS: AP " MACSTR
2695                                    " changed type %d -> %d",
2696                                    MAC2STR(res->bssid), ap->type, type);
2697                         ap->type = type;
2698                         if (type != WPS_AP_NOT_SEL_REG)
2699                                 wpa_blacklist_del(wpa_s, ap->bssid);
2700                 }
2701                 return;
2702         }
2703
2704         ap = os_realloc_array(wpa_s->wps_ap, wpa_s->num_wps_ap + 1,
2705                               sizeof(struct wps_ap_info));
2706         if (ap == NULL)
2707                 return;
2708
2709         wpa_s->wps_ap = ap;
2710         ap = &wpa_s->wps_ap[wpa_s->num_wps_ap];
2711         wpa_s->num_wps_ap++;
2712
2713         os_memset(ap, 0, sizeof(*ap));
2714         os_memcpy(ap->bssid, res->bssid, ETH_ALEN);
2715         ap->type = type;
2716         wpa_printf(MSG_DEBUG, "WPS: AP " MACSTR " type %d added",
2717                    MAC2STR(ap->bssid), ap->type);
2718 }
2719
2720
2721 void wpas_wps_update_ap_info(struct wpa_supplicant *wpa_s,
2722                              struct wpa_scan_results *scan_res)
2723 {
2724         size_t i;
2725
2726         for (i = 0; i < scan_res->num; i++)
2727                 wpas_wps_update_ap_info_bss(wpa_s, scan_res->res[i]);
2728
2729         wpas_wps_dump_ap_info(wpa_s);
2730 }
2731
2732
2733 void wpas_wps_notify_assoc(struct wpa_supplicant *wpa_s, const u8 *bssid)
2734 {
2735         struct wps_ap_info *ap;
2736
2737         wpa_s->after_wps = 0;
2738
2739         if (!wpa_s->wps_ap_iter)
2740                 return;
2741         ap = wpas_wps_get_ap_info(wpa_s, bssid);
2742         if (ap == NULL)
2743                 return;
2744         ap->tries++;
2745         os_get_reltime(&ap->last_attempt);
2746 }