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
rtw_wow_show_wakeup_reason(struct rtw_dev * rtwdev)13 static 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
rtw_wow_pattern_write_cam(struct rtw_dev * rtwdev,u8 addr,u32 wdata)37 static 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
rtw_wow_pattern_write_cam_ent(struct rtw_dev * rtwdev,u8 id,struct rtw_wow_pattern * rtw_pattern)48 static 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 */
__rtw_cal_crc16(u8 data,u16 crc)84 static 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
rtw_calc_crc(u8 * pdata,int length)122 static 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
rtw_wow_pattern_generate(struct rtw_dev * rtwdev,struct rtw_vif * rtwvif,const struct cfg80211_pkt_pattern * pkt_pattern,struct rtw_wow_pattern * rtw_pattern)134 static 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
rtw_wow_pattern_clear_cam(struct rtw_dev * rtwdev)213 static 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
rtw_wow_pattern_write(struct rtw_dev * rtwdev)225 static 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
rtw_wow_pattern_clear(struct rtw_dev * rtwdev)235 static 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
rtw_wow_bb_stop(struct rtw_dev * rtwdev)245 static 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
rtw_wow_bb_start(struct rtw_dev * rtwdev)260 static 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
rtw_wow_rx_dma_stop(struct rtw_dev * rtwdev)268 static 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
rtw_wow_rx_dma_start(struct rtw_dev * rtwdev)279 static void rtw_wow_rx_dma_start(struct rtw_dev *rtwdev)
280 {
281 rtw_write32_clr(rtwdev, REG_RXPKT_NUM, BIT_RW_RELEASE);
282 }
283
rtw_wow_check_fw_status(struct rtw_dev * rtwdev,bool wow_enable)284 static 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
311 wow_fail:
312 rtw_err(rtwdev, "failed to check wow status %s\n",
313 wow_enable ? "enabled" : "disabled");
314 return -EBUSY;
315 }
316
rtw_wow_fw_security_type_iter(struct ieee80211_hw * hw,struct ieee80211_vif * vif,struct ieee80211_sta * sta,struct ieee80211_key_conf * key,void * data)317 static 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
rtw_wow_fw_security_type(struct rtw_dev * rtwdev)357 static 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
rtw_wow_fw_start(struct rtw_dev * rtwdev)369 static 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
rtw_wow_fw_stop(struct rtw_dev * rtwdev)389 static 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
rtw_wow_avoid_reset_mac(struct rtw_dev * rtwdev)406 static 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
rtw_wow_fw_media_status_iter(void * data,struct ieee80211_sta * sta)426 static 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
rtw_wow_fw_media_status(struct rtw_dev * rtwdev,bool connect)435 static 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
rtw_wow_config_pno_rsvd_page(struct rtw_dev * rtwdev,struct rtw_vif * rtwvif)445 static 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
rtw_wow_config_linked_rsvd_page(struct rtw_dev * rtwdev,struct rtw_vif * rtwvif)451 static 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
rtw_wow_config_rsvd_page(struct rtw_dev * rtwdev,struct rtw_vif * rtwvif)457 static 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
rtw_wow_dl_fw_rsvd_page(struct rtw_dev * rtwdev)470 static 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
rtw_wow_swap_fw(struct rtw_dev * rtwdev,enum rtw_fw_type type)480 static 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
507 out:
508 return ret;
509 }
510
rtw_wow_check_pno(struct rtw_dev * rtwdev,struct cfg80211_sched_scan_request * nd_config)511 static 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
546 channel_err:
547 kfree(pno_req->match_sets);
548
549 err:
550 rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: net-detect is disabled\n");
551 }
552
rtw_wow_leave_linked_ps(struct rtw_dev * rtwdev)553 static 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
rtw_wow_leave_no_link_ps(struct rtw_dev * rtwdev)563 static 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
rtw_wow_leave_ps(struct rtw_dev * rtwdev)583 static 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
rtw_wow_restore_ps(struct rtw_dev * rtwdev)595 static 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
rtw_wow_enter_linked_ps(struct rtw_dev * rtwdev)605 static 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
rtw_wow_enter_no_link_ps(struct rtw_dev * rtwdev)616 static 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
rtw_wow_enter_ps(struct rtw_dev * rtwdev)624 static 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
rtw_wow_stop_trx(struct rtw_dev * rtwdev)636 static 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
rtw_wow_start(struct rtw_dev * rtwdev)642 static 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
654 out:
655 return ret;
656 }
657
rtw_wow_enable(struct rtw_dev * rtwdev)658 static 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
686 error:
687 clear_bit(RTW_FLAG_WOWLAN, rtwdev->flags);
688 return ret;
689 }
690
rtw_wow_stop(struct rtw_dev * rtwdev)691 static 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
rtw_wow_resume_trx(struct rtw_dev * rtwdev)719 static 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
rtw_wow_disable(struct rtw_dev * rtwdev)727 static 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
749 out:
750 rtw_wow_resume_trx(rtwdev);
751 return ret;
752 }
753
rtw_wow_vif_iter(void * data,u8 * mac,struct ieee80211_vif * vif)754 static 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
rtw_wow_set_wakeups(struct rtw_dev * rtwdev,struct cfg80211_wowlan * wowlan)779 static 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
rtw_wow_clear_wakeups(struct rtw_dev * rtwdev)813 static 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
rtw_wow_suspend(struct rtw_dev * rtwdev,struct cfg80211_wowlan * wowlan)826 int 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
853 out:
854 return ret;
855 }
856
rtw_wow_resume(struct rtw_dev * rtwdev)857 int 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
886 out:
887 rtw_wow_clear_wakeups(rtwdev);
888 return ret;
889 }
890