]>
Commit | Line | Data |
---|---|---|
1 | // -*- mode: cpp; mode: fold -*- | |
2 | // Description /*{{{*/ | |
3 | // $Id: md5.h,v 1.6 2001/05/07 05:06:52 jgg Exp $ | |
4 | /* ###################################################################### | |
5 | ||
6 | MD5SumValue - Storage for a MD5Sum | |
7 | MD5Summation - MD5 Message Digest Algorithm. | |
8 | ||
9 | This is a C++ interface to a set of MD5Sum functions. The class can | |
10 | store a MD5Sum in 16 bytes of memory. | |
11 | ||
12 | A MD5Sum is used to generate a (hopefully) unique 16 byte number for a | |
13 | block of data. This can be used to gaurd against corruption of a file. | |
14 | MD5 should not be used for tamper protection, use SHA or something more | |
15 | secure. | |
16 | ||
17 | There are two classes because computing a MD5 is not a continual | |
18 | operation unless 64 byte blocks are used. Also the summation requires an | |
19 | extra 18*4 bytes to operate. | |
20 | ||
21 | ##################################################################### */ | |
22 | /*}}}*/ | |
23 | #ifndef APTPKG_MD5_H | |
24 | #define APTPKG_MD5_H | |
25 | ||
26 | ||
27 | #include <string> | |
28 | #include <cstring> | |
29 | #include <algorithm> | |
30 | #include <stdint.h> | |
31 | ||
32 | using std::string; | |
33 | using std::min; | |
34 | ||
35 | class MD5Summation; | |
36 | ||
37 | class MD5SumValue | |
38 | { | |
39 | friend class MD5Summation; | |
40 | unsigned char Sum[4*4]; | |
41 | ||
42 | public: | |
43 | ||
44 | // Accessors | |
45 | bool operator ==(const MD5SumValue &rhs) const; | |
46 | string Value() const; | |
47 | inline void Value(unsigned char S[16]) | |
48 | {for (int I = 0; I != sizeof(Sum); I++) S[I] = Sum[I];}; | |
49 | inline operator string() const {return Value();}; | |
50 | bool Set(string Str); | |
51 | inline void Set(unsigned char S[16]) | |
52 | {for (int I = 0; I != sizeof(Sum); I++) Sum[I] = S[I];}; | |
53 | ||
54 | MD5SumValue(string Str); | |
55 | MD5SumValue(); | |
56 | }; | |
57 | ||
58 | class MD5Summation | |
59 | { | |
60 | uint32_t Buf[4]; | |
61 | unsigned char Bytes[2*4]; | |
62 | unsigned char In[16*4]; | |
63 | bool Done; | |
64 | ||
65 | public: | |
66 | ||
67 | bool Add(const unsigned char *Data,unsigned long Size); | |
68 | inline bool Add(const char *Data) {return Add((unsigned char *)Data,strlen(Data));}; | |
69 | bool AddFD(int Fd,unsigned long Size); | |
70 | inline bool Add(const unsigned char *Beg,const unsigned char *End) | |
71 | {return Add(Beg,End-Beg);}; | |
72 | MD5SumValue Result(); | |
73 | ||
74 | MD5Summation(); | |
75 | }; | |
76 | ||
77 | #endif |