#pragma once

#include "KMemDriver.h"

#include <string>
#include <vector>


typedef bool(*map_file_cb)(IN MODULE_DATA&, OUT PVOID * const,
	OUT SIZE_T * const, IN PVOID const);
typedef bool(*map_file_cleanup_cb)(IN MODULE_DATA&,
	IN PVOID, IN PVOID const);

struct map_file_data {
	map_file_cb map_file;
	map_file_cleanup_cb map_file_cleanup;
	bool in_memory_module;
};

struct loadlib_user_data {
	std::vector<std::string> additionalDllSearchDirectories;
};
bool map_file_loadlib(MODULE_DATA& module, PVOID * const buffer,
	SIZE_T * const size, PVOID const user_ptr);
bool map_file_loadlib_cleanup(MODULE_DATA& module, PVOID buffer,
	PVOID const user_ptr);
bool map_file_kmem(MODULE_DATA& module, PVOID * const buffer,
	SIZE_T * const size, PVOID const user_ptr);
bool map_file_kmem_cleanup(MODULE_DATA& module, PVOID buffer,
	PVOID const user_ptr);

extern const struct map_file_data loadlib_data;
extern const struct map_file_data kmem_data;

class PatternScanner
{
public:
	explicit PatternScanner(struct map_file_data const * const mfd = &loadlib_data, PVOID map_file_user_data = NULL);
	~PatternScanner();
	void SetScanLowAddress(UINT64 startAddress) {
		m_LowAddress = startAddress;
	}
	void SetScanAddress(UINT64 startAddress) {
		m_LowAddress = startAddress;
	}
	bool Scan(MODULE_DATA& module, const char * const pattern);
private:
	bool checkPattern(MODULE_DATA& module, const char * const pattern, std::string& result);
	bool doScan(UINT8 *buf, SIZE_T size, std::vector<UINT64>& foundOffsets);

	struct map_file_data const * const mfd;
	UINT64 m_LowAddress = 0x0;
	UINT64 m_HighAddress = ((UINT64)-1);
	PVOID map_file_user_data;
};