mirror of
https://github.com/brain-hackers/u-boot-brain
synced 2024-08-20 17:08:52 +09:00
![Jeroen Hofstee](/assets/img/avatar_default.png)
commit ecd729500
"Add parameter to md5sum to save the md5 sum"
adds support to build a string to be saved in the env and tries
to zero end it with str_ptr = '\0'; This does actually set the
pointer to the end of the buffer itself to zero. Since the
string was already zero terminated by the sprintf before it,
just remove the line, preventing a clang warning.
cc: Joe Hershberger <joe.hershberger@ni.com>
Signed-off-by: Jeroen Hofstee <jeroen@myspectrum.nl>
175 lines
3.4 KiB
C
175 lines
3.4 KiB
C
/*
|
|
* (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;
|
|
}
|
|
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("\n");
|
|
|
|
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\n",
|
|
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 **\n");
|
|
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("\n");
|
|
|
|
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]\n"
|
|
" - compute MD5 message digest [save to sum]\n"
|
|
"md5sum -v address count [*]sum\n"
|
|
" - verify md5sum of memory area"
|
|
);
|
|
#else
|
|
U_BOOT_CMD(
|
|
md5sum, 4, 1, do_md5sum,
|
|
"compute MD5 message digest",
|
|
"address count [[*]sum]\n"
|
|
" - compute MD5 message digest [save to sum]"
|
|
);
|
|
#endif
|