blob: 6e4ae14ec783d4e12693887aa166cdf514fc4d0d [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;
Paul Kocialkowskif3f431a2015-07-26 18:48:15 +020091 time_t time;
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053092
93 image_header_t * hdr = (image_header_t *)ptr;
94
Stephen Warren712fbcf2011-10-18 11:11:49 +000095 checksum = crc32(0,
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053096 (const unsigned char *)(ptr +
97 sizeof(image_header_t)),
98 sbuf->st_size - sizeof(image_header_t));
99
Vagrant Cascadian58470842016-06-16 12:28:40 -0700100 time = imagetool_get_source_date(params, sbuf->st_mtime);
Paul Kocialkowskif3f431a2015-07-26 18:48:15 +0200101
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530102 /* Build new header */
Stephen Warren712fbcf2011-10-18 11:11:49 +0000103 image_set_magic(hdr, IH_MAGIC);
Paul Kocialkowskif3f431a2015-07-26 18:48:15 +0200104 image_set_time(hdr, time);
Stephen Warren712fbcf2011-10-18 11:11:49 +0000105 image_set_size(hdr, sbuf->st_size - sizeof(image_header_t));
106 image_set_load(hdr, params->addr);
107 image_set_ep(hdr, params->ep);
108 image_set_dcrc(hdr, checksum);
109 image_set_os(hdr, params->os);
110 image_set_arch(hdr, params->arch);
111 image_set_type(hdr, params->type);
112 image_set_comp(hdr, params->comp);
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530113
Stephen Warren712fbcf2011-10-18 11:11:49 +0000114 image_set_name(hdr, params->imagename);
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530115
Stephen Warren712fbcf2011-10-18 11:11:49 +0000116 checksum = crc32(0, (const unsigned char *)hdr,
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530117 sizeof(image_header_t));
118
Stephen Warren712fbcf2011-10-18 11:11:49 +0000119 image_set_hcrc(hdr, checksum);
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530120}
121
Guilherme Maciel Ferreira67f946c2015-01-15 02:54:41 -0200122static int image_extract_subimage(void *ptr, struct image_tool_params *params)
Guilherme Maciel Ferreiraa804b5c2013-12-01 12:43:11 -0700123{
124 const image_header_t *hdr = (const image_header_t *)ptr;
125 ulong file_data;
126 ulong file_len;
127
128 if (image_check_type(hdr, IH_TYPE_MULTI)) {
129 ulong idx = params->pflag;
130 ulong count;
131
132 /* get the number of data files present in the image */
133 count = image_multi_count(hdr);
134
135 /* retrieve the "data file" at the idx position */
136 image_multi_getimg(hdr, idx, &file_data, &file_len);
137
138 if ((file_len == 0) || (idx >= count)) {
139 fprintf(stderr, "%s: No such data file %ld in \"%s\"\n",
140 params->cmdname, idx, params->imagefile);
141 return -1;
142 }
143 } else {
144 file_data = image_get_data(hdr);
145 file_len = image_get_size(hdr);
146 }
147
148 /* save the "data file" into the file system */
Guilherme Maciel Ferreira67f946c2015-01-15 02:54:41 -0200149 return imagetool_save_subimage(params->outfile, file_data, file_len);
Guilherme Maciel Ferreiraa804b5c2013-12-01 12:43:11 -0700150}
151
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530152/*
153 * Default image type parameters definition
154 */
Guilherme Maciel Ferreiraa93648d2015-01-15 02:48:07 -0200155U_BOOT_IMAGE_TYPE(
156 defimage,
157 "Default Image support",
158 sizeof(image_header_t),
159 (void *)&header,
160 image_check_params,
161 image_verify_header,
162 image_print_contents,
163 image_set_header,
Guilherme Maciel Ferreira67f946c2015-01-15 02:54:41 -0200164 image_extract_subimage,
Guilherme Maciel Ferreiraa93648d2015-01-15 02:48:07 -0200165 image_check_image_types,
166 NULL,
167 NULL
168);