[people/ms/u-boot.git] / board / mcc200 / auto_update.c

/*
 * (C) Copyright 2006
 * Wolfgang Denk, DENX Software Engineering, wd@denx.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
 */
#include <common.h>
#include <command.h>
#include <malloc.h>
#include <image.h>
#include <asm/byteorder.h>
#include <usb.h>
#include <part.h>

#ifdef CFG_HUSH_PARSER
#include <hush.h>
#endif


#ifdef CONFIG_AUTO_UPDATE

#ifndef CONFIG_USB_OHCI
#error "must define CONFIG_USB_OHCI"
#endif

#ifndef CONFIG_USB_STORAGE
#error "must define CONFIG_USB_STORAGE"
#endif

#ifndef CFG_HUSH_PARSER
#error "must define CFG_HUSH_PARSER"
#endif

#if !defined(CONFIG_CMD_FAT)
#error "must define CONFIG_CMD_FAT"
#endif

#undef AU_DEBUG

#undef debug
#ifdef	AU_DEBUG
#define debug(fmt,args...)	printf (fmt ,##args)
#else
#define debug(fmt,args...)
#endif	/* AU_DEBUG */

/* possible names of files on the USB stick. */
#define AU_FIRMWARE	"u-boot.img"
#define AU_KERNEL	"kernel.img"
#define AU_ROOTFS	"rootfs.img"

struct flash_layout {
	long start;
	long end;
};

/* layout of the FLASH. ST = start address, ND = end address. */
#define AU_FL_FIRMWARE_ST	0xfC000000
#define AU_FL_FIRMWARE_ND	0xfC03FFFF
#define AU_FL_KERNEL_ST		0xfC0C0000
#define AU_FL_KERNEL_ND		0xfC1BFFFF
#define AU_FL_ROOTFS_ST		0xFC1C0000
#define AU_FL_ROOTFS_ND		0xFCFBFFFF

static int au_usb_stor_curr_dev; /* current device */

/* index of each file in the following arrays */
#define IDX_FIRMWARE	0
#define IDX_KERNEL	1
#define IDX_ROOTFS	2

/* max. number of files which could interest us */
#define AU_MAXFILES 3

/* pointers to file names */
char *aufile[AU_MAXFILES] = {
	AU_FIRMWARE,
	AU_KERNEL,
	AU_ROOTFS
};

/* sizes of flash areas for each file */
long ausize[AU_MAXFILES] = {
	(AU_FL_FIRMWARE_ND + 1) - AU_FL_FIRMWARE_ST,
	(AU_FL_KERNEL_ND   + 1) - AU_FL_KERNEL_ST,
	(AU_FL_ROOTFS_ND   + 1) - AU_FL_ROOTFS_ST,
};

/* array of flash areas start and end addresses */
struct flash_layout aufl_layout[AU_MAXFILES] = {
	{ AU_FL_FIRMWARE_ST,	AU_FL_FIRMWARE_ND, },
	{ AU_FL_KERNEL_ST,	AU_FL_KERNEL_ND,   },
	{ AU_FL_ROOTFS_ST,	AU_FL_ROOTFS_ND,   },
};

ulong totsize;

/* where to load files into memory */
#define LOAD_ADDR ((unsigned char *)0x00200000)

/* the root file system is the largest image */
#define MAX_LOADSZ ausize[IDX_ROOTFS]

/*i2c address of the keypad status*/
#define I2C_PSOC_KEYPAD_ADDR	0x53

/* keypad mask */
#define KEYPAD_ROW	2
#define KEYPAD_COL	2
#define KEYPAD_MASK_LO	((1<<(KEYPAD_COL-1+(KEYPAD_ROW*3-3)))&0xFF)
#define KEYPAD_MASK_HI	((1<<(KEYPAD_COL-1+(KEYPAD_ROW*3-3)))>>8)

/* externals */
extern int fat_register_device(block_dev_desc_t *, int);
extern int file_fat_detectfs(void);
extern long file_fat_read(const char *, void *, unsigned long);
extern int i2c_read (unsigned char, unsigned int, int , unsigned char* , int);
extern int flash_sect_erase(ulong, ulong);
extern int flash_sect_protect (int, ulong, ulong);
extern int flash_write (char *, ulong, ulong);
extern int u_boot_hush_start(void);
#ifdef CONFIG_PROGRESSBAR
extern void show_progress(int, int);
extern void lcd_puts (char *);
extern void lcd_enable(void);
#endif

int au_check_cksum_valid(int idx, long nbytes)
{
	image_header_t *hdr;

	hdr = (image_header_t *)LOAD_ADDR;
#if defined(CONFIG_FIT)
	if (gen_image_get_format ((void *)hdr) != IMAGE_FORMAT_LEGACY) {
		puts ("Non legacy image format not supported\n");
		return -1;
	}
#endif

	if (nbytes != image_get_image_size (hdr)) {
		printf ("Image %s bad total SIZE\n", aufile[idx]);
		return -1;
	}
	/* check the data CRC */
	if (!image_check_dcrc (hdr)) {
		printf ("Image %s bad data checksum\n", aufile[idx]);
		return -1;
	}
	return 0;
}

int au_check_header_valid(int idx, long nbytes)
{
	image_header_t *hdr;
	unsigned long checksum, fsize;

	hdr = (image_header_t *)LOAD_ADDR;
#if defined(CONFIG_FIT)
	if (gen_image_get_format ((void *)hdr) != IMAGE_FORMAT_LEGACY) {
		puts ("Non legacy image format not supported\n");
		return -1;
	}
#endif

	/* check the easy ones first */
#undef CHECK_VALID_DEBUG
#ifdef CHECK_VALID_DEBUG
	printf("magic %#x %#x ", image_get_magic (hdr), IH_MAGIC);
	printf("arch %#x %#x ", image_get_arch (hdr), IH_ARCH_ARM);
	printf("size %#x %#lx ", image_get_data_size (hdr), nbytes);
	printf("type %#x %#x ", image_get_type (hdr), IH_TYPE_KERNEL);
#endif
	if (nbytes < image_get_header_size ()) {
		printf ("Image %s bad header SIZE\n", aufile[idx]);
		ausize[idx] = 0;
		return -1;
	}
	if (!image_check_magic (hdr) || !image_check_arch (hdr, IH_ARCH_PPC)) {
		printf ("Image %s bad MAGIC or ARCH\n", aufile[idx]);
		ausize[idx] = 0;
		return -1;
	}
	/* check the hdr CRC */
	if (!image_check_hcrc (hdr)) {
		printf ("Image %s bad header checksum\n", aufile[idx]);
		ausize[idx] = 0;
		return -1;
	}
	/* check the type - could do this all in one gigantic if() */
	if ((idx == IDX_FIRMWARE) && !image_check_type (hdr, IH_TYPE_FIRMWARE)) {
		printf ("Image %s wrong type\n", aufile[idx]);
		ausize[idx] = 0;
		return -1;
	}
	if ((idx == IDX_KERNEL) && !image_check_type (hdr, IH_TYPE_KERNEL)) {
		printf ("Image %s wrong type\n", aufile[idx]);
		ausize[idx] = 0;
		return -1;
	}
	if ((idx == IDX_ROOTFS) &&
			(!image_check_type (hdr, IH_TYPE_RAMDISK) &&
			!image_check_type (hdr, IH_TYPE_FILESYSTEM))) {
		printf ("Image %s wrong type\n", aufile[idx]);
		ausize[idx] = 0;
		return -1;
	}
	/* recycle checksum */
	checksum = image_get_data_size (hdr);

	fsize = checksum + image_get_header_size ();
	/* for kernel and ramdisk the image header must also fit into flash */
	if (idx == IDX_KERNEL || image_check_type (hdr, IH_TYPE_RAMDISK))
		checksum += image_get_header_size ();

	/* check the size does not exceed space in flash. HUSH scripts */
	if ((ausize[idx] != 0) && (ausize[idx] < checksum)) {
		printf ("Image %s is bigger than FLASH\n", aufile[idx]);
		ausize[idx] = 0;
		return -1;
	}
	/* Update with the real filesize */
	ausize[idx] = fsize;

	return checksum; /* return size to be written to flash */
}

int au_do_update(int idx, long sz)
{
	image_header_t *hdr;
	char *addr;
	long start, end;
	int off, rc;
	uint nbytes;

	hdr = (image_header_t *)LOAD_ADDR;
#if defined(CONFIG_FIT)
	if (gen_image_get_format ((void *)hdr) != IMAGE_FORMAT_LEGACY) {
		puts ("Non legacy image format not supported\n");
		return -1;
	}
#endif

	/* execute a script */
	if (image_check_type (hdr, IH_TYPE_SCRIPT)) {
		addr = (char *)((char *)hdr + image_get_header_size ());
		/* stick a NULL at the end of the script, otherwise */
		/* parse_string_outer() runs off the end. */
		addr[image_get_data_size (hdr)] = 0;
		addr += 8;
		parse_string_outer(addr, FLAG_PARSE_SEMICOLON);
		return 0;
	}

	start = aufl_layout[idx].start;
	end = aufl_layout[idx].end;

	/* unprotect the address range */
	/* this assumes that ONLY the firmware is protected! */
	if (idx == IDX_FIRMWARE) {
#undef AU_UPDATE_TEST
#ifdef AU_UPDATE_TEST
		/* erase it where Linux goes */
		start = aufl_layout[1].start;
		end = aufl_layout[1].end;
#endif
		flash_sect_protect(0, start, end);
	}

	/*
	 * erase the address range.
	 */
	debug ("flash_sect_erase(%lx, %lx);\n", start, end);
	flash_sect_erase(start, end);
	wait_ms(100);
#ifdef CONFIG_PROGRESSBAR
	show_progress(end - start, totsize);
#endif

	/* strip the header - except for the kernel and ramdisk */
	if (image_check_type (hdr, IH_TYPE_KERNEL) ||
			image_check_type (hdr, IH_TYPE_RAMDISK)) {
		addr = (char *)hdr;
		off = image_get_header_size ();
		nbytes = image_get_image_size (hdr);
	} else {
		addr = (char *)((char *)hdr + image_get_header_size ());
#ifdef AU_UPDATE_TEST
		/* copy it to where Linux goes */
		if (idx == IDX_FIRMWARE)
			start = aufl_layout[1].start;
#endif
		off = 0;
		nbytes = image_get_data_size (hdr);
	}

	/* copy the data from RAM to FLASH */
	debug ("flash_write(%p, %lx %x)\n", addr, start, nbytes);
	rc = flash_write(addr, start, nbytes);
	if (rc != 0) {
		printf("Flashing failed due to error %d\n", rc);
		return -1;
	}

#ifdef CONFIG_PROGRESSBAR
	show_progress(nbytes, totsize);
#endif

	/* check the data CRC of the copy */
	if (crc32 (0, (uchar *)(start + off), image_get_data_size (hdr)) !=
	    image_get_dcrc (hdr)) {
		printf ("Image %s Bad Data Checksum after COPY\n", aufile[idx]);
		return -1;
	}

	/* protect the address range */
	/* this assumes that ONLY the firmware is protected! */
	if (idx == IDX_FIRMWARE)
		flash_sect_protect(1, start, end);
	return 0;
}

/*
 * this is called from board_init() after the hardware has been set up
 * and is usable. That seems like a good time to do this.
 * Right now the return value is ignored.
 */
int do_auto_update(void)
{
	block_dev_desc_t *stor_dev;
	long sz;
	int i, res = 0, bitmap_first, cnt, old_ctrlc, got_ctrlc;
	char *env;
	long start, end;

#if 0 /* disable key-press detection to speed up boot-up time */
	uchar keypad_status1[2] = {0,0}, keypad_status2[2] = {0,0};

	/*
	 * Read keypad status
	 */
	i2c_read(I2C_PSOC_KEYPAD_ADDR, 0, 0, keypad_status1, 2);
	wait_ms(500);
	i2c_read(I2C_PSOC_KEYPAD_ADDR, 0, 0, keypad_status2, 2);

	/*
	 * Check keypad
	 */
	if ( !(keypad_status1[1] & KEYPAD_MASK_LO) ||
	      (keypad_status1[1] != keypad_status2[1])) {
		return 0;
	}

#endif
	au_usb_stor_curr_dev = -1;
	/* start USB */
	if (usb_stop() < 0) {
		debug ("usb_stop failed\n");
		return -1;
	}
	if (usb_init() < 0) {
		debug ("usb_init failed\n");
		return -1;
	}
	/*
	 * check whether a storage device is attached (assume that it's
	 * a USB memory stick, since nothing else should be attached).
	 */
	au_usb_stor_curr_dev = usb_stor_scan(0);
	if (au_usb_stor_curr_dev == -1) {
		debug ("No device found. Not initialized?\n");
		res = -1;
		goto xit;
	}
	/* check whether it has a partition table */
	stor_dev = get_dev("usb", 0);
	if (stor_dev == NULL) {
		debug ("uknown device type\n");
		res = -1;
		goto xit;
	}
	if (fat_register_device(stor_dev, 1) != 0) {
		debug ("Unable to use USB %d:%d for fatls\n",
			au_usb_stor_curr_dev, 1);
		res = -1;
		goto xit;
	}
	if (file_fat_detectfs() != 0) {
		debug ("file_fat_detectfs failed\n");
	}

	/*
	 * now check whether start and end are defined using environment
	 * variables.
	 */
	start = -1;
	end = 0;
	env = getenv("firmware_st");
	if (env != NULL)
		start = simple_strtoul(env, NULL, 16);
	env = getenv("firmware_nd");
	if (env != NULL)
		end = simple_strtoul(env, NULL, 16);
	if (start >= 0 && end && end > start) {
		ausize[IDX_FIRMWARE] = (end + 1) - start;
		aufl_layout[IDX_FIRMWARE].start = start;
		aufl_layout[IDX_FIRMWARE].end = end;
	}
	start = -1;
	end = 0;
	env = getenv("kernel_st");
	if (env != NULL)
		start = simple_strtoul(env, NULL, 16);
	env = getenv("kernel_nd");
	if (env != NULL)
		end = simple_strtoul(env, NULL, 16);
	if (start >= 0 && end && end > start) {
		ausize[IDX_KERNEL] = (end + 1) - start;
		aufl_layout[IDX_KERNEL].start = start;
		aufl_layout[IDX_KERNEL].end = end;
	}
	start = -1;
	end = 0;
	env = getenv("rootfs_st");
	if (env != NULL)
		start = simple_strtoul(env, NULL, 16);
	env = getenv("rootfs_nd");
	if (env != NULL)
		end = simple_strtoul(env, NULL, 16);
	if (start >= 0 && end && end > start) {
		ausize[IDX_ROOTFS] = (end + 1) - start;
		aufl_layout[IDX_ROOTFS].start = start;
		aufl_layout[IDX_ROOTFS].end = end;
	}

	/* make certain that HUSH is runnable */
	u_boot_hush_start();
	/* make sure that we see CTRL-C and save the old state */
	old_ctrlc = disable_ctrlc(0);

	bitmap_first = 0;

	/* validate the images first */
	for (i = 0; i < AU_MAXFILES; i++) {
		ulong imsize;
		/* just read the header */
		sz = file_fat_read(aufile[i], LOAD_ADDR, image_get_header_size ());
		debug ("read %s sz %ld hdr %d\n",
			aufile[i], sz, image_get_header_size ());
		if (sz <= 0 || sz < image_get_header_size ()) {
			debug ("%s not found\n", aufile[i]);
			ausize[i] = 0;
			continue;
		}
		/* au_check_header_valid() updates ausize[] */
		if ((imsize = au_check_header_valid(i, sz)) < 0) {
			debug ("%s header not valid\n", aufile[i]);
			continue;
		}
		/* totsize accounts for image size and flash erase size */
		totsize += (imsize + (aufl_layout[i].end - aufl_layout[i].start));
	}

#ifdef CONFIG_PROGRESSBAR
	if (totsize) {
		lcd_puts(" Update in progress\n");
		lcd_enable();
	}
#endif

	/* just loop thru all the possible files */
	for (i = 0; i < AU_MAXFILES && totsize; i++) {
		if (!ausize[i]) {
			continue;
		}
		sz = file_fat_read(aufile[i], LOAD_ADDR, ausize[i]);

		debug ("read %s sz %ld hdr %d\n",
			aufile[i], sz, image_get_header_size ());

		if (sz != ausize[i]) {
			printf ("%s: size %d read %d?\n", aufile[i], ausize[i], sz);
			continue;
		}

		if (sz <= 0 || sz <= image_get_header_size ()) {
			debug ("%s not found\n", aufile[i]);
			continue;
		}
		if (au_check_cksum_valid(i, sz) < 0) {
			debug ("%s checksum not valid\n", aufile[i]);
			continue;
		}
		/* this is really not a good idea, but it's what the */
		/* customer wants. */
		cnt = 0;
		got_ctrlc = 0;
		do {
			res = au_do_update(i, sz);
			/* let the user break out of the loop */
			if (ctrlc() || had_ctrlc()) {
				clear_ctrlc();
				if (res < 0)
					got_ctrlc = 1;
				break;
			}
			cnt++;
#ifdef AU_TEST_ONLY
		} while (res < 0 && cnt < (AU_MAXFILES + 1));
		if (cnt < (AU_MAXFILES + 1))
#else
		} while (res < 0);
#endif
	}

	/* restore the old state */
	disable_ctrlc(old_ctrlc);
#ifdef CONFIG_PROGRESSBAR
	if (totsize) {
		if (!res) {
			lcd_puts("\n  Update completed\n");
		} else {
			lcd_puts("\n   Update error\n");
		}
		lcd_enable();
	}
#endif
 xit:
	usb_stop();
	return res;
}
#endif /* CONFIG_AUTO_UPDATE */
Commit	Line	Data
cdb97a66 AS	1	/*
	2	* (C) Copyright 2006
	3	* Wolfgang Denk, DENX Software Engineering, wd@denx.de.
	4	*
	5	* This program is free software; you can redistribute it and/or
	6	* modify it under the terms of the GNU General Public License as
	7	* published by the Free Software Foundation; either version 2 of
	8	* the License, or (at your option) any later version.
	9	*
	10	* This program is distributed in the hope that it will be useful,
	11	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	13	* GNU General Public License for more details.
	14	*
	15	* You should have received a copy of the GNU General Public License
	16	* along with this program; if not, write to the Free Software
	17	* Foundation, Inc., 59 Temple Place, Suite 330, Boston,
	18	* MA 02111-1307 USA
	19	*/
	20	#include <common.h>
	21	#include <command.h>
	22	#include <malloc.h>
	23	#include <image.h>
	24	#include <asm/byteorder.h>
	25	#include <usb.h>
735dd97b	26	#include <part.h>
cdb97a66 AS	27
	28	#ifdef CFG_HUSH_PARSER
	29	#include <hush.h>
	30	#endif
	31
	32
	33	#ifdef CONFIG_AUTO_UPDATE
	34
	35	#ifndef CONFIG_USB_OHCI
	36	#error "must define CONFIG_USB_OHCI"
	37	#endif
	38
	39	#ifndef CONFIG_USB_STORAGE
	40	#error "must define CONFIG_USB_STORAGE"
	41	#endif
	42
	43	#ifndef CFG_HUSH_PARSER
	44	#error "must define CFG_HUSH_PARSER"
	45	#endif
	46
3fe00109	47	#if !defined(CONFIG_CMD_FAT)
d39b5741	48	#error "must define CONFIG_CMD_FAT"
cdb97a66 AS	49	#endif
cdb97a66 AS	50
cdb97a66 AS	51	#undef AU_DEBUG
	52
	53	#undef debug
	54	#ifdef AU_DEBUG
	55	#define debug(fmt,args...) printf (fmt ,##args)
	56	#else
	57	#define debug(fmt,args...)
	58	#endif /* AU_DEBUG */
	59
	60	/* possible names of files on the USB stick. */
	61	#define AU_FIRMWARE "u-boot.img"
	62	#define AU_KERNEL "kernel.img"
489c696a	63	#define AU_ROOTFS "rootfs.img"
cdb97a66	64
82e5236a	65	struct flash_layout {
cdb97a66 AS	66	long start;
	67	long end;
	68	};
	69
	70	/* layout of the FLASH. ST = start address, ND = end address. */
	71	#define AU_FL_FIRMWARE_ST 0xfC000000
	72	#define AU_FL_FIRMWARE_ND 0xfC03FFFF
	73	#define AU_FL_KERNEL_ST 0xfC0C0000
	74	#define AU_FL_KERNEL_ND 0xfC1BFFFF
489c696a SP	75	#define AU_FL_ROOTFS_ST 0xFC1C0000
489c696a SP	76	#define AU_FL_ROOTFS_ND 0xFCFBFFFF
cdb97a66 AS	77
	78	static int au_usb_stor_curr_dev; /* current device */
	79
	80	/* index of each file in the following arrays */
	81	#define IDX_FIRMWARE 0
	82	#define IDX_KERNEL 1
489c696a	83	#define IDX_ROOTFS 2
cdb97a66 AS	84
cdb97a66 AS	85	/* max. number of files which could interest us */
489c696a	86	#define AU_MAXFILES 3
cdb97a66 AS	87
cdb97a66 AS	88	/* pointers to file names */
489c696a SP	89	char *aufile[AU_MAXFILES] = {
	90	AU_FIRMWARE,
	91	AU_KERNEL,
	92	AU_ROOTFS
	93	};
cdb97a66 AS	94
cdb97a66 AS	95	/* sizes of flash areas for each file */
489c696a SP	96	long ausize[AU_MAXFILES] = {
489c696a SP	97	(AU_FL_FIRMWARE_ND + 1) - AU_FL_FIRMWARE_ST,
f5fcc3c2 WD	98	(AU_FL_KERNEL_ND + 1) - AU_FL_KERNEL_ST,
f5fcc3c2 WD	99	(AU_FL_ROOTFS_ND + 1) - AU_FL_ROOTFS_ST,
489c696a	100	};
cdb97a66 AS	101
cdb97a66 AS	102	/* array of flash areas start and end addresses */
489c696a	103	struct flash_layout aufl_layout[AU_MAXFILES] = {
f5fcc3c2 WD	104	{ AU_FL_FIRMWARE_ST, AU_FL_FIRMWARE_ND, },
	105	{ AU_FL_KERNEL_ST, AU_FL_KERNEL_ND, },
	106	{ AU_FL_ROOTFS_ST, AU_FL_ROOTFS_ND, },
cdb97a66 AS	107	};
cdb97a66 AS	108
638dd145 SP	109	ulong totsize;
638dd145 SP	110
cdb97a66 AS	111	/* where to load files into memory */
	112	#define LOAD_ADDR ((unsigned char *)0x00200000)
	113
f5fcc3c2	114	/* the root file system is the largest image */
489c696a	115	#define MAX_LOADSZ ausize[IDX_ROOTFS]
cdb97a66 AS	116
	117	/i2c address of the keypad status/
	118	#define I2C_PSOC_KEYPAD_ADDR 0x53
	119
	120	/* keypad mask */
787fa158 WD	121	#define KEYPAD_ROW 2
	122	#define KEYPAD_COL 2
	123	#define KEYPAD_MASK_LO ((1<<(KEYPAD_COL-1+(KEYPAD_ROW*3-3)))&0xFF)
	124	#define KEYPAD_MASK_HI ((1<<(KEYPAD_COL-1+(KEYPAD_ROW*3-3)))>>8)
cdb97a66 AS	125
	126	/* externals */
	127	extern int fat_register_device(block_dev_desc_t *, int);
	128	extern int file_fat_detectfs(void);
	129	extern long file_fat_read(const char , void , unsigned long);
	130	extern int i2c_read (unsigned char, unsigned int, int , unsigned char* , int);
	131	extern int flash_sect_erase(ulong, ulong);
	132	extern int flash_sect_protect (int, ulong, ulong);
	133	extern int flash_write (char *, ulong, ulong);
cdb97a66	134	extern int u_boot_hush_start(void);
638dd145 SP	135	#ifdef CONFIG_PROGRESSBAR
	136	extern void show_progress(int, int);
	137	extern void lcd_puts (char *);
	138	extern void lcd_enable(void);
	139	#endif
cdb97a66	140
82e5236a	141	int au_check_cksum_valid(int idx, long nbytes)
cdb97a66 AS	142	{
cdb97a66 AS	143	image_header_t *hdr;
cdb97a66 AS	144
cdb97a66 AS	145	hdr = (image_header_t *)LOAD_ADDR;
d5934ad7 MB	146	#if defined(CONFIG_FIT)
	147	if (gen_image_get_format ((void *)hdr) != IMAGE_FORMAT_LEGACY) {
	148	puts ("Non legacy image format not supported\n");
	149	return -1;
	150	}
	151	#endif
cdb97a66	152
b97a2a0a	153	if (nbytes != image_get_image_size (hdr)) {
cdb97a66 AS	154	printf ("Image %s bad total SIZE\n", aufile[idx]);
	155	return -1;
	156	}
	157	/* check the data CRC */
b97a2a0a	158	if (!image_check_dcrc (hdr)) {
cdb97a66 AS	159	printf ("Image %s bad data checksum\n", aufile[idx]);
	160	return -1;
	161	}
	162	return 0;
	163	}
	164
82e5236a	165	int au_check_header_valid(int idx, long nbytes)
cdb97a66 AS	166	{
cdb97a66 AS	167	image_header_t *hdr;
e344568b	168	unsigned long checksum, fsize;
cdb97a66 AS	169
cdb97a66 AS	170	hdr = (image_header_t *)LOAD_ADDR;
d5934ad7 MB	171	#if defined(CONFIG_FIT)
	172	if (gen_image_get_format ((void *)hdr) != IMAGE_FORMAT_LEGACY) {
	173	puts ("Non legacy image format not supported\n");
	174	return -1;
	175	}
	176	#endif
	177
cdb97a66 AS	178	/* check the easy ones first */
	179	#undef CHECK_VALID_DEBUG
	180	#ifdef CHECK_VALID_DEBUG
b97a2a0a MB	181	printf("magic %#x %#x ", image_get_magic (hdr), IH_MAGIC);
	182	printf("arch %#x %#x ", image_get_arch (hdr), IH_ARCH_ARM);
	183	printf("size %#x %#lx ", image_get_data_size (hdr), nbytes);
	184	printf("type %#x %#x ", image_get_type (hdr), IH_TYPE_KERNEL);
cdb97a66	185	#endif
b97a2a0a	186	if (nbytes < image_get_header_size ()) {
cdb97a66	187	printf ("Image %s bad header SIZE\n", aufile[idx]);
638dd145	188	ausize[idx] = 0;
cdb97a66 AS	189	return -1;
cdb97a66 AS	190	}
b97a2a0a	191	if (!image_check_magic (hdr) \|\| !image_check_arch (hdr, IH_ARCH_PPC)) {
cdb97a66	192	printf ("Image %s bad MAGIC or ARCH\n", aufile[idx]);
638dd145	193	ausize[idx] = 0;
cdb97a66 AS	194	return -1;
	195	}
	196	/* check the hdr CRC */
b97a2a0a	197	if (!image_check_hcrc (hdr)) {
cdb97a66	198	printf ("Image %s bad header checksum\n", aufile[idx]);
638dd145	199	ausize[idx] = 0;
cdb97a66 AS	200	return -1;
cdb97a66 AS	201	}
cdb97a66	202	/* check the type - could do this all in one gigantic if() */
b97a2a0a	203	if ((idx == IDX_FIRMWARE) && !image_check_type (hdr, IH_TYPE_FIRMWARE)) {
cdb97a66	204	printf ("Image %s wrong type\n", aufile[idx]);
638dd145	205	ausize[idx] = 0;
cdb97a66 AS	206	return -1;
cdb97a66 AS	207	}
b97a2a0a	208	if ((idx == IDX_KERNEL) && !image_check_type (hdr, IH_TYPE_KERNEL)) {
cdb97a66	209	printf ("Image %s wrong type\n", aufile[idx]);
638dd145	210	ausize[idx] = 0;
cdb97a66 AS	211	return -1;
cdb97a66 AS	212	}
f5fcc3c2	213	if ((idx == IDX_ROOTFS) &&
b97a2a0a MB	214	(!image_check_type (hdr, IH_TYPE_RAMDISK) &&
b97a2a0a MB	215	!image_check_type (hdr, IH_TYPE_FILESYSTEM))) {
489c696a	216	printf ("Image %s wrong type\n", aufile[idx]);
638dd145	217	ausize[idx] = 0;
489c696a SP	218	return -1;
489c696a SP	219	}
cdb97a66	220	/* recycle checksum */
b97a2a0a	221	checksum = image_get_data_size (hdr);
e344568b	222
b97a2a0a	223	fsize = checksum + image_get_header_size ();
e344568b	224	/* for kernel and ramdisk the image header must also fit into flash */
b97a2a0a MB	225	if (idx == IDX_KERNEL \|\| image_check_type (hdr, IH_TYPE_RAMDISK))
b97a2a0a MB	226	checksum += image_get_header_size ();
638dd145	227
cdb97a66	228	/* check the size does not exceed space in flash. HUSH scripts */
cdb97a66 AS	229	if ((ausize[idx] != 0) && (ausize[idx] < checksum)) {
cdb97a66 AS	230	printf ("Image %s is bigger than FLASH\n", aufile[idx]);
638dd145	231	ausize[idx] = 0;
cdb97a66 AS	232	return -1;
cdb97a66 AS	233	}
638dd145	234	/* Update with the real filesize */
e344568b	235	ausize[idx] = fsize;
638dd145 SP	236
638dd145 SP	237	return checksum; /* return size to be written to flash */
cdb97a66 AS	238	}
cdb97a66 AS	239
82e5236a	240	int au_do_update(int idx, long sz)
cdb97a66 AS	241	{
	242	image_header_t *hdr;
	243	char *addr;
	244	long start, end;
	245	int off, rc;
	246	uint nbytes;
	247
	248	hdr = (image_header_t *)LOAD_ADDR;
d5934ad7 MB	249	#if defined(CONFIG_FIT)
	250	if (gen_image_get_format ((void *)hdr) != IMAGE_FORMAT_LEGACY) {
	251	puts ("Non legacy image format not supported\n");
	252	return -1;
	253	}
	254	#endif
cdb97a66 AS	255
cdb97a66 AS	256	/* execute a script */
b97a2a0a MB	257	if (image_check_type (hdr, IH_TYPE_SCRIPT)) {
b97a2a0a MB	258	addr = (char )((char )hdr + image_get_header_size ());
cdb97a66 AS	259	/* stick a NULL at the end of the script, otherwise */
cdb97a66 AS	260	/* parse_string_outer() runs off the end. */
b97a2a0a	261	addr[image_get_data_size (hdr)] = 0;
cdb97a66 AS	262	addr += 8;
	263	parse_string_outer(addr, FLAG_PARSE_SEMICOLON);
	264	return 0;
	265	}
	266
	267	start = aufl_layout[idx].start;
	268	end = aufl_layout[idx].end;
	269
	270	/* unprotect the address range */
	271	/* this assumes that ONLY the firmware is protected! */
	272	if (idx == IDX_FIRMWARE) {
	273	#undef AU_UPDATE_TEST
	274	#ifdef AU_UPDATE_TEST
	275	/* erase it where Linux goes */
	276	start = aufl_layout[1].start;
	277	end = aufl_layout[1].end;
	278	#endif
	279	flash_sect_protect(0, start, end);
	280	}
	281
	282	/*
	283	* erase the address range.
	284	*/
	285	debug ("flash_sect_erase(%lx, %lx);\n", start, end);
	286	flash_sect_erase(start, end);
	287	wait_ms(100);
638dd145 SP	288	#ifdef CONFIG_PROGRESSBAR
	289	show_progress(end - start, totsize);
	290	#endif
	291
cdb97a66	292	/* strip the header - except for the kernel and ramdisk */
b97a2a0a MB	293	if (image_check_type (hdr, IH_TYPE_KERNEL) \|\|
b97a2a0a MB	294	image_check_type (hdr, IH_TYPE_RAMDISK)) {
cdb97a66	295	addr = (char *)hdr;
b97a2a0a MB	296	off = image_get_header_size ();
b97a2a0a MB	297	nbytes = image_get_image_size (hdr);
cdb97a66	298	} else {
b97a2a0a	299	addr = (char )((char )hdr + image_get_header_size ());
cdb97a66 AS	300	#ifdef AU_UPDATE_TEST
	301	/* copy it to where Linux goes */
	302	if (idx == IDX_FIRMWARE)
	303	start = aufl_layout[1].start;
	304	#endif
	305	off = 0;
b97a2a0a	306	nbytes = image_get_data_size (hdr);
cdb97a66 AS	307	}
	308
	309	/* copy the data from RAM to FLASH */
	310	debug ("flash_write(%p, %lx %x)\n", addr, start, nbytes);
	311	rc = flash_write(addr, start, nbytes);
	312	if (rc != 0) {
	313	printf("Flashing failed due to error %d\n", rc);
	314	return -1;
	315	}
	316
638dd145 SP	317	#ifdef CONFIG_PROGRESSBAR
	318	show_progress(nbytes, totsize);
	319	#endif
	320
f5fcc3c2	321	/* check the data CRC of the copy */
b97a2a0a MB	322	if (crc32 (0, (uchar *)(start + off), image_get_data_size (hdr)) !=
b97a2a0a MB	323	image_get_dcrc (hdr)) {
f5fcc3c2	324	printf ("Image %s Bad Data Checksum after COPY\n", aufile[idx]);
cdb97a66 AS	325	return -1;
	326	}
	327
	328	/* protect the address range */
	329	/* this assumes that ONLY the firmware is protected! */
	330	if (idx == IDX_FIRMWARE)
	331	flash_sect_protect(1, start, end);
	332	return 0;
	333	}
	334
cdb97a66 AS	335	/*
	336	* this is called from board_init() after the hardware has been set up
	337	* and is usable. That seems like a good time to do this.
	338	* Right now the return value is ignored.
	339	*/
82e5236a	340	int do_auto_update(void)
cdb97a66 AS	341	{
	342	block_dev_desc_t *stor_dev;
	343	long sz;
638dd145	344	int i, res = 0, bitmap_first, cnt, old_ctrlc, got_ctrlc;
cdb97a66 AS	345	char *env;
cdb97a66 AS	346	long start, end;
a4d2636f WD	347
a4d2636f WD	348	#if 0 /* disable key-press detection to speed up boot-up time */
489c696a	349	uchar keypad_status1[2] = {0,0}, keypad_status2[2] = {0,0};
cdb97a66 AS	350
	351	/*
	352	* Read keypad status
	353	*/
	354	i2c_read(I2C_PSOC_KEYPAD_ADDR, 0, 0, keypad_status1, 2);
	355	wait_ms(500);
	356	i2c_read(I2C_PSOC_KEYPAD_ADDR, 0, 0, keypad_status2, 2);
	357
	358	/*
	359	* Check keypad
	360	*/
cdb97a66 AS	361	if ( !(keypad_status1[1] & KEYPAD_MASK_LO) \|\|
	362	(keypad_status1[1] != keypad_status2[1])) {
	363	return 0;
	364	}
638dd145	365
a4d2636f	366	#endif
cdb97a66 AS	367	au_usb_stor_curr_dev = -1;
	368	/* start USB */
	369	if (usb_stop() < 0) {
	370	debug ("usb_stop failed\n");
	371	return -1;
	372	}
	373	if (usb_init() < 0) {
	374	debug ("usb_init failed\n");
	375	return -1;
	376	}
	377	/*
	378	* check whether a storage device is attached (assume that it's
	379	* a USB memory stick, since nothing else should be attached).
	380	*/
	381	au_usb_stor_curr_dev = usb_stor_scan(0);
	382	if (au_usb_stor_curr_dev == -1) {
	383	debug ("No device found. Not initialized?\n");
a4d2636f WD	384	res = -1;
a4d2636f WD	385	goto xit;
cdb97a66 AS	386	}
	387	/* check whether it has a partition table */
	388	stor_dev = get_dev("usb", 0);
	389	if (stor_dev == NULL) {
	390	debug ("uknown device type\n");
a4d2636f WD	391	res = -1;
a4d2636f WD	392	goto xit;
cdb97a66 AS	393	}
	394	if (fat_register_device(stor_dev, 1) != 0) {
	395	debug ("Unable to use USB %d:%d for fatls\n",
	396	au_usb_stor_curr_dev, 1);
a4d2636f WD	397	res = -1;
a4d2636f WD	398	goto xit;
cdb97a66 AS	399	}
	400	if (file_fat_detectfs() != 0) {
	401	debug ("file_fat_detectfs failed\n");
	402	}
	403
cdb97a66 AS	404	/*
	405	* now check whether start and end are defined using environment
	406	* variables.
	407	*/
	408	start = -1;
	409	end = 0;
	410	env = getenv("firmware_st");
	411	if (env != NULL)
	412	start = simple_strtoul(env, NULL, 16);
	413	env = getenv("firmware_nd");
	414	if (env != NULL)
	415	end = simple_strtoul(env, NULL, 16);
	416	if (start >= 0 && end && end > start) {
	417	ausize[IDX_FIRMWARE] = (end + 1) - start;
489c696a SP	418	aufl_layout[IDX_FIRMWARE].start = start;
489c696a SP	419	aufl_layout[IDX_FIRMWARE].end = end;
cdb97a66 AS	420	}
	421	start = -1;
	422	end = 0;
	423	env = getenv("kernel_st");
	424	if (env != NULL)
	425	start = simple_strtoul(env, NULL, 16);
	426	env = getenv("kernel_nd");
	427	if (env != NULL)
	428	end = simple_strtoul(env, NULL, 16);
	429	if (start >= 0 && end && end > start) {
	430	ausize[IDX_KERNEL] = (end + 1) - start;
489c696a SP	431	aufl_layout[IDX_KERNEL].start = start;
489c696a SP	432	aufl_layout[IDX_KERNEL].end = end;
cdb97a66	433	}
489c696a SP	434	start = -1;
	435	end = 0;
	436	env = getenv("rootfs_st");
	437	if (env != NULL)
	438	start = simple_strtoul(env, NULL, 16);
	439	env = getenv("rootfs_nd");
	440	if (env != NULL)
	441	end = simple_strtoul(env, NULL, 16);
	442	if (start >= 0 && end && end > start) {
	443	ausize[IDX_ROOTFS] = (end + 1) - start;
	444	aufl_layout[IDX_ROOTFS].start = start;
	445	aufl_layout[IDX_ROOTFS].end = end;
	446	}
	447
cdb97a66 AS	448	/* make certain that HUSH is runnable */
	449	u_boot_hush_start();
	450	/* make sure that we see CTRL-C and save the old state */
	451	old_ctrlc = disable_ctrlc(0);
	452
	453	bitmap_first = 0;
638dd145	454
74357114	455	/* validate the images first */
cdb97a66	456	for (i = 0; i < AU_MAXFILES; i++) {
638dd145	457	ulong imsize;
cdb97a66	458	/* just read the header */
b97a2a0a	459	sz = file_fat_read(aufile[i], LOAD_ADDR, image_get_header_size ());
cdb97a66	460	debug ("read %s sz %ld hdr %d\n",
b97a2a0a MB	461	aufile[i], sz, image_get_header_size ());
b97a2a0a MB	462	if (sz <= 0 \|\| sz < image_get_header_size ()) {
cdb97a66	463	debug ("%s not found\n", aufile[i]);
638dd145	464	ausize[i] = 0;
cdb97a66 AS	465	continue;
cdb97a66 AS	466	}
638dd145 SP	467	/* au_check_header_valid() updates ausize[] */
638dd145 SP	468	if ((imsize = au_check_header_valid(i, sz)) < 0) {
cdb97a66 AS	469	debug ("%s header not valid\n", aufile[i]);
	470	continue;
	471	}
638dd145	472	/* totsize accounts for image size and flash erase size */
74357114	473	totsize += (imsize + (aufl_layout[i].end - aufl_layout[i].start));
638dd145 SP	474	}
	475
	476	#ifdef CONFIG_PROGRESSBAR
	477	if (totsize) {
	478	lcd_puts(" Update in progress\n");
	479	lcd_enable();
	480	}
	481	#endif
	482
	483	/* just loop thru all the possible files */
	484	for (i = 0; i < AU_MAXFILES && totsize; i++) {
	485	if (!ausize[i]) {
	486	continue;
	487	}
	488	sz = file_fat_read(aufile[i], LOAD_ADDR, ausize[i]);
	489
cdb97a66	490	debug ("read %s sz %ld hdr %d\n",
b97a2a0a	491	aufile[i], sz, image_get_header_size ());
638dd145 SP	492
	493	if (sz != ausize[i]) {
	494	printf ("%s: size %d read %d?\n", aufile[i], ausize[i], sz);
74357114 WD	495	continue;
74357114 WD	496	}
638dd145	497
b97a2a0a	498	if (sz <= 0 \|\| sz <= image_get_header_size ()) {
cdb97a66 AS	499	debug ("%s not found\n", aufile[i]);
	500	continue;
	501	}
	502	if (au_check_cksum_valid(i, sz) < 0) {
	503	debug ("%s checksum not valid\n", aufile[i]);
	504	continue;
	505	}
	506	/* this is really not a good idea, but it's what the */
	507	/* customer wants. */
	508	cnt = 0;
	509	got_ctrlc = 0;
	510	do {
	511	res = au_do_update(i, sz);
	512	/* let the user break out of the loop */
	513	if (ctrlc() \|\| had_ctrlc()) {
	514	clear_ctrlc();
	515	if (res < 0)
	516	got_ctrlc = 1;
	517	break;
	518	}
	519	cnt++;
	520	#ifdef AU_TEST_ONLY
489c696a SP	521	} while (res < 0 && cnt < (AU_MAXFILES + 1));
489c696a SP	522	if (cnt < (AU_MAXFILES + 1))
cdb97a66 AS	523	#else
	524	} while (res < 0);
	525	#endif
	526	}
a4d2636f	527
cdb97a66 AS	528	/* restore the old state */
cdb97a66 AS	529	disable_ctrlc(old_ctrlc);
638dd145 SP	530	#ifdef CONFIG_PROGRESSBAR
	531	if (totsize) {
	532	if (!res) {
	533	lcd_puts("\n Update completed\n");
	534	} else {
	535	lcd_puts("\n Update error\n");
	536	}
	537	lcd_enable();
	538	}
	539	#endif
a4d2636f WD	540	xit:
	541	usb_stop();
	542	return res;
cdb97a66 AS	543	}
cdb97a66 AS	544	#endif /* CONFIG_AUTO_UPDATE */