2 * (C) Copyright 2008 Semihalf
4 * (C) Copyright 2000-2004
5 * DENX Software Engineering
6 * Wolfgang Denk, wd@denx.de
8 * Updated-by: Prafulla Wadaskar <prafulla@marvell.com>
9 * default_image specific code abstracted from mkimage.c
10 * some functions added to address abstraction
12 * All rights reserved.
14 * SPDX-License-Identifier: GPL-2.0+
17 #include "imagetool.h"
19 #include <u-boot/crc.h>
21 static image_header_t header
;
23 static int image_check_image_types(uint8_t type
)
25 if (((type
> IH_TYPE_INVALID
) && (type
< IH_TYPE_FLATDT
)) ||
26 (type
== IH_TYPE_KERNEL_NOLOAD
))
32 static int image_check_params(struct image_tool_params
*params
)
34 return ((params
->dflag
&& (params
->fflag
|| params
->lflag
)) ||
35 (params
->fflag
&& (params
->dflag
|| params
->lflag
)) ||
36 (params
->lflag
&& (params
->dflag
|| params
->fflag
)));
39 static int image_verify_header(unsigned char *ptr
, int image_size
,
40 struct image_tool_params
*params
)
43 const unsigned char *data
;
45 image_header_t header
;
46 image_header_t
*hdr
= &header
;
49 * create copy of header so that we can blank out the
50 * checksum field for checking - this can't be done
51 * on the PROT_READ mapped data.
53 memcpy(hdr
, ptr
, sizeof(image_header_t
));
55 if (be32_to_cpu(hdr
->ih_magic
) != IH_MAGIC
) {
57 "%s: Bad Magic Number: \"%s\" is no valid image\n",
58 params
->cmdname
, params
->imagefile
);
59 return -FDT_ERR_BADMAGIC
;
62 data
= (const unsigned char *)hdr
;
63 len
= sizeof(image_header_t
);
65 checksum
= be32_to_cpu(hdr
->ih_hcrc
);
66 hdr
->ih_hcrc
= cpu_to_be32(0); /* clear for re-calculation */
68 if (crc32(0, data
, len
) != checksum
) {
70 "%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
) {
81 "%s: ERROR: \"%s\" has corrupted data!\n",
82 params
->cmdname
, params
->imagefile
);
83 return -FDT_ERR_BADSTRUCTURE
;
88 static void image_set_header(void *ptr
, struct stat
*sbuf
, int ifd
,
89 struct image_tool_params
*params
)
93 image_header_t
* hdr
= (image_header_t
*)ptr
;
96 (const unsigned char *)(ptr
+
97 sizeof(image_header_t
)),
98 sbuf
->st_size
- sizeof(image_header_t
));
100 /* Build new header */
101 image_set_magic(hdr
, IH_MAGIC
);
102 image_set_time(hdr
, sbuf
->st_mtime
);
103 image_set_size(hdr
, sbuf
->st_size
- sizeof(image_header_t
));
104 image_set_load(hdr
, params
->addr
);
105 image_set_ep(hdr
, params
->ep
);
106 image_set_dcrc(hdr
, checksum
);
107 image_set_os(hdr
, params
->os
);
108 image_set_arch(hdr
, params
->arch
);
109 image_set_type(hdr
, params
->type
);
110 image_set_comp(hdr
, params
->comp
);
112 image_set_name(hdr
, params
->imagename
);
114 checksum
= crc32(0, (const unsigned char *)hdr
,
115 sizeof(image_header_t
));
117 image_set_hcrc(hdr
, checksum
);
120 static int image_extract_datafile(void *ptr
, struct image_tool_params
*params
)
122 const image_header_t
*hdr
= (const image_header_t
*)ptr
;
126 if (image_check_type(hdr
, IH_TYPE_MULTI
)) {
127 ulong idx
= params
->pflag
;
130 /* get the number of data files present in the image */
131 count
= image_multi_count(hdr
);
133 /* retrieve the "data file" at the idx position */
134 image_multi_getimg(hdr
, idx
, &file_data
, &file_len
);
136 if ((file_len
== 0) || (idx
>= count
)) {
137 fprintf(stderr
, "%s: No such data file %ld in \"%s\"\n",
138 params
->cmdname
, idx
, params
->imagefile
);
142 file_data
= image_get_data(hdr
);
143 file_len
= image_get_size(hdr
);
146 /* save the "data file" into the file system */
147 return imagetool_save_datafile(params
->outfile
, file_data
, file_len
);
151 * Default image type parameters definition
153 static struct image_type_params defimage_params
= {
154 .name
= "Default Image support",
155 .header_size
= sizeof(image_header_t
),
156 .hdr
= (void*)&header
,
157 .check_image_type
= image_check_image_types
,
158 .verify_header
= image_verify_header
,
159 .print_header
= image_print_contents
,
160 .set_header
= image_set_header
,
161 .extract_datafile
= image_extract_datafile
,
162 .check_params
= image_check_params
,
165 void init_default_image_type(void)
167 register_image_type(&defimage_params
);