nu/eeprom.c: Add emulated EEPROM device

This commit is contained in:
Tau 2018-11-07 16:45:48 -05:00
parent 278108a914
commit 5aaab2879d
4 changed files with 170 additions and 0 deletions

154
nu/eeprom.c Normal file
View File

@ -0,0 +1,154 @@
#include <windows.h>
#include "hook/iohook.h"
#include "nu/eeprom.h"
#include "nu/nvram.h"
#include "util/dprintf.h"
#include "util/setupapi.h"
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 HANDLE eeprom_file;
HRESULT eeprom_hook_init(void)
{
HRESULT hr;
hr = nvram_open_file(&eeprom_file, L"DEVICE\\eeprom.bin", 0x2000);
if (FAILED(hr)) {
return hr;
}
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)
{
if (wcscmp(irp->open_filename, L"$eeprom") != 0) {
return iohook_invoke_next(irp);
}
dprintf("EEPROM: Open device\n");
irp->fd = eeprom_file;
return S_OK;
}
static HRESULT eeprom_handle_close(struct irp *irp)
{
dprintf("EEPROM: Close device\n");
return S_OK;
}
static HRESULT eeprom_handle_ioctl(struct irp *irp)
{
switch (irp->ioctl) {
case IOCTL_DISK_GET_DRIVE_GEOMETRY:
return eeprom_ioctl_get_geometry(irp);
default:
dprintf("EEPROM: Unknown ioctl %x, write %i read %i\n",
irp->ioctl,
irp->write.nbytes,
irp->read.nbytes);
return HRESULT_FROM_WIN32(ERROR_INVALID_FUNCTION);
}
}
static HRESULT eeprom_ioctl_get_geometry(struct irp *irp)
{
DISK_GEOMETRY *out;
if (irp->read.nbytes < sizeof(*out)) {
dprintf("EEPROM: Invalid ioctl response buffer size\n");
return HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER);
}
dprintf("EEPROM: Get geometry\n");
/* Not the real values, just bullshitting something for now */
out = (DISK_GEOMETRY *) irp->read.bytes;
out->Cylinders.QuadPart = 0x800;
out->MediaType = 0;
out->TracksPerCylinder = 1;
out->SectorsPerTrack = 2;
out->BytesPerSector = 4;
irp->read.pos = sizeof(*out);
return S_OK;
}
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,
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,
irp->write.nbytes);
return iohook_invoke_next(irp);
}

12
nu/eeprom.h Normal file
View File

@ -0,0 +1,12 @@
#pragma once
#include <windows.h>
DEFINE_GUID(
eeprom_guid,
0xB7970F0C,
0x31C4,
0x45FF,
0x96, 0x18, 0x0A, 0x24, 0x00, 0x94, 0xB2, 0x71);
HRESULT eeprom_hook_init(void);

View File

@ -1,2 +1,4 @@
#include <windows.h> #include <windows.h>
#include <initguid.h> #include <initguid.h>
#include "nu/eeprom.h"

View File

@ -8,6 +8,8 @@ nu_lib = static_library(
capnhook.get_variable('hooklib_dep'), capnhook.get_variable('hooklib_dep'),
], ],
sources : [ sources : [
'eeprom.c',
'eeprom.h',
'guid.c', 'guid.c',
'nvram.c', 'nvram.c',
'nvram.h', 'nvram.h',