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.
142 lines
4.5 KiB
142 lines
4.5 KiB
/*
|
|
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.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <cstdint>
|
|
#include <cstring>
|
|
#include <vector>
|
|
#include "common.hpp"
|
|
|
|
namespace randomx {
|
|
|
|
class Program;
|
|
struct ProgramConfiguration;
|
|
class SuperscalarProgram;
|
|
class JitCompilerX86;
|
|
class Instruction;
|
|
|
|
typedef void(JitCompilerX86::*InstructionGeneratorX86)(Instruction&, int);
|
|
|
|
class JitCompilerX86 {
|
|
public:
|
|
JitCompilerX86();
|
|
~JitCompilerX86();
|
|
void generateProgram(Program&, ProgramConfiguration&);
|
|
void generateProgramLight(Program&, ProgramConfiguration&, uint32_t);
|
|
template<size_t N>
|
|
void generateSuperscalarHash(SuperscalarProgram (&programs)[N], std::vector<uint64_t> &);
|
|
void generateDatasetInitCode();
|
|
ProgramFunc* getProgramFunc() {
|
|
return (ProgramFunc*)code;
|
|
}
|
|
DatasetInitFunc* getDatasetInitFunc() {
|
|
return (DatasetInitFunc*)code;
|
|
}
|
|
uint8_t* getCode() {
|
|
return code;
|
|
}
|
|
size_t getCodeSize();
|
|
void enableWriting();
|
|
void enableExecution();
|
|
void enableAll();
|
|
private:
|
|
static InstructionGeneratorX86 engine[256];
|
|
std::vector<int32_t> instructionOffsets;
|
|
int registerUsage[RegistersCount];
|
|
uint8_t* code;
|
|
int32_t codePos;
|
|
|
|
void generateProgramPrologue(Program&, ProgramConfiguration&);
|
|
void generateProgramEpilogue(Program&, ProgramConfiguration&);
|
|
void genAddressReg(Instruction&, bool);
|
|
void genAddressRegDst(Instruction&);
|
|
void genAddressImm(Instruction&);
|
|
void genSIB(int scale, int index, int base);
|
|
|
|
void generateCode(Instruction&, int);
|
|
void generateSuperscalarCode(Instruction &, std::vector<uint64_t> &);
|
|
|
|
void emitByte(uint8_t val) {
|
|
code[codePos] = val;
|
|
codePos++;
|
|
}
|
|
|
|
void emit32(uint32_t val) {
|
|
memcpy(code + codePos, &val, sizeof val);
|
|
codePos += sizeof val;
|
|
}
|
|
|
|
void emit64(uint64_t val) {
|
|
memcpy(code + codePos, &val, sizeof val);
|
|
codePos += sizeof val;
|
|
}
|
|
|
|
template<size_t N>
|
|
void emit(const uint8_t (&src)[N]) {
|
|
emit(src, N);
|
|
}
|
|
|
|
void emit(const uint8_t* src, size_t count) {
|
|
memcpy(code + codePos, src, count);
|
|
codePos += count;
|
|
}
|
|
|
|
void h_IADD_RS(Instruction&, int);
|
|
void h_IADD_M(Instruction&, int);
|
|
void h_ISUB_R(Instruction&, int);
|
|
void h_ISUB_M(Instruction&, int);
|
|
void h_IMUL_R(Instruction&, int);
|
|
void h_IMUL_M(Instruction&, int);
|
|
void h_IMULH_R(Instruction&, int);
|
|
void h_IMULH_M(Instruction&, int);
|
|
void h_ISMULH_R(Instruction&, int);
|
|
void h_ISMULH_M(Instruction&, int);
|
|
void h_IMUL_RCP(Instruction&, int);
|
|
void h_INEG_R(Instruction&, int);
|
|
void h_IXOR_R(Instruction&, int);
|
|
void h_IXOR_M(Instruction&, int);
|
|
void h_IROR_R(Instruction&, int);
|
|
void h_IROL_R(Instruction&, int);
|
|
void h_ISWAP_R(Instruction&, int);
|
|
void h_FSWAP_R(Instruction&, int);
|
|
void h_FADD_R(Instruction&, int);
|
|
void h_FADD_M(Instruction&, int);
|
|
void h_FSUB_R(Instruction&, int);
|
|
void h_FSUB_M(Instruction&, int);
|
|
void h_FSCAL_R(Instruction&, int);
|
|
void h_FMUL_R(Instruction&, int);
|
|
void h_FDIV_M(Instruction&, int);
|
|
void h_FSQRT_R(Instruction&, int);
|
|
void h_CBRANCH(Instruction&, int);
|
|
void h_CFROUND(Instruction&, int);
|
|
void h_ISTORE(Instruction&, int);
|
|
void h_NOP(Instruction&, int);
|
|
};
|
|
|
|
} |