wpa_supplicant/wnm_sta.c

   1 /*
   2  * wpa_supplicant - WNM
   3  * Copyright (c) 2011-2013, Qualcomm Atheros, Inc.
   4  *
   5  * This software may be distributed under the terms of the BSD license.
   6  * See README for more details.
   7  */
   8
   9 #include "utils/includes.h"
  10
  11 #include "utils/common.h"
  12 #include "common/ieee802_11_defs.h"
  13 #include "common/ieee802_11_common.h"
  14 #include "common/wpa_ctrl.h"
  15 #include "rsn_supp/wpa.h"
  16 #include "wpa_supplicant_i.h"
  17 #include "driver_i.h"
  18 #include "scan.h"
  19 #include "ctrl_iface.h"
  20 #include "bss.h"
  21 #include "wnm_sta.h"
  22 #include "hs20_supplicant.h"
  23
  24 #define MAX_TFS_IE_LEN  1024
  25 #define WNM_MAX_NEIGHBOR_REPORT 10
  26
  27
  28 /* get the TFS IE from driver */
  29 static int ieee80211_11_get_tfs_ie(struct wpa_supplicant *wpa_s, u8 *buf,
  30                                    u16 *buf_len, enum wnm_oper oper)
  31 {
  32         wpa_printf(MSG_DEBUG, "%s: TFS get operation %d", __func__, oper);
  33
  34         return wpa_drv_wnm_oper(wpa_s, oper, wpa_s->bssid, buf, buf_len);
  35 }
  36
  37
  38 /* set the TFS IE to driver */
  39 static int ieee80211_11_set_tfs_ie(struct wpa_supplicant *wpa_s,
  40                                    const u8 *addr, const u8 *buf, u16 buf_len,
  41                                    enum wnm_oper oper)
  42 {
  43         u16 len = buf_len;
  44
  45         wpa_printf(MSG_DEBUG, "%s: TFS set operation %d", __func__, oper);
  46
  47         return wpa_drv_wnm_oper(wpa_s, oper, addr, (u8 *) buf, &len);
  48 }
  49
  50
  51 /* MLME-SLEEPMODE.request */
  52 int ieee802_11_send_wnmsleep_req(struct wpa_supplicant *wpa_s,
  53                                  u8 action, u16 intval, struct wpabuf *tfs_req)
  54 {
  55         struct ieee80211_mgmt *mgmt;
  56         int res;
  57         size_t len;
  58         struct wnm_sleep_element *wnmsleep_ie;
  59         u8 *wnmtfs_ie;
  60         u8 wnmsleep_ie_len;
  61         u16 wnmtfs_ie_len;  /* possibly multiple IE(s) */
  62         enum wnm_oper tfs_oper = action == 0 ? WNM_SLEEP_TFS_REQ_IE_ADD :
  63                 WNM_SLEEP_TFS_REQ_IE_NONE;
  64
  65         wpa_printf(MSG_DEBUG, "WNM: Request to send WNM-Sleep Mode Request "
  66                    "action=%s to " MACSTR,
  67                    action == 0 ? "enter" : "exit",
  68                    MAC2STR(wpa_s->bssid));
  69
  70         /* WNM-Sleep Mode IE */
  71         wnmsleep_ie_len = sizeof(struct wnm_sleep_element);
  72         wnmsleep_ie = os_zalloc(sizeof(struct wnm_sleep_element));
  73         if (wnmsleep_ie == NULL)
  74                 return -1;
  75         wnmsleep_ie->eid = WLAN_EID_WNMSLEEP;
  76         wnmsleep_ie->len = wnmsleep_ie_len - 2;
  77         wnmsleep_ie->action_type = action;
  78         wnmsleep_ie->status = WNM_STATUS_SLEEP_ACCEPT;
  79         wnmsleep_ie->intval = host_to_le16(intval);
  80         wpa_hexdump(MSG_DEBUG, "WNM: WNM-Sleep Mode element",
  81                     (u8 *) wnmsleep_ie, wnmsleep_ie_len);
  82
  83         /* TFS IE(s) */
  84         if (tfs_req) {
  85                 wnmtfs_ie_len = wpabuf_len(tfs_req);
  86                 wnmtfs_ie = os_malloc(wnmtfs_ie_len);
  87                 if (wnmtfs_ie == NULL) {
  88                         os_free(wnmsleep_ie);
  89                         return -1;
  90                 }
  91                 os_memcpy(wnmtfs_ie, wpabuf_head(tfs_req), wnmtfs_ie_len);
  92         } else {
  93                 wnmtfs_ie = os_zalloc(MAX_TFS_IE_LEN);
  94                 if (wnmtfs_ie == NULL) {
  95                         os_free(wnmsleep_ie);
  96                         return -1;
  97                 }
  98                 if (ieee80211_11_get_tfs_ie(wpa_s, wnmtfs_ie, &wnmtfs_ie_len,
  99                                             tfs_oper)) {
 100                         wnmtfs_ie_len = 0;
 101                         os_free(wnmtfs_ie);
 102                         wnmtfs_ie = NULL;
 103                 }
 104         }
 105         wpa_hexdump(MSG_DEBUG, "WNM: TFS Request element",
 106                     (u8 *) wnmtfs_ie, wnmtfs_ie_len);
 107
 108         mgmt = os_zalloc(sizeof(*mgmt) + wnmsleep_ie_len + wnmtfs_ie_len);
 109         if (mgmt == NULL) {
 110                 wpa_printf(MSG_DEBUG, "MLME: Failed to allocate buffer for "
 111                            "WNM-Sleep Request action frame");
 112                 os_free(wnmsleep_ie);
 113                 os_free(wnmtfs_ie);
 114                 return -1;
 115         }
 116
 117         os_memcpy(mgmt->da, wpa_s->bssid, ETH_ALEN);
 118         os_memcpy(mgmt->sa, wpa_s->own_addr, ETH_ALEN);
 119         os_memcpy(mgmt->bssid, wpa_s->bssid, ETH_ALEN);
 120         mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
 121                                            WLAN_FC_STYPE_ACTION);
 122         mgmt->u.action.category = WLAN_ACTION_WNM;
 123         mgmt->u.action.u.wnm_sleep_req.action = WNM_SLEEP_MODE_REQ;
 124         mgmt->u.action.u.wnm_sleep_req.dialogtoken = 1;
 125         os_memcpy(mgmt->u.action.u.wnm_sleep_req.variable, wnmsleep_ie,
 126                   wnmsleep_ie_len);
 127         /* copy TFS IE here */
 128         if (wnmtfs_ie_len > 0) {
 129                 os_memcpy(mgmt->u.action.u.wnm_sleep_req.variable +
 130                           wnmsleep_ie_len, wnmtfs_ie, wnmtfs_ie_len);
 131         }
 132
 133         len = 1 + sizeof(mgmt->u.action.u.wnm_sleep_req) + wnmsleep_ie_len +
 134                 wnmtfs_ie_len;
 135
 136         res = wpa_drv_send_action(wpa_s, wpa_s->assoc_freq, 0, wpa_s->bssid,
 137                                   wpa_s->own_addr, wpa_s->bssid,
 138                                   &mgmt->u.action.category, len, 0);
 139         if (res < 0)
 140                 wpa_printf(MSG_DEBUG, "Failed to send WNM-Sleep Request "
 141                            "(action=%d, intval=%d)", action, intval);
 142         else
 143                 wpa_s->wnmsleep_used = 1;
 144
 145         os_free(wnmsleep_ie);
 146         os_free(wnmtfs_ie);
 147         os_free(mgmt);
 148
 149         return res;
 150 }
 151
 152
 153 static void wnm_sleep_mode_enter_success(struct wpa_supplicant *wpa_s,
 154                                          const u8 *tfsresp_ie_start,
 155                                          const u8 *tfsresp_ie_end)
 156 {
 157         wpa_drv_wnm_oper(wpa_s, WNM_SLEEP_ENTER_CONFIRM,
 158                          wpa_s->bssid, NULL, NULL);
 159         /* remove GTK/IGTK ?? */
 160
 161         /* set the TFS Resp IE(s) */
 162         if (tfsresp_ie_start && tfsresp_ie_end &&
 163             tfsresp_ie_end - tfsresp_ie_start >= 0) {
 164                 u16 tfsresp_ie_len;
 165                 tfsresp_ie_len = (tfsresp_ie_end + tfsresp_ie_end[1] + 2) -
 166                         tfsresp_ie_start;
 167                 wpa_printf(MSG_DEBUG, "TFS Resp IE(s) found");
 168                 /* pass the TFS Resp IE(s) to driver for processing */
 169                 if (ieee80211_11_set_tfs_ie(wpa_s, wpa_s->bssid,
 170                                             tfsresp_ie_start,
 171                                             tfsresp_ie_len,
 172                                             WNM_SLEEP_TFS_RESP_IE_SET))
 173                         wpa_printf(MSG_DEBUG, "WNM: Fail to set TFS Resp IE");
 174         }
 175 }
 176
 177
 178 static void wnm_sleep_mode_exit_success(struct wpa_supplicant *wpa_s,
 179                                         const u8 *frm, u16 key_len_total)
 180 {
 181         u8 *ptr, *end;
 182         u8 gtk_len;
 183
 184         wpa_drv_wnm_oper(wpa_s, WNM_SLEEP_EXIT_CONFIRM,  wpa_s->bssid,
 185                          NULL, NULL);
 186
 187         /* Install GTK/IGTK */
 188
 189         /* point to key data field */
 190         ptr = (u8 *) frm + 1 + 2;
 191         end = ptr + key_len_total;
 192         wpa_hexdump_key(MSG_DEBUG, "WNM: Key Data", ptr, key_len_total);
 193
 194         if (key_len_total && !wpa_sm_pmf_enabled(wpa_s->wpa)) {
 195                 wpa_msg(wpa_s, MSG_INFO,
 196                         "WNM: Ignore Key Data in WNM-Sleep Mode Response - PMF not enabled");
 197                 return;
 198         }
 199
 200         while (end - ptr > 1) {
 201                 if (2 + ptr[1] > end - ptr) {
 202                         wpa_printf(MSG_DEBUG, "WNM: Invalid Key Data element "
 203                                    "length");
 204                         if (end > ptr) {
 205                                 wpa_hexdump(MSG_DEBUG, "WNM: Remaining data",
 206                                             ptr, end - ptr);
 207                         }
 208                         break;
 209                 }
 210                 if (*ptr == WNM_SLEEP_SUBELEM_GTK) {
 211                         if (ptr[1] < 11 + 5) {
 212                                 wpa_printf(MSG_DEBUG, "WNM: Too short GTK "
 213                                            "subelem");
 214                                 break;
 215                         }
 216                         gtk_len = *(ptr + 4);
 217                         if (ptr[1] < 11 + gtk_len ||
 218                             gtk_len < 5 || gtk_len > 32) {
 219                                 wpa_printf(MSG_DEBUG, "WNM: Invalid GTK "
 220                                            "subelem");
 221                                 break;
 222                         }
 223                         wpa_wnmsleep_install_key(
 224                                 wpa_s->wpa,
 225                                 WNM_SLEEP_SUBELEM_GTK,
 226                                 ptr);
 227                         ptr += 13 + gtk_len;
 228 #ifdef CONFIG_IEEE80211W
 229                 } else if (*ptr == WNM_SLEEP_SUBELEM_IGTK) {
 230                         if (ptr[1] < 2 + 6 + WPA_IGTK_LEN) {
 231                                 wpa_printf(MSG_DEBUG, "WNM: Too short IGTK "
 232                                            "subelem");
 233                                 break;
 234                         }
 235                         wpa_wnmsleep_install_key(wpa_s->wpa,
 236                                                  WNM_SLEEP_SUBELEM_IGTK, ptr);
 237                         ptr += 10 + WPA_IGTK_LEN;
 238 #endif /* CONFIG_IEEE80211W */
 239                 } else
 240                         break; /* skip the loop */
 241         }
 242 }
 243
 244
 245 static void ieee802_11_rx_wnmsleep_resp(struct wpa_supplicant *wpa_s,
 246                                         const u8 *frm, int len)
 247 {
 248         /*
 249          * Action [1] | Dialog Token [1] | Key Data Len [2] | Key Data |
 250          * WNM-Sleep Mode IE | TFS Response IE
 251          */
 252         const u8 *pos = frm; /* point to payload after the action field */
 253         u16 key_len_total;
 254         struct wnm_sleep_element *wnmsleep_ie = NULL;
 255         /* multiple TFS Resp IE (assuming consecutive) */
 256         const u8 *tfsresp_ie_start = NULL;
 257         const u8 *tfsresp_ie_end = NULL;
 258         size_t left;
 259
 260         if (!wpa_s->wnmsleep_used) {
 261                 wpa_printf(MSG_DEBUG,
 262                            "WNM: Ignore WNM-Sleep Mode Response frame since WNM-Sleep Mode has not been used in this association");
 263                 return;
 264         }
 265
 266         if (len < 3)
 267                 return;
 268         key_len_total = WPA_GET_LE16(frm + 1);
 269
 270         wpa_printf(MSG_DEBUG, "WNM-Sleep Mode Response token=%u key_len_total=%d",
 271                    frm[0], key_len_total);
 272         left = len - 3;
 273         if (key_len_total > left) {
 274                 wpa_printf(MSG_INFO, "WNM: Too short frame for Key Data field");
 275                 return;
 276         }
 277         pos += 3 + key_len_total;
 278         while (pos - frm + 1 < len) {
 279                 u8 ie_len = *(pos + 1);
 280                 if (2 + ie_len > frm + len - pos) {
 281                         wpa_printf(MSG_INFO, "WNM: Invalid IE len %u", ie_len);
 282                         break;
 283                 }
 284                 wpa_hexdump(MSG_DEBUG, "WNM: Element", pos, 2 + ie_len);
 285                 if (*pos == WLAN_EID_WNMSLEEP && ie_len >= 4)
 286                         wnmsleep_ie = (struct wnm_sleep_element *) pos;
 287                 else if (*pos == WLAN_EID_TFS_RESP) {
 288                         if (!tfsresp_ie_start)
 289                                 tfsresp_ie_start = pos;
 290                         tfsresp_ie_end = pos;
 291                 } else
 292                         wpa_printf(MSG_DEBUG, "EID %d not recognized", *pos);
 293                 pos += ie_len + 2;
 294         }
 295
 296         if (!wnmsleep_ie) {
 297                 wpa_printf(MSG_DEBUG, "No WNM-Sleep IE found");
 298                 return;
 299         }
 300
 301         if (wnmsleep_ie->status == WNM_STATUS_SLEEP_ACCEPT ||
 302             wnmsleep_ie->status == WNM_STATUS_SLEEP_EXIT_ACCEPT_GTK_UPDATE) {
 303                 wpa_printf(MSG_DEBUG, "Successfully recv WNM-Sleep Response "
 304                            "frame (action=%d, intval=%d)",
 305                            wnmsleep_ie->action_type, wnmsleep_ie->intval);
 306                 if (wnmsleep_ie->action_type == WNM_SLEEP_MODE_ENTER) {
 307                         wnm_sleep_mode_enter_success(wpa_s, tfsresp_ie_start,
 308                                                      tfsresp_ie_end);
 309                 } else if (wnmsleep_ie->action_type == WNM_SLEEP_MODE_EXIT) {
 310                         wnm_sleep_mode_exit_success(wpa_s, frm, key_len_total);
 311                 }
 312         } else {
 313                 wpa_printf(MSG_DEBUG, "Reject recv WNM-Sleep Response frame "
 314                            "(action=%d, intval=%d)",
 315                            wnmsleep_ie->action_type, wnmsleep_ie->intval);
 316                 if (wnmsleep_ie->action_type == WNM_SLEEP_MODE_ENTER)
 317                         wpa_drv_wnm_oper(wpa_s, WNM_SLEEP_ENTER_FAIL,
 318                                          wpa_s->bssid, NULL, NULL);
 319                 else if (wnmsleep_ie->action_type == WNM_SLEEP_MODE_EXIT)
 320                         wpa_drv_wnm_oper(wpa_s, WNM_SLEEP_EXIT_FAIL,
 321                                          wpa_s->bssid, NULL, NULL);
 322         }
 323 }
 324
 325
 326 void wnm_deallocate_memory(struct wpa_supplicant *wpa_s)
 327 {
 328         int i;
 329
 330         for (i = 0; i < wpa_s->wnm_num_neighbor_report; i++) {
 331                 os_free(wpa_s->wnm_neighbor_report_elements[i].meas_pilot);
 332                 os_free(wpa_s->wnm_neighbor_report_elements[i].mul_bssid);
 333         }
 334
 335         wpa_s->wnm_num_neighbor_report = 0;
 336         os_free(wpa_s->wnm_neighbor_report_elements);
 337         wpa_s->wnm_neighbor_report_elements = NULL;
 338 }
 339
 340
 341 static void wnm_parse_neighbor_report_elem(struct neighbor_report *rep,
 342                                            u8 id, u8 elen, const u8 *pos)
 343 {
 344         switch (id) {
 345         case WNM_NEIGHBOR_TSF:
 346                 if (elen < 2 + 2) {
 347                         wpa_printf(MSG_DEBUG, "WNM: Too short TSF");
 348                         break;
 349                 }
 350                 rep->tsf_offset = WPA_GET_LE16(pos);
 351                 rep->beacon_int = WPA_GET_LE16(pos + 2);
 352                 rep->tsf_present = 1;
 353                 break;
 354         case WNM_NEIGHBOR_CONDENSED_COUNTRY_STRING:
 355                 if (elen < 2) {
 356                         wpa_printf(MSG_DEBUG, "WNM: Too short condensed "
 357                                    "country string");
 358                         break;
 359                 }
 360                 os_memcpy(rep->country, pos, 2);
 361                 rep->country_present = 1;
 362                 break;
 363         case WNM_NEIGHBOR_BSS_TRANSITION_CANDIDATE:
 364                 if (elen < 1) {
 365                         wpa_printf(MSG_DEBUG, "WNM: Too short BSS transition "
 366                                    "candidate");
 367                         break;
 368                 }
 369                 rep->preference = pos[0];
 370                 rep->preference_present = 1;
 371                 break;
 372         case WNM_NEIGHBOR_BSS_TERMINATION_DURATION:
 373                 rep->bss_term_tsf = WPA_GET_LE64(pos);
 374                 rep->bss_term_dur = WPA_GET_LE16(pos + 8);
 375                 rep->bss_term_present = 1;
 376                 break;
 377         case WNM_NEIGHBOR_BEARING:
 378                 if (elen < 8) {
 379                         wpa_printf(MSG_DEBUG, "WNM: Too short neighbor "
 380                                    "bearing");
 381                         break;
 382                 }
 383                 rep->bearing = WPA_GET_LE16(pos);
 384                 rep->distance = WPA_GET_LE32(pos + 2);
 385                 rep->rel_height = WPA_GET_LE16(pos + 2 + 4);
 386                 rep->bearing_present = 1;
 387                 break;
 388         case WNM_NEIGHBOR_MEASUREMENT_PILOT:
 389                 if (elen < 1) {
 390                         wpa_printf(MSG_DEBUG, "WNM: Too short measurement "
 391                                    "pilot");
 392                         break;
 393                 }
 394                 os_free(rep->meas_pilot);
 395                 rep->meas_pilot = os_zalloc(sizeof(struct measurement_pilot));
 396                 if (rep->meas_pilot == NULL)
 397                         break;
 398                 rep->meas_pilot->measurement_pilot = pos[0];
 399                 rep->meas_pilot->subelem_len = elen - 1;
 400                 os_memcpy(rep->meas_pilot->subelems, pos + 1, elen - 1);
 401                 break;
 402         case WNM_NEIGHBOR_RRM_ENABLED_CAPABILITIES:
 403                 if (elen < 5) {
 404                         wpa_printf(MSG_DEBUG, "WNM: Too short RRM enabled "
 405                                    "capabilities");
 406                         break;
 407                 }
 408                 os_memcpy(rep->rm_capab, pos, 5);
 409                 rep->rm_capab_present = 1;
 410                 break;
 411         case WNM_NEIGHBOR_MULTIPLE_BSSID:
 412                 if (elen < 1) {
 413                         wpa_printf(MSG_DEBUG, "WNM: Too short multiple BSSID");
 414                         break;
 415                 }
 416                 os_free(rep->mul_bssid);
 417                 rep->mul_bssid = os_zalloc(sizeof(struct multiple_bssid));
 418                 if (rep->mul_bssid == NULL)
 419                         break;
 420                 rep->mul_bssid->max_bssid_indicator = pos[0];
 421                 rep->mul_bssid->subelem_len = elen - 1;
 422                 os_memcpy(rep->mul_bssid->subelems, pos + 1, elen - 1);
 423                 break;
 424         }
 425 }
 426
 427
 428 static int wnm_nei_get_chan(struct wpa_supplicant *wpa_s, u8 op_class, u8 chan)
 429 {
 430         struct wpa_bss *bss = wpa_s->current_bss;
 431         const char *country = NULL;
 432         int freq;
 433
 434         if (bss) {
 435                 const u8 *elem = wpa_bss_get_ie(bss, WLAN_EID_COUNTRY);
 436
 437                 if (elem && elem[1] >= 2)
 438                         country = (const char *) (elem + 2);
 439         }
 440
 441         freq = ieee80211_chan_to_freq(country, op_class, chan);
 442         if (freq <= 0 && op_class == 0) {
 443                 /*
 444                  * Some APs do not advertise correct operating class
 445                  * information. Try to determine the most likely operating
 446                  * frequency based on the channel number.
 447                  */
 448                 if (chan >= 1 && chan <= 13)
 449                         freq = 2407 + chan * 5;
 450                 else if (chan == 14)
 451                         freq = 2484;
 452                 else if (chan >= 36 && chan <= 169)
 453                         freq = 5000 + chan * 5;
 454         }
 455         return freq;
 456 }
 457
 458
 459 static void wnm_parse_neighbor_report(struct wpa_supplicant *wpa_s,
 460                                       const u8 *pos, u8 len,
 461                                       struct neighbor_report *rep)
 462 {
 463         u8 left = len;
 464
 465         if (left < 13) {
 466                 wpa_printf(MSG_DEBUG, "WNM: Too short neighbor report");
 467                 return;
 468         }
 469
 470         os_memcpy(rep->bssid, pos, ETH_ALEN);
 471         rep->bssid_info = WPA_GET_LE32(pos + ETH_ALEN);
 472         rep->regulatory_class = *(pos + 10);
 473         rep->channel_number = *(pos + 11);
 474         rep->phy_type = *(pos + 12);
 475
 476         pos += 13;
 477         left -= 13;
 478
 479         while (left >= 2) {
 480                 u8 id, elen;
 481
 482                 id = *pos++;
 483                 elen = *pos++;
 484                 wpa_printf(MSG_DEBUG, "WNM: Subelement id=%u len=%u", id, elen);
 485                 left -= 2;
 486                 if (elen > left) {
 487                         wpa_printf(MSG_DEBUG,
 488                                    "WNM: Truncated neighbor report subelement");
 489                         break;
 490                 }
 491                 wnm_parse_neighbor_report_elem(rep, id, elen, pos);
 492                 left -= elen;
 493                 pos += elen;
 494         }
 495
 496         rep->freq = wnm_nei_get_chan(wpa_s, rep->regulatory_class,
 497                                      rep->channel_number);
 498 }
 499
 500
 501 static struct wpa_bss *
 502 compare_scan_neighbor_results(struct wpa_supplicant *wpa_s)
 503 {
 504
 505         u8 i;
 506         struct wpa_bss *bss = wpa_s->current_bss;
 507         struct wpa_bss *target;
 508
 509         if (!bss)
 510                 return 0;
 511
 512         wpa_printf(MSG_DEBUG, "WNM: Current BSS " MACSTR " RSSI %d",
 513                    MAC2STR(wpa_s->bssid), bss->level);
 514
 515         for (i = 0; i < wpa_s->wnm_num_neighbor_report; i++) {
 516                 struct neighbor_report *nei;
 517
 518                 nei = &wpa_s->wnm_neighbor_report_elements[i];
 519                 if (nei->preference_present && nei->preference == 0) {
 520                         wpa_printf(MSG_DEBUG, "Skip excluded BSS " MACSTR,
 521                                    MAC2STR(nei->bssid));
 522                         continue;
 523                 }
 524
 525                 target = wpa_bss_get_bssid(wpa_s, nei->bssid);
 526                 if (!target) {
 527                         wpa_printf(MSG_DEBUG, "Candidate BSS " MACSTR
 528                                    " (pref %d) not found in scan results",
 529                                    MAC2STR(nei->bssid),
 530                                    nei->preference_present ? nei->preference :
 531                                    -1);
 532                         continue;
 533                 }
 534
 535                 if (bss->ssid_len != target->ssid_len ||
 536                     os_memcmp(bss->ssid, target->ssid, bss->ssid_len) != 0) {
 537                         /*
 538                          * TODO: Could consider allowing transition to another
 539                          * ESS if PMF was enabled for the association.
 540                          */
 541                         wpa_printf(MSG_DEBUG, "Candidate BSS " MACSTR
 542                                    " (pref %d) in different ESS",
 543                                    MAC2STR(nei->bssid),
 544                                    nei->preference_present ? nei->preference :
 545                                    -1);
 546                         continue;
 547                 }
 548
 549                 if (wpa_is_bss_tmp_disallowed(wpa_s, target->bssid)) {
 550                         wpa_printf(MSG_DEBUG,
 551                                    "MBO: Candidate BSS " MACSTR
 552                                    " retry delay is not over yet",
 553                                    MAC2STR(nei->bssid));
 554                         continue;
 555                 }
 556
 557                 if (target->level < bss->level && target->level < -80) {
 558                         wpa_printf(MSG_DEBUG, "Candidate BSS " MACSTR
 559                                    " (pref %d) does not have sufficient signal level (%d)",
 560                                    MAC2STR(nei->bssid),
 561                                    nei->preference_present ? nei->preference :
 562                                    -1,
 563                                    target->level);
 564                         continue;
 565                 }
 566
 567                 wpa_printf(MSG_DEBUG,
 568                            "WNM: Found an acceptable preferred transition candidate BSS "
 569                            MACSTR " (RSSI %d)",
 570                            MAC2STR(nei->bssid), target->level);
 571                 return target;
 572         }
 573
 574         return NULL;
 575 }
 576
 577
 578 static void wnm_send_bss_transition_mgmt_resp(
 579         struct wpa_supplicant *wpa_s, u8 dialog_token,
 580         enum bss_trans_mgmt_status_code status, u8 delay,
 581         const u8 *target_bssid)
 582 {
 583         u8 buf[1000], *pos;
 584         struct ieee80211_mgmt *mgmt;
 585         size_t len;
 586         int res;
 587
 588         wpa_printf(MSG_DEBUG, "WNM: Send BSS Transition Management Response "
 589                    "to " MACSTR " dialog_token=%u status=%u delay=%d",
 590                    MAC2STR(wpa_s->bssid), dialog_token, status, delay);
 591         if (!wpa_s->current_bss) {
 592                 wpa_printf(MSG_DEBUG,
 593                            "WNM: Current BSS not known - drop response");
 594                 return;
 595         }
 596
 597         mgmt = (struct ieee80211_mgmt *) buf;
 598         os_memset(&buf, 0, sizeof(buf));
 599         os_memcpy(mgmt->da, wpa_s->bssid, ETH_ALEN);
 600         os_memcpy(mgmt->sa, wpa_s->own_addr, ETH_ALEN);
 601         os_memcpy(mgmt->bssid, wpa_s->bssid, ETH_ALEN);
 602         mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
 603                                            WLAN_FC_STYPE_ACTION);
 604         mgmt->u.action.category = WLAN_ACTION_WNM;
 605         mgmt->u.action.u.bss_tm_resp.action = WNM_BSS_TRANS_MGMT_RESP;
 606         mgmt->u.action.u.bss_tm_resp.dialog_token = dialog_token;
 607         mgmt->u.action.u.bss_tm_resp.status_code = status;
 608         mgmt->u.action.u.bss_tm_resp.bss_termination_delay = delay;
 609         pos = mgmt->u.action.u.bss_tm_resp.variable;
 610         if (target_bssid) {
 611                 os_memcpy(pos, target_bssid, ETH_ALEN);
 612                 pos += ETH_ALEN;
 613         } else if (status == WNM_BSS_TM_ACCEPT) {
 614                 /*
 615                  * P802.11-REVmc clarifies that the Target BSSID field is always
 616                  * present when status code is zero, so use a fake value here if
 617                  * no BSSID is yet known.
 618                  */
 619                 os_memset(pos, 0, ETH_ALEN);
 620                 pos += ETH_ALEN;
 621         }
 622
 623 #ifdef CONFIG_MBO
 624         if (status != WNM_BSS_TM_ACCEPT) {
 625                 pos += wpas_mbo_ie_bss_trans_reject(
 626                         wpa_s, pos, buf + sizeof(buf) - pos,
 627                         MBO_TRANSITION_REJECT_REASON_UNSPECIFIED);
 628         }
 629 #endif /* CONFIG_MBO */
 630
 631         len = pos - (u8 *) &mgmt->u.action.category;
 632
 633         res = wpa_drv_send_action(wpa_s, wpa_s->assoc_freq, 0, wpa_s->bssid,
 634                                   wpa_s->own_addr, wpa_s->bssid,
 635                                   &mgmt->u.action.category, len, 0);
 636         if (res < 0) {
 637                 wpa_printf(MSG_DEBUG,
 638                            "WNM: Failed to send BSS Transition Management Response");
 639         }
 640 }
 641
 642
 643 int wnm_scan_process(struct wpa_supplicant *wpa_s, int reply_on_fail)
 644 {
 645         struct wpa_bss *bss;
 646         struct wpa_ssid *ssid = wpa_s->current_ssid;
 647         enum bss_trans_mgmt_status_code status = WNM_BSS_TM_REJECT_UNSPECIFIED;
 648
 649         if (!wpa_s->wnm_neighbor_report_elements)
 650                 return 0;
 651
 652         if (os_reltime_before(&wpa_s->wnm_cand_valid_until,
 653                               &wpa_s->scan_trigger_time)) {
 654                 wpa_printf(MSG_DEBUG, "WNM: Previously stored BSS transition candidate list is not valid anymore - drop it");
 655                 wnm_deallocate_memory(wpa_s);
 656                 return 0;
 657         }
 658
 659         if (!wpa_s->current_bss ||
 660             os_memcmp(wpa_s->wnm_cand_from_bss, wpa_s->current_bss->bssid,
 661                       ETH_ALEN) != 0) {
 662                 wpa_printf(MSG_DEBUG, "WNM: Stored BSS transition candidate list not from the current BSS - ignore it");
 663                 return 0;
 664         }
 665
 666         /* Compare the Neighbor Report and scan results */
 667         bss = compare_scan_neighbor_results(wpa_s);
 668         if (!bss) {
 669                 wpa_printf(MSG_DEBUG, "WNM: No BSS transition candidate match found");
 670                 status = WNM_BSS_TM_REJECT_NO_SUITABLE_CANDIDATES;
 671                 goto send_bss_resp_fail;
 672         }
 673
 674         /* Associate to the network */
 675         /* Send the BSS Management Response - Accept */
 676         if (wpa_s->wnm_reply) {
 677                 wpa_s->wnm_reply = 0;
 678                 wnm_send_bss_transition_mgmt_resp(wpa_s,
 679                                                   wpa_s->wnm_dialog_token,
 680                                                   WNM_BSS_TM_ACCEPT,
 681                                                   0, bss->bssid);
 682         }
 683
 684         if (bss == wpa_s->current_bss) {
 685                 wpa_printf(MSG_DEBUG,
 686                            "WNM: Already associated with the preferred candidate");
 687                 wnm_deallocate_memory(wpa_s);
 688                 return 1;
 689         }
 690
 691         wpa_s->reassociate = 1;
 692         wpa_supplicant_connect(wpa_s, bss, ssid);
 693         wnm_deallocate_memory(wpa_s);
 694         return 1;
 695
 696 send_bss_resp_fail:
 697         if (!reply_on_fail)
 698                 return 0;
 699
 700         /* Send reject response for all the failures */
 701
 702         if (wpa_s->wnm_reply) {
 703                 wpa_s->wnm_reply = 0;
 704                 wnm_send_bss_transition_mgmt_resp(wpa_s,
 705                                                   wpa_s->wnm_dialog_token,
 706                                                   status, 0, NULL);
 707         }
 708         wnm_deallocate_memory(wpa_s);
 709
 710         return 0;
 711 }
 712
 713
 714 static int cand_pref_compar(const void *a, const void *b)
 715 {
 716         const struct neighbor_report *aa = a;
 717         const struct neighbor_report *bb = b;
 718
 719         if (!aa->preference_present && !bb->preference_present)
 720                 return 0;
 721         if (!aa->preference_present)
 722                 return 1;
 723         if (!bb->preference_present)
 724                 return -1;
 725         if (bb->preference > aa->preference)
 726                 return 1;
 727         if (bb->preference < aa->preference)
 728                 return -1;
 729         return 0;
 730 }
 731
 732
 733 static void wnm_sort_cand_list(struct wpa_supplicant *wpa_s)
 734 {
 735         if (!wpa_s->wnm_neighbor_report_elements)
 736                 return;
 737         qsort(wpa_s->wnm_neighbor_report_elements,
 738               wpa_s->wnm_num_neighbor_report, sizeof(struct neighbor_report),
 739               cand_pref_compar);
 740 }
 741
 742
 743 static void wnm_dump_cand_list(struct wpa_supplicant *wpa_s)
 744 {
 745         unsigned int i;
 746
 747         wpa_printf(MSG_DEBUG, "WNM: BSS Transition Candidate List");
 748         if (!wpa_s->wnm_neighbor_report_elements)
 749                 return;
 750         for (i = 0; i < wpa_s->wnm_num_neighbor_report; i++) {
 751                 struct neighbor_report *nei;
 752
 753                 nei = &wpa_s->wnm_neighbor_report_elements[i];
 754                 wpa_printf(MSG_DEBUG, "%u: " MACSTR
 755                            " info=0x%x op_class=%u chan=%u phy=%u pref=%d freq=%d",
 756                            i, MAC2STR(nei->bssid), nei->bssid_info,
 757                            nei->regulatory_class,
 758                            nei->channel_number, nei->phy_type,
 759                            nei->preference_present ? nei->preference : -1,
 760                            nei->freq);
 761         }
 762 }
 763
 764
 765 static int chan_supported(struct wpa_supplicant *wpa_s, int freq)
 766 {
 767         unsigned int i;
 768
 769         for (i = 0; i < wpa_s->hw.num_modes; i++) {
 770                 struct hostapd_hw_modes *mode = &wpa_s->hw.modes[i];
 771                 int j;
 772
 773                 for (j = 0; j < mode->num_channels; j++) {
 774                         struct hostapd_channel_data *chan;
 775
 776                         chan = &mode->channels[j];
 777                         if (chan->freq == freq &&
 778                             !(chan->flag & HOSTAPD_CHAN_DISABLED))
 779                                 return 1;
 780                 }
 781         }
 782
 783         return 0;
 784 }
 785
 786
 787 static void wnm_set_scan_freqs(struct wpa_supplicant *wpa_s)
 788 {
 789         int *freqs;
 790         int num_freqs = 0;
 791         unsigned int i;
 792
 793         if (!wpa_s->wnm_neighbor_report_elements)
 794                 return;
 795
 796         if (wpa_s->hw.modes == NULL)
 797                 return;
 798
 799         os_free(wpa_s->next_scan_freqs);
 800         wpa_s->next_scan_freqs = NULL;
 801
 802         freqs = os_calloc(wpa_s->wnm_num_neighbor_report + 1, sizeof(int));
 803         if (freqs == NULL)
 804                 return;
 805
 806         for (i = 0; i < wpa_s->wnm_num_neighbor_report; i++) {
 807                 struct neighbor_report *nei;
 808
 809                 nei = &wpa_s->wnm_neighbor_report_elements[i];
 810                 if (nei->freq <= 0) {
 811                         wpa_printf(MSG_DEBUG,
 812                                    "WNM: Unknown neighbor operating frequency for "
 813                                    MACSTR " - scan all channels",
 814                                    MAC2STR(nei->bssid));
 815                         os_free(freqs);
 816                         return;
 817                 }
 818                 if (chan_supported(wpa_s, nei->freq))
 819                         add_freq(freqs, &num_freqs, nei->freq);
 820         }
 821
 822         if (num_freqs == 0) {
 823                 os_free(freqs);
 824                 return;
 825         }
 826
 827         wpa_printf(MSG_DEBUG,
 828                    "WNM: Scan %d frequencies based on transition candidate list",
 829                    num_freqs);
 830         wpa_s->next_scan_freqs = freqs;
 831 }
 832
 833
 834 static void ieee802_11_rx_bss_trans_mgmt_req(struct wpa_supplicant *wpa_s,
 835                                              const u8 *pos, const u8 *end,
 836                                              int reply)
 837 {
 838         unsigned int beacon_int;
 839         u8 valid_int;
 840 #ifdef CONFIG_MBO
 841         const u8 *vendor;
 842 #endif /* CONFIG_MBO */
 843
 844         if (end - pos < 5)
 845                 return;
 846
 847         if (wpa_s->current_bss)
 848                 beacon_int = wpa_s->current_bss->beacon_int;
 849         else
 850                 beacon_int = 100; /* best guess */
 851
 852         wpa_s->wnm_dialog_token = pos[0];
 853         wpa_s->wnm_mode = pos[1];
 854         wpa_s->wnm_dissoc_timer = WPA_GET_LE16(pos + 2);
 855         valid_int = pos[4];
 856         wpa_s->wnm_reply = reply;
 857
 858         wpa_printf(MSG_DEBUG, "WNM: BSS Transition Management Request: "
 859                    "dialog_token=%u request_mode=0x%x "
 860                    "disassoc_timer=%u validity_interval=%u",
 861                    wpa_s->wnm_dialog_token, wpa_s->wnm_mode,
 862                    wpa_s->wnm_dissoc_timer, valid_int);
 863
 864         pos += 5;
 865
 866         if (wpa_s->wnm_mode & WNM_BSS_TM_REQ_BSS_TERMINATION_INCLUDED) {
 867                 if (end - pos < 12) {
 868                         wpa_printf(MSG_DEBUG, "WNM: Too short BSS TM Request");
 869                         return;
 870                 }
 871                 os_memcpy(wpa_s->wnm_bss_termination_duration, pos, 12);
 872                 pos += 12; /* BSS Termination Duration */
 873         }
 874
 875         if (wpa_s->wnm_mode & WNM_BSS_TM_REQ_ESS_DISASSOC_IMMINENT) {
 876                 char url[256];
 877
 878                 if (end - pos < 1 || 1 + pos[0] > end - pos) {
 879                         wpa_printf(MSG_DEBUG, "WNM: Invalid BSS Transition "
 880                                    "Management Request (URL)");
 881                         return;
 882                 }
 883                 os_memcpy(url, pos + 1, pos[0]);
 884                 url[pos[0]] = '\0';
 885                 pos += 1 + pos[0];
 886
 887                 wpa_msg(wpa_s, MSG_INFO, ESS_DISASSOC_IMMINENT "%d %u %s",
 888                         wpa_sm_pmf_enabled(wpa_s->wpa),
 889                         wpa_s->wnm_dissoc_timer * beacon_int * 128 / 125, url);
 890         }
 891
 892         if (wpa_s->wnm_mode & WNM_BSS_TM_REQ_DISASSOC_IMMINENT) {
 893                 wpa_msg(wpa_s, MSG_INFO, "WNM: Disassociation Imminent - "
 894                         "Disassociation Timer %u", wpa_s->wnm_dissoc_timer);
 895                 if (wpa_s->wnm_dissoc_timer && !wpa_s->scanning) {
 896                         /* TODO: mark current BSS less preferred for
 897                          * selection */
 898                         wpa_printf(MSG_DEBUG, "Trying to find another BSS");
 899                         wpa_supplicant_req_scan(wpa_s, 0, 0);
 900                 }
 901         }
 902
 903 #ifdef CONFIG_MBO
 904         vendor = get_ie(pos, end - pos, WLAN_EID_VENDOR_SPECIFIC);
 905         if (vendor)
 906                 wpas_mbo_ie_trans_req(wpa_s, vendor + 2, vendor[1]);
 907 #endif /* CONFIG_MBO */
 908
 909         if (wpa_s->wnm_mode & WNM_BSS_TM_REQ_PREF_CAND_LIST_INCLUDED) {
 910                 unsigned int valid_ms;
 911
 912                 wpa_msg(wpa_s, MSG_INFO, "WNM: Preferred List Available");
 913                 wnm_deallocate_memory(wpa_s);
 914                 wpa_s->wnm_neighbor_report_elements = os_calloc(
 915                         WNM_MAX_NEIGHBOR_REPORT,
 916                         sizeof(struct neighbor_report));
 917                 if (wpa_s->wnm_neighbor_report_elements == NULL)
 918                         return;
 919
 920                 while (end - pos >= 2 &&
 921                        wpa_s->wnm_num_neighbor_report < WNM_MAX_NEIGHBOR_REPORT)
 922                 {
 923                         u8 tag = *pos++;
 924                         u8 len = *pos++;
 925
 926                         wpa_printf(MSG_DEBUG, "WNM: Neighbor report tag %u",
 927                                    tag);
 928                         if (len > end - pos) {
 929                                 wpa_printf(MSG_DEBUG, "WNM: Truncated request");
 930                                 return;
 931                         }
 932                         if (tag == WLAN_EID_NEIGHBOR_REPORT) {
 933                                 struct neighbor_report *rep;
 934                                 rep = &wpa_s->wnm_neighbor_report_elements[
 935                                         wpa_s->wnm_num_neighbor_report];
 936                                 wnm_parse_neighbor_report(wpa_s, pos, len, rep);
 937                         }
 938
 939                         pos += len;
 940                         wpa_s->wnm_num_neighbor_report++;
 941                 }
 942                 wnm_sort_cand_list(wpa_s);
 943                 wnm_dump_cand_list(wpa_s);
 944                 valid_ms = valid_int * beacon_int * 128 / 125;
 945                 wpa_printf(MSG_DEBUG, "WNM: Candidate list valid for %u ms",
 946                            valid_ms);
 947                 os_get_reltime(&wpa_s->wnm_cand_valid_until);
 948                 wpa_s->wnm_cand_valid_until.sec += valid_ms / 1000;
 949                 wpa_s->wnm_cand_valid_until.usec += (valid_ms % 1000) * 1000;
 950                 wpa_s->wnm_cand_valid_until.sec +=
 951                         wpa_s->wnm_cand_valid_until.usec / 1000000;
 952                 wpa_s->wnm_cand_valid_until.usec %= 1000000;
 953                 os_memcpy(wpa_s->wnm_cand_from_bss, wpa_s->bssid, ETH_ALEN);
 954
 955                 if (wpa_s->last_scan_res_used > 0) {
 956                         struct os_reltime now;
 957
 958                         os_get_reltime(&now);
 959                         if (!os_reltime_expired(&now, &wpa_s->last_scan, 10)) {
 960                                 wpa_printf(MSG_DEBUG,
 961                                            "WNM: Try to use recent scan results");
 962                                 if (wnm_scan_process(wpa_s, 0) > 0)
 963                                         return;
 964                                 wpa_printf(MSG_DEBUG,
 965                                            "WNM: No match in previous scan results - try a new scan");
 966                         }
 967                 }
 968
 969                 wnm_set_scan_freqs(wpa_s);
 970                 wpa_supplicant_req_scan(wpa_s, 0, 0);
 971         } else if (reply) {
 972                 enum bss_trans_mgmt_status_code status;
 973                 if (wpa_s->wnm_mode & WNM_BSS_TM_REQ_ESS_DISASSOC_IMMINENT)
 974                         status = WNM_BSS_TM_ACCEPT;
 975                 else {
 976                         wpa_msg(wpa_s, MSG_INFO, "WNM: BSS Transition Management Request did not include candidates");
 977                         status = WNM_BSS_TM_REJECT_UNSPECIFIED;
 978                 }
 979                 wnm_send_bss_transition_mgmt_resp(wpa_s,
 980                                                   wpa_s->wnm_dialog_token,
 981                                                   status, 0, NULL);
 982         }
 983 }
 984
 985
 986 int wnm_send_bss_transition_mgmt_query(struct wpa_supplicant *wpa_s,
 987                                        u8 query_reason)
 988 {
 989         u8 buf[1000], *pos;
 990         struct ieee80211_mgmt *mgmt;
 991         size_t len;
 992         int ret;
 993
 994         wpa_printf(MSG_DEBUG, "WNM: Send BSS Transition Management Query to "
 995                    MACSTR " query_reason=%u",
 996                    MAC2STR(wpa_s->bssid), query_reason);
 997
 998         mgmt = (struct ieee80211_mgmt *) buf;
 999         os_memset(&buf, 0, sizeof(buf));
1000         os_memcpy(mgmt->da, wpa_s->bssid, ETH_ALEN);
1001         os_memcpy(mgmt->sa, wpa_s->own_addr, ETH_ALEN);
1002         os_memcpy(mgmt->bssid, wpa_s->bssid, ETH_ALEN);
1003         mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
1004                                            WLAN_FC_STYPE_ACTION);
1005         mgmt->u.action.category = WLAN_ACTION_WNM;
1006         mgmt->u.action.u.bss_tm_query.action = WNM_BSS_TRANS_MGMT_QUERY;
1007         mgmt->u.action.u.bss_tm_query.dialog_token = 1;
1008         mgmt->u.action.u.bss_tm_query.query_reason = query_reason;
1009         pos = mgmt->u.action.u.bss_tm_query.variable;
1010
1011         len = pos - (u8 *) &mgmt->u.action.category;
1012
1013         ret = wpa_drv_send_action(wpa_s, wpa_s->assoc_freq, 0, wpa_s->bssid,
1014                                   wpa_s->own_addr, wpa_s->bssid,
1015                                   &mgmt->u.action.category, len, 0);
1016
1017         return ret;
1018 }
1019
1020
1021 static void ieee802_11_rx_wnm_notif_req_wfa(struct wpa_supplicant *wpa_s,
1022                                             const u8 *sa, const u8 *data,
1023                                             int len)
1024 {
1025         const u8 *pos, *end, *next;
1026         u8 ie, ie_len;
1027
1028         pos = data;
1029         end = data + len;
1030
1031         while (end - pos > 1) {
1032                 ie = *pos++;
1033                 ie_len = *pos++;
1034                 wpa_printf(MSG_DEBUG, "WNM: WFA subelement %u len %u",
1035                            ie, ie_len);
1036                 if (ie_len > end - pos) {
1037                         wpa_printf(MSG_DEBUG, "WNM: Not enough room for "
1038                                    "subelement");
1039                         break;
1040                 }
1041                 next = pos + ie_len;
1042                 if (ie_len < 4) {
1043                         pos = next;
1044                         continue;
1045                 }
1046                 wpa_printf(MSG_DEBUG, "WNM: Subelement OUI %06x type %u",
1047                            WPA_GET_BE24(pos), pos[3]);
1048
1049 #ifdef CONFIG_HS20
1050                 if (ie == WLAN_EID_VENDOR_SPECIFIC && ie_len >= 5 &&
1051                     WPA_GET_BE24(pos) == OUI_WFA &&
1052                     pos[3] == HS20_WNM_SUB_REM_NEEDED) {
1053                         /* Subscription Remediation subelement */
1054                         const u8 *ie_end;
1055                         u8 url_len;
1056                         char *url;
1057                         u8 osu_method;
1058
1059                         wpa_printf(MSG_DEBUG, "WNM: Subscription Remediation "
1060                                    "subelement");
1061                         ie_end = pos + ie_len;
1062                         pos += 4;
1063                         url_len = *pos++;
1064                         if (url_len == 0) {
1065                                 wpa_printf(MSG_DEBUG, "WNM: No Server URL included");
1066                                 url = NULL;
1067                                 osu_method = 1;
1068                         } else {
1069                                 if (url_len + 1 > ie_end - pos) {
1070                                         wpa_printf(MSG_DEBUG, "WNM: Not enough room for Server URL (len=%u) and Server Method (left %d)",
1071                                                    url_len,
1072                                                    (int) (ie_end - pos));
1073                                         break;
1074                                 }
1075                                 url = os_malloc(url_len + 1);
1076                                 if (url == NULL)
1077                                         break;
1078                                 os_memcpy(url, pos, url_len);
1079                                 url[url_len] = '\0';
1080                                 osu_method = pos[url_len];
1081                         }
1082                         hs20_rx_subscription_remediation(wpa_s, url,
1083                                                          osu_method);
1084                         os_free(url);
1085                         pos = next;
1086                         continue;
1087                 }
1088
1089                 if (ie == WLAN_EID_VENDOR_SPECIFIC && ie_len >= 8 &&
1090                     WPA_GET_BE24(pos) == OUI_WFA &&
1091                     pos[3] == HS20_WNM_DEAUTH_IMMINENT_NOTICE) {
1092                         const u8 *ie_end;
1093                         u8 url_len;
1094                         char *url;
1095                         u8 code;
1096                         u16 reauth_delay;
1097
1098                         ie_end = pos + ie_len;
1099                         pos += 4;
1100                         code = *pos++;
1101                         reauth_delay = WPA_GET_LE16(pos);
1102                         pos += 2;
1103                         url_len = *pos++;
1104                         wpa_printf(MSG_DEBUG, "WNM: HS 2.0 Deauthentication "
1105                                    "Imminent - Reason Code %u   "
1106                                    "Re-Auth Delay %u  URL Length %u",
1107                                    code, reauth_delay, url_len);
1108                         if (url_len > ie_end - pos)
1109                                 break;
1110                         url = os_malloc(url_len + 1);
1111                         if (url == NULL)
1112                                 break;
1113                         os_memcpy(url, pos, url_len);
1114                         url[url_len] = '\0';
1115                         hs20_rx_deauth_imminent_notice(wpa_s, code,
1116                                                        reauth_delay, url);
1117                         os_free(url);
1118                         pos = next;
1119                         continue;
1120                 }
1121 #endif /* CONFIG_HS20 */
1122
1123                 pos = next;
1124         }
1125 }
1126
1127
1128 static void ieee802_11_rx_wnm_notif_req(struct wpa_supplicant *wpa_s,
1129                                         const u8 *sa, const u8 *frm, int len)
1130 {
1131         const u8 *pos, *end;
1132         u8 dialog_token, type;
1133
1134         /* Dialog Token [1] | Type [1] | Subelements */
1135
1136         if (len < 2 || sa == NULL)
1137                 return;
1138         end = frm + len;
1139         pos = frm;
1140         dialog_token = *pos++;
1141         type = *pos++;
1142
1143         wpa_dbg(wpa_s, MSG_DEBUG, "WNM: Received WNM-Notification Request "
1144                 "(dialog_token %u type %u sa " MACSTR ")",
1145                 dialog_token, type, MAC2STR(sa));
1146         wpa_hexdump(MSG_DEBUG, "WNM-Notification Request subelements",
1147                     pos, end - pos);
1148
1149         if (wpa_s->wpa_state != WPA_COMPLETED ||
1150             os_memcmp(sa, wpa_s->bssid, ETH_ALEN) != 0) {
1151                 wpa_dbg(wpa_s, MSG_DEBUG, "WNM: WNM-Notification frame not "
1152                         "from our AP - ignore it");
1153                 return;
1154         }
1155
1156         switch (type) {
1157         case 1:
1158                 ieee802_11_rx_wnm_notif_req_wfa(wpa_s, sa, pos, end - pos);
1159                 break;
1160         default:
1161                 wpa_dbg(wpa_s, MSG_DEBUG, "WNM: Ignore unknown "
1162                         "WNM-Notification type %u", type);
1163                 break;
1164         }
1165 }
1166
1167
1168 void ieee802_11_rx_wnm_action(struct wpa_supplicant *wpa_s,
1169                               const struct ieee80211_mgmt *mgmt, size_t len)
1170 {
1171         const u8 *pos, *end;
1172         u8 act;
1173
1174         if (len < IEEE80211_HDRLEN + 2)
1175                 return;
1176
1177         pos = ((const u8 *) mgmt) + IEEE80211_HDRLEN + 1;
1178         act = *pos++;
1179         end = ((const u8 *) mgmt) + len;
1180
1181         wpa_printf(MSG_DEBUG, "WNM: RX action %u from " MACSTR,
1182                    act, MAC2STR(mgmt->sa));
1183         if (wpa_s->wpa_state < WPA_ASSOCIATED ||
1184             os_memcmp(mgmt->sa, wpa_s->bssid, ETH_ALEN) != 0) {
1185                 wpa_printf(MSG_DEBUG, "WNM: Ignore unexpected WNM Action "
1186                            "frame");
1187                 return;
1188         }
1189
1190         switch (act) {
1191         case WNM_BSS_TRANS_MGMT_REQ:
1192                 ieee802_11_rx_bss_trans_mgmt_req(wpa_s, pos, end,
1193                                                  !(mgmt->da[0] & 0x01));
1194                 break;
1195         case WNM_SLEEP_MODE_RESP:
1196                 ieee802_11_rx_wnmsleep_resp(wpa_s, pos, end - pos);
1197                 break;
1198         case WNM_NOTIFICATION_REQ:
1199                 ieee802_11_rx_wnm_notif_req(wpa_s, mgmt->sa, pos, end - pos);
1200                 break;
1201         default:
1202                 wpa_printf(MSG_ERROR, "WNM: Unknown request");
1203                 break;
1204         }
1205 }