segatools2/amex/sram.c

138 lines
2.8 KiB
C
Raw Normal View History

2018-11-08 01:03:25 +00:00
#include <windows.h>
2018-11-23 23:24:41 +00:00
#ifdef __GNUC__
#include <ntdef.h>
#else
#include <winnt.h>
#endif
#include <devioctl.h>
#include <ntdddisk.h>
2018-11-23 19:37:42 +00:00
#include <assert.h>
#include "amex/config.h"
#include "amex/sram.h"
#include "amex/nvram.h"
2018-11-08 01:03:25 +00:00
#include "hook/iohook.h"
2018-11-08 01:03:25 +00:00
#include "hooklib/setupapi.h"
2018-11-08 01:03:25 +00:00
#include "util/dprintf.h"
2019-05-13 20:49:18 +00:00
#include "util/str.h"
2018-11-08 01:03:25 +00:00
static HRESULT sram_handle_irp(struct irp *irp);
static HRESULT sram_handle_open(struct irp *irp);
static HRESULT sram_handle_close(struct irp *irp);
static HRESULT sram_handle_ioctl(struct irp *irp);
static HRESULT sram_ioctl_get_geometry(struct irp *irp);
static HANDLE sram_file;
2019-05-04 17:12:20 +00:00
HRESULT sram_hook_init(const struct sram_config *cfg)
2018-11-08 01:03:25 +00:00
{
HRESULT hr;
2019-05-04 17:12:20 +00:00
assert(cfg != NULL);
if (!cfg->enable) {
return S_FALSE;
}
hr = nvram_open_file(&sram_file, cfg->path, 0x80000);
2018-11-08 01:03:25 +00:00
if (FAILED(hr)) {
return hr;
}
hr = iohook_push_handler(sram_handle_irp);
if (FAILED(hr)) {
return hr;
}
hr = setupapi_add_phantom_dev(&sram_guid, L"$sram");
if (FAILED(hr)) {
return hr;
}
return S_OK;
}
static HRESULT sram_handle_irp(struct irp *irp)
{
assert(irp != NULL);
if (irp->op != IRP_OP_OPEN && irp->fd != sram_file) {
return iohook_invoke_next(irp);
}
switch (irp->op) {
case IRP_OP_OPEN: return sram_handle_open(irp);
case IRP_OP_CLOSE: return sram_handle_close(irp);
case IRP_OP_IOCTL: return sram_handle_ioctl(irp);
default: return iohook_invoke_next(irp);
}
}
static HRESULT sram_handle_open(struct irp *irp)
{
2019-05-13 20:49:18 +00:00
if (!wstr_eq(irp->open_filename, L"$sram")) {
2018-11-08 01:03:25 +00:00
return iohook_invoke_next(irp);
}
dprintf("SRAM: Open device\n");
irp->fd = sram_file;
return S_OK;
}
static HRESULT sram_handle_close(struct irp *irp)
{
dprintf("SRAM: Close device\n");
return S_OK;
}
static HRESULT sram_handle_ioctl(struct irp *irp)
{
switch (irp->ioctl) {
case IOCTL_DISK_GET_DRIVE_GEOMETRY:
return sram_ioctl_get_geometry(irp);
default:
dprintf("SRAM: Unknown ioctl %x, write %i read %i\n",
irp->ioctl,
2018-12-17 22:34:15 +00:00
(int) irp->write.nbytes,
(int) irp->read.nbytes);
2018-11-08 01:03:25 +00:00
return HRESULT_FROM_WIN32(ERROR_INVALID_FUNCTION);
}
}
static HRESULT sram_ioctl_get_geometry(struct irp *irp)
{
DISK_GEOMETRY *out;
if (irp->read.nbytes < sizeof(*out)) {
dprintf("SRAM: Invalid ioctl response buffer size\n");
return HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER);
}
dprintf("SRAM: Get geometry\n");
out = (DISK_GEOMETRY *) irp->read.bytes;
out->Cylinders.QuadPart = 0x20000;
out->MediaType = 0;
out->TracksPerCylinder = 1;
out->SectorsPerTrack = 1;
out->BytesPerSector = 4;
irp->read.pos = sizeof(*out);
return S_OK;
}