17dd7cddfSDavid du Colombier #include <u.h>
27dd7cddfSDavid du Colombier #include <libc.h>
37dd7cddfSDavid du Colombier #include <bio.h>
459cc4ca5SDavid du Colombier #include <libsec.h>
57dd7cddfSDavid du Colombier
659cc4ca5SDavid du Colombier #pragma varargck type "M" uchar*
77dd7cddfSDavid du Colombier
859cc4ca5SDavid du Colombier static int
digestfmt(Fmt * fmt)99a747e4fSDavid du Colombier digestfmt(Fmt *fmt)
107dd7cddfSDavid du Colombier {
1159cc4ca5SDavid du Colombier char buf[MD5dlen*2+1];
1259cc4ca5SDavid du Colombier uchar *p;
1359cc4ca5SDavid du Colombier int i;
147dd7cddfSDavid du Colombier
159a747e4fSDavid du Colombier p = va_arg(fmt->args, uchar*);
1659cc4ca5SDavid du Colombier for(i=0; i<MD5dlen; i++)
1759cc4ca5SDavid du Colombier sprint(buf+2*i, "%.2ux", p[i]);
189a747e4fSDavid du Colombier return fmtstrcpy(fmt, buf);
197dd7cddfSDavid du Colombier }
207dd7cddfSDavid du Colombier
217dd7cddfSDavid du Colombier static void
sum(int fd,char * name)227dd7cddfSDavid du Colombier sum(int fd, char *name)
237dd7cddfSDavid du Colombier {
2459cc4ca5SDavid du Colombier int n;
2559cc4ca5SDavid du Colombier uchar buf[8192], digest[MD5dlen];
2659cc4ca5SDavid du Colombier DigestState *s;
277dd7cddfSDavid du Colombier
2859cc4ca5SDavid du Colombier s = md5(nil, 0, nil, nil);
2959cc4ca5SDavid du Colombier while((n = read(fd, buf, sizeof buf)) > 0)
3059cc4ca5SDavid du Colombier md5(buf, n, nil, s);
31*225077b0SDavid du Colombier if(n < 0){
32*225077b0SDavid du Colombier fprint(2, "reading %s: %r\n", name ? name : "stdin");
33*225077b0SDavid du Colombier return;
34*225077b0SDavid du Colombier }
3559cc4ca5SDavid du Colombier md5(nil, 0, digest, s);
3659cc4ca5SDavid du Colombier if(name == nil)
3759cc4ca5SDavid du Colombier print("%M\n", digest);
387dd7cddfSDavid du Colombier else
3959cc4ca5SDavid du Colombier print("%M\t%s\n", digest, name);
407dd7cddfSDavid du Colombier }
417dd7cddfSDavid du Colombier
4259cc4ca5SDavid du Colombier void
main(int argc,char * argv[])4359cc4ca5SDavid du Colombier main(int argc, char *argv[])
447dd7cddfSDavid du Colombier {
4559cc4ca5SDavid du Colombier int i, fd;
467dd7cddfSDavid du Colombier
4759cc4ca5SDavid du Colombier ARGBEGIN{
4859cc4ca5SDavid du Colombier default:
4959cc4ca5SDavid du Colombier fprint(2, "usage: md5sum [file...]\n");
5059cc4ca5SDavid du Colombier exits("usage");
5159cc4ca5SDavid du Colombier }ARGEND
5259cc4ca5SDavid du Colombier
539a747e4fSDavid du Colombier fmtinstall('M', digestfmt);
5459cc4ca5SDavid du Colombier
5559cc4ca5SDavid du Colombier if(argc == 0)
5659cc4ca5SDavid du Colombier sum(0, nil);
5759cc4ca5SDavid du Colombier else for(i = 0; i < argc; i++){
5859cc4ca5SDavid du Colombier fd = open(argv[i], OREAD);
5959cc4ca5SDavid du Colombier if(fd < 0){
6059cc4ca5SDavid du Colombier fprint(2, "md5sum: can't open %s: %r\n", argv[i]);
6159cc4ca5SDavid du Colombier continue;
627dd7cddfSDavid du Colombier }
6359cc4ca5SDavid du Colombier sum(fd, argv[i]);
6459cc4ca5SDavid du Colombier close(fd);
657dd7cddfSDavid du Colombier }
6659cc4ca5SDavid du Colombier exits(nil);
677dd7cddfSDavid du Colombier }
68