/*

   * (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 <u-boot/md5.h>

  /*
   * Store the resulting sum to an address or variable
   */
  static void store_result(const u8 *sum, const char *dest)
  {
  	unsigned int i;
  
  	if (*dest == '*') {
  		u8 *ptr;
  
  		ptr = (u8 *)simple_strtoul(dest + 1, NULL, 16);
  		for (i = 0; i < 16; i++)
  			*ptr++ = sum[i];
  	} else {
  		char str_output[33];
  		char *str_ptr = str_output;
  
  		for (i = 0; i < 16; i++) {
  			sprintf(str_ptr, "%02x", sum[i]);
  			str_ptr += 2;
  		}
  		str_ptr = '\0';
  		setenv(dest, str_output);
  	}
  }

  #ifdef CONFIG_MD5SUM_VERIFY
  static int parse_verify_sum(char *verify_str, u8 *vsum)
  {
  	if (*verify_str == '*') {
  		u8 *ptr;
  
  		ptr = (u8 *)simple_strtoul(verify_str + 1, NULL, 16);
  		memcpy(vsum, ptr, 16);
  	} else {
  		unsigned int i;
  		char *vsum_str;
  
  		if (strlen(verify_str) == 32)
  			vsum_str = verify_str;
  		else {
  			vsum_str = getenv(verify_str);
  			if (vsum_str == NULL || strlen(vsum_str) != 32)
  				return 1;
  		}
  
  		for (i = 0; i < 16; i++) {
  			char *nullp = vsum_str + (i + 1) * 2;
  			char end = *nullp;
  
  			*nullp = '\0';
  			*(u8 *)(vsum + i) =
  				simple_strtoul(vsum_str + (i * 2), NULL, 16);
  			*nullp = end;
  		}
  	}
  	return 0;
  }
  
  int do_md5sum(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  {
  	ulong addr, len;
  	unsigned int i;
  	u8 output[16];
  	u8 vsum[16];
  	int verify = 0;
  	int ac;
  	char * const *av;
  
  	if (argc < 3)
  		return CMD_RET_USAGE;
  
  	av = argv + 1;
  	ac = argc - 1;
  	if (strcmp(*av, "-v") == 0) {
  		verify = 1;
  		av++;
  		ac--;
  		if (ac < 3)
  			return CMD_RET_USAGE;
  	}
  
  	addr = simple_strtoul(*av++, NULL, 16);
  	len = simple_strtoul(*av++, NULL, 16);
  
  	md5_wd((unsigned char *) addr, len, output, CHUNKSZ_MD5);
  
  	if (!verify) {
  		printf("md5 for %08lx ... %08lx ==> ", addr, addr + len - 1);
  		for (i = 0; i < 16; i++)
  			printf("%02x", output[i]);
  		printf("
  ");

  
  		if (ac > 2)
  			store_result(output, *av);

  	} else {
  		char *verify_str = *av++;
  
  		if (parse_verify_sum(verify_str, vsum)) {
  			printf("ERROR: %s does not contain a valid md5 sum
  ",
  				verify_str);
  			return 1;
  		}
  		if (memcmp(output, vsum, 16) != 0) {
  			printf("md5 for %08lx ... %08lx ==> ", addr,
  				addr + len - 1);
  			for (i = 0; i < 16; i++)
  				printf("%02x", output[i]);
  			printf(" != ");
  			for (i = 0; i < 16; i++)
  				printf("%02x", vsum[i]);
  			printf(" ** ERROR **
  ");
  			return 1;
  		}
  	}
  
  	return 0;
  }
  #else

  static int do_md5sum(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  {
  	unsigned long addr, len;
  	unsigned int i;
  	u8 output[16];
  
  	if (argc < 3)

  		return CMD_RET_USAGE;

  
  	addr = simple_strtoul(argv[1], NULL, 16);
  	len = simple_strtoul(argv[2], NULL, 16);

  	md5_wd((unsigned char *) addr, len, output, CHUNKSZ_MD5);

  	printf("md5 for %08lx ... %08lx ==> ", addr, addr + len - 1);
  	for (i = 0; i < 16; i++)
  		printf("%02x", output[i]);
  	printf("
  ");

  	if (argc > 3)
  		store_result(output, argv[3]);

  	return 0;
  }

  #endif

  #ifdef CONFIG_MD5SUM_VERIFY
  U_BOOT_CMD(
  	md5sum,	5,	1,	do_md5sum,
  	"compute MD5 message digest",

  	"address count [[*]sum]
  "
  		"    - compute MD5 message digest [save to sum]
  "

  	"md5sum -v address count [*]sum
  "
  		"    - verify md5sum of memory area"
  );
  #else

  U_BOOT_CMD(

  	md5sum,	4,	1,	do_md5sum,

  	"compute MD5 message digest",

  	"address count [[*]sum]
  "
  		"    - compute MD5 message digest [save to sum]"

  );

  #endif