Back to home page

OSCL-LXR

 
 

    


0001 // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
0002 /* Copyright(c) 2018-2019  Realtek Corporation
0003  */
0004 
0005 #include "main.h"
0006 #include "fw.h"
0007 #include "wow.h"
0008 #include "reg.h"
0009 #include "debug.h"
0010 #include "mac.h"
0011 #include "ps.h"
0012 
0013 static void rtw_wow_show_wakeup_reason(struct rtw_dev *rtwdev)
0014 {
0015     struct cfg80211_wowlan_nd_info nd_info;
0016     struct cfg80211_wowlan_wakeup wakeup = {
0017         .pattern_idx = -1,
0018     };
0019     u8 reason;
0020 
0021     reason = rtw_read8(rtwdev, REG_WOWLAN_WAKE_REASON);
0022 
0023     switch (reason) {
0024     case RTW_WOW_RSN_RX_DEAUTH:
0025         wakeup.disconnect = true;
0026         rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: Rx deauth\n");
0027         break;
0028     case RTW_WOW_RSN_DISCONNECT:
0029         wakeup.disconnect = true;
0030         rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: AP is off\n");
0031         break;
0032     case RTW_WOW_RSN_RX_MAGIC_PKT:
0033         wakeup.magic_pkt = true;
0034         rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: Rx magic packet\n");
0035         break;
0036     case RTW_WOW_RSN_RX_GTK_REKEY:
0037         wakeup.gtk_rekey_failure = true;
0038         rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: Rx gtk rekey\n");
0039         break;
0040     case RTW_WOW_RSN_RX_PATTERN_MATCH:
0041         /* Current firmware and driver don't report pattern index
0042          * Use pattern_idx to 0 defaultly.
0043          */
0044         wakeup.pattern_idx = 0;
0045         rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: Rx pattern match packet\n");
0046         break;
0047     case RTW_WOW_RSN_RX_NLO:
0048         /* Current firmware and driver don't report ssid index.
0049          * Use 0 for n_matches based on its comment.
0050          */
0051         nd_info.n_matches = 0;
0052         wakeup.net_detect = &nd_info;
0053         rtw_dbg(rtwdev, RTW_DBG_WOW, "Rx NLO\n");
0054         break;
0055     default:
0056         rtw_warn(rtwdev, "Unknown wakeup reason %x\n", reason);
0057         ieee80211_report_wowlan_wakeup(rtwdev->wow.wow_vif, NULL,
0058                            GFP_KERNEL);
0059         return;
0060     }
0061     ieee80211_report_wowlan_wakeup(rtwdev->wow.wow_vif, &wakeup,
0062                        GFP_KERNEL);
0063 }
0064 
0065 static void rtw_wow_pattern_write_cam(struct rtw_dev *rtwdev, u8 addr,
0066                       u32 wdata)
0067 {
0068     rtw_write32(rtwdev, REG_WKFMCAM_RWD, wdata);
0069     rtw_write32(rtwdev, REG_WKFMCAM_CMD, BIT_WKFCAM_POLLING_V1 |
0070             BIT_WKFCAM_WE | BIT_WKFCAM_ADDR_V2(addr));
0071 
0072     if (!check_hw_ready(rtwdev, REG_WKFMCAM_CMD, BIT_WKFCAM_POLLING_V1, 0))
0073         rtw_err(rtwdev, "failed to write pattern cam\n");
0074 }
0075 
0076 static void rtw_wow_pattern_write_cam_ent(struct rtw_dev *rtwdev, u8 id,
0077                       struct rtw_wow_pattern *rtw_pattern)
0078 {
0079     int i;
0080     u8 addr;
0081     u32 wdata;
0082 
0083     for (i = 0; i < RTW_MAX_PATTERN_MASK_SIZE / 4; i++) {
0084         addr = (id << 3) + i;
0085         wdata = rtw_pattern->mask[i * 4];
0086         wdata |= rtw_pattern->mask[i * 4 + 1] << 8;
0087         wdata |= rtw_pattern->mask[i * 4 + 2] << 16;
0088         wdata |= rtw_pattern->mask[i * 4 + 3] << 24;
0089         rtw_wow_pattern_write_cam(rtwdev, addr, wdata);
0090     }
0091 
0092     wdata = rtw_pattern->crc;
0093     addr = (id << 3) + RTW_MAX_PATTERN_MASK_SIZE / 4;
0094 
0095     switch (rtw_pattern->type) {
0096     case RTW_PATTERN_BROADCAST:
0097         wdata |= BIT_WKFMCAM_BC | BIT_WKFMCAM_VALID;
0098         break;
0099     case RTW_PATTERN_MULTICAST:
0100         wdata |= BIT_WKFMCAM_MC | BIT_WKFMCAM_VALID;
0101         break;
0102     case RTW_PATTERN_UNICAST:
0103         wdata |= BIT_WKFMCAM_UC | BIT_WKFMCAM_VALID;
0104         break;
0105     default:
0106         break;
0107     }
0108     rtw_wow_pattern_write_cam(rtwdev, addr, wdata);
0109 }
0110 
0111 /* RTK internal CRC16 for Pattern Cam */
0112 static u16 __rtw_cal_crc16(u8 data, u16 crc)
0113 {
0114     u8 shift_in, data_bit;
0115     u8 crc_bit4, crc_bit11, crc_bit15;
0116     u16 crc_result;
0117     int index;
0118 
0119     for (index = 0; index < 8; index++) {
0120         crc_bit15 = ((crc & BIT(15)) ? 1 : 0);
0121         data_bit = (data & (BIT(0) << index) ? 1 : 0);
0122         shift_in = crc_bit15 ^ data_bit;
0123 
0124         crc_result = crc << 1;
0125 
0126         if (shift_in == 0)
0127             crc_result &= (~BIT(0));
0128         else
0129             crc_result |= BIT(0);
0130 
0131         crc_bit11 = ((crc & BIT(11)) ? 1 : 0) ^ shift_in;
0132 
0133         if (crc_bit11 == 0)
0134             crc_result &= (~BIT(12));
0135         else
0136             crc_result |= BIT(12);
0137 
0138         crc_bit4 = ((crc & BIT(4)) ? 1 : 0) ^ shift_in;
0139 
0140         if (crc_bit4 == 0)
0141             crc_result &= (~BIT(5));
0142         else
0143             crc_result |= BIT(5);
0144 
0145         crc = crc_result;
0146     }
0147     return crc;
0148 }
0149 
0150 static u16 rtw_calc_crc(u8 *pdata, int length)
0151 {
0152     u16 crc = 0xffff;
0153     int i;
0154 
0155     for (i = 0; i < length; i++)
0156         crc = __rtw_cal_crc16(pdata[i], crc);
0157 
0158     /* get 1' complement */
0159     return ~crc;
0160 }
0161 
0162 static void rtw_wow_pattern_generate(struct rtw_dev *rtwdev,
0163                      struct rtw_vif *rtwvif,
0164                      const struct cfg80211_pkt_pattern *pkt_pattern,
0165                      struct rtw_wow_pattern *rtw_pattern)
0166 {
0167     const u8 *mask;
0168     const u8 *pattern;
0169     u8 mask_hw[RTW_MAX_PATTERN_MASK_SIZE] = {0};
0170     u8 content[RTW_MAX_PATTERN_SIZE] = {0};
0171     u8 mac_addr[ETH_ALEN] = {0};
0172     u8 mask_len;
0173     u16 count;
0174     int len;
0175     int i;
0176 
0177     pattern = pkt_pattern->pattern;
0178     len = pkt_pattern->pattern_len;
0179     mask = pkt_pattern->mask;
0180 
0181     ether_addr_copy(mac_addr, rtwvif->mac_addr);
0182     memset(rtw_pattern, 0, sizeof(*rtw_pattern));
0183 
0184     mask_len = DIV_ROUND_UP(len, 8);
0185 
0186     if (is_broadcast_ether_addr(pattern))
0187         rtw_pattern->type = RTW_PATTERN_BROADCAST;
0188     else if (is_multicast_ether_addr(pattern))
0189         rtw_pattern->type = RTW_PATTERN_MULTICAST;
0190     else if (ether_addr_equal(pattern, mac_addr))
0191         rtw_pattern->type = RTW_PATTERN_UNICAST;
0192     else
0193         rtw_pattern->type = RTW_PATTERN_INVALID;
0194 
0195     /* translate mask from os to mask for hw
0196      * pattern from OS uses 'ethenet frame', like this:
0197      * |    6   |    6   |   2  |     20    |  Variable  |  4  |
0198      * |--------+--------+------+-----------+------------+-----|
0199      * |    802.3 Mac Header    | IP Header | TCP Packet | FCS |
0200      * |   DA   |   SA   | Type |
0201      *
0202      * BUT, packet catched by our HW is in '802.11 frame', begin from LLC
0203      * |     24 or 30      |    6   |   2  |     20    |  Variable  |  4  |
0204      * |-------------------+--------+------+-----------+------------+-----|
0205      * | 802.11 MAC Header |       LLC     | IP Header | TCP Packet | FCS |
0206      *             | Others | Tpye |
0207      *
0208      * Therefore, we need translate mask_from_OS to mask_to_hw.
0209      * We should left-shift mask by 6 bits, then set the new bit[0~5] = 0,
0210      * because new mask[0~5] means 'SA', but our HW packet begins from LLC,
0211      * bit[0~5] corresponds to first 6 Bytes in LLC, they just don't match.
0212      */
0213 
0214     /* Shift 6 bits */
0215     for (i = 0; i < mask_len - 1; i++) {
0216         mask_hw[i] = u8_get_bits(mask[i], GENMASK(7, 6));
0217         mask_hw[i] |= u8_get_bits(mask[i + 1], GENMASK(5, 0)) << 2;
0218     }
0219     mask_hw[i] = u8_get_bits(mask[i], GENMASK(7, 6));
0220 
0221     /* Set bit 0-5 to zero */
0222     mask_hw[0] &= (~GENMASK(5, 0));
0223 
0224     memcpy(rtw_pattern->mask, mask_hw, RTW_MAX_PATTERN_MASK_SIZE);
0225 
0226     /* To get the wake up pattern from the mask.
0227      * We do not count first 12 bits which means
0228      * DA[6] and SA[6] in the pattern to match HW design.
0229      */
0230     count = 0;
0231     for (i = 12; i < len; i++) {
0232         if ((mask[i / 8] >> (i % 8)) & 0x01) {
0233             content[count] = pattern[i];
0234             count++;
0235         }
0236     }
0237 
0238     rtw_pattern->crc = rtw_calc_crc(content, count);
0239 }
0240 
0241 static void rtw_wow_pattern_clear_cam(struct rtw_dev *rtwdev)
0242 {
0243     bool ret;
0244 
0245     rtw_write32(rtwdev, REG_WKFMCAM_CMD, BIT_WKFCAM_POLLING_V1 |
0246             BIT_WKFCAM_CLR_V1);
0247 
0248     ret = check_hw_ready(rtwdev, REG_WKFMCAM_CMD, BIT_WKFCAM_POLLING_V1, 0);
0249     if (!ret)
0250         rtw_err(rtwdev, "failed to clean pattern cam\n");
0251 }
0252 
0253 static void rtw_wow_pattern_write(struct rtw_dev *rtwdev)
0254 {
0255     struct rtw_wow_param *rtw_wow = &rtwdev->wow;
0256     struct rtw_wow_pattern *rtw_pattern = rtw_wow->patterns;
0257     int i = 0;
0258 
0259     for (i = 0; i < rtw_wow->pattern_cnt; i++)
0260         rtw_wow_pattern_write_cam_ent(rtwdev, i, rtw_pattern + i);
0261 }
0262 
0263 static void rtw_wow_pattern_clear(struct rtw_dev *rtwdev)
0264 {
0265     struct rtw_wow_param *rtw_wow = &rtwdev->wow;
0266 
0267     rtw_wow_pattern_clear_cam(rtwdev);
0268 
0269     rtw_wow->pattern_cnt = 0;
0270     memset(rtw_wow->patterns, 0, sizeof(rtw_wow->patterns));
0271 }
0272 
0273 static void rtw_wow_bb_stop(struct rtw_dev *rtwdev)
0274 {
0275     struct rtw_wow_param *rtw_wow = &rtwdev->wow;
0276 
0277     /* wait 100ms for firmware to finish TX */
0278     msleep(100);
0279 
0280     if (!rtw_read32_mask(rtwdev, REG_BCNQ_INFO, BIT_MGQ_CPU_EMPTY))
0281         rtw_warn(rtwdev, "Wrong status of MGQ_CPU empty!\n");
0282 
0283     rtw_wow->txpause = rtw_read8(rtwdev, REG_TXPAUSE);
0284     rtw_write8(rtwdev, REG_TXPAUSE, 0xff);
0285     rtw_write8_clr(rtwdev, REG_SYS_FUNC_EN, BIT_FEN_BB_RSTB);
0286 }
0287 
0288 static void rtw_wow_bb_start(struct rtw_dev *rtwdev)
0289 {
0290     struct rtw_wow_param *rtw_wow = &rtwdev->wow;
0291 
0292     rtw_write8_set(rtwdev, REG_SYS_FUNC_EN, BIT_FEN_BB_RSTB);
0293     rtw_write8(rtwdev, REG_TXPAUSE, rtw_wow->txpause);
0294 }
0295 
0296 static void rtw_wow_rx_dma_stop(struct rtw_dev *rtwdev)
0297 {
0298     /* wait 100ms for HW to finish rx dma */
0299     msleep(100);
0300 
0301     rtw_write32_set(rtwdev, REG_RXPKT_NUM, BIT_RW_RELEASE);
0302 
0303     if (!check_hw_ready(rtwdev, REG_RXPKT_NUM, BIT_RXDMA_IDLE, 1))
0304         rtw_err(rtwdev, "failed to stop rx dma\n");
0305 }
0306 
0307 static void rtw_wow_rx_dma_start(struct rtw_dev *rtwdev)
0308 {
0309     rtw_write32_clr(rtwdev, REG_RXPKT_NUM, BIT_RW_RELEASE);
0310 }
0311 
0312 static int rtw_wow_check_fw_status(struct rtw_dev *rtwdev, bool wow_enable)
0313 {
0314     int ret;
0315     u8 check;
0316     u32 check_dis;
0317 
0318     if (wow_enable) {
0319         ret = read_poll_timeout(rtw_read8, check, !check, 1000,
0320                     100000, true, rtwdev,
0321                     REG_WOWLAN_WAKE_REASON);
0322         if (ret)
0323             goto wow_fail;
0324     } else {
0325         ret = read_poll_timeout(rtw_read32_mask, check_dis,
0326                     !check_dis, 1000, 100000, true, rtwdev,
0327                     REG_FE1IMR, BIT_FS_RXDONE);
0328         if (ret)
0329             goto wow_fail;
0330         ret = read_poll_timeout(rtw_read32_mask, check_dis,
0331                     !check_dis, 1000, 100000, false, rtwdev,
0332                     REG_RXPKT_NUM, BIT_RW_RELEASE);
0333         if (ret)
0334             goto wow_fail;
0335     }
0336 
0337     return 0;
0338 
0339 wow_fail:
0340     rtw_err(rtwdev, "failed to check wow status %s\n",
0341         wow_enable ? "enabled" : "disabled");
0342     return -EBUSY;
0343 }
0344 
0345 static void rtw_wow_fw_security_type_iter(struct ieee80211_hw *hw,
0346                       struct ieee80211_vif *vif,
0347                       struct ieee80211_sta *sta,
0348                       struct ieee80211_key_conf *key,
0349                       void *data)
0350 {
0351     struct rtw_fw_key_type_iter_data *iter_data = data;
0352     struct rtw_dev *rtwdev = hw->priv;
0353     u8 hw_key_type;
0354 
0355     if (vif != rtwdev->wow.wow_vif)
0356         return;
0357 
0358     switch (key->cipher) {
0359     case WLAN_CIPHER_SUITE_WEP40:
0360         hw_key_type = RTW_CAM_WEP40;
0361         break;
0362     case WLAN_CIPHER_SUITE_WEP104:
0363         hw_key_type = RTW_CAM_WEP104;
0364         break;
0365     case WLAN_CIPHER_SUITE_TKIP:
0366         hw_key_type = RTW_CAM_TKIP;
0367         key->flags |= IEEE80211_KEY_FLAG_GENERATE_MMIC;
0368         break;
0369     case WLAN_CIPHER_SUITE_CCMP:
0370         hw_key_type = RTW_CAM_AES;
0371         key->flags |= IEEE80211_KEY_FLAG_SW_MGMT_TX;
0372         break;
0373     default:
0374         rtw_err(rtwdev, "Unsupported key type for wowlan mode: %#x\n",
0375             key->cipher);
0376         hw_key_type = 0;
0377         break;
0378     }
0379 
0380     if (sta)
0381         iter_data->pairwise_key_type = hw_key_type;
0382     else
0383         iter_data->group_key_type = hw_key_type;
0384 }
0385 
0386 static void rtw_wow_fw_security_type(struct rtw_dev *rtwdev)
0387 {
0388     struct rtw_fw_key_type_iter_data data = {};
0389     struct ieee80211_vif *wow_vif = rtwdev->wow.wow_vif;
0390 
0391     data.rtwdev = rtwdev;
0392     rtw_iterate_keys(rtwdev, wow_vif,
0393              rtw_wow_fw_security_type_iter, &data);
0394     rtw_fw_set_aoac_global_info_cmd(rtwdev, data.pairwise_key_type,
0395                     data.group_key_type);
0396 }
0397 
0398 static int rtw_wow_fw_start(struct rtw_dev *rtwdev)
0399 {
0400     if (rtw_wow_mgd_linked(rtwdev)) {
0401         rtw_send_rsvd_page_h2c(rtwdev);
0402         rtw_wow_pattern_write(rtwdev);
0403         rtw_wow_fw_security_type(rtwdev);
0404         rtw_fw_set_disconnect_decision_cmd(rtwdev, true);
0405         rtw_fw_set_keep_alive_cmd(rtwdev, true);
0406     } else if (rtw_wow_no_link(rtwdev)) {
0407         rtw_fw_set_nlo_info(rtwdev, true);
0408         rtw_fw_update_pkt_probe_req(rtwdev, NULL);
0409         rtw_fw_channel_switch(rtwdev, true);
0410     }
0411 
0412     rtw_fw_set_wowlan_ctrl_cmd(rtwdev, true);
0413     rtw_fw_set_remote_wake_ctrl_cmd(rtwdev, true);
0414 
0415     return rtw_wow_check_fw_status(rtwdev, true);
0416 }
0417 
0418 static int rtw_wow_fw_stop(struct rtw_dev *rtwdev)
0419 {
0420     if (rtw_wow_mgd_linked(rtwdev)) {
0421         rtw_fw_set_disconnect_decision_cmd(rtwdev, false);
0422         rtw_fw_set_keep_alive_cmd(rtwdev, false);
0423         rtw_wow_pattern_clear(rtwdev);
0424     } else if (rtw_wow_no_link(rtwdev)) {
0425         rtw_fw_channel_switch(rtwdev, false);
0426         rtw_fw_set_nlo_info(rtwdev, false);
0427     }
0428 
0429     rtw_fw_set_wowlan_ctrl_cmd(rtwdev, false);
0430     rtw_fw_set_remote_wake_ctrl_cmd(rtwdev, false);
0431 
0432     return rtw_wow_check_fw_status(rtwdev, false);
0433 }
0434 
0435 static void rtw_wow_avoid_reset_mac(struct rtw_dev *rtwdev)
0436 {
0437     /* When resuming from wowlan mode, some hosts issue signal
0438      * (PCIE: PREST, USB: SE0RST) to device, and lead to reset
0439      * mac core. If it happens, the connection to AP will be lost.
0440      * Setting REG_RSV_CTRL Register can avoid this process.
0441      */
0442     switch (rtw_hci_type(rtwdev)) {
0443     case RTW_HCI_TYPE_PCIE:
0444     case RTW_HCI_TYPE_USB:
0445         rtw_write8(rtwdev, REG_RSV_CTRL, BIT_WLOCK_1C_B6);
0446         rtw_write8(rtwdev, REG_RSV_CTRL,
0447                BIT_WLOCK_1C_B6 | BIT_R_DIS_PRST);
0448         break;
0449     default:
0450         rtw_warn(rtwdev, "Unsupported hci type to disable reset MAC\n");
0451         break;
0452     }
0453 }
0454 
0455 static void rtw_wow_fw_media_status_iter(void *data, struct ieee80211_sta *sta)
0456 {
0457     struct rtw_sta_info *si = (struct rtw_sta_info *)sta->drv_priv;
0458     struct rtw_fw_media_status_iter_data *iter_data = data;
0459     struct rtw_dev *rtwdev = iter_data->rtwdev;
0460 
0461     rtw_fw_media_status_report(rtwdev, si->mac_id, iter_data->connect);
0462 }
0463 
0464 static void rtw_wow_fw_media_status(struct rtw_dev *rtwdev, bool connect)
0465 {
0466     struct rtw_fw_media_status_iter_data data;
0467 
0468     data.rtwdev = rtwdev;
0469     data.connect = connect;
0470 
0471     rtw_iterate_stas_atomic(rtwdev, rtw_wow_fw_media_status_iter, &data);
0472 }
0473 
0474 static int rtw_wow_config_wow_fw_rsvd_page(struct rtw_dev *rtwdev)
0475 {
0476     struct ieee80211_vif *wow_vif = rtwdev->wow.wow_vif;
0477     struct rtw_vif *rtwvif = (struct rtw_vif *)wow_vif->drv_priv;
0478 
0479     rtw_remove_rsvd_page(rtwdev, rtwvif);
0480 
0481     if (rtw_wow_no_link(rtwdev))
0482         rtw_add_rsvd_page_pno(rtwdev, rtwvif);
0483     else
0484         rtw_add_rsvd_page_sta(rtwdev, rtwvif);
0485 
0486     return rtw_fw_download_rsvd_page(rtwdev);
0487 }
0488 
0489 static int rtw_wow_config_normal_fw_rsvd_page(struct rtw_dev *rtwdev)
0490 {
0491     struct ieee80211_vif *wow_vif = rtwdev->wow.wow_vif;
0492     struct rtw_vif *rtwvif = (struct rtw_vif *)wow_vif->drv_priv;
0493 
0494     rtw_remove_rsvd_page(rtwdev, rtwvif);
0495     rtw_add_rsvd_page_sta(rtwdev, rtwvif);
0496 
0497     if (rtw_wow_no_link(rtwdev))
0498         return 0;
0499 
0500     return rtw_fw_download_rsvd_page(rtwdev);
0501 }
0502 
0503 static int rtw_wow_swap_fw(struct rtw_dev *rtwdev, enum rtw_fw_type type)
0504 {
0505     struct rtw_fw_state *fw;
0506     int ret;
0507 
0508     switch (type) {
0509     case RTW_WOWLAN_FW:
0510         fw = &rtwdev->wow_fw;
0511         break;
0512 
0513     case RTW_NORMAL_FW:
0514         fw = &rtwdev->fw;
0515         break;
0516 
0517     default:
0518         rtw_warn(rtwdev, "unsupported firmware type to swap\n");
0519         return -ENOENT;
0520     }
0521 
0522     ret = rtw_download_firmware(rtwdev, fw);
0523     if (ret)
0524         goto out;
0525 
0526     rtw_fw_send_general_info(rtwdev);
0527     rtw_fw_send_phydm_info(rtwdev);
0528     rtw_wow_fw_media_status(rtwdev, true);
0529 
0530 out:
0531     return ret;
0532 }
0533 
0534 static void rtw_wow_check_pno(struct rtw_dev *rtwdev,
0535                   struct cfg80211_sched_scan_request *nd_config)
0536 {
0537     struct rtw_wow_param *rtw_wow = &rtwdev->wow;
0538     struct rtw_pno_request *pno_req = &rtw_wow->pno_req;
0539     struct ieee80211_channel *channel;
0540     int i, size;
0541 
0542     if (!nd_config->n_match_sets || !nd_config->n_channels)
0543         goto err;
0544 
0545     pno_req->match_set_cnt = nd_config->n_match_sets;
0546     size = sizeof(*pno_req->match_sets) * pno_req->match_set_cnt;
0547     pno_req->match_sets = kmemdup(nd_config->match_sets, size, GFP_KERNEL);
0548     if (!pno_req->match_sets)
0549         goto err;
0550 
0551     pno_req->channel_cnt = nd_config->n_channels;
0552     size = sizeof(*nd_config->channels[0]) * nd_config->n_channels;
0553     pno_req->channels = kmalloc(size, GFP_KERNEL);
0554     if (!pno_req->channels)
0555         goto channel_err;
0556 
0557     for (i = 0 ; i < pno_req->channel_cnt; i++) {
0558         channel = pno_req->channels + i;
0559         memcpy(channel, nd_config->channels[i], sizeof(*channel));
0560     }
0561 
0562     pno_req->scan_plan = *nd_config->scan_plans;
0563     pno_req->inited = true;
0564 
0565     rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: net-detect is enabled\n");
0566 
0567     return;
0568 
0569 channel_err:
0570     kfree(pno_req->match_sets);
0571 
0572 err:
0573     rtw_dbg(rtwdev, RTW_DBG_WOW, "WOW: net-detect is disabled\n");
0574 }
0575 
0576 static int rtw_wow_leave_linked_ps(struct rtw_dev *rtwdev)
0577 {
0578     if (!test_bit(RTW_FLAG_WOWLAN, rtwdev->flags))
0579         cancel_delayed_work_sync(&rtwdev->watch_dog_work);
0580 
0581     rtw_leave_lps(rtwdev);
0582 
0583     return 0;
0584 }
0585 
0586 static int rtw_wow_leave_no_link_ps(struct rtw_dev *rtwdev)
0587 {
0588     struct rtw_wow_param *rtw_wow = &rtwdev->wow;
0589     int ret = 0;
0590 
0591     if (test_bit(RTW_FLAG_WOWLAN, rtwdev->flags)) {
0592         if (rtw_get_lps_deep_mode(rtwdev) != LPS_DEEP_MODE_NONE)
0593             rtw_leave_lps_deep(rtwdev);
0594     } else {
0595         if (test_bit(RTW_FLAG_INACTIVE_PS, rtwdev->flags)) {
0596             rtw_wow->ips_enabled = true;
0597             ret = rtw_leave_ips(rtwdev);
0598             if (ret)
0599                 return ret;
0600         }
0601     }
0602 
0603     return 0;
0604 }
0605 
0606 static int rtw_wow_leave_ps(struct rtw_dev *rtwdev)
0607 {
0608     int ret = 0;
0609 
0610     if (rtw_wow_mgd_linked(rtwdev))
0611         ret = rtw_wow_leave_linked_ps(rtwdev);
0612     else if (rtw_wow_no_link(rtwdev))
0613         ret = rtw_wow_leave_no_link_ps(rtwdev);
0614 
0615     return ret;
0616 }
0617 
0618 static int rtw_wow_restore_ps(struct rtw_dev *rtwdev)
0619 {
0620     int ret = 0;
0621 
0622     if (rtw_wow_no_link(rtwdev) && rtwdev->wow.ips_enabled)
0623         ret = rtw_enter_ips(rtwdev);
0624 
0625     return ret;
0626 }
0627 
0628 static int rtw_wow_enter_linked_ps(struct rtw_dev *rtwdev)
0629 {
0630     struct rtw_wow_param *rtw_wow = &rtwdev->wow;
0631     struct ieee80211_vif *wow_vif = rtw_wow->wow_vif;
0632     struct rtw_vif *rtwvif = (struct rtw_vif *)wow_vif->drv_priv;
0633 
0634     rtw_enter_lps(rtwdev, rtwvif->port);
0635 
0636     return 0;
0637 }
0638 
0639 static int rtw_wow_enter_no_link_ps(struct rtw_dev *rtwdev)
0640 {
0641     /* firmware enters deep ps by itself if supported */
0642     set_bit(RTW_FLAG_LEISURE_PS_DEEP, rtwdev->flags);
0643 
0644     return 0;
0645 }
0646 
0647 static int rtw_wow_enter_ps(struct rtw_dev *rtwdev)
0648 {
0649     int ret = 0;
0650 
0651     if (rtw_wow_mgd_linked(rtwdev))
0652         ret = rtw_wow_enter_linked_ps(rtwdev);
0653     else if (rtw_wow_no_link(rtwdev) &&
0654          rtw_get_lps_deep_mode(rtwdev) != LPS_DEEP_MODE_NONE)
0655         ret = rtw_wow_enter_no_link_ps(rtwdev);
0656 
0657     return ret;
0658 }
0659 
0660 static void rtw_wow_stop_trx(struct rtw_dev *rtwdev)
0661 {
0662     rtw_wow_bb_stop(rtwdev);
0663     rtw_wow_rx_dma_stop(rtwdev);
0664 }
0665 
0666 static int rtw_wow_start(struct rtw_dev *rtwdev)
0667 {
0668     int ret;
0669 
0670     ret = rtw_wow_fw_start(rtwdev);
0671     if (ret)
0672         goto out;
0673 
0674     rtw_hci_stop(rtwdev);
0675     rtw_wow_bb_start(rtwdev);
0676     rtw_wow_avoid_reset_mac(rtwdev);
0677 
0678 out:
0679     return ret;
0680 }
0681 
0682 static int rtw_wow_enable(struct rtw_dev *rtwdev)
0683 {
0684     int ret = 0;
0685 
0686     rtw_wow_stop_trx(rtwdev);
0687 
0688     ret = rtw_wow_swap_fw(rtwdev, RTW_WOWLAN_FW);
0689     if (ret) {
0690         rtw_err(rtwdev, "failed to swap wow fw\n");
0691         goto error;
0692     }
0693 
0694     set_bit(RTW_FLAG_WOWLAN, rtwdev->flags);
0695 
0696     ret = rtw_wow_config_wow_fw_rsvd_page(rtwdev);
0697     if (ret) {
0698         rtw_err(rtwdev, "failed to download wowlan rsvd page\n");
0699         goto error;
0700     }
0701 
0702     ret = rtw_wow_start(rtwdev);
0703     if (ret) {
0704         rtw_err(rtwdev, "failed to start wow\n");
0705         goto error;
0706     }
0707 
0708     return ret;
0709 
0710 error:
0711     clear_bit(RTW_FLAG_WOWLAN, rtwdev->flags);
0712     return ret;
0713 }
0714 
0715 static int rtw_wow_stop(struct rtw_dev *rtwdev)
0716 {
0717     int ret;
0718 
0719     /* some HCI related registers will be reset after resume,
0720      * need to set them again.
0721      */
0722     ret = rtw_hci_setup(rtwdev);
0723     if (ret) {
0724         rtw_err(rtwdev, "failed to setup hci\n");
0725         return ret;
0726     }
0727 
0728     ret = rtw_hci_start(rtwdev);
0729     if (ret) {
0730         rtw_err(rtwdev, "failed to start hci\n");
0731         return ret;
0732     }
0733 
0734     ret = rtw_wow_fw_stop(rtwdev);
0735     if (ret)
0736         rtw_err(rtwdev, "failed to stop wowlan fw\n");
0737 
0738     rtw_wow_bb_stop(rtwdev);
0739 
0740     return ret;
0741 }
0742 
0743 static void rtw_wow_resume_trx(struct rtw_dev *rtwdev)
0744 {
0745     rtw_wow_rx_dma_start(rtwdev);
0746     rtw_wow_bb_start(rtwdev);
0747     ieee80211_queue_delayed_work(rtwdev->hw, &rtwdev->watch_dog_work,
0748                      RTW_WATCH_DOG_DELAY_TIME);
0749 }
0750 
0751 static int rtw_wow_disable(struct rtw_dev *rtwdev)
0752 {
0753     int ret;
0754 
0755     clear_bit(RTW_FLAG_WOWLAN, rtwdev->flags);
0756 
0757     ret = rtw_wow_stop(rtwdev);
0758     if (ret) {
0759         rtw_err(rtwdev, "failed to stop wow\n");
0760         goto out;
0761     }
0762 
0763     ret = rtw_wow_swap_fw(rtwdev, RTW_NORMAL_FW);
0764     if (ret) {
0765         rtw_err(rtwdev, "failed to swap normal fw\n");
0766         goto out;
0767     }
0768 
0769     ret = rtw_wow_config_normal_fw_rsvd_page(rtwdev);
0770     if (ret)
0771         rtw_err(rtwdev, "failed to download normal rsvd page\n");
0772 
0773 out:
0774     rtw_wow_resume_trx(rtwdev);
0775     return ret;
0776 }
0777 
0778 static void rtw_wow_vif_iter(void *data, u8 *mac, struct ieee80211_vif *vif)
0779 {
0780     struct rtw_dev *rtwdev = data;
0781     struct rtw_vif *rtwvif = (struct rtw_vif *)vif->drv_priv;
0782     struct rtw_wow_param *rtw_wow = &rtwdev->wow;
0783 
0784     /* Current wowlan function support setting of only one STATION vif.
0785      * So when one suitable vif is found, stop the iteration.
0786      */
0787     if (rtw_wow->wow_vif || vif->type != NL80211_IFTYPE_STATION)
0788         return;
0789 
0790     switch (rtwvif->net_type) {
0791     case RTW_NET_MGD_LINKED:
0792         rtw_wow->wow_vif = vif;
0793         break;
0794     case RTW_NET_NO_LINK:
0795         if (rtw_wow->pno_req.inited)
0796             rtwdev->wow.wow_vif = vif;
0797         break;
0798     default:
0799         break;
0800     }
0801 }
0802 
0803 static int rtw_wow_set_wakeups(struct rtw_dev *rtwdev,
0804                    struct cfg80211_wowlan *wowlan)
0805 {
0806     struct rtw_wow_param *rtw_wow = &rtwdev->wow;
0807     struct rtw_wow_pattern *rtw_patterns = rtw_wow->patterns;
0808     struct rtw_vif *rtwvif;
0809     int i;
0810 
0811     if (wowlan->disconnect)
0812         set_bit(RTW_WOW_FLAG_EN_DISCONNECT, rtw_wow->flags);
0813     if (wowlan->magic_pkt)
0814         set_bit(RTW_WOW_FLAG_EN_MAGIC_PKT, rtw_wow->flags);
0815     if (wowlan->gtk_rekey_failure)
0816         set_bit(RTW_WOW_FLAG_EN_REKEY_PKT, rtw_wow->flags);
0817 
0818     if (wowlan->nd_config)
0819         rtw_wow_check_pno(rtwdev, wowlan->nd_config);
0820 
0821     rtw_iterate_vifs_atomic(rtwdev, rtw_wow_vif_iter, rtwdev);
0822     if (!rtw_wow->wow_vif)
0823         return -EPERM;
0824 
0825     rtwvif = (struct rtw_vif *)rtw_wow->wow_vif->drv_priv;
0826     if (wowlan->n_patterns && wowlan->patterns) {
0827         rtw_wow->pattern_cnt = wowlan->n_patterns;
0828         for (i = 0; i < wowlan->n_patterns; i++)
0829             rtw_wow_pattern_generate(rtwdev, rtwvif,
0830                          wowlan->patterns + i,
0831                          rtw_patterns + i);
0832     }
0833 
0834     return 0;
0835 }
0836 
0837 static void rtw_wow_clear_wakeups(struct rtw_dev *rtwdev)
0838 {
0839     struct rtw_wow_param *rtw_wow = &rtwdev->wow;
0840     struct rtw_pno_request *pno_req = &rtw_wow->pno_req;
0841 
0842     if (pno_req->inited) {
0843         kfree(pno_req->channels);
0844         kfree(pno_req->match_sets);
0845     }
0846 
0847     memset(rtw_wow, 0, sizeof(rtwdev->wow));
0848 }
0849 
0850 int rtw_wow_suspend(struct rtw_dev *rtwdev, struct cfg80211_wowlan *wowlan)
0851 {
0852     int ret = 0;
0853 
0854     ret = rtw_wow_set_wakeups(rtwdev, wowlan);
0855     if (ret) {
0856         rtw_err(rtwdev, "failed to set wakeup event\n");
0857         goto out;
0858     }
0859 
0860     ret = rtw_wow_leave_ps(rtwdev);
0861     if (ret) {
0862         rtw_err(rtwdev, "failed to leave ps from normal mode\n");
0863         goto out;
0864     }
0865 
0866     ret = rtw_wow_enable(rtwdev);
0867     if (ret) {
0868         rtw_err(rtwdev, "failed to enable wow\n");
0869         rtw_wow_restore_ps(rtwdev);
0870         goto out;
0871     }
0872 
0873     ret = rtw_wow_enter_ps(rtwdev);
0874     if (ret)
0875         rtw_err(rtwdev, "failed to enter ps for wow\n");
0876 
0877 out:
0878     return ret;
0879 }
0880 
0881 int rtw_wow_resume(struct rtw_dev *rtwdev)
0882 {
0883     int ret;
0884 
0885     /* If wowlan mode is not enabled, do nothing */
0886     if (!test_bit(RTW_FLAG_WOWLAN, rtwdev->flags)) {
0887         rtw_err(rtwdev, "wow is not enabled\n");
0888         ret = -EPERM;
0889         goto out;
0890     }
0891 
0892     ret = rtw_wow_leave_ps(rtwdev);
0893     if (ret) {
0894         rtw_err(rtwdev, "failed to leave ps from wowlan mode\n");
0895         goto out;
0896     }
0897 
0898     rtw_wow_show_wakeup_reason(rtwdev);
0899 
0900     ret = rtw_wow_disable(rtwdev);
0901     if (ret) {
0902         rtw_err(rtwdev, "failed to disable wow\n");
0903         goto out;
0904     }
0905 
0906     ret = rtw_wow_restore_ps(rtwdev);
0907     if (ret)
0908         rtw_err(rtwdev, "failed to restore ps to normal mode\n");
0909 
0910 out:
0911     rtw_wow_clear_wakeups(rtwdev);
0912     return ret;
0913 }