]>
Commit | Line | Data |
---|---|---|
d32d94db XC |
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 | ||
d32d94db XC |
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; | |
615a5d55 JM |
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; | |
d32d94db XC |
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 */ | |
4da10640 | 107 | if (!wpa_auth_uses_mfp(sta->wpa_sm) || |
615a5d55 | 108 | action_type != WNM_SLEEP_MODE_EXIT) { |
d32d94db XC |
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; | |
a8e93a1a JM |
135 | if (wnmtfs_ie) |
136 | os_memcpy(pos, wnmtfs_ie, wnmtfs_ie_len); | |
d32d94db XC |
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 && | |
615a5d55 | 156 | wnmsleep_ie.action_type == WNM_SLEEP_MODE_ENTER) { |
d32d94db XC |
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 | */ | |
4da10640 JM |
166 | if ((wnmsleep_ie.status == WNM_STATUS_SLEEP_ACCEPT || |
167 | wnmsleep_ie.status == | |
168 | WNM_STATUS_SLEEP_EXIT_ACCEPT_GTK_UPDATE) && | |
615a5d55 | 169 | wnmsleep_ie.action_type == WNM_SLEEP_MODE_EXIT) { |
d32d94db XC |
170 | wpa_set_wnmsleep(sta->wpa_sm, 0); |
171 | hostapd_drv_wnm_oper(hapd, WNM_SLEEP_EXIT_CONFIRM, | |
172 | addr, NULL, NULL); | |
4da10640 | 173 | if (!wpa_auth_uses_mfp(sta->wpa_sm)) |
d32d94db XC |
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 | { | |
c79938a5 JM |
190 | /* Dialog Token [1] | WNM-Sleep Mode IE | TFS Response IE */ |
191 | const u8 *pos = frm; | |
192 | u8 dialog_token; | |
d32d94db XC |
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 | ||
c79938a5 JM |
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; | |
d32d94db | 204 | if (*pos == WLAN_EID_WNMSLEEP) |
c79938a5 | 205 | wnmsleep_ie = (struct wnm_sleep_element *) pos; |
d32d94db XC |
206 | else if (*pos == WLAN_EID_TFS_REQ) { |
207 | if (!tfsreq_ie_start) | |
c79938a5 JM |
208 | tfsreq_ie_start = (u8 *) pos; |
209 | tfsreq_ie_end = (u8 *) pos; | |
d32d94db | 210 | } else |
c79938a5 JM |
211 | wpa_printf(MSG_DEBUG, "WNM: EID %d not recognized", |
212 | *pos); | |
d32d94db XC |
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 | ||
615a5d55 JM |
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) { | |
d32d94db XC |
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 | ||
615a5d55 | 238 | if (wnmsleep_ie->action_type == WNM_SLEEP_MODE_EXIT) { |
d32d94db XC |
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 | ||
c79938a5 JM |
246 | int ieee802_11_rx_wnm_action_ap(struct hostapd_data *hapd, |
247 | struct rx_action *action) | |
d32d94db | 248 | { |
c79938a5 JM |
249 | if (action->len < 1 || action->data == NULL) |
250 | return -1; | |
d32d94db | 251 | |
c79938a5 | 252 | switch (action->data[0]) { |
d32d94db XC |
253 | case WNM_SLEEP_MODE_REQ: |
254 | ieee802_11_rx_wnmsleep_req(hapd, action->sa, action->data + 1, | |
c79938a5 JM |
255 | action->len - 1); |
256 | return 0; | |
d32d94db | 257 | } |
c79938a5 JM |
258 | |
259 | wpa_printf(MSG_DEBUG, "WNM: Unsupported WNM Action %u from " MACSTR, | |
260 | action->data[0], MAC2STR(action->sa)); | |
261 | return -1; | |
d32d94db | 262 | } |