push f2fb4b43e6a5ad960b5745b9c70e0a8fefaca935
[wine/hacks.git] / dlls / ntoskrnl.exe / ntoskrnl.c
blobdcc2769b77a226fbe144f3c1f9c481c121457e88
1 /*
2 * ntoskrnl.exe implementation
4 * Copyright (C) 2007 Alexandre Julliard
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #include "config.h"
22 #include "wine/port.h"
24 #include <stdarg.h>
26 #define NONAMELESSUNION
27 #define NONAMELESSSTRUCT
29 #include "ntstatus.h"
30 #define WIN32_NO_STATUS
31 #include "windef.h"
32 #include "winternl.h"
33 #include "excpt.h"
34 #include "ddk/wdm.h"
35 #include "wine/unicode.h"
36 #include "wine/server.h"
37 #include "wine/debug.h"
39 WINE_DEFAULT_DEBUG_CHANNEL(ntoskrnl);
40 WINE_DECLARE_DEBUG_CHANNEL(relay);
43 KSYSTEM_TIME KeTickCount;
45 typedef struct _KSERVICE_TABLE_DESCRIPTOR
47 PULONG_PTR Base;
48 PULONG Count;
49 ULONG Limit;
50 PUCHAR Number;
51 } KSERVICE_TABLE_DESCRIPTOR, *PKSERVICE_TABLE_DESCRIPTOR;
53 KSERVICE_TABLE_DESCRIPTOR KeServiceDescriptorTable[4];
55 typedef void (WINAPI *PCREATE_PROCESS_NOTIFY_ROUTINE)(HANDLE,HANDLE,BOOLEAN);
57 #ifdef __i386__
58 #define DEFINE_FASTCALL1_ENTRYPOINT( name ) \
59 __ASM_GLOBAL_FUNC( name, \
60 "popl %eax\n\t" \
61 "pushl %ecx\n\t" \
62 "pushl %eax\n\t" \
63 "jmp " __ASM_NAME("__regs_") #name )
64 #define DEFINE_FASTCALL2_ENTRYPOINT( name ) \
65 __ASM_GLOBAL_FUNC( name, \
66 "popl %eax\n\t" \
67 "pushl %edx\n\t" \
68 "pushl %ecx\n\t" \
69 "pushl %eax\n\t" \
70 "jmp " __ASM_NAME("__regs_") #name )
71 #endif
73 static inline LPCSTR debugstr_us( const UNICODE_STRING *us )
75 if (!us) return "<null>";
76 return debugstr_wn( us->Buffer, us->Length / sizeof(WCHAR) );
79 static HANDLE get_device_manager(void)
81 static HANDLE device_manager;
82 HANDLE handle = 0, ret = device_manager;
84 if (!ret)
86 SERVER_START_REQ( create_device_manager )
88 req->access = SYNCHRONIZE;
89 req->attributes = 0;
90 if (!wine_server_call( req )) handle = reply->handle;
92 SERVER_END_REQ;
94 if (!handle)
96 ERR( "failed to create the device manager\n" );
97 return 0;
99 if (!(ret = InterlockedCompareExchangePointer( &device_manager, handle, 0 )))
100 ret = handle;
101 else
102 NtClose( handle ); /* somebody beat us to it */
104 return ret;
107 /* exception handler for emulation of privileged instructions */
108 static LONG CALLBACK vectored_handler( EXCEPTION_POINTERS *ptrs )
110 extern DWORD __wine_emulate_instruction( EXCEPTION_RECORD *rec, CONTEXT86 *context );
112 EXCEPTION_RECORD *record = ptrs->ExceptionRecord;
113 CONTEXT86 *context = ptrs->ContextRecord;
115 if (record->ExceptionCode == EXCEPTION_ACCESS_VIOLATION ||
116 record->ExceptionCode == EXCEPTION_PRIV_INSTRUCTION)
118 if (__wine_emulate_instruction( record, context ) == ExceptionContinueExecution)
119 return EXCEPTION_CONTINUE_EXECUTION;
121 return EXCEPTION_CONTINUE_SEARCH;
124 /* process an ioctl request for a given device */
125 static NTSTATUS process_ioctl( DEVICE_OBJECT *device, ULONG code, void *in_buff, ULONG in_size,
126 void *out_buff, ULONG *out_size )
128 IRP irp;
129 MDL mdl;
130 IO_STACK_LOCATION irpsp;
131 PDRIVER_DISPATCH dispatch = device->DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL];
132 NTSTATUS status;
133 LARGE_INTEGER count;
135 TRACE( "ioctl %x device %p in_size %u out_size %u\n", code, device, in_size, *out_size );
137 /* so we can spot things that we should initialize */
138 memset( &irp, 0x55, sizeof(irp) );
139 memset( &irpsp, 0x66, sizeof(irpsp) );
140 memset( &mdl, 0x77, sizeof(mdl) );
142 irp.RequestorMode = UserMode;
143 irp.AssociatedIrp.SystemBuffer = in_buff;
144 irp.UserBuffer = out_buff;
145 irp.MdlAddress = &mdl;
146 irp.Tail.Overlay.s.u.CurrentStackLocation = &irpsp;
148 irpsp.MajorFunction = IRP_MJ_DEVICE_CONTROL;
149 irpsp.Parameters.DeviceIoControl.OutputBufferLength = *out_size;
150 irpsp.Parameters.DeviceIoControl.InputBufferLength = in_size;
151 irpsp.Parameters.DeviceIoControl.IoControlCode = code;
152 irpsp.Parameters.DeviceIoControl.Type3InputBuffer = in_buff;
153 irpsp.DeviceObject = device;
155 mdl.Next = NULL;
156 mdl.Size = 0;
157 mdl.StartVa = out_buff;
158 mdl.ByteCount = *out_size;
159 mdl.ByteOffset = 0;
161 device->CurrentIrp = &irp;
163 KeQueryTickCount( &count ); /* update the global KeTickCount */
165 if (TRACE_ON(relay))
166 DPRINTF( "%04x:Call driver dispatch %p (device=%p,irp=%p)\n",
167 GetCurrentThreadId(), dispatch, device, &irp );
169 status = dispatch( device, &irp );
171 if (TRACE_ON(relay))
172 DPRINTF( "%04x:Ret driver dispatch %p (device=%p,irp=%p) retval=%08x\n",
173 GetCurrentThreadId(), dispatch, device, &irp, status );
175 *out_size = (irp.IoStatus.u.Status >= 0) ? irp.IoStatus.Information : 0;
176 return irp.IoStatus.u.Status;
180 /***********************************************************************
181 * wine_ntoskrnl_main_loop (Not a Windows API)
183 NTSTATUS wine_ntoskrnl_main_loop( HANDLE stop_event )
185 HANDLE manager = get_device_manager();
186 HANDLE ioctl = 0;
187 NTSTATUS status = STATUS_SUCCESS;
188 ULONG code = 0;
189 void *in_buff, *out_buff = NULL;
190 DEVICE_OBJECT *device = NULL;
191 ULONG in_size = 4096, out_size = 0;
192 HANDLE handles[2];
194 if (!(in_buff = HeapAlloc( GetProcessHeap(), 0, in_size )))
196 ERR( "failed to allocate buffer\n" );
197 return STATUS_NO_MEMORY;
200 handles[0] = stop_event;
201 handles[1] = manager;
203 for (;;)
205 SERVER_START_REQ( get_next_device_request )
207 req->manager = manager;
208 req->prev = ioctl;
209 req->status = status;
210 wine_server_add_data( req, out_buff, out_size );
211 wine_server_set_reply( req, in_buff, in_size );
212 if (!(status = wine_server_call( req )))
214 code = reply->code;
215 ioctl = reply->next;
216 device = reply->user_ptr;
217 in_size = reply->in_size;
218 out_size = reply->out_size;
220 else
222 ioctl = 0; /* no previous ioctl */
223 out_size = 0;
224 in_size = reply->in_size;
227 SERVER_END_REQ;
229 switch(status)
231 case STATUS_SUCCESS:
232 HeapFree( GetProcessHeap(), 0, out_buff );
233 if (out_size) out_buff = HeapAlloc( GetProcessHeap(), 0, out_size );
234 else out_buff = NULL;
235 status = process_ioctl( device, code, in_buff, in_size, out_buff, &out_size );
236 break;
237 case STATUS_BUFFER_OVERFLOW:
238 HeapFree( GetProcessHeap(), 0, in_buff );
239 in_buff = HeapAlloc( GetProcessHeap(), 0, in_size );
240 /* restart with larger buffer */
241 break;
242 case STATUS_PENDING:
243 if (WaitForMultipleObjects( 2, handles, FALSE, INFINITE ) == WAIT_OBJECT_0)
244 return STATUS_SUCCESS;
245 break;
250 /***********************************************************************
251 * IoAllocateMdl (NTOSKRNL.EXE.@)
253 PMDL WINAPI IoAllocateMdl( PVOID VirtualAddress, ULONG Length, BOOLEAN SecondaryBuffer, BOOLEAN ChargeQuota, PIRP Irp )
255 FIXME( "stub: %p, %u, %i, %i, %p\n", VirtualAddress, Length, SecondaryBuffer, ChargeQuota, Irp );
256 return NULL;
260 /***********************************************************************
261 * IoAllocateWorkItem (NTOSKRNL.EXE.@)
263 PIO_WORKITEM WINAPI IoAllocateWorkItem( PDEVICE_OBJECT DeviceObject )
265 FIXME( "stub: %p\n", DeviceObject );
266 return NULL;
270 /***********************************************************************
271 * IoCreateDriver (NTOSKRNL.EXE.@)
273 NTSTATUS WINAPI IoCreateDriver( UNICODE_STRING *name, PDRIVER_INITIALIZE init )
275 DRIVER_OBJECT *driver;
276 DRIVER_EXTENSION *extension;
277 NTSTATUS status;
279 if (!(driver = RtlAllocateHeap( GetProcessHeap(), HEAP_ZERO_MEMORY,
280 sizeof(*driver) + sizeof(*extension) )))
281 return STATUS_NO_MEMORY;
283 if ((status = RtlDuplicateUnicodeString( 1, name, &driver->DriverName )))
285 RtlFreeHeap( GetProcessHeap(), 0, driver );
286 return status;
289 extension = (DRIVER_EXTENSION *)(driver + 1);
290 driver->Size = sizeof(*driver);
291 driver->DriverInit = init;
292 driver->DriverExtension = extension;
293 extension->DriverObject = driver;
294 extension->ServiceKeyName = driver->DriverName;
296 status = driver->DriverInit( driver, name );
298 if (status)
300 RtlFreeUnicodeString( &driver->DriverName );
301 RtlFreeHeap( GetProcessHeap(), 0, driver );
303 return status;
307 /***********************************************************************
308 * IoDeleteDriver (NTOSKRNL.EXE.@)
310 void WINAPI IoDeleteDriver( DRIVER_OBJECT *driver )
312 RtlFreeUnicodeString( &driver->DriverName );
313 RtlFreeHeap( GetProcessHeap(), 0, driver );
317 /***********************************************************************
318 * IoCreateDevice (NTOSKRNL.EXE.@)
320 NTSTATUS WINAPI IoCreateDevice( DRIVER_OBJECT *driver, ULONG ext_size,
321 UNICODE_STRING *name, DEVICE_TYPE type,
322 ULONG characteristics, BOOLEAN exclusive,
323 DEVICE_OBJECT **ret_device )
325 NTSTATUS status;
326 DEVICE_OBJECT *device;
327 HANDLE handle = 0;
328 HANDLE manager = get_device_manager();
330 TRACE( "(%p, %u, %s, %u, %x, %u, %p)\n",
331 driver, ext_size, debugstr_us(name), type, characteristics, exclusive, ret_device );
333 if (!(device = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*device) + ext_size )))
334 return STATUS_NO_MEMORY;
336 SERVER_START_REQ( create_device )
338 req->access = 0;
339 req->attributes = 0;
340 req->rootdir = 0;
341 req->manager = manager;
342 req->user_ptr = device;
343 if (name) wine_server_add_data( req, name->Buffer, name->Length );
344 if (!(status = wine_server_call( req ))) handle = reply->handle;
346 SERVER_END_REQ;
348 if (status == STATUS_SUCCESS)
350 device->DriverObject = driver;
351 device->DeviceExtension = device + 1;
352 device->DeviceType = type;
353 device->Reserved = handle;
355 device->NextDevice = driver->DeviceObject;
356 driver->DeviceObject = device;
358 *ret_device = device;
360 else HeapFree( GetProcessHeap(), 0, device );
362 return status;
366 /***********************************************************************
367 * IoDeleteDevice (NTOSKRNL.EXE.@)
369 void WINAPI IoDeleteDevice( DEVICE_OBJECT *device )
371 NTSTATUS status;
373 TRACE( "%p\n", device );
375 SERVER_START_REQ( delete_device )
377 req->handle = device->Reserved;
378 status = wine_server_call( req );
380 SERVER_END_REQ;
382 if (status == STATUS_SUCCESS)
384 DEVICE_OBJECT **prev = &device->DriverObject->DeviceObject;
385 while (*prev && *prev != device) prev = &(*prev)->NextDevice;
386 if (*prev) *prev = (*prev)->NextDevice;
387 NtClose( device->Reserved );
388 HeapFree( GetProcessHeap(), 0, device );
393 /***********************************************************************
394 * IoCreateSymbolicLink (NTOSKRNL.EXE.@)
396 NTSTATUS WINAPI IoCreateSymbolicLink( UNICODE_STRING *name, UNICODE_STRING *target )
398 HANDLE handle;
399 OBJECT_ATTRIBUTES attr;
401 attr.Length = sizeof(attr);
402 attr.RootDirectory = 0;
403 attr.ObjectName = name;
404 attr.Attributes = OBJ_CASE_INSENSITIVE | OBJ_OPENIF;
405 attr.SecurityDescriptor = NULL;
406 attr.SecurityQualityOfService = NULL;
408 TRACE( "%s -> %s\n", debugstr_us(name), debugstr_us(target) );
409 /* FIXME: store handle somewhere */
410 return NtCreateSymbolicLinkObject( &handle, SYMBOLIC_LINK_ALL_ACCESS, &attr, target );
414 /***********************************************************************
415 * IofCompleteRequest (NTOSKRNL.EXE.@)
417 #ifdef DEFINE_FASTCALL2_ENTRYPOINT
418 DEFINE_FASTCALL2_ENTRYPOINT( IofCompleteRequest )
419 void WINAPI __regs_IofCompleteRequest( IRP *irp, UCHAR priority_boost )
420 #else
421 void WINAPI IofCompleteRequest( IRP *irp, UCHAR priority_boost )
422 #endif
424 TRACE( "%p %u\n", irp, priority_boost );
425 /* nothing to do for now */
429 /***********************************************************************
430 * ExAllocatePool (NTOSKRNL.EXE.@)
432 PVOID WINAPI ExAllocatePool( POOL_TYPE type, SIZE_T size )
434 return ExAllocatePoolWithTag( type, size, 0 );
438 /***********************************************************************
439 * ExAllocatePoolWithQuota (NTOSKRNL.EXE.@)
441 PVOID WINAPI ExAllocatePoolWithQuota( POOL_TYPE type, SIZE_T size )
443 return ExAllocatePoolWithTag( type, size, 0 );
447 /***********************************************************************
448 * ExAllocatePoolWithTag (NTOSKRNL.EXE.@)
450 PVOID WINAPI ExAllocatePoolWithTag( POOL_TYPE type, SIZE_T size, ULONG tag )
452 /* FIXME: handle page alignment constraints */
453 void *ret = HeapAlloc( GetProcessHeap(), 0, size );
454 TRACE( "%lu pool %u -> %p\n", size, type, ret );
455 return ret;
459 /***********************************************************************
460 * ExAllocatePoolWithQuotaTag (NTOSKRNL.EXE.@)
462 PVOID WINAPI ExAllocatePoolWithQuotaTag( POOL_TYPE type, SIZE_T size, ULONG tag )
464 return ExAllocatePoolWithTag( type, size, tag );
468 /***********************************************************************
469 * ExFreePool (NTOSKRNL.EXE.@)
471 void WINAPI ExFreePool( void *ptr )
473 ExFreePoolWithTag( ptr, 0 );
477 /***********************************************************************
478 * ExFreePoolWithTag (NTOSKRNL.EXE.@)
480 void WINAPI ExFreePoolWithTag( void *ptr, ULONG tag )
482 TRACE( "%p\n", ptr );
483 HeapFree( GetProcessHeap(), 0, ptr );
487 /***********************************************************************
488 * KeInitializeSpinLock (NTOSKRNL.EXE.@)
490 void WINAPI KeInitializeSpinLock( PKSPIN_LOCK SpinLock )
492 FIXME("%p\n", SpinLock);
496 /***********************************************************************
497 * KeInitializeTimerEx (NTOSKRNL.EXE.@)
499 void WINAPI KeInitializeTimerEx( PKTIMER Timer, TIMER_TYPE Type )
501 FIXME("%p %d\n", Timer, Type);
505 /***********************************************************************
506 * KeInitializeTimer (NTOSKRNL.EXE.@)
508 void WINAPI KeInitializeTimer( PKTIMER Timer )
510 KeInitializeTimerEx(Timer, NotificationTimer);
514 /***********************************************************************
515 * KeQuerySystemTime (NTOSKRNL.EXE.@)
517 void WINAPI KeQuerySystemTime( LARGE_INTEGER *time )
519 NtQuerySystemTime( time );
523 /***********************************************************************
524 * KeQueryTickCount (NTOSKRNL.EXE.@)
526 void WINAPI KeQueryTickCount( LARGE_INTEGER *count )
528 count->QuadPart = NtGetTickCount();
529 /* update the global variable too */
530 KeTickCount.LowPart = count->u.LowPart;
531 KeTickCount.High1Time = count->u.HighPart;
532 KeTickCount.High2Time = count->u.HighPart;
536 /***********************************************************************
537 * KeQueryTimeIncrement (NTOSKRNL.EXE.@)
539 ULONG WINAPI KeQueryTimeIncrement(void)
541 return 10000;
545 /***********************************************************************
546 * MmAllocateNonCachedMemory (NTOSKRNL.EXE.@)
548 PVOID WINAPI MmAllocateNonCachedMemory( SIZE_T size )
550 TRACE( "%lu\n", size );
551 return VirtualAlloc( NULL, size, MEM_RESERVE|MEM_COMMIT, PAGE_READWRITE|PAGE_NOCACHE );
555 /***********************************************************************
556 * MmFreeNonCachedMemory (NTOSKRNL.EXE.@)
558 void WINAPI MmFreeNonCachedMemory( void *addr, SIZE_T size )
560 TRACE( "%p %lu\n", addr, size );
561 VirtualFree( addr, 0, MEM_RELEASE );
564 /***********************************************************************
565 * PsCreateSystemThread (NTOSKRNL.EXE.@)
567 NTSTATUS WINAPI PsCreateSystemThread(PHANDLE ThreadHandle, ULONG DesiredAccess,
568 POBJECT_ATTRIBUTES ObjectAttributes,
569 HANDLE ProcessHandle, PCLIENT_ID ClientId,
570 PKSTART_ROUTINE StartRoutine, PVOID StartContext)
572 if (!ProcessHandle) ProcessHandle = GetCurrentProcess();
573 return RtlCreateUserThread(ProcessHandle, 0, FALSE, 0, 0,
574 0, StartRoutine, StartContext,
575 ThreadHandle, ClientId);
578 /***********************************************************************
579 * PsGetCurrentProcessId (NTOSKRNL.EXE.@)
581 HANDLE WINAPI PsGetCurrentProcessId(void)
583 return (HANDLE)GetCurrentProcessId(); /* FIXME: not quite right... */
587 /***********************************************************************
588 * PsGetCurrentThreadId (NTOSKRNL.EXE.@)
590 HANDLE WINAPI PsGetCurrentThreadId(void)
592 return (HANDLE)GetCurrentThreadId(); /* FIXME: not quite right... */
596 /***********************************************************************
597 * PsGetVersion (NTOSKRNL.EXE.@)
599 BOOLEAN WINAPI PsGetVersion(ULONG *major, ULONG *minor, ULONG *build, UNICODE_STRING *version )
601 RTL_OSVERSIONINFOEXW info;
603 RtlGetVersion( &info );
604 if (major) *major = info.dwMajorVersion;
605 if (minor) *minor = info.dwMinorVersion;
606 if (build) *build = info.dwBuildNumber;
608 if (version)
610 #if 0 /* FIXME: GameGuard passes an uninitialized pointer in version->Buffer */
611 size_t len = min( strlenW(info.szCSDVersion)*sizeof(WCHAR), version->MaximumLength );
612 memcpy( version->Buffer, info.szCSDVersion, len );
613 if (len < version->MaximumLength) version->Buffer[len / sizeof(WCHAR)] = 0;
614 version->Length = len;
615 #endif
617 return TRUE;
621 /***********************************************************************
622 * PsSetCreateProcessNotifyRoutine (NTOSKRNL.EXE.@)
624 NTSTATUS WINAPI PsSetCreateProcessNotifyRoutine( PCREATE_PROCESS_NOTIFY_ROUTINE callback, BOOLEAN remove )
626 FIXME( "stub: %p %d\n", callback, remove );
627 return STATUS_SUCCESS;
631 /*****************************************************
632 * DllMain
634 BOOL WINAPI DllMain( HINSTANCE inst, DWORD reason, LPVOID reserved )
636 LARGE_INTEGER count;
638 switch(reason)
640 case DLL_PROCESS_ATTACH:
641 DisableThreadLibraryCalls( inst );
642 RtlAddVectoredExceptionHandler( TRUE, vectored_handler );
643 KeQueryTickCount( &count ); /* initialize the global KeTickCount */
644 break;
646 return TRUE;