/*
 * Copyright (C) 2006-2009 Freescale Semiconductor, Inc.
 *
 * Dave Liu <daveliu@freescale.com>
 * based on source code of Shlomi Gridish
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of
 * the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
 * MA 02111-1307 USA
 */

#include "common.h"
#include <command.h>
#include "asm/errno.h"
#include "asm/io.h"
#include "asm/immap_qe.h"
#include "qe.h"

qe_map_t                *qe_immr = NULL;
static qe_snum_t        snums[QE_NUM_OF_SNUM];

DECLARE_GLOBAL_DATA_PTR;

void qe_issue_cmd(uint cmd, uint sbc, u8 mcn, u32 cmd_data)
{
        u32 cecr;

        if (cmd == QE_RESET) {
                out_be32(&qe_immr->cp.cecr,(u32) (cmd | QE_CR_FLG));
        } else {
                out_be32(&qe_immr->cp.cecdr, cmd_data);
                out_be32(&qe_immr->cp.cecr, (sbc | QE_CR_FLG |
                         ((u32) mcn<<QE_CR_PROTOCOL_SHIFT) | cmd));
        }
        /* Wait for the QE_CR_FLG to clear */
        do {
                cecr = in_be32(&qe_immr->cp.cecr);
        } while (cecr & QE_CR_FLG);

        return;
}

uint qe_muram_alloc(uint size, uint align)
{
        uint    retloc;
        uint    align_mask, off;
        uint    savebase;

        align_mask = align - 1;
        savebase = gd->mp_alloc_base;

        if ((off = (gd->mp_alloc_base & align_mask)) != 0)
                gd->mp_alloc_base += (align - off);

        if ((off = size & align_mask) != 0)
                size += (align - off);

        if ((gd->mp_alloc_base + size) >= gd->mp_alloc_top) {
                gd->mp_alloc_base = savebase;
                printf("%s: ran out of ram.\n",  __FUNCTION__);
        }

        retloc = gd->mp_alloc_base;
        gd->mp_alloc_base += size;

        memset((void *)&qe_immr->muram[retloc], 0, size);

        __asm__ __volatile__("sync");

        return retloc;
}

void *qe_muram_addr(uint offset)
{
        return (void *)&qe_immr->muram[offset];
}

static void qe_sdma_init(void)
{
        volatile sdma_t *p;
        uint            sdma_buffer_base;

        p = (volatile sdma_t *)&qe_immr->sdma;

        /* All of DMA transaction in bus 1 */
        out_be32(&p->sdaqr, 0);
        out_be32(&p->sdaqmr, 0);

        /* Allocate 2KB temporary buffer for sdma */
        sdma_buffer_base = qe_muram_alloc(2048, 4096);
        out_be32(&p->sdwbcr, sdma_buffer_base & QE_SDEBCR_BA_MASK);

        /* Clear sdma status */
        out_be32(&p->sdsr, 0x03000000);

        /* Enable global mode on bus 1, and 2KB buffer size */
        out_be32(&p->sdmr, QE_SDMR_GLB_1_MSK | (0x3 << QE_SDMR_CEN_SHIFT));
}

/* This table is a list of the serial numbers of the Threads, taken from the
 * "SNUM Table" chart in the QE Reference Manual. The order is not important,
 * we just need to know what the SNUMs are for the threads.
 */
static u8 thread_snum[] = {
        0x04, 0x05, 0x0c, 0x0d,
        0x14, 0x15, 0x1c, 0x1d,
        0x24, 0x25, 0x2c, 0x2d,
        0x34, 0x35, 0x88, 0x89,
        0x98, 0x99, 0xa8, 0xa9,
        0xb8, 0xb9, 0xc8, 0xc9,
        0xd8, 0xd9, 0xe8, 0xe9,
        0x08, 0x09, 0x18, 0x19,
        0x28, 0x29, 0x38, 0x39,
        0x48, 0x49, 0x58, 0x59,
        0x68, 0x69, 0x78, 0x79,
        0x80, 0x81
};

static void qe_snums_init(void)
{
        int     i;

        for (i = 0; i < QE_NUM_OF_SNUM; i++) {
                snums[i].state = QE_SNUM_STATE_FREE;
                snums[i].num   = thread_snum[i];
        }
}

int qe_get_snum(void)
{
        int     snum = -EBUSY;
        int     i;

        for (i = 0; i < QE_NUM_OF_SNUM; i++) {
                if (snums[i].state == QE_SNUM_STATE_FREE) {
                        snums[i].state = QE_SNUM_STATE_USED;
                        snum = snums[i].num;
                        break;
                }
        }

        return snum;
}

void qe_put_snum(u8 snum)
{
        int     i;

        for (i = 0; i < QE_NUM_OF_SNUM; i++) {
                if (snums[i].num == snum) {
                        snums[i].state = QE_SNUM_STATE_FREE;
                        break;
                }
        }
}

void qe_init(uint qe_base)
{
        /* Init the QE IMMR base */
        qe_immr = (qe_map_t *)qe_base;

#ifdef CONFIG_SYS_QE_FMAN_FW_IN_NOR
        /*
         * Upload microcode to IRAM for those SOCs which do not have ROM in QE.
         */
        qe_upload_firmware((const void *)CONFIG_SYS_QE_FMAN_FW_ADDR);

        /* enable the microcode in IRAM */
        out_be32(&qe_immr->iram.iready,QE_IRAM_READY);
#endif

        gd->mp_alloc_base = QE_DATAONLY_BASE;
        gd->mp_alloc_top = gd->mp_alloc_base + QE_DATAONLY_SIZE;

        qe_sdma_init();
        qe_snums_init();
}

void qe_reset(void)
{
        qe_issue_cmd(QE_RESET, QE_CR_SUBBLOCK_INVALID,
                         (u8) QE_CR_PROTOCOL_UNSPECIFIED, 0);
}

void qe_assign_page(uint snum, uint para_ram_base)
{
        u32     cecr;

        out_be32(&qe_immr->cp.cecdr, para_ram_base);
        out_be32(&qe_immr->cp.cecr, ((u32) snum<<QE_CR_ASSIGN_PAGE_SNUM_SHIFT)
                                         | QE_CR_FLG | QE_ASSIGN_PAGE);

        /* Wait for the QE_CR_FLG to clear */
        do {
                cecr = in_be32(&qe_immr->cp.cecr);
        } while (cecr & QE_CR_FLG );

        return;
}

/*
 * brg: 0~15 as BRG1~BRG16
   rate: baud rate
 * BRG input clock comes from the BRGCLK (internal clock generated from
   the QE clock, it is one-half of the QE clock), If need the clock source
   from CLKn pin, we have te change the function.
 */

#define BRG_CLK         (gd->brg_clk)

int qe_set_brg(uint brg, uint rate)
{
        volatile uint   *bp;
        u32             divisor;
        int             div16 = 0;

        if (brg >= QE_NUM_OF_BRGS)
                return -EINVAL;
        bp = (uint *)&qe_immr->brg.brgc1;
        bp += brg;

        divisor = (BRG_CLK / rate);
        if (divisor > QE_BRGC_DIVISOR_MAX + 1) {
                div16 = 1;
                divisor /= 16;
        }

        *bp = ((divisor - 1) << QE_BRGC_DIVISOR_SHIFT) | QE_BRGC_ENABLE;
        __asm__ __volatile__("sync");

        if (div16) {
                *bp |= QE_BRGC_DIV16;
                __asm__ __volatile__("sync");
        }

        return 0;
}

/* Set ethernet MII clock master
*/
int qe_set_mii_clk_src(int ucc_num)
{
        u32     cmxgcr;

        /* check if the UCC number is in range. */
        if ((ucc_num > UCC_MAX_NUM - 1) || (ucc_num < 0)) {
                printf("%s: ucc num not in ranges\n", __FUNCTION__);
                return -EINVAL;
        }

        cmxgcr = in_be32(&qe_immr->qmx.cmxgcr);
        cmxgcr &= ~QE_CMXGCR_MII_ENET_MNG_MASK;
        cmxgcr |= (ucc_num <<QE_CMXGCR_MII_ENET_MNG_SHIFT);
        out_be32(&qe_immr->qmx.cmxgcr, cmxgcr);

        return 0;
}

/* Firmware information stored here for qe_get_firmware_info() */
static struct qe_firmware_info qe_firmware_info;

/*
 * Set to 1 if QE firmware has been uploaded, and therefore
 * qe_firmware_info contains valid data.
 */
static int qe_firmware_uploaded;

/*
 * Upload a QE microcode
 *
 * This function is a worker function for qe_upload_firmware().  It does
 * the actual uploading of the microcode.
 */
static void qe_upload_microcode(const void *base,
        const struct qe_microcode *ucode)
{
        const u32 *code = base + be32_to_cpu(ucode->code_offset);
        unsigned int i;

        if (ucode->major || ucode->minor || ucode->revision)
                printf("QE: uploading microcode '%s' version %u.%u.%u\n",
                        ucode->id, ucode->major, ucode->minor, ucode->revision);
        else
                printf("QE: uploading microcode '%s'\n", ucode->id);

        /* Use auto-increment */
        out_be32(&qe_immr->iram.iadd, be32_to_cpu(ucode->iram_offset) |
                QE_IRAM_IADD_AIE | QE_IRAM_IADD_BADDR);

        for (i = 0; i < be32_to_cpu(ucode->count); i++)
                out_be32(&qe_immr->iram.idata, be32_to_cpu(code[i]));
}

/*
 * Upload a microcode to the I-RAM at a specific address.
 *
 * See docs/README.qe_firmware for information on QE microcode uploading.
 *
 * Currently, only version 1 is supported, so the 'version' field must be
 * set to 1.
 *
 * The SOC model and revision are not validated, they are only displayed for
 * informational purposes.
 *
 * 'calc_size' is the calculated size, in bytes, of the firmware structure and
 * all of the microcode structures, minus the CRC.
 *
 * 'length' is the size that the structure says it is, including the CRC.
 */
int qe_upload_firmware(const struct qe_firmware *firmware)
{
        unsigned int i;
        unsigned int j;
        u32 crc;
        size_t calc_size = sizeof(struct qe_firmware);
        size_t length;
        const struct qe_header *hdr;

        if (!firmware) {
                printf("Invalid address\n");
                return -EINVAL;
        }

        hdr = &firmware->header;
        length = be32_to_cpu(hdr->length);

        /* Check the magic */
        if ((hdr->magic[0] != 'Q') || (hdr->magic[1] != 'E') ||
            (hdr->magic[2] != 'F')) {
                printf("Not a microcode\n");
                return -EPERM;
        }

        /* Check the version */
        if (hdr->version != 1) {
                printf("Unsupported version\n");
                return -EPERM;
        }

        /* Validate some of the fields */
        if ((firmware->count < 1) || (firmware->count > MAX_QE_RISC)) {
                printf("Invalid data\n");
                return -EINVAL;
        }

        /* Validate the length and check if there's a CRC */
        calc_size += (firmware->count - 1) * sizeof(struct qe_microcode);

        for (i = 0; i < firmware->count; i++)
                /*
                 * For situations where the second RISC uses the same microcode
                 * as the first, the 'code_offset' and 'count' fields will be
                 * zero, so it's okay to add those.
                 */
                calc_size += sizeof(u32) *
                        be32_to_cpu(firmware->microcode[i].count);

        /* Validate the length */
        if (length != calc_size + sizeof(u32)) {
                printf("Invalid length\n");
                return -EPERM;
        }

        /*
         * Validate the CRC.  We would normally call crc32_no_comp(), but that
         * function isn't available unless you turn on JFFS support.
         */
        crc = be32_to_cpu(*(u32 *)((void *)firmware + calc_size));
        if (crc != (crc32(-1, (const void *) firmware, calc_size) ^ -1)) {
                printf("Firmware CRC is invalid\n");
                return -EIO;
        }

        /*
         * If the microcode calls for it, split the I-RAM.
         */
        if (!firmware->split) {
                out_be16(&qe_immr->cp.cercr,
                        in_be16(&qe_immr->cp.cercr) | QE_CP_CERCR_CIR);
        }

        if (firmware->soc.model)
                printf("Firmware '%s' for %u V%u.%u\n",
                        firmware->id, be16_to_cpu(firmware->soc.model),
                        firmware->soc.major, firmware->soc.minor);
        else
                printf("Firmware '%s'\n", firmware->id);

        /*
         * The QE only supports one microcode per RISC, so clear out all the
         * saved microcode information and put in the new.
         */
        memset(&qe_firmware_info, 0, sizeof(qe_firmware_info));
        strcpy(qe_firmware_info.id, (char *)firmware->id);
        qe_firmware_info.extended_modes = firmware->extended_modes;
        memcpy(qe_firmware_info.vtraps, firmware->vtraps,
                sizeof(firmware->vtraps));
        qe_firmware_uploaded = 1;

        /* Loop through each microcode. */
        for (i = 0; i < firmware->count; i++) {
                const struct qe_microcode *ucode = &firmware->microcode[i];

                /* Upload a microcode if it's present */
                if (ucode->code_offset)
                        qe_upload_microcode(firmware, ucode);

                /* Program the traps for this processor */
                for (j = 0; j < 16; j++) {
                        u32 trap = be32_to_cpu(ucode->traps[j]);

                        if (trap)
                                out_be32(&qe_immr->rsp[i].tibcr[j], trap);
                }

                /* Enable traps */
                out_be32(&qe_immr->rsp[i].eccr, be32_to_cpu(ucode->eccr));
        }

        return 0;
}

struct qe_firmware_info *qe_get_firmware_info(void)
{
        return qe_firmware_uploaded ? &qe_firmware_info : NULL;
}

static int qe_cmd(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
        ulong addr;

        if (argc < 3)
                return cmd_usage(cmdtp);

        if (strcmp(argv[1], "fw") == 0) {
                addr = simple_strtoul(argv[2], NULL, 16);

                if (!addr) {
                        printf("Invalid address\n");
                        return -EINVAL;
                }

                /*
                 * If a length was supplied, compare that with the 'length'
                 * field.
                 */

                if (argc > 3) {
                        ulong length = simple_strtoul(argv[3], NULL, 16);
                        struct qe_firmware *firmware = (void *) addr;

                        if (length != be32_to_cpu(firmware->header.length)) {
                                printf("Length mismatch\n");
                                return -EINVAL;
                        }
                }

                return qe_upload_firmware((const struct qe_firmware *) addr);
        }

        return cmd_usage(cmdtp);
}

U_BOOT_CMD(
        qe, 4, 0, qe_cmd,
        "QUICC Engine commands",
        "fw <addr> [<length>] - Upload firmware binary at address <addr> to "
                "the QE,\n"
        "\twith optional length <length> verification."
);