2018-12-11 20:00:30 +00:00
|
|
|
/*
|
2019-05-18 12:21:47 +00:00
|
|
|
Copyright (c) 2018-2019, tevador <tevador@gmail.com>
|
|
|
|
|
|
|
|
All rights reserved.
|
|
|
|
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
|
|
modification, are permitted provided that the following conditions are met:
|
|
|
|
* Redistributions of source code must retain the above copyright
|
|
|
|
notice, this list of conditions and the following disclaimer.
|
|
|
|
* Redistributions in binary form must reproduce the above copyright
|
|
|
|
notice, this list of conditions and the following disclaimer in the
|
|
|
|
documentation and/or other materials provided with the distribution.
|
|
|
|
* Neither the name of the copyright holder nor the
|
|
|
|
names of its contributors may be used to endorse or promote products
|
|
|
|
derived from this software without specific prior written permission.
|
|
|
|
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
|
|
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
|
|
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
|
|
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
|
|
|
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
|
|
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
|
|
|
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
|
|
|
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
|
|
|
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
2018-12-11 20:00:30 +00:00
|
|
|
*/
|
|
|
|
|
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include <cstdint>
|
2019-04-20 09:08:01 +00:00
|
|
|
#include <vector>
|
2019-04-28 10:44:28 +00:00
|
|
|
#include <type_traits>
|
2018-12-11 20:00:30 +00:00
|
|
|
#include "common.hpp"
|
2019-04-20 09:08:01 +00:00
|
|
|
#include "superscalar_program.hpp"
|
|
|
|
#include "allocator.hpp"
|
2019-10-05 22:43:01 +00:00
|
|
|
#include "argon2.h"
|
2018-12-11 20:00:30 +00:00
|
|
|
|
2019-04-20 14:53:06 +00:00
|
|
|
/* Global scope for C binding */
|
2019-04-20 09:08:01 +00:00
|
|
|
struct randomx_dataset {
|
|
|
|
uint8_t* memory = nullptr;
|
2019-04-28 16:17:29 +00:00
|
|
|
randomx::DatasetDeallocFunc* dealloc;
|
2019-04-20 09:08:01 +00:00
|
|
|
};
|
2018-12-11 20:00:30 +00:00
|
|
|
|
2019-04-20 14:53:06 +00:00
|
|
|
/* Global scope for C binding */
|
2019-04-28 10:44:28 +00:00
|
|
|
struct randomx_cache {
|
|
|
|
uint8_t* memory = nullptr;
|
2019-04-28 16:17:29 +00:00
|
|
|
randomx::CacheDeallocFunc* dealloc;
|
2019-05-12 20:31:48 +00:00
|
|
|
randomx::JitCompiler* jit;
|
2019-04-28 16:17:29 +00:00
|
|
|
randomx::CacheInitializeFunc* initialize;
|
|
|
|
randomx::DatasetInitFunc* datasetInit;
|
2019-04-20 09:08:01 +00:00
|
|
|
randomx::SuperscalarProgram programs[RANDOMX_CACHE_ACCESSES];
|
|
|
|
std::vector<uint64_t> reciprocalCache;
|
2019-10-04 06:31:32 +00:00
|
|
|
std::string cacheKey;
|
2019-10-05 22:43:01 +00:00
|
|
|
randomx_argon2_impl* argonImpl;
|
2019-06-29 16:53:49 +00:00
|
|
|
|
|
|
|
bool isInitialized() {
|
|
|
|
return programs[0].getSize() != 0;
|
|
|
|
}
|
2019-04-20 09:08:01 +00:00
|
|
|
};
|
2018-12-11 20:00:30 +00:00
|
|
|
|
2019-04-28 10:44:28 +00:00
|
|
|
//A pointer to a standard-layout struct object points to its initial member
|
|
|
|
static_assert(std::is_standard_layout<randomx_dataset>(), "randomx_dataset must be a standard-layout struct");
|
2019-11-22 17:24:16 +00:00
|
|
|
|
2019-09-22 20:18:28 +00:00
|
|
|
//the following assert fails when compiling Debug in Visual Studio (JIT mode will crash in Debug)
|
2019-11-22 17:24:16 +00:00
|
|
|
#if defined(_MSC_VER) && !defined(__INTEL_COMPILER) && defined(_DEBUG)
|
|
|
|
#define TO_STR(x) #x
|
|
|
|
#define STR(x) TO_STR(x)
|
|
|
|
#pragma message ( __FILE__ "(" STR(__LINE__) ") warning: check std::is_standard_layout<randomx_cache>() is disabled for Debug configuration. JIT mode will crash." )
|
|
|
|
#undef STR
|
|
|
|
#undef TO_STR
|
|
|
|
#else
|
2019-04-28 10:44:28 +00:00
|
|
|
static_assert(std::is_standard_layout<randomx_cache>(), "randomx_cache must be a standard-layout struct");
|
2019-11-22 17:24:16 +00:00
|
|
|
#endif
|
2018-12-11 20:00:30 +00:00
|
|
|
|
2019-04-28 10:44:28 +00:00
|
|
|
namespace randomx {
|
2018-12-11 20:00:30 +00:00
|
|
|
|
2019-04-28 10:44:28 +00:00
|
|
|
using DefaultAllocator = AlignedAllocator<CacheLineSize>;
|
2019-01-14 23:01:11 +00:00
|
|
|
|
2019-04-20 09:08:01 +00:00
|
|
|
template<class Allocator>
|
2019-04-28 10:44:28 +00:00
|
|
|
void deallocDataset(randomx_dataset* dataset) {
|
|
|
|
if (dataset->memory != nullptr)
|
|
|
|
Allocator::freeMemory(dataset->memory, DatasetSize);
|
|
|
|
}
|
2019-04-20 09:08:01 +00:00
|
|
|
|
|
|
|
template<class Allocator>
|
2019-05-17 22:14:43 +00:00
|
|
|
void deallocCache(randomx_cache* cache);
|
2019-04-20 09:08:01 +00:00
|
|
|
|
2019-04-28 10:44:28 +00:00
|
|
|
void initCache(randomx_cache*, const void*, size_t);
|
|
|
|
void initCacheCompile(randomx_cache*, const void*, size_t);
|
2019-04-22 13:13:41 +00:00
|
|
|
void initDatasetItem(randomx_cache* cache, uint8_t* out, uint64_t blockNumber);
|
2019-04-20 09:08:01 +00:00
|
|
|
void initDataset(randomx_cache* cache, uint8_t* dataset, uint32_t startBlock, uint32_t endBlock);
|
2019-10-05 22:43:01 +00:00
|
|
|
|
|
|
|
inline randomx_argon2_impl* selectArgonImpl(randomx_flags flags) {
|
2019-10-09 19:37:14 +00:00
|
|
|
if (flags & RANDOMX_FLAG_ARGON2_AVX2) {
|
|
|
|
return randomx_argon2_impl_avx2();
|
2019-10-05 22:43:01 +00:00
|
|
|
}
|
2019-10-09 19:37:14 +00:00
|
|
|
if (flags & RANDOMX_FLAG_ARGON2_SSSE3) {
|
|
|
|
return randomx_argon2_impl_ssse3();
|
2019-10-05 22:43:01 +00:00
|
|
|
}
|
2019-10-09 19:37:14 +00:00
|
|
|
return &randomx_argon2_fill_segment_ref;
|
2019-10-05 22:43:01 +00:00
|
|
|
}
|
2019-04-20 09:08:01 +00:00
|
|
|
}
|