cmd_nvedit.c 14 KB

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