plan9fox/sys/src/cmd/md5sum.c

71 lines
1.2 KiB
C

#include <u.h>
#include <libc.h>
#include <libsec.h>
#pragma varargck type "M" uchar*
static char exitstr[ERRMAX];
static int
digestfmt(Fmt *fmt)
{
char buf[MD5dlen*2+1];
uchar *p;
int i;
p = va_arg(fmt->args, uchar*);
for(i=0; i<MD5dlen; i++)
sprint(buf+2*i, "%.2ux", p[i]);
return fmtstrcpy(fmt, buf);
}
static void
sum(int fd, char *name)
{
int n;
uchar buf[8192], digest[MD5dlen];
DigestState *s;
s = md5(nil, 0, nil, nil);
while((n = read(fd, buf, sizeof buf)) > 0)
md5(buf, n, nil, s);
if(n < 0){
snprint(exitstr, sizeof(exitstr), "reading %s: %r\n", name ? name : "stdin");
fprint(2, "%s", exitstr);
return;
}
md5(nil, 0, digest, s);
if(name == nil)
print("%M\n", digest);
else
print("%M\t%s\n", digest, name);
}
void
main(int argc, char *argv[])
{
int i, fd;
ARGBEGIN{
default:
fprint(2, "usage: md5sum [file...]\n");
exits("usage");
}ARGEND
fmtinstall('M', digestfmt);
if(argc == 0)
sum(0, nil);
else for(i = 0; i < argc; i++){
fd = open(argv[i], OREAD);
if(fd < 0){
snprint(exitstr, sizeof(exitstr), "can't open %s: %r", argv[i]);
fprint(2, "%s: %s\n", argv0, exitstr);
continue;
}
sum(fd, argv[i]);
close(fd);
}
exits(exitstr);
}