mirror of
https://gitea.tendokyu.moe/Hay1tsme/segatools.git
synced 2024-12-12 16:21:04 +01:00
195 lines
4.2 KiB
C
195 lines
4.2 KiB
C
#include <windows.h>
|
|
|
|
#ifdef __GNUC__
|
|
#include <ntdef.h>
|
|
#else
|
|
#include <winnt.h>
|
|
#endif
|
|
#include <devioctl.h>
|
|
#include <ntdddisk.h>
|
|
|
|
#include <assert.h>
|
|
|
|
#include "amex/eeprom.h"
|
|
#include "amex/nvram.h"
|
|
|
|
#include "hook/iohook.h"
|
|
|
|
#include "hooklib/setupapi.h"
|
|
|
|
#include "util/dprintf.h"
|
|
#include "util/str.h"
|
|
|
|
enum {
|
|
EEPROM_IOCTL_GET_ABI_VERSION = 0x80006000,
|
|
};
|
|
|
|
static HRESULT eeprom_handle_irp(struct irp *irp);
|
|
static HRESULT eeprom_handle_open(struct irp *irp);
|
|
static HRESULT eeprom_handle_close(struct irp *irp);
|
|
static HRESULT eeprom_handle_ioctl(struct irp *irp);
|
|
static HRESULT eeprom_handle_read(struct irp *irp);
|
|
static HRESULT eeprom_handle_write(struct irp *irp);
|
|
|
|
static HRESULT eeprom_ioctl_get_geometry(struct irp *irp);
|
|
static HRESULT eeprom_ioctl_get_abi_version(struct irp *irp);
|
|
|
|
static struct eeprom_config eeprom_config;
|
|
static HANDLE eeprom_file;
|
|
|
|
HRESULT eeprom_hook_init(const struct eeprom_config *cfg)
|
|
{
|
|
HRESULT hr;
|
|
|
|
assert(cfg != NULL);
|
|
|
|
if (!cfg->enable) {
|
|
return S_FALSE;
|
|
}
|
|
|
|
memcpy(&eeprom_config, cfg, sizeof(*cfg));
|
|
|
|
hr = iohook_push_handler(eeprom_handle_irp);
|
|
|
|
if (FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
|
|
hr = setupapi_add_phantom_dev(&eeprom_guid, L"$eeprom");
|
|
|
|
if (FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT eeprom_handle_irp(struct irp *irp)
|
|
{
|
|
assert(irp != NULL);
|
|
|
|
if (irp->op != IRP_OP_OPEN && irp->fd != eeprom_file) {
|
|
return iohook_invoke_next(irp);
|
|
}
|
|
|
|
switch (irp->op) {
|
|
case IRP_OP_OPEN: return eeprom_handle_open(irp);
|
|
case IRP_OP_CLOSE: return eeprom_handle_close(irp);
|
|
case IRP_OP_IOCTL: return eeprom_handle_ioctl(irp);
|
|
case IRP_OP_READ: return eeprom_handle_read(irp);
|
|
case IRP_OP_WRITE: return eeprom_handle_write(irp);
|
|
default: return iohook_invoke_next(irp);
|
|
}
|
|
}
|
|
|
|
static HRESULT eeprom_handle_open(struct irp *irp)
|
|
{
|
|
HRESULT hr;
|
|
|
|
if (!wstr_eq(irp->open_filename, L"$eeprom") != 0) {
|
|
return iohook_invoke_next(irp);
|
|
}
|
|
|
|
if (eeprom_file != NULL) {
|
|
dprintf("EEPROM: Already open\n");
|
|
|
|
return HRESULT_FROM_WIN32(ERROR_SHARING_VIOLATION);
|
|
}
|
|
|
|
dprintf("EEPROM: Open device\n");
|
|
hr = nvram_open_file(&eeprom_file, eeprom_config.path, 0x2000);
|
|
|
|
if (FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
|
|
irp->fd = eeprom_file;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT eeprom_handle_close(struct irp *irp)
|
|
{
|
|
dprintf("EEPROM: Close device\n");
|
|
eeprom_file = NULL;
|
|
|
|
return iohook_invoke_next(irp);
|
|
}
|
|
|
|
static HRESULT eeprom_handle_ioctl(struct irp *irp)
|
|
{
|
|
switch (irp->ioctl) {
|
|
case IOCTL_DISK_GET_DRIVE_GEOMETRY:
|
|
return eeprom_ioctl_get_geometry(irp);
|
|
|
|
case EEPROM_IOCTL_GET_ABI_VERSION:
|
|
return eeprom_ioctl_get_abi_version(irp);
|
|
|
|
default:
|
|
dprintf("EEPROM: Unknown ioctl %x, write %i read %i\n",
|
|
irp->ioctl,
|
|
(int) irp->write.nbytes,
|
|
(int) irp->read.nbytes);
|
|
|
|
return HRESULT_FROM_WIN32(ERROR_INVALID_FUNCTION);
|
|
}
|
|
}
|
|
|
|
static HRESULT eeprom_ioctl_get_geometry(struct irp *irp)
|
|
{
|
|
DISK_GEOMETRY out;
|
|
HRESULT hr;
|
|
|
|
dprintf("EEPROM: Get geometry\n");
|
|
|
|
memset(&out, 0, sizeof(out));
|
|
out.Cylinders.QuadPart = 1;
|
|
out.MediaType = FixedMedia;
|
|
out.TracksPerCylinder = 224;
|
|
out.SectorsPerTrack = 32;
|
|
out.BytesPerSector = 1;
|
|
|
|
hr = iobuf_write(&irp->read, &out, sizeof(out));
|
|
|
|
if (FAILED(hr)) {
|
|
dprintf("EEPROM: Get geometry failed: %08x\n", (int) hr);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT eeprom_ioctl_get_abi_version(struct irp *irp)
|
|
{
|
|
return iobuf_write_le16(&irp->read, 256);
|
|
}
|
|
|
|
static HRESULT eeprom_handle_read(struct irp *irp)
|
|
{
|
|
if (irp->ovl == NULL) {
|
|
dprintf("EEPROM: Synchronous read..?\n");
|
|
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
dprintf("EEPROM: Read off %x len %x\n",
|
|
(int) irp->ovl->Offset,
|
|
(int) irp->read.nbytes);
|
|
|
|
return iohook_invoke_next(irp);
|
|
}
|
|
|
|
static HRESULT eeprom_handle_write(struct irp *irp)
|
|
{
|
|
if (irp->ovl == NULL) {
|
|
dprintf("EEPROM: Synchronous write..?\n");
|
|
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
dprintf("EEPROM: Write off %x len %x\n",
|
|
(int) irp->ovl->Offset,
|
|
(int) irp->write.nbytes);
|
|
|
|
return iohook_invoke_next(irp);
|
|
}
|