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