xref: /linux/drivers/net/wireless/realtek/rtw89/wow.c (revision a19d0236f466f1ce8f44a04a96c302d3023eebf4)
1 // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2 /* Copyright(c) 2019-2022  Realtek Corporation
3  */
4 #include "cam.h"
5 #include "core.h"
6 #include "debug.h"
7 #include "fw.h"
8 #include "mac.h"
9 #include "phy.h"
10 #include "ps.h"
11 #include "reg.h"
12 #include "util.h"
13 #include "wow.h"
14 
15 void rtw89_wow_parse_akm(struct rtw89_dev *rtwdev, struct sk_buff *skb)
16 {
17 	struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *)skb->data;
18 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
19 	const u8 *rsn, *ies = mgmt->u.assoc_req.variable;
20 	struct rtw89_rsn_ie *rsn_ie;
21 
22 	rsn = cfg80211_find_ie(WLAN_EID_RSN, ies, skb->len);
23 	if (!rsn)
24 		return;
25 
26 	rsn_ie = (struct rtw89_rsn_ie *)rsn;
27 	rtw_wow->akm = rsn_ie->akm_cipher_suite.type;
28 }
29 
30 #define RTW89_CIPHER_INFO_DEF(cipher) \
31 	{WLAN_CIPHER_SUITE_ ## cipher, .fw_alg = RTW89_WOW_FW_ALG_ ## cipher, \
32 	 .len = WLAN_KEY_LEN_ ## cipher}
33 
34 static const struct rtw89_cipher_info rtw89_cipher_info_defs[] = {
35 	RTW89_CIPHER_INFO_DEF(WEP40),
36 	RTW89_CIPHER_INFO_DEF(WEP104),
37 	RTW89_CIPHER_INFO_DEF(TKIP),
38 	RTW89_CIPHER_INFO_DEF(CCMP),
39 	RTW89_CIPHER_INFO_DEF(GCMP),
40 	RTW89_CIPHER_INFO_DEF(CCMP_256),
41 	RTW89_CIPHER_INFO_DEF(GCMP_256),
42 	RTW89_CIPHER_INFO_DEF(AES_CMAC),
43 };
44 
45 #undef RTW89_CIPHER_INFO_DEF
46 
47 static const
48 struct rtw89_cipher_info *rtw89_cipher_alg_recognize(u32 cipher)
49 {
50 	const struct rtw89_cipher_info *cipher_info_defs;
51 	int i;
52 
53 	for (i = 0; i < ARRAY_SIZE(rtw89_cipher_info_defs); i++) {
54 		cipher_info_defs = &rtw89_cipher_info_defs[i];
55 		if (cipher_info_defs->cipher == cipher)
56 			return cipher_info_defs;
57 	}
58 
59 	return NULL;
60 }
61 
62 static int _pn_to_iv(struct rtw89_dev *rtwdev, struct ieee80211_key_conf *key,
63 		     u8 *iv, u64 pn, u8 key_idx)
64 {
65 	switch (key->cipher) {
66 	case WLAN_CIPHER_SUITE_TKIP:
67 		iv[0] = u64_get_bits(pn, RTW89_KEY_PN_1);
68 		iv[1] = (u64_get_bits(pn, RTW89_KEY_PN_1) | 0x20) & 0x7f;
69 		iv[2] = u64_get_bits(pn, RTW89_KEY_PN_0);
70 		break;
71 	case WLAN_CIPHER_SUITE_CCMP:
72 	case WLAN_CIPHER_SUITE_GCMP:
73 	case WLAN_CIPHER_SUITE_CCMP_256:
74 	case WLAN_CIPHER_SUITE_GCMP_256:
75 		iv[0] = u64_get_bits(pn, RTW89_KEY_PN_0);
76 		iv[1] = u64_get_bits(pn, RTW89_KEY_PN_1);
77 		iv[2] = 0;
78 		break;
79 	default:
80 		return -EINVAL;
81 	}
82 
83 	iv[3] = BIT(5) | ((key_idx & 0x3) << 6);
84 	iv[4] = u64_get_bits(pn, RTW89_KEY_PN_2);
85 	iv[5] = u64_get_bits(pn, RTW89_KEY_PN_3);
86 	iv[6] = u64_get_bits(pn, RTW89_KEY_PN_4);
87 	iv[7] = u64_get_bits(pn, RTW89_KEY_PN_5);
88 
89 	return 0;
90 }
91 
92 static int rtw89_rx_pn_to_iv(struct rtw89_dev *rtwdev,
93 			     struct ieee80211_key_conf *key,
94 			     u8 *iv)
95 {
96 	struct ieee80211_key_seq seq;
97 	int err;
98 	u64 pn;
99 
100 	ieee80211_get_key_rx_seq(key, 0, &seq);
101 
102 	/* seq.ccmp.pn[] is BE order array */
103 	pn = u64_encode_bits(seq.ccmp.pn[0], RTW89_KEY_PN_5) |
104 	     u64_encode_bits(seq.ccmp.pn[1], RTW89_KEY_PN_4) |
105 	     u64_encode_bits(seq.ccmp.pn[2], RTW89_KEY_PN_3) |
106 	     u64_encode_bits(seq.ccmp.pn[3], RTW89_KEY_PN_2) |
107 	     u64_encode_bits(seq.ccmp.pn[4], RTW89_KEY_PN_1) |
108 	     u64_encode_bits(seq.ccmp.pn[5], RTW89_KEY_PN_0);
109 
110 	err = _pn_to_iv(rtwdev, key, iv, pn, key->keyidx);
111 	if (err)
112 		return err;
113 
114 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "%s key %d pn-%llx to iv-%*ph\n",
115 		    __func__, key->keyidx, pn, 8, iv);
116 
117 	return 0;
118 }
119 
120 static int rtw89_tx_pn_to_iv(struct rtw89_dev *rtwdev,
121 			     struct ieee80211_key_conf *key,
122 			     u8 *iv)
123 {
124 	int err;
125 	u64 pn;
126 
127 	pn = atomic64_inc_return(&key->tx_pn);
128 	err = _pn_to_iv(rtwdev, key, iv, pn, key->keyidx);
129 	if (err)
130 		return err;
131 
132 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "%s key %d pn-%llx to iv-%*ph\n",
133 		    __func__, key->keyidx, pn, 8, iv);
134 
135 	return 0;
136 }
137 
138 static int _iv_to_pn(struct rtw89_dev *rtwdev, u8 *iv, u64 *pn, u8 *key_id,
139 		     struct ieee80211_key_conf *key)
140 {
141 	switch (key->cipher) {
142 	case WLAN_CIPHER_SUITE_TKIP:
143 		*pn = u64_encode_bits(iv[2], RTW89_KEY_PN_0) |
144 		      u64_encode_bits(iv[0], RTW89_KEY_PN_1);
145 		break;
146 	case WLAN_CIPHER_SUITE_CCMP:
147 	case WLAN_CIPHER_SUITE_GCMP:
148 	case WLAN_CIPHER_SUITE_CCMP_256:
149 	case WLAN_CIPHER_SUITE_GCMP_256:
150 		*pn = u64_encode_bits(iv[0], RTW89_KEY_PN_0) |
151 		      u64_encode_bits(iv[1], RTW89_KEY_PN_1);
152 		break;
153 	default:
154 		return -EINVAL;
155 	}
156 
157 	*pn |= u64_encode_bits(iv[4], RTW89_KEY_PN_2) |
158 	       u64_encode_bits(iv[5], RTW89_KEY_PN_3) |
159 	       u64_encode_bits(iv[6], RTW89_KEY_PN_4) |
160 	       u64_encode_bits(iv[7], RTW89_KEY_PN_5);
161 
162 	if (key_id)
163 		*key_id = *(iv + 3) >> 6;
164 
165 	return 0;
166 }
167 
168 static int rtw89_rx_iv_to_pn(struct rtw89_dev *rtwdev,
169 			     struct ieee80211_key_conf *key,
170 			     u8 *iv)
171 {
172 	struct ieee80211_key_seq seq;
173 	int err;
174 	u64 pn;
175 
176 	err = _iv_to_pn(rtwdev, iv, &pn, NULL, key);
177 	if (err)
178 		return err;
179 
180 	/* seq.ccmp.pn[] is BE order array */
181 	seq.ccmp.pn[0] = u64_get_bits(pn, RTW89_KEY_PN_5);
182 	seq.ccmp.pn[1] = u64_get_bits(pn, RTW89_KEY_PN_4);
183 	seq.ccmp.pn[2] = u64_get_bits(pn, RTW89_KEY_PN_3);
184 	seq.ccmp.pn[3] = u64_get_bits(pn, RTW89_KEY_PN_2);
185 	seq.ccmp.pn[4] = u64_get_bits(pn, RTW89_KEY_PN_1);
186 	seq.ccmp.pn[5] = u64_get_bits(pn, RTW89_KEY_PN_0);
187 
188 	ieee80211_set_key_rx_seq(key, 0, &seq);
189 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "%s key %d iv-%*ph to pn-%*ph\n",
190 		    __func__, key->keyidx, 8, iv, 6, seq.ccmp.pn);
191 
192 	return 0;
193 }
194 
195 static int rtw89_tx_iv_to_pn(struct rtw89_dev *rtwdev,
196 			     struct ieee80211_key_conf *key,
197 			     u8 *iv)
198 {
199 	int err;
200 	u64 pn;
201 
202 	err = _iv_to_pn(rtwdev, iv, &pn, NULL, key);
203 	if (err)
204 		return err;
205 
206 	atomic64_set(&key->tx_pn, pn);
207 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "%s key %d iv-%*ph to pn-%llx\n",
208 		    __func__, key->keyidx, 8, iv, pn);
209 
210 	return 0;
211 }
212 
213 static int rtw89_rx_pn_get_pmf(struct rtw89_dev *rtwdev,
214 			       struct ieee80211_key_conf *key,
215 			       struct rtw89_wow_gtk_info *gtk_info)
216 {
217 	struct ieee80211_key_seq seq;
218 	u64 pn;
219 
220 	if (key->keyidx == 4)
221 		memcpy(gtk_info->igtk[0], key->key, key->keylen);
222 	else if (key->keyidx == 5)
223 		memcpy(gtk_info->igtk[1], key->key, key->keylen);
224 	else
225 		return -EINVAL;
226 
227 	ieee80211_get_key_rx_seq(key, 0, &seq);
228 
229 	/* seq.ccmp.pn[] is BE order array */
230 	pn = u64_encode_bits(seq.ccmp.pn[0], RTW89_KEY_PN_5) |
231 	     u64_encode_bits(seq.ccmp.pn[1], RTW89_KEY_PN_4) |
232 	     u64_encode_bits(seq.ccmp.pn[2], RTW89_KEY_PN_3) |
233 	     u64_encode_bits(seq.ccmp.pn[3], RTW89_KEY_PN_2) |
234 	     u64_encode_bits(seq.ccmp.pn[4], RTW89_KEY_PN_1) |
235 	     u64_encode_bits(seq.ccmp.pn[5], RTW89_KEY_PN_0);
236 	gtk_info->ipn = cpu_to_le64(pn);
237 	gtk_info->igtk_keyid = cpu_to_le32(key->keyidx);
238 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "%s key %d pn-%llx\n",
239 		    __func__, key->keyidx, pn);
240 
241 	return 0;
242 }
243 
244 static int rtw89_rx_pn_set_pmf(struct rtw89_dev *rtwdev,
245 			       struct ieee80211_key_conf *key,
246 			       u64 pn)
247 {
248 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
249 	struct rtw89_wow_aoac_report *aoac_rpt = &rtw_wow->aoac_rpt;
250 	struct ieee80211_key_seq seq;
251 
252 	if (key->keyidx != aoac_rpt->igtk_key_id)
253 		return 0;
254 
255 	/* seq.ccmp.pn[] is BE order array */
256 	seq.ccmp.pn[0] = u64_get_bits(pn, RTW89_KEY_PN_5);
257 	seq.ccmp.pn[1] = u64_get_bits(pn, RTW89_KEY_PN_4);
258 	seq.ccmp.pn[2] = u64_get_bits(pn, RTW89_KEY_PN_3);
259 	seq.ccmp.pn[3] = u64_get_bits(pn, RTW89_KEY_PN_2);
260 	seq.ccmp.pn[4] = u64_get_bits(pn, RTW89_KEY_PN_1);
261 	seq.ccmp.pn[5] = u64_get_bits(pn, RTW89_KEY_PN_0);
262 
263 	ieee80211_set_key_rx_seq(key, 0, &seq);
264 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "%s key %d pn-%*ph\n",
265 		    __func__, key->keyidx, 6, seq.ccmp.pn);
266 
267 	return 0;
268 }
269 
270 static void rtw89_wow_get_key_info_iter(struct ieee80211_hw *hw,
271 					struct ieee80211_vif *vif,
272 					struct ieee80211_sta *sta,
273 					struct ieee80211_key_conf *key,
274 					void *data)
275 {
276 	struct rtw89_dev *rtwdev = hw->priv;
277 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
278 	struct rtw89_wow_key_info *key_info = &rtw_wow->key_info;
279 	struct rtw89_wow_gtk_info *gtk_info = &rtw_wow->gtk_info;
280 	const struct rtw89_cipher_info *cipher_info;
281 	bool *err = data;
282 	int ret;
283 
284 	cipher_info = rtw89_cipher_alg_recognize(key->cipher);
285 
286 	switch (key->cipher) {
287 	case WLAN_CIPHER_SUITE_TKIP:
288 	case WLAN_CIPHER_SUITE_CCMP:
289 	case WLAN_CIPHER_SUITE_GCMP:
290 	case WLAN_CIPHER_SUITE_CCMP_256:
291 	case WLAN_CIPHER_SUITE_GCMP_256:
292 		if (sta) {
293 			ret = rtw89_tx_pn_to_iv(rtwdev, key,
294 						key_info->ptk_tx_iv);
295 			if (ret)
296 				goto err;
297 			ret = rtw89_rx_pn_to_iv(rtwdev, key,
298 						key_info->ptk_rx_iv);
299 			if (ret)
300 				goto err;
301 
302 			rtw_wow->ptk_alg = cipher_info->fw_alg;
303 			rtw_wow->ptk_keyidx = key->keyidx;
304 		} else {
305 			ret = rtw89_rx_pn_to_iv(rtwdev, key,
306 						key_info->gtk_rx_iv[key->keyidx]);
307 			if (ret)
308 				goto err;
309 
310 			rtw_wow->gtk_alg = cipher_info->fw_alg;
311 			key_info->gtk_keyidx = key->keyidx;
312 		}
313 		break;
314 	case WLAN_CIPHER_SUITE_AES_CMAC:
315 		ret = rtw89_rx_pn_get_pmf(rtwdev, key, gtk_info);
316 		if (ret)
317 			goto err;
318 		break;
319 	case WLAN_CIPHER_SUITE_WEP40:
320 	case WLAN_CIPHER_SUITE_WEP104:
321 		/* WEP only set group key in mac80211, but fw need to set
322 		 * both of pairwise key and group key.
323 		 */
324 		rtw_wow->ptk_alg = cipher_info->fw_alg;
325 		rtw_wow->ptk_keyidx = key->keyidx;
326 		rtw_wow->gtk_alg = cipher_info->fw_alg;
327 		key_info->gtk_keyidx = key->keyidx;
328 		break;
329 	default:
330 		rtw89_debug(rtwdev, RTW89_DBG_WOW, "unsupport cipher %x\n",
331 			    key->cipher);
332 		goto err;
333 	}
334 
335 	return;
336 err:
337 	*err = true;
338 }
339 
340 static void rtw89_wow_set_key_info_iter(struct ieee80211_hw *hw,
341 					struct ieee80211_vif *vif,
342 					struct ieee80211_sta *sta,
343 					struct ieee80211_key_conf *key,
344 					void *data)
345 {
346 	struct rtw89_dev *rtwdev = hw->priv;
347 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
348 	struct rtw89_wow_aoac_report *aoac_rpt = &rtw_wow->aoac_rpt;
349 	struct rtw89_set_key_info_iter_data *iter_data = data;
350 	bool update_tx_key_info = iter_data->rx_ready;
351 	int ret;
352 
353 	switch (key->cipher) {
354 	case WLAN_CIPHER_SUITE_TKIP:
355 	case WLAN_CIPHER_SUITE_CCMP:
356 	case WLAN_CIPHER_SUITE_GCMP:
357 	case WLAN_CIPHER_SUITE_CCMP_256:
358 	case WLAN_CIPHER_SUITE_GCMP_256:
359 		if (sta && !update_tx_key_info) {
360 			ret = rtw89_rx_iv_to_pn(rtwdev, key,
361 						aoac_rpt->ptk_rx_iv);
362 			if (ret)
363 				goto err;
364 		}
365 
366 		if (sta && update_tx_key_info) {
367 			ret = rtw89_tx_iv_to_pn(rtwdev, key,
368 						aoac_rpt->ptk_tx_iv);
369 			if (ret)
370 				goto err;
371 		}
372 
373 		if (!sta && !update_tx_key_info) {
374 			ret = rtw89_rx_iv_to_pn(rtwdev, key,
375 						aoac_rpt->gtk_rx_iv[key->keyidx]);
376 			if (ret)
377 				goto err;
378 		}
379 
380 		if (!sta && update_tx_key_info && aoac_rpt->rekey_ok)
381 			iter_data->gtk_cipher = key->cipher;
382 		break;
383 	case WLAN_CIPHER_SUITE_AES_CMAC:
384 		if (update_tx_key_info) {
385 			if (aoac_rpt->rekey_ok)
386 				iter_data->igtk_cipher = key->cipher;
387 		} else {
388 			ret = rtw89_rx_pn_set_pmf(rtwdev, key,
389 						  aoac_rpt->igtk_ipn);
390 			if (ret)
391 				goto err;
392 		}
393 		break;
394 	case WLAN_CIPHER_SUITE_WEP40:
395 	case WLAN_CIPHER_SUITE_WEP104:
396 		break;
397 	default:
398 		rtw89_debug(rtwdev, RTW89_DBG_WOW, "unsupport cipher %x\n",
399 			    key->cipher);
400 		goto err;
401 	}
402 
403 	return;
404 
405 err:
406 	iter_data->error = true;
407 }
408 
409 static void rtw89_wow_key_clear(struct rtw89_dev *rtwdev)
410 {
411 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
412 
413 	memset(&rtw_wow->aoac_rpt, 0, sizeof(rtw_wow->aoac_rpt));
414 	memset(&rtw_wow->gtk_info, 0, sizeof(rtw_wow->gtk_info));
415 	memset(&rtw_wow->key_info, 0, sizeof(rtw_wow->key_info));
416 	rtw_wow->ptk_alg = 0;
417 	rtw_wow->gtk_alg = 0;
418 }
419 
420 static void rtw89_wow_construct_key_info(struct rtw89_dev *rtwdev)
421 {
422 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
423 	struct rtw89_wow_key_info *key_info = &rtw_wow->key_info;
424 	struct rtw89_vif_link *rtwvif_link = rtwdev->wow.rtwvif_link;
425 	struct ieee80211_vif *wow_vif = rtwvif_link_to_vif(rtwvif_link);
426 	bool err = false;
427 
428 	rcu_read_lock();
429 	ieee80211_iter_keys_rcu(rtwdev->hw, wow_vif,
430 				rtw89_wow_get_key_info_iter, &err);
431 	rcu_read_unlock();
432 
433 	if (err) {
434 		rtw89_wow_key_clear(rtwdev);
435 		return;
436 	}
437 
438 	key_info->valid_check = RTW89_WOW_VALID_CHECK;
439 	key_info->symbol_check_en = RTW89_WOW_SYMBOL_CHK_PTK |
440 				    RTW89_WOW_SYMBOL_CHK_GTK;
441 }
442 
443 static void rtw89_wow_debug_aoac_rpt(struct rtw89_dev *rtwdev)
444 {
445 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
446 	struct rtw89_wow_aoac_report *aoac_rpt = &rtw_wow->aoac_rpt;
447 
448 	if (!rtw89_debug_is_enabled(rtwdev, RTW89_DBG_WOW))
449 		return;
450 
451 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] rpt_ver = %d\n",
452 		    aoac_rpt->rpt_ver);
453 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] sec_type = %d\n",
454 		    aoac_rpt->sec_type);
455 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] key_idx = %d\n",
456 		    aoac_rpt->key_idx);
457 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] pattern_idx = %d\n",
458 		    aoac_rpt->pattern_idx);
459 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] rekey_ok = %d\n",
460 		    aoac_rpt->rekey_ok);
461 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] ptk_tx_iv = %*ph\n",
462 		    8, aoac_rpt->ptk_tx_iv);
463 	rtw89_debug(rtwdev, RTW89_DBG_WOW,
464 		    "[aoac_rpt] eapol_key_replay_count = %*ph\n",
465 		    8, aoac_rpt->eapol_key_replay_count);
466 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] ptk_rx_iv = %*ph\n",
467 		    8, aoac_rpt->ptk_rx_iv);
468 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] gtk_rx_iv[0] = %*ph\n",
469 		    8, aoac_rpt->gtk_rx_iv[0]);
470 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] gtk_rx_iv[1] = %*ph\n",
471 		    8, aoac_rpt->gtk_rx_iv[1]);
472 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] gtk_rx_iv[2] = %*ph\n",
473 		    8, aoac_rpt->gtk_rx_iv[2]);
474 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] gtk_rx_iv[3] = %*ph\n",
475 		    8, aoac_rpt->gtk_rx_iv[3]);
476 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] igtk_key_id = %llu\n",
477 		    aoac_rpt->igtk_key_id);
478 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] igtk_ipn = %llu\n",
479 		    aoac_rpt->igtk_ipn);
480 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "[aoac_rpt] igtk = %*ph\n",
481 		    32, aoac_rpt->igtk);
482 }
483 
484 static int rtw89_wow_get_aoac_rpt_reg(struct rtw89_dev *rtwdev)
485 {
486 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
487 	struct rtw89_wow_aoac_report *aoac_rpt = &rtw_wow->aoac_rpt;
488 	struct rtw89_mac_c2h_info c2h_info = {};
489 	struct rtw89_mac_h2c_info h2c_info = {};
490 	u8 igtk_ipn[8];
491 	u8 key_idx;
492 	int ret;
493 
494 	h2c_info.id = RTW89_FWCMD_H2CREG_FUNC_AOAC_RPT_1;
495 	h2c_info.content_len = 2;
496 	ret = rtw89_fw_msg_reg(rtwdev, &h2c_info, &c2h_info);
497 	if (ret)
498 		return ret;
499 
500 	aoac_rpt->key_idx =
501 		u32_get_bits(c2h_info.u.c2hreg[0], RTW89_C2HREG_AOAC_RPT_1_W0_KEY_IDX);
502 	key_idx = aoac_rpt->key_idx;
503 	aoac_rpt->gtk_rx_iv[key_idx][0] =
504 		u32_get_bits(c2h_info.u.c2hreg[1], RTW89_C2HREG_AOAC_RPT_1_W1_IV_0);
505 	aoac_rpt->gtk_rx_iv[key_idx][1] =
506 		u32_get_bits(c2h_info.u.c2hreg[1], RTW89_C2HREG_AOAC_RPT_1_W1_IV_1);
507 	aoac_rpt->gtk_rx_iv[key_idx][2] =
508 		u32_get_bits(c2h_info.u.c2hreg[1], RTW89_C2HREG_AOAC_RPT_1_W1_IV_2);
509 	aoac_rpt->gtk_rx_iv[key_idx][3] =
510 		u32_get_bits(c2h_info.u.c2hreg[1], RTW89_C2HREG_AOAC_RPT_1_W1_IV_3);
511 	aoac_rpt->gtk_rx_iv[key_idx][4] =
512 		u32_get_bits(c2h_info.u.c2hreg[2], RTW89_C2HREG_AOAC_RPT_1_W2_IV_4);
513 	aoac_rpt->gtk_rx_iv[key_idx][5] =
514 		u32_get_bits(c2h_info.u.c2hreg[2], RTW89_C2HREG_AOAC_RPT_1_W2_IV_5);
515 	aoac_rpt->gtk_rx_iv[key_idx][6] =
516 		u32_get_bits(c2h_info.u.c2hreg[2], RTW89_C2HREG_AOAC_RPT_1_W2_IV_6);
517 	aoac_rpt->gtk_rx_iv[key_idx][7] =
518 		u32_get_bits(c2h_info.u.c2hreg[2], RTW89_C2HREG_AOAC_RPT_1_W2_IV_7);
519 	aoac_rpt->ptk_rx_iv[0] =
520 		u32_get_bits(c2h_info.u.c2hreg[3], RTW89_C2HREG_AOAC_RPT_1_W3_PTK_IV_0);
521 	aoac_rpt->ptk_rx_iv[1] =
522 		u32_get_bits(c2h_info.u.c2hreg[3], RTW89_C2HREG_AOAC_RPT_1_W3_PTK_IV_1);
523 	aoac_rpt->ptk_rx_iv[2] =
524 		u32_get_bits(c2h_info.u.c2hreg[3], RTW89_C2HREG_AOAC_RPT_1_W3_PTK_IV_2);
525 	aoac_rpt->ptk_rx_iv[3] =
526 		u32_get_bits(c2h_info.u.c2hreg[3], RTW89_C2HREG_AOAC_RPT_1_W3_PTK_IV_3);
527 
528 	h2c_info.id = RTW89_FWCMD_H2CREG_FUNC_AOAC_RPT_2;
529 	h2c_info.content_len = 2;
530 	ret = rtw89_fw_msg_reg(rtwdev, &h2c_info, &c2h_info);
531 	if (ret)
532 		return ret;
533 
534 	aoac_rpt->ptk_rx_iv[4] =
535 		u32_get_bits(c2h_info.u.c2hreg[0], RTW89_C2HREG_AOAC_RPT_2_W0_PTK_IV_4);
536 	aoac_rpt->ptk_rx_iv[5] =
537 		u32_get_bits(c2h_info.u.c2hreg[0], RTW89_C2HREG_AOAC_RPT_2_W0_PTK_IV_5);
538 	aoac_rpt->ptk_rx_iv[6] =
539 		u32_get_bits(c2h_info.u.c2hreg[1], RTW89_C2HREG_AOAC_RPT_2_W1_PTK_IV_6);
540 	aoac_rpt->ptk_rx_iv[7] =
541 		u32_get_bits(c2h_info.u.c2hreg[1], RTW89_C2HREG_AOAC_RPT_2_W1_PTK_IV_7);
542 	igtk_ipn[0] =
543 		u32_get_bits(c2h_info.u.c2hreg[1], RTW89_C2HREG_AOAC_RPT_2_W1_IGTK_IPN_IV_0);
544 	igtk_ipn[1] =
545 		u32_get_bits(c2h_info.u.c2hreg[1], RTW89_C2HREG_AOAC_RPT_2_W1_IGTK_IPN_IV_1);
546 	igtk_ipn[2] =
547 		u32_get_bits(c2h_info.u.c2hreg[2], RTW89_C2HREG_AOAC_RPT_2_W2_IGTK_IPN_IV_2);
548 	igtk_ipn[3] =
549 		u32_get_bits(c2h_info.u.c2hreg[2], RTW89_C2HREG_AOAC_RPT_2_W2_IGTK_IPN_IV_3);
550 	igtk_ipn[4] =
551 		u32_get_bits(c2h_info.u.c2hreg[2], RTW89_C2HREG_AOAC_RPT_2_W2_IGTK_IPN_IV_4);
552 	igtk_ipn[5] =
553 		u32_get_bits(c2h_info.u.c2hreg[2], RTW89_C2HREG_AOAC_RPT_2_W2_IGTK_IPN_IV_5);
554 	igtk_ipn[6] =
555 		u32_get_bits(c2h_info.u.c2hreg[3], RTW89_C2HREG_AOAC_RPT_2_W3_IGTK_IPN_IV_6);
556 	igtk_ipn[7] =
557 		u32_get_bits(c2h_info.u.c2hreg[3], RTW89_C2HREG_AOAC_RPT_2_W3_IGTK_IPN_IV_7);
558 	aoac_rpt->igtk_ipn = u64_encode_bits(igtk_ipn[0], RTW89_IGTK_IPN_0) |
559 			     u64_encode_bits(igtk_ipn[1], RTW89_IGTK_IPN_1) |
560 			     u64_encode_bits(igtk_ipn[2], RTW89_IGTK_IPN_2) |
561 			     u64_encode_bits(igtk_ipn[3], RTW89_IGTK_IPN_3) |
562 			     u64_encode_bits(igtk_ipn[4], RTW89_IGTK_IPN_4) |
563 			     u64_encode_bits(igtk_ipn[5], RTW89_IGTK_IPN_5) |
564 			     u64_encode_bits(igtk_ipn[6], RTW89_IGTK_IPN_6) |
565 			     u64_encode_bits(igtk_ipn[7], RTW89_IGTK_IPN_7);
566 
567 	return 0;
568 }
569 
570 static int rtw89_wow_get_aoac_rpt(struct rtw89_dev *rtwdev, bool rx_ready)
571 {
572 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
573 	int ret;
574 
575 	if (!rtw_wow->ptk_alg)
576 		return -EPERM;
577 
578 	if (!rx_ready) {
579 		ret = rtw89_wow_get_aoac_rpt_reg(rtwdev);
580 		if (ret) {
581 			rtw89_err(rtwdev, "wow: failed to get aoac rpt by reg\n");
582 			return ret;
583 		}
584 	} else {
585 		ret = rtw89_fw_h2c_wow_request_aoac(rtwdev);
586 		if (ret) {
587 			rtw89_err(rtwdev, "wow: failed to get aoac rpt by pkt\n");
588 			return ret;
589 		}
590 	}
591 
592 	rtw89_wow_debug_aoac_rpt(rtwdev);
593 
594 	return 0;
595 }
596 
597 static struct ieee80211_key_conf *rtw89_wow_gtk_rekey(struct rtw89_dev *rtwdev,
598 						      u32 cipher, u8 keyidx, u8 *gtk)
599 {
600 	struct rtw89_vif_link *rtwvif_link = rtwdev->wow.rtwvif_link;
601 	struct ieee80211_vif *wow_vif = rtwvif_link_to_vif(rtwvif_link);
602 	const struct rtw89_cipher_info *cipher_info;
603 	struct ieee80211_key_conf *rekey_conf;
604 	struct ieee80211_key_conf *key;
605 	u8 sz;
606 
607 	cipher_info = rtw89_cipher_alg_recognize(cipher);
608 	sz = struct_size(rekey_conf, key, cipher_info->len);
609 	rekey_conf = kmalloc(sz, GFP_KERNEL);
610 	if (!rekey_conf)
611 		return NULL;
612 
613 	rekey_conf->cipher = cipher;
614 	rekey_conf->keyidx = keyidx;
615 	rekey_conf->keylen = cipher_info->len;
616 	memcpy(rekey_conf->key, gtk,
617 	       flex_array_size(rekey_conf, key, cipher_info->len));
618 
619 	/* ieee80211_gtk_rekey_add() will call set_key(), therefore we
620 	 * need to unlock mutex
621 	 */
622 	mutex_unlock(&rtwdev->mutex);
623 	if (ieee80211_vif_is_mld(wow_vif))
624 		key = ieee80211_gtk_rekey_add(wow_vif, rekey_conf, rtwvif_link->link_id);
625 	else
626 		key = ieee80211_gtk_rekey_add(wow_vif, rekey_conf, -1);
627 	mutex_lock(&rtwdev->mutex);
628 
629 	kfree(rekey_conf);
630 	if (IS_ERR(key)) {
631 		rtw89_err(rtwdev, "ieee80211_gtk_rekey_add failed\n");
632 		return NULL;
633 	}
634 
635 	return key;
636 }
637 
638 static void rtw89_wow_update_key_info(struct rtw89_dev *rtwdev, bool rx_ready)
639 {
640 	struct rtw89_vif_link *rtwvif_link = rtwdev->wow.rtwvif_link;
641 	struct ieee80211_vif *wow_vif = rtwvif_link_to_vif(rtwvif_link);
642 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
643 	struct rtw89_wow_aoac_report *aoac_rpt = &rtw_wow->aoac_rpt;
644 	struct rtw89_set_key_info_iter_data data = {.error = false,
645 						    .rx_ready = rx_ready};
646 	struct ieee80211_bss_conf *bss_conf;
647 	struct ieee80211_key_conf *key;
648 
649 	rcu_read_lock();
650 	ieee80211_iter_keys_rcu(rtwdev->hw, wow_vif,
651 				rtw89_wow_set_key_info_iter, &data);
652 	rcu_read_unlock();
653 
654 	if (data.error) {
655 		rtw89_debug(rtwdev, RTW89_DBG_WOW, "%s error\n", __func__);
656 		return;
657 	}
658 
659 	if (!data.gtk_cipher)
660 		return;
661 
662 	key = rtw89_wow_gtk_rekey(rtwdev, data.gtk_cipher, aoac_rpt->key_idx,
663 				  aoac_rpt->gtk);
664 	if (!key)
665 		return;
666 
667 	rtw89_rx_iv_to_pn(rtwdev, key,
668 			  aoac_rpt->gtk_rx_iv[key->keyidx]);
669 
670 	if (!data.igtk_cipher)
671 		return;
672 
673 	key = rtw89_wow_gtk_rekey(rtwdev, data.igtk_cipher, aoac_rpt->igtk_key_id,
674 				  aoac_rpt->igtk);
675 	if (!key)
676 		return;
677 
678 	rtw89_rx_pn_set_pmf(rtwdev, key, aoac_rpt->igtk_ipn);
679 
680 	rcu_read_lock();
681 
682 	bss_conf = rtw89_vif_rcu_dereference_link(rtwvif_link, true);
683 	ieee80211_gtk_rekey_notify(wow_vif, bss_conf->bssid,
684 				   aoac_rpt->eapol_key_replay_count,
685 				   GFP_ATOMIC);
686 
687 	rcu_read_unlock();
688 }
689 
690 static void rtw89_wow_leave_deep_ps(struct rtw89_dev *rtwdev)
691 {
692 	__rtw89_leave_ps_mode(rtwdev);
693 }
694 
695 static void rtw89_wow_enter_deep_ps(struct rtw89_dev *rtwdev)
696 {
697 	__rtw89_enter_ps_mode(rtwdev);
698 }
699 
700 static void rtw89_wow_enter_ps(struct rtw89_dev *rtwdev)
701 {
702 	struct rtw89_vif_link *rtwvif_link = rtwdev->wow.rtwvif_link;
703 
704 	if (rtw89_wow_mgd_linked(rtwdev))
705 		rtw89_enter_lps(rtwdev, rtwvif_link->rtwvif, false);
706 	else if (rtw89_wow_no_link(rtwdev))
707 		rtw89_fw_h2c_fwips(rtwdev, rtwvif_link, true);
708 }
709 
710 static void rtw89_wow_leave_ps(struct rtw89_dev *rtwdev, bool enable_wow)
711 {
712 	struct rtw89_vif_link *rtwvif_link = rtwdev->wow.rtwvif_link;
713 
714 	if (rtw89_wow_mgd_linked(rtwdev)) {
715 		rtw89_leave_lps(rtwdev);
716 	} else if (rtw89_wow_no_link(rtwdev)) {
717 		if (enable_wow)
718 			rtw89_leave_ips(rtwdev);
719 		else
720 			rtw89_fw_h2c_fwips(rtwdev, rtwvif_link, false);
721 	}
722 }
723 
724 static int rtw89_wow_config_mac(struct rtw89_dev *rtwdev, bool enable_wow)
725 {
726 	const struct rtw89_mac_gen_def *mac = rtwdev->chip->mac_def;
727 
728 	return mac->wow_config_mac(rtwdev, enable_wow);
729 }
730 
731 static void rtw89_wow_set_rx_filter(struct rtw89_dev *rtwdev, bool enable)
732 {
733 	const struct rtw89_mac_gen_def *mac = rtwdev->chip->mac_def;
734 	enum rtw89_mac_fwd_target fwd_target = enable ?
735 					       RTW89_FWD_DONT_CARE :
736 					       RTW89_FWD_TO_HOST;
737 
738 	mac->typ_fltr_opt(rtwdev, RTW89_MGNT, fwd_target, RTW89_MAC_0);
739 	mac->typ_fltr_opt(rtwdev, RTW89_CTRL, fwd_target, RTW89_MAC_0);
740 	mac->typ_fltr_opt(rtwdev, RTW89_DATA, fwd_target, RTW89_MAC_0);
741 }
742 
743 static void rtw89_wow_show_wakeup_reason(struct rtw89_dev *rtwdev)
744 {
745 	struct rtw89_vif_link *rtwvif_link = rtwdev->wow.rtwvif_link;
746 	struct ieee80211_vif *wow_vif = rtwvif_link_to_vif(rtwvif_link);
747 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
748 	struct rtw89_wow_aoac_report *aoac_rpt = &rtw_wow->aoac_rpt;
749 	struct cfg80211_wowlan_nd_info nd_info;
750 	struct cfg80211_wowlan_wakeup wakeup = {
751 		.pattern_idx = -1,
752 	};
753 	u32 wow_reason_reg;
754 	u8 reason;
755 
756 	if (RTW89_CHK_FW_FEATURE(WOW_REASON_V1, &rtwdev->fw))
757 		wow_reason_reg = rtwdev->chip->wow_reason_reg[RTW89_WOW_REASON_V1];
758 	else
759 		wow_reason_reg = rtwdev->chip->wow_reason_reg[RTW89_WOW_REASON_V0];
760 
761 	reason = rtw89_read8(rtwdev, wow_reason_reg);
762 	switch (reason) {
763 	case RTW89_WOW_RSN_RX_DEAUTH:
764 		wakeup.disconnect = true;
765 		rtw89_debug(rtwdev, RTW89_DBG_WOW, "WOW: Rx deauth\n");
766 		break;
767 	case RTW89_WOW_RSN_DISCONNECT:
768 		wakeup.disconnect = true;
769 		rtw89_debug(rtwdev, RTW89_DBG_WOW, "WOW: AP is off\n");
770 		break;
771 	case RTW89_WOW_RSN_RX_MAGIC_PKT:
772 		wakeup.magic_pkt = true;
773 		rtw89_debug(rtwdev, RTW89_DBG_WOW, "WOW: Rx magic packet\n");
774 		break;
775 	case RTW89_WOW_RSN_RX_GTK_REKEY:
776 		wakeup.gtk_rekey_failure = true;
777 		rtw89_debug(rtwdev, RTW89_DBG_WOW, "WOW: Rx gtk rekey\n");
778 		break;
779 	case RTW89_WOW_RSN_RX_PATTERN_MATCH:
780 		wakeup.pattern_idx = aoac_rpt->pattern_idx;
781 		rtw89_debug(rtwdev, RTW89_DBG_WOW, "WOW: Rx pattern match packet\n");
782 		break;
783 	case RTW89_WOW_RSN_RX_NLO:
784 		/* Current firmware and driver don't report ssid index.
785 		 * Use 0 for n_matches based on its comment.
786 		 */
787 		nd_info.n_matches = 0;
788 		wakeup.net_detect = &nd_info;
789 		rtw89_debug(rtwdev, RTW89_DBG_WOW, "Rx NLO\n");
790 		break;
791 	default:
792 		rtw89_warn(rtwdev, "Unknown wakeup reason %x\n", reason);
793 		ieee80211_report_wowlan_wakeup(wow_vif, NULL, GFP_KERNEL);
794 		return;
795 	}
796 
797 	ieee80211_report_wowlan_wakeup(wow_vif, &wakeup, GFP_KERNEL);
798 }
799 
800 static void rtw89_wow_vif_iter(struct rtw89_dev *rtwdev,
801 			       struct rtw89_vif_link *rtwvif_link)
802 {
803 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
804 	struct ieee80211_vif *vif = rtwvif_link_to_vif(rtwvif_link);
805 
806 	/* Current WoWLAN function support setting of only vif in
807 	 * infra mode or no link mode. When one suitable vif is found,
808 	 * stop the iteration.
809 	 */
810 	if (rtw_wow->rtwvif_link || vif->type != NL80211_IFTYPE_STATION)
811 		return;
812 
813 	switch (rtwvif_link->net_type) {
814 	case RTW89_NET_TYPE_INFRA:
815 		if (rtw_wow_has_mgd_features(rtwdev))
816 			rtw_wow->rtwvif_link = rtwvif_link;
817 		break;
818 	case RTW89_NET_TYPE_NO_LINK:
819 		if (rtw_wow->pno_inited)
820 			rtw_wow->rtwvif_link = rtwvif_link;
821 		break;
822 	default:
823 		break;
824 	}
825 }
826 
827 static u16 __rtw89_cal_crc16(u8 data, u16 crc)
828 {
829 	u8 shift_in, data_bit;
830 	u8 crc_bit4, crc_bit11, crc_bit15;
831 	u16 crc_result;
832 	int index;
833 
834 	for (index = 0; index < 8; index++) {
835 		crc_bit15 = crc & BIT(15) ? 1 : 0;
836 		data_bit = data & BIT(index) ? 1 : 0;
837 		shift_in = crc_bit15 ^ data_bit;
838 
839 		crc_result = crc << 1;
840 
841 		if (shift_in == 0)
842 			crc_result &= ~BIT(0);
843 		else
844 			crc_result |= BIT(0);
845 
846 		crc_bit11 = (crc & BIT(11) ? 1 : 0) ^ shift_in;
847 
848 		if (crc_bit11 == 0)
849 			crc_result &= ~BIT(12);
850 		else
851 			crc_result |= BIT(12);
852 
853 		crc_bit4 = (crc & BIT(4) ? 1 : 0) ^ shift_in;
854 
855 		if (crc_bit4 == 0)
856 			crc_result &= ~BIT(5);
857 		else
858 			crc_result |= BIT(5);
859 
860 		crc = crc_result;
861 	}
862 	return crc;
863 }
864 
865 static u16 rtw89_calc_crc(u8 *pdata, int length)
866 {
867 	u16 crc = 0xffff;
868 	int i;
869 
870 	for (i = 0; i < length; i++)
871 		crc = __rtw89_cal_crc16(pdata[i], crc);
872 
873 	/* get 1' complement */
874 	return ~crc;
875 }
876 
877 static int rtw89_wow_pattern_get_type(struct rtw89_vif_link *rtwvif_link,
878 				      struct rtw89_wow_cam_info *rtw_pattern,
879 				      const u8 *pattern, u8 da_mask)
880 {
881 	u8 da[ETH_ALEN];
882 
883 	ether_addr_copy_mask(da, pattern, da_mask);
884 
885 	/* Each pattern is divided into different kinds by DA address
886 	 *  a. DA is broadcast address: set bc = 0;
887 	 *  b. DA is multicast address: set mc = 0
888 	 *  c. DA is unicast address same as dev's mac address: set uc = 0
889 	 *  d. DA is unmasked. Also called wildcard type: set uc = bc = mc = 0
890 	 *  e. Others is invalid type.
891 	 */
892 
893 	if (is_broadcast_ether_addr(da))
894 		rtw_pattern->bc = true;
895 	else if (is_multicast_ether_addr(da))
896 		rtw_pattern->mc = true;
897 	else if (ether_addr_equal(da, rtwvif_link->mac_addr) &&
898 		 da_mask == GENMASK(5, 0))
899 		rtw_pattern->uc = true;
900 	else if (!da_mask) /*da_mask == 0 mean wildcard*/
901 		return 0;
902 	else
903 		return -EPERM;
904 
905 	return 0;
906 }
907 
908 static int rtw89_wow_pattern_generate(struct rtw89_dev *rtwdev,
909 				      struct rtw89_vif_link *rtwvif_link,
910 				      const struct cfg80211_pkt_pattern *pkt_pattern,
911 				      struct rtw89_wow_cam_info *rtw_pattern)
912 {
913 	u8 mask_hw[RTW89_MAX_PATTERN_MASK_SIZE * 4] = {0};
914 	u8 content[RTW89_MAX_PATTERN_SIZE] = {0};
915 	const u8 *mask;
916 	const u8 *pattern;
917 	u8 mask_len;
918 	u16 count;
919 	u32 len;
920 	int i, ret;
921 
922 	pattern = pkt_pattern->pattern;
923 	len = pkt_pattern->pattern_len;
924 	mask = pkt_pattern->mask;
925 	mask_len = DIV_ROUND_UP(len, 8);
926 	memset(rtw_pattern, 0, sizeof(*rtw_pattern));
927 
928 	ret = rtw89_wow_pattern_get_type(rtwvif_link, rtw_pattern, pattern,
929 					 mask[0] & GENMASK(5, 0));
930 	if (ret)
931 		return ret;
932 
933 	/* translate mask from os to mask for hw
934 	 * pattern from OS uses 'ethenet frame', like this:
935 	 * |    6   |    6   |   2  |     20    |  Variable  |  4  |
936 	 * |--------+--------+------+-----------+------------+-----|
937 	 * |    802.3 Mac Header    | IP Header | TCP Packet | FCS |
938 	 * |   DA   |   SA   | Type |
939 	 *
940 	 * BUT, packet catched by our HW is in '802.11 frame', begin from LLC
941 	 * |     24 or 30      |    6   |   2  |     20    |  Variable  |  4  |
942 	 * |-------------------+--------+------+-----------+------------+-----|
943 	 * | 802.11 MAC Header |       LLC     | IP Header | TCP Packet | FCS |
944 	 *		       | Others | Tpye |
945 	 *
946 	 * Therefore, we need translate mask_from_OS to mask_to_hw.
947 	 * We should left-shift mask by 6 bits, then set the new bit[0~5] = 0,
948 	 * because new mask[0~5] means 'SA', but our HW packet begins from LLC,
949 	 * bit[0~5] corresponds to first 6 Bytes in LLC, they just don't match.
950 	 */
951 
952 	/* Shift 6 bits */
953 	for (i = 0; i < mask_len - 1; i++) {
954 		mask_hw[i] = u8_get_bits(mask[i], GENMASK(7, 6)) |
955 			     u8_get_bits(mask[i + 1], GENMASK(5, 0)) << 2;
956 	}
957 	mask_hw[i] = u8_get_bits(mask[i], GENMASK(7, 6));
958 
959 	/* Set bit 0-5 to zero */
960 	mask_hw[0] &= ~GENMASK(5, 0);
961 
962 	memcpy(rtw_pattern->mask, mask_hw, sizeof(rtw_pattern->mask));
963 
964 	/* To get the wake up pattern from the mask.
965 	 * We do not count first 12 bits which means
966 	 * DA[6] and SA[6] in the pattern to match HW design.
967 	 */
968 	count = 0;
969 	for (i = 12; i < len; i++) {
970 		if ((mask[i / 8] >> (i % 8)) & 0x01) {
971 			content[count] = pattern[i];
972 			count++;
973 		}
974 	}
975 
976 	rtw_pattern->crc = rtw89_calc_crc(content, count);
977 
978 	return 0;
979 }
980 
981 static int rtw89_wow_parse_patterns(struct rtw89_dev *rtwdev,
982 				    struct rtw89_vif_link *rtwvif_link,
983 				    struct cfg80211_wowlan *wowlan)
984 {
985 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
986 	struct rtw89_wow_cam_info *rtw_pattern = rtw_wow->patterns;
987 	int i;
988 	int ret;
989 
990 	if (!wowlan->n_patterns || !wowlan->patterns)
991 		return 0;
992 
993 	for (i = 0; i < wowlan->n_patterns; i++) {
994 		rtw_pattern = &rtw_wow->patterns[i];
995 		ret = rtw89_wow_pattern_generate(rtwdev, rtwvif_link,
996 						 &wowlan->patterns[i],
997 						 rtw_pattern);
998 		if (ret) {
999 			rtw89_err(rtwdev, "failed to generate pattern(%d)\n", i);
1000 			rtw_wow->pattern_cnt = 0;
1001 			return ret;
1002 		}
1003 
1004 		rtw_pattern->r_w = true;
1005 		rtw_pattern->idx = i;
1006 		rtw_pattern->negative_pattern_match = false;
1007 		rtw_pattern->skip_mac_hdr = true;
1008 		rtw_pattern->valid = true;
1009 	}
1010 	rtw_wow->pattern_cnt = wowlan->n_patterns;
1011 
1012 	return 0;
1013 }
1014 
1015 static void rtw89_wow_pattern_clear_cam(struct rtw89_dev *rtwdev)
1016 {
1017 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1018 	struct rtw89_wow_cam_info *rtw_pattern = rtw_wow->patterns;
1019 	int i = 0;
1020 
1021 	for (i = 0; i < rtw_wow->pattern_cnt; i++) {
1022 		rtw_pattern = &rtw_wow->patterns[i];
1023 		rtw_pattern->valid = false;
1024 		rtw89_fw_wow_cam_update(rtwdev, rtw_pattern);
1025 	}
1026 }
1027 
1028 static void rtw89_wow_pattern_write(struct rtw89_dev *rtwdev)
1029 {
1030 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1031 	struct rtw89_wow_cam_info *rtw_pattern = rtw_wow->patterns;
1032 	int i;
1033 
1034 	for (i = 0; i < rtw_wow->pattern_cnt; i++)
1035 		rtw89_fw_wow_cam_update(rtwdev, rtw_pattern + i);
1036 }
1037 
1038 static void rtw89_wow_pattern_clear(struct rtw89_dev *rtwdev)
1039 {
1040 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1041 
1042 	rtw89_wow_pattern_clear_cam(rtwdev);
1043 
1044 	rtw_wow->pattern_cnt = 0;
1045 	memset(rtw_wow->patterns, 0, sizeof(rtw_wow->patterns));
1046 }
1047 
1048 static void rtw89_wow_clear_wakeups(struct rtw89_dev *rtwdev)
1049 {
1050 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1051 
1052 	rtw_wow->rtwvif_link = NULL;
1053 	rtw89_core_release_all_bits_map(rtw_wow->flags, RTW89_WOW_FLAG_NUM);
1054 	rtw_wow->pattern_cnt = 0;
1055 	rtw_wow->pno_inited = false;
1056 }
1057 
1058 static void rtw89_wow_init_pno(struct rtw89_dev *rtwdev,
1059 			       struct cfg80211_sched_scan_request *nd_config)
1060 {
1061 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1062 
1063 	if (!nd_config->n_match_sets || !nd_config->n_channels)
1064 		return;
1065 
1066 	rtw_wow->nd_config = nd_config;
1067 	rtw_wow->pno_inited = true;
1068 
1069 	INIT_LIST_HEAD(&rtw_wow->pno_pkt_list);
1070 
1071 	rtw89_debug(rtwdev, RTW89_DBG_WOW, "WOW: net-detect is enabled\n");
1072 }
1073 
1074 static int rtw89_wow_set_wakeups(struct rtw89_dev *rtwdev,
1075 				 struct cfg80211_wowlan *wowlan)
1076 {
1077 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1078 	struct rtw89_vif_link *rtwvif_link;
1079 	struct rtw89_vif *rtwvif;
1080 
1081 	if (wowlan->disconnect)
1082 		set_bit(RTW89_WOW_FLAG_EN_DISCONNECT, rtw_wow->flags);
1083 	if (wowlan->magic_pkt)
1084 		set_bit(RTW89_WOW_FLAG_EN_MAGIC_PKT, rtw_wow->flags);
1085 	if (wowlan->n_patterns && wowlan->patterns)
1086 		set_bit(RTW89_WOW_FLAG_EN_PATTERN, rtw_wow->flags);
1087 
1088 	if (wowlan->nd_config)
1089 		rtw89_wow_init_pno(rtwdev, wowlan->nd_config);
1090 
1091 	rtw89_for_each_rtwvif(rtwdev, rtwvif) {
1092 		/* use the link on HW-0 to do wow flow */
1093 		rtwvif_link = rtw89_vif_get_link_inst(rtwvif, 0);
1094 		if (!rtwvif_link)
1095 			continue;
1096 
1097 		rtw89_wow_vif_iter(rtwdev, rtwvif_link);
1098 	}
1099 
1100 	rtwvif_link = rtw_wow->rtwvif_link;
1101 	if (!rtwvif_link)
1102 		return -EPERM;
1103 
1104 	return rtw89_wow_parse_patterns(rtwdev, rtwvif_link, wowlan);
1105 }
1106 
1107 static int rtw89_wow_cfg_wake_pno(struct rtw89_dev *rtwdev, bool wow)
1108 {
1109 	struct rtw89_vif_link *rtwvif_link = rtwdev->wow.rtwvif_link;
1110 	int ret;
1111 
1112 	ret = rtw89_fw_h2c_cfg_pno(rtwdev, rtwvif_link, true);
1113 	if (ret) {
1114 		rtw89_err(rtwdev, "failed to config pno\n");
1115 		return ret;
1116 	}
1117 
1118 	ret = rtw89_fw_h2c_wow_wakeup_ctrl(rtwdev, rtwvif_link, wow);
1119 	if (ret) {
1120 		rtw89_err(rtwdev, "failed to fw wow wakeup ctrl\n");
1121 		return ret;
1122 	}
1123 
1124 	ret = rtw89_fw_h2c_wow_global(rtwdev, rtwvif_link, wow);
1125 	if (ret) {
1126 		rtw89_err(rtwdev, "failed to fw wow global\n");
1127 		return ret;
1128 	}
1129 
1130 	return 0;
1131 }
1132 
1133 static int rtw89_wow_cfg_wake(struct rtw89_dev *rtwdev, bool wow)
1134 {
1135 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1136 	struct rtw89_vif_link *rtwvif_link = rtw_wow->rtwvif_link;
1137 	struct ieee80211_vif *wow_vif = rtwvif_link_to_vif(rtwvif_link);
1138 	struct ieee80211_sta *wow_sta;
1139 	struct rtw89_sta_link *rtwsta_link = NULL;
1140 	struct rtw89_sta *rtwsta;
1141 	int ret;
1142 
1143 	wow_sta = ieee80211_find_sta(wow_vif, wow_vif->cfg.ap_addr);
1144 	if (wow_sta) {
1145 		rtwsta = sta_to_rtwsta(wow_sta);
1146 		rtwsta_link = rtwsta->links[rtwvif_link->link_id];
1147 		if (!rtwsta_link)
1148 			return -ENOLINK;
1149 	}
1150 
1151 	if (wow) {
1152 		if (rtw_wow->pattern_cnt)
1153 			rtwvif_link->wowlan_pattern = true;
1154 		if (test_bit(RTW89_WOW_FLAG_EN_MAGIC_PKT, rtw_wow->flags))
1155 			rtwvif_link->wowlan_magic = true;
1156 	} else {
1157 		rtwvif_link->wowlan_pattern = false;
1158 		rtwvif_link->wowlan_magic = false;
1159 	}
1160 
1161 	ret = rtw89_fw_h2c_wow_wakeup_ctrl(rtwdev, rtwvif_link, wow);
1162 	if (ret) {
1163 		rtw89_err(rtwdev, "failed to fw wow wakeup ctrl\n");
1164 		return ret;
1165 	}
1166 
1167 	if (wow) {
1168 		ret = rtw89_chip_h2c_dctl_sec_cam(rtwdev, rtwvif_link, rtwsta_link);
1169 		if (ret) {
1170 			rtw89_err(rtwdev, "failed to update dctl cam sec entry: %d\n",
1171 				  ret);
1172 			return ret;
1173 		}
1174 	}
1175 
1176 	ret = rtw89_fw_h2c_cam(rtwdev, rtwvif_link, rtwsta_link, NULL);
1177 	if (ret) {
1178 		rtw89_warn(rtwdev, "failed to send h2c cam\n");
1179 		return ret;
1180 	}
1181 
1182 	ret = rtw89_fw_h2c_wow_global(rtwdev, rtwvif_link, wow);
1183 	if (ret) {
1184 		rtw89_err(rtwdev, "failed to fw wow global\n");
1185 		return ret;
1186 	}
1187 
1188 	return 0;
1189 }
1190 
1191 static int rtw89_wow_check_fw_status(struct rtw89_dev *rtwdev, bool wow_enable)
1192 {
1193 	const struct rtw89_mac_gen_def *mac = rtwdev->chip->mac_def;
1194 	u8 polling;
1195 	int ret;
1196 
1197 	ret = read_poll_timeout_atomic(rtw89_read8_mask, polling,
1198 				       wow_enable == !!polling,
1199 				       50, 50000, false, rtwdev,
1200 				       mac->wow_ctrl.addr, mac->wow_ctrl.mask);
1201 	if (ret)
1202 		rtw89_err(rtwdev, "failed to check wow status %s\n",
1203 			  wow_enable ? "enabled" : "disabled");
1204 	return ret;
1205 }
1206 
1207 static int rtw89_wow_swap_fw(struct rtw89_dev *rtwdev, bool wow)
1208 {
1209 	enum rtw89_fw_type fw_type = wow ? RTW89_FW_WOWLAN : RTW89_FW_NORMAL;
1210 	enum rtw89_chip_gen chip_gen = rtwdev->chip->chip_gen;
1211 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1212 	struct rtw89_vif_link *rtwvif_link = rtw_wow->rtwvif_link;
1213 	struct ieee80211_vif *wow_vif = rtwvif_link_to_vif(rtwvif_link);
1214 	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
1215 	const struct rtw89_chip_info *chip = rtwdev->chip;
1216 	bool include_bb = !!chip->bbmcu_nr;
1217 	bool disable_intr_for_dlfw = false;
1218 	struct ieee80211_sta *wow_sta;
1219 	struct rtw89_sta_link *rtwsta_link = NULL;
1220 	struct rtw89_sta *rtwsta;
1221 	bool is_conn = true;
1222 	int ret;
1223 
1224 	if (chip_id == RTL8852C || chip_id == RTL8922A)
1225 		disable_intr_for_dlfw = true;
1226 
1227 	wow_sta = ieee80211_find_sta(wow_vif, wow_vif->cfg.ap_addr);
1228 	if (wow_sta) {
1229 		rtwsta = sta_to_rtwsta(wow_sta);
1230 		rtwsta_link = rtwsta->links[rtwvif_link->link_id];
1231 		if (!rtwsta_link)
1232 			return -ENOLINK;
1233 	} else {
1234 		is_conn = false;
1235 	}
1236 
1237 	if (disable_intr_for_dlfw)
1238 		rtw89_hci_disable_intr(rtwdev);
1239 
1240 	ret = rtw89_fw_download(rtwdev, fw_type, include_bb);
1241 	if (ret) {
1242 		rtw89_warn(rtwdev, "download fw failed\n");
1243 		return ret;
1244 	}
1245 
1246 	if (disable_intr_for_dlfw)
1247 		rtw89_hci_enable_intr(rtwdev);
1248 
1249 	rtw89_phy_init_rf_reg(rtwdev, true);
1250 
1251 	ret = rtw89_fw_h2c_role_maintain(rtwdev, rtwvif_link, rtwsta_link,
1252 					 RTW89_ROLE_FW_RESTORE);
1253 	if (ret) {
1254 		rtw89_warn(rtwdev, "failed to send h2c role maintain\n");
1255 		return ret;
1256 	}
1257 
1258 	ret = rtw89_chip_h2c_assoc_cmac_tbl(rtwdev, rtwvif_link, rtwsta_link);
1259 	if (ret) {
1260 		rtw89_warn(rtwdev, "failed to send h2c assoc cmac tbl\n");
1261 		return ret;
1262 	}
1263 
1264 	if (!is_conn)
1265 		rtw89_cam_reset_keys(rtwdev);
1266 
1267 	ret = rtw89_fw_h2c_join_info(rtwdev, rtwvif_link, rtwsta_link, !is_conn);
1268 	if (ret) {
1269 		rtw89_warn(rtwdev, "failed to send h2c join info\n");
1270 		return ret;
1271 	}
1272 
1273 	ret = rtw89_fw_h2c_cam(rtwdev, rtwvif_link, rtwsta_link, NULL);
1274 	if (ret) {
1275 		rtw89_warn(rtwdev, "failed to send h2c cam\n");
1276 		return ret;
1277 	}
1278 
1279 	if (is_conn) {
1280 		ret = rtw89_fw_h2c_general_pkt(rtwdev, rtwvif_link, rtwsta_link->mac_id);
1281 		if (ret) {
1282 			rtw89_warn(rtwdev, "failed to send h2c general packet\n");
1283 			return ret;
1284 		}
1285 		rtw89_phy_ra_assoc(rtwdev, rtwsta_link);
1286 		rtw89_phy_set_bss_color(rtwdev, rtwvif_link);
1287 		rtw89_chip_cfg_txpwr_ul_tb_offset(rtwdev, rtwvif_link);
1288 	}
1289 
1290 	if (chip_gen == RTW89_CHIP_BE)
1291 		rtw89_phy_rfk_pre_ntfy_and_wait(rtwdev, RTW89_PHY_0, 5);
1292 
1293 	rtw89_mac_hw_mgnt_sec(rtwdev, wow);
1294 
1295 	return 0;
1296 }
1297 
1298 static int rtw89_wow_enable_trx_pre(struct rtw89_dev *rtwdev)
1299 {
1300 	int ret;
1301 
1302 	rtw89_hci_ctrl_txdma_ch(rtwdev, false);
1303 	rtw89_hci_ctrl_txdma_fw_ch(rtwdev, true);
1304 
1305 	rtw89_mac_ptk_drop_by_band_and_wait(rtwdev, RTW89_MAC_0);
1306 
1307 	ret = rtw89_hci_poll_txdma_ch_idle(rtwdev);
1308 	if (ret) {
1309 		rtw89_err(rtwdev, "txdma ch busy\n");
1310 		return ret;
1311 	}
1312 	rtw89_wow_set_rx_filter(rtwdev, true);
1313 
1314 	ret = rtw89_mac_cfg_ppdu_status(rtwdev, RTW89_MAC_0, false);
1315 	if (ret) {
1316 		rtw89_err(rtwdev, "cfg ppdu status\n");
1317 		return ret;
1318 	}
1319 
1320 	return 0;
1321 }
1322 
1323 static int rtw89_wow_enable_trx_post(struct rtw89_dev *rtwdev)
1324 {
1325 	int ret;
1326 
1327 	rtw89_hci_disable_intr(rtwdev);
1328 	rtw89_hci_ctrl_trxhci(rtwdev, false);
1329 
1330 	ret = rtw89_hci_poll_txdma_ch_idle(rtwdev);
1331 	if (ret) {
1332 		rtw89_err(rtwdev, "failed to poll txdma ch idle pcie\n");
1333 		return ret;
1334 	}
1335 
1336 	ret = rtw89_wow_config_mac(rtwdev, true);
1337 	if (ret) {
1338 		rtw89_err(rtwdev, "failed to config mac\n");
1339 		return ret;
1340 	}
1341 
1342 	rtw89_wow_set_rx_filter(rtwdev, false);
1343 	rtw89_hci_reset(rtwdev);
1344 
1345 	return 0;
1346 }
1347 
1348 static int rtw89_wow_disable_trx_pre(struct rtw89_dev *rtwdev)
1349 {
1350 	int ret;
1351 
1352 	rtw89_hci_clr_idx_all(rtwdev);
1353 
1354 	ret = rtw89_hci_rst_bdram(rtwdev);
1355 	if (ret) {
1356 		rtw89_warn(rtwdev, "reset bdram busy\n");
1357 		return ret;
1358 	}
1359 
1360 	rtw89_hci_ctrl_trxhci(rtwdev, true);
1361 	rtw89_hci_ctrl_txdma_ch(rtwdev, true);
1362 
1363 	ret = rtw89_wow_config_mac(rtwdev, false);
1364 	if (ret) {
1365 		rtw89_err(rtwdev, "failed to config mac\n");
1366 		return ret;
1367 	}
1368 
1369 	/* Before enabling interrupt, we need to get AOAC report by reg due to RX
1370 	 * not enabled yet. Also, we need to sync RX related IV from firmware to
1371 	 * mac80211 before receiving RX packets from driver.
1372 	 * After enabling interrupt, we can get AOAC report from h2c and c2h, and
1373 	 * can get TX IV and complete rekey info. We need to update TX related IV
1374 	 * and new GTK info if rekey happened.
1375 	 */
1376 	ret = rtw89_wow_get_aoac_rpt(rtwdev, false);
1377 	if (!ret)
1378 		rtw89_wow_update_key_info(rtwdev, false);
1379 
1380 	rtw89_hci_enable_intr(rtwdev);
1381 	ret = rtw89_wow_get_aoac_rpt(rtwdev, true);
1382 	if (!ret)
1383 		rtw89_wow_update_key_info(rtwdev, true);
1384 
1385 	return 0;
1386 }
1387 
1388 static int rtw89_wow_disable_trx_post(struct rtw89_dev *rtwdev)
1389 {
1390 	struct rtw89_vif_link *rtwvif_link = rtwdev->wow.rtwvif_link;
1391 	int ret;
1392 
1393 	ret = rtw89_mac_cfg_ppdu_status(rtwdev, RTW89_MAC_0, true);
1394 	if (ret)
1395 		rtw89_err(rtwdev, "cfg ppdu status\n");
1396 
1397 	rtw89_fw_h2c_set_bcn_fltr_cfg(rtwdev, rtwvif_link, true);
1398 
1399 	return ret;
1400 }
1401 
1402 static void rtw89_fw_release_pno_pkt_list(struct rtw89_dev *rtwdev,
1403 					  struct rtw89_vif_link *rtwvif_link)
1404 {
1405 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1406 	struct list_head *pkt_list = &rtw_wow->pno_pkt_list;
1407 	struct rtw89_pktofld_info *info, *tmp;
1408 
1409 	list_for_each_entry_safe(info, tmp, pkt_list, list) {
1410 		rtw89_fw_h2c_del_pkt_offload(rtwdev, info->id);
1411 		list_del(&info->list);
1412 		kfree(info);
1413 	}
1414 }
1415 
1416 static int rtw89_pno_scan_update_probe_req(struct rtw89_dev *rtwdev,
1417 					   struct rtw89_vif_link *rtwvif_link)
1418 {
1419 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1420 	struct cfg80211_sched_scan_request *nd_config = rtw_wow->nd_config;
1421 	u8 num = nd_config->n_match_sets, i;
1422 	struct rtw89_pktofld_info *info;
1423 	struct sk_buff *skb;
1424 	int ret;
1425 
1426 	for (i = 0; i < num; i++) {
1427 		skb = ieee80211_probereq_get(rtwdev->hw, rtwvif_link->mac_addr,
1428 					     nd_config->match_sets[i].ssid.ssid,
1429 					     nd_config->match_sets[i].ssid.ssid_len,
1430 					     nd_config->ie_len);
1431 		if (!skb)
1432 			return -ENOMEM;
1433 
1434 		skb_put_data(skb, nd_config->ie, nd_config->ie_len);
1435 
1436 		info = kzalloc(sizeof(*info), GFP_KERNEL);
1437 		if (!info) {
1438 			kfree_skb(skb);
1439 			rtw89_fw_release_pno_pkt_list(rtwdev, rtwvif_link);
1440 			return -ENOMEM;
1441 		}
1442 
1443 		ret = rtw89_fw_h2c_add_pkt_offload(rtwdev, &info->id, skb);
1444 		if (ret) {
1445 			kfree_skb(skb);
1446 			kfree(info);
1447 			rtw89_fw_release_pno_pkt_list(rtwdev, rtwvif_link);
1448 			return ret;
1449 		}
1450 
1451 		list_add_tail(&info->list, &rtw_wow->pno_pkt_list);
1452 		kfree_skb(skb);
1453 	}
1454 
1455 	return 0;
1456 }
1457 
1458 static int rtw89_pno_scan_offload(struct rtw89_dev *rtwdev, bool enable)
1459 {
1460 	const struct rtw89_mac_gen_def *mac = rtwdev->chip->mac_def;
1461 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1462 	struct rtw89_vif_link *rtwvif_link = rtwdev->wow.rtwvif_link;
1463 	int interval = rtw_wow->nd_config->scan_plans[0].interval;
1464 	struct rtw89_scan_option opt = {};
1465 	int ret;
1466 
1467 	if (enable) {
1468 		ret = rtw89_pno_scan_update_probe_req(rtwdev, rtwvif_link);
1469 		if (ret) {
1470 			rtw89_err(rtwdev, "Update probe request failed\n");
1471 			return ret;
1472 		}
1473 
1474 		ret = mac->add_chan_list_pno(rtwdev, rtwvif_link);
1475 		if (ret) {
1476 			rtw89_err(rtwdev, "Update channel list failed\n");
1477 			return ret;
1478 		}
1479 	}
1480 
1481 	opt.enable = enable;
1482 	opt.repeat = RTW89_SCAN_NORMAL;
1483 	opt.norm_pd = max(interval, 1) * 10; /* in unit of 100ms */
1484 	opt.delay = max(rtw_wow->nd_config->delay, 1);
1485 
1486 	if (rtwdev->chip->chip_gen == RTW89_CHIP_BE) {
1487 		opt.operation = enable ? RTW89_SCAN_OP_START : RTW89_SCAN_OP_STOP;
1488 		opt.scan_mode = RTW89_SCAN_MODE_SA;
1489 		opt.band = RTW89_PHY_0;
1490 		opt.num_macc_role = 0;
1491 		opt.mlo_mode = rtwdev->mlo_dbcc_mode;
1492 		opt.num_opch = 0;
1493 		opt.opch_end = RTW89_CHAN_INVALID;
1494 	}
1495 
1496 	mac->scan_offload(rtwdev, &opt, rtwvif_link, true);
1497 
1498 	return 0;
1499 }
1500 
1501 static int rtw89_wow_fw_start(struct rtw89_dev *rtwdev)
1502 {
1503 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1504 	struct rtw89_vif_link *rtwvif_link = rtw_wow->rtwvif_link;
1505 	int ret;
1506 
1507 	if (rtw89_wow_no_link(rtwdev)) {
1508 		ret = rtw89_pno_scan_offload(rtwdev, false);
1509 		if (ret) {
1510 			rtw89_err(rtwdev, "wow: failed to disable pno scan offload\n");
1511 			return ret;
1512 		}
1513 
1514 		ret = rtw89_pno_scan_offload(rtwdev, true);
1515 		if (ret) {
1516 			rtw89_err(rtwdev, "wow: failed to enable pno scan offload\n");
1517 			return ret;
1518 		}
1519 	} else {
1520 		rtw89_wow_pattern_write(rtwdev);
1521 		rtw89_wow_construct_key_info(rtwdev);
1522 
1523 		ret = rtw89_fw_h2c_keep_alive(rtwdev, rtwvif_link, true);
1524 		if (ret) {
1525 			rtw89_err(rtwdev, "wow: failed to enable keep alive\n");
1526 			return ret;
1527 		}
1528 
1529 		ret = rtw89_fw_h2c_disconnect_detect(rtwdev, rtwvif_link, true);
1530 		if (ret) {
1531 			rtw89_err(rtwdev, "wow: failed to enable disconnect detect\n");
1532 			return ret;
1533 		}
1534 
1535 		ret = rtw89_fw_h2c_wow_gtk_ofld(rtwdev, rtwvif_link, true);
1536 		if (ret) {
1537 			rtw89_err(rtwdev, "wow: failed to enable GTK offload\n");
1538 			return ret;
1539 		}
1540 
1541 		ret = rtw89_fw_h2c_arp_offload(rtwdev, rtwvif_link, true);
1542 		if (ret)
1543 			rtw89_warn(rtwdev, "wow: failed to enable arp offload\n");
1544 	}
1545 
1546 	if (rtw89_wow_no_link(rtwdev)) {
1547 		ret = rtw89_wow_cfg_wake_pno(rtwdev, true);
1548 		if (ret) {
1549 			rtw89_err(rtwdev, "wow: failed to config wake PNO\n");
1550 			return ret;
1551 		}
1552 	} else {
1553 		ret = rtw89_wow_cfg_wake(rtwdev, true);
1554 		if (ret) {
1555 			rtw89_err(rtwdev, "wow: failed to config wake\n");
1556 			return ret;
1557 		}
1558 	}
1559 
1560 	ret = rtw89_wow_check_fw_status(rtwdev, true);
1561 	if (ret) {
1562 		rtw89_err(rtwdev, "wow: failed to check enable fw ready\n");
1563 		return ret;
1564 	}
1565 
1566 	return 0;
1567 }
1568 
1569 static int rtw89_wow_fw_stop(struct rtw89_dev *rtwdev)
1570 {
1571 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
1572 	struct rtw89_vif_link *rtwvif_link = rtw_wow->rtwvif_link;
1573 	int ret;
1574 
1575 	if (rtw89_wow_no_link(rtwdev)) {
1576 		ret = rtw89_pno_scan_offload(rtwdev, false);
1577 		if (ret) {
1578 			rtw89_err(rtwdev, "wow: failed to disable pno scan offload\n");
1579 			return ret;
1580 		}
1581 
1582 		ret = rtw89_fw_h2c_cfg_pno(rtwdev, rtwvif_link, false);
1583 		if (ret) {
1584 			rtw89_err(rtwdev, "wow: failed to disable pno\n");
1585 			return ret;
1586 		}
1587 
1588 		rtw89_fw_release_pno_pkt_list(rtwdev, rtwvif_link);
1589 	} else {
1590 		rtw89_wow_pattern_clear(rtwdev);
1591 
1592 		ret = rtw89_fw_h2c_keep_alive(rtwdev, rtwvif_link, false);
1593 		if (ret) {
1594 			rtw89_err(rtwdev, "wow: failed to disable keep alive\n");
1595 			return ret;
1596 		}
1597 
1598 		ret = rtw89_fw_h2c_disconnect_detect(rtwdev, rtwvif_link, false);
1599 		if (ret) {
1600 			rtw89_err(rtwdev, "wow: failed to disable disconnect detect\n");
1601 			return ret;
1602 		}
1603 
1604 		ret = rtw89_fw_h2c_wow_gtk_ofld(rtwdev, rtwvif_link, false);
1605 		if (ret) {
1606 			rtw89_err(rtwdev, "wow: failed to disable GTK offload\n");
1607 			return ret;
1608 		}
1609 
1610 		ret = rtw89_fw_h2c_arp_offload(rtwdev, rtwvif_link, false);
1611 		if (ret)
1612 			rtw89_warn(rtwdev, "wow: failed to disable arp offload\n");
1613 
1614 		rtw89_wow_key_clear(rtwdev);
1615 		rtw89_fw_release_general_pkt_list(rtwdev, true);
1616 	}
1617 
1618 
1619 	ret = rtw89_wow_cfg_wake(rtwdev, false);
1620 	if (ret) {
1621 		rtw89_err(rtwdev, "wow: failed to disable config wake\n");
1622 		return ret;
1623 	}
1624 
1625 	ret = rtw89_wow_check_fw_status(rtwdev, false);
1626 	if (ret) {
1627 		rtw89_err(rtwdev, "wow: failed to check disable fw ready\n");
1628 		return ret;
1629 	}
1630 
1631 	return 0;
1632 }
1633 
1634 static int rtw89_wow_enable(struct rtw89_dev *rtwdev)
1635 {
1636 	int ret;
1637 
1638 	set_bit(RTW89_FLAG_WOWLAN, rtwdev->flags);
1639 
1640 	ret = rtw89_wow_enable_trx_pre(rtwdev);
1641 	if (ret) {
1642 		rtw89_err(rtwdev, "wow: failed to enable trx_pre\n");
1643 		goto out;
1644 	}
1645 
1646 	rtw89_fw_release_general_pkt_list(rtwdev, true);
1647 
1648 	ret = rtw89_wow_swap_fw(rtwdev, true);
1649 	if (ret) {
1650 		rtw89_err(rtwdev, "wow: failed to swap to wow fw\n");
1651 		goto out;
1652 	}
1653 
1654 	ret = rtw89_wow_fw_start(rtwdev);
1655 	if (ret) {
1656 		rtw89_err(rtwdev, "wow: failed to let wow fw start\n");
1657 		goto out;
1658 	}
1659 
1660 	rtw89_wow_enter_ps(rtwdev);
1661 
1662 	ret = rtw89_wow_enable_trx_post(rtwdev);
1663 	if (ret) {
1664 		rtw89_err(rtwdev, "wow: failed to enable trx_post\n");
1665 		goto out;
1666 	}
1667 
1668 	return 0;
1669 
1670 out:
1671 	clear_bit(RTW89_FLAG_WOWLAN, rtwdev->flags);
1672 	return ret;
1673 }
1674 
1675 static int rtw89_wow_disable(struct rtw89_dev *rtwdev)
1676 {
1677 	int ret;
1678 
1679 	ret = rtw89_wow_disable_trx_pre(rtwdev);
1680 	if (ret) {
1681 		rtw89_err(rtwdev, "wow: failed to disable trx_pre\n");
1682 		goto out;
1683 	}
1684 
1685 	rtw89_wow_leave_ps(rtwdev, false);
1686 
1687 	ret = rtw89_wow_fw_stop(rtwdev);
1688 	if (ret) {
1689 		rtw89_err(rtwdev, "wow: failed to swap to normal fw\n");
1690 		goto out;
1691 	}
1692 
1693 	ret = rtw89_wow_swap_fw(rtwdev, false);
1694 	if (ret) {
1695 		rtw89_err(rtwdev, "wow: failed to disable trx_post\n");
1696 		goto out;
1697 	}
1698 
1699 	ret = rtw89_wow_disable_trx_post(rtwdev);
1700 	if (ret) {
1701 		rtw89_err(rtwdev, "wow: failed to disable trx_pre\n");
1702 		goto out;
1703 	}
1704 
1705 out:
1706 	clear_bit(RTW89_FLAG_WOWLAN, rtwdev->flags);
1707 	return ret;
1708 }
1709 
1710 static void rtw89_wow_restore_ps(struct rtw89_dev *rtwdev)
1711 {
1712 	if (rtw89_wow_no_link(rtwdev))
1713 		rtw89_enter_ips(rtwdev);
1714 }
1715 
1716 int rtw89_wow_resume(struct rtw89_dev *rtwdev)
1717 {
1718 	int ret;
1719 
1720 	if (!test_bit(RTW89_FLAG_WOWLAN, rtwdev->flags)) {
1721 		rtw89_err(rtwdev, "wow is not enabled\n");
1722 		ret = -EPERM;
1723 		goto out;
1724 	}
1725 
1726 	if (!rtw89_mac_get_power_state(rtwdev)) {
1727 		rtw89_err(rtwdev, "chip is no power when resume\n");
1728 		ret = -EPERM;
1729 		goto out;
1730 	}
1731 
1732 	rtw89_wow_leave_deep_ps(rtwdev);
1733 
1734 	rtw89_wow_show_wakeup_reason(rtwdev);
1735 
1736 	ret = rtw89_wow_disable(rtwdev);
1737 	if (ret)
1738 		rtw89_err(rtwdev, "failed to disable wow\n");
1739 
1740 	rtw89_wow_restore_ps(rtwdev);
1741 out:
1742 	rtw89_wow_clear_wakeups(rtwdev);
1743 	return ret;
1744 }
1745 
1746 int rtw89_wow_suspend(struct rtw89_dev *rtwdev, struct cfg80211_wowlan *wowlan)
1747 {
1748 	int ret;
1749 
1750 	ret = rtw89_wow_set_wakeups(rtwdev, wowlan);
1751 	if (ret) {
1752 		rtw89_err(rtwdev, "failed to set wakeup event\n");
1753 		return ret;
1754 	}
1755 
1756 	rtw89_wow_leave_ps(rtwdev, true);
1757 
1758 	ret = rtw89_wow_enable(rtwdev);
1759 	if (ret) {
1760 		rtw89_err(rtwdev, "failed to enable wow\n");
1761 		return ret;
1762 	}
1763 
1764 	rtw89_wow_enter_deep_ps(rtwdev);
1765 
1766 	return 0;
1767 }
1768