Fixed up base64_dump() users and interfaces to fix memory leaks
[jigit.git] / md5.c
1 /*
2  * This code implements the MD5 message-digest algorithm.
3  * The algorithm is due to Ron Rivest.  This code was
4  * written by Colin Plumb in 1993, no copyright is claimed.
5  * This code is in the public domain; do with it what you wish.
6  *
7  * Equivalent code is available from RSA Data Security, Inc.
8  * This code has been tested against that, and is equivalent,
9  * except that you don't need to include two pages of legalese
10  * with every copy.
11  *
12  * To compute the message digest of a chunk of bytes, declare an
13  * MD5Context structure, pass it to MD5Init, call MD5Update as
14  * needed on buffers full of bytes, and then call MD5Final, which
15  * will fill a supplied 16-byte array with the digest.
16  */
17
18 /* This code was modified in 1997 by Jim Kingdon of Cyclic Software to
19    not require an integer type which is exactly 32 bits.  This work
20    draws on the changes for the same purpose by Tatu Ylonen
21    <ylo@cs.hut.fi> as part of SSH, but since I didn't actually use
22    that code, there is no copyright issue.  I hereby disclaim
23    copyright in any changes I have made; this code remains in the
24    public domain.  */
25
26 /* Note regarding cvs_* namespace: this avoids potential conflicts
27    with libraries such as some versions of Kerberos.  No particular
28    need to worry about whether the system supplies an MD5 library, as
29    this file is only about 3k of object code.  */
30
31 /* Steve McIntyre, 2004/05/31: borrowed this code from the CVS
32    library. s/cvs_/mk_/ across the source */
33
34 #ifdef HAVE_CONFIG_H
35 #include "config.h"
36 #endif
37
38 #include <string.h>     /* for memcpy() and memset() */
39 #include <stdlib.h> /* for calloc() */
40
41 #include "md5.h"
42
43 /* Little-endian byte-swapping routines.  Note that these do not
44    depend on the size of datatypes such as mk_uint32, nor do they require
45    us to detect the endianness of the machine we are running on.  It
46    is possible they should be macros for speed, but I would be
47    surprised if they were a performance bottleneck for MD5.  */
48
49 static mk_uint32
50 getu32 (addr)
51      const unsigned char *addr;
52 {
53         return (((((unsigned long)addr[3] << 8) | addr[2]) << 8)
54                 | addr[1]) << 8 | addr[0];
55 }
56
57 static void
58 putu32 (data, addr)
59      mk_uint32 data;
60      unsigned char *addr;
61 {
62         addr[0] = (unsigned char)data;
63         addr[1] = (unsigned char)(data >> 8);
64         addr[2] = (unsigned char)(data >> 16);
65         addr[3] = (unsigned char)(data >> 24);
66 }
67
68 /*
69  * Start MD5 accumulation.  Set bit count to 0 and buffer to mysterious
70  * initialization constants.
71  */
72 void
73 mk_MD5Init (ctx)
74      struct mk_MD5Context *ctx;
75 {
76         ctx->buf[0] = 0x67452301;
77         ctx->buf[1] = 0xefcdab89;
78         ctx->buf[2] = 0x98badcfe;
79         ctx->buf[3] = 0x10325476;
80
81         ctx->bits[0] = 0;
82         ctx->bits[1] = 0;
83 }
84
85 /*
86  * Update context to reflect the concatenation of another buffer full
87  * of bytes.
88  */
89 void
90 mk_MD5Update (ctx, buf, len)
91      struct mk_MD5Context *ctx;
92      unsigned char const *buf;
93      unsigned len;
94 {
95         mk_uint32 t;
96
97         /* Update bitcount */
98
99         t = ctx->bits[0];
100         if ((ctx->bits[0] = (t + ((mk_uint32)len << 3)) & 0xffffffff) < t)
101                 ctx->bits[1]++; /* Carry from low to high */
102         ctx->bits[1] += len >> 29;
103
104         t = (t >> 3) & 0x3f;    /* Bytes already in shsInfo->data */
105
106         /* Handle any leading odd-sized chunks */
107
108         if ( t ) {
109                 unsigned char *p = ctx->in + t;
110
111                 t = 64-t;
112                 if (len < t) {
113                         memcpy(p, buf, len);
114                         return;
115                 }
116                 memcpy(p, buf, t);
117                 mk_MD5Transform (ctx->buf, ctx->in);
118                 buf += t;
119                 len -= t;
120         }
121
122         /* Process data in 64-byte chunks */
123
124         while (len >= 64) {
125                 memcpy(ctx->in, buf, 64);
126                 mk_MD5Transform (ctx->buf, ctx->in);
127                 buf += 64;
128                 len -= 64;
129         }
130
131         /* Handle any remaining bytes of data. */
132
133         memcpy(ctx->in, buf, len);
134 }
135
136 /*
137  * Final wrapup - pad to 64-byte boundary with the bit pattern 
138  * 1 0* (64-bit count of bits processed, MSB-first)
139  */
140 void
141 mk_MD5Final (digest, ctx)
142      unsigned char digest[16];
143      struct mk_MD5Context *ctx;
144 {
145         unsigned count;
146         unsigned char *p;
147
148         /* Compute number of bytes mod 64 */
149         count = (ctx->bits[0] >> 3) & 0x3F;
150
151         /* Set the first char of padding to 0x80.  This is safe since there is
152            always at least one byte free */
153         p = ctx->in + count;
154         *p++ = 0x80;
155
156         /* Bytes of padding needed to make 64 bytes */
157         count = 64 - 1 - count;
158
159         /* Pad out to 56 mod 64 */
160         if (count < 8) {
161                 /* Two lots of padding:  Pad the first block to 64 bytes */
162                 memset(p, 0, count);
163                 mk_MD5Transform (ctx->buf, ctx->in);
164
165                 /* Now fill the next block with 56 bytes */
166                 memset(ctx->in, 0, 56);
167         } else {
168                 /* Pad block to 56 bytes */
169                 memset(p, 0, count-8);
170         }
171
172         /* Append length in bits and transform */
173         putu32(ctx->bits[0], ctx->in + 56);
174         putu32(ctx->bits[1], ctx->in + 60);
175
176         mk_MD5Transform (ctx->buf, ctx->in);
177         putu32(ctx->buf[0], digest);
178         putu32(ctx->buf[1], digest + 4);
179         putu32(ctx->buf[2], digest + 8);
180         putu32(ctx->buf[3], digest + 12);
181         memset(ctx, 0, sizeof(ctx));    /* In case it's sensitive */
182 }
183
184 #ifndef ASM_MD5
185
186 /* The four core functions - F1 is optimized somewhat */
187
188 /* #define F1(x, y, z) (x & y | ~x & z) */
189 #define F1(x, y, z) (z ^ (x & (y ^ z)))
190 #define F2(x, y, z) F1(z, x, y)
191 #define F3(x, y, z) (x ^ y ^ z)
192 #define F4(x, y, z) (y ^ (x | ~z))
193
194 /* This is the central step in the MD5 algorithm. */
195 #define MD5STEP(f, w, x, y, z, data, s) \
196         ( w += f(x, y, z) + data, w &= 0xffffffff, w = w<<s | w>>(32-s), w += x )
197
198 /*
199  * The core of the MD5 algorithm, this alters an existing MD5 hash to
200  * reflect the addition of 16 longwords of new data.  MD5Update blocks
201  * the data and converts bytes into longwords for this routine.
202  */
203 void
204 mk_MD5Transform (buf, inraw)
205      mk_uint32 buf[4];
206      const unsigned char inraw[64];
207 {
208         register mk_uint32 a, b, c, d;
209         mk_uint32 in[16];
210         int i;
211
212         for (i = 0; i < 16; ++i)
213                 in[i] = getu32 (inraw + 4 * i);
214
215         a = buf[0];
216         b = buf[1];
217         c = buf[2];
218         d = buf[3];
219
220         MD5STEP(F1, a, b, c, d, in[ 0]+0xd76aa478,  7);
221         MD5STEP(F1, d, a, b, c, in[ 1]+0xe8c7b756, 12);
222         MD5STEP(F1, c, d, a, b, in[ 2]+0x242070db, 17);
223         MD5STEP(F1, b, c, d, a, in[ 3]+0xc1bdceee, 22);
224         MD5STEP(F1, a, b, c, d, in[ 4]+0xf57c0faf,  7);
225         MD5STEP(F1, d, a, b, c, in[ 5]+0x4787c62a, 12);
226         MD5STEP(F1, c, d, a, b, in[ 6]+0xa8304613, 17);
227         MD5STEP(F1, b, c, d, a, in[ 7]+0xfd469501, 22);
228         MD5STEP(F1, a, b, c, d, in[ 8]+0x698098d8,  7);
229         MD5STEP(F1, d, a, b, c, in[ 9]+0x8b44f7af, 12);
230         MD5STEP(F1, c, d, a, b, in[10]+0xffff5bb1, 17);
231         MD5STEP(F1, b, c, d, a, in[11]+0x895cd7be, 22);
232         MD5STEP(F1, a, b, c, d, in[12]+0x6b901122,  7);
233         MD5STEP(F1, d, a, b, c, in[13]+0xfd987193, 12);
234         MD5STEP(F1, c, d, a, b, in[14]+0xa679438e, 17);
235         MD5STEP(F1, b, c, d, a, in[15]+0x49b40821, 22);
236
237         MD5STEP(F2, a, b, c, d, in[ 1]+0xf61e2562,  5);
238         MD5STEP(F2, d, a, b, c, in[ 6]+0xc040b340,  9);
239         MD5STEP(F2, c, d, a, b, in[11]+0x265e5a51, 14);
240         MD5STEP(F2, b, c, d, a, in[ 0]+0xe9b6c7aa, 20);
241         MD5STEP(F2, a, b, c, d, in[ 5]+0xd62f105d,  5);
242         MD5STEP(F2, d, a, b, c, in[10]+0x02441453,  9);
243         MD5STEP(F2, c, d, a, b, in[15]+0xd8a1e681, 14);
244         MD5STEP(F2, b, c, d, a, in[ 4]+0xe7d3fbc8, 20);
245         MD5STEP(F2, a, b, c, d, in[ 9]+0x21e1cde6,  5);
246         MD5STEP(F2, d, a, b, c, in[14]+0xc33707d6,  9);
247         MD5STEP(F2, c, d, a, b, in[ 3]+0xf4d50d87, 14);
248         MD5STEP(F2, b, c, d, a, in[ 8]+0x455a14ed, 20);
249         MD5STEP(F2, a, b, c, d, in[13]+0xa9e3e905,  5);
250         MD5STEP(F2, d, a, b, c, in[ 2]+0xfcefa3f8,  9);
251         MD5STEP(F2, c, d, a, b, in[ 7]+0x676f02d9, 14);
252         MD5STEP(F2, b, c, d, a, in[12]+0x8d2a4c8a, 20);
253
254         MD5STEP(F3, a, b, c, d, in[ 5]+0xfffa3942,  4);
255         MD5STEP(F3, d, a, b, c, in[ 8]+0x8771f681, 11);
256         MD5STEP(F3, c, d, a, b, in[11]+0x6d9d6122, 16);
257         MD5STEP(F3, b, c, d, a, in[14]+0xfde5380c, 23);
258         MD5STEP(F3, a, b, c, d, in[ 1]+0xa4beea44,  4);
259         MD5STEP(F3, d, a, b, c, in[ 4]+0x4bdecfa9, 11);
260         MD5STEP(F3, c, d, a, b, in[ 7]+0xf6bb4b60, 16);
261         MD5STEP(F3, b, c, d, a, in[10]+0xbebfbc70, 23);
262         MD5STEP(F3, a, b, c, d, in[13]+0x289b7ec6,  4);
263         MD5STEP(F3, d, a, b, c, in[ 0]+0xeaa127fa, 11);
264         MD5STEP(F3, c, d, a, b, in[ 3]+0xd4ef3085, 16);
265         MD5STEP(F3, b, c, d, a, in[ 6]+0x04881d05, 23);
266         MD5STEP(F3, a, b, c, d, in[ 9]+0xd9d4d039,  4);
267         MD5STEP(F3, d, a, b, c, in[12]+0xe6db99e5, 11);
268         MD5STEP(F3, c, d, a, b, in[15]+0x1fa27cf8, 16);
269         MD5STEP(F3, b, c, d, a, in[ 2]+0xc4ac5665, 23);
270
271         MD5STEP(F4, a, b, c, d, in[ 0]+0xf4292244,  6);
272         MD5STEP(F4, d, a, b, c, in[ 7]+0x432aff97, 10);
273         MD5STEP(F4, c, d, a, b, in[14]+0xab9423a7, 15);
274         MD5STEP(F4, b, c, d, a, in[ 5]+0xfc93a039, 21);
275         MD5STEP(F4, a, b, c, d, in[12]+0x655b59c3,  6);
276         MD5STEP(F4, d, a, b, c, in[ 3]+0x8f0ccc92, 10);
277         MD5STEP(F4, c, d, a, b, in[10]+0xffeff47d, 15);
278         MD5STEP(F4, b, c, d, a, in[ 1]+0x85845dd1, 21);
279         MD5STEP(F4, a, b, c, d, in[ 8]+0x6fa87e4f,  6);
280         MD5STEP(F4, d, a, b, c, in[15]+0xfe2ce6e0, 10);
281         MD5STEP(F4, c, d, a, b, in[ 6]+0xa3014314, 15);
282         MD5STEP(F4, b, c, d, a, in[13]+0x4e0811a1, 21);
283         MD5STEP(F4, a, b, c, d, in[ 4]+0xf7537e82,  6);
284         MD5STEP(F4, d, a, b, c, in[11]+0xbd3af235, 10);
285         MD5STEP(F4, c, d, a, b, in[ 2]+0x2ad7d2bb, 15);
286         MD5STEP(F4, b, c, d, a, in[ 9]+0xeb86d391, 21);
287
288         buf[0] += a;
289         buf[1] += b;
290         buf[2] += c;
291         buf[3] += d;
292 }
293 #endif
294
295 char *base64_dump(unsigned char *buf, size_t buf_size)
296 {
297     const char *b64_enc = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_";
298     int value = 0;
299     unsigned int i;
300     int bits = 0;
301     char *out = calloc(1, 30);
302     unsigned int out_pos = 0;
303
304     if (!out)
305         return NULL;
306
307     for (i = 0; i < buf_size ; i++)
308     {
309         value = (value << 8) | buf[i];
310         bits += 2;
311         out[out_pos++] = b64_enc[(value >> bits) & 63U];
312         if (bits >= 6) {
313             bits -= 6;
314             out[out_pos++] = b64_enc[(value >> bits) & 63U];
315         }
316     }
317     if (bits > 0)
318     {
319         value <<= 6 - bits;
320         out[out_pos++] = b64_enc[value & 63U];
321     }
322     return out;
323 }
324
325 #ifdef TEST
326 /* Simple test program.  Can use it to manually run the tests from
327    RFC1321 for example.  */
328 #include <stdio.h>
329
330 int
331 main (int argc, char **argv)
332 {
333         struct mk_MD5Context context;
334         unsigned char checksum[16];
335         int i;
336         int j;
337
338         if (argc < 2)
339         {
340                 fprintf (stderr, "usage: %s string-to-hash\n", argv[0]);
341                 exit (1);
342         }
343         for (j = 1; j < argc; ++j)
344         {
345                 printf ("MD5 (\"%s\") = ", argv[j]);
346                 mk_MD5Init (&context);
347                 mk_MD5Update (&context, argv[j], strlen (argv[j]));
348                 mk_MD5Final (checksum, &context);
349                 for (i = 0; i < 16; i++)
350                 {
351                         printf ("%02x", (unsigned int) checksum[i]);
352                 }
353                 printf ("\n");
354         }
355         return 0;
356 }
357 #endif /* TEST */