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

#include "../wifi.h"
#include "../pci.h"
#include "../base.h"
#include "../stats.h"
#include "reg.h"
#include "def.h"
#include "phy.h"
#include "trx.h"
#include "led.h"

static u8 _rtl8723e_map_hwqueue_to_fwqueue(struct sk_buff *skb, u8 hw_queue)
{
        __le16 fc = rtl_get_fc(skb);

        if (unlikely(ieee80211_is_beacon(fc)))
                return QSLT_BEACON;
        if (ieee80211_is_mgmt(fc) || ieee80211_is_ctl(fc))
                return QSLT_MGNT;

        return skb->priority;
}

static void _rtl8723e_query_rxphystatus(struct ieee80211_hw *hw,
                                        struct rtl_stats *pstatus, u8 *pdesc,
                                        struct rx_fwinfo_8723e *p_drvinfo,
                                        bool bpacket_match_bssid,
                                        bool bpacket_toself, bool packet_beacon)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_ps_ctl *ppsc = rtl_psc(rtlpriv);
        struct phy_sts_cck_8723e_t *cck_buf;
        s8 rx_pwr_all = 0, rx_pwr[4];
        u8 rf_rx_num = 0, evm, pwdb_all;
        u8 i, max_spatial_stream;
        u32 rssi, total_rssi = 0;
        bool is_cck = pstatus->is_cck;

        /* Record it for next packet processing */
        pstatus->packet_matchbssid = bpacket_match_bssid;
        pstatus->packet_toself = bpacket_toself;
        pstatus->packet_beacon = packet_beacon;
        pstatus->rx_mimo_signalquality[0] = -1;
        pstatus->rx_mimo_signalquality[1] = -1;

        if (is_cck) {
                u8 report, cck_highpwr;

                /* CCK Driver info Structure is not the same as OFDM packet. */
                cck_buf = (struct phy_sts_cck_8723e_t *)p_drvinfo;

                /* (1)Hardware does not provide RSSI for CCK */
                /* (2)PWDB, Average PWDB calculated by
                 * hardware (for rate adaptive)
                 */
                if (ppsc->rfpwr_state == ERFON)
                        cck_highpwr = (u8)rtl_get_bbreg(hw,
                                        RFPGA0_XA_HSSIPARAMETER2,
                                        BIT(9));
                else
                        cck_highpwr = false;

                if (!cck_highpwr) {
                        u8 cck_agc_rpt = cck_buf->cck_agc_rpt;
                        report = cck_buf->cck_agc_rpt & 0xc0;
                        report = report >> 6;
                        switch (report) {
                        case 0x3:
                                rx_pwr_all = -46 - (cck_agc_rpt & 0x3e);
                                break;
                        case 0x2:
                                rx_pwr_all = -26 - (cck_agc_rpt & 0x3e);
                                break;
                        case 0x1:
                                rx_pwr_all = -12 - (cck_agc_rpt & 0x3e);
                                break;
                        case 0x0:
                                rx_pwr_all = 16 - (cck_agc_rpt & 0x3e);
                                break;
                        }
                } else {
                        u8 cck_agc_rpt = cck_buf->cck_agc_rpt;
                        report = p_drvinfo->cfosho[0] & 0x60;
                        report = report >> 5;
                        switch (report) {
                        case 0x3:
                                rx_pwr_all = -46 - ((cck_agc_rpt & 0x1f) << 1);
                                break;
                        case 0x2:
                                rx_pwr_all = -26 - ((cck_agc_rpt & 0x1f) << 1);
                                break;
                        case 0x1:
                                rx_pwr_all = -12 - ((cck_agc_rpt & 0x1f) << 1);
                                break;
                        case 0x0:
                                rx_pwr_all = 16 - ((cck_agc_rpt & 0x1f) << 1);
                                break;
                        }
                }

                pwdb_all = rtl_query_rxpwrpercentage(rx_pwr_all);
                /* CCK gain is smaller than OFDM/MCS gain,  */
                /* so we add gain diff by experiences,
                 * the val is 6
                 */
                pwdb_all += 6;
                if (pwdb_all > 100)
                        pwdb_all = 100;
                /* modify the offset to make the same
                 * gain index with OFDM.
                 */
                if (pwdb_all > 34 && pwdb_all <= 42)
                        pwdb_all -= 2;
                else if (pwdb_all > 26 && pwdb_all <= 34)
                        pwdb_all -= 6;
                else if (pwdb_all > 14 && pwdb_all <= 26)
                        pwdb_all -= 8;
                else if (pwdb_all > 4 && pwdb_all <= 14)
                        pwdb_all -= 4;

                pstatus->rx_pwdb_all = pwdb_all;
                pstatus->recvsignalpower = rx_pwr_all;

                /* (3) Get Signal Quality (EVM) */
                if (bpacket_match_bssid) {
                        u8 sq;

                        if (pstatus->rx_pwdb_all > 40)
                                sq = 100;
                        else {
                                sq = cck_buf->sq_rpt;
                                if (sq > 64)
                                        sq = 0;
                                else if (sq < 20)
                                        sq = 100;
                                else
                                        sq = ((64 - sq) * 100) / 44;
                        }

                        pstatus->signalquality = sq;
                        pstatus->rx_mimo_signalquality[0] = sq;
                        pstatus->rx_mimo_signalquality[1] = -1;
                }
        } else {
                rtlpriv->dm.rfpath_rxenable[0] =
                    rtlpriv->dm.rfpath_rxenable[1] = true;

                /* (1)Get RSSI for HT rate */
                for (i = RF90_PATH_A; i < RF6052_MAX_PATH; i++) {

                        /* we will judge RF RX path now. */
                        if (rtlpriv->dm.rfpath_rxenable[i])
                                rf_rx_num++;

                        rx_pwr[i] = ((p_drvinfo->gain_trsw[i] &
                                      0x3f) * 2) - 110;

                        /* Translate DBM to percentage. */
                        rssi = rtl_query_rxpwrpercentage(rx_pwr[i]);
                        total_rssi += rssi;

                        /* Get Rx snr value in DB */
                        rtlpriv->stats.rx_snr_db[i] =
                                (long)(p_drvinfo->rxsnr[i] / 2);

                        /* Record Signal Strength for next packet */
                        if (bpacket_match_bssid)
                                pstatus->rx_mimo_signalstrength[i] = (u8)rssi;
                }

                /* (2)PWDB, Average PWDB calculated by
                 * hardware (for rate adaptive)
                 */
                rx_pwr_all = ((p_drvinfo->pwdb_all >> 1) & 0x7f) - 110;

                pwdb_all = rtl_query_rxpwrpercentage(rx_pwr_all);
                pstatus->rx_pwdb_all = pwdb_all;
                pstatus->rxpower = rx_pwr_all;
                pstatus->recvsignalpower = rx_pwr_all;

                /* (3)EVM of HT rate */
                if (pstatus->is_ht && pstatus->rate >= DESC92C_RATEMCS8 &&
                    pstatus->rate <= DESC92C_RATEMCS15)
                        max_spatial_stream = 2;
                else
                        max_spatial_stream = 1;

                for (i = 0; i < max_spatial_stream; i++) {
                        evm = rtl_evm_db_to_percentage(p_drvinfo->rxevm[i]);

                        if (bpacket_match_bssid) {
                                /* Fill value in RFD, Get the first
                                 * spatial stream only
                                 */
                                if (i == 0)
                                        pstatus->signalquality =
                                                (u8)(evm & 0xff);
                                pstatus->rx_mimo_signalquality[i] =
                                        (u8)(evm & 0xff);
                        }
                }
        }

        /* UI BSS List signal strength(in percentage),
         * make it good looking, from 0~100.
         */
        if (is_cck)
                pstatus->signalstrength = (u8)(rtl_signal_scale_mapping(hw,
                        pwdb_all));
        else if (rf_rx_num != 0)
                pstatus->signalstrength = (u8)(rtl_signal_scale_mapping(hw,
                        total_rssi /= rf_rx_num));
}

static void translate_rx_signal_stuff(struct ieee80211_hw *hw,
                                      struct sk_buff *skb,
                                      struct rtl_stats *pstatus, u8 *pdesc,
                                      struct rx_fwinfo_8723e *p_drvinfo)
{
        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
        struct rtl_efuse *rtlefuse = rtl_efuse(rtl_priv(hw));
        struct ieee80211_hdr *hdr;
        u8 *tmp_buf;
        u8 *praddr;
        /*u8 *psaddr;*/
        u16 fc, type;
        bool packet_matchbssid, packet_toself, packet_beacon;

        tmp_buf = skb->data + pstatus->rx_drvinfo_size + pstatus->rx_bufshift;

        hdr = (struct ieee80211_hdr *)tmp_buf;
        fc = le16_to_cpu(hdr->frame_control);
        type = WLAN_FC_GET_TYPE(hdr->frame_control);
        praddr = hdr->addr1;

        packet_matchbssid = ((IEEE80211_FTYPE_CTL != type) &&
                (ether_addr_equal(mac->bssid, (fc & IEEE80211_FCTL_TODS) ?
                 hdr->addr1 : (fc & IEEE80211_FCTL_FROMDS) ?
                 hdr->addr2 : hdr->addr3)) &&
                 (!pstatus->hwerror) &&
                 (!pstatus->crc) && (!pstatus->icv));

        packet_toself = packet_matchbssid &&
            (ether_addr_equal(praddr, rtlefuse->dev_addr));

        if (ieee80211_is_beacon(hdr->frame_control))
                packet_beacon = true;
        else
                packet_beacon = false;

        _rtl8723e_query_rxphystatus(hw, pstatus, pdesc, p_drvinfo,
                                    packet_matchbssid, packet_toself,
                                    packet_beacon);

        rtl_process_phyinfo(hw, tmp_buf, pstatus);
}

bool rtl8723e_rx_query_desc(struct ieee80211_hw *hw,
                            struct rtl_stats *status,
                            struct ieee80211_rx_status *rx_status,
                            u8 *pdesc8, struct sk_buff *skb)
{
        struct rx_fwinfo_8723e *p_drvinfo;
        struct ieee80211_hdr *hdr;
        __le32 *pdesc = (__le32 *)pdesc8;
        u32 phystatus = get_rx_desc_physt(pdesc);

        status->length = (u16)get_rx_desc_pkt_len(pdesc);
        status->rx_drvinfo_size = (u8)get_rx_desc_drv_info_size(pdesc) *
            RX_DRV_INFO_SIZE_UNIT;
        status->rx_bufshift = (u8)(get_rx_desc_shift(pdesc) & 0x03);
        status->icv = (u16)get_rx_desc_icv(pdesc);
        status->crc = (u16)get_rx_desc_crc32(pdesc);
        status->hwerror = (status->crc | status->icv);
        status->decrypted = !get_rx_desc_swdec(pdesc);
        status->rate = (u8)get_rx_desc_rxmcs(pdesc);
        status->shortpreamble = (u16)get_rx_desc_splcp(pdesc);
        status->isampdu = (bool)(get_rx_desc_paggr(pdesc) == 1);
        status->isfirst_ampdu = (bool)((get_rx_desc_paggr(pdesc) == 1) &&
                                       (get_rx_desc_faggr(pdesc) == 1));
        status->timestamp_low = get_rx_desc_tsfl(pdesc);
        status->rx_is40mhzpacket = (bool)get_rx_desc_bw(pdesc);
        status->is_ht = (bool)get_rx_desc_rxht(pdesc);

        status->is_cck = RX_HAL_IS_CCK_RATE(status->rate);

        rx_status->freq = hw->conf.chandef.chan->center_freq;
        rx_status->band = hw->conf.chandef.chan->band;

        hdr = (struct ieee80211_hdr *)(skb->data + status->rx_drvinfo_size
                        + status->rx_bufshift);

        if (status->crc)
                rx_status->flag |= RX_FLAG_FAILED_FCS_CRC;

        if (status->rx_is40mhzpacket)
                rx_status->bw = RATE_INFO_BW_40;

        if (status->is_ht)
                rx_status->encoding = RX_ENC_HT;

        rx_status->flag |= RX_FLAG_MACTIME_START;

        /* hw will set status->decrypted true, if it finds the
         * frame is open data frame or mgmt frame.
         * So hw will not decryption robust managment frame
         * for IEEE80211w but still set status->decrypted
         * true, so here we should set it back to undecrypted
         * for IEEE80211w frame, and mac80211 sw will help
         * to decrypt it
         */
        if (status->decrypted) {
                if ((!_ieee80211_is_robust_mgmt_frame(hdr)) &&
                    (ieee80211_has_protected(hdr->frame_control)))
                        rx_status->flag |= RX_FLAG_DECRYPTED;
                else
                        rx_status->flag &= ~RX_FLAG_DECRYPTED;
        }

        /* rate_idx: index of data rate into band's
         * supported rates or MCS index if HT rates
         * are use (RX_FLAG_HT)
         * Notice: this is diff with windows define
         */
        rx_status->rate_idx = rtlwifi_rate_mapping(hw, status->is_ht,
                                                   false, status->rate);

        rx_status->mactime = status->timestamp_low;
        if (phystatus == true) {
                p_drvinfo = (struct rx_fwinfo_8723e *)(skb->data +
                                                     status->rx_bufshift);

                translate_rx_signal_stuff(hw, skb, status, pdesc8, p_drvinfo);
        }
        rx_status->signal = status->recvsignalpower + 10;
        return true;
}

void rtl8723e_tx_fill_desc(struct ieee80211_hw *hw,
                           struct ieee80211_hdr *hdr, u8 *pdesc_tx,
                           u8 *txbd, struct ieee80211_tx_info *info,
                           struct ieee80211_sta *sta,
                           struct sk_buff *skb,
                           u8 hw_queue, struct rtl_tcb_desc *ptcb_desc)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
        struct rtl_pci *rtlpci = rtl_pcidev(rtl_pcipriv(hw));
        struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
        /* bool b_trigger_ac = false; */
        u8 *pdesc8 = (u8 *)pdesc_tx;
        __le32 *pdesc = (__le32 *)pdesc8;
        u16 seq_number;
        __le16 fc = hdr->frame_control;
        u8 fw_qsel = _rtl8723e_map_hwqueue_to_fwqueue(skb, hw_queue);
        bool firstseg = ((hdr->seq_ctrl &
                          cpu_to_le16(IEEE80211_SCTL_FRAG)) == 0);

        bool lastseg = ((hdr->frame_control &
                         cpu_to_le16(IEEE80211_FCTL_MOREFRAGS)) == 0);

        dma_addr_t mapping = dma_map_single(&rtlpci->pdev->dev, skb->data,
                                            skb->len, DMA_TO_DEVICE);
        u8 bw_40 = 0;

        if (dma_mapping_error(&rtlpci->pdev->dev, mapping)) {
                rtl_dbg(rtlpriv, COMP_SEND, DBG_TRACE,
                        "DMA mapping error\n");
                return;
        }
        if (mac->opmode == NL80211_IFTYPE_STATION) {
                bw_40 = mac->bw_40;
        } else if (mac->opmode == NL80211_IFTYPE_AP ||
                mac->opmode == NL80211_IFTYPE_ADHOC) {
                if (sta)
                        bw_40 = sta->deflink.ht_cap.cap &
                                IEEE80211_HT_CAP_SUP_WIDTH_20_40;
        }

        seq_number = (le16_to_cpu(hdr->seq_ctrl) & IEEE80211_SCTL_SEQ) >> 4;

        rtl_get_tcb_desc(hw, info, sta, skb, ptcb_desc);

        clear_pci_tx_desc_content(pdesc, sizeof(struct tx_desc_8723e));

        if (ieee80211_is_nullfunc(fc) || ieee80211_is_ctl(fc)) {
                firstseg = true;
                lastseg = true;
        }

        if (firstseg) {
                set_tx_desc_offset(pdesc, USB_HWDESC_HEADER_LEN);

                set_tx_desc_tx_rate(pdesc, ptcb_desc->hw_rate);

                if (ptcb_desc->use_shortgi || ptcb_desc->use_shortpreamble)
                        set_tx_desc_data_shortgi(pdesc, 1);

                if (info->flags & IEEE80211_TX_CTL_AMPDU) {
                        set_tx_desc_agg_break(pdesc, 1);
                        set_tx_desc_max_agg_num(pdesc, 0x14);
                }
                set_tx_desc_seq(pdesc, seq_number);

                set_tx_desc_rts_enable(pdesc,
                                       ((ptcb_desc->rts_enable &&
                                        !ptcb_desc->cts_enable) ? 1 : 0));
                set_tx_desc_hw_rts_enable(pdesc,
                                          ((ptcb_desc->rts_enable ||
                                          ptcb_desc->cts_enable) ? 1 : 0));
                set_tx_desc_cts2self(pdesc,
                                     ((ptcb_desc->cts_enable) ? 1 : 0));
                set_tx_desc_rts_stbc(pdesc,
                                     ((ptcb_desc->rts_stbc) ? 1 : 0));

                set_tx_desc_rts_rate(pdesc, ptcb_desc->rts_rate);
                set_tx_desc_rts_bw(pdesc, 0);
                set_tx_desc_rts_sc(pdesc, ptcb_desc->rts_sc);
                set_tx_desc_rts_short(pdesc,
                                ((ptcb_desc->rts_rate <= DESC92C_RATE54M) ?
                                (ptcb_desc->rts_use_shortpreamble ? 1 : 0)
                                : (ptcb_desc->rts_use_shortgi ? 1 : 0)));

                if (bw_40) {
                        if (ptcb_desc->packet_bw == HT_CHANNEL_WIDTH_20_40) {
                                set_tx_desc_data_bw(pdesc, 1);
                                set_tx_desc_tx_sub_carrier(pdesc, 3);
                        } else {
                                set_tx_desc_data_bw(pdesc, 0);
                                set_tx_desc_tx_sub_carrier(pdesc,
                                        mac->cur_40_prime_sc);
                        }
                } else {
                        set_tx_desc_data_bw(pdesc, 0);
                        set_tx_desc_tx_sub_carrier(pdesc, 0);
                }

                set_tx_desc_linip(pdesc, 0);
                set_tx_desc_pkt_size(pdesc, (u16)skb->len);

                if (sta) {
                        u8 ampdu_density = sta->deflink.ht_cap.ampdu_density;
                        set_tx_desc_ampdu_density(pdesc, ampdu_density);
                }

                if (info->control.hw_key) {
                        struct ieee80211_key_conf *keyconf =
                            info->control.hw_key;

                        switch (keyconf->cipher) {
                        case WLAN_CIPHER_SUITE_WEP40:
                        case WLAN_CIPHER_SUITE_WEP104:
                        case WLAN_CIPHER_SUITE_TKIP:
                                set_tx_desc_sec_type(pdesc, 0x1);
                                break;
                        case WLAN_CIPHER_SUITE_CCMP:
                                set_tx_desc_sec_type(pdesc, 0x3);
                                break;
                        default:
                                set_tx_desc_sec_type(pdesc, 0x0);
                                break;

                        }
                }

                set_tx_desc_pkt_id(pdesc, 0);
                set_tx_desc_queue_sel(pdesc, fw_qsel);

                set_tx_desc_data_rate_fb_limit(pdesc, 0x1F);
                set_tx_desc_rts_rate_fb_limit(pdesc, 0xF);
                set_tx_desc_disable_fb(pdesc, 0);
                set_tx_desc_use_rate(pdesc, ptcb_desc->use_driver_rate ? 1 : 0);

                if (ieee80211_is_data_qos(fc)) {
                        if (mac->rdg_en) {
                                rtl_dbg(rtlpriv, COMP_SEND, DBG_TRACE,
                                        "Enable RDG function.\n");
                                set_tx_desc_rdg_enable(pdesc, 1);
                                set_tx_desc_htc(pdesc, 1);
                        }
                }
        }

        set_tx_desc_first_seg(pdesc, (firstseg ? 1 : 0));
        set_tx_desc_last_seg(pdesc, (lastseg ? 1 : 0));

        set_tx_desc_tx_buffer_size(pdesc, (u16)skb->len);

        set_tx_desc_tx_buffer_address(pdesc, mapping);

        if (rtlpriv->dm.useramask) {
                set_tx_desc_rate_id(pdesc, ptcb_desc->ratr_index);
                set_tx_desc_macid(pdesc, ptcb_desc->mac_id);
        } else {
                set_tx_desc_rate_id(pdesc, 0xC + ptcb_desc->ratr_index);
                set_tx_desc_macid(pdesc, ptcb_desc->ratr_index);
        }

        if ((!ieee80211_is_data_qos(fc)) && ppsc->fwctrl_lps) {
                set_tx_desc_hwseq_en_8723(pdesc, 1);
                /* set_tx_desc_hwseq_en(pdesc, 1); */
                /* set_tx_desc_pkt_id(pdesc, 8); */
                /* set_tx_desc_qos(pdesc, 1); */
        }

        set_tx_desc_more_frag(pdesc, (lastseg ? 0 : 1));

        if (is_multicast_ether_addr(ieee80211_get_DA(hdr)) ||
            is_broadcast_ether_addr(ieee80211_get_DA(hdr))) {
                set_tx_desc_bmc(pdesc, 1);
        }

        rtl_dbg(rtlpriv, COMP_SEND, DBG_TRACE, "\n");
}

void rtl8723e_tx_fill_cmddesc(struct ieee80211_hw *hw, u8 *pdesc8,
                              struct sk_buff *skb)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_pci *rtlpci = rtl_pcidev(rtl_pcipriv(hw));
        u8 fw_queue = QSLT_BEACON;
        __le32 *pdesc = (__le32 *)pdesc8;

        struct ieee80211_hdr *hdr = rtl_get_hdr(skb);
        __le16 fc = hdr->frame_control;

        dma_addr_t mapping = dma_map_single(&rtlpci->pdev->dev, skb->data,
                                            skb->len, DMA_TO_DEVICE);

        if (dma_mapping_error(&rtlpci->pdev->dev, mapping)) {
                rtl_dbg(rtlpriv, COMP_SEND, DBG_TRACE,
                        "DMA mapping error\n");
                return;
        }
        clear_pci_tx_desc_content(pdesc, TX_DESC_SIZE);

        set_tx_desc_offset(pdesc, USB_HWDESC_HEADER_LEN);

        set_tx_desc_tx_rate(pdesc, DESC92C_RATE1M);

        set_tx_desc_seq(pdesc, 0);

        set_tx_desc_linip(pdesc, 0);

        set_tx_desc_queue_sel(pdesc, fw_queue);

        set_tx_desc_first_seg(pdesc, 1);
        set_tx_desc_last_seg(pdesc, 1);

        set_tx_desc_tx_buffer_size(pdesc, (u16)(skb->len));

        set_tx_desc_tx_buffer_address(pdesc, mapping);

        set_tx_desc_rate_id(pdesc, 7);
        set_tx_desc_macid(pdesc, 0);

        set_tx_desc_own(pdesc, 1);

        set_tx_desc_pkt_size(pdesc, (u16)(skb->len));

        set_tx_desc_first_seg(pdesc, 1);
        set_tx_desc_last_seg(pdesc, 1);

        set_tx_desc_offset(pdesc, 0x20);

        set_tx_desc_use_rate(pdesc, 1);

        if (!ieee80211_is_data_qos(fc)) {
                set_tx_desc_hwseq_en_8723(pdesc, 1);
                /* set_tx_desc_hwseq_en(pdesc, 1); */
                /* set_tx_desc_pkt_id(pdesc, 8); */
        }

        RT_PRINT_DATA(rtlpriv, COMP_CMD, DBG_LOUD,
                      "H2C Tx Cmd Content\n",
                      pdesc, TX_DESC_SIZE);
}

void rtl8723e_set_desc(struct ieee80211_hw *hw, u8 *pdesc8,
                       bool istx, u8 desc_name, u8 *val)
{
        __le32 *pdesc = (__le32 *)pdesc8;

        if (istx) {
                switch (desc_name) {
                case HW_DESC_OWN:
                        set_tx_desc_own(pdesc, 1);
                        break;
                case HW_DESC_TX_NEXTDESC_ADDR:
                        set_tx_desc_next_desc_address(pdesc, *(u32 *)val);
                        break;
                default:
                        WARN_ONCE(true, "rtl8723ae: ERR txdesc :%d not processed\n",
                                  desc_name);
                        break;
                }
        } else {
                switch (desc_name) {
                case HW_DESC_RXOWN:
                        set_rx_desc_own(pdesc, 1);
                        break;
                case HW_DESC_RXBUFF_ADDR:
                        set_rx_desc_buff_addr(pdesc, *(u32 *)val);
                        break;
                case HW_DESC_RXPKT_LEN:
                        set_rx_desc_pkt_len(pdesc, *(u32 *)val);
                        break;
                case HW_DESC_RXERO:
                        set_rx_desc_eor(pdesc, 1);
                        break;
                default:
                        WARN_ONCE(true, "rtl8723ae: ERR rxdesc :%d not processed\n",
                                  desc_name);
                        break;
                }
        }
}

u64 rtl8723e_get_desc(struct ieee80211_hw *hw,
                      u8 *pdesc8, bool istx, u8 desc_name)
{
        u32 ret = 0;
        __le32 *pdesc = (__le32 *)pdesc8;

        if (istx) {
                switch (desc_name) {
                case HW_DESC_OWN:
                        ret = get_tx_desc_own(pdesc);
                        break;
                case HW_DESC_TXBUFF_ADDR:
                        ret = get_tx_desc_tx_buffer_address(pdesc);
                        break;
                default:
                        WARN_ONCE(true, "rtl8723ae: ERR txdesc :%d not processed\n",
                                  desc_name);
                        break;
                }
        } else {
                switch (desc_name) {
                case HW_DESC_OWN:
                        ret = get_rx_desc_own(pdesc);
                        break;
                case HW_DESC_RXPKT_LEN:
                        ret = get_rx_desc_pkt_len(pdesc);
                        break;
                case HW_DESC_RXBUFF_ADDR:
                        ret = get_rx_desc_buff_addr(pdesc);
                        break;
                default:
                        WARN_ONCE(true, "rtl8723ae: ERR rxdesc :%d not processed\n",
                                  desc_name);
                        break;
                }
        }
        return ret;
}

bool rtl8723e_is_tx_desc_closed(struct ieee80211_hw *hw,
                                u8 hw_queue, u16 index)
{
        struct rtl_pci *rtlpci = rtl_pcidev(rtl_pcipriv(hw));
        struct rtl8192_tx_ring *ring = &rtlpci->tx_ring[hw_queue];
        u8 *entry = (u8 *)(&ring->desc[ring->idx]);
        u8 own = (u8)rtl8723e_get_desc(hw, entry, true, HW_DESC_OWN);

        /**
         *beacon packet will only use the first
         *descriptor defautly,and the own may not
         *be cleared by the hardware
         */
        if (own)
                return false;
        return true;
}

void rtl8723e_tx_polling(struct ieee80211_hw *hw, u8 hw_queue)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        if (hw_queue == BEACON_QUEUE) {
                rtl_write_word(rtlpriv, REG_PCIE_CTRL_REG, BIT(4));
        } else {
                rtl_write_word(rtlpriv, REG_PCIE_CTRL_REG,
                               BIT(0) << (hw_queue));
        }
}