/*
 * (C) Copyright 2017 Theobroma Systems Design und Consulting GmbH
 *
 * SPDX-License-Identifier:     GPL-2.0+
 */

#include <common.h>
#include <dm.h>
#include <misc.h>
#include <spl.h>
#include <syscon.h>
#include <usb.h>
#include <dm/pinctrl.h>
#include <dm/uclass-internal.h>
#include <asm/io.h>
#include <asm/gpio.h>
#include <asm/setup.h>
#include <asm/arch/clock.h>
#include <asm/arch/cru_rk3399.h>
#include <asm/arch/hardware.h>
#include <asm/arch/grf_rk3399.h>
#include <asm/arch/periph.h>
#include <power/regulator.h>
#include <u-boot/sha256.h>

DECLARE_GLOBAL_DATA_PTR;

int board_init(void)
{
        int ret;

        /*
         * We need to call into regulators_enable_boot_on() again, as the call
         * during SPL may have not included all regulators.
         */
        ret = regulators_enable_boot_on(false);
        if (ret)
                debug("%s: Cannot enable boot on regulator\n", __func__);

        return 0;
}

static void rk3399_force_power_on_reset(void)
{
        ofnode node;
        struct gpio_desc sysreset_gpio;

        debug("%s: trying to force a power-on reset\n", __func__);

        node = ofnode_path("/config");
        if (!ofnode_valid(node)) {
                debug("%s: no /config node?\n", __func__);
                return;
        }

        if (gpio_request_by_name_nodev(node, "sysreset-gpio", 0,
                                       &sysreset_gpio, GPIOD_IS_OUT)) {
                debug("%s: could not find a /config/sysreset-gpio\n", __func__);
                return;
        }

        dm_gpio_set_value(&sysreset_gpio, 1);
}

void spl_board_init(void)
{
        int  ret;
        struct rk3399_cru *cru = rockchip_get_cru();

        /*
         * The RK3399 resets only 'almost all logic' (see also in the TRM
         * "3.9.4 Global software reset"), when issuing a software reset.
         * This may cause issues during boot-up for some configurations of
         * the application software stack.
         *
         * To work around this, we test whether the last reset reason was
         * a power-on reset and (if not) issue an overtemp-reset to reset
         * the entire module.
         *
         * While this was previously fixed by modifying the various places
         * that could generate a software reset (e.g. U-Boot's sysreset
         * driver, the ATF or Linux), we now have it here to ensure that
         * we no longer have to track this through the various components.
         */
        if (cru->glb_rst_st != 0)
                rk3399_force_power_on_reset();

        /*
         * Turning the eMMC and SPI back on (if disabled via the Qseven
         * BIOS_ENABLE) signal is done through a always-on regulator).
         */
        ret = regulators_enable_boot_on(false);
        if (ret)
                debug("%s: Cannot enable boot on regulator\n", __func__);

        preloader_console_init();
}

static void setup_macaddr(void)
{
#if CONFIG_IS_ENABLED(CMD_NET)
        int ret;
        const char *cpuid = env_get("cpuid#");
        u8 hash[SHA256_SUM_LEN];
        int size = sizeof(hash);
        u8 mac_addr[6];

        /* Only generate a MAC address, if none is set in the environment */
        if (env_get("ethaddr"))
                return;

        if (!cpuid) {
                debug("%s: could not retrieve 'cpuid#'\n", __func__);
                return;
        }

        ret = hash_block("sha256", (void *)cpuid, strlen(cpuid), hash, &size);
        if (ret) {
                debug("%s: failed to calculate SHA256\n", __func__);
                return;
        }

        /* Copy 6 bytes of the hash to base the MAC address on */
        memcpy(mac_addr, hash, 6);

        /* Make this a valid MAC address and set it */
        mac_addr[0] &= 0xfe;  /* clear multicast bit */
        mac_addr[0] |= 0x02;  /* set local assignment bit (IEEE802) */
        eth_env_set_enetaddr("ethaddr", mac_addr);
#endif
}

static void setup_serial(void)
{
#if CONFIG_IS_ENABLED(ROCKCHIP_EFUSE)
        const u32 cpuid_offset = 0x7;
        const u32 cpuid_length = 0x10;

        struct udevice *dev;
        int ret, i;
        u8 cpuid[cpuid_length];
        u8 low[cpuid_length/2], high[cpuid_length/2];
        char cpuid_str[cpuid_length * 2 + 1];
        u64 serialno;
        char serialno_str[17];

        /* retrieve the device */
        ret = uclass_get_device_by_driver(UCLASS_MISC,
                                          DM_GET_DRIVER(rockchip_efuse), &dev);
        if (ret) {
                debug("%s: could not find efuse device\n", __func__);
                return;
        }

        /* read the cpu_id range from the efuses */
        ret = misc_read(dev, cpuid_offset, &cpuid, sizeof(cpuid));
        if (ret) {
                debug("%s: reading cpuid from the efuses failed\n",
                      __func__);
                return;
        }

        memset(cpuid_str, 0, sizeof(cpuid_str));
        for (i = 0; i < 16; i++)
                sprintf(&cpuid_str[i * 2], "%02x", cpuid[i]);

        debug("cpuid: %s\n", cpuid_str);

        /*
         * Mix the cpuid bytes using the same rules as in
         *   ${linux}/drivers/soc/rockchip/rockchip-cpuinfo.c
         */
        for (i = 0; i < 8; i++) {
                low[i] = cpuid[1 + (i << 1)];
                high[i] = cpuid[i << 1];
        }

        serialno = crc32_no_comp(0, low, 8);
        serialno |= (u64)crc32_no_comp(serialno, high, 8) << 32;
        snprintf(serialno_str, sizeof(serialno_str), "%016llx", serialno);

        env_set("cpuid#", cpuid_str);
        env_set("serial#", serialno_str);
#endif
}

static void setup_iodomain(void)
{
        const u32 GRF_IO_VSEL_GPIO4CD_SHIFT = 3;
        struct rk3399_grf_regs *grf =
            syscon_get_first_range(ROCKCHIP_SYSCON_GRF);

        /*
         * Set bit 3 in GRF_IO_VSEL so PCIE_RST# works (pin GPIO4_C6).
         * Linux assumes that PCIE_RST# works out of the box as it probes
         * PCIe before loading the iodomain driver.
         */
        rk_setreg(&grf->io_vsel, 1 << GRF_IO_VSEL_GPIO4CD_SHIFT);
}

int misc_init_r(void)
{
        setup_serial();
        setup_macaddr();
        setup_iodomain();

        return 0;
}

#ifdef CONFIG_SERIAL_TAG
void get_board_serial(struct tag_serialnr *serialnr)
{
        char *serial_string;
        u64 serial = 0;

        serial_string = env_get("serial#");

        if (serial_string)
                serial = simple_strtoull(serial_string, NULL, 16);

        serialnr->high = (u32)(serial >> 32);
        serialnr->low = (u32)(serial & 0xffffffff);
}
#endif

/**
 * Switch power at an external regulator (for our root hub).
 *
 * @param ctrl pointer to the xHCI controller
 * @param port port number as in the control message (one-based)
 * @param enable boolean indicating whether to enable or disable power
 * @return returns 0 on success, an error-code on failure
 */
static int board_usb_port_power_set(struct udevice *dev, int port,
                                    bool enable)
{
#if CONFIG_IS_ENABLED(OF_CONTROL) && CONFIG_IS_ENABLED(DM_REGULATOR)
        /* We start counting ports at 0, while USB counts from 1. */
        int index = port - 1;
        const char *regname = NULL;
        struct udevice *regulator;
        const char *prop = "tsd,usb-port-power";
        int ret;

        debug("%s: ctrl '%s' port %d enable %s\n", __func__,
              dev_read_name(dev), port, enable ? "true" : "false");

        ret = dev_read_string_index(dev, prop, index, &regname);
        if (ret < 0) {
                debug("%s: ctrl '%s' port %d: no entry in '%s'\n",
                      __func__, dev_read_name(dev), port, prop);
                return ret;
        }

        ret = regulator_get_by_platname(regname, &regulator);
        if (ret) {
                debug("%s: ctrl '%s' port %d: could not get regulator '%s'\n",
                      __func__, dev_read_name(dev), port, regname);
                return ret;
        }

        regulator_set_enable(regulator, enable);
        return 0;
#else
        return -ENOTSUPP;
#endif
}

void usb_hub_reset_devices(struct usb_hub_device *hub, int port)
{
        struct udevice *dev = hub->pusb_dev->dev;
        struct udevice *ctrl;

        /* We are only interested in our root-hubs */
        if (usb_hub_is_root_hub(dev) == false)
                return;

        ctrl = usb_get_bus(dev);
        if (!ctrl) {
                debug("%s: could not retrieve ctrl for hub\n", __func__);
                return;
        }

        /*
         * To work around an incompatibility between the single-threaded
         * USB stack in U-Boot and (a strange low-power mode of) the USB
         * hub we have on-module, we need to delay powering on the hub
         * until the first time the port is probed.
         */
        board_usb_port_power_set(ctrl, port, true);
}