bootm.c 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916
  1. /*
  2. * (C) Copyright 2000-2009
  3. * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
  4. *
  5. * SPDX-License-Identifier: GPL-2.0+
  6. */
  7. #ifndef USE_HOSTCC
  8. #include <common.h>
  9. #include <bootstage.h>
  10. #include <bzlib.h>
  11. #include <errno.h>
  12. #include <fdt_support.h>
  13. #include <lmb.h>
  14. #include <malloc.h>
  15. #include <asm/io.h>
  16. #include <linux/lzo.h>
  17. #include <lzma/LzmaTypes.h>
  18. #include <lzma/LzmaDec.h>
  19. #include <lzma/LzmaTools.h>
  20. #if defined(CONFIG_CMD_USB)
  21. #include <usb.h>
  22. #endif
  23. #else
  24. #include "mkimage.h"
  25. #endif
  26. #include <command.h>
  27. #include <bootm.h>
  28. #include <image.h>
  29. #ifndef CONFIG_SYS_BOOTM_LEN
  30. /* use 8MByte as default max gunzip size */
  31. #define CONFIG_SYS_BOOTM_LEN 0x800000
  32. #endif
  33. #define IH_INITRD_ARCH IH_ARCH_DEFAULT
  34. #ifndef USE_HOSTCC
  35. DECLARE_GLOBAL_DATA_PTR;
  36. static const void *boot_get_kernel(cmd_tbl_t *cmdtp, int flag, int argc,
  37. char * const argv[], bootm_headers_t *images,
  38. ulong *os_data, ulong *os_len);
  39. #ifdef CONFIG_LMB
  40. static void boot_start_lmb(bootm_headers_t *images)
  41. {
  42. ulong mem_start;
  43. phys_size_t mem_size;
  44. lmb_init(&images->lmb);
  45. mem_start = getenv_bootm_low();
  46. mem_size = getenv_bootm_size();
  47. lmb_add(&images->lmb, (phys_addr_t)mem_start, mem_size);
  48. arch_lmb_reserve(&images->lmb);
  49. board_lmb_reserve(&images->lmb);
  50. }
  51. #else
  52. #define lmb_reserve(lmb, base, size)
  53. static inline void boot_start_lmb(bootm_headers_t *images) { }
  54. #endif
  55. static int bootm_start(cmd_tbl_t *cmdtp, int flag, int argc,
  56. char * const argv[])
  57. {
  58. memset((void *)&images, 0, sizeof(images));
  59. images.verify = getenv_yesno("verify");
  60. boot_start_lmb(&images);
  61. bootstage_mark_name(BOOTSTAGE_ID_BOOTM_START, "bootm_start");
  62. images.state = BOOTM_STATE_START;
  63. return 0;
  64. }
  65. static int bootm_find_os(cmd_tbl_t *cmdtp, int flag, int argc,
  66. char * const argv[])
  67. {
  68. const void *os_hdr;
  69. bool ep_found = false;
  70. int ret;
  71. /* get kernel image header, start address and length */
  72. os_hdr = boot_get_kernel(cmdtp, flag, argc, argv,
  73. &images, &images.os.image_start, &images.os.image_len);
  74. if (images.os.image_len == 0) {
  75. puts("ERROR: can't get kernel image!\n");
  76. return 1;
  77. }
  78. /* get image parameters */
  79. switch (genimg_get_format(os_hdr)) {
  80. #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
  81. case IMAGE_FORMAT_LEGACY:
  82. images.os.type = image_get_type(os_hdr);
  83. images.os.comp = image_get_comp(os_hdr);
  84. images.os.os = image_get_os(os_hdr);
  85. images.os.end = image_get_image_end(os_hdr);
  86. images.os.load = image_get_load(os_hdr);
  87. images.os.arch = image_get_arch(os_hdr);
  88. break;
  89. #endif
  90. #if defined(CONFIG_FIT)
  91. case IMAGE_FORMAT_FIT:
  92. if (fit_image_get_type(images.fit_hdr_os,
  93. images.fit_noffset_os,
  94. &images.os.type)) {
  95. puts("Can't get image type!\n");
  96. bootstage_error(BOOTSTAGE_ID_FIT_TYPE);
  97. return 1;
  98. }
  99. if (fit_image_get_comp(images.fit_hdr_os,
  100. images.fit_noffset_os,
  101. &images.os.comp)) {
  102. puts("Can't get image compression!\n");
  103. bootstage_error(BOOTSTAGE_ID_FIT_COMPRESSION);
  104. return 1;
  105. }
  106. if (fit_image_get_os(images.fit_hdr_os, images.fit_noffset_os,
  107. &images.os.os)) {
  108. puts("Can't get image OS!\n");
  109. bootstage_error(BOOTSTAGE_ID_FIT_OS);
  110. return 1;
  111. }
  112. if (fit_image_get_arch(images.fit_hdr_os,
  113. images.fit_noffset_os,
  114. &images.os.arch)) {
  115. puts("Can't get image ARCH!\n");
  116. return 1;
  117. }
  118. images.os.end = fit_get_end(images.fit_hdr_os);
  119. if (fit_image_get_load(images.fit_hdr_os, images.fit_noffset_os,
  120. &images.os.load)) {
  121. puts("Can't get image load address!\n");
  122. bootstage_error(BOOTSTAGE_ID_FIT_LOADADDR);
  123. return 1;
  124. }
  125. break;
  126. #endif
  127. #ifdef CONFIG_ANDROID_BOOT_IMAGE
  128. case IMAGE_FORMAT_ANDROID:
  129. images.os.type = IH_TYPE_KERNEL;
  130. images.os.comp = IH_COMP_NONE;
  131. images.os.os = IH_OS_LINUX;
  132. images.os.end = android_image_get_end(os_hdr);
  133. images.os.load = android_image_get_kload(os_hdr);
  134. images.ep = images.os.load;
  135. ep_found = true;
  136. break;
  137. #endif
  138. default:
  139. puts("ERROR: unknown image format type!\n");
  140. return 1;
  141. }
  142. /* If we have a valid setup.bin, we will use that for entry (x86) */
  143. if (images.os.arch == IH_ARCH_I386 ||
  144. images.os.arch == IH_ARCH_X86_64) {
  145. ulong len;
  146. ret = boot_get_setup(&images, IH_ARCH_I386, &images.ep, &len);
  147. if (ret < 0 && ret != -ENOENT) {
  148. puts("Could not find a valid setup.bin for x86\n");
  149. return 1;
  150. }
  151. /* Kernel entry point is the setup.bin */
  152. } else if (images.legacy_hdr_valid) {
  153. images.ep = image_get_ep(&images.legacy_hdr_os_copy);
  154. #if defined(CONFIG_FIT)
  155. } else if (images.fit_uname_os) {
  156. int ret;
  157. ret = fit_image_get_entry(images.fit_hdr_os,
  158. images.fit_noffset_os, &images.ep);
  159. if (ret) {
  160. puts("Can't get entry point property!\n");
  161. return 1;
  162. }
  163. #endif
  164. } else if (!ep_found) {
  165. puts("Could not find kernel entry point!\n");
  166. return 1;
  167. }
  168. if (images.os.type == IH_TYPE_KERNEL_NOLOAD) {
  169. images.os.load = images.os.image_start;
  170. images.ep += images.os.load;
  171. }
  172. images.os.start = (ulong)os_hdr;
  173. return 0;
  174. }
  175. static int bootm_find_ramdisk(int flag, int argc, char * const argv[])
  176. {
  177. int ret;
  178. /* find ramdisk */
  179. ret = boot_get_ramdisk(argc, argv, &images, IH_INITRD_ARCH,
  180. &images.rd_start, &images.rd_end);
  181. if (ret) {
  182. puts("Ramdisk image is corrupt or invalid\n");
  183. return 1;
  184. }
  185. return 0;
  186. }
  187. #if defined(CONFIG_OF_LIBFDT)
  188. static int bootm_find_fdt(int flag, int argc, char * const argv[])
  189. {
  190. int ret;
  191. /* find flattened device tree */
  192. ret = boot_get_fdt(flag, argc, argv, IH_ARCH_DEFAULT, &images,
  193. &images.ft_addr, &images.ft_len);
  194. if (ret) {
  195. puts("Could not find a valid device tree\n");
  196. return 1;
  197. }
  198. set_working_fdt_addr(images.ft_addr);
  199. return 0;
  200. }
  201. #endif
  202. int bootm_find_ramdisk_fdt(int flag, int argc, char * const argv[])
  203. {
  204. if (bootm_find_ramdisk(flag, argc, argv))
  205. return 1;
  206. #if defined(CONFIG_OF_LIBFDT)
  207. if (bootm_find_fdt(flag, argc, argv))
  208. return 1;
  209. #endif
  210. return 0;
  211. }
  212. static int bootm_find_other(cmd_tbl_t *cmdtp, int flag, int argc,
  213. char * const argv[])
  214. {
  215. if (((images.os.type == IH_TYPE_KERNEL) ||
  216. (images.os.type == IH_TYPE_KERNEL_NOLOAD) ||
  217. (images.os.type == IH_TYPE_MULTI)) &&
  218. (images.os.os == IH_OS_LINUX ||
  219. images.os.os == IH_OS_VXWORKS))
  220. return bootm_find_ramdisk_fdt(flag, argc, argv);
  221. return 0;
  222. }
  223. #endif /* USE_HOSTCC */
  224. /**
  225. * decomp_image() - decompress the operating system
  226. *
  227. * @comp: Compression algorithm that is used (IH_COMP_...)
  228. * @load: Destination load address in U-Boot memory
  229. * @image_start Image start address (where we are decompressing from)
  230. * @type: OS type (IH_OS_...)
  231. * @load_bug: Place to decompress to
  232. * @image_buf: Address to decompress from
  233. * @return 0 if OK, -ve on error (BOOTM_ERR_...)
  234. */
  235. static int decomp_image(int comp, ulong load, ulong image_start, int type,
  236. void *load_buf, void *image_buf, ulong image_len,
  237. ulong *load_end)
  238. {
  239. const char *type_name = genimg_get_type_name(type);
  240. __attribute__((unused)) uint unc_len = CONFIG_SYS_BOOTM_LEN;
  241. *load_end = load;
  242. switch (comp) {
  243. case IH_COMP_NONE:
  244. if (load == image_start) {
  245. printf(" XIP %s ... ", type_name);
  246. } else {
  247. printf(" Loading %s ... ", type_name);
  248. memmove_wd(load_buf, image_buf, image_len, CHUNKSZ);
  249. }
  250. *load_end = load + image_len;
  251. break;
  252. #ifdef CONFIG_GZIP
  253. case IH_COMP_GZIP:
  254. printf(" Uncompressing %s ... ", type_name);
  255. if (gunzip(load_buf, unc_len, image_buf, &image_len) != 0) {
  256. puts("GUNZIP: uncompress, out-of-mem or overwrite error - must RESET board to recover\n");
  257. return BOOTM_ERR_RESET;
  258. }
  259. *load_end = load + image_len;
  260. break;
  261. #endif /* CONFIG_GZIP */
  262. #ifdef CONFIG_BZIP2
  263. case IH_COMP_BZIP2:
  264. printf(" Uncompressing %s ... ", type_name);
  265. /*
  266. * If we've got less than 4 MB of malloc() space,
  267. * use slower decompression algorithm which requires
  268. * at most 2300 KB of memory.
  269. */
  270. int i = BZ2_bzBuffToBuffDecompress(load_buf, &unc_len,
  271. image_buf, image_len,
  272. CONFIG_SYS_MALLOC_LEN < (4096 * 1024), 0);
  273. if (i != BZ_OK) {
  274. printf("BUNZIP2: uncompress or overwrite error %d - must RESET board to recover\n",
  275. i);
  276. return BOOTM_ERR_RESET;
  277. }
  278. *load_end = load + unc_len;
  279. break;
  280. #endif /* CONFIG_BZIP2 */
  281. #ifdef CONFIG_LZMA
  282. case IH_COMP_LZMA: {
  283. SizeT lzma_len = unc_len;
  284. int ret;
  285. printf(" Uncompressing %s ... ", type_name);
  286. ret = lzmaBuffToBuffDecompress(load_buf, &lzma_len,
  287. image_buf, image_len);
  288. unc_len = lzma_len;
  289. if (ret != SZ_OK) {
  290. printf("LZMA: uncompress or overwrite error %d - must RESET board to recover\n",
  291. ret);
  292. bootstage_error(BOOTSTAGE_ID_DECOMP_IMAGE);
  293. return BOOTM_ERR_RESET;
  294. }
  295. *load_end = load + unc_len;
  296. break;
  297. }
  298. #endif /* CONFIG_LZMA */
  299. #ifdef CONFIG_LZO
  300. case IH_COMP_LZO: {
  301. size_t size = unc_len;
  302. int ret;
  303. printf(" Uncompressing %s ... ", type_name);
  304. ret = lzop_decompress(image_buf, image_len, load_buf, &size);
  305. if (ret != LZO_E_OK) {
  306. printf("LZO: uncompress or overwrite error %d - must RESET board to recover\n",
  307. ret);
  308. return BOOTM_ERR_RESET;
  309. }
  310. *load_end = load + size;
  311. break;
  312. }
  313. #endif /* CONFIG_LZO */
  314. default:
  315. printf("Unimplemented compression type %d\n", comp);
  316. return BOOTM_ERR_UNIMPLEMENTED;
  317. }
  318. puts("OK\n");
  319. return 0;
  320. }
  321. #ifndef USE_HOSTCC
  322. static int bootm_load_os(bootm_headers_t *images, unsigned long *load_end,
  323. int boot_progress)
  324. {
  325. image_info_t os = images->os;
  326. ulong load = os.load;
  327. ulong blob_start = os.start;
  328. ulong blob_end = os.end;
  329. ulong image_start = os.image_start;
  330. ulong image_len = os.image_len;
  331. bool no_overlap;
  332. void *load_buf, *image_buf;
  333. int err;
  334. load_buf = map_sysmem(load, 0);
  335. image_buf = map_sysmem(os.image_start, image_len);
  336. err = decomp_image(os.comp, load, os.image_start, os.type, load_buf,
  337. image_buf, image_len, load_end);
  338. if (err) {
  339. bootstage_error(BOOTSTAGE_ID_DECOMP_IMAGE);
  340. return err;
  341. }
  342. flush_cache(load, (*load_end - load) * sizeof(ulong));
  343. debug(" kernel loaded at 0x%08lx, end = 0x%08lx\n", load, *load_end);
  344. bootstage_mark(BOOTSTAGE_ID_KERNEL_LOADED);
  345. no_overlap = (os.comp == IH_COMP_NONE && load == image_start);
  346. if (!no_overlap && (load < blob_end) && (*load_end > blob_start)) {
  347. debug("images.os.start = 0x%lX, images.os.end = 0x%lx\n",
  348. blob_start, blob_end);
  349. debug("images.os.load = 0x%lx, load_end = 0x%lx\n", load,
  350. *load_end);
  351. /* Check what type of image this is. */
  352. if (images->legacy_hdr_valid) {
  353. if (image_get_type(&images->legacy_hdr_os_copy)
  354. == IH_TYPE_MULTI)
  355. puts("WARNING: legacy format multi component image overwritten\n");
  356. return BOOTM_ERR_OVERLAP;
  357. } else {
  358. puts("ERROR: new format image overwritten - must RESET the board to recover\n");
  359. bootstage_error(BOOTSTAGE_ID_OVERWRITTEN);
  360. return BOOTM_ERR_RESET;
  361. }
  362. }
  363. return 0;
  364. }
  365. /**
  366. * bootm_disable_interrupts() - Disable interrupts in preparation for load/boot
  367. *
  368. * @return interrupt flag (0 if interrupts were disabled, non-zero if they were
  369. * enabled)
  370. */
  371. ulong bootm_disable_interrupts(void)
  372. {
  373. ulong iflag;
  374. /*
  375. * We have reached the point of no return: we are going to
  376. * overwrite all exception vector code, so we cannot easily
  377. * recover from any failures any more...
  378. */
  379. iflag = disable_interrupts();
  380. #ifdef CONFIG_NETCONSOLE
  381. /* Stop the ethernet stack if NetConsole could have left it up */
  382. eth_halt();
  383. eth_unregister(eth_get_dev());
  384. #endif
  385. #if defined(CONFIG_CMD_USB)
  386. /*
  387. * turn off USB to prevent the host controller from writing to the
  388. * SDRAM while Linux is booting. This could happen (at least for OHCI
  389. * controller), because the HCCA (Host Controller Communication Area)
  390. * lies within the SDRAM and the host controller writes continously to
  391. * this area (as busmaster!). The HccaFrameNumber is for example
  392. * updated every 1 ms within the HCCA structure in SDRAM! For more
  393. * details see the OpenHCI specification.
  394. */
  395. usb_stop();
  396. #endif
  397. return iflag;
  398. }
  399. #if defined(CONFIG_SILENT_CONSOLE) && !defined(CONFIG_SILENT_U_BOOT_ONLY)
  400. #define CONSOLE_ARG "console="
  401. #define CONSOLE_ARG_LEN (sizeof(CONSOLE_ARG) - 1)
  402. static void fixup_silent_linux(void)
  403. {
  404. char *buf;
  405. const char *env_val;
  406. char *cmdline = getenv("bootargs");
  407. int want_silent;
  408. /*
  409. * Only fix cmdline when requested. The environment variable can be:
  410. *
  411. * no - we never fixup
  412. * yes - we always fixup
  413. * unset - we rely on the console silent flag
  414. */
  415. want_silent = getenv_yesno("silent_linux");
  416. if (want_silent == 0)
  417. return;
  418. else if (want_silent == -1 && !(gd->flags & GD_FLG_SILENT))
  419. return;
  420. debug("before silent fix-up: %s\n", cmdline);
  421. if (cmdline && (cmdline[0] != '\0')) {
  422. char *start = strstr(cmdline, CONSOLE_ARG);
  423. /* Allocate space for maximum possible new command line */
  424. buf = malloc(strlen(cmdline) + 1 + CONSOLE_ARG_LEN + 1);
  425. if (!buf) {
  426. debug("%s: out of memory\n", __func__);
  427. return;
  428. }
  429. if (start) {
  430. char *end = strchr(start, ' ');
  431. int num_start_bytes = start - cmdline + CONSOLE_ARG_LEN;
  432. strncpy(buf, cmdline, num_start_bytes);
  433. if (end)
  434. strcpy(buf + num_start_bytes, end);
  435. else
  436. buf[num_start_bytes] = '\0';
  437. } else {
  438. sprintf(buf, "%s %s", cmdline, CONSOLE_ARG);
  439. }
  440. env_val = buf;
  441. } else {
  442. buf = NULL;
  443. env_val = CONSOLE_ARG;
  444. }
  445. setenv("bootargs", env_val);
  446. debug("after silent fix-up: %s\n", env_val);
  447. free(buf);
  448. }
  449. #endif /* CONFIG_SILENT_CONSOLE */
  450. /**
  451. * Execute selected states of the bootm command.
  452. *
  453. * Note the arguments to this state must be the first argument, Any 'bootm'
  454. * or sub-command arguments must have already been taken.
  455. *
  456. * Note that if states contains more than one flag it MUST contain
  457. * BOOTM_STATE_START, since this handles and consumes the command line args.
  458. *
  459. * Also note that aside from boot_os_fn functions and bootm_load_os no other
  460. * functions we store the return value of in 'ret' may use a negative return
  461. * value, without special handling.
  462. *
  463. * @param cmdtp Pointer to bootm command table entry
  464. * @param flag Command flags (CMD_FLAG_...)
  465. * @param argc Number of subcommand arguments (0 = no arguments)
  466. * @param argv Arguments
  467. * @param states Mask containing states to run (BOOTM_STATE_...)
  468. * @param images Image header information
  469. * @param boot_progress 1 to show boot progress, 0 to not do this
  470. * @return 0 if ok, something else on error. Some errors will cause this
  471. * function to perform a reboot! If states contains BOOTM_STATE_OS_GO
  472. * then the intent is to boot an OS, so this function will not return
  473. * unless the image type is standalone.
  474. */
  475. int do_bootm_states(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
  476. int states, bootm_headers_t *images, int boot_progress)
  477. {
  478. boot_os_fn *boot_fn;
  479. ulong iflag = 0;
  480. int ret = 0, need_boot_fn;
  481. images->state |= states;
  482. /*
  483. * Work through the states and see how far we get. We stop on
  484. * any error.
  485. */
  486. if (states & BOOTM_STATE_START)
  487. ret = bootm_start(cmdtp, flag, argc, argv);
  488. if (!ret && (states & BOOTM_STATE_FINDOS))
  489. ret = bootm_find_os(cmdtp, flag, argc, argv);
  490. if (!ret && (states & BOOTM_STATE_FINDOTHER)) {
  491. ret = bootm_find_other(cmdtp, flag, argc, argv);
  492. argc = 0; /* consume the args */
  493. }
  494. /* Load the OS */
  495. if (!ret && (states & BOOTM_STATE_LOADOS)) {
  496. ulong load_end;
  497. iflag = bootm_disable_interrupts();
  498. ret = bootm_load_os(images, &load_end, 0);
  499. if (ret == 0)
  500. lmb_reserve(&images->lmb, images->os.load,
  501. (load_end - images->os.load));
  502. else if (ret && ret != BOOTM_ERR_OVERLAP)
  503. goto err;
  504. else if (ret == BOOTM_ERR_OVERLAP)
  505. ret = 0;
  506. #if defined(CONFIG_SILENT_CONSOLE) && !defined(CONFIG_SILENT_U_BOOT_ONLY)
  507. if (images->os.os == IH_OS_LINUX)
  508. fixup_silent_linux();
  509. #endif
  510. }
  511. /* Relocate the ramdisk */
  512. #ifdef CONFIG_SYS_BOOT_RAMDISK_HIGH
  513. if (!ret && (states & BOOTM_STATE_RAMDISK)) {
  514. ulong rd_len = images->rd_end - images->rd_start;
  515. ret = boot_ramdisk_high(&images->lmb, images->rd_start,
  516. rd_len, &images->initrd_start, &images->initrd_end);
  517. if (!ret) {
  518. setenv_hex("initrd_start", images->initrd_start);
  519. setenv_hex("initrd_end", images->initrd_end);
  520. }
  521. }
  522. #endif
  523. #if defined(CONFIG_OF_LIBFDT) && defined(CONFIG_LMB)
  524. if (!ret && (states & BOOTM_STATE_FDT)) {
  525. boot_fdt_add_mem_rsv_regions(&images->lmb, images->ft_addr);
  526. ret = boot_relocate_fdt(&images->lmb, &images->ft_addr,
  527. &images->ft_len);
  528. }
  529. #endif
  530. /* From now on, we need the OS boot function */
  531. if (ret)
  532. return ret;
  533. boot_fn = bootm_os_get_boot_func(images->os.os);
  534. need_boot_fn = states & (BOOTM_STATE_OS_CMDLINE |
  535. BOOTM_STATE_OS_BD_T | BOOTM_STATE_OS_PREP |
  536. BOOTM_STATE_OS_FAKE_GO | BOOTM_STATE_OS_GO);
  537. if (boot_fn == NULL && need_boot_fn) {
  538. if (iflag)
  539. enable_interrupts();
  540. printf("ERROR: booting os '%s' (%d) is not supported\n",
  541. genimg_get_os_name(images->os.os), images->os.os);
  542. bootstage_error(BOOTSTAGE_ID_CHECK_BOOT_OS);
  543. return 1;
  544. }
  545. /* Call various other states that are not generally used */
  546. if (!ret && (states & BOOTM_STATE_OS_CMDLINE))
  547. ret = boot_fn(BOOTM_STATE_OS_CMDLINE, argc, argv, images);
  548. if (!ret && (states & BOOTM_STATE_OS_BD_T))
  549. ret = boot_fn(BOOTM_STATE_OS_BD_T, argc, argv, images);
  550. if (!ret && (states & BOOTM_STATE_OS_PREP))
  551. ret = boot_fn(BOOTM_STATE_OS_PREP, argc, argv, images);
  552. #ifdef CONFIG_TRACE
  553. /* Pretend to run the OS, then run a user command */
  554. if (!ret && (states & BOOTM_STATE_OS_FAKE_GO)) {
  555. char *cmd_list = getenv("fakegocmd");
  556. ret = boot_selected_os(argc, argv, BOOTM_STATE_OS_FAKE_GO,
  557. images, boot_fn);
  558. if (!ret && cmd_list)
  559. ret = run_command_list(cmd_list, -1, flag);
  560. }
  561. #endif
  562. /* Check for unsupported subcommand. */
  563. if (ret) {
  564. puts("subcommand not supported\n");
  565. return ret;
  566. }
  567. /* Now run the OS! We hope this doesn't return */
  568. if (!ret && (states & BOOTM_STATE_OS_GO))
  569. ret = boot_selected_os(argc, argv, BOOTM_STATE_OS_GO,
  570. images, boot_fn);
  571. /* Deal with any fallout */
  572. err:
  573. if (iflag)
  574. enable_interrupts();
  575. if (ret == BOOTM_ERR_UNIMPLEMENTED)
  576. bootstage_error(BOOTSTAGE_ID_DECOMP_UNIMPL);
  577. else if (ret == BOOTM_ERR_RESET)
  578. do_reset(cmdtp, flag, argc, argv);
  579. return ret;
  580. }
  581. #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
  582. /**
  583. * image_get_kernel - verify legacy format kernel image
  584. * @img_addr: in RAM address of the legacy format image to be verified
  585. * @verify: data CRC verification flag
  586. *
  587. * image_get_kernel() verifies legacy image integrity and returns pointer to
  588. * legacy image header if image verification was completed successfully.
  589. *
  590. * returns:
  591. * pointer to a legacy image header if valid image was found
  592. * otherwise return NULL
  593. */
  594. static image_header_t *image_get_kernel(ulong img_addr, int verify)
  595. {
  596. image_header_t *hdr = (image_header_t *)img_addr;
  597. if (!image_check_magic(hdr)) {
  598. puts("Bad Magic Number\n");
  599. bootstage_error(BOOTSTAGE_ID_CHECK_MAGIC);
  600. return NULL;
  601. }
  602. bootstage_mark(BOOTSTAGE_ID_CHECK_HEADER);
  603. if (!image_check_hcrc(hdr)) {
  604. puts("Bad Header Checksum\n");
  605. bootstage_error(BOOTSTAGE_ID_CHECK_HEADER);
  606. return NULL;
  607. }
  608. bootstage_mark(BOOTSTAGE_ID_CHECK_CHECKSUM);
  609. image_print_contents(hdr);
  610. if (verify) {
  611. puts(" Verifying Checksum ... ");
  612. if (!image_check_dcrc(hdr)) {
  613. printf("Bad Data CRC\n");
  614. bootstage_error(BOOTSTAGE_ID_CHECK_CHECKSUM);
  615. return NULL;
  616. }
  617. puts("OK\n");
  618. }
  619. bootstage_mark(BOOTSTAGE_ID_CHECK_ARCH);
  620. if (!image_check_target_arch(hdr)) {
  621. printf("Unsupported Architecture 0x%x\n", image_get_arch(hdr));
  622. bootstage_error(BOOTSTAGE_ID_CHECK_ARCH);
  623. return NULL;
  624. }
  625. return hdr;
  626. }
  627. #endif
  628. /**
  629. * boot_get_kernel - find kernel image
  630. * @os_data: pointer to a ulong variable, will hold os data start address
  631. * @os_len: pointer to a ulong variable, will hold os data length
  632. *
  633. * boot_get_kernel() tries to find a kernel image, verifies its integrity
  634. * and locates kernel data.
  635. *
  636. * returns:
  637. * pointer to image header if valid image was found, plus kernel start
  638. * address and length, otherwise NULL
  639. */
  640. static const void *boot_get_kernel(cmd_tbl_t *cmdtp, int flag, int argc,
  641. char * const argv[], bootm_headers_t *images,
  642. ulong *os_data, ulong *os_len)
  643. {
  644. #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
  645. image_header_t *hdr;
  646. #endif
  647. ulong img_addr;
  648. const void *buf;
  649. const char *fit_uname_config = NULL;
  650. const char *fit_uname_kernel = NULL;
  651. #if defined(CONFIG_FIT)
  652. int os_noffset;
  653. #endif
  654. img_addr = genimg_get_kernel_addr_fit(argc < 1 ? NULL : argv[0],
  655. &fit_uname_config,
  656. &fit_uname_kernel);
  657. bootstage_mark(BOOTSTAGE_ID_CHECK_MAGIC);
  658. /* copy from dataflash if needed */
  659. img_addr = genimg_get_image(img_addr);
  660. /* check image type, for FIT images get FIT kernel node */
  661. *os_data = *os_len = 0;
  662. buf = map_sysmem(img_addr, 0);
  663. switch (genimg_get_format(buf)) {
  664. #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
  665. case IMAGE_FORMAT_LEGACY:
  666. printf("## Booting kernel from Legacy Image at %08lx ...\n",
  667. img_addr);
  668. hdr = image_get_kernel(img_addr, images->verify);
  669. if (!hdr)
  670. return NULL;
  671. bootstage_mark(BOOTSTAGE_ID_CHECK_IMAGETYPE);
  672. /* get os_data and os_len */
  673. switch (image_get_type(hdr)) {
  674. case IH_TYPE_KERNEL:
  675. case IH_TYPE_KERNEL_NOLOAD:
  676. *os_data = image_get_data(hdr);
  677. *os_len = image_get_data_size(hdr);
  678. break;
  679. case IH_TYPE_MULTI:
  680. image_multi_getimg(hdr, 0, os_data, os_len);
  681. break;
  682. case IH_TYPE_STANDALONE:
  683. *os_data = image_get_data(hdr);
  684. *os_len = image_get_data_size(hdr);
  685. break;
  686. default:
  687. printf("Wrong Image Type for %s command\n",
  688. cmdtp->name);
  689. bootstage_error(BOOTSTAGE_ID_CHECK_IMAGETYPE);
  690. return NULL;
  691. }
  692. /*
  693. * copy image header to allow for image overwrites during
  694. * kernel decompression.
  695. */
  696. memmove(&images->legacy_hdr_os_copy, hdr,
  697. sizeof(image_header_t));
  698. /* save pointer to image header */
  699. images->legacy_hdr_os = hdr;
  700. images->legacy_hdr_valid = 1;
  701. bootstage_mark(BOOTSTAGE_ID_DECOMP_IMAGE);
  702. break;
  703. #endif
  704. #if defined(CONFIG_FIT)
  705. case IMAGE_FORMAT_FIT:
  706. os_noffset = fit_image_load(images, img_addr,
  707. &fit_uname_kernel, &fit_uname_config,
  708. IH_ARCH_DEFAULT, IH_TYPE_KERNEL,
  709. BOOTSTAGE_ID_FIT_KERNEL_START,
  710. FIT_LOAD_IGNORED, os_data, os_len);
  711. if (os_noffset < 0)
  712. return NULL;
  713. images->fit_hdr_os = map_sysmem(img_addr, 0);
  714. images->fit_uname_os = fit_uname_kernel;
  715. images->fit_uname_cfg = fit_uname_config;
  716. images->fit_noffset_os = os_noffset;
  717. break;
  718. #endif
  719. #ifdef CONFIG_ANDROID_BOOT_IMAGE
  720. case IMAGE_FORMAT_ANDROID:
  721. printf("## Booting Android Image at 0x%08lx ...\n", img_addr);
  722. if (android_image_get_kernel(buf, images->verify,
  723. os_data, os_len))
  724. return NULL;
  725. break;
  726. #endif
  727. default:
  728. printf("Wrong Image Format for %s command\n", cmdtp->name);
  729. bootstage_error(BOOTSTAGE_ID_FIT_KERNEL_INFO);
  730. return NULL;
  731. }
  732. debug(" kernel data at 0x%08lx, len = 0x%08lx (%ld)\n",
  733. *os_data, *os_len, *os_len);
  734. return buf;
  735. }
  736. #else /* USE_HOSTCC */
  737. void memmove_wd(void *to, void *from, size_t len, ulong chunksz)
  738. {
  739. memmove(to, from, len);
  740. }
  741. static int bootm_host_load_image(const void *fit, int req_image_type)
  742. {
  743. const char *fit_uname_config = NULL;
  744. ulong data, len;
  745. bootm_headers_t images;
  746. int noffset;
  747. ulong load_end;
  748. uint8_t image_type;
  749. uint8_t imape_comp;
  750. void *load_buf;
  751. int ret;
  752. memset(&images, '\0', sizeof(images));
  753. images.verify = 1;
  754. noffset = fit_image_load(&images, (ulong)fit,
  755. NULL, &fit_uname_config,
  756. IH_ARCH_DEFAULT, req_image_type, -1,
  757. FIT_LOAD_IGNORED, &data, &len);
  758. if (noffset < 0)
  759. return noffset;
  760. if (fit_image_get_type(fit, noffset, &image_type)) {
  761. puts("Can't get image type!\n");
  762. return -EINVAL;
  763. }
  764. if (fit_image_get_comp(fit, noffset, &imape_comp)) {
  765. puts("Can't get image compression!\n");
  766. return -EINVAL;
  767. }
  768. /* Allow the image to expand by a factor of 4, should be safe */
  769. load_buf = malloc((1 << 20) + len * 4);
  770. ret = decomp_image(imape_comp, 0, data, image_type, load_buf,
  771. (void *)data, len, &load_end);
  772. free(load_buf);
  773. if (ret && ret != BOOTM_ERR_UNIMPLEMENTED)
  774. return ret;
  775. return 0;
  776. }
  777. int bootm_host_load_images(const void *fit, int cfg_noffset)
  778. {
  779. static uint8_t image_types[] = {
  780. IH_TYPE_KERNEL,
  781. IH_TYPE_FLATDT,
  782. IH_TYPE_RAMDISK,
  783. };
  784. int err = 0;
  785. int i;
  786. for (i = 0; i < ARRAY_SIZE(image_types); i++) {
  787. int ret;
  788. ret = bootm_host_load_image(fit, image_types[i]);
  789. if (!err && ret && ret != -ENOENT)
  790. err = ret;
  791. }
  792. /* Return the first error we found */
  793. return err;
  794. }
  795. #endif /* ndef USE_HOSTCC */