You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
397 lines
11 KiB
397 lines
11 KiB
#include <cuda.h> |
|
#include "cuda_runtime.h" |
|
#include "device_launch_parameters.h" |
|
|
|
#include <stdio.h> |
|
#include <memory.h> |
|
|
|
// Folgende Definitionen später durch header ersetzen |
|
typedef unsigned int uint32_t; |
|
typedef unsigned char uint8_t; |
|
typedef unsigned short uint16_t; |
|
|
|
// globaler Speicher für alle HeftyHashes aller Threads |
|
uint32_t *d_heftyHashes[8]; |
|
|
|
/* Hash-Tabellen */ |
|
__constant__ uint32_t hefty_gpu_constantTable[64]; |
|
|
|
// muss expandiert werden |
|
__constant__ uint32_t hefty_gpu_blockHeader[16]; // 2x512 Bit Message |
|
__constant__ uint32_t hefty_gpu_register[8]; |
|
__constant__ uint32_t hefty_gpu_sponge[4]; |
|
|
|
uint32_t hefty_cpu_hashTable[] = { 0x6a09e667UL, |
|
0xbb67ae85UL, |
|
0x3c6ef372UL, |
|
0xa54ff53aUL, |
|
0x510e527fUL, |
|
0x9b05688cUL, |
|
0x1f83d9abUL, |
|
0x5be0cd19UL }; |
|
uint32_t hefty_cpu_constantTable[] = { |
|
0x428a2f98UL, 0x71374491UL, 0xb5c0fbcfUL, 0xe9b5dba5UL, |
|
0x3956c25bUL, 0x59f111f1UL, 0x923f82a4UL, 0xab1c5ed5UL, |
|
0xd807aa98UL, 0x12835b01UL, 0x243185beUL, 0x550c7dc3UL, |
|
0x72be5d74UL, 0x80deb1feUL, 0x9bdc06a7UL, 0xc19bf174UL, |
|
0xe49b69c1UL, 0xefbe4786UL, 0x0fc19dc6UL, 0x240ca1ccUL, |
|
0x2de92c6fUL, 0x4a7484aaUL, 0x5cb0a9dcUL, 0x76f988daUL, |
|
0x983e5152UL, 0xa831c66dUL, 0xb00327c8UL, 0xbf597fc7UL, |
|
0xc6e00bf3UL, 0xd5a79147UL, 0x06ca6351UL, 0x14292967UL, |
|
0x27b70a85UL, 0x2e1b2138UL, 0x4d2c6dfcUL, 0x53380d13UL, |
|
0x650a7354UL, 0x766a0abbUL, 0x81c2c92eUL, 0x92722c85UL, |
|
0xa2bfe8a1UL, 0xa81a664bUL, 0xc24b8b70UL, 0xc76c51a3UL, |
|
0xd192e819UL, 0xd6990624UL, 0xf40e3585UL, 0x106aa070UL, |
|
0x19a4c116UL, 0x1e376c08UL, 0x2748774cUL, 0x34b0bcb5UL, |
|
0x391c0cb3UL, 0x4ed8aa4aUL, 0x5b9cca4fUL, 0x682e6ff3UL, |
|
0x748f82eeUL, 0x78a5636fUL, 0x84c87814UL, 0x8cc70208UL, |
|
0x90befffaUL, 0xa4506cebUL, 0xbef9a3f7UL, 0xc67178f2UL |
|
}; |
|
|
|
#define S(x, n) (((x) >> (n)) | ((x) << (32 - (n)))) |
|
#define R(x, n) ((x) >> (n)) |
|
#define Ch(x, y, z) ((x & (y ^ z)) ^ z) |
|
#define Maj(x, y, z) ((x & (y | z)) | (y & z)) |
|
#define S0(x) (S(x, 2) ^ S(x, 13) ^ S(x, 22)) |
|
#define S1(x) (S(x, 6) ^ S(x, 11) ^ S(x, 25)) |
|
#define s0(x) (S(x, 7) ^ S(x, 18) ^ R(x, 3)) |
|
#define s1(x) (S(x, 17) ^ S(x, 19) ^ R(x, 10)) |
|
|
|
#define SWAB32(x) ( ((x & 0x000000FF) << 24) | ((x & 0x0000FF00) << 8) | ((x & 0x00FF0000) >> 8) | ((x & 0xFF000000) >> 24) ) |
|
|
|
// uint8_t |
|
#define smoosh4(x) ( ((x)>>4) ^ ((x) & 0x0F) ) |
|
__host__ __forceinline__ __device__ uint8_t smoosh2(uint32_t x) |
|
{ |
|
uint16_t w = (x >> 16) ^ (x & 0xffff); |
|
uint8_t n = smoosh4( (uint8_t)( (w >> 8) ^ (w & 0xFF) ) ); |
|
return (n >> 2) ^ (n & 0x03); |
|
} |
|
// 4 auf einmal |
|
#define smoosh4Quad(x) ( (((x)>>4) ^ (x)) & 0x0F0F0F0F ) |
|
#define getByte(x,y) ( ((x) >> (y)) & 0xFF ) |
|
|
|
__host__ __device__ void Mangle(uint32_t *inp) |
|
{ |
|
uint32_t r = smoosh4Quad(inp[0]); |
|
//uint8_t r0 = smoosh4( (uint8_t)(inp[0] >> 24) ); |
|
//uint8_t r1 = smoosh4( (uint8_t)(inp[0] >> 16) ); |
|
//uint8_t r2 = smoosh4( (uint8_t)(inp[0] >> 8) ); |
|
//uint8_t r3 = smoosh4( (uint8_t)(inp[0] & 0xFF) ); |
|
|
|
inp[1] = inp[1] ^ S(inp[0], getByte(r, 24)); |
|
|
|
switch (smoosh2(inp[1])) { |
|
case 0: inp[2] ^= S(inp[0], 1 + getByte(r,24)); break; |
|
case 1: inp[2] += S(~inp[0], 1 + getByte(r,16)); break; |
|
case 2: inp[2] &= S(~inp[0], 1 + getByte(r,8)); break; |
|
case 3: inp[2] ^= S(inp[0], 1 + getByte(r,0)); break; |
|
} |
|
|
|
uint32_t tmp = smoosh2(inp[1] ^ inp[2]); |
|
switch (tmp) { |
|
case 0: inp[3] ^= S(inp[0], 2 + getByte(r,24)); break; |
|
case 1: inp[3] += S(~inp[0], 2 + getByte(r,16)); break; |
|
case 2: inp[3] &= S(~inp[0], 2 + getByte(r,8)); break; |
|
case 3: inp[3] ^= S(inp[0], 2 + getByte(r,0)); break; |
|
} |
|
|
|
inp[0] ^= (inp[1] ^ inp[2]) + inp[3]; |
|
} |
|
|
|
__host__ __forceinline__ __device__ void Absorb(uint32_t *inp, uint32_t x) |
|
{ |
|
inp[0] ^= x; |
|
Mangle(inp); |
|
} |
|
|
|
__host__ __forceinline__ __device__ uint32_t Squeeze(uint32_t *inp) |
|
{ |
|
uint32_t y = inp[0]; |
|
Mangle(inp); |
|
return y; |
|
} |
|
|
|
__host__ __forceinline__ __device__ uint32_t Br(uint32_t *sponge, uint32_t x) |
|
{ |
|
uint32_t r = Squeeze(sponge); |
|
|
|
//uint8_t r0 = r >> 8; |
|
uint8_t r1 = r & 0xFF; |
|
uint32_t y = 1 << ((r >> 8) & 0x1F); |
|
|
|
//uint32_t retVal; |
|
//retVal = x; |
|
|
|
uint32_t resArr[4]; |
|
resArr[0] = x; |
|
resArr[1] = x & ~y; |
|
resArr[2] = x | y; |
|
resArr[3] = x ^ y; |
|
return resArr[r1 & 0x03]; |
|
|
|
/* |
|
switch(r1 & 0x03) |
|
{ |
|
case 0: |
|
break; |
|
case 1: |
|
retVal = x & ~y; |
|
break; |
|
case 2: |
|
retVal = x | y; |
|
break; |
|
case 3: |
|
retVal = x ^ y; |
|
break; |
|
} |
|
return retVal; |
|
*/ |
|
} |
|
|
|
__forceinline__ __device__ void hefty_gpu_round(uint32_t *regs, uint32_t W, uint32_t K, uint32_t *sponge) |
|
{ |
|
uint32_t tmpBr; |
|
|
|
uint32_t brG = Br(sponge, regs[6]); |
|
uint32_t brF = Br(sponge, regs[5]); |
|
uint32_t tmp1 = Ch(regs[4], brF, brG) + regs[7] + W + K; |
|
uint32_t brE = Br(sponge, regs[4]); |
|
uint32_t tmp2 = tmp1 + S1(brE); |
|
uint32_t brC = Br(sponge, regs[2]); |
|
uint32_t brB = Br(sponge, regs[1]); |
|
uint32_t brA = Br(sponge, regs[0]); |
|
uint32_t tmp3 = Maj(brA, brB, brC); |
|
tmpBr = Br(sponge, regs[0]); |
|
uint32_t tmp4 = tmp3 + S0(tmpBr); |
|
tmpBr = Br(sponge, tmp2); |
|
|
|
#pragma unroll 7 |
|
for (int k=6; k >= 0; k--) regs[k+1] = regs[k]; |
|
regs[0] = tmp2 + tmp4; |
|
regs[4] += tmpBr; |
|
} |
|
|
|
__host__ void hefty_cpu_round(uint32_t *regs, uint32_t W, uint32_t K, uint32_t *sponge) |
|
{ |
|
uint32_t tmpBr; |
|
|
|
uint32_t brG = Br(sponge, regs[6]); |
|
uint32_t brF = Br(sponge, regs[5]); |
|
uint32_t tmp1 = Ch(regs[4], brF, brG) + regs[7] + W + K; |
|
uint32_t brE = Br(sponge, regs[4]); |
|
uint32_t tmp2 = tmp1 + S1(brE); |
|
uint32_t brC = Br(sponge, regs[2]); |
|
uint32_t brB = Br(sponge, regs[1]); |
|
uint32_t brA = Br(sponge, regs[0]); |
|
uint32_t tmp3 = Maj(brA, brB, brC); |
|
tmpBr = Br(sponge, regs[0]); |
|
uint32_t tmp4 = tmp3 + S0(tmpBr); |
|
tmpBr = Br(sponge, tmp2); |
|
|
|
for (int k=6; k >= 0; k--) regs[k+1] = regs[k]; |
|
regs[0] = tmp2 + tmp4; |
|
regs[4] += tmpBr; |
|
} |
|
|
|
// Die Hash-Funktion |
|
__global__ void hefty_gpu_hash(int threads, uint32_t startNounce, void *outputHash) |
|
{ |
|
int thread = (blockDim.x * blockIdx.x + threadIdx.x); |
|
if (thread < threads) |
|
{ |
|
// bestimme den aktuellen Zähler |
|
uint32_t nounce = startNounce + thread; |
|
|
|
// jeder thread in diesem Block bekommt sein eigenes W Array im Shared memory |
|
#if USE_SHARED |
|
extern __shared__ unsigned char s[]; |
|
uint32_t *W = (uint32_t *)(&s[W_ALIGNMENT * sizeof(uint32_t) * threadIdx.x]); |
|
#else |
|
// reduktion von 256 byte auf 128 byte |
|
uint32_t W1[16]; |
|
uint32_t W2[16]; |
|
#endif |
|
|
|
// Initialisiere die register a bis h mit der Hash-Tabelle |
|
uint32_t regs[8]; |
|
uint32_t hash[8]; |
|
uint32_t sponge[4]; |
|
|
|
#pragma unroll 4 |
|
for(int k=0; k < 4; k++) |
|
sponge[k] = hefty_gpu_sponge[k]; |
|
|
|
// pre |
|
#pragma unroll 8 |
|
for (int k=0; k < 8; k++) |
|
{ |
|
regs[k] = hefty_gpu_register[k]; |
|
hash[k] = regs[k]; |
|
} |
|
|
|
//memcpy(W, &hefty_gpu_blockHeader[0], sizeof(uint32_t) * 16); // verbleibende 20 bytes aus Block 2 plus padding |
|
#pragma unroll 16 |
|
for(int k=0;k<16;k++) |
|
W1[k] = hefty_gpu_blockHeader[k]; |
|
W1[3] = SWAB32(nounce); |
|
|
|
|
|
// 2. Runde |
|
#pragma unroll 16 |
|
for(int j=0;j<16;j++) |
|
Absorb(sponge, W1[j] ^ hefty_gpu_constantTable[j]); |
|
|
|
// Progress W1 (Bytes 0...63) |
|
#pragma unroll 16 |
|
for(int j=0;j<16;j++) |
|
{ |
|
Absorb(sponge, regs[3] ^ regs[7]); |
|
hefty_gpu_round(regs, W1[j], hefty_gpu_constantTable[j], sponge); |
|
} |
|
|
|
// Progress W2 (Bytes 64...127) then W3 (Bytes 128...191) ... |
|
|
|
#pragma unroll 3 |
|
for(int k=0;k<3;k++) |
|
{ |
|
#pragma unroll 2 |
|
for(int j=0;j<2;j++) |
|
W2[j] = s1(W1[14+j]) + W1[9+j] + s0(W1[1+j]) + W1[j]; |
|
#pragma unroll 5 |
|
for(int j=2;j<7;j++) |
|
W2[j] = s1(W2[j-2]) + W1[9+j] + s0(W1[1+j]) + W1[j]; |
|
|
|
#pragma unroll 8 |
|
for(int j=7;j<15;j++) |
|
W2[j] = s1(W2[j-2]) + W2[j-7] + s0(W1[1+j]) + W1[j]; |
|
|
|
W2[15] = s1(W2[13]) + W2[8] + s0(W2[0]) + W1[15]; |
|
|
|
#pragma unroll 16 |
|
for(int j=0;j<16;j++) |
|
{ |
|
Absorb(sponge, regs[3] + regs[7]); |
|
hefty_gpu_round(regs, W2[j], hefty_gpu_constantTable[j + 16 * (k+1)], sponge); |
|
} |
|
#pragma unroll 16 |
|
for(int j=0;j<16;j++) |
|
W1[j] = W2[j]; |
|
} |
|
|
|
|
|
#pragma unroll 8 |
|
for(int k=0;k<8;k++) |
|
hash[k] += regs[k]; |
|
|
|
#pragma unroll 8 |
|
for(int k=0;k<8;k++) |
|
((uint32_t*)outputHash)[8*thread+k] = SWAB32(hash[k]); |
|
} |
|
} |
|
|
|
// Setup-Funktionen |
|
__host__ void hefty_cpu_init(int thr_id, int threads) |
|
{ |
|
cudaSetDevice(thr_id); |
|
|
|
// Kopiere die Hash-Tabellen in den GPU-Speicher |
|
cudaMemcpyToSymbol( hefty_gpu_constantTable, |
|
hefty_cpu_constantTable, |
|
sizeof(uint32_t) * 64 ); |
|
|
|
// Speicher für alle Hefty1 hashes belegen |
|
cudaMalloc(&d_heftyHashes[thr_id], 8 * sizeof(uint32_t) * threads); |
|
} |
|
|
|
__host__ void hefty_cpu_setBlock(int thr_id, int threads, void *data) |
|
// data muss 84-Byte haben! |
|
{ |
|
// Nachricht expandieren und setzen |
|
uint32_t msgBlock[32]; |
|
|
|
memset(msgBlock, 0, sizeof(uint32_t) * 32); |
|
memcpy(&msgBlock[0], data, 84); |
|
msgBlock[21] |= 0x80; |
|
msgBlock[31] = 672; // bitlen |
|
|
|
for(int i=0;i<31;i++) // Byteorder drehen |
|
msgBlock[i] = SWAB32(msgBlock[i]); |
|
|
|
// die erste Runde wird auf der CPU durchgeführt, da diese für |
|
// alle Threads gleich ist. Der Hash wird dann an die Threads |
|
// übergeben |
|
|
|
// Erstelle expandierten Block W |
|
uint32_t W[64]; |
|
memcpy(W, &msgBlock[0], sizeof(uint32_t) * 16); |
|
for(int j=16;j<64;j++) |
|
W[j] = s1(W[j-2]) + W[j-7] + s0(W[j-15]) + W[j-16]; |
|
|
|
// Initialisiere die register a bis h mit der Hash-Tabelle |
|
uint32_t regs[8]; |
|
uint32_t hash[8]; |
|
uint32_t sponge[4]; |
|
|
|
// pre |
|
memset(sponge, 0, sizeof(uint32_t) * 4); |
|
for (int k=0; k < 8; k++) |
|
{ |
|
regs[k] = hefty_cpu_hashTable[k]; |
|
hash[k] = regs[k]; |
|
} |
|
|
|
// 1. Runde |
|
for(int j=0;j<16;j++) |
|
Absorb(sponge, W[j] ^ hefty_cpu_constantTable[j]); |
|
|
|
for(int j=0;j<16;j++) |
|
{ |
|
Absorb(sponge, regs[3] ^ regs[7]); |
|
hefty_cpu_round(regs, W[j], hefty_cpu_constantTable[j], sponge); |
|
} |
|
|
|
for(int j=16;j<64;j++) |
|
{ |
|
Absorb(sponge, regs[3] + regs[7]); |
|
hefty_cpu_round(regs, W[j], hefty_cpu_constantTable[j], sponge); |
|
} |
|
|
|
for(int k=0;k<8;k++) |
|
hash[k] += regs[k]; |
|
|
|
// sponge speichern |
|
|
|
cudaMemcpyToSymbol( hefty_gpu_sponge, |
|
sponge, |
|
sizeof(uint32_t) * 4 ); |
|
// hash speichern |
|
cudaMemcpyToSymbol( hefty_gpu_register, |
|
hash, |
|
sizeof(uint32_t) * 8 ); |
|
|
|
// Blockheader setzen (korrekte Nonce fehlt da drin noch) |
|
cudaMemcpyToSymbol( hefty_gpu_blockHeader, |
|
&msgBlock[16], |
|
64); |
|
} |
|
|
|
__host__ void hefty_cpu_hash(int thr_id, int threads, int startNounce) |
|
{ |
|
const int threadsperblock = 128; |
|
|
|
// berechne wie viele Thread Blocks wir brauchen |
|
dim3 grid((threads + threadsperblock-1)/threadsperblock); |
|
dim3 block(threadsperblock); |
|
|
|
// Größe des dynamischen Shared Memory Bereichs (abhängig von der Threadanzahl) |
|
#if USE_SHARED |
|
size_t shared_size = W_ALIGNMENT*sizeof(uint32_t)*threadsperblock; // ein uint32_t eingefügt gegen Bank Konflikte |
|
#else |
|
size_t shared_size = 0; |
|
#endif |
|
|
|
// fprintf(stderr, "threads=%d, %d blocks, %d threads per block, %d bytes shared\n", threads, grid.x, block.x, shared_size); |
|
|
|
hefty_gpu_hash<<<grid, block, shared_size>>>(threads, startNounce, (void*)d_heftyHashes[thr_id]); |
|
}
|
|
|