/*
 * Driver for NAND support, Rick Bronson
 * borrowed heavily from:
 * (c) 1999 Machine Vision Holdings, Inc.
 * (c) 1999, 2000 David Woodhouse <dwmw2@infradead.org>
 *
 * Ported 'dynenv' to 'nand env.oob' command
 * (C) 2010 Nanometrics, Inc.
 * 'dynenv' -- Dynamic environment offset in NAND OOB
 * (C) Copyright 2006-2007 OpenMoko, Inc.
 * Added 16-bit nand support
 * (C) 2004 Texas Instruments
 *
 * Copyright 2010 Freescale Semiconductor
 * The portions of this file whose copyright is held by Freescale and which
 * are not considered a derived work of GPL v2-only code may be distributed
 * and/or modified 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.
 */

#include <common.h>
#include <command.h>
#include <fastboot.h>
#include <sys_partition.h>

#define  SUNXI_FLASH_READ_FIRST_SIZE      (32 * 1024)

static int sunxi_flash_read_all(u32 start, ulong buf, const char *part_name)
{
	int ret;
	u32 rbytes, rblock;
	u32 start_block = start;
	void *addr;
	struct fastboot_boot_img_hdr *fb_hdr;

	addr = (void *)buf;
	ret = sunxi_flash_read(start_block, SUNXI_FLASH_READ_FIRST_SIZE/512, addr);
	if(!ret)
	{
		printf("read all error: start=%x, addr=%x\n", start_block, (u32)addr);

		return 1;
	}
	fb_hdr = (struct fastboot_boot_img_hdr *)addr;
	if (memcmp(fb_hdr->magic, FASTBOOT_BOOT_MAGIC, 8))
	{
		printf("boota: bad boot image magic, maybe not a boot.img?\n");
		printf("try to read all\n");
		debug("part name=%s\n", part_name);
		rbytes = sunxi_partition_get_size_byname(part_name) * 512;
	}
	else
	{
		rbytes = fb_hdr->kernel_size + fb_hdr->ramdisk_size + fb_hdr->second_size + 1024 * 1024 + 511;
	}
	rblock = rbytes/512 - SUNXI_FLASH_READ_FIRST_SIZE/512;
	debug("rblock=%d, start=%d\n", rblock, start_block);
	start_block += SUNXI_FLASH_READ_FIRST_SIZE/512;
	addr = (void *)(buf + SUNXI_FLASH_READ_FIRST_SIZE);

	ret = sunxi_flash_read(start_block, rblock, addr);

	tick_printf("sunxi flash read :offset %x, %d bytes %s\n", start<<9, rbytes,
		       ret ? "OK" : "ERROR");

	return ret == 0 ? 1 : 0;

}

int do_sunxi_flash(cmd_tbl_t * cmdtp, int flag, int argc, char * const argv[])
{
	int ret = 0;
	ulong addr;
	char *cmd;
	char *part_name;
	u32 start_block;
	u32 rblock;
	int readall_flag = 0;

	/* at least four arguments please */
	if ((argc != 4) && (argc != 5))
		goto usage;

	cmd = argv[1];
	part_name = argv[3];

	if (strncmp(cmd, "read", 4) == 0)
	{

		addr = (ulong)simple_strtoul(argv[2], NULL, 16);

		if((!strncmp(part_name, "boot", 4)) || (!strncmp(part_name, "recovery", 8)))
		{
			readall_flag = 1;
		}
		start_block = sunxi_partition_get_offset_byname((const char *)part_name);
		if(!start_block)
		{
			printf("cant find part named %s\n", (char *)part_name);

			goto usage;
		}
		if(argc == 4)
		{
			if(readall_flag)
			{
				puts("read boot or recovery all\n");

				return sunxi_flash_read_all(start_block, addr, (const char *)part_name);
			}
			rblock = sunxi_partition_get_size_byname((const char *)part_name);
		}
		else
		{
			rblock = (u32)simple_strtoul(argv[4], NULL, 16)/512;
		}
#if DEBUG
		printf("part name   = %s\n", part_name);
		printf("start block = %x\n", start_block);
		printf("     nblock = %x\n", rblock);
#endif
		ret = sunxi_flash_read(start_block, rblock, (void *)addr);

		tick_printf("sunxi flash read :offset %x, %d bytes %s\n", start_block<<9, rblock<<9,
		       ret ? "OK" : "ERROR");

		return ret == 0 ? 1 : 0;
	}
	else if(strncmp(cmd, "log_read", strlen("log_read")) == 0)
	{

        printf("read logical\n");

		addr = (ulong)simple_strtoul(argv[2], NULL, 16);
		start_block = (ulong)simple_strtoul(argv[3], NULL, 16);
		rblock = (ulong)simple_strtoul(argv[4], NULL, 16);

		ret = sunxi_flash_read(start_block, rblock, (void *)addr);

		tick_printf("sunxi flash log_read :offset %x, %d sectors %s\n", start_block, rblock,
		       ret ? "OK" : "ERROR");

		return ret == 0 ? 1 : 0;
	}
	else if(strncmp(cmd, "phy_read", strlen("phy_read")) == 0)
	{

        printf("read physical\n");

		addr = (ulong)simple_strtoul(argv[2], NULL, 16);
		start_block = (ulong)simple_strtoul(argv[3], NULL, 16);
		rblock = (ulong)simple_strtoul(argv[4], NULL, 16);

		ret = sunxi_flash_phyread(start_block, rblock, (void *)addr);

		tick_printf("sunxi flash phy_read :offset %x, %d sectors %s\n", start_block, rblock,
		       ret ? "OK" : "ERROR");

		return ret == 0 ? 1 : 0;
	}
	else if (strncmp(cmd, "write", 4) == 0)
	{
		addr = (ulong)simple_strtoul(argv[2], NULL, 16);
		/* write size: indecated on partemeter 1 */
		if (argc == 4) {
			start_block = sunxi_partition_get_offset_byname((const char *)part_name);
			rblock = sunxi_partition_get_size_byname((const char *)part_name);
		} else {
			/* write size: partemeter 2 */
			start_block = (u32)simple_strtoul(argv[3], NULL, 16)/512;
			rblock = (u32)simple_strtoul(argv[4], NULL, 16)/512;
		}
#if DEBUG
		printf("part name   = %s\n", part_name);
		printf("start block = %x\n", start_block);
		printf("     nblock = %x\n", rblock);
#endif
		ret = sunxi_flash_write(start_block, rblock, (void *)addr);

		tick_printf("sunxi flash write :offset %x, %d bytes %s\n", start_block<<9, rblock<<9,
		            ret ? "OK" : "ERROR");

		return ret == 0 ? 1 : 0;
	}

usage:
	return cmd_usage(cmdtp);
}

U_BOOT_CMD(
	sunxi_flash, CONFIG_SYS_MAXARGS, 1, do_sunxi_flash,
	"sunxi_flash sub-system",
	"read command parmeters : \n"
	"parmeters 0 : addr to load(hex only)\n"
	"parmeters 1 : the name of the part to be load\n"
	"[parmeters 2] : the number of bytes to be load(hex only)\n"
	"if [parmeters 2] not exist, the number of bytes to be load "
	"is the size of the part indecated on partemeter 1"
        "\nwrite command parmeters : \n"
        "parmeters 0 : addr to save(hex only)\n"
        "parmeters 1 : the name of the part to be write or the flash offset\n"
        "[parmeters 2] : the number of bytes to be write(hex only)\n"
        "if [parmeters 2] not exist, the number of bytes to be write "
        "is the size of the part indecated on partemeter 1"
);