76 lines
2.0 KiB
C++
76 lines
2.0 KiB
C++
|
#include "rom.h"
|
||
|
|
#include "util/logging.h"
|
||
|
|
#include "util/utils.h"
|
||
|
|
#include "avs/game.h"
|
||
|
|
#include "hooks/devicehook.h"
|
||
|
|
|
||
|
|
namespace hooks::rom {
|
||
|
|
|
||
|
|
static std::string MODEL;
|
||
|
|
|
||
|
|
class ROMFileHandle : public CustomHandle {
|
||
|
|
private:
|
||
|
|
int offset = 0;
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
bool open(LPCWSTR lpFileName) override {
|
||
|
|
if (wcsicmp(lpFileName, L"D:\\001rom.txt")
|
||
|
|
&& wcsicmp(lpFileName, L"D:\\\\001rom.txt")) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
log_info("romhook", "opened 001rom.txt");
|
||
|
|
offset = 0;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
int read(LPVOID lpBuffer, DWORD nNumberOfBytesToRead) override {
|
||
|
|
int ret = 0;
|
||
|
|
for (int i = 0; i < (signed) MIN(nNumberOfBytesToRead, MODEL.length() - offset); i++) {
|
||
|
|
*((char*) lpBuffer + i) = MODEL[i + offset];
|
||
|
|
ret++;
|
||
|
|
}
|
||
|
|
offset += ret;
|
||
|
|
if (offset == (int) MODEL.length()) {
|
||
|
|
log_info("romhook", "read complete: {}", MODEL);
|
||
|
|
}
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
int write(LPCVOID lpBuffer, DWORD nNumberOfBytesToWrite) override {
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int device_io(DWORD dwIoControlCode, LPVOID lpInBuffer, DWORD nInBufferSize,
|
||
|
|
LPVOID lpOutBuffer, DWORD nOutBufferSize) override {
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool close() override {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
void file_info(LPBY_HANDLE_FILE_INFORMATION lpFileInformation) override {
|
||
|
|
*lpFileInformation = BY_HANDLE_FILE_INFORMATION {};
|
||
|
|
lpFileInformation->nFileSizeLow = MODEL.length();
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
void init() {
|
||
|
|
log_info("romhook", "init");
|
||
|
|
|
||
|
|
// populate model
|
||
|
|
if (MODEL.empty()) {
|
||
|
|
MODEL = avs::game::MODEL;
|
||
|
|
}
|
||
|
|
|
||
|
|
// add device hook
|
||
|
|
devicehook_init();
|
||
|
|
devicehook_add(new ROMFileHandle());
|
||
|
|
}
|
||
|
|
|
||
|
|
void set_model(const std::string &model) {
|
||
|
|
MODEL = model;
|
||
|
|
}
|
||
|
|
}
|