xref: /linux/drivers/net/wireless/realtek/rtw89/cam.c (revision 1a9239bb4253f9076b5b4b2a1a4e8d7defd77a95)
1 // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2 /* Copyright(c) 2019-2020  Realtek Corporation
3  */
4 
5 #include "cam.h"
6 #include "debug.h"
7 #include "fw.h"
8 #include "mac.h"
9 
10 static struct sk_buff *
rtw89_cam_get_sec_key_cmd(struct rtw89_dev * rtwdev,struct rtw89_sec_cam_entry * sec_cam,bool ext_key)11 rtw89_cam_get_sec_key_cmd(struct rtw89_dev *rtwdev,
12 			  struct rtw89_sec_cam_entry *sec_cam,
13 			  bool ext_key)
14 {
15 	struct sk_buff *skb;
16 	u32 cmd_len = H2C_SEC_CAM_LEN;
17 	u32 key32[4];
18 	u8 *cmd;
19 	int i, j;
20 
21 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, cmd_len);
22 	if (!skb)
23 		return NULL;
24 
25 	skb_put_zero(skb, cmd_len);
26 
27 	for (i = 0; i < 4; i++) {
28 		j = i * 4;
29 		j += ext_key ? 16 : 0;
30 		key32[i] = FIELD_PREP(GENMASK(7, 0), sec_cam->key[j + 0]) |
31 			   FIELD_PREP(GENMASK(15, 8), sec_cam->key[j + 1]) |
32 			   FIELD_PREP(GENMASK(23, 16), sec_cam->key[j + 2]) |
33 			   FIELD_PREP(GENMASK(31, 24), sec_cam->key[j + 3]);
34 	}
35 
36 	cmd = skb->data;
37 	RTW89_SET_FWCMD_SEC_IDX(cmd, sec_cam->sec_cam_idx + (ext_key ? 1 : 0));
38 	RTW89_SET_FWCMD_SEC_OFFSET(cmd, sec_cam->offset);
39 	RTW89_SET_FWCMD_SEC_LEN(cmd, sec_cam->len);
40 	RTW89_SET_FWCMD_SEC_TYPE(cmd, sec_cam->type);
41 	RTW89_SET_FWCMD_SEC_EXT_KEY(cmd, ext_key);
42 	RTW89_SET_FWCMD_SEC_SPP_MODE(cmd, sec_cam->spp_mode);
43 	RTW89_SET_FWCMD_SEC_KEY0(cmd, key32[0]);
44 	RTW89_SET_FWCMD_SEC_KEY1(cmd, key32[1]);
45 	RTW89_SET_FWCMD_SEC_KEY2(cmd, key32[2]);
46 	RTW89_SET_FWCMD_SEC_KEY3(cmd, key32[3]);
47 
48 	return skb;
49 }
50 
rtw89_cam_send_sec_key_cmd(struct rtw89_dev * rtwdev,struct rtw89_sec_cam_entry * sec_cam)51 static int rtw89_cam_send_sec_key_cmd(struct rtw89_dev *rtwdev,
52 				      struct rtw89_sec_cam_entry *sec_cam)
53 {
54 	struct sk_buff *skb, *ext_skb;
55 	int ret;
56 
57 	skb = rtw89_cam_get_sec_key_cmd(rtwdev, sec_cam, false);
58 	if (!skb) {
59 		rtw89_err(rtwdev, "failed to get sec key command\n");
60 		return -ENOMEM;
61 	}
62 
63 	rtw89_h2c_pkt_set_hdr(rtwdev, skb,
64 			      FWCMD_TYPE_H2C,
65 			      H2C_CAT_MAC,
66 			      H2C_CL_MAC_SEC_CAM,
67 			      H2C_FUNC_MAC_SEC_UPD, 1, 0,
68 			      H2C_SEC_CAM_LEN);
69 	ret = rtw89_h2c_tx(rtwdev, skb, false);
70 	if (ret) {
71 		rtw89_err(rtwdev, "failed to send sec key h2c: %d\n", ret);
72 		dev_kfree_skb(skb);
73 		return ret;
74 	}
75 
76 	if (!sec_cam->ext_key)
77 		return 0;
78 
79 	ext_skb = rtw89_cam_get_sec_key_cmd(rtwdev, sec_cam, true);
80 	if (!ext_skb) {
81 		rtw89_err(rtwdev, "failed to get ext sec key command\n");
82 		return -ENOMEM;
83 	}
84 
85 	rtw89_h2c_pkt_set_hdr(rtwdev, ext_skb,
86 			      FWCMD_TYPE_H2C,
87 			      H2C_CAT_MAC,
88 			      H2C_CL_MAC_SEC_CAM,
89 			      H2C_FUNC_MAC_SEC_UPD,
90 			      1, 0, H2C_SEC_CAM_LEN);
91 	ret = rtw89_h2c_tx(rtwdev, ext_skb, false);
92 	if (ret) {
93 		rtw89_err(rtwdev, "failed to send ext sec key h2c: %d\n", ret);
94 		dev_kfree_skb(ext_skb);
95 		return ret;
96 	}
97 
98 	return 0;
99 }
100 
rtw89_cam_get_avail_sec_cam(struct rtw89_dev * rtwdev,u8 * sec_cam_idx,bool ext_key)101 static int rtw89_cam_get_avail_sec_cam(struct rtw89_dev *rtwdev,
102 				       u8 *sec_cam_idx, bool ext_key)
103 {
104 	const struct rtw89_chip_info *chip = rtwdev->chip;
105 	struct rtw89_cam_info *cam_info = &rtwdev->cam_info;
106 	u8 sec_cam_num = chip->scam_num;
107 	u8 idx = 0;
108 
109 	if (!ext_key) {
110 		idx = find_first_zero_bit(cam_info->sec_cam_map, sec_cam_num);
111 		if (idx >= sec_cam_num)
112 			return -EBUSY;
113 
114 		set_bit(idx, cam_info->sec_cam_map);
115 		*sec_cam_idx = idx;
116 
117 		return 0;
118 	}
119 
120 again:
121 	idx = find_next_zero_bit(cam_info->sec_cam_map, sec_cam_num, idx);
122 	if (idx >= sec_cam_num - 1)
123 		return -EBUSY;
124 	/* ext keys need two cam entries for 256-bit key */
125 	if (test_bit(idx + 1, cam_info->sec_cam_map)) {
126 		idx++;
127 		goto again;
128 	}
129 
130 	set_bit(idx, cam_info->sec_cam_map);
131 	set_bit(idx + 1, cam_info->sec_cam_map);
132 	*sec_cam_idx = idx;
133 
134 	return 0;
135 }
136 
rtw89_cam_get_addr_cam_key_idx(struct rtw89_addr_cam_entry * addr_cam,const struct rtw89_sec_cam_entry * sec_cam,const struct ieee80211_key_conf * key,u8 * key_idx)137 static int rtw89_cam_get_addr_cam_key_idx(struct rtw89_addr_cam_entry *addr_cam,
138 					  const struct rtw89_sec_cam_entry *sec_cam,
139 					  const struct ieee80211_key_conf *key,
140 					  u8 *key_idx)
141 {
142 	u8 idx;
143 
144 	/* RTW89_ADDR_CAM_SEC_NONE	: not enabled
145 	 * RTW89_ADDR_CAM_SEC_ALL_UNI	: 0 - 6 unicast
146 	 * RTW89_ADDR_CAM_SEC_NORMAL	: 0 - 1 unicast, 2 - 4 group, 5 - 6 BIP
147 	 * RTW89_ADDR_CAM_SEC_4GROUP	: 0 - 1 unicast, 2 - 5 group, 6 BIP
148 	 */
149 	switch (addr_cam->sec_ent_mode) {
150 	case RTW89_ADDR_CAM_SEC_NONE:
151 		return -EINVAL;
152 	case RTW89_ADDR_CAM_SEC_ALL_UNI:
153 		idx = find_first_zero_bit(addr_cam->sec_cam_map,
154 					  RTW89_SEC_CAM_IN_ADDR_CAM);
155 		if (idx >= RTW89_SEC_CAM_IN_ADDR_CAM)
156 			return -EBUSY;
157 		*key_idx = idx;
158 		break;
159 	case RTW89_ADDR_CAM_SEC_NORMAL:
160 		if (sec_cam->type == RTW89_SEC_KEY_TYPE_BIP_CCMP128) {
161 			idx = find_next_zero_bit(addr_cam->sec_cam_map,
162 						 RTW89_SEC_CAM_IN_ADDR_CAM, 5);
163 			if (idx > 6)
164 				return -EBUSY;
165 			*key_idx = idx;
166 			break;
167 		}
168 
169 		if (key->flags & IEEE80211_KEY_FLAG_PAIRWISE) {
170 			idx = find_next_zero_bit(addr_cam->sec_cam_map,
171 						 RTW89_SEC_CAM_IN_ADDR_CAM, 0);
172 			if (idx > 1)
173 				return -EBUSY;
174 			*key_idx = idx;
175 			break;
176 		}
177 
178 		/* Group keys */
179 		idx = find_next_zero_bit(addr_cam->sec_cam_map,
180 					 RTW89_SEC_CAM_IN_ADDR_CAM, 2);
181 		if (idx > 4)
182 			return -EBUSY;
183 		*key_idx = idx;
184 		break;
185 	case RTW89_ADDR_CAM_SEC_4GROUP:
186 		if (sec_cam->type == RTW89_SEC_KEY_TYPE_BIP_CCMP128) {
187 			if (test_bit(6, addr_cam->sec_cam_map))
188 				return -EINVAL;
189 			*key_idx = 6;
190 			break;
191 		}
192 
193 		if (key->flags & IEEE80211_KEY_FLAG_PAIRWISE) {
194 			idx = find_next_zero_bit(addr_cam->sec_cam_map,
195 						 RTW89_SEC_CAM_IN_ADDR_CAM, 0);
196 			if (idx > 1)
197 				return -EBUSY;
198 			*key_idx = idx;
199 			break;
200 		}
201 
202 		/* Group keys */
203 		idx = find_next_zero_bit(addr_cam->sec_cam_map,
204 					 RTW89_SEC_CAM_IN_ADDR_CAM, 2);
205 		if (idx > 5)
206 			return -EBUSY;
207 		*key_idx = idx;
208 		break;
209 	}
210 
211 	return 0;
212 }
213 
__rtw89_cam_detach_sec_cam(struct rtw89_dev * rtwdev,struct rtw89_vif_link * rtwvif_link,struct rtw89_sta_link * rtwsta_link,const struct rtw89_sec_cam_entry * sec_cam,bool inform_fw)214 static int __rtw89_cam_detach_sec_cam(struct rtw89_dev *rtwdev,
215 				      struct rtw89_vif_link *rtwvif_link,
216 				      struct rtw89_sta_link *rtwsta_link,
217 				      const struct rtw89_sec_cam_entry *sec_cam,
218 				      bool inform_fw)
219 {
220 	struct rtw89_addr_cam_entry *addr_cam;
221 	unsigned int i;
222 	int ret = 0;
223 
224 	addr_cam = rtw89_get_addr_cam_of(rtwvif_link, rtwsta_link);
225 
226 	for_each_set_bit(i, addr_cam->sec_cam_map, RTW89_SEC_CAM_IN_ADDR_CAM) {
227 		if (addr_cam->sec_ent[i] != sec_cam->sec_cam_idx)
228 			continue;
229 
230 		clear_bit(i, addr_cam->sec_cam_map);
231 	}
232 
233 	if (inform_fw) {
234 		ret = rtw89_chip_h2c_dctl_sec_cam(rtwdev, rtwvif_link, rtwsta_link);
235 		if (ret)
236 			rtw89_err(rtwdev,
237 				  "failed to update dctl cam del key: %d\n", ret);
238 		ret = rtw89_fw_h2c_cam(rtwdev, rtwvif_link, rtwsta_link, NULL);
239 		if (ret)
240 			rtw89_err(rtwdev, "failed to update cam del key: %d\n", ret);
241 	}
242 
243 	return ret;
244 }
245 
__rtw89_cam_attach_sec_cam(struct rtw89_dev * rtwdev,struct rtw89_vif_link * rtwvif_link,struct rtw89_sta_link * rtwsta_link,const struct ieee80211_key_conf * key,const struct rtw89_sec_cam_entry * sec_cam)246 static int __rtw89_cam_attach_sec_cam(struct rtw89_dev *rtwdev,
247 				      struct rtw89_vif_link *rtwvif_link,
248 				      struct rtw89_sta_link *rtwsta_link,
249 				      const struct ieee80211_key_conf *key,
250 				      const struct rtw89_sec_cam_entry *sec_cam)
251 {
252 	struct rtw89_addr_cam_entry *addr_cam;
253 	u8 key_idx = 0;
254 	int ret;
255 
256 	addr_cam = rtw89_get_addr_cam_of(rtwvif_link, rtwsta_link);
257 
258 	if (key->cipher == WLAN_CIPHER_SUITE_WEP40 ||
259 	    key->cipher == WLAN_CIPHER_SUITE_WEP104)
260 		addr_cam->sec_ent_mode = RTW89_ADDR_CAM_SEC_ALL_UNI;
261 
262 	ret = rtw89_cam_get_addr_cam_key_idx(addr_cam, sec_cam, key, &key_idx);
263 	if (ret) {
264 		rtw89_err(rtwdev, "failed to get addr cam key idx %d, %d\n",
265 			  addr_cam->sec_ent_mode, sec_cam->type);
266 		return ret;
267 	}
268 
269 	addr_cam->sec_ent_keyid[key_idx] = key->keyidx;
270 	addr_cam->sec_ent[key_idx] = sec_cam->sec_cam_idx;
271 	set_bit(key_idx, addr_cam->sec_cam_map);
272 	ret = rtw89_chip_h2c_dctl_sec_cam(rtwdev, rtwvif_link, rtwsta_link);
273 	if (ret) {
274 		rtw89_err(rtwdev, "failed to update dctl cam sec entry: %d\n",
275 			  ret);
276 		return ret;
277 	}
278 	ret = rtw89_fw_h2c_cam(rtwdev, rtwvif_link, rtwsta_link, NULL);
279 	if (ret) {
280 		rtw89_err(rtwdev, "failed to update addr cam sec entry: %d\n",
281 			  ret);
282 		clear_bit(key_idx, addr_cam->sec_cam_map);
283 		return ret;
284 	}
285 
286 	return 0;
287 }
288 
rtw89_cam_attach_link_sec_cam(struct rtw89_dev * rtwdev,struct rtw89_vif_link * rtwvif_link,struct rtw89_sta_link * rtwsta_link,u8 sec_cam_idx)289 int rtw89_cam_attach_link_sec_cam(struct rtw89_dev *rtwdev,
290 				  struct rtw89_vif_link *rtwvif_link,
291 				  struct rtw89_sta_link *rtwsta_link,
292 				  u8 sec_cam_idx)
293 {
294 	struct rtw89_cam_info *cam_info = &rtwdev->cam_info;
295 	const struct rtw89_sec_cam_entry *sec_cam;
296 
297 	sec_cam = cam_info->sec_entries[sec_cam_idx];
298 	if (!sec_cam)
299 		return -ENOENT;
300 
301 	return __rtw89_cam_attach_sec_cam(rtwdev, rtwvif_link, rtwsta_link,
302 					  sec_cam->key_conf, sec_cam);
303 }
304 
rtw89_cam_detach_sec_cam(struct rtw89_dev * rtwdev,struct ieee80211_vif * vif,struct ieee80211_sta * sta,const struct rtw89_sec_cam_entry * sec_cam,bool inform_fw)305 static int rtw89_cam_detach_sec_cam(struct rtw89_dev *rtwdev,
306 				    struct ieee80211_vif *vif,
307 				    struct ieee80211_sta *sta,
308 				    const struct rtw89_sec_cam_entry *sec_cam,
309 				    bool inform_fw)
310 {
311 	struct rtw89_sta *rtwsta = sta_to_rtwsta_safe(sta);
312 	struct rtw89_sta_link *rtwsta_link;
313 	struct rtw89_vif_link *rtwvif_link;
314 	struct rtw89_vif *rtwvif;
315 	unsigned int link_id;
316 	int ret;
317 
318 	if (!vif) {
319 		rtw89_err(rtwdev, "No iface for deleting sec cam\n");
320 		return -EINVAL;
321 	}
322 
323 	rtwvif = vif_to_rtwvif(vif);
324 
325 	if (rtwsta)
326 		clear_bit(sec_cam->sec_cam_idx, rtwsta->pairwise_sec_cam_map);
327 
328 	rtw89_vif_for_each_link(rtwvif, rtwvif_link, link_id) {
329 		rtwsta_link = rtwsta ? rtwsta->links[link_id] : NULL;
330 		if (rtwsta && !rtwsta_link)
331 			continue;
332 
333 		ret = __rtw89_cam_detach_sec_cam(rtwdev, rtwvif_link, rtwsta_link,
334 						 sec_cam, inform_fw);
335 		if (ret)
336 			return ret;
337 	}
338 
339 	return 0;
340 }
341 
rtw89_cam_attach_sec_cam(struct rtw89_dev * rtwdev,struct ieee80211_vif * vif,struct ieee80211_sta * sta,struct ieee80211_key_conf * key,struct rtw89_sec_cam_entry * sec_cam)342 static int rtw89_cam_attach_sec_cam(struct rtw89_dev *rtwdev,
343 				    struct ieee80211_vif *vif,
344 				    struct ieee80211_sta *sta,
345 				    struct ieee80211_key_conf *key,
346 				    struct rtw89_sec_cam_entry *sec_cam)
347 {
348 	struct rtw89_sta *rtwsta = sta_to_rtwsta_safe(sta);
349 	struct rtw89_sta_link *rtwsta_link;
350 	struct rtw89_vif_link *rtwvif_link;
351 	struct rtw89_vif *rtwvif;
352 	unsigned int link_id;
353 	int key_link_id;
354 	int ret;
355 
356 	if (!vif) {
357 		rtw89_err(rtwdev, "No iface for adding sec cam\n");
358 		return -EINVAL;
359 	}
360 
361 	rtwvif = vif_to_rtwvif(vif);
362 
363 	key_link_id = ieee80211_vif_is_mld(vif) ? key->link_id : 0;
364 	if (key_link_id >= 0) {
365 		rtwvif_link = rtwvif->links[key_link_id];
366 		rtwsta_link = rtwsta ? rtwsta->links[key_link_id] : NULL;
367 
368 		if (!rtwvif_link || (rtwsta && !rtwsta_link)) {
369 			rtw89_err(rtwdev, "No drv link for adding sec cam\n");
370 			return -ENOLINK;
371 		}
372 
373 		return __rtw89_cam_attach_sec_cam(rtwdev, rtwvif_link,
374 						  rtwsta_link, key, sec_cam);
375 	}
376 
377 	/* key_link_id < 0: MLD pairwise key */
378 	if (!rtwsta) {
379 		rtw89_err(rtwdev, "No sta for adding MLD pairwise sec cam\n");
380 		return -EINVAL;
381 	}
382 
383 	rtw89_sta_for_each_link(rtwsta, rtwsta_link, link_id) {
384 		rtwvif_link = rtwsta_link->rtwvif_link;
385 		ret = __rtw89_cam_attach_sec_cam(rtwdev, rtwvif_link,
386 						 rtwsta_link, key, sec_cam);
387 		if (ret)
388 			return ret;
389 	}
390 
391 	set_bit(sec_cam->sec_cam_idx, rtwsta->pairwise_sec_cam_map);
392 
393 	return 0;
394 }
395 
rtw89_cam_sec_key_install(struct rtw89_dev * rtwdev,struct ieee80211_vif * vif,struct ieee80211_sta * sta,struct ieee80211_key_conf * key,u8 hw_key_type,bool ext_key)396 static int rtw89_cam_sec_key_install(struct rtw89_dev *rtwdev,
397 				     struct ieee80211_vif *vif,
398 				     struct ieee80211_sta *sta,
399 				     struct ieee80211_key_conf *key,
400 				     u8 hw_key_type, bool ext_key)
401 {
402 	struct rtw89_sec_cam_entry *sec_cam = NULL;
403 	struct rtw89_cam_info *cam_info = &rtwdev->cam_info;
404 	u8 sec_cam_idx;
405 	int ret;
406 
407 	/* maximum key length 256-bit */
408 	if (key->keylen > 32) {
409 		rtw89_err(rtwdev, "invalid sec key length %d\n", key->keylen);
410 		return -EINVAL;
411 	}
412 
413 	ret = rtw89_cam_get_avail_sec_cam(rtwdev, &sec_cam_idx, ext_key);
414 	if (ret) {
415 		rtw89_warn(rtwdev, "no available sec cam: %d ext: %d\n",
416 			   ret, ext_key);
417 		return ret;
418 	}
419 
420 	sec_cam = kzalloc(sizeof(*sec_cam), GFP_KERNEL);
421 	if (!sec_cam) {
422 		ret = -ENOMEM;
423 		goto err_release_cam;
424 	}
425 
426 	key->hw_key_idx = sec_cam_idx;
427 	cam_info->sec_entries[sec_cam_idx] = sec_cam;
428 
429 	sec_cam->sec_cam_idx = sec_cam_idx;
430 	sec_cam->type = hw_key_type;
431 	sec_cam->len = RTW89_SEC_CAM_LEN;
432 	sec_cam->ext_key = ext_key;
433 	memcpy(sec_cam->key, key->key, key->keylen);
434 
435 	sec_cam->key_conf = key;
436 
437 	ret = rtw89_cam_send_sec_key_cmd(rtwdev, sec_cam);
438 	if (ret) {
439 		rtw89_err(rtwdev, "failed to send sec key cmd: %d\n", ret);
440 		goto err_release_cam;
441 	}
442 
443 	/* associate with addr cam */
444 	ret = rtw89_cam_attach_sec_cam(rtwdev, vif, sta, key, sec_cam);
445 	if (ret) {
446 		rtw89_err(rtwdev, "failed to attach sec cam: %d\n", ret);
447 		goto err_release_cam;
448 	}
449 
450 	return 0;
451 
452 err_release_cam:
453 	cam_info->sec_entries[sec_cam_idx] = NULL;
454 	kfree(sec_cam);
455 	clear_bit(sec_cam_idx, cam_info->sec_cam_map);
456 	if (ext_key)
457 		clear_bit(sec_cam_idx + 1, cam_info->sec_cam_map);
458 
459 	return ret;
460 }
461 
rtw89_cam_sec_key_add(struct rtw89_dev * rtwdev,struct ieee80211_vif * vif,struct ieee80211_sta * sta,struct ieee80211_key_conf * key)462 int rtw89_cam_sec_key_add(struct rtw89_dev *rtwdev,
463 			  struct ieee80211_vif *vif,
464 			  struct ieee80211_sta *sta,
465 			  struct ieee80211_key_conf *key)
466 {
467 	const struct rtw89_chip_info *chip = rtwdev->chip;
468 	u8 hw_key_type;
469 	bool ext_key = false;
470 	int ret;
471 
472 	switch (key->cipher) {
473 	case WLAN_CIPHER_SUITE_WEP40:
474 		hw_key_type = RTW89_SEC_KEY_TYPE_WEP40;
475 		break;
476 	case WLAN_CIPHER_SUITE_WEP104:
477 		hw_key_type = RTW89_SEC_KEY_TYPE_WEP104;
478 		break;
479 	case WLAN_CIPHER_SUITE_TKIP:
480 		if (!chip->hw_tkip_crypto)
481 			return -EOPNOTSUPP;
482 		hw_key_type = RTW89_SEC_KEY_TYPE_TKIP;
483 		key->flags |= IEEE80211_KEY_FLAG_GENERATE_MMIC;
484 		break;
485 	case WLAN_CIPHER_SUITE_CCMP:
486 		hw_key_type = RTW89_SEC_KEY_TYPE_CCMP128;
487 		if (!chip->hw_mgmt_tx_encrypt)
488 			key->flags |= IEEE80211_KEY_FLAG_SW_MGMT_TX;
489 		break;
490 	case WLAN_CIPHER_SUITE_CCMP_256:
491 		hw_key_type = RTW89_SEC_KEY_TYPE_CCMP256;
492 		if (!chip->hw_mgmt_tx_encrypt)
493 			key->flags |= IEEE80211_KEY_FLAG_SW_MGMT_TX;
494 		ext_key = true;
495 		break;
496 	case WLAN_CIPHER_SUITE_GCMP:
497 		hw_key_type = RTW89_SEC_KEY_TYPE_GCMP128;
498 		if (!chip->hw_mgmt_tx_encrypt)
499 			key->flags |= IEEE80211_KEY_FLAG_SW_MGMT_TX;
500 		break;
501 	case WLAN_CIPHER_SUITE_GCMP_256:
502 		hw_key_type = RTW89_SEC_KEY_TYPE_GCMP256;
503 		if (!chip->hw_mgmt_tx_encrypt)
504 			key->flags |= IEEE80211_KEY_FLAG_SW_MGMT_TX;
505 		ext_key = true;
506 		break;
507 	case WLAN_CIPHER_SUITE_AES_CMAC:
508 		hw_key_type = RTW89_SEC_KEY_TYPE_BIP_CCMP128;
509 		break;
510 	default:
511 		return -EOPNOTSUPP;
512 	}
513 
514 	if (!chip->hw_sec_hdr)
515 		key->flags |= IEEE80211_KEY_FLAG_GENERATE_IV;
516 
517 	ret = rtw89_cam_sec_key_install(rtwdev, vif, sta, key, hw_key_type,
518 					ext_key);
519 	if (ret) {
520 		rtw89_err(rtwdev, "failed to install key type %d ext %d: %d\n",
521 			  hw_key_type, ext_key, ret);
522 		return ret;
523 	}
524 
525 	return 0;
526 }
527 
rtw89_cam_sec_key_del(struct rtw89_dev * rtwdev,struct ieee80211_vif * vif,struct ieee80211_sta * sta,struct ieee80211_key_conf * key,bool inform_fw)528 int rtw89_cam_sec_key_del(struct rtw89_dev *rtwdev,
529 			  struct ieee80211_vif *vif,
530 			  struct ieee80211_sta *sta,
531 			  struct ieee80211_key_conf *key,
532 			  bool inform_fw)
533 {
534 	struct rtw89_cam_info *cam_info = &rtwdev->cam_info;
535 	const struct rtw89_sec_cam_entry *sec_cam;
536 	u8 sec_cam_idx;
537 	int ret;
538 
539 	sec_cam_idx = key->hw_key_idx;
540 	sec_cam = cam_info->sec_entries[sec_cam_idx];
541 	if (!sec_cam)
542 		return -EINVAL;
543 
544 	ret = rtw89_cam_detach_sec_cam(rtwdev, vif, sta, sec_cam, inform_fw);
545 
546 	/* clear valid bit in addr cam will disable sec cam,
547 	 * so we don't need to send H2C command again
548 	 */
549 	cam_info->sec_entries[sec_cam_idx] = NULL;
550 	clear_bit(sec_cam_idx, cam_info->sec_cam_map);
551 	if (sec_cam->ext_key)
552 		clear_bit(sec_cam_idx + 1, cam_info->sec_cam_map);
553 
554 	kfree(sec_cam);
555 
556 	return ret;
557 }
558 
rtw89_cam_reset_key_iter(struct ieee80211_hw * hw,struct ieee80211_vif * vif,struct ieee80211_sta * sta,struct ieee80211_key_conf * key,void * data)559 static void rtw89_cam_reset_key_iter(struct ieee80211_hw *hw,
560 				     struct ieee80211_vif *vif,
561 				     struct ieee80211_sta *sta,
562 				     struct ieee80211_key_conf *key,
563 				     void *data)
564 {
565 	struct rtw89_dev *rtwdev = (struct rtw89_dev *)data;
566 
567 	rtw89_cam_sec_key_del(rtwdev, vif, sta, key, false);
568 }
569 
rtw89_cam_deinit_addr_cam(struct rtw89_dev * rtwdev,struct rtw89_addr_cam_entry * addr_cam)570 void rtw89_cam_deinit_addr_cam(struct rtw89_dev *rtwdev,
571 			       struct rtw89_addr_cam_entry *addr_cam)
572 {
573 	struct rtw89_cam_info *cam_info = &rtwdev->cam_info;
574 
575 	addr_cam->valid = false;
576 	clear_bit(addr_cam->addr_cam_idx, cam_info->addr_cam_map);
577 }
578 
rtw89_cam_deinit_bssid_cam(struct rtw89_dev * rtwdev,struct rtw89_bssid_cam_entry * bssid_cam)579 void rtw89_cam_deinit_bssid_cam(struct rtw89_dev *rtwdev,
580 				struct rtw89_bssid_cam_entry *bssid_cam)
581 {
582 	struct rtw89_cam_info *cam_info = &rtwdev->cam_info;
583 
584 	bssid_cam->valid = false;
585 	clear_bit(bssid_cam->bssid_cam_idx, cam_info->bssid_cam_map);
586 }
587 
rtw89_cam_deinit(struct rtw89_dev * rtwdev,struct rtw89_vif_link * rtwvif_link)588 void rtw89_cam_deinit(struct rtw89_dev *rtwdev, struct rtw89_vif_link *rtwvif_link)
589 {
590 	struct rtw89_addr_cam_entry *addr_cam = &rtwvif_link->addr_cam;
591 	struct rtw89_bssid_cam_entry *bssid_cam = &rtwvif_link->bssid_cam;
592 
593 	rtw89_cam_deinit_addr_cam(rtwdev, addr_cam);
594 	rtw89_cam_deinit_bssid_cam(rtwdev, bssid_cam);
595 }
596 
rtw89_cam_reset_keys(struct rtw89_dev * rtwdev)597 void rtw89_cam_reset_keys(struct rtw89_dev *rtwdev)
598 {
599 	rcu_read_lock();
600 	ieee80211_iter_keys_rcu(rtwdev->hw, NULL, rtw89_cam_reset_key_iter, rtwdev);
601 	rcu_read_unlock();
602 }
603 
rtw89_cam_get_avail_addr_cam(struct rtw89_dev * rtwdev,u8 * addr_cam_idx)604 static int rtw89_cam_get_avail_addr_cam(struct rtw89_dev *rtwdev,
605 					u8 *addr_cam_idx)
606 {
607 	const struct rtw89_chip_info *chip = rtwdev->chip;
608 	struct rtw89_cam_info *cam_info = &rtwdev->cam_info;
609 	u8 addr_cam_num = chip->acam_num;
610 	u8 idx;
611 
612 	idx = find_first_zero_bit(cam_info->addr_cam_map, addr_cam_num);
613 	if (idx >= addr_cam_num)
614 		return -EBUSY;
615 
616 	set_bit(idx, cam_info->addr_cam_map);
617 	*addr_cam_idx = idx;
618 
619 	return 0;
620 }
621 
rtw89_get_addr_cam_entry_size(struct rtw89_dev * rtwdev)622 static u8 rtw89_get_addr_cam_entry_size(struct rtw89_dev *rtwdev)
623 {
624 	const struct rtw89_chip_info *chip = rtwdev->chip;
625 
626 	switch (chip->chip_id) {
627 	case RTL8852A:
628 	case RTL8852B:
629 	case RTL8851B:
630 	case RTL8852BT:
631 		return ADDR_CAM_ENT_SIZE;
632 	default:
633 		return ADDR_CAM_ENT_SHORT_SIZE;
634 	}
635 }
636 
rtw89_cam_init_addr_cam(struct rtw89_dev * rtwdev,struct rtw89_addr_cam_entry * addr_cam,const struct rtw89_bssid_cam_entry * bssid_cam)637 int rtw89_cam_init_addr_cam(struct rtw89_dev *rtwdev,
638 			    struct rtw89_addr_cam_entry *addr_cam,
639 			    const struct rtw89_bssid_cam_entry *bssid_cam)
640 {
641 	u8 addr_cam_idx;
642 	int i;
643 	int ret;
644 
645 	if (unlikely(addr_cam->valid)) {
646 		rtw89_debug(rtwdev, RTW89_DBG_FW,
647 			    "addr cam is already valid; skip init\n");
648 		return 0;
649 	}
650 
651 	ret = rtw89_cam_get_avail_addr_cam(rtwdev, &addr_cam_idx);
652 	if (ret) {
653 		rtw89_err(rtwdev, "failed to get available addr cam\n");
654 		return ret;
655 	}
656 
657 	addr_cam->addr_cam_idx = addr_cam_idx;
658 	addr_cam->len = rtw89_get_addr_cam_entry_size(rtwdev);
659 	addr_cam->offset = 0;
660 	addr_cam->valid = true;
661 	addr_cam->addr_mask = 0;
662 	addr_cam->mask_sel = RTW89_NO_MSK;
663 	addr_cam->sec_ent_mode = RTW89_ADDR_CAM_SEC_NORMAL;
664 	bitmap_zero(addr_cam->sec_cam_map, RTW89_SEC_CAM_IN_ADDR_CAM);
665 
666 	for (i = 0; i < RTW89_SEC_CAM_IN_ADDR_CAM; i++) {
667 		addr_cam->sec_ent_keyid[i] = 0;
668 		addr_cam->sec_ent[i] = 0;
669 	}
670 
671 	/* associate addr cam with bssid cam */
672 	addr_cam->bssid_cam_idx = bssid_cam->bssid_cam_idx;
673 
674 	return 0;
675 }
676 
rtw89_cam_get_avail_bssid_cam(struct rtw89_dev * rtwdev,u8 * bssid_cam_idx)677 static int rtw89_cam_get_avail_bssid_cam(struct rtw89_dev *rtwdev,
678 					 u8 *bssid_cam_idx)
679 {
680 	const struct rtw89_chip_info *chip = rtwdev->chip;
681 	struct rtw89_cam_info *cam_info = &rtwdev->cam_info;
682 	u8 bssid_cam_num = chip->bcam_num;
683 	u8 idx;
684 
685 	idx = find_first_zero_bit(cam_info->bssid_cam_map, bssid_cam_num);
686 	if (idx >= bssid_cam_num)
687 		return -EBUSY;
688 
689 	set_bit(idx, cam_info->bssid_cam_map);
690 	*bssid_cam_idx = idx;
691 
692 	return 0;
693 }
694 
rtw89_cam_init_bssid_cam(struct rtw89_dev * rtwdev,struct rtw89_vif_link * rtwvif_link,struct rtw89_bssid_cam_entry * bssid_cam,const u8 * bssid)695 int rtw89_cam_init_bssid_cam(struct rtw89_dev *rtwdev,
696 			     struct rtw89_vif_link *rtwvif_link,
697 			     struct rtw89_bssid_cam_entry *bssid_cam,
698 			     const u8 *bssid)
699 {
700 	u8 bssid_cam_idx;
701 	int ret;
702 
703 	if (unlikely(bssid_cam->valid)) {
704 		rtw89_debug(rtwdev, RTW89_DBG_FW,
705 			    "bssid cam is already valid; skip init\n");
706 		return 0;
707 	}
708 
709 	ret = rtw89_cam_get_avail_bssid_cam(rtwdev, &bssid_cam_idx);
710 	if (ret) {
711 		rtw89_err(rtwdev, "failed to get available bssid cam\n");
712 		return ret;
713 	}
714 
715 	bssid_cam->bssid_cam_idx = bssid_cam_idx;
716 	bssid_cam->phy_idx = rtwvif_link->phy_idx;
717 	bssid_cam->len = BSSID_CAM_ENT_SIZE;
718 	bssid_cam->offset = 0;
719 	bssid_cam->valid = true;
720 	ether_addr_copy(bssid_cam->bssid, bssid);
721 
722 	return 0;
723 }
724 
rtw89_cam_bssid_changed(struct rtw89_dev * rtwdev,struct rtw89_vif_link * rtwvif_link)725 void rtw89_cam_bssid_changed(struct rtw89_dev *rtwdev, struct rtw89_vif_link *rtwvif_link)
726 {
727 	struct rtw89_bssid_cam_entry *bssid_cam = &rtwvif_link->bssid_cam;
728 
729 	ether_addr_copy(bssid_cam->bssid, rtwvif_link->bssid);
730 }
731 
rtw89_cam_init(struct rtw89_dev * rtwdev,struct rtw89_vif_link * rtwvif_link)732 int rtw89_cam_init(struct rtw89_dev *rtwdev, struct rtw89_vif_link *rtwvif_link)
733 {
734 	struct rtw89_addr_cam_entry *addr_cam = &rtwvif_link->addr_cam;
735 	struct rtw89_bssid_cam_entry *bssid_cam = &rtwvif_link->bssid_cam;
736 	int ret;
737 
738 	ret = rtw89_cam_init_bssid_cam(rtwdev, rtwvif_link, bssid_cam,
739 				       rtwvif_link->bssid);
740 	if (ret) {
741 		rtw89_err(rtwdev, "failed to init bssid cam\n");
742 		return ret;
743 	}
744 
745 	ret = rtw89_cam_init_addr_cam(rtwdev, addr_cam, bssid_cam);
746 	if (ret) {
747 		rtw89_err(rtwdev, "failed to init addr cam\n");
748 		return ret;
749 	}
750 
751 	return 0;
752 }
753 
rtw89_cam_fill_bssid_cam_info(struct rtw89_dev * rtwdev,struct rtw89_vif_link * rtwvif_link,struct rtw89_sta_link * rtwsta_link,u8 * cmd)754 int rtw89_cam_fill_bssid_cam_info(struct rtw89_dev *rtwdev,
755 				  struct rtw89_vif_link *rtwvif_link,
756 				  struct rtw89_sta_link *rtwsta_link, u8 *cmd)
757 {
758 	struct rtw89_bssid_cam_entry *bssid_cam = rtw89_get_bssid_cam_of(rtwvif_link,
759 									 rtwsta_link);
760 	struct ieee80211_bss_conf *bss_conf;
761 	u8 bss_color;
762 	u8 bss_mask;
763 
764 	rcu_read_lock();
765 
766 	bss_conf = rtw89_vif_rcu_dereference_link(rtwvif_link, false);
767 	bss_color = bss_conf->he_bss_color.color;
768 
769 	if (bss_conf->nontransmitted)
770 		bss_mask = RTW89_BSSID_MATCH_5_BYTES;
771 	else
772 		bss_mask = RTW89_BSSID_MATCH_ALL;
773 
774 	rcu_read_unlock();
775 
776 	FWCMD_SET_ADDR_BSSID_IDX(cmd, bssid_cam->bssid_cam_idx);
777 	FWCMD_SET_ADDR_BSSID_OFFSET(cmd, bssid_cam->offset);
778 	FWCMD_SET_ADDR_BSSID_LEN(cmd, bssid_cam->len);
779 	FWCMD_SET_ADDR_BSSID_VALID(cmd, bssid_cam->valid);
780 	FWCMD_SET_ADDR_BSSID_MASK(cmd, bss_mask);
781 	FWCMD_SET_ADDR_BSSID_BB_SEL(cmd, bssid_cam->phy_idx);
782 	FWCMD_SET_ADDR_BSSID_BSS_COLOR(cmd, bss_color);
783 
784 	FWCMD_SET_ADDR_BSSID_BSSID0(cmd, bssid_cam->bssid[0]);
785 	FWCMD_SET_ADDR_BSSID_BSSID1(cmd, bssid_cam->bssid[1]);
786 	FWCMD_SET_ADDR_BSSID_BSSID2(cmd, bssid_cam->bssid[2]);
787 	FWCMD_SET_ADDR_BSSID_BSSID3(cmd, bssid_cam->bssid[3]);
788 	FWCMD_SET_ADDR_BSSID_BSSID4(cmd, bssid_cam->bssid[4]);
789 	FWCMD_SET_ADDR_BSSID_BSSID5(cmd, bssid_cam->bssid[5]);
790 
791 	return 0;
792 }
793 
rtw89_cam_addr_hash(u8 start,const u8 * addr)794 static u8 rtw89_cam_addr_hash(u8 start, const u8 *addr)
795 {
796 	u8 hash = 0;
797 	u8 i;
798 
799 	for (i = start; i < ETH_ALEN; i++)
800 		hash ^= addr[i];
801 
802 	return hash;
803 }
804 
rtw89_cam_fill_addr_cam_info(struct rtw89_dev * rtwdev,struct rtw89_vif_link * rtwvif_link,struct rtw89_sta_link * rtwsta_link,const u8 * scan_mac_addr,u8 * cmd)805 void rtw89_cam_fill_addr_cam_info(struct rtw89_dev *rtwdev,
806 				  struct rtw89_vif_link *rtwvif_link,
807 				  struct rtw89_sta_link *rtwsta_link,
808 				  const u8 *scan_mac_addr,
809 				  u8 *cmd)
810 {
811 	struct ieee80211_vif *vif = rtwvif_link_to_vif(rtwvif_link);
812 	struct rtw89_addr_cam_entry *addr_cam =
813 		rtw89_get_addr_cam_of(rtwvif_link, rtwsta_link);
814 	struct ieee80211_sta *sta = rtwsta_link_to_sta_safe(rtwsta_link);
815 	struct ieee80211_link_sta *link_sta;
816 	const u8 *sma = scan_mac_addr ? scan_mac_addr : rtwvif_link->mac_addr;
817 	u8 sma_hash, tma_hash, addr_msk_start;
818 	u8 sma_start = 0;
819 	u8 tma_start = 0;
820 	const u8 *tma;
821 
822 	rcu_read_lock();
823 
824 	if (sta) {
825 		link_sta = rtw89_sta_rcu_dereference_link(rtwsta_link, true);
826 		tma = link_sta->addr;
827 	} else {
828 		tma = rtwvif_link->bssid;
829 	}
830 
831 	if (addr_cam->addr_mask != 0) {
832 		addr_msk_start = __ffs(addr_cam->addr_mask);
833 		if (addr_cam->mask_sel == RTW89_SMA)
834 			sma_start = addr_msk_start;
835 		else if (addr_cam->mask_sel == RTW89_TMA)
836 			tma_start = addr_msk_start;
837 	}
838 	sma_hash = rtw89_cam_addr_hash(sma_start, sma);
839 	tma_hash = rtw89_cam_addr_hash(tma_start, tma);
840 
841 	FWCMD_SET_ADDR_IDX(cmd, addr_cam->addr_cam_idx);
842 	FWCMD_SET_ADDR_OFFSET(cmd, addr_cam->offset);
843 	FWCMD_SET_ADDR_LEN(cmd, addr_cam->len);
844 
845 	FWCMD_SET_ADDR_VALID(cmd, addr_cam->valid);
846 	FWCMD_SET_ADDR_NET_TYPE(cmd, rtwvif_link->net_type);
847 	FWCMD_SET_ADDR_BCN_HIT_COND(cmd, rtwvif_link->bcn_hit_cond);
848 	FWCMD_SET_ADDR_HIT_RULE(cmd, rtwvif_link->hit_rule);
849 	FWCMD_SET_ADDR_BB_SEL(cmd, rtwvif_link->phy_idx);
850 	FWCMD_SET_ADDR_ADDR_MASK(cmd, addr_cam->addr_mask);
851 	FWCMD_SET_ADDR_MASK_SEL(cmd, addr_cam->mask_sel);
852 	FWCMD_SET_ADDR_SMA_HASH(cmd, sma_hash);
853 	FWCMD_SET_ADDR_TMA_HASH(cmd, tma_hash);
854 
855 	FWCMD_SET_ADDR_BSSID_CAM_IDX(cmd, addr_cam->bssid_cam_idx);
856 
857 	FWCMD_SET_ADDR_SMA0(cmd, sma[0]);
858 	FWCMD_SET_ADDR_SMA1(cmd, sma[1]);
859 	FWCMD_SET_ADDR_SMA2(cmd, sma[2]);
860 	FWCMD_SET_ADDR_SMA3(cmd, sma[3]);
861 	FWCMD_SET_ADDR_SMA4(cmd, sma[4]);
862 	FWCMD_SET_ADDR_SMA5(cmd, sma[5]);
863 
864 	FWCMD_SET_ADDR_TMA0(cmd, tma[0]);
865 	FWCMD_SET_ADDR_TMA1(cmd, tma[1]);
866 	FWCMD_SET_ADDR_TMA2(cmd, tma[2]);
867 	FWCMD_SET_ADDR_TMA3(cmd, tma[3]);
868 	FWCMD_SET_ADDR_TMA4(cmd, tma[4]);
869 	FWCMD_SET_ADDR_TMA5(cmd, tma[5]);
870 
871 	FWCMD_SET_ADDR_PORT_INT(cmd, rtwvif_link->port);
872 	FWCMD_SET_ADDR_TSF_SYNC(cmd, rtwvif_link->port);
873 	FWCMD_SET_ADDR_TF_TRS(cmd, rtwvif_link->trigger);
874 	FWCMD_SET_ADDR_LSIG_TXOP(cmd, rtwvif_link->lsig_txop);
875 	FWCMD_SET_ADDR_TGT_IND(cmd, rtwvif_link->tgt_ind);
876 	FWCMD_SET_ADDR_FRM_TGT_IND(cmd, rtwvif_link->frm_tgt_ind);
877 	FWCMD_SET_ADDR_MACID(cmd, rtwsta_link ? rtwsta_link->mac_id :
878 						rtwvif_link->mac_id);
879 	if (rtwvif_link->net_type == RTW89_NET_TYPE_INFRA)
880 		FWCMD_SET_ADDR_AID12(cmd, vif->cfg.aid & 0xfff);
881 	else if (rtwvif_link->net_type == RTW89_NET_TYPE_AP_MODE)
882 		FWCMD_SET_ADDR_AID12(cmd, sta ? sta->aid & 0xfff : 0);
883 	FWCMD_SET_ADDR_WOL_PATTERN(cmd, rtwvif_link->wowlan_pattern);
884 	FWCMD_SET_ADDR_WOL_UC(cmd, rtwvif_link->wowlan_uc);
885 	FWCMD_SET_ADDR_WOL_MAGIC(cmd, rtwvif_link->wowlan_magic);
886 	FWCMD_SET_ADDR_WAPI(cmd, addr_cam->wapi);
887 	FWCMD_SET_ADDR_SEC_ENT_MODE(cmd, addr_cam->sec_ent_mode);
888 	FWCMD_SET_ADDR_SEC_ENT0_KEYID(cmd, addr_cam->sec_ent_keyid[0]);
889 	FWCMD_SET_ADDR_SEC_ENT1_KEYID(cmd, addr_cam->sec_ent_keyid[1]);
890 	FWCMD_SET_ADDR_SEC_ENT2_KEYID(cmd, addr_cam->sec_ent_keyid[2]);
891 	FWCMD_SET_ADDR_SEC_ENT3_KEYID(cmd, addr_cam->sec_ent_keyid[3]);
892 	FWCMD_SET_ADDR_SEC_ENT4_KEYID(cmd, addr_cam->sec_ent_keyid[4]);
893 	FWCMD_SET_ADDR_SEC_ENT5_KEYID(cmd, addr_cam->sec_ent_keyid[5]);
894 	FWCMD_SET_ADDR_SEC_ENT6_KEYID(cmd, addr_cam->sec_ent_keyid[6]);
895 
896 	FWCMD_SET_ADDR_SEC_ENT_VALID(cmd, addr_cam->sec_cam_map[0] & 0xff);
897 	FWCMD_SET_ADDR_SEC_ENT0(cmd, addr_cam->sec_ent[0]);
898 	FWCMD_SET_ADDR_SEC_ENT1(cmd, addr_cam->sec_ent[1]);
899 	FWCMD_SET_ADDR_SEC_ENT2(cmd, addr_cam->sec_ent[2]);
900 	FWCMD_SET_ADDR_SEC_ENT3(cmd, addr_cam->sec_ent[3]);
901 	FWCMD_SET_ADDR_SEC_ENT4(cmd, addr_cam->sec_ent[4]);
902 	FWCMD_SET_ADDR_SEC_ENT5(cmd, addr_cam->sec_ent[5]);
903 	FWCMD_SET_ADDR_SEC_ENT6(cmd, addr_cam->sec_ent[6]);
904 
905 	rcu_read_unlock();
906 }
907 
rtw89_cam_fill_dctl_sec_cam_info_v1(struct rtw89_dev * rtwdev,struct rtw89_vif_link * rtwvif_link,struct rtw89_sta_link * rtwsta_link,struct rtw89_h2c_dctlinfo_ud_v1 * h2c)908 void rtw89_cam_fill_dctl_sec_cam_info_v1(struct rtw89_dev *rtwdev,
909 					 struct rtw89_vif_link *rtwvif_link,
910 					 struct rtw89_sta_link *rtwsta_link,
911 					 struct rtw89_h2c_dctlinfo_ud_v1 *h2c)
912 {
913 	struct rtw89_addr_cam_entry *addr_cam =
914 		rtw89_get_addr_cam_of(rtwvif_link, rtwsta_link);
915 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
916 	u8 *ptk_tx_iv = rtw_wow->key_info.ptk_tx_iv;
917 
918 	h2c->c0 = le32_encode_bits(rtwsta_link ? rtwsta_link->mac_id :
919 						 rtwvif_link->mac_id,
920 				   DCTLINFO_V1_C0_MACID) |
921 		  le32_encode_bits(1, DCTLINFO_V1_C0_OP);
922 
923 	h2c->w4 = le32_encode_bits(addr_cam->sec_ent_keyid[0],
924 				   DCTLINFO_V1_W4_SEC_ENT0_KEYID) |
925 		  le32_encode_bits(addr_cam->sec_ent_keyid[1],
926 				   DCTLINFO_V1_W4_SEC_ENT1_KEYID) |
927 		  le32_encode_bits(addr_cam->sec_ent_keyid[2],
928 				   DCTLINFO_V1_W4_SEC_ENT2_KEYID) |
929 		  le32_encode_bits(addr_cam->sec_ent_keyid[3],
930 				   DCTLINFO_V1_W4_SEC_ENT3_KEYID) |
931 		  le32_encode_bits(addr_cam->sec_ent_keyid[4],
932 				   DCTLINFO_V1_W4_SEC_ENT4_KEYID) |
933 		  le32_encode_bits(addr_cam->sec_ent_keyid[5],
934 				   DCTLINFO_V1_W4_SEC_ENT5_KEYID) |
935 		  le32_encode_bits(addr_cam->sec_ent_keyid[6],
936 				   DCTLINFO_V1_W4_SEC_ENT6_KEYID);
937 	h2c->m4 = cpu_to_le32(DCTLINFO_V1_W4_SEC_ENT0_KEYID |
938 			      DCTLINFO_V1_W4_SEC_ENT1_KEYID |
939 			      DCTLINFO_V1_W4_SEC_ENT2_KEYID |
940 			      DCTLINFO_V1_W4_SEC_ENT3_KEYID |
941 			      DCTLINFO_V1_W4_SEC_ENT4_KEYID |
942 			      DCTLINFO_V1_W4_SEC_ENT5_KEYID |
943 			      DCTLINFO_V1_W4_SEC_ENT6_KEYID);
944 
945 	h2c->w5 = le32_encode_bits(addr_cam->sec_cam_map[0] & 0xff,
946 				   DCTLINFO_V1_W5_SEC_ENT_VALID) |
947 		  le32_encode_bits(addr_cam->sec_ent[0],
948 				   DCTLINFO_V1_W5_SEC_ENT0) |
949 		  le32_encode_bits(addr_cam->sec_ent[1],
950 				   DCTLINFO_V1_W5_SEC_ENT1) |
951 		  le32_encode_bits(addr_cam->sec_ent[2],
952 				   DCTLINFO_V1_W5_SEC_ENT2);
953 	h2c->m5 = cpu_to_le32(DCTLINFO_V1_W5_SEC_ENT_VALID |
954 			      DCTLINFO_V1_W5_SEC_ENT0      |
955 			      DCTLINFO_V1_W5_SEC_ENT1      |
956 			      DCTLINFO_V1_W5_SEC_ENT2);
957 
958 	h2c->w6 = le32_encode_bits(addr_cam->sec_ent[3],
959 				   DCTLINFO_V1_W6_SEC_ENT3) |
960 		  le32_encode_bits(addr_cam->sec_ent[4],
961 				   DCTLINFO_V1_W6_SEC_ENT4) |
962 		  le32_encode_bits(addr_cam->sec_ent[5],
963 				   DCTLINFO_V1_W6_SEC_ENT5) |
964 		  le32_encode_bits(addr_cam->sec_ent[6],
965 				   DCTLINFO_V1_W6_SEC_ENT6);
966 	h2c->m6 = cpu_to_le32(DCTLINFO_V1_W6_SEC_ENT3 |
967 			      DCTLINFO_V1_W6_SEC_ENT4 |
968 			      DCTLINFO_V1_W6_SEC_ENT5 |
969 			      DCTLINFO_V1_W6_SEC_ENT6);
970 
971 	if (rtw_wow->ptk_alg) {
972 		h2c->w0 = le32_encode_bits(ptk_tx_iv[0] | ptk_tx_iv[1] << 8,
973 					   DCTLINFO_V1_W0_AES_IV_L);
974 		h2c->m0 = cpu_to_le32(DCTLINFO_V1_W0_AES_IV_L);
975 
976 		h2c->w1 = le32_encode_bits(ptk_tx_iv[4]       |
977 					   ptk_tx_iv[5] << 8  |
978 					   ptk_tx_iv[6] << 16 |
979 					   ptk_tx_iv[7] << 24,
980 					   DCTLINFO_V1_W1_AES_IV_H);
981 		h2c->m1 = cpu_to_le32(DCTLINFO_V1_W1_AES_IV_H);
982 
983 		h2c->w4 |= le32_encode_bits(rtw_wow->ptk_keyidx,
984 					    DCTLINFO_V1_W4_SEC_KEY_ID);
985 		h2c->m4 |= cpu_to_le32(DCTLINFO_V1_W4_SEC_KEY_ID);
986 	}
987 }
988 
rtw89_cam_fill_dctl_sec_cam_info_v2(struct rtw89_dev * rtwdev,struct rtw89_vif_link * rtwvif_link,struct rtw89_sta_link * rtwsta_link,struct rtw89_h2c_dctlinfo_ud_v2 * h2c)989 void rtw89_cam_fill_dctl_sec_cam_info_v2(struct rtw89_dev *rtwdev,
990 					 struct rtw89_vif_link *rtwvif_link,
991 					 struct rtw89_sta_link *rtwsta_link,
992 					 struct rtw89_h2c_dctlinfo_ud_v2 *h2c)
993 {
994 	struct ieee80211_sta *sta = rtwsta_link_to_sta_safe(rtwsta_link);
995 	struct ieee80211_vif *vif = rtwvif_to_vif(rtwvif_link->rtwvif);
996 	struct rtw89_vif *rtwvif = rtwvif_link->rtwvif;
997 	struct rtw89_addr_cam_entry *addr_cam =
998 		rtw89_get_addr_cam_of(rtwvif_link, rtwsta_link);
999 	bool is_mld = sta ? sta->mlo : ieee80211_vif_is_mld(vif);
1000 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1001 	u8 *ptk_tx_iv = rtw_wow->key_info.ptk_tx_iv;
1002 	u8 *mld_sma, *mld_tma, *mld_bssid;
1003 
1004 	h2c->c0 = le32_encode_bits(rtwsta_link ? rtwsta_link->mac_id :
1005 						 rtwvif_link->mac_id,
1006 				   DCTLINFO_V2_C0_MACID) |
1007 		  le32_encode_bits(1, DCTLINFO_V2_C0_OP);
1008 
1009 	h2c->w2 = le32_encode_bits(is_mld, DCTLINFO_V2_W2_IS_MLD);
1010 	h2c->m2 = cpu_to_le32(DCTLINFO_V2_W2_IS_MLD);
1011 
1012 	h2c->w4 = le32_encode_bits(addr_cam->sec_ent_keyid[0],
1013 				   DCTLINFO_V2_W4_SEC_ENT0_KEYID) |
1014 		  le32_encode_bits(addr_cam->sec_ent_keyid[1],
1015 				   DCTLINFO_V2_W4_SEC_ENT1_KEYID) |
1016 		  le32_encode_bits(addr_cam->sec_ent_keyid[2],
1017 				   DCTLINFO_V2_W4_SEC_ENT2_KEYID) |
1018 		  le32_encode_bits(addr_cam->sec_ent_keyid[3],
1019 				   DCTLINFO_V2_W4_SEC_ENT3_KEYID) |
1020 		  le32_encode_bits(addr_cam->sec_ent_keyid[4],
1021 				   DCTLINFO_V2_W4_SEC_ENT4_KEYID) |
1022 		  le32_encode_bits(addr_cam->sec_ent_keyid[5],
1023 				   DCTLINFO_V2_W4_SEC_ENT5_KEYID) |
1024 		  le32_encode_bits(addr_cam->sec_ent_keyid[6],
1025 				   DCTLINFO_V2_W4_SEC_ENT6_KEYID);
1026 	h2c->m4 = cpu_to_le32(DCTLINFO_V2_W4_SEC_ENT0_KEYID |
1027 			      DCTLINFO_V2_W4_SEC_ENT1_KEYID |
1028 			      DCTLINFO_V2_W4_SEC_ENT2_KEYID |
1029 			      DCTLINFO_V2_W4_SEC_ENT3_KEYID |
1030 			      DCTLINFO_V2_W4_SEC_ENT4_KEYID |
1031 			      DCTLINFO_V2_W4_SEC_ENT5_KEYID |
1032 			      DCTLINFO_V2_W4_SEC_ENT6_KEYID);
1033 
1034 	h2c->w5 = le32_encode_bits(addr_cam->sec_cam_map[0],
1035 				   DCTLINFO_V2_W5_SEC_ENT_VALID_V1) |
1036 		  le32_encode_bits(addr_cam->sec_ent[0],
1037 				   DCTLINFO_V2_W5_SEC_ENT0_V1);
1038 	h2c->m5 = cpu_to_le32(DCTLINFO_V2_W5_SEC_ENT_VALID_V1 |
1039 			      DCTLINFO_V2_W5_SEC_ENT0_V1);
1040 
1041 	h2c->w6 = le32_encode_bits(addr_cam->sec_ent[1],
1042 				   DCTLINFO_V2_W6_SEC_ENT1_V1) |
1043 		  le32_encode_bits(addr_cam->sec_ent[2],
1044 				   DCTLINFO_V2_W6_SEC_ENT2_V1) |
1045 		  le32_encode_bits(addr_cam->sec_ent[3],
1046 				   DCTLINFO_V2_W6_SEC_ENT3_V1) |
1047 		  le32_encode_bits(addr_cam->sec_ent[4],
1048 				   DCTLINFO_V2_W6_SEC_ENT4_V1);
1049 	h2c->m6 = cpu_to_le32(DCTLINFO_V2_W6_SEC_ENT1_V1 |
1050 			      DCTLINFO_V2_W6_SEC_ENT2_V1 |
1051 			      DCTLINFO_V2_W6_SEC_ENT3_V1 |
1052 			      DCTLINFO_V2_W6_SEC_ENT4_V1);
1053 
1054 	h2c->w7 = le32_encode_bits(addr_cam->sec_ent[5],
1055 				   DCTLINFO_V2_W7_SEC_ENT5_V1) |
1056 		  le32_encode_bits(addr_cam->sec_ent[6],
1057 				   DCTLINFO_V2_W7_SEC_ENT6_V1);
1058 	h2c->m7 = cpu_to_le32(DCTLINFO_V2_W7_SEC_ENT5_V1 |
1059 			      DCTLINFO_V2_W7_SEC_ENT6_V1);
1060 
1061 	if (rtw_wow->ptk_alg) {
1062 		h2c->w0 = le32_encode_bits(ptk_tx_iv[0] | ptk_tx_iv[1] << 8,
1063 					   DCTLINFO_V2_W0_AES_IV_L);
1064 		h2c->m0 = cpu_to_le32(DCTLINFO_V2_W0_AES_IV_L);
1065 
1066 		h2c->w1 = le32_encode_bits(ptk_tx_iv[4] |
1067 					   ptk_tx_iv[5] << 8 |
1068 					   ptk_tx_iv[6] << 16 |
1069 					   ptk_tx_iv[7] << 24,
1070 					   DCTLINFO_V2_W1_AES_IV_H);
1071 		h2c->m1 = cpu_to_le32(DCTLINFO_V2_W1_AES_IV_H);
1072 
1073 		h2c->w4 |= le32_encode_bits(rtw_wow->ptk_keyidx,
1074 					    DCTLINFO_V2_W4_SEC_KEY_ID);
1075 		h2c->m4 |= cpu_to_le32(DCTLINFO_V2_W4_SEC_KEY_ID);
1076 	}
1077 
1078 	if (!is_mld)
1079 		return;
1080 
1081 	if (rtwvif_link->net_type == RTW89_NET_TYPE_INFRA) {
1082 		mld_sma = rtwvif->mac_addr;
1083 		mld_tma = vif->cfg.ap_addr;
1084 		mld_bssid = vif->cfg.ap_addr;
1085 	} else if (rtwvif_link->net_type == RTW89_NET_TYPE_AP_MODE && sta) {
1086 		mld_sma = rtwvif->mac_addr;
1087 		mld_tma = sta->addr;
1088 		mld_bssid = rtwvif->mac_addr;
1089 	} else {
1090 		return;
1091 	}
1092 
1093 	h2c->w8 = le32_encode_bits(mld_sma[0], DCTLINFO_V2_W8_MLD_SMA_0) |
1094 		  le32_encode_bits(mld_sma[1], DCTLINFO_V2_W8_MLD_SMA_1) |
1095 		  le32_encode_bits(mld_sma[2], DCTLINFO_V2_W8_MLD_SMA_2) |
1096 		  le32_encode_bits(mld_sma[3], DCTLINFO_V2_W8_MLD_SMA_3);
1097 	h2c->m8 = cpu_to_le32(DCTLINFO_V2_W8_ALL);
1098 
1099 	h2c->w9 = le32_encode_bits(mld_sma[4], DCTLINFO_V2_W9_MLD_SMA_4) |
1100 		  le32_encode_bits(mld_sma[5], DCTLINFO_V2_W9_MLD_SMA_5) |
1101 		  le32_encode_bits(mld_tma[0], DCTLINFO_V2_W9_MLD_TMA_0) |
1102 		  le32_encode_bits(mld_tma[1], DCTLINFO_V2_W9_MLD_TMA_1);
1103 	h2c->m9 = cpu_to_le32(DCTLINFO_V2_W9_ALL);
1104 
1105 	h2c->w10 = le32_encode_bits(mld_tma[2], DCTLINFO_V2_W10_MLD_TMA_2) |
1106 		   le32_encode_bits(mld_tma[3], DCTLINFO_V2_W10_MLD_TMA_3) |
1107 		   le32_encode_bits(mld_tma[4], DCTLINFO_V2_W10_MLD_TMA_4) |
1108 		   le32_encode_bits(mld_tma[5], DCTLINFO_V2_W10_MLD_TMA_5);
1109 	h2c->m10 = cpu_to_le32(DCTLINFO_V2_W10_ALL);
1110 
1111 	h2c->w11 = le32_encode_bits(mld_bssid[0], DCTLINFO_V2_W11_MLD_BSSID_0) |
1112 		   le32_encode_bits(mld_bssid[1], DCTLINFO_V2_W11_MLD_BSSID_1) |
1113 		   le32_encode_bits(mld_bssid[2], DCTLINFO_V2_W11_MLD_BSSID_2) |
1114 		   le32_encode_bits(mld_bssid[3], DCTLINFO_V2_W11_MLD_BSSID_3);
1115 	h2c->m11 = cpu_to_le32(DCTLINFO_V2_W11_ALL);
1116 
1117 	h2c->w12 = le32_encode_bits(mld_bssid[4], DCTLINFO_V2_W12_MLD_BSSID_4) |
1118 		   le32_encode_bits(mld_bssid[5], DCTLINFO_V2_W12_MLD_BSSID_5);
1119 	h2c->m12 = cpu_to_le32(DCTLINFO_V2_W12_ALL);
1120 }
1121