/*
   * Copyright (C) 2006-2009 Freescale Semiconductor, Inc.
   *
   * Dave Liu <daveliu@freescale.com>
   * based on source code of Shlomi Gridish
   *
   * SPDX-License-Identifier:	GPL-2.0+
   */
  
  #include "common.h"
  #include <command.h>
  #include "asm/errno.h"
  #include "asm/io.h"
  #include "linux/immap_qe.h"
  #include "qe.h"
  #ifdef CONFIG_LS102XA
  #include <asm/arch/immap_ls102xa.h>
  #endif
  
  #define MPC85xx_DEVDISR_QE_DISABLE	0x1
  
  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;
  }
  
  #ifdef CONFIG_QE
  uint qe_muram_alloc(uint size, uint align)
  {
  	uint	retloc;
  	uint	align_mask, off;
  	uint	savebase;
  
  	align_mask = align - 1;
  	savebase = gd->arch.mp_alloc_base;
  
  	off = gd->arch.mp_alloc_base & align_mask;
  	if (off != 0)
  		gd->arch.mp_alloc_base += (align - off);
  
  	if ((off = size & align_mask) != 0)
  		size += (align - off);
  
  	if ((gd->arch.mp_alloc_base + size) >= gd->arch.mp_alloc_top) {
  		gd->arch.mp_alloc_base = savebase;
  		printf("%s: ran out of ram.
  ",  __FUNCTION__);
  	}
  
  	retloc = gd->arch.mp_alloc_base;
  	gd->arch.mp_alloc_base += size;
  
  	memset((void *)&qe_immr->muram[retloc], 0, size);
  
  	__asm__ __volatile__("sync");
  
  	return retloc;
  }
  #endif
  
  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[] = {
  /* Evthreads 16-29 are not supported in MPC8309 */
  #if !defined(CONFIG_MPC8309)
  	0x04, 0x05, 0x0c, 0x0d,
  	0x14, 0x15, 0x1c, 0x1d,
  	0x24, 0x25, 0x2c, 0x2d,
  	0x34, 0x35,
  #endif
  	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_FW_ADDR);
  
  	/* enable the microcode in IRAM */
  	out_be32(&qe_immr->iram.iready,QE_IRAM_READY);
  #endif
  
  	gd->arch.mp_alloc_base = QE_DATAONLY_BASE;
  	gd->arch.mp_alloc_top = gd->arch.mp_alloc_base + QE_DATAONLY_SIZE;
  
  	qe_sdma_init();
  	qe_snums_init();
  }
  
  #ifdef CONFIG_U_QE
  void u_qe_init(void)
  {
  	uint qe_base = CONFIG_SYS_IMMR + 0x01400000; /* QE immr base */
  	qe_immr = (qe_map_t *)qe_base;
  
  	u_qe_upload_firmware((const void *)CONFIG_SYS_QE_FW_ADDR);
  	out_be32(&qe_immr->iram.iready, QE_IRAM_READY);
  }
  #endif
  
  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->arch.brg_clk)
  
  #ifdef CONFIG_QE
  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;
  }
  #endif
  
  /* 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
  ", __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
  ",
  			ucode->id, ucode->major, ucode->minor, ucode->revision);
  	else
  		printf("QE: uploading microcode '%s'
  ", 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;
  #ifdef CONFIG_DEEP_SLEEP
  #ifdef CONFIG_LS102XA
  	struct ccsr_gur __iomem *gur = (void *)CONFIG_SYS_FSL_GUTS_ADDR;
  #else
  	ccsr_gur_t *gur = (void *)(CONFIG_SYS_MPC85xx_GUTS_ADDR);
  #endif
  #endif
  	if (!firmware) {
  		printf("Invalid address
  ");
  		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("QE microcode not found
  ");
  #ifdef CONFIG_DEEP_SLEEP
  		setbits_be32(&gur->devdisr, MPC85xx_DEVDISR_QE_DISABLE);
  #endif
  		return -EPERM;
  	}
  
  	/* Check the version */
  	if (hdr->version != 1) {
  		printf("Unsupported version
  ");
  		return -EPERM;
  	}
  
  	/* Validate some of the fields */
  	if ((firmware->count < 1) || (firmware->count > MAX_QE_RISC)) {
  		printf("Invalid data
  ");
  		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
  ");
  		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
  ");
  		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
  ",
  			firmware->id, be16_to_cpu(firmware->soc.model),
  			firmware->soc.major, firmware->soc.minor);
  	else
  		printf("Firmware '%s'
  ", 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;
  }
  
  #ifdef CONFIG_U_QE
  /*
   * 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 u_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;
  #ifdef CONFIG_DEEP_SLEEP
  #ifdef CONFIG_LS102XA
  	struct ccsr_gur __iomem *gur = (void *)CONFIG_SYS_FSL_GUTS_ADDR;
  #else
  	ccsr_gur_t __iomem *gur = (void *)(CONFIG_SYS_MPC85xx_GUTS_ADDR);
  #endif
  #endif
  	if (!firmware) {
  		printf("Invalid address
  ");
  		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
  ");
  #ifdef CONFIG_DEEP_SLEEP
  		setbits_be32(&gur->devdisr, MPC85xx_DEVDISR_QE_DISABLE);
  #endif
  		return -EPERM;
  	}
  
  	/* Check the version */
  	if (hdr->version != 1) {
  		printf("Unsupported version
  ");
  		return -EPERM;
  	}
  
  	/* Validate some of the fields */
  	if ((firmware->count < 1) || (firmware->count > MAX_QE_RISC)) {
  		printf("Invalid data
  ");
  		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
  ");
  		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
  ");
  		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
  ",
  		       firmware->id, be16_to_cpu(firmware->soc.model),
  		       firmware->soc.major, firmware->soc.minor);
  	else
  		printf("Firmware '%s'
  ", firmware->id);
  
  	/* 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;
  }
  #endif
  
  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
  ");
  			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
  ");
  				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,
  "
  	"\twith optional length <length> verification."
  );