[people/ms/u-boot.git] / common / hash.c

/*
 * Copyright (c) 2012 The Chromium OS Authors.
 *
 * (C) Copyright 2011
 * Joe Hershberger, National Instruments, joe.hershberger@ni.com
 *
 * (C) Copyright 2000
 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

#include <common.h>
#include <command.h>
#include <malloc.h>
#include <hw_sha.h>
#include <hash.h>
#include <sha1.h>
#include <sha256.h>
#include <asm/io.h>
#include <asm/errno.h>

#ifdef CONFIG_CMD_SHA1SUM
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

#ifdef CONFIG_SHA256
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

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;
}

/*
 * These are the hash algorithms we support. Chips which support accelerated
 * crypto could perhaps add named version of these algorithms here. Note that
 * algorithm names must be in lower case.
 */
static struct hash_algo hash_algo[] = {
	/*
	 * CONFIG_SHA_HW_ACCEL is defined if hardware acceleration is
	 * available.
	 */
#ifdef CONFIG_SHA_HW_ACCEL
	{
		"sha1",
		SHA1_SUM_LEN,
		hw_sha1,
		CHUNKSZ_SHA1,
	}, {
		"sha256",
		SHA256_SUM_LEN,
		hw_sha256,
		CHUNKSZ_SHA256,
	},
#endif
	/*
	 * This is CONFIG_CMD_SHA1SUM instead of CONFIG_SHA1 since otherwise
	 * it bloats the code for boards which use SHA1 but not the 'hash'
	 * or 'sha1sum' commands.
	 */
#ifdef CONFIG_CMD_SHA1SUM
	{
		"sha1",
		SHA1_SUM_LEN,
		sha1_csum_wd,
		CHUNKSZ_SHA1,
		hash_init_sha1,
		hash_update_sha1,
		hash_finish_sha1,
	},
#define MULTI_HASH
#endif
#ifdef CONFIG_SHA256
	{
		"sha256",
		SHA256_SUM_LEN,
		sha256_csum_wd,
		CHUNKSZ_SHA256,
		hash_init_sha256,
		hash_update_sha256,
		hash_finish_sha256,
	},
#define MULTI_HASH
#endif
	{
		"crc32",
		4,
		crc32_wd_buf,
		CHUNKSZ_CRC32,
		hash_init_crc32,
		hash_update_crc32,
		hash_finish_crc32,
	},
};

#if defined(CONFIG_HASH_VERIFY) || defined(CONFIG_CMD_HASH)
#define MULTI_HASH
#endif

/* Try to minimize code size for boards that don't want much hashing */
#ifdef MULTI_HASH
#define multi_hash()	1
#else
#define multi_hash()	0
#endif

/**
 * 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 u8 *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';
		setenv(dest, str_output);
	} else {
		ulong addr;
		void *buf;

		addr = simple_strtoul(dest, NULL, 16);
		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, u8 *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 = simple_strtoul(verify_str, NULL, 16);
		buf = map_sysmem(addr, algo->digest_size);
		memcpy(vsum, buf, algo->digest_size);
	} else {
		unsigned int i;
		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 = getenv(verify_str);
			if (vsum_str == NULL || strlen(vsum_str) != digits) {
				printf("Expected %d hex digits in env var\n",
				       digits);
				return 1;
			}
		}

		for (i = 0; i < algo->digest_size; i++) {
			char *nullp = vsum_str + (i + 1) * 2;
			char end = *nullp;

			*nullp = '\0';
			vsum[i] = simple_strtoul(vsum_str + (i * 2), NULL, 16);
			*nullp = end;
		}
	}
	return 0;
}

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

	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;
}

static void show_hash(struct hash_algo *algo, ulong addr, ulong len,
		      u8 *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_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;
}

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

	if (argc < 2)
		return CMD_RET_USAGE;

	addr = simple_strtoul(*argv++, NULL, 16);
	len = simple_strtoul(*argv++, NULL, 16);

	if (multi_hash()) {
		struct hash_algo *algo;
		u8 output[HASH_MAX_DIGEST_SIZE];
		u8 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;
		}

		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 */
#ifdef CONFIG_HASH_VERIFY
		if (flags & HASH_FLAG_VERIFY) {
#else
		if (0) {
#endif
			if (!argc)
				return CMD_RET_USAGE;
			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;

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

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

	/* 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 *)simple_strtoul(argv[0], NULL, 16);
			*ptr = crc;
		}
	}

	return 0;
}
Commit	Line	Data
460408ef SG	1	/*
	2	* Copyright (c) 2012 The Chromium OS Authors.
	3	*
	4	* (C) Copyright 2011
	5	* Joe Hershberger, National Instruments, joe.hershberger@ni.com
	6	*
	7	* (C) Copyright 2000
	8	* Wolfgang Denk, DENX Software Engineering, wd@denx.de.
	9	*
1a459660	10	* SPDX-License-Identifier: GPL-2.0+
460408ef SG	11	*/
	12
	13	#include <common.h>
	14	#include <command.h>
bf007ebb	15	#include <malloc.h>
1f9c9280	16	#include <hw_sha.h>
460408ef SG	17	#include <hash.h>
	18	#include <sha1.h>
	19	#include <sha256.h>
bd091b67	20	#include <asm/io.h>
6f907b42	21	#include <asm/errno.h>
460408ef	22
bf007ebb HT	23	#ifdef CONFIG_CMD_SHA1SUM
	24	static int hash_init_sha1(struct hash_algo algo, void *ctxp)
	25	{
	26	sha1_context *ctx = malloc(sizeof(sha1_context));
	27	sha1_starts(ctx);
	28	*ctxp = ctx;
	29	return 0;
	30	}
	31
	32	static int hash_update_sha1(struct hash_algo algo, void ctx, const void *buf,
	33	unsigned int size, int is_last)
	34	{
	35	sha1_update((sha1_context *)ctx, buf, size);
	36	return 0;
	37	}
	38
	39	static int hash_finish_sha1(struct hash_algo algo, void ctx, void *dest_buf,
	40	int size)
	41	{
	42	if (size < algo->digest_size)
	43	return -1;
	44
	45	sha1_finish((sha1_context *)ctx, dest_buf);
	46	free(ctx);
	47	return 0;
	48	}
	49	#endif
	50
	51	#ifdef CONFIG_SHA256
	52	static int hash_init_sha256(struct hash_algo algo, void *ctxp)
	53	{
	54	sha256_context *ctx = malloc(sizeof(sha256_context));
	55	sha256_starts(ctx);
	56	*ctxp = ctx;
	57	return 0;
	58	}
	59
	60	static int hash_update_sha256(struct hash_algo algo, void ctx,
	61	const void *buf, unsigned int size, int is_last)
	62	{
	63	sha256_update((sha256_context *)ctx, buf, size);
	64	return 0;
	65	}
	66
	67	static int hash_finish_sha256(struct hash_algo algo, void ctx, void
	68	*dest_buf, int size)
	69	{
	70	if (size < algo->digest_size)
	71	return -1;
	72
	73	sha256_finish((sha256_context *)ctx, dest_buf);
	74	free(ctx);
	75	return 0;
	76	}
	77	#endif
	78
	79	static int hash_init_crc32(struct hash_algo algo, void *ctxp)
	80	{
	81	uint32_t *ctx = malloc(sizeof(uint32_t));
	82	*ctx = 0;
	83	*ctxp = ctx;
	84	return 0;
	85	}
	86
87	static int hash_update_crc32(struct hash_algo algo, void ctx,
88	const void *buf, unsigned int size, int is_last)
89	{
90	((uint32_t )ctx) = crc32(((uint32_t )ctx), buf, size);
91	return 0;
92	}
93
94	static int hash_finish_crc32(struct hash_algo algo, void ctx, void *dest_buf,
95	int size)
96	{
97	if (size < algo->digest_size)
98	return -1;
99
100	((uint32_t )dest_buf) = ((uint32_t )ctx);
101	free(ctx);
102	return 0;
103	}
104
460408ef SG	105	/*
460408ef SG	106	* These are the hash algorithms we support. Chips which support accelerated
218da0f3 SG	107	* crypto could perhaps add named version of these algorithms here. Note that
218da0f3 SG	108	* algorithm names must be in lower case.
460408ef SG	109	*/
460408ef SG	110	static struct hash_algo hash_algo[] = {
1f9c9280 AS	111	/*
	112	* CONFIG_SHA_HW_ACCEL is defined if hardware acceleration is
	113	* available.
	114	*/
	115	#ifdef CONFIG_SHA_HW_ACCEL
	116	{
	117	"sha1",
	118	SHA1_SUM_LEN,
	119	hw_sha1,
	120	CHUNKSZ_SHA1,
	121	}, {
	122	"sha256",
	123	SHA256_SUM_LEN,
	124	hw_sha256,
	125	CHUNKSZ_SHA256,
	126	},
	127	#endif
d20a40de SG	128	/*
	129	* This is CONFIG_CMD_SHA1SUM instead of CONFIG_SHA1 since otherwise
	130	* it bloats the code for boards which use SHA1 but not the 'hash'
	131	* or 'sha1sum' commands.
	132	*/
	133	#ifdef CONFIG_CMD_SHA1SUM
460408ef	134	{
218da0f3	135	"sha1",
460408ef SG	136	SHA1_SUM_LEN,
	137	sha1_csum_wd,
	138	CHUNKSZ_SHA1,
bf007ebb HT	139	hash_init_sha1,
	140	hash_update_sha1,
	141	hash_finish_sha1,
460408ef	142	},
d20a40de	143	#define MULTI_HASH
460408ef SG	144	#endif
	145	#ifdef CONFIG_SHA256
	146	{
218da0f3	147	"sha256",
460408ef SG	148	SHA256_SUM_LEN,
	149	sha256_csum_wd,
	150	CHUNKSZ_SHA256,
bf007ebb HT	151	hash_init_sha256,
	152	hash_update_sha256,
	153	hash_finish_sha256,
460408ef	154	},
d20a40de	155	#define MULTI_HASH
460408ef	156	#endif
d20a40de	157	{
218da0f3	158	"crc32",
d20a40de SG	159	4,
	160	crc32_wd_buf,
	161	CHUNKSZ_CRC32,
bf007ebb HT	162	hash_init_crc32,
	163	hash_update_crc32,
	164	hash_finish_crc32,
d20a40de	165	},
460408ef SG	166	};
460408ef SG	167
d20a40de SG	168	#if defined(CONFIG_HASH_VERIFY) \|\| defined(CONFIG_CMD_HASH)
	169	#define MULTI_HASH
	170	#endif
	171
	172	/* Try to minimize code size for boards that don't want much hashing */
	173	#ifdef MULTI_HASH
	174	#define multi_hash() 1
	175	#else
	176	#define multi_hash() 0
	177	#endif
	178
460408ef SG	179	/**
	180	* store_result: Store the resulting sum to an address or variable
	181	*
	182	* @algo: Hash algorithm being used
	183	* @sum: Hash digest (algo->digest_size bytes)
	184	* @dest: Destination, interpreted as a hex address if it starts
d5b76673 SG	185	* with * (or allow_env_vars is 0) or otherwise as an
	186	* environment variable.
	187	* @allow_env_vars: non-zero to permit storing the result to an
	188	* variable environment
460408ef SG	189	*/
460408ef SG	190	static void store_result(struct hash_algo algo, const u8 sum,
d5b76673	191	const char *dest, int allow_env_vars)
460408ef SG	192	{
460408ef SG	193	unsigned int i;
d5b76673	194	int env_var = 0;
460408ef	195
d5b76673 SG	196	/*
	197	* If environment variables are allowed, then we assume that 'dest'
	198	* is an environment variable, unless it starts with *, in which
	199	* case we assume it is an address. If not allowed, it is always an
	200	* address. This is to support the crc32 command.
	201	*/
	202	if (allow_env_vars) {
	203	if (dest == '')
	204	dest++;
	205	else
	206	env_var = 1;
	207	}
460408ef	208
d5b76673	209	if (env_var) {
460408ef SG	210	char str_output[HASH_MAX_DIGEST_SIZE * 2 + 1];
	211	char *str_ptr = str_output;
	212
	213	for (i = 0; i < algo->digest_size; i++) {
	214	sprintf(str_ptr, "%02x", sum[i]);
	215	str_ptr += 2;
	216	}
	217	str_ptr = '\0';
	218	setenv(dest, str_output);
d5b76673	219	} else {
bd091b67 SG	220	ulong addr;
bd091b67 SG	221	void *buf;
d5b76673	222
bd091b67 SG	223	addr = simple_strtoul(dest, NULL, 16);
	224	buf = map_sysmem(addr, algo->digest_size);
	225	memcpy(buf, sum, algo->digest_size);
	226	unmap_sysmem(buf);
460408ef SG	227	}
	228	}
	229
	230	/**
	231	* parse_verify_sum: Parse a hash verification parameter
	232	*
	233	* @algo: Hash algorithm being used
	234	* @verify_str: Argument to parse. If it starts with * then it is
	235	* interpreted as a hex address containing the hash.
	236	* If the length is exactly the right number of hex digits
	237	* for the digest size, then we assume it is a hex digest.
	238	* Otherwise we assume it is an environment variable, and
	239	* look up its value (it must contain a hex digest).
	240	* @vsum: Returns binary digest value (algo->digest_size bytes)
d5b76673 SG	241	* @allow_env_vars: non-zero to permit storing the result to an environment
	242	* variable. If 0 then verify_str is assumed to be an
	243	* address, and the * prefix is not expected.
460408ef SG	244	* @return 0 if ok, non-zero on error
460408ef SG	245	*/
d5b76673 SG	246	static int parse_verify_sum(struct hash_algo algo, char verify_str, u8 *vsum,
d5b76673 SG	247	int allow_env_vars)
460408ef	248	{
d5b76673 SG	249	int env_var = 0;
	250
	251	/* See comment above in store_result() */
	252	if (allow_env_vars) {
	253	if (verify_str == '')
	254	verify_str++;
	255	else
	256	env_var = 1;
	257	}
	258
	259	if (env_var) {
bd091b67 SG	260	ulong addr;
bd091b67 SG	261	void *buf;
460408ef	262
bd091b67 SG	263	addr = simple_strtoul(verify_str, NULL, 16);
	264	buf = map_sysmem(addr, algo->digest_size);
	265	memcpy(vsum, buf, algo->digest_size);
460408ef SG	266	} else {
	267	unsigned int i;
	268	char *vsum_str;
	269	int digits = algo->digest_size * 2;
	270
	271	/*
	272	* As with the original code from sha1sum.c, we assume that a
	273	* string which matches the digest size exactly is a hex
	274	* string and not an environment variable.
	275	*/
	276	if (strlen(verify_str) == digits)
	277	vsum_str = verify_str;
	278	else {
	279	vsum_str = getenv(verify_str);
	280	if (vsum_str == NULL \|\| strlen(vsum_str) != digits) {
	281	printf("Expected %d hex digits in env var\n",
	282	digits);
	283	return 1;
	284	}
	285	}
	286
	287	for (i = 0; i < algo->digest_size; i++) {
	288	char nullp = vsum_str + (i + 1) 2;
	289	char end = *nullp;
	290
	291	*nullp = '\0';
	292	vsum[i] = simple_strtoul(vsum_str + (i * 2), NULL, 16);
	293	*nullp = end;
	294	}
	295	}
	296	return 0;
	297	}
	298
bf007ebb	299	int hash_lookup_algo(const char algo_name, struct hash_algo *algop)
460408ef SG	300	{
	301	int i;
	302
	303	for (i = 0; i < ARRAY_SIZE(hash_algo); i++) {
bf007ebb HT	304	if (!strcmp(algo_name, hash_algo[i].name)) {
	305	*algop = &hash_algo[i];
	306	return 0;
	307	}
460408ef SG	308	}
460408ef SG	309
bf007ebb HT	310	debug("Unknown hash algorithm '%s'\n", algo_name);
bf007ebb HT	311	return -EPROTONOSUPPORT;
460408ef SG	312	}
	313
	314	static void show_hash(struct hash_algo *algo, ulong addr, ulong len,
	315	u8 *output)
	316	{
	317	int i;
	318
	319	printf("%s for %08lx ... %08lx ==> ", algo->name, addr, addr + len - 1);
	320	for (i = 0; i < algo->digest_size; i++)
	321	printf("%02x", output[i]);
	322	}
	323
6f907b42 SG	324	int hash_block(const char algo_name, const void data, unsigned int len,
	325	uint8_t output, int output_size)
	326	{
	327	struct hash_algo *algo;
bf007ebb HT	328	int ret;
	329
	330	ret = hash_lookup_algo(algo_name, &algo);
	331	if (ret)
	332	return ret;
6f907b42	333
6f907b42 SG	334	if (output_size && *output_size < algo->digest_size) {
	335	debug("Output buffer size %d too small (need %d bytes)",
	336	*output_size, algo->digest_size);
	337	return -ENOSPC;
	338	}
	339	if (output_size)
	340	*output_size = algo->digest_size;
	341	algo->hash_func_ws(data, len, output, algo->chunk_size);
	342
	343	return 0;
	344	}
	345
d5b76673	346	int hash_command(const char algo_name, int flags, cmd_tbl_t cmdtp, int flag,
460408ef SG	347	int argc, char * const argv[])
460408ef SG	348	{
460408ef	349	ulong addr, len;
460408ef SG	350
	351	if (argc < 2)
	352	return CMD_RET_USAGE;
	353
d5b76673 SG	354	addr = simple_strtoul(*argv++, NULL, 16);
	355	len = simple_strtoul(*argv++, NULL, 16);
	356
d20a40de SG	357	if (multi_hash()) {
	358	struct hash_algo *algo;
	359	u8 output[HASH_MAX_DIGEST_SIZE];
	360	u8 vsum[HASH_MAX_DIGEST_SIZE];
bd091b67	361	void *buf;
460408ef	362
bf007ebb	363	if (hash_lookup_algo(algo_name, &algo)) {
d20a40de SG	364	printf("Unknown hash algorithm '%s'\n", algo_name);
	365	return CMD_RET_USAGE;
	366	}
	367	argc -= 2;
	368
	369	if (algo->digest_size > HASH_MAX_DIGEST_SIZE) {
	370	puts("HASH_MAX_DIGEST_SIZE exceeded\n");
	371	return 1;
	372	}
460408ef	373
bd091b67 SG	374	buf = map_sysmem(addr, len);
	375	algo->hash_func_ws(buf, len, output, algo->chunk_size);
	376	unmap_sysmem(buf);
460408ef	377
d20a40de	378	/* Try to avoid code bloat when verify is not needed */
460408ef	379	#ifdef CONFIG_HASH_VERIFY
d20a40de	380	if (flags & HASH_FLAG_VERIFY) {
460408ef	381	#else
d20a40de	382	if (0) {
460408ef	383	#endif
d20a40de SG	384	if (!argc)
	385	return CMD_RET_USAGE;
	386	if (parse_verify_sum(algo, *argv, vsum,
d5b76673	387	flags & HASH_FLAG_ENV)) {
d20a40de SG	388	printf("ERROR: %s does not contain a valid "
	389	"%s sum\n", *argv, algo->name);
	390	return 1;
	391	}
	392	if (memcmp(output, vsum, algo->digest_size) != 0) {
	393	int i;
460408ef	394
d20a40de SG	395	show_hash(algo, addr, len, output);
	396	printf(" != ");
	397	for (i = 0; i < algo->digest_size; i++)
	398	printf("%02x", vsum[i]);
	399	puts(" ERROR \n");
	400	return 1;
	401	}
	402	} else {
460408ef	403	show_hash(algo, addr, len, output);
d20a40de SG	404	printf("\n");
	405
	406	if (argc) {
	407	store_result(algo, output, *argv,
	408	flags & HASH_FLAG_ENV);
	409	}
460408ef	410	}
d20a40de SG	411
d20a40de SG	412	/* Horrible code size hack for boards that just want crc32 */
460408ef	413	} else {
d20a40de SG	414	ulong crc;
	415	ulong *ptr;
	416
	417	crc = crc32_wd(0, (const uchar *)addr, len, CHUNKSZ_CRC32);
	418
	419	printf("CRC32 for %08lx ... %08lx ==> %08lx\n",
	420	addr, addr + len - 1, crc);
460408ef	421
4b756b01 TR	422	if (argc >= 3) {
4b756b01 TR	423	ptr = (ulong *)simple_strtoul(argv[0], NULL, 16);
d20a40de	424	*ptr = crc;
d5b76673	425	}
460408ef SG	426	}
	427
	428	return 0;
	429	}