/*-
 * Copyright (c) 2017 Kevin Lo <kevlo@FreeBSD.org>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/cdefs.h>
#include "opt_wlan.h"

#include <sys/param.h>
#include <sys/lock.h>
#include <sys/mutex.h>
#include <sys/mbuf.h>
#include <sys/kernel.h>
#include <sys/socket.h>
#include <sys/systm.h>
#include <sys/malloc.h>
#include <sys/queue.h>
#include <sys/taskqueue.h>
#include <sys/bus.h>
#include <sys/endian.h>
#include <sys/linker.h>

#include <net/if.h>
#include <net/ethernet.h>
#include <net/if_media.h>

#include <net80211/ieee80211_var.h>
#include <net80211/ieee80211_radiotap.h>

#include <dev/rtwn/if_rtwnreg.h>
#include <dev/rtwn/if_rtwnvar.h>

#include <dev/rtwn/if_rtwn_debug.h>
#include <dev/rtwn/if_rtwn_ridx.h>
#include <dev/rtwn/if_rtwn_rx.h>

#include <dev/rtwn/rtl8192c/r92c.h>

#include <dev/rtwn/rtl8192e/r92e.h>
#include <dev/rtwn/rtl8192e/r92e_reg.h>
#include <dev/rtwn/rtl8192e/r92e_var.h>

static int
r92e_get_power_group(struct rtwn_softc *sc, struct ieee80211_channel *c)
{
        uint8_t chan;
        int group;

        chan = rtwn_chan2centieee(c);
        if (IEEE80211_IS_CHAN_2GHZ(c)) {
                if (chan <= 2)                  group = 0;
                else if (chan <= 5)             group = 1;
                else if (chan <= 8)             group = 2;
                else if (chan <= 11)            group = 3;
                else if (chan <= 14)            group = 4;
                else {
                        KASSERT(0, ("wrong 2GHz channel %d!\n", chan));
                        return (-1);
                }
        } else {
                KASSERT(0, ("wrong channel band (flags %08X)\n", c->ic_flags));
                return (-1);
        }

        return (group);
}

static void
r92e_get_txpower(struct rtwn_softc *sc, int chain, struct ieee80211_channel *c,
    uint8_t power[RTWN_RIDX_COUNT])
{
        const struct ieee80211com *ic = &sc->sc_ic;
        struct r92e_softc *rs = sc->sc_priv;
        int i, ridx, group, max_mcs;

        /* Determine channel group. */
        group = r92e_get_power_group(sc, c);
        if (group == -1) {      /* shouldn't happen */
                device_printf(sc->sc_dev, "%s: incorrect channel\n", __func__);
                return;
        }

        max_mcs = RTWN_RIDX_HT_MCS(sc->ntxchains * 8 - 1);

        /* XXX regulatory */

        for (ridx = RTWN_RIDX_CCK1; ridx <= RTWN_RIDX_CCK11; ridx++) {
                power[ridx] = rs->cck_tx_pwr[chain][group];
                if (power[ridx] > ic->ic_txpowlimit)
                        power[ridx] = ic->ic_txpowlimit;
        }
        for (ridx = RTWN_RIDX_OFDM6; ridx <= max_mcs; ridx++) {
                power[ridx] = rs->ht40_tx_pwr_2g[chain][group];
                if (power[ridx] > ic->ic_txpowlimit)
                        power[ridx] = ic->ic_txpowlimit;
        }

        for (ridx = RTWN_RIDX_OFDM6; ridx <= RTWN_RIDX_OFDM54; ridx++) {
                /* Ensure we don't underflow if the power delta is -ve */
                int8_t pwr;

                pwr = power[ridx] + rs->ofdm_tx_pwr_diff_2g[chain][0];
                if (pwr < 0)
                        pwr = 0;

                power[ridx] = pwr;
        }

        for (i = 0; i < sc->ntxchains; i++) {
                uint8_t min_mcs;
                int8_t pwr_diff, pwr;

                if (IEEE80211_IS_CHAN_HT40(c))
                        pwr_diff = rs->bw40_tx_pwr_diff_2g[chain][i];
                else
                        pwr_diff = rs->bw20_tx_pwr_diff_2g[chain][i];

                min_mcs = RTWN_RIDX_HT_MCS(i * 8);
                for (ridx = min_mcs; ridx <= max_mcs; ridx++) {
                        /* Ensure we don't underflow */
                        pwr = power[ridx] + pwr_diff;
                        if (pwr < 0)
                                pwr = 0;
                        power[ridx] = pwr;
                }
        }

        /* Apply max limit. */
        for (ridx = RTWN_RIDX_CCK1; ridx <= max_mcs; ridx++) {
                if (power[ridx] > R92C_MAX_TX_PWR)
                        power[ridx] = R92C_MAX_TX_PWR;
        }
}

static void
r92e_set_txpower(struct rtwn_softc *sc, struct ieee80211_channel *c)
{
        uint8_t power[RTWN_RIDX_COUNT];
        int i;

        for (i = 0; i < sc->ntxchains; i++) {
                memset(power, 0, sizeof(power));
                /* Compute per-rate Tx power values. */
                r92e_get_txpower(sc, i, c, power);
                /* Optionally print out the power table */
                r92c_dump_txpower(sc, i, power);
                /* Write per-rate Tx power values to hardware. */
                r92c_write_txpower(sc, i, power);
        }
}

int
r92e_set_tx_power(struct rtwn_softc *sc, struct ieee80211vap *vap)
{

        if (vap->iv_bss == NULL)
                return (EINVAL);
        if (vap->iv_bss->ni_chan == IEEE80211_CHAN_ANYC)
                return (EINVAL);

        r92e_set_txpower(sc, vap->iv_bss->ni_chan);
        return (0);
}

static void
r92e_set_bw40(struct rtwn_softc *sc, uint8_t chan, int prichlo)
{
        int i;

        rtwn_setbits_2(sc, R92C_WMAC_TRXPTCL_CTL, 0x100, 0x80);
        rtwn_write_1(sc, R12A_DATA_SEC,
            prichlo ? R12A_DATA_SEC_PRIM_DOWN_20 : R12A_DATA_SEC_PRIM_UP_20);

        rtwn_bb_setbits(sc, R92C_FPGA0_RFMOD, 0, R92C_RFMOD_40MHZ);
        rtwn_bb_setbits(sc, R92C_FPGA1_RFMOD, 0, R92C_RFMOD_40MHZ);

        /* Select 40MHz bandwidth. */
        for (i = 0; i < sc->nrxchains; i++)
                rtwn_rf_setbits(sc, i, R92C_RF_CHNLBW,
                    R88E_RF_CHNLBW_BW20, 0x400);

        /* Set CCK side band. */
        rtwn_bb_setbits(sc, R92C_CCK0_SYSTEM,
            R92C_CCK0_SYSTEM_CCK_SIDEBAND, (prichlo ? 0 : 1) << 4);
                
        rtwn_bb_setbits(sc, R92C_OFDM1_LSTF, 0x0c00, (prichlo ? 1 : 2) << 10);

        rtwn_bb_setbits(sc, R92C_FPGA0_ANAPARAM2,
            R92C_FPGA0_ANAPARAM2_CBW20, 0);

        rtwn_bb_setbits(sc, 0x818, 0x0c000000, (prichlo ? 2 : 1) << 26);
}

static void
r92e_set_bw20(struct rtwn_softc *sc, uint8_t chan)
{
        int i;

        rtwn_setbits_2(sc, R92C_WMAC_TRXPTCL_CTL, 0x180, 0);
        rtwn_write_1(sc, R12A_DATA_SEC, R12A_DATA_SEC_NO_EXT);

        rtwn_bb_setbits(sc, R92C_FPGA0_RFMOD, R92C_RFMOD_40MHZ, 0);
        rtwn_bb_setbits(sc, R92C_FPGA1_RFMOD, R92C_RFMOD_40MHZ, 0);

        /* Select 20MHz bandwidth. */
        for (i = 0; i < sc->nrxchains; i++)
                rtwn_rf_setbits(sc, i, R92C_RF_CHNLBW,
                    R88E_RF_CHNLBW_BW20, 0xc00);

        rtwn_bb_setbits(sc, R92C_OFDM0_TXPSEUDONOISEWGT, 0xc0000000, 0);
}

void
r92e_set_chan(struct rtwn_softc *sc, struct ieee80211_channel *c)
{
        struct r92e_softc *rs = sc->sc_priv;
        u_int chan;
        int i;

        chan = rtwn_chan2centieee(c);

        for (i = 0; i < sc->nrxchains; i++) {
                rtwn_rf_write(sc, i, R92C_RF_CHNLBW,
                    RW(rs->rf_chnlbw[0], R92C_RF_CHNLBW_CHNL, chan));
        }

        if (IEEE80211_IS_CHAN_HT40(c))
                r92e_set_bw40(sc, chan, IEEE80211_IS_CHAN_HT40U(c));
        else
                r92e_set_bw20(sc, chan);

        /* Set Tx power for this new channel. */
        r92e_set_txpower(sc, c);

        /*
         * Work around some timing issues with RTL8192EU on faster
         * CPUs / USB-3 ports by sleeping for 10ms.
         *
         * Without this delay the initial frame send during authentication
         * doesn't occur.
         *
         * My (adrian) guess is that there's a race condition between
         * everything being programmed into the hardware and the first
         * send.  Notably, TXPAUSE isn't 0x0 after rf init,
         * which the rtl8xxxu driver has a commit to address (c6015bf3ff1ff)
         * - wifi: rtl8xxxu: fixing transmission failure for rtl8192eu
         *
         * Although it's hard to do due to locking constraints, reading
         * TXPAUSE during scan / association shows it's non-zero, which
         * needs to be looked at in more depth.
         *
         * Linux doesn't have a delay here, however it does try multiple
         * times to send an authentication frame.
         *
         * See PR/247528 for more info.
         */
        rtwn_delay(sc, 10000);
}