contrib/wpa/src/ap/wnm_ap.c

   1 /*
   2  * hostapd - WNM
   3  * Copyright (c) 2011-2012, 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 "ap/hostapd.h"
  14 #include "ap/sta_info.h"
  15 #include "ap/ap_config.h"
  16 #include "ap/ap_drv_ops.h"
  17 #include "ap/wpa_auth.h"
  18 #include "wnm_ap.h"
  19
  20 #define MAX_TFS_IE_LEN  1024
  21
  22
  23 /* get the TFS IE from driver */
  24 static int ieee80211_11_get_tfs_ie(struct hostapd_data *hapd, const u8 *addr,
  25                                    u8 *buf, u16 *buf_len, enum wnm_oper oper)
  26 {
  27         wpa_printf(MSG_DEBUG, "%s: TFS get operation %d", __func__, oper);
  28
  29         return hostapd_drv_wnm_oper(hapd, oper, addr, buf, buf_len);
  30 }
  31
  32
  33 /* set the TFS IE to driver */
  34 static int ieee80211_11_set_tfs_ie(struct hostapd_data *hapd, const u8 *addr,
  35                                    u8 *buf, u16 *buf_len, enum wnm_oper oper)
  36 {
  37         wpa_printf(MSG_DEBUG, "%s: TFS set operation %d", __func__, oper);
  38
  39         return hostapd_drv_wnm_oper(hapd, oper, addr, buf, buf_len);
  40 }
  41
  42
  43 /* MLME-SLEEPMODE.response */
  44 static int ieee802_11_send_wnmsleep_resp(struct hostapd_data *hapd,
  45                                          const u8 *addr, u8 dialog_token,
  46                                          u8 action_type, u16 intval)
  47 {
  48         struct ieee80211_mgmt *mgmt;
  49         int res;
  50         size_t len;
  51         size_t gtk_elem_len = 0;
  52         size_t igtk_elem_len = 0;
  53         struct wnm_sleep_element wnmsleep_ie;
  54         u8 *wnmtfs_ie;
  55         u8 wnmsleep_ie_len;
  56         u16 wnmtfs_ie_len;
  57         u8 *pos;
  58         struct sta_info *sta;
  59         enum wnm_oper tfs_oper = action_type == WNM_SLEEP_MODE_ENTER ?
  60                 WNM_SLEEP_TFS_RESP_IE_ADD : WNM_SLEEP_TFS_RESP_IE_NONE;
  61
  62         sta = ap_get_sta(hapd, addr);
  63         if (sta == NULL) {
  64                 wpa_printf(MSG_DEBUG, "%s: station not found", __func__);
  65                 return -EINVAL;
  66         }
  67
  68         /* WNM-Sleep Mode IE */
  69         os_memset(&wnmsleep_ie, 0, sizeof(struct wnm_sleep_element));
  70         wnmsleep_ie_len = sizeof(struct wnm_sleep_element);
  71         wnmsleep_ie.eid = WLAN_EID_WNMSLEEP;
  72         wnmsleep_ie.len = wnmsleep_ie_len - 2;
  73         wnmsleep_ie.action_type = action_type;
  74         wnmsleep_ie.status = WNM_STATUS_SLEEP_ACCEPT;
  75         wnmsleep_ie.intval = intval;
  76
  77         /* TFS IE(s) */
  78         wnmtfs_ie = os_zalloc(MAX_TFS_IE_LEN);
  79         if (wnmtfs_ie == NULL)
  80                 return -1;
  81         if (ieee80211_11_get_tfs_ie(hapd, addr, wnmtfs_ie, &wnmtfs_ie_len,
  82                                     tfs_oper)) {
  83                 wnmtfs_ie_len = 0;
  84                 os_free(wnmtfs_ie);
  85                 wnmtfs_ie = NULL;
  86         }
  87
  88 #define MAX_GTK_SUBELEM_LEN 45
  89 #define MAX_IGTK_SUBELEM_LEN 26
  90         mgmt = os_zalloc(sizeof(*mgmt) + wnmsleep_ie_len +
  91                          MAX_GTK_SUBELEM_LEN + MAX_IGTK_SUBELEM_LEN);
  92         if (mgmt == NULL) {
  93                 wpa_printf(MSG_DEBUG, "MLME: Failed to allocate buffer for "
  94                            "WNM-Sleep Response action frame");
  95                 return -1;
  96         }
  97         os_memcpy(mgmt->da, addr, ETH_ALEN);
  98         os_memcpy(mgmt->sa, hapd->own_addr, ETH_ALEN);
  99         os_memcpy(mgmt->bssid, hapd->own_addr, ETH_ALEN);
 100         mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
 101                                            WLAN_FC_STYPE_ACTION);
 102         mgmt->u.action.category = WLAN_ACTION_WNM;
 103         mgmt->u.action.u.wnm_sleep_resp.action = WNM_SLEEP_MODE_RESP;
 104         mgmt->u.action.u.wnm_sleep_resp.dialogtoken = dialog_token;
 105         pos = (u8 *)mgmt->u.action.u.wnm_sleep_resp.variable;
 106         /* add key data if MFP is enabled */
 107         if (!wpa_auth_uses_mfp(sta->wpa_sm) ||
 108             action_type != WNM_SLEEP_MODE_EXIT) {
 109                 mgmt->u.action.u.wnm_sleep_resp.keydata_len = 0;
 110         } else {
 111                 gtk_elem_len = wpa_wnmsleep_gtk_subelem(sta->wpa_sm, pos);
 112                 pos += gtk_elem_len;
 113                 wpa_printf(MSG_DEBUG, "Pass 4, gtk_len = %d",
 114                            (int) gtk_elem_len);
 115 #ifdef CONFIG_IEEE80211W
 116                 res = wpa_wnmsleep_igtk_subelem(sta->wpa_sm, pos);
 117                 if (res < 0) {
 118                         os_free(wnmtfs_ie);
 119                         os_free(mgmt);
 120                         return -1;
 121                 }
 122                 igtk_elem_len = res;
 123                 pos += igtk_elem_len;
 124                 wpa_printf(MSG_DEBUG, "Pass 4 igtk_len = %d",
 125                            (int) igtk_elem_len);
 126 #endif /* CONFIG_IEEE80211W */
 127
 128                 WPA_PUT_LE16((u8 *)
 129                              &mgmt->u.action.u.wnm_sleep_resp.keydata_len,
 130                              gtk_elem_len + igtk_elem_len);
 131         }
 132         os_memcpy(pos, &wnmsleep_ie, wnmsleep_ie_len);
 133         /* copy TFS IE here */
 134         pos += wnmsleep_ie_len;
 135         if (wnmtfs_ie)
 136                 os_memcpy(pos, wnmtfs_ie, wnmtfs_ie_len);
 137
 138         len = 1 + sizeof(mgmt->u.action.u.wnm_sleep_resp) + gtk_elem_len +
 139                 igtk_elem_len + wnmsleep_ie_len + wnmtfs_ie_len;
 140
 141         /* In driver, response frame should be forced to sent when STA is in
 142          * PS mode */
 143         res = hostapd_drv_send_action(hapd, hapd->iface->freq, 0,
 144                                       mgmt->da, &mgmt->u.action.category, len);
 145
 146         if (!res) {
 147                 wpa_printf(MSG_DEBUG, "Successfully send WNM-Sleep Response "
 148                            "frame");
 149
 150                 /* when entering wnmsleep
 151                  * 1. pause the node in driver
 152                  * 2. mark the node so that AP won't update GTK/IGTK during
 153                  * WNM Sleep
 154                  */
 155                 if (wnmsleep_ie.status == WNM_STATUS_SLEEP_ACCEPT &&
 156                     wnmsleep_ie.action_type == WNM_SLEEP_MODE_ENTER) {
 157                         hostapd_drv_wnm_oper(hapd, WNM_SLEEP_ENTER_CONFIRM,
 158                                              addr, NULL, NULL);
 159                         wpa_set_wnmsleep(sta->wpa_sm, 1);
 160                 }
 161                 /* when exiting wnmsleep
 162                  * 1. unmark the node
 163                  * 2. start GTK/IGTK update if MFP is not used
 164                  * 3. unpause the node in driver
 165                  */
 166                 if ((wnmsleep_ie.status == WNM_STATUS_SLEEP_ACCEPT ||
 167                      wnmsleep_ie.status ==
 168                      WNM_STATUS_SLEEP_EXIT_ACCEPT_GTK_UPDATE) &&
 169                     wnmsleep_ie.action_type == WNM_SLEEP_MODE_EXIT) {
 170                         wpa_set_wnmsleep(sta->wpa_sm, 0);
 171                         hostapd_drv_wnm_oper(hapd, WNM_SLEEP_EXIT_CONFIRM,
 172                                              addr, NULL, NULL);
 173                         if (!wpa_auth_uses_mfp(sta->wpa_sm))
 174                                 wpa_wnmsleep_rekey_gtk(sta->wpa_sm);
 175                 }
 176         } else
 177                 wpa_printf(MSG_DEBUG, "Fail to send WNM-Sleep Response frame");
 178
 179 #undef MAX_GTK_SUBELEM_LEN
 180 #undef MAX_IGTK_SUBELEM_LEN
 181         os_free(wnmtfs_ie);
 182         os_free(mgmt);
 183         return res;
 184 }
 185
 186
 187 static void ieee802_11_rx_wnmsleep_req(struct hostapd_data *hapd,
 188                                        const u8 *addr, const u8 *frm, int len)
 189 {
 190         /* Dialog Token [1] | WNM-Sleep Mode IE | TFS Response IE */
 191         const u8 *pos = frm;
 192         u8 dialog_token;
 193         struct wnm_sleep_element *wnmsleep_ie = NULL;
 194         /* multiple TFS Req IE (assuming consecutive) */
 195         u8 *tfsreq_ie_start = NULL;
 196         u8 *tfsreq_ie_end = NULL;
 197         u16 tfsreq_ie_len = 0;
 198
 199         dialog_token = *pos++;
 200         while (pos + 1 < frm + len) {
 201                 u8 ie_len = pos[1];
 202                 if (pos + 2 + ie_len > frm + len)
 203                         break;
 204                 if (*pos == WLAN_EID_WNMSLEEP)
 205                         wnmsleep_ie = (struct wnm_sleep_element *) pos;
 206                 else if (*pos == WLAN_EID_TFS_REQ) {
 207                         if (!tfsreq_ie_start)
 208                                 tfsreq_ie_start = (u8 *) pos;
 209                         tfsreq_ie_end = (u8 *) pos;
 210                 } else
 211                         wpa_printf(MSG_DEBUG, "WNM: EID %d not recognized",
 212                                    *pos);
 213                 pos += ie_len + 2;
 214         }
 215
 216         if (!wnmsleep_ie) {
 217                 wpa_printf(MSG_DEBUG, "No WNM-Sleep IE found");
 218                 return;
 219         }
 220
 221         if (wnmsleep_ie->action_type == WNM_SLEEP_MODE_ENTER &&
 222             tfsreq_ie_start && tfsreq_ie_end &&
 223             tfsreq_ie_end - tfsreq_ie_start >= 0) {
 224                 tfsreq_ie_len = (tfsreq_ie_end + tfsreq_ie_end[1] + 2) -
 225                         tfsreq_ie_start;
 226                 wpa_printf(MSG_DEBUG, "TFS Req IE(s) found");
 227                 /* pass the TFS Req IE(s) to driver for processing */
 228                 if (ieee80211_11_set_tfs_ie(hapd, addr, tfsreq_ie_start,
 229                                             &tfsreq_ie_len,
 230                                             WNM_SLEEP_TFS_REQ_IE_SET))
 231                         wpa_printf(MSG_DEBUG, "Fail to set TFS Req IE");
 232         }
 233
 234         ieee802_11_send_wnmsleep_resp(hapd, addr, dialog_token,
 235                                       wnmsleep_ie->action_type,
 236                                       wnmsleep_ie->intval);
 237
 238         if (wnmsleep_ie->action_type == WNM_SLEEP_MODE_EXIT) {
 239                 /* clear the tfs after sending the resp frame */
 240                 ieee80211_11_set_tfs_ie(hapd, addr, tfsreq_ie_start,
 241                                         &tfsreq_ie_len, WNM_SLEEP_TFS_IE_DEL);
 242         }
 243 }
 244
 245
 246 int ieee802_11_rx_wnm_action_ap(struct hostapd_data *hapd,
 247                                 struct rx_action *action)
 248 {
 249         if (action->len < 1 || action->data == NULL)
 250                 return -1;
 251
 252         switch (action->data[0]) {
 253         case WNM_BSS_TRANS_MGMT_QUERY:
 254                 wpa_printf(MSG_DEBUG, "WNM: BSS Transition Management Query");
 255                 /* TODO */
 256                 return -1;
 257         case WNM_BSS_TRANS_MGMT_RESP:
 258                 wpa_printf(MSG_DEBUG, "WNM: BSS Transition Management "
 259                            "Response");
 260                 /* TODO */
 261                 return -1;
 262         case WNM_SLEEP_MODE_REQ:
 263                 ieee802_11_rx_wnmsleep_req(hapd, action->sa, action->data + 1,
 264                                            action->len - 1);
 265                 return 0;
 266         }
 267
 268         wpa_printf(MSG_DEBUG, "WNM: Unsupported WNM Action %u from " MACSTR,
 269                    action->data[0], MAC2STR(action->sa));
 270         return -1;
 271 }