| 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 |