apt-pkg/contrib/fileutl.cc: Add WriteAtomic mode.

[apt.git] / apt-pkg / contrib / hashes.h

// -*- mode: cpp; mode: fold -*-
// Description								/*{{{*/
// $Id: hashes.h,v 1.2 2001/03/11 05:30:20 jgg Exp $
/* ######################################################################

   Hashes - Simple wrapper around the hash functions
   
   This is just used to make building the methods simpler, this is the
   only interface required..
   
   ##################################################################### */
									/*}}}*/
#ifndef APTPKG_HASHES_H
#define APTPKG_HASHES_H


#include <apt-pkg/md5.h>
#include <apt-pkg/sha1.h>
#include <apt-pkg/sha256.h>

#include <algorithm>
#include <vector>
#include <cstring>

using std::min;
using std::vector;

// helper class that contains hash function name
// and hash
class HashString
{
 protected:
   string Type;
   string Hash;
   static const char * _SupportedHashes[10];

 public:
   HashString(string Type, string Hash);
   HashString(string StringedHashString);  // init from str as "type:hash"
   HashString();

   // get hash type used
   string HashType() { return Type; };

   // verify the given filename against the currently loaded hash
   bool VerifyFile(string filename) const;

   // helper
   string toStr() const;                    // convert to str as "type:hash"
   bool empty() const;

   // return the list of hashes we support
   static const char** SupportedHashes();
};

class Hashes
{
   public:

   MD5Summation MD5;
   SHA1Summation SHA1;
   SHA256Summation SHA256;
   
   inline bool Add(const unsigned char *Data,unsigned long Size)
   {
      return MD5.Add(Data,Size) && SHA1.Add(Data,Size) && SHA256.Add(Data,Size);
   };
   inline bool Add(const char *Data) {return Add((unsigned char *)Data,strlen(Data));};
   bool AddFD(int Fd,unsigned long Size);
   inline bool Add(const unsigned char *Beg,const unsigned char *End) 
                  {return Add(Beg,End-Beg);};
};

#endif