drivers/net/wireless/rtlwifi/ps.c

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