/*
 * 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>


static int sunxi_flash_read_all(u32 start, ulong buf)
{
	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, 4 * 1024 * 1024/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;
	rbytes = fb_hdr->kernel_size + fb_hdr->ramdisk_size + fb_hdr->second_size + 1024 * 1024;
	if(rbytes <= 4 * 1024 * 1024)
	{
		return 0;
	}
	rbytes += 511;
	rblock = (rbytes/512) - (4 * 1024 * 1024/512);
	start_block += 4 * 1024 * 1024/512;
	addr = (void *)(buf + 4 * 1024 * 1024);

	ret = sunxi_flash_read(start_block, rblock, addr);

	printf("sunxi flash read :offset %x, %d bytes %s\n", start<<9, rbytes - 511,
		       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;

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

	cmd = argv[1];
	part_name = argv[3];
/*
************************************************
*************  read only   *********************
************************************************
*/

	if (strncmp(cmd, "read", 4) == 0) {
		u32 start_block;
		u32 rblock;
		int readall_flag = 0;

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

		if((!strncmp(part_name, "boot", 4)) || (!strncmp(part_name, "recovery", 4)))
		{
			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);
			}
			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);

		printf("sunxi flash read :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"
);