RandomWOW/src/VirtualMachine.hpp

/*
Copyright (c) 2018 tevador

This file is part of RandomX.

RandomX is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

RandomX is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with RandomX.  If not, see<http://www.gnu.org/licenses/>.
*/

#pragma once
#include <cstdint>
#include "common.hpp"

namespace RandomX {

	class VirtualMachine {
	public:
		VirtualMachine(bool softAes);
		virtual ~VirtualMachine() {}
		virtual void initializeDataset(const void* seed, bool light = false);
		void initializeScratchpad(uint32_t index);
		virtual void initializeProgram(const void* seed) = 0;
		virtual void execute() = 0;
		void getResult(void*);
		const RegisterFile& getRegisterFile() const {
			return reg;
		}
		const convertible_t* getScratchpad() const {
			return scratchpad;
		}
		const void* getCache() {
			if (lightClient) {
				return mem.lcm->cache;
			}
			return nullptr;
		}
		const __m128i* getKeys() {
			if (lightClient) {
				return mem.lcm->keys;
			}
			return nullptr;
		}
	protected:
		bool softAes, lightClient;
		RegisterFile reg;
		MemoryRegisters mem;
		DatasetReadFunc readDataset;
		alignas(16) convertible_t scratchpad[ScratchpadLength];
	};
}