// SPDX-License-Identifier: GPL-2.0
/* Copyright(c) 2009-2012  Realtek Corporation.*/

#include "wifi.h"
#include "base.h"
#include "ps.h"
#include <linux/export.h>
#include "btcoexist/rtl_btc.h"

bool rtl_ps_enable_nic(struct ieee80211_hw *hw)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
        struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
        struct rtl_mac *rtlmac = rtl_mac(rtl_priv(hw));

        /*<1> reset trx ring */
        if (rtlhal->interface == INTF_PCI)
                rtlpriv->intf_ops->reset_trx_ring(hw);

        if (is_hal_stop(rtlhal))
                rtl_dbg(rtlpriv, COMP_ERR, DBG_WARNING,
                        "Driver is already down!\n");

        /*<2> Enable Adapter */
        if (rtlpriv->cfg->ops->hw_init(hw))
                return false;
        rtlpriv->cfg->ops->set_hw_reg(hw, HW_VAR_RETRY_LIMIT,
                        &rtlmac->retry_long);
        RT_CLEAR_PS_LEVEL(ppsc, RT_RF_OFF_LEVL_HALT_NIC);

        rtlpriv->cfg->ops->switch_channel(hw);
        rtlpriv->cfg->ops->set_channel_access(hw);
        rtlpriv->cfg->ops->set_bw_mode(hw,
                        cfg80211_get_chandef_type(&hw->conf.chandef));

        /*<3> Enable Interrupt */
        rtlpriv->cfg->ops->enable_interrupt(hw);

        /*<enable timer> */
        rtl_watch_dog_timer_callback(&rtlpriv->works.watchdog_timer);

        return true;
}
EXPORT_SYMBOL(rtl_ps_enable_nic);

bool rtl_ps_disable_nic(struct ieee80211_hw *hw)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        /*<1> Stop all timer */
        rtl_deinit_deferred_work(hw, true);

        /*<2> Disable Interrupt */
        rtlpriv->cfg->ops->disable_interrupt(hw);
        tasklet_kill(&rtlpriv->works.irq_tasklet);

        /*<3> Disable Adapter */
        rtlpriv->cfg->ops->hw_disable(hw);

        return true;
}
EXPORT_SYMBOL(rtl_ps_disable_nic);

static bool rtl_ps_set_rf_state(struct ieee80211_hw *hw,
                                enum rf_pwrstate state_toset,
                                u32 changesource)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
        bool actionallowed = false;
        u16 rfwait_cnt = 0;

        /*Only one thread can change
         *the RF state at one time, and others
         *should wait to be executed.
         */
        while (true) {
                spin_lock(&rtlpriv->locks.rf_ps_lock);
                if (ppsc->rfchange_inprogress) {
                        spin_unlock(&rtlpriv->locks.rf_ps_lock);

                        rtl_dbg(rtlpriv, COMP_ERR, DBG_WARNING,
                                "RF Change in progress! Wait to set..state_toset(%d).\n",
                                state_toset);

                        /* Set RF after the previous action is done.  */
                        while (ppsc->rfchange_inprogress) {
                                rfwait_cnt++;
                                mdelay(1);
                                /*Wait too long, return false to avoid
                                 *to be stuck here.
                                 */
                                if (rfwait_cnt > 100)
                                        return false;
                        }
                } else {
                        ppsc->rfchange_inprogress = true;
                        spin_unlock(&rtlpriv->locks.rf_ps_lock);
                        break;
                }
        }

        switch (state_toset) {
        case ERFON:
                ppsc->rfoff_reason &= (~changesource);

                if ((changesource == RF_CHANGE_BY_HW) &&
                    (ppsc->hwradiooff)) {
                        ppsc->hwradiooff = false;
                }

                if (!ppsc->rfoff_reason) {
                        ppsc->rfoff_reason = 0;
                        actionallowed = true;
                }

                break;

        case ERFOFF:

                if ((changesource == RF_CHANGE_BY_HW) && !ppsc->hwradiooff) {
                        ppsc->hwradiooff = true;
                }

                ppsc->rfoff_reason |= changesource;
                actionallowed = true;
                break;

        case ERFSLEEP:
                ppsc->rfoff_reason |= changesource;
                actionallowed = true;
                break;

        default:
                pr_err("switch case %#x not processed\n", state_toset);
                break;
        }

        if (actionallowed)
                rtlpriv->cfg->ops->set_rf_power_state(hw, state_toset);

        spin_lock(&rtlpriv->locks.rf_ps_lock);
        ppsc->rfchange_inprogress = false;
        spin_unlock(&rtlpriv->locks.rf_ps_lock);

        return actionallowed;
}

static void _rtl_ps_inactive_ps(struct ieee80211_hw *hw)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));

        ppsc->swrf_processing = true;

        if (ppsc->inactive_pwrstate == ERFON &&
            rtlhal->interface == INTF_PCI) {
                if ((ppsc->reg_rfps_level & RT_RF_OFF_LEVL_ASPM) &&
                    RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM)) {
                        rtlpriv->intf_ops->disable_aspm(hw);
                        RT_CLEAR_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
                }
        }

        rtl_ps_set_rf_state(hw, ppsc->inactive_pwrstate,
                            RF_CHANGE_BY_IPS);

        if (ppsc->inactive_pwrstate == ERFOFF &&
            rtlhal->interface == INTF_PCI) {
                if (ppsc->reg_rfps_level & RT_RF_OFF_LEVL_ASPM &&
                    !RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM)) {
                        rtlpriv->intf_ops->enable_aspm(hw);
                        RT_SET_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
                }
        }

        ppsc->swrf_processing = false;
}

void rtl_ips_nic_off_wq_callback(struct work_struct *work)
{
        struct rtl_works *rtlworks = container_of(work, struct rtl_works,
                                                  ips_nic_off_wq.work);
        struct ieee80211_hw *hw = rtlworks->hw;
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
        enum rf_pwrstate rtstate;

        if (mac->opmode != NL80211_IFTYPE_STATION) {
                rtl_dbg(rtlpriv, COMP_ERR, DBG_WARNING,
                        "not station return\n");
                return;
        }

        if (mac->p2p_in_use)
                return;

        if (mac->link_state > MAC80211_NOLINK)
                return;

        if (is_hal_stop(rtlhal))
                return;

        if (rtlpriv->sec.being_setkey)
                return;

        if (rtlpriv->cfg->ops->bt_coex_off_before_lps)
                rtlpriv->cfg->ops->bt_coex_off_before_lps(hw);

        if (ppsc->inactiveps) {
                rtstate = ppsc->rfpwr_state;

                /*
                 *Do not enter IPS in the following conditions:
                 *(1) RF is already OFF or Sleep
                 *(2) swrf_processing (indicates the IPS is still under going)
                 *(3) Connectted (only disconnected can trigger IPS)
                 *(4) IBSS (send Beacon)
                 *(5) AP mode (send Beacon)
                 *(6) monitor mode (rcv packet)
                 */

                if (rtstate == ERFON &&
                    !ppsc->swrf_processing &&
                    (mac->link_state == MAC80211_NOLINK) &&
                    !mac->act_scanning) {
                        rtl_dbg(rtlpriv, COMP_RF, DBG_TRACE,
                                "IPSEnter(): Turn off RF\n");

                        ppsc->inactive_pwrstate = ERFOFF;
                        ppsc->in_powersavemode = true;

                        /* call before RF off */
                        if (rtlpriv->cfg->ops->get_btc_status())
                                rtlpriv->btcoexist.btc_ops->btc_ips_notify(rtlpriv,
                                                                        ppsc->inactive_pwrstate);

                        /*rtl_pci_reset_trx_ring(hw); */
                        _rtl_ps_inactive_ps(hw);
                }
        }
}

void rtl_ips_nic_off(struct ieee80211_hw *hw)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        /* because when link with ap, mac80211 will ask us
         * to disable nic quickly after scan before linking,
         * this will cause link failed, so we delay 100ms here
         */
        queue_delayed_work(rtlpriv->works.rtl_wq,
                           &rtlpriv->works.ips_nic_off_wq, MSECS(100));
}

/* NOTICE: any opmode should exc nic_on, or disable without
 * nic_on may something wrong, like adhoc TP
 */
void rtl_ips_nic_on(struct ieee80211_hw *hw)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
        enum rf_pwrstate rtstate;

        cancel_delayed_work_sync(&rtlpriv->works.ips_nic_off_wq);

        mutex_lock(&rtlpriv->locks.ips_mutex);
        if (ppsc->inactiveps) {
                rtstate = ppsc->rfpwr_state;

                if (rtstate != ERFON &&
                    !ppsc->swrf_processing &&
                    ppsc->rfoff_reason <= RF_CHANGE_BY_IPS) {

                        ppsc->inactive_pwrstate = ERFON;
                        ppsc->in_powersavemode = false;
                        _rtl_ps_inactive_ps(hw);
                        /* call after RF on */
                        if (rtlpriv->cfg->ops->get_btc_status())
                                rtlpriv->btcoexist.btc_ops->btc_ips_notify(rtlpriv,
                                                                        ppsc->inactive_pwrstate);
                }
        }
        mutex_unlock(&rtlpriv->locks.ips_mutex);
}
EXPORT_SYMBOL_GPL(rtl_ips_nic_on);

/*for FW LPS*/

/*
 *Determine if we can set Fw into PS mode
 *in current condition.Return TRUE if it
 *can enter PS mode.
 */
static bool rtl_get_fwlps_doze(struct ieee80211_hw *hw)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
        u32 ps_timediff;

        ps_timediff = jiffies_to_msecs(jiffies -
                                       ppsc->last_delaylps_stamp_jiffies);

        if (ps_timediff < 2000) {
                rtl_dbg(rtlpriv, COMP_POWER, DBG_LOUD,
                        "Delay enter Fw LPS for DHCP, ARP, or EAPOL exchanging state\n");
                return false;
        }

        if (mac->link_state != MAC80211_LINKED)
                return false;

        if (mac->opmode == NL80211_IFTYPE_ADHOC)
                return false;

        return true;
}

/* Change current and default preamble mode.*/
void rtl_lps_set_psmode(struct ieee80211_hw *hw, u8 rt_psmode)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
        bool enter_fwlps;

        if (mac->opmode == NL80211_IFTYPE_ADHOC)
                return;

        if (mac->link_state != MAC80211_LINKED)
                return;

        if (ppsc->dot11_psmode == rt_psmode && rt_psmode == EACTIVE)
                return;

        /* Update power save mode configured. */
        ppsc->dot11_psmode = rt_psmode;

        /*
         *<FW control LPS>
         *1. Enter PS mode
         *   Set RPWM to Fw to turn RF off and send H2C fw_pwrmode
         *   cmd to set Fw into PS mode.
         *2. Leave PS mode
         *   Send H2C fw_pwrmode cmd to Fw to set Fw into Active
         *   mode and set RPWM to turn RF on.
         */

        if ((ppsc->fwctrl_lps) && ppsc->report_linked) {
                if (ppsc->dot11_psmode == EACTIVE) {
                        rtl_dbg(rtlpriv, COMP_RF, DBG_DMESG,
                                "FW LPS leave ps_mode:%x\n",
                                FW_PS_ACTIVE_MODE);
                        enter_fwlps = false;
                        ppsc->pwr_mode = FW_PS_ACTIVE_MODE;
                        ppsc->smart_ps = 0;
                        rtlpriv->cfg->ops->set_hw_reg(hw, HW_VAR_FW_LPS_ACTION,
                                                      (u8 *)(&enter_fwlps));
                        if (ppsc->p2p_ps_info.opp_ps)
                                rtl_p2p_ps_cmd(hw , P2P_PS_ENABLE);

                        if (rtlpriv->cfg->ops->get_btc_status())
                                rtlpriv->btcoexist.btc_ops->btc_lps_notify(rtlpriv, rt_psmode);
                } else {
                        if (rtl_get_fwlps_doze(hw)) {
                                rtl_dbg(rtlpriv, COMP_RF, DBG_DMESG,
                                        "FW LPS enter ps_mode:%x\n",
                                        ppsc->fwctrl_psmode);
                                if (rtlpriv->cfg->ops->get_btc_status())
                                        rtlpriv->btcoexist.btc_ops->btc_lps_notify(rtlpriv, rt_psmode);
                                enter_fwlps = true;
                                ppsc->pwr_mode = ppsc->fwctrl_psmode;
                                ppsc->smart_ps = 2;
                                rtlpriv->cfg->ops->set_hw_reg(hw,
                                                        HW_VAR_FW_LPS_ACTION,
                                                        (u8 *)(&enter_fwlps));

                        } else {
                                /* Reset the power save related parameters. */
                                ppsc->dot11_psmode = EACTIVE;
                        }
                }
        }
}

/* Interrupt safe routine to enter the leisure power save mode.*/
static void rtl_lps_enter_core(struct ieee80211_hw *hw)
{
        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        if (!ppsc->fwctrl_lps)
                return;

        if (rtlpriv->sec.being_setkey)
                return;

        if (rtlpriv->link_info.busytraffic)
                return;

        /*sleep after linked 10s, to let DHCP and 4-way handshake ok enough!! */
        if (mac->cnt_after_linked < 5)
                return;

        if (mac->opmode == NL80211_IFTYPE_ADHOC)
                return;

        if (mac->link_state != MAC80211_LINKED)
                return;

        mutex_lock(&rtlpriv->locks.lps_mutex);

        /* Don't need to check (ppsc->dot11_psmode == EACTIVE), because
         * bt_ccoexist may ask to enter lps.
         * In normal case, this constraint move to rtl_lps_set_psmode().
         */
        rtl_dbg(rtlpriv, COMP_POWER, DBG_LOUD,
                "Enter 802.11 power save mode...\n");
        rtl_lps_set_psmode(hw, EAUTOPS);

        mutex_unlock(&rtlpriv->locks.lps_mutex);
}

/* Interrupt safe routine to leave the leisure power save mode.*/
static void rtl_lps_leave_core(struct ieee80211_hw *hw)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
        struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));

        mutex_lock(&rtlpriv->locks.lps_mutex);

        if (ppsc->fwctrl_lps) {
                if (ppsc->dot11_psmode != EACTIVE) {

                        /*FIX ME */
                        /*rtlpriv->cfg->ops->enable_interrupt(hw); */

                        if (ppsc->reg_rfps_level & RT_RF_LPS_LEVEL_ASPM &&
                            RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM) &&
                            rtlhal->interface == INTF_PCI) {
                                rtlpriv->intf_ops->disable_aspm(hw);
                                RT_CLEAR_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
                        }

                        rtl_dbg(rtlpriv, COMP_POWER, DBG_LOUD,
                                "Busy Traffic,Leave 802.11 power save..\n");

                        rtl_lps_set_psmode(hw, EACTIVE);
                }
        }
        mutex_unlock(&rtlpriv->locks.lps_mutex);
}

/* For sw LPS*/
void rtl_swlps_beacon(struct ieee80211_hw *hw, void *data, unsigned int len)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
        struct ieee80211_hdr *hdr = data;
        struct ieee80211_tim_ie *tim_ie;
        u8 *tim;
        u8 tim_len;
        bool u_buffed;
        bool m_buffed;

        if (mac->opmode != NL80211_IFTYPE_STATION)
                return;

        if (!rtlpriv->psc.swctrl_lps)
                return;

        if (rtlpriv->mac80211.link_state != MAC80211_LINKED)
                return;

        if (!rtlpriv->psc.sw_ps_enabled)
                return;

        if (rtlpriv->psc.fwctrl_lps)
                return;

        if (likely(!(hw->conf.flags & IEEE80211_CONF_PS)))
                return;

        /* check if this really is a beacon */
        if (!ieee80211_is_beacon(hdr->frame_control))
                return;

        /* min. beacon length + FCS_LEN */
        if (len <= 40 + FCS_LEN)
                return;

        /* and only beacons from the associated BSSID, please */
        if (!ether_addr_equal_64bits(hdr->addr3, rtlpriv->mac80211.bssid))
                return;

        rtlpriv->psc.last_beacon = jiffies;

        tim = rtl_find_ie(data, len - FCS_LEN, WLAN_EID_TIM);
        if (!tim)
                return;

        if (tim[1] < sizeof(*tim_ie))
                return;

        tim_len = tim[1];
        tim_ie = (struct ieee80211_tim_ie *) &tim[2];

        if (!WARN_ON_ONCE(!hw->conf.ps_dtim_period))
                rtlpriv->psc.dtim_counter = tim_ie->dtim_count;

        /* Check whenever the PHY can be turned off again. */

        /* 1. What about buffered unicast traffic for our AID? */
        u_buffed = ieee80211_check_tim(tim_ie, tim_len,
                                       rtlpriv->mac80211.assoc_id, false);

        /* 2. Maybe the AP wants to send multicast/broadcast data? */
        m_buffed = tim_ie->bitmap_ctrl & 0x01;
        rtlpriv->psc.multi_buffered = m_buffed;

        /* unicast will process by mac80211 through
         * set ~IEEE80211_CONF_PS, So we just check
         * multicast frames here */
        if (!m_buffed) {
                /* back to low-power land. and delay is
                 * prevent null power save frame tx fail */
                queue_delayed_work(rtlpriv->works.rtl_wq,
                                   &rtlpriv->works.ps_work, MSECS(5));
        } else {
                rtl_dbg(rtlpriv, COMP_POWER, DBG_DMESG,
                        "u_bufferd: %x, m_buffered: %x\n", u_buffed, m_buffed);
        }
}
EXPORT_SYMBOL_GPL(rtl_swlps_beacon);

void rtl_swlps_rf_awake(struct ieee80211_hw *hw)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));

        if (!rtlpriv->psc.swctrl_lps)
                return;
        if (mac->link_state != MAC80211_LINKED)
                return;

        if (ppsc->reg_rfps_level & RT_RF_LPS_LEVEL_ASPM &&
            RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM)) {
                rtlpriv->intf_ops->disable_aspm(hw);
                RT_CLEAR_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
        }

        mutex_lock(&rtlpriv->locks.lps_mutex);
        rtl_ps_set_rf_state(hw, ERFON, RF_CHANGE_BY_PS);
        mutex_unlock(&rtlpriv->locks.lps_mutex);
}

void rtl_swlps_rfon_wq_callback(struct work_struct *work)
{
        struct rtl_works *rtlworks = container_of(work, struct rtl_works,
                                                  ps_rfon_wq.work);
        struct ieee80211_hw *hw = rtlworks->hw;

        rtl_swlps_rf_awake(hw);
}

void rtl_swlps_rf_sleep(struct ieee80211_hw *hw)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
        u8 sleep_intv;

        if (!rtlpriv->psc.sw_ps_enabled)
                return;

        if ((rtlpriv->sec.being_setkey) ||
            (mac->opmode == NL80211_IFTYPE_ADHOC))
                return;

        /*sleep after linked 10s, to let DHCP and 4-way handshake ok enough!! */
        if ((mac->link_state != MAC80211_LINKED) || (mac->cnt_after_linked < 5))
                return;

        if (rtlpriv->link_info.busytraffic)
                return;

        spin_lock(&rtlpriv->locks.rf_ps_lock);
        if (rtlpriv->psc.rfchange_inprogress) {
                spin_unlock(&rtlpriv->locks.rf_ps_lock);
                return;
        }
        spin_unlock(&rtlpriv->locks.rf_ps_lock);

        mutex_lock(&rtlpriv->locks.lps_mutex);
        rtl_ps_set_rf_state(hw, ERFSLEEP, RF_CHANGE_BY_PS);
        mutex_unlock(&rtlpriv->locks.lps_mutex);

        if (ppsc->reg_rfps_level & RT_RF_OFF_LEVL_ASPM &&
            !RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM)) {
                rtlpriv->intf_ops->enable_aspm(hw);
                RT_SET_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
        }

        /* here is power save alg, when this beacon is DTIM
         * we will set sleep time to dtim_period * n;
         * when this beacon is not DTIM, we will set sleep
         * time to sleep_intv = rtlpriv->psc.dtim_counter or
         * MAX_SW_LPS_SLEEP_INTV(default set to 5) */

        if (rtlpriv->psc.dtim_counter == 0) {
                if (hw->conf.ps_dtim_period == 1)
                        sleep_intv = hw->conf.ps_dtim_period * 2;
                else
                        sleep_intv = hw->conf.ps_dtim_period;
        } else {
                sleep_intv = rtlpriv->psc.dtim_counter;
        }

        if (sleep_intv > MAX_SW_LPS_SLEEP_INTV)
                sleep_intv = MAX_SW_LPS_SLEEP_INTV;

        /* this print should always be dtim_conter = 0 &
         * sleep  = dtim_period, that meaons, we should
         * awake before every dtim */
        rtl_dbg(rtlpriv, COMP_POWER, DBG_DMESG,
                "dtim_counter:%x will sleep :%d beacon_intv\n",
                rtlpriv->psc.dtim_counter, sleep_intv);

        /* we tested that 40ms is enough for sw & hw sw delay */
        queue_delayed_work(rtlpriv->works.rtl_wq, &rtlpriv->works.ps_rfon_wq,
                        MSECS(sleep_intv * mac->vif->bss_conf.beacon_int - 40));
}

void rtl_lps_change_work_callback(struct work_struct *work)
{
        struct rtl_works *rtlworks =
            container_of(work, struct rtl_works, lps_change_work);
        struct ieee80211_hw *hw = rtlworks->hw;
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        if (rtlpriv->enter_ps)
                rtl_lps_enter_core(hw);
        else
                rtl_lps_leave_core(hw);
}
EXPORT_SYMBOL_GPL(rtl_lps_change_work_callback);

void rtl_lps_enter(struct ieee80211_hw *hw, bool may_block)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        if (may_block)
                return rtl_lps_enter_core(hw);
        rtlpriv->enter_ps = true;
        schedule_work(&rtlpriv->works.lps_change_work);
}
EXPORT_SYMBOL_GPL(rtl_lps_enter);

void rtl_lps_leave(struct ieee80211_hw *hw, bool may_block)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        if (may_block)
                return rtl_lps_leave_core(hw);
        rtlpriv->enter_ps = false;
        schedule_work(&rtlpriv->works.lps_change_work);
}
EXPORT_SYMBOL_GPL(rtl_lps_leave);

void rtl_swlps_wq_callback(struct work_struct *work)
{
        struct rtl_works *rtlworks = container_of(work, struct rtl_works,
                                                  ps_work.work);
        struct ieee80211_hw *hw = rtlworks->hw;
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        /* we can sleep after ps null send ok */
        if (rtlpriv->psc.state_inap)
                rtl_swlps_rf_sleep(hw);
}

static void rtl_p2p_noa_ie(struct ieee80211_hw *hw, void *data,
                           unsigned int len)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct ieee80211_mgmt *mgmt = data;
        struct rtl_p2p_ps_info *p2pinfo = &(rtlpriv->psc.p2p_ps_info);
        u8 *pos, *end, *ie;
        u16 noa_len;
        static u8 p2p_oui_ie_type[4] = {0x50, 0x6f, 0x9a, 0x09};
        u8 noa_num, index , i, noa_index = 0;
        bool find_p2p_ie = false , find_p2p_ps_ie = false;

        pos = (u8 *)mgmt->u.beacon.variable;
        end = data + len;
        ie = NULL;

        while (pos + 1 < end) {
                if (pos + 2 + pos[1] > end)
                        return;

                if (pos[0] == 221 && pos[1] > 4) {
                        if (memcmp(&pos[2], p2p_oui_ie_type, 4) == 0) {
                                ie = pos + 2+4;
                                break;
                        }
                }
                pos += 2 + pos[1];
        }

        if (ie == NULL)
                return;
        find_p2p_ie = true;
        /*to find noa ie*/
        while (ie + 1 < end) {
                noa_len = le16_to_cpu(*((__le16 *)&ie[1]));
                if (ie + 3 + ie[1] > end)
                        return;

                if (ie[0] == 12) {
                        find_p2p_ps_ie = true;
                        if ((noa_len - 2) % 13 != 0) {
                                rtl_dbg(rtlpriv, COMP_INIT, DBG_LOUD,
                                        "P2P notice of absence: invalid length.%d\n",
                                        noa_len);
                                return;
                        } else {
                                noa_num = (noa_len - 2) / 13;
                                if (noa_num > P2P_MAX_NOA_NUM)
                                        noa_num = P2P_MAX_NOA_NUM;

                        }
                        noa_index = ie[3];
                        if (rtlpriv->psc.p2p_ps_info.p2p_ps_mode ==
                            P2P_PS_NONE || noa_index != p2pinfo->noa_index) {
                                rtl_dbg(rtlpriv, COMP_FW, DBG_LOUD,
                                        "update NOA ie.\n");
                                p2pinfo->noa_index = noa_index;
                                p2pinfo->opp_ps = (ie[4] >> 7);
                                p2pinfo->ctwindow = ie[4] & 0x7F;
                                p2pinfo->noa_num = noa_num;
                                index = 5;
                                for (i = 0; i < noa_num; i++) {
                                        p2pinfo->noa_count_type[i] =
                                         *(u8 *)(ie + index);
                                        index += 1;
                                        p2pinfo->noa_duration[i] =
                                         le32_to_cpu(*(__le32 *)(ie + index));
                                        index += 4;
                                        p2pinfo->noa_interval[i] =
                                         le32_to_cpu(*(__le32 *)(ie + index));
                                        index += 4;
                                        p2pinfo->noa_start_time[i] =
                                         le32_to_cpu(*(__le32 *)(ie + index));
                                        index += 4;
                                }

                                if (p2pinfo->opp_ps == 1) {
                                        p2pinfo->p2p_ps_mode = P2P_PS_CTWINDOW;
                                        /* Driver should wait LPS entering
                                         * CTWindow
                                         */
                                        if (rtlpriv->psc.fw_current_inpsmode)
                                                rtl_p2p_ps_cmd(hw,
                                                               P2P_PS_ENABLE);
                                } else if (p2pinfo->noa_num > 0) {
                                        p2pinfo->p2p_ps_mode = P2P_PS_NOA;
                                        rtl_p2p_ps_cmd(hw, P2P_PS_ENABLE);
                                } else if (p2pinfo->p2p_ps_mode > P2P_PS_NONE) {
                                        rtl_p2p_ps_cmd(hw, P2P_PS_DISABLE);
                                }
                        }
                        break;
                }
                ie += 3 + noa_len;
        }

        if (find_p2p_ie) {
                if ((p2pinfo->p2p_ps_mode > P2P_PS_NONE) &&
                    (!find_p2p_ps_ie))
                        rtl_p2p_ps_cmd(hw, P2P_PS_DISABLE);
        }
}

static void rtl_p2p_action_ie(struct ieee80211_hw *hw, void *data,
                              unsigned int len)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct ieee80211_mgmt *mgmt = data;
        struct rtl_p2p_ps_info *p2pinfo = &(rtlpriv->psc.p2p_ps_info);
        u8 noa_num, index , i , noa_index = 0;
        u8 *pos, *end, *ie;
        u16 noa_len;
        static u8 p2p_oui_ie_type[4] = {0x50, 0x6f, 0x9a, 0x09};

        pos = (u8 *)&mgmt->u.action.category;
        end = data + len;
        ie = NULL;

        if (pos[0] == 0x7f) {
                if (memcmp(&pos[1], p2p_oui_ie_type, 4) == 0)
                        ie = pos + 3+4;
        }

        if (ie == NULL)
                return;

        rtl_dbg(rtlpriv, COMP_FW, DBG_LOUD, "action frame find P2P IE.\n");
        /*to find noa ie*/
        while (ie + 1 < end) {
                noa_len = le16_to_cpu(*(__le16 *)&ie[1]);
                if (ie + 3 + ie[1] > end)
                        return;

                if (ie[0] == 12) {
                        rtl_dbg(rtlpriv, COMP_FW, DBG_LOUD, "find NOA IE.\n");
                        RT_PRINT_DATA(rtlpriv, COMP_FW, DBG_LOUD, "noa ie ",
                                      ie, noa_len);
                        if ((noa_len - 2) % 13 != 0) {
                                rtl_dbg(rtlpriv, COMP_FW, DBG_LOUD,
                                        "P2P notice of absence: invalid length.%d\n",
                                        noa_len);
                                return;
                        } else {
                                noa_num = (noa_len - 2) / 13;
                                if (noa_num > P2P_MAX_NOA_NUM)
                                        noa_num = P2P_MAX_NOA_NUM;

                        }
                        noa_index = ie[3];
                        if (rtlpriv->psc.p2p_ps_info.p2p_ps_mode ==
                            P2P_PS_NONE || noa_index != p2pinfo->noa_index) {
                                p2pinfo->noa_index = noa_index;
                                p2pinfo->opp_ps = (ie[4] >> 7);
                                p2pinfo->ctwindow = ie[4] & 0x7F;
                                p2pinfo->noa_num = noa_num;
                                index = 5;
                                for (i = 0; i < noa_num; i++) {
                                        p2pinfo->noa_count_type[i] =
                                         *(u8 *)(ie + index);
                                        index += 1;
                                        p2pinfo->noa_duration[i] =
                                         le32_to_cpu(*(__le32 *)(ie + index));
                                        index += 4;
                                        p2pinfo->noa_interval[i] =
                                         le32_to_cpu(*(__le32 *)(ie + index));
                                        index += 4;
                                        p2pinfo->noa_start_time[i] =
                                         le32_to_cpu(*(__le32 *)(ie + index));
                                        index += 4;
                                }

                                if (p2pinfo->opp_ps == 1) {
                                        p2pinfo->p2p_ps_mode = P2P_PS_CTWINDOW;
                                        /* Driver should wait LPS entering
                                         * CTWindow
                                         */
                                        if (rtlpriv->psc.fw_current_inpsmode)
                                                rtl_p2p_ps_cmd(hw,
                                                               P2P_PS_ENABLE);
                                } else if (p2pinfo->noa_num > 0) {
                                        p2pinfo->p2p_ps_mode = P2P_PS_NOA;
                                        rtl_p2p_ps_cmd(hw, P2P_PS_ENABLE);
                                } else if (p2pinfo->p2p_ps_mode > P2P_PS_NONE) {
                                        rtl_p2p_ps_cmd(hw, P2P_PS_DISABLE);
                                }
                        }
                        break;
                }
                ie += 3 + noa_len;
        }
}

void rtl_p2p_ps_cmd(struct ieee80211_hw *hw , u8 p2p_ps_state)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_ps_ctl *rtlps = rtl_psc(rtl_priv(hw));
        struct rtl_p2p_ps_info  *p2pinfo = &(rtlpriv->psc.p2p_ps_info);

        rtl_dbg(rtlpriv, COMP_FW, DBG_LOUD, " p2p state %x\n", p2p_ps_state);
        switch (p2p_ps_state) {
        case P2P_PS_DISABLE:
                p2pinfo->p2p_ps_state = p2p_ps_state;
                rtlpriv->cfg->ops->set_hw_reg(hw, HW_VAR_H2C_FW_P2P_PS_OFFLOAD,
                                              &p2p_ps_state);
                p2pinfo->noa_index = 0;
                p2pinfo->ctwindow = 0;
                p2pinfo->opp_ps = 0;
                p2pinfo->noa_num = 0;
                p2pinfo->p2p_ps_mode = P2P_PS_NONE;
                if (rtlps->fw_current_inpsmode) {
                        if (rtlps->smart_ps == 0) {
                                rtlps->smart_ps = 2;
                                rtlpriv->cfg->ops->set_hw_reg(hw,
                                         HW_VAR_H2C_FW_PWRMODE,
                                         &rtlps->pwr_mode);
                        }

                }
                break;
        case P2P_PS_ENABLE:
                if (p2pinfo->p2p_ps_mode > P2P_PS_NONE) {
                        p2pinfo->p2p_ps_state = p2p_ps_state;

                        if (p2pinfo->ctwindow > 0) {
                                if (rtlps->smart_ps != 0) {
                                        rtlps->smart_ps = 0;
                                        rtlpriv->cfg->ops->set_hw_reg(hw,
                                                 HW_VAR_H2C_FW_PWRMODE,
                                                 &rtlps->pwr_mode);
                                }
                        }
                        rtlpriv->cfg->ops->set_hw_reg(hw,
                                 HW_VAR_H2C_FW_P2P_PS_OFFLOAD,
                                 &p2p_ps_state);

                }
                break;
        case P2P_PS_SCAN:
        case P2P_PS_SCAN_DONE:
        case P2P_PS_ALLSTASLEEP:
                if (p2pinfo->p2p_ps_mode > P2P_PS_NONE) {
                        p2pinfo->p2p_ps_state = p2p_ps_state;
                        rtlpriv->cfg->ops->set_hw_reg(hw,
                                 HW_VAR_H2C_FW_P2P_PS_OFFLOAD,
                                 &p2p_ps_state);
                }
                break;
        default:
                break;
        }
        rtl_dbg(rtlpriv, COMP_FW, DBG_LOUD,
                "ctwindow %x oppps %x\n",
                p2pinfo->ctwindow, p2pinfo->opp_ps);
        rtl_dbg(rtlpriv, COMP_FW, DBG_LOUD,
                "count %x duration %x index %x interval %x start time %x noa num %x\n",
                p2pinfo->noa_count_type[0],
                p2pinfo->noa_duration[0],
                p2pinfo->noa_index,
                p2pinfo->noa_interval[0],
                p2pinfo->noa_start_time[0],
                p2pinfo->noa_num);
        rtl_dbg(rtlpriv, COMP_FW, DBG_LOUD, "end\n");
}

void rtl_p2p_info(struct ieee80211_hw *hw, void *data, unsigned int len)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
        struct ieee80211_hdr *hdr = data;

        if (!mac->p2p)
                return;
        if (mac->link_state != MAC80211_LINKED)
                return;
        /* min. beacon length + FCS_LEN */
        if (len <= 40 + FCS_LEN)
                return;

        /* and only beacons from the associated BSSID, please */
        if (!ether_addr_equal_64bits(hdr->addr3, rtlpriv->mac80211.bssid))
                return;

        /* check if this really is a beacon */
        if (!(ieee80211_is_beacon(hdr->frame_control) ||
              ieee80211_is_probe_resp(hdr->frame_control) ||
              ieee80211_is_action(hdr->frame_control)))
                return;

        if (ieee80211_is_action(hdr->frame_control))
                rtl_p2p_action_ie(hw , data , len - FCS_LEN);
        else
                rtl_p2p_noa_ie(hw , data , len - FCS_LEN);
}
EXPORT_SYMBOL_GPL(rtl_p2p_info);