Module: wine Branch: master Commit: 4cbe5b78d3f2406776e512b7688853c60ccf907c URL: http://source.winehq.org/git/wine.git/?a=commit;h=4cbe5b78d3f2406776e512b768...
Author: Alexandre Julliard julliard@winehq.org Date: Wed May 16 21:13:34 2007 +0200
ntoskrnl.exe: Export a Wine-internal function to run the device request handling loop.
---
dlls/ntoskrnl.exe/ntoskrnl.c | 124 +++++++++++++++++++++++++++++++++++ dlls/ntoskrnl.exe/ntoskrnl.exe.spec | 8 ++ 2 files changed, 132 insertions(+), 0 deletions(-)
diff --git a/dlls/ntoskrnl.exe/ntoskrnl.c b/dlls/ntoskrnl.exe/ntoskrnl.c index 46cf820..b14fca5 100644 --- a/dlls/ntoskrnl.exe/ntoskrnl.c +++ b/dlls/ntoskrnl.exe/ntoskrnl.c @@ -36,6 +36,7 @@ #include "wine/debug.h"
WINE_DEFAULT_DEBUG_CHANNEL(ntoskrnl); +WINE_DECLARE_DEBUG_CHANNEL(relay);
KSYSTEM_TIME KeTickCount; @@ -101,6 +102,129 @@ static HANDLE get_device_manager(void) return ret; }
+/* process an ioctl request for a given device */ +static NTSTATUS process_ioctl( DEVICE_OBJECT *device, ULONG code, void *in_buff, ULONG in_size, + void *out_buff, ULONG *out_size ) +{ + IRP irp; + MDL mdl; + IO_STACK_LOCATION irpsp; + PDRIVER_DISPATCH dispatch = device->DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL]; + NTSTATUS status; + + TRACE( "ioctl %x device %p in_size %u out_size %u\n", code, device, in_size, *out_size ); + + /* so we can spot things that we should initialize */ + memset( &irp, 0x55, sizeof(irp) ); + memset( &irpsp, 0x66, sizeof(irpsp) ); + memset( &mdl, 0x77, sizeof(mdl) ); + + irp.RequestorMode = UserMode; + irp.AssociatedIrp.SystemBuffer = in_buff; + irp.UserBuffer = out_buff; + irp.MdlAddress = &mdl; + irp.Tail.Overlay.s.u.CurrentStackLocation = &irpsp; + + irpsp.MajorFunction = IRP_MJ_DEVICE_CONTROL; + irpsp.Parameters.DeviceIoControl.OutputBufferLength = *out_size; + irpsp.Parameters.DeviceIoControl.InputBufferLength = in_size; + irpsp.Parameters.DeviceIoControl.IoControlCode = code; + irpsp.Parameters.DeviceIoControl.Type3InputBuffer = in_buff; + irpsp.DeviceObject = device; + + mdl.Next = NULL; + mdl.Size = 0; + mdl.StartVa = out_buff; + mdl.ByteCount = *out_size; + mdl.ByteOffset = 0; + + device->CurrentIrp = &irp; + + if (TRACE_ON(relay)) + DPRINTF( "%04x:Call driver dispatch %p (device=%p,irp=%p)\n", + GetCurrentThreadId(), dispatch, device, &irp ); + + status = dispatch( device, &irp ); + + if (TRACE_ON(relay)) + DPRINTF( "%04x:Ret driver dispatch %p (device=%p,irp=%p) retval=%08x\n", + GetCurrentThreadId(), dispatch, device, &irp, status ); + + *out_size = irp.IoStatus.u.Status ? 0 : irp.IoStatus.Information; + return irp.IoStatus.u.Status; +} + + +/*********************************************************************** + * wine_ntoskrnl_main_loop (Not a Windows API) + */ +NTSTATUS wine_ntoskrnl_main_loop( HANDLE stop_event ) +{ + HANDLE manager = get_device_manager(); + HANDLE ioctl = 0; + NTSTATUS status = STATUS_SUCCESS; + ULONG code = 0; + void *in_buff, *out_buff = NULL; + DEVICE_OBJECT *device = NULL; + ULONG in_size = 4096, out_size = 0; + HANDLE handles[2]; + + if (!(in_buff = HeapAlloc( GetProcessHeap(), 0, in_size ))) + { + ERR( "failed to allocate buffer\n" ); + return STATUS_NO_MEMORY; + } + + handles[0] = stop_event; + handles[1] = manager; + + for (;;) + { + SERVER_START_REQ( get_next_device_request ) + { + req->manager = manager; + req->prev = ioctl; + req->status = status; + wine_server_add_data( req, out_buff, out_size ); + wine_server_set_reply( req, in_buff, in_size ); + if (!(status = wine_server_call( req ))) + { + code = reply->code; + ioctl = reply->next; + device = reply->user_ptr; + in_size = reply->in_size; + out_size = reply->out_size; + } + else + { + ioctl = 0; /* no previous ioctl */ + out_size = 0; + in_size = reply->in_size; + } + } + SERVER_END_REQ; + + switch(status) + { + case STATUS_SUCCESS: + HeapFree( GetProcessHeap(), 0, out_buff ); + if (out_size) out_buff = HeapAlloc( GetProcessHeap(), 0, out_size ); + else out_buff = NULL; + status = process_ioctl( device, code, in_buff, in_size, out_buff, &out_size ); + break; + case STATUS_BUFFER_OVERFLOW: + HeapFree( GetProcessHeap(), 0, in_buff ); + in_buff = HeapAlloc( GetProcessHeap(), 0, in_size ); + /* restart with larger buffer */ + break; + case STATUS_PENDING: + if (WaitForMultipleObjects( 2, handles, FALSE, INFINITE ) == WAIT_OBJECT_0) + return STATUS_SUCCESS; + break; + } + } +} +
/*********************************************************************** * IoCreateDevice (NTOSKRNL.EXE.@) diff --git a/dlls/ntoskrnl.exe/ntoskrnl.exe.spec b/dlls/ntoskrnl.exe/ntoskrnl.exe.spec index 963a2db..9aaa01e 100644 --- a/dlls/ntoskrnl.exe/ntoskrnl.exe.spec +++ b/dlls/ntoskrnl.exe/ntoskrnl.exe.spec @@ -1481,3 +1481,11 @@ @ cdecl -private wcsstr(wstr wstr) msvcrt.wcsstr @ cdecl -private wcstombs(ptr ptr long) msvcrt.wcstombs @ cdecl -private wctomb(ptr long) msvcrt.wctomb + +################################################################ +# Wine internal extensions +# +# All functions must be prefixed with '__wine_' (for internal functions) +# or 'wine_' (for user-visible functions) to avoid namespace conflicts. + +@ cdecl wine_ntoskrnl_main_loop(long)