1/*
2 * mac80211 TDLS handling code
3 *
4 * Copyright 2006-2010	Johannes Berg <johannes@sipsolutions.net>
5 * Copyright 2014, Intel Corporation
6 * Copyright 2014  Intel Mobile Communications GmbH
7 *
8 * This file is GPLv2 as found in COPYING.
9 */
10
11#include <linux/ieee80211.h>
12#include <linux/log2.h>
13#include <net/cfg80211.h>
14#include "ieee80211_i.h"
15#include "driver-ops.h"
16
17/* give usermode some time for retries in setting up the TDLS session */
18#define TDLS_PEER_SETUP_TIMEOUT	(15 * HZ)
19
20void ieee80211_tdls_peer_del_work(struct work_struct *wk)
21{
22	struct ieee80211_sub_if_data *sdata;
23	struct ieee80211_local *local;
24
25	sdata = container_of(wk, struct ieee80211_sub_if_data,
26			     u.mgd.tdls_peer_del_work.work);
27	local = sdata->local;
28
29	mutex_lock(&local->mtx);
30	if (!is_zero_ether_addr(sdata->u.mgd.tdls_peer)) {
31		tdls_dbg(sdata, "TDLS del peer %pM\n", sdata->u.mgd.tdls_peer);
32		sta_info_destroy_addr(sdata, sdata->u.mgd.tdls_peer);
33		eth_zero_addr(sdata->u.mgd.tdls_peer);
34	}
35	mutex_unlock(&local->mtx);
36}
37
38static void ieee80211_tdls_add_ext_capab(struct ieee80211_local *local,
39					 struct sk_buff *skb)
40{
41	u8 *pos = (void *)skb_put(skb, 7);
42	bool chan_switch = local->hw.wiphy->features &
43			   NL80211_FEATURE_TDLS_CHANNEL_SWITCH;
44
45	*pos++ = WLAN_EID_EXT_CAPABILITY;
46	*pos++ = 5; /* len */
47	*pos++ = 0x0;
48	*pos++ = 0x0;
49	*pos++ = 0x0;
50	*pos++ = chan_switch ? WLAN_EXT_CAPA4_TDLS_CHAN_SWITCH : 0;
51	*pos++ = WLAN_EXT_CAPA5_TDLS_ENABLED;
52}
53
54static u8
55ieee80211_tdls_add_subband(struct ieee80211_sub_if_data *sdata,
56			   struct sk_buff *skb, u16 start, u16 end,
57			   u16 spacing)
58{
59	u8 subband_cnt = 0, ch_cnt = 0;
60	struct ieee80211_channel *ch;
61	struct cfg80211_chan_def chandef;
62	int i, subband_start;
63
64	for (i = start; i <= end; i += spacing) {
65		if (!ch_cnt)
66			subband_start = i;
67
68		ch = ieee80211_get_channel(sdata->local->hw.wiphy, i);
69		if (ch) {
70			/* we will be active on the channel */
71			cfg80211_chandef_create(&chandef, ch,
72						NL80211_CHAN_NO_HT);
73			if (cfg80211_reg_can_beacon(sdata->local->hw.wiphy,
74						    &chandef,
75						    sdata->wdev.iftype)) {
76				ch_cnt++;
77				/*
78				 * check if the next channel is also part of
79				 * this allowed range
80				 */
81				continue;
82			}
83		}
84
85		/*
86		 * we've reached the end of a range, with allowed channels
87		 * found
88		 */
89		if (ch_cnt) {
90			u8 *pos = skb_put(skb, 2);
91			*pos++ = ieee80211_frequency_to_channel(subband_start);
92			*pos++ = ch_cnt;
93
94			subband_cnt++;
95			ch_cnt = 0;
96		}
97	}
98
99	/* all channels in the requested range are allowed - add them here */
100	if (ch_cnt) {
101		u8 *pos = skb_put(skb, 2);
102		*pos++ = ieee80211_frequency_to_channel(subband_start);
103		*pos++ = ch_cnt;
104
105		subband_cnt++;
106	}
107
108	return subband_cnt;
109}
110
111static void
112ieee80211_tdls_add_supp_channels(struct ieee80211_sub_if_data *sdata,
113				 struct sk_buff *skb)
114{
115	/*
116	 * Add possible channels for TDLS. These are channels that are allowed
117	 * to be active.
118	 */
119	u8 subband_cnt;
120	u8 *pos = skb_put(skb, 2);
121
122	*pos++ = WLAN_EID_SUPPORTED_CHANNELS;
123
124	/*
125	 * 5GHz and 2GHz channels numbers can overlap. Ignore this for now, as
126	 * this doesn't happen in real world scenarios.
127	 */
128
129	/* 2GHz, with 5MHz spacing */
130	subband_cnt = ieee80211_tdls_add_subband(sdata, skb, 2412, 2472, 5);
131
132	/* 5GHz, with 20MHz spacing */
133	subband_cnt += ieee80211_tdls_add_subband(sdata, skb, 5000, 5825, 20);
134
135	/* length */
136	*pos = 2 * subband_cnt;
137}
138
139static void ieee80211_tdls_add_oper_classes(struct ieee80211_sub_if_data *sdata,
140					    struct sk_buff *skb)
141{
142	u8 *pos;
143	u8 op_class;
144
145	if (!ieee80211_chandef_to_operating_class(&sdata->vif.bss_conf.chandef,
146						  &op_class))
147		return;
148
149	pos = skb_put(skb, 4);
150	*pos++ = WLAN_EID_SUPPORTED_REGULATORY_CLASSES;
151	*pos++ = 2; /* len */
152
153	*pos++ = op_class;
154	*pos++ = op_class; /* give current operating class as alternate too */
155}
156
157static void ieee80211_tdls_add_bss_coex_ie(struct sk_buff *skb)
158{
159	u8 *pos = (void *)skb_put(skb, 3);
160
161	*pos++ = WLAN_EID_BSS_COEX_2040;
162	*pos++ = 1; /* len */
163
164	*pos++ = WLAN_BSS_COEX_INFORMATION_REQUEST;
165}
166
167static u16 ieee80211_get_tdls_sta_capab(struct ieee80211_sub_if_data *sdata,
168					u16 status_code)
169{
170	struct ieee80211_local *local = sdata->local;
171	u16 capab;
172
173	/* The capability will be 0 when sending a failure code */
174	if (status_code != 0)
175		return 0;
176
177	capab = 0;
178	if (ieee80211_get_sdata_band(sdata) != IEEE80211_BAND_2GHZ)
179		return capab;
180
181	if (!(local->hw.flags & IEEE80211_HW_2GHZ_SHORT_SLOT_INCAPABLE))
182		capab |= WLAN_CAPABILITY_SHORT_SLOT_TIME;
183	if (!(local->hw.flags & IEEE80211_HW_2GHZ_SHORT_PREAMBLE_INCAPABLE))
184		capab |= WLAN_CAPABILITY_SHORT_PREAMBLE;
185
186	return capab;
187}
188
189static void ieee80211_tdls_add_link_ie(struct ieee80211_sub_if_data *sdata,
190				       struct sk_buff *skb, const u8 *peer,
191				       bool initiator)
192{
193	struct ieee80211_tdls_lnkie *lnkid;
194	const u8 *init_addr, *rsp_addr;
195
196	if (initiator) {
197		init_addr = sdata->vif.addr;
198		rsp_addr = peer;
199	} else {
200		init_addr = peer;
201		rsp_addr = sdata->vif.addr;
202	}
203
204	lnkid = (void *)skb_put(skb, sizeof(struct ieee80211_tdls_lnkie));
205
206	lnkid->ie_type = WLAN_EID_LINK_ID;
207	lnkid->ie_len = sizeof(struct ieee80211_tdls_lnkie) - 2;
208
209	memcpy(lnkid->bssid, sdata->u.mgd.bssid, ETH_ALEN);
210	memcpy(lnkid->init_sta, init_addr, ETH_ALEN);
211	memcpy(lnkid->resp_sta, rsp_addr, ETH_ALEN);
212}
213
214static void
215ieee80211_tdls_add_aid(struct ieee80211_sub_if_data *sdata, struct sk_buff *skb)
216{
217	struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
218	u8 *pos = (void *)skb_put(skb, 4);
219
220	*pos++ = WLAN_EID_AID;
221	*pos++ = 2; /* len */
222	put_unaligned_le16(ifmgd->aid, pos);
223}
224
225/* translate numbering in the WMM parameter IE to the mac80211 notation */
226static enum ieee80211_ac_numbers ieee80211_ac_from_wmm(int ac)
227{
228	switch (ac) {
229	default:
230		WARN_ON_ONCE(1);
231	case 0:
232		return IEEE80211_AC_BE;
233	case 1:
234		return IEEE80211_AC_BK;
235	case 2:
236		return IEEE80211_AC_VI;
237	case 3:
238		return IEEE80211_AC_VO;
239	}
240}
241
242static u8 ieee80211_wmm_aci_aifsn(int aifsn, bool acm, int aci)
243{
244	u8 ret;
245
246	ret = aifsn & 0x0f;
247	if (acm)
248		ret |= 0x10;
249	ret |= (aci << 5) & 0x60;
250	return ret;
251}
252
253static u8 ieee80211_wmm_ecw(u16 cw_min, u16 cw_max)
254{
255	return ((ilog2(cw_min + 1) << 0x0) & 0x0f) |
256	       ((ilog2(cw_max + 1) << 0x4) & 0xf0);
257}
258
259static void ieee80211_tdls_add_wmm_param_ie(struct ieee80211_sub_if_data *sdata,
260					    struct sk_buff *skb)
261{
262	struct ieee80211_wmm_param_ie *wmm;
263	struct ieee80211_tx_queue_params *txq;
264	int i;
265
266	wmm = (void *)skb_put(skb, sizeof(*wmm));
267	memset(wmm, 0, sizeof(*wmm));
268
269	wmm->element_id = WLAN_EID_VENDOR_SPECIFIC;
270	wmm->len = sizeof(*wmm) - 2;
271
272	wmm->oui[0] = 0x00; /* Microsoft OUI 00:50:F2 */
273	wmm->oui[1] = 0x50;
274	wmm->oui[2] = 0xf2;
275	wmm->oui_type = 2; /* WME */
276	wmm->oui_subtype = 1; /* WME param */
277	wmm->version = 1; /* WME ver */
278	wmm->qos_info = 0; /* U-APSD not in use */
279
280	/*
281	 * Use the EDCA parameters defined for the BSS, or default if the AP
282	 * doesn't support it, as mandated by 802.11-2012 section 10.22.4
283	 */
284	for (i = 0; i < IEEE80211_NUM_ACS; i++) {
285		txq = &sdata->tx_conf[ieee80211_ac_from_wmm(i)];
286		wmm->ac[i].aci_aifsn = ieee80211_wmm_aci_aifsn(txq->aifs,
287							       txq->acm, i);
288		wmm->ac[i].cw = ieee80211_wmm_ecw(txq->cw_min, txq->cw_max);
289		wmm->ac[i].txop_limit = cpu_to_le16(txq->txop);
290	}
291}
292
293static void
294ieee80211_tdls_add_setup_start_ies(struct ieee80211_sub_if_data *sdata,
295				   struct sk_buff *skb, const u8 *peer,
296				   u8 action_code, bool initiator,
297				   const u8 *extra_ies, size_t extra_ies_len)
298{
299	enum ieee80211_band band = ieee80211_get_sdata_band(sdata);
300	struct ieee80211_local *local = sdata->local;
301	struct ieee80211_supported_band *sband;
302	struct ieee80211_sta_ht_cap ht_cap;
303	struct ieee80211_sta_vht_cap vht_cap;
304	struct sta_info *sta = NULL;
305	size_t offset = 0, noffset;
306	u8 *pos;
307
308	ieee80211_add_srates_ie(sdata, skb, false, band);
309	ieee80211_add_ext_srates_ie(sdata, skb, false, band);
310	ieee80211_tdls_add_supp_channels(sdata, skb);
311
312	/* add any custom IEs that go before Extended Capabilities */
313	if (extra_ies_len) {
314		static const u8 before_ext_cap[] = {
315			WLAN_EID_SUPP_RATES,
316			WLAN_EID_COUNTRY,
317			WLAN_EID_EXT_SUPP_RATES,
318			WLAN_EID_SUPPORTED_CHANNELS,
319			WLAN_EID_RSN,
320		};
321		noffset = ieee80211_ie_split(extra_ies, extra_ies_len,
322					     before_ext_cap,
323					     ARRAY_SIZE(before_ext_cap),
324					     offset);
325		pos = skb_put(skb, noffset - offset);
326		memcpy(pos, extra_ies + offset, noffset - offset);
327		offset = noffset;
328	}
329
330	ieee80211_tdls_add_ext_capab(local, skb);
331
332	/* add the QoS element if we support it */
333	if (local->hw.queues >= IEEE80211_NUM_ACS &&
334	    action_code != WLAN_PUB_ACTION_TDLS_DISCOVER_RES)
335		ieee80211_add_wmm_info_ie(skb_put(skb, 9), 0); /* no U-APSD */
336
337	/* add any custom IEs that go before HT capabilities */
338	if (extra_ies_len) {
339		static const u8 before_ht_cap[] = {
340			WLAN_EID_SUPP_RATES,
341			WLAN_EID_COUNTRY,
342			WLAN_EID_EXT_SUPP_RATES,
343			WLAN_EID_SUPPORTED_CHANNELS,
344			WLAN_EID_RSN,
345			WLAN_EID_EXT_CAPABILITY,
346			WLAN_EID_QOS_CAPA,
347			WLAN_EID_FAST_BSS_TRANSITION,
348			WLAN_EID_TIMEOUT_INTERVAL,
349			WLAN_EID_SUPPORTED_REGULATORY_CLASSES,
350		};
351		noffset = ieee80211_ie_split(extra_ies, extra_ies_len,
352					     before_ht_cap,
353					     ARRAY_SIZE(before_ht_cap),
354					     offset);
355		pos = skb_put(skb, noffset - offset);
356		memcpy(pos, extra_ies + offset, noffset - offset);
357		offset = noffset;
358	}
359
360	rcu_read_lock();
361
362	/* we should have the peer STA if we're already responding */
363	if (action_code == WLAN_TDLS_SETUP_RESPONSE) {
364		sta = sta_info_get(sdata, peer);
365		if (WARN_ON_ONCE(!sta)) {
366			rcu_read_unlock();
367			return;
368		}
369	}
370
371	ieee80211_tdls_add_oper_classes(sdata, skb);
372
373	/*
374	 * with TDLS we can switch channels, and HT-caps are not necessarily
375	 * the same on all bands. The specification limits the setup to a
376	 * single HT-cap, so use the current band for now.
377	 */
378	sband = local->hw.wiphy->bands[band];
379	memcpy(&ht_cap, &sband->ht_cap, sizeof(ht_cap));
380
381	if ((action_code == WLAN_TDLS_SETUP_REQUEST ||
382	     action_code == WLAN_PUB_ACTION_TDLS_DISCOVER_RES) &&
383	    ht_cap.ht_supported) {
384		ieee80211_apply_htcap_overrides(sdata, &ht_cap);
385
386		/* disable SMPS in TDLS initiator */
387		ht_cap.cap |= WLAN_HT_CAP_SM_PS_DISABLED
388				<< IEEE80211_HT_CAP_SM_PS_SHIFT;
389
390		pos = skb_put(skb, sizeof(struct ieee80211_ht_cap) + 2);
391		ieee80211_ie_build_ht_cap(pos, &ht_cap, ht_cap.cap);
392	} else if (action_code == WLAN_TDLS_SETUP_RESPONSE &&
393		   ht_cap.ht_supported && sta->sta.ht_cap.ht_supported) {
394		/* disable SMPS in TDLS responder */
395		sta->sta.ht_cap.cap |= WLAN_HT_CAP_SM_PS_DISABLED
396					<< IEEE80211_HT_CAP_SM_PS_SHIFT;
397
398		/* the peer caps are already intersected with our own */
399		memcpy(&ht_cap, &sta->sta.ht_cap, sizeof(ht_cap));
400
401		pos = skb_put(skb, sizeof(struct ieee80211_ht_cap) + 2);
402		ieee80211_ie_build_ht_cap(pos, &ht_cap, ht_cap.cap);
403	}
404
405	if (ht_cap.ht_supported &&
406	    (ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40))
407		ieee80211_tdls_add_bss_coex_ie(skb);
408
409	ieee80211_tdls_add_link_ie(sdata, skb, peer, initiator);
410
411	/* add any custom IEs that go before VHT capabilities */
412	if (extra_ies_len) {
413		static const u8 before_vht_cap[] = {
414			WLAN_EID_SUPP_RATES,
415			WLAN_EID_COUNTRY,
416			WLAN_EID_EXT_SUPP_RATES,
417			WLAN_EID_SUPPORTED_CHANNELS,
418			WLAN_EID_RSN,
419			WLAN_EID_EXT_CAPABILITY,
420			WLAN_EID_QOS_CAPA,
421			WLAN_EID_FAST_BSS_TRANSITION,
422			WLAN_EID_TIMEOUT_INTERVAL,
423			WLAN_EID_SUPPORTED_REGULATORY_CLASSES,
424			WLAN_EID_MULTI_BAND,
425		};
426		noffset = ieee80211_ie_split(extra_ies, extra_ies_len,
427					     before_vht_cap,
428					     ARRAY_SIZE(before_vht_cap),
429					     offset);
430		pos = skb_put(skb, noffset - offset);
431		memcpy(pos, extra_ies + offset, noffset - offset);
432		offset = noffset;
433	}
434
435	/* build the VHT-cap similarly to the HT-cap */
436	memcpy(&vht_cap, &sband->vht_cap, sizeof(vht_cap));
437	if ((action_code == WLAN_TDLS_SETUP_REQUEST ||
438	     action_code == WLAN_PUB_ACTION_TDLS_DISCOVER_RES) &&
439	    vht_cap.vht_supported) {
440		ieee80211_apply_vhtcap_overrides(sdata, &vht_cap);
441
442		/* the AID is present only when VHT is implemented */
443		if (action_code == WLAN_TDLS_SETUP_REQUEST)
444			ieee80211_tdls_add_aid(sdata, skb);
445
446		pos = skb_put(skb, sizeof(struct ieee80211_vht_cap) + 2);
447		ieee80211_ie_build_vht_cap(pos, &vht_cap, vht_cap.cap);
448	} else if (action_code == WLAN_TDLS_SETUP_RESPONSE &&
449		   vht_cap.vht_supported && sta->sta.vht_cap.vht_supported) {
450		/* the peer caps are already intersected with our own */
451		memcpy(&vht_cap, &sta->sta.vht_cap, sizeof(vht_cap));
452
453		/* the AID is present only when VHT is implemented */
454		ieee80211_tdls_add_aid(sdata, skb);
455
456		pos = skb_put(skb, sizeof(struct ieee80211_vht_cap) + 2);
457		ieee80211_ie_build_vht_cap(pos, &vht_cap, vht_cap.cap);
458	}
459
460	rcu_read_unlock();
461
462	/* add any remaining IEs */
463	if (extra_ies_len) {
464		noffset = extra_ies_len;
465		pos = skb_put(skb, noffset - offset);
466		memcpy(pos, extra_ies + offset, noffset - offset);
467	}
468
469}
470
471static void
472ieee80211_tdls_add_setup_cfm_ies(struct ieee80211_sub_if_data *sdata,
473				 struct sk_buff *skb, const u8 *peer,
474				 bool initiator, const u8 *extra_ies,
475				 size_t extra_ies_len)
476{
477	struct ieee80211_local *local = sdata->local;
478	struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
479	size_t offset = 0, noffset;
480	struct sta_info *sta, *ap_sta;
481	enum ieee80211_band band = ieee80211_get_sdata_band(sdata);
482	u8 *pos;
483
484	rcu_read_lock();
485
486	sta = sta_info_get(sdata, peer);
487	ap_sta = sta_info_get(sdata, ifmgd->bssid);
488	if (WARN_ON_ONCE(!sta || !ap_sta)) {
489		rcu_read_unlock();
490		return;
491	}
492
493	/* add any custom IEs that go before the QoS IE */
494	if (extra_ies_len) {
495		static const u8 before_qos[] = {
496			WLAN_EID_RSN,
497		};
498		noffset = ieee80211_ie_split(extra_ies, extra_ies_len,
499					     before_qos,
500					     ARRAY_SIZE(before_qos),
501					     offset);
502		pos = skb_put(skb, noffset - offset);
503		memcpy(pos, extra_ies + offset, noffset - offset);
504		offset = noffset;
505	}
506
507	/* add the QoS param IE if both the peer and we support it */
508	if (local->hw.queues >= IEEE80211_NUM_ACS && sta->sta.wme)
509		ieee80211_tdls_add_wmm_param_ie(sdata, skb);
510
511	/* add any custom IEs that go before HT operation */
512	if (extra_ies_len) {
513		static const u8 before_ht_op[] = {
514			WLAN_EID_RSN,
515			WLAN_EID_QOS_CAPA,
516			WLAN_EID_FAST_BSS_TRANSITION,
517			WLAN_EID_TIMEOUT_INTERVAL,
518		};
519		noffset = ieee80211_ie_split(extra_ies, extra_ies_len,
520					     before_ht_op,
521					     ARRAY_SIZE(before_ht_op),
522					     offset);
523		pos = skb_put(skb, noffset - offset);
524		memcpy(pos, extra_ies + offset, noffset - offset);
525		offset = noffset;
526	}
527
528	/* if HT support is only added in TDLS, we need an HT-operation IE */
529	if (!ap_sta->sta.ht_cap.ht_supported && sta->sta.ht_cap.ht_supported) {
530		struct ieee80211_chanctx_conf *chanctx_conf =
531				rcu_dereference(sdata->vif.chanctx_conf);
532		if (!WARN_ON(!chanctx_conf)) {
533			pos = skb_put(skb, 2 +
534				      sizeof(struct ieee80211_ht_operation));
535			/* send an empty HT operation IE */
536			ieee80211_ie_build_ht_oper(pos, &sta->sta.ht_cap,
537						   &chanctx_conf->def, 0);
538		}
539	}
540
541	ieee80211_tdls_add_link_ie(sdata, skb, peer, initiator);
542
543	/* only include VHT-operation if not on the 2.4GHz band */
544	if (band != IEEE80211_BAND_2GHZ && !ap_sta->sta.vht_cap.vht_supported &&
545	    sta->sta.vht_cap.vht_supported) {
546		struct ieee80211_chanctx_conf *chanctx_conf =
547				rcu_dereference(sdata->vif.chanctx_conf);
548		if (!WARN_ON(!chanctx_conf)) {
549			pos = skb_put(skb, 2 +
550				      sizeof(struct ieee80211_vht_operation));
551			ieee80211_ie_build_vht_oper(pos, &sta->sta.vht_cap,
552						    &chanctx_conf->def);
553		}
554	}
555
556	rcu_read_unlock();
557
558	/* add any remaining IEs */
559	if (extra_ies_len) {
560		noffset = extra_ies_len;
561		pos = skb_put(skb, noffset - offset);
562		memcpy(pos, extra_ies + offset, noffset - offset);
563	}
564}
565
566static void
567ieee80211_tdls_add_chan_switch_req_ies(struct ieee80211_sub_if_data *sdata,
568				       struct sk_buff *skb, const u8 *peer,
569				       bool initiator, const u8 *extra_ies,
570				       size_t extra_ies_len, u8 oper_class,
571				       struct cfg80211_chan_def *chandef)
572{
573	struct ieee80211_tdls_data *tf;
574	size_t offset = 0, noffset;
575	u8 *pos;
576
577	if (WARN_ON_ONCE(!chandef))
578		return;
579
580	tf = (void *)skb->data;
581	tf->u.chan_switch_req.target_channel =
582		ieee80211_frequency_to_channel(chandef->chan->center_freq);
583	tf->u.chan_switch_req.oper_class = oper_class;
584
585	if (extra_ies_len) {
586		static const u8 before_lnkie[] = {
587			WLAN_EID_SECONDARY_CHANNEL_OFFSET,
588		};
589		noffset = ieee80211_ie_split(extra_ies, extra_ies_len,
590					     before_lnkie,
591					     ARRAY_SIZE(before_lnkie),
592					     offset);
593		pos = skb_put(skb, noffset - offset);
594		memcpy(pos, extra_ies + offset, noffset - offset);
595		offset = noffset;
596	}
597
598	ieee80211_tdls_add_link_ie(sdata, skb, peer, initiator);
599
600	/* add any remaining IEs */
601	if (extra_ies_len) {
602		noffset = extra_ies_len;
603		pos = skb_put(skb, noffset - offset);
604		memcpy(pos, extra_ies + offset, noffset - offset);
605	}
606}
607
608static void
609ieee80211_tdls_add_chan_switch_resp_ies(struct ieee80211_sub_if_data *sdata,
610					struct sk_buff *skb, const u8 *peer,
611					u16 status_code, bool initiator,
612					const u8 *extra_ies,
613					size_t extra_ies_len)
614{
615	if (status_code == 0)
616		ieee80211_tdls_add_link_ie(sdata, skb, peer, initiator);
617
618	if (extra_ies_len)
619		memcpy(skb_put(skb, extra_ies_len), extra_ies, extra_ies_len);
620}
621
622static void ieee80211_tdls_add_ies(struct ieee80211_sub_if_data *sdata,
623				   struct sk_buff *skb, const u8 *peer,
624				   u8 action_code, u16 status_code,
625				   bool initiator, const u8 *extra_ies,
626				   size_t extra_ies_len, u8 oper_class,
627				   struct cfg80211_chan_def *chandef)
628{
629	switch (action_code) {
630	case WLAN_TDLS_SETUP_REQUEST:
631	case WLAN_TDLS_SETUP_RESPONSE:
632	case WLAN_PUB_ACTION_TDLS_DISCOVER_RES:
633		if (status_code == 0)
634			ieee80211_tdls_add_setup_start_ies(sdata, skb, peer,
635							   action_code,
636							   initiator,
637							   extra_ies,
638							   extra_ies_len);
639		break;
640	case WLAN_TDLS_SETUP_CONFIRM:
641		if (status_code == 0)
642			ieee80211_tdls_add_setup_cfm_ies(sdata, skb, peer,
643							 initiator, extra_ies,
644							 extra_ies_len);
645		break;
646	case WLAN_TDLS_TEARDOWN:
647	case WLAN_TDLS_DISCOVERY_REQUEST:
648		if (extra_ies_len)
649			memcpy(skb_put(skb, extra_ies_len), extra_ies,
650			       extra_ies_len);
651		if (status_code == 0 || action_code == WLAN_TDLS_TEARDOWN)
652			ieee80211_tdls_add_link_ie(sdata, skb, peer, initiator);
653		break;
654	case WLAN_TDLS_CHANNEL_SWITCH_REQUEST:
655		ieee80211_tdls_add_chan_switch_req_ies(sdata, skb, peer,
656						       initiator, extra_ies,
657						       extra_ies_len,
658						       oper_class, chandef);
659		break;
660	case WLAN_TDLS_CHANNEL_SWITCH_RESPONSE:
661		ieee80211_tdls_add_chan_switch_resp_ies(sdata, skb, peer,
662							status_code,
663							initiator, extra_ies,
664							extra_ies_len);
665		break;
666	}
667
668}
669
670static int
671ieee80211_prep_tdls_encap_data(struct wiphy *wiphy, struct net_device *dev,
672			       const u8 *peer, u8 action_code, u8 dialog_token,
673			       u16 status_code, struct sk_buff *skb)
674{
675	struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
676	struct ieee80211_tdls_data *tf;
677
678	tf = (void *)skb_put(skb, offsetof(struct ieee80211_tdls_data, u));
679
680	memcpy(tf->da, peer, ETH_ALEN);
681	memcpy(tf->sa, sdata->vif.addr, ETH_ALEN);
682	tf->ether_type = cpu_to_be16(ETH_P_TDLS);
683	tf->payload_type = WLAN_TDLS_SNAP_RFTYPE;
684
685	/* network header is after the ethernet header */
686	skb_set_network_header(skb, ETH_HLEN);
687
688	switch (action_code) {
689	case WLAN_TDLS_SETUP_REQUEST:
690		tf->category = WLAN_CATEGORY_TDLS;
691		tf->action_code = WLAN_TDLS_SETUP_REQUEST;
692
693		skb_put(skb, sizeof(tf->u.setup_req));
694		tf->u.setup_req.dialog_token = dialog_token;
695		tf->u.setup_req.capability =
696			cpu_to_le16(ieee80211_get_tdls_sta_capab(sdata,
697								 status_code));
698		break;
699	case WLAN_TDLS_SETUP_RESPONSE:
700		tf->category = WLAN_CATEGORY_TDLS;
701		tf->action_code = WLAN_TDLS_SETUP_RESPONSE;
702
703		skb_put(skb, sizeof(tf->u.setup_resp));
704		tf->u.setup_resp.status_code = cpu_to_le16(status_code);
705		tf->u.setup_resp.dialog_token = dialog_token;
706		tf->u.setup_resp.capability =
707			cpu_to_le16(ieee80211_get_tdls_sta_capab(sdata,
708								 status_code));
709		break;
710	case WLAN_TDLS_SETUP_CONFIRM:
711		tf->category = WLAN_CATEGORY_TDLS;
712		tf->action_code = WLAN_TDLS_SETUP_CONFIRM;
713
714		skb_put(skb, sizeof(tf->u.setup_cfm));
715		tf->u.setup_cfm.status_code = cpu_to_le16(status_code);
716		tf->u.setup_cfm.dialog_token = dialog_token;
717		break;
718	case WLAN_TDLS_TEARDOWN:
719		tf->category = WLAN_CATEGORY_TDLS;
720		tf->action_code = WLAN_TDLS_TEARDOWN;
721
722		skb_put(skb, sizeof(tf->u.teardown));
723		tf->u.teardown.reason_code = cpu_to_le16(status_code);
724		break;
725	case WLAN_TDLS_DISCOVERY_REQUEST:
726		tf->category = WLAN_CATEGORY_TDLS;
727		tf->action_code = WLAN_TDLS_DISCOVERY_REQUEST;
728
729		skb_put(skb, sizeof(tf->u.discover_req));
730		tf->u.discover_req.dialog_token = dialog_token;
731		break;
732	case WLAN_TDLS_CHANNEL_SWITCH_REQUEST:
733		tf->category = WLAN_CATEGORY_TDLS;
734		tf->action_code = WLAN_TDLS_CHANNEL_SWITCH_REQUEST;
735
736		skb_put(skb, sizeof(tf->u.chan_switch_req));
737		break;
738	case WLAN_TDLS_CHANNEL_SWITCH_RESPONSE:
739		tf->category = WLAN_CATEGORY_TDLS;
740		tf->action_code = WLAN_TDLS_CHANNEL_SWITCH_RESPONSE;
741
742		skb_put(skb, sizeof(tf->u.chan_switch_resp));
743		tf->u.chan_switch_resp.status_code = cpu_to_le16(status_code);
744		break;
745	default:
746		return -EINVAL;
747	}
748
749	return 0;
750}
751
752static int
753ieee80211_prep_tdls_direct(struct wiphy *wiphy, struct net_device *dev,
754			   const u8 *peer, u8 action_code, u8 dialog_token,
755			   u16 status_code, struct sk_buff *skb)
756{
757	struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
758	struct ieee80211_mgmt *mgmt;
759
760	mgmt = (void *)skb_put(skb, 24);
761	memset(mgmt, 0, 24);
762	memcpy(mgmt->da, peer, ETH_ALEN);
763	memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
764	memcpy(mgmt->bssid, sdata->u.mgd.bssid, ETH_ALEN);
765
766	mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
767					  IEEE80211_STYPE_ACTION);
768
769	switch (action_code) {
770	case WLAN_PUB_ACTION_TDLS_DISCOVER_RES:
771		skb_put(skb, 1 + sizeof(mgmt->u.action.u.tdls_discover_resp));
772		mgmt->u.action.category = WLAN_CATEGORY_PUBLIC;
773		mgmt->u.action.u.tdls_discover_resp.action_code =
774			WLAN_PUB_ACTION_TDLS_DISCOVER_RES;
775		mgmt->u.action.u.tdls_discover_resp.dialog_token =
776			dialog_token;
777		mgmt->u.action.u.tdls_discover_resp.capability =
778			cpu_to_le16(ieee80211_get_tdls_sta_capab(sdata,
779								 status_code));
780		break;
781	default:
782		return -EINVAL;
783	}
784
785	return 0;
786}
787
788static struct sk_buff *
789ieee80211_tdls_build_mgmt_packet_data(struct ieee80211_sub_if_data *sdata,
790				      const u8 *peer, u8 action_code,
791				      u8 dialog_token, u16 status_code,
792				      bool initiator, const u8 *extra_ies,
793				      size_t extra_ies_len, u8 oper_class,
794				      struct cfg80211_chan_def *chandef)
795{
796	struct ieee80211_local *local = sdata->local;
797	struct sk_buff *skb;
798	int ret;
799
800	skb = netdev_alloc_skb(sdata->dev,
801			       local->hw.extra_tx_headroom +
802			       max(sizeof(struct ieee80211_mgmt),
803				   sizeof(struct ieee80211_tdls_data)) +
804			       50 + /* supported rates */
805			       7 + /* ext capab */
806			       26 + /* max(WMM-info, WMM-param) */
807			       2 + max(sizeof(struct ieee80211_ht_cap),
808				       sizeof(struct ieee80211_ht_operation)) +
809			       2 + max(sizeof(struct ieee80211_vht_cap),
810				       sizeof(struct ieee80211_vht_operation)) +
811			       50 + /* supported channels */
812			       3 + /* 40/20 BSS coex */
813			       4 + /* AID */
814			       4 + /* oper classes */
815			       extra_ies_len +
816			       sizeof(struct ieee80211_tdls_lnkie));
817	if (!skb)
818		return NULL;
819
820	skb_reserve(skb, local->hw.extra_tx_headroom);
821
822	switch (action_code) {
823	case WLAN_TDLS_SETUP_REQUEST:
824	case WLAN_TDLS_SETUP_RESPONSE:
825	case WLAN_TDLS_SETUP_CONFIRM:
826	case WLAN_TDLS_TEARDOWN:
827	case WLAN_TDLS_DISCOVERY_REQUEST:
828	case WLAN_TDLS_CHANNEL_SWITCH_REQUEST:
829	case WLAN_TDLS_CHANNEL_SWITCH_RESPONSE:
830		ret = ieee80211_prep_tdls_encap_data(local->hw.wiphy,
831						     sdata->dev, peer,
832						     action_code, dialog_token,
833						     status_code, skb);
834		break;
835	case WLAN_PUB_ACTION_TDLS_DISCOVER_RES:
836		ret = ieee80211_prep_tdls_direct(local->hw.wiphy, sdata->dev,
837						 peer, action_code,
838						 dialog_token, status_code,
839						 skb);
840		break;
841	default:
842		ret = -ENOTSUPP;
843		break;
844	}
845
846	if (ret < 0)
847		goto fail;
848
849	ieee80211_tdls_add_ies(sdata, skb, peer, action_code, status_code,
850			       initiator, extra_ies, extra_ies_len, oper_class,
851			       chandef);
852	return skb;
853
854fail:
855	dev_kfree_skb(skb);
856	return NULL;
857}
858
859static int
860ieee80211_tdls_prep_mgmt_packet(struct wiphy *wiphy, struct net_device *dev,
861				const u8 *peer, u8 action_code, u8 dialog_token,
862				u16 status_code, u32 peer_capability,
863				bool initiator, const u8 *extra_ies,
864				size_t extra_ies_len, u8 oper_class,
865				struct cfg80211_chan_def *chandef)
866{
867	struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
868	struct sk_buff *skb = NULL;
869	struct sta_info *sta;
870	u32 flags = 0;
871	int ret = 0;
872
873	rcu_read_lock();
874	sta = sta_info_get(sdata, peer);
875
876	/* infer the initiator if we can, to support old userspace */
877	switch (action_code) {
878	case WLAN_TDLS_SETUP_REQUEST:
879		if (sta) {
880			set_sta_flag(sta, WLAN_STA_TDLS_INITIATOR);
881			sta->sta.tdls_initiator = false;
882		}
883		/* fall-through */
884	case WLAN_TDLS_SETUP_CONFIRM:
885	case WLAN_TDLS_DISCOVERY_REQUEST:
886		initiator = true;
887		break;
888	case WLAN_TDLS_SETUP_RESPONSE:
889		/*
890		 * In some testing scenarios, we send a request and response.
891		 * Make the last packet sent take effect for the initiator
892		 * value.
893		 */
894		if (sta) {
895			clear_sta_flag(sta, WLAN_STA_TDLS_INITIATOR);
896			sta->sta.tdls_initiator = true;
897		}
898		/* fall-through */
899	case WLAN_PUB_ACTION_TDLS_DISCOVER_RES:
900		initiator = false;
901		break;
902	case WLAN_TDLS_TEARDOWN:
903	case WLAN_TDLS_CHANNEL_SWITCH_REQUEST:
904	case WLAN_TDLS_CHANNEL_SWITCH_RESPONSE:
905		/* any value is ok */
906		break;
907	default:
908		ret = -ENOTSUPP;
909		break;
910	}
911
912	if (sta && test_sta_flag(sta, WLAN_STA_TDLS_INITIATOR))
913		initiator = true;
914
915	rcu_read_unlock();
916	if (ret < 0)
917		goto fail;
918
919	skb = ieee80211_tdls_build_mgmt_packet_data(sdata, peer, action_code,
920						    dialog_token, status_code,
921						    initiator, extra_ies,
922						    extra_ies_len, oper_class,
923						    chandef);
924	if (!skb) {
925		ret = -EINVAL;
926		goto fail;
927	}
928
929	if (action_code == WLAN_PUB_ACTION_TDLS_DISCOVER_RES) {
930		ieee80211_tx_skb(sdata, skb);
931		return 0;
932	}
933
934	/*
935	 * According to 802.11z: Setup req/resp are sent in AC_BK, otherwise
936	 * we should default to AC_VI.
937	 */
938	switch (action_code) {
939	case WLAN_TDLS_SETUP_REQUEST:
940	case WLAN_TDLS_SETUP_RESPONSE:
941		skb_set_queue_mapping(skb, IEEE80211_AC_BK);
942		skb->priority = 2;
943		break;
944	default:
945		skb_set_queue_mapping(skb, IEEE80211_AC_VI);
946		skb->priority = 5;
947		break;
948	}
949
950	/*
951	 * Set the WLAN_TDLS_TEARDOWN flag to indicate a teardown in progress.
952	 * Later, if no ACK is returned from peer, we will re-send the teardown
953	 * packet through the AP.
954	 */
955	if ((action_code == WLAN_TDLS_TEARDOWN) &&
956	    (sdata->local->hw.flags & IEEE80211_HW_REPORTS_TX_ACK_STATUS)) {
957		bool try_resend; /* Should we keep skb for possible resend */
958
959		/* If not sending directly to peer - no point in keeping skb */
960		rcu_read_lock();
961		sta = sta_info_get(sdata, peer);
962		try_resend = sta && test_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH);
963		rcu_read_unlock();
964
965		spin_lock_bh(&sdata->u.mgd.teardown_lock);
966		if (try_resend && !sdata->u.mgd.teardown_skb) {
967			/* Mark it as requiring TX status callback  */
968			flags |= IEEE80211_TX_CTL_REQ_TX_STATUS |
969				 IEEE80211_TX_INTFL_MLME_CONN_TX;
970
971			/*
972			 * skb is copied since mac80211 will later set
973			 * properties that might not be the same as the AP,
974			 * such as encryption, QoS, addresses, etc.
975			 *
976			 * No problem if skb_copy() fails, so no need to check.
977			 */
978			sdata->u.mgd.teardown_skb = skb_copy(skb, GFP_ATOMIC);
979			sdata->u.mgd.orig_teardown_skb = skb;
980		}
981		spin_unlock_bh(&sdata->u.mgd.teardown_lock);
982	}
983
984	/* disable bottom halves when entering the Tx path */
985	local_bh_disable();
986	__ieee80211_subif_start_xmit(skb, dev, flags);
987	local_bh_enable();
988
989	return ret;
990
991fail:
992	dev_kfree_skb(skb);
993	return ret;
994}
995
996static int
997ieee80211_tdls_mgmt_setup(struct wiphy *wiphy, struct net_device *dev,
998			  const u8 *peer, u8 action_code, u8 dialog_token,
999			  u16 status_code, u32 peer_capability, bool initiator,
1000			  const u8 *extra_ies, size_t extra_ies_len)
1001{
1002	struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1003	struct ieee80211_local *local = sdata->local;
1004	int ret;
1005
1006	mutex_lock(&local->mtx);
1007
1008	/* we don't support concurrent TDLS peer setups */
1009	if (!is_zero_ether_addr(sdata->u.mgd.tdls_peer) &&
1010	    !ether_addr_equal(sdata->u.mgd.tdls_peer, peer)) {
1011		ret = -EBUSY;
1012		goto out_unlock;
1013	}
1014
1015	/*
1016	 * make sure we have a STA representing the peer so we drop or buffer
1017	 * non-TDLS-setup frames to the peer. We can't send other packets
1018	 * during setup through the AP path.
1019	 * Allow error packets to be sent - sometimes we don't even add a STA
1020	 * before failing the setup.
1021	 */
1022	if (status_code == 0) {
1023		rcu_read_lock();
1024		if (!sta_info_get(sdata, peer)) {
1025			rcu_read_unlock();
1026			ret = -ENOLINK;
1027			goto out_unlock;
1028		}
1029		rcu_read_unlock();
1030	}
1031
1032	ieee80211_flush_queues(local, sdata, false);
1033	memcpy(sdata->u.mgd.tdls_peer, peer, ETH_ALEN);
1034	mutex_unlock(&local->mtx);
1035
1036	/* we cannot take the mutex while preparing the setup packet */
1037	ret = ieee80211_tdls_prep_mgmt_packet(wiphy, dev, peer, action_code,
1038					      dialog_token, status_code,
1039					      peer_capability, initiator,
1040					      extra_ies, extra_ies_len, 0,
1041					      NULL);
1042	if (ret < 0) {
1043		mutex_lock(&local->mtx);
1044		eth_zero_addr(sdata->u.mgd.tdls_peer);
1045		mutex_unlock(&local->mtx);
1046		return ret;
1047	}
1048
1049	ieee80211_queue_delayed_work(&sdata->local->hw,
1050				     &sdata->u.mgd.tdls_peer_del_work,
1051				     TDLS_PEER_SETUP_TIMEOUT);
1052	return 0;
1053
1054out_unlock:
1055	mutex_unlock(&local->mtx);
1056	return ret;
1057}
1058
1059static int
1060ieee80211_tdls_mgmt_teardown(struct wiphy *wiphy, struct net_device *dev,
1061			     const u8 *peer, u8 action_code, u8 dialog_token,
1062			     u16 status_code, u32 peer_capability,
1063			     bool initiator, const u8 *extra_ies,
1064			     size_t extra_ies_len)
1065{
1066	struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1067	struct ieee80211_local *local = sdata->local;
1068	struct sta_info *sta;
1069	int ret;
1070
1071	/*
1072	 * No packets can be transmitted to the peer via the AP during setup -
1073	 * the STA is set as a TDLS peer, but is not authorized.
1074	 * During teardown, we prevent direct transmissions by stopping the
1075	 * queues and flushing all direct packets.
1076	 */
1077	ieee80211_stop_vif_queues(local, sdata,
1078				  IEEE80211_QUEUE_STOP_REASON_TDLS_TEARDOWN);
1079	ieee80211_flush_queues(local, sdata, false);
1080
1081	ret = ieee80211_tdls_prep_mgmt_packet(wiphy, dev, peer, action_code,
1082					      dialog_token, status_code,
1083					      peer_capability, initiator,
1084					      extra_ies, extra_ies_len, 0,
1085					      NULL);
1086	if (ret < 0)
1087		sdata_err(sdata, "Failed sending TDLS teardown packet %d\n",
1088			  ret);
1089
1090	/*
1091	 * Remove the STA AUTH flag to force further traffic through the AP. If
1092	 * the STA was unreachable, it was already removed.
1093	 */
1094	rcu_read_lock();
1095	sta = sta_info_get(sdata, peer);
1096	if (sta)
1097		clear_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH);
1098	rcu_read_unlock();
1099
1100	ieee80211_wake_vif_queues(local, sdata,
1101				  IEEE80211_QUEUE_STOP_REASON_TDLS_TEARDOWN);
1102
1103	return 0;
1104}
1105
1106int ieee80211_tdls_mgmt(struct wiphy *wiphy, struct net_device *dev,
1107			const u8 *peer, u8 action_code, u8 dialog_token,
1108			u16 status_code, u32 peer_capability,
1109			bool initiator, const u8 *extra_ies,
1110			size_t extra_ies_len)
1111{
1112	struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1113	int ret;
1114
1115	if (!(wiphy->flags & WIPHY_FLAG_SUPPORTS_TDLS))
1116		return -ENOTSUPP;
1117
1118	/* make sure we are in managed mode, and associated */
1119	if (sdata->vif.type != NL80211_IFTYPE_STATION ||
1120	    !sdata->u.mgd.associated)
1121		return -EINVAL;
1122
1123	switch (action_code) {
1124	case WLAN_TDLS_SETUP_REQUEST:
1125	case WLAN_TDLS_SETUP_RESPONSE:
1126		ret = ieee80211_tdls_mgmt_setup(wiphy, dev, peer, action_code,
1127						dialog_token, status_code,
1128						peer_capability, initiator,
1129						extra_ies, extra_ies_len);
1130		break;
1131	case WLAN_TDLS_TEARDOWN:
1132		ret = ieee80211_tdls_mgmt_teardown(wiphy, dev, peer,
1133						   action_code, dialog_token,
1134						   status_code,
1135						   peer_capability, initiator,
1136						   extra_ies, extra_ies_len);
1137		break;
1138	case WLAN_TDLS_DISCOVERY_REQUEST:
1139		/*
1140		 * Protect the discovery so we can hear the TDLS discovery
1141		 * response frame. It is transmitted directly and not buffered
1142		 * by the AP.
1143		 */
1144		drv_mgd_protect_tdls_discover(sdata->local, sdata);
1145		/* fall-through */
1146	case WLAN_TDLS_SETUP_CONFIRM:
1147	case WLAN_PUB_ACTION_TDLS_DISCOVER_RES:
1148		/* no special handling */
1149		ret = ieee80211_tdls_prep_mgmt_packet(wiphy, dev, peer,
1150						      action_code,
1151						      dialog_token,
1152						      status_code,
1153						      peer_capability,
1154						      initiator, extra_ies,
1155						      extra_ies_len, 0, NULL);
1156		break;
1157	default:
1158		ret = -EOPNOTSUPP;
1159		break;
1160	}
1161
1162	tdls_dbg(sdata, "TDLS mgmt action %d peer %pM status %d\n",
1163		 action_code, peer, ret);
1164	return ret;
1165}
1166
1167int ieee80211_tdls_oper(struct wiphy *wiphy, struct net_device *dev,
1168			const u8 *peer, enum nl80211_tdls_operation oper)
1169{
1170	struct sta_info *sta;
1171	struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1172	struct ieee80211_local *local = sdata->local;
1173	int ret;
1174
1175	if (!(wiphy->flags & WIPHY_FLAG_SUPPORTS_TDLS))
1176		return -ENOTSUPP;
1177
1178	if (sdata->vif.type != NL80211_IFTYPE_STATION)
1179		return -EINVAL;
1180
1181	switch (oper) {
1182	case NL80211_TDLS_ENABLE_LINK:
1183	case NL80211_TDLS_DISABLE_LINK:
1184		break;
1185	case NL80211_TDLS_TEARDOWN:
1186	case NL80211_TDLS_SETUP:
1187	case NL80211_TDLS_DISCOVERY_REQ:
1188		/* We don't support in-driver setup/teardown/discovery */
1189		return -ENOTSUPP;
1190	}
1191
1192	mutex_lock(&local->mtx);
1193	tdls_dbg(sdata, "TDLS oper %d peer %pM\n", oper, peer);
1194
1195	switch (oper) {
1196	case NL80211_TDLS_ENABLE_LINK:
1197		rcu_read_lock();
1198		sta = sta_info_get(sdata, peer);
1199		if (!sta) {
1200			rcu_read_unlock();
1201			ret = -ENOLINK;
1202			break;
1203		}
1204
1205		set_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH);
1206		rcu_read_unlock();
1207
1208		WARN_ON_ONCE(is_zero_ether_addr(sdata->u.mgd.tdls_peer) ||
1209			     !ether_addr_equal(sdata->u.mgd.tdls_peer, peer));
1210		ret = 0;
1211		break;
1212	case NL80211_TDLS_DISABLE_LINK:
1213		/*
1214		 * The teardown message in ieee80211_tdls_mgmt_teardown() was
1215		 * created while the queues were stopped, so it might still be
1216		 * pending. Before flushing the queues we need to be sure the
1217		 * message is handled by the tasklet handling pending messages,
1218		 * otherwise we might start destroying the station before
1219		 * sending the teardown packet.
1220		 * Note that this only forces the tasklet to flush pendings -
1221		 * not to stop the tasklet from rescheduling itself.
1222		 */
1223		tasklet_kill(&local->tx_pending_tasklet);
1224		/* flush a potentially queued teardown packet */
1225		ieee80211_flush_queues(local, sdata, false);
1226
1227		ret = sta_info_destroy_addr(sdata, peer);
1228		break;
1229	default:
1230		ret = -ENOTSUPP;
1231		break;
1232	}
1233
1234	if (ret == 0 && ether_addr_equal(sdata->u.mgd.tdls_peer, peer)) {
1235		cancel_delayed_work(&sdata->u.mgd.tdls_peer_del_work);
1236		eth_zero_addr(sdata->u.mgd.tdls_peer);
1237	}
1238
1239	mutex_unlock(&local->mtx);
1240	return ret;
1241}
1242
1243void ieee80211_tdls_oper_request(struct ieee80211_vif *vif, const u8 *peer,
1244				 enum nl80211_tdls_operation oper,
1245				 u16 reason_code, gfp_t gfp)
1246{
1247	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1248
1249	if (vif->type != NL80211_IFTYPE_STATION || !vif->bss_conf.assoc) {
1250		sdata_err(sdata, "Discarding TDLS oper %d - not STA or disconnected\n",
1251			  oper);
1252		return;
1253	}
1254
1255	cfg80211_tdls_oper_request(sdata->dev, peer, oper, reason_code, gfp);
1256}
1257EXPORT_SYMBOL(ieee80211_tdls_oper_request);
1258
1259static void
1260iee80211_tdls_add_ch_switch_timing(u8 *buf, u16 switch_time, u16 switch_timeout)
1261{
1262	struct ieee80211_ch_switch_timing *ch_sw;
1263
1264	*buf++ = WLAN_EID_CHAN_SWITCH_TIMING;
1265	*buf++ = sizeof(struct ieee80211_ch_switch_timing);
1266
1267	ch_sw = (void *)buf;
1268	ch_sw->switch_time = cpu_to_le16(switch_time);
1269	ch_sw->switch_timeout = cpu_to_le16(switch_timeout);
1270}
1271
1272/* find switch timing IE in SKB ready for Tx */
1273static const u8 *ieee80211_tdls_find_sw_timing_ie(struct sk_buff *skb)
1274{
1275	struct ieee80211_tdls_data *tf;
1276	const u8 *ie_start;
1277
1278	/*
1279	 * Get the offset for the new location of the switch timing IE.
1280	 * The SKB network header will now point to the "payload_type"
1281	 * element of the TDLS data frame struct.
1282	 */
1283	tf = container_of(skb->data + skb_network_offset(skb),
1284			  struct ieee80211_tdls_data, payload_type);
1285	ie_start = tf->u.chan_switch_req.variable;
1286	return cfg80211_find_ie(WLAN_EID_CHAN_SWITCH_TIMING, ie_start,
1287				skb->len - (ie_start - skb->data));
1288}
1289
1290static struct sk_buff *
1291ieee80211_tdls_ch_sw_tmpl_get(struct sta_info *sta, u8 oper_class,
1292			      struct cfg80211_chan_def *chandef,
1293			      u32 *ch_sw_tm_ie_offset)
1294{
1295	struct ieee80211_sub_if_data *sdata = sta->sdata;
1296	u8 extra_ies[2 + sizeof(struct ieee80211_sec_chan_offs_ie) +
1297		     2 + sizeof(struct ieee80211_ch_switch_timing)];
1298	int extra_ies_len = 2 + sizeof(struct ieee80211_ch_switch_timing);
1299	u8 *pos = extra_ies;
1300	struct sk_buff *skb;
1301
1302	/*
1303	 * if chandef points to a wide channel add a Secondary-Channel
1304	 * Offset information element
1305	 */
1306	if (chandef->width == NL80211_CHAN_WIDTH_40) {
1307		struct ieee80211_sec_chan_offs_ie *sec_chan_ie;
1308		bool ht40plus;
1309
1310		*pos++ = WLAN_EID_SECONDARY_CHANNEL_OFFSET;
1311		*pos++ = sizeof(*sec_chan_ie);
1312		sec_chan_ie = (void *)pos;
1313
1314		ht40plus = cfg80211_get_chandef_type(chandef) ==
1315							NL80211_CHAN_HT40PLUS;
1316		sec_chan_ie->sec_chan_offs = ht40plus ?
1317					     IEEE80211_HT_PARAM_CHA_SEC_ABOVE :
1318					     IEEE80211_HT_PARAM_CHA_SEC_BELOW;
1319		pos += sizeof(*sec_chan_ie);
1320
1321		extra_ies_len += 2 + sizeof(struct ieee80211_sec_chan_offs_ie);
1322	}
1323
1324	/* just set the values to 0, this is a template */
1325	iee80211_tdls_add_ch_switch_timing(pos, 0, 0);
1326
1327	skb = ieee80211_tdls_build_mgmt_packet_data(sdata, sta->sta.addr,
1328					      WLAN_TDLS_CHANNEL_SWITCH_REQUEST,
1329					      0, 0, !sta->sta.tdls_initiator,
1330					      extra_ies, extra_ies_len,
1331					      oper_class, chandef);
1332	if (!skb)
1333		return NULL;
1334
1335	skb = ieee80211_build_data_template(sdata, skb, 0);
1336	if (IS_ERR(skb)) {
1337		tdls_dbg(sdata, "Failed building TDLS channel switch frame\n");
1338		return NULL;
1339	}
1340
1341	if (ch_sw_tm_ie_offset) {
1342		const u8 *tm_ie = ieee80211_tdls_find_sw_timing_ie(skb);
1343
1344		if (!tm_ie) {
1345			tdls_dbg(sdata, "No switch timing IE in TDLS switch\n");
1346			dev_kfree_skb_any(skb);
1347			return NULL;
1348		}
1349
1350		*ch_sw_tm_ie_offset = tm_ie - skb->data;
1351	}
1352
1353	tdls_dbg(sdata,
1354		 "TDLS channel switch request template for %pM ch %d width %d\n",
1355		 sta->sta.addr, chandef->chan->center_freq, chandef->width);
1356	return skb;
1357}
1358
1359int
1360ieee80211_tdls_channel_switch(struct wiphy *wiphy, struct net_device *dev,
1361			      const u8 *addr, u8 oper_class,
1362			      struct cfg80211_chan_def *chandef)
1363{
1364	struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1365	struct ieee80211_local *local = sdata->local;
1366	struct sta_info *sta;
1367	struct sk_buff *skb = NULL;
1368	u32 ch_sw_tm_ie;
1369	int ret;
1370
1371	mutex_lock(&local->sta_mtx);
1372	sta = sta_info_get(sdata, addr);
1373	if (!sta) {
1374		tdls_dbg(sdata,
1375			 "Invalid TDLS peer %pM for channel switch request\n",
1376			 addr);
1377		ret = -ENOENT;
1378		goto out;
1379	}
1380
1381	if (!test_sta_flag(sta, WLAN_STA_TDLS_CHAN_SWITCH)) {
1382		tdls_dbg(sdata, "TDLS channel switch unsupported by %pM\n",
1383			 addr);
1384		ret = -ENOTSUPP;
1385		goto out;
1386	}
1387
1388	skb = ieee80211_tdls_ch_sw_tmpl_get(sta, oper_class, chandef,
1389					    &ch_sw_tm_ie);
1390	if (!skb) {
1391		ret = -ENOENT;
1392		goto out;
1393	}
1394
1395	ret = drv_tdls_channel_switch(local, sdata, &sta->sta, oper_class,
1396				      chandef, skb, ch_sw_tm_ie);
1397	if (!ret)
1398		set_sta_flag(sta, WLAN_STA_TDLS_OFF_CHANNEL);
1399
1400out:
1401	mutex_unlock(&local->sta_mtx);
1402	dev_kfree_skb_any(skb);
1403	return ret;
1404}
1405
1406void
1407ieee80211_tdls_cancel_channel_switch(struct wiphy *wiphy,
1408				     struct net_device *dev,
1409				     const u8 *addr)
1410{
1411	struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1412	struct ieee80211_local *local = sdata->local;
1413	struct sta_info *sta;
1414
1415	mutex_lock(&local->sta_mtx);
1416	sta = sta_info_get(sdata, addr);
1417	if (!sta) {
1418		tdls_dbg(sdata,
1419			 "Invalid TDLS peer %pM for channel switch cancel\n",
1420			 addr);
1421		goto out;
1422	}
1423
1424	if (!test_sta_flag(sta, WLAN_STA_TDLS_OFF_CHANNEL)) {
1425		tdls_dbg(sdata, "TDLS channel switch not initiated by %pM\n",
1426			 addr);
1427		goto out;
1428	}
1429
1430	drv_tdls_cancel_channel_switch(local, sdata, &sta->sta);
1431	clear_sta_flag(sta, WLAN_STA_TDLS_OFF_CHANNEL);
1432
1433out:
1434	mutex_unlock(&local->sta_mtx);
1435}
1436
1437static struct sk_buff *
1438ieee80211_tdls_ch_sw_resp_tmpl_get(struct sta_info *sta,
1439				   u32 *ch_sw_tm_ie_offset)
1440{
1441	struct ieee80211_sub_if_data *sdata = sta->sdata;
1442	struct sk_buff *skb;
1443	u8 extra_ies[2 + sizeof(struct ieee80211_ch_switch_timing)];
1444
1445	/* initial timing are always zero in the template */
1446	iee80211_tdls_add_ch_switch_timing(extra_ies, 0, 0);
1447
1448	skb = ieee80211_tdls_build_mgmt_packet_data(sdata, sta->sta.addr,
1449					WLAN_TDLS_CHANNEL_SWITCH_RESPONSE,
1450					0, 0, !sta->sta.tdls_initiator,
1451					extra_ies, sizeof(extra_ies), 0, NULL);
1452	if (!skb)
1453		return NULL;
1454
1455	skb = ieee80211_build_data_template(sdata, skb, 0);
1456	if (IS_ERR(skb)) {
1457		tdls_dbg(sdata,
1458			 "Failed building TDLS channel switch resp frame\n");
1459		return NULL;
1460	}
1461
1462	if (ch_sw_tm_ie_offset) {
1463		const u8 *tm_ie = ieee80211_tdls_find_sw_timing_ie(skb);
1464
1465		if (!tm_ie) {
1466			tdls_dbg(sdata,
1467				 "No switch timing IE in TDLS switch resp\n");
1468			dev_kfree_skb_any(skb);
1469			return NULL;
1470		}
1471
1472		*ch_sw_tm_ie_offset = tm_ie - skb->data;
1473	}
1474
1475	tdls_dbg(sdata, "TDLS get channel switch response template for %pM\n",
1476		 sta->sta.addr);
1477	return skb;
1478}
1479
1480static int
1481ieee80211_process_tdls_channel_switch_resp(struct ieee80211_sub_if_data *sdata,
1482					   struct sk_buff *skb)
1483{
1484	struct ieee80211_local *local = sdata->local;
1485	struct ieee802_11_elems elems;
1486	struct sta_info *sta;
1487	struct ieee80211_tdls_data *tf = (void *)skb->data;
1488	bool local_initiator;
1489	struct ieee80211_rx_status *rx_status = IEEE80211_SKB_RXCB(skb);
1490	int baselen = offsetof(typeof(*tf), u.chan_switch_resp.variable);
1491	struct ieee80211_tdls_ch_sw_params params = {};
1492	int ret;
1493
1494	params.action_code = WLAN_TDLS_CHANNEL_SWITCH_RESPONSE;
1495	params.timestamp = rx_status->device_timestamp;
1496
1497	if (skb->len < baselen) {
1498		tdls_dbg(sdata, "TDLS channel switch resp too short: %d\n",
1499			 skb->len);
1500		return -EINVAL;
1501	}
1502
1503	mutex_lock(&local->sta_mtx);
1504	sta = sta_info_get(sdata, tf->sa);
1505	if (!sta || !test_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH)) {
1506		tdls_dbg(sdata, "TDLS chan switch from non-peer sta %pM\n",
1507			 tf->sa);
1508		ret = -EINVAL;
1509		goto out;
1510	}
1511
1512	params.sta = &sta->sta;
1513	params.status = le16_to_cpu(tf->u.chan_switch_resp.status_code);
1514	if (params.status != 0) {
1515		ret = 0;
1516		goto call_drv;
1517	}
1518
1519	ieee802_11_parse_elems(tf->u.chan_switch_resp.variable,
1520			       skb->len - baselen, false, &elems);
1521	if (elems.parse_error) {
1522		tdls_dbg(sdata, "Invalid IEs in TDLS channel switch resp\n");
1523		ret = -EINVAL;
1524		goto out;
1525	}
1526
1527	if (!elems.ch_sw_timing || !elems.lnk_id) {
1528		tdls_dbg(sdata, "TDLS channel switch resp - missing IEs\n");
1529		ret = -EINVAL;
1530		goto out;
1531	}
1532
1533	/* validate the initiator is set correctly */
1534	local_initiator =
1535		!memcmp(elems.lnk_id->init_sta, sdata->vif.addr, ETH_ALEN);
1536	if (local_initiator == sta->sta.tdls_initiator) {
1537		tdls_dbg(sdata, "TDLS chan switch invalid lnk-id initiator\n");
1538		ret = -EINVAL;
1539		goto out;
1540	}
1541
1542	params.switch_time = le16_to_cpu(elems.ch_sw_timing->switch_time);
1543	params.switch_timeout = le16_to_cpu(elems.ch_sw_timing->switch_timeout);
1544
1545	params.tmpl_skb =
1546		ieee80211_tdls_ch_sw_resp_tmpl_get(sta, &params.ch_sw_tm_ie);
1547	if (!params.tmpl_skb) {
1548		ret = -ENOENT;
1549		goto out;
1550	}
1551
1552call_drv:
1553	drv_tdls_recv_channel_switch(sdata->local, sdata, &params);
1554
1555	tdls_dbg(sdata,
1556		 "TDLS channel switch response received from %pM status %d\n",
1557		 tf->sa, params.status);
1558
1559out:
1560	mutex_unlock(&local->sta_mtx);
1561	dev_kfree_skb_any(params.tmpl_skb);
1562	return ret;
1563}
1564
1565static int
1566ieee80211_process_tdls_channel_switch_req(struct ieee80211_sub_if_data *sdata,
1567					  struct sk_buff *skb)
1568{
1569	struct ieee80211_local *local = sdata->local;
1570	struct ieee802_11_elems elems;
1571	struct cfg80211_chan_def chandef;
1572	struct ieee80211_channel *chan;
1573	enum nl80211_channel_type chan_type;
1574	int freq;
1575	u8 target_channel, oper_class;
1576	bool local_initiator;
1577	struct sta_info *sta;
1578	enum ieee80211_band band;
1579	struct ieee80211_tdls_data *tf = (void *)skb->data;
1580	struct ieee80211_rx_status *rx_status = IEEE80211_SKB_RXCB(skb);
1581	int baselen = offsetof(typeof(*tf), u.chan_switch_req.variable);
1582	struct ieee80211_tdls_ch_sw_params params = {};
1583	int ret = 0;
1584
1585	params.action_code = WLAN_TDLS_CHANNEL_SWITCH_REQUEST;
1586	params.timestamp = rx_status->device_timestamp;
1587
1588	if (skb->len < baselen) {
1589		tdls_dbg(sdata, "TDLS channel switch req too short: %d\n",
1590			 skb->len);
1591		return -EINVAL;
1592	}
1593
1594	target_channel = tf->u.chan_switch_req.target_channel;
1595	oper_class = tf->u.chan_switch_req.oper_class;
1596
1597	/*
1598	 * We can't easily infer the channel band. The operating class is
1599	 * ambiguous - there are multiple tables (US/Europe/JP/Global). The
1600	 * solution here is to treat channels with number >14 as 5GHz ones,
1601	 * and specifically check for the (oper_class, channel) combinations
1602	 * where this doesn't hold. These are thankfully unique according to
1603	 * IEEE802.11-2012.
1604	 * We consider only the 2GHz and 5GHz bands and 20MHz+ channels as
1605	 * valid here.
1606	 */
1607	if ((oper_class == 112 || oper_class == 2 || oper_class == 3 ||
1608	     oper_class == 4 || oper_class == 5 || oper_class == 6) &&
1609	     target_channel < 14)
1610		band = IEEE80211_BAND_5GHZ;
1611	else
1612		band = target_channel < 14 ? IEEE80211_BAND_2GHZ :
1613					     IEEE80211_BAND_5GHZ;
1614
1615	freq = ieee80211_channel_to_frequency(target_channel, band);
1616	if (freq == 0) {
1617		tdls_dbg(sdata, "Invalid channel in TDLS chan switch: %d\n",
1618			 target_channel);
1619		return -EINVAL;
1620	}
1621
1622	chan = ieee80211_get_channel(sdata->local->hw.wiphy, freq);
1623	if (!chan) {
1624		tdls_dbg(sdata,
1625			 "Unsupported channel for TDLS chan switch: %d\n",
1626			 target_channel);
1627		return -EINVAL;
1628	}
1629
1630	ieee802_11_parse_elems(tf->u.chan_switch_req.variable,
1631			       skb->len - baselen, false, &elems);
1632	if (elems.parse_error) {
1633		tdls_dbg(sdata, "Invalid IEs in TDLS channel switch req\n");
1634		return -EINVAL;
1635	}
1636
1637	if (!elems.ch_sw_timing || !elems.lnk_id) {
1638		tdls_dbg(sdata, "TDLS channel switch req - missing IEs\n");
1639		return -EINVAL;
1640	}
1641
1642	mutex_lock(&local->sta_mtx);
1643	sta = sta_info_get(sdata, tf->sa);
1644	if (!sta || !test_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH)) {
1645		tdls_dbg(sdata, "TDLS chan switch from non-peer sta %pM\n",
1646			 tf->sa);
1647		ret = -EINVAL;
1648		goto out;
1649	}
1650
1651	params.sta = &sta->sta;
1652
1653	/* validate the initiator is set correctly */
1654	local_initiator =
1655		!memcmp(elems.lnk_id->init_sta, sdata->vif.addr, ETH_ALEN);
1656	if (local_initiator == sta->sta.tdls_initiator) {
1657		tdls_dbg(sdata, "TDLS chan switch invalid lnk-id initiator\n");
1658		ret = -EINVAL;
1659		goto out;
1660	}
1661
1662	if (!sta->sta.ht_cap.ht_supported) {
1663		chan_type = NL80211_CHAN_NO_HT;
1664	} else if (!elems.sec_chan_offs) {
1665		chan_type = NL80211_CHAN_HT20;
1666	} else {
1667		switch (elems.sec_chan_offs->sec_chan_offs) {
1668		case IEEE80211_HT_PARAM_CHA_SEC_ABOVE:
1669			chan_type = NL80211_CHAN_HT40PLUS;
1670			break;
1671		case IEEE80211_HT_PARAM_CHA_SEC_BELOW:
1672			chan_type = NL80211_CHAN_HT40MINUS;
1673			break;
1674		default:
1675			chan_type = NL80211_CHAN_HT20;
1676			break;
1677		}
1678	}
1679
1680	cfg80211_chandef_create(&chandef, chan, chan_type);
1681	params.chandef = &chandef;
1682
1683	params.switch_time = le16_to_cpu(elems.ch_sw_timing->switch_time);
1684	params.switch_timeout = le16_to_cpu(elems.ch_sw_timing->switch_timeout);
1685
1686	params.tmpl_skb =
1687		ieee80211_tdls_ch_sw_resp_tmpl_get(sta,
1688						   &params.ch_sw_tm_ie);
1689	if (!params.tmpl_skb) {
1690		ret = -ENOENT;
1691		goto out;
1692	}
1693
1694	drv_tdls_recv_channel_switch(sdata->local, sdata, &params);
1695
1696	tdls_dbg(sdata,
1697		 "TDLS ch switch request received from %pM ch %d width %d\n",
1698		 tf->sa, params.chandef->chan->center_freq,
1699		 params.chandef->width);
1700out:
1701	mutex_unlock(&local->sta_mtx);
1702	dev_kfree_skb_any(params.tmpl_skb);
1703	return ret;
1704}
1705
1706void ieee80211_process_tdls_channel_switch(struct ieee80211_sub_if_data *sdata,
1707					   struct sk_buff *skb)
1708{
1709	struct ieee80211_tdls_data *tf = (void *)skb->data;
1710	struct wiphy *wiphy = sdata->local->hw.wiphy;
1711
1712	/* make sure the driver supports it */
1713	if (!(wiphy->features & NL80211_FEATURE_TDLS_CHANNEL_SWITCH))
1714		return;
1715
1716	/* we want to access the entire packet */
1717	if (skb_linearize(skb))
1718		return;
1719	/*
1720	 * The packet/size was already validated by mac80211 Rx path, only look
1721	 * at the action type.
1722	 */
1723	switch (tf->action_code) {
1724	case WLAN_TDLS_CHANNEL_SWITCH_REQUEST:
1725		ieee80211_process_tdls_channel_switch_req(sdata, skb);
1726		break;
1727	case WLAN_TDLS_CHANNEL_SWITCH_RESPONSE:
1728		ieee80211_process_tdls_channel_switch_resp(sdata, skb);
1729		break;
1730	default:
1731		WARN_ON_ONCE(1);
1732		return;
1733	}
1734}
1735