src/scrypt.cpp

   1 #include <stdlib.h>
   2
   3 #include "scrypt.h"
   4 #include "pbkdf2.h"
   5
   6 #include "util.h"
   7 #include "net.h"
   8
   9 #define SCRYPT_BUFFER_SIZE (131072 + 63)
  10
  11 extern "C" void scrypt_core(unsigned int *X, unsigned int *V);
  12
  13 /* cpu and memory intensive function to transform a 80 byte buffer into a 32 byte output
  14    scratchpad size needs to be at least 63 + (128 * r * p) + (256 * r + 64) + (128 * r * N) bytes
  15    r = 1, p = 1, N = 1024
  16  */
  17
  18 uint256 scrypt_nosalt(const void* input, size_t inputlen, void *scratchpad)
  19 {
  20     unsigned int *V;
  21     unsigned int X[32];
  22     uint256 result = 0;
  23     V = (unsigned int *)(((uintptr_t)(scratchpad) + 63) & ~ (uintptr_t)(63));
  24
  25     PBKDF2_SHA256((const uint8_t*)input, inputlen, (const uint8_t*)input, inputlen, 1, (uint8_t *)X, 128);
  26     scrypt_core(X, V);
  27     PBKDF2_SHA256((const uint8_t*)input, inputlen, (uint8_t *)X, 128, 1, (uint8_t*)&result, 32);
  28
  29     return result;
  30 }
  31
  32 uint256 scrypt(const void* data, size_t datalen, const void* salt, size_t saltlen, void *scratchpad)
  33 {
  34     unsigned int *V;
  35     unsigned int X[32];
  36     uint256 result = 0;
  37     V = (unsigned int *)(((uintptr_t)(scratchpad) + 63) & ~ (uintptr_t)(63));
  38
  39     PBKDF2_SHA256((const uint8_t*)data, datalen, (const uint8_t*)salt, saltlen, 1, (uint8_t *)X, 128);
  40     scrypt_core(X, V);
  41     PBKDF2_SHA256((const uint8_t*)data, datalen, (uint8_t *)X, 128, 1, (uint8_t*)&result, 32);
  42
  43     return result;
  44 }
  45
  46 uint256 scrypt_hash(const void* input, size_t inputlen)
  47 {
  48     unsigned char scratchpad[SCRYPT_BUFFER_SIZE];
  49     return scrypt_nosalt(input, inputlen, scratchpad);
  50 }
  51
  52 uint256 scrypt_salted_hash(const void* input, size_t inputlen, const void* salt, size_t saltlen)
  53 {
  54     unsigned char scratchpad[SCRYPT_BUFFER_SIZE];
  55     return scrypt(input, inputlen, salt, saltlen, scratchpad);
  56 }
  57
  58 uint256 scrypt_salted_multiround_hash(const void* input, size_t inputlen, const void* salt, size_t saltlen, const unsigned int nRounds)
  59 {
  60     uint256 resultHash = scrypt_salted_hash(input, inputlen, salt, saltlen);
  61     uint256 transitionalHash = resultHash;
  62
  63     for(unsigned int i = 1; i < nRounds; i++)
  64     {
  65         resultHash = scrypt_salted_hash(input, inputlen, (const void*)&transitionalHash, 32);
  66         transitionalHash = resultHash;
  67     }
  68
  69     return resultHash;
  70 }
  71
  72 uint256 scrypt_blockhash(const void* input)
  73 {
  74     unsigned char scratchpad[SCRYPT_BUFFER_SIZE];
  75     return scrypt_nosalt(input, 80, scratchpad);
  76 }
  77