[J-u-boot.git] / common / hash.c

// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright (c) 2012 The Chromium OS Authors.
 *
 * (C) Copyright 2011
 * Joe Hershberger, National Instruments, [email protected]
 *
 * (C) Copyright 2000
 * Wolfgang Denk, DENX Software Engineering, [email protected].
 */

#ifndef USE_HOSTCC
#include <common.h>
#include <command.h>
#include <env.h>
#include <log.h>
#include <malloc.h>
#include <mapmem.h>
#include <hw_sha.h>
#include <asm/cache.h>
#include <asm/global_data.h>
#include <asm/io.h>
#include <linux/errno.h>
#include <u-boot/crc.h>
#else
#include "mkimage.h"
#include <time.h>
#endif /* !USE_HOSTCC*/

#include <hash.h>
#include <image.h>
#include <u-boot/crc.h>
#include <u-boot/sha1.h>
#include <u-boot/sha256.h>
#include <u-boot/sha512.h>
#include <u-boot/md5.h>

#if !defined(USE_HOSTCC) && defined(CONFIG_NEEDS_MANUAL_RELOC)
DECLARE_GLOBAL_DATA_PTR;
#endif

static void reloc_update(void);

#if defined(CONFIG_SHA1) && !defined(CONFIG_SHA_PROG_HW_ACCEL)
static int hash_init_sha1(struct hash_algo *algo, void **ctxp)
{
	sha1_context *ctx = malloc(sizeof(sha1_context));
	sha1_starts(ctx);
	*ctxp = ctx;
	return 0;
}

static int hash_update_sha1(struct hash_algo *algo, void *ctx, const void *buf,
			    unsigned int size, int is_last)
{
	sha1_update((sha1_context *)ctx, buf, size);
	return 0;
}

static int hash_finish_sha1(struct hash_algo *algo, void *ctx, void *dest_buf,
			    int size)
{
	if (size < algo->digest_size)
		return -1;

	sha1_finish((sha1_context *)ctx, dest_buf);
	free(ctx);
	return 0;
}
#endif

#if defined(CONFIG_SHA256) && !defined(CONFIG_SHA_PROG_HW_ACCEL)
static int hash_init_sha256(struct hash_algo *algo, void **ctxp)
{
	sha256_context *ctx = malloc(sizeof(sha256_context));
	sha256_starts(ctx);
	*ctxp = ctx;
	return 0;
}

static int hash_update_sha256(struct hash_algo *algo, void *ctx,
			      const void *buf, unsigned int size, int is_last)
{
	sha256_update((sha256_context *)ctx, buf, size);
	return 0;
}

static int hash_finish_sha256(struct hash_algo *algo, void *ctx, void
			      *dest_buf, int size)
{
	if (size < algo->digest_size)
		return -1;

	sha256_finish((sha256_context *)ctx, dest_buf);
	free(ctx);
	return 0;
}
#endif

#if defined(CONFIG_SHA384) && !defined(CONFIG_SHA_PROG_HW_ACCEL)
static int hash_init_sha384(struct hash_algo *algo, void **ctxp)
{
	sha512_context *ctx = malloc(sizeof(sha512_context));
	sha384_starts(ctx);
	*ctxp = ctx;
	return 0;
}

static int hash_update_sha384(struct hash_algo *algo, void *ctx,
			      const void *buf, unsigned int size, int is_last)
{
	sha384_update((sha512_context *)ctx, buf, size);
	return 0;
}

static int hash_finish_sha384(struct hash_algo *algo, void *ctx, void
			      *dest_buf, int size)
{
	if (size < algo->digest_size)
		return -1;

	sha384_finish((sha512_context *)ctx, dest_buf);
	free(ctx);
	return 0;
}
#endif

#if defined(CONFIG_SHA512) && !defined(CONFIG_SHA_PROG_HW_ACCEL)
static int hash_init_sha512(struct hash_algo *algo, void **ctxp)
{
	sha512_context *ctx = malloc(sizeof(sha512_context));
	sha512_starts(ctx);
	*ctxp = ctx;
	return 0;
}

static int hash_update_sha512(struct hash_algo *algo, void *ctx,
			      const void *buf, unsigned int size, int is_last)
{
	sha512_update((sha512_context *)ctx, buf, size);
	return 0;
}

static int hash_finish_sha512(struct hash_algo *algo, void *ctx, void
			      *dest_buf, int size)
{
	if (size < algo->digest_size)
		return -1;

	sha512_finish((sha512_context *)ctx, dest_buf);
	free(ctx);
	return 0;
}
#endif


static int hash_init_crc16_ccitt(struct hash_algo *algo, void **ctxp)
{
	uint16_t *ctx = malloc(sizeof(uint16_t));
	*ctx = 0;
	*ctxp = ctx;
	return 0;
}

static int hash_update_crc16_ccitt(struct hash_algo *algo, void *ctx,
				   const void *buf, unsigned int size,
				   int is_last)
{
	*((uint16_t *)ctx) = crc16_ccitt(*((uint16_t *)ctx), buf, size);
	return 0;
}

static int hash_finish_crc16_ccitt(struct hash_algo *algo, void *ctx,
				   void *dest_buf, int size)
{
	if (size < algo->digest_size)
		return -1;

	*((uint16_t *)dest_buf) = *((uint16_t *)ctx);
	free(ctx);
	return 0;
}

static int hash_init_crc32(struct hash_algo *algo, void **ctxp)
{
	uint32_t *ctx = malloc(sizeof(uint32_t));
	*ctx = 0;
	*ctxp = ctx;
	return 0;
}

static int hash_update_crc32(struct hash_algo *algo, void *ctx,
			     const void *buf, unsigned int size, int is_last)
{
	*((uint32_t *)ctx) = crc32(*((uint32_t *)ctx), buf, size);
	return 0;
}

static int hash_finish_crc32(struct hash_algo *algo, void *ctx, void *dest_buf,
			     int size)
{
	if (size < algo->digest_size)
		return -1;

	*((uint32_t *)dest_buf) = *((uint32_t *)ctx);
	free(ctx);
	return 0;
}

#ifdef USE_HOSTCC
# define I_WANT_MD5	1
#else
# define I_WANT_MD5	CONFIG_IS_ENABLED(MD5)
#endif
/*
 * These are the hash algorithms we support.  If we have hardware acceleration
 * is enable we will use that, otherwise a software version of the algorithm.
 * Note that algorithm names must be in lower case.
 */
static struct hash_algo hash_algo[] = {
#if I_WANT_MD5
	{
		.name		= "md5",
		.digest_size	= MD5_SUM_LEN,
		.chunk_size	= CHUNKSZ_MD5,
		.hash_func_ws	= md5_wd,
	},
#endif
#ifdef CONFIG_SHA1
	{
		.name		= "sha1",
		.digest_size	= SHA1_SUM_LEN,
		.chunk_size	= CHUNKSZ_SHA1,
#ifdef CONFIG_SHA_HW_ACCEL
		.hash_func_ws	= hw_sha1,
#else
		.hash_func_ws	= sha1_csum_wd,
#endif
#ifdef CONFIG_SHA_PROG_HW_ACCEL
		.hash_init	= hw_sha_init,
		.hash_update	= hw_sha_update,
		.hash_finish	= hw_sha_finish,
#else
		.hash_init	= hash_init_sha1,
		.hash_update	= hash_update_sha1,
		.hash_finish	= hash_finish_sha1,
#endif
	},
#endif
#ifdef CONFIG_SHA256
	{
		.name		= "sha256",
		.digest_size	= SHA256_SUM_LEN,
		.chunk_size	= CHUNKSZ_SHA256,
#ifdef CONFIG_SHA_HW_ACCEL
		.hash_func_ws	= hw_sha256,
#else
		.hash_func_ws	= sha256_csum_wd,
#endif
#ifdef CONFIG_SHA_PROG_HW_ACCEL
		.hash_init	= hw_sha_init,
		.hash_update	= hw_sha_update,
		.hash_finish	= hw_sha_finish,
#else
		.hash_init	= hash_init_sha256,
		.hash_update	= hash_update_sha256,
		.hash_finish	= hash_finish_sha256,
#endif
	},
#endif
#ifdef CONFIG_SHA384
	{
		.name		= "sha384",
		.digest_size	= SHA384_SUM_LEN,
		.chunk_size	= CHUNKSZ_SHA384,
#ifdef CONFIG_SHA512_HW_ACCEL
		.hash_func_ws	= hw_sha384,
#else
		.hash_func_ws	= sha384_csum_wd,
#endif
#if defined(CONFIG_SHA512_HW_ACCEL) && defined(CONFIG_SHA_PROG_HW_ACCEL)
		.hash_init	= hw_sha_init,
		.hash_update	= hw_sha_update,
		.hash_finish	= hw_sha_finish,
#else
		.hash_init	= hash_init_sha384,
		.hash_update	= hash_update_sha384,
		.hash_finish	= hash_finish_sha384,
#endif
	},
#endif
#ifdef CONFIG_SHA512
	{
		.name		= "sha512",
		.digest_size	= SHA512_SUM_LEN,
		.chunk_size	= CHUNKSZ_SHA512,
#ifdef CONFIG_SHA512_HW_ACCEL
		.hash_func_ws	= hw_sha512,
#else
		.hash_func_ws	= sha512_csum_wd,
#endif
#if defined(CONFIG_SHA512_HW_ACCEL) && defined(CONFIG_SHA_PROG_HW_ACCEL)
		.hash_init	= hw_sha_init,
		.hash_update	= hw_sha_update,
		.hash_finish	= hw_sha_finish,
#else
		.hash_init	= hash_init_sha512,
		.hash_update	= hash_update_sha512,
		.hash_finish	= hash_finish_sha512,
#endif
	},
#endif
	{
		.name		= "crc16-ccitt",
		.digest_size	= 2,
		.chunk_size	= CHUNKSZ,
		.hash_func_ws	= crc16_ccitt_wd_buf,
		.hash_init	= hash_init_crc16_ccitt,
		.hash_update	= hash_update_crc16_ccitt,
		.hash_finish	= hash_finish_crc16_ccitt,
	},
	{
		.name		= "crc32",
		.digest_size	= 4,
		.chunk_size	= CHUNKSZ_CRC32,
		.hash_func_ws	= crc32_wd_buf,
		.hash_init	= hash_init_crc32,
		.hash_update	= hash_update_crc32,
		.hash_finish	= hash_finish_crc32,
	},
};

/* Try to minimize code size for boards that don't want much hashing */
#if defined(CONFIG_SHA256) || defined(CONFIG_CMD_SHA1SUM) || \
	defined(CONFIG_CRC32_VERIFY) || defined(CONFIG_CMD_HASH) || \
	defined(CONFIG_SHA384) || defined(CONFIG_SHA512)
#define multi_hash()	1
#else
#define multi_hash()	0
#endif

static void reloc_update(void)
{
#if !defined(USE_HOSTCC) && defined(CONFIG_NEEDS_MANUAL_RELOC)
	int i;
	static bool done;

	if (!done) {
		done = true;
		for (i = 0; i < ARRAY_SIZE(hash_algo); i++) {
			hash_algo[i].name += gd->reloc_off;
			hash_algo[i].hash_func_ws += gd->reloc_off;
			hash_algo[i].hash_init += gd->reloc_off;
			hash_algo[i].hash_update += gd->reloc_off;
			hash_algo[i].hash_finish += gd->reloc_off;
		}
	}
#endif
}

int hash_lookup_algo(const char *algo_name, struct hash_algo **algop)
{
	int i;

	reloc_update();

	for (i = 0; i < ARRAY_SIZE(hash_algo); i++) {
		if (!strcmp(algo_name, hash_algo[i].name)) {
			*algop = &hash_algo[i];
			return 0;
		}
	}

	debug("Unknown hash algorithm '%s'\n", algo_name);
	return -EPROTONOSUPPORT;
}

int hash_progressive_lookup_algo(const char *algo_name,
				 struct hash_algo **algop)
{
	int i;

	reloc_update();

	for (i = 0; i < ARRAY_SIZE(hash_algo); i++) {
		if (!strcmp(algo_name, hash_algo[i].name)) {
			if (hash_algo[i].hash_init) {
				*algop = &hash_algo[i];
				return 0;
			}
		}
	}

	debug("Unknown hash algorithm '%s'\n", algo_name);
	return -EPROTONOSUPPORT;
}

#ifndef USE_HOSTCC
int hash_parse_string(const char *algo_name, const char *str, uint8_t *result)
{
	struct hash_algo *algo;
	int ret;
	int i;

	ret = hash_lookup_algo(algo_name, &algo);
	if (ret)
		return ret;

	for (i = 0; i < algo->digest_size; i++) {
		char chr[3];

		strlcpy(chr, &str[i * 2], 3);
		result[i] = hextoul(chr, NULL);
	}

	return 0;
}

int hash_block(const char *algo_name, const void *data, unsigned int len,
	       uint8_t *output, int *output_size)
{
	struct hash_algo *algo;
	int ret;

	ret = hash_lookup_algo(algo_name, &algo);
	if (ret)
		return ret;

	if (output_size && *output_size < algo->digest_size) {
		debug("Output buffer size %d too small (need %d bytes)",
		      *output_size, algo->digest_size);
		return -ENOSPC;
	}
	if (output_size)
		*output_size = algo->digest_size;
	algo->hash_func_ws(data, len, output, algo->chunk_size);

	return 0;
}

#if defined(CONFIG_CMD_HASH) || defined(CONFIG_CMD_SHA1SUM) || defined(CONFIG_CMD_CRC32)
/**
 * store_result: Store the resulting sum to an address or variable
 *
 * @algo:		Hash algorithm being used
 * @sum:		Hash digest (algo->digest_size bytes)
 * @dest:		Destination, interpreted as a hex address if it starts
 *			with * (or allow_env_vars is 0) or otherwise as an
 *			environment variable.
 * @allow_env_vars:	non-zero to permit storing the result to an
 *			variable environment
 */
static void store_result(struct hash_algo *algo, const uint8_t *sum,
			 const char *dest, int allow_env_vars)
{
	unsigned int i;
	int env_var = 0;

	/*
	 * If environment variables are allowed, then we assume that 'dest'
	 * is an environment variable, unless it starts with *, in which
	 * case we assume it is an address. If not allowed, it is always an
	 * address. This is to support the crc32 command.
	 */
	if (allow_env_vars) {
		if (*dest == '*')
			dest++;
		else
			env_var = 1;
	}

	if (env_var) {
		char str_output[HASH_MAX_DIGEST_SIZE * 2 + 1];
		char *str_ptr = str_output;

		for (i = 0; i < algo->digest_size; i++) {
			sprintf(str_ptr, "%02x", sum[i]);
			str_ptr += 2;
		}
		*str_ptr = '\0';
		env_set(dest, str_output);
	} else {
		ulong addr;
		void *buf;

		addr = hextoul(dest, NULL);
		buf = map_sysmem(addr, algo->digest_size);
		memcpy(buf, sum, algo->digest_size);
		unmap_sysmem(buf);
	}
}

/**
 * parse_verify_sum: Parse a hash verification parameter
 *
 * @algo:		Hash algorithm being used
 * @verify_str:		Argument to parse. If it starts with * then it is
 *			interpreted as a hex address containing the hash.
 *			If the length is exactly the right number of hex digits
 *			for the digest size, then we assume it is a hex digest.
 *			Otherwise we assume it is an environment variable, and
 *			look up its value (it must contain a hex digest).
 * @vsum:		Returns binary digest value (algo->digest_size bytes)
 * @allow_env_vars:	non-zero to permit storing the result to an environment
 *			variable. If 0 then verify_str is assumed to be an
 *			address, and the * prefix is not expected.
 * @return 0 if ok, non-zero on error
 */
static int parse_verify_sum(struct hash_algo *algo, char *verify_str,
			    uint8_t *vsum, int allow_env_vars)
{
	int env_var = 0;

	/* See comment above in store_result() */
	if (allow_env_vars) {
		if (*verify_str == '*')
			verify_str++;
		else
			env_var = 1;
	}

	if (!env_var) {
		ulong addr;
		void *buf;

		addr = hextoul(verify_str, NULL);
		buf = map_sysmem(addr, algo->digest_size);
		memcpy(vsum, buf, algo->digest_size);
	} else {
		char *vsum_str;
		int digits = algo->digest_size * 2;

		/*
		 * As with the original code from sha1sum.c, we assume that a
		 * string which matches the digest size exactly is a hex
		 * string and not an environment variable.
		 */
		if (strlen(verify_str) == digits)
			vsum_str = verify_str;
		else {
			vsum_str = env_get(verify_str);
			if (vsum_str == NULL || strlen(vsum_str) != digits) {
				printf("Expected %d hex digits in env var\n",
				       digits);
				return 1;
			}
		}

		hash_parse_string(algo->name, vsum_str, vsum);
	}
	return 0;
}

static void hash_show(struct hash_algo *algo, ulong addr, ulong len, uint8_t *output)
{
	int i;

	printf("%s for %08lx ... %08lx ==> ", algo->name, addr, addr + len - 1);
	for (i = 0; i < algo->digest_size; i++)
		printf("%02x", output[i]);
}

int hash_command(const char *algo_name, int flags, struct cmd_tbl *cmdtp,
		 int flag, int argc, char *const argv[])
{
	ulong addr, len;

	if ((argc < 2) || ((flags & HASH_FLAG_VERIFY) && (argc < 3)))
		return CMD_RET_USAGE;

	addr = hextoul(*argv++, NULL);
	len = hextoul(*argv++, NULL);

	if (multi_hash()) {
		struct hash_algo *algo;
		u8 *output;
		uint8_t vsum[HASH_MAX_DIGEST_SIZE];
		void *buf;

		if (hash_lookup_algo(algo_name, &algo)) {
			printf("Unknown hash algorithm '%s'\n", algo_name);
			return CMD_RET_USAGE;
		}
		argc -= 2;

		if (algo->digest_size > HASH_MAX_DIGEST_SIZE) {
			puts("HASH_MAX_DIGEST_SIZE exceeded\n");
			return 1;
		}

		output = memalign(ARCH_DMA_MINALIGN,
				  sizeof(uint32_t) * HASH_MAX_DIGEST_SIZE);

		buf = map_sysmem(addr, len);
		algo->hash_func_ws(buf, len, output, algo->chunk_size);
		unmap_sysmem(buf);

		/* Try to avoid code bloat when verify is not needed */
#if defined(CONFIG_CRC32_VERIFY) || defined(CONFIG_SHA1SUM_VERIFY) || \
	defined(CONFIG_HASH_VERIFY)
		if (flags & HASH_FLAG_VERIFY) {
#else
		if (0) {
#endif
			if (parse_verify_sum(algo, *argv, vsum,
					flags & HASH_FLAG_ENV)) {
				printf("ERROR: %s does not contain a valid "
					"%s sum\n", *argv, algo->name);
				return 1;
			}
			if (memcmp(output, vsum, algo->digest_size) != 0) {
				int i;

				hash_show(algo, addr, len, output);
				printf(" != ");
				for (i = 0; i < algo->digest_size; i++)
					printf("%02x", vsum[i]);
				puts(" ** ERROR **\n");
				return 1;
			}
		} else {
			hash_show(algo, addr, len, output);
			printf("\n");

			if (argc) {
				store_result(algo, output, *argv,
					flags & HASH_FLAG_ENV);
			}
		unmap_sysmem(output);

		}

	/* Horrible code size hack for boards that just want crc32 */
	} else {
		ulong crc;
		ulong *ptr;

		crc = crc32_wd(0, (const uchar *)addr, len, CHUNKSZ_CRC32);

		printf("CRC32 for %08lx ... %08lx ==> %08lx\n",
				addr, addr + len - 1, crc);

		if (argc >= 3) {
			ptr = (ulong *)hextoul(argv[0], NULL);
			*ptr = crc;
		}
	}

	return 0;
}
#endif /* CONFIG_CMD_HASH || CONFIG_CMD_SHA1SUM || CONFIG_CMD_CRC32) */
#endif /* !USE_HOSTCC */
Commit	Line	Data
83d290c5	1	// SPDX-License-Identifier: GPL-2.0+
460408ef SG	2	/*
	3	* Copyright (c) 2012 The Chromium OS Authors.
	4	*
	5	* (C) Copyright 2011
	6	* Joe Hershberger, National Instruments, [email protected]
	7	*
	8	* (C) Copyright 2000
	9	* Wolfgang Denk, DENX Software Engineering, [email protected].
460408ef SG	10	*/
460408ef SG	11
2dd90027	12	#ifndef USE_HOSTCC
460408ef SG	13	#include <common.h>
460408ef SG	14	#include <command.h>
9fb625ce	15	#include <env.h>
f7ae49fc	16	#include <log.h>
bf007ebb	17	#include <malloc.h>
0eb25b61	18	#include <mapmem.h>
1f9c9280	19	#include <hw_sha.h>
90526e9f	20	#include <asm/cache.h>
401d1c4f	21	#include <asm/global_data.h>
2dd90027	22	#include <asm/io.h>
1221ce45	23	#include <linux/errno.h>
3db71108	24	#include <u-boot/crc.h>
2dd90027 RG	25	#else
	26	#include "mkimage.h"
	27	#include <time.h>
2dd90027 RG	28	#endif /* !USE_HOSTCC*/
2dd90027 RG	29
460408ef	30	#include <hash.h>
4d72caa5	31	#include <image.h>
2dd90027	32	#include <u-boot/crc.h>
2b9912e6 JH	33	#include <u-boot/sha1.h>
2b9912e6 JH	34	#include <u-boot/sha256.h>
d16b38f4	35	#include <u-boot/sha512.h>
2dd90027	36	#include <u-boot/md5.h>
460408ef	37
10088fb0 KR	38	#if !defined(USE_HOSTCC) && defined(CONFIG_NEEDS_MANUAL_RELOC)
	39	DECLARE_GLOBAL_DATA_PTR;
	40	#endif
	41
	42	static void reloc_update(void);
	43
78eda89e	44	#if defined(CONFIG_SHA1) && !defined(CONFIG_SHA_PROG_HW_ACCEL)
bf007ebb HT	45	static int hash_init_sha1(struct hash_algo algo, void *ctxp)
	46	{
	47	sha1_context *ctx = malloc(sizeof(sha1_context));
	48	sha1_starts(ctx);
	49	*ctxp = ctx;
	50	return 0;
	51	}
	52
	53	static int hash_update_sha1(struct hash_algo algo, void ctx, const void *buf,
	54	unsigned int size, int is_last)
	55	{
	56	sha1_update((sha1_context *)ctx, buf, size);
	57	return 0;
	58	}
	59
	60	static int hash_finish_sha1(struct hash_algo algo, void ctx, void *dest_buf,
	61	int size)
	62	{
	63	if (size < algo->digest_size)
	64	return -1;
	65
	66	sha1_finish((sha1_context *)ctx, dest_buf);
	67	free(ctx);
	68	return 0;
	69	}
	70	#endif
	71
78eda89e	72	#if defined(CONFIG_SHA256) && !defined(CONFIG_SHA_PROG_HW_ACCEL)
bf007ebb HT	73	static int hash_init_sha256(struct hash_algo algo, void *ctxp)
	74	{
	75	sha256_context *ctx = malloc(sizeof(sha256_context));
	76	sha256_starts(ctx);
	77	*ctxp = ctx;
	78	return 0;
	79	}
	80
	81	static int hash_update_sha256(struct hash_algo algo, void ctx,
	82	const void *buf, unsigned int size, int is_last)
	83	{
	84	sha256_update((sha256_context *)ctx, buf, size);
	85	return 0;
	86	}
	87
	88	static int hash_finish_sha256(struct hash_algo algo, void ctx, void
	89	*dest_buf, int size)
	90	{
	91	if (size < algo->digest_size)
	92	return -1;
	93
	94	sha256_finish((sha256_context *)ctx, dest_buf);
	95	free(ctx);
	96	return 0;
	97	}
	98	#endif
	99
a479f103	100	#if defined(CONFIG_SHA384) && !defined(CONFIG_SHA_PROG_HW_ACCEL)
d16b38f4 RD	101	static int hash_init_sha384(struct hash_algo algo, void *ctxp)
	102	{
	103	sha512_context *ctx = malloc(sizeof(sha512_context));
	104	sha384_starts(ctx);
	105	*ctxp = ctx;
	106	return 0;
	107	}
	108
	109	static int hash_update_sha384(struct hash_algo algo, void ctx,
	110	const void *buf, unsigned int size, int is_last)
	111	{
	112	sha384_update((sha512_context *)ctx, buf, size);
	113	return 0;
	114	}
	115
	116	static int hash_finish_sha384(struct hash_algo algo, void ctx, void
	117	*dest_buf, int size)
	118	{
	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	}
	126	#endif
	127
a479f103	128	#if defined(CONFIG_SHA512) && !defined(CONFIG_SHA_PROG_HW_ACCEL)
d16b38f4 RD	129	static int hash_init_sha512(struct hash_algo algo, void *ctxp)
	130	{
	131	sha512_context *ctx = malloc(sizeof(sha512_context));
	132	sha512_starts(ctx);
	133	*ctxp = ctx;
	134	return 0;
	135	}
	136
	137	static int hash_update_sha512(struct hash_algo algo, void ctx,
	138	const void *buf, unsigned int size, int is_last)
	139	{
	140	sha512_update((sha512_context *)ctx, buf, size);
	141	return 0;
	142	}
	143
	144	static int hash_finish_sha512(struct hash_algo algo, void ctx, void
	145	*dest_buf, int size)
	146	{
	147	if (size < algo->digest_size)
	148	return -1;
	149
d16b38f4 RD	150	sha512_finish((sha512_context *)ctx, dest_buf);
	151	free(ctx);
	152	return 0;
	153	}
	154	#endif
	155
	156
51c2345b PT	157	static int hash_init_crc16_ccitt(struct hash_algo algo, void *ctxp)
	158	{
	159	uint16_t *ctx = malloc(sizeof(uint16_t));
	160	*ctx = 0;
	161	*ctxp = ctx;
	162	return 0;
	163	}
	164
	165	static int hash_update_crc16_ccitt(struct hash_algo algo, void ctx,
	166	const void *buf, unsigned int size,
	167	int is_last)
	168	{
	169	((uint16_t )ctx) = crc16_ccitt(((uint16_t )ctx), buf, size);
	170	return 0;
	171	}
	172
	173	static int hash_finish_crc16_ccitt(struct hash_algo algo, void ctx,
	174	void *dest_buf, int size)
	175	{
	176	if (size < algo->digest_size)
	177	return -1;
	178
	179	((uint16_t )dest_buf) = ((uint16_t )ctx);
	180	free(ctx);
	181	return 0;
	182	}
	183
bf007ebb HT	184	static int hash_init_crc32(struct hash_algo algo, void *ctxp)
	185	{
	186	uint32_t *ctx = malloc(sizeof(uint32_t));
	187	*ctx = 0;
	188	*ctxp = ctx;
	189	return 0;
	190	}
	191
	192	static int hash_update_crc32(struct hash_algo algo, void ctx,
	193	const void *buf, unsigned int size, int is_last)
	194	{
	195	((uint32_t )ctx) = crc32(((uint32_t )ctx), buf, size);
	196	return 0;
	197	}
	198
	199	static int hash_finish_crc32(struct hash_algo algo, void ctx, void *dest_buf,
	200	int size)
	201	{
	202	if (size < algo->digest_size)
	203	return -1;
	204
	205	((uint32_t )dest_buf) = ((uint32_t )ctx);
	206	free(ctx);
	207	return 0;
	208	}
	209
fe54aeaa AG	210	#ifdef USE_HOSTCC
	211	# define I_WANT_MD5 1
	212	#else
	213	# define I_WANT_MD5 CONFIG_IS_ENABLED(MD5)
	214	#endif
460408ef	215	/*
78eda89e TR	216	* These are the hash algorithms we support. If we have hardware acceleration
	217	* is enable we will use that, otherwise a software version of the algorithm.
	218	* Note that algorithm names must be in lower case.
460408ef SG	219	*/
460408ef SG	220	static struct hash_algo hash_algo[] = {
fe54aeaa AG	221	#if I_WANT_MD5
	222	{
	223	.name = "md5",
	224	.digest_size = MD5_SUM_LEN,
	225	.chunk_size = CHUNKSZ_MD5,
	226	.hash_func_ws = md5_wd,
	227	},
	228	#endif
78eda89e	229	#ifdef CONFIG_SHA1
1f9c9280	230	{
0cf207ec	231	.name = "sha1",
78eda89e TR	232	.digest_size = SHA1_SUM_LEN,
	233	.chunk_size = CHUNKSZ_SHA1,
	234	#ifdef CONFIG_SHA_HW_ACCEL
	235	.hash_func_ws = hw_sha1,
	236	#else
	237	.hash_func_ws = sha1_csum_wd,
94e3c8c4	238	#endif
94e3c8c4	239	#ifdef CONFIG_SHA_PROG_HW_ACCEL
78eda89e TR	240	.hash_init = hw_sha_init,
	241	.hash_update = hw_sha_update,
	242	.hash_finish = hw_sha_finish,
	243	#else
	244	.hash_init = hash_init_sha1,
	245	.hash_update = hash_update_sha1,
	246	.hash_finish = hash_finish_sha1,
1f9c9280	247	#endif
460408ef SG	248	},
	249	#endif
	250	#ifdef CONFIG_SHA256
	251	{
78eda89e TR	252	.name = "sha256",
	253	.digest_size = SHA256_SUM_LEN,
	254	.chunk_size = CHUNKSZ_SHA256,
	255	#ifdef CONFIG_SHA_HW_ACCEL
	256	.hash_func_ws = hw_sha256,
	257	#else
	258	.hash_func_ws = sha256_csum_wd,
	259	#endif
	260	#ifdef CONFIG_SHA_PROG_HW_ACCEL
	261	.hash_init = hw_sha_init,
	262	.hash_update = hw_sha_update,
	263	.hash_finish = hw_sha_finish,
	264	#else
	265	.hash_init = hash_init_sha256,
	266	.hash_update = hash_update_sha256,
	267	.hash_finish = hash_finish_sha256,
	268	#endif
460408ef	269	},
d16b38f4 RD	270	#endif
	271	#ifdef CONFIG_SHA384
	272	{
	273	.name = "sha384",
	274	.digest_size = SHA384_SUM_LEN,
	275	.chunk_size = CHUNKSZ_SHA384,
e2ae483c	276	#ifdef CONFIG_SHA512_HW_ACCEL
a479f103 JS	277	.hash_func_ws = hw_sha384,
a479f103 JS	278	#else
d16b38f4	279	.hash_func_ws = sha384_csum_wd,
a479f103	280	#endif
e2ae483c	281	#if defined(CONFIG_SHA512_HW_ACCEL) && defined(CONFIG_SHA_PROG_HW_ACCEL)
a479f103 JS	282	.hash_init = hw_sha_init,
	283	.hash_update = hw_sha_update,
	284	.hash_finish = hw_sha_finish,
	285	#else
d16b38f4 RD	286	.hash_init = hash_init_sha384,
	287	.hash_update = hash_update_sha384,
	288	.hash_finish = hash_finish_sha384,
a479f103	289	#endif
d16b38f4 RD	290	},
	291	#endif
	292	#ifdef CONFIG_SHA512
	293	{
	294	.name = "sha512",
	295	.digest_size = SHA512_SUM_LEN,
	296	.chunk_size = CHUNKSZ_SHA512,
e2ae483c	297	#ifdef CONFIG_SHA512_HW_ACCEL
a479f103 JS	298	.hash_func_ws = hw_sha512,
a479f103 JS	299	#else
d16b38f4	300	.hash_func_ws = sha512_csum_wd,
a479f103	301	#endif
e2ae483c	302	#if defined(CONFIG_SHA512_HW_ACCEL) && defined(CONFIG_SHA_PROG_HW_ACCEL)
a479f103 JS	303	.hash_init = hw_sha_init,
	304	.hash_update = hw_sha_update,
	305	.hash_finish = hw_sha_finish,
	306	#else
d16b38f4 RD	307	.hash_init = hash_init_sha512,
	308	.hash_update = hash_update_sha512,
	309	.hash_finish = hash_finish_sha512,
a479f103	310	#endif
d16b38f4	311	},
460408ef	312	#endif
51c2345b PT	313	{
	314	.name = "crc16-ccitt",
	315	.digest_size = 2,
	316	.chunk_size = CHUNKSZ,
	317	.hash_func_ws = crc16_ccitt_wd_buf,
	318	.hash_init = hash_init_crc16_ccitt,
	319	.hash_update = hash_update_crc16_ccitt,
	320	.hash_finish = hash_finish_crc16_ccitt,
	321	},
d20a40de	322	{
78eda89e TR	323	.name = "crc32",
	324	.digest_size = 4,
	325	.chunk_size = CHUNKSZ_CRC32,
	326	.hash_func_ws = crc32_wd_buf,
	327	.hash_init = hash_init_crc32,
	328	.hash_update = hash_update_crc32,
	329	.hash_finish = hash_finish_crc32,
d20a40de	330	},
460408ef SG	331	};
460408ef SG	332
d20a40de	333	/* Try to minimize code size for boards that don't want much hashing */
221a949e	334	#if defined(CONFIG_SHA256) \|\| defined(CONFIG_CMD_SHA1SUM) \|\| \
d16b38f4 RD	335	defined(CONFIG_CRC32_VERIFY) \|\| defined(CONFIG_CMD_HASH) \|\| \
d16b38f4 RD	336	defined(CONFIG_SHA384) \|\| defined(CONFIG_SHA512)
d20a40de SG	337	#define multi_hash() 1
	338	#else
	339	#define multi_hash() 0
	340	#endif
	341
10088fb0 KR	342	static void reloc_update(void)
	343	{
	344	#if !defined(USE_HOSTCC) && defined(CONFIG_NEEDS_MANUAL_RELOC)
	345	int i;
	346	static bool done;
	347
	348	if (!done) {
	349	done = true;
	350	for (i = 0; i < ARRAY_SIZE(hash_algo); i++) {
	351	hash_algo[i].name += gd->reloc_off;
	352	hash_algo[i].hash_func_ws += gd->reloc_off;
	353	hash_algo[i].hash_init += gd->reloc_off;
	354	hash_algo[i].hash_update += gd->reloc_off;
	355	hash_algo[i].hash_finish += gd->reloc_off;
	356	}
	357	}
	358	#endif
	359	}
	360
2dd90027 RG	361	int hash_lookup_algo(const char algo_name, struct hash_algo *algop)
	362	{
	363	int i;
	364
10088fb0 KR	365	reloc_update();
10088fb0 KR	366
2dd90027 RG	367	for (i = 0; i < ARRAY_SIZE(hash_algo); i++) {
	368	if (!strcmp(algo_name, hash_algo[i].name)) {
	369	*algop = &hash_algo[i];
	370	return 0;
	371	}
	372	}
	373
	374	debug("Unknown hash algorithm '%s'\n", algo_name);
	375	return -EPROTONOSUPPORT;
	376	}
	377
	378	int hash_progressive_lookup_algo(const char *algo_name,
	379	struct hash_algo **algop)
	380	{
	381	int i;
	382
10088fb0 KR	383	reloc_update();
10088fb0 KR	384
2dd90027 RG	385	for (i = 0; i < ARRAY_SIZE(hash_algo); i++) {
	386	if (!strcmp(algo_name, hash_algo[i].name)) {
	387	if (hash_algo[i].hash_init) {
	388	*algop = &hash_algo[i];
	389	return 0;
	390	}
	391	}
	392	}
	393
	394	debug("Unknown hash algorithm '%s'\n", algo_name);
	395	return -EPROTONOSUPPORT;
	396	}
	397
	398	#ifndef USE_HOSTCC
8f0b1e24 SR	399	int hash_parse_string(const char algo_name, const char str, uint8_t *result)
	400	{
	401	struct hash_algo *algo;
	402	int ret;
	403	int i;
	404
	405	ret = hash_lookup_algo(algo_name, &algo);
	406	if (ret)
	407	return ret;
	408
	409	for (i = 0; i < algo->digest_size; i++) {
	410	char chr[3];
	411
031725f8	412	strlcpy(chr, &str[i * 2], 3);
7e5f460e	413	result[i] = hextoul(chr, NULL);
8f0b1e24 SR	414	}
	415
	416	return 0;
	417	}
	418
48ad68de TR	419	int hash_block(const char algo_name, const void data, unsigned int len,
	420	uint8_t output, int output_size)
	421	{
	422	struct hash_algo *algo;
	423	int ret;
	424
	425	ret = hash_lookup_algo(algo_name, &algo);
	426	if (ret)
	427	return ret;
	428
	429	if (output_size && *output_size < algo->digest_size) {
	430	debug("Output buffer size %d too small (need %d bytes)",
	431	*output_size, algo->digest_size);
	432	return -ENOSPC;
	433	}
	434	if (output_size)
	435	*output_size = algo->digest_size;
	436	algo->hash_func_ws(data, len, output, algo->chunk_size);
	437
	438	return 0;
	439	}
	440
	441	#if defined(CONFIG_CMD_HASH) \|\| defined(CONFIG_CMD_SHA1SUM) \|\| defined(CONFIG_CMD_CRC32)
460408ef SG	442	/**
	443	* store_result: Store the resulting sum to an address or variable
	444	*
	445	* @algo: Hash algorithm being used
	446	* @sum: Hash digest (algo->digest_size bytes)
	447	* @dest: Destination, interpreted as a hex address if it starts
d5b76673 SG	448	* with * (or allow_env_vars is 0) or otherwise as an
	449	* environment variable.
	450	* @allow_env_vars: non-zero to permit storing the result to an
	451	* variable environment
460408ef	452	*/
04819a4f	453	static void store_result(struct hash_algo algo, const uint8_t sum,
d5b76673	454	const char *dest, int allow_env_vars)
460408ef SG	455	{
460408ef SG	456	unsigned int i;
d5b76673	457	int env_var = 0;
460408ef	458
d5b76673 SG	459	/*
	460	* If environment variables are allowed, then we assume that 'dest'
	461	* is an environment variable, unless it starts with *, in which
	462	* case we assume it is an address. If not allowed, it is always an
	463	* address. This is to support the crc32 command.
	464	*/
	465	if (allow_env_vars) {
	466	if (dest == '')
	467	dest++;
	468	else
	469	env_var = 1;
	470	}
460408ef	471
d5b76673	472	if (env_var) {
460408ef SG	473	char str_output[HASH_MAX_DIGEST_SIZE * 2 + 1];
	474	char *str_ptr = str_output;
	475
	476	for (i = 0; i < algo->digest_size; i++) {
	477	sprintf(str_ptr, "%02x", sum[i]);
	478	str_ptr += 2;
	479	}
8b9cc866	480	*str_ptr = '\0';
382bee57	481	env_set(dest, str_output);
d5b76673	482	} else {
bd091b67 SG	483	ulong addr;
bd091b67 SG	484	void *buf;
d5b76673	485
7e5f460e	486	addr = hextoul(dest, NULL);
bd091b67 SG	487	buf = map_sysmem(addr, algo->digest_size);
	488	memcpy(buf, sum, algo->digest_size);
	489	unmap_sysmem(buf);
460408ef SG	490	}
	491	}
	492
	493	/**
	494	* parse_verify_sum: Parse a hash verification parameter
	495	*
	496	* @algo: Hash algorithm being used
	497	* @verify_str: Argument to parse. If it starts with * then it is
	498	* interpreted as a hex address containing the hash.
	499	* If the length is exactly the right number of hex digits
	500	* for the digest size, then we assume it is a hex digest.
	501	* Otherwise we assume it is an environment variable, and
	502	* look up its value (it must contain a hex digest).
	503	* @vsum: Returns binary digest value (algo->digest_size bytes)
d5b76673 SG	504	* @allow_env_vars: non-zero to permit storing the result to an environment
	505	* variable. If 0 then verify_str is assumed to be an
	506	* address, and the * prefix is not expected.
460408ef SG	507	* @return 0 if ok, non-zero on error
460408ef SG	508	*/
04819a4f SG	509	static int parse_verify_sum(struct hash_algo algo, char verify_str,
04819a4f SG	510	uint8_t *vsum, int allow_env_vars)
460408ef	511	{
d5b76673 SG	512	int env_var = 0;
	513
	514	/* See comment above in store_result() */
	515	if (allow_env_vars) {
	516	if (verify_str == '')
	517	verify_str++;
	518	else
	519	env_var = 1;
	520	}
	521
3ef46a99	522	if (!env_var) {
bd091b67 SG	523	ulong addr;
bd091b67 SG	524	void *buf;
460408ef	525
7e5f460e	526	addr = hextoul(verify_str, NULL);
bd091b67 SG	527	buf = map_sysmem(addr, algo->digest_size);
bd091b67 SG	528	memcpy(vsum, buf, algo->digest_size);
460408ef	529	} else {
460408ef SG	530	char *vsum_str;
	531	int digits = algo->digest_size * 2;
	532
	533	/*
	534	* As with the original code from sha1sum.c, we assume that a
	535	* string which matches the digest size exactly is a hex
	536	* string and not an environment variable.
	537	*/
	538	if (strlen(verify_str) == digits)
	539	vsum_str = verify_str;
	540	else {
00caae6d	541	vsum_str = env_get(verify_str);
460408ef SG	542	if (vsum_str == NULL \|\| strlen(vsum_str) != digits) {
	543	printf("Expected %d hex digits in env var\n",
	544	digits);
	545	return 1;
	546	}
	547	}
	548
8f0b1e24	549	hash_parse_string(algo->name, vsum_str, vsum);
460408ef SG	550	}
	551	return 0;
	552	}
	553
48ad68de	554	static void hash_show(struct hash_algo algo, ulong addr, ulong len, uint8_t output)
460408ef SG	555	{
	556	int i;
	557
	558	printf("%s for %08lx ... %08lx ==> ", algo->name, addr, addr + len - 1);
	559	for (i = 0; i < algo->digest_size; i++)
	560	printf("%02x", output[i]);
	561	}
	562
09140113 SG	563	int hash_command(const char algo_name, int flags, struct cmd_tbl cmdtp,
09140113 SG	564	int flag, int argc, char *const argv[])
460408ef	565	{
460408ef	566	ulong addr, len;
460408ef	567
3ef46a99	568	if ((argc < 2) \|\| ((flags & HASH_FLAG_VERIFY) && (argc < 3)))
460408ef SG	569	return CMD_RET_USAGE;
460408ef SG	570
7e5f460e SG	571	addr = hextoul(*argv++, NULL);
7e5f460e SG	572	len = hextoul(*argv++, NULL);
d5b76673	573
d20a40de SG	574	if (multi_hash()) {
d20a40de SG	575	struct hash_algo *algo;
d7af2baa	576	u8 *output;
04819a4f	577	uint8_t vsum[HASH_MAX_DIGEST_SIZE];
bd091b67	578	void *buf;
460408ef	579
bf007ebb	580	if (hash_lookup_algo(algo_name, &algo)) {
d20a40de SG	581	printf("Unknown hash algorithm '%s'\n", algo_name);
	582	return CMD_RET_USAGE;
	583	}
	584	argc -= 2;
	585
	586	if (algo->digest_size > HASH_MAX_DIGEST_SIZE) {
	587	puts("HASH_MAX_DIGEST_SIZE exceeded\n");
	588	return 1;
	589	}
460408ef	590
d7af2baa BL	591	output = memalign(ARCH_DMA_MINALIGN,
	592	sizeof(uint32_t) * HASH_MAX_DIGEST_SIZE);
	593
bd091b67 SG	594	buf = map_sysmem(addr, len);
	595	algo->hash_func_ws(buf, len, output, algo->chunk_size);
	596	unmap_sysmem(buf);
460408ef	597
d20a40de	598	/* Try to avoid code bloat when verify is not needed */
221a949e DT	599	#if defined(CONFIG_CRC32_VERIFY) \|\| defined(CONFIG_SHA1SUM_VERIFY) \|\| \
221a949e DT	600	defined(CONFIG_HASH_VERIFY)
d20a40de	601	if (flags & HASH_FLAG_VERIFY) {
460408ef	602	#else
d20a40de	603	if (0) {
460408ef	604	#endif
d20a40de	605	if (parse_verify_sum(algo, *argv, vsum,
d5b76673	606	flags & HASH_FLAG_ENV)) {
d20a40de SG	607	printf("ERROR: %s does not contain a valid "
	608	"%s sum\n", *argv, algo->name);
	609	return 1;
	610	}
	611	if (memcmp(output, vsum, algo->digest_size) != 0) {
	612	int i;
460408ef	613
31890ae2	614	hash_show(algo, addr, len, output);
d20a40de SG	615	printf(" != ");
	616	for (i = 0; i < algo->digest_size; i++)
	617	printf("%02x", vsum[i]);
	618	puts(" ERROR \n");
	619	return 1;
	620	}
	621	} else {
31890ae2	622	hash_show(algo, addr, len, output);
d20a40de SG	623	printf("\n");
	624
	625	if (argc) {
	626	store_result(algo, output, *argv,
	627	flags & HASH_FLAG_ENV);
	628	}
d7af2baa BL	629	unmap_sysmem(output);
d7af2baa BL	630
460408ef	631	}
d20a40de SG	632
d20a40de SG	633	/* Horrible code size hack for boards that just want crc32 */
460408ef	634	} else {
d20a40de SG	635	ulong crc;
	636	ulong *ptr;
	637
	638	crc = crc32_wd(0, (const uchar *)addr, len, CHUNKSZ_CRC32);
	639
	640	printf("CRC32 for %08lx ... %08lx ==> %08lx\n",
	641	addr, addr + len - 1, crc);
460408ef	642
4b756b01	643	if (argc >= 3) {
7e5f460e	644	ptr = (ulong *)hextoul(argv[0], NULL);
d20a40de	645	*ptr = crc;
d5b76673	646	}
460408ef SG	647	}
	648
	649	return 0;
	650	}
d70f919e SG	651	#endif /* CONFIG_CMD_HASH \|\| CONFIG_CMD_SHA1SUM \|\| CONFIG_CMD_CRC32) */
d70f919e SG	652	#endif /* !USE_HOSTCC */