net: mvneta: Fix an out of bounds check
[platform/kernel/linux-starfive.git] / net / mac80211 / wme.c
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
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  */
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
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 */
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
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. */
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  */
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 }