blob: cf5c0d4393b0937b55bb684023a43a2c0d62f9cd [file] [log] [blame]
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +05301/*
2 * (C) Copyright 2008 Semihalf
3 *
4 * (C) Copyright 2000-2004
5 * DENX Software Engineering
6 * Wolfgang Denk, wd@denx.de
7 *
8 * Updated-by: Prafulla Wadaskar <prafulla@marvell.com>
9 * default_image specific code abstracted from mkimage.c
10 * some functions added to address abstraction
11 *
12 * All rights reserved.
13 *
Wolfgang Denk1a459662013-07-08 09:37:19 +020014 * SPDX-License-Identifier: GPL-2.0+
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053015 */
16
Guilherme Maciel Ferreiraf86ed6a2013-12-01 12:43:10 -070017#include "imagetool.h"
Guilherme Maciel Ferreira26621792015-01-15 02:54:43 -020018#include "mkimage.h"
19
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053020#include <image.h>
21#include <u-boot/crc.h>
22
23static image_header_t header;
24
Stephen Warren712fbcf2011-10-18 11:11:49 +000025static int image_check_image_types(uint8_t type)
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053026{
Stephen Warrenb9b50e892011-11-10 13:17:53 -070027 if (((type > IH_TYPE_INVALID) && (type < IH_TYPE_FLATDT)) ||
28 (type == IH_TYPE_KERNEL_NOLOAD))
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053029 return EXIT_SUCCESS;
30 else
31 return EXIT_FAILURE;
32}
33
Guilherme Maciel Ferreiraf86ed6a2013-12-01 12:43:10 -070034static int image_check_params(struct image_tool_params *params)
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053035{
36 return ((params->dflag && (params->fflag || params->lflag)) ||
37 (params->fflag && (params->dflag || params->lflag)) ||
38 (params->lflag && (params->dflag || params->fflag)));
39}
40
Stephen Warren712fbcf2011-10-18 11:11:49 +000041static int image_verify_header(unsigned char *ptr, int image_size,
Guilherme Maciel Ferreiraf86ed6a2013-12-01 12:43:10 -070042 struct image_tool_params *params)
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053043{
44 uint32_t len;
45 const unsigned char *data;
46 uint32_t checksum;
47 image_header_t header;
48 image_header_t *hdr = &header;
49
50 /*
51 * create copy of header so that we can blank out the
52 * checksum field for checking - this can't be done
53 * on the PROT_READ mapped data.
54 */
Stephen Warren712fbcf2011-10-18 11:11:49 +000055 memcpy(hdr, ptr, sizeof(image_header_t));
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053056
57 if (be32_to_cpu(hdr->ih_magic) != IH_MAGIC) {
Guilherme Maciel Ferreira26621792015-01-15 02:54:43 -020058 debug("%s: Bad Magic Number: \"%s\" is no valid image\n",
59 params->cmdname, params->imagefile);
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053060 return -FDT_ERR_BADMAGIC;
61 }
62
63 data = (const unsigned char *)hdr;
64 len = sizeof(image_header_t);
65
66 checksum = be32_to_cpu(hdr->ih_hcrc);
67 hdr->ih_hcrc = cpu_to_be32(0); /* clear for re-calculation */
68
Stephen Warren712fbcf2011-10-18 11:11:49 +000069 if (crc32(0, data, len) != checksum) {
Guilherme Maciel Ferreira26621792015-01-15 02:54:43 -020070 debug("%s: ERROR: \"%s\" has bad header checksum!\n",
71 params->cmdname, params->imagefile);
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053072 return -FDT_ERR_BADSTATE;
73 }
74
75 data = (const unsigned char *)ptr + sizeof(image_header_t);
76 len = image_size - sizeof(image_header_t) ;
77
78 checksum = be32_to_cpu(hdr->ih_dcrc);
Stephen Warren712fbcf2011-10-18 11:11:49 +000079 if (crc32(0, data, len) != checksum) {
Guilherme Maciel Ferreira26621792015-01-15 02:54:43 -020080 debug("%s: ERROR: \"%s\" has corrupted data!\n",
81 params->cmdname, params->imagefile);
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053082 return -FDT_ERR_BADSTRUCTURE;
83 }
84 return 0;
85}
86
Stephen Warren712fbcf2011-10-18 11:11:49 +000087static void image_set_header(void *ptr, struct stat *sbuf, int ifd,
Guilherme Maciel Ferreiraf86ed6a2013-12-01 12:43:10 -070088 struct image_tool_params *params)
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053089{
90 uint32_t checksum;
91
92 image_header_t * hdr = (image_header_t *)ptr;
93
Stephen Warren712fbcf2011-10-18 11:11:49 +000094 checksum = crc32(0,
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053095 (const unsigned char *)(ptr +
96 sizeof(image_header_t)),
97 sbuf->st_size - sizeof(image_header_t));
98
99 /* Build new header */
Stephen Warren712fbcf2011-10-18 11:11:49 +0000100 image_set_magic(hdr, IH_MAGIC);
101 image_set_time(hdr, sbuf->st_mtime);
102 image_set_size(hdr, sbuf->st_size - sizeof(image_header_t));
103 image_set_load(hdr, params->addr);
104 image_set_ep(hdr, params->ep);
105 image_set_dcrc(hdr, checksum);
106 image_set_os(hdr, params->os);
107 image_set_arch(hdr, params->arch);
108 image_set_type(hdr, params->type);
109 image_set_comp(hdr, params->comp);
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530110
Stephen Warren712fbcf2011-10-18 11:11:49 +0000111 image_set_name(hdr, params->imagename);
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530112
Stephen Warren712fbcf2011-10-18 11:11:49 +0000113 checksum = crc32(0, (const unsigned char *)hdr,
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530114 sizeof(image_header_t));
115
Stephen Warren712fbcf2011-10-18 11:11:49 +0000116 image_set_hcrc(hdr, checksum);
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530117}
118
Guilherme Maciel Ferreira67f946c2015-01-15 02:54:41 -0200119static int image_extract_subimage(void *ptr, struct image_tool_params *params)
Guilherme Maciel Ferreiraa804b5c2013-12-01 12:43:11 -0700120{
121 const image_header_t *hdr = (const image_header_t *)ptr;
122 ulong file_data;
123 ulong file_len;
124
125 if (image_check_type(hdr, IH_TYPE_MULTI)) {
126 ulong idx = params->pflag;
127 ulong count;
128
129 /* get the number of data files present in the image */
130 count = image_multi_count(hdr);
131
132 /* retrieve the "data file" at the idx position */
133 image_multi_getimg(hdr, idx, &file_data, &file_len);
134
135 if ((file_len == 0) || (idx >= count)) {
136 fprintf(stderr, "%s: No such data file %ld in \"%s\"\n",
137 params->cmdname, idx, params->imagefile);
138 return -1;
139 }
140 } else {
141 file_data = image_get_data(hdr);
142 file_len = image_get_size(hdr);
143 }
144
145 /* save the "data file" into the file system */
Guilherme Maciel Ferreira67f946c2015-01-15 02:54:41 -0200146 return imagetool_save_subimage(params->outfile, file_data, file_len);
Guilherme Maciel Ferreiraa804b5c2013-12-01 12:43:11 -0700147}
148
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530149/*
150 * Default image type parameters definition
151 */
Guilherme Maciel Ferreiraa93648d2015-01-15 02:48:07 -0200152U_BOOT_IMAGE_TYPE(
153 defimage,
154 "Default Image support",
155 sizeof(image_header_t),
156 (void *)&header,
157 image_check_params,
158 image_verify_header,
159 image_print_contents,
160 image_set_header,
Guilherme Maciel Ferreira67f946c2015-01-15 02:54:41 -0200161 image_extract_subimage,
Guilherme Maciel Ferreiraa93648d2015-01-15 02:48:07 -0200162 image_check_image_types,
163 NULL,
164 NULL
165);