- initial import of revision 374 from cnc
[apt.git] / apt-pkg / contrib / md5.h
1 // -*- mode: cpp; mode: fold -*-
2 // Description                                                          /*{{{*/
3 // $Id: md5.h,v 1.1 2002/07/23 17:54:51 niemeyer 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
32 using std::string;
33
34 class MD5Summation;
35
36 class MD5SumValue
37 {
38    friend class MD5Summation;
39    unsigned char Sum[4*4];
40    
41    public:
42
43    // Accessors
44    bool operator ==(const MD5SumValue &rhs) const; 
45    string Value() const;
46    inline void Value(unsigned char S[16]) 
47          {for (int I = 0; I != sizeof(Sum); I++) S[I] = Sum[I];};
48    inline operator string() const {return Value();};
49    bool Set(string Str);
50    inline void Set(unsigned char S[16]) 
51          {for (int I = 0; I != sizeof(Sum); I++) Sum[I] = S[I];};
52
53    MD5SumValue(string Str);
54    MD5SumValue();
55 };
56
57 class MD5Summation
58 {
59    unsigned char Buf[4*4];
60    unsigned char Bytes[2*4];
61    unsigned char In[16*4];
62    bool Done;
63    
64    public:
65
66    bool Add(const unsigned char *Data,unsigned long Size);
67    inline bool Add(const char *Data) {return Add((unsigned char *)Data,strlen(Data));};
68    bool AddFD(int Fd,unsigned long Size);
69    inline bool Add(const unsigned char *Beg,const unsigned char *End) 
70                   {return Add(Beg,End-Beg);};
71    MD5SumValue Result();
72    
73    MD5Summation();
74 };
75
76 #endif