1 c9ed5273 2014-11-17 rsc #include <u.h>
 2 c9ed5273 2014-11-17 rsc #include <libc.h>
 3 c9ed5273 2014-11-17 rsc #include <bio.h>
 4 c9ed5273 2014-11-17 rsc #include <libsec.h>
 5 c9ed5273 2014-11-17 rsc 
 6 c9ed5273 2014-11-17 rsc static int
 7 c9ed5273 2014-11-17 rsc digestfmt(Fmt *fmt)
 8 c9ed5273 2014-11-17 rsc {
 9 c9ed5273 2014-11-17 rsc 	char buf[MD5dlen*2+1];
10 c9ed5273 2014-11-17 rsc 	uchar *p;
11 c9ed5273 2014-11-17 rsc 	int i;
12 c9ed5273 2014-11-17 rsc 
13 c9ed5273 2014-11-17 rsc 	p = va_arg(fmt->args, uchar*);
14 c9ed5273 2014-11-17 rsc 	for(i=0; i<MD5dlen; i++)
15 c9ed5273 2014-11-17 rsc 		sprint(buf+2*i, "%.2ux", p[i]);
16 c9ed5273 2014-11-17 rsc 	return fmtstrcpy(fmt, buf);
17 c9ed5273 2014-11-17 rsc }
18 c9ed5273 2014-11-17 rsc 
19 c9ed5273 2014-11-17 rsc static void
20 c9ed5273 2014-11-17 rsc sum(int fd, char *name)
21 c9ed5273 2014-11-17 rsc {
22 c9ed5273 2014-11-17 rsc 	int n;
23 c9ed5273 2014-11-17 rsc 	uchar buf[8192], digest[MD5dlen];
24 c9ed5273 2014-11-17 rsc 	DigestState *s;
25 c9ed5273 2014-11-17 rsc 
26 c9ed5273 2014-11-17 rsc 	s = md5(nil, 0, nil, nil);
27 c9ed5273 2014-11-17 rsc 	while((n = read(fd, buf, sizeof buf)) > 0)
28 c9ed5273 2014-11-17 rsc 		md5(buf, n, nil, s);
29 c9ed5273 2014-11-17 rsc 	md5(nil, 0, digest, s);
30 c9ed5273 2014-11-17 rsc 	if(name == nil)
31 c9ed5273 2014-11-17 rsc 		print("%M\n", digest);
32 c9ed5273 2014-11-17 rsc 	else
33 c9ed5273 2014-11-17 rsc 		print("%M\t%s\n", digest, name);
34 c9ed5273 2014-11-17 rsc }
35 c9ed5273 2014-11-17 rsc 
36 c9ed5273 2014-11-17 rsc void
37 c9ed5273 2014-11-17 rsc main(int argc, char *argv[])
38 c9ed5273 2014-11-17 rsc {
39 c9ed5273 2014-11-17 rsc 	int i, fd;
40 c9ed5273 2014-11-17 rsc 
41 c9ed5273 2014-11-17 rsc 	ARGBEGIN{
42 c9ed5273 2014-11-17 rsc 	default:
43 c9ed5273 2014-11-17 rsc 		fprint(2, "usage: md5sum [file...]\n");
44 c9ed5273 2014-11-17 rsc 		exits("usage");
45 c9ed5273 2014-11-17 rsc 	}ARGEND
46 c9ed5273 2014-11-17 rsc 
47 c9ed5273 2014-11-17 rsc 	fmtinstall('M', digestfmt);
48 c9ed5273 2014-11-17 rsc 
49 c9ed5273 2014-11-17 rsc 	if(argc == 0)
50 c9ed5273 2014-11-17 rsc 		sum(0, nil);
51 c9ed5273 2014-11-17 rsc 	else for(i = 0; i < argc; i++){
52 c9ed5273 2014-11-17 rsc 		fd = open(argv[i], OREAD);
53 c9ed5273 2014-11-17 rsc 		if(fd < 0){
54 c9ed5273 2014-11-17 rsc 			fprint(2, "md5sum: can't open %s: %r\n", argv[i]);
55 c9ed5273 2014-11-17 rsc 			continue;
56 c9ed5273 2014-11-17 rsc 		}
57 c9ed5273 2014-11-17 rsc 		sum(fd, argv[i]);
58 c9ed5273 2014-11-17 rsc 		close(fd);
59 c9ed5273 2014-11-17 rsc 	}
60 c9ed5273 2014-11-17 rsc 	exits(nil);
61 c9ed5273 2014-11-17 rsc }