xref: /linux/net/mac80211/mesh_plink.c (revision 9f69e8a71026839d4bd2e0c6d269600bfaa6f84d)
1 /*
2  * Copyright (c) 2008, 2009 open80211s Ltd.
3  * Author:     Luis Carlos Cobo <luisca@cozybit.com>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation.
8  */
9 #include <linux/gfp.h>
10 #include <linux/kernel.h>
11 #include <linux/random.h>
12 #include "ieee80211_i.h"
13 #include "rate.h"
14 #include "mesh.h"
15 
16 #define PLINK_GET_LLID(p) (p + 2)
17 #define PLINK_GET_PLID(p) (p + 4)
18 
19 #define mod_plink_timer(s, t) (mod_timer(&s->plink_timer, \
20 				jiffies + msecs_to_jiffies(t)))
21 
22 enum plink_event {
23 	PLINK_UNDEFINED,
24 	OPN_ACPT,
25 	OPN_RJCT,
26 	OPN_IGNR,
27 	CNF_ACPT,
28 	CNF_RJCT,
29 	CNF_IGNR,
30 	CLS_ACPT,
31 	CLS_IGNR
32 };
33 
34 static const char * const mplstates[] = {
35 	[NL80211_PLINK_LISTEN] = "LISTEN",
36 	[NL80211_PLINK_OPN_SNT] = "OPN-SNT",
37 	[NL80211_PLINK_OPN_RCVD] = "OPN-RCVD",
38 	[NL80211_PLINK_CNF_RCVD] = "CNF_RCVD",
39 	[NL80211_PLINK_ESTAB] = "ESTAB",
40 	[NL80211_PLINK_HOLDING] = "HOLDING",
41 	[NL80211_PLINK_BLOCKED] = "BLOCKED"
42 };
43 
44 static const char * const mplevents[] = {
45 	[PLINK_UNDEFINED] = "NONE",
46 	[OPN_ACPT] = "OPN_ACPT",
47 	[OPN_RJCT] = "OPN_RJCT",
48 	[OPN_IGNR] = "OPN_IGNR",
49 	[CNF_ACPT] = "CNF_ACPT",
50 	[CNF_RJCT] = "CNF_RJCT",
51 	[CNF_IGNR] = "CNF_IGNR",
52 	[CLS_ACPT] = "CLS_ACPT",
53 	[CLS_IGNR] = "CLS_IGNR"
54 };
55 
56 static int mesh_plink_frame_tx(struct ieee80211_sub_if_data *sdata,
57 			       enum ieee80211_self_protected_actioncode action,
58 			       u8 *da, u16 llid, u16 plid, u16 reason);
59 
60 
61 /* We only need a valid sta if user configured a minimum rssi_threshold. */
62 static bool rssi_threshold_check(struct ieee80211_sub_if_data *sdata,
63 				 struct sta_info *sta)
64 {
65 	s32 rssi_threshold = sdata->u.mesh.mshcfg.rssi_threshold;
66 	return rssi_threshold == 0 ||
67 	       (sta && (s8) -ewma_read(&sta->avg_signal) > rssi_threshold);
68 }
69 
70 /**
71  * mesh_plink_fsm_restart - restart a mesh peer link finite state machine
72  *
73  * @sta: mesh peer link to restart
74  *
75  * Locking: this function must be called holding sta->lock
76  */
77 static inline void mesh_plink_fsm_restart(struct sta_info *sta)
78 {
79 	sta->plink_state = NL80211_PLINK_LISTEN;
80 	sta->llid = sta->plid = sta->reason = 0;
81 	sta->plink_retries = 0;
82 }
83 
84 /*
85  * mesh_set_short_slot_time - enable / disable ERP short slot time.
86  *
87  * The standard indirectly mandates mesh STAs to turn off short slot time by
88  * disallowing advertising this (802.11-2012 8.4.1.4), but that doesn't mean we
89  * can't be sneaky about it. Enable short slot time if all mesh STAs in the
90  * MBSS support ERP rates.
91  *
92  * Returns BSS_CHANGED_ERP_SLOT or 0 for no change.
93  */
94 static u32 mesh_set_short_slot_time(struct ieee80211_sub_if_data *sdata)
95 {
96 	struct ieee80211_local *local = sdata->local;
97 	enum ieee80211_band band = ieee80211_get_sdata_band(sdata);
98 	struct ieee80211_supported_band *sband = local->hw.wiphy->bands[band];
99 	struct sta_info *sta;
100 	u32 erp_rates = 0, changed = 0;
101 	int i;
102 	bool short_slot = false;
103 
104 	if (band == IEEE80211_BAND_5GHZ) {
105 		/* (IEEE 802.11-2012 19.4.5) */
106 		short_slot = true;
107 		goto out;
108 	} else if (band != IEEE80211_BAND_2GHZ ||
109 		   (band == IEEE80211_BAND_2GHZ &&
110 		    local->hw.flags & IEEE80211_HW_2GHZ_SHORT_SLOT_INCAPABLE))
111 		goto out;
112 
113 	for (i = 0; i < sband->n_bitrates; i++)
114 		if (sband->bitrates[i].flags & IEEE80211_RATE_ERP_G)
115 			erp_rates |= BIT(i);
116 
117 	if (!erp_rates)
118 		goto out;
119 
120 	rcu_read_lock();
121 	list_for_each_entry_rcu(sta, &local->sta_list, list) {
122 		if (sdata != sta->sdata ||
123 		    sta->plink_state != NL80211_PLINK_ESTAB)
124 			continue;
125 
126 		short_slot = false;
127 		if (erp_rates & sta->sta.supp_rates[band])
128 			short_slot = true;
129 		 else
130 			break;
131 	}
132 	rcu_read_unlock();
133 
134 out:
135 	if (sdata->vif.bss_conf.use_short_slot != short_slot) {
136 		sdata->vif.bss_conf.use_short_slot = short_slot;
137 		changed = BSS_CHANGED_ERP_SLOT;
138 		mpl_dbg(sdata, "mesh_plink %pM: ERP short slot time %d\n",
139 			sdata->vif.addr, short_slot);
140 	}
141 	return changed;
142 }
143 
144 /**
145  * mesh_set_ht_prot_mode - set correct HT protection mode
146  *
147  * Section 9.23.3.5 of IEEE 80211-2012 describes the protection rules for HT
148  * mesh STA in a MBSS. Three HT protection modes are supported for now, non-HT
149  * mixed mode, 20MHz-protection and no-protection mode. non-HT mixed mode is
150  * selected if any non-HT peers are present in our MBSS.  20MHz-protection mode
151  * is selected if all peers in our 20/40MHz MBSS support HT and atleast one
152  * HT20 peer is present. Otherwise no-protection mode is selected.
153  */
154 static u32 mesh_set_ht_prot_mode(struct ieee80211_sub_if_data *sdata)
155 {
156 	struct ieee80211_local *local = sdata->local;
157 	struct sta_info *sta;
158 	u16 ht_opmode;
159 	bool non_ht_sta = false, ht20_sta = false;
160 
161 	switch (sdata->vif.bss_conf.chandef.width) {
162 	case NL80211_CHAN_WIDTH_20_NOHT:
163 	case NL80211_CHAN_WIDTH_5:
164 	case NL80211_CHAN_WIDTH_10:
165 		return 0;
166 	default:
167 		break;
168 	}
169 
170 	rcu_read_lock();
171 	list_for_each_entry_rcu(sta, &local->sta_list, list) {
172 		if (sdata != sta->sdata ||
173 		    sta->plink_state != NL80211_PLINK_ESTAB)
174 			continue;
175 
176 		if (sta->sta.bandwidth > IEEE80211_STA_RX_BW_20)
177 			continue;
178 
179 		if (!sta->sta.ht_cap.ht_supported) {
180 			mpl_dbg(sdata, "nonHT sta (%pM) is present\n",
181 				       sta->sta.addr);
182 			non_ht_sta = true;
183 			break;
184 		}
185 
186 		mpl_dbg(sdata, "HT20 sta (%pM) is present\n", sta->sta.addr);
187 		ht20_sta = true;
188 	}
189 	rcu_read_unlock();
190 
191 	if (non_ht_sta)
192 		ht_opmode = IEEE80211_HT_OP_MODE_PROTECTION_NONHT_MIXED;
193 	else if (ht20_sta &&
194 		 sdata->vif.bss_conf.chandef.width > NL80211_CHAN_WIDTH_20)
195 		ht_opmode = IEEE80211_HT_OP_MODE_PROTECTION_20MHZ;
196 	else
197 		ht_opmode = IEEE80211_HT_OP_MODE_PROTECTION_NONE;
198 
199 	if (sdata->vif.bss_conf.ht_operation_mode == ht_opmode)
200 		return 0;
201 
202 	sdata->vif.bss_conf.ht_operation_mode = ht_opmode;
203 	sdata->u.mesh.mshcfg.ht_opmode = ht_opmode;
204 	mpl_dbg(sdata, "selected new HT protection mode %d\n", ht_opmode);
205 	return BSS_CHANGED_HT;
206 }
207 
208 /**
209  * __mesh_plink_deactivate - deactivate mesh peer link
210  *
211  * @sta: mesh peer link to deactivate
212  *
213  * All mesh paths with this peer as next hop will be flushed
214  * Returns beacon changed flag if the beacon content changed.
215  *
216  * Locking: the caller must hold sta->lock
217  */
218 static u32 __mesh_plink_deactivate(struct sta_info *sta)
219 {
220 	struct ieee80211_sub_if_data *sdata = sta->sdata;
221 	u32 changed = 0;
222 
223 	if (sta->plink_state == NL80211_PLINK_ESTAB)
224 		changed = mesh_plink_dec_estab_count(sdata);
225 	sta->plink_state = NL80211_PLINK_BLOCKED;
226 	mesh_path_flush_by_nexthop(sta);
227 
228 	ieee80211_mps_sta_status_update(sta);
229 	changed |= ieee80211_mps_set_sta_local_pm(sta,
230 			NL80211_MESH_POWER_UNKNOWN);
231 
232 	return changed;
233 }
234 
235 /**
236  * mesh_plink_deactivate - deactivate mesh peer link
237  *
238  * @sta: mesh peer link to deactivate
239  *
240  * All mesh paths with this peer as next hop will be flushed
241  */
242 u32 mesh_plink_deactivate(struct sta_info *sta)
243 {
244 	struct ieee80211_sub_if_data *sdata = sta->sdata;
245 	u32 changed;
246 
247 	spin_lock_bh(&sta->lock);
248 	changed = __mesh_plink_deactivate(sta);
249 	sta->reason = WLAN_REASON_MESH_PEER_CANCELED;
250 	mesh_plink_frame_tx(sdata, WLAN_SP_MESH_PEERING_CLOSE,
251 			    sta->sta.addr, sta->llid, sta->plid,
252 			    sta->reason);
253 	spin_unlock_bh(&sta->lock);
254 
255 	return changed;
256 }
257 
258 static int mesh_plink_frame_tx(struct ieee80211_sub_if_data *sdata,
259 			       enum ieee80211_self_protected_actioncode action,
260 			       u8 *da, u16 llid, u16 plid, u16 reason)
261 {
262 	struct ieee80211_local *local = sdata->local;
263 	struct sk_buff *skb;
264 	struct ieee80211_tx_info *info;
265 	struct ieee80211_mgmt *mgmt;
266 	bool include_plid = false;
267 	u16 peering_proto = 0;
268 	u8 *pos, ie_len = 4;
269 	int hdr_len = offsetof(struct ieee80211_mgmt, u.action.u.self_prot) +
270 		      sizeof(mgmt->u.action.u.self_prot);
271 	int err = -ENOMEM;
272 
273 	skb = dev_alloc_skb(local->tx_headroom +
274 			    hdr_len +
275 			    2 + /* capability info */
276 			    2 + /* AID */
277 			    2 + 8 + /* supported rates */
278 			    2 + (IEEE80211_MAX_SUPP_RATES - 8) +
279 			    2 + sdata->u.mesh.mesh_id_len +
280 			    2 + sizeof(struct ieee80211_meshconf_ie) +
281 			    2 + sizeof(struct ieee80211_ht_cap) +
282 			    2 + sizeof(struct ieee80211_ht_operation) +
283 			    2 + 8 + /* peering IE */
284 			    sdata->u.mesh.ie_len);
285 	if (!skb)
286 		return err;
287 	info = IEEE80211_SKB_CB(skb);
288 	skb_reserve(skb, local->tx_headroom);
289 	mgmt = (struct ieee80211_mgmt *) skb_put(skb, hdr_len);
290 	memset(mgmt, 0, hdr_len);
291 	mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
292 					  IEEE80211_STYPE_ACTION);
293 	memcpy(mgmt->da, da, ETH_ALEN);
294 	memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
295 	memcpy(mgmt->bssid, sdata->vif.addr, ETH_ALEN);
296 	mgmt->u.action.category = WLAN_CATEGORY_SELF_PROTECTED;
297 	mgmt->u.action.u.self_prot.action_code = action;
298 
299 	if (action != WLAN_SP_MESH_PEERING_CLOSE) {
300 		enum ieee80211_band band = ieee80211_get_sdata_band(sdata);
301 
302 		/* capability info */
303 		pos = skb_put(skb, 2);
304 		memset(pos, 0, 2);
305 		if (action == WLAN_SP_MESH_PEERING_CONFIRM) {
306 			/* AID */
307 			pos = skb_put(skb, 2);
308 			put_unaligned_le16(plid, pos + 2);
309 		}
310 		if (ieee80211_add_srates_ie(sdata, skb, true, band) ||
311 		    ieee80211_add_ext_srates_ie(sdata, skb, true, band) ||
312 		    mesh_add_rsn_ie(sdata, skb) ||
313 		    mesh_add_meshid_ie(sdata, skb) ||
314 		    mesh_add_meshconf_ie(sdata, skb))
315 			goto free;
316 	} else {	/* WLAN_SP_MESH_PEERING_CLOSE */
317 		info->flags |= IEEE80211_TX_CTL_NO_ACK;
318 		if (mesh_add_meshid_ie(sdata, skb))
319 			goto free;
320 	}
321 
322 	/* Add Mesh Peering Management element */
323 	switch (action) {
324 	case WLAN_SP_MESH_PEERING_OPEN:
325 		break;
326 	case WLAN_SP_MESH_PEERING_CONFIRM:
327 		ie_len += 2;
328 		include_plid = true;
329 		break;
330 	case WLAN_SP_MESH_PEERING_CLOSE:
331 		if (plid) {
332 			ie_len += 2;
333 			include_plid = true;
334 		}
335 		ie_len += 2;	/* reason code */
336 		break;
337 	default:
338 		err = -EINVAL;
339 		goto free;
340 	}
341 
342 	if (WARN_ON(skb_tailroom(skb) < 2 + ie_len))
343 		goto free;
344 
345 	pos = skb_put(skb, 2 + ie_len);
346 	*pos++ = WLAN_EID_PEER_MGMT;
347 	*pos++ = ie_len;
348 	memcpy(pos, &peering_proto, 2);
349 	pos += 2;
350 	put_unaligned_le16(llid, pos);
351 	pos += 2;
352 	if (include_plid) {
353 		put_unaligned_le16(plid, pos);
354 		pos += 2;
355 	}
356 	if (action == WLAN_SP_MESH_PEERING_CLOSE) {
357 		put_unaligned_le16(reason, pos);
358 		pos += 2;
359 	}
360 
361 	if (action != WLAN_SP_MESH_PEERING_CLOSE) {
362 		if (mesh_add_ht_cap_ie(sdata, skb) ||
363 		    mesh_add_ht_oper_ie(sdata, skb))
364 			goto free;
365 	}
366 
367 	if (mesh_add_vendor_ies(sdata, skb))
368 		goto free;
369 
370 	ieee80211_tx_skb(sdata, skb);
371 	return 0;
372 free:
373 	kfree_skb(skb);
374 	return err;
375 }
376 
377 static void mesh_sta_info_init(struct ieee80211_sub_if_data *sdata,
378 			       struct sta_info *sta,
379 			       struct ieee802_11_elems *elems, bool insert)
380 {
381 	struct ieee80211_local *local = sdata->local;
382 	enum ieee80211_band band = ieee80211_get_sdata_band(sdata);
383 	struct ieee80211_supported_band *sband;
384 	u32 rates, basic_rates = 0, changed = 0;
385 	enum ieee80211_sta_rx_bandwidth bw = sta->sta.bandwidth;
386 
387 	sband = local->hw.wiphy->bands[band];
388 	rates = ieee80211_sta_get_rates(sdata, elems, band, &basic_rates);
389 
390 	spin_lock_bh(&sta->lock);
391 	sta->last_rx = jiffies;
392 
393 	/* rates and capabilities don't change during peering */
394 	if (sta->plink_state == NL80211_PLINK_ESTAB)
395 		goto out;
396 
397 	if (sta->sta.supp_rates[band] != rates)
398 		changed |= IEEE80211_RC_SUPP_RATES_CHANGED;
399 	sta->sta.supp_rates[band] = rates;
400 
401 	if (ieee80211_ht_cap_ie_to_sta_ht_cap(sdata, sband,
402 					      elems->ht_cap_elem, sta))
403 		changed |= IEEE80211_RC_BW_CHANGED;
404 
405 	if (bw != sta->sta.bandwidth)
406 		changed |= IEEE80211_RC_BW_CHANGED;
407 
408 	/* HT peer is operating 20MHz-only */
409 	if (elems->ht_operation &&
410 	    !(elems->ht_operation->ht_param &
411 	      IEEE80211_HT_PARAM_CHAN_WIDTH_ANY)) {
412 		if (sta->sta.bandwidth != IEEE80211_STA_RX_BW_20)
413 			changed |= IEEE80211_RC_BW_CHANGED;
414 		sta->sta.bandwidth = IEEE80211_STA_RX_BW_20;
415 	}
416 
417 	if (insert)
418 		rate_control_rate_init(sta);
419 	else
420 		rate_control_rate_update(local, sband, sta, changed);
421 out:
422 	spin_unlock_bh(&sta->lock);
423 }
424 
425 static struct sta_info *
426 __mesh_sta_info_alloc(struct ieee80211_sub_if_data *sdata, u8 *hw_addr)
427 {
428 	struct sta_info *sta;
429 
430 	if (sdata->local->num_sta >= MESH_MAX_PLINKS)
431 		return NULL;
432 
433 	sta = sta_info_alloc(sdata, hw_addr, GFP_KERNEL);
434 	if (!sta)
435 		return NULL;
436 
437 	sta->plink_state = NL80211_PLINK_LISTEN;
438 	sta->sta.wme = true;
439 
440 	sta_info_pre_move_state(sta, IEEE80211_STA_AUTH);
441 	sta_info_pre_move_state(sta, IEEE80211_STA_ASSOC);
442 	sta_info_pre_move_state(sta, IEEE80211_STA_AUTHORIZED);
443 
444 	return sta;
445 }
446 
447 static struct sta_info *
448 mesh_sta_info_alloc(struct ieee80211_sub_if_data *sdata, u8 *addr,
449 		    struct ieee802_11_elems *elems)
450 {
451 	struct sta_info *sta = NULL;
452 
453 	/* Userspace handles station allocation */
454 	if (sdata->u.mesh.user_mpm ||
455 	    sdata->u.mesh.security & IEEE80211_MESH_SEC_AUTHED)
456 		cfg80211_notify_new_peer_candidate(sdata->dev, addr,
457 						   elems->ie_start,
458 						   elems->total_len,
459 						   GFP_KERNEL);
460 	else
461 		sta = __mesh_sta_info_alloc(sdata, addr);
462 
463 	return sta;
464 }
465 
466 /*
467  * mesh_sta_info_get - return mesh sta info entry for @addr.
468  *
469  * @sdata: local meshif
470  * @addr: peer's address
471  * @elems: IEs from beacon or mesh peering frame.
472  *
473  * Return existing or newly allocated sta_info under RCU read lock.
474  * (re)initialize with given IEs.
475  */
476 static struct sta_info *
477 mesh_sta_info_get(struct ieee80211_sub_if_data *sdata,
478 		  u8 *addr, struct ieee802_11_elems *elems) __acquires(RCU)
479 {
480 	struct sta_info *sta = NULL;
481 
482 	rcu_read_lock();
483 	sta = sta_info_get(sdata, addr);
484 	if (sta) {
485 		mesh_sta_info_init(sdata, sta, elems, false);
486 	} else {
487 		rcu_read_unlock();
488 		/* can't run atomic */
489 		sta = mesh_sta_info_alloc(sdata, addr, elems);
490 		if (!sta) {
491 			rcu_read_lock();
492 			return NULL;
493 		}
494 
495 		mesh_sta_info_init(sdata, sta, elems, true);
496 
497 		if (sta_info_insert_rcu(sta))
498 			return NULL;
499 	}
500 
501 	return sta;
502 }
503 
504 /*
505  * mesh_neighbour_update - update or initialize new mesh neighbor.
506  *
507  * @sdata: local meshif
508  * @addr: peer's address
509  * @elems: IEs from beacon or mesh peering frame
510  *
511  * Initiates peering if appropriate.
512  */
513 void mesh_neighbour_update(struct ieee80211_sub_if_data *sdata,
514 			   u8 *hw_addr,
515 			   struct ieee802_11_elems *elems)
516 {
517 	struct sta_info *sta;
518 	u32 changed = 0;
519 
520 	sta = mesh_sta_info_get(sdata, hw_addr, elems);
521 	if (!sta)
522 		goto out;
523 
524 	if (mesh_peer_accepts_plinks(elems) &&
525 	    sta->plink_state == NL80211_PLINK_LISTEN &&
526 	    sdata->u.mesh.accepting_plinks &&
527 	    sdata->u.mesh.mshcfg.auto_open_plinks &&
528 	    rssi_threshold_check(sdata, sta))
529 		changed = mesh_plink_open(sta);
530 
531 	ieee80211_mps_frame_release(sta, elems);
532 out:
533 	rcu_read_unlock();
534 	ieee80211_mbss_info_change_notify(sdata, changed);
535 }
536 
537 static void mesh_plink_timer(unsigned long data)
538 {
539 	struct sta_info *sta;
540 	u16 reason = 0;
541 	struct ieee80211_sub_if_data *sdata;
542 	struct mesh_config *mshcfg;
543 	enum ieee80211_self_protected_actioncode action = 0;
544 
545 	/*
546 	 * This STA is valid because sta_info_destroy() will
547 	 * del_timer_sync() this timer after having made sure
548 	 * it cannot be readded (by deleting the plink.)
549 	 */
550 	sta = (struct sta_info *) data;
551 
552 	if (sta->sdata->local->quiescing)
553 		return;
554 
555 	spin_lock_bh(&sta->lock);
556 
557 	/* If a timer fires just before a state transition on another CPU,
558 	 * we may have already extended the timeout and changed state by the
559 	 * time we've acquired the lock and arrived  here.  In that case,
560 	 * skip this timer and wait for the new one.
561 	 */
562 	if (time_before(jiffies, sta->plink_timer.expires)) {
563 		mpl_dbg(sta->sdata,
564 			"Ignoring timer for %pM in state %s (timer adjusted)",
565 			sta->sta.addr, mplstates[sta->plink_state]);
566 		spin_unlock_bh(&sta->lock);
567 		return;
568 	}
569 
570 	/* del_timer() and handler may race when entering these states */
571 	if (sta->plink_state == NL80211_PLINK_LISTEN ||
572 	    sta->plink_state == NL80211_PLINK_ESTAB) {
573 		mpl_dbg(sta->sdata,
574 			"Ignoring timer for %pM in state %s (timer deleted)",
575 			sta->sta.addr, mplstates[sta->plink_state]);
576 		spin_unlock_bh(&sta->lock);
577 		return;
578 	}
579 
580 	mpl_dbg(sta->sdata,
581 		"Mesh plink timer for %pM fired on state %s\n",
582 		sta->sta.addr, mplstates[sta->plink_state]);
583 	sdata = sta->sdata;
584 	mshcfg = &sdata->u.mesh.mshcfg;
585 
586 	switch (sta->plink_state) {
587 	case NL80211_PLINK_OPN_RCVD:
588 	case NL80211_PLINK_OPN_SNT:
589 		/* retry timer */
590 		if (sta->plink_retries < mshcfg->dot11MeshMaxRetries) {
591 			u32 rand;
592 			mpl_dbg(sta->sdata,
593 				"Mesh plink for %pM (retry, timeout): %d %d\n",
594 				sta->sta.addr, sta->plink_retries,
595 				sta->plink_timeout);
596 			get_random_bytes(&rand, sizeof(u32));
597 			sta->plink_timeout = sta->plink_timeout +
598 					     rand % sta->plink_timeout;
599 			++sta->plink_retries;
600 			mod_plink_timer(sta, sta->plink_timeout);
601 			action = WLAN_SP_MESH_PEERING_OPEN;
602 			break;
603 		}
604 		reason = WLAN_REASON_MESH_MAX_RETRIES;
605 		/* fall through on else */
606 	case NL80211_PLINK_CNF_RCVD:
607 		/* confirm timer */
608 		if (!reason)
609 			reason = WLAN_REASON_MESH_CONFIRM_TIMEOUT;
610 		sta->plink_state = NL80211_PLINK_HOLDING;
611 		mod_plink_timer(sta, mshcfg->dot11MeshHoldingTimeout);
612 		action = WLAN_SP_MESH_PEERING_CLOSE;
613 		break;
614 	case NL80211_PLINK_HOLDING:
615 		/* holding timer */
616 		del_timer(&sta->plink_timer);
617 		mesh_plink_fsm_restart(sta);
618 		break;
619 	default:
620 		break;
621 	}
622 	spin_unlock_bh(&sta->lock);
623 	if (action)
624 		mesh_plink_frame_tx(sdata, action, sta->sta.addr,
625 				    sta->llid, sta->plid, reason);
626 }
627 
628 static inline void mesh_plink_timer_set(struct sta_info *sta, u32 timeout)
629 {
630 	sta->plink_timer.expires = jiffies + msecs_to_jiffies(timeout);
631 	sta->plink_timer.data = (unsigned long) sta;
632 	sta->plink_timer.function = mesh_plink_timer;
633 	sta->plink_timeout = timeout;
634 	add_timer(&sta->plink_timer);
635 }
636 
637 static bool llid_in_use(struct ieee80211_sub_if_data *sdata,
638 			u16 llid)
639 {
640 	struct ieee80211_local *local = sdata->local;
641 	bool in_use = false;
642 	struct sta_info *sta;
643 
644 	rcu_read_lock();
645 	list_for_each_entry_rcu(sta, &local->sta_list, list) {
646 		if (!memcmp(&sta->llid, &llid, sizeof(llid))) {
647 			in_use = true;
648 			break;
649 		}
650 	}
651 	rcu_read_unlock();
652 
653 	return in_use;
654 }
655 
656 static u16 mesh_get_new_llid(struct ieee80211_sub_if_data *sdata)
657 {
658 	u16 llid;
659 
660 	do {
661 		get_random_bytes(&llid, sizeof(llid));
662 		/* for mesh PS we still only have the AID range for TIM bits */
663 		llid = (llid % IEEE80211_MAX_AID) + 1;
664 	} while (llid_in_use(sdata, llid));
665 
666 	return llid;
667 }
668 
669 u32 mesh_plink_open(struct sta_info *sta)
670 {
671 	struct ieee80211_sub_if_data *sdata = sta->sdata;
672 	u32 changed;
673 
674 	if (!test_sta_flag(sta, WLAN_STA_AUTH))
675 		return 0;
676 
677 	spin_lock_bh(&sta->lock);
678 	sta->llid = mesh_get_new_llid(sdata);
679 	if (sta->plink_state != NL80211_PLINK_LISTEN &&
680 	    sta->plink_state != NL80211_PLINK_BLOCKED) {
681 		spin_unlock_bh(&sta->lock);
682 		return 0;
683 	}
684 	sta->plink_state = NL80211_PLINK_OPN_SNT;
685 	mesh_plink_timer_set(sta, sdata->u.mesh.mshcfg.dot11MeshRetryTimeout);
686 	spin_unlock_bh(&sta->lock);
687 	mpl_dbg(sdata,
688 		"Mesh plink: starting establishment with %pM\n",
689 		sta->sta.addr);
690 
691 	/* set the non-peer mode to active during peering */
692 	changed = ieee80211_mps_local_status_update(sdata);
693 
694 	mesh_plink_frame_tx(sdata, WLAN_SP_MESH_PEERING_OPEN,
695 			    sta->sta.addr, sta->llid, 0, 0);
696 	return changed;
697 }
698 
699 u32 mesh_plink_block(struct sta_info *sta)
700 {
701 	u32 changed;
702 
703 	spin_lock_bh(&sta->lock);
704 	changed = __mesh_plink_deactivate(sta);
705 	sta->plink_state = NL80211_PLINK_BLOCKED;
706 	spin_unlock_bh(&sta->lock);
707 
708 	return changed;
709 }
710 
711 static void mesh_plink_close(struct ieee80211_sub_if_data *sdata,
712 			     struct sta_info *sta,
713 			     enum plink_event event)
714 {
715 	struct mesh_config *mshcfg = &sdata->u.mesh.mshcfg;
716 
717 	u16 reason = (event == CLS_ACPT) ?
718 		     WLAN_REASON_MESH_CLOSE : WLAN_REASON_MESH_CONFIG;
719 
720 	sta->reason = reason;
721 	sta->plink_state = NL80211_PLINK_HOLDING;
722 	mod_plink_timer(sta, mshcfg->dot11MeshHoldingTimeout);
723 }
724 
725 static u32 mesh_plink_establish(struct ieee80211_sub_if_data *sdata,
726 				struct sta_info *sta)
727 {
728 	struct mesh_config *mshcfg = &sdata->u.mesh.mshcfg;
729 	u32 changed = 0;
730 
731 	del_timer(&sta->plink_timer);
732 	sta->plink_state = NL80211_PLINK_ESTAB;
733 	changed |= mesh_plink_inc_estab_count(sdata);
734 	changed |= mesh_set_ht_prot_mode(sdata);
735 	changed |= mesh_set_short_slot_time(sdata);
736 	mpl_dbg(sdata, "Mesh plink with %pM ESTABLISHED\n", sta->sta.addr);
737 	ieee80211_mps_sta_status_update(sta);
738 	changed |= ieee80211_mps_set_sta_local_pm(sta, mshcfg->power_mode);
739 	return changed;
740 }
741 
742 /**
743  * mesh_plink_fsm - step @sta MPM based on @event
744  *
745  * @sdata: interface
746  * @sta: mesh neighbor
747  * @event: peering event
748  *
749  * Return: changed MBSS flags
750  */
751 static u32 mesh_plink_fsm(struct ieee80211_sub_if_data *sdata,
752 			  struct sta_info *sta, enum plink_event event)
753 {
754 	struct mesh_config *mshcfg = &sdata->u.mesh.mshcfg;
755 	enum ieee80211_self_protected_actioncode action = 0;
756 	u32 changed = 0;
757 
758 	mpl_dbg(sdata, "peer %pM in state %s got event %s\n", sta->sta.addr,
759 		mplstates[sta->plink_state], mplevents[event]);
760 
761 	spin_lock_bh(&sta->lock);
762 	switch (sta->plink_state) {
763 	case NL80211_PLINK_LISTEN:
764 		switch (event) {
765 		case CLS_ACPT:
766 			mesh_plink_fsm_restart(sta);
767 			break;
768 		case OPN_ACPT:
769 			sta->plink_state = NL80211_PLINK_OPN_RCVD;
770 			sta->llid = mesh_get_new_llid(sdata);
771 			mesh_plink_timer_set(sta,
772 					     mshcfg->dot11MeshRetryTimeout);
773 
774 			/* set the non-peer mode to active during peering */
775 			changed |= ieee80211_mps_local_status_update(sdata);
776 			action = WLAN_SP_MESH_PEERING_OPEN;
777 			break;
778 		default:
779 			break;
780 		}
781 		break;
782 	case NL80211_PLINK_OPN_SNT:
783 		switch (event) {
784 		case OPN_RJCT:
785 		case CNF_RJCT:
786 		case CLS_ACPT:
787 			mesh_plink_close(sdata, sta, event);
788 			action = WLAN_SP_MESH_PEERING_CLOSE;
789 			break;
790 		case OPN_ACPT:
791 			/* retry timer is left untouched */
792 			sta->plink_state = NL80211_PLINK_OPN_RCVD;
793 			action = WLAN_SP_MESH_PEERING_CONFIRM;
794 			break;
795 		case CNF_ACPT:
796 			sta->plink_state = NL80211_PLINK_CNF_RCVD;
797 			mod_plink_timer(sta, mshcfg->dot11MeshConfirmTimeout);
798 			break;
799 		default:
800 			break;
801 		}
802 		break;
803 	case NL80211_PLINK_OPN_RCVD:
804 		switch (event) {
805 		case OPN_RJCT:
806 		case CNF_RJCT:
807 		case CLS_ACPT:
808 			mesh_plink_close(sdata, sta, event);
809 			action = WLAN_SP_MESH_PEERING_CLOSE;
810 			break;
811 		case OPN_ACPT:
812 			action = WLAN_SP_MESH_PEERING_CONFIRM;
813 			break;
814 		case CNF_ACPT:
815 			changed |= mesh_plink_establish(sdata, sta);
816 			break;
817 		default:
818 			break;
819 		}
820 		break;
821 	case NL80211_PLINK_CNF_RCVD:
822 		switch (event) {
823 		case OPN_RJCT:
824 		case CNF_RJCT:
825 		case CLS_ACPT:
826 			mesh_plink_close(sdata, sta, event);
827 			action = WLAN_SP_MESH_PEERING_CLOSE;
828 			break;
829 		case OPN_ACPT:
830 			changed |= mesh_plink_establish(sdata, sta);
831 			action = WLAN_SP_MESH_PEERING_CONFIRM;
832 			break;
833 		default:
834 			break;
835 		}
836 		break;
837 	case NL80211_PLINK_ESTAB:
838 		switch (event) {
839 		case CLS_ACPT:
840 			changed |= __mesh_plink_deactivate(sta);
841 			changed |= mesh_set_ht_prot_mode(sdata);
842 			changed |= mesh_set_short_slot_time(sdata);
843 			mesh_plink_close(sdata, sta, event);
844 			action = WLAN_SP_MESH_PEERING_CLOSE;
845 			break;
846 		case OPN_ACPT:
847 			action = WLAN_SP_MESH_PEERING_CONFIRM;
848 			break;
849 		default:
850 			break;
851 		}
852 		break;
853 	case NL80211_PLINK_HOLDING:
854 		switch (event) {
855 		case CLS_ACPT:
856 			del_timer(&sta->plink_timer);
857 			mesh_plink_fsm_restart(sta);
858 			break;
859 		case OPN_ACPT:
860 		case CNF_ACPT:
861 		case OPN_RJCT:
862 		case CNF_RJCT:
863 			action = WLAN_SP_MESH_PEERING_CLOSE;
864 			break;
865 		default:
866 			break;
867 		}
868 		break;
869 	default:
870 		/* should not get here, PLINK_BLOCKED is dealt with at the
871 		 * beginning of the function
872 		 */
873 		break;
874 	}
875 	spin_unlock_bh(&sta->lock);
876 	if (action) {
877 		mesh_plink_frame_tx(sdata, action, sta->sta.addr,
878 				    sta->llid, sta->plid, sta->reason);
879 
880 		/* also send confirm in open case */
881 		if (action == WLAN_SP_MESH_PEERING_OPEN) {
882 			mesh_plink_frame_tx(sdata,
883 					    WLAN_SP_MESH_PEERING_CONFIRM,
884 					    sta->sta.addr, sta->llid,
885 					    sta->plid, 0);
886 		}
887 	}
888 
889 	return changed;
890 }
891 
892 /*
893  * mesh_plink_get_event - get correct MPM event
894  *
895  * @sdata: interface
896  * @sta: peer, leave NULL if processing a frame from a new suitable peer
897  * @elems: peering management IEs
898  * @ftype: frame type
899  * @llid: peer's peer link ID
900  * @plid: peer's local link ID
901  *
902  * Return: new peering event for @sta, but PLINK_UNDEFINED should be treated as
903  * an error.
904  */
905 static enum plink_event
906 mesh_plink_get_event(struct ieee80211_sub_if_data *sdata,
907 		     struct sta_info *sta,
908 		     struct ieee802_11_elems *elems,
909 		     enum ieee80211_self_protected_actioncode ftype,
910 		     u16 llid, u16 plid)
911 {
912 	enum plink_event event = PLINK_UNDEFINED;
913 	u8 ie_len = elems->peering_len;
914 	bool matches_local;
915 
916 	matches_local = (ftype == WLAN_SP_MESH_PEERING_CLOSE ||
917 			 mesh_matches_local(sdata, elems));
918 
919 	/* deny open request from non-matching peer */
920 	if (!matches_local && !sta) {
921 		event = OPN_RJCT;
922 		goto out;
923 	}
924 
925 	if (!sta) {
926 		if (ftype != WLAN_SP_MESH_PEERING_OPEN) {
927 			mpl_dbg(sdata, "Mesh plink: cls or cnf from unknown peer\n");
928 			goto out;
929 		}
930 		/* ftype == WLAN_SP_MESH_PEERING_OPEN */
931 		if (!mesh_plink_free_count(sdata)) {
932 			mpl_dbg(sdata, "Mesh plink error: no more free plinks\n");
933 			goto out;
934 		}
935 	} else {
936 		if (!test_sta_flag(sta, WLAN_STA_AUTH)) {
937 			mpl_dbg(sdata, "Mesh plink: Action frame from non-authed peer\n");
938 			goto out;
939 		}
940 		if (sta->plink_state == NL80211_PLINK_BLOCKED)
941 			goto out;
942 	}
943 
944 	/* new matching peer */
945 	if (!sta) {
946 		event = OPN_ACPT;
947 		goto out;
948 	}
949 
950 	switch (ftype) {
951 	case WLAN_SP_MESH_PEERING_OPEN:
952 		if (!matches_local)
953 			event = OPN_RJCT;
954 		if (!mesh_plink_free_count(sdata) ||
955 		    (sta->plid && sta->plid != plid))
956 			event = OPN_IGNR;
957 		else
958 			event = OPN_ACPT;
959 		break;
960 	case WLAN_SP_MESH_PEERING_CONFIRM:
961 		if (!matches_local)
962 			event = CNF_RJCT;
963 		if (!mesh_plink_free_count(sdata) ||
964 		    sta->llid != llid ||
965 		    (sta->plid && sta->plid != plid))
966 			event = CNF_IGNR;
967 		else
968 			event = CNF_ACPT;
969 		break;
970 	case WLAN_SP_MESH_PEERING_CLOSE:
971 		if (sta->plink_state == NL80211_PLINK_ESTAB)
972 			/* Do not check for llid or plid. This does not
973 			 * follow the standard but since multiple plinks
974 			 * per sta are not supported, it is necessary in
975 			 * order to avoid a livelock when MP A sees an
976 			 * establish peer link to MP B but MP B does not
977 			 * see it. This can be caused by a timeout in
978 			 * B's peer link establishment or B beign
979 			 * restarted.
980 			 */
981 			event = CLS_ACPT;
982 		else if (sta->plid != plid)
983 			event = CLS_IGNR;
984 		else if (ie_len == 8 && sta->llid != llid)
985 			event = CLS_IGNR;
986 		else
987 			event = CLS_ACPT;
988 		break;
989 	default:
990 		mpl_dbg(sdata, "Mesh plink: unknown frame subtype\n");
991 		break;
992 	}
993 
994 out:
995 	return event;
996 }
997 
998 static void
999 mesh_process_plink_frame(struct ieee80211_sub_if_data *sdata,
1000 			 struct ieee80211_mgmt *mgmt,
1001 			 struct ieee802_11_elems *elems)
1002 {
1003 
1004 	struct sta_info *sta;
1005 	enum plink_event event;
1006 	enum ieee80211_self_protected_actioncode ftype;
1007 	u32 changed = 0;
1008 	u8 ie_len = elems->peering_len;
1009 	u16 plid, llid = 0;
1010 
1011 	if (!elems->peering) {
1012 		mpl_dbg(sdata,
1013 			"Mesh plink: missing necessary peer link ie\n");
1014 		return;
1015 	}
1016 
1017 	if (elems->rsn_len &&
1018 	    sdata->u.mesh.security == IEEE80211_MESH_SEC_NONE) {
1019 		mpl_dbg(sdata,
1020 			"Mesh plink: can't establish link with secure peer\n");
1021 		return;
1022 	}
1023 
1024 	ftype = mgmt->u.action.u.self_prot.action_code;
1025 	if ((ftype == WLAN_SP_MESH_PEERING_OPEN && ie_len != 4) ||
1026 	    (ftype == WLAN_SP_MESH_PEERING_CONFIRM && ie_len != 6) ||
1027 	    (ftype == WLAN_SP_MESH_PEERING_CLOSE && ie_len != 6
1028 							&& ie_len != 8)) {
1029 		mpl_dbg(sdata,
1030 			"Mesh plink: incorrect plink ie length %d %d\n",
1031 			ftype, ie_len);
1032 		return;
1033 	}
1034 
1035 	if (ftype != WLAN_SP_MESH_PEERING_CLOSE &&
1036 	    (!elems->mesh_id || !elems->mesh_config)) {
1037 		mpl_dbg(sdata, "Mesh plink: missing necessary ie\n");
1038 		return;
1039 	}
1040 	/* Note the lines below are correct, the llid in the frame is the plid
1041 	 * from the point of view of this host.
1042 	 */
1043 	plid = get_unaligned_le16(PLINK_GET_LLID(elems->peering));
1044 	if (ftype == WLAN_SP_MESH_PEERING_CONFIRM ||
1045 	    (ftype == WLAN_SP_MESH_PEERING_CLOSE && ie_len == 8))
1046 		llid = get_unaligned_le16(PLINK_GET_PLID(elems->peering));
1047 
1048 	/* WARNING: Only for sta pointer, is dropped & re-acquired */
1049 	rcu_read_lock();
1050 
1051 	sta = sta_info_get(sdata, mgmt->sa);
1052 
1053 	if (ftype == WLAN_SP_MESH_PEERING_OPEN &&
1054 	    !rssi_threshold_check(sdata, sta)) {
1055 		mpl_dbg(sdata, "Mesh plink: %pM does not meet rssi threshold\n",
1056 			mgmt->sa);
1057 		goto unlock_rcu;
1058 	}
1059 
1060 	/* Now we will figure out the appropriate event... */
1061 	event = mesh_plink_get_event(sdata, sta, elems, ftype, llid, plid);
1062 
1063 	if (event == OPN_ACPT) {
1064 		rcu_read_unlock();
1065 		/* allocate sta entry if necessary and update info */
1066 		sta = mesh_sta_info_get(sdata, mgmt->sa, elems);
1067 		if (!sta) {
1068 			mpl_dbg(sdata, "Mesh plink: failed to init peer!\n");
1069 			goto unlock_rcu;
1070 		}
1071 		sta->plid = plid;
1072 	} else if (!sta && event == OPN_RJCT) {
1073 		mesh_plink_frame_tx(sdata, WLAN_SP_MESH_PEERING_CLOSE,
1074 				    mgmt->sa, 0, plid,
1075 				    WLAN_REASON_MESH_CONFIG);
1076 		goto unlock_rcu;
1077 	} else if (!sta || event == PLINK_UNDEFINED) {
1078 		/* something went wrong */
1079 		goto unlock_rcu;
1080 	}
1081 
1082 	/* 802.11-2012 13.3.7.2 - update plid on CNF if not set */
1083 	if (!sta->plid && event == CNF_ACPT)
1084 		sta->plid = plid;
1085 
1086 	changed |= mesh_plink_fsm(sdata, sta, event);
1087 
1088 unlock_rcu:
1089 	rcu_read_unlock();
1090 
1091 	if (changed)
1092 		ieee80211_mbss_info_change_notify(sdata, changed);
1093 }
1094 
1095 void mesh_rx_plink_frame(struct ieee80211_sub_if_data *sdata,
1096 			 struct ieee80211_mgmt *mgmt, size_t len,
1097 			 struct ieee80211_rx_status *rx_status)
1098 {
1099 	struct ieee802_11_elems elems;
1100 	size_t baselen;
1101 	u8 *baseaddr;
1102 
1103 	/* need action_code, aux */
1104 	if (len < IEEE80211_MIN_ACTION_SIZE + 3)
1105 		return;
1106 
1107 	if (sdata->u.mesh.user_mpm)
1108 		/* userspace must register for these */
1109 		return;
1110 
1111 	if (is_multicast_ether_addr(mgmt->da)) {
1112 		mpl_dbg(sdata,
1113 			"Mesh plink: ignore frame from multicast address\n");
1114 		return;
1115 	}
1116 
1117 	baseaddr = mgmt->u.action.u.self_prot.variable;
1118 	baselen = (u8 *) mgmt->u.action.u.self_prot.variable - (u8 *) mgmt;
1119 	if (mgmt->u.action.u.self_prot.action_code ==
1120 						WLAN_SP_MESH_PEERING_CONFIRM) {
1121 		baseaddr += 4;
1122 		baselen += 4;
1123 	}
1124 	ieee802_11_parse_elems(baseaddr, len - baselen, true, &elems);
1125 	mesh_process_plink_frame(sdata, mgmt, &elems);
1126 }
1127