apt-pkg/contrib/md5.h

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