1*83d290c5STom Rini // SPDX-License-Identifier: GPL-2.0+
22e192b24SSimon Glass /*
32e192b24SSimon Glass * (C) Copyright 2011
42e192b24SSimon Glass * Joe Hershberger, National Instruments, joe.hershberger@ni.com
52e192b24SSimon Glass *
62e192b24SSimon Glass * (C) Copyright 2000
72e192b24SSimon Glass * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
82e192b24SSimon Glass */
92e192b24SSimon Glass
102e192b24SSimon Glass #include <common.h>
112e192b24SSimon Glass #include <command.h>
122e192b24SSimon Glass #include <mapmem.h>
132e192b24SSimon Glass #include <u-boot/md5.h>
142e192b24SSimon Glass #include <asm/io.h>
152e192b24SSimon Glass
162e192b24SSimon Glass /*
172e192b24SSimon Glass * Store the resulting sum to an address or variable
182e192b24SSimon Glass */
store_result(const u8 * sum,const char * dest)192e192b24SSimon Glass static void store_result(const u8 *sum, const char *dest)
202e192b24SSimon Glass {
212e192b24SSimon Glass unsigned int i;
222e192b24SSimon Glass
232e192b24SSimon Glass if (*dest == '*') {
242e192b24SSimon Glass u8 *ptr;
252e192b24SSimon Glass
262e192b24SSimon Glass ptr = (u8 *)simple_strtoul(dest + 1, NULL, 16);
272e192b24SSimon Glass for (i = 0; i < 16; i++)
282e192b24SSimon Glass *ptr++ = sum[i];
292e192b24SSimon Glass } else {
302e192b24SSimon Glass char str_output[33];
312e192b24SSimon Glass char *str_ptr = str_output;
322e192b24SSimon Glass
332e192b24SSimon Glass for (i = 0; i < 16; i++) {
342e192b24SSimon Glass sprintf(str_ptr, "%02x", sum[i]);
352e192b24SSimon Glass str_ptr += 2;
362e192b24SSimon Glass }
37382bee57SSimon Glass env_set(dest, str_output);
382e192b24SSimon Glass }
392e192b24SSimon Glass }
402e192b24SSimon Glass
412e192b24SSimon Glass #ifdef CONFIG_MD5SUM_VERIFY
parse_verify_sum(char * verify_str,u8 * vsum)422e192b24SSimon Glass static int parse_verify_sum(char *verify_str, u8 *vsum)
432e192b24SSimon Glass {
442e192b24SSimon Glass if (*verify_str == '*') {
452e192b24SSimon Glass u8 *ptr;
462e192b24SSimon Glass
472e192b24SSimon Glass ptr = (u8 *)simple_strtoul(verify_str + 1, NULL, 16);
482e192b24SSimon Glass memcpy(vsum, ptr, 16);
492e192b24SSimon Glass } else {
502e192b24SSimon Glass unsigned int i;
512e192b24SSimon Glass char *vsum_str;
522e192b24SSimon Glass
532e192b24SSimon Glass if (strlen(verify_str) == 32)
542e192b24SSimon Glass vsum_str = verify_str;
552e192b24SSimon Glass else {
5600caae6dSSimon Glass vsum_str = env_get(verify_str);
572e192b24SSimon Glass if (vsum_str == NULL || strlen(vsum_str) != 32)
582e192b24SSimon Glass return 1;
592e192b24SSimon Glass }
602e192b24SSimon Glass
612e192b24SSimon Glass for (i = 0; i < 16; i++) {
622e192b24SSimon Glass char *nullp = vsum_str + (i + 1) * 2;
632e192b24SSimon Glass char end = *nullp;
642e192b24SSimon Glass
652e192b24SSimon Glass *nullp = '\0';
662e192b24SSimon Glass *(u8 *)(vsum + i) =
672e192b24SSimon Glass simple_strtoul(vsum_str + (i * 2), NULL, 16);
682e192b24SSimon Glass *nullp = end;
692e192b24SSimon Glass }
702e192b24SSimon Glass }
712e192b24SSimon Glass return 0;
722e192b24SSimon Glass }
732e192b24SSimon Glass
do_md5sum(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])742e192b24SSimon Glass int do_md5sum(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
752e192b24SSimon Glass {
762e192b24SSimon Glass ulong addr, len;
772e192b24SSimon Glass unsigned int i;
782e192b24SSimon Glass u8 output[16];
792e192b24SSimon Glass u8 vsum[16];
802e192b24SSimon Glass int verify = 0;
812e192b24SSimon Glass int ac;
822e192b24SSimon Glass char * const *av;
832e192b24SSimon Glass void *buf;
842e192b24SSimon Glass
852e192b24SSimon Glass if (argc < 3)
862e192b24SSimon Glass return CMD_RET_USAGE;
872e192b24SSimon Glass
882e192b24SSimon Glass av = argv + 1;
892e192b24SSimon Glass ac = argc - 1;
902e192b24SSimon Glass if (strcmp(*av, "-v") == 0) {
912e192b24SSimon Glass verify = 1;
922e192b24SSimon Glass av++;
932e192b24SSimon Glass ac--;
942e192b24SSimon Glass if (ac < 3)
952e192b24SSimon Glass return CMD_RET_USAGE;
962e192b24SSimon Glass }
972e192b24SSimon Glass
982e192b24SSimon Glass addr = simple_strtoul(*av++, NULL, 16);
992e192b24SSimon Glass len = simple_strtoul(*av++, NULL, 16);
1002e192b24SSimon Glass
1012e192b24SSimon Glass buf = map_sysmem(addr, len);
1022e192b24SSimon Glass md5_wd(buf, len, output, CHUNKSZ_MD5);
1032e192b24SSimon Glass unmap_sysmem(buf);
1042e192b24SSimon Glass
1052e192b24SSimon Glass if (!verify) {
1062e192b24SSimon Glass printf("md5 for %08lx ... %08lx ==> ", addr, addr + len - 1);
1072e192b24SSimon Glass for (i = 0; i < 16; i++)
1082e192b24SSimon Glass printf("%02x", output[i]);
1092e192b24SSimon Glass printf("\n");
1102e192b24SSimon Glass
1112e192b24SSimon Glass if (ac > 2)
1122e192b24SSimon Glass store_result(output, *av);
1132e192b24SSimon Glass } else {
1142e192b24SSimon Glass char *verify_str = *av++;
1152e192b24SSimon Glass
1162e192b24SSimon Glass if (parse_verify_sum(verify_str, vsum)) {
1172e192b24SSimon Glass printf("ERROR: %s does not contain a valid md5 sum\n",
1182e192b24SSimon Glass verify_str);
1192e192b24SSimon Glass return 1;
1202e192b24SSimon Glass }
1212e192b24SSimon Glass if (memcmp(output, vsum, 16) != 0) {
1222e192b24SSimon Glass printf("md5 for %08lx ... %08lx ==> ", addr,
1232e192b24SSimon Glass addr + len - 1);
1242e192b24SSimon Glass for (i = 0; i < 16; i++)
1252e192b24SSimon Glass printf("%02x", output[i]);
1262e192b24SSimon Glass printf(" != ");
1272e192b24SSimon Glass for (i = 0; i < 16; i++)
1282e192b24SSimon Glass printf("%02x", vsum[i]);
1292e192b24SSimon Glass printf(" ** ERROR **\n");
1302e192b24SSimon Glass return 1;
1312e192b24SSimon Glass }
1322e192b24SSimon Glass }
1332e192b24SSimon Glass
1342e192b24SSimon Glass return 0;
1352e192b24SSimon Glass }
1362e192b24SSimon Glass #else
do_md5sum(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])1372e192b24SSimon Glass static int do_md5sum(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
1382e192b24SSimon Glass {
1392e192b24SSimon Glass unsigned long addr, len;
1402e192b24SSimon Glass unsigned int i;
1412e192b24SSimon Glass u8 output[16];
1422e192b24SSimon Glass void *buf;
1432e192b24SSimon Glass
1442e192b24SSimon Glass if (argc < 3)
1452e192b24SSimon Glass return CMD_RET_USAGE;
1462e192b24SSimon Glass
1472e192b24SSimon Glass addr = simple_strtoul(argv[1], NULL, 16);
1482e192b24SSimon Glass len = simple_strtoul(argv[2], NULL, 16);
1492e192b24SSimon Glass
1502e192b24SSimon Glass buf = map_sysmem(addr, len);
1512e192b24SSimon Glass md5_wd(buf, len, output, CHUNKSZ_MD5);
1522e192b24SSimon Glass unmap_sysmem(buf);
1532e192b24SSimon Glass
1542e192b24SSimon Glass printf("md5 for %08lx ... %08lx ==> ", addr, addr + len - 1);
1552e192b24SSimon Glass for (i = 0; i < 16; i++)
1562e192b24SSimon Glass printf("%02x", output[i]);
1572e192b24SSimon Glass printf("\n");
1582e192b24SSimon Glass
1592e192b24SSimon Glass if (argc > 3)
1602e192b24SSimon Glass store_result(output, argv[3]);
1612e192b24SSimon Glass
1622e192b24SSimon Glass return 0;
1632e192b24SSimon Glass }
1642e192b24SSimon Glass #endif
1652e192b24SSimon Glass
1662e192b24SSimon Glass #ifdef CONFIG_MD5SUM_VERIFY
1672e192b24SSimon Glass U_BOOT_CMD(
1682e192b24SSimon Glass md5sum, 5, 1, do_md5sum,
1692e192b24SSimon Glass "compute MD5 message digest",
1702e192b24SSimon Glass "address count [[*]sum]\n"
1712e192b24SSimon Glass " - compute MD5 message digest [save to sum]\n"
1722e192b24SSimon Glass "md5sum -v address count [*]sum\n"
1732e192b24SSimon Glass " - verify md5sum of memory area"
1742e192b24SSimon Glass );
1752e192b24SSimon Glass #else
1762e192b24SSimon Glass U_BOOT_CMD(
1772e192b24SSimon Glass md5sum, 4, 1, do_md5sum,
1782e192b24SSimon Glass "compute MD5 message digest",
1792e192b24SSimon Glass "address count [[*]sum]\n"
1802e192b24SSimon Glass " - compute MD5 message digest [save to sum]"
1812e192b24SSimon Glass );
1822e192b24SSimon Glass #endif
183