Merge branch 'u-boot/master' into 'u-boot-arm/master'
[platform/kernel/u-boot.git] / tools / image-host.c
1 /*
2  * Copyright (c) 2013, Google Inc.
3  *
4  * (C) Copyright 2008 Semihalf
5  *
6  * (C) Copyright 2000-2006
7  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
8  *
9  * See file CREDITS for list of people who contributed to this
10  * project.
11  *
12  * This program is free software; you can redistribute it and/or
13  * modify it under the terms of the GNU General Public License as
14  * published by the Free Software Foundation; either version 2 of
15  * the License, or (at your option) any later version.
16  *
17  * This program is distributed in the hope that it will be useful,
18  * but WITHOUT ANY WARRANTY; without even the implied warranty of
19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20  * GNU General Public License for more details.
21  *
22  * You should have received a copy of the GNU General Public License
23  * along with this program; if not, write to the Free Software
24  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
25  * MA 02111-1307 USA
26  */
27
28 #include "mkimage.h"
29 #include <bootstage.h>
30 #include <image.h>
31 #include <sha1.h>
32 #include <time.h>
33 #include <u-boot/crc.h>
34 #include <u-boot/md5.h>
35
36 /**
37  * fit_set_hash_value - set hash value in requested has node
38  * @fit: pointer to the FIT format image header
39  * @noffset: hash node offset
40  * @value: hash value to be set
41  * @value_len: hash value length
42  *
43  * fit_set_hash_value() attempts to set hash value in a node at offset
44  * given and returns operation status to the caller.
45  *
46  * returns
47  *     0, on success
48  *     -1, on failure
49  */
50 static int fit_set_hash_value(void *fit, int noffset, uint8_t *value,
51                                 int value_len)
52 {
53         int ret;
54
55         ret = fdt_setprop(fit, noffset, FIT_VALUE_PROP, value, value_len);
56         if (ret) {
57                 printf("Can't set hash '%s' property for '%s' node(%s)\n",
58                        FIT_VALUE_PROP, fit_get_name(fit, noffset, NULL),
59                        fdt_strerror(ret));
60                 return -1;
61         }
62
63         return 0;
64 }
65
66 /**
67  * fit_image_process_hash - Process a single subnode of the images/ node
68  *
69  * Check each subnode and process accordingly. For hash nodes we generate
70  * a hash of the supplised data and store it in the node.
71  *
72  * @fit:        pointer to the FIT format image header
73  * @image_name: name of image being processes (used to display errors)
74  * @noffset:    subnode offset
75  * @data:       data to process
76  * @size:       size of data in bytes
77  * @return 0 if ok, -1 on error
78  */
79 static int fit_image_process_hash(void *fit, const char *image_name,
80                 int noffset, const void *data, size_t size)
81 {
82         uint8_t value[FIT_MAX_HASH_LEN];
83         const char *node_name;
84         int value_len;
85         char *algo;
86
87         node_name = fit_get_name(fit, noffset, NULL);
88
89         if (fit_image_hash_get_algo(fit, noffset, &algo)) {
90                 printf("Can't get hash algo property for '%s' hash node in '%s' image node\n",
91                        node_name, image_name);
92                 return -1;
93         }
94
95         if (calculate_hash(data, size, algo, value, &value_len)) {
96                 printf("Unsupported hash algorithm (%s) for '%s' hash node in '%s' image node\n",
97                        algo, node_name, image_name);
98                 return -1;
99         }
100
101         if (fit_set_hash_value(fit, noffset, value, value_len)) {
102                 printf("Can't set hash value for '%s' hash node in '%s' image node\n",
103                        node_name, image_name);
104                 return -1;
105         }
106
107         return 0;
108 }
109
110 /**
111  * fit_image_add_verification_data() - calculate/set hash data for image node
112  *
113  * This adds hash values for a component image node.
114  *
115  * All existing hash subnodes are checked, if algorithm property is set to
116  * one of the supported hash algorithms, hash value is computed and
117  * corresponding hash node property is set, for example:
118  *
119  * Input component image node structure:
120  *
121  * o image@1 (at image_noffset)
122  *   | - data = [binary data]
123  *   o hash@1
124  *     |- algo = "sha1"
125  *
126  * Output component image node structure:
127  *
128  * o image@1 (at image_noffset)
129  *   | - data = [binary data]
130  *   o hash@1
131  *     |- algo = "sha1"
132  *     |- value = sha1(data)
133  *
134  * For signature details, please see doc/uImage.FIT/signature.txt
135  *
136  * @fit:        Pointer to the FIT format image header
137  * @image_noffset: Requested component image node
138  * @return: 0 on success, <0 on failure
139  */
140 int fit_image_add_verification_data(void *fit, int image_noffset)
141 {
142         const char *image_name;
143         const void *data;
144         size_t size;
145         int noffset;
146
147         /* Get image data and data length */
148         if (fit_image_get_data(fit, image_noffset, &data, &size)) {
149                 printf("Can't get image data/size\n");
150                 return -1;
151         }
152
153         image_name = fit_get_name(fit, image_noffset, NULL);
154
155         /* Process all hash subnodes of the component image node */
156         for (noffset = fdt_first_subnode(fit, image_noffset);
157              noffset >= 0;
158              noffset = fdt_next_subnode(fit, noffset)) {
159                 const char *node_name;
160                 int ret = 0;
161
162                 /*
163                  * Check subnode name, must be equal to "hash" or "signature".
164                  * Multiple hash nodes require unique unit node
165                  * names, e.g. hash@1, hash@2, signature@1, etc.
166                  */
167                 node_name = fit_get_name(fit, noffset, NULL);
168                 if (!strncmp(node_name, FIT_HASH_NODENAME,
169                              strlen(FIT_HASH_NODENAME))) {
170                         ret = fit_image_process_hash(fit, image_name, noffset,
171                                                 data, size);
172                 }
173                 if (ret)
174                         return -1;
175         }
176
177         return 0;
178 }
179
180 int fit_add_verification_data(void *fit)
181 {
182         int images_noffset;
183         int noffset;
184         int ret;
185
186         /* Find images parent node offset */
187         images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
188         if (images_noffset < 0) {
189                 printf("Can't find images parent node '%s' (%s)\n",
190                        FIT_IMAGES_PATH, fdt_strerror(images_noffset));
191                 return images_noffset;
192         }
193
194         /* Process its subnodes, print out component images details */
195         for (noffset = fdt_first_subnode(fit, images_noffset);
196              noffset >= 0;
197              noffset = fdt_next_subnode(fit, noffset)) {
198                 /*
199                  * Direct child node of the images parent node,
200                  * i.e. component image node.
201                  */
202                 ret = fit_image_add_verification_data(fit, noffset);
203                 if (ret)
204                         return ret;
205         }
206
207         return 0;
208 }