LXR linux/drivers/net/wireless/realtek/rtlwifi/ps.c

   1// SPDX-License-Identifier: GPL-2.0
   2/* Copyright(c) 2009-2012  Realtek Corporation.*/
   3
   4#include "wifi.h"
   5#include "base.h"
   6#include "ps.h"
   7#include <linux/export.h>
   8#include "btcoexist/rtl_btc.h"
   9
  10bool rtl_ps_enable_nic(struct ieee80211_hw *hw)
  11{
  12        struct rtl_priv *rtlpriv = rtl_priv(hw);
  13        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
  14        struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
  15        struct rtl_mac *rtlmac = rtl_mac(rtl_priv(hw));
  16
  17        /*<1> reset trx ring */
  18        if (rtlhal->interface == INTF_PCI)
  19                rtlpriv->intf_ops->reset_trx_ring(hw);
  20
  21        if (is_hal_stop(rtlhal))
  22                RT_TRACE(rtlpriv, COMP_ERR, DBG_WARNING,
  23                         "Driver is already down!\n");
  24
  25        /*<2> Enable Adapter */
  26        if (rtlpriv->cfg->ops->hw_init(hw))
  27                return false;
  28        rtlpriv->cfg->ops->set_hw_reg(hw, HW_VAR_RETRY_LIMIT,
  29                        &rtlmac->retry_long);
  30        RT_CLEAR_PS_LEVEL(ppsc, RT_RF_OFF_LEVL_HALT_NIC);
  31
  32        rtlpriv->cfg->ops->switch_channel(hw);
  33        rtlpriv->cfg->ops->set_channel_access(hw);
  34        rtlpriv->cfg->ops->set_bw_mode(hw,
  35                        cfg80211_get_chandef_type(&hw->conf.chandef));
  36
  37        /*<3> Enable Interrupt */
  38        rtlpriv->cfg->ops->enable_interrupt(hw);
  39
  40        /*<enable timer> */
  41        rtl_watch_dog_timer_callback(&rtlpriv->works.watchdog_timer);
  42
  43        return true;
  44}
  45EXPORT_SYMBOL(rtl_ps_enable_nic);
  46
  47bool rtl_ps_disable_nic(struct ieee80211_hw *hw)
  48{
  49        struct rtl_priv *rtlpriv = rtl_priv(hw);
  50
  51        /*<1> Stop all timer */
  52        rtl_deinit_deferred_work(hw, true);
  53
  54        /*<2> Disable Interrupt */
  55        rtlpriv->cfg->ops->disable_interrupt(hw);
  56        tasklet_kill(&rtlpriv->works.irq_tasklet);
  57
  58        /*<3> Disable Adapter */
  59        rtlpriv->cfg->ops->hw_disable(hw);
  60
  61        return true;
  62}
  63EXPORT_SYMBOL(rtl_ps_disable_nic);
  64
  65static bool rtl_ps_set_rf_state(struct ieee80211_hw *hw,
  66                                enum rf_pwrstate state_toset,
  67                                u32 changesource)
  68{
  69        struct rtl_priv *rtlpriv = rtl_priv(hw);
  70        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
  71        enum rf_pwrstate rtstate;
  72        bool actionallowed = false;
  73        u16 rfwait_cnt = 0;
  74
  75        /*Only one thread can change
  76         *the RF state at one time, and others
  77         *should wait to be executed.
  78         */
  79        while (true) {
  80                spin_lock(&rtlpriv->locks.rf_ps_lock);
  81                if (ppsc->rfchange_inprogress) {
  82                        spin_unlock(&rtlpriv->locks.rf_ps_lock);
  83
  84                        RT_TRACE(rtlpriv, COMP_ERR, DBG_WARNING,
  85                                 "RF Change in progress! Wait to set..state_toset(%d).\n",
  86                                  state_toset);
  87
  88                        /* Set RF after the previous action is done.  */
  89                        while (ppsc->rfchange_inprogress) {
  90                                rfwait_cnt++;
  91                                mdelay(1);
  92                                /*Wait too long, return false to avoid
  93                                 *to be stuck here.
  94                                 */
  95                                if (rfwait_cnt > 100)
  96                                        return false;
  97                        }
  98                } else {
  99                        ppsc->rfchange_inprogress = true;
 100                        spin_unlock(&rtlpriv->locks.rf_ps_lock);
 101                        break;
 102                }
 103        }
 104
 105        rtstate = ppsc->rfpwr_state;
 106
 107        switch (state_toset) {
 108        case ERFON:
 109                ppsc->rfoff_reason &= (~changesource);
 110
 111                if ((changesource == RF_CHANGE_BY_HW) &&
 112                    (ppsc->hwradiooff)) {
 113                        ppsc->hwradiooff = false;
 114                }
 115
 116                if (!ppsc->rfoff_reason) {
 117                        ppsc->rfoff_reason = 0;
 118                        actionallowed = true;
 119                }
 120
 121                break;
 122
 123        case ERFOFF:
 124
 125                if ((changesource == RF_CHANGE_BY_HW) && !ppsc->hwradiooff) {
 126                        ppsc->hwradiooff = true;
 127                }
 128
 129                ppsc->rfoff_reason |= changesource;
 130                actionallowed = true;
 131                break;
 132
 133        case ERFSLEEP:
 134                ppsc->rfoff_reason |= changesource;
 135                actionallowed = true;
 136                break;
 137
 138        default:
 139                pr_err("switch case %#x not processed\n", state_toset);
 140                break;
 141        }
 142
 143        if (actionallowed)
 144                rtlpriv->cfg->ops->set_rf_power_state(hw, state_toset);
 145
 146        spin_lock(&rtlpriv->locks.rf_ps_lock);
 147        ppsc->rfchange_inprogress = false;
 148        spin_unlock(&rtlpriv->locks.rf_ps_lock);
 149
 150        return actionallowed;
 151}
 152
 153static void _rtl_ps_inactive_ps(struct ieee80211_hw *hw)
 154{
 155        struct rtl_priv *rtlpriv = rtl_priv(hw);
 156        struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
 157        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
 158
 159        ppsc->swrf_processing = true;
 160
 161        if (ppsc->inactive_pwrstate == ERFON &&
 162            rtlhal->interface == INTF_PCI) {
 163                if ((ppsc->reg_rfps_level & RT_RF_OFF_LEVL_ASPM) &&
 164                    RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM) &&
 165                    rtlhal->interface == INTF_PCI) {
 166                        rtlpriv->intf_ops->disable_aspm(hw);
 167                        RT_CLEAR_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
 168                }
 169        }
 170
 171        rtl_ps_set_rf_state(hw, ppsc->inactive_pwrstate,
 172                            RF_CHANGE_BY_IPS);
 173
 174        if (ppsc->inactive_pwrstate == ERFOFF &&
 175            rtlhal->interface == INTF_PCI) {
 176                if (ppsc->reg_rfps_level & RT_RF_OFF_LEVL_ASPM &&
 177                    !RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM)) {
 178                        rtlpriv->intf_ops->enable_aspm(hw);
 179                        RT_SET_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
 180                }
 181        }
 182
 183        ppsc->swrf_processing = false;
 184}
 185
 186void rtl_ips_nic_off_wq_callback(void *data)
 187{
 188        struct rtl_works *rtlworks =
 189            container_of_dwork_rtl(data, struct rtl_works, ips_nic_off_wq);
 190        struct ieee80211_hw *hw = rtlworks->hw;
 191        struct rtl_priv *rtlpriv = rtl_priv(hw);
 192        struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
 193        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
 194        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
 195        enum rf_pwrstate rtstate;
 196
 197        if (mac->opmode != NL80211_IFTYPE_STATION) {
 198                RT_TRACE(rtlpriv, COMP_ERR, DBG_WARNING,
 199                         "not station return\n");
 200                return;
 201        }
 202
 203        if (mac->p2p_in_use)
 204                return;
 205
 206        if (mac->link_state > MAC80211_NOLINK)
 207                return;
 208
 209        if (is_hal_stop(rtlhal))
 210                return;
 211
 212        if (rtlpriv->sec.being_setkey)
 213                return;
 214
 215        if (rtlpriv->cfg->ops->bt_coex_off_before_lps)
 216                rtlpriv->cfg->ops->bt_coex_off_before_lps(hw);
 217
 218        if (ppsc->inactiveps) {
 219                rtstate = ppsc->rfpwr_state;
 220
 221                /*
 222                 *Do not enter IPS in the following conditions:
 223                 *(1) RF is already OFF or Sleep
 224                 *(2) swrf_processing (indicates the IPS is still under going)
 225                 *(3) Connectted (only disconnected can trigger IPS)
 226                 *(4) IBSS (send Beacon)
 227                 *(5) AP mode (send Beacon)
 228                 *(6) monitor mode (rcv packet)
 229                 */
 230
 231                if (rtstate == ERFON &&
 232                    !ppsc->swrf_processing &&
 233                    (mac->link_state == MAC80211_NOLINK) &&
 234                    !mac->act_scanning) {
 235                        RT_TRACE(rtlpriv, COMP_RF, DBG_TRACE,
 236                                 "IPSEnter(): Turn off RF\n");
 237
 238                        ppsc->inactive_pwrstate = ERFOFF;
 239                        ppsc->in_powersavemode = true;
 240
 241                        /* call before RF off */
 242                        if (rtlpriv->cfg->ops->get_btc_status())
 243                                rtlpriv->btcoexist.btc_ops->btc_ips_notify(rtlpriv,
 244                                                                        ppsc->inactive_pwrstate);
 245
 246                        /*rtl_pci_reset_trx_ring(hw); */
 247                        _rtl_ps_inactive_ps(hw);
 248                }
 249        }
 250}
 251
 252void rtl_ips_nic_off(struct ieee80211_hw *hw)
 253{
 254        struct rtl_priv *rtlpriv = rtl_priv(hw);
 255
 256        /* because when link with ap, mac80211 will ask us
 257         * to disable nic quickly after scan before linking,
 258         * this will cause link failed, so we delay 100ms here
 259         */
 260        queue_delayed_work(rtlpriv->works.rtl_wq,
 261                           &rtlpriv->works.ips_nic_off_wq, MSECS(100));
 262}
 263
 264/* NOTICE: any opmode should exc nic_on, or disable without
 265 * nic_on may something wrong, like adhoc TP
 266 */
 267void rtl_ips_nic_on(struct ieee80211_hw *hw)
 268{
 269        struct rtl_priv *rtlpriv = rtl_priv(hw);
 270        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
 271        enum rf_pwrstate rtstate;
 272
 273        cancel_delayed_work_sync(&rtlpriv->works.ips_nic_off_wq);
 274
 275        mutex_lock(&rtlpriv->locks.ips_mutex);
 276        if (ppsc->inactiveps) {
 277                rtstate = ppsc->rfpwr_state;
 278
 279                if (rtstate != ERFON &&
 280                    !ppsc->swrf_processing &&
 281                    ppsc->rfoff_reason <= RF_CHANGE_BY_IPS) {
 282
 283                        ppsc->inactive_pwrstate = ERFON;
 284                        ppsc->in_powersavemode = false;
 285                        _rtl_ps_inactive_ps(hw);
 286                        /* call after RF on */
 287                        if (rtlpriv->cfg->ops->get_btc_status())
 288                                rtlpriv->btcoexist.btc_ops->btc_ips_notify(rtlpriv,
 289                                                                        ppsc->inactive_pwrstate);
 290                }
 291        }
 292        mutex_unlock(&rtlpriv->locks.ips_mutex);
 293}
 294EXPORT_SYMBOL_GPL(rtl_ips_nic_on);
 295
 296/*for FW LPS*/
 297
 298/*
 299 *Determine if we can set Fw into PS mode
 300 *in current condition.Return TRUE if it
 301 *can enter PS mode.
 302 */
 303static bool rtl_get_fwlps_doze(struct ieee80211_hw *hw)
 304{
 305        struct rtl_priv *rtlpriv = rtl_priv(hw);
 306        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
 307        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
 308        u32 ps_timediff;
 309
 310        ps_timediff = jiffies_to_msecs(jiffies -
 311                                       ppsc->last_delaylps_stamp_jiffies);
 312
 313        if (ps_timediff < 2000) {
 314                RT_TRACE(rtlpriv, COMP_POWER, DBG_LOUD,
 315                         "Delay enter Fw LPS for DHCP, ARP, or EAPOL exchanging state\n");
 316                return false;
 317        }
 318
 319        if (mac->link_state != MAC80211_LINKED)
 320                return false;
 321
 322        if (mac->opmode == NL80211_IFTYPE_ADHOC)
 323                return false;
 324
 325        return true;
 326}
 327
 328/* Change current and default preamble mode.*/
 329void rtl_lps_set_psmode(struct ieee80211_hw *hw, u8 rt_psmode)
 330{
 331        struct rtl_priv *rtlpriv = rtl_priv(hw);
 332        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
 333        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
 334        bool enter_fwlps;
 335
 336        if (mac->opmode == NL80211_IFTYPE_ADHOC)
 337                return;
 338
 339        if (mac->link_state != MAC80211_LINKED)
 340                return;
 341
 342        if (ppsc->dot11_psmode == rt_psmode && rt_psmode == EACTIVE)
 343                return;
 344
 345        /* Update power save mode configured. */
 346        ppsc->dot11_psmode = rt_psmode;
 347
 348        /*
 349         *<FW control LPS>
 350         *1. Enter PS mode
 351         *   Set RPWM to Fw to turn RF off and send H2C fw_pwrmode
 352         *   cmd to set Fw into PS mode.
 353         *2. Leave PS mode
 354         *   Send H2C fw_pwrmode cmd to Fw to set Fw into Active
 355         *   mode and set RPWM to turn RF on.
 356         */
 357
 358        if ((ppsc->fwctrl_lps) && ppsc->report_linked) {
 359                if (ppsc->dot11_psmode == EACTIVE) {
 360                        RT_TRACE(rtlpriv, COMP_RF, DBG_DMESG,
 361                                 "FW LPS leave ps_mode:%x\n",
 362                                  FW_PS_ACTIVE_MODE);
 363                        enter_fwlps = false;
 364                        ppsc->pwr_mode = FW_PS_ACTIVE_MODE;
 365                        ppsc->smart_ps = 0;
 366                        rtlpriv->cfg->ops->set_hw_reg(hw, HW_VAR_FW_LPS_ACTION,
 367                                                      (u8 *)(&enter_fwlps));
 368                        if (ppsc->p2p_ps_info.opp_ps)
 369                                rtl_p2p_ps_cmd(hw , P2P_PS_ENABLE);
 370
 371                        if (rtlpriv->cfg->ops->get_btc_status())
 372                                rtlpriv->btcoexist.btc_ops->btc_lps_notify(rtlpriv, rt_psmode);
 373                } else {
 374                        if (rtl_get_fwlps_doze(hw)) {
 375                                RT_TRACE(rtlpriv, COMP_RF, DBG_DMESG,
 376                                         "FW LPS enter ps_mode:%x\n",
 377                                         ppsc->fwctrl_psmode);
 378                                if (rtlpriv->cfg->ops->get_btc_status())
 379                                        rtlpriv->btcoexist.btc_ops->btc_lps_notify(rtlpriv, rt_psmode);
 380                                enter_fwlps = true;
 381                                ppsc->pwr_mode = ppsc->fwctrl_psmode;
 382                                ppsc->smart_ps = 2;
 383                                rtlpriv->cfg->ops->set_hw_reg(hw,
 384                                                        HW_VAR_FW_LPS_ACTION,
 385                                                        (u8 *)(&enter_fwlps));
 386
 387                        } else {
 388                                /* Reset the power save related parameters. */
 389                                ppsc->dot11_psmode = EACTIVE;
 390                        }
 391                }
 392        }
 393}
 394
 395/* Interrupt safe routine to enter the leisure power save mode.*/
 396static void rtl_lps_enter_core(struct ieee80211_hw *hw)
 397{
 398        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
 399        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
 400        struct rtl_priv *rtlpriv = rtl_priv(hw);
 401
 402        if (!ppsc->fwctrl_lps)
 403                return;
 404
 405        if (rtlpriv->sec.being_setkey)
 406                return;
 407
 408        if (rtlpriv->link_info.busytraffic)
 409                return;
 410
 411        /*sleep after linked 10s, to let DHCP and 4-way handshake ok enough!! */
 412        if (mac->cnt_after_linked < 5)
 413                return;
 414
 415        if (mac->opmode == NL80211_IFTYPE_ADHOC)
 416                return;
 417
 418        if (mac->link_state != MAC80211_LINKED)
 419                return;
 420
 421        mutex_lock(&rtlpriv->locks.lps_mutex);
 422
 423        /* Don't need to check (ppsc->dot11_psmode == EACTIVE), because
 424         * bt_ccoexist may ask to enter lps.
 425         * In normal case, this constraint move to rtl_lps_set_psmode().
 426         */
 427        RT_TRACE(rtlpriv, COMP_POWER, DBG_LOUD,
 428                 "Enter 802.11 power save mode...\n");
 429        rtl_lps_set_psmode(hw, EAUTOPS);
 430
 431        mutex_unlock(&rtlpriv->locks.lps_mutex);
 432}
 433
 434/* Interrupt safe routine to leave the leisure power save mode.*/
 435static void rtl_lps_leave_core(struct ieee80211_hw *hw)
 436{
 437        struct rtl_priv *rtlpriv = rtl_priv(hw);
 438        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
 439        struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
 440
 441        mutex_lock(&rtlpriv->locks.lps_mutex);
 442
 443        if (ppsc->fwctrl_lps) {
 444                if (ppsc->dot11_psmode != EACTIVE) {
 445
 446                        /*FIX ME */
 447                        /*rtlpriv->cfg->ops->enable_interrupt(hw); */
 448
 449                        if (ppsc->reg_rfps_level & RT_RF_LPS_LEVEL_ASPM &&
 450                            RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM) &&
 451                            rtlhal->interface == INTF_PCI) {
 452                                rtlpriv->intf_ops->disable_aspm(hw);
 453                                RT_CLEAR_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
 454                        }
 455
 456                        RT_TRACE(rtlpriv, COMP_POWER, DBG_LOUD,
 457                                 "Busy Traffic,Leave 802.11 power save..\n");
 458
 459                        rtl_lps_set_psmode(hw, EACTIVE);
 460                }
 461        }
 462        mutex_unlock(&rtlpriv->locks.lps_mutex);
 463}
 464
 465/* For sw LPS*/
 466void rtl_swlps_beacon(struct ieee80211_hw *hw, void *data, unsigned int len)
 467{
 468        struct rtl_priv *rtlpriv = rtl_priv(hw);
 469        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
 470        struct ieee80211_hdr *hdr = data;
 471        struct ieee80211_tim_ie *tim_ie;
 472        u8 *tim;
 473        u8 tim_len;
 474        bool u_buffed;
 475        bool m_buffed;
 476
 477        if (mac->opmode != NL80211_IFTYPE_STATION)
 478                return;
 479
 480        if (!rtlpriv->psc.swctrl_lps)
 481                return;
 482
 483        if (rtlpriv->mac80211.link_state != MAC80211_LINKED)
 484                return;
 485
 486        if (!rtlpriv->psc.sw_ps_enabled)
 487                return;
 488
 489        if (rtlpriv->psc.fwctrl_lps)
 490                return;
 491
 492        if (likely(!(hw->conf.flags & IEEE80211_CONF_PS)))
 493                return;
 494
 495        /* check if this really is a beacon */
 496        if (!ieee80211_is_beacon(hdr->frame_control))
 497                return;
 498
 499        /* min. beacon length + FCS_LEN */
 500        if (len <= 40 + FCS_LEN)
 501                return;
 502
 503        /* and only beacons from the associated BSSID, please */
 504        if (!ether_addr_equal_64bits(hdr->addr3, rtlpriv->mac80211.bssid))
 505                return;
 506
 507        rtlpriv->psc.last_beacon = jiffies;
 508
 509        tim = rtl_find_ie(data, len - FCS_LEN, WLAN_EID_TIM);
 510        if (!tim)
 511                return;
 512
 513        if (tim[1] < sizeof(*tim_ie))
 514                return;
 515
 516        tim_len = tim[1];
 517        tim_ie = (struct ieee80211_tim_ie *) &tim[2];
 518
 519        if (!WARN_ON_ONCE(!hw->conf.ps_dtim_period))
 520                rtlpriv->psc.dtim_counter = tim_ie->dtim_count;
 521
 522        /* Check whenever the PHY can be turned off again. */
 523
 524        /* 1. What about buffered unicast traffic for our AID? */
 525        u_buffed = ieee80211_check_tim(tim_ie, tim_len,
 526                                       rtlpriv->mac80211.assoc_id);
 527
 528        /* 2. Maybe the AP wants to send multicast/broadcast data? */
 529        m_buffed = tim_ie->bitmap_ctrl & 0x01;
 530        rtlpriv->psc.multi_buffered = m_buffed;
 531
 532        /* unicast will process by mac80211 through
 533         * set ~IEEE80211_CONF_PS, So we just check
 534         * multicast frames here */
 535        if (!m_buffed) {
 536                /* back to low-power land. and delay is
 537                 * prevent null power save frame tx fail */
 538                queue_delayed_work(rtlpriv->works.rtl_wq,
 539                                   &rtlpriv->works.ps_work, MSECS(5));
 540        } else {
 541                RT_TRACE(rtlpriv, COMP_POWER, DBG_DMESG,
 542                         "u_bufferd: %x, m_buffered: %x\n", u_buffed, m_buffed);
 543        }
 544}
 545EXPORT_SYMBOL_GPL(rtl_swlps_beacon);
 546
 547void rtl_swlps_rf_awake(struct ieee80211_hw *hw)
 548{
 549        struct rtl_priv *rtlpriv = rtl_priv(hw);
 550        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
 551        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
 552
 553        if (!rtlpriv->psc.swctrl_lps)
 554                return;
 555        if (mac->link_state != MAC80211_LINKED)
 556                return;
 557
 558        if (ppsc->reg_rfps_level & RT_RF_LPS_LEVEL_ASPM &&
 559            RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM)) {
 560                rtlpriv->intf_ops->disable_aspm(hw);
 561                RT_CLEAR_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
 562        }
 563
 564        mutex_lock(&rtlpriv->locks.lps_mutex);
 565        rtl_ps_set_rf_state(hw, ERFON, RF_CHANGE_BY_PS);
 566        mutex_unlock(&rtlpriv->locks.lps_mutex);
 567}
 568
 569void rtl_swlps_rfon_wq_callback(void *data)
 570{
 571        struct rtl_works *rtlworks =
 572            container_of_dwork_rtl(data, struct rtl_works, ps_rfon_wq);
 573        struct ieee80211_hw *hw = rtlworks->hw;
 574
 575        rtl_swlps_rf_awake(hw);
 576}
 577
 578void rtl_swlps_rf_sleep(struct ieee80211_hw *hw)
 579{
 580        struct rtl_priv *rtlpriv = rtl_priv(hw);
 581        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
 582        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
 583        u8 sleep_intv;
 584
 585        if (!rtlpriv->psc.sw_ps_enabled)
 586                return;
 587
 588        if ((rtlpriv->sec.being_setkey) ||
 589            (mac->opmode == NL80211_IFTYPE_ADHOC))
 590                return;
 591
 592        /*sleep after linked 10s, to let DHCP and 4-way handshake ok enough!! */
 593        if ((mac->link_state != MAC80211_LINKED) || (mac->cnt_after_linked < 5))
 594                return;
 595
 596        if (rtlpriv->link_info.busytraffic)
 597                return;
 598
 599        spin_lock(&rtlpriv->locks.rf_ps_lock);
 600        if (rtlpriv->psc.rfchange_inprogress) {
 601                spin_unlock(&rtlpriv->locks.rf_ps_lock);
 602                return;
 603        }
 604        spin_unlock(&rtlpriv->locks.rf_ps_lock);
 605
 606        mutex_lock(&rtlpriv->locks.lps_mutex);
 607        rtl_ps_set_rf_state(hw, ERFSLEEP, RF_CHANGE_BY_PS);
 608        mutex_unlock(&rtlpriv->locks.lps_mutex);
 609
 610        if (ppsc->reg_rfps_level & RT_RF_OFF_LEVL_ASPM &&
 611            !RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM)) {
 612                rtlpriv->intf_ops->enable_aspm(hw);
 613                RT_SET_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
 614        }
 615
 616        /* here is power save alg, when this beacon is DTIM
 617         * we will set sleep time to dtim_period * n;
 618         * when this beacon is not DTIM, we will set sleep
 619         * time to sleep_intv = rtlpriv->psc.dtim_counter or
 620         * MAX_SW_LPS_SLEEP_INTV(default set to 5) */
 621
 622        if (rtlpriv->psc.dtim_counter == 0) {
 623                if (hw->conf.ps_dtim_period == 1)
 624                        sleep_intv = hw->conf.ps_dtim_period * 2;
 625                else
 626                        sleep_intv = hw->conf.ps_dtim_period;
 627        } else {
 628                sleep_intv = rtlpriv->psc.dtim_counter;
 629        }
 630
 631        if (sleep_intv > MAX_SW_LPS_SLEEP_INTV)
 632                sleep_intv = MAX_SW_LPS_SLEEP_INTV;
 633
 634        /* this print should always be dtim_conter = 0 &
 635         * sleep  = dtim_period, that meaons, we should
 636         * awake before every dtim */
 637        RT_TRACE(rtlpriv, COMP_POWER, DBG_DMESG,
 638                 "dtim_counter:%x will sleep :%d beacon_intv\n",
 639                  rtlpriv->psc.dtim_counter, sleep_intv);
 640
 641        /* we tested that 40ms is enough for sw & hw sw delay */
 642        queue_delayed_work(rtlpriv->works.rtl_wq, &rtlpriv->works.ps_rfon_wq,
 643                        MSECS(sleep_intv * mac->vif->bss_conf.beacon_int - 40));
 644}
 645
 646void rtl_lps_change_work_callback(struct work_struct *work)
 647{
 648        struct rtl_works *rtlworks =
 649            container_of(work, struct rtl_works, lps_change_work);
 650        struct ieee80211_hw *hw = rtlworks->hw;
 651        struct rtl_priv *rtlpriv = rtl_priv(hw);
 652
 653        if (rtlpriv->enter_ps)
 654                rtl_lps_enter_core(hw);
 655        else
 656                rtl_lps_leave_core(hw);
 657}
 658EXPORT_SYMBOL_GPL(rtl_lps_change_work_callback);
 659
 660void rtl_lps_enter(struct ieee80211_hw *hw)
 661{
 662        struct rtl_priv *rtlpriv = rtl_priv(hw);
 663
 664        if (!in_interrupt())
 665                return rtl_lps_enter_core(hw);
 666        rtlpriv->enter_ps = true;
 667        schedule_work(&rtlpriv->works.lps_change_work);
 668}
 669EXPORT_SYMBOL_GPL(rtl_lps_enter);
 670
 671void rtl_lps_leave(struct ieee80211_hw *hw)
 672{
 673        struct rtl_priv *rtlpriv = rtl_priv(hw);
 674
 675        if (!in_interrupt())
 676                return rtl_lps_leave_core(hw);
 677        rtlpriv->enter_ps = false;
 678        schedule_work(&rtlpriv->works.lps_change_work);
 679}
 680EXPORT_SYMBOL_GPL(rtl_lps_leave);
 681
 682void rtl_swlps_wq_callback(void *data)
 683{
 684        struct rtl_works *rtlworks = container_of_dwork_rtl(data,
 685                                     struct rtl_works,
 686                                     ps_work);
 687        struct ieee80211_hw *hw = rtlworks->hw;
 688        struct rtl_priv *rtlpriv = rtl_priv(hw);
 689        bool ps = false;
 690
 691        ps = (hw->conf.flags & IEEE80211_CONF_PS);
 692
 693        /* we can sleep after ps null send ok */
 694        if (rtlpriv->psc.state_inap) {
 695                rtl_swlps_rf_sleep(hw);
 696
 697                if (rtlpriv->psc.state && !ps) {
 698                        rtlpriv->psc.sleep_ms = jiffies_to_msecs(jiffies -
 699                                                 rtlpriv->psc.last_action);
 700                }
 701
 702                if (ps)
 703                        rtlpriv->psc.last_slept = jiffies;
 704
 705                rtlpriv->psc.last_action = jiffies;
 706                rtlpriv->psc.state = ps;
 707        }
 708}
 709
 710static void rtl_p2p_noa_ie(struct ieee80211_hw *hw, void *data,
 711                           unsigned int len)
 712{
 713        struct rtl_priv *rtlpriv = rtl_priv(hw);
 714        struct ieee80211_mgmt *mgmt = data;
 715        struct rtl_p2p_ps_info *p2pinfo = &(rtlpriv->psc.p2p_ps_info);
 716        u8 *pos, *end, *ie;
 717        u16 noa_len;
 718        static u8 p2p_oui_ie_type[4] = {0x50, 0x6f, 0x9a, 0x09};
 719        u8 noa_num, index , i, noa_index = 0;
 720        bool find_p2p_ie = false , find_p2p_ps_ie = false;
 721
 722        pos = (u8 *)mgmt->u.beacon.variable;
 723        end = data + len;
 724        ie = NULL;
 725
 726        while (pos + 1 < end) {
 727                if (pos + 2 + pos[1] > end)
 728                        return;
 729
 730                if (pos[0] == 221 && pos[1] > 4) {
 731                        if (memcmp(&pos[2], p2p_oui_ie_type, 4) == 0) {
 732                                ie = pos + 2+4;
 733                                break;
 734                        }
 735                }
 736                pos += 2 + pos[1];
 737        }
 738
 739        if (ie == NULL)
 740                return;
 741        find_p2p_ie = true;
 742        /*to find noa ie*/
 743        while (ie + 1 < end) {
 744                noa_len = READEF2BYTE((__le16 *)&ie[1]);
 745                if (ie + 3 + ie[1] > end)
 746                        return;
 747
 748                if (ie[0] == 12) {
 749                        find_p2p_ps_ie = true;
 750                        if ((noa_len - 2) % 13 != 0) {
 751                                RT_TRACE(rtlpriv, COMP_INIT, DBG_LOUD,
 752                                         "P2P notice of absence: invalid length.%d\n",
 753                                         noa_len);
 754                                return;
 755                        } else {
 756                                noa_num = (noa_len - 2) / 13;
 757                        }
 758                        noa_index = ie[3];
 759                        if (rtlpriv->psc.p2p_ps_info.p2p_ps_mode ==
 760                            P2P_PS_NONE || noa_index != p2pinfo->noa_index) {
 761                                RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD,
 762                                         "update NOA ie.\n");
 763                                p2pinfo->noa_index = noa_index;
 764                                p2pinfo->opp_ps = (ie[4] >> 7);
 765                                p2pinfo->ctwindow = ie[4] & 0x7F;
 766                                p2pinfo->noa_num = noa_num;
 767                                index = 5;
 768                                for (i = 0; i < noa_num; i++) {
 769                                        p2pinfo->noa_count_type[i] =
 770                                                        READEF1BYTE(ie+index);
 771                                        index += 1;
 772                                        p2pinfo->noa_duration[i] =
 773                                                 READEF4BYTE((__le32 *)ie+index);
 774                                        index += 4;
 775                                        p2pinfo->noa_interval[i] =
 776                                                 READEF4BYTE((__le32 *)ie+index);
 777                                        index += 4;
 778                                        p2pinfo->noa_start_time[i] =
 779                                                 READEF4BYTE((__le32 *)ie+index);
 780                                        index += 4;
 781                                }
 782
 783                                if (p2pinfo->opp_ps == 1) {
 784                                        p2pinfo->p2p_ps_mode = P2P_PS_CTWINDOW;
 785                                        /* Driver should wait LPS entering
 786                                         * CTWindow
 787                                         */
 788                                        if (rtlpriv->psc.fw_current_inpsmode)
 789                                                rtl_p2p_ps_cmd(hw,
 790                                                               P2P_PS_ENABLE);
 791                                } else if (p2pinfo->noa_num > 0) {
 792                                        p2pinfo->p2p_ps_mode = P2P_PS_NOA;
 793                                        rtl_p2p_ps_cmd(hw, P2P_PS_ENABLE);
 794                                } else if (p2pinfo->p2p_ps_mode > P2P_PS_NONE) {
 795                                        rtl_p2p_ps_cmd(hw, P2P_PS_DISABLE);
 796                                }
 797                        }
 798                        break;
 799                }
 800                ie += 3 + noa_len;
 801        }
 802
 803        if (find_p2p_ie == true) {
 804                if ((p2pinfo->p2p_ps_mode > P2P_PS_NONE) &&
 805                    (find_p2p_ps_ie == false))
 806                        rtl_p2p_ps_cmd(hw, P2P_PS_DISABLE);
 807        }
 808}
 809
 810static void rtl_p2p_action_ie(struct ieee80211_hw *hw, void *data,
 811                              unsigned int len)
 812{
 813        struct rtl_priv *rtlpriv = rtl_priv(hw);
 814        struct ieee80211_mgmt *mgmt = data;
 815        struct rtl_p2p_ps_info *p2pinfo = &(rtlpriv->psc.p2p_ps_info);
 816        u8 noa_num, index , i , noa_index = 0;
 817        u8 *pos, *end, *ie;
 818        u16 noa_len;
 819        static u8 p2p_oui_ie_type[4] = {0x50, 0x6f, 0x9a, 0x09};
 820
 821        pos = (u8 *)&mgmt->u.action.category;
 822        end = data + len;
 823        ie = NULL;
 824
 825        if (pos[0] == 0x7f) {
 826                if (memcmp(&pos[1], p2p_oui_ie_type, 4) == 0)
 827                        ie = pos + 3+4;
 828        }
 829
 830        if (ie == NULL)
 831                return;
 832
 833        RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD, "action frame find P2P IE.\n");
 834        /*to find noa ie*/
 835        while (ie + 1 < end) {
 836                noa_len = READEF2BYTE((__le16 *)&ie[1]);
 837                if (ie + 3 + ie[1] > end)
 838                        return;
 839
 840                if (ie[0] == 12) {
 841                        RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD, "find NOA IE.\n");
 842                        RT_PRINT_DATA(rtlpriv, COMP_FW, DBG_LOUD, "noa ie ",
 843                                      ie, noa_len);
 844                        if ((noa_len - 2) % 13 != 0) {
 845                                RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD,
 846                                         "P2P notice of absence: invalid length.%d\n",
 847                                         noa_len);
 848                                return;
 849                        } else {
 850                                noa_num = (noa_len - 2) / 13;
 851                        }
 852                        noa_index = ie[3];
 853                        if (rtlpriv->psc.p2p_ps_info.p2p_ps_mode ==
 854                            P2P_PS_NONE || noa_index != p2pinfo->noa_index) {
 855                                p2pinfo->noa_index = noa_index;
 856                                p2pinfo->opp_ps = (ie[4] >> 7);
 857                                p2pinfo->ctwindow = ie[4] & 0x7F;
 858                                p2pinfo->noa_num = noa_num;
 859                                index = 5;
 860                                for (i = 0; i < noa_num; i++) {
 861                                        p2pinfo->noa_count_type[i] =
 862                                                        READEF1BYTE(ie+index);
 863                                        index += 1;
 864                                        p2pinfo->noa_duration[i] =
 865                                                         READEF4BYTE((__le32 *)ie+index);
 866                                        index += 4;
 867                                        p2pinfo->noa_interval[i] =
 868                                                         READEF4BYTE((__le32 *)ie+index);
 869                                        index += 4;
 870                                        p2pinfo->noa_start_time[i] =
 871                                                         READEF4BYTE((__le32 *)ie+index);
 872                                        index += 4;
 873                                }
 874
 875                                if (p2pinfo->opp_ps == 1) {
 876                                        p2pinfo->p2p_ps_mode = P2P_PS_CTWINDOW;
 877                                        /* Driver should wait LPS entering
 878                                         * CTWindow
 879                                         */
 880                                        if (rtlpriv->psc.fw_current_inpsmode)
 881                                                rtl_p2p_ps_cmd(hw,
 882                                                               P2P_PS_ENABLE);
 883                                } else if (p2pinfo->noa_num > 0) {
 884                                        p2pinfo->p2p_ps_mode = P2P_PS_NOA;
 885                                        rtl_p2p_ps_cmd(hw, P2P_PS_ENABLE);
 886                                } else if (p2pinfo->p2p_ps_mode > P2P_PS_NONE) {
 887                                        rtl_p2p_ps_cmd(hw, P2P_PS_DISABLE);
 888                                }
 889                        }
 890                        break;
 891                }
 892                ie += 3 + noa_len;
 893        }
 894}
 895
 896void rtl_p2p_ps_cmd(struct ieee80211_hw *hw , u8 p2p_ps_state)
 897{
 898        struct rtl_priv *rtlpriv = rtl_priv(hw);
 899        struct rtl_ps_ctl *rtlps = rtl_psc(rtl_priv(hw));
 900        struct rtl_p2p_ps_info  *p2pinfo = &(rtlpriv->psc.p2p_ps_info);
 901
 902        RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD, " p2p state %x\n" , p2p_ps_state);
 903        switch (p2p_ps_state) {
 904        case P2P_PS_DISABLE:
 905                p2pinfo->p2p_ps_state = p2p_ps_state;
 906                rtlpriv->cfg->ops->set_hw_reg(hw, HW_VAR_H2C_FW_P2P_PS_OFFLOAD,
 907                                              &p2p_ps_state);
 908                p2pinfo->noa_index = 0;
 909                p2pinfo->ctwindow = 0;
 910                p2pinfo->opp_ps = 0;
 911                p2pinfo->noa_num = 0;
 912                p2pinfo->p2p_ps_mode = P2P_PS_NONE;
 913                if (rtlps->fw_current_inpsmode) {
 914                        if (rtlps->smart_ps == 0) {
 915                                rtlps->smart_ps = 2;
 916                                rtlpriv->cfg->ops->set_hw_reg(hw,
 917                                         HW_VAR_H2C_FW_PWRMODE,
 918                                         &rtlps->pwr_mode);
 919                        }
 920
 921                }
 922                break;
 923        case P2P_PS_ENABLE:
 924                if (p2pinfo->p2p_ps_mode > P2P_PS_NONE) {
 925                        p2pinfo->p2p_ps_state = p2p_ps_state;
 926
 927                        if (p2pinfo->ctwindow > 0) {
 928                                if (rtlps->smart_ps != 0) {
 929                                        rtlps->smart_ps = 0;
 930                                        rtlpriv->cfg->ops->set_hw_reg(hw,
 931                                                 HW_VAR_H2C_FW_PWRMODE,
 932                                                 &rtlps->pwr_mode);
 933                                }
 934                        }
 935                        rtlpriv->cfg->ops->set_hw_reg(hw,
 936                                 HW_VAR_H2C_FW_P2P_PS_OFFLOAD,
 937                                 &p2p_ps_state);
 938
 939                }
 940                break;
 941        case P2P_PS_SCAN:
 942        case P2P_PS_SCAN_DONE:
 943        case P2P_PS_ALLSTASLEEP:
 944                if (p2pinfo->p2p_ps_mode > P2P_PS_NONE) {
 945                        p2pinfo->p2p_ps_state = p2p_ps_state;
 946                        rtlpriv->cfg->ops->set_hw_reg(hw,
 947                                 HW_VAR_H2C_FW_P2P_PS_OFFLOAD,
 948                                 &p2p_ps_state);
 949                }
 950                break;
 951        default:
 952                break;
 953        }
 954        RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD,
 955                 "ctwindow %x oppps %x\n",
 956                 p2pinfo->ctwindow , p2pinfo->opp_ps);
 957        RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD,
 958                 "count %x duration %x index %x interval %x start time %x noa num %x\n",
 959                 p2pinfo->noa_count_type[0],
 960                 p2pinfo->noa_duration[0],
 961                 p2pinfo->noa_index,
 962                 p2pinfo->noa_interval[0],
 963                 p2pinfo->noa_start_time[0],
 964                 p2pinfo->noa_num);
 965        RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD, "end\n");
 966}
 967
 968void rtl_p2p_info(struct ieee80211_hw *hw, void *data, unsigned int len)
 969{
 970        struct rtl_priv *rtlpriv = rtl_priv(hw);
 971        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
 972        struct ieee80211_hdr *hdr = data;
 973
 974        if (!mac->p2p)
 975                return;
 976        if (mac->link_state != MAC80211_LINKED)
 977                return;
 978        /* min. beacon length + FCS_LEN */
 979        if (len <= 40 + FCS_LEN)
 980                return;
 981
 982        /* and only beacons from the associated BSSID, please */
 983        if (!ether_addr_equal_64bits(hdr->addr3, rtlpriv->mac80211.bssid))
 984                return;
 985
 986        /* check if this really is a beacon */
 987        if (!(ieee80211_is_beacon(hdr->frame_control) ||
 988              ieee80211_is_probe_resp(hdr->frame_control) ||
 989              ieee80211_is_action(hdr->frame_control)))
 990                return;
 991
 992        if (ieee80211_is_action(hdr->frame_control))
 993                rtl_p2p_action_ie(hw , data , len - FCS_LEN);
 994        else
 995                rtl_p2p_noa_ie(hw , data , len - FCS_LEN);
 996}
 997EXPORT_SYMBOL_GPL(rtl_p2p_info);
 998