aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--misc/rfc1321-md5.txt1179
-rw-r--r--src/compat/win32.cpp22
-rw-r--r--src/compat/win32.h2
-rw-r--r--src/md5.cpp323
-rw-r--r--src/md5.h17
-rw-r--r--src/plugger.cpp4
-rw-r--r--src/plugger.h107
-rw-r--r--src/tcpsocket.h1
-rw-r--r--src/tests/udpsocket.cpp86
-rw-r--r--src/udpsocket.cpp240
-rw-r--r--src/udpsocket.h81
-rw-r--r--src/unit/md5.unit82
12 files changed, 1975 insertions, 169 deletions
diff --git a/misc/rfc1321-md5.txt b/misc/rfc1321-md5.txt
new file mode 100644
index 0000000..68af27d
--- /dev/null
+++ b/misc/rfc1321-md5.txt
@@ -0,0 +1,1179 @@
1
2
3
4
5
6
7Network Working Group R. Rivest
8Request for Comments: 1321 MIT Laboratory for Computer Science
9 and RSA Data Security, Inc.
10 April 1992
11
12
13 The MD5 Message-Digest Algorithm
14
15Status of this Memo
16
17 This memo provides information for the Internet community. It does
18 not specify an Internet standard. Distribution of this memo is
19 unlimited.
20
21Acknowlegements
22
23 We would like to thank Don Coppersmith, Burt Kaliski, Ralph Merkle,
24 David Chaum, and Noam Nisan for numerous helpful comments and
25 suggestions.
26
27Table of Contents
28
29 1. Executive Summary 1
30 2. Terminology and Notation 2
31 3. MD5 Algorithm Description 3
32 4. Summary 6
33 5. Differences Between MD4 and MD5 6
34 References 7
35 APPENDIX A - Reference Implementation 7
36 Security Considerations 21
37 Author's Address 21
38
391. Executive Summary
40
41 This document describes the MD5 message-digest algorithm. The
42 algorithm takes as input a message of arbitrary length and produces
43 as output a 128-bit "fingerprint" or "message digest" of the input.
44 It is conjectured that it is computationally infeasible to produce
45 two messages having the same message digest, or to produce any
46 message having a given prespecified target message digest. The MD5
47 algorithm is intended for digital signature applications, where a
48 large file must be "compressed" in a secure manner before being
49 encrypted with a private (secret) key under a public-key cryptosystem
50 such as RSA.
51
52
53
54
55
56
57
58Rivest [Page 1]
59
60RFC 1321 MD5 Message-Digest Algorithm April 1992
61
62
63 The MD5 algorithm is designed to be quite fast on 32-bit machines. In
64 addition, the MD5 algorithm does not require any large substitution
65 tables; the algorithm can be coded quite compactly.
66
67 The MD5 algorithm is an extension of the MD4 message-digest algorithm
68 1,2]. MD5 is slightly slower than MD4, but is more "conservative" in
69 design. MD5 was designed because it was felt that MD4 was perhaps
70 being adopted for use more quickly than justified by the existing
71 critical review; because MD4 was designed to be exceptionally fast,
72 it is "at the edge" in terms of risking successful cryptanalytic
73 attack. MD5 backs off a bit, giving up a little in speed for a much
74 greater likelihood of ultimate security. It incorporates some
75 suggestions made by various reviewers, and contains additional
76 optimizations. The MD5 algorithm is being placed in the public domain
77 for review and possible adoption as a standard.
78
79 For OSI-based applications, MD5's object identifier is
80
81 md5 OBJECT IDENTIFIER ::=
82 iso(1) member-body(2) US(840) rsadsi(113549) digestAlgorithm(2) 5}
83
84 In the X.509 type AlgorithmIdentifier [3], the parameters for MD5
85 should have type NULL.
86
872. Terminology and Notation
88
89 In this document a "word" is a 32-bit quantity and a "byte" is an
90 eight-bit quantity. A sequence of bits can be interpreted in a
91 natural manner as a sequence of bytes, where each consecutive group
92 of eight bits is interpreted as a byte with the high-order (most
93 significant) bit of each byte listed first. Similarly, a sequence of
94 bytes can be interpreted as a sequence of 32-bit words, where each
95 consecutive group of four bytes is interpreted as a word with the
96 low-order (least significant) byte given first.
97
98 Let x_i denote "x sub i". If the subscript is an expression, we
99 surround it in braces, as in x_{i+1}. Similarly, we use ^ for
100 superscripts (exponentiation), so that x^i denotes x to the i-th
101 power.
102
103 Let the symbol "+" denote addition of words (i.e., modulo-2^32
104 addition). Let X <<< s denote the 32-bit value obtained by circularly
105 shifting (rotating) X left by s bit positions. Let not(X) denote the
106 bit-wise complement of X, and let X v Y denote the bit-wise OR of X
107 and Y. Let X xor Y denote the bit-wise XOR of X and Y, and let XY
108 denote the bit-wise AND of X and Y.
109
110
111
112
113
114Rivest [Page 2]
115
116RFC 1321 MD5 Message-Digest Algorithm April 1992
117
118
1193. MD5 Algorithm Description
120
121 We begin by supposing that we have a b-bit message as input, and that
122 we wish to find its message digest. Here b is an arbitrary
123 nonnegative integer; b may be zero, it need not be a multiple of
124 eight, and it may be arbitrarily large. We imagine the bits of the
125 message written down as follows:
126
127 m_0 m_1 ... m_{b-1}
128
129 The following five steps are performed to compute the message digest
130 of the message.
131
1323.1 Step 1. Append Padding Bits
133
134 The message is "padded" (extended) so that its length (in bits) is
135 congruent to 448, modulo 512. That is, the message is extended so
136 that it is just 64 bits shy of being a multiple of 512 bits long.
137 Padding is always performed, even if the length of the message is
138 already congruent to 448, modulo 512.
139
140 Padding is performed as follows: a single "1" bit is appended to the
141 message, and then "0" bits are appended so that the length in bits of
142 the padded message becomes congruent to 448, modulo 512. In all, at
143 least one bit and at most 512 bits are appended.
144
1453.2 Step 2. Append Length
146
147 A 64-bit representation of b (the length of the message before the
148 padding bits were added) is appended to the result of the previous
149 step. In the unlikely event that b is greater than 2^64, then only
150 the low-order 64 bits of b are used. (These bits are appended as two
151 32-bit words and appended low-order word first in accordance with the
152 previous conventions.)
153
154 At this point the resulting message (after padding with bits and with
155 b) has a length that is an exact multiple of 512 bits. Equivalently,
156 this message has a length that is an exact multiple of 16 (32-bit)
157 words. Let M[0 ... N-1] denote the words of the resulting message,
158 where N is a multiple of 16.
159
1603.3 Step 3. Initialize MD Buffer
161
162 A four-word buffer (A,B,C,D) is used to compute the message digest.
163 Here each of A, B, C, D is a 32-bit register. These registers are
164 initialized to the following values in hexadecimal, low-order bytes
165 first):
166
167
168
169
170Rivest [Page 3]
171
172RFC 1321 MD5 Message-Digest Algorithm April 1992
173
174
175 word A: 01 23 45 67
176 word B: 89 ab cd ef
177 word C: fe dc ba 98
178 word D: 76 54 32 10
179
1803.4 Step 4. Process Message in 16-Word Blocks
181
182 We first define four auxiliary functions that each take as input
183 three 32-bit words and produce as output one 32-bit word.
184
185 F(X,Y,Z) = XY v not(X) Z
186 G(X,Y,Z) = XZ v Y not(Z)
187 H(X,Y,Z) = X xor Y xor Z
188 I(X,Y,Z) = Y xor (X v not(Z))
189
190 In each bit position F acts as a conditional: if X then Y else Z.
191 The function F could have been defined using + instead of v since XY
192 and not(X)Z will never have 1's in the same bit position.) It is
193 interesting to note that if the bits of X, Y, and Z are independent
194 and unbiased, the each bit of F(X,Y,Z) will be independent and
195 unbiased.
196
197 The functions G, H, and I are similar to the function F, in that they
198 act in "bitwise parallel" to produce their output from the bits of X,
199 Y, and Z, in such a manner that if the corresponding bits of X, Y,
200 and Z are independent and unbiased, then each bit of G(X,Y,Z),
201 H(X,Y,Z), and I(X,Y,Z) will be independent and unbiased. Note that
202 the function H is the bit-wise "xor" or "parity" function of its
203 inputs.
204
205 This step uses a 64-element table T[1 ... 64] constructed from the
206 sine function. Let T[i] denote the i-th element of the table, which
207 is equal to the integer part of 4294967296 times abs(sin(i)), where i
208 is in radians. The elements of the table are given in the appendix.
209
210 Do the following:
211
212 /* Process each 16-word block. */
213 For i = 0 to N/16-1 do
214
215 /* Copy block i into X. */
216 For j = 0 to 15 do
217 Set X[j] to M[i*16+j].
218 end /* of loop on j */
219
220 /* Save A as AA, B as BB, C as CC, and D as DD. */
221 AA = A
222 BB = B
223
224
225
226Rivest [Page 4]
227
228RFC 1321 MD5 Message-Digest Algorithm April 1992
229
230
231 CC = C
232 DD = D
233
234 /* Round 1. */
235 /* Let [abcd k s i] denote the operation
236 a = b + ((a + F(b,c,d) + X[k] + T[i]) <<< s). */
237 /* Do the following 16 operations. */
238 [ABCD 0 7 1] [DABC 1 12 2] [CDAB 2 17 3] [BCDA 3 22 4]
239 [ABCD 4 7 5] [DABC 5 12 6] [CDAB 6 17 7] [BCDA 7 22 8]
240 [ABCD 8 7 9] [DABC 9 12 10] [CDAB 10 17 11] [BCDA 11 22 12]
241 [ABCD 12 7 13] [DABC 13 12 14] [CDAB 14 17 15] [BCDA 15 22 16]
242
243 /* Round 2. */
244 /* Let [abcd k s i] denote the operation
245 a = b + ((a + G(b,c,d) + X[k] + T[i]) <<< s). */
246 /* Do the following 16 operations. */
247 [ABCD 1 5 17] [DABC 6 9 18] [CDAB 11 14 19] [BCDA 0 20 20]
248 [ABCD 5 5 21] [DABC 10 9 22] [CDAB 15 14 23] [BCDA 4 20 24]
249 [ABCD 9 5 25] [DABC 14 9 26] [CDAB 3 14 27] [BCDA 8 20 28]
250 [ABCD 13 5 29] [DABC 2 9 30] [CDAB 7 14 31] [BCDA 12 20 32]
251
252 /* Round 3. */
253 /* Let [abcd k s t] denote the operation
254 a = b + ((a + H(b,c,d) + X[k] + T[i]) <<< s). */
255 /* Do the following 16 operations. */
256 [ABCD 5 4 33] [DABC 8 11 34] [CDAB 11 16 35] [BCDA 14 23 36]
257 [ABCD 1 4 37] [DABC 4 11 38] [CDAB 7 16 39] [BCDA 10 23 40]
258 [ABCD 13 4 41] [DABC 0 11 42] [CDAB 3 16 43] [BCDA 6 23 44]
259 [ABCD 9 4 45] [DABC 12 11 46] [CDAB 15 16 47] [BCDA 2 23 48]
260
261 /* Round 4. */
262 /* Let [abcd k s t] denote the operation
263 a = b + ((a + I(b,c,d) + X[k] + T[i]) <<< s). */
264 /* Do the following 16 operations. */
265 [ABCD 0 6 49] [DABC 7 10 50] [CDAB 14 15 51] [BCDA 5 21 52]
266 [ABCD 12 6 53] [DABC 3 10 54] [CDAB 10 15 55] [BCDA 1 21 56]
267 [ABCD 8 6 57] [DABC 15 10 58] [CDAB 6 15 59] [BCDA 13 21 60]
268 [ABCD 4 6 61] [DABC 11 10 62] [CDAB 2 15 63] [BCDA 9 21 64]
269
270 /* Then perform the following additions. (That is increment each
271 of the four registers by the value it had before this block
272 was started.) */
273 A = A + AA
274 B = B + BB
275 C = C + CC
276 D = D + DD
277
278 end /* of loop on i */
279
280
281
282Rivest [Page 5]
283
284RFC 1321 MD5 Message-Digest Algorithm April 1992
285
286
2873.5 Step 5. Output
288
289 The message digest produced as output is A, B, C, D. That is, we
290 begin with the low-order byte of A, and end with the high-order byte
291 of D.
292
293 This completes the description of MD5. A reference implementation in
294 C is given in the appendix.
295
2964. Summary
297
298 The MD5 message-digest algorithm is simple to implement, and provides
299 a "fingerprint" or message digest of a message of arbitrary length.
300 It is conjectured that the difficulty of coming up with two messages
301 having the same message digest is on the order of 2^64 operations,
302 and that the difficulty of coming up with any message having a given
303 message digest is on the order of 2^128 operations. The MD5 algorithm
304 has been carefully scrutinized for weaknesses. It is, however, a
305 relatively new algorithm and further security analysis is of course
306 justified, as is the case with any new proposal of this sort.
307
3085. Differences Between MD4 and MD5
309
310 The following are the differences between MD4 and MD5:
311
312 1. A fourth round has been added.
313
314 2. Each step now has a unique additive constant.
315
316 3. The function g in round 2 was changed from (XY v XZ v YZ) to
317 (XZ v Y not(Z)) to make g less symmetric.
318
319 4. Each step now adds in the result of the previous step. This
320 promotes a faster "avalanche effect".
321
322 5. The order in which input words are accessed in rounds 2 and
323 3 is changed, to make these patterns less like each other.
324
325 6. The shift amounts in each round have been approximately
326 optimized, to yield a faster "avalanche effect." The shifts in
327 different rounds are distinct.
328
329
330
331
332
333
334
335
336
337
338Rivest [Page 6]
339
340RFC 1321 MD5 Message-Digest Algorithm April 1992
341
342
343References
344
345 [1] Rivest, R., "The MD4 Message Digest Algorithm", RFC 1320, MIT and
346 RSA Data Security, Inc., April 1992.
347
348 [2] Rivest, R., "The MD4 message digest algorithm", in A.J. Menezes
349 and S.A. Vanstone, editors, Advances in Cryptology - CRYPTO '90
350 Proceedings, pages 303-311, Springer-Verlag, 1991.
351
352 [3] CCITT Recommendation X.509 (1988), "The Directory -
353 Authentication Framework."
354
355APPENDIX A - Reference Implementation
356
357 This appendix contains the following files taken from RSAREF: A
358 Cryptographic Toolkit for Privacy-Enhanced Mail:
359
360 global.h -- global header file
361
362 md5.h -- header file for MD5
363
364 md5c.c -- source code for MD5
365
366 For more information on RSAREF, send email to <rsaref@rsa.com>.
367
368 The appendix also includes the following file:
369
370 mddriver.c -- test driver for MD2, MD4 and MD5
371
372 The driver compiles for MD5 by default but can compile for MD2 or MD4
373 if the symbol MD is defined on the C compiler command line as 2 or 4.
374
375 The implementation is portable and should work on many different
376 plaforms. However, it is not difficult to optimize the implementation
377 on particular platforms, an exercise left to the reader. For example,
378 on "little-endian" platforms where the lowest-addressed byte in a 32-
379 bit word is the least significant and there are no alignment
380 restrictions, the call to Decode in MD5Transform can be replaced with
381 a typecast.
382
383A.1 global.h
384
385/* GLOBAL.H - RSAREF types and constants
386 */
387
388/* PROTOTYPES should be set to one if and only if the compiler supports
389 function argument prototyping.
390The following makes PROTOTYPES default to 0 if it has not already
391
392
393
394Rivest [Page 7]
395
396RFC 1321 MD5 Message-Digest Algorithm April 1992
397
398
399 been defined with C compiler flags.
400 */
401#ifndef PROTOTYPES
402#define PROTOTYPES 0
403#endif
404
405/* POINTER defines a generic pointer type */
406typedef unsigned char *POINTER;
407
408/* UINT2 defines a two byte word */
409typedef unsigned short int UINT2;
410
411/* UINT4 defines a four byte word */
412typedef unsigned long int UINT4;
413
414/* PROTO_LIST is defined depending on how PROTOTYPES is defined above.
415If using PROTOTYPES, then PROTO_LIST returns the list, otherwise it
416 returns an empty list.
417 */
418#if PROTOTYPES
419#define PROTO_LIST(list) list
420#else
421#define PROTO_LIST(list) ()
422#endif
423
424A.2 md5.h
425
426/* MD5.H - header file for MD5C.C
427 */
428
429/* Copyright (C) 1991-2, RSA Data Security, Inc. Created 1991. All
430rights reserved.
431
432License to copy and use this software is granted provided that it
433is identified as the "RSA Data Security, Inc. MD5 Message-Digest
434Algorithm" in all material mentioning or referencing this software
435or this function.
436
437License is also granted to make and use derivative works provided
438that such works are identified as "derived from the RSA Data
439Security, Inc. MD5 Message-Digest Algorithm" in all material
440mentioning or referencing the derived work.
441
442RSA Data Security, Inc. makes no representations concerning either
443the merchantability of this software or the suitability of this
444software for any particular purpose. It is provided "as is"
445without express or implied warranty of any kind.
446
447
448
449
450Rivest [Page 8]
451
452RFC 1321 MD5 Message-Digest Algorithm April 1992
453
454
455These notices must be retained in any copies of any part of this
456documentation and/or software.
457 */
458
459/* MD5 context. */
460typedef struct {
461 UINT4 state[4]; /* state (ABCD) */
462 UINT4 count[2]; /* number of bits, modulo 2^64 (lsb first) */
463 unsigned char buffer[64]; /* input buffer */
464} MD5_CTX;
465
466void MD5Init PROTO_LIST ((MD5_CTX *));
467void MD5Update PROTO_LIST
468 ((MD5_CTX *, unsigned char *, unsigned int));
469void MD5Final PROTO_LIST ((unsigned char [16], MD5_CTX *));
470
471A.3 md5c.c
472
473/* MD5C.C - RSA Data Security, Inc., MD5 message-digest algorithm
474 */
475
476/* Copyright (C) 1991-2, RSA Data Security, Inc. Created 1991. All
477rights reserved.
478
479License to copy and use this software is granted provided that it
480is identified as the "RSA Data Security, Inc. MD5 Message-Digest
481Algorithm" in all material mentioning or referencing this software
482or this function.
483
484License is also granted to make and use derivative works provided
485that such works are identified as "derived from the RSA Data
486Security, Inc. MD5 Message-Digest Algorithm" in all material
487mentioning or referencing the derived work.
488
489RSA Data Security, Inc. makes no representations concerning either
490the merchantability of this software or the suitability of this
491software for any particular purpose. It is provided "as is"
492without express or implied warranty of any kind.
493
494These notices must be retained in any copies of any part of this
495documentation and/or software.
496 */
497
498#include "global.h"
499#include "md5.h"
500
501/* Constants for MD5Transform routine.
502 */
503
504
505
506Rivest [Page 9]
507
508RFC 1321 MD5 Message-Digest Algorithm April 1992
509
510
511#define S11 7
512#define S12 12
513#define S13 17
514#define S14 22
515#define S21 5
516#define S22 9
517#define S23 14
518#define S24 20
519#define S31 4
520#define S32 11
521#define S33 16
522#define S34 23
523#define S41 6
524#define S42 10
525#define S43 15
526#define S44 21
527
528static void MD5Transform PROTO_LIST ((UINT4 [4], unsigned char [64]));
529static void Encode PROTO_LIST
530 ((unsigned char *, UINT4 *, unsigned int));
531static void Decode PROTO_LIST
532 ((UINT4 *, unsigned char *, unsigned int));
533static void MD5_memcpy PROTO_LIST ((POINTER, POINTER, unsigned int));
534static void MD5_memset PROTO_LIST ((POINTER, int, unsigned int));
535
536static unsigned char PADDING[64] = {
537 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
538 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
539 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
540};
541
542/* F, G, H and I are basic MD5 functions.
543 */
544#define F(x, y, z) (((x) & (y)) | ((~x) & (z)))
545#define G(x, y, z) (((x) & (z)) | ((y) & (~z)))
546#define H(x, y, z) ((x) ^ (y) ^ (z))
547#define I(x, y, z) ((y) ^ ((x) | (~z)))
548
549/* ROTATE_LEFT rotates x left n bits.
550 */
551#define ROTATE_LEFT(x, n) (((x) << (n)) | ((x) >> (32-(n))))
552
553/* FF, GG, HH, and II transformations for rounds 1, 2, 3, and 4.
554Rotation is separate from addition to prevent recomputation.
555 */
556#define FF(a, b, c, d, x, s, ac) { \
557 (a) += F ((b), (c), (d)) + (x) + (UINT4)(ac); \
558 (a) = ROTATE_LEFT ((a), (s)); \
559
560
561
562Rivest [Page 10]
563
564RFC 1321 MD5 Message-Digest Algorithm April 1992
565
566
567 (a) += (b); \
568 }
569#define GG(a, b, c, d, x, s, ac) { \
570 (a) += G ((b), (c), (d)) + (x) + (UINT4)(ac); \
571 (a) = ROTATE_LEFT ((a), (s)); \
572 (a) += (b); \
573 }
574#define HH(a, b, c, d, x, s, ac) { \
575 (a) += H ((b), (c), (d)) + (x) + (UINT4)(ac); \
576 (a) = ROTATE_LEFT ((a), (s)); \
577 (a) += (b); \
578 }
579#define II(a, b, c, d, x, s, ac) { \
580 (a) += I ((b), (c), (d)) + (x) + (UINT4)(ac); \
581 (a) = ROTATE_LEFT ((a), (s)); \
582 (a) += (b); \
583 }
584
585/* MD5 initialization. Begins an MD5 operation, writing a new context.
586 */
587void MD5Init (context)
588MD5_CTX *context; /* context */
589{
590 context->count[0] = context->count[1] = 0;
591 /* Load magic initialization constants.
592*/
593 context->state[0] = 0x67452301;
594 context->state[1] = 0xefcdab89;
595 context->state[2] = 0x98badcfe;
596 context->state[3] = 0x10325476;
597}
598
599/* MD5 block update operation. Continues an MD5 message-digest
600 operation, processing another message block, and updating the
601 context.
602 */
603void MD5Update (context, input, inputLen)
604MD5_CTX *context; /* context */
605unsigned char *input; /* input block */
606unsigned int inputLen; /* length of input block */
607{
608 unsigned int i, index, partLen;
609
610 /* Compute number of bytes mod 64 */
611 index = (unsigned int)((context->count[0] >> 3) & 0x3F);
612
613 /* Update number of bits */
614 if ((context->count[0] += ((UINT4)inputLen << 3))
615
616
617
618Rivest [Page 11]
619
620RFC 1321 MD5 Message-Digest Algorithm April 1992
621
622
623 < ((UINT4)inputLen << 3))
624 context->count[1]++;
625 context->count[1] += ((UINT4)inputLen >> 29);
626
627 partLen = 64 - index;
628
629 /* Transform as many times as possible.
630*/
631 if (inputLen >= partLen) {
632 MD5_memcpy
633 ((POINTER)&context->buffer[index], (POINTER)input, partLen);
634 MD5Transform (context->state, context->buffer);
635
636 for (i = partLen; i + 63 < inputLen; i += 64)
637 MD5Transform (context->state, &input[i]);
638
639 index = 0;
640 }
641 else
642 i = 0;
643
644 /* Buffer remaining input */
645 MD5_memcpy
646 ((POINTER)&context->buffer[index], (POINTER)&input[i],
647 inputLen-i);
648}
649
650/* MD5 finalization. Ends an MD5 message-digest operation, writing the
651 the message digest and zeroizing the context.
652 */
653void MD5Final (digest, context)
654unsigned char digest[16]; /* message digest */
655MD5_CTX *context; /* context */
656{
657 unsigned char bits[8];
658 unsigned int index, padLen;
659
660 /* Save number of bits */
661 Encode (bits, context->count, 8);
662
663 /* Pad out to 56 mod 64.
664*/
665 index = (unsigned int)((context->count[0] >> 3) & 0x3f);
666 padLen = (index < 56) ? (56 - index) : (120 - index);
667 MD5Update (context, PADDING, padLen);
668
669 /* Append length (before padding) */
670 MD5Update (context, bits, 8);
671
672
673
674Rivest [Page 12]
675
676RFC 1321 MD5 Message-Digest Algorithm April 1992
677
678
679 /* Store state in digest */
680 Encode (digest, context->state, 16);
681
682 /* Zeroize sensitive information.
683*/
684 MD5_memset ((POINTER)context, 0, sizeof (*context));
685}
686
687/* MD5 basic transformation. Transforms state based on block.
688 */
689static void MD5Transform (state, block)
690UINT4 state[4];
691unsigned char block[64];
692{
693 UINT4 a = state[0], b = state[1], c = state[2], d = state[3], x[16];
694
695 Decode (x, block, 64);
696
697 /* Round 1 */
698 FF (a, b, c, d, x[ 0], S11, 0xd76aa478); /* 1 */
699 FF (d, a, b, c, x[ 1], S12, 0xe8c7b756); /* 2 */
700 FF (c, d, a, b, x[ 2], S13, 0x242070db); /* 3 */
701 FF (b, c, d, a, x[ 3], S14, 0xc1bdceee); /* 4 */
702 FF (a, b, c, d, x[ 4], S11, 0xf57c0faf); /* 5 */
703 FF (d, a, b, c, x[ 5], S12, 0x4787c62a); /* 6 */
704 FF (c, d, a, b, x[ 6], S13, 0xa8304613); /* 7 */
705 FF (b, c, d, a, x[ 7], S14, 0xfd469501); /* 8 */
706 FF (a, b, c, d, x[ 8], S11, 0x698098d8); /* 9 */
707 FF (d, a, b, c, x[ 9], S12, 0x8b44f7af); /* 10 */
708 FF (c, d, a, b, x[10], S13, 0xffff5bb1); /* 11 */
709 FF (b, c, d, a, x[11], S14, 0x895cd7be); /* 12 */
710 FF (a, b, c, d, x[12], S11, 0x6b901122); /* 13 */
711 FF (d, a, b, c, x[13], S12, 0xfd987193); /* 14 */
712 FF (c, d, a, b, x[14], S13, 0xa679438e); /* 15 */
713 FF (b, c, d, a, x[15], S14, 0x49b40821); /* 16 */
714
715 /* Round 2 */
716 GG (a, b, c, d, x[ 1], S21, 0xf61e2562); /* 17 */
717 GG (d, a, b, c, x[ 6], S22, 0xc040b340); /* 18 */
718 GG (c, d, a, b, x[11], S23, 0x265e5a51); /* 19 */
719 GG (b, c, d, a, x[ 0], S24, 0xe9b6c7aa); /* 20 */
720 GG (a, b, c, d, x[ 5], S21, 0xd62f105d); /* 21 */
721 GG (d, a, b, c, x[10], S22, 0x2441453); /* 22 */
722 GG (c, d, a, b, x[15], S23, 0xd8a1e681); /* 23 */
723 GG (b, c, d, a, x[ 4], S24, 0xe7d3fbc8); /* 24 */
724 GG (a, b, c, d, x[ 9], S21, 0x21e1cde6); /* 25 */
725 GG (d, a, b, c, x[14], S22, 0xc33707d6); /* 26 */
726 GG (c, d, a, b, x[ 3], S23, 0xf4d50d87); /* 27 */
727
728
729
730Rivest [Page 13]
731
732RFC 1321 MD5 Message-Digest Algorithm April 1992
733
734
735 GG (b, c, d, a, x[ 8], S24, 0x455a14ed); /* 28 */
736 GG (a, b, c, d, x[13], S21, 0xa9e3e905); /* 29 */
737 GG (d, a, b, c, x[ 2], S22, 0xfcefa3f8); /* 30 */
738 GG (c, d, a, b, x[ 7], S23, 0x676f02d9); /* 31 */
739 GG (b, c, d, a, x[12], S24, 0x8d2a4c8a); /* 32 */
740
741 /* Round 3 */
742 HH (a, b, c, d, x[ 5], S31, 0xfffa3942); /* 33 */
743 HH (d, a, b, c, x[ 8], S32, 0x8771f681); /* 34 */
744 HH (c, d, a, b, x[11], S33, 0x6d9d6122); /* 35 */
745 HH (b, c, d, a, x[14], S34, 0xfde5380c); /* 36 */
746 HH (a, b, c, d, x[ 1], S31, 0xa4beea44); /* 37 */
747 HH (d, a, b, c, x[ 4], S32, 0x4bdecfa9); /* 38 */
748 HH (c, d, a, b, x[ 7], S33, 0xf6bb4b60); /* 39 */
749 HH (b, c, d, a, x[10], S34, 0xbebfbc70); /* 40 */
750 HH (a, b, c, d, x[13], S31, 0x289b7ec6); /* 41 */
751 HH (d, a, b, c, x[ 0], S32, 0xeaa127fa); /* 42 */
752 HH (c, d, a, b, x[ 3], S33, 0xd4ef3085); /* 43 */
753 HH (b, c, d, a, x[ 6], S34, 0x4881d05); /* 44 */
754 HH (a, b, c, d, x[ 9], S31, 0xd9d4d039); /* 45 */
755 HH (d, a, b, c, x[12], S32, 0xe6db99e5); /* 46 */
756 HH (c, d, a, b, x[15], S33, 0x1fa27cf8); /* 47 */
757 HH (b, c, d, a, x[ 2], S34, 0xc4ac5665); /* 48 */
758
759 /* Round 4 */
760 II (a, b, c, d, x[ 0], S41, 0xf4292244); /* 49 */
761 II (d, a, b, c, x[ 7], S42, 0x432aff97); /* 50 */
762 II (c, d, a, b, x[14], S43, 0xab9423a7); /* 51 */
763 II (b, c, d, a, x[ 5], S44, 0xfc93a039); /* 52 */
764 II (a, b, c, d, x[12], S41, 0x655b59c3); /* 53 */
765 II (d, a, b, c, x[ 3], S42, 0x8f0ccc92); /* 54 */
766 II (c, d, a, b, x[10], S43, 0xffeff47d); /* 55 */
767 II (b, c, d, a, x[ 1], S44, 0x85845dd1); /* 56 */
768 II (a, b, c, d, x[ 8], S41, 0x6fa87e4f); /* 57 */
769 II (d, a, b, c, x[15], S42, 0xfe2ce6e0); /* 58 */
770 II (c, d, a, b, x[ 6], S43, 0xa3014314); /* 59 */
771 II (b, c, d, a, x[13], S44, 0x4e0811a1); /* 60 */
772 II (a, b, c, d, x[ 4], S41, 0xf7537e82); /* 61 */
773 II (d, a, b, c, x[11], S42, 0xbd3af235); /* 62 */
774 II (c, d, a, b, x[ 2], S43, 0x2ad7d2bb); /* 63 */
775 II (b, c, d, a, x[ 9], S44, 0xeb86d391); /* 64 */
776
777 state[0] += a;
778 state[1] += b;
779 state[2] += c;
780 state[3] += d;
781
782 /* Zeroize sensitive information.
783
784
785
786Rivest [Page 14]
787
788RFC 1321 MD5 Message-Digest Algorithm April 1992
789
790
791*/
792 MD5_memset ((POINTER)x, 0, sizeof (x));
793}
794
795/* Encodes input (UINT4) into output (unsigned char). Assumes len is
796 a multiple of 4.
797 */
798static void Encode (output, input, len)
799unsigned char *output;
800UINT4 *input;
801unsigned int len;
802{
803 unsigned int i, j;
804
805 for (i = 0, j = 0; j < len; i++, j += 4) {
806 output[j] = (unsigned char)(input[i] & 0xff);
807 output[j+1] = (unsigned char)((input[i] >> 8) & 0xff);
808 output[j+2] = (unsigned char)((input[i] >> 16) & 0xff);
809 output[j+3] = (unsigned char)((input[i] >> 24) & 0xff);
810 }
811}
812
813/* Decodes input (unsigned char) into output (UINT4). Assumes len is
814 a multiple of 4.
815 */
816static void Decode (output, input, len)
817UINT4 *output;
818unsigned char *input;
819unsigned int len;
820{
821 unsigned int i, j;
822
823 for (i = 0, j = 0; j < len; i++, j += 4)
824 output[i] = ((UINT4)input[j]) | (((UINT4)input[j+1]) << 8) |
825 (((UINT4)input[j+2]) << 16) | (((UINT4)input[j+3]) << 24);
826}
827
828/* Note: Replace "for loop" with standard memcpy if possible.
829 */
830
831static void MD5_memcpy (output, input, len)
832POINTER output;
833POINTER input;
834unsigned int len;
835{
836 unsigned int i;
837
838 for (i = 0; i < len; i++)
839
840
841
842Rivest [Page 15]
843
844RFC 1321 MD5 Message-Digest Algorithm April 1992
845
846
847 output[i] = input[i];
848}
849
850/* Note: Replace "for loop" with standard memset if possible.
851 */
852static void MD5_memset (output, value, len)
853POINTER output;
854int value;
855unsigned int len;
856{
857 unsigned int i;
858
859 for (i = 0; i < len; i++)
860 ((char *)output)[i] = (char)value;
861}
862
863A.4 mddriver.c
864
865/* MDDRIVER.C - test driver for MD2, MD4 and MD5
866 */
867
868/* Copyright (C) 1990-2, RSA Data Security, Inc. Created 1990. All
869rights reserved.
870
871RSA Data Security, Inc. makes no representations concerning either
872the merchantability of this software or the suitability of this
873software for any particular purpose. It is provided "as is"
874without express or implied warranty of any kind.
875
876These notices must be retained in any copies of any part of this
877documentation and/or software.
878 */
879
880/* The following makes MD default to MD5 if it has not already been
881 defined with C compiler flags.
882 */
883#ifndef MD
884#define MD MD5
885#endif
886
887#include <stdio.h>
888#include <time.h>
889#include <string.h>
890#include "global.h"
891#if MD == 2
892#include "md2.h"
893#endif
894#if MD == 4
895
896
897
898Rivest [Page 16]
899
900RFC 1321 MD5 Message-Digest Algorithm April 1992
901
902
903#include "md4.h"
904#endif
905#if MD == 5
906#include "md5.h"
907#endif
908
909/* Length of test block, number of test blocks.
910 */
911#define TEST_BLOCK_LEN 1000
912#define TEST_BLOCK_COUNT 1000
913
914static void MDString PROTO_LIST ((char *));
915static void MDTimeTrial PROTO_LIST ((void));
916static void MDTestSuite PROTO_LIST ((void));
917static void MDFile PROTO_LIST ((char *));
918static void MDFilter PROTO_LIST ((void));
919static void MDPrint PROTO_LIST ((unsigned char [16]));
920
921#if MD == 2
922#define MD_CTX MD2_CTX
923#define MDInit MD2Init
924#define MDUpdate MD2Update
925#define MDFinal MD2Final
926#endif
927#if MD == 4
928#define MD_CTX MD4_CTX
929#define MDInit MD4Init
930#define MDUpdate MD4Update
931#define MDFinal MD4Final
932#endif
933#if MD == 5
934#define MD_CTX MD5_CTX
935#define MDInit MD5Init
936#define MDUpdate MD5Update
937#define MDFinal MD5Final
938#endif
939
940/* Main driver.
941
942Arguments (may be any combination):
943 -sstring - digests string
944 -t - runs time trial
945 -x - runs test script
946 filename - digests file
947 (none) - digests standard input
948 */
949int main (argc, argv)
950int argc;
951
952
953
954Rivest [Page 17]
955
956RFC 1321 MD5 Message-Digest Algorithm April 1992
957
958
959char *argv[];
960{
961 int i;
962
963 if (argc > 1)
964 for (i = 1; i < argc; i++)
965 if (argv[i][0] == '-' && argv[i][1] == 's')
966 MDString (argv[i] + 2);
967 else if (strcmp (argv[i], "-t") == 0)
968 MDTimeTrial ();
969 else if (strcmp (argv[i], "-x") == 0)
970 MDTestSuite ();
971 else
972 MDFile (argv[i]);
973 else
974 MDFilter ();
975
976 return (0);
977}
978
979/* Digests a string and prints the result.
980 */
981static void MDString (string)
982char *string;
983{
984 MD_CTX context;
985 unsigned char digest[16];
986 unsigned int len = strlen (string);
987
988 MDInit (&context);
989 MDUpdate (&context, string, len);
990 MDFinal (digest, &context);
991
992 printf ("MD%d (\"%s\") = ", MD, string);
993 MDPrint (digest);
994 printf ("\n");
995}
996
997/* Measures the time to digest TEST_BLOCK_COUNT TEST_BLOCK_LEN-byte
998 blocks.
999 */
1000static void MDTimeTrial ()
1001{
1002 MD_CTX context;
1003 time_t endTime, startTime;
1004 unsigned char block[TEST_BLOCK_LEN], digest[16];
1005 unsigned int i;
1006
1007
1008
1009
1010Rivest [Page 18]
1011
1012RFC 1321 MD5 Message-Digest Algorithm April 1992
1013
1014
1015 printf
1016 ("MD%d time trial. Digesting %d %d-byte blocks ...", MD,
1017 TEST_BLOCK_LEN, TEST_BLOCK_COUNT);
1018
1019 /* Initialize block */
1020 for (i = 0; i < TEST_BLOCK_LEN; i++)
1021 block[i] = (unsigned char)(i & 0xff);
1022
1023 /* Start timer */
1024 time (&startTime);
1025
1026 /* Digest blocks */
1027 MDInit (&context);
1028 for (i = 0; i < TEST_BLOCK_COUNT; i++)
1029 MDUpdate (&context, block, TEST_BLOCK_LEN);
1030 MDFinal (digest, &context);
1031
1032 /* Stop timer */
1033 time (&endTime);
1034
1035 printf (" done\n");
1036 printf ("Digest = ");
1037 MDPrint (digest);
1038 printf ("\nTime = %ld seconds\n", (long)(endTime-startTime));
1039 printf
1040 ("Speed = %ld bytes/second\n",
1041 (long)TEST_BLOCK_LEN * (long)TEST_BLOCK_COUNT/(endTime-startTime));
1042}
1043
1044/* Digests a reference suite of strings and prints the results.
1045 */
1046static void MDTestSuite ()
1047{
1048 printf ("MD%d test suite:\n", MD);
1049
1050 MDString ("");
1051 MDString ("a");
1052 MDString ("abc");
1053 MDString ("message digest");
1054 MDString ("abcdefghijklmnopqrstuvwxyz");
1055 MDString
1056 ("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789");
1057 MDString
1058 ("1234567890123456789012345678901234567890\
10591234567890123456789012345678901234567890");
1060}
1061
1062/* Digests a file and prints the result.
1063
1064
1065
1066Rivest [Page 19]
1067
1068RFC 1321 MD5 Message-Digest Algorithm April 1992
1069
1070
1071 */
1072static void MDFile (filename)
1073char *filename;
1074{
1075 FILE *file;
1076 MD_CTX context;
1077 int len;
1078 unsigned char buffer[1024], digest[16];
1079
1080 if ((file = fopen (filename, "rb")) == NULL)
1081 printf ("%s can't be opened\n", filename);
1082
1083 else {
1084 MDInit (&context);
1085 while (len = fread (buffer, 1, 1024, file))
1086 MDUpdate (&context, buffer, len);
1087 MDFinal (digest, &context);
1088
1089 fclose (file);
1090
1091 printf ("MD%d (%s) = ", MD, filename);
1092 MDPrint (digest);
1093 printf ("\n");
1094 }
1095}
1096
1097/* Digests the standard input and prints the result.
1098 */
1099static void MDFilter ()
1100{
1101 MD_CTX context;
1102 int len;
1103 unsigned char buffer[16], digest[16];
1104
1105 MDInit (&context);
1106 while (len = fread (buffer, 1, 16, stdin))
1107 MDUpdate (&context, buffer, len);
1108 MDFinal (digest, &context);
1109
1110 MDPrint (digest);
1111 printf ("\n");
1112}
1113
1114/* Prints a message digest in hexadecimal.
1115 */
1116static void MDPrint (digest)
1117unsigned char digest[16];
1118{
1119
1120
1121
1122Rivest [Page 20]
1123
1124RFC 1321 MD5 Message-Digest Algorithm April 1992
1125
1126
1127 unsigned int i;
1128
1129 for (i = 0; i < 16; i++)
1130 printf ("%02x", digest[i]);
1131}
1132
1133A.5 Test suite
1134
1135 The MD5 test suite (driver option "-x") should print the following
1136 results:
1137
1138MD5 test suite:
1139MD5 ("") = d41d8cd98f00b204e9800998ecf8427e
1140MD5 ("a") = 0cc175b9c0f1b6a831c399e269772661
1141MD5 ("abc") = 900150983cd24fb0d6963f7d28e17f72
1142MD5 ("message digest") = f96b697d7cb7938d525a2f31aaf161d0
1143MD5 ("abcdefghijklmnopqrstuvwxyz") = c3fcd3d76192e4007dfb496cca67e13b
1144MD5 ("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789") =
1145d174ab98d277d9f5a5611c2c9f419d9f
1146MD5 ("123456789012345678901234567890123456789012345678901234567890123456
114778901234567890") = 57edf4a22be3c955ac49da2e2107b67a
1148
1149Security Considerations
1150
1151 The level of security discussed in this memo is considered to be
1152 sufficient for implementing very high security hybrid digital-
1153 signature schemes based on MD5 and a public-key cryptosystem.
1154
1155Author's Address
1156
1157 Ronald L. Rivest
1158 Massachusetts Institute of Technology
1159 Laboratory for Computer Science
1160 NE43-324
1161 545 Technology Square
1162 Cambridge, MA 02139-1986
1163
1164 Phone: (617) 253-5880
1165 EMail: rivest@theory.lcs.mit.edu
1166
1167
1168
1169
1170
1171
1172
1173
1174
1175
1176
1177
1178Rivest [Page 21]
1179 \ No newline at end of file
diff --git a/src/compat/win32.cpp b/src/compat/win32.cpp
index 2a43ff9..b689322 100644
--- a/src/compat/win32.cpp
+++ b/src/compat/win32.cpp
@@ -162,5 +162,27 @@ int Bu::Winsock2::__WSAFDIsSet( SOCKET s, fd_set *set ) {
162 return (*Bu::Winsock2::_fnptr___WSAFDIsSet)( s, set ); 162 return (*Bu::Winsock2::_fnptr___WSAFDIsSet)( s, set );
163} 163}
164 164
165Bu::String Bu::getLastWinError()
166{
167 LPVOID lpMsgBuf;
168 DWORD dw = GetLastError();
169
170 FormatMessageA(
171 FORMAT_MESSAGE_ALLOCATE_BUFFER |
172 FORMAT_MESSAGE_FROM_SYSTEM |
173 FORMAT_MESSAGE_IGNORE_INSERTS,
174 NULL,
175 dw,
176 MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
177 (LPSTR) &lpMsgBuf,
178 0, NULL );
179
180 Bu::String sRet( (char *)lpMsgBuf );
181
182 LocalFree(lpMsgBuf);
183
184 return sRet;
185}
186
165#endif 187#endif
166 188
diff --git a/src/compat/win32.h b/src/compat/win32.h
index 67d80d3..1d82fbe 100644
--- a/src/compat/win32.h
+++ b/src/compat/win32.h
@@ -89,6 +89,8 @@ namespace Bu
89 static char scode[15]; 89 static char scode[15];
90 static char *gai_strerror( int iCode ); 90 static char *gai_strerror( int iCode );
91 }; 91 };
92
93 Bu::String getLastWinError();
92}; 94};
93 95
94#ifdef FD_ISSET 96#ifdef FD_ISSET
diff --git a/src/md5.cpp b/src/md5.cpp
index f971f23..71f17a0 100644
--- a/src/md5.cpp
+++ b/src/md5.cpp
@@ -11,31 +11,11 @@
11#include "bu/md5.h" 11#include "bu/md5.h"
12#include "bu/stream.h" 12#include "bu/stream.h"
13 13
14 14#ifdef SYSTEM_BIG_ENDIAN
15// This performs a wrapping bitwise shift, kinda' fun! 15# define toLittleEndian( a, b ) _toLittleEndian( a, b )
16 16#else
17#define bit_roll( num, cnt ) \ 17# define toLittleEndian( a, b ) (void)0
18 (((num) << (cnt)) | (((num) >> (32 - (cnt))) & ~(-1<<(cnt)))) 18#endif
19
20//#define md5_cmn( q, a, b, x, s, t ) (bit_roll((a + q + x + t), s) + b)
21
22// The following are handy wrappers for the cmn function
23#define md5_ff( a, b, c, d, x, s, t ) \
24 (md5_cmn((b & c) | ((~b) & d), a, b, x, s, t))
25
26#define md5_gg( a, b, c, d, x, s, t ) \
27 (md5_cmn((b & d) | (c & (~d)), a, b, x, s, t))
28
29#define md5_hh( a, b, c, d, x, s, t ) \
30 (md5_cmn(b ^ c ^ d, a, b, x, s, t))
31
32#define md5_ii( a, b, c, d, x, s, t ) \
33 (md5_cmn(c ^ (b | (~d)), a, b, x, s, t))
34
35inline long md5_cmn( long q, long a, long b, long x, long s, long t )
36{
37 return bit_roll((a + q + x + t), s) + b;
38}
39 19
40Bu::Md5::Md5() 20Bu::Md5::Md5()
41{ 21{
@@ -50,14 +30,13 @@ void Bu::Md5::reset()
50{ 30{
51 // These are the magic seed numbers... 31 // These are the magic seed numbers...
52 32
53 sum[0] = 1732584193; 33 sum[0] = 0x67452301U;
54 sum[1] = -271733879; 34 sum[1] = 0xEFCDAB89U;
55 sum[2] = -1732584194; 35 sum[2] = 0x98BADCFEU;
56 sum[3] = 271733878; 36 sum[3] = 0x10325476U;
57 37
58 iBytes = 0; 38 uBits[0] = 0;
59 memset( inbuf, 0, 4*16 ); 39 uBits[1] = 0;
60 iFill = 0;
61} 40}
62 41
63void Bu::Md5::setSalt( const Bu::String & /*sSalt*/ ) 42void Bu::Md5::setSalt( const Bu::String & /*sSalt*/ )
@@ -67,137 +46,201 @@ void Bu::Md5::setSalt( const Bu::String & /*sSalt*/ )
67void Bu::Md5::addData( const void *sVData, int iSize ) 46void Bu::Md5::addData( const void *sVData, int iSize )
68{ 47{
69 const char *sData = (const char *)sVData; 48 const char *sData = (const char *)sVData;
49 uint32_t t;
50
51 t = uBits[0];
52 if( (uBits[0] = t + ((uint32_t)iSize << 3)) < t )
53 uBits[1]++;
54 uBits[1] += iSize >> 29;
70 55
71 int iInPos = 0; 56 t = (t >> 3) & 0x3f; /* How many bytes we have buffered */
72 for(;;) 57
58 /* Handle any leading odd-sized chunks */
59 if( t )
73 { 60 {
74 for( ; iFill < 16*4 && iInPos < iSize; iFill++, iInPos++ ) 61 unsigned char *p = (unsigned char *) inbuf + t;
75 { 62
76 inbuf[iFill>>2] |= ((long)sData[iInPos]) << ((iFill*8)%32); 63 t = 64 - t;
64 if( iSize < t ) {
65 memcpy( p, sData, iSize );
66 return;
77 } 67 }
78 if( iFill < 16*4 ) 68 memcpy( p, sData, t );
79 break; 69 toLittleEndian( inbuf, 16 );
80 compBlock( inbuf, sum ); 70 compBlock( sum, (uint32_t *)inbuf );
81 memset( inbuf, 0, 4*16 ); 71 sData += t;
82 iFill = 0; 72 iSize -= t;
83 } 73 }
84 iBytes += iSize; 74
75 /* Process data in 64-byte chunks */
76 while( iSize >= 64 )
77 {
78 memcpy( inbuf, sData, 64 );
79 toLittleEndian( inbuf, 16 );
80 compBlock( sum, (uint32_t *)inbuf );
81 sData += 64;
82 iSize -= 64;
83 }
84
85 /* Handle any remaining bytes of data. */
86 memcpy( inbuf, sData, iSize );
85} 87}
86 88
87Bu::String Bu::Md5::getResult() 89Bu::String Bu::Md5::getResult()
88{ 90{
89 long lsum[4]; 91 uint32_t lsum[4];
90 compCap( lsum ); 92 compCap( lsum );
91 return Bu::String( (const char *)lsum, 4*4 ); 93 return Bu::String( (const char *)lsum, 4*4 );
92} 94}
93 95
94void Bu::Md5::writeResult( Bu::Stream &sOut ) 96void Bu::Md5::writeResult( Bu::Stream &sOut )
95{ 97{
96 long lsum[4]; 98 uint32_t lsum[4];
97 compCap( lsum ); 99 compCap( lsum );
98 sOut.write( lsum, 4*4 ); 100 sOut.write( lsum, 4*4 );
99} 101}
100 102
101void Bu::Md5::compCap( long *sumout ) 103void Bu::Md5::compCap( uint32_t *sumout )
102{ 104{
105 uint8_t tmpbuf[64];
103 memcpy( sumout, sum, 4*4 ); 106 memcpy( sumout, sum, 4*4 );
104 long lbuf[16]; 107 memcpy( tmpbuf, inbuf, 64 );
105 memcpy( lbuf, inbuf, 4*16 ); 108
106 109 uint32_t count;
107 lbuf[iFill>>2] |= 0x80 << ((iFill*8)%32); 110 uint8_t *p;
108 uint64_t iBits = iBytes*8; 111
109 if( iBytes > 0 && iFill>>2 >= 14 ) 112 /* Compute number of bytes mod 64 */
110 { 113 count = (uBits[0] >> 3) & 0x3F;
111 compBlock( lbuf, sumout ); 114
112 memset( lbuf, 0, 4*16 ); 115 /* Set the first char of padding to 0x80. This is safe since there is
113 memcpy( lbuf+14, &iBits, 8 ); 116 always at least one byte free */
114 compBlock( lbuf, sumout ); 117 p = tmpbuf + count;
115 } 118 *p++ = 0x80;
116 else 119
117 { 120 /* Bytes of padding needed to make 64 bytes */
118 memcpy( lbuf+14, &iBits, 8 ); 121 count = 64 - 1 - count;
119 compBlock( lbuf, sumout ); 122
123 /* Pad out to 56 mod 64 */
124 if (count < 8) {
125 /* Two lots of padding: Pad the first block to 64 bytes */
126 memset( p, 0, count );
127 toLittleEndian( tmpbuf, 16 );
128 compBlock( sumout, (uint32_t *)tmpbuf );
129
130 /* Now fill the next block with 56 bytes */
131 memset( tmpbuf, 0, 56);
132 } else {
133 /* Pad block to 56 bytes */
134 memset( p, 0, count - 8);
120 } 135 }
136 toLittleEndian( tmpbuf, 14 );
137
138 /* Append length in bits and transform */
139 ((uint32_t *) tmpbuf)[14] = uBits[0];
140 ((uint32_t *) tmpbuf)[15] = uBits[1];
141
142 compBlock( sumout, (uint32_t *)tmpbuf );
143 toLittleEndian((unsigned char *)sumout, 4);
121} 144}
122 145
123void Bu::Md5::compBlock( long *x, long *lsum ) 146#define F1(x, y, z) (z ^ (x & (y ^ z)))
147#define F2(x, y, z) F1(z, x, y)
148#define F3(x, y, z) (x ^ y ^ z)
149#define F4(x, y, z) (y ^ (x | ~z))
150
151/* This is the central step in the MD5 algorithm. */
152#define MD5STEP(f, w, x, y, z, data, s) \
153 ( w += f(x, y, z) + data, w = w<<s | w>>(32-s), w += x )
154
155void Bu::Md5::compBlock( uint32_t *lsum, uint32_t *x )
124{ 156{
125 long a = lsum[0]; 157 register uint32_t a, b, c, d;
126 long b = lsum[1]; 158 a = lsum[0];
127 long c = lsum[2]; 159 b = lsum[1];
128 long d = lsum[3]; 160 c = lsum[2];
129 161 d = lsum[3];
130 a = md5_ff(a, b, c, d, x[ 0], 7 , -680876936); 162
131 d = md5_ff(d, a, b, c, x[ 1], 12, -389564586); 163 MD5STEP(F1, a, b, c, d, x[0] + 0xd76aa478, 7);
132 c = md5_ff(c, d, a, b, x[ 2], 17, 606105819); 164 MD5STEP(F1, d, a, b, c, x[1] + 0xe8c7b756, 12);
133 b = md5_ff(b, c, d, a, x[ 3], 22, -1044525330); 165 MD5STEP(F1, c, d, a, b, x[2] + 0x242070db, 17);
134 a = md5_ff(a, b, c, d, x[ 4], 7 , -176418897); 166 MD5STEP(F1, b, c, d, a, x[3] + 0xc1bdceee, 22);
135 d = md5_ff(d, a, b, c, x[ 5], 12, 1200080426); 167 MD5STEP(F1, a, b, c, d, x[4] + 0xf57c0faf, 7);
136 c = md5_ff(c, d, a, b, x[ 6], 17, -1473231341); 168 MD5STEP(F1, d, a, b, c, x[5] + 0x4787c62a, 12);
137 b = md5_ff(b, c, d, a, x[ 7], 22, -45705983); 169 MD5STEP(F1, c, d, a, b, x[6] + 0xa8304613, 17);
138 a = md5_ff(a, b, c, d, x[ 8], 7 , 1770035416); 170 MD5STEP(F1, b, c, d, a, x[7] + 0xfd469501, 22);
139 d = md5_ff(d, a, b, c, x[ 9], 12, -1958414417); 171 MD5STEP(F1, a, b, c, d, x[8] + 0x698098d8, 7);
140 c = md5_ff(c, d, a, b, x[10], 17, -42063); 172 MD5STEP(F1, d, a, b, c, x[9] + 0x8b44f7af, 12);
141 b = md5_ff(b, c, d, a, x[11], 22, -1990404162); 173 MD5STEP(F1, c, d, a, b, x[10] + 0xffff5bb1, 17);
142 a = md5_ff(a, b, c, d, x[12], 7 , 1804603682); 174 MD5STEP(F1, b, c, d, a, x[11] + 0x895cd7be, 22);
143 d = md5_ff(d, a, b, c, x[13], 12, -40341101); 175 MD5STEP(F1, a, b, c, d, x[12] + 0x6b901122, 7);
144 c = md5_ff(c, d, a, b, x[14], 17, -1502002290); 176 MD5STEP(F1, d, a, b, c, x[13] + 0xfd987193, 12);
145 b = md5_ff(b, c, d, a, x[15], 22, 1236535329); 177 MD5STEP(F1, c, d, a, b, x[14] + 0xa679438e, 17);
146 178 MD5STEP(F1, b, c, d, a, x[15] + 0x49b40821, 22);
147 a = md5_gg(a, b, c, d, x[ 1], 5 , -165796510); 179
148 d = md5_gg(d, a, b, c, x[ 6], 9 , -1069501632); 180 MD5STEP(F2, a, b, c, d, x[1] + 0xf61e2562, 5);
149 c = md5_gg(c, d, a, b, x[11], 14, 643717713); 181 MD5STEP(F2, d, a, b, c, x[6] + 0xc040b340, 9);
150 b = md5_gg(b, c, d, a, x[ 0], 20, -373897302); 182 MD5STEP(F2, c, d, a, b, x[11] + 0x265e5a51, 14);
151 a = md5_gg(a, b, c, d, x[ 5], 5 , -701558691); 183 MD5STEP(F2, b, c, d, a, x[0] + 0xe9b6c7aa, 20);
152 d = md5_gg(d, a, b, c, x[10], 9 , 38016083); 184 MD5STEP(F2, a, b, c, d, x[5] + 0xd62f105d, 5);
153 c = md5_gg(c, d, a, b, x[15], 14, -660478335); 185 MD5STEP(F2, d, a, b, c, x[10] + 0x02441453, 9);
154 b = md5_gg(b, c, d, a, x[ 4], 20, -405537848); 186 MD5STEP(F2, c, d, a, b, x[15] + 0xd8a1e681, 14);
155 a = md5_gg(a, b, c, d, x[ 9], 5 , 568446438); 187 MD5STEP(F2, b, c, d, a, x[4] + 0xe7d3fbc8, 20);
156 d = md5_gg(d, a, b, c, x[14], 9 , -1019803690); 188 MD5STEP(F2, a, b, c, d, x[9] + 0x21e1cde6, 5);
157 c = md5_gg(c, d, a, b, x[ 3], 14, -187363961); 189 MD5STEP(F2, d, a, b, c, x[14] + 0xc33707d6, 9);
158 b = md5_gg(b, c, d, a, x[ 8], 20, 1163531501); 190 MD5STEP(F2, c, d, a, b, x[3] + 0xf4d50d87, 14);
159 a = md5_gg(a, b, c, d, x[13], 5 , -1444681467); 191 MD5STEP(F2, b, c, d, a, x[8] + 0x455a14ed, 20);
160 d = md5_gg(d, a, b, c, x[ 2], 9 , -51403784); 192 MD5STEP(F2, a, b, c, d, x[13] + 0xa9e3e905, 5);
161 c = md5_gg(c, d, a, b, x[ 7], 14, 1735328473); 193 MD5STEP(F2, d, a, b, c, x[2] + 0xfcefa3f8, 9);
162 b = md5_gg(b, c, d, a, x[12], 20, -1926607734); 194 MD5STEP(F2, c, d, a, b, x[7] + 0x676f02d9, 14);
163 195 MD5STEP(F2, b, c, d, a, x[12] + 0x8d2a4c8a, 20);
164 a = md5_hh(a, b, c, d, x[ 5], 4 , -378558); 196
165 d = md5_hh(d, a, b, c, x[ 8], 11, -2022574463); 197 MD5STEP(F3, a, b, c, d, x[5] + 0xfffa3942, 4);
166 c = md5_hh(c, d, a, b, x[11], 16, 1839030562); 198 MD5STEP(F3, d, a, b, c, x[8] + 0x8771f681, 11);
167 b = md5_hh(b, c, d, a, x[14], 23, -35309556); 199 MD5STEP(F3, c, d, a, b, x[11] + 0x6d9d6122, 16);
168 a = md5_hh(a, b, c, d, x[ 1], 4 , -1530992060); 200 MD5STEP(F3, b, c, d, a, x[14] + 0xfde5380c, 23);
169 d = md5_hh(d, a, b, c, x[ 4], 11, 1272893353); 201 MD5STEP(F3, a, b, c, d, x[1] + 0xa4beea44, 4);
170 c = md5_hh(c, d, a, b, x[ 7], 16, -155497632); 202 MD5STEP(F3, d, a, b, c, x[4] + 0x4bdecfa9, 11);
171 b = md5_hh(b, c, d, a, x[10], 23, -1094730640); 203 MD5STEP(F3, c, d, a, b, x[7] + 0xf6bb4b60, 16);
172 a = md5_hh(a, b, c, d, x[13], 4 , 681279174); 204 MD5STEP(F3, b, c, d, a, x[10] + 0xbebfbc70, 23);
173 d = md5_hh(d, a, b, c, x[ 0], 11, -358537222); 205 MD5STEP(F3, a, b, c, d, x[13] + 0x289b7ec6, 4);
174 c = md5_hh(c, d, a, b, x[ 3], 16, -722521979); 206 MD5STEP(F3, d, a, b, c, x[0] + 0xeaa127fa, 11);
175 b = md5_hh(b, c, d, a, x[ 6], 23, 76029189); 207 MD5STEP(F3, c, d, a, b, x[3] + 0xd4ef3085, 16);
176 a = md5_hh(a, b, c, d, x[ 9], 4 , -640364487); 208 MD5STEP(F3, b, c, d, a, x[6] + 0x04881d05, 23);
177 d = md5_hh(d, a, b, c, x[12], 11, -421815835); 209 MD5STEP(F3, a, b, c, d, x[9] + 0xd9d4d039, 4);
178 c = md5_hh(c, d, a, b, x[15], 16, 530742520); 210 MD5STEP(F3, d, a, b, c, x[12] + 0xe6db99e5, 11);
179 b = md5_hh(b, c, d, a, x[ 2], 23, -995338651); 211 MD5STEP(F3, c, d, a, b, x[15] + 0x1fa27cf8, 16);
180 212 MD5STEP(F3, b, c, d, a, x[2] + 0xc4ac5665, 23);
181 a = md5_ii(a, b, c, d, x[ 0], 6 , -198630844); 213
182 d = md5_ii(d, a, b, c, x[ 7], 10, 1126891415); 214 MD5STEP(F4, a, b, c, d, x[0] + 0xf4292244, 6);
183 c = md5_ii(c, d, a, b, x[14], 15, -1416354905); 215 MD5STEP(F4, d, a, b, c, x[7] + 0x432aff97, 10);
184 b = md5_ii(b, c, d, a, x[ 5], 21, -57434055); 216 MD5STEP(F4, c, d, a, b, x[14] + 0xab9423a7, 15);
185 a = md5_ii(a, b, c, d, x[12], 6 , 1700485571); 217 MD5STEP(F4, b, c, d, a, x[5] + 0xfc93a039, 21);
186 d = md5_ii(d, a, b, c, x[ 3], 10, -1894986606); 218 MD5STEP(F4, a, b, c, d, x[12] + 0x655b59c3, 6);
187 c = md5_ii(c, d, a, b, x[10], 15, -1051523); 219 MD5STEP(F4, d, a, b, c, x[3] + 0x8f0ccc92, 10);
188 b = md5_ii(b, c, d, a, x[ 1], 21, -2054922799); 220 MD5STEP(F4, c, d, a, b, x[10] + 0xffeff47d, 15);
189 a = md5_ii(a, b, c, d, x[ 8], 6 , 1873313359); 221 MD5STEP(F4, b, c, d, a, x[1] + 0x85845dd1, 21);
190 d = md5_ii(d, a, b, c, x[15], 10, -30611744); 222 MD5STEP(F4, a, b, c, d, x[8] + 0x6fa87e4f, 6);
191 c = md5_ii(c, d, a, b, x[ 6], 15, -1560198380); 223 MD5STEP(F4, d, a, b, c, x[15] + 0xfe2ce6e0, 10);
192 b = md5_ii(b, c, d, a, x[13], 21, 1309151649); 224 MD5STEP(F4, c, d, a, b, x[6] + 0xa3014314, 15);
193 a = md5_ii(a, b, c, d, x[ 4], 6 , -145523070); 225 MD5STEP(F4, b, c, d, a, x[13] + 0x4e0811a1, 21);
194 d = md5_ii(d, a, b, c, x[11], 10, -1120210379); 226 MD5STEP(F4, a, b, c, d, x[4] + 0xf7537e82, 6);
195 c = md5_ii(c, d, a, b, x[ 2], 15, 718787259); 227 MD5STEP(F4, d, a, b, c, x[11] + 0xbd3af235, 10);
196 b = md5_ii(b, c, d, a, x[ 9], 21, -343485551); 228 MD5STEP(F4, c, d, a, b, x[2] + 0x2ad7d2bb, 15);
197 229 MD5STEP(F4, b, c, d, a, x[9] + 0xeb86d391, 21);
198 lsum[0] = a + lsum[0]; 230
199 lsum[1] = b + lsum[1]; 231 lsum[0] += a;
200 lsum[2] = c + lsum[2]; 232 lsum[1] += b;
201 lsum[3] = d + lsum[3]; 233 lsum[2] += c;
234 lsum[3] += d;
202} 235}
203 236
237void Bu::Md5::_toLittleEndian( uint8_t *buf, uint32_t count )
238{
239 uint32_t t;
240 do {
241 t = (uint32_t) ((unsigned) buf[3] << 8 | buf[2]) << 16 |
242 ((unsigned) buf[1] << 8 | buf[0]);
243 *(uint32_t *) buf = t;
244 buf += 4;
245 } while( --count );
246}
diff --git a/src/md5.h b/src/md5.h
index d4a82ec..b7597fd 100644
--- a/src/md5.h
+++ b/src/md5.h
@@ -14,6 +14,8 @@ namespace Bu
14{ 14{
15 /** 15 /**
16 * Class for easily calculating MD5 sums of just about any data. 16 * Class for easily calculating MD5 sums of just about any data.
17 * This code is based on some public domain code written by Colin Plumb in
18 * 1993.
17 *@author Mike Buland 19 *@author Mike Buland
18 */ 20 */
19 class Md5 : public Bu::CryptoHash 21 class Md5 : public Bu::CryptoHash
@@ -36,13 +38,16 @@ namespace Bu
36 /** 38 /**
37 * Compute one block of input data. 39 * Compute one block of input data.
38 */ 40 */
39 void compBlock( long *x, long *lsum ); 41 void compBlock( uint32_t *lsum, uint32_t *x );
40 void compCap( long *sumout ); 42 void compCap( uint32_t *sumout );
43
44 void _addData( uint8_t *target, int &iCurFill, const void *sData,
45 int iSize );
46 void _toLittleEndian( uint8_t *buf, uint32_t count );
41 47
42 long inbuf[16]; 48 uint8_t inbuf[64];
43 long iFill; 49 uint32_t sum[4];
44 long sum[4]; 50 uint32_t uBits[2];
45 uint64_t iBytes;
46 }; 51 };
47}; 52};
48 53
diff --git a/src/plugger.cpp b/src/plugger.cpp
index 6e49b4e..fb6850a 100644
--- a/src/plugger.cpp
+++ b/src/plugger.cpp
@@ -7,9 +7,5 @@
7 7
8#include "bu/plugger.h" 8#include "bu/plugger.h"
9 9
10#ifndef WIN32
11
12namespace Bu { subExceptionDef( PluginException ) } 10namespace Bu { subExceptionDef( PluginException ) }
13 11
14#endif
15
diff --git a/src/plugger.h b/src/plugger.h
index 65ac4bb..2ba7b4b 100644
--- a/src/plugger.h
+++ b/src/plugger.h
@@ -8,15 +8,20 @@
8#ifndef BU_PLUGGER_H 8#ifndef BU_PLUGGER_H
9#define BU_PLUGGER_H 9#define BU_PLUGGER_H
10 10
11#ifndef WIN32 //yeah, this one is going to take some work...
12
13#include "bu/hash.h" 11#include "bu/hash.h"
14#include "bu/list.h" 12#include "bu/list.h"
15#include <dlfcn.h>
16#include "bu/exceptionbase.h" 13#include "bu/exceptionbase.h"
17#include "bu/string.h" 14#include "bu/string.h"
18#include <stddef.h> 15#include <stddef.h>
19 16
17#include "bu/config.h"
18
19#ifdef WIN32
20# include <windows.h>
21#else
22# include <dlfcn.h>
23#endif
24
20namespace Bu 25namespace Bu
21{ 26{
22 subExceptionDecl( PluginException ); 27 subExceptionDecl( PluginException );
@@ -34,7 +39,11 @@ namespace Bu
34 typedef struct PluginReg 39 typedef struct PluginReg
35 { 40 {
36 bool bBuiltin; 41 bool bBuiltin;
42#ifdef WIN32
43 HMODULE dlHandle;
44#else
37 void *dlHandle; 45 void *dlHandle;
46#endif
38 PluginInfo *pInfo; 47 PluginInfo *pInfo;
39 } PluginReg; 48 } PluginReg;
40 49
@@ -69,6 +78,20 @@ namespace Bu
69 (void (*)( void * ))(destroy ##classname) }; \ 78 (void (*)( void * ))(destroy ##classname) }; \
70 } 79 }
71 80
81//
82// This is probably the main interface to use, I'll describe it some here...
83// structname - The name of the structure, this is what you have to pass to
84// register. Depending on how you build your dll/so files this
85// will need to be unique (generally not)
86// pluginname - This is what will be used by the plugin system to refer to
87// your plugin once it's loaded. This should be unique, but not
88// a string
89// classname - The name of the class that is the plugin
90// baseclass - The name of the base class that is the parent of the plugin
91// name - The name of the author of this class (or company)
92// ver - an integer version number for the plugin
93// rev - an integer revision number for the plugin
94//
72#define PluginInterface3( structname, pluginname, classname, baseclass, name, ver, rev ) \ 95#define PluginInterface3( structname, pluginname, classname, baseclass, name, ver, rev ) \
73 extern "C" { \ 96 extern "C" { \
74 baseclass *create ##classname() \ 97 baseclass *create ##classname() \
@@ -85,6 +108,38 @@ namespace Bu
85 (void (*)( void * ))(destroy ##classname) }; \ 108 (void (*)( void * ))(destroy ##classname) }; \
86 } 109 }
87 110
111 /**
112 * A complete dynamic plugin manager system. This will allow you to design
113 * and use plugins that are compiled into your program and dynamically
114 * linked to your program interchangably. It works on windows and on *nix
115 * and bsd type systems (anything that supports dlopen). Basically you
116 * create a base class that will be the basic interface of your plugins.
117 * Then you create some classes that inherit from it, and use the
118 * PluginInterface3 macro to create the required data structures for it.
119 *
120 * Once you have plugins you can create a Plugger, by passing in the base
121 * class as it's template parameter. Once it's created, you can register
122 * plugins. To register a plugin that is builtin, you just need to pass
123 * a pointer to it's interface structure to the registerBuiltinPlugin
124 * function. To register a plugin that is in a shared object or dll file
125 * you just pass the filename (with path, probably), and the name of the
126 * structure to load and you're all set.
127 *
128 * To instantiate an object from a plugin simply call instantiate with the
129 * name of the plugin as specified in the interface macro. To destroy an
130 * object crated with the plugger do not delete it, instead pass it into
131 * Plugger's destroy function.
132 *
133 * Any objects not destroyed when the plugger is deleted will be destroyed
134 * automatically.
135 *
136 * It is important to note that some systems (linux at least) partition off
137 * the memory allocated by objects linked in at run time into a seperate
138 * segment that, while it can be accessed by the main program, cannot be
139 * safely or reliably freed by the main program. With that in mind it is
140 * a good idea to free all memory allocated by a plugin object in the plugin
141 * object and not allow the calling program to delete it.
142 */
88 template<class T> 143 template<class T>
89 class Plugger 144 class Plugger
90 { 145 {
@@ -111,7 +166,11 @@ namespace Bu
111 { 166 {
112 if( (*i)->bBuiltin == false ) 167 if( (*i)->bBuiltin == false )
113 { 168 {
169#ifdef WIN32
170 FreeLibrary( (*i)->dlHandle );
171#else
114 dlclose( (*i)->dlHandle ); 172 dlclose( (*i)->dlHandle );
173#endif
115 } 174 }
116 delete (*i); 175 delete (*i);
117 } 176 }
@@ -129,32 +188,42 @@ namespace Bu
129 const Bu::String &sPluginName ) 188 const Bu::String &sPluginName )
130 { 189 {
131 PluginReg *pReg; 190 PluginReg *pReg;
132 try { 191 if( hPlugin.has( sPluginName ) )
133 pReg = (PluginReg *)hPlugin[sPluginName]; 192 throw Bu::ExceptionBase("A plugin with name '%s' is already "
134 hPlugin.erase( sPluginName ); 193 "loaded.", sPluginName.getStr() );
135 dlclose( pReg->dlHandle );
136 delete pReg;
137 pReg = NULL;
138 } catch( Bu::HashException &e )
139 {
140 }
141 194
142 pReg = new PluginReg; 195 pReg = new PluginReg;
143 196
144 pReg->bBuiltin = false; 197 pReg->bBuiltin = false;
198#ifdef WIN32
199 pReg->dlHandle = LoadLibrary( sFName.getStr() );
200 if( pReg->dlHandle == NULL )
201 {
202 throw PluginException( 1, "Error opening %s: %s",
203 sFName.getStr(), Bu::getLastWinError().getStr() );
204 }
205 pReg->pInfo = (PluginInfo *)GetProcAddress( pReg->dlHandle,
206 sPluginName.getStr() );
207 if( pReg->pInfo == NULL )
208 {
209 throw PluginException( 2, "Error mapping %s: %s",
210 sFName.getStr(), Bu::getLastWinError().getStr() );
211 }
212#else
145 pReg->dlHandle = dlopen( sFName.getStr(), RTLD_NOW ); 213 pReg->dlHandle = dlopen( sFName.getStr(), RTLD_NOW );
146 if( pReg->dlHandle == NULL ) 214 if( pReg->dlHandle == NULL )
147 { 215 {
148 throw PluginException( 1, "Error on %s: %s", sFName.getStr(), 216 throw PluginException( 1, "Error opening %s: %s",
149 dlerror() ); 217 sFName.getStr(), dlerror() );
150 } 218 }
151 pReg->pInfo = (PluginInfo *)dlsym( pReg->dlHandle, 219 pReg->pInfo = (PluginInfo *)dlsym( pReg->dlHandle,
152 sPluginName.getStr() ); 220 sPluginName.getStr() );
153 if( pReg->pInfo == NULL ) 221 if( pReg->pInfo == NULL )
154 { 222 {
155 throw PluginException( 2, "Error on %s: %s", sFName.getStr(), 223 throw PluginException( 2, "Error mapping %s: %s",
156 dlerror() ); 224 sFName.getStr(), dlerror() );
157 } 225 }
226#endif
158 hPlugin.insert( pReg->pInfo->sID, pReg ); 227 hPlugin.insert( pReg->pInfo->sID, pReg );
159 } 228 }
160 229
@@ -195,7 +264,11 @@ namespace Bu
195 { 264 {
196 if( (*i)->bBuiltin == false ) 265 if( (*i)->bBuiltin == false )
197 { 266 {
267#ifdef WIN32
268 FreeLibrary( (*i)->dlHandle );
269#else
198 dlclose( (*i)->dlHandle ); 270 dlclose( (*i)->dlHandle );
271#endif
199 } 272 }
200 delete (*i); 273 delete (*i);
201 } 274 }
@@ -213,6 +286,4 @@ namespace Bu
213 }; 286 };
214} 287}
215 288
216#endif //#ifndef WIN32 //yeah, this one is going to take some work...
217
218#endif 289#endif
diff --git a/src/tcpsocket.h b/src/tcpsocket.h
index a05580a..dbaaa5e 100644
--- a/src/tcpsocket.h
+++ b/src/tcpsocket.h
@@ -66,7 +66,6 @@ namespace Bu
66 virtual ~TcpSocket(); 66 virtual ~TcpSocket();
67 67
68 virtual void close(); 68 virtual void close();
69 //virtual void read();
70 virtual size read( void *pBuf, size nBytes ); 69 virtual size read( void *pBuf, size nBytes );
71 virtual size read( void *pBuf, size nBytes, 70 virtual size read( void *pBuf, size nBytes,
72 uint32_t nSec, uint32_t nUSec=0 ); 71 uint32_t nSec, uint32_t nUSec=0 );
diff --git a/src/tests/udpsocket.cpp b/src/tests/udpsocket.cpp
new file mode 100644
index 0000000..2a74acf
--- /dev/null
+++ b/src/tests/udpsocket.cpp
@@ -0,0 +1,86 @@
1/*
2 * Copyright (C) 2007-2011 Xagasoft, All rights reserved.
3 *
4 * This file is part of the libbu++ library and is released under the
5 * terms of the license contained in the file LICENSE.
6 */
7
8#include "bu/udpsocket.h"
9#include "bu/sio.h"
10
11#include <errno.h>
12#include <arpa/inet.h>
13#include <sys/socket.h>
14#include <netinet/in.h>
15#include <sys/utsname.h>
16
17using namespace Bu;
18
19int main( int argc, char *argv[] )
20{
21 sio << Fmt::hex(8) << INADDR_ANY << sio.nl << Fmt::hex(8) << inet_addr("0.0.0.0") << sio.nl;
22 if( argc == 1 )
23 {
24 sio << "Options are 'l' for listening and 'b' for broadcasting."
25 << sio.nl;
26 }
27 else if( argv[1][0] == 'l' )
28 {
29 sio << "Listening..." << sio.nl;
30 Bu::UdpSocket udp( "0.0.0.0", 6688, UdpSocket::Read|UdpSocket::Broadcast );
31
32 for(;;)
33 {
34 char buf[1501];
35 int iRead = udp.read( buf, 1500 );
36 if( iRead >= 0 )
37 {
38 buf[iRead] = '\0';
39 sio << "Read(" << iRead << "): '" << buf << "'" << sio.nl;
40 }
41 else
42 {
43 sio << "Got " << iRead << ": " << strerror( errno ) << sio.nl;
44 }
45 }
46 }
47 else if( argv[1][0] == 'L' )
48 {
49 sio << "Listening..." << sio.nl;
50 Bu::UdpSocket udp( "0.0.0.0", 6688, UdpSocket::Read|UdpSocket::Broadcast );
51
52 for(;;)
53 {
54 char buf[1501];
55 Bu::UdpSocket::addr aHost;
56 int iPort;
57 int iRead = udp.read( buf, 1500, aHost, iPort );
58 if( iRead >= 0 )
59 {
60 buf[iRead] = '\0';
61 sio << "Read(" << iRead << ") from " << Bu::UdpSocket::addrToStr( aHost ) << ":" << iPort << ": '" << buf << "'" << sio.nl;
62 }
63 }
64 }
65 else if( argv[1][0] == 'b' )
66 {
67 sio << "Broadcasting..." << sio.nl;
68 Bu::UdpSocket udp("255.255.255.255", 6688,
69 UdpSocket::Write|UdpSocket::Broadcast );
70
71 for(;;)
72 {
73 int iWrote = udp.write("hello", 5 );
74 sio << "Wrote(" << iWrote << "): " << strerror( errno ) << sio.nl;
75 usleep( 250000 );
76 }
77 }
78 else
79 {
80 sio << "Options are 'l' for listening and 'b' for broadcasting."
81 << sio.nl;
82 }
83
84 return 0;
85}
86
diff --git a/src/udpsocket.cpp b/src/udpsocket.cpp
new file mode 100644
index 0000000..91e04c1
--- /dev/null
+++ b/src/udpsocket.cpp
@@ -0,0 +1,240 @@
1/*
2 * Copyright (C) 2007-2011 Xagasoft, All rights reserved.
3 *
4 * This file is part of the libbu++ library and is released under the
5 * terms of the license contained in the file LICENSE.
6 */
7
8#include "bu/udpsocket.h"
9
10#include "bu/sio.h"
11using namespace Bu;
12#include <fcntl.h>
13
14#include <errno.h>
15#include <arpa/inet.h>
16#include <sys/socket.h>
17#include <netinet/in.h>
18#include <sys/utsname.h>
19
20namespace Bu { subExceptionDef( UdpSocketException ) }
21
22#define saTarget ( *((struct sockaddr_in *)paTarget) )
23
24Bu::UdpSocket::UdpSocket( int iUdpSocket ) :
25 iUdpSocket( iUdpSocket ),
26 paTarget( NULL ),
27 bBound( false )
28{
29}
30
31Bu::UdpSocket::UdpSocket( const Bu::String &sAddr, int iPort, int iFlags ) :
32 iUdpSocket( 0 ),
33 paTarget( NULL ),
34 bBound( false )
35{
36 iUdpSocket = socket( PF_INET, SOCK_DGRAM, 0 );
37 if( iUdpSocket < 0 )
38 {
39 throw UdpSocketException("Couldn't open udp socket: %s",
40 strerror( errno )
41 );
42 }
43
44 if( (iFlags&Broadcast) )
45 {
46 int broadcast = 1;
47 if( (setsockopt( iUdpSocket, SOL_SOCKET, SO_BROADCAST,
48 &broadcast, sizeof(broadcast) )) == -1)
49 {
50 throw UdpSocketException("Couldn't set udp socket to broadcast: %s",
51 strerror( errno )
52 );
53 }
54 }
55
56 paTarget = new struct sockaddr_in;
57 saTarget.sin_family = AF_INET;
58 saTarget.sin_port = htons( iPort );
59 saTarget.sin_addr.s_addr = inet_addr( sAddr.getStr() ); // INADDR_ANY;
60 memset( saTarget.sin_zero, '\0', sizeof(saTarget.sin_zero) );
61
62 if( (iFlags&Read) )
63 {
64 if( bind( iUdpSocket, (struct sockaddr*)paTarget, sizeof(struct sockaddr_in) )
65 == -1 )
66 {
67 throw UdpSocketException("Couldn't bind port to udp socket: %s",
68 strerror( errno )
69 );
70 }
71 bBound = true;
72 }
73}
74
75Bu::UdpSocket::~UdpSocket()
76{
77 close();
78 delete (struct sockaddr_in *)paTarget;
79 paTarget = NULL;
80}
81
82Bu::String Bu::UdpSocket::addrToStr( const addr &a )
83{
84 Bu::String sOut;
85 sOut.format("%d.%d.%d.%d",
86 (a&0xff),
87 (a&0xff00)>>8,
88 (a&0xff0000)>>16,
89 (a&0xff000000)>>24
90 );
91
92 return sOut;
93}
94
95void Bu::UdpSocket::close()
96{
97 ::close( iUdpSocket );
98}
99
100Bu::size Bu::UdpSocket::read( void *pBuf, Bu::size nBytes )
101{
102 return recv( iUdpSocket, pBuf, nBytes, 0 );
103}
104
105Bu::size Bu::UdpSocket::read( void *pBuf, Bu::size nBytes,
106 Bu::UdpSocket::addr &aHost, int &iPort )
107{
108 sockaddr_in name;
109 size_t size = sizeof(name);
110 Bu::size ret = recvfrom( iUdpSocket, pBuf, nBytes, 0,
111 (struct sockaddr *)&name, &size );
112 aHost = name.sin_addr.s_addr;
113 iPort = ntohs(name.sin_port);
114 return ret;
115}
116
117Bu::size Bu::UdpSocket::write( const void *pBuf, Bu::size nBytes )
118{
119 if( bBound )
120 {
121 return sendto( iUdpSocket, pBuf, nBytes, 0, NULL, 0 );
122 }
123 else
124 {
125 return sendto( iUdpSocket, pBuf, nBytes, 0,
126 (struct sockaddr*)paTarget, sizeof(struct sockaddr_in) );
127 }
128}
129
130Bu::size Bu::UdpSocket::tell()
131{
132 throw Bu::UnsupportedException();
133}
134
135void Bu::UdpSocket::seek( Bu::size )
136{
137 throw Bu::UnsupportedException();
138}
139
140void Bu::UdpSocket::setPos( Bu::size )
141{
142 throw Bu::UnsupportedException();
143}
144
145void Bu::UdpSocket::setPosEnd( Bu::size )
146{
147 throw Bu::UnsupportedException();
148}
149
150bool Bu::UdpSocket::isEos()
151{
152 return false;
153}
154
155bool Bu::UdpSocket::isOpen()
156{
157 return true;
158}
159
160void Bu::UdpSocket::flush()
161{
162}
163
164bool Bu::UdpSocket::canRead()
165{
166 return bBound;
167}
168
169bool Bu::UdpSocket::canWrite()
170{
171 return true;
172}
173
174bool Bu::UdpSocket::isReadable()
175{
176 return bBound;
177}
178
179bool Bu::UdpSocket::isWritable()
180{
181 return true;
182}
183
184bool Bu::UdpSocket::isSeekable()
185{
186 return false;
187}
188
189bool Bu::UdpSocket::isBlocking()
190{
191 return true;
192}
193
194void Bu::UdpSocket::setBlocking( bool bBlocking )
195{
196#ifndef WIN32
197 if( bBlocking )
198 {
199 fcntl( iUdpSocket, F_SETFL, fcntl( iUdpSocket, F_GETFL, 0 ) & (~O_NONBLOCK) );
200 }
201 else
202 {
203 fcntl( iUdpSocket, F_SETFL, fcntl( iUdpSocket, F_GETFL, 0 ) | O_NONBLOCK );
204 }
205#else
206 u_long iMode;
207 if( bBlocking )
208 iMode = 0;
209 else
210 iMode = 1;
211 //-------------------------
212 // Set the socket I/O mode: In this case FIONBIO
213 // enables or disables the blocking mode for the
214 // socket based on the numerical value of iMode.
215 // If iMode = 0, blocking is enabled;
216 // If iMode != 0, non-blocking mode is enabled.
217 bu_ioctlsocket(iUdpSocket, FIONBIO, &iMode);
218#endif
219}
220
221void Bu::UdpSocket::setSize( Bu::size )
222{
223 throw Bu::UnsupportedException();
224}
225
226Bu::size Bu::UdpSocket::getSize() const
227{
228 throw Bu::UnsupportedException();
229}
230
231Bu::size Bu::UdpSocket::getBlockSize() const
232{
233 return 1500;
234}
235
236Bu::String Bu::UdpSocket::getLocation() const
237{
238 throw Bu::UnsupportedException();
239}
240
diff --git a/src/udpsocket.h b/src/udpsocket.h
new file mode 100644
index 0000000..8fe114d
--- /dev/null
+++ b/src/udpsocket.h
@@ -0,0 +1,81 @@
1/*
2 * Copyright (C) 2007-2011 Xagasoft, All rights reserved.
3 *
4 * This file is part of the libbu++ library and is released under the
5 * terms of the license contained in the file LICENSE.
6 */
7
8#ifndef BU_UDP_SOCKET_H
9#define BU_UDP_SOCKET_H
10
11#include <stdint.h>
12
13#include "bu/stream.h"
14
15namespace Bu
16{
17 subExceptionDecl( UdpSocketException );
18
19 class UdpSocket : public Stream
20 {
21 public:
22 UdpSocket( int iUdpSocket );
23 UdpSocket( const Bu::String &sAddr, int iPort, int iFlags );
24 virtual ~UdpSocket();
25
26 typedef uint32_t addr;
27
28 static Bu::String addrToStr( const addr &a );
29
30 virtual void close();
31 virtual Bu::size read( void *pBuf, Bu::size nBytes );
32 virtual Bu::size read( void *pBuf, Bu::size nBytes,
33 addr &sHost, int &iPort );
34 virtual Bu::size write( const void *pBuf, Bu::size nBytes );
35 using Stream::write;
36
37 virtual Bu::size tell();
38 virtual void seek( Bu::size offset );
39 virtual void setPos( Bu::size pos );
40 virtual void setPosEnd( Bu::size pos );
41 virtual bool isEos();
42 virtual bool isOpen();
43
44 virtual void flush();
45
46 virtual bool canRead();
47 virtual bool canWrite();
48
49 virtual bool isReadable();
50 virtual bool isWritable();
51 virtual bool isSeekable();
52
53 virtual bool isBlocking();
54 virtual void setBlocking( bool bBlocking=true );
55
56 virtual void setSize( Bu::size iSize );
57
58 enum {
59 // Flags
60 Read = 0x01, ///< Open udp socket for reading
61 Write = 0x02, ///< Open udp socket for writing
62 ReadWrite = 0x03, ///< Open for both read and write
63 Broadcast = 0x04, ///< Open for broadcast
64 };
65
66 virtual size getSize() const;
67 virtual size getBlockSize() const;
68 virtual Bu::String getLocation() const;
69
70 private:
71#ifdef WIN32
72 unsigned int iUdpSocket;
73#else
74 int iUdpSocket;
75#endif
76 void *paTarget;
77 bool bBound;
78 };
79};
80
81#endif
diff --git a/src/unit/md5.unit b/src/unit/md5.unit
new file mode 100644
index 0000000..248aaaf
--- /dev/null
+++ b/src/unit/md5.unit
@@ -0,0 +1,82 @@
1// vim: syntax=cpp
2/*
3 * Copyright (C) 2007-2010 Xagasoft, All rights reserved.
4 *
5 * This file is part of the libbu++ library and is released under the
6 * terms of the license contained in the file LICENSE.
7 */
8
9#include "bu/md5.h"
10
11#include <stdlib.h>
12
13suite Md5
14{
15 test basics
16 {
17#define tryStr( a, b ) \
18 { Bu::Md5 m; m.addData(a); unitTest( m.getHexResult() == b ); } (void)0
19 tryStr("", "d41d8cd98f00b204e9800998ecf8427e");
20 tryStr("a", "0cc175b9c0f1b6a831c399e269772661");
21 tryStr("abc", "900150983cd24fb0d6963f7d28e17f72");
22 tryStr("message digest", "f96b697d7cb7938d525a2f31aaf161d0");
23 tryStr("abcdefghijklmnopqrstuvwxyz",
24 "c3fcd3d76192e4007dfb496cca67e13b");
25 tryStr("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789",
26 "d174ab98d277d9f5a5611c2c9f419d9f");
27 tryStr("12345678901234567890123456789012345"
28 "678901234567890123456789012345678901234567890",
29 "57edf4a22be3c955ac49da2e2107b67a");
30 }
31
32 test twoChunks
33 {
34 Bu::Md5 m;
35 m.addData("12345678901234567890123456789012345");
36 m.addData("678901234567890123456789012345678901234567890");
37 unitTest( m.getHexResult() == "57edf4a22be3c955ac49da2e2107b67a" );
38 }
39
40 test biggerBlocks
41 {
42 const char *sums[33] = {
43 "75fcf199abe516903321095a588b938d",
44 "e26a863c96d6bdba6601175aedaae108",
45 "2b207fdcb222078d3ebfeb8d5e7c9315",
46 "b08683aaa465add72cc2b43ae42f4f70",
47 "638bb73963b2d925771c3579ccb5e879",
48 "c727bd4b48a88e3df5924a2604de0790",
49 "f33d21203c80490f7342e5853c5550eb",
50 "db449faca66a177aae59b1e36a19d053",
51 "c800d429afb5f5c820f75c2c94e2e2bb",
52 "43b79c70b9a6a11e823ffbfa0f45a4db",
53 "0177ffc483cf598ae3966b3a5ae00c8c",
54 "1a68fdf4b17a3820d48d101e9355a818"
55 };
56
57 char block[128];
58 for( int i = 0; i < 128; i++ )
59 block[i] = i*2;
60
61 const char **curSum = sums;
62 for( int j = 1; j < 4096; j*=2 )
63 {
64 /*
65 Bu::File fOut("temp", Bu::File::WriteNew );
66 for( int b = 0; b < j; b++ )
67 {
68 fOut.write( block, 128 );
69 }
70 fOut.close();
71 system("md5sum -b temp");
72 */
73 Bu::Md5 m;
74 for( int b = 0; b < j; b++ )
75 {
76 m.addData( block, 128 );
77 }
78 unitTest( m.getHexResult() == *curSum );
79 curSum++;
80 }
81 }
82}