blob: 5580c118878ae4fc76724eb9651311838e6737db [file] [log] [blame]
Andrei Safronovcdb97a62006-12-08 16:23:08 +01001/*
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>
Grant Likely735dd972007-02-20 09:04:34 +010026#include <part.h>
Andrei Safronovcdb97a62006-12-08 16:23:08 +010027
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
Jon Loeliger3fe00102007-07-09 18:38:39 -050047#if !defined(CONFIG_CMD_FAT)
Jon Loeligerd39b5742007-07-10 10:48:22 -050048#error "must define CONFIG_CMD_FAT"
Andrei Safronovcdb97a62006-12-08 16:23:08 +010049#endif
50
Andrei Safronovcdb97a62006-12-08 16:23:08 +010051#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"
Sergei Poselenov489c6962007-02-14 14:30:28 +030063#define AU_ROOTFS "rootfs.img"
Andrei Safronovcdb97a62006-12-08 16:23:08 +010064
Wolfgang Denk82e52362006-12-22 10:30:26 +010065struct flash_layout {
Andrei Safronovcdb97a62006-12-08 16:23:08 +010066 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
Sergei Poselenov489c6962007-02-14 14:30:28 +030075#define AU_FL_ROOTFS_ST 0xFC1C0000
76#define AU_FL_ROOTFS_ND 0xFCFBFFFF
Andrei Safronovcdb97a62006-12-08 16:23:08 +010077
78static 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
Sergei Poselenov489c6962007-02-14 14:30:28 +030083#define IDX_ROOTFS 2
Andrei Safronovcdb97a62006-12-08 16:23:08 +010084
85/* max. number of files which could interest us */
Sergei Poselenov489c6962007-02-14 14:30:28 +030086#define AU_MAXFILES 3
Andrei Safronovcdb97a62006-12-08 16:23:08 +010087
88/* pointers to file names */
Sergei Poselenov489c6962007-02-14 14:30:28 +030089char *aufile[AU_MAXFILES] = {
90 AU_FIRMWARE,
91 AU_KERNEL,
92 AU_ROOTFS
93};
Andrei Safronovcdb97a62006-12-08 16:23:08 +010094
95/* sizes of flash areas for each file */
Sergei Poselenov489c6962007-02-14 14:30:28 +030096long ausize[AU_MAXFILES] = {
97 (AU_FL_FIRMWARE_ND + 1) - AU_FL_FIRMWARE_ST,
Wolfgang Denkf5fcc3c2007-02-19 23:09:51 +010098 (AU_FL_KERNEL_ND + 1) - AU_FL_KERNEL_ST,
99 (AU_FL_ROOTFS_ND + 1) - AU_FL_ROOTFS_ST,
Sergei Poselenov489c6962007-02-14 14:30:28 +0300100};
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100101
102/* array of flash areas start and end addresses */
Sergei Poselenov489c6962007-02-14 14:30:28 +0300103struct flash_layout aufl_layout[AU_MAXFILES] = {
Wolfgang Denkf5fcc3c2007-02-19 23:09:51 +0100104 { 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, },
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100107};
108
Sergei Poselenov638dd142007-02-27 12:40:16 +0300109ulong totsize;
110
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100111/* where to load files into memory */
112#define LOAD_ADDR ((unsigned char *)0x00200000)
113
Wolfgang Denkf5fcc3c2007-02-19 23:09:51 +0100114/* the root file system is the largest image */
Sergei Poselenov489c6962007-02-14 14:30:28 +0300115#define MAX_LOADSZ ausize[IDX_ROOTFS]
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100116
117/*i2c address of the keypad status*/
118#define I2C_PSOC_KEYPAD_ADDR 0x53
119
120/* keypad mask */
Wolfgang Denk787fa152007-01-10 01:28:39 +0100121#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)
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100125
126/* externals */
127extern int fat_register_device(block_dev_desc_t *, int);
128extern int file_fat_detectfs(void);
129extern long file_fat_read(const char *, void *, unsigned long);
130extern int i2c_read (unsigned char, unsigned int, int , unsigned char* , int);
131extern int flash_sect_erase(ulong, ulong);
132extern int flash_sect_protect (int, ulong, ulong);
133extern int flash_write (char *, ulong, ulong);
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100134extern int u_boot_hush_start(void);
Sergei Poselenov638dd142007-02-27 12:40:16 +0300135#ifdef CONFIG_PROGRESSBAR
136extern void show_progress(int, int);
137extern void lcd_puts (char *);
138extern void lcd_enable(void);
139#endif
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100140
Wolfgang Denk82e52362006-12-22 10:30:26 +0100141int au_check_cksum_valid(int idx, long nbytes)
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100142{
143 image_header_t *hdr;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100144
145 hdr = (image_header_t *)LOAD_ADDR;
Marian Balakowiczd5934ad2008-02-04 08:28:09 +0100146#if defined(CONFIG_FIT)
Marian Balakowicz9a4daad2008-02-29 14:58:34 +0100147 if (genimg_get_format ((void *)hdr) != IMAGE_FORMAT_LEGACY) {
Marian Balakowiczd5934ad2008-02-04 08:28:09 +0100148 puts ("Non legacy image format not supported\n");
149 return -1;
150 }
151#endif
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100152
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100153 if (nbytes != image_get_image_size (hdr)) {
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100154 printf ("Image %s bad total SIZE\n", aufile[idx]);
155 return -1;
156 }
157 /* check the data CRC */
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100158 if (!image_check_dcrc (hdr)) {
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100159 printf ("Image %s bad data checksum\n", aufile[idx]);
160 return -1;
161 }
162 return 0;
163}
164
Wolfgang Denk82e52362006-12-22 10:30:26 +0100165int au_check_header_valid(int idx, long nbytes)
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100166{
167 image_header_t *hdr;
Sergei Poselenove3445682007-02-27 20:15:30 +0300168 unsigned long checksum, fsize;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100169
170 hdr = (image_header_t *)LOAD_ADDR;
Marian Balakowiczd5934ad2008-02-04 08:28:09 +0100171#if defined(CONFIG_FIT)
Marian Balakowicz9a4daad2008-02-29 14:58:34 +0100172 if (genimg_get_format ((void *)hdr) != IMAGE_FORMAT_LEGACY) {
Marian Balakowiczd5934ad2008-02-04 08:28:09 +0100173 puts ("Non legacy image format not supported\n");
174 return -1;
175 }
176#endif
177
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100178 /* check the easy ones first */
179#undef CHECK_VALID_DEBUG
180#ifdef CHECK_VALID_DEBUG
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100181 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);
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100185#endif
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100186 if (nbytes < image_get_header_size ()) {
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100187 printf ("Image %s bad header SIZE\n", aufile[idx]);
Sergei Poselenov638dd142007-02-27 12:40:16 +0300188 ausize[idx] = 0;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100189 return -1;
190 }
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100191 if (!image_check_magic (hdr) || !image_check_arch (hdr, IH_ARCH_PPC)) {
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100192 printf ("Image %s bad MAGIC or ARCH\n", aufile[idx]);
Sergei Poselenov638dd142007-02-27 12:40:16 +0300193 ausize[idx] = 0;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100194 return -1;
195 }
196 /* check the hdr CRC */
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100197 if (!image_check_hcrc (hdr)) {
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100198 printf ("Image %s bad header checksum\n", aufile[idx]);
Sergei Poselenov638dd142007-02-27 12:40:16 +0300199 ausize[idx] = 0;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100200 return -1;
201 }
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100202 /* check the type - could do this all in one gigantic if() */
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100203 if ((idx == IDX_FIRMWARE) && !image_check_type (hdr, IH_TYPE_FIRMWARE)) {
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100204 printf ("Image %s wrong type\n", aufile[idx]);
Sergei Poselenov638dd142007-02-27 12:40:16 +0300205 ausize[idx] = 0;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100206 return -1;
207 }
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100208 if ((idx == IDX_KERNEL) && !image_check_type (hdr, IH_TYPE_KERNEL)) {
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100209 printf ("Image %s wrong type\n", aufile[idx]);
Sergei Poselenov638dd142007-02-27 12:40:16 +0300210 ausize[idx] = 0;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100211 return -1;
212 }
Wolfgang Denkf5fcc3c2007-02-19 23:09:51 +0100213 if ((idx == IDX_ROOTFS) &&
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100214 (!image_check_type (hdr, IH_TYPE_RAMDISK) &&
215 !image_check_type (hdr, IH_TYPE_FILESYSTEM))) {
Sergei Poselenov489c6962007-02-14 14:30:28 +0300216 printf ("Image %s wrong type\n", aufile[idx]);
Sergei Poselenov638dd142007-02-27 12:40:16 +0300217 ausize[idx] = 0;
Sergei Poselenov489c6962007-02-14 14:30:28 +0300218 return -1;
219 }
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100220 /* recycle checksum */
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100221 checksum = image_get_data_size (hdr);
Sergei Poselenove3445682007-02-27 20:15:30 +0300222
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100223 fsize = checksum + image_get_header_size ();
Sergei Poselenove3445682007-02-27 20:15:30 +0300224 /* for kernel and ramdisk the image header must also fit into flash */
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100225 if (idx == IDX_KERNEL || image_check_type (hdr, IH_TYPE_RAMDISK))
226 checksum += image_get_header_size ();
Sergei Poselenov638dd142007-02-27 12:40:16 +0300227
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100228 /* check the size does not exceed space in flash. HUSH scripts */
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100229 if ((ausize[idx] != 0) && (ausize[idx] < checksum)) {
230 printf ("Image %s is bigger than FLASH\n", aufile[idx]);
Sergei Poselenov638dd142007-02-27 12:40:16 +0300231 ausize[idx] = 0;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100232 return -1;
233 }
Sergei Poselenov638dd142007-02-27 12:40:16 +0300234 /* Update with the real filesize */
Sergei Poselenove3445682007-02-27 20:15:30 +0300235 ausize[idx] = fsize;
Sergei Poselenov638dd142007-02-27 12:40:16 +0300236
237 return checksum; /* return size to be written to flash */
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100238}
239
Wolfgang Denk82e52362006-12-22 10:30:26 +0100240int au_do_update(int idx, long sz)
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100241{
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;
Marian Balakowiczd5934ad2008-02-04 08:28:09 +0100249#if defined(CONFIG_FIT)
Marian Balakowicz9a4daad2008-02-29 14:58:34 +0100250 if (genimg_get_format ((void *)hdr) != IMAGE_FORMAT_LEGACY) {
Marian Balakowiczd5934ad2008-02-04 08:28:09 +0100251 puts ("Non legacy image format not supported\n");
252 return -1;
253 }
254#endif
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100255
256 /* execute a script */
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100257 if (image_check_type (hdr, IH_TYPE_SCRIPT)) {
258 addr = (char *)((char *)hdr + image_get_header_size ());
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100259 /* stick a NULL at the end of the script, otherwise */
260 /* parse_string_outer() runs off the end. */
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100261 addr[image_get_data_size (hdr)] = 0;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100262 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);
Sergei Poselenov638dd142007-02-27 12:40:16 +0300288#ifdef CONFIG_PROGRESSBAR
289 show_progress(end - start, totsize);
290#endif
291
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100292 /* strip the header - except for the kernel and ramdisk */
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100293 if (image_check_type (hdr, IH_TYPE_KERNEL) ||
294 image_check_type (hdr, IH_TYPE_RAMDISK)) {
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100295 addr = (char *)hdr;
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100296 off = image_get_header_size ();
297 nbytes = image_get_image_size (hdr);
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100298 } else {
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100299 addr = (char *)((char *)hdr + image_get_header_size ());
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100300#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;
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100306 nbytes = image_get_data_size (hdr);
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100307 }
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
Sergei Poselenov638dd142007-02-27 12:40:16 +0300317#ifdef CONFIG_PROGRESSBAR
318 show_progress(nbytes, totsize);
319#endif
320
Wolfgang Denkf5fcc3c2007-02-19 23:09:51 +0100321 /* check the data CRC of the copy */
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100322 if (crc32 (0, (uchar *)(start + off), image_get_data_size (hdr)) !=
323 image_get_dcrc (hdr)) {
Wolfgang Denkf5fcc3c2007-02-19 23:09:51 +0100324 printf ("Image %s Bad Data Checksum after COPY\n", aufile[idx]);
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100325 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
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100335/*
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 */
Wolfgang Denk82e52362006-12-22 10:30:26 +0100340int do_auto_update(void)
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100341{
342 block_dev_desc_t *stor_dev;
343 long sz;
Sergei Poselenov638dd142007-02-27 12:40:16 +0300344 int i, res = 0, bitmap_first, cnt, old_ctrlc, got_ctrlc;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100345 char *env;
346 long start, end;
Wolfgang Denka4d26362007-08-12 15:11:38 +0200347
348#if 0 /* disable key-press detection to speed up boot-up time */
Sergei Poselenov489c6962007-02-14 14:30:28 +0300349 uchar keypad_status1[2] = {0,0}, keypad_status2[2] = {0,0};
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100350
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 */
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100361 if ( !(keypad_status1[1] & KEYPAD_MASK_LO) ||
362 (keypad_status1[1] != keypad_status2[1])) {
363 return 0;
364 }
Sergei Poselenov638dd142007-02-27 12:40:16 +0300365
Wolfgang Denka4d26362007-08-12 15:11:38 +0200366#endif
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100367 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");
Wolfgang Denka4d26362007-08-12 15:11:38 +0200384 res = -1;
385 goto xit;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100386 }
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");
Wolfgang Denka4d26362007-08-12 15:11:38 +0200391 res = -1;
392 goto xit;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100393 }
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);
Wolfgang Denka4d26362007-08-12 15:11:38 +0200397 res = -1;
398 goto xit;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100399 }
400 if (file_fat_detectfs() != 0) {
401 debug ("file_fat_detectfs failed\n");
402 }
403
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100404 /*
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;
Sergei Poselenov489c6962007-02-14 14:30:28 +0300418 aufl_layout[IDX_FIRMWARE].start = start;
419 aufl_layout[IDX_FIRMWARE].end = end;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100420 }
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;
Sergei Poselenov489c6962007-02-14 14:30:28 +0300431 aufl_layout[IDX_KERNEL].start = start;
432 aufl_layout[IDX_KERNEL].end = end;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100433 }
Sergei Poselenov489c6962007-02-14 14:30:28 +0300434 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
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100448 /* 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;
Sergei Poselenov638dd142007-02-27 12:40:16 +0300454
Wolfgang Denk74357112007-02-27 14:26:04 +0100455 /* validate the images first */
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100456 for (i = 0; i < AU_MAXFILES; i++) {
Sergei Poselenov638dd142007-02-27 12:40:16 +0300457 ulong imsize;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100458 /* just read the header */
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100459 sz = file_fat_read(aufile[i], LOAD_ADDR, image_get_header_size ());
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100460 debug ("read %s sz %ld hdr %d\n",
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100461 aufile[i], sz, image_get_header_size ());
462 if (sz <= 0 || sz < image_get_header_size ()) {
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100463 debug ("%s not found\n", aufile[i]);
Sergei Poselenov638dd142007-02-27 12:40:16 +0300464 ausize[i] = 0;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100465 continue;
466 }
Sergei Poselenov638dd142007-02-27 12:40:16 +0300467 /* au_check_header_valid() updates ausize[] */
468 if ((imsize = au_check_header_valid(i, sz)) < 0) {
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100469 debug ("%s header not valid\n", aufile[i]);
470 continue;
471 }
Sergei Poselenov638dd142007-02-27 12:40:16 +0300472 /* totsize accounts for image size and flash erase size */
Wolfgang Denk74357112007-02-27 14:26:04 +0100473 totsize += (imsize + (aufl_layout[i].end - aufl_layout[i].start));
Sergei Poselenov638dd142007-02-27 12:40:16 +0300474 }
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
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100490 debug ("read %s sz %ld hdr %d\n",
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100491 aufile[i], sz, image_get_header_size ());
Sergei Poselenov638dd142007-02-27 12:40:16 +0300492
493 if (sz != ausize[i]) {
494 printf ("%s: size %d read %d?\n", aufile[i], ausize[i], sz);
Wolfgang Denk74357112007-02-27 14:26:04 +0100495 continue;
496 }
Sergei Poselenov638dd142007-02-27 12:40:16 +0300497
Marian Balakowiczb97a2a02008-01-08 18:14:09 +0100498 if (sz <= 0 || sz <= image_get_header_size ()) {
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100499 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
Sergei Poselenov489c6962007-02-14 14:30:28 +0300521 } while (res < 0 && cnt < (AU_MAXFILES + 1));
522 if (cnt < (AU_MAXFILES + 1))
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100523#else
524 } while (res < 0);
525#endif
526 }
Wolfgang Denka4d26362007-08-12 15:11:38 +0200527
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100528 /* restore the old state */
529 disable_ctrlc(old_ctrlc);
Sergei Poselenov638dd142007-02-27 12:40:16 +0300530#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
Wolfgang Denka4d26362007-08-12 15:11:38 +0200540 xit:
541 usb_stop();
542 return res;
Andrei Safronovcdb97a62006-12-08 16:23:08 +0100543}
544#endif /* CONFIG_AUTO_UPDATE */