| /* |
| * (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> |
| #include <asm/io.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; |
| void *buf; |
| |
| 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); |
| |
| buf = map_sysmem(addr, len); |
| md5_wd(buf, len, output, CHUNKSZ_MD5); |
| unmap_sysmem(buf); |
| |
| 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]; |
| void *buf; |
| |
| if (argc < 3) |
| return CMD_RET_USAGE; |
| |
| addr = simple_strtoul(argv[1], NULL, 16); |
| len = simple_strtoul(argv[2], NULL, 16); |
| |
| buf = map_sysmem(addr, len); |
| md5_wd(buf, len, output, CHUNKSZ_MD5); |
| unmap_sysmem(buf); |
| |
| 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 |