cmd_arc.c 8.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407
  1. // SPDX-License-Identifier: GPL-2.0+ OR BSD-3-Clause
  2. /*
  3. * Command for accessing Arcturus factory environment.
  4. *
  5. * Copyright 2013-2019 Arcturus Networks Inc.
  6. * https://www.arcturusnetworks.com/products/
  7. * by Oleksandr G Zhadan et al.
  8. *
  9. */
  10. #include <common.h>
  11. #include <command.h>
  12. #include <cpu_func.h>
  13. #include <div64.h>
  14. #include <env.h>
  15. #include <flash.h>
  16. #include <malloc.h>
  17. #include <spi_flash.h>
  18. #include <mmc.h>
  19. #include <version.h>
  20. #include <asm/io.h>
  21. static ulong fwenv_addr[MAX_FWENV_ADDR];
  22. const char mystrerr[] = "ERROR: Failed to save factory info";
  23. static int ishwaddr(char *hwaddr)
  24. {
  25. if (strlen(hwaddr) == MAX_HWADDR_SIZE)
  26. if (hwaddr[2] == ':' &&
  27. hwaddr[5] == ':' &&
  28. hwaddr[8] == ':' &&
  29. hwaddr[11] == ':' &&
  30. hwaddr[14] == ':')
  31. return 0;
  32. return -1;
  33. }
  34. #if (FWENV_TYPE == FWENV_MMC)
  35. static char smac[29][18] __attribute__ ((aligned(0x200))); /* 1 MMC block is 512 bytes */
  36. int set_mmc_arc_product(int argc, char *const argv[])
  37. {
  38. struct mmc *mmc;
  39. u32 blk, cnt, n;
  40. int i, err = 1;
  41. void *addr;
  42. const u8 mmc_dev_num = CONFIG_SYS_MMC_ENV_DEV;
  43. mmc = find_mmc_device(mmc_dev_num);
  44. if (!mmc) {
  45. printf("No SD/MMC/eMMC card found\n");
  46. return 0;
  47. }
  48. if (mmc_init(mmc)) {
  49. printf("%s(%d) init failed\n", IS_SD(mmc) ? "SD" : "MMC",
  50. mmc_dev_num);
  51. return 0;
  52. }
  53. if (mmc_getwp(mmc) == 1) {
  54. printf("Error: card is write protected!\n");
  55. return CMD_RET_FAILURE;
  56. }
  57. /* Save factory defaults */
  58. addr = (void *)smac;
  59. cnt = 1; /* One 512 bytes block */
  60. for (i = 0; i < MAX_FWENV_ADDR; i++)
  61. if (fwenv_addr[i] != -1) {
  62. blk = fwenv_addr[i] / 512;
  63. n = blk_dwrite(mmc_get_blk_desc(mmc), blk, cnt, addr);
  64. if (n != cnt)
  65. printf("%s: %s [%d]\n", __func__, mystrerr, i);
  66. else
  67. err = 0;
  68. }
  69. if (err)
  70. return -2;
  71. return err;
  72. }
  73. static int read_mmc_arc_info(void)
  74. {
  75. struct mmc *mmc;
  76. u32 blk, cnt, n;
  77. int i;
  78. void *addr;
  79. const u8 mmc_dev_num = CONFIG_SYS_MMC_ENV_DEV;
  80. mmc = find_mmc_device(mmc_dev_num);
  81. if (!mmc) {
  82. printf("No SD/MMC/eMMC card found\n");
  83. return 0;
  84. }
  85. if (mmc_init(mmc)) {
  86. printf("%s(%d) init failed\n", IS_SD(mmc) ? "SD" : "MMC",
  87. mmc_dev_num);
  88. return 0;
  89. }
  90. addr = (void *)smac;
  91. cnt = 1; /* One 512 bytes block */
  92. for (i = 0; i < MAX_FWENV_ADDR; i++)
  93. if (fwenv_addr[i] != -1) {
  94. blk = fwenv_addr[i] / 512;
  95. n = blk_dread(mmc_get_blk_desc(mmc), blk, cnt, addr);
  96. flush_cache((ulong) addr, 512);
  97. if (n == cnt)
  98. return (i + 1);
  99. }
  100. return 0;
  101. }
  102. #endif
  103. #if (FWENV_TYPE == FWENV_SPI_FLASH)
  104. static struct spi_flash *flash;
  105. static char smac[4][18];
  106. int set_spi_arc_product(int argc, char *const argv[])
  107. {
  108. int i, err = 1;
  109. flash = spi_flash_probe(CONFIG_ENV_SPI_BUS, CONFIG_ENV_SPI_CS,
  110. CONFIG_ENV_SPI_MAX_HZ, CONFIG_ENV_SPI_MODE);
  111. if (!flash) {
  112. printf("Failed to initialize SPI flash at %u:%u\n",
  113. CONFIG_ENV_SPI_BUS, CONFIG_ENV_SPI_CS);
  114. return -1;
  115. }
  116. /* Save factory defaults */
  117. for (i = 0; i < MAX_FWENV_ADDR; i++)
  118. if (fwenv_addr[i] != -1)
  119. if (spi_flash_write
  120. (flash, fwenv_addr[i], sizeof(smac), smac))
  121. printf("%s: %s [%d]\n", __func__, mystrerr, i);
  122. else
  123. err = 0;
  124. if (err)
  125. return -2;
  126. return err;
  127. }
  128. static int read_spi_arc_info(void)
  129. {
  130. int i;
  131. flash = spi_flash_probe(CONFIG_ENV_SPI_BUS, CONFIG_ENV_SPI_CS,
  132. CONFIG_ENV_SPI_MAX_HZ, CONFIG_ENV_SPI_MODE);
  133. if (!flash) {
  134. printf("Failed to initialize SPI flash at %u:%u\n",
  135. CONFIG_ENV_SPI_BUS, CONFIG_ENV_SPI_CS);
  136. return 0;
  137. }
  138. for (i = 0; i < MAX_FWENV_ADDR; i++)
  139. if (fwenv_addr[i] != -1)
  140. if (!spi_flash_read
  141. (flash, fwenv_addr[i], sizeof(smac), smac))
  142. return (i + 1);
  143. return 0;
  144. }
  145. #endif
  146. #if (FWENV_TYPE == FWENV_NOR_FLASH)
  147. static char smac[4][18];
  148. int set_nor_arc_product(int argc, char *const argv[])
  149. {
  150. int i, err = 1;
  151. /* Save factory defaults */
  152. for (i = 0; i < MAX_FWENV_ADDR; i++)
  153. if (fwenv_addr[i] != -1) {
  154. ulong fwenv_end = fwenv_addr[i] + 4;
  155. flash_sect_roundb(&fwenv_end);
  156. flash_sect_protect(0, fwenv_addr[i], fwenv_end);
  157. if (flash_write
  158. ((char *)smac, fwenv_addr[i], sizeof(smac)))
  159. printf("%s: %s [%d]\n", __func__, mystrerr, i);
  160. else
  161. err = 0;
  162. flash_sect_protect(1, fwenv_addr[i], fwenv_end);
  163. }
  164. if (err)
  165. return -2;
  166. return err;
  167. }
  168. static int read_nor_arc_info(void)
  169. {
  170. int i;
  171. for (i = 0; i < MAX_FWENV_ADDR; i++)
  172. if (fwenv_addr[i] != -1) {
  173. memcpy(smac, (void *)fwenv_addr[i], sizeof(smac));
  174. return (i + 1);
  175. }
  176. return 0;
  177. }
  178. #endif
  179. int set_arc_product(int argc, char *const argv[])
  180. {
  181. if (argc != 5)
  182. return -1;
  183. /* Check serial number */
  184. if (strlen(argv[1]) != MAX_SERIAL_SIZE)
  185. return -1;
  186. /* Check HWaddrs */
  187. if (ishwaddr(argv[2]) || ishwaddr(argv[3]) || ishwaddr(argv[4]))
  188. return -1;
  189. strcpy(smac[0], argv[1]);
  190. strcpy(smac[1], argv[2]);
  191. strcpy(smac[2], argv[3]);
  192. strcpy(smac[3], argv[4]);
  193. #if (FWENV_TYPE == FWENV_NOR_FLASH)
  194. return set_nor_arc_product(argc, argv);
  195. #endif
  196. #if (FWENV_TYPE == FWENV_SPI_FLASH)
  197. return set_spi_arc_product(argc, argv);
  198. #endif
  199. #if (FWENV_TYPE == FWENV_MMC)
  200. return set_mmc_arc_product(argc, argv);
  201. #endif
  202. return -2;
  203. }
  204. static int read_arc_info(void)
  205. {
  206. #if (FWENV_TYPE == FWENV_NOR_FLASH)
  207. return read_nor_arc_info();
  208. #endif
  209. #if (FWENV_TYPE == FWENV_SPI_FLASH)
  210. return read_spi_arc_info();
  211. #endif
  212. #if (FWENV_TYPE == FWENV_MMC)
  213. return read_mmc_arc_info();
  214. #endif
  215. return 0;
  216. }
  217. static int do_get_arc_info(void)
  218. {
  219. int l = read_arc_info();
  220. char *oldserial = env_get("SERIAL");
  221. char *oldversion = env_get("VERSION");
  222. if (oldversion != NULL)
  223. if (strcmp(oldversion, U_BOOT_VERSION) != 0)
  224. oldversion = NULL;
  225. if (l == 0) {
  226. printf("%s: failed to read factory info\n", __func__);
  227. return -2;
  228. }
  229. printf("\rSERIAL: ");
  230. if (smac[0][0] == EMPY_CHAR) {
  231. printf("<not found>\n");
  232. } else {
  233. printf("%s\n", smac[0]);
  234. env_set("SERIAL", smac[0]);
  235. }
  236. if (strcmp(smac[1], "00:00:00:00:00:00") == 0) {
  237. env_set("ethaddr", NULL);
  238. env_set("eth1addr", NULL);
  239. env_set("eth2addr", NULL);
  240. goto done;
  241. }
  242. printf("HWADDR0: ");
  243. if (smac[1][0] == EMPY_CHAR) {
  244. printf("<not found>\n");
  245. } else {
  246. char *ret = env_get("ethaddr");
  247. if (ret == NULL) {
  248. env_set("ethaddr", smac[1]);
  249. printf("%s\n", smac[1]);
  250. } else if (strcmp(ret, __stringify(CONFIG_ETHADDR)) == 0) {
  251. env_set("ethaddr", smac[1]);
  252. printf("%s (factory)\n", smac[1]);
  253. } else {
  254. printf("%s\n", ret);
  255. }
  256. }
  257. if (strcmp(smac[2], "00:00:00:00:00:00") == 0) {
  258. env_set("eth1addr", NULL);
  259. env_set("eth2addr", NULL);
  260. goto done;
  261. }
  262. printf("HWADDR1: ");
  263. if (smac[2][0] == EMPY_CHAR) {
  264. printf("<not found>\n");
  265. } else {
  266. char *ret = env_get("eth1addr");
  267. if (ret == NULL) {
  268. env_set("ethaddr", smac[2]);
  269. printf("%s\n", smac[2]);
  270. } else if (strcmp(ret, __stringify(CONFIG_ETH1ADDR)) == 0) {
  271. env_set("eth1addr", smac[2]);
  272. printf("%s (factory)\n", smac[2]);
  273. } else {
  274. printf("%s\n", ret);
  275. }
  276. }
  277. if (strcmp(smac[3], "00:00:00:00:00:00") == 0) {
  278. env_set("eth2addr", NULL);
  279. goto done;
  280. }
  281. printf("HWADDR2: ");
  282. if (smac[3][0] == EMPY_CHAR) {
  283. printf("<not found>\n");
  284. } else {
  285. char *ret = env_get("eth2addr");
  286. if (ret == NULL) {
  287. env_set("ethaddr", smac[3]);
  288. printf("%s\n", smac[3]);
  289. } else if (strcmp(ret, __stringify(CONFIG_ETH2ADDR)) == 0) {
  290. env_set("eth2addr", smac[3]);
  291. printf("%s (factory)\n", smac[3]);
  292. } else {
  293. printf("%s\n", ret);
  294. }
  295. }
  296. done:
  297. if (oldserial == NULL || oldversion == NULL) {
  298. if (oldversion == NULL)
  299. env_set("VERSION", U_BOOT_VERSION);
  300. env_save();
  301. }
  302. return 0;
  303. }
  304. static int init_fwenv(void)
  305. {
  306. int i, ret = -1;
  307. fwenv_addr[0] = FWENV_ADDR1;
  308. fwenv_addr[1] = FWENV_ADDR2;
  309. fwenv_addr[2] = FWENV_ADDR3;
  310. fwenv_addr[3] = FWENV_ADDR4;
  311. for (i = 0; i < MAX_FWENV_ADDR; i++)
  312. if (fwenv_addr[i] != -1)
  313. ret = 0;
  314. if (ret)
  315. printf("%s: No firmfare info storage address is defined\n",
  316. __func__);
  317. return ret;
  318. }
  319. void get_arc_info(void)
  320. {
  321. if (!init_fwenv())
  322. do_get_arc_info();
  323. }
  324. static int do_arc_cmd(struct cmd_tbl *cmdtp, int flag, int argc,
  325. char *const argv[])
  326. {
  327. const char *cmd;
  328. int ret = -1;
  329. cmd = argv[1];
  330. --argc;
  331. ++argv;
  332. if (init_fwenv())
  333. return ret;
  334. if (strcmp(cmd, "product") == 0)
  335. ret = set_arc_product(argc, argv);
  336. else if (strcmp(cmd, "info") == 0)
  337. ret = do_get_arc_info();
  338. if (ret == -1)
  339. return CMD_RET_USAGE;
  340. return ret;
  341. }
  342. U_BOOT_CMD(arc, 6, 1, do_arc_cmd,
  343. "Arcturus product command sub-system",
  344. "product serial hwaddr0 hwaddr1 hwaddr2 - save Arcturus factory env\n"
  345. "info - show Arcturus factory env\n\n");