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