blob: 79202e18a2a3dedf1bbb480c970a6c66e41b85c2 [file] [log] [blame]
Tom Rini83d290c2018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
Simon Glass460408e2012-12-05 14:46:36 +00002/*
3 * Copyright (c) 2012 The Chromium OS Authors.
4 *
5 * (C) Copyright 2011
6 * Joe Hershberger, National Instruments, joe.hershberger@ni.com
7 *
8 * (C) Copyright 2000
9 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
Simon Glass460408e2012-12-05 14:46:36 +000010 */
11
Ruchika Gupta2dd90022015-01-23 16:01:58 +053012#ifndef USE_HOSTCC
Simon Glass460408e2012-12-05 14:46:36 +000013#include <common.h>
14#include <command.h>
Simon Glass9fb625c2019-08-01 09:46:51 -060015#include <env.h>
Simon Glassf7ae49f2020-05-10 11:40:05 -060016#include <log.h>
Hung-ying Tyanbf007eb2014-03-03 12:19:28 +010017#include <malloc.h>
Joe Hershberger0eb25b62015-03-22 17:08:59 -050018#include <mapmem.h>
Akshay Saraswat1f9c9282013-03-20 21:00:58 +000019#include <hw_sha.h>
Simon Glass90526e92020-05-10 11:39:56 -060020#include <asm/cache.h>
Simon Glass401d1c42020-10-30 21:38:53 -060021#include <asm/global_data.h>
Simon Glassbd091b62013-02-24 17:33:31 +000022#include <asm/io.h>
Masahiro Yamada1221ce42016-09-21 11:28:55 +090023#include <linux/errno.h>
Simon Glass3db71102019-11-14 12:57:16 -070024#include <u-boot/crc.h>
Ruchika Gupta2dd90022015-01-23 16:01:58 +053025#else
26#include "mkimage.h"
Simon Glassd54f7e32021-09-25 19:43:19 -060027#include <linux/compiler_attributes.h>
Ruchika Gupta2dd90022015-01-23 16:01:58 +053028#include <time.h>
Simon Glass2c212562021-09-25 19:43:18 -060029#include <linux/kconfig.h>
Ruchika Gupta2dd90022015-01-23 16:01:58 +053030#endif /* !USE_HOSTCC*/
31
32#include <hash.h>
Simon Glass4d72caa2020-05-10 11:40:01 -060033#include <image.h>
Ruchika Gupta2dd90022015-01-23 16:01:58 +053034#include <u-boot/crc.h>
35#include <u-boot/sha1.h>
36#include <u-boot/sha256.h>
Reuben Dowled16b38f2020-04-16 17:36:52 +120037#include <u-boot/sha512.h>
Ruchika Gupta2dd90022015-01-23 16:01:58 +053038#include <u-boot/md5.h>
Simon Glass460408e2012-12-05 14:46:36 +000039
T Karthik Reddy10088fb2019-03-16 15:23:01 +053040#if !defined(USE_HOSTCC) && defined(CONFIG_NEEDS_MANUAL_RELOC)
41DECLARE_GLOBAL_DATA_PTR;
42#endif
43
44static void reloc_update(void);
45
Simon Glassd54f7e32021-09-25 19:43:19 -060046static int __maybe_unused hash_init_sha1(struct hash_algo *algo, void **ctxp)
Hung-ying Tyanbf007eb2014-03-03 12:19:28 +010047{
48 sha1_context *ctx = malloc(sizeof(sha1_context));
49 sha1_starts(ctx);
50 *ctxp = ctx;
51 return 0;
52}
53
Simon Glassd54f7e32021-09-25 19:43:19 -060054static int __maybe_unused hash_update_sha1(struct hash_algo *algo, void *ctx,
55 const void *buf, unsigned int size,
56 int is_last)
Hung-ying Tyanbf007eb2014-03-03 12:19:28 +010057{
58 sha1_update((sha1_context *)ctx, buf, size);
59 return 0;
60}
61
Simon Glassd54f7e32021-09-25 19:43:19 -060062static int __maybe_unused hash_finish_sha1(struct hash_algo *algo, void *ctx,
63 void *dest_buf, int size)
Hung-ying Tyanbf007eb2014-03-03 12:19:28 +010064{
65 if (size < algo->digest_size)
66 return -1;
67
68 sha1_finish((sha1_context *)ctx, dest_buf);
69 free(ctx);
70 return 0;
71}
Hung-ying Tyanbf007eb2014-03-03 12:19:28 +010072
Simon Glassd54f7e32021-09-25 19:43:19 -060073static int __maybe_unused hash_init_sha256(struct hash_algo *algo, void **ctxp)
Hung-ying Tyanbf007eb2014-03-03 12:19:28 +010074{
75 sha256_context *ctx = malloc(sizeof(sha256_context));
76 sha256_starts(ctx);
77 *ctxp = ctx;
78 return 0;
79}
80
Simon Glassd54f7e32021-09-25 19:43:19 -060081static int __maybe_unused hash_update_sha256(struct hash_algo *algo, void *ctx,
82 const void *buf, uint size,
83 int is_last)
Hung-ying Tyanbf007eb2014-03-03 12:19:28 +010084{
85 sha256_update((sha256_context *)ctx, buf, size);
86 return 0;
87}
88
Simon Glassd54f7e32021-09-25 19:43:19 -060089static int __maybe_unused hash_finish_sha256(struct hash_algo *algo, void *ctx,
90 void *dest_buf, int size)
Hung-ying Tyanbf007eb2014-03-03 12:19:28 +010091{
92 if (size < algo->digest_size)
93 return -1;
94
95 sha256_finish((sha256_context *)ctx, dest_buf);
96 free(ctx);
97 return 0;
98}
Hung-ying Tyanbf007eb2014-03-03 12:19:28 +010099
Simon Glassd54f7e32021-09-25 19:43:19 -0600100static int __maybe_unused hash_init_sha384(struct hash_algo *algo, void **ctxp)
Reuben Dowled16b38f2020-04-16 17:36:52 +1200101{
102 sha512_context *ctx = malloc(sizeof(sha512_context));
103 sha384_starts(ctx);
104 *ctxp = ctx;
105 return 0;
106}
107
Simon Glassd54f7e32021-09-25 19:43:19 -0600108static int __maybe_unused hash_update_sha384(struct hash_algo *algo, void *ctx,
109 const void *buf, uint size,
110 int is_last)
Reuben Dowled16b38f2020-04-16 17:36:52 +1200111{
112 sha384_update((sha512_context *)ctx, buf, size);
113 return 0;
114}
115
Simon Glassd54f7e32021-09-25 19:43:19 -0600116static int __maybe_unused hash_finish_sha384(struct hash_algo *algo, void *ctx,
117 void *dest_buf, int size)
Reuben Dowled16b38f2020-04-16 17:36:52 +1200118{
119 if (size < algo->digest_size)
120 return -1;
121
122 sha384_finish((sha512_context *)ctx, dest_buf);
123 free(ctx);
124 return 0;
125}
Reuben Dowled16b38f2020-04-16 17:36:52 +1200126
Simon Glassd54f7e32021-09-25 19:43:19 -0600127static int __maybe_unused hash_init_sha512(struct hash_algo *algo, void **ctxp)
Reuben Dowled16b38f2020-04-16 17:36:52 +1200128{
129 sha512_context *ctx = malloc(sizeof(sha512_context));
130 sha512_starts(ctx);
131 *ctxp = ctx;
132 return 0;
133}
134
Simon Glassd54f7e32021-09-25 19:43:19 -0600135static int __maybe_unused hash_update_sha512(struct hash_algo *algo, void *ctx,
136 const void *buf, uint size,
137 int is_last)
Reuben Dowled16b38f2020-04-16 17:36:52 +1200138{
139 sha512_update((sha512_context *)ctx, buf, size);
140 return 0;
141}
142
Simon Glassd54f7e32021-09-25 19:43:19 -0600143static int __maybe_unused hash_finish_sha512(struct hash_algo *algo, void *ctx,
144 void *dest_buf, int size)
Reuben Dowled16b38f2020-04-16 17:36:52 +1200145{
146 if (size < algo->digest_size)
147 return -1;
148
Reuben Dowled16b38f2020-04-16 17:36:52 +1200149 sha512_finish((sha512_context *)ctx, dest_buf);
150 free(ctx);
151 return 0;
152}
Reuben Dowled16b38f2020-04-16 17:36:52 +1200153
Philipp Tomsich51c23452018-11-25 19:22:19 +0100154static int hash_init_crc16_ccitt(struct hash_algo *algo, void **ctxp)
155{
156 uint16_t *ctx = malloc(sizeof(uint16_t));
157 *ctx = 0;
158 *ctxp = ctx;
159 return 0;
160}
161
162static int hash_update_crc16_ccitt(struct hash_algo *algo, void *ctx,
163 const void *buf, unsigned int size,
164 int is_last)
165{
166 *((uint16_t *)ctx) = crc16_ccitt(*((uint16_t *)ctx), buf, size);
167 return 0;
168}
169
170static int hash_finish_crc16_ccitt(struct hash_algo *algo, void *ctx,
171 void *dest_buf, int size)
172{
173 if (size < algo->digest_size)
174 return -1;
175
176 *((uint16_t *)dest_buf) = *((uint16_t *)ctx);
177 free(ctx);
178 return 0;
179}
180
Simon Glasse7d285b2021-09-25 19:43:24 -0600181static int __maybe_unused hash_init_crc32(struct hash_algo *algo, void **ctxp)
Hung-ying Tyanbf007eb2014-03-03 12:19:28 +0100182{
183 uint32_t *ctx = malloc(sizeof(uint32_t));
184 *ctx = 0;
185 *ctxp = ctx;
186 return 0;
187}
188
Simon Glasse7d285b2021-09-25 19:43:24 -0600189static int __maybe_unused hash_update_crc32(struct hash_algo *algo, void *ctx,
190 const void *buf, unsigned int size,
191 int is_last)
Hung-ying Tyanbf007eb2014-03-03 12:19:28 +0100192{
193 *((uint32_t *)ctx) = crc32(*((uint32_t *)ctx), buf, size);
194 return 0;
195}
196
Simon Glasse7d285b2021-09-25 19:43:24 -0600197static int __maybe_unused hash_finish_crc32(struct hash_algo *algo, void *ctx,
198 void *dest_buf, int size)
Hung-ying Tyanbf007eb2014-03-03 12:19:28 +0100199{
200 if (size < algo->digest_size)
201 return -1;
202
203 *((uint32_t *)dest_buf) = *((uint32_t *)ctx);
204 free(ctx);
205 return 0;
206}
207
Simon Glass460408e2012-12-05 14:46:36 +0000208/*
Tom Rini78eda892017-08-14 16:38:07 -0400209 * These are the hash algorithms we support. If we have hardware acceleration
210 * is enable we will use that, otherwise a software version of the algorithm.
211 * Note that algorithm names must be in lower case.
Simon Glass460408e2012-12-05 14:46:36 +0000212 */
213static struct hash_algo hash_algo[] = {
Simon Glass2c212562021-09-25 19:43:18 -0600214#if CONFIG_IS_ENABLED(MD5)
Alexandru Gagniucfe54aea2021-09-02 19:54:20 -0500215 {
216 .name = "md5",
217 .digest_size = MD5_SUM_LEN,
218 .chunk_size = CHUNKSZ_MD5,
219 .hash_func_ws = md5_wd,
220 },
221#endif
Simon Glass2c212562021-09-25 19:43:18 -0600222#if CONFIG_IS_ENABLED(SHA1)
Simon Glass460408e2012-12-05 14:46:36 +0000223 {
Wolfgang Denk0cf207e2021-09-27 17:42:39 +0200224 .name = "sha1",
Tom Rini78eda892017-08-14 16:38:07 -0400225 .digest_size = SHA1_SUM_LEN,
226 .chunk_size = CHUNKSZ_SHA1,
Simon Glass2c212562021-09-25 19:43:18 -0600227#if CONFIG_IS_ENABLED(SHA_HW_ACCEL)
Tom Rini78eda892017-08-14 16:38:07 -0400228 .hash_func_ws = hw_sha1,
229#else
230 .hash_func_ws = sha1_csum_wd,
231#endif
Simon Glass2c212562021-09-25 19:43:18 -0600232#if CONFIG_IS_ENABLED(SHA_PROG_HW_ACCEL)
Tom Rini78eda892017-08-14 16:38:07 -0400233 .hash_init = hw_sha_init,
234 .hash_update = hw_sha_update,
235 .hash_finish = hw_sha_finish,
236#else
237 .hash_init = hash_init_sha1,
238 .hash_update = hash_update_sha1,
239 .hash_finish = hash_finish_sha1,
240#endif
Simon Glass460408e2012-12-05 14:46:36 +0000241 },
242#endif
Simon Glass2c212562021-09-25 19:43:18 -0600243#if CONFIG_IS_ENABLED(SHA256)
Simon Glass460408e2012-12-05 14:46:36 +0000244 {
Tom Rini78eda892017-08-14 16:38:07 -0400245 .name = "sha256",
246 .digest_size = SHA256_SUM_LEN,
247 .chunk_size = CHUNKSZ_SHA256,
Simon Glass2c212562021-09-25 19:43:18 -0600248#if CONFIG_IS_ENABLED(SHA_HW_ACCEL)
Tom Rini78eda892017-08-14 16:38:07 -0400249 .hash_func_ws = hw_sha256,
250#else
251 .hash_func_ws = sha256_csum_wd,
252#endif
Simon Glass2c212562021-09-25 19:43:18 -0600253#if CONFIG_IS_ENABLED(SHA_PROG_HW_ACCEL)
Tom Rini78eda892017-08-14 16:38:07 -0400254 .hash_init = hw_sha_init,
255 .hash_update = hw_sha_update,
256 .hash_finish = hw_sha_finish,
257#else
258 .hash_init = hash_init_sha256,
259 .hash_update = hash_update_sha256,
260 .hash_finish = hash_finish_sha256,
261#endif
Simon Glass460408e2012-12-05 14:46:36 +0000262 },
263#endif
Simon Glass2c212562021-09-25 19:43:18 -0600264#if CONFIG_IS_ENABLED(SHA384)
Reuben Dowled16b38f2020-04-16 17:36:52 +1200265 {
266 .name = "sha384",
267 .digest_size = SHA384_SUM_LEN,
268 .chunk_size = CHUNKSZ_SHA384,
Simon Glass2c212562021-09-25 19:43:18 -0600269#if CONFIG_IS_ENABLED(SHA512_HW_ACCEL)
Joel Stanleya479f102021-02-17 13:50:42 +1030270 .hash_func_ws = hw_sha384,
271#else
Reuben Dowled16b38f2020-04-16 17:36:52 +1200272 .hash_func_ws = sha384_csum_wd,
Joel Stanleya479f102021-02-17 13:50:42 +1030273#endif
Simon Glass2c212562021-09-25 19:43:18 -0600274#if CONFIG_IS_ENABLED(SHA512_HW_ACCEL) && CONFIG_IS_ENABLED(SHA_PROG_HW_ACCEL)
Joel Stanleya479f102021-02-17 13:50:42 +1030275 .hash_init = hw_sha_init,
276 .hash_update = hw_sha_update,
277 .hash_finish = hw_sha_finish,
278#else
Reuben Dowled16b38f2020-04-16 17:36:52 +1200279 .hash_init = hash_init_sha384,
280 .hash_update = hash_update_sha384,
281 .hash_finish = hash_finish_sha384,
Joel Stanleya479f102021-02-17 13:50:42 +1030282#endif
Reuben Dowled16b38f2020-04-16 17:36:52 +1200283 },
284#endif
Simon Glass2c212562021-09-25 19:43:18 -0600285#if CONFIG_IS_ENABLED(SHA512)
Reuben Dowled16b38f2020-04-16 17:36:52 +1200286 {
287 .name = "sha512",
288 .digest_size = SHA512_SUM_LEN,
289 .chunk_size = CHUNKSZ_SHA512,
Simon Glass2c212562021-09-25 19:43:18 -0600290#if CONFIG_IS_ENABLED(SHA512_HW_ACCEL)
Joel Stanleya479f102021-02-17 13:50:42 +1030291 .hash_func_ws = hw_sha512,
292#else
Reuben Dowled16b38f2020-04-16 17:36:52 +1200293 .hash_func_ws = sha512_csum_wd,
Joel Stanleya479f102021-02-17 13:50:42 +1030294#endif
Simon Glass2c212562021-09-25 19:43:18 -0600295#if CONFIG_IS_ENABLED(SHA512_HW_ACCEL) && CONFIG_IS_ENABLED(SHA_PROG_HW_ACCEL)
Joel Stanleya479f102021-02-17 13:50:42 +1030296 .hash_init = hw_sha_init,
297 .hash_update = hw_sha_update,
298 .hash_finish = hw_sha_finish,
299#else
Reuben Dowled16b38f2020-04-16 17:36:52 +1200300 .hash_init = hash_init_sha512,
301 .hash_update = hash_update_sha512,
302 .hash_finish = hash_finish_sha512,
Joel Stanleya479f102021-02-17 13:50:42 +1030303#endif
Reuben Dowled16b38f2020-04-16 17:36:52 +1200304 },
305#endif
Simon Glassd20a40d2013-02-24 20:30:22 +0000306 {
Philipp Tomsich51c23452018-11-25 19:22:19 +0100307 .name = "crc16-ccitt",
308 .digest_size = 2,
309 .chunk_size = CHUNKSZ,
310 .hash_func_ws = crc16_ccitt_wd_buf,
311 .hash_init = hash_init_crc16_ccitt,
312 .hash_update = hash_update_crc16_ccitt,
313 .hash_finish = hash_finish_crc16_ccitt,
314 },
Simon Glasse7d285b2021-09-25 19:43:24 -0600315#if CONFIG_IS_ENABLED(CRC32)
Philipp Tomsich51c23452018-11-25 19:22:19 +0100316 {
Tom Rini78eda892017-08-14 16:38:07 -0400317 .name = "crc32",
318 .digest_size = 4,
319 .chunk_size = CHUNKSZ_CRC32,
320 .hash_func_ws = crc32_wd_buf,
321 .hash_init = hash_init_crc32,
322 .hash_update = hash_update_crc32,
323 .hash_finish = hash_finish_crc32,
Simon Glassd20a40d2013-02-24 20:30:22 +0000324 },
Simon Glasse7d285b2021-09-25 19:43:24 -0600325#endif
Simon Glass460408e2012-12-05 14:46:36 +0000326};
327
Simon Glassd20a40d2013-02-24 20:30:22 +0000328/* Try to minimize code size for boards that don't want much hashing */
Simon Glass2c212562021-09-25 19:43:18 -0600329#if CONFIG_IS_ENABLED(SHA256) || CONFIG_IS_ENABLED(CMD_SHA1SUM) || \
330 CONFIG_IS_ENABLED(CRC32_VERIFY) || CONFIG_IS_ENABLED(CMD_HASH) || \
331 CONFIG_IS_ENABLED(SHA384) || CONFIG_IS_ENABLED(SHA512)
Simon Glassd20a40d2013-02-24 20:30:22 +0000332#define multi_hash() 1
333#else
334#define multi_hash() 0
335#endif
336
T Karthik Reddy10088fb2019-03-16 15:23:01 +0530337static void reloc_update(void)
338{
339#if !defined(USE_HOSTCC) && defined(CONFIG_NEEDS_MANUAL_RELOC)
340 int i;
341 static bool done;
342
343 if (!done) {
344 done = true;
345 for (i = 0; i < ARRAY_SIZE(hash_algo); i++) {
346 hash_algo[i].name += gd->reloc_off;
347 hash_algo[i].hash_func_ws += gd->reloc_off;
348 hash_algo[i].hash_init += gd->reloc_off;
349 hash_algo[i].hash_update += gd->reloc_off;
350 hash_algo[i].hash_finish += gd->reloc_off;
351 }
352 }
353#endif
354}
355
Ruchika Gupta2dd90022015-01-23 16:01:58 +0530356int hash_lookup_algo(const char *algo_name, struct hash_algo **algop)
357{
358 int i;
359
T Karthik Reddy10088fb2019-03-16 15:23:01 +0530360 reloc_update();
361
Ruchika Gupta2dd90022015-01-23 16:01:58 +0530362 for (i = 0; i < ARRAY_SIZE(hash_algo); i++) {
363 if (!strcmp(algo_name, hash_algo[i].name)) {
364 *algop = &hash_algo[i];
365 return 0;
366 }
367 }
368
369 debug("Unknown hash algorithm '%s'\n", algo_name);
370 return -EPROTONOSUPPORT;
371}
372
373int hash_progressive_lookup_algo(const char *algo_name,
374 struct hash_algo **algop)
375{
376 int i;
377
T Karthik Reddy10088fb2019-03-16 15:23:01 +0530378 reloc_update();
379
Ruchika Gupta2dd90022015-01-23 16:01:58 +0530380 for (i = 0; i < ARRAY_SIZE(hash_algo); i++) {
381 if (!strcmp(algo_name, hash_algo[i].name)) {
382 if (hash_algo[i].hash_init) {
383 *algop = &hash_algo[i];
384 return 0;
385 }
386 }
387 }
388
389 debug("Unknown hash algorithm '%s'\n", algo_name);
390 return -EPROTONOSUPPORT;
391}
392
393#ifndef USE_HOSTCC
Stefan Roese8f0b1e22015-05-18 14:08:24 +0200394int hash_parse_string(const char *algo_name, const char *str, uint8_t *result)
395{
396 struct hash_algo *algo;
397 int ret;
398 int i;
399
400 ret = hash_lookup_algo(algo_name, &algo);
401 if (ret)
402 return ret;
403
404 for (i = 0; i < algo->digest_size; i++) {
405 char chr[3];
406
Simon Glass031725f2021-07-24 09:03:28 -0600407 strlcpy(chr, &str[i * 2], 3);
Simon Glass7e5f4602021-07-24 09:03:29 -0600408 result[i] = hextoul(chr, NULL);
Stefan Roese8f0b1e22015-05-18 14:08:24 +0200409 }
410
411 return 0;
412}
413
Tom Rini48ad68d2016-01-05 08:47:48 -0500414int hash_block(const char *algo_name, const void *data, unsigned int len,
415 uint8_t *output, int *output_size)
416{
417 struct hash_algo *algo;
418 int ret;
419
420 ret = hash_lookup_algo(algo_name, &algo);
421 if (ret)
422 return ret;
423
424 if (output_size && *output_size < algo->digest_size) {
425 debug("Output buffer size %d too small (need %d bytes)",
426 *output_size, algo->digest_size);
427 return -ENOSPC;
428 }
429 if (output_size)
430 *output_size = algo->digest_size;
431 algo->hash_func_ws(data, len, output, algo->chunk_size);
432
433 return 0;
434}
435
Simon Glass2c212562021-09-25 19:43:18 -0600436#if !defined(CONFIG_SPL_BUILD) && (defined(CONFIG_CMD_HASH) || \
437 defined(CONFIG_CMD_SHA1SUM) || defined(CONFIG_CMD_CRC32))
Simon Glass460408e2012-12-05 14:46:36 +0000438/**
439 * store_result: Store the resulting sum to an address or variable
440 *
441 * @algo: Hash algorithm being used
442 * @sum: Hash digest (algo->digest_size bytes)
443 * @dest: Destination, interpreted as a hex address if it starts
Simon Glassd5b76672013-02-24 17:33:26 +0000444 * with * (or allow_env_vars is 0) or otherwise as an
445 * environment variable.
446 * @allow_env_vars: non-zero to permit storing the result to an
447 * variable environment
Simon Glass460408e2012-12-05 14:46:36 +0000448 */
Simon Glass04819a42014-06-12 07:24:41 -0600449static void store_result(struct hash_algo *algo, const uint8_t *sum,
Simon Glassd5b76672013-02-24 17:33:26 +0000450 const char *dest, int allow_env_vars)
Simon Glass460408e2012-12-05 14:46:36 +0000451{
452 unsigned int i;
Simon Glassd5b76672013-02-24 17:33:26 +0000453 int env_var = 0;
Simon Glass460408e2012-12-05 14:46:36 +0000454
Simon Glassd5b76672013-02-24 17:33:26 +0000455 /*
456 * If environment variables are allowed, then we assume that 'dest'
457 * is an environment variable, unless it starts with *, in which
458 * case we assume it is an address. If not allowed, it is always an
459 * address. This is to support the crc32 command.
460 */
461 if (allow_env_vars) {
462 if (*dest == '*')
463 dest++;
464 else
465 env_var = 1;
466 }
Simon Glass460408e2012-12-05 14:46:36 +0000467
Simon Glassd5b76672013-02-24 17:33:26 +0000468 if (env_var) {
Simon Glass460408e2012-12-05 14:46:36 +0000469 char str_output[HASH_MAX_DIGEST_SIZE * 2 + 1];
470 char *str_ptr = str_output;
471
472 for (i = 0; i < algo->digest_size; i++) {
473 sprintf(str_ptr, "%02x", sum[i]);
474 str_ptr += 2;
475 }
Jeroen Hofstee8b9cc862014-06-09 11:02:02 +0200476 *str_ptr = '\0';
Simon Glass382bee52017-08-03 12:22:09 -0600477 env_set(dest, str_output);
Simon Glassd5b76672013-02-24 17:33:26 +0000478 } else {
Simon Glassbd091b62013-02-24 17:33:31 +0000479 ulong addr;
480 void *buf;
Simon Glassd5b76672013-02-24 17:33:26 +0000481
Simon Glass7e5f4602021-07-24 09:03:29 -0600482 addr = hextoul(dest, NULL);
Simon Glassbd091b62013-02-24 17:33:31 +0000483 buf = map_sysmem(addr, algo->digest_size);
484 memcpy(buf, sum, algo->digest_size);
485 unmap_sysmem(buf);
Simon Glass460408e2012-12-05 14:46:36 +0000486 }
487}
488
489/**
490 * parse_verify_sum: Parse a hash verification parameter
491 *
492 * @algo: Hash algorithm being used
493 * @verify_str: Argument to parse. If it starts with * then it is
494 * interpreted as a hex address containing the hash.
495 * If the length is exactly the right number of hex digits
496 * for the digest size, then we assume it is a hex digest.
497 * Otherwise we assume it is an environment variable, and
498 * look up its value (it must contain a hex digest).
499 * @vsum: Returns binary digest value (algo->digest_size bytes)
Simon Glassd5b76672013-02-24 17:33:26 +0000500 * @allow_env_vars: non-zero to permit storing the result to an environment
501 * variable. If 0 then verify_str is assumed to be an
502 * address, and the * prefix is not expected.
Simon Glass460408e2012-12-05 14:46:36 +0000503 * @return 0 if ok, non-zero on error
504 */
Simon Glass04819a42014-06-12 07:24:41 -0600505static int parse_verify_sum(struct hash_algo *algo, char *verify_str,
506 uint8_t *vsum, int allow_env_vars)
Simon Glass460408e2012-12-05 14:46:36 +0000507{
Simon Glassd5b76672013-02-24 17:33:26 +0000508 int env_var = 0;
509
510 /* See comment above in store_result() */
511 if (allow_env_vars) {
512 if (*verify_str == '*')
513 verify_str++;
514 else
515 env_var = 1;
516 }
517
Nikolay Dimitrov3ef46a92014-12-12 20:01:23 +0200518 if (!env_var) {
Simon Glassbd091b62013-02-24 17:33:31 +0000519 ulong addr;
520 void *buf;
Simon Glass460408e2012-12-05 14:46:36 +0000521
Simon Glass7e5f4602021-07-24 09:03:29 -0600522 addr = hextoul(verify_str, NULL);
Simon Glassbd091b62013-02-24 17:33:31 +0000523 buf = map_sysmem(addr, algo->digest_size);
524 memcpy(vsum, buf, algo->digest_size);
Simon Glass460408e2012-12-05 14:46:36 +0000525 } else {
Simon Glass460408e2012-12-05 14:46:36 +0000526 char *vsum_str;
527 int digits = algo->digest_size * 2;
528
529 /*
530 * As with the original code from sha1sum.c, we assume that a
531 * string which matches the digest size exactly is a hex
532 * string and not an environment variable.
533 */
534 if (strlen(verify_str) == digits)
535 vsum_str = verify_str;
536 else {
Simon Glass00caae62017-08-03 12:22:12 -0600537 vsum_str = env_get(verify_str);
Simon Glass460408e2012-12-05 14:46:36 +0000538 if (vsum_str == NULL || strlen(vsum_str) != digits) {
539 printf("Expected %d hex digits in env var\n",
540 digits);
541 return 1;
542 }
543 }
544
Stefan Roese8f0b1e22015-05-18 14:08:24 +0200545 hash_parse_string(algo->name, vsum_str, vsum);
Simon Glass460408e2012-12-05 14:46:36 +0000546 }
547 return 0;
548}
549
Tom Rini48ad68d2016-01-05 08:47:48 -0500550static void hash_show(struct hash_algo *algo, ulong addr, ulong len, uint8_t *output)
Simon Glass460408e2012-12-05 14:46:36 +0000551{
552 int i;
553
554 printf("%s for %08lx ... %08lx ==> ", algo->name, addr, addr + len - 1);
555 for (i = 0; i < algo->digest_size; i++)
556 printf("%02x", output[i]);
557}
558
Simon Glass09140112020-05-10 11:40:03 -0600559int hash_command(const char *algo_name, int flags, struct cmd_tbl *cmdtp,
560 int flag, int argc, char *const argv[])
Simon Glass460408e2012-12-05 14:46:36 +0000561{
Simon Glass460408e2012-12-05 14:46:36 +0000562 ulong addr, len;
Simon Glass460408e2012-12-05 14:46:36 +0000563
Nikolay Dimitrov3ef46a92014-12-12 20:01:23 +0200564 if ((argc < 2) || ((flags & HASH_FLAG_VERIFY) && (argc < 3)))
Simon Glass460408e2012-12-05 14:46:36 +0000565 return CMD_RET_USAGE;
566
Simon Glass7e5f4602021-07-24 09:03:29 -0600567 addr = hextoul(*argv++, NULL);
568 len = hextoul(*argv++, NULL);
Simon Glassd5b76672013-02-24 17:33:26 +0000569
Simon Glassd20a40d2013-02-24 20:30:22 +0000570 if (multi_hash()) {
571 struct hash_algo *algo;
Breno Limad7af2ba2018-01-17 10:03:45 -0200572 u8 *output;
Simon Glass04819a42014-06-12 07:24:41 -0600573 uint8_t vsum[HASH_MAX_DIGEST_SIZE];
Simon Glassbd091b62013-02-24 17:33:31 +0000574 void *buf;
Simon Glass460408e2012-12-05 14:46:36 +0000575
Hung-ying Tyanbf007eb2014-03-03 12:19:28 +0100576 if (hash_lookup_algo(algo_name, &algo)) {
Simon Glassd20a40d2013-02-24 20:30:22 +0000577 printf("Unknown hash algorithm '%s'\n", algo_name);
Simon Glass460408e2012-12-05 14:46:36 +0000578 return CMD_RET_USAGE;
Simon Glassd20a40d2013-02-24 20:30:22 +0000579 }
580 argc -= 2;
581
582 if (algo->digest_size > HASH_MAX_DIGEST_SIZE) {
583 puts("HASH_MAX_DIGEST_SIZE exceeded\n");
584 return 1;
585 }
586
Breno Limad7af2ba2018-01-17 10:03:45 -0200587 output = memalign(ARCH_DMA_MINALIGN,
588 sizeof(uint32_t) * HASH_MAX_DIGEST_SIZE);
589
Simon Glassbd091b62013-02-24 17:33:31 +0000590 buf = map_sysmem(addr, len);
591 algo->hash_func_ws(buf, len, output, algo->chunk_size);
592 unmap_sysmem(buf);
Simon Glassd20a40d2013-02-24 20:30:22 +0000593
594 /* Try to avoid code bloat when verify is not needed */
Daniel Thompson221a9492017-05-19 17:26:58 +0100595#if defined(CONFIG_CRC32_VERIFY) || defined(CONFIG_SHA1SUM_VERIFY) || \
596 defined(CONFIG_HASH_VERIFY)
Simon Glassd20a40d2013-02-24 20:30:22 +0000597 if (flags & HASH_FLAG_VERIFY) {
598#else
599 if (0) {
600#endif
Simon Glassd20a40d2013-02-24 20:30:22 +0000601 if (parse_verify_sum(algo, *argv, vsum,
Simon Glassd5b76672013-02-24 17:33:26 +0000602 flags & HASH_FLAG_ENV)) {
Simon Glassd20a40d2013-02-24 20:30:22 +0000603 printf("ERROR: %s does not contain a valid "
604 "%s sum\n", *argv, algo->name);
605 return 1;
606 }
607 if (memcmp(output, vsum, algo->digest_size) != 0) {
608 int i;
Simon Glass460408e2012-12-05 14:46:36 +0000609
Simon Glass31890ae2014-06-02 22:04:49 -0600610 hash_show(algo, addr, len, output);
Simon Glassd20a40d2013-02-24 20:30:22 +0000611 printf(" != ");
612 for (i = 0; i < algo->digest_size; i++)
613 printf("%02x", vsum[i]);
614 puts(" ** ERROR **\n");
615 return 1;
616 }
617 } else {
Simon Glass31890ae2014-06-02 22:04:49 -0600618 hash_show(algo, addr, len, output);
Simon Glassd20a40d2013-02-24 20:30:22 +0000619 printf("\n");
Simon Glass460408e2012-12-05 14:46:36 +0000620
Simon Glassd20a40d2013-02-24 20:30:22 +0000621 if (argc) {
622 store_result(algo, output, *argv,
623 flags & HASH_FLAG_ENV);
624 }
Breno Limad7af2ba2018-01-17 10:03:45 -0200625 unmap_sysmem(output);
626
Simon Glassd20a40d2013-02-24 20:30:22 +0000627 }
628
629 /* Horrible code size hack for boards that just want crc32 */
630 } else {
631 ulong crc;
632 ulong *ptr;
633
634 crc = crc32_wd(0, (const uchar *)addr, len, CHUNKSZ_CRC32);
635
636 printf("CRC32 for %08lx ... %08lx ==> %08lx\n",
637 addr, addr + len - 1, crc);
638
Tom Rini4b756b02013-11-07 07:39:48 -0500639 if (argc >= 3) {
Simon Glass7e5f4602021-07-24 09:03:29 -0600640 ptr = (ulong *)hextoul(argv[0], NULL);
Simon Glassd20a40d2013-02-24 20:30:22 +0000641 *ptr = crc;
Simon Glassd5b76672013-02-24 17:33:26 +0000642 }
Simon Glass460408e2012-12-05 14:46:36 +0000643 }
644
645 return 0;
646}
Simon Glassd70f9192017-05-17 09:05:34 -0600647#endif /* CONFIG_CMD_HASH || CONFIG_CMD_SHA1SUM || CONFIG_CMD_CRC32) */
648#endif /* !USE_HOSTCC */