1// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2/* Copyright(c) 2018-2019  Realtek Corporation
3 */
4
5#include "main.h"
6#include "fw.h"
7#include "wow.h"
8#include "reg.h"
9#include "debug.h"
10#include "mac.h"
11#include "ps.h"
12
13static void rtw_wow_show_wakeup_reason(struct rtw_dev *rtwdev)
14{
15	u8 reason;
16
17	reason = rtw_read8(rtwdev, REG_WOWLAN_WAKE_REASON);
18
19	if (reason == RTW_WOW_RSN_RX_DEAUTH)
20		rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: Rx deauth\n");
21	else if (reason == RTW_WOW_RSN_DISCONNECT)
22		rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: AP is off\n");
23	else if (reason == RTW_WOW_RSN_RX_MAGIC_PKT)
24		rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: Rx magic packet\n");
25	else if (reason == RTW_WOW_RSN_RX_GTK_REKEY)
26		rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: Rx gtk rekey\n");
27	else if (reason == RTW_WOW_RSN_RX_PTK_REKEY)
28		rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: Rx ptk rekey\n");
29	else if (reason == RTW_WOW_RSN_RX_PATTERN_MATCH)
30		rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: Rx pattern match packet\n");
31	else if (reason == RTW_WOW_RSN_RX_NLO)
32		rtw_dbg(rtwdev, RTW_DBG_WOW, "Rx NLO\n");
33	else
34		rtw_warn(rtwdev, "Unknown wakeup reason %x\n", reason);
35}
36
37static void rtw_wow_pattern_write_cam(struct rtw_dev *rtwdev, u8 addr,
38				      u32 wdata)
39{
40	rtw_write32(rtwdev, REG_WKFMCAM_RWD, wdata);
41	rtw_write32(rtwdev, REG_WKFMCAM_CMD, BIT_WKFCAM_POLLING_V1 |
42		    BIT_WKFCAM_WE | BIT_WKFCAM_ADDR_V2(addr));
43
44	if (!check_hw_ready(rtwdev, REG_WKFMCAM_CMD, BIT_WKFCAM_POLLING_V1, 0))
45		rtw_err(rtwdev, "failed to write pattern cam\n");
46}
47
48static void rtw_wow_pattern_write_cam_ent(struct rtw_dev *rtwdev, u8 id,
49					  struct rtw_wow_pattern *rtw_pattern)
50{
51	int i;
52	u8 addr;
53	u32 wdata;
54
55	for (i = 0; i < RTW_MAX_PATTERN_MASK_SIZE / 4; i++) {
56		addr = (id << 3) + i;
57		wdata = rtw_pattern->mask[i * 4];
58		wdata |= rtw_pattern->mask[i * 4 + 1] << 8;
59		wdata |= rtw_pattern->mask[i * 4 + 2] << 16;
60		wdata |= rtw_pattern->mask[i * 4 + 3] << 24;
61		rtw_wow_pattern_write_cam(rtwdev, addr, wdata);
62	}
63
64	wdata = rtw_pattern->crc;
65	addr = (id << 3) + RTW_MAX_PATTERN_MASK_SIZE / 4;
66
67	switch (rtw_pattern->type) {
68	case RTW_PATTERN_BROADCAST:
69		wdata |= BIT_WKFMCAM_BC | BIT_WKFMCAM_VALID;
70		break;
71	case RTW_PATTERN_MULTICAST:
72		wdata |= BIT_WKFMCAM_MC | BIT_WKFMCAM_VALID;
73		break;
74	case RTW_PATTERN_UNICAST:
75		wdata |= BIT_WKFMCAM_UC | BIT_WKFMCAM_VALID;
76		break;
77	default:
78		break;
79	}
80	rtw_wow_pattern_write_cam(rtwdev, addr, wdata);
81}
82
83/* RTK internal CRC16 for Pattern Cam */
84static u16 __rtw_cal_crc16(u8 data, u16 crc)
85{
86	u8 shift_in, data_bit;
87	u8 crc_bit4, crc_bit11, crc_bit15;
88	u16 crc_result;
89	int index;
90
91	for (index = 0; index < 8; index++) {
92		crc_bit15 = ((crc & BIT(15)) ? 1 : 0);
93		data_bit = (data & (BIT(0) << index) ? 1 : 0);
94		shift_in = crc_bit15 ^ data_bit;
95
96		crc_result = crc << 1;
97
98		if (shift_in == 0)
99			crc_result &= (~BIT(0));
100		else
101			crc_result |= BIT(0);
102
103		crc_bit11 = ((crc & BIT(11)) ? 1 : 0) ^ shift_in;
104
105		if (crc_bit11 == 0)
106			crc_result &= (~BIT(12));
107		else
108			crc_result |= BIT(12);
109
110		crc_bit4 = ((crc & BIT(4)) ? 1 : 0) ^ shift_in;
111
112		if (crc_bit4 == 0)
113			crc_result &= (~BIT(5));
114		else
115			crc_result |= BIT(5);
116
117		crc = crc_result;
118	}
119	return crc;
120}
121
122static u16 rtw_calc_crc(u8 *pdata, int length)
123{
124	u16 crc = 0xffff;
125	int i;
126
127	for (i = 0; i < length; i++)
128		crc = __rtw_cal_crc16(pdata[i], crc);
129
130	/* get 1' complement */
131	return ~crc;
132}
133
134static void rtw_wow_pattern_generate(struct rtw_dev *rtwdev,
135				     struct rtw_vif *rtwvif,
136				     const struct cfg80211_pkt_pattern *pkt_pattern,
137				     struct rtw_wow_pattern *rtw_pattern)
138{
139	const u8 *mask;
140	const u8 *pattern;
141	u8 mask_hw[RTW_MAX_PATTERN_MASK_SIZE] = {0};
142	u8 content[RTW_MAX_PATTERN_SIZE] = {0};
143	u8 mac_addr[ETH_ALEN] = {0};
144	u8 mask_len;
145	u16 count;
146	int len;
147	int i;
148
149	pattern = pkt_pattern->pattern;
150	len = pkt_pattern->pattern_len;
151	mask = pkt_pattern->mask;
152
153	ether_addr_copy(mac_addr, rtwvif->mac_addr);
154	memset(rtw_pattern, 0, sizeof(*rtw_pattern));
155
156	mask_len = DIV_ROUND_UP(len, 8);
157
158	if (is_broadcast_ether_addr(pattern))
159		rtw_pattern->type = RTW_PATTERN_BROADCAST;
160	else if (is_multicast_ether_addr(pattern))
161		rtw_pattern->type = RTW_PATTERN_MULTICAST;
162	else if (ether_addr_equal(pattern, mac_addr))
163		rtw_pattern->type = RTW_PATTERN_UNICAST;
164	else
165		rtw_pattern->type = RTW_PATTERN_INVALID;
166
167	/* translate mask from os to mask for hw
168	 * pattern from OS uses 'ethenet frame', like this:
169	 * |    6   |    6   |   2  |     20    |  Variable  |  4  |
170	 * |--------+--------+------+-----------+------------+-----|
171	 * |    802.3 Mac Header    | IP Header | TCP Packet | FCS |
172	 * |   DA   |   SA   | Type |
173	 *
174	 * BUT, packet catched by our HW is in '802.11 frame', begin from LLC
175	 * |     24 or 30      |    6   |   2  |     20    |  Variable  |  4  |
176	 * |-------------------+--------+------+-----------+------------+-----|
177	 * | 802.11 MAC Header |       LLC     | IP Header | TCP Packet | FCS |
178	 *		       | Others | Tpye |
179	 *
180	 * Therefore, we need translate mask_from_OS to mask_to_hw.
181	 * We should left-shift mask by 6 bits, then set the new bit[0~5] = 0,
182	 * because new mask[0~5] means 'SA', but our HW packet begins from LLC,
183	 * bit[0~5] corresponds to first 6 Bytes in LLC, they just don't match.
184	 */
185
186	/* Shift 6 bits */
187	for (i = 0; i < mask_len - 1; i++) {
188		mask_hw[i] = u8_get_bits(mask[i], GENMASK(7, 6));
189		mask_hw[i] |= u8_get_bits(mask[i + 1], GENMASK(5, 0)) << 2;
190	}
191	mask_hw[i] = u8_get_bits(mask[i], GENMASK(7, 6));
192
193	/* Set bit 0-5 to zero */
194	mask_hw[0] &= (~GENMASK(5, 0));
195
196	memcpy(rtw_pattern->mask, mask_hw, RTW_MAX_PATTERN_MASK_SIZE);
197
198	/* To get the wake up pattern from the mask.
199	 * We do not count first 12 bits which means
200	 * DA[6] and SA[6] in the pattern to match HW design.
201	 */
202	count = 0;
203	for (i = 12; i < len; i++) {
204		if ((mask[i / 8] >> (i % 8)) & 0x01) {
205			content[count] = pattern[i];
206			count++;
207		}
208	}
209
210	rtw_pattern->crc = rtw_calc_crc(content, count);
211}
212
213static void rtw_wow_pattern_clear_cam(struct rtw_dev *rtwdev)
214{
215	bool ret;
216
217	rtw_write32(rtwdev, REG_WKFMCAM_CMD, BIT_WKFCAM_POLLING_V1 |
218		    BIT_WKFCAM_CLR_V1);
219
220	ret = check_hw_ready(rtwdev, REG_WKFMCAM_CMD, BIT_WKFCAM_POLLING_V1, 0);
221	if (!ret)
222		rtw_err(rtwdev, "failed to clean pattern cam\n");
223}
224
225static void rtw_wow_pattern_write(struct rtw_dev *rtwdev)
226{
227	struct rtw_wow_param *rtw_wow = &rtwdev->wow;
228	struct rtw_wow_pattern *rtw_pattern = rtw_wow->patterns;
229	int i = 0;
230
231	for (i = 0; i < rtw_wow->pattern_cnt; i++)
232		rtw_wow_pattern_write_cam_ent(rtwdev, i, rtw_pattern + i);
233}
234
235static void rtw_wow_pattern_clear(struct rtw_dev *rtwdev)
236{
237	struct rtw_wow_param *rtw_wow = &rtwdev->wow;
238
239	rtw_wow_pattern_clear_cam(rtwdev);
240
241	rtw_wow->pattern_cnt = 0;
242	memset(rtw_wow->patterns, 0, sizeof(rtw_wow->patterns));
243}
244
245static void rtw_wow_bb_stop(struct rtw_dev *rtwdev)
246{
247	struct rtw_wow_param *rtw_wow = &rtwdev->wow;
248
249	/* wait 100ms for firmware to finish TX */
250	msleep(100);
251
252	if (!rtw_read32_mask(rtwdev, REG_BCNQ_INFO, BIT_MGQ_CPU_EMPTY))
253		rtw_warn(rtwdev, "Wrong status of MGQ_CPU empty!\n");
254
255	rtw_wow->txpause = rtw_read8(rtwdev, REG_TXPAUSE);
256	rtw_write8(rtwdev, REG_TXPAUSE, 0xff);
257	rtw_write8_clr(rtwdev, REG_SYS_FUNC_EN, BIT_FEN_BB_RSTB);
258}
259
260static void rtw_wow_bb_start(struct rtw_dev *rtwdev)
261{
262	struct rtw_wow_param *rtw_wow = &rtwdev->wow;
263
264	rtw_write8_set(rtwdev, REG_SYS_FUNC_EN, BIT_FEN_BB_RSTB);
265	rtw_write8(rtwdev, REG_TXPAUSE, rtw_wow->txpause);
266}
267
268static void rtw_wow_rx_dma_stop(struct rtw_dev *rtwdev)
269{
270	/* wait 100ms for HW to finish rx dma */
271	msleep(100);
272
273	rtw_write32_set(rtwdev, REG_RXPKT_NUM, BIT_RW_RELEASE);
274
275	if (!check_hw_ready(rtwdev, REG_RXPKT_NUM, BIT_RXDMA_IDLE, 1))
276		rtw_err(rtwdev, "failed to stop rx dma\n");
277}
278
279static void rtw_wow_rx_dma_start(struct rtw_dev *rtwdev)
280{
281	rtw_write32_clr(rtwdev, REG_RXPKT_NUM, BIT_RW_RELEASE);
282}
283
284static int rtw_wow_check_fw_status(struct rtw_dev *rtwdev, bool wow_enable)
285{
286	int ret;
287	u8 check;
288	u32 check_dis;
289
290	if (wow_enable) {
291		ret = read_poll_timeout(rtw_read8, check, !check, 1000,
292					100000, true, rtwdev,
293					REG_WOWLAN_WAKE_REASON);
294		if (ret)
295			goto wow_fail;
296	} else {
297		ret = read_poll_timeout(rtw_read32_mask, check_dis,
298					!check_dis, 1000, 100000, true, rtwdev,
299					REG_FE1IMR, BIT_FS_RXDONE);
300		if (ret)
301			goto wow_fail;
302		ret = read_poll_timeout(rtw_read32_mask, check_dis,
303					!check_dis, 1000, 100000, false, rtwdev,
304					REG_RXPKT_NUM, BIT_RW_RELEASE);
305		if (ret)
306			goto wow_fail;
307	}
308
309	return 0;
310
311wow_fail:
312	rtw_err(rtwdev, "failed to check wow status %s\n",
313		wow_enable ? "enabled" : "disabled");
314	return -EBUSY;
315}
316
317static void rtw_wow_fw_security_type_iter(struct ieee80211_hw *hw,
318					  struct ieee80211_vif *vif,
319					  struct ieee80211_sta *sta,
320					  struct ieee80211_key_conf *key,
321					  void *data)
322{
323	struct rtw_fw_key_type_iter_data *iter_data = data;
324	struct rtw_dev *rtwdev = hw->priv;
325	u8 hw_key_type;
326
327	if (vif != rtwdev->wow.wow_vif)
328		return;
329
330	switch (key->cipher) {
331	case WLAN_CIPHER_SUITE_WEP40:
332		hw_key_type = RTW_CAM_WEP40;
333		break;
334	case WLAN_CIPHER_SUITE_WEP104:
335		hw_key_type = RTW_CAM_WEP104;
336		break;
337	case WLAN_CIPHER_SUITE_TKIP:
338		hw_key_type = RTW_CAM_TKIP;
339		key->flags |= IEEE80211_KEY_FLAG_GENERATE_MMIC;
340		break;
341	case WLAN_CIPHER_SUITE_CCMP:
342		hw_key_type = RTW_CAM_AES;
343		key->flags |= IEEE80211_KEY_FLAG_SW_MGMT_TX;
344		break;
345	default:
346		rtw_err(rtwdev, "Unsupported key type for wowlan mode\n");
347		hw_key_type = 0;
348		break;
349	}
350
351	if (sta)
352		iter_data->pairwise_key_type = hw_key_type;
353	else
354		iter_data->group_key_type = hw_key_type;
355}
356
357static void rtw_wow_fw_security_type(struct rtw_dev *rtwdev)
358{
359	struct rtw_fw_key_type_iter_data data = {};
360	struct ieee80211_vif *wow_vif = rtwdev->wow.wow_vif;
361
362	data.rtwdev = rtwdev;
363	rtw_iterate_keys(rtwdev, wow_vif,
364			 rtw_wow_fw_security_type_iter, &data);
365	rtw_fw_set_aoac_global_info_cmd(rtwdev, data.pairwise_key_type,
366					data.group_key_type);
367}
368
369static int rtw_wow_fw_start(struct rtw_dev *rtwdev)
370{
371	if (rtw_wow_mgd_linked(rtwdev)) {
372		rtw_send_rsvd_page_h2c(rtwdev);
373		rtw_wow_pattern_write(rtwdev);
374		rtw_wow_fw_security_type(rtwdev);
375		rtw_fw_set_disconnect_decision_cmd(rtwdev, true);
376		rtw_fw_set_keep_alive_cmd(rtwdev, true);
377	} else if (rtw_wow_no_link(rtwdev)) {
378		rtw_fw_set_nlo_info(rtwdev, true);
379		rtw_fw_update_pkt_probe_req(rtwdev, NULL);
380		rtw_fw_channel_switch(rtwdev, true);
381	}
382
383	rtw_fw_set_wowlan_ctrl_cmd(rtwdev, true);
384	rtw_fw_set_remote_wake_ctrl_cmd(rtwdev, true);
385
386	return rtw_wow_check_fw_status(rtwdev, true);
387}
388
389static int rtw_wow_fw_stop(struct rtw_dev *rtwdev)
390{
391	if (rtw_wow_mgd_linked(rtwdev)) {
392		rtw_fw_set_disconnect_decision_cmd(rtwdev, false);
393		rtw_fw_set_keep_alive_cmd(rtwdev, false);
394		rtw_wow_pattern_clear(rtwdev);
395	} else if (rtw_wow_no_link(rtwdev)) {
396		rtw_fw_channel_switch(rtwdev, false);
397		rtw_fw_set_nlo_info(rtwdev, false);
398	}
399
400	rtw_fw_set_wowlan_ctrl_cmd(rtwdev, false);
401	rtw_fw_set_remote_wake_ctrl_cmd(rtwdev, false);
402
403	return rtw_wow_check_fw_status(rtwdev, false);
404}
405
406static void rtw_wow_avoid_reset_mac(struct rtw_dev *rtwdev)
407{
408	/* When resuming from wowlan mode, some hosts issue signal
409	 * (PCIE: PREST, USB: SE0RST) to device, and lead to reset
410	 * mac core. If it happens, the connection to AP will be lost.
411	 * Setting REG_RSV_CTRL Register can avoid this process.
412	 */
413	switch (rtw_hci_type(rtwdev)) {
414	case RTW_HCI_TYPE_PCIE:
415	case RTW_HCI_TYPE_USB:
416		rtw_write8(rtwdev, REG_RSV_CTRL, BIT_WLOCK_1C_B6);
417		rtw_write8(rtwdev, REG_RSV_CTRL,
418			   BIT_WLOCK_1C_B6 | BIT_R_DIS_PRST);
419		break;
420	default:
421		rtw_warn(rtwdev, "Unsupported hci type to disable reset MAC\n");
422		break;
423	}
424}
425
426static void rtw_wow_fw_media_status_iter(void *data, struct ieee80211_sta *sta)
427{
428	struct rtw_sta_info *si = (struct rtw_sta_info *)sta->drv_priv;
429	struct rtw_fw_media_status_iter_data *iter_data = data;
430	struct rtw_dev *rtwdev = iter_data->rtwdev;
431
432	rtw_fw_media_status_report(rtwdev, si->mac_id, iter_data->connect);
433}
434
435static void rtw_wow_fw_media_status(struct rtw_dev *rtwdev, bool connect)
436{
437	struct rtw_fw_media_status_iter_data data;
438
439	data.rtwdev = rtwdev;
440	data.connect = connect;
441
442	rtw_iterate_stas_atomic(rtwdev, rtw_wow_fw_media_status_iter, &data);
443}
444
445static void rtw_wow_config_pno_rsvd_page(struct rtw_dev *rtwdev,
446					 struct rtw_vif *rtwvif)
447{
448	rtw_add_rsvd_page_pno(rtwdev, rtwvif);
449}
450
451static void rtw_wow_config_linked_rsvd_page(struct rtw_dev *rtwdev,
452					   struct rtw_vif *rtwvif)
453{
454	rtw_add_rsvd_page_sta(rtwdev, rtwvif);
455}
456
457static void rtw_wow_config_rsvd_page(struct rtw_dev *rtwdev,
458				     struct rtw_vif *rtwvif)
459{
460	rtw_remove_rsvd_page(rtwdev, rtwvif);
461
462	if (rtw_wow_mgd_linked(rtwdev)) {
463		rtw_wow_config_linked_rsvd_page(rtwdev, rtwvif);
464	} else if (test_bit(RTW_FLAG_WOWLAN, rtwdev->flags) &&
465		   rtw_wow_no_link(rtwdev)) {
466		rtw_wow_config_pno_rsvd_page(rtwdev, rtwvif);
467	}
468}
469
470static int rtw_wow_dl_fw_rsvd_page(struct rtw_dev *rtwdev)
471{
472	struct ieee80211_vif *wow_vif = rtwdev->wow.wow_vif;
473	struct rtw_vif *rtwvif = (struct rtw_vif *)wow_vif->drv_priv;
474
475	rtw_wow_config_rsvd_page(rtwdev, rtwvif);
476
477	return rtw_fw_download_rsvd_page(rtwdev);
478}
479
480static int rtw_wow_swap_fw(struct rtw_dev *rtwdev, enum rtw_fw_type type)
481{
482	struct rtw_fw_state *fw;
483	int ret;
484
485	switch (type) {
486	case RTW_WOWLAN_FW:
487		fw = &rtwdev->wow_fw;
488		break;
489
490	case RTW_NORMAL_FW:
491		fw = &rtwdev->fw;
492		break;
493
494	default:
495		rtw_warn(rtwdev, "unsupported firmware type to swap\n");
496		return -ENOENT;
497	}
498
499	ret = rtw_download_firmware(rtwdev, fw);
500	if (ret)
501		goto out;
502
503	rtw_fw_send_general_info(rtwdev);
504	rtw_fw_send_phydm_info(rtwdev);
505	rtw_wow_fw_media_status(rtwdev, true);
506
507out:
508	return ret;
509}
510
511static void rtw_wow_check_pno(struct rtw_dev *rtwdev,
512			      struct cfg80211_sched_scan_request *nd_config)
513{
514	struct rtw_wow_param *rtw_wow = &rtwdev->wow;
515	struct rtw_pno_request *pno_req = &rtw_wow->pno_req;
516	struct ieee80211_channel *channel;
517	int i, size;
518
519	if (!nd_config->n_match_sets || !nd_config->n_channels)
520		goto err;
521
522	pno_req->match_set_cnt = nd_config->n_match_sets;
523	size = sizeof(*pno_req->match_sets) * pno_req->match_set_cnt;
524	pno_req->match_sets = kmemdup(nd_config->match_sets, size, GFP_KERNEL);
525	if (!pno_req->match_sets)
526		goto err;
527
528	pno_req->channel_cnt = nd_config->n_channels;
529	size = sizeof(*nd_config->channels[0]) * nd_config->n_channels;
530	pno_req->channels = kmalloc(size, GFP_KERNEL);
531	if (!pno_req->channels)
532		goto channel_err;
533
534	for (i = 0 ; i < pno_req->channel_cnt; i++) {
535		channel = pno_req->channels + i;
536		memcpy(channel, nd_config->channels[i], sizeof(*channel));
537	}
538
539	pno_req->scan_plan = *nd_config->scan_plans;
540	pno_req->inited = true;
541
542	rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: net-detect is enabled\n");
543
544	return;
545
546channel_err:
547	kfree(pno_req->match_sets);
548
549err:
550	rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: net-detect is disabled\n");
551}
552
553static int rtw_wow_leave_linked_ps(struct rtw_dev *rtwdev)
554{
555	if (!test_bit(RTW_FLAG_WOWLAN, rtwdev->flags))
556		cancel_delayed_work_sync(&rtwdev->watch_dog_work);
557
558	rtw_leave_lps(rtwdev);
559
560	return 0;
561}
562
563static int rtw_wow_leave_no_link_ps(struct rtw_dev *rtwdev)
564{
565	struct rtw_wow_param *rtw_wow = &rtwdev->wow;
566	int ret = 0;
567
568	if (test_bit(RTW_FLAG_WOWLAN, rtwdev->flags)) {
569		if (rtw_fw_lps_deep_mode)
570			rtw_leave_lps_deep(rtwdev);
571	} else {
572		if (test_bit(RTW_FLAG_INACTIVE_PS, rtwdev->flags)) {
573			rtw_wow->ips_enabled = true;
574			ret = rtw_leave_ips(rtwdev);
575			if (ret)
576				return ret;
577		}
578	}
579
580	return 0;
581}
582
583static int rtw_wow_leave_ps(struct rtw_dev *rtwdev)
584{
585	int ret = 0;
586
587	if (rtw_wow_mgd_linked(rtwdev))
588		ret = rtw_wow_leave_linked_ps(rtwdev);
589	else if (rtw_wow_no_link(rtwdev))
590		ret = rtw_wow_leave_no_link_ps(rtwdev);
591
592	return ret;
593}
594
595static int rtw_wow_restore_ps(struct rtw_dev *rtwdev)
596{
597	int ret = 0;
598
599	if (rtw_wow_no_link(rtwdev) && rtwdev->wow.ips_enabled)
600		ret = rtw_enter_ips(rtwdev);
601
602	return ret;
603}
604
605static int rtw_wow_enter_linked_ps(struct rtw_dev *rtwdev)
606{
607	struct rtw_wow_param *rtw_wow = &rtwdev->wow;
608	struct ieee80211_vif *wow_vif = rtw_wow->wow_vif;
609	struct rtw_vif *rtwvif = (struct rtw_vif *)wow_vif->drv_priv;
610
611	rtw_enter_lps(rtwdev, rtwvif->port);
612
613	return 0;
614}
615
616static int rtw_wow_enter_no_link_ps(struct rtw_dev *rtwdev)
617{
618	/* firmware enters deep ps by itself if supported */
619	set_bit(RTW_FLAG_LEISURE_PS_DEEP, rtwdev->flags);
620
621	return 0;
622}
623
624static int rtw_wow_enter_ps(struct rtw_dev *rtwdev)
625{
626	int ret = 0;
627
628	if (rtw_wow_mgd_linked(rtwdev))
629		ret = rtw_wow_enter_linked_ps(rtwdev);
630	else if (rtw_wow_no_link(rtwdev) && rtw_fw_lps_deep_mode)
631		ret = rtw_wow_enter_no_link_ps(rtwdev);
632
633	return ret;
634}
635
636static void rtw_wow_stop_trx(struct rtw_dev *rtwdev)
637{
638	rtw_wow_bb_stop(rtwdev);
639	rtw_wow_rx_dma_stop(rtwdev);
640}
641
642static int rtw_wow_start(struct rtw_dev *rtwdev)
643{
644	int ret;
645
646	ret = rtw_wow_fw_start(rtwdev);
647	if (ret)
648		goto out;
649
650	rtw_hci_stop(rtwdev);
651	rtw_wow_bb_start(rtwdev);
652	rtw_wow_avoid_reset_mac(rtwdev);
653
654out:
655	return ret;
656}
657
658static int rtw_wow_enable(struct rtw_dev *rtwdev)
659{
660	int ret = 0;
661
662	rtw_wow_stop_trx(rtwdev);
663
664	ret = rtw_wow_swap_fw(rtwdev, RTW_WOWLAN_FW);
665	if (ret) {
666		rtw_err(rtwdev, "failed to swap wow fw\n");
667		goto error;
668	}
669
670	set_bit(RTW_FLAG_WOWLAN, rtwdev->flags);
671
672	ret = rtw_wow_dl_fw_rsvd_page(rtwdev);
673	if (ret) {
674		rtw_err(rtwdev, "failed to download wowlan rsvd page\n");
675		goto error;
676	}
677
678	ret = rtw_wow_start(rtwdev);
679	if (ret) {
680		rtw_err(rtwdev, "failed to start wow\n");
681		goto error;
682	}
683
684	return ret;
685
686error:
687	clear_bit(RTW_FLAG_WOWLAN, rtwdev->flags);
688	return ret;
689}
690
691static int rtw_wow_stop(struct rtw_dev *rtwdev)
692{
693	int ret;
694
695	/* some HCI related registers will be reset after resume,
696	 * need to set them again.
697	 */
698	ret = rtw_hci_setup(rtwdev);
699	if (ret) {
700		rtw_err(rtwdev, "failed to setup hci\n");
701		return ret;
702	}
703
704	ret = rtw_hci_start(rtwdev);
705	if (ret) {
706		rtw_err(rtwdev, "failed to start hci\n");
707		return ret;
708	}
709
710	ret = rtw_wow_fw_stop(rtwdev);
711	if (ret)
712		rtw_err(rtwdev, "failed to stop wowlan fw\n");
713
714	rtw_wow_bb_stop(rtwdev);
715
716	return ret;
717}
718
719static void rtw_wow_resume_trx(struct rtw_dev *rtwdev)
720{
721	rtw_wow_rx_dma_start(rtwdev);
722	rtw_wow_bb_start(rtwdev);
723	ieee80211_queue_delayed_work(rtwdev->hw, &rtwdev->watch_dog_work,
724				     RTW_WATCH_DOG_DELAY_TIME);
725}
726
727static int rtw_wow_disable(struct rtw_dev *rtwdev)
728{
729	int ret;
730
731	clear_bit(RTW_FLAG_WOWLAN, rtwdev->flags);
732
733	ret = rtw_wow_stop(rtwdev);
734	if (ret) {
735		rtw_err(rtwdev, "failed to stop wow\n");
736		goto out;
737	}
738
739	ret = rtw_wow_swap_fw(rtwdev, RTW_NORMAL_FW);
740	if (ret) {
741		rtw_err(rtwdev, "failed to swap normal fw\n");
742		goto out;
743	}
744
745	ret = rtw_wow_dl_fw_rsvd_page(rtwdev);
746	if (ret)
747		rtw_err(rtwdev, "failed to download normal rsvd page\n");
748
749out:
750	rtw_wow_resume_trx(rtwdev);
751	return ret;
752}
753
754static void rtw_wow_vif_iter(void *data, u8 *mac, struct ieee80211_vif *vif)
755{
756	struct rtw_dev *rtwdev = data;
757	struct rtw_vif *rtwvif = (struct rtw_vif *)vif->drv_priv;
758	struct rtw_wow_param *rtw_wow = &rtwdev->wow;
759
760	/* Current wowlan function support setting of only one STATION vif.
761	 * So when one suitable vif is found, stop the iteration.
762	 */
763	if (rtw_wow->wow_vif || vif->type != NL80211_IFTYPE_STATION)
764		return;
765
766	switch (rtwvif->net_type) {
767	case RTW_NET_MGD_LINKED:
768		rtw_wow->wow_vif = vif;
769		break;
770	case RTW_NET_NO_LINK:
771		if (rtw_wow->pno_req.inited)
772			rtwdev->wow.wow_vif = vif;
773		break;
774	default:
775		break;
776	}
777}
778
779static int rtw_wow_set_wakeups(struct rtw_dev *rtwdev,
780			       struct cfg80211_wowlan *wowlan)
781{
782	struct rtw_wow_param *rtw_wow = &rtwdev->wow;
783	struct rtw_wow_pattern *rtw_patterns = rtw_wow->patterns;
784	struct rtw_vif *rtwvif;
785	int i;
786
787	if (wowlan->disconnect)
788		set_bit(RTW_WOW_FLAG_EN_DISCONNECT, rtw_wow->flags);
789	if (wowlan->magic_pkt)
790		set_bit(RTW_WOW_FLAG_EN_MAGIC_PKT, rtw_wow->flags);
791	if (wowlan->gtk_rekey_failure)
792		set_bit(RTW_WOW_FLAG_EN_REKEY_PKT, rtw_wow->flags);
793
794	if (wowlan->nd_config)
795		rtw_wow_check_pno(rtwdev, wowlan->nd_config);
796
797	rtw_iterate_vifs_atomic(rtwdev, rtw_wow_vif_iter, rtwdev);
798	if (!rtw_wow->wow_vif)
799		return -EPERM;
800
801	rtwvif = (struct rtw_vif *)rtw_wow->wow_vif->drv_priv;
802	if (wowlan->n_patterns && wowlan->patterns) {
803		rtw_wow->pattern_cnt = wowlan->n_patterns;
804		for (i = 0; i < wowlan->n_patterns; i++)
805			rtw_wow_pattern_generate(rtwdev, rtwvif,
806						 wowlan->patterns + i,
807						 rtw_patterns + i);
808	}
809
810	return 0;
811}
812
813static void rtw_wow_clear_wakeups(struct rtw_dev *rtwdev)
814{
815	struct rtw_wow_param *rtw_wow = &rtwdev->wow;
816	struct rtw_pno_request *pno_req = &rtw_wow->pno_req;
817
818	if (pno_req->inited) {
819		kfree(pno_req->channels);
820		kfree(pno_req->match_sets);
821	}
822
823	memset(rtw_wow, 0, sizeof(rtwdev->wow));
824}
825
826int rtw_wow_suspend(struct rtw_dev *rtwdev, struct cfg80211_wowlan *wowlan)
827{
828	int ret = 0;
829
830	ret = rtw_wow_set_wakeups(rtwdev, wowlan);
831	if (ret) {
832		rtw_err(rtwdev, "failed to set wakeup event\n");
833		goto out;
834	}
835
836	ret = rtw_wow_leave_ps(rtwdev);
837	if (ret) {
838		rtw_err(rtwdev, "failed to leave ps from normal mode\n");
839		goto out;
840	}
841
842	ret = rtw_wow_enable(rtwdev);
843	if (ret) {
844		rtw_err(rtwdev, "failed to enable wow\n");
845		rtw_wow_restore_ps(rtwdev);
846		goto out;
847	}
848
849	ret = rtw_wow_enter_ps(rtwdev);
850	if (ret)
851		rtw_err(rtwdev, "failed to enter ps for wow\n");
852
853out:
854	return ret;
855}
856
857int rtw_wow_resume(struct rtw_dev *rtwdev)
858{
859	int ret;
860
861	/* If wowlan mode is not enabled, do nothing */
862	if (!test_bit(RTW_FLAG_WOWLAN, rtwdev->flags)) {
863		rtw_err(rtwdev, "wow is not enabled\n");
864		ret = -EPERM;
865		goto out;
866	}
867
868	ret = rtw_wow_leave_ps(rtwdev);
869	if (ret) {
870		rtw_err(rtwdev, "failed to leave ps from wowlan mode\n");
871		goto out;
872	}
873
874	rtw_wow_show_wakeup_reason(rtwdev);
875
876	ret = rtw_wow_disable(rtwdev);
877	if (ret) {
878		rtw_err(rtwdev, "failed to disable wow\n");
879		goto out;
880	}
881
882	ret = rtw_wow_restore_ps(rtwdev);
883	if (ret)
884		rtw_err(rtwdev, "failed to restore ps to normal mode\n");
885
886out:
887	rtw_wow_clear_wakeups(rtwdev);
888	return ret;
889}
890