tizen 2.4 release
[profile/mobile/platform/kernel/u-boot-tm1.git] / 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 -p 200 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                 close (tfd);
109                 return (EXIT_FAILURE);
110         }
111
112         ptr = mmap (0, sbuf.st_size, PROT_READ|PROT_WRITE, MAP_SHARED,
113                                 tfd, 0);
114         if (ptr == MAP_FAILED) {
115                 fprintf (stderr, "%s: Can't read %s: %s\n",
116                                 params->cmdname, tmpfile, strerror(errno));
117                 unlink (tmpfile);
118                 close (tfd);
119                 return (EXIT_FAILURE);
120         }
121
122         /* check if ptr has a valid blob */
123         if (fdt_check_header (ptr)) {
124                 fprintf (stderr, "%s: Invalid FIT blob\n", params->cmdname);
125                 unlink (tmpfile);
126                 close (tfd);
127                 return (EXIT_FAILURE);
128         }
129
130         /* set hashes for images in the blob */
131         if (fit_set_hashes (ptr)) {
132                 fprintf (stderr, "%s Can't add hashes to FIT blob",
133                                 params->cmdname);
134                 unlink (tmpfile);
135                 close (tfd);
136                 return (EXIT_FAILURE);
137         }
138
139         /* add a timestamp at offset 0 i.e., root  */
140         if (fit_set_timestamp (ptr, 0, sbuf.st_mtime)) {
141                 fprintf (stderr, "%s: Can't add image timestamp\n",
142                                 params->cmdname);
143                 unlink (tmpfile);
144               close (tfd);
145                 return (EXIT_FAILURE);
146         }
147         debug ("Added timestamp successfully\n");
148
149         munmap ((void *)ptr, sbuf.st_size);
150         close (tfd);
151
152         if (rename (tmpfile, params->imagefile) == -1) {
153                 fprintf (stderr, "%s: Can't rename %s to %s: %s\n",
154                                 params->cmdname, tmpfile, params->imagefile,
155                                 strerror (errno));
156                 unlink (tmpfile);
157                 unlink (params->imagefile);
158                 return (EXIT_FAILURE);
159         }
160         return (EXIT_SUCCESS);
161 }
162
163 static int fit_check_params (struct mkimage_params *params)
164 {
165         return  ((params->dflag && (params->fflag || params->lflag)) ||
166                 (params->fflag && (params->dflag || params->lflag)) ||
167                 (params->lflag && (params->dflag || params->fflag)));
168 }
169
170 static struct image_type_params fitimage_params = {
171         .name = "FIT Image support",
172         .header_size = sizeof(image_header_t),
173         .hdr = (void*)&header,
174         .verify_header = fit_verify_header,
175         .print_header = fit_print_contents,
176         .check_image_type = fit_check_image_types,
177         .fflag_handle = fit_handle_file,
178         .set_header = NULL,     /* FIT images use DTB header */
179         .check_params = fit_check_params,
180 };
181
182 void init_fit_image_type (void)
183 {
184         mkimage_register (&fitimage_params);
185 }