scryptjane/scrypt-jane-hash_sha256.h

   1 #define SCRYPT_HASH "SHA-2-256"
   2 #define SCRYPT_HASH_BLOCK_SIZE 64
   3 #define SCRYPT_HASH_DIGEST_SIZE 32
   4
   5 typedef uint8_t scrypt_hash_digest[SCRYPT_HASH_DIGEST_SIZE];
   6
   7 typedef struct scrypt_hash_state_t {
   8         uint32_t H[8];
   9         uint64_t T;
  10         uint32_t leftover;
  11         uint8_t buffer[SCRYPT_HASH_BLOCK_SIZE];
  12 } scrypt_hash_state;
  13
  14 static const uint32_t sha256_constants[64] = {
  15         0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5, 0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5,
  16         0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3, 0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174,
  17         0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc, 0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da,
  18         0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7, 0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967,
  19         0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13, 0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85,
  20         0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3, 0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070,
  21         0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5, 0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3,
  22         0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208, 0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2
  23 };
  24
  25 #define Ch(x,y,z)  (z ^ (x & (y ^ z)))
  26 #define Maj(x,y,z) (((x | y) & z) | (x & y))
  27 #define S0(x)      (ROTR32(x,  2) ^ ROTR32(x, 13) ^ ROTR32(x, 22))
  28 #define S1(x)      (ROTR32(x,  6) ^ ROTR32(x, 11) ^ ROTR32(x, 25))
  29 #define G0(x)      (ROTR32(x,  7) ^ ROTR32(x, 18) ^ (x >>  3))
  30 #define G1(x)      (ROTR32(x, 17) ^ ROTR32(x, 19) ^ (x >> 10))
  31 #define W0(in,i)   (U8TO32_BE(&in[i * 4]))
  32 #define W1(i)      (G1(w[i - 2]) + w[i - 7] + G0(w[i - 15]) + w[i - 16])
  33 #define STEP(i) \
  34         t1 = S0(r[0]) + Maj(r[0], r[1], r[2]); \
  35         t0 = r[7] + S1(r[4]) + Ch(r[4], r[5], r[6]) + sha256_constants[i] + w[i]; \
  36         r[7] = r[6]; \
  37         r[6] = r[5]; \
  38         r[5] = r[4]; \
  39         r[4] = r[3] + t0; \
  40         r[3] = r[2]; \
  41         r[2] = r[1]; \
  42         r[1] = r[0]; \
  43         r[0] = t0 + t1;
  44
  45 static void
  46 sha256_blocks(scrypt_hash_state *S, const uint8_t *in, size_t blocks) {
  47         uint32_t r[8], w[64], t0, t1;
  48         size_t i;
  49
  50         for (i = 0; i < 8; i++) r[i] = S->H[i];
  51
  52         while (blocks--) {
  53                 for (i =  0; i < 16; i++) { w[i] = W0(in, i); }
  54                 for (i = 16; i < 64; i++) { w[i] = W1(i); }
  55                 for (i =  0; i < 64; i++) { STEP(i); }
  56                 for (i =  0; i <  8; i++) { r[i] += S->H[i]; S->H[i] = r[i]; }
  57                 S->T += SCRYPT_HASH_BLOCK_SIZE * 8;
  58                 in += SCRYPT_HASH_BLOCK_SIZE;
  59         }
  60 }
  61
  62 static void
  63 scrypt_hash_init(scrypt_hash_state *S) {
  64         S->H[0] = 0x6a09e667;
  65         S->H[1] = 0xbb67ae85;
  66         S->H[2] = 0x3c6ef372;
  67         S->H[3] = 0xa54ff53a;
  68         S->H[4] = 0x510e527f;
  69         S->H[5] = 0x9b05688c;
  70         S->H[6] = 0x1f83d9ab;
  71         S->H[7] = 0x5be0cd19;
  72         S->T = 0;
  73         S->leftover = 0;
  74 }
  75
  76 static void
  77 scrypt_hash_update(scrypt_hash_state *S, const uint8_t *in, size_t inlen) {
  78         size_t blocks, want;
  79
  80         /* handle the previous data */
  81         if (S->leftover) {
  82                 want = (SCRYPT_HASH_BLOCK_SIZE - S->leftover);
  83                 want = (want < inlen) ? want : inlen;
  84                 memcpy(S->buffer + S->leftover, in, want);
  85                 S->leftover += (uint32_t)want;
  86                 if (S->leftover < SCRYPT_HASH_BLOCK_SIZE)
  87                         return;
  88                 in += want;
  89                 inlen -= want;
  90                 sha256_blocks(S, S->buffer, 1);
  91         }
  92
  93         /* handle the current data */
  94         blocks = (inlen & ~(SCRYPT_HASH_BLOCK_SIZE - 1));
  95         S->leftover = (uint32_t)(inlen - blocks);
  96         if (blocks) {
  97                 sha256_blocks(S, in, blocks / SCRYPT_HASH_BLOCK_SIZE);
  98                 in += blocks;
  99         }
 100
 101         /* handle leftover data */
 102         if (S->leftover)
 103                 memcpy(S->buffer, in, S->leftover);
 104 }
 105
 106 static void
 107 scrypt_hash_finish(scrypt_hash_state *S, uint8_t *hash) {
 108         uint64_t t = S->T + (S->leftover * 8);
 109
 110         S->buffer[S->leftover] = 0x80;
 111         if (S->leftover <= 55) {
 112                 memset(S->buffer + S->leftover + 1, 0, 55 - S->leftover);
 113         } else {
 114                 memset(S->buffer + S->leftover + 1, 0, 63 - S->leftover);
 115                 sha256_blocks(S, S->buffer, 1);
 116                 memset(S->buffer, 0, 56);
 117         }
 118
 119         U64TO8_BE(S->buffer + 56, t);
 120         sha256_blocks(S, S->buffer, 1);
 121
 122         U32TO8_BE(&hash[ 0], S->H[0]);
 123         U32TO8_BE(&hash[ 4], S->H[1]);
 124         U32TO8_BE(&hash[ 8], S->H[2]);
 125         U32TO8_BE(&hash[12], S->H[3]);
 126         U32TO8_BE(&hash[16], S->H[4]);
 127         U32TO8_BE(&hash[20], S->H[5]);
 128         U32TO8_BE(&hash[24], S->H[6]);
 129         U32TO8_BE(&hash[28], S->H[7]);
 130 }
 131
 132 static const uint8_t scrypt_test_hash_expected[SCRYPT_HASH_DIGEST_SIZE] = {
 133         0xee,0x36,0xae,0xa6,0x65,0xf0,0x28,0x7d,0xc9,0xde,0xd8,0xad,0x48,0x33,0x7d,0xbf,
 134         0xcb,0xc0,0x48,0xfa,0x5f,0x92,0xfd,0x0a,0x95,0x6f,0x34,0x8e,0x8c,0x1e,0x73,0xad,
 135 };