Merge tag 'ti-v2020.07-next' of https://gitlab.denx.de/u-boot/custodians/u-boot-ti...
[oweals/u-boot.git] / common / image-fit-sig.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (c) 2013, Google Inc.
4  */
5
6 #ifdef USE_HOSTCC
7 #include "mkimage.h"
8 #include <time.h>
9 #else
10 #include <common.h>
11 #include <malloc.h>
12 DECLARE_GLOBAL_DATA_PTR;
13 #endif /* !USE_HOSTCC*/
14 #include <image.h>
15 #include <u-boot/rsa.h>
16 #include <u-boot/rsa-checksum.h>
17
18 #define IMAGE_MAX_HASHED_NODES          100
19
20 #ifdef USE_HOSTCC
21 void *host_blob;
22
23 void image_set_host_blob(void *blob)
24 {
25         host_blob = blob;
26 }
27
28 void *image_get_host_blob(void)
29 {
30         return host_blob;
31 }
32 #endif
33
34 /**
35  * fit_region_make_list() - Make a list of image regions
36  *
37  * Given a list of fdt_regions, create a list of image_regions. This is a
38  * simple conversion routine since the FDT and image code use different
39  * structures.
40  *
41  * @fit: FIT image
42  * @fdt_regions: Pointer to FDT regions
43  * @count: Number of FDT regions
44  * @region: Pointer to image regions, which must hold @count records. If
45  * region is NULL, then (except for an SPL build) the array will be
46  * allocated.
47  * @return: Pointer to image regions
48  */
49 struct image_region *fit_region_make_list(const void *fit,
50                                           struct fdt_region *fdt_regions,
51                                           int count,
52                                           struct image_region *region)
53 {
54         int i;
55
56         debug("Hash regions:\n");
57         debug("%10s %10s\n", "Offset", "Size");
58
59         /*
60          * Use malloc() except in SPL (to save code size). In SPL the caller
61          * must allocate the array.
62          */
63 #ifndef CONFIG_SPL_BUILD
64         if (!region)
65                 region = calloc(sizeof(*region), count);
66 #endif
67         if (!region)
68                 return NULL;
69         for (i = 0; i < count; i++) {
70                 debug("%10x %10x\n", fdt_regions[i].offset,
71                       fdt_regions[i].size);
72                 region[i].data = fit + fdt_regions[i].offset;
73                 region[i].size = fdt_regions[i].size;
74         }
75
76         return region;
77 }
78
79 static int fit_image_setup_verify(struct image_sign_info *info,
80                                   const void *fit, int noffset,
81                                   int required_keynode, char **err_msgp)
82 {
83         char *algo_name;
84         const char *padding_name;
85
86         if (fdt_totalsize(fit) > CONFIG_FIT_SIGNATURE_MAX_SIZE) {
87                 *err_msgp = "Total size too large";
88                 return 1;
89         }
90
91         if (fit_image_hash_get_algo(fit, noffset, &algo_name)) {
92                 *err_msgp = "Can't get hash algo property";
93                 return -1;
94         }
95
96         padding_name = fdt_getprop(fit, noffset, "padding", NULL);
97         if (!padding_name)
98                 padding_name = RSA_DEFAULT_PADDING_NAME;
99
100         memset(info, '\0', sizeof(*info));
101         info->keyname = fdt_getprop(fit, noffset, "key-name-hint", NULL);
102         info->fit = (void *)fit;
103         info->node_offset = noffset;
104         info->name = algo_name;
105         info->checksum = image_get_checksum_algo(algo_name);
106         info->crypto = image_get_crypto_algo(algo_name);
107         info->padding = image_get_padding_algo(padding_name);
108         info->fdt_blob = gd_fdt_blob();
109         info->required_keynode = required_keynode;
110         printf("%s:%s", algo_name, info->keyname);
111
112         if (!info->checksum || !info->crypto || !info->padding) {
113                 *err_msgp = "Unknown signature algorithm";
114                 return -1;
115         }
116
117         return 0;
118 }
119
120 int fit_image_check_sig(const void *fit, int noffset, const void *data,
121                         size_t size, int required_keynode, char **err_msgp)
122 {
123         struct image_sign_info info;
124         struct image_region region;
125         uint8_t *fit_value;
126         int fit_value_len;
127
128         *err_msgp = NULL;
129         if (fit_image_setup_verify(&info, fit, noffset, required_keynode,
130                                    err_msgp))
131                 return -1;
132
133         if (fit_image_hash_get_value(fit, noffset, &fit_value,
134                                      &fit_value_len)) {
135                 *err_msgp = "Can't get hash value property";
136                 return -1;
137         }
138
139         region.data = data;
140         region.size = size;
141
142         if (info.crypto->verify(&info, &region, 1, fit_value, fit_value_len)) {
143                 *err_msgp = "Verification failed";
144                 return -1;
145         }
146
147         return 0;
148 }
149
150 static int fit_image_verify_sig(const void *fit, int image_noffset,
151                                 const char *data, size_t size,
152                                 const void *sig_blob, int sig_offset)
153 {
154         int noffset;
155         char *err_msg = "";
156         int verified = 0;
157         int ret;
158
159         /* Process all hash subnodes of the component image node */
160         fdt_for_each_subnode(noffset, fit, image_noffset) {
161                 const char *name = fit_get_name(fit, noffset, NULL);
162
163                 if (!strncmp(name, FIT_SIG_NODENAME,
164                              strlen(FIT_SIG_NODENAME))) {
165                         ret = fit_image_check_sig(fit, noffset, data,
166                                                   size, -1, &err_msg);
167                         if (ret) {
168                                 puts("- ");
169                         } else {
170                                 puts("+ ");
171                                 verified = 1;
172                                 break;
173                         }
174                 }
175         }
176
177         if (noffset == -FDT_ERR_TRUNCATED || noffset == -FDT_ERR_BADSTRUCTURE) {
178                 err_msg = "Corrupted or truncated tree";
179                 goto error;
180         }
181
182         return verified ? 0 : -EPERM;
183
184 error:
185         printf(" error!\n%s for '%s' hash node in '%s' image node\n",
186                err_msg, fit_get_name(fit, noffset, NULL),
187                fit_get_name(fit, image_noffset, NULL));
188         return -1;
189 }
190
191 int fit_image_verify_required_sigs(const void *fit, int image_noffset,
192                                    const char *data, size_t size,
193                                    const void *sig_blob, int *no_sigsp)
194 {
195         int verify_count = 0;
196         int noffset;
197         int sig_node;
198
199         /* Work out what we need to verify */
200         *no_sigsp = 1;
201         sig_node = fdt_subnode_offset(sig_blob, 0, FIT_SIG_NODENAME);
202         if (sig_node < 0) {
203                 debug("%s: No signature node found: %s\n", __func__,
204                       fdt_strerror(sig_node));
205                 return 0;
206         }
207
208         fdt_for_each_subnode(noffset, sig_blob, sig_node) {
209                 const char *required;
210                 int ret;
211
212                 required = fdt_getprop(sig_blob, noffset, "required", NULL);
213                 if (!required || strcmp(required, "image"))
214                         continue;
215                 ret = fit_image_verify_sig(fit, image_noffset, data, size,
216                                            sig_blob, noffset);
217                 if (ret) {
218                         printf("Failed to verify required signature '%s'\n",
219                                fit_get_name(sig_blob, noffset, NULL));
220                         return ret;
221                 }
222                 verify_count++;
223         }
224
225         if (verify_count)
226                 *no_sigsp = 0;
227
228         return 0;
229 }
230
231 int fit_config_check_sig(const void *fit, int noffset, int required_keynode,
232                          char **err_msgp)
233 {
234         char * const exc_prop[] = {"data"};
235         const char *prop, *end, *name;
236         struct image_sign_info info;
237         const uint32_t *strings;
238         uint8_t *fit_value;
239         int fit_value_len;
240         int max_regions;
241         int i, prop_len;
242         char path[200];
243         int count;
244
245         debug("%s: fdt=%p, conf='%s', sig='%s'\n", __func__, gd_fdt_blob(),
246               fit_get_name(fit, noffset, NULL),
247               fit_get_name(gd_fdt_blob(), required_keynode, NULL));
248         *err_msgp = NULL;
249         if (fit_image_setup_verify(&info, fit, noffset, required_keynode,
250                                    err_msgp))
251                 return -1;
252
253         if (fit_image_hash_get_value(fit, noffset, &fit_value,
254                                      &fit_value_len)) {
255                 *err_msgp = "Can't get hash value property";
256                 return -1;
257         }
258
259         /* Count the number of strings in the property */
260         prop = fdt_getprop(fit, noffset, "hashed-nodes", &prop_len);
261         end = prop ? prop + prop_len : prop;
262         for (name = prop, count = 0; name < end; name++)
263                 if (!*name)
264                         count++;
265         if (!count) {
266                 *err_msgp = "Can't get hashed-nodes property";
267                 return -1;
268         }
269
270         if (prop && prop_len > 0 && prop[prop_len - 1] != '\0') {
271                 *err_msgp = "hashed-nodes property must be null-terminated";
272                 return -1;
273         }
274
275         /* Add a sanity check here since we are using the stack */
276         if (count > IMAGE_MAX_HASHED_NODES) {
277                 *err_msgp = "Number of hashed nodes exceeds maximum";
278                 return -1;
279         }
280
281         /* Create a list of node names from those strings */
282         char *node_inc[count];
283
284         debug("Hash nodes (%d):\n", count);
285         for (name = prop, i = 0; name < end; name += strlen(name) + 1, i++) {
286                 debug("   '%s'\n", name);
287                 node_inc[i] = (char *)name;
288         }
289
290         /*
291          * Each node can generate one region for each sub-node. Allow for
292          * 7 sub-nodes (hash-1, signature-1, etc.) and some extra.
293          */
294         max_regions = 20 + count * 7;
295         struct fdt_region fdt_regions[max_regions];
296
297         /* Get a list of regions to hash */
298         count = fdt_find_regions(fit, node_inc, count,
299                                  exc_prop, ARRAY_SIZE(exc_prop),
300                                  fdt_regions, max_regions - 1,
301                                  path, sizeof(path), 0);
302         if (count < 0) {
303                 *err_msgp = "Failed to hash configuration";
304                 return -1;
305         }
306         if (count == 0) {
307                 *err_msgp = "No data to hash";
308                 return -1;
309         }
310         if (count >= max_regions - 1) {
311                 *err_msgp = "Too many hash regions";
312                 return -1;
313         }
314
315         /* Add the strings */
316         strings = fdt_getprop(fit, noffset, "hashed-strings", NULL);
317         if (strings) {
318                 /*
319                  * The strings region offset must be a static 0x0.
320                  * This is set in tool/image-host.c
321                  */
322                 fdt_regions[count].offset = fdt_off_dt_strings(fit);
323                 fdt_regions[count].size = fdt32_to_cpu(strings[1]);
324                 count++;
325         }
326
327         /* Allocate the region list on the stack */
328         struct image_region region[count];
329
330         fit_region_make_list(fit, fdt_regions, count, region);
331         if (info.crypto->verify(&info, region, count, fit_value,
332                                 fit_value_len)) {
333                 *err_msgp = "Verification failed";
334                 return -1;
335         }
336
337         return 0;
338 }
339
340 static int fit_config_verify_sig(const void *fit, int conf_noffset,
341                                  const void *sig_blob, int sig_offset)
342 {
343         int noffset;
344         char *err_msg = "";
345         int verified = 0;
346         int ret;
347
348         /* Process all hash subnodes of the component conf node */
349         fdt_for_each_subnode(noffset, fit, conf_noffset) {
350                 const char *name = fit_get_name(fit, noffset, NULL);
351
352                 if (!strncmp(name, FIT_SIG_NODENAME,
353                              strlen(FIT_SIG_NODENAME))) {
354                         ret = fit_config_check_sig(fit, noffset, sig_offset,
355                                                    &err_msg);
356                         if (ret) {
357                                 puts("- ");
358                         } else {
359                                 puts("+ ");
360                                 verified = 1;
361                                 break;
362                         }
363                 }
364         }
365
366         if (noffset == -FDT_ERR_TRUNCATED || noffset == -FDT_ERR_BADSTRUCTURE) {
367                 err_msg = "Corrupted or truncated tree";
368                 goto error;
369         }
370
371         return verified ? 0 : -EPERM;
372
373 error:
374         printf(" error!\n%s for '%s' hash node in '%s' config node\n",
375                err_msg, fit_get_name(fit, noffset, NULL),
376                fit_get_name(fit, conf_noffset, NULL));
377         return -1;
378 }
379
380 int fit_config_verify_required_sigs(const void *fit, int conf_noffset,
381                                     const void *sig_blob)
382 {
383         int noffset;
384         int sig_node;
385
386         /* Work out what we need to verify */
387         sig_node = fdt_subnode_offset(sig_blob, 0, FIT_SIG_NODENAME);
388         if (sig_node < 0) {
389                 debug("%s: No signature node found: %s\n", __func__,
390                       fdt_strerror(sig_node));
391                 return 0;
392         }
393
394         fdt_for_each_subnode(noffset, sig_blob, sig_node) {
395                 const char *required;
396                 int ret;
397
398                 required = fdt_getprop(sig_blob, noffset, "required", NULL);
399                 if (!required || strcmp(required, "conf"))
400                         continue;
401                 ret = fit_config_verify_sig(fit, conf_noffset, sig_blob,
402                                             noffset);
403                 if (ret) {
404                         printf("Failed to verify required signature '%s'\n",
405                                fit_get_name(sig_blob, noffset, NULL));
406                         return ret;
407                 }
408         }
409
410         return 0;
411 }
412
413 int fit_config_verify(const void *fit, int conf_noffset)
414 {
415         return fit_config_verify_required_sigs(fit, conf_noffset,
416                                                gd_fdt_blob());
417 }