1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright 2004, Instant802 Networks, Inc.
4  * Copyright 2013-2014  Intel Mobile Communications GmbH
5  * Copyright (C) 2022 Intel Corporation
6  */
7 
8 #include <linux/netdevice.h>
9 #include <linux/skbuff.h>
10 #include <linux/module.h>
11 #include <linux/if_arp.h>
12 #include <linux/types.h>
13 #include <net/ip.h>
14 #include <net/pkt_sched.h>
15 
16 #include <net/mac80211.h>
17 #include "ieee80211_i.h"
18 #include "wme.h"
19 
20 /* Default mapping in classifier to work with default
21  * queue setup.
22  */
23 const int ieee802_1d_to_ac[8] = {
24 	IEEE80211_AC_BE,
25 	IEEE80211_AC_BK,
26 	IEEE80211_AC_BK,
27 	IEEE80211_AC_BE,
28 	IEEE80211_AC_VI,
29 	IEEE80211_AC_VI,
30 	IEEE80211_AC_VO,
31 	IEEE80211_AC_VO
32 };
33 
wme_downgrade_ac(struct sk_buff * skb)34 static int wme_downgrade_ac(struct sk_buff *skb)
35 {
36 	switch (skb->priority) {
37 	case 6:
38 	case 7:
39 		skb->priority = 5; /* VO -> VI */
40 		return 0;
41 	case 4:
42 	case 5:
43 		skb->priority = 3; /* VI -> BE */
44 		return 0;
45 	case 0:
46 	case 3:
47 		skb->priority = 2; /* BE -> BK */
48 		return 0;
49 	default:
50 		return -1;
51 	}
52 }
53 
54 /**
55  * ieee80211_fix_reserved_tid - return the TID to use if this one is reserved
56  * @tid: the assumed-reserved TID
57  *
58  * Returns: the alternative TID to use, or 0 on error
59  */
ieee80211_fix_reserved_tid(u8 tid)60 static inline u8 ieee80211_fix_reserved_tid(u8 tid)
61 {
62 	switch (tid) {
63 	case 0:
64 		return 3;
65 	case 1:
66 		return 2;
67 	case 2:
68 		return 1;
69 	case 3:
70 		return 0;
71 	case 4:
72 		return 5;
73 	case 5:
74 		return 4;
75 	case 6:
76 		return 7;
77 	case 7:
78 		return 6;
79 	}
80 
81 	return 0;
82 }
83 
ieee80211_downgrade_queue(struct ieee80211_sub_if_data * sdata,struct sta_info * sta,struct sk_buff * skb)84 static u16 ieee80211_downgrade_queue(struct ieee80211_sub_if_data *sdata,
85 				     struct sta_info *sta, struct sk_buff *skb)
86 {
87 	struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
88 
89 	/* in case we are a client verify acm is not set for this ac */
90 	while (sdata->wmm_acm & BIT(skb->priority)) {
91 		int ac = ieee802_1d_to_ac[skb->priority];
92 
93 		if (ifmgd->tx_tspec[ac].admitted_time &&
94 		    skb->priority == ifmgd->tx_tspec[ac].up)
95 			return ac;
96 
97 		if (wme_downgrade_ac(skb)) {
98 			/*
99 			 * This should not really happen. The AP has marked all
100 			 * lower ACs to require admission control which is not
101 			 * a reasonable configuration. Allow the frame to be
102 			 * transmitted using AC_BK as a workaround.
103 			 */
104 			break;
105 		}
106 	}
107 
108 	/* Check to see if this is a reserved TID */
109 	if (sta && sta->reserved_tid == skb->priority)
110 		skb->priority = ieee80211_fix_reserved_tid(skb->priority);
111 
112 	/* look up which queue to use for frames with this 1d tag */
113 	return ieee802_1d_to_ac[skb->priority];
114 }
115 
116 /* Indicate which queue to use for this fully formed 802.11 frame */
ieee80211_select_queue_80211(struct ieee80211_sub_if_data * sdata,struct sk_buff * skb,struct ieee80211_hdr * hdr)117 u16 ieee80211_select_queue_80211(struct ieee80211_sub_if_data *sdata,
118 				 struct sk_buff *skb,
119 				 struct ieee80211_hdr *hdr)
120 {
121 	struct ieee80211_local *local = sdata->local;
122 	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
123 	u8 *p;
124 
125 	if ((info->control.flags & IEEE80211_TX_CTRL_DONT_REORDER) ||
126 	    local->hw.queues < IEEE80211_NUM_ACS)
127 		return 0;
128 
129 	if (!ieee80211_is_data(hdr->frame_control)) {
130 		skb->priority = 7;
131 		return ieee802_1d_to_ac[skb->priority];
132 	}
133 	if (!ieee80211_is_data_qos(hdr->frame_control)) {
134 		skb->priority = 0;
135 		return ieee802_1d_to_ac[skb->priority];
136 	}
137 
138 	p = ieee80211_get_qos_ctl(hdr);
139 	skb->priority = *p & IEEE80211_QOS_CTL_TAG1D_MASK;
140 
141 	return ieee80211_downgrade_queue(sdata, NULL, skb);
142 }
143 
__ieee80211_select_queue(struct ieee80211_sub_if_data * sdata,struct sta_info * sta,struct sk_buff * skb)144 u16 __ieee80211_select_queue(struct ieee80211_sub_if_data *sdata,
145 			     struct sta_info *sta, struct sk_buff *skb)
146 {
147 	struct mac80211_qos_map *qos_map;
148 	bool qos;
149 
150 	/* all mesh/ocb stations are required to support WME */
151 	if (sta && (sdata->vif.type == NL80211_IFTYPE_MESH_POINT ||
152 		    sdata->vif.type == NL80211_IFTYPE_OCB))
153 		qos = true;
154 	else if (sta)
155 		qos = sta->sta.wme;
156 	else
157 		qos = false;
158 
159 	if (!qos) {
160 		skb->priority = 0; /* required for correct WPA/11i MIC */
161 		return IEEE80211_AC_BE;
162 	}
163 
164 	if (skb->protocol == sdata->control_port_protocol) {
165 		skb->priority = 7;
166 		goto downgrade;
167 	}
168 
169 	/* use the data classifier to determine what 802.1d tag the
170 	 * data frame has */
171 	qos_map = rcu_dereference(sdata->qos_map);
172 	skb->priority = cfg80211_classify8021d(skb, qos_map ?
173 					       &qos_map->qos_map : NULL);
174 
175  downgrade:
176 	return ieee80211_downgrade_queue(sdata, sta, skb);
177 }
178 
179 
180 /* Indicate which queue to use. */
ieee80211_select_queue(struct ieee80211_sub_if_data * sdata,struct sk_buff * skb)181 u16 ieee80211_select_queue(struct ieee80211_sub_if_data *sdata,
182 			   struct sk_buff *skb)
183 {
184 	struct ieee80211_local *local = sdata->local;
185 	struct sta_info *sta = NULL;
186 	const u8 *ra = NULL;
187 	u16 ret;
188 
189 	/* when using iTXQ, we can do this later */
190 	if (local->ops->wake_tx_queue)
191 		return 0;
192 
193 	if (local->hw.queues < IEEE80211_NUM_ACS || skb->len < 6) {
194 		skb->priority = 0; /* required for correct WPA/11i MIC */
195 		return 0;
196 	}
197 
198 	rcu_read_lock();
199 	switch (sdata->vif.type) {
200 	case NL80211_IFTYPE_AP_VLAN:
201 		sta = rcu_dereference(sdata->u.vlan.sta);
202 		if (sta)
203 			break;
204 		fallthrough;
205 	case NL80211_IFTYPE_AP:
206 		ra = skb->data;
207 		break;
208 	case NL80211_IFTYPE_STATION:
209 		/* might be a TDLS station */
210 		sta = sta_info_get(sdata, skb->data);
211 		if (sta)
212 			break;
213 
214 		ra = sdata->deflink.u.mgd.bssid;
215 		break;
216 	case NL80211_IFTYPE_ADHOC:
217 		ra = skb->data;
218 		break;
219 	default:
220 		break;
221 	}
222 
223 	if (!sta && ra && !is_multicast_ether_addr(ra))
224 		sta = sta_info_get(sdata, ra);
225 
226 	ret = __ieee80211_select_queue(sdata, sta, skb);
227 
228 	rcu_read_unlock();
229 	return ret;
230 }
231 
232 /**
233  * ieee80211_set_qos_hdr - Fill in the QoS header if there is one.
234  *
235  * @sdata: local subif
236  * @skb: packet to be updated
237  */
ieee80211_set_qos_hdr(struct ieee80211_sub_if_data * sdata,struct sk_buff * skb)238 void ieee80211_set_qos_hdr(struct ieee80211_sub_if_data *sdata,
239 			   struct sk_buff *skb)
240 {
241 	struct ieee80211_hdr *hdr = (void *)skb->data;
242 	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
243 	u8 tid = skb->priority & IEEE80211_QOS_CTL_TAG1D_MASK;
244 	u8 flags;
245 	u8 *p;
246 
247 	if (!ieee80211_is_data_qos(hdr->frame_control))
248 		return;
249 
250 	p = ieee80211_get_qos_ctl(hdr);
251 
252 	/* don't overwrite the QoS field of injected frames */
253 	if (info->flags & IEEE80211_TX_CTL_INJECTED) {
254 		/* do take into account Ack policy of injected frames */
255 		if (*p & IEEE80211_QOS_CTL_ACK_POLICY_NOACK)
256 			info->flags |= IEEE80211_TX_CTL_NO_ACK;
257 		return;
258 	}
259 
260 	/* set up the first byte */
261 
262 	/*
263 	 * preserve everything but the TID and ACK policy
264 	 * (which we both write here)
265 	 */
266 	flags = *p & ~(IEEE80211_QOS_CTL_TID_MASK |
267 		       IEEE80211_QOS_CTL_ACK_POLICY_MASK);
268 
269 	if (is_multicast_ether_addr(hdr->addr1) ||
270 	    sdata->noack_map & BIT(tid)) {
271 		flags |= IEEE80211_QOS_CTL_ACK_POLICY_NOACK;
272 		info->flags |= IEEE80211_TX_CTL_NO_ACK;
273 	}
274 
275 	*p = flags | tid;
276 
277 	/* set up the second byte */
278 	p++;
279 
280 	if (ieee80211_vif_is_mesh(&sdata->vif)) {
281 		/* preserve RSPI and Mesh PS Level bit */
282 		*p &= ((IEEE80211_QOS_CTL_RSPI |
283 			IEEE80211_QOS_CTL_MESH_PS_LEVEL) >> 8);
284 
285 		/* Nulls don't have a mesh header (frame body) */
286 		if (!ieee80211_is_qos_nullfunc(hdr->frame_control))
287 			*p |= (IEEE80211_QOS_CTL_MESH_CONTROL_PRESENT >> 8);
288 	} else {
289 		*p = 0;
290 	}
291 }
292