source/sha2.h

   1 #ifndef MSP_CRYPTO_SHA2_H_
   2 #define MSP_CRYPTO_SHA2_H_
   3
   4 #include <msp/core/inttypes.h>
   5 #include "blockhash.h"
   6
   7 namespace Msp {
   8 namespace Crypto {
   9
  10 template<typename C>
  11 class SHA2: public BlockHash<C::BLOCK_SIZE>
  12 {
  13 private:
  14         typedef C Constants;
  15         typedef typename Constants::WordType WordType;
  16
  17         typename Constants::WordType buffer[8];
  18         UInt64 processed_bytes;
  19
  20 public:
  21         SHA2();
  22         SHA2(const char *, unsigned);
  23         SHA2(const std::string &);
  24 private:
  25         void init();
  26
  27 public:
  28         virtual unsigned get_digest_size() const { return Constants::DIGEST_SIZE; }
  29
  30         virtual unsigned get_digest(char *, unsigned) const;
  31
  32 private:
  33         virtual void process_block(const char *);
  34 };
  35
  36
  37 struct SHA256Constants
  38 {
  39         typedef UInt32 WordType;
  40
  41         enum
  42         {
  43                 WORD_SIZE = sizeof(WordType),
  44                 BLOCK_SIZE = 64,   // 512 bits
  45                 DIGEST_SIZE = 32,  // 256 bits
  46                 MIN_PADDING = 8,
  47                 N_ROUNDS = 64
  48         };
  49
  50         static const WordType initial[8];
  51         static const WordType round_constants[N_ROUNDS];
  52         static const unsigned sigma_constants[12];
  53 };
  54
  55 struct SHA512Constants
  56 {
  57         typedef UInt64 WordType;
  58
  59         enum
  60         {
  61                 WORD_SIZE = sizeof(WordType),
  62                 BLOCK_SIZE = 128,  // 1024 bits
  63                 DIGEST_SIZE = 64,  // 512 bits
  64                 MIN_PADDING = 16,
  65                 N_ROUNDS = 80
  66         };
  67
  68         static const WordType initial[8];
  69         static const WordType round_constants[N_ROUNDS];
  70         static const unsigned sigma_constants[12];
  71 };
  72
  73 typedef SHA2<SHA256Constants> SHA256;
  74 typedef SHA2<SHA512Constants> SHA512;
  75
  76 } // namespace Crypto
  77 } // namespace Msp
  78
  79 #endif