cmd_nvedit.c 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637
  1. /*
  2. * (C) Copyright 2000-2002
  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. * See file CREDITS for list of people who contributed to this
  8. * project.
  9. *
  10. * This program is free software; you can redistribute it and/or
  11. * modify it under the terms of the GNU General Public License as
  12. * published by the Free Software Foundation; either version 2 of
  13. * the License, or (at your option) any later version.
  14. *
  15. * This program is distributed in the hope that it will be useful,
  16. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  18. * GNU General Public License for more details.
  19. *
  20. * You should have received a copy of the GNU General Public License
  21. * along with this program; if not, write to the Free Software
  22. * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
  23. * MA 02111-1307 USA
  24. */
  25. /**************************************************************************
  26. *
  27. * Support for persistent environment data
  28. *
  29. * The "environment" is stored as a list of '\0' terminated
  30. * "name=value" strings. The end of the list is marked by a double
  31. * '\0'. New entries are always added at the end. Deleting an entry
  32. * shifts the remaining entries to the front. Replacing an entry is a
  33. * combination of deleting the old value and adding the new one.
  34. *
  35. * The environment is preceeded by a 32 bit CRC over the data part.
  36. *
  37. **************************************************************************
  38. */
  39. #include <common.h>
  40. #include <command.h>
  41. #include <environment.h>
  42. #include <watchdog.h>
  43. #include <serial.h>
  44. #include <linux/stddef.h>
  45. #include <asm/byteorder.h>
  46. #if defined(CONFIG_CMD_NET)
  47. #include <net.h>
  48. #endif
  49. DECLARE_GLOBAL_DATA_PTR;
  50. #if !defined(CONFIG_ENV_IS_IN_NVRAM) && \
  51. !defined(CONFIG_ENV_IS_IN_EEPROM) && \
  52. !defined(CONFIG_ENV_IS_IN_FLASH) && \
  53. !defined(CONFIG_ENV_IS_IN_DATAFLASH) && \
  54. !defined(CONFIG_ENV_IS_IN_NAND) && \
  55. !defined(CONFIG_ENV_IS_IN_ONENAND) && \
  56. !defined(CONFIG_ENV_IS_IN_SPI_FLASH) && \
  57. !defined(CONFIG_ENV_IS_NOWHERE)
  58. # error Define one of CFG_ENV_IS_IN_{NVRAM|EEPROM|FLASH|DATAFLASH|ONENAND|SPI_FLASH|NOWHERE}
  59. #endif
  60. #define XMK_STR(x) #x
  61. #define MK_STR(x) XMK_STR(x)
  62. /************************************************************************
  63. ************************************************************************/
  64. /*
  65. * Table with supported baudrates (defined in config_xyz.h)
  66. */
  67. static const unsigned long baudrate_table[] = CFG_BAUDRATE_TABLE;
  68. #define N_BAUDRATES (sizeof(baudrate_table) / sizeof(baudrate_table[0]))
  69. /************************************************************************
  70. * Command interface: print one or all environment variables
  71. */
  72. int do_printenv (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
  73. {
  74. int i, j, k, nxt;
  75. int rcode = 0;
  76. if (argc == 1) { /* Print all env variables */
  77. for (i=0; env_get_char(i) != '\0'; i=nxt+1) {
  78. for (nxt=i; env_get_char(nxt) != '\0'; ++nxt)
  79. ;
  80. for (k=i; k<nxt; ++k)
  81. putc(env_get_char(k));
  82. putc ('\n');
  83. if (ctrlc()) {
  84. puts ("\n ** Abort\n");
  85. return 1;
  86. }
  87. }
  88. printf("\nEnvironment size: %d/%ld bytes\n",
  89. i, (ulong)ENV_SIZE);
  90. return 0;
  91. }
  92. for (i=1; i<argc; ++i) { /* print single env variables */
  93. char *name = argv[i];
  94. k = -1;
  95. for (j=0; env_get_char(j) != '\0'; j=nxt+1) {
  96. for (nxt=j; env_get_char(nxt) != '\0'; ++nxt)
  97. ;
  98. k = envmatch((uchar *)name, j);
  99. if (k < 0) {
  100. continue;
  101. }
  102. puts (name);
  103. putc ('=');
  104. while (k < nxt)
  105. putc(env_get_char(k++));
  106. putc ('\n');
  107. break;
  108. }
  109. if (k < 0) {
  110. printf ("## Error: \"%s\" not defined\n", name);
  111. rcode ++;
  112. }
  113. }
  114. return rcode;
  115. }
  116. /************************************************************************
  117. * Set a new environment variable,
  118. * or replace or delete an existing one.
  119. *
  120. * This function will ONLY work with a in-RAM copy of the environment
  121. */
  122. int _do_setenv (int flag, int argc, char *argv[])
  123. {
  124. int i, len, oldval;
  125. int console = -1;
  126. uchar *env, *nxt = NULL;
  127. char *name;
  128. bd_t *bd = gd->bd;
  129. uchar *env_data = env_get_addr(0);
  130. if (!env_data) /* need copy in RAM */
  131. return 1;
  132. name = argv[1];
  133. if (strchr(name, '=')) {
  134. printf ("## Error: illegal character '=' in variable name \"%s\"\n", name);
  135. return 1;
  136. }
  137. /*
  138. * search if variable with this name already exists
  139. */
  140. oldval = -1;
  141. for (env=env_data; *env; env=nxt+1) {
  142. for (nxt=env; *nxt; ++nxt)
  143. ;
  144. if ((oldval = envmatch((uchar *)name, env-env_data)) >= 0)
  145. break;
  146. }
  147. /*
  148. * Delete any existing definition
  149. */
  150. if (oldval >= 0) {
  151. #ifndef CONFIG_ENV_OVERWRITE
  152. /*
  153. * Ethernet Address and serial# can be set only once,
  154. * ver is readonly.
  155. */
  156. if (
  157. #ifdef CONFIG_HAS_UID
  158. /* Allow serial# forced overwrite with 0xdeaf4add flag */
  159. ((strcmp (name, "serial#") == 0) && (flag != 0xdeaf4add)) ||
  160. #else
  161. (strcmp (name, "serial#") == 0) ||
  162. #endif
  163. ((strcmp (name, "ethaddr") == 0)
  164. #if defined(CONFIG_OVERWRITE_ETHADDR_ONCE) && defined(CONFIG_ETHADDR)
  165. && (strcmp ((char *)env_get_addr(oldval),MK_STR(CONFIG_ETHADDR)) != 0)
  166. #endif /* CONFIG_OVERWRITE_ETHADDR_ONCE && CONFIG_ETHADDR */
  167. ) ) {
  168. printf ("Can't overwrite \"%s\"\n", name);
  169. return 1;
  170. }
  171. #endif
  172. /* Check for console redirection */
  173. if (strcmp(name,"stdin") == 0) {
  174. console = stdin;
  175. } else if (strcmp(name,"stdout") == 0) {
  176. console = stdout;
  177. } else if (strcmp(name,"stderr") == 0) {
  178. console = stderr;
  179. }
  180. if (console != -1) {
  181. if (argc < 3) { /* Cannot delete it! */
  182. printf("Can't delete \"%s\"\n", name);
  183. return 1;
  184. }
  185. /* Try assigning specified device */
  186. if (console_assign (console, argv[2]) < 0)
  187. return 1;
  188. #ifdef CONFIG_SERIAL_MULTI
  189. if (serial_assign (argv[2]) < 0)
  190. return 1;
  191. #endif
  192. }
  193. /*
  194. * Switch to new baudrate if new baudrate is supported
  195. */
  196. if (strcmp(argv[1],"baudrate") == 0) {
  197. int baudrate = simple_strtoul(argv[2], NULL, 10);
  198. int i;
  199. for (i=0; i<N_BAUDRATES; ++i) {
  200. if (baudrate == baudrate_table[i])
  201. break;
  202. }
  203. if (i == N_BAUDRATES) {
  204. printf ("## Baudrate %d bps not supported\n",
  205. baudrate);
  206. return 1;
  207. }
  208. printf ("## Switch baudrate to %d bps and press ENTER ...\n",
  209. baudrate);
  210. udelay(50000);
  211. gd->baudrate = baudrate;
  212. #if defined(CONFIG_PPC) || defined(CONFIG_MCF52x2)
  213. gd->bd->bi_baudrate = baudrate;
  214. #endif
  215. serial_setbrg ();
  216. udelay(50000);
  217. for (;;) {
  218. if (getc() == '\r')
  219. break;
  220. }
  221. }
  222. if (*++nxt == '\0') {
  223. if (env > env_data) {
  224. env--;
  225. } else {
  226. *env = '\0';
  227. }
  228. } else {
  229. for (;;) {
  230. *env = *nxt++;
  231. if ((*env == '\0') && (*nxt == '\0'))
  232. break;
  233. ++env;
  234. }
  235. }
  236. *++env = '\0';
  237. }
  238. #ifdef CONFIG_NET_MULTI
  239. if (strncmp(name, "eth", 3) == 0) {
  240. char *end;
  241. int num = simple_strtoul(name+3, &end, 10);
  242. if (strcmp(end, "addr") == 0) {
  243. eth_set_enetaddr(num, argv[2]);
  244. }
  245. }
  246. #endif
  247. /* Delete only ? */
  248. if ((argc < 3) || argv[2] == NULL) {
  249. env_crc_update ();
  250. return 0;
  251. }
  252. /*
  253. * Append new definition at the end
  254. */
  255. for (env=env_data; *env || *(env+1); ++env)
  256. ;
  257. if (env > env_data)
  258. ++env;
  259. /*
  260. * Overflow when:
  261. * "name" + "=" + "val" +"\0\0" > ENV_SIZE - (env-env_data)
  262. */
  263. len = strlen(name) + 2;
  264. /* add '=' for first arg, ' ' for all others */
  265. for (i=2; i<argc; ++i) {
  266. len += strlen(argv[i]) + 1;
  267. }
  268. if (len > (&env_data[ENV_SIZE]-env)) {
  269. printf ("## Error: environment overflow, \"%s\" deleted\n", name);
  270. return 1;
  271. }
  272. while ((*env = *name++) != '\0')
  273. env++;
  274. for (i=2; i<argc; ++i) {
  275. char *val = argv[i];
  276. *env = (i==2) ? '=' : ' ';
  277. while ((*++env = *val++) != '\0')
  278. ;
  279. }
  280. /* end is marked with double '\0' */
  281. *++env = '\0';
  282. /* Update CRC */
  283. env_crc_update ();
  284. /*
  285. * Some variables should be updated when the corresponding
  286. * entry in the enviornment is changed
  287. */
  288. if (strcmp(argv[1],"ethaddr") == 0) {
  289. char *s = argv[2]; /* always use only one arg */
  290. char *e;
  291. for (i=0; i<6; ++i) {
  292. bd->bi_enetaddr[i] = s ? simple_strtoul(s, &e, 16) : 0;
  293. if (s) s = (*e) ? e+1 : e;
  294. }
  295. #ifdef CONFIG_NET_MULTI
  296. eth_set_enetaddr(0, argv[2]);
  297. #endif
  298. return 0;
  299. }
  300. if (strcmp(argv[1],"ipaddr") == 0) {
  301. char *s = argv[2]; /* always use only one arg */
  302. char *e;
  303. unsigned long addr;
  304. bd->bi_ip_addr = 0;
  305. for (addr=0, i=0; i<4; ++i) {
  306. ulong val = s ? simple_strtoul(s, &e, 10) : 0;
  307. addr <<= 8;
  308. addr |= (val & 0xFF);
  309. if (s) s = (*e) ? e+1 : e;
  310. }
  311. bd->bi_ip_addr = htonl(addr);
  312. return 0;
  313. }
  314. if (strcmp(argv[1],"loadaddr") == 0) {
  315. load_addr = simple_strtoul(argv[2], NULL, 16);
  316. return 0;
  317. }
  318. #if defined(CONFIG_CMD_NET)
  319. if (strcmp(argv[1],"bootfile") == 0) {
  320. copy_filename (BootFile, argv[2], sizeof(BootFile));
  321. return 0;
  322. }
  323. #endif
  324. #ifdef CONFIG_AMIGAONEG3SE
  325. if (strcmp(argv[1], "vga_fg_color") == 0 ||
  326. strcmp(argv[1], "vga_bg_color") == 0 ) {
  327. extern void video_set_color(unsigned char attr);
  328. extern unsigned char video_get_attr(void);
  329. video_set_color(video_get_attr());
  330. return 0;
  331. }
  332. #endif /* CONFIG_AMIGAONEG3SE */
  333. return 0;
  334. }
  335. int setenv (char *varname, char *varvalue)
  336. {
  337. char *argv[4] = { "setenv", varname, varvalue, NULL };
  338. if (varvalue == NULL)
  339. return _do_setenv (0, 2, argv);
  340. else
  341. return _do_setenv (0, 3, argv);
  342. }
  343. #ifdef CONFIG_HAS_UID
  344. void forceenv (char *varname, char *varvalue)
  345. {
  346. char *argv[4] = { "forceenv", varname, varvalue, NULL };
  347. _do_setenv (0xdeaf4add, 3, argv);
  348. }
  349. #endif
  350. int do_setenv (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
  351. {
  352. if (argc < 2) {
  353. printf ("Usage:\n%s\n", cmdtp->usage);
  354. return 1;
  355. }
  356. return _do_setenv (flag, argc, argv);
  357. }
  358. /************************************************************************
  359. * Prompt for environment variable
  360. */
  361. #if defined(CONFIG_CMD_ASKENV)
  362. int do_askenv ( cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
  363. {
  364. extern char console_buffer[CFG_CBSIZE];
  365. char message[CFG_CBSIZE];
  366. int size = CFG_CBSIZE - 1;
  367. int len;
  368. char *local_args[4];
  369. local_args[0] = argv[0];
  370. local_args[1] = argv[1];
  371. local_args[2] = NULL;
  372. local_args[3] = NULL;
  373. if (argc < 2) {
  374. printf ("Usage:\n%s\n", cmdtp->usage);
  375. return 1;
  376. }
  377. /* Check the syntax */
  378. switch (argc) {
  379. case 1:
  380. printf ("Usage:\n%s\n", cmdtp->usage);
  381. return 1;
  382. case 2: /* askenv envname */
  383. sprintf (message, "Please enter '%s':", argv[1]);
  384. break;
  385. case 3: /* askenv envname size */
  386. sprintf (message, "Please enter '%s':", argv[1]);
  387. size = simple_strtoul (argv[2], NULL, 10);
  388. break;
  389. default: /* askenv envname message1 ... messagen size */
  390. {
  391. int i;
  392. int pos = 0;
  393. for (i = 2; i < argc - 1; i++) {
  394. if (pos) {
  395. message[pos++] = ' ';
  396. }
  397. strcpy (message+pos, argv[i]);
  398. pos += strlen(argv[i]);
  399. }
  400. message[pos] = '\0';
  401. size = simple_strtoul (argv[argc - 1], NULL, 10);
  402. }
  403. break;
  404. }
  405. if (size >= CFG_CBSIZE)
  406. size = CFG_CBSIZE - 1;
  407. if (size <= 0)
  408. return 1;
  409. /* prompt for input */
  410. len = readline (message);
  411. if (size < len)
  412. console_buffer[size] = '\0';
  413. len = 2;
  414. if (console_buffer[0] != '\0') {
  415. local_args[2] = console_buffer;
  416. len = 3;
  417. }
  418. /* Continue calling setenv code */
  419. return _do_setenv (flag, len, local_args);
  420. }
  421. #endif
  422. /************************************************************************
  423. * Look up variable from environment,
  424. * return address of storage for that variable,
  425. * or NULL if not found
  426. */
  427. char *getenv (char *name)
  428. {
  429. int i, nxt;
  430. WATCHDOG_RESET();
  431. for (i=0; env_get_char(i) != '\0'; i=nxt+1) {
  432. int val;
  433. for (nxt=i; env_get_char(nxt) != '\0'; ++nxt) {
  434. if (nxt >= CFG_ENV_SIZE) {
  435. return (NULL);
  436. }
  437. }
  438. if ((val=envmatch((uchar *)name, i)) < 0)
  439. continue;
  440. return ((char *)env_get_addr(val));
  441. }
  442. return (NULL);
  443. }
  444. int getenv_r (char *name, char *buf, unsigned len)
  445. {
  446. int i, nxt;
  447. for (i=0; env_get_char(i) != '\0'; i=nxt+1) {
  448. int val, n;
  449. for (nxt=i; env_get_char(nxt) != '\0'; ++nxt) {
  450. if (nxt >= CFG_ENV_SIZE) {
  451. return (-1);
  452. }
  453. }
  454. if ((val=envmatch((uchar *)name, i)) < 0)
  455. continue;
  456. /* found; copy out */
  457. n = 0;
  458. while ((len > n++) && (*buf++ = env_get_char(val++)) != '\0')
  459. ;
  460. if (len == n)
  461. *buf = '\0';
  462. return (n);
  463. }
  464. return (-1);
  465. }
  466. #if ((defined(CONFIG_ENV_IS_IN_NVRAM) || defined(CONFIG_ENV_IS_IN_EEPROM) \
  467. || (defined(CONFIG_CMD_ENV) && defined(CONFIG_CMD_FLASH)) \
  468. || (defined(CONFIG_CMD_ENV) && defined(CONFIG_CMD_NAND)) \
  469. || (defined(CONFIG_CMD_ENV) && defined(CONFIG_CMD_ONENAND))) \
  470. && !defined(CONFIG_ENV_IS_NOWHERE))
  471. int do_saveenv (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
  472. {
  473. extern char * env_name_spec;
  474. printf ("Saving Environment to %s...\n", env_name_spec);
  475. return (saveenv() ? 1 : 0);
  476. }
  477. #endif
  478. /************************************************************************
  479. * Match a name / name=value pair
  480. *
  481. * s1 is either a simple 'name', or a 'name=value' pair.
  482. * i2 is the environment index for a 'name2=value2' pair.
  483. * If the names match, return the index for the value2, else NULL.
  484. */
  485. int envmatch (uchar *s1, int i2)
  486. {
  487. while (*s1 == env_get_char(i2++))
  488. if (*s1++ == '=')
  489. return(i2);
  490. if (*s1 == '\0' && env_get_char(i2-1) == '=')
  491. return(i2);
  492. return(-1);
  493. }
  494. /**************************************************/
  495. U_BOOT_CMD(
  496. printenv, CFG_MAXARGS, 1, do_printenv,
  497. "printenv- print environment variables\n",
  498. "\n - print values of all environment variables\n"
  499. "printenv name ...\n"
  500. " - print value of environment variable 'name'\n"
  501. );
  502. U_BOOT_CMD(
  503. setenv, CFG_MAXARGS, 0, do_setenv,
  504. "setenv - set environment variables\n",
  505. "name value ...\n"
  506. " - set environment variable 'name' to 'value ...'\n"
  507. "setenv name\n"
  508. " - delete environment variable 'name'\n"
  509. );
  510. #if ((defined(CONFIG_ENV_IS_IN_NVRAM) || defined(CONFIG_ENV_IS_IN_EEPROM) \
  511. || (defined(CONFIG_CMD_ENV) && defined(CONFIG_CMD_FLASH)) \
  512. || (defined(CONFIG_CMD_ENV) && defined(CONFIG_CMD_NAND)) \
  513. || (defined(CONFIG_CMD_ENV) && defined(CONFIG_CMD_ONENAND))) \
  514. && !defined(CONFIG_ENV_IS_NOWHERE))
  515. U_BOOT_CMD(
  516. saveenv, 1, 0, do_saveenv,
  517. "saveenv - save environment variables to persistent storage\n",
  518. NULL
  519. );
  520. #endif
  521. #if defined(CONFIG_CMD_ASKENV)
  522. U_BOOT_CMD(
  523. askenv, CFG_MAXARGS, 1, do_askenv,
  524. "askenv - get environment variables from stdin\n",
  525. "name [message] [size]\n"
  526. " - get environment variable 'name' from stdin (max 'size' chars)\n"
  527. "askenv name\n"
  528. " - get environment variable 'name' from stdin\n"
  529. "askenv name size\n"
  530. " - get environment variable 'name' from stdin (max 'size' chars)\n"
  531. "askenv name [message] size\n"
  532. " - display 'message' string and get environment variable 'name'"
  533. "from stdin (max 'size' chars)\n"
  534. );
  535. #endif
  536. #if defined(CONFIG_CMD_RUN)
  537. int do_run (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[]);
  538. U_BOOT_CMD(
  539. run, CFG_MAXARGS, 1, do_run,
  540. "run - run commands in an environment variable\n",
  541. "var [...]\n"
  542. " - run the commands in the environment variable(s) 'var'\n"
  543. );
  544. #endif