SHA256.h

Go to the documentation of this file.

#ifndef SHA256_H
#define SHA256_H
#include <string>
#include <cstddef>
#include "../TypeDefs.h"

using namespace std;

#define SHA2_SHFR(x, n)    ((x) >> (n))
#define SHA2_ROTR(x, n)   (((x) >> (n)) | ((x) << ((sizeof(x) << 3) - (n))))
#define SHA2_ROTL(x, n)   (((x) << (n)) | ((x) >> ((sizeof(x) << 3) - (n))))
#define SHA2_CH(x, y, z)  (((x) & (y)) ^ (~(x) & (z)))
#define SHA2_MAJ(x, y, z) (((x) & (y)) ^ ((x) & (z)) ^ ((y) & (z)))
#define SHA256_F1(x) (SHA2_ROTR(x,  2) ^ SHA2_ROTR(x, 13) ^ SHA2_ROTR(x, 22))
#define SHA256_F2(x) (SHA2_ROTR(x,  6) ^ SHA2_ROTR(x, 11) ^ SHA2_ROTR(x, 25))
#define SHA256_F3(x) (SHA2_ROTR(x,  7) ^ SHA2_ROTR(x, 18) ^ SHA2_SHFR(x,  3))
#define SHA256_F4(x) (SHA2_ROTR(x, 17) ^ SHA2_ROTR(x, 19) ^ SHA2_SHFR(x, 10))
#define SHA2_UNPACK32(x, str)                 \
{                                             \
    *((str) + 3) = (uint8_t) ((x)      );       \
    *((str) + 2) = (uint8_t) ((x) >>  8);       \
    *((str) + 1) = (uint8_t) ((x) >> 16);       \
    *((str) + 0) = (uint8_t) ((x) >> 24);       \
}
#define SHA2_PACK32(str, x)                   \
{                                             \
    *(x) =   ((uint32_t) *((str) + 3)      )    \
           | ((uint32_t) *((str) + 2) <<  8)    \
           | ((uint32_t) *((str) + 1) << 16)    \
           | ((uint32_t) *((str) + 0) << 24);   \
}

class SHA256
{
protected:
    typedef unsigned char uint8;
    typedef unsigned int  uint32;
    typedef unsigned long long uint64;

    const static uint32_t     sha256_k[];
    static const unsigned int SHA224_256_BLOCK_SIZE = (512 / 8);
    unsigned     int          m_tot_len;
    unsigned     int          m_len;
    unsigned     char         m_block[2 * SHA224_256_BLOCK_SIZE];
    uint32_t     m_h[8];

    void transform(const unsigned char *message, unsigned int block_nb);

public:
    static const unsigned int DIGEST_SIZE = ( 256 / 8);

    void init();
    void update(const unsigned char* message, uint len);
    void final(uint8_t* digest);
};

extern uint8_t* sha256(unsigned char* input, uint inputLen, uint8_t* digest);
extern uint8_t* sha256(std::string input, uint8_t* digest);

#endif