]>
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 | #ifdef __GNUG__ | |
27 | #pragma interface "apt-pkg/md5.h" | |
28 | #endif | |
29 | ||
30 | #include <string> | |
31 | #include <algorithm> | |
32 | #include <stdint.h> | |
33 | ||
34 | using std::string; | |
35 | using std::min; | |
36 | ||
37 | class MD5Summation; | |
38 | ||
39 | class MD5SumValue | |
40 | { | |
41 | friend class MD5Summation; | |
42 | unsigned char Sum[4*4]; | |
43 | ||
44 | public: | |
45 | ||
46 | // Accessors | |
47 | bool operator ==(const MD5SumValue &rhs) const; | |
48 | string Value() const; | |
49 | inline void Value(unsigned char S[16]) | |
50 | {for (int I = 0; I != sizeof(Sum); I++) S[I] = Sum[I];}; | |
51 | inline operator string() const {return Value();}; | |
52 | bool Set(string Str); | |
53 | inline void Set(unsigned char S[16]) | |
54 | {for (int I = 0; I != sizeof(Sum); I++) Sum[I] = S[I];}; | |
55 | ||
56 | MD5SumValue(string Str); | |
57 | MD5SumValue(); | |
58 | }; | |
59 | ||
60 | class MD5Summation | |
61 | { | |
62 | uint32_t Buf[4]; | |
63 | unsigned char Bytes[2*4]; | |
64 | unsigned char In[16*4]; | |
65 | bool Done; | |
66 | ||
67 | public: | |
68 | ||
69 | bool Add(const unsigned char *Data,unsigned long Size); | |
70 | inline bool Add(const char *Data) {return Add((unsigned char *)Data,strlen(Data));}; | |
71 | bool AddFD(int Fd,unsigned long Size); | |
72 | inline bool Add(const unsigned char *Beg,const unsigned char *End) | |
73 | {return Add(Beg,End-Beg);}; | |
74 | MD5SumValue Result(); | |
75 | ||
76 | MD5Summation(); | |
77 | }; | |
78 | ||
79 | #endif |