2 * Unix SMB/Netbios implementation.
4 * An implementation of MD4 designed for use in the SMB authentication protocol.
6 * Copyright (C) 1997-1998 Andrew Tridgell
7 * Copyright (C) 2005 Wayne Davison
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License along
20 * with this program; if not, write to the Free Software Foundation, Inc.,
21 * 51 Franklin Street - Fifth Floor, Boston, MA 02110-1301, USA.
26 /* NOTE: This code makes no attempt to be fast!
28 * It assumes that a int is at least 32 bits long. */
30 static struct mdfour *m;
32 #define MASK32 (0xffffffff)
34 #define F(X,Y,Z) ((((X)&(Y)) | ((~(X))&(Z))))
35 #define G(X,Y,Z) ((((X)&(Y)) | ((X)&(Z)) | ((Y)&(Z))))
36 #define H(X,Y,Z) (((X)^(Y)^(Z)))
37 #define lshift(x,s) (((((x)<<(s))&MASK32) | (((x)>>(32-(s)))&MASK32)))
39 #define ROUND1(a,b,c,d,k,s) a = lshift((a + F(b,c,d) + M[k])&MASK32, s)
40 #define ROUND2(a,b,c,d,k,s) a = lshift((a + G(b,c,d) + M[k] + 0x5A827999)&MASK32,s)
41 #define ROUND3(a,b,c,d,k,s) a = lshift((a + H(b,c,d) + M[k] + 0x6ED9EBA1)&MASK32,s)
43 /* this applies md4 to 64 byte chunks */
44 static void mdfour64(uint32 *M)
46 uint32 AA, BB, CC, DD;
49 A = m->A; B = m->B; C = m->C; D = m->D;
50 AA = A; BB = B; CC = C; DD = D;
52 ROUND1(A,B,C,D, 0, 3); ROUND1(D,A,B,C, 1, 7);
53 ROUND1(C,D,A,B, 2, 11); ROUND1(B,C,D,A, 3, 19);
54 ROUND1(A,B,C,D, 4, 3); ROUND1(D,A,B,C, 5, 7);
55 ROUND1(C,D,A,B, 6, 11); ROUND1(B,C,D,A, 7, 19);
56 ROUND1(A,B,C,D, 8, 3); ROUND1(D,A,B,C, 9, 7);
57 ROUND1(C,D,A,B, 10, 11); ROUND1(B,C,D,A, 11, 19);
58 ROUND1(A,B,C,D, 12, 3); ROUND1(D,A,B,C, 13, 7);
59 ROUND1(C,D,A,B, 14, 11); ROUND1(B,C,D,A, 15, 19);
62 ROUND2(A,B,C,D, 0, 3); ROUND2(D,A,B,C, 4, 5);
63 ROUND2(C,D,A,B, 8, 9); ROUND2(B,C,D,A, 12, 13);
64 ROUND2(A,B,C,D, 1, 3); ROUND2(D,A,B,C, 5, 5);
65 ROUND2(C,D,A,B, 9, 9); ROUND2(B,C,D,A, 13, 13);
66 ROUND2(A,B,C,D, 2, 3); ROUND2(D,A,B,C, 6, 5);
67 ROUND2(C,D,A,B, 10, 9); ROUND2(B,C,D,A, 14, 13);
68 ROUND2(A,B,C,D, 3, 3); ROUND2(D,A,B,C, 7, 5);
69 ROUND2(C,D,A,B, 11, 9); ROUND2(B,C,D,A, 15, 13);
71 ROUND3(A,B,C,D, 0, 3); ROUND3(D,A,B,C, 8, 9);
72 ROUND3(C,D,A,B, 4, 11); ROUND3(B,C,D,A, 12, 15);
73 ROUND3(A,B,C,D, 2, 3); ROUND3(D,A,B,C, 10, 9);
74 ROUND3(C,D,A,B, 6, 11); ROUND3(B,C,D,A, 14, 15);
75 ROUND3(A,B,C,D, 1, 3); ROUND3(D,A,B,C, 9, 9);
76 ROUND3(C,D,A,B, 5, 11); ROUND3(B,C,D,A, 13, 15);
77 ROUND3(A,B,C,D, 3, 3); ROUND3(D,A,B,C, 11, 9);
78 ROUND3(C,D,A,B, 7, 11); ROUND3(B,C,D,A, 15, 15);
83 A &= MASK32; B &= MASK32;
84 C &= MASK32; D &= MASK32;
86 m->A = A; m->B = B; m->C = C; m->D = D;
89 static void copy64(uint32 *M, unsigned char *in)
94 M[i] = (in[i*4+3]<<24) | (in[i*4+2]<<16) |
95 (in[i*4+1]<<8) | (in[i*4+0]<<0);
98 static void copy4(unsigned char *out,uint32 x)
101 out[1] = (x>>8)&0xFF;
102 out[2] = (x>>16)&0xFF;
103 out[3] = (x>>24)&0xFF;
106 void mdfour_begin(struct mdfour *md)
117 static void mdfour_tail(unsigned char *in, uint32 n)
119 unsigned char buf[128];
121 extern int protocol_version;
124 * Count total number of bits, modulo 2^64
127 if (m->totalN < (n << 3)) {
130 m->totalN2 += n >> 29;
133 if (n) memcpy(buf, in, n);
137 copy4(buf+56, m->totalN);
139 * Prior to protocol version 27 only the number of bits
140 * modulo 2^32 was included. MD4 requires the number
141 * of bits modulo 2^64, which was fixed starting with
142 * protocol version 27.
144 if (protocol_version >= 27) {
145 copy4(buf+60, m->totalN2);
150 copy4(buf+120, m->totalN);
152 * Prior to protocol version 27 only the number of bits
153 * modulo 2^32 was included. MD4 requires the number
154 * of bits modulo 2^64, which was fixed starting with
155 * protocol version 27.
157 if (protocol_version >= 27) {
158 copy4(buf+124, m->totalN2);
167 void mdfour_update(struct mdfour *md, unsigned char *in, uint32 n)
173 if (n == 0) mdfour_tail(in, n);
180 m->totalN += 64 << 3;
181 if (m->totalN < 64 << 3) {
186 if (n) mdfour_tail(in, n);
190 void mdfour_result(struct mdfour *md, unsigned char *out)
201 void mdfour(unsigned char *out, unsigned char *in, int n)
205 mdfour_update(&md, in, n);
206 mdfour_result(&md, out);
210 int protocol_version = 28;
212 static void file_checksum1(char *fname)
214 int fd, i, was_multiple_of_64 = 1;
216 unsigned char buf[64*1024], sum[16];
218 fd = open(fname,O_RDONLY);
227 int n = read(fd, buf, sizeof(buf));
230 was_multiple_of_64 = !(n % 64);
231 mdfour_update(&md, buf, n);
233 if (was_multiple_of_64 && protocol_version >= 27)
234 mdfour_update(&md, buf, 0);
238 mdfour_result(&md, sum);
241 printf("%02X", sum[i]);
248 static void file_checksum2(char *fname)
252 unsigned char buf[64], sum[16];
254 fd = open(fname,O_RDONLY);
263 int n = read(fd, buf, sizeof(buf));
265 MDupdate(&md, buf, n*8);
269 MDupdate(&md, buf, 0);
274 memcpy(sum, md.buffer, 16);
277 printf("%02X", sum[i]);
282 int main(int argc, char *argv[])
284 file_checksum1(argv[1]);
286 file_checksum2(argv[1]);