]> git.kernelconcepts.de Git - karo-tx-uboot.git/blob - common/cmd_nvedit.c
Merge branch 'spi' of git://git.denx.de/u-boot-blackfin
[karo-tx-uboot.git] / common / cmd_nvedit.c
1 /*
2  * (C) Copyright 2000-2010
3  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
4  *
5  * (C) Copyright 2001 Sysgo Real-Time Solutions, GmbH <www.elinos.com>
6  * Andreas Heppel <aheppel@sysgo.de>
7
8  * See file CREDITS for list of people who contributed to this
9  * project.
10  *
11  * This program is free software; you can redistribute it and/or
12  * modify it under the terms of the GNU General Public License as
13  * published by the Free Software Foundation; either version 2 of
14  * the License, or (at your option) any later version.
15  *
16  * This program is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19  * GNU General Public License for more details.
20  *
21  * You should have received a copy of the GNU General Public License
22  * along with this program; if not, write to the Free Software
23  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
24  * MA 02111-1307 USA
25  */
26
27 /*
28  * Support for persistent environment data
29  *
30  * The "environment" is stored on external storage as a list of '\0'
31  * terminated "name=value" strings. The end of the list is marked by
32  * a double '\0'. The environment is preceeded by a 32 bit CRC over
33  * the data part and, in case of redundant environment, a byte of
34  * flags.
35  *
36  * This linearized representation will also be used before
37  * relocation, i. e. as long as we don't have a full C runtime
38  * environment. After that, we use a hash table.
39  */
40
41 #include <common.h>
42 #include <command.h>
43 #include <environment.h>
44 #include <search.h>
45 #include <errno.h>
46 #include <malloc.h>
47 #include <watchdog.h>
48 #include <serial.h>
49 #include <linux/stddef.h>
50 #include <asm/byteorder.h>
51 #if defined(CONFIG_CMD_NET)
52 #include <net.h>
53 #endif
54
55 DECLARE_GLOBAL_DATA_PTR;
56
57 #if !defined(CONFIG_ENV_IS_IN_EEPROM)   && \
58     !defined(CONFIG_ENV_IS_IN_FLASH)    && \
59     !defined(CONFIG_ENV_IS_IN_DATAFLASH)        && \
60     !defined(CONFIG_ENV_IS_IN_MG_DISK)  && \
61     !defined(CONFIG_ENV_IS_IN_MMC)  && \
62     !defined(CONFIG_ENV_IS_IN_NAND)     && \
63     !defined(CONFIG_ENV_IS_IN_NVRAM)    && \
64     !defined(CONFIG_ENV_IS_IN_ONENAND)  && \
65     !defined(CONFIG_ENV_IS_IN_SPI_FLASH)        && \
66     !defined(CONFIG_ENV_IS_NOWHERE)
67 # error Define one of CONFIG_ENV_IS_IN_{EEPROM|FLASH|DATAFLASH|ONENAND|\
68 SPI_FLASH|MG_DISK|NVRAM|MMC} or CONFIG_ENV_IS_NOWHERE
69 #endif
70
71 #define XMK_STR(x)      #x
72 #define MK_STR(x)       XMK_STR(x)
73
74 /*
75  * Maximum expected input data size for import command
76  */
77 #define MAX_ENV_SIZE    (1 << 20)       /* 1 MiB */
78
79 ulong load_addr = CONFIG_SYS_LOAD_ADDR; /* Default Load Address */
80
81 /*
82  * Table with supported baudrates (defined in config_xyz.h)
83  */
84 static const unsigned long baudrate_table[] = CONFIG_SYS_BAUDRATE_TABLE;
85 #define N_BAUDRATES (sizeof(baudrate_table) / sizeof(baudrate_table[0]))
86
87 /*
88  * This variable is incremented on each do_env_set(), so it can
89  * be used via get_env_id() as an indication, if the environment
90  * has changed or not. So it is possible to reread an environment
91  * variable only if the environment was changed ... done so for
92  * example in NetInitLoop()
93  */
94 static int env_id = 1;
95
96 int get_env_id (void)
97 {
98         return env_id;
99 }
100
101 /*
102  * Command interface: print one or all environment variables
103  *
104  * Returns 0 in case of error, or length of printed string
105  */
106 static int env_print(char *name)
107 {
108         char *res = NULL;
109         size_t len;
110
111         if (name) {             /* print a single name */
112                 ENTRY e, *ep;
113
114                 e.key = name;
115                 e.data = NULL;
116                 hsearch_r(e, FIND, &ep, &env_htab);
117                 if (ep == NULL)
118                         return 0;
119                 len = printf ("%s=%s\n", ep->key, ep->data);
120                 return len;
121         }
122
123         /* print whole list */
124         len = hexport_r(&env_htab, '\n', &res, 0);
125
126         if (len > 0) {
127                 puts(res);
128                 free(res);
129                 return len;
130         }
131
132         /* should never happen */
133         return 0;
134 }
135
136 int do_env_print (cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
137 {
138         int i;
139         int rcode = 0;
140
141         if (argc == 1) {
142                 /* print all env vars */
143                 rcode = env_print(NULL);
144                 if (!rcode)
145                         return 1;
146                 printf("\nEnvironment size: %d/%ld bytes\n",
147                         rcode, (ulong)ENV_SIZE);
148                 return 0;
149         }
150
151         /* print selected env vars */
152         for (i = 1; i < argc; ++i) {
153                 int rc = env_print(argv[i]);
154                 if (!rc) {
155                         printf("## Error: \"%s\" not defined\n", argv[i]);
156                         ++rcode;
157                 }
158         }
159
160         return rcode;
161 }
162
163 /*
164  * Set a new environment variable,
165  * or replace or delete an existing one.
166  */
167
168 int _do_env_set (int flag, int argc, char * const argv[])
169 {
170         bd_t  *bd = gd->bd;
171         int   i, len;
172         int   console = -1;
173         char  *name, *value, *s;
174         ENTRY e, *ep;
175
176         name = argv[1];
177
178         if (strchr(name, '=')) {
179                 printf ("## Error: illegal character '=' in variable name \"%s\"\n", name);
180                 return 1;
181         }
182
183         env_id++;
184         /*
185          * search if variable with this name already exists
186          */
187         e.key = name;
188         e.data = NULL;
189         hsearch_r(e, FIND, &ep, &env_htab);
190
191         /* Check for console redirection */
192         if (strcmp(name,"stdin") == 0) {
193                 console = stdin;
194         } else if (strcmp(name,"stdout") == 0) {
195                 console = stdout;
196         } else if (strcmp(name,"stderr") == 0) {
197                 console = stderr;
198         }
199
200         if (console != -1) {
201                 if (argc < 3) {         /* Cannot delete it! */
202                         printf("Can't delete \"%s\"\n", name);
203                         return 1;
204                 }
205
206 #ifdef CONFIG_CONSOLE_MUX
207                 i = iomux_doenv(console, argv[2]);
208                 if (i)
209                         return i;
210 #else
211                 /* Try assigning specified device */
212                 if (console_assign (console, argv[2]) < 0)
213                         return 1;
214
215 #ifdef CONFIG_SERIAL_MULTI
216                 if (serial_assign (argv[2]) < 0)
217                         return 1;
218 #endif
219 #endif /* CONFIG_CONSOLE_MUX */
220         }
221
222         /*
223          * Some variables like "ethaddr" and "serial#" can be set only
224          * once and cannot be deleted; also, "ver" is readonly.
225          */
226         if (ep) {               /* variable exists */
227 #ifndef CONFIG_ENV_OVERWRITE
228                 if ((strcmp (name, "serial#") == 0) ||
229                     ((strcmp (name, "ethaddr") == 0)
230 #if defined(CONFIG_OVERWRITE_ETHADDR_ONCE) && defined(CONFIG_ETHADDR)
231                      && (strcmp (ep->data,MK_STR(CONFIG_ETHADDR)) != 0)
232 #endif  /* CONFIG_OVERWRITE_ETHADDR_ONCE && CONFIG_ETHADDR */
233                     ) ) {
234                         printf ("Can't overwrite \"%s\"\n", name);
235                         return 1;
236                 }
237 #endif
238                 /*
239                  * Switch to new baudrate if new baudrate is supported
240                  */
241                 if (strcmp(name,"baudrate") == 0) {
242                         int baudrate = simple_strtoul(argv[2], NULL, 10);
243                         int i;
244                         for (i=0; i<N_BAUDRATES; ++i) {
245                                 if (baudrate == baudrate_table[i])
246                                         break;
247                         }
248                         if (i == N_BAUDRATES) {
249                                 printf ("## Baudrate %d bps not supported\n",
250                                         baudrate);
251                                 return 1;
252                         }
253                         printf ("## Switch baudrate to %d bps and press ENTER ...\n",
254                                 baudrate);
255                         udelay(50000);
256                         gd->baudrate = baudrate;
257 #if defined(CONFIG_PPC) || defined(CONFIG_MCF52x2)
258                         gd->bd->bi_baudrate = baudrate;
259 #endif
260
261                         serial_setbrg ();
262                         udelay(50000);
263                         for (;;) {
264                                 if (getc() == '\r')
265                                       break;
266                         }
267                 }
268         }
269
270         /* Delete only ? */
271         if ((argc < 3) || argv[2] == NULL) {
272                 int rc = hdelete_r(name, &env_htab);
273                 return !rc;
274         }
275
276         /*
277          * Insert / replace new value
278          */
279         for (i=2,len=0; i<argc; ++i) {
280                 len += strlen(argv[i]) + 1;
281         }
282         if ((value = malloc(len)) == NULL) {
283                 printf("## Can't malloc %d bytes\n", len);
284                 return 1;
285         }
286         for (i=2,s=value; i<argc; ++i) {
287                 char *v = argv[i];
288
289                 while ((*s++ = *v++) != '\0')
290                         ;
291                 *(s-1) = ' ';
292         }
293         if (s != value)
294                 *--s = '\0';
295
296         e.key  = name;
297         e.data = value;
298         hsearch_r(e, ENTER, &ep, &env_htab);
299         free(value);
300         if (!ep) {
301                 printf("## Error inserting \"%s\" variable, errno=%d\n",
302                         name, errno);
303                 return 1;
304         }
305
306         /*
307          * Some variables should be updated when the corresponding
308          * entry in the environment is changed
309          */
310
311         if (strcmp(name,"ipaddr") == 0) {
312                 char *s = argv[2];      /* always use only one arg */
313                 char *e;
314                 unsigned long addr;
315                 bd->bi_ip_addr = 0;
316                 for (addr=0, i=0; i<4; ++i) {
317                         ulong val = s ? simple_strtoul(s, &e, 10) : 0;
318                         addr <<= 8;
319                         addr  |= (val & 0xFF);
320                         if (s) s = (*e) ? e+1 : e;
321                 }
322                 bd->bi_ip_addr = htonl(addr);
323                 return 0;
324         } else if (strcmp(argv[1],"loadaddr") == 0) {
325                 load_addr = simple_strtoul(argv[2], NULL, 16);
326                 return 0;
327         }
328 #if defined(CONFIG_CMD_NET)
329         else if (strcmp(argv[1],"bootfile") == 0) {
330                 copy_filename (BootFile, argv[2], sizeof(BootFile));
331                 return 0;
332         }
333 #endif
334         return 0;
335 }
336
337 int setenv (char *varname, char *varvalue)
338 {
339         char * const argv[4] = { "setenv", varname, varvalue, NULL };
340         if ((varvalue == NULL) || (varvalue[0] == '\0'))
341                 return _do_env_set(0, 2, argv);
342         else
343                 return _do_env_set(0, 3, argv);
344 }
345
346 int do_env_set (cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
347 {
348         if (argc < 2)
349                 return cmd_usage(cmdtp);
350
351         return _do_env_set(flag, argc, argv);
352 }
353
354 /*
355  * Prompt for environment variable
356  */
357 #if defined(CONFIG_CMD_ASKENV)
358 int do_env_ask ( cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
359 {
360         extern char console_buffer[CONFIG_SYS_CBSIZE];
361         char message[CONFIG_SYS_CBSIZE];
362         int size = CONFIG_SYS_CBSIZE - 1;
363         int i, len, pos;
364         char *local_args[4];
365
366         local_args[0] = argv[0];
367         local_args[1] = argv[1];
368         local_args[2] = NULL;
369         local_args[3] = NULL;
370
371         /* Check the syntax */
372         switch (argc) {
373         case 1:
374                 return cmd_usage(cmdtp);
375
376         case 2:         /* env_ask envname */
377                 sprintf(message, "Please enter '%s':", argv[1]);
378                 break;
379
380         case 3:         /* env_ask envname size */
381                 sprintf(message, "Please enter '%s':", argv[1]);
382                 size = simple_strtoul(argv[2], NULL, 10);
383                 break;
384
385         default:        /* env_ask envname message1 ... messagen size */
386                 for (i=2,pos=0; i < argc - 1; i++) {
387                         if (pos) {
388                                 message[pos++] = ' ';
389                         }
390                         strcpy(message+pos, argv[i]);
391                         pos += strlen(argv[i]);
392                 }
393                 message[pos] = '\0';
394                 size = simple_strtoul(argv[argc - 1], NULL, 10);
395                 break;
396         }
397
398         if (size >= CONFIG_SYS_CBSIZE)
399                 size = CONFIG_SYS_CBSIZE - 1;
400
401         if (size <= 0)
402                 return 1;
403
404         /* prompt for input */
405         len = readline(message);
406
407         if (size < len)
408                 console_buffer[size] = '\0';
409
410         len = 2;
411         if (console_buffer[0] != '\0') {
412                 local_args[2] = console_buffer;
413                 len = 3;
414         }
415
416         /* Continue calling setenv code */
417         return _do_env_set(flag, len, local_args);
418 }
419 #endif
420
421 /*
422  * Interactively edit an environment variable
423  */
424 #if defined(CONFIG_CMD_EDITENV)
425 int do_env_edit(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
426 {
427         char buffer[CONFIG_SYS_CBSIZE];
428         char *init_val;
429         int len;
430
431         if (argc < 2)
432                 return cmd_usage(cmdtp);
433
434         /* Set read buffer to initial value or empty sting */
435         init_val = getenv(argv[1]);
436         if (init_val)
437                 len = sprintf(buffer, "%s", init_val);
438         else
439                 buffer[0] = '\0';
440
441         readline_into_buffer("edit: ", buffer);
442
443         return setenv(argv[1], buffer);
444 }
445 #endif /* CONFIG_CMD_EDITENV */
446
447 /*
448  * Look up variable from environment,
449  * return address of storage for that variable,
450  * or NULL if not found
451  */
452 char *getenv (char *name)
453 {
454         if (gd->flags & GD_FLG_ENV_READY) {     /* after import into hashtable */
455                 ENTRY e, *ep;
456
457                 WATCHDOG_RESET();
458
459                 e.key  = name;
460                 e.data = NULL;
461                 hsearch_r(e, FIND, &ep, &env_htab);
462
463                 return (ep ? ep->data : NULL);
464         }
465
466         /* restricted capabilities before import */
467
468         if (getenv_f(name, (char *)(gd->env_buf), sizeof(gd->env_buf)) > 0)
469                 return (char *)(gd->env_buf);
470
471         return NULL;
472 }
473
474 /*
475  * Look up variable from environment for restricted C runtime env.
476  */
477 int getenv_f (char *name, char *buf, unsigned len)
478 {
479         int i, nxt;
480
481         for (i=0; env_get_char(i) != '\0'; i=nxt+1) {
482                 int val, n;
483
484                 for (nxt=i; env_get_char(nxt) != '\0'; ++nxt) {
485                         if (nxt >= CONFIG_ENV_SIZE) {
486                                 return (-1);
487                         }
488                 }
489                 if ((val=envmatch((uchar *)name, i)) < 0)
490                         continue;
491
492                 /* found; copy out */
493                 for (n=0; n<len; ++n, ++buf) {
494                         if ((*buf = env_get_char(val++)) == '\0')
495                                 return n;
496                 }
497
498                 if (n)
499                         *--buf = '\0';
500
501                 printf("env_buf too small [%d]\n", len);
502
503                 return n;
504         }
505         return (-1);
506 }
507
508 #if defined(CONFIG_CMD_SAVEENV) && !defined(CONFIG_ENV_IS_NOWHERE)
509
510 int do_env_save (cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
511 {
512         extern char * env_name_spec;
513
514         printf ("Saving Environment to %s...\n", env_name_spec);
515
516         return (saveenv() ? 1 : 0);
517 }
518
519 U_BOOT_CMD(
520         saveenv, 1, 0,  do_env_save,
521         "save environment variables to persistent storage",
522         ""
523 );
524
525 #endif
526
527
528 /*
529  * Match a name / name=value pair
530  *
531  * s1 is either a simple 'name', or a 'name=value' pair.
532  * i2 is the environment index for a 'name2=value2' pair.
533  * If the names match, return the index for the value2, else NULL.
534  */
535
536 int envmatch (uchar *s1, int i2)
537 {
538
539         while (*s1 == env_get_char(i2++))
540                 if (*s1++ == '=')
541                         return(i2);
542         if (*s1 == '\0' && env_get_char(i2-1) == '=')
543                 return(i2);
544         return(-1);
545 }
546
547 static int do_env_default(cmd_tbl_t * cmdtp, int flag, int argc, char * const argv[])
548 {
549         if ((argc != 2) || (strcmp(argv[1], "-f") != 0)) {
550                 return cmd_usage(cmdtp);
551         }
552         set_default_env("## Resetting to default environment\n");
553         return 0;
554 }
555
556 static int do_env_delete(cmd_tbl_t * cmdtp, int flag, int argc, char * const argv[])
557 {
558         printf("Not implemented yet\n");
559         return 0;
560 }
561
562 #ifdef CONFIG_CMD_EXPORTENV
563 /*
564  * env export [-t | -b | -c] addr [size]
565  *      -t:     export as text format; if size is given, data will be
566  *              padded with '\0' bytes; if not, one terminating '\0'
567  *              will be added (which is included in the "filesize"
568  *              setting so you can for exmple copy this to flash and
569  *              keep the termination).
570  *      -b:     export as binary format (name=value pairs separated by
571  *              '\0', list end marked by double "\0\0")
572  *      -c:     export as checksum protected environment format as
573  *              used for example by "saveenv" command
574  *      addr:   memory address where environment gets stored
575  *      size:   size of output buffer
576  *
577  * With "-c" and size is NOT given, then the export command will
578  * format the data as currently used for the persistent storage,
579  * i. e. it will use CONFIG_ENV_SECT_SIZE as output block size and
580  * prepend a valid CRC32 checksum and, in case of resundant
581  * environment, a "current" redundancy flag. If size is given, this
582  * value will be used instead of CONFIG_ENV_SECT_SIZE; again, CRC32
583  * checksum and redundancy flag will be inserted.
584  *
585  * With "-b" and "-t", always only the real data (including a
586  * terminating '\0' byte) will be written; here the optional size
587  * argument will be used to make sure not to overflow the user
588  * provided buffer; the command will abort if the size is not
589  * sufficient. Any remainign space will be '\0' padded.
590  *
591  * On successful return, the variable "filesize" will be set.
592  * Note that filesize includes the trailing/terminating '\0' byte(s).
593  *
594  * Usage szenario:  create a text snapshot/backup of the current settings:
595  *
596  *      => env export -t 100000
597  *      => era ${backup_addr} +${filesize}
598  *      => cp.b 100000 ${backup_addr} ${filesize}
599  *
600  * Re-import this snapshot, deleting all other settings:
601  *
602  *      => env import -d -t ${backup_addr}
603  */
604 static int do_env_export(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
605 {
606         char    buf[32];
607         char    *addr, *cmd, *res;
608         size_t  size;
609         ssize_t len;
610         env_t   *envp;
611         char    sep = '\n';
612         int     chk = 0;
613         int     fmt = 0;
614
615         cmd = *argv;
616
617         while (--argc > 0 && **++argv == '-') {
618                 char *arg = *argv;
619                 while (*++arg) {
620                         switch (*arg) {
621                         case 'b':               /* raw binary format */
622                                 if (fmt++)
623                                         goto sep_err;
624                                 sep = '\0';
625                                 break;
626                         case 'c':               /* external checksum format */
627                                 if (fmt++)
628                                         goto sep_err;
629                                 sep = '\0';
630                                 chk = 1;
631                                 break;
632                         case 't':               /* text format */
633                                 if (fmt++)
634                                         goto sep_err;
635                                 sep = '\n';
636                                 break;
637                         default:
638                                 return cmd_usage(cmdtp);
639                         }
640                 }
641         }
642
643         if (argc < 1) {
644                 return cmd_usage(cmdtp);
645         }
646
647         addr = (char *)simple_strtoul(argv[0], NULL, 16);
648
649         if (argc == 2) {
650                 size = simple_strtoul(argv[1], NULL, 16);
651                 memset(addr, '\0', size);
652         } else {
653                 size = 0;
654         }
655
656         if (sep) {              /* export as text file */
657                 len = hexport_r(&env_htab, sep, &addr, size);
658                 if (len < 0) {
659                         error("Cannot export environment: errno = %d\n",
660                                 errno);
661                         return 1;
662                 }
663                 sprintf(buf, "%zX", (size_t)len);
664                 setenv("filesize", buf);
665
666                 return 0;
667         }
668
669         envp = (env_t *)addr;
670
671         if (chk)                /* export as checksum protected block */
672                 res = (char *)envp->data;
673         else                    /* export as raw binary data */
674                 res = addr;
675
676         len = hexport_r(&env_htab, '\0', &res, ENV_SIZE);
677         if (len < 0) {
678                 error("Cannot export environment: errno = %d\n",
679                         errno);
680                 return 1;
681         }
682
683         if (chk) {
684                 envp->crc   = crc32(0, envp->data, ENV_SIZE);
685 #ifdef CONFIG_ENV_ADDR_REDUND
686                 envp->flags = ACTIVE_FLAG;
687 #endif
688         }
689         sprintf(buf, "%zX", (size_t)(len + offsetof(env_t,data)));
690         setenv("filesize", buf);
691
692         return 0;
693
694 sep_err:
695         printf("## %s: only one of \"-b\", \"-c\" or \"-t\" allowed\n",
696                 cmd);
697         return 1;
698 }
699 #endif
700
701 #ifdef CONFIG_CMD_IMPORTENV
702 /*
703  * env import [-d] [-t | -b | -c] addr [size]
704  *      -d:     delete existing environment before importing;
705  *              otherwise overwrite / append to existion definitions
706  *      -t:     assume text format; either "size" must be given or the
707  *              text data must be '\0' terminated
708  *      -b:     assume binary format ('\0' separated, "\0\0" terminated)
709  *      -c:     assume checksum protected environment format
710  *      addr:   memory address to read from
711  *      size:   length of input data; if missing, proper '\0'
712  *              termination is mandatory
713  */
714 static int do_env_import(cmd_tbl_t * cmdtp, int flag, int argc, char * const argv[])
715 {
716         char    *cmd, *addr;
717         char    sep = '\n';
718         int     chk = 0;
719         int     fmt = 0;
720         int     del = 0;
721         size_t  size;
722
723         cmd = *argv;
724
725         while (--argc > 0 && **++argv == '-') {
726                 char *arg = *argv;
727                 while (*++arg) {
728                         switch (*arg) {
729                         case 'b':               /* raw binary format */
730                                 if (fmt++)
731                                         goto sep_err;
732                                 sep = '\0';
733                                 break;
734                         case 'c':               /* external checksum format */
735                                 if (fmt++)
736                                         goto sep_err;
737                                 sep = '\0';
738                                 chk = 1;
739                                 break;
740                         case 't':               /* text format */
741                                 if (fmt++)
742                                         goto sep_err;
743                                 sep = '\n';
744                                 break;
745                         case 'd':
746                                 del = 1;
747                                 break;
748                         default:
749                                 return cmd_usage(cmdtp);
750                         }
751                 }
752         }
753
754         if (argc < 1) {
755                 return cmd_usage(cmdtp);
756         }
757
758         if (!fmt)
759                 printf("## Warning: defaulting to text format\n");
760
761         addr = (char *)simple_strtoul(argv[0], NULL, 16);
762
763         if (argc == 2) {
764                 size = simple_strtoul(argv[1], NULL, 16);
765         } else {
766                 char *s = addr;
767
768                 size = 0;
769
770                 while (size < MAX_ENV_SIZE) {
771                         if ((*s == sep) && (*(s+1) == '\0'))
772                                 break;
773                         ++s;
774                         ++size;
775                 }
776                 if (size == MAX_ENV_SIZE) {
777                         printf("## Warning: Input data exceeds %d bytes"
778                                 " - truncated\n", MAX_ENV_SIZE);
779                 }
780                 ++size;
781                 printf("## Info: input data size = %zd = 0x%zX\n", size, size);
782         }
783
784         if (chk) {
785                 uint32_t crc;
786                 env_t *ep = (env_t *)addr;
787
788                 size -= offsetof(env_t, data);
789                 memcpy(&crc, &ep->crc, sizeof(crc));
790
791                 if (crc32(0, ep->data, size) != crc) {
792                         puts("## Error: bad CRC, import failed\n");
793                         return 1;
794                 }
795                 addr = (char *)ep->data;
796         }
797
798         if (himport_r(&env_htab, addr, size, sep, del ? 0 : H_NOCLEAR) == 0) {
799                 error("Environment import failed: errno = %d\n", errno);
800                 return 1;
801         }
802         gd->flags |= GD_FLG_ENV_READY;
803
804         return 0;
805
806 sep_err:
807         printf("## %s: only one of \"-b\", \"-c\" or \"-t\" allowed\n",
808                 cmd);
809         return 1;
810 }
811 #endif
812
813 #if defined(CONFIG_CMD_RUN)
814 extern int do_run (cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]);
815 #endif
816
817 /*
818  * New command line interface: "env" command with subcommands
819  */
820 static cmd_tbl_t cmd_env_sub[] = {
821 #if defined(CONFIG_CMD_ASKENV)
822         U_BOOT_CMD_MKENT(ask, CONFIG_SYS_MAXARGS, 1, do_env_ask, "", ""),
823 #endif
824         U_BOOT_CMD_MKENT(default, 1, 0, do_env_default, "", ""),
825         U_BOOT_CMD_MKENT(delete, 2, 0, do_env_delete, "", ""),
826 #if defined(CONFIG_CMD_EDITENV)
827         U_BOOT_CMD_MKENT(edit, 2, 0, do_env_edit, "", ""),
828 #endif
829 #if defined(CONFIG_CMD_EXPORTENV)
830         U_BOOT_CMD_MKENT(export, 4, 0, do_env_export, "", ""),
831 #endif
832 #if defined(CONFIG_CMD_IMPORTENV)
833         U_BOOT_CMD_MKENT(import, 5, 0, do_env_import, "", ""),
834 #endif
835         U_BOOT_CMD_MKENT(print, CONFIG_SYS_MAXARGS, 1, do_env_print, "", ""),
836 #if defined(CONFIG_CMD_RUN)
837         U_BOOT_CMD_MKENT(run, CONFIG_SYS_MAXARGS, 1, do_run, "", ""),
838 #endif
839 #if defined(CONFIG_CMD_SAVEENV) && !defined(CONFIG_ENV_IS_NOWHERE)
840         U_BOOT_CMD_MKENT(save, 1, 0, do_env_save, "", ""),
841 #endif
842         U_BOOT_CMD_MKENT(set, CONFIG_SYS_MAXARGS, 0, do_env_set, "", ""),
843 };
844
845 #if defined(CONFIG_NEEDS_MANUAL_RELOC)
846 void env_reloc(void)
847 {
848         fixup_cmdtable(cmd_env_sub, ARRAY_SIZE(cmd_env_sub));
849 }
850 #endif
851
852 static int do_env (cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
853 {
854         cmd_tbl_t *cp;
855
856         if (argc < 2)
857                 return cmd_usage(cmdtp);
858
859         /* drop initial "env" arg */
860         argc--;
861         argv++;
862
863         cp = find_cmd_tbl(argv[0], cmd_env_sub, ARRAY_SIZE(cmd_env_sub));
864
865         if (cp)
866                 return cp->cmd(cmdtp, flag, argc, argv);
867
868         return cmd_usage(cmdtp);
869 }
870
871 U_BOOT_CMD(
872         env, CONFIG_SYS_MAXARGS, 1, do_env,
873         "environment handling commands",
874 #if defined(CONFIG_CMD_ASKENV)
875         "ask name [message] [size] - ask for environment variable\nenv "
876 #endif
877         "default -f - reset default environment\n"
878 #if defined(CONFIG_CMD_EDITENV)
879         "env edit name - edit environment variable\n"
880 #endif
881         "env export [-t | -b | -c] addr [size] - export environmnt\n"
882         "env import [-d] [-t | -b | -c] addr [size] - import environmnt\n"
883         "env print [name ...] - print environment\n"
884 #if defined(CONFIG_CMD_RUN)
885         "env run var [...] - run commands in an environment variable\n"
886 #endif
887         "env save - save environment\n"
888         "env set [-f] name [arg ...]\n"
889 );
890
891 /*
892  * Old command line interface, kept for compatibility
893  */
894
895 #if defined(CONFIG_CMD_EDITENV)
896 U_BOOT_CMD_COMPLETE(
897         editenv, 2, 0,  do_env_edit,
898         "edit environment variable",
899         "name\n"
900         "    - edit environment variable 'name'",
901         var_complete
902 );
903 #endif
904
905 U_BOOT_CMD_COMPLETE(
906         printenv, CONFIG_SYS_MAXARGS, 1,        do_env_print,
907         "print environment variables",
908         "\n    - print values of all environment variables\n"
909         "printenv name ...\n"
910         "    - print value of environment variable 'name'",
911         var_complete
912 );
913
914 U_BOOT_CMD_COMPLETE(
915         setenv, CONFIG_SYS_MAXARGS, 0,  do_env_set,
916         "set environment variables",
917         "name value ...\n"
918         "    - set environment variable 'name' to 'value ...'\n"
919         "setenv name\n"
920         "    - delete environment variable 'name'",
921         var_complete
922 );
923
924 #if defined(CONFIG_CMD_ASKENV)
925
926 U_BOOT_CMD(
927         askenv, CONFIG_SYS_MAXARGS,     1,      do_env_ask,
928         "get environment variables from stdin",
929         "name [message] [size]\n"
930         "    - get environment variable 'name' from stdin (max 'size' chars)\n"
931         "askenv name\n"
932         "    - get environment variable 'name' from stdin\n"
933         "askenv name size\n"
934         "    - get environment variable 'name' from stdin (max 'size' chars)\n"
935         "askenv name [message] size\n"
936         "    - display 'message' string and get environment variable 'name'"
937         "from stdin (max 'size' chars)"
938 );
939 #endif
940
941 #if defined(CONFIG_CMD_RUN)
942 U_BOOT_CMD_COMPLETE(
943         run,    CONFIG_SYS_MAXARGS,     1,      do_run,
944         "run commands in an environment variable",
945         "var [...]\n"
946         "    - run the commands in the environment variable(s) 'var'",
947         var_complete
948 );
949 #endif