mirror of
				https://git.wownero.com/wownero/RandomWOW.git
				synced 2024-08-15 00:23:14 +00:00 
			
		
		
		
	Merge pull request #265 from tevador/pr-randomx-v2
RandomX commitments with double-hashing
This commit is contained in:
		
						commit
						1c603a23fa
					
				
					 4 changed files with 71 additions and 7 deletions
				
			
		|  | @ -417,4 +417,15 @@ extern "C" { | |||
| 		machine->run(machine->tempHash); | ||||
| 		machine->getFinalResult(output, RANDOMX_HASH_SIZE); | ||||
| 	} | ||||
| 
 | ||||
| 	void randomx_calculate_commitment(const void* input, size_t inputSize, const void* hash_in, void* com_out) { | ||||
| 		assert(inputSize == 0 || input != nullptr); | ||||
| 		assert(hash_in != nullptr); | ||||
| 		assert(com_out != nullptr); | ||||
| 		blake2b_state state; | ||||
| 		blake2b_init(&state, RANDOMX_HASH_SIZE); | ||||
| 		blake2b_update(&state, input, inputSize); | ||||
| 		blake2b_update(&state, hash_in, RANDOMX_HASH_SIZE); | ||||
| 		blake2b_final(&state, com_out, RANDOMX_HASH_SIZE); | ||||
| 	} | ||||
| } | ||||
|  |  | |||
|  | @ -260,6 +260,17 @@ RANDOMX_EXPORT void randomx_calculate_hash_first(randomx_vm* machine, const void | |||
| RANDOMX_EXPORT void randomx_calculate_hash_next(randomx_vm* machine, const void* nextInput, size_t nextInputSize, void* output); | ||||
| RANDOMX_EXPORT void randomx_calculate_hash_last(randomx_vm* machine, void* output); | ||||
| 
 | ||||
| /**
 | ||||
|  * Calculate a RandomX commitment from a RandomX hash and its input. | ||||
|  * | ||||
|  * @param input is a pointer to memory that was hashed. Must not be NULL. | ||||
|  * @param inputSize is the number of bytes in the input. | ||||
|  * @param hash_in is the output from randomx_calculate_hash* (RANDOMX_HASH_SIZE bytes). | ||||
|  * @param com_out is a pointer to memory where the commitment will be stored. Must not | ||||
|  *        be NULL and at least RANDOMX_HASH_SIZE bytes must be available for writing. | ||||
| */ | ||||
| RANDOMX_EXPORT void randomx_calculate_commitment(const void* input, size_t inputSize, const void* hash_in, void* com_out); | ||||
| 
 | ||||
| #if defined(__cplusplus) | ||||
| } | ||||
| #endif | ||||
|  |  | |||
|  | @ -96,6 +96,7 @@ void printUsage(const char* executable) { | |||
| 	std::cout << "  --avx2        use optimized Argon2 for AVX2 CPUs" << std::endl; | ||||
| 	std::cout << "  --auto        select the best options for the current CPU" << std::endl; | ||||
| 	std::cout << "  --noBatch     calculate hashes one by one (default: batch)" << std::endl; | ||||
| 	std::cout << "  --commit      calculate commitments instead of hashes (default: hashes)" << std::endl; | ||||
| } | ||||
| 
 | ||||
| struct MemoryException : public std::exception { | ||||
|  | @ -113,7 +114,7 @@ struct DatasetAllocException : public MemoryException { | |||
| 
 | ||||
| using MineFunc = void(randomx_vm * vm, std::atomic<uint32_t> & atomicNonce, AtomicHash & result, uint32_t noncesCount, int thread, int cpuid); | ||||
| 
 | ||||
| template<bool batch> | ||||
| template<bool batch, bool commit> | ||||
| void mine(randomx_vm* vm, std::atomic<uint32_t>& atomicNonce, AtomicHash& result, uint32_t noncesCount, int thread, int cpuid = -1) { | ||||
| 	if (cpuid >= 0) { | ||||
| 		int rc = set_thread_affinity(cpuid); | ||||
|  | @ -138,6 +139,9 @@ void mine(randomx_vm* vm, std::atomic<uint32_t>& atomicNonce, AtomicHash& result | |||
| 		} | ||||
| 		store32(noncePtr, nonce); | ||||
| 		(batch ? randomx_calculate_hash_next : randomx_calculate_hash)(vm, blockTemplate, sizeof(blockTemplate), &hash); | ||||
| 		if (commit) { | ||||
| 			randomx_calculate_commitment(blockTemplate, sizeof(blockTemplate), &hash, &hash); | ||||
| 		} | ||||
| 		result.xorWith(hash); | ||||
| 		if (!batch) { | ||||
| 			nonce = atomicNonce.fetch_add(1); | ||||
|  | @ -146,7 +150,7 @@ void mine(randomx_vm* vm, std::atomic<uint32_t>& atomicNonce, AtomicHash& result | |||
| } | ||||
| 
 | ||||
| int main(int argc, char** argv) { | ||||
| 	bool softAes, miningMode, verificationMode, help, largePages, jit, secure; | ||||
| 	bool softAes, miningMode, verificationMode, help, largePages, jit, secure, commit; | ||||
| 	bool ssse3, avx2, autoFlags, noBatch; | ||||
| 	int noncesCount, threadCount, initThreadCount; | ||||
| 	uint64_t threadAffinity; | ||||
|  | @ -172,10 +176,11 @@ int main(int argc, char** argv) { | |||
| 	readOption("--avx2", argc, argv, avx2); | ||||
| 	readOption("--auto", argc, argv, autoFlags); | ||||
| 	readOption("--noBatch", argc, argv, noBatch); | ||||
| 	readOption("--commit", argc, argv, commit); | ||||
| 
 | ||||
| 	store32(&seed, seedValue); | ||||
| 
 | ||||
| 	std::cout << "RandomX benchmark v1.1.11" << std::endl; | ||||
| 	std::cout << "RandomX benchmark v1.1.12" << std::endl; | ||||
| 
 | ||||
| 	if (help) { | ||||
| 		printUsage(argv[0]); | ||||
|  | @ -280,11 +285,24 @@ int main(int argc, char** argv) { | |||
| 	MineFunc* func; | ||||
| 
 | ||||
| 	if (noBatch) { | ||||
| 		func = &mine<false>; | ||||
| 		if (commit) { | ||||
| 			std::cout << " - hash commitments" << std::endl; | ||||
| 			func = &mine<false, true>; | ||||
| 		} | ||||
| 		else { | ||||
| 			func = &mine<false, false>; | ||||
| 		} | ||||
| 	} | ||||
| 	else { | ||||
| 		func = &mine<true>; | ||||
| 		std::cout << " - batch mode" << std::endl; | ||||
| 		if (commit) { | ||||
| 			//TODO: support batch mode with commitments
 | ||||
| 			std::cout << " - hash commitments" << std::endl; | ||||
| 			func = &mine<false, true>; | ||||
| 		} | ||||
| 		else { | ||||
| 			std::cout << " - batch mode" << std::endl; | ||||
| 			func = &mine<true, false>; | ||||
| 		} | ||||
| 	} | ||||
| 
 | ||||
| 	std::cout << "Initializing"; | ||||
|  | @ -376,7 +394,7 @@ int main(int argc, char** argv) { | |||
| 			randomx_release_cache(cache); | ||||
| 		std::cout << "Calculated result: "; | ||||
| 		result.print(std::cout); | ||||
| 		if (noncesCount == 1000 && seedValue == 0) | ||||
| 		if (noncesCount == 1000 && seedValue == 0 && !commit) | ||||
| 			std::cout << "Reference result:  10b649a3f15c7c7f88277812f2e74b337a0f20ce909af09199cccb960771cfa1" << std::endl; | ||||
| 		if (!miningMode) { | ||||
| 			std::cout << "Performance: " << 1000 * elapsed / noncesCount << " ms per hash" << std::endl; | ||||
|  |  | |||
|  | @ -34,6 +34,14 @@ void calcStringHash(const char(&key)[K], const char(&input)[H], void* output) { | |||
| 	randomx_calculate_hash(vm, input, H - 1, output); | ||||
| } | ||||
| 
 | ||||
| template<size_t K, size_t H> | ||||
| void calcStringCommitment(const char(&key)[K], const char(&input)[H], void* output) { | ||||
| 	initCache(key); | ||||
| 	assert(vm != nullptr); | ||||
| 	randomx_calculate_hash(vm, input, H - 1, output); | ||||
| 	randomx_calculate_commitment(input, H - 1, output, output); | ||||
| } | ||||
| 
 | ||||
| template<size_t K, size_t H> | ||||
| void calcHexHash(const char(&key)[K], const char(&hex)[H], void* output) { | ||||
| 	initCache(key); | ||||
|  | @ -1082,6 +1090,22 @@ int main() { | |||
| 		assert(rx_get_rounding_mode() == RoundToNearest); | ||||
| 	}); | ||||
| 
 | ||||
| 	if (RANDOMX_HAVE_COMPILER) { | ||||
| 		randomx_destroy_vm(vm); | ||||
| 		vm = nullptr; | ||||
| #ifdef RANDOMX_FORCE_SECURE | ||||
| 		vm = randomx_create_vm(RANDOMX_FLAG_DEFAULT | RANDOMX_FLAG_SECURE, cache, nullptr); | ||||
| #else | ||||
| 		vm = randomx_create_vm(RANDOMX_FLAG_DEFAULT, cache, nullptr); | ||||
| #endif | ||||
| 	} | ||||
| 
 | ||||
| 	runTest("Commitment test", stringsEqual(RANDOMX_ARGON_SALT, "RandomX\x03"), []() { | ||||
| 		char hash[RANDOMX_HASH_SIZE]; | ||||
| 		calcStringCommitment("test key 000", "This is a test", &hash); | ||||
| 		assert(equalsHex(hash, "d53ccf348b75291b7be76f0a7ac8208bbced734b912f6fca60539ab6f86be919")); | ||||
| 	}); | ||||
| 
 | ||||
| 	randomx_destroy_vm(vm); | ||||
| 	vm = nullptr; | ||||
| 
 | ||||
|  |  | |||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue