diff options
Diffstat (limited to 'km/src/special/nt_path_converter/device.c')
-rw-r--r-- | km/src/special/nt_path_converter/device.c | 203 |
1 files changed, 0 insertions, 203 deletions
diff --git a/km/src/special/nt_path_converter/device.c b/km/src/special/nt_path_converter/device.c deleted file mode 100644 index a179984..0000000 --- a/km/src/special/nt_path_converter/device.c +++ /dev/null @@ -1,203 +0,0 @@ -/* - * Copyright (c) 2015 Egor Tensin <Egor.Tensin@gmail.com> - * This file is part of the "Windows 7 drivers" project. - * For details, see https://github.com/egor-tensin/windows7-drivers. - * Distributed under the MIT License. - */ - -#include "control_codes.h" -#include "device.h" -#include "nt2dos.h" - -#include <ntddk.h> - -static NTSTATUS device_open(DEVICE_OBJECT *device_object, IRP *irp) -{ - NTSTATUS status = STATUS_SUCCESS; - - UNREFERENCED_PARAMETER(device_object); - - irp->IoStatus.Status = status; - irp->IoStatus.Information = 0; - IoCompleteRequest(irp, IO_NO_INCREMENT); - return status; -} - -typedef NTSTATUS (*ioctl_handler)( - void *, unsigned long, - void *, unsigned long, - ULONG_PTR *); - -static NTSTATUS handle_convert_nt_path( - void *in_buf, - unsigned long in_buf_size, - void *out_buf, - unsigned long out_buf_size, - ULONG_PTR *nbwritten) -{ - UNICODE_STRING uUnresolved, uResolved; - NTSTATUS status = STATUS_SUCCESS; - - DbgPrint("nt_path_converter: unresolved path: %ws\n", (WCHAR *) in_buf); - DbgPrint("nt_path_converter: unresolved size: %lu\n", in_buf_size); - - RtlInitUnicodeString(&uUnresolved, (WCHAR *) in_buf); - status = nt2dos(&uResolved, &uUnresolved); - - if (!NT_SUCCESS(status)) - return status; - - *nbwritten = uResolved.Length + sizeof(WCHAR); - - DbgPrint("nt_path_converter: resolved path: %wZ\n", &uResolved); - DbgPrint("nt_path_converter: resolved size: %Iu\n", *nbwritten); - - if (out_buf_size < *nbwritten) - { - status = STATUS_BUFFER_OVERFLOW; - goto free_resolved; - } - - RtlFillMemory(out_buf, *nbwritten, L'\0'); - RtlCopyMemory(out_buf, uResolved.Buffer, uResolved.Length); - -free_resolved: - ExFreePool(uResolved.Buffer); - - return status; -} - -static NTSTATUS device_ioctl(DEVICE_OBJECT *device_object, IRP *irp) -{ - IO_STACK_LOCATION *io_stack_loc; - void* in_buf, *out_buf; - unsigned long in_buf_size, out_buf_size; - ioctl_handler handler; - NTSTATUS status = STATUS_UNSUCCESSFUL; - - UNREFERENCED_PARAMETER(device_object); - - irp->IoStatus.Status = status; - irp->IoStatus.Information = 0; - io_stack_loc = IoGetCurrentIrpStackLocation(irp); - - in_buf = out_buf = irp->AssociatedIrp.SystemBuffer; - in_buf_size = io_stack_loc->Parameters.DeviceIoControl.InputBufferLength; - out_buf_size = io_stack_loc->Parameters.DeviceIoControl.OutputBufferLength; - - switch (io_stack_loc->Parameters.DeviceIoControl.IoControlCode) - { - case CONVERT_NT_PATH: - handler = handle_convert_nt_path; - break; - - default: - status = irp->IoStatus.Status = STATUS_NOT_SUPPORTED; - goto complete_request; - } - - status = irp->IoStatus.Status = handler( - in_buf, in_buf_size, - out_buf, out_buf_size, - &irp->IoStatus.Information); - -complete_request: - IoCompleteRequest(irp, IO_NO_INCREMENT); - - return status; -} - -typedef struct -{ - const wchar_t *path; - const wchar_t *symlink; -} -DeviceInfo; - -typedef struct -{ - DEVICE_OBJECT *object; - UNICODE_STRING path; - UNICODE_STRING symlink; -} -Device; - -#define NUMOF_DEVICES 1 - -static DeviceInfo devices_info[NUMOF_DEVICES] = -{ - { - L"\\Device\\nt_path_converter", - L"\\DosDevices\\nt_path_converter", - }, -}; - -static Device devices[NUMOF_DEVICES]; - -static void destroy_device(int i) -{ - IoDeleteSymbolicLink(&devices[i].symlink); - IoDeleteDevice(devices[i].object); -} - -void destroy_devices() -{ - int i; - for (i = 0; i < NUMOF_DEVICES; ++i) - destroy_device(i); -} - -static NTSTATUS set_up_device(DRIVER_OBJECT *driver_object, int i) -{ - NTSTATUS status = STATUS_SUCCESS; - - RtlInitUnicodeString(&devices[i].path, devices_info[i].path); - RtlInitUnicodeString(&devices[i].symlink, devices_info[i].symlink); - - status = IoCreateDevice( - driver_object, - 0, - &devices[i].path, - FILE_DEVICE_UNKNOWN, - FILE_DEVICE_SECURE_OPEN, - FALSE, - &devices[i].object); - - if (!NT_SUCCESS(status)) - return status; - - devices[i].object->Flags |= DO_BUFFERED_IO; - devices[i].object->Flags &= ~DO_DEVICE_INITIALIZING; - - if (!NT_SUCCESS(status = IoCreateSymbolicLink( - &devices[i].symlink, &devices[i].path))) - goto delete_device; - - return status; - -delete_device: - IoDeleteDevice(devices[i].object); - - return status; -} - -NTSTATUS set_up_devices(DRIVER_OBJECT *driver_object) -{ - int i, j; - NTSTATUS status = STATUS_SUCCESS; - - for (i = 0; i < NUMOF_DEVICES; ++i) - if (!NT_SUCCESS(status = set_up_device(driver_object, i))) - goto destroy_devices; - - driver_object->MajorFunction[IRP_MJ_CREATE] = device_open; - driver_object->MajorFunction[IRP_MJ_DEVICE_CONTROL] = device_ioctl; - - return status; - -destroy_devices: - for (j = 0; j < i; ++j) - destroy_device(j); - - return status; -} |