digest.h

Go to the documentation of this file.

// Copyright (C) 1999-2002 Open Source Telecom Corporation.
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
//
// As a special exception to the GNU General Public License, permission is
// granted for additional uses of the text contained in its release
// of Common C++.
//
// The exception is that, if you link the Common C++ library with other
// files to produce an executable, this does not by itself cause the
// resulting executable to be covered by the GNU General Public License.
// Your use of that executable is in no way restricted on account of
// linking the Common C++ library code into it.
//
// This exception does not however invalidate any other reasons why
// the executable file might be covered by the GNU General Public License.
//
// This exception applies only to the code released under the
// name Common C++.  If you copy code from other releases into a copy of
// Common C++, as the General Public License permits, the exception does
// not apply to the code that you add in this way.  To avoid misleading
// anyone as to the status of such modified files, you must delete
// this exception notice from them.
//
// If you write modifications of your own for Common C++, it is your choice
// whether to permit this exception to apply to your modifications.
// If you do not wish that, delete this exception notice.

#ifndef CCXX_DIGEST_H_
#define CCXX_DIGEST_H_

#ifndef CCXX_CONFIG_H_
#include <cc++/config.h>
#endif

#ifndef CCXX_THREAD_H_
#include <cc++/thread.h>
#endif

#include <iostream>
#include <fstream>

#ifdef  CCXX_NAMESPACES
namespace ost {
#endif

#ifdef WIN32
class CCXX_CLASS_EXPORT Digest;
class CCXX_CLASS_EXPORT ChecksumDigest;
class CCXX_CLASS_EXPORT CRC16Digest;
class CCXX_CLASS_EXPORT CRC32Digest;
class CCXX_CLASS_EXPORT MD5Digest;
class CCXX_CLASS_EXPORT SHAConstant;
class CCXX_CLASS_EXPORT SHA64DigestHelper;
class CCXX_CLASS_EXPORT SHA1Digest;
class CCXX_CLASS_EXPORT SHA256Digest;
#endif

#if defined(__KCC)
#define ostream ostream_withassign
#endif
class Digest : protected std::streambuf, public std::ostream
{
protected:
        Digest();

        virtual void initDigest(void) = 0;

        virtual unsigned getSize(void) = 0;

        virtual unsigned getDigest(unsigned char *buffer) = 0;

        virtual void putDigest(const unsigned char *buffer, unsigned length) = 0;

        virtual std::ostream &strDigest(std::ostream &os) = 0;

        friend std::ostream &operator<<(std::ostream &os, Digest &ia)
                {return ia.strDigest(os);};
};

class ChecksumDigest : public Digest
{
private:
        unsigned char csum;

protected:
        int overflow(int c);
        std::ostream &strDigest(std::ostream &os);

public:
        ChecksumDigest();

        void initDigest(void)
                {csum = 0;};

        unsigned getSize(void)
                {return 1;};

        unsigned getDigest(unsigned char *buffer);

        void putDigest(const unsigned char *buffer, unsigned length);
};

class CRC16Digest : public Digest
{
private:
        unsigned short crc16;

protected:
        int overflow(int c);

        std::ostream &strDigest(std::ostream &os);

public:
        CRC16Digest();

        inline void initDigest(void)
                {crc16 = 0;};

        inline unsigned getSize(void)
                {return 2;};

        unsigned getDigest(unsigned char *buffer);

        void putDigest(const unsigned char *buffer, unsigned length);
};

class CRC32Digest : public Digest
{
private:
   unsigned long crc_table[256];
   unsigned long crc_reg;
   unsigned long crc32;
   
protected:
   unsigned char overflow(unsigned char octet);

   std::ostream &strDigest(std::ostream &os);
   
public:
   CRC32Digest();
   
   void initDigest(void);
   
   inline unsigned getSize(void) {return 4;}
   
   unsigned getDigest(unsigned char *buffer);
   
   void putDigest(const unsigned char *buffer, unsigned length);
};

class MD5Digest : public Digest
{
private:
        unsigned long state[4];
        unsigned long count[2];
        unsigned char buf[64];
        unsigned bpos;
        unsigned char md5[16];
        bool updated;

protected:
        int overflow(int c);

        void update(void);

        void commit(void);

        std::ostream &strDigest(std::ostream &os);

public:
        MD5Digest();

        void initDigest(void);

        inline unsigned getSize(void)
                {return 16;};

        unsigned getDigest(unsigned char *buffer);

        void putDigest(const unsigned char *buffer, unsigned len);
};

class DigestException : public Exception {
public:
        DigestException(const std::string & what_arg) throw();
        virtual ~DigestException() throw();
};
 
template <class int_type>
class CCXX_CLASS_EXPORT SHATumbler {
public:
        SHATumbler(int);

        SHATumbler(const SHATumbler&);
        SHATumbler& operator=(const SHATumbler&);

        int_type& operator[](int);

        ~SHATumbler();
  
        SHATumbler operator+(const SHATumbler& addend) const;
        SHATumbler& operator+=(const SHATumbler& addend);
        std::ostream & toString(std::ostream & os);

        friend std::ostream &operator<<(std::ostream &os, SHATumbler<int_type>& ia)
        {return ia.toString(os);};

        unsigned getSize();
        unsigned placeInBuffer(unsigned char *);
    
private:
        int_type * h;
        int size;

        char * tmp_buff;

        static char format_string[];
};

class SHAConstant {
protected:
        const static uint64 K[];
};

template <class uint_type, unsigned blockSizeInBytes>
class CCXX_CLASS_EXPORT SHADigest : public Digest {
private:
        uint_type totalLengthInBits;

        void copyTempBlock(const SHADigest &);

protected:
        unsigned char tempBlock[blockSizeInBytes];
        void initDigest(void);

        virtual void processBlock(const unsigned char * buffer) = 0;
        void padBuffer(unsigned char * buffer);

        bool completed;

        SHADigest();
        SHADigest(const SHADigest & other);
        SHADigest & operator=(const SHADigest & other);

public:
        unsigned getSize(void) = 0;
        void putDigest(const unsigned char * buffer, unsigned length) 
                throw(DigestException);
        std::ostream & strDigest(std::ostream &os) = 0;

};

class SHA64DigestHelper: public SHADigest<uint64, 64> {
protected:
        SHATumbler<uint32> h;
        SHATumbler<uint32> a;

        SHA64DigestHelper(unsigned);
        SHATumbler<uint32> getDigest();

public:
        unsigned getDigest(unsigned char * buffer) 
        { return getDigest().placeInBuffer(buffer); }
        std::ostream & strDigest(std::ostream & os);

        SHA64DigestHelper(const SHA64DigestHelper & other);
        SHA64DigestHelper & operator=(const SHA64DigestHelper & other);
};

class SHA1Digest : public SHA64DigestHelper {
protected:
        void processBlock(const unsigned char * buffer);

public:
        SHA1Digest();
        void initDigest();
        unsigned getSize() { return 20; }
        SHA1Digest(const SHA1Digest & other);
        SHA1Digest & operator=(const SHA1Digest & other)
        { *((SHA64DigestHelper*)this) = other; return *this; }
};

class SHA256Digest : public SHA64DigestHelper, public SHAConstant {
protected:
        void processBlock(const unsigned char * buffer);

public:
        SHA256Digest();
        SHA256Digest(const SHA256Digest &);
        void initDigest();
        unsigned getSize() { return 32; }
        SHA256Digest & operator=(const SHA256Digest & other)
        { *((SHA64DigestHelper*)this) = other; return *this; }
};

#ifdef  CCXX_NAMESPACES
};
#endif

#endif

---