/***************************************************************************
 *   Copyright (C) 2007 by Dominic Rath                                    *
 *   Dominic.Rath@gmx.de                                                   *
 *                                                                         *
 *   partially based on                                                    *
 * 	 drivers/mtd/nand_ids.c                                                *
 *                                                                         *
 *   Copyright (C) 2002 Thomas Gleixner (tglx@linutronix.de)               *
 *                                                                         *
 *   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.             *
 ***************************************************************************/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "replacements.h"
#include "log.h"

#include <stdlib.h>
#include <string.h>

#include <errno.h>

#include "nand.h"
#include "flash.h"
#include "time_support.h"
#include "fileio.h"

int nand_register_commands(struct command_context_s *cmd_ctx);
int handle_nand_list_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc);
int handle_nand_probe_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc);
int handle_nand_check_bad_blocks_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc);
int handle_nand_info_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc);
int handle_nand_copy_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc);
int handle_nand_write_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc);
int handle_nand_dump_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc);
int handle_nand_erase_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc);

int handle_nand_raw_access_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc);

int nand_read_page_raw(struct nand_device_s *device, u32 page, u8 *data, u32 data_size, u8 *oob, u32 oob_size);
int nand_read_page(struct nand_device_s *device, u32 page, u8 *data, u32 data_size, u8 *oob, u32 oob_size);
int nand_read_plain(struct nand_device_s *device, u32 address, u8 *data, u32 data_size);

int nand_write_page_raw(struct nand_device_s *device, u32 page, u8 *data, u32 data_size, u8 *oob, u32 oob_size);
int nand_write_page(struct nand_device_s *device, u32 page, u8 *data, u32 data_size, u8 *oob, u32 oob_size);

/* NAND flash controller
 */
extern nand_flash_controller_t lpc3180_nand_controller;
/* extern nand_flash_controller_t s3c2410_nand_controller; */
/* extern nand_flash_controller_t boundary_scan_nand_controller; */

nand_flash_controller_t *nand_flash_controllers[] =
{
	&lpc3180_nand_controller,
/*	&s3c2410_nand_controller, */
/*	&boundary_scan_nand_controller, */
	NULL
};

/* configured NAND devices and NAND Flash command handler */
nand_device_t *nand_devices = NULL;
static command_t *nand_cmd;

/*	Chip ID list
 *
 *	Name, ID code, pagesize, chipsize in MegaByte, eraseblock size,
 *	options
 *
 *	Pagesize; 0, 256, 512
 *	0	get this information from the extended chip ID
 *	256	256 Byte page size
 *	512	512 Byte page size
 */
nand_info_t nand_flash_ids[] =
{
	{"NAND 1MiB 5V 8-bit",		0x6e, 256, 1, 0x1000, 0},
	{"NAND 2MiB 5V 8-bit",		0x64, 256, 2, 0x1000, 0},
	{"NAND 4MiB 5V 8-bit",		0x6b, 512, 4, 0x2000, 0},
	{"NAND 1MiB 3,3V 8-bit",	0xe8, 256, 1, 0x1000, 0},
	{"NAND 1MiB 3,3V 8-bit",	0xec, 256, 1, 0x1000, 0},
	{"NAND 2MiB 3,3V 8-bit",	0xea, 256, 2, 0x1000, 0},
	{"NAND 4MiB 3,3V 8-bit",	0xd5, 512, 4, 0x2000, 0},
	{"NAND 4MiB 3,3V 8-bit",	0xe3, 512, 4, 0x2000, 0},
	{"NAND 4MiB 3,3V 8-bit",	0xe5, 512, 4, 0x2000, 0},
	{"NAND 8MiB 3,3V 8-bit",	0xd6, 512, 8, 0x2000, 0},

	{"NAND 8MiB 1,8V 8-bit",	0x39, 512, 8, 0x2000, 0},
	{"NAND 8MiB 3,3V 8-bit",	0xe6, 512, 8, 0x2000, 0},
	{"NAND 8MiB 1,8V 16-bit",	0x49, 512, 8, 0x2000, NAND_BUSWIDTH_16},
	{"NAND 8MiB 3,3V 16-bit",	0x59, 512, 8, 0x2000, NAND_BUSWIDTH_16},

	{"NAND 16MiB 1,8V 8-bit",	0x33, 512, 16, 0x4000, 0},
	{"NAND 16MiB 3,3V 8-bit",	0x73, 512, 16, 0x4000, 0},
	{"NAND 16MiB 1,8V 16-bit",	0x43, 512, 16, 0x4000, NAND_BUSWIDTH_16},
	{"NAND 16MiB 3,3V 16-bit",	0x53, 512, 16, 0x4000, NAND_BUSWIDTH_16},

	{"NAND 32MiB 1,8V 8-bit",	0x35, 512, 32, 0x4000, 0},
	{"NAND 32MiB 3,3V 8-bit",	0x75, 512, 32, 0x4000, 0},
	{"NAND 32MiB 1,8V 16-bit",	0x45, 512, 32, 0x4000, NAND_BUSWIDTH_16},
	{"NAND 32MiB 3,3V 16-bit",	0x55, 512, 32, 0x4000, NAND_BUSWIDTH_16},

	{"NAND 64MiB 1,8V 8-bit",	0x36, 512, 64, 0x4000, 0},
	{"NAND 64MiB 3,3V 8-bit",	0x76, 512, 64, 0x4000, 0},
	{"NAND 64MiB 1,8V 16-bit",	0x46, 512, 64, 0x4000, NAND_BUSWIDTH_16},
	{"NAND 64MiB 3,3V 16-bit",	0x56, 512, 64, 0x4000, NAND_BUSWIDTH_16},

	{"NAND 128MiB 1,8V 8-bit",	0x78, 512, 128, 0x4000, 0},
	{"NAND 128MiB 1,8V 8-bit",	0x39, 512, 128, 0x4000, 0},
	{"NAND 128MiB 3,3V 8-bit",	0x79, 512, 128, 0x4000, 0},
	{"NAND 128MiB 1,8V 16-bit",	0x72, 512, 128, 0x4000, NAND_BUSWIDTH_16},
	{"NAND 128MiB 1,8V 16-bit",	0x49, 512, 128, 0x4000, NAND_BUSWIDTH_16},
	{"NAND 128MiB 3,3V 16-bit",	0x74, 512, 128, 0x4000, NAND_BUSWIDTH_16},
	{"NAND 128MiB 3,3V 16-bit",	0x59, 512, 128, 0x4000, NAND_BUSWIDTH_16},

	{"NAND 256MiB 3,3V 8-bit",	0x71, 512, 256, 0x4000, 0},

	{"NAND 64MiB 1,8V 8-bit",	0xA2, 0,  64, 0, LP_OPTIONS},
	{"NAND 64MiB 3,3V 8-bit",	0xF2, 0,  64, 0, LP_OPTIONS},
	{"NAND 64MiB 1,8V 16-bit",	0xB2, 0,  64, 0, LP_OPTIONS16},
	{"NAND 64MiB 3,3V 16-bit",	0xC2, 0,  64, 0, LP_OPTIONS16},

	{"NAND 128MiB 1,8V 8-bit",	0xA1, 0, 128, 0, LP_OPTIONS},
	{"NAND 128MiB 3,3V 8-bit",	0xF1, 0, 128, 0, LP_OPTIONS},
	{"NAND 128MiB 1,8V 16-bit",	0xB1, 0, 128, 0, LP_OPTIONS16},
	{"NAND 128MiB 3,3V 16-bit",	0xC1, 0, 128, 0, LP_OPTIONS16},

	{"NAND 256MiB 1,8V 8-bit",	0xAA, 0, 256, 0, LP_OPTIONS},
	{"NAND 256MiB 3,3V 8-bit",	0xDA, 0, 256, 0, LP_OPTIONS},
	{"NAND 256MiB 1,8V 16-bit",	0xBA, 0, 256, 0, LP_OPTIONS16},
	{"NAND 256MiB 3,3V 16-bit",	0xCA, 0, 256, 0, LP_OPTIONS16},

	{"NAND 512MiB 1,8V 8-bit",	0xAC, 0, 512, 0, LP_OPTIONS},
	{"NAND 512MiB 3,3V 8-bit",	0xDC, 0, 512, 0, LP_OPTIONS},
	{"NAND 512MiB 1,8V 16-bit",	0xBC, 0, 512, 0, LP_OPTIONS16},
	{"NAND 512MiB 3,3V 16-bit",	0xCC, 0, 512, 0, LP_OPTIONS16},

	{"NAND 1GiB 1,8V 8-bit",	0xA3, 0, 1024, 0, LP_OPTIONS},
	{"NAND 1GiB 3,3V 8-bit",	0xD3, 0, 1024, 0, LP_OPTIONS},
	{"NAND 1GiB 1,8V 16-bit",	0xB3, 0, 1024, 0, LP_OPTIONS16},
	{"NAND 1GiB 3,3V 16-bit",	0xC3, 0, 1024, 0, LP_OPTIONS16},

	{"NAND 2GiB 1,8V 8-bit",	0xA5, 0, 2048, 0, LP_OPTIONS},
	{"NAND 2GiB 3,3V 8-bit",	0xD5, 0, 2048, 0, LP_OPTIONS},
	{"NAND 2GiB 1,8V 16-bit",	0xB5, 0, 2048, 0, LP_OPTIONS16},
	{"NAND 2GiB 3,3V 16-bit",	0xC5, 0, 2048, 0, LP_OPTIONS16},

	{NULL, 0,}
};

/* Manufacturer ID list
 */
nand_manufacturer_t nand_manuf_ids[] =
{
	{0x0, "unknown"},
	{NAND_MFR_TOSHIBA, "Toshiba"},
	{NAND_MFR_SAMSUNG, "Samsung"},
	{NAND_MFR_FUJITSU, "Fujitsu"},
	{NAND_MFR_NATIONAL, "National"},
	{NAND_MFR_RENESAS, "Renesas"},
	{NAND_MFR_STMICRO, "ST Micro"},
	{NAND_MFR_HYNIX, "Hynix"},
	{0x0, NULL},
};

/* nand device <nand_controller> [controller options]
 */
int handle_nand_device_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc)
{
	int i;
	int retval;
		
	if (argc < 1)
	{
		WARNING("incomplete flash device nand configuration");
		return ERROR_FLASH_BANK_INVALID;
	}
	
	for (i = 0; nand_flash_controllers[i]; i++)
	{
		nand_device_t *p, *c;
		
		if (strcmp(args[0], nand_flash_controllers[i]->name) == 0)
		{
			/* register flash specific commands */
			if (nand_flash_controllers[i]->register_commands(cmd_ctx) != ERROR_OK)
			{
				ERROR("couldn't register '%s' commands", args[0]);
				exit(-1);
			}
	
			c = malloc(sizeof(nand_device_t));

			c->controller = nand_flash_controllers[i];
			c->controller_priv = NULL;
			c->manufacturer = NULL;
			c->device = NULL;
			c->bus_width = 0;
			c->address_cycles = 0;
			c->page_size = 0;
			c->use_raw = 0;
			c->next = NULL;

			if ((retval = nand_flash_controllers[i]->nand_device_command(cmd_ctx, cmd, args, argc, c)) != ERROR_OK)
			{
				ERROR("'%s' driver rejected nand flash", c->controller->name);
				free(c);
				return ERROR_OK;
			}
			
			/* put NAND device in linked list */
			if (nand_devices)
			{
				/* find last flash device */
				for (p = nand_devices; p && p->next; p = p->next);
				if (p)
					p->next = c;
			}
			else
			{
				nand_devices = c;
			}
			
			return ERROR_OK;
		}
	}

	/* no valid NAND controller was found (i.e. the configuration option,
	 * didn't match one of the compiled-in controllers)
	 */
	ERROR("No valid NAND flash controller found (%s)", args[0]);
	ERROR("compiled-in NAND flash controllers:");
	for (i = 0; nand_flash_controllers[i]; i++)
	{
		ERROR("%i: %s", i, nand_flash_controllers[i]->name);
	}
	
	return ERROR_OK;
}

int nand_register_commands(struct command_context_s *cmd_ctx)
{
	nand_cmd = register_command(cmd_ctx, NULL, "nand", NULL, COMMAND_ANY, NULL);
	
	register_command(cmd_ctx, nand_cmd, "device", handle_nand_device_command, COMMAND_CONFIG, NULL);
	
	return ERROR_OK;
}

int nand_init(struct command_context_s *cmd_ctx)
{
	if (nand_devices)
	{
		register_command(cmd_ctx, nand_cmd, "list", handle_nand_list_command, COMMAND_EXEC,
						 "list configured NAND flash devices");
		register_command(cmd_ctx, nand_cmd, "info", handle_nand_info_command, COMMAND_EXEC,
						 "print info about NAND flash device <num>");
		register_command(cmd_ctx, nand_cmd, "probe", handle_nand_probe_command, COMMAND_EXEC,
						 "identify NAND flash device <num>");
		register_command(cmd_ctx, nand_cmd, "check_bad_blocks", handle_nand_check_bad_blocks_command, COMMAND_EXEC,
						 "check NAND flash device <num> for bad blocks [<first> <last>]");
		register_command(cmd_ctx, nand_cmd, "erase", handle_nand_erase_command, COMMAND_EXEC,
						 "erase blocks on NAND flash device <num> <first> <last>");
		register_command(cmd_ctx, nand_cmd, "copy", handle_nand_copy_command, COMMAND_EXEC,
						 "copy from NAND flash device <num> <offset> <length> <ram-address>");
		register_command(cmd_ctx, nand_cmd, "dump", handle_nand_dump_command, COMMAND_EXEC,
						 "dump from NAND flash device <num> <filename> <offset> <size> [options]");
		register_command(cmd_ctx, nand_cmd, "write", handle_nand_write_command, COMMAND_EXEC,
						 "write to NAND flash device <num> <filename> <offset> [options]");
		register_command(cmd_ctx, nand_cmd, "raw_access", handle_nand_raw_access_command, COMMAND_EXEC,
						 "raw access to NAND flash device <num> ['enable'|'disable']");
	}
	
	return ERROR_OK;
}

nand_device_t *get_nand_device_by_num(int num)
{
	nand_device_t *p;
	int i = 0;

	for (p = nand_devices; p; p = p->next)
	{
		if (i++ == num)
		{
			return p;
		}
	}
	
	return NULL;
}

int nand_build_bbt(struct nand_device_s *device, int first, int last)
{
	u32 page = 0x0;
	int i;
	u8 *oob;
	
	oob = malloc(6);
	
	if ((first < 0) || (first >= device->num_blocks))
		first = 0;
	
	if ((last >= device->num_blocks) || (last == -1))
		last = device->num_blocks - 1;
	
	for (i = first; i < last; i++)
	{
		nand_read_page(device, page, NULL, 0, oob, 6);
		
		if (((device->device->options & NAND_BUSWIDTH_16) && ((oob[0] & oob[1]) != 0xff))
			|| (((device->page_size == 512) && (oob[5] != 0xff)) ||
				((device->page_size == 2048) && (oob[0] != 0xff))))
		{
			WARNING("invalid block: %i", i);
			device->blocks[i].is_bad = 1;
		}
		else
		{
			device->blocks[i].is_bad = 0;
		}
		
		page += (device->erase_size / device->page_size);
	}
	
	return ERROR_OK;
}

int nand_read_status(struct nand_device_s *device, u8 *status)
{
	if (!device->device)
		return ERROR_NAND_DEVICE_NOT_PROBED;
		
	/* Send read status command */
	device->controller->command(device, NAND_CMD_STATUS);
	
	usleep(1000);
	
	/* read status */
	if (device->device->options & NAND_BUSWIDTH_16)
	{
		u16 data;
		device->controller->read_data(device, &data);
		*status = data & 0xff;
	}
	else
	{
		device->controller->read_data(device, status);
	}
			
	return ERROR_OK;
}

int nand_probe(struct nand_device_s *device)
{
	u8 manufacturer_id, device_id;
	int retval;
	int i;

	/* clear device data */
	device->device = NULL;
	device->manufacturer = NULL;
	
	/* clear device parameters */
	device->bus_width = 0;
	device->address_cycles = 0;
	device->page_size = 0;
	device->erase_size = 0;
	
	/* initialize controller (device parameters are zero, use controller default) */
	if ((retval = device->controller->init(device) != ERROR_OK))
	{
		switch (retval)
		{
			case ERROR_NAND_OPERATION_FAILED:
				DEBUG("controller initialization failed");
				return ERROR_NAND_OPERATION_FAILED;
			case ERROR_NAND_OPERATION_NOT_SUPPORTED:
				ERROR("BUG: controller reported that it doesn't support default parameters");
				return ERROR_NAND_OPERATION_FAILED;
			default:
				ERROR("BUG: unknown controller initialization failure");
				return ERROR_NAND_OPERATION_FAILED;
		}
	}
	
	device->controller->command(device, NAND_CMD_RESET);
	device->controller->reset(device);

	device->controller->command(device, NAND_CMD_READID);
	device->controller->address(device, 0x0);
	
	if (device->bus_width == 8)
	{
		device->controller->read_data(device, &manufacturer_id);
		device->controller->read_data(device, &device_id);
	}
	else
	{
		u16 data_buf;
		device->controller->read_data(device, &data_buf);
		manufacturer_id = data_buf & 0xff;
		device->controller->read_data(device, &data_buf);
		device_id = data_buf & 0xff;
	}
		
	for (i = 0; nand_flash_ids[i].name; i++)
	{
		if (nand_flash_ids[i].id == device_id)
		{
			device->device = &nand_flash_ids[i];
			break;
		}
	}
	
	for (i = 0; nand_manuf_ids[i].name; i++)
	{
		if (nand_manuf_ids[i].id == manufacturer_id)
		{
			device->manufacturer = &nand_manuf_ids[i];
			break;
		}
	}
	
	if (!device->manufacturer)
	{
		device->manufacturer = &nand_manuf_ids[0];
		device->manufacturer->id = manufacturer_id;
	}
	
	if (!device->device)
	{
		ERROR("unknown NAND flash device found, manufacturer id: 0x%2.2x device id: 0x%2.2x",
			manufacturer_id, device_id);
		return ERROR_NAND_OPERATION_FAILED;
	}
	
	DEBUG("found %s (%s)", device->device->name, device->manufacturer->name);
	
	/* initialize device parameters */
	
	/* bus width */ 
	if (device->device->options & NAND_BUSWIDTH_16)
		device->bus_width = 16;
	else
		device->bus_width = 8;
		
	/* page size */
	if (device->device->page_size == 0)
	{
		/* TODO: support reading extended chip id to determine page size */
		return ERROR_NAND_OPERATION_FAILED;
	}
	else if (device->device->page_size == 256)
	{
		ERROR("NAND flashes with 256 byte pagesize are not supported");
		return ERROR_NAND_OPERATION_FAILED;
	}
	else
	{
		device->page_size = device->device->page_size;
	}
	
	/* number of address cycles */
	if (device->page_size <= 512)
	{
		/* small page devices */
		if (device->device->chip_size <= 32)
			device->address_cycles = 3;
		else if (device->device->chip_size <= 8*1024)
			device->address_cycles = 4;
		else
		{
			ERROR("BUG: small page NAND device with more than 8 GiB encountered");
			device->address_cycles = 5;
		}
	}
	else
	{
		/* large page devices */
		if (device->device->chip_size <= 128)
			device->address_cycles = 4;
		else if (device->device->chip_size <= 32*1024)
			device->address_cycles = 5;
		else
		{
			ERROR("BUG: small page NAND device with more than 32 GiB encountered");
			device->address_cycles = 6;
		}
	}
	
	/* erase size */
	if (device->device->erase_size == 0)
	{
		/* TODO: support reading extended chip id to determine erase size */
	}
	else
	{
		device->erase_size = device->device->erase_size;
	}
	
	/* initialize controller, but leave parameters at the controllers default */
	if ((retval = device->controller->init(device) != ERROR_OK))
	{
		switch (retval)
		{
			case ERROR_NAND_OPERATION_FAILED:
				DEBUG("controller initialization failed");
				return ERROR_NAND_OPERATION_FAILED;
			case ERROR_NAND_OPERATION_NOT_SUPPORTED:
				ERROR("controller doesn't support requested parameters (buswidth: %i, address cycles: %i, page size: %i)",
					device->bus_width, device->address_cycles, device->page_size);
				return ERROR_NAND_OPERATION_FAILED;
			default:
				ERROR("BUG: unknown controller initialization failure");
				return ERROR_NAND_OPERATION_FAILED;
		}
	}
	
	device->num_blocks = (device->device->chip_size * 1024) / (device->erase_size / 1024);
	device->blocks = malloc(sizeof(nand_block_t) * device->num_blocks);
	
	for (i = 0; i < device->num_blocks; i++)
	{
		device->blocks[i].size = device->erase_size;
		device->blocks[i].offset = i * device->erase_size;
		device->blocks[i].is_erased = -1;
		device->blocks[i].is_bad = -1;
	}
	
	return ERROR_OK;
}

int nand_erase(struct nand_device_s *device, int first_block, int last_block)
{
	int i;
	u32 page;
	u8 status;
	int retval;
	
	if (!device->device)
		return ERROR_NAND_DEVICE_NOT_PROBED;
	
	if ((first_block < 0) || (last_block > device->num_blocks))
		return ERROR_INVALID_ARGUMENTS;
	
	/* make sure we know if a block is bad before erasing it */
	for (i = first_block; i <= last_block; i++)
	{
		if (device->blocks[i].is_bad == -1)
		{
			nand_build_bbt(device, i, last_block);
			break;
		}
	}
	
	for (i = first_block; i <= last_block; i++)
	{
		/* Send erase setup command */
		device->controller->command(device, NAND_CMD_ERASE1);
		
		page = i * (device->erase_size / device->page_size);
		
		/* Send page address */
		if (device->page_size <= 512)
		{
			/* row */
			device->controller->address(device, page & 0xff);
			device->controller->address(device, (page >> 8) & 0xff);
			
			/* 3rd cycle only on devices with more than 32 MiB */
			if (device->address_cycles >= 4)
				device->controller->address(device, (page >> 16) & 0xff);
	
			/* 4th cycle only on devices with more than 8 GiB */
			if (device->address_cycles >= 5)
				device->controller->address(device, (page >> 24) & 0xff);
		}
		else
		{
			/* row */
			device->controller->address(device, page & 0xff);
			device->controller->address(device, (page >> 8) & 0xff);
	
			/* 3rd cycle only on devices with more than 128 MiB */
			if (device->address_cycles >= 5)
				device->controller->address(device, (page >> 16) & 0xff);
		}
		
		/* Send erase confirm command */
		device->controller->command(device, NAND_CMD_ERASE2);
		
		if (!device->controller->nand_ready(device, 1000))
		{
			ERROR("timeout waiting for NAND flash block erase to complete");
			return ERROR_NAND_OPERATION_TIMEOUT;
		}
		
		if ((retval = nand_read_status(device, &status)) != ERROR_OK)
		{
			ERROR("couldn't read status");
			return ERROR_NAND_OPERATION_FAILED;
		}
		
		if (status & 0x1)
		{
			ERROR("erase operation didn't pass, status: 0x%2.2x", status);
			return ERROR_NAND_OPERATION_FAILED;
		}
	}
	
	return ERROR_OK;
}

int nand_read_plain(struct nand_device_s *device, u32 address, u8 *data, u32 data_size)
{
	u8 *page;
	
	if (!device->device)
		return ERROR_NAND_DEVICE_NOT_PROBED;
		
	if (address % device->page_size)
	{
		ERROR("reads need to be page aligned");
		return ERROR_NAND_OPERATION_FAILED;
	}
	
	page = malloc(device->page_size);
	
	while (data_size > 0 )
	{
		u32 thisrun_size = (data_size > device->page_size) ? device->page_size : data_size;
		u32 page_address;
		
		
		page_address = address / device->page_size;
		
		nand_read_page(device, page_address, page, device->page_size, NULL, 0);

		memcpy(data, page, thisrun_size);
		
		address += thisrun_size;
		data += thisrun_size;
		data_size -= thisrun_size;
	}
	
	free(page);
	
	return ERROR_OK;
}

int nand_write_plain(struct nand_device_s *device, u32 address, u8 *data, u32 data_size)
{
	u8 *page;
	
	if (!device->device)
		return ERROR_NAND_DEVICE_NOT_PROBED;
		
	if (address % device->page_size)
	{
		ERROR("writes need to be page aligned");
		return ERROR_NAND_OPERATION_FAILED;
	}
	
	page = malloc(device->page_size);
	
	while (data_size > 0 )
	{
		u32 thisrun_size = (data_size > device->page_size) ? device->page_size : data_size;
		u32 page_address;
		
		memset(page, 0xff, device->page_size);
		memcpy(page, data, thisrun_size);
		
		page_address = address / device->page_size;
		
		nand_write_page(device, page_address, page, device->page_size, NULL, 0);
		
		address += thisrun_size;
		data += thisrun_size;
		data_size -= thisrun_size;
	}
	
	free(page);
	
	return ERROR_OK;
}

int nand_write_page(struct nand_device_s *device, u32 page, u8 *data, u32 data_size, u8 *oob, u32 oob_size)
{
	if (!device->device)
		return ERROR_NAND_DEVICE_NOT_PROBED;
		
	if (device->use_raw)
		return nand_write_page_raw(device, page, data, data_size, oob, oob_size);
	else
		return device->controller->write_page(device, page, data, data_size, oob, oob_size);
}

int nand_read_page(struct nand_device_s *device, u32 page, u8 *data, u32 data_size, u8 *oob, u32 oob_size)
{
	if (!device->device)
		return ERROR_NAND_DEVICE_NOT_PROBED;
		
	if (device->use_raw)
		return nand_read_page_raw(device, page, data, data_size, oob, oob_size);
	else
		return device->controller->read_page(device, page, data, data_size, oob, oob_size);
}

int nand_read_page_raw(struct nand_device_s *device, u32 page, u8 *data, u32 data_size, u8 *oob, u32 oob_size)
{
	int i;
	
	if (!device->device)
		return ERROR_NAND_DEVICE_NOT_PROBED;

	if (device->page_size <= 512)
	{
		/* small page device */
		if (data)
			device->controller->command(device, NAND_CMD_READ0);
		else
			device->controller->command(device, NAND_CMD_READOOB);
		
		/* column (always 0, we start at the beginning of a page/OOB area) */
		device->controller->address(device, 0x0);
		
		/* row */
		device->controller->address(device, page & 0xff);
		device->controller->address(device, (page >> 8) & 0xff);
		
		/* 4th cycle only on devices with more than 32 MiB */
		if (device->address_cycles >= 4)
			device->controller->address(device, (page >> 16) & 0xff);

		/* 5th cycle only on devices with more than 8 GiB */
		if (device->address_cycles >= 5)
			device->controller->address(device, (page >> 24) & 0xff);
	}
	else
	{
		/* large page device */
		device->controller->command(device, NAND_CMD_READ0);
		
		/* column (0 when we start at the beginning of a page,
		 * or 2048 for the beginning of OOB area)
		 */
		device->controller->address(device, 0x0);
		device->controller->address(device, 0x8);
		
		/* row */
		device->controller->address(device, page & 0xff);
		device->controller->address(device, (page >> 8) & 0xff);

		/* 5th cycle only on devices with more than 128 MiB */
		if (device->address_cycles >= 5)
			device->controller->address(device, (page >> 16) & 0xff);

		/* large page devices need a start command */
		device->controller->command(device, NAND_CMD_READSTART);
	}
	
	if (!device->controller->nand_ready(device, 100))
		return ERROR_NAND_OPERATION_TIMEOUT;
	
	if (data)
	{
		for (i = 0; i < data_size;)
		{
			if (device->device->options & NAND_BUSWIDTH_16)
			{
				device->controller->read_data(device, data);
				data += 2;
				i += 2;
			}
			else
			{
				device->controller->read_data(device, data);
				data += 1;
				i += 1;
			}
		}
	}
	
	if (oob)
	{
		for (i = 0; i < oob_size;)
		{
			if (device->device->options & NAND_BUSWIDTH_16)
			{
				device->controller->read_data(device, oob);
				oob += 2;
				i += 2;
			}
			else
			{
				device->controller->read_data(device, oob);
				oob += 1;
				i += 1;
			}
		}
	}
	
	return ERROR_OK;	
}

int nand_write_page_raw(struct nand_device_s *device, u32 page, u8 *data, u32 data_size, u8 *oob, u32 oob_size)
{
	int i;
	int retval;
	u8 status;
	
	if (!device->device)
		return ERROR_NAND_DEVICE_NOT_PROBED;

	device->controller->command(device, NAND_CMD_SEQIN);
	
	if (device->page_size <= 512)
	{
		/* column (always 0, we start at the beginning of a page/OOB area) */
		device->controller->address(device, 0x0);
		
		/* row */
		device->controller->address(device, page & 0xff);
		device->controller->address(device, (page >> 8) & 0xff);
		
		/* 4th cycle only on devices with more than 32 MiB */
		if (device->address_cycles >= 4)
			device->controller->address(device, (page >> 16) & 0xff);

		/* 5th cycle only on devices with more than 8 GiB */
		if (device->address_cycles >= 5)
			device->controller->address(device, (page >> 24) & 0xff);
	}
	else
	{
		/* column (0 when we start at the beginning of a page,
		 * or 2048 for the beginning of OOB area)
		 */
		device->controller->address(device, 0x0);
		device->controller->address(device, 0x8);
		
		/* row */
		device->controller->address(device, page & 0xff);
		device->controller->address(device, (page >> 8) & 0xff);

		/* 5th cycle only on devices with more than 128 MiB */
		if (device->address_cycles >= 5)
			device->controller->address(device, (page >> 16) & 0xff);
	}
	
	if (data)
	{
		for (i = 0; i < data_size;)
		{
			if (device->device->options & NAND_BUSWIDTH_16)
			{
				u16 data_buf = le_to_h_u16(data);
				device->controller->write_data(device, data_buf);
				data += 2;
				i += 2;
			}
			else
			{
				device->controller->write_data(device, *data);
				data += 1;
				i += 1;
			}
		}
	}
	
	if (oob)
	{
		for (i = 0; i < oob_size;)
		{
			if (device->device->options & NAND_BUSWIDTH_16)
			{
				u16 oob_buf = le_to_h_u16(data);
				device->controller->write_data(device, oob_buf);
				oob += 2;
				i += 2;
			}
			else
			{
				device->controller->write_data(device, *oob);
				oob += 1;
				i += 1;
			}
		}
	}
	
	device->controller->command(device, NAND_CMD_PAGEPROG);
	
	if (!device->controller->nand_ready(device, 100))
		return ERROR_NAND_OPERATION_TIMEOUT;
	
	if ((retval = nand_read_status(device, &status)) != ERROR_OK)
	{
		ERROR("couldn't read status");
		return ERROR_NAND_OPERATION_FAILED;
	}
		
	if (status & NAND_STATUS_FAIL)
	{
		ERROR("write operation didn't pass, status: 0x%2.2x", status);
		return ERROR_NAND_OPERATION_FAILED;
	}
	
	return ERROR_OK;	
}

int handle_nand_list_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc)
{
	nand_device_t *p;
	int i = 0;
	
	if (!nand_devices)
	{
		command_print(cmd_ctx, "no NAND flash devices configured");
		return ERROR_OK;
	}
	
	for (p = nand_devices; p; p = p->next)
	{
		if (p->device)
			command_print(cmd_ctx, "#%i: %s (%s) pagesize: %i, buswidth: %i, erasesize: %i",
				i++, p->device->name, p->manufacturer->name, p->page_size, p->bus_width, p->erase_size);
		else
			command_print(cmd_ctx, "#%i: not probed");
	}
	
	return ERROR_OK;
}

int handle_nand_info_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc)
{
	nand_device_t *p;
	int i = 0;
	int j = 0;
	int first = -1;
	int last = -1;
		
	if ((argc < 1) || (argc > 3))
	{
		command_print(cmd_ctx, "usage: nand info <num> [<first> <last>]");
		return ERROR_OK;
	}
	
	if (argc == 2)
	{
		first = last = strtoul(args[1], NULL, 0);
	}
	else if (argc == 3)
	{
		first = strtoul(args[1], NULL, 0);
		last = strtoul(args[2], NULL, 0);
	}
		
	p = get_nand_device_by_num(strtoul(args[0], NULL, 0));
	if (p)
	{
		if (p->device)
		{
			if (first >= p->num_blocks)
				first = p->num_blocks - 1;
			
			if (last >= p->num_blocks)
				last = p->num_blocks - 1;
			
			command_print(cmd_ctx, "#%i: %s (%s) pagesize: %i, buswidth: %i, erasesize: %i",
				i++, p->device->name, p->manufacturer->name, p->page_size, p->bus_width, p->erase_size);
			
			for (j = first; j <= last; j++)
			{
				char *erase_state, *bad_state;
				
				if (p->blocks[j].is_erased == 0)
					erase_state = "not erased";
				else if (p->blocks[j].is_erased == 1)
					erase_state = "erased";
				else
					erase_state = "erase state unknown";
				
				if (p->blocks[j].is_bad == 0)
					bad_state = "";
				else if (p->blocks[j].is_bad == 1)
					bad_state = " (marked bad)";
				else
					bad_state = " (block condition unknown)";

				command_print(cmd_ctx, "\t#%i: 0x%8.8x (0x%xkB) %s%s",
							j, p->blocks[j].offset, p->blocks[j].size / 1024,
							erase_state, bad_state);
			}
		}
		else
		{
			command_print(cmd_ctx, "#%i: not probed");
		}
	}
	
	return ERROR_OK;
}

int handle_nand_probe_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc)
{
	nand_device_t *p;
	int retval;
		
	if (argc != 1)
	{
		command_print(cmd_ctx, "usage: nand probe <num>");
		return ERROR_OK;
	}
	
	p = get_nand_device_by_num(strtoul(args[0], NULL, 0));
	if (p)
	{
		if ((retval = nand_probe(p)) == ERROR_OK)
		{
			command_print(cmd_ctx, "NAND flash device '%s' found", p->device->name);
		}
		else if (retval == ERROR_NAND_OPERATION_FAILED)
		{
			command_print(cmd_ctx, "probing failed for NAND flash device");
		}
		else
		{
			command_print(cmd_ctx, "unknown error when probing NAND flash device");
		}
	}
	else
	{
		command_print(cmd_ctx, "NAND flash device '#%s' is out of bounds", args[0]);
	}
	
	return ERROR_OK;
}

int handle_nand_erase_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc)
{
	nand_device_t *p;
	int retval;
		
	if (argc != 3)
	{
		command_print(cmd_ctx, "usage: nand erase <num> <first> <last>");
		return ERROR_OK;
	}
	
	p = get_nand_device_by_num(strtoul(args[0], NULL, 0));
	if (p)
	{
		int first = strtoul(args[1], NULL, 0);
		int last = strtoul(args[2], NULL, 0);
		
		if ((retval = nand_erase(p, first, last)) == ERROR_OK)
		{
			command_print(cmd_ctx, "successfully erased blocks %i to %i on NAND flash device '%s'", first, last, p->device->name);
		}
		else if (retval == ERROR_NAND_OPERATION_FAILED)
		{
			command_print(cmd_ctx, "erase failed");
		}
		else
		{
			command_print(cmd_ctx, "unknown error when erasing NAND flash device");
		}
	}
	else
	{
		command_print(cmd_ctx, "NAND flash device '#%s' is out of bounds", args[0]);
	}
	
	return ERROR_OK;
}

int handle_nand_check_bad_blocks_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc)
{
	nand_device_t *p;
	int retval;
	int first = -1;
	int last = -1;
		
	if ((argc < 1) || (argc > 3) || (argc == 2))
	{
		command_print(cmd_ctx, "usage: nand check_bad_blocks <num> [<first> <last>]");
		return ERROR_OK;
	}
	
	if (argc == 3)
	{
		first = strtoul(args[1], NULL, 0);
		last = strtoul(args[2], NULL, 0);
	}
	
	p = get_nand_device_by_num(strtoul(args[0], NULL, 0));
	if (p)
	{
		if ((retval = nand_build_bbt(p, first, last)) == ERROR_OK)
		{
			command_print(cmd_ctx, "checked NAND flash device for bad blocks, use \"nand info\" command to list blocks", p->device->name);
		}
		else if (retval == ERROR_NAND_OPERATION_FAILED)
		{
			command_print(cmd_ctx, "error when checking for bad blocks on NAND flash device");
		}
		else
		{
			command_print(cmd_ctx, "unknown error when checking for bad blocks on NAND flash device");
		}
	}
	else
	{
		command_print(cmd_ctx, "NAND flash device '#%s' is out of bounds", args[0]);
	}
	
	return ERROR_OK;
}

int handle_nand_copy_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc)
{
	nand_device_t *p;
		
	if (argc != 4)
	{
		command_print(cmd_ctx, "usage: nand copy <num> <offset> <length> <ram-address>");
		return ERROR_OK;
	}
	
	p = get_nand_device_by_num(strtoul(args[0], NULL, 0));
	if (p)
	{

	}
	else
	{
		command_print(cmd_ctx, "NAND flash device '#%s' is out of bounds", args[0]);
	}
	
	return ERROR_OK;
}

int handle_nand_write_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc)
{
	u32 offset;
	u32 binary_size;
	u32 buf_cnt;
	enum oob_formats oob_format = NAND_OOB_NONE;
	
	fileio_t file;
	fileio_image_t image_info;
	int sec_type_identified = 0;
	enum fileio_sec_type sec_type;
	
	duration_t duration;
	char *duration_text;
	
	nand_device_t *p;
		
	if (argc < 3)
	{
		command_print(cmd_ctx, "usage: nand write <num> <file> <offset> [options]");
		return ERROR_OK;
	}
	
	p = get_nand_device_by_num(strtoul(args[0], NULL, 0));
	if (p)
	{
		u8 *page = NULL;
		u32 page_size = 0;
		u8 *oob = NULL;
		u32 oob_size = 0;
			
		duration_start_measure(&duration);
		strtoul(args[2], NULL, 0);
		
		if (argc > 3)
		{
			int i;
			for (i = 3; i < argc; i++)
			{
				if (!strcmp(args[i], "oob_raw"))
					oob_format |= NAND_OOB_RAW;
				else if (!strcmp(args[i], "oob_only"))
					oob_format |= NAND_OOB_RAW | NAND_OOB_ONLY;
				else
				{
					if (fileio_identify_image_type(&sec_type, args[i]) == ERROR_OK)
					{
						sec_type_identified = 1;
					}
					else
					{
						command_print(cmd_ctx, "unknown option: %s", args[i]);
					}
				}
			}
		}
		
		/* if no image type option was encountered, set the default */
		if (!sec_type_identified)
		{
			
			fileio_identify_image_type(&sec_type, NULL);
			sec_type_identified = 1;
		}

		image_info.base_address = strtoul(args[2], NULL, 0);
		image_info.has_start_address = 0;
	
		if (fileio_open(&file, args[1], FILEIO_READ, 
			FILEIO_IMAGE, &image_info, sec_type) != ERROR_OK)
		{
			command_print(cmd_ctx, "flash write error: %s", file.error_str);
			return ERROR_OK;
		}
	
		/* the offset might have been overwritten by the image base address */
		offset = image_info.base_address;
		
		buf_cnt = binary_size = file.size;
		
		if (!(oob_format & NAND_OOB_ONLY))
		{
			page_size = p->page_size;
			page = malloc(p->page_size);
		}

		if (oob_format & NAND_OOB_RAW)
		{
			if (p->page_size == 512)
				oob_size = 16;
			else if (p->page_size == 2048)
				oob_size = 64;
			oob = malloc(oob_size);
		}
		
		if (offset % p->page_size)
		{
			command_print(cmd_ctx, "only page size aligned offsets and sizes are supported");
			return ERROR_OK;
		}
		
		while (buf_cnt > 0)
		{
			u32 size_read;
			
			if (page)
			{
				fileio_read(&file, page_size, page, &size_read);
				buf_cnt -= size_read;
				if (size_read < page_size)
				{
					memset(page + size_read, 0xff, page_size - size_read);
				}
			}
				
			if (oob)
			{
				fileio_read(&file, oob_size, oob, &size_read);
				buf_cnt -= size_read;
				if (size_read < oob_size)
				{
					memset(oob + size_read, 0xff, oob_size - size_read);
				}
			}
			
			if (nand_write_page(p, offset / p->page_size, page, page_size, oob, oob_size) != ERROR_OK)
			{
				command_print(cmd_ctx, "failed writing file %s to NAND flash %s at offset 0x%8.8x",
					file.url, args[0], offset);
				return ERROR_OK;
			}
			offset += page_size;
		}

		duration_stop_measure(&duration, &duration_text);
		command_print(cmd_ctx, "wrote file %s to NAND flash %s at offset 0x%8.8x in %s",
			file.url, args[0], image_info.base_address, duration_text);
		free(duration_text);
	}
	else
	{
		command_print(cmd_ctx, "NAND flash device '#%s' is out of bounds", args[0]);
	}
	
	return ERROR_OK;
}

int handle_nand_dump_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc)
{
	nand_device_t *p;
			
	if (argc < 4)
	{
		command_print(cmd_ctx, "usage: nand dump <num> <filename> <address> <size> [options]");
		return ERROR_OK;
	}
	
	p = get_nand_device_by_num(strtoul(args[0], NULL, 0));
	if (p)
	{
		if (p->device)
		{
			fileio_t file;
			fileio_image_t image_info;
			duration_t duration;
			char *duration_text;
			int retval;
			
			u8 *page = NULL;
			u32 page_size = 0;
			u8 *oob = NULL;
			u32 oob_size = 0;
			u32 address = strtoul(args[2], NULL, 0);
			u32 size = strtoul(args[3], NULL, 0);
			u32 bytes_done = 0;
			enum oob_formats oob_format = NAND_OOB_NONE;
			
			if (argc > 4)
			{
				int i;
				for (i = 4; i < argc; i++)
				{
					if (!strcmp(args[i], "oob_raw"))
						oob_format |= NAND_OOB_RAW;
					else if (!strcmp(args[i], "oob_only"))
						oob_format |= NAND_OOB_RAW | NAND_OOB_ONLY;
					else
						command_print(cmd_ctx, "unknown option: '%s'", args[i]); 
				}
			}
			
			if ((address % p->page_size) || (size % p->page_size))
			{
				command_print(cmd_ctx, "only page size aligned addresses and sizes are supported");
				return ERROR_OK;
			}
		
			if (!(oob_format & NAND_OOB_ONLY))
			{
				page_size = p->page_size;
				page = malloc(p->page_size);
			}

			if (oob_format & NAND_OOB_RAW)
			{
				if (p->page_size == 512)
					oob_size = 16;
				else if (p->page_size == 2048)
					oob_size = 64;
				oob = malloc(oob_size);
			}

			image_info.base_address = address;
			image_info.has_start_address = 0;
			
			if (fileio_open(&file, args[1], FILEIO_WRITE, 
				FILEIO_IMAGE, &image_info, FILEIO_PLAIN) != ERROR_OK)
			{
				command_print(cmd_ctx, "dump_image error: %s", file.error_str);
				return ERROR_OK;
			}
	
			duration_start_measure(&duration);
			
			while (size > 0)
			{
				u32 size_written;
				if ((retval = nand_read_page(p, address / p->page_size, page, page_size, oob, oob_size)) != ERROR_OK)
				{
					command_print(cmd_ctx, "reading NAND flash page failed");
					return ERROR_OK;
				}
				
				if (page)
				{
					fileio_write(&file, page_size, page, &size_written);
					bytes_done += page_size;
				}
					
				if (oob)
				{
					fileio_write(&file, oob_size, oob, &size_written);
					bytes_done += oob_size;
				}
					
				size -= p->page_size;
				address += p->page_size;
			}
			
			if (page)
				free(page);
				
			if (oob)
				free(oob);
			
			fileio_close(&file);

			duration_stop_measure(&duration, &duration_text);
			command_print(cmd_ctx, "dumped %lli byte in %s", file.size, duration_text);
			free(duration_text);
		}
		else
		{
			command_print(cmd_ctx, "#%i: not probed");
		}
	}
	else
	{
		command_print(cmd_ctx, "NAND flash device '#%s' is out of bounds", args[0]);
	}
	
	return ERROR_OK;
}

int handle_nand_raw_access_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc)
{
	nand_device_t *p;
		
	if ((argc < 1) || (argc > 2))
	{
		command_print(cmd_ctx, "usage: nand raw_access <num> ['enable'|'disable']");
		return ERROR_OK;
	}
	
	p = get_nand_device_by_num(strtoul(args[0], NULL, 0));
	if (p)
	{
		if (p->device)
		{
			if (argc == 2)
			{
				if (strcmp("enable", args[1]) == 0)
				{
					p->use_raw = 1;
				}
				else if (strcmp("disable", args[1]) == 0)
				{
					p->use_raw = 0;
				}
				else
				{
					command_print(cmd_ctx, "usage: nand raw_access ['enable'|disable']");
				}
			}
	
			command_print(cmd_ctx, "raw access is %s", (p->use_raw) ? "enabled" : "disabled");
		}
		else
		{
			command_print(cmd_ctx, "#%i: not probed");
		}
	}
	else
	{
		command_print(cmd_ctx, "NAND flash device '#%s' is out of bounds", args[0]);
	}
	
	return ERROR_OK;
}