/* $NetBSD: mesong12_usb3pciephy.c,v 1.3 2024/02/07 04:20:26 msaitoh Exp $ */

/*
 * Copyright (c) 2021 Ryo Shimizu
 * 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 ``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 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>
__KERNEL_RCSID(0, "$NetBSD: mesong12_usb3pciephy.c,v 1.3 2024/02/07 04:20:26 msaitoh Exp $");

#include <sys/param.h>
#include <sys/types.h>
#include <sys/bus.h>
#include <sys/device.h>

#include <dev/fdt/fdtvar.h>

#define USB3PCIEPHY_R0_REG                              0x00
#define  USB3PCIEPHY_R0_PCIE_USB3_SWITCH                __BITS(6,5)
#define  USB3PCIEPHY_R0_PCIE_POWER_STATE                __BITS(4,0)
#define USB3PCIEPHY_R1_REG                              0x04
#define  USB3PCIEPHY_R1_PHY_MPLL_MULTIPLIER             __BITS(31,25)
#define  USB3PCIEPHY_R1_PHY_REF_CLKDIV2                 __BIT(24)
#define  USB3PCIEPHY_R1_PHY_LOS_BIAS                    __BITS(23,21)
#define  USB3PCIEPHY_R1_PHY_LOS_LEVEL                   __BITS(20,16)
#define  USB3PCIEPHY_R1_PHY_RX0_EQ                      __BITS(15,13)
#define  USB3PCIEPHY_R1_PHY_RX1_EQ                      __BITS(12,10)
#define  USB3PCIEPHY_R1_PHY_TX0_TERM_OFFSET             __BITS(9,5)
#define  USB3PCIEPHY_R1_PHY_TX1_TERM_OFFSET             __BITS(4,0)
#define USB3PCIEPHY_R2_REG                              0x08
#define  USB3PCIEPHY_R2_PHY_TX_VBOOST_LVL               __BITS(20,18)
#define  USB3PCIEPHY_R2_PCS_TX_DEEMPH_GEN1              __BITS(17,12)
#define  USB3PCIEPHY_R2_PCS_TX_DEEMPH_GEN2_3P5DB        __BITS(11,6)
#define  USB3PCIEPHY_R2_PCS_TX_DEEMPH_GEN2_6DB          __BITS(5,0)
#define USB3PCIEPHY_R4_REG                              0x10
#define  USB3PCIEPHY_R4_PHY_CR_CAP_ADDR                 __BIT(19)
#define  USB3PCIEPHY_R4_PHY_CR_CAP_DATA                 __BIT(18)
#define  USB3PCIEPHY_R4_PHY_CR_DATA_IN                  __BITS(17,2)
#define  USB3PCIEPHY_R4_PHY_CR_READ                     __BIT(1)
#define  USB3PCIEPHY_R4_PHY_CR_WRITE                    __BIT(0)
#define USB3PCIEPHY_R5_REG                              0x14
#define  USB3PCIEPHY_R5_PHY_BS_OUT                      __BIT(17)
#define  USB3PCIEPHY_R5_PHY_CR_ACK                      __BIT(16)
#define  USB3PCIEPHY_R5_PHY_CR_DATA_OUT                 __BITS(15,0)

#define PHY_READ_REG(sc, reg)   \
        bus_space_read_4((sc)->sc_bst, (sc)->sc_bsh, (reg))
#define PHY_WRITE_REG(sc, reg, val)     \
        bus_space_write_4((sc)->sc_bst, (sc)->sc_bsh, (reg), (val))


/* The values must be matched to those in dt-bindings/phy/phy.h */
#define PHY_NONE        0
#define PHY_TYPE_PCIE   2
#define PHY_TYPE_USB3   4

struct mesong12_usb3pciephy_softc {
        device_t sc_dev;
        bus_space_tag_t sc_bst;
        bus_space_handle_t sc_bsh;
        struct clk *sc_clk;
        struct fdtbus_reset *sc_reset;
        struct fdtbus_regulator *sc_supply;
        int sc_phandle;
        int sc_phy_type;
};

static void *
mesong12_usb3pciephy_acquire(device_t dev, const void *data, size_t len)
{
        struct mesong12_usb3pciephy_softc * const sc = device_private(dev);
        const uint32_t *p = data;

        /* already acquired? */
        if (sc->sc_phy_type != PHY_NONE)
                return NULL;

        if (len != sizeof(uint32_t))
                return NULL;

        switch (be32toh(p[0])) {
        case PHY_TYPE_USB3:
                sc->sc_phy_type = PHY_TYPE_USB3;
                break;
        case PHY_TYPE_PCIE:
                return NULL;    /* PCIe mode is not supported */
        default:
                return NULL;
        }

        return sc;
}

static void
mesong12_usb3pciephy_release(device_t dev, void *priv)
{
        struct mesong12_usb3pciephy_softc * const sc = device_private(dev);

        sc->sc_phy_type = PHY_NONE;
}

static inline int
mesong12_usb3pciephy_ack(struct mesong12_usb3pciephy_softc *sc, bool ack,
    const char *str)
{
        int timeout;
        uint32_t val;

        for (timeout = 1000; timeout > 0; timeout--) {
                val = PHY_READ_REG(sc, USB3PCIEPHY_R5_REG);
                if (!(val & USB3PCIEPHY_R5_PHY_CR_ACK) == !ack)
                        return 0;
                delay(5);
        }
        device_printf(sc->sc_dev, "phy %s %s timeout\n",
            str, ack ? "ack" : "nack");
        return ETIMEDOUT;
}

static void
mesong12_usb3pciephy_addr(struct mesong12_usb3pciephy_softc *sc,
    bus_addr_t addr)
{
        uint32_t val;

        val = __SHIFTIN(addr, USB3PCIEPHY_R4_PHY_CR_DATA_IN);
        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, val);
        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, val);
        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, val |
            USB3PCIEPHY_R4_PHY_CR_CAP_ADDR);
        if (mesong12_usb3pciephy_ack(sc, true, "addr") != 0)
                return;

        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, val);
        mesong12_usb3pciephy_ack(sc, false, "addr");
}

static uint16_t
mesong12_usb3pciephy_read(struct mesong12_usb3pciephy_softc *sc,
    bus_addr_t addr)
{
        uint32_t val;

        mesong12_usb3pciephy_addr(sc, addr);

        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, 0);
        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, USB3PCIEPHY_R4_PHY_CR_READ);
        if (mesong12_usb3pciephy_ack(sc, true, "read data") != 0)
                return 0;

        val = PHY_READ_REG(sc, USB3PCIEPHY_R5_REG);
        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, 0);
        if (mesong12_usb3pciephy_ack(sc, false, "read data") != 0)
                return 0;

        return __SHIFTOUT(val, USB3PCIEPHY_R5_PHY_CR_DATA_OUT);
}

static void
mesong12_usb3pciephy_write(struct mesong12_usb3pciephy_softc *sc,
    bus_addr_t addr, uint16_t data)
{
        uint32_t val;

        mesong12_usb3pciephy_addr(sc, addr);

        val = __SHIFTIN(addr, USB3PCIEPHY_R4_PHY_CR_DATA_IN);
        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, val);
        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, val);
        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, val |
            USB3PCIEPHY_R4_PHY_CR_CAP_DATA);
        if (mesong12_usb3pciephy_ack(sc, true, "write addr") != 0)
                return;
        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, val);
        if (mesong12_usb3pciephy_ack(sc, false, "write addr") != 0)
                return;

        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, val);
        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, val |
            USB3PCIEPHY_R4_PHY_CR_WRITE);
        if (mesong12_usb3pciephy_ack(sc, true, "write data") != 0)
                return;
        PHY_WRITE_REG(sc, USB3PCIEPHY_R4_REG, val);
        if (mesong12_usb3pciephy_ack(sc, false, "write data") != 0)
                return;
}

static int
mesong12_usb3pciephy_enable(device_t dev, void *priv, bool enable)
{
        struct mesong12_usb3pciephy_softc * const sc = device_private(dev);
        uint32_t val;

        fdtbus_clock_assign(sc->sc_phandle);
        if (sc->sc_reset != NULL) {
                fdtbus_reset_assert(sc->sc_reset);
                delay(10);
                fdtbus_reset_deassert(sc->sc_reset);
        }

        if (!enable)
                return 0;

        /* switch to USB3.0 */
        val = PHY_READ_REG(sc, USB3PCIEPHY_R0_REG);
        val &= ~USB3PCIEPHY_R0_PCIE_USB3_SWITCH;
        val |= __SHIFTIN(3, USB3PCIEPHY_R0_PCIE_USB3_SWITCH);
        PHY_WRITE_REG(sc, USB3PCIEPHY_R0_REG, val);
        delay(10);

#if 0 /* XXX: doesn't work? */
        /* workaround for SSPHY(SuperSpeedPHY) suspend bug */
        val = mesong12_usb3pciephy_read(sc, 0x102d);
        val |= __BIT(7);
        mesong12_usb3pciephy_write(sc, 0x102d, val);
#endif

        val = mesong12_usb3pciephy_read(sc, 0x1010);
        val &= ~__BITS(11,4);
        val |= __SHIFTIN(2, __BITS(11,4));
        mesong12_usb3pciephy_write(sc, 0x1010, val);

#if 0 /* XXX: doesn't work? */
        /* Rx equalization magic */
        val = mesong12_usb3pciephy_read(sc, 0x1006);
        val &= ~__BITS(7,6);
        val |= __SHIFTIN(2, __BITS(7,6));
        val &= ~__BITS(10,8);
        val |= __SHIFTIN(3, __BITS(10,8));
        val |= __BIT(11);
        mesong12_usb3pciephy_write(sc, 0x1006, val);
#endif

        /* Tx equalization magic */
        val = mesong12_usb3pciephy_read(sc, 0x1002);
        val &= ~__BITS(13,7);
        val |= __SHIFTIN(0x16, __BITS(13,7));
        val &= ~__BITS(6,0);
        val |= __SHIFTIN(0x7f, __BITS(6,0));
        val |= __BIT(14);
        mesong12_usb3pciephy_write(sc, 0x1002, val);

        /* MPLL loop magic */
        val = mesong12_usb3pciephy_read(sc, 0x30);
        val &= ~__BITS(7,4);
        val |= __SHIFTIN(8, __BITS(7,4));
        mesong12_usb3pciephy_write(sc, 0x30, val);


        val = PHY_READ_REG(sc, USB3PCIEPHY_R2_REG);
        val &= ~USB3PCIEPHY_R2_PHY_TX_VBOOST_LVL;
        val |= __SHIFTIN(4, USB3PCIEPHY_R2_PHY_TX_VBOOST_LVL);
        PHY_WRITE_REG(sc, USB3PCIEPHY_R2_REG, val);

        val = PHY_READ_REG(sc, USB3PCIEPHY_R1_REG);
        val &= ~USB3PCIEPHY_R1_PHY_LOS_BIAS;
        val |= __SHIFTIN(4, USB3PCIEPHY_R1_PHY_LOS_BIAS);
        val &= ~USB3PCIEPHY_R1_PHY_LOS_LEVEL;
        val |= __SHIFTIN(9, USB3PCIEPHY_R1_PHY_LOS_LEVEL);
        PHY_WRITE_REG(sc, USB3PCIEPHY_R1_REG, val);

        return 0;
}

static const struct device_compatible_entry compat_data[] = {
        { .compat = "amlogic,g12a-usb3-pcie-phy" },
        DEVICE_COMPAT_EOL
};

static int
mesong12_usb3pciephy_match(device_t parent, cfdata_t cf, void *aux)
{
        struct fdt_attach_args * const faa = aux;

        return of_compatible_match(faa->faa_phandle, compat_data);
}

static const struct fdtbus_phy_controller_func mesong12_usb3pciephy_funcs = {
        .acquire = mesong12_usb3pciephy_acquire,
        .release = mesong12_usb3pciephy_release,
        .enable = mesong12_usb3pciephy_enable
};

static void
mesong12_usb3pciephy_attach(device_t parent, device_t self, void *aux)
{
        struct mesong12_usb3pciephy_softc * const sc = device_private(self);
        struct fdt_attach_args * const faa = aux;
        const int phandle = faa->faa_phandle;
        bus_addr_t addr;
        bus_size_t size;

        sc->sc_dev = self;
        sc->sc_bst = faa->faa_bst;
        sc->sc_phandle = phandle;

        if (fdtbus_get_reg(phandle, 0, &addr, &size) != 0) {
                aprint_error(": couldn't get registers\n");
                return;
        }
        if (bus_space_map(sc->sc_bst, addr, size, 0, &sc->sc_bsh) != 0) {
                aprint_error(": couldn't map registers\n");
                return;
        }

        sc->sc_clk = fdtbus_clock_get_index(phandle, 0);
        if (sc->sc_clk == NULL) {
                aprint_error(": couldn't get clock\n");
                goto attach_failure;
        }
        if (clk_enable(sc->sc_clk) != 0) {
                aprint_error(": couldn't enable clock\n");
                goto attach_failure;
        }

        sc->sc_reset = fdtbus_reset_get_index(phandle, 0);
        sc->sc_supply = fdtbus_regulator_acquire(phandle, "phy-supply");
        if (sc->sc_supply != NULL)
                fdtbus_regulator_enable(sc->sc_supply);

        aprint_naive("\n");
        aprint_normal(": USB3 PCIe PHY\n");

        fdtbus_register_phy_controller(self, phandle,
            &mesong12_usb3pciephy_funcs);
        return;

 attach_failure:
        bus_space_unmap(sc->sc_bst, sc->sc_bsh, size);
        return;
}

CFATTACH_DECL_NEW(mesong12_usb3pciephy,
    sizeof(struct mesong12_usb3pciephy_softc),
    mesong12_usb3pciephy_match, mesong12_usb3pciephy_attach, NULL, NULL);