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