1 // SPDX-License-Identifier: GPL-2.0+
3 * (C) Copyright 2008 Semihalf
5 * (C) Copyright 2000-2004
6 * DENX Software Engineering
7 * Wolfgang Denk, wd@denx.de
9 * Updated-by: Prafulla Wadaskar <prafulla@marvell.com>
10 * default_image specific code abstracted from mkimage.c
11 * some functions added to address abstraction
13 * All rights reserved.
16 #include "imagetool.h"
20 #include <tee/optee.h>
21 #include <u-boot/crc.h>
23 static image_header_t header
;
25 static int image_check_image_types(uint8_t type
)
27 if (((type
> IH_TYPE_INVALID
) && (type
< IH_TYPE_FLATDT
)) ||
28 (type
== IH_TYPE_KERNEL_NOLOAD
) || (type
== IH_TYPE_FIRMWARE_IVT
))
34 static int image_check_params(struct image_tool_params
*params
)
36 return ((params
->dflag
&& (params
->fflag
|| params
->lflag
)) ||
37 (params
->fflag
&& (params
->dflag
|| params
->lflag
)) ||
38 (params
->lflag
&& (params
->dflag
|| params
->fflag
)));
41 static int image_verify_header(unsigned char *ptr
, int image_size
,
42 struct image_tool_params
*params
)
45 const unsigned char *data
;
47 image_header_t header
;
48 image_header_t
*hdr
= &header
;
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.
55 memcpy(hdr
, ptr
, sizeof(image_header_t
));
57 if (be32_to_cpu(hdr
->ih_magic
) != IH_MAGIC
) {
58 debug("%s: Bad Magic Number: \"%s\" is no valid image\n",
59 params
->cmdname
, params
->imagefile
);
60 return -FDT_ERR_BADMAGIC
;
63 data
= (const unsigned char *)hdr
;
64 len
= sizeof(image_header_t
);
66 checksum
= be32_to_cpu(hdr
->ih_hcrc
);
67 hdr
->ih_hcrc
= cpu_to_be32(0); /* clear for re-calculation */
69 if (crc32(0, data
, len
) != checksum
) {
70 debug("%s: ERROR: \"%s\" has bad header checksum!\n",
71 params
->cmdname
, params
->imagefile
);
72 return -FDT_ERR_BADSTATE
;
75 data
= (const unsigned char *)ptr
+ sizeof(image_header_t
);
76 len
= image_size
- sizeof(image_header_t
) ;
78 checksum
= be32_to_cpu(hdr
->ih_dcrc
);
79 if (crc32(0, data
, len
) != checksum
) {
80 debug("%s: ERROR: \"%s\" has corrupted data!\n",
81 params
->cmdname
, params
->imagefile
);
82 return -FDT_ERR_BADSTRUCTURE
;
87 static void image_set_header(void *ptr
, struct stat
*sbuf
, int ifd
,
88 struct image_tool_params
*params
)
96 image_header_t
* hdr
= (image_header_t
*)ptr
;
99 (const unsigned char *)(ptr
+
100 sizeof(image_header_t
)),
101 sbuf
->st_size
- sizeof(image_header_t
));
103 time
= imagetool_get_source_date(params
->cmdname
, sbuf
->st_mtime
);
107 if (params
->type
== IH_TYPE_FIRMWARE_IVT
)
108 /* Add size of CSF minus IVT */
109 imagesize
= sbuf
->st_size
- sizeof(image_header_t
) + 0x1FE0;
111 imagesize
= sbuf
->st_size
- sizeof(image_header_t
);
113 if (params
->os
== IH_OS_TEE
) {
114 addr
= optee_image_get_load_addr(hdr
);
115 ep
= optee_image_get_entry_point(hdr
);
118 /* Build new header */
119 image_set_magic(hdr
, IH_MAGIC
);
120 image_set_time(hdr
, time
);
121 image_set_size(hdr
, imagesize
);
122 image_set_load(hdr
, addr
);
123 image_set_ep(hdr
, ep
);
124 image_set_dcrc(hdr
, checksum
);
125 image_set_os(hdr
, params
->os
);
126 image_set_arch(hdr
, params
->arch
);
127 image_set_type(hdr
, params
->type
);
128 image_set_comp(hdr
, params
->comp
);
130 image_set_name(hdr
, params
->imagename
);
132 checksum
= crc32(0, (const unsigned char *)hdr
,
133 sizeof(image_header_t
));
135 image_set_hcrc(hdr
, checksum
);
138 static int image_extract_subimage(void *ptr
, struct image_tool_params
*params
)
140 const image_header_t
*hdr
= (const image_header_t
*)ptr
;
144 if (image_check_type(hdr
, IH_TYPE_MULTI
)) {
145 ulong idx
= params
->pflag
;
148 /* get the number of data files present in the image */
149 count
= image_multi_count(hdr
);
151 /* retrieve the "data file" at the idx position */
152 image_multi_getimg(hdr
, idx
, &file_data
, &file_len
);
154 if ((file_len
== 0) || (idx
>= count
)) {
155 fprintf(stderr
, "%s: No such data file %ld in \"%s\"\n",
156 params
->cmdname
, idx
, params
->imagefile
);
160 file_data
= image_get_data(hdr
);
161 file_len
= image_get_size(hdr
);
164 /* save the "data file" into the file system */
165 return imagetool_save_subimage(params
->outfile
, file_data
, file_len
);
169 * Default image type parameters definition
173 "Default Image support",
174 sizeof(image_header_t
),
178 image_print_contents
,
180 image_extract_subimage
,
181 image_check_image_types
,