]> git.sur5r.net Git - u-boot/blob - tools/fit_image.c
palmas: rename init_settings to an generic palmas init
[u-boot] / tools / fit_image.c
1 /*
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  *              FIT image specific code abstracted from mkimage.c
10  *              some functions added to address abstraction
11  *
12  * All rights reserved.
13  *
14  * This program is free software; you can redistribute it and/or
15  * modify it under the terms of the GNU General Public License as
16  * published by the Free Software Foundation; either version 2 of
17  * the License, or (at your option) any later version.
18  *
19  * This program is distributed in the hope that it will be useful,
20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
22  * GNU General Public License for more details.
23  *
24  * You should have received a copy of the GNU General Public License
25  * along with this program; if not, write to the Free Software
26  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
27  * MA 02111-1307 USA
28  */
29
30 #include "mkimage.h"
31 #include <image.h>
32 #include <u-boot/crc.h>
33
34 static image_header_t header;
35
36 static int fit_verify_header (unsigned char *ptr, int image_size,
37                         struct mkimage_params *params)
38 {
39         return fdt_check_header ((void *)ptr);
40 }
41
42 static int fit_check_image_types (uint8_t type)
43 {
44         if (type == IH_TYPE_FLATDT)
45                 return EXIT_SUCCESS;
46         else
47                 return EXIT_FAILURE;
48 }
49
50 /**
51  * fit_handle_file - main FIT file processing function
52  *
53  * fit_handle_file() runs dtc to convert .its to .itb, includes
54  * binary data, updates timestamp property and calculates hashes.
55  *
56  * datafile  - .its file
57  * imagefile - .itb file
58  *
59  * returns:
60  *     only on success, otherwise calls exit (EXIT_FAILURE);
61  */
62 static int fit_handle_file (struct mkimage_params *params)
63 {
64         char tmpfile[MKIMAGE_MAX_TMPFILE_LEN];
65         char cmd[MKIMAGE_MAX_DTC_CMDLINE_LEN];
66         int tfd;
67         struct stat sbuf;
68         unsigned char *ptr;
69
70         /* Flattened Image Tree (FIT) format  handling */
71         debug ("FIT format handling\n");
72
73         /* call dtc to include binary properties into the tmp file */
74         if (strlen (params->imagefile) +
75                 strlen (MKIMAGE_TMPFILE_SUFFIX) + 1 > sizeof (tmpfile)) {
76                 fprintf (stderr, "%s: Image file name (%s) too long, "
77                                 "can't create tmpfile",
78                                 params->imagefile, params->cmdname);
79                 return (EXIT_FAILURE);
80         }
81         sprintf (tmpfile, "%s%s", params->imagefile, MKIMAGE_TMPFILE_SUFFIX);
82
83         /* dtc -I dts -O dtb -p 500 datafile > tmpfile */
84         sprintf (cmd, "%s %s %s > %s",
85                 MKIMAGE_DTC, params->dtc, params->datafile, tmpfile);
86         debug ("Trying to execute \"%s\"\n", cmd);
87         if (system (cmd) == -1) {
88                 fprintf (stderr, "%s: system(%s) failed: %s\n",
89                                 params->cmdname, cmd, strerror(errno));
90                 unlink (tmpfile);
91                 return (EXIT_FAILURE);
92         }
93
94         /* load FIT blob into memory */
95         tfd = open (tmpfile, O_RDWR|O_BINARY);
96
97         if (tfd < 0) {
98                 fprintf (stderr, "%s: Can't open %s: %s\n",
99                                 params->cmdname, tmpfile, strerror(errno));
100                 unlink (tmpfile);
101                 return (EXIT_FAILURE);
102         }
103
104         if (fstat (tfd, &sbuf) < 0) {
105                 fprintf (stderr, "%s: Can't stat %s: %s\n",
106                                 params->cmdname, tmpfile, strerror(errno));
107                 unlink (tmpfile);
108                 return (EXIT_FAILURE);
109         }
110
111         ptr = mmap (0, sbuf.st_size, PROT_READ|PROT_WRITE, MAP_SHARED,
112                                 tfd, 0);
113         if (ptr == MAP_FAILED) {
114                 fprintf (stderr, "%s: Can't read %s: %s\n",
115                                 params->cmdname, tmpfile, strerror(errno));
116                 unlink (tmpfile);
117                 return (EXIT_FAILURE);
118         }
119
120         /* check if ptr has a valid blob */
121         if (fdt_check_header (ptr)) {
122                 fprintf (stderr, "%s: Invalid FIT blob\n", params->cmdname);
123                 unlink (tmpfile);
124                 return (EXIT_FAILURE);
125         }
126
127         /* set hashes for images in the blob */
128         if (fit_set_hashes (ptr)) {
129                 fprintf (stderr, "%s Can't add hashes to FIT blob",
130                                 params->cmdname);
131                 unlink (tmpfile);
132                 return (EXIT_FAILURE);
133         }
134
135         /* add a timestamp at offset 0 i.e., root  */
136         if (fit_set_timestamp (ptr, 0, sbuf.st_mtime)) {
137                 fprintf (stderr, "%s: Can't add image timestamp\n",
138                                 params->cmdname);
139                 unlink (tmpfile);
140                 return (EXIT_FAILURE);
141         }
142         debug ("Added timestamp successfully\n");
143
144         munmap ((void *)ptr, sbuf.st_size);
145         close (tfd);
146
147         if (rename (tmpfile, params->imagefile) == -1) {
148                 fprintf (stderr, "%s: Can't rename %s to %s: %s\n",
149                                 params->cmdname, tmpfile, params->imagefile,
150                                 strerror (errno));
151                 unlink (tmpfile);
152                 unlink (params->imagefile);
153                 return (EXIT_FAILURE);
154         }
155         return (EXIT_SUCCESS);
156 }
157
158 static int fit_check_params (struct mkimage_params *params)
159 {
160         return  ((params->dflag && (params->fflag || params->lflag)) ||
161                 (params->fflag && (params->dflag || params->lflag)) ||
162                 (params->lflag && (params->dflag || params->fflag)));
163 }
164
165 static struct image_type_params fitimage_params = {
166         .name = "FIT Image support",
167         .header_size = sizeof(image_header_t),
168         .hdr = (void*)&header,
169         .verify_header = fit_verify_header,
170         .print_header = fit_print_contents,
171         .check_image_type = fit_check_image_types,
172         .fflag_handle = fit_handle_file,
173         .set_header = NULL,     /* FIT images use DTB header */
174         .check_params = fit_check_params,
175 };
176
177 void init_fit_image_type (void)
178 {
179         mkimage_register (&fitimage_params);
180 }