/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2022 Adrian Chadd <adrian@FreeBSD.org>.
 *
 * 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/param.h>
#include <sys/bus.h>
#include <sys/errno.h>
#include <sys/kernel.h>
#include <sys/malloc.h>
#include <sys/module.h>
#include <sys/socket.h>
#include <sys/sockio.h>
#include <sys/sysctl.h>
#include <sys/systm.h>

#include <net/if.h>
#include <net/if_var.h>
#include <net/if_arp.h>
#include <net/ethernet.h>
#include <net/if_dl.h>
#include <net/if_media.h>
#include <net/if_types.h>

#include <machine/bus.h>
#include <dev/iicbus/iic.h>
#include <dev/iicbus/iiconf.h>
#include <dev/iicbus/iicbus.h>
#include <dev/mii/mii.h>
#include <dev/mii/miivar.h>
#include <dev/mdio/mdio.h>
#include <dev/clk/clk.h>
#include <dev/hwreset/hwreset.h>

#include <dev/fdt/fdt_common.h>
#include <dev/ofw/ofw_bus.h>
#include <dev/ofw/ofw_bus_subr.h>

#include <dev/etherswitch/etherswitch.h>

#include <dev/etherswitch/ar40xx/ar40xx_var.h>
#include <dev/etherswitch/ar40xx/ar40xx_reg.h>
#include <dev/etherswitch/ar40xx/ar40xx_hw.h>
#include <dev/etherswitch/ar40xx/ar40xx_debug.h>

/*
 * XXX these are here for now; move the code using these
 * into main.c once this is all done!
 */
#include <dev/etherswitch/ar40xx/ar40xx_hw_vtu.h>
#include <dev/etherswitch/ar40xx/ar40xx_hw_port.h>
#include <dev/etherswitch/ar40xx/ar40xx_hw_mirror.h>

#include "mdio_if.h"
#include "miibus_if.h"
#include "etherswitch_if.h"

/*
 * Reset the ESS switch.  This also resets the ESS ethernet
 * and PSGMII block.
 */
int
ar40xx_hw_ess_reset(struct ar40xx_softc *sc)
{
        int ret;

        AR40XX_DPRINTF(sc, AR40XX_DBG_HW_RESET, "%s: called\n", __func__);

        ret = hwreset_assert(sc->sc_ess_rst);
        if (ret != 0) {
                device_printf(sc->sc_dev, "ERROR: failed to assert reset\n");
                return ret;
        }
        DELAY(10*1000);

        ret = hwreset_deassert(sc->sc_ess_rst);
        if (ret != 0) {
                device_printf(sc->sc_dev,
                    "ERROR: failed to deassert reset\n");
                return ret;
        }

        DELAY(10*1000);

        return (0);
}

int
ar40xx_hw_init_globals(struct ar40xx_softc *sc)
{
        uint32_t reg;

        AR40XX_DPRINTF(sc, AR40XX_DBG_HW_INIT, "%s: called\n", __func__);

        /* enable CPU port and disable mirror port */
        reg = AR40XX_FWD_CTRL0_CPU_PORT_EN
             | AR40XX_FWD_CTRL0_MIRROR_PORT;
        AR40XX_REG_WRITE(sc, AR40XX_REG_FWD_CTRL0, reg);

        /* forward multicast and broadcast frames to CPU */
        reg = (AR40XX_PORTS_ALL << AR40XX_FWD_CTRL1_UC_FLOOD_S)
            | (AR40XX_PORTS_ALL << AR40XX_FWD_CTRL1_MC_FLOOD_S)
            | (AR40XX_PORTS_ALL << AR40XX_FWD_CTRL1_BC_FLOOD_S);
        AR40XX_REG_WRITE(sc, AR40XX_REG_FWD_CTRL1, reg);

        /* enable jumbo frames */
        reg = AR40XX_REG_READ(sc, AR40XX_REG_MAX_FRAME_SIZE);
        reg &= ~AR40XX_MAX_FRAME_SIZE_MTU;
        reg |= 9018 + 8 + 2;
        AR40XX_REG_WRITE(sc, AR40XX_REG_MAX_FRAME_SIZE, reg);

        /* Enable MIB counters */
        reg = AR40XX_REG_READ(sc, AR40XX_REG_MODULE_EN);
        reg |= AR40XX_MODULE_EN_MIB;
        AR40XX_REG_WRITE(sc, AR40XX_REG_MODULE_EN, reg);

        /* Disable AZ */
        AR40XX_REG_WRITE(sc, AR40XX_REG_EEE_CTRL, 0);

        /* set flowctrl thershold for cpu port */
        reg = (AR40XX_PORT0_FC_THRESH_ON_DFLT << 16)
            | AR40XX_PORT0_FC_THRESH_OFF_DFLT;
        AR40XX_REG_WRITE(sc, AR40XX_REG_PORT_FLOWCTRL_THRESH(0), reg);

        AR40XX_REG_BARRIER_WRITE(sc);

        return (0);
}

int
ar40xx_hw_vlan_init(struct ar40xx_softc *sc)
{
        int i;

        AR40XX_DPRINTF(sc, AR40XX_DBG_HW_INIT, "%s: called\n", __func__);

        /* Enable VLANs by default */
        sc->sc_vlan.vlan = 1;

        /* Configure initial LAN/WAN bitmap and include CPU port as tagged */
        sc->sc_vlan.vlan_id[AR40XX_LAN_VLAN] = AR40XX_LAN_VLAN
            | ETHERSWITCH_VID_VALID;
        sc->sc_vlan.vlan_id[AR40XX_WAN_VLAN] = AR40XX_WAN_VLAN
            | ETHERSWITCH_VID_VALID;

        sc->sc_vlan.vlan_ports[AR40XX_LAN_VLAN] =
            sc->sc_config.switch_cpu_bmp | sc->sc_config.switch_lan_bmp;
        sc->sc_vlan.vlan_untagged[AR40XX_LAN_VLAN] =
            sc->sc_config.switch_lan_bmp;

        sc->sc_vlan.vlan_ports[AR40XX_WAN_VLAN] =
            sc->sc_config.switch_cpu_bmp | sc->sc_config.switch_wan_bmp;
        sc->sc_vlan.vlan_untagged[AR40XX_WAN_VLAN] =
            sc->sc_config.switch_wan_bmp;

        /* Populate the per-port PVID - pvid[] is an index into vlan_id[] */
        for (i = 0; i < AR40XX_NUM_PORTS; i++) {
                if (sc->sc_config.switch_lan_bmp & (1U << i))
                        sc->sc_vlan.pvid[i] = AR40XX_LAN_VLAN;
                if (sc->sc_config.switch_wan_bmp & (1U << i))
                        sc->sc_vlan.pvid[i] = AR40XX_WAN_VLAN;
        }

        return (0);
}

/*
 * Apply the per-port and global configuration from software.
 *
 * This is useful if we ever start doing the linux switch framework
 * thing of updating the config in one hit and pushing it to the
 * hardware.  For now it's just used in the reset path.
 */
int
ar40xx_hw_sw_hw_apply(struct ar40xx_softc *sc)
{
        uint8_t portmask[AR40XX_NUM_PORTS];
        int i, j, ret;

        AR40XX_DPRINTF(sc, AR40XX_DBG_HW_INIT, "%s: called\n", __func__);

        /*
         * Flush the VTU configuration.
         */
        ret = ar40xx_hw_vtu_flush(sc);
        if (ret != 0) {
                device_printf(sc->sc_dev,
                    "ERROR: couldn't apply config; vtu flush failed (%d)\n",
                    ret);
                return (ret);
        }

        memset(portmask, 0, sizeof(portmask));

        /*
         * Configure the ports based on whether it's 802.1q
         * VLANs, or just straight up per-port VLANs.
         */
        if (sc->sc_vlan.vlan) {
                device_printf(sc->sc_dev, "%s: configuring 802.1q VLANs\n",
                    __func__);
                for (j = 0; j < AR40XX_NUM_VTU_ENTRIES; j++) {
                        uint8_t vp = sc->sc_vlan.vlan_ports[j];

                        if (!vp)
                                continue;
                        if ((sc->sc_vlan.vlan_id[j]
                            & ETHERSWITCH_VID_VALID) == 0)
                                continue;

                        for (i = 0; i < AR40XX_NUM_PORTS; i++) {
                                uint8_t mask = (1U << i);

                                if (vp & mask)
                                        portmask[i] |= vp & ~mask;
                        }

                        ar40xx_hw_vtu_load_vlan(sc,
                            sc->sc_vlan.vlan_id[j] & ETHERSWITCH_VID_MASK,
                            sc->sc_vlan.vlan_ports[j],
                            sc->sc_vlan.vlan_untagged[j]);
                }
        } else {
                device_printf(sc->sc_dev, "%s: configuring per-port VLANs\n",
                    __func__);
                for (i = 0; i < AR40XX_NUM_PORTS; i++) {
                        if (i == AR40XX_PORT_CPU)
                                continue;

                        portmask[i] = (1U << AR40XX_PORT_CPU);
                        portmask[AR40XX_PORT_CPU] |= (1U << i);
                }
        }

        /*
         * Update per-port destination mask, vlan tag settings
         */
        for (i = 0; i < AR40XX_NUM_PORTS; i++)
                (void) ar40xx_hw_port_setup(sc, i, portmask[i]);

        /* Set the mirror register config */
        ret = ar40xx_hw_mirror_set_registers(sc);
        if (ret != 0) {
                device_printf(sc->sc_dev,
                    "ERROR: couldn't apply config; mirror config failed"
                    " (%d)\n",
                    ret);
                return (ret);
        }

        return (0);
}

int
ar40xx_hw_wait_bit(struct ar40xx_softc *sc, int reg, uint32_t mask,
    uint32_t val)
{
        int timeout = 20;
        uint32_t t;

        while (true) {
                AR40XX_REG_BARRIER_READ(sc);
                t = AR40XX_REG_READ(sc, reg);
                if ((t & mask) == val)
                        return 0;

                if (timeout-- <= 0)
                        break;

                DELAY(20);
        }

        device_printf(sc->sc_dev, "ERROR: timeout for reg "
            "%08x: %08x & %08x != %08x\n",
            (unsigned int)reg, t, mask, val);
        return (ETIMEDOUT);
}

/*
 * Read the switch MAC address.
 */
int
ar40xx_hw_read_switch_mac_address(struct ar40xx_softc *sc,
    struct ether_addr *ea)
{
        uint32_t ret0, ret1;
        char *s;

        s = (void *) ea;

        AR40XX_LOCK_ASSERT(sc);

        AR40XX_REG_BARRIER_READ(sc);
        ret0 = AR40XX_REG_READ(sc, AR40XX_REG_SW_MAC_ADDR0);
        ret1 = AR40XX_REG_READ(sc, AR40XX_REG_SW_MAC_ADDR1);

        s[5] = MS(ret0, AR40XX_REG_SW_MAC_ADDR0_BYTE5);
        s[4] = MS(ret0, AR40XX_REG_SW_MAC_ADDR0_BYTE4);
        s[3] = MS(ret1, AR40XX_REG_SW_MAC_ADDR1_BYTE3);
        s[2] = MS(ret1, AR40XX_REG_SW_MAC_ADDR1_BYTE2);
        s[1] = MS(ret1, AR40XX_REG_SW_MAC_ADDR1_BYTE1);
        s[0] = MS(ret1, AR40XX_REG_SW_MAC_ADDR1_BYTE0);

        return (0);
}

/*
 * Set the switch MAC address.
 */
int
ar40xx_hw_write_switch_mac_address(struct ar40xx_softc *sc,
    struct ether_addr *ea)
{
        uint32_t ret0 = 0, ret1 = 0;
        char *s;

        s = (void *) ea;

        AR40XX_LOCK_ASSERT(sc);

        ret0 |= SM(s[5], AR40XX_REG_SW_MAC_ADDR0_BYTE5);
        ret0 |= SM(s[4], AR40XX_REG_SW_MAC_ADDR0_BYTE4);

        ret1 |= SM(s[3], AR40XX_REG_SW_MAC_ADDR1_BYTE3);
        ret1 |= SM(s[2], AR40XX_REG_SW_MAC_ADDR1_BYTE2);
        ret1 |= SM(s[1], AR40XX_REG_SW_MAC_ADDR1_BYTE1);
        ret1 |= SM(s[0], AR40XX_REG_SW_MAC_ADDR1_BYTE0);

        AR40XX_REG_WRITE(sc, AR40XX_REG_SW_MAC_ADDR0, ret0);
        AR40XX_REG_WRITE(sc, AR40XX_REG_SW_MAC_ADDR1, ret1);

        AR40XX_REG_BARRIER_WRITE(sc);

        return (0);
}