9
0
Fork 0

commands: add md5/sha1/sha256sum commands using the digest api

The interface emulates the Linux commands, except that you can
specify a sub area - E.G.:

barebox:/ md5sum /dev/fd0 2M+1M /env/config /env/bin/boot 10+2
61c4c0180b044191d28f27545f43562f  /dev/fd0	0x00200000 ... 0x00300000
908b84bcbadd2f263583a65ff31d1cad  /env/config	0x00000000 ... 0x000003a7
f23bd15825cc5006cf5f9fd486d82d2d  /env/bin/boot	0x0000000a ... 0x0000000c

Adds around 1400 bytes (+ size of digest code) with everything enabled.

Signed-off-by: Peter Korsgaard <jacmet@sunsite.dk>
Signed-off-by: Sascha Hauer <s.hauer@pengutronix.de>
This commit is contained in:
Peter Korsgaard 2011-05-18 09:18:20 +02:00 committed by Sascha Hauer
parent c2bd07f6a9
commit 373de3c1c4
3 changed files with 206 additions and 0 deletions

View File

@ -222,6 +222,28 @@ config CMD_CRC_CMP
depends on CMD_CRC
prompt "compare 2 files crc"
config CMD_DIGEST
tristate
select DIGEST
config CMD_MD5SUM
tristate
select CMD_DIGEST
select MD5
prompt "md5sum"
config CMD_SHA1SUM
tristate
select CMD_DIGEST
select SHA1
prompt "sha1sum"
config CMD_SHA256SUM
tristate
select CMD_DIGEST
select SHA256
prompt "sha256sum"
config CMD_MTEST
tristate
prompt "mtest"

View File

@ -26,6 +26,7 @@ obj-$(CONFIG_CMD_MOUNT) += mount.o
obj-$(CONFIG_CMD_UMOUNT) += umount.o
obj-$(CONFIG_CMD_REGINFO) += reginfo.o
obj-$(CONFIG_CMD_CRC) += crc.o
obj-$(CONFIG_CMD_DIGEST) += digest.o
obj-$(CONFIG_CMD_CLEAR) += clear.o
obj-$(CONFIG_CMD_TEST) += test.o
obj-$(CONFIG_CMD_FLASH) += flash.o

183
commands/digest.c Normal file
View File

@ -0,0 +1,183 @@
/*
* digest.c - Calculate a md5/sha1/sha256 checksum of a memory area
*
* Copyright (c) 2011 Peter Korsgaard <jacmet@sunsite.dk>
*
* See file CREDITS for list of people who contributed to this
* project.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
#include <common.h>
#include <command.h>
#include <fs.h>
#include <fcntl.h>
#include <errno.h>
#include <xfuncs.h>
#include <malloc.h>
#include <digest.h>
static int file_digest(struct digest *d, char *filename,
ulong start, ulong size)
{
ulong len = 0;
int fd, now, i, ret = 0;
unsigned char *buf;
d->init(d);
fd = open(filename, O_RDONLY);
if (fd < 0) {
perror(filename);
return fd;
}
if (start > 0) {
ret = lseek(fd, start, SEEK_SET);
if (ret == -1) {
perror("lseek");
goto out;
}
}
buf = xmalloc(4096);
while (size) {
now = min((ulong)4096, size);
now = read(fd, buf, now);
if (now < 0) {
ret = now;
perror("read");
goto out_free;
}
if (!now)
break;
if (ctrlc()) {
ret = -EINTR;
goto out_free;
}
d->update(d, buf, now);
size -= now;
len += now;
}
d->final(d, buf);
for (i = 0; i < d->length; i++)
printf("%02x", buf[i]);
printf(" %s\t0x%08lx ... 0x%08lx\n", filename, start, start + len);
out_free:
free(buf);
out:
close(fd);
return ret;
}
static int do_digest(char *algorithm, int argc, char *argv[])
{
struct digest *d;
int ret = 0;
d = digest_get_by_name(algorithm);
BUG_ON(!d);
if (argc < 2)
return COMMAND_ERROR_USAGE;
argv++;
while (*argv) {
char *filename = "/dev/mem";
ulong start = 0, size = ~0;
/* arguments are either file, file+area or area */
if (parse_area_spec(*argv, &start, &size)) {
filename = *argv;
if (argv[1] && !parse_area_spec(argv[1], &start, &size))
argv++;
}
if (file_digest(d, filename, start, size) < 0)
ret = 1;
argv++;
}
return ret;
}
#ifdef CONFIG_CMD_MD5SUM
static int do_md5(struct command *cmdtp, int argc, char *argv[])
{
return do_digest("md5", argc, argv);
}
BAREBOX_CMD_HELP_START(md5sum)
BAREBOX_CMD_HELP_USAGE("md5sum [[FILE] [AREA]]...\n")
BAREBOX_CMD_HELP_SHORT("Calculate a md5 checksum of a memory area.\n")
BAREBOX_CMD_HELP_END
BAREBOX_CMD_START(md5sum)
.cmd = do_md5,
.usage = "md5 checksum calculation",
BAREBOX_CMD_HELP(cmd_md5sum_help)
BAREBOX_CMD_END
#endif /* CMD_CMD_MD5SUM */
#ifdef CONFIG_CMD_SHA1SUM
static int do_sha1(struct command *cmdtp, int argc, char *argv[])
{
return do_digest("sha1", argc, argv);
}
BAREBOX_CMD_HELP_START(sha1sum)
BAREBOX_CMD_HELP_USAGE("sha1sum [[FILE] [AREA]]...\n")
BAREBOX_CMD_HELP_SHORT("Calculate a sha1 checksum of a memory area.\n")
BAREBOX_CMD_HELP_END
BAREBOX_CMD_START(sha1sum)
.cmd = do_sha1,
.usage = "sha1 checksum calculation",
BAREBOX_CMD_HELP(cmd_sha1sum_help)
BAREBOX_CMD_END
#endif /* CMD_CMD_SHA1SUM */
#ifdef CONFIG_CMD_SHA256SUM
static int do_sha256(struct command *cmdtp, int argc, char *argv[])
{
return do_digest("sha256", argc, argv);
}
BAREBOX_CMD_HELP_START(sha256sum)
BAREBOX_CMD_HELP_USAGE("sha256sum [[FILE] [AREA]]...\n")
BAREBOX_CMD_HELP_SHORT("Calculate a sha256 checksum of a memory area.\n")
BAREBOX_CMD_HELP_END
BAREBOX_CMD_START(sha256sum)
.cmd = do_sha256,
.usage = "sha256 checksum calculation",
BAREBOX_CMD_HELP(cmd_sha256sum_help)
BAREBOX_CMD_END
#endif /* CMD_CMD_SHA256SUM */