apt-pkg/contrib/md5.h

   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 <algorithm>
  29 #include <stdint.h>
  30
  31 using std::string;
  32 using std::min;
  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    uint32_t Buf[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