Subversion Repositories planix.SVN

#include <u.h>

#include <libc.h>

#include <bio.h>

#include <libsec.h>

#pragma	varargck	type	"M"	uchar*

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){

		fprint(2, "reading %s: %r\n", name ? name : "stdin");

		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){

			fprint(2, "md5sum: can't open %s: %r\n", argv[i]);

			continue;

		}

		sum(fd, argv[i]);

		close(fd);

	}

	exits(nil);

}