rpcrt4: Implement RpcAsyncInitializeHandle.
[wine] / dlls / ntoskrnl.exe / ntoskrnl.c
1 /*
2  * ntoskrnl.exe implementation
3  *
4  * Copyright (C) 2007 Alexandre Julliard
5  *
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.
10  *
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.
15  *
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
19  */
20
21 #include "config.h"
22 #include "wine/port.h"
23
24 #include <stdarg.h>
25
26 #define NONAMELESSUNION
27 #define NONAMELESSSTRUCT
28
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"
38
39 WINE_DEFAULT_DEBUG_CHANNEL(ntoskrnl);
40 WINE_DECLARE_DEBUG_CHANNEL(relay);
41
42
43 KSYSTEM_TIME KeTickCount;
44
45 typedef struct _KSERVICE_TABLE_DESCRIPTOR
46 {
47     PULONG_PTR Base;
48     PULONG Count;
49     ULONG Limit;
50     PUCHAR Number;
51 } KSERVICE_TABLE_DESCRIPTOR, *PKSERVICE_TABLE_DESCRIPTOR;
52
53 KSERVICE_TABLE_DESCRIPTOR KeServiceDescriptorTable[4];
54
55 typedef void (WINAPI *PCREATE_PROCESS_NOTIFY_ROUTINE)(HANDLE,HANDLE,BOOLEAN);
56
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
72
73 static inline LPCSTR debugstr_us( const UNICODE_STRING *us )
74 {
75     if (!us) return "<null>";
76     return debugstr_wn( us->Buffer, us->Length / sizeof(WCHAR) );
77 }
78
79 static HANDLE get_device_manager(void)
80 {
81     static HANDLE device_manager;
82     HANDLE handle = 0, ret = device_manager;
83
84     if (!ret)
85     {
86         SERVER_START_REQ( create_device_manager )
87         {
88             req->access     = SYNCHRONIZE;
89             req->attributes = 0;
90             if (!wine_server_call( req )) handle = reply->handle;
91         }
92         SERVER_END_REQ;
93
94         if (!handle)
95         {
96             ERR( "failed to create the device manager\n" );
97             return 0;
98         }
99         if (!(ret = InterlockedCompareExchangePointer( &device_manager, handle, 0 )))
100             ret = handle;
101         else
102             NtClose( handle );  /* somebody beat us to it */
103     }
104     return ret;
105 }
106
107 /* exception handler for emulation of privileged instructions */
108 static LONG CALLBACK vectored_handler( EXCEPTION_POINTERS *ptrs )
109 {
110     extern DWORD __wine_emulate_instruction( EXCEPTION_RECORD *rec, CONTEXT86 *context );
111
112     EXCEPTION_RECORD *record = ptrs->ExceptionRecord;
113     CONTEXT86 *context = ptrs->ContextRecord;
114
115     if (record->ExceptionCode == EXCEPTION_ACCESS_VIOLATION ||
116         record->ExceptionCode == EXCEPTION_PRIV_INSTRUCTION)
117     {
118         if (__wine_emulate_instruction( record, context ) == ExceptionContinueExecution)
119             return EXCEPTION_CONTINUE_EXECUTION;
120     }
121     return EXCEPTION_CONTINUE_SEARCH;
122 }
123
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 )
127 {
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;
134
135     TRACE( "ioctl %x device %p in_size %u out_size %u\n", code, device, in_size, *out_size );
136
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) );
141
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;
147
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;
154
155     mdl.Next = NULL;
156     mdl.Size = 0;
157     mdl.StartVa = out_buff;
158     mdl.ByteCount = *out_size;
159     mdl.ByteOffset = 0;
160
161     device->CurrentIrp = &irp;
162
163     KeQueryTickCount( &count );  /* update the global KeTickCount */
164
165     if (TRACE_ON(relay))
166         DPRINTF( "%04x:Call driver dispatch %p (device=%p,irp=%p)\n",
167                  GetCurrentThreadId(), dispatch, device, &irp );
168
169     status = dispatch( device, &irp );
170
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 );
174
175     *out_size = (irp.IoStatus.u.Status >= 0) ? irp.IoStatus.Information : 0;
176     return irp.IoStatus.u.Status;
177 }
178
179
180 /***********************************************************************
181  *           wine_ntoskrnl_main_loop   (Not a Windows API)
182  */
183 NTSTATUS wine_ntoskrnl_main_loop( HANDLE stop_event )
184 {
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];
193
194     if (!(in_buff = HeapAlloc( GetProcessHeap(), 0, in_size )))
195     {
196         ERR( "failed to allocate buffer\n" );
197         return STATUS_NO_MEMORY;
198     }
199
200     handles[0] = stop_event;
201     handles[1] = manager;
202
203     for (;;)
204     {
205         SERVER_START_REQ( get_next_device_request )
206         {
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 )))
213             {
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;
219             }
220             else
221             {
222                 ioctl = 0; /* no previous ioctl */
223                 out_size = 0;
224                 in_size = reply->in_size;
225             }
226         }
227         SERVER_END_REQ;
228
229         switch(status)
230         {
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;
246         }
247     }
248 }
249
250 /***********************************************************************
251  *           IoAllocateMdl  (NTOSKRNL.EXE.@)
252  */
253 PMDL WINAPI IoAllocateMdl( PVOID VirtualAddress, ULONG Length, BOOLEAN SecondaryBuffer, BOOLEAN ChargeQuota, PIRP Irp )
254 {
255     FIXME( "stub: %p, %u, %i, %i, %p\n", VirtualAddress, Length, SecondaryBuffer, ChargeQuota, Irp );
256     return NULL;
257 }
258
259
260 /***********************************************************************
261  *           IoAllocateWorkItem  (NTOSKRNL.EXE.@)
262  */
263 PIO_WORKITEM WINAPI IoAllocateWorkItem( PDEVICE_OBJECT DeviceObject )
264 {
265     FIXME( "stub: %p\n", DeviceObject );
266     return NULL;
267 }
268
269
270 /***********************************************************************
271  *           IoCreateDriver   (NTOSKRNL.EXE.@)
272  */
273 NTSTATUS WINAPI IoCreateDriver( UNICODE_STRING *name, PDRIVER_INITIALIZE init )
274 {
275     DRIVER_OBJECT *driver;
276     DRIVER_EXTENSION *extension;
277     NTSTATUS status;
278
279     if (!(driver = RtlAllocateHeap( GetProcessHeap(), HEAP_ZERO_MEMORY,
280                                     sizeof(*driver) + sizeof(*extension) )))
281         return STATUS_NO_MEMORY;
282
283     if ((status = RtlDuplicateUnicodeString( 1, name, &driver->DriverName )))
284     {
285         RtlFreeHeap( GetProcessHeap(), 0, driver );
286         return status;
287     }
288
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;
295
296     status = driver->DriverInit( driver, name );
297
298     if (status)
299     {
300         RtlFreeUnicodeString( &driver->DriverName );
301         RtlFreeHeap( GetProcessHeap(), 0, driver );
302     }
303     return status;
304 }
305
306
307 /***********************************************************************
308  *           IoDeleteDriver   (NTOSKRNL.EXE.@)
309  */
310 void WINAPI IoDeleteDriver( DRIVER_OBJECT *driver )
311 {
312     RtlFreeUnicodeString( &driver->DriverName );
313     RtlFreeHeap( GetProcessHeap(), 0, driver );
314 }
315
316
317 /***********************************************************************
318  *           IoCreateDevice   (NTOSKRNL.EXE.@)
319  */
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 )
324 {
325     NTSTATUS status;
326     DEVICE_OBJECT *device;
327     HANDLE handle = 0;
328     HANDLE manager = get_device_manager();
329
330     TRACE( "(%p, %u, %s, %u, %x, %u, %p)\n",
331            driver, ext_size, debugstr_us(name), type, characteristics, exclusive, ret_device );
332
333     if (!(device = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*device) + ext_size )))
334         return STATUS_NO_MEMORY;
335
336     SERVER_START_REQ( create_device )
337     {
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;
345     }
346     SERVER_END_REQ;
347
348     if (status == STATUS_SUCCESS)
349     {
350         device->DriverObject    = driver;
351         device->DeviceExtension = device + 1;
352         device->DeviceType      = type;
353         device->Reserved        = handle;
354
355         device->NextDevice   = driver->DeviceObject;
356         driver->DeviceObject = device;
357
358         *ret_device = device;
359     }
360     else HeapFree( GetProcessHeap(), 0, device );
361
362     return status;
363 }
364
365
366 /***********************************************************************
367  *           IoDeleteDevice   (NTOSKRNL.EXE.@)
368  */
369 void WINAPI IoDeleteDevice( DEVICE_OBJECT *device )
370 {
371     NTSTATUS status;
372
373     TRACE( "%p\n", device );
374
375     SERVER_START_REQ( delete_device )
376     {
377         req->handle = device->Reserved;
378         status = wine_server_call( req );
379     }
380     SERVER_END_REQ;
381
382     if (status == STATUS_SUCCESS)
383     {
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 );
389     }
390 }
391
392
393 /***********************************************************************
394  *           IoCreateSymbolicLink   (NTOSKRNL.EXE.@)
395  */
396 NTSTATUS WINAPI IoCreateSymbolicLink( UNICODE_STRING *name, UNICODE_STRING *target )
397 {
398     HANDLE handle;
399     OBJECT_ATTRIBUTES attr;
400
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;
407
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 );
411 }
412
413
414 /***********************************************************************
415  *           IofCompleteRequest   (NTOSKRNL.EXE.@)
416  */
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
423 {
424     TRACE( "%p %u\n", irp, priority_boost );
425     /* nothing to do for now */
426 }
427
428
429 /***********************************************************************
430  *           ExAllocatePool   (NTOSKRNL.EXE.@)
431  */
432 PVOID WINAPI ExAllocatePool( POOL_TYPE type, SIZE_T size )
433 {
434     return ExAllocatePoolWithTag( type, size, 0 );
435 }
436
437
438 /***********************************************************************
439  *           ExAllocatePoolWithQuota   (NTOSKRNL.EXE.@)
440  */
441 PVOID WINAPI ExAllocatePoolWithQuota( POOL_TYPE type, SIZE_T size )
442 {
443     return ExAllocatePoolWithTag( type, size, 0 );
444 }
445
446
447 /***********************************************************************
448  *           ExAllocatePoolWithTag   (NTOSKRNL.EXE.@)
449  */
450 PVOID WINAPI ExAllocatePoolWithTag( POOL_TYPE type, SIZE_T size, ULONG tag )
451 {
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;
456 }
457
458
459 /***********************************************************************
460  *           ExAllocatePoolWithQuotaTag   (NTOSKRNL.EXE.@)
461  */
462 PVOID WINAPI ExAllocatePoolWithQuotaTag( POOL_TYPE type, SIZE_T size, ULONG tag )
463 {
464     return ExAllocatePoolWithTag( type, size, tag );
465 }
466
467
468 /***********************************************************************
469  *           ExFreePool   (NTOSKRNL.EXE.@)
470  */
471 void WINAPI ExFreePool( void *ptr )
472 {
473     ExFreePoolWithTag( ptr, 0 );
474 }
475
476
477 /***********************************************************************
478  *           ExFreePoolWithTag   (NTOSKRNL.EXE.@)
479  */
480 void WINAPI ExFreePoolWithTag( void *ptr, ULONG tag )
481 {
482     TRACE( "%p\n", ptr );
483     HeapFree( GetProcessHeap(), 0, ptr );
484 }
485
486
487 /***********************************************************************
488  *           KeInitializeTimer   (NTOSKRNL.EXE.@)
489  */
490 void WINAPI KeInitializeTimer( PKTIMER Timer )
491 {
492     FIXME("%p\n", Timer);
493 }
494
495
496 /***********************************************************************
497  *           KeQuerySystemTime   (NTOSKRNL.EXE.@)
498  */
499 void WINAPI KeQuerySystemTime( LARGE_INTEGER *time )
500 {
501     NtQuerySystemTime( time );
502 }
503
504
505 /***********************************************************************
506  *           KeQueryTickCount   (NTOSKRNL.EXE.@)
507  */
508 void WINAPI KeQueryTickCount( LARGE_INTEGER *count )
509 {
510     count->QuadPart = NtGetTickCount();
511     /* update the global variable too */
512     KeTickCount.LowPart   = count->u.LowPart;
513     KeTickCount.High1Time = count->u.HighPart;
514     KeTickCount.High2Time = count->u.HighPart;
515 }
516
517
518 /***********************************************************************
519  *           KeQueryTimeIncrement   (NTOSKRNL.EXE.@)
520  */
521 ULONG WINAPI KeQueryTimeIncrement(void)
522 {
523     return 10000;
524 }
525
526
527 /***********************************************************************
528  *           MmAllocateNonCachedMemory   (NTOSKRNL.EXE.@)
529  */
530 PVOID WINAPI MmAllocateNonCachedMemory( SIZE_T size )
531 {
532     TRACE( "%lu\n", size );
533     return VirtualAlloc( NULL, size, MEM_RESERVE|MEM_COMMIT, PAGE_READWRITE|PAGE_NOCACHE );
534 }
535
536
537 /***********************************************************************
538  *           MmFreeNonCachedMemory   (NTOSKRNL.EXE.@)
539  */
540 void WINAPI MmFreeNonCachedMemory( void *addr, SIZE_T size )
541 {
542     TRACE( "%p %lu\n", addr, size );
543     VirtualFree( addr, 0, MEM_RELEASE );
544 }
545
546 /***********************************************************************
547  *           PsCreateSystemThread   (NTOSKRNL.EXE.@)
548  */
549 NTSTATUS WINAPI PsCreateSystemThread(PHANDLE ThreadHandle, ULONG DesiredAccess,
550                                      POBJECT_ATTRIBUTES ObjectAttributes,
551                                      HANDLE ProcessHandle, PCLIENT_ID ClientId,
552                                      PKSTART_ROUTINE StartRoutine, PVOID StartContext)
553 {
554     if (!ProcessHandle) ProcessHandle = GetCurrentProcess();
555     return RtlCreateUserThread(ProcessHandle, 0, FALSE, 0, 0,
556                                0, StartRoutine, StartContext,
557                                ThreadHandle, ClientId);
558 }
559
560 /***********************************************************************
561  *           PsGetCurrentProcessId   (NTOSKRNL.EXE.@)
562  */
563 HANDLE WINAPI PsGetCurrentProcessId(void)
564 {
565     return (HANDLE)GetCurrentProcessId();  /* FIXME: not quite right... */
566 }
567
568
569 /***********************************************************************
570  *           PsGetCurrentThreadId   (NTOSKRNL.EXE.@)
571  */
572 HANDLE WINAPI PsGetCurrentThreadId(void)
573 {
574     return (HANDLE)GetCurrentThreadId();  /* FIXME: not quite right... */
575 }
576
577
578 /***********************************************************************
579  *           PsGetVersion   (NTOSKRNL.EXE.@)
580  */
581 BOOLEAN WINAPI PsGetVersion(ULONG *major, ULONG *minor, ULONG *build, UNICODE_STRING *version )
582 {
583     RTL_OSVERSIONINFOEXW info;
584
585     RtlGetVersion( &info );
586     if (major) *major = info.dwMajorVersion;
587     if (minor) *minor = info.dwMinorVersion;
588     if (build) *build = info.dwBuildNumber;
589
590     if (version)
591     {
592 #if 0  /* FIXME: GameGuard passes an uninitialized pointer in version->Buffer */
593         size_t len = min( strlenW(info.szCSDVersion)*sizeof(WCHAR), version->MaximumLength );
594         memcpy( version->Buffer, info.szCSDVersion, len );
595         if (len < version->MaximumLength) version->Buffer[len / sizeof(WCHAR)] = 0;
596         version->Length = len;
597 #endif
598     }
599     return TRUE;
600 }
601
602
603 /***********************************************************************
604  *           PsSetCreateProcessNotifyRoutine   (NTOSKRNL.EXE.@)
605  */
606 NTSTATUS WINAPI PsSetCreateProcessNotifyRoutine( PCREATE_PROCESS_NOTIFY_ROUTINE callback, BOOLEAN remove )
607 {
608     FIXME( "stub: %p %d\n", callback, remove );
609     return STATUS_SUCCESS;
610 }
611
612
613 /*****************************************************
614  *           DllMain
615  */
616 BOOL WINAPI DllMain( HINSTANCE inst, DWORD reason, LPVOID reserved )
617 {
618     LARGE_INTEGER count;
619
620     switch(reason)
621     {
622     case DLL_PROCESS_ATTACH:
623         DisableThreadLibraryCalls( inst );
624         RtlAddVectoredExceptionHandler( TRUE, vectored_handler );
625         KeQueryTickCount( &count );  /* initialize the global KeTickCount */
626         break;
627     }
628     return TRUE;
629 }