blob: 5ae3ddf85f82074960b7bdfbef83f9df5242da7a [file] [log] [blame]
Tom Rini83d290c2018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
Mike Frysingerc3d2a172011-04-02 21:40:19 -04002/*
Joe Hershberger5ab177b2012-08-17 11:00:47 +00003 * (C) Copyright 2011
4 * Joe Hershberger, National Instruments, joe.hershberger@ni.com
5 *
Mike Frysingerc3d2a172011-04-02 21:40:19 -04006 * (C) Copyright 2000
7 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
Mike Frysingerc3d2a172011-04-02 21:40:19 -04008 */
9
10#include <common.h>
11#include <command.h>
Simon Glass9fb625c2019-08-01 09:46:51 -060012#include <env.h>
Simon Glass4d72caa2020-05-10 11:40:01 -060013#include <image.h>
Joe Hershberger0eb25b62015-03-22 17:08:59 -050014#include <mapmem.h>
Mike Frysingerc3d2a172011-04-02 21:40:19 -040015#include <u-boot/md5.h>
Suriyan Ramasami79444952014-11-17 14:39:34 -080016#include <asm/io.h>
Mike Frysingerc3d2a172011-04-02 21:40:19 -040017
Joe Hershbergerecd72952012-08-17 11:00:48 +000018/*
19 * Store the resulting sum to an address or variable
20 */
21static void store_result(const u8 *sum, const char *dest)
22{
23 unsigned int i;
24
25 if (*dest == '*') {
26 u8 *ptr;
27
28 ptr = (u8 *)simple_strtoul(dest + 1, NULL, 16);
29 for (i = 0; i < 16; i++)
30 *ptr++ = sum[i];
31 } else {
32 char str_output[33];
33 char *str_ptr = str_output;
34
35 for (i = 0; i < 16; i++) {
36 sprintf(str_ptr, "%02x", sum[i]);
37 str_ptr += 2;
38 }
Simon Glass382bee52017-08-03 12:22:09 -060039 env_set(dest, str_output);
Joe Hershbergerecd72952012-08-17 11:00:48 +000040 }
41}
42
Joe Hershberger5ab177b2012-08-17 11:00:47 +000043#ifdef CONFIG_MD5SUM_VERIFY
44static int parse_verify_sum(char *verify_str, u8 *vsum)
45{
46 if (*verify_str == '*') {
47 u8 *ptr;
48
49 ptr = (u8 *)simple_strtoul(verify_str + 1, NULL, 16);
50 memcpy(vsum, ptr, 16);
51 } else {
52 unsigned int i;
53 char *vsum_str;
54
55 if (strlen(verify_str) == 32)
56 vsum_str = verify_str;
57 else {
Simon Glass00caae62017-08-03 12:22:12 -060058 vsum_str = env_get(verify_str);
Joe Hershberger5ab177b2012-08-17 11:00:47 +000059 if (vsum_str == NULL || strlen(vsum_str) != 32)
60 return 1;
61 }
62
63 for (i = 0; i < 16; i++) {
64 char *nullp = vsum_str + (i + 1) * 2;
65 char end = *nullp;
66
67 *nullp = '\0';
68 *(u8 *)(vsum + i) =
69 simple_strtoul(vsum_str + (i * 2), NULL, 16);
70 *nullp = end;
71 }
72 }
73 return 0;
74}
75
Simon Glass09140112020-05-10 11:40:03 -060076int do_md5sum(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
Joe Hershberger5ab177b2012-08-17 11:00:47 +000077{
78 ulong addr, len;
79 unsigned int i;
80 u8 output[16];
81 u8 vsum[16];
82 int verify = 0;
83 int ac;
84 char * const *av;
Suriyan Ramasami79444952014-11-17 14:39:34 -080085 void *buf;
Joe Hershberger5ab177b2012-08-17 11:00:47 +000086
87 if (argc < 3)
88 return CMD_RET_USAGE;
89
90 av = argv + 1;
91 ac = argc - 1;
92 if (strcmp(*av, "-v") == 0) {
93 verify = 1;
94 av++;
95 ac--;
96 if (ac < 3)
97 return CMD_RET_USAGE;
98 }
99
100 addr = simple_strtoul(*av++, NULL, 16);
101 len = simple_strtoul(*av++, NULL, 16);
102
Suriyan Ramasami79444952014-11-17 14:39:34 -0800103 buf = map_sysmem(addr, len);
104 md5_wd(buf, len, output, CHUNKSZ_MD5);
105 unmap_sysmem(buf);
Joe Hershberger5ab177b2012-08-17 11:00:47 +0000106
107 if (!verify) {
108 printf("md5 for %08lx ... %08lx ==> ", addr, addr + len - 1);
109 for (i = 0; i < 16; i++)
110 printf("%02x", output[i]);
111 printf("\n");
Joe Hershbergerecd72952012-08-17 11:00:48 +0000112
113 if (ac > 2)
114 store_result(output, *av);
Joe Hershberger5ab177b2012-08-17 11:00:47 +0000115 } else {
116 char *verify_str = *av++;
117
118 if (parse_verify_sum(verify_str, vsum)) {
119 printf("ERROR: %s does not contain a valid md5 sum\n",
120 verify_str);
121 return 1;
122 }
123 if (memcmp(output, vsum, 16) != 0) {
124 printf("md5 for %08lx ... %08lx ==> ", addr,
125 addr + len - 1);
126 for (i = 0; i < 16; i++)
127 printf("%02x", output[i]);
128 printf(" != ");
129 for (i = 0; i < 16; i++)
130 printf("%02x", vsum[i]);
131 printf(" ** ERROR **\n");
132 return 1;
133 }
134 }
135
136 return 0;
137}
138#else
Simon Glass09140112020-05-10 11:40:03 -0600139static int do_md5sum(struct cmd_tbl *cmdtp, int flag, int argc,
140 char *const argv[])
Mike Frysingerc3d2a172011-04-02 21:40:19 -0400141{
142 unsigned long addr, len;
143 unsigned int i;
144 u8 output[16];
Suriyan Ramasami79444952014-11-17 14:39:34 -0800145 void *buf;
Mike Frysingerc3d2a172011-04-02 21:40:19 -0400146
147 if (argc < 3)
Simon Glass4c12eeb2011-12-10 08:44:01 +0000148 return CMD_RET_USAGE;
Mike Frysingerc3d2a172011-04-02 21:40:19 -0400149
150 addr = simple_strtoul(argv[1], NULL, 16);
151 len = simple_strtoul(argv[2], NULL, 16);
152
Suriyan Ramasami79444952014-11-17 14:39:34 -0800153 buf = map_sysmem(addr, len);
154 md5_wd(buf, len, output, CHUNKSZ_MD5);
155 unmap_sysmem(buf);
156
Mike Frysingerc3d2a172011-04-02 21:40:19 -0400157 printf("md5 for %08lx ... %08lx ==> ", addr, addr + len - 1);
158 for (i = 0; i < 16; i++)
159 printf("%02x", output[i]);
160 printf("\n");
161
Joe Hershbergerecd72952012-08-17 11:00:48 +0000162 if (argc > 3)
163 store_result(output, argv[3]);
164
Mike Frysingerc3d2a172011-04-02 21:40:19 -0400165 return 0;
166}
Joe Hershberger5ab177b2012-08-17 11:00:47 +0000167#endif
Mike Frysingerc3d2a172011-04-02 21:40:19 -0400168
Joe Hershberger5ab177b2012-08-17 11:00:47 +0000169#ifdef CONFIG_MD5SUM_VERIFY
170U_BOOT_CMD(
171 md5sum, 5, 1, do_md5sum,
172 "compute MD5 message digest",
Joe Hershbergerecd72952012-08-17 11:00:48 +0000173 "address count [[*]sum]\n"
174 " - compute MD5 message digest [save to sum]\n"
Joe Hershberger5ab177b2012-08-17 11:00:47 +0000175 "md5sum -v address count [*]sum\n"
176 " - verify md5sum of memory area"
177);
178#else
Mike Frysingerc3d2a172011-04-02 21:40:19 -0400179U_BOOT_CMD(
Joe Hershbergerecd72952012-08-17 11:00:48 +0000180 md5sum, 4, 1, do_md5sum,
Mike Frysingerc3d2a172011-04-02 21:40:19 -0400181 "compute MD5 message digest",
Joe Hershbergerecd72952012-08-17 11:00:48 +0000182 "address count [[*]sum]\n"
183 " - compute MD5 message digest [save to sum]"
Mike Frysingerc3d2a172011-04-02 21:40:19 -0400184);
Joe Hershberger5ab177b2012-08-17 11:00:47 +0000185#endif