2021-04-02 17:53:41 +02:00
|
|
|
/*
|
|
|
|
* ntoskrnl.exe testing framework
|
|
|
|
*
|
|
|
|
* Copyright 2020 Zebediah Figura
|
|
|
|
*
|
|
|
|
* This library is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
|
|
* License as published by the Free Software Foundation; either
|
|
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This library is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
|
* Lesser General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
|
|
* License along with this library; if not, write to the Free Software
|
|
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <stdarg.h>
|
2021-04-08 06:56:42 +02:00
|
|
|
#include <stdio.h>
|
2021-04-02 17:53:41 +02:00
|
|
|
|
|
|
|
#include "ntstatus.h"
|
|
|
|
#define WIN32_NO_STATUS
|
|
|
|
#include "windef.h"
|
|
|
|
#include "winbase.h"
|
|
|
|
#include "winternl.h"
|
|
|
|
#include "winioctl.h"
|
|
|
|
#include "ddk/wdm.h"
|
|
|
|
|
|
|
|
#include "driver.h"
|
2021-04-08 06:56:42 +02:00
|
|
|
#include "utils.h"
|
2021-04-02 17:53:41 +02:00
|
|
|
|
|
|
|
static const GUID control_class = {0xdeadbeef, 0x29ef, 0x4538, {0xa5, 0xfd, 0xb6, 0x95, 0x73, 0xa3, 0x62, 0xc0}};
|
|
|
|
static UNICODE_STRING control_symlink;
|
|
|
|
|
|
|
|
static DEVICE_OBJECT *bus_fdo, *bus_pdo;
|
|
|
|
|
|
|
|
static NTSTATUS fdo_pnp(IRP *irp)
|
|
|
|
{
|
|
|
|
IO_STACK_LOCATION *stack = IoGetCurrentIrpStackLocation(irp);
|
|
|
|
NTSTATUS ret;
|
|
|
|
|
|
|
|
switch (stack->MinorFunction)
|
|
|
|
{
|
|
|
|
case IRP_MN_START_DEVICE:
|
|
|
|
irp->IoStatus.Status = IoSetDeviceInterfaceState(&control_symlink, TRUE);
|
|
|
|
break;
|
|
|
|
|
|
|
|
case IRP_MN_SURPRISE_REMOVAL:
|
|
|
|
irp->IoStatus.Status = STATUS_SUCCESS;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case IRP_MN_REMOVE_DEVICE:
|
|
|
|
IoSetDeviceInterfaceState(&control_symlink, FALSE);
|
|
|
|
irp->IoStatus.Status = STATUS_SUCCESS;
|
|
|
|
IoSkipCurrentIrpStackLocation(irp);
|
|
|
|
ret = IoCallDriver(bus_pdo, irp);
|
|
|
|
IoDetachDevice(bus_pdo);
|
|
|
|
IoDeleteDevice(bus_fdo);
|
|
|
|
RtlFreeUnicodeString(&control_symlink);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
IoSkipCurrentIrpStackLocation(irp);
|
|
|
|
return IoCallDriver(bus_pdo, irp);
|
|
|
|
}
|
|
|
|
|
|
|
|
static NTSTATUS pdo_pnp(DEVICE_OBJECT *device_obj, IRP *irp)
|
|
|
|
{
|
|
|
|
IO_STACK_LOCATION *stack = IoGetCurrentIrpStackLocation(irp);
|
|
|
|
NTSTATUS ret = irp->IoStatus.Status;
|
|
|
|
|
|
|
|
switch (stack->MinorFunction)
|
|
|
|
{
|
|
|
|
case IRP_MN_START_DEVICE:
|
|
|
|
ret = STATUS_SUCCESS;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case IRP_MN_QUERY_CAPABILITIES:
|
|
|
|
case IRP_MN_SURPRISE_REMOVAL:
|
|
|
|
ret = STATUS_SUCCESS;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
irp->IoStatus.Status = ret;
|
|
|
|
IoCompleteRequest(irp, IO_NO_INCREMENT);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
static NTSTATUS WINAPI driver_pnp(DEVICE_OBJECT *device, IRP *irp)
|
|
|
|
{
|
|
|
|
if (device == bus_fdo)
|
|
|
|
return fdo_pnp(irp);
|
|
|
|
return pdo_pnp(device, irp);
|
|
|
|
}
|
|
|
|
|
2021-04-08 06:56:42 +02:00
|
|
|
static void test_bus_query_id(DEVICE_OBJECT *top_device)
|
|
|
|
{
|
|
|
|
IO_STACK_LOCATION *stack;
|
|
|
|
IO_STATUS_BLOCK io;
|
|
|
|
NTSTATUS ret;
|
|
|
|
KEVENT event;
|
|
|
|
IRP *irp;
|
|
|
|
|
|
|
|
KeInitializeEvent(&event, NotificationEvent, FALSE);
|
|
|
|
|
|
|
|
irp = IoBuildSynchronousFsdRequest(IRP_MJ_PNP, top_device, NULL, 0, NULL, &event, &io);
|
|
|
|
irp->IoStatus.Status = STATUS_NOT_SUPPORTED;
|
|
|
|
stack = IoGetNextIrpStackLocation(irp);
|
|
|
|
stack->MinorFunction = IRP_MN_QUERY_ID;
|
|
|
|
stack->Parameters.QueryId.IdType = BusQueryDeviceID;
|
|
|
|
ret = IoCallDriver(top_device, irp);
|
|
|
|
ok(ret == STATUS_SUCCESS, "got %#x\n", ret);
|
|
|
|
ok(io.Status == STATUS_SUCCESS, "got %#x\n", ret);
|
|
|
|
ok(!wcscmp((WCHAR *)io.Information, L"ROOT\\WINETEST"), "got id '%ls'\n", (WCHAR *)io.Information);
|
|
|
|
ExFreePool((WCHAR *)io.Information);
|
|
|
|
|
|
|
|
irp = IoBuildSynchronousFsdRequest(IRP_MJ_PNP, top_device, NULL, 0, NULL, &event, &io);
|
|
|
|
irp->IoStatus.Status = STATUS_NOT_SUPPORTED;
|
|
|
|
stack = IoGetNextIrpStackLocation(irp);
|
|
|
|
stack->MinorFunction = IRP_MN_QUERY_ID;
|
|
|
|
stack->Parameters.QueryId.IdType = BusQueryInstanceID;
|
|
|
|
ret = IoCallDriver(top_device, irp);
|
|
|
|
ok(ret == STATUS_SUCCESS, "got %#x\n", ret);
|
|
|
|
ok(io.Status == STATUS_SUCCESS, "got %#x\n", ret);
|
|
|
|
ok(!wcscmp((WCHAR *)io.Information, L"0"), "got id '%ls'\n", (WCHAR *)io.Information);
|
|
|
|
ExFreePool((WCHAR *)io.Information);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void test_bus_query(void)
|
|
|
|
{
|
|
|
|
DEVICE_OBJECT *top_device;
|
|
|
|
|
|
|
|
top_device = IoGetAttachedDeviceReference(bus_pdo);
|
|
|
|
ok(top_device == bus_fdo, "wrong top device\n");
|
|
|
|
|
|
|
|
test_bus_query_id(top_device);
|
|
|
|
|
|
|
|
ObDereferenceObject(top_device);
|
|
|
|
}
|
|
|
|
|
|
|
|
static NTSTATUS fdo_ioctl(IRP *irp, IO_STACK_LOCATION *stack, ULONG code)
|
|
|
|
{
|
|
|
|
switch (code)
|
|
|
|
{
|
|
|
|
case IOCTL_WINETEST_BUS_MAIN:
|
|
|
|
test_bus_query();
|
|
|
|
return STATUS_SUCCESS;
|
|
|
|
|
|
|
|
default:
|
|
|
|
ok(0, "Unexpected ioctl %#x.\n", code);
|
|
|
|
return STATUS_NOT_IMPLEMENTED;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static NTSTATUS WINAPI driver_ioctl(DEVICE_OBJECT *device, IRP *irp)
|
|
|
|
{
|
|
|
|
IO_STACK_LOCATION *stack = IoGetCurrentIrpStackLocation(irp);
|
|
|
|
ULONG code = stack->Parameters.DeviceIoControl.IoControlCode;
|
|
|
|
NTSTATUS status = STATUS_NOT_IMPLEMENTED;
|
|
|
|
|
|
|
|
if (device == bus_fdo)
|
|
|
|
status = fdo_ioctl(irp, stack, code);
|
|
|
|
|
|
|
|
irp->IoStatus.Status = status;
|
|
|
|
IoCompleteRequest(irp, IO_NO_INCREMENT);
|
|
|
|
return status;
|
|
|
|
}
|
|
|
|
|
2021-04-02 17:53:41 +02:00
|
|
|
static NTSTATUS WINAPI driver_add_device(DRIVER_OBJECT *driver, DEVICE_OBJECT *pdo)
|
|
|
|
{
|
|
|
|
NTSTATUS ret;
|
|
|
|
|
|
|
|
if ((ret = IoCreateDevice(driver, 0, NULL, FILE_DEVICE_BUS_EXTENDER, 0, FALSE, &bus_fdo)))
|
|
|
|
return ret;
|
|
|
|
|
|
|
|
if ((ret = IoRegisterDeviceInterface(pdo, &control_class, NULL, &control_symlink)))
|
|
|
|
{
|
|
|
|
IoDeleteDevice(bus_fdo);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
IoAttachDeviceToDeviceStack(bus_fdo, pdo);
|
|
|
|
bus_pdo = pdo;
|
|
|
|
bus_fdo->Flags &= ~DO_DEVICE_INITIALIZING;
|
|
|
|
return STATUS_SUCCESS;
|
|
|
|
}
|
|
|
|
|
|
|
|
static NTSTATUS WINAPI driver_create(DEVICE_OBJECT *device, IRP *irp)
|
|
|
|
{
|
|
|
|
irp->IoStatus.Status = STATUS_SUCCESS;
|
|
|
|
IoCompleteRequest(irp, IO_NO_INCREMENT);
|
|
|
|
return STATUS_SUCCESS;
|
|
|
|
}
|
|
|
|
|
|
|
|
static NTSTATUS WINAPI driver_close(DEVICE_OBJECT *device, IRP *irp)
|
|
|
|
{
|
|
|
|
irp->IoStatus.Status = STATUS_SUCCESS;
|
|
|
|
IoCompleteRequest(irp, IO_NO_INCREMENT);
|
|
|
|
return STATUS_SUCCESS;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void WINAPI driver_unload(DRIVER_OBJECT *driver)
|
|
|
|
{
|
2021-04-08 06:56:42 +02:00
|
|
|
winetest_cleanup();
|
2021-04-02 17:53:41 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
NTSTATUS WINAPI DriverEntry(DRIVER_OBJECT *driver, UNICODE_STRING *registry)
|
|
|
|
{
|
2021-04-08 06:56:42 +02:00
|
|
|
NTSTATUS ret;
|
|
|
|
|
|
|
|
if ((ret = winetest_init()))
|
|
|
|
return ret;
|
|
|
|
|
2021-04-02 17:53:41 +02:00
|
|
|
driver->DriverExtension->AddDevice = driver_add_device;
|
|
|
|
driver->DriverUnload = driver_unload;
|
|
|
|
driver->MajorFunction[IRP_MJ_PNP] = driver_pnp;
|
2021-04-08 06:56:42 +02:00
|
|
|
driver->MajorFunction[IRP_MJ_DEVICE_CONTROL] = driver_ioctl;
|
2021-04-02 17:53:41 +02:00
|
|
|
driver->MajorFunction[IRP_MJ_CREATE] = driver_create;
|
|
|
|
driver->MajorFunction[IRP_MJ_CLOSE] = driver_close;
|
|
|
|
|
|
|
|
return STATUS_SUCCESS;
|
|
|
|
}
|