Md5Sum.h

Go to the documentation of this file.

#ifndef _utl_Md5Sum_h_
#define _utl_Md5Sum_h_

#include <string>
#include <sstream>
#include <utility>
#include <utl/AugerException.h>


namespace utl {

  class Md5Sum {
  public:
    typedef const unsigned char* Iterator;
    typedef std::pair<Iterator, Iterator> IteratorPair;
    static const unsigned int kBufferSize = 64;
    static const unsigned int kDigestSize = 16;

    class WrongStageException : public AugerException {
    public:
      WrongStageException(const std::string& message = std::string())
        : AugerException(message) { }

      virtual std::string GetExceptionName() const
      { return "Invalid processing stage exception"; }
    };

    Md5Sum() { Clear(); }

    Md5Sum(const std::string& data) { Clear(); AddData(data); }

    void AddDataRange(const unsigned char* const begin, const unsigned char* const end);

    void AddDataRange(const char* const begin, const char* const end)
    { AddDataRange((const unsigned char*)begin, (const unsigned char*)end); }

    void
    AddData(const std::string& s)
    {
      const char* const p = s.c_str();
      AddDataRange(p, p + s.length());
    }

    void
    AddData(const char* const s)
    {
      const char* p;
      for (p = s; *p; ++p) { }
      AddDataRange(s, p);
    }

    IteratorPair GetRawDigest();

    std::string GetHexDigest();

    bool IsFinalized() const
    { return fFinalized; }

    void Clear();

  private:
    void ProcessBuffer();

    unsigned int BufferIndex() const
    { return (fCountLow >> 3) % kBufferSize; }

    bool fFinalized;

    unsigned int fStateA;
    unsigned int fStateB;
    unsigned int fStateC;
    unsigned int fStateD;
    unsigned int fCountLow;
    unsigned int fCountHigh;
    unsigned char fBuffer[kBufferSize];
    unsigned char fDigest[kDigestSize];
  };
}


#endif