ntoskrnl.exe: Add stub for IoGetDeviceObjectPointer().
[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 = { 0, 0, 0 };
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] = { { 0 } };
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 /***********************************************************************
252  *           IoInitializeIrp  (NTOSKRNL.EXE.@)
253  */
254 void WINAPI IoInitializeIrp( IRP *irp, USHORT size, CCHAR stack_size )
255 {
256     FIXME( "%p, %u, %d\n", irp, size, stack_size );
257 }
258
259
260 /***********************************************************************
261  *           IoAllocateIrp  (NTOSKRNL.EXE.@)
262  */
263 PIRP WINAPI IoAllocateIrp( CCHAR stack_size, BOOLEAN charge_quota )
264 {
265     FIXME( "%d, %d\n", stack_size, charge_quota );
266     return NULL;
267 }
268
269
270 /***********************************************************************
271  *           IoFreeIrp  (NTOSKRNL.EXE.@)
272  */
273 void WINAPI IoFreeIrp( IRP *irp )
274 {
275     FIXME( "%p\n", irp );
276 }
277
278
279 /***********************************************************************
280  *           IoAllocateMdl  (NTOSKRNL.EXE.@)
281  */
282 PMDL WINAPI IoAllocateMdl( PVOID VirtualAddress, ULONG Length, BOOLEAN SecondaryBuffer, BOOLEAN ChargeQuota, PIRP Irp )
283 {
284     FIXME( "stub: %p, %u, %i, %i, %p\n", VirtualAddress, Length, SecondaryBuffer, ChargeQuota, Irp );
285     return NULL;
286 }
287
288
289 /***********************************************************************
290  *           IoAllocateWorkItem  (NTOSKRNL.EXE.@)
291  */
292 PIO_WORKITEM WINAPI IoAllocateWorkItem( PDEVICE_OBJECT DeviceObject )
293 {
294     FIXME( "stub: %p\n", DeviceObject );
295     return NULL;
296 }
297
298
299 /***********************************************************************
300  *           IoCreateDriver   (NTOSKRNL.EXE.@)
301  */
302 NTSTATUS WINAPI IoCreateDriver( UNICODE_STRING *name, PDRIVER_INITIALIZE init )
303 {
304     DRIVER_OBJECT *driver;
305     DRIVER_EXTENSION *extension;
306     NTSTATUS status;
307
308     if (!(driver = RtlAllocateHeap( GetProcessHeap(), HEAP_ZERO_MEMORY,
309                                     sizeof(*driver) + sizeof(*extension) )))
310         return STATUS_NO_MEMORY;
311
312     if ((status = RtlDuplicateUnicodeString( 1, name, &driver->DriverName )))
313     {
314         RtlFreeHeap( GetProcessHeap(), 0, driver );
315         return status;
316     }
317
318     extension = (DRIVER_EXTENSION *)(driver + 1);
319     driver->Size            = sizeof(*driver);
320     driver->DriverInit      = init;
321     driver->DriverExtension = extension;
322     extension->DriverObject   = driver;
323     extension->ServiceKeyName = driver->DriverName;
324
325     status = driver->DriverInit( driver, name );
326
327     if (status)
328     {
329         RtlFreeUnicodeString( &driver->DriverName );
330         RtlFreeHeap( GetProcessHeap(), 0, driver );
331     }
332     return status;
333 }
334
335
336 /***********************************************************************
337  *           IoDeleteDriver   (NTOSKRNL.EXE.@)
338  */
339 void WINAPI IoDeleteDriver( DRIVER_OBJECT *driver )
340 {
341     RtlFreeUnicodeString( &driver->DriverName );
342     RtlFreeHeap( GetProcessHeap(), 0, driver );
343 }
344
345
346 /***********************************************************************
347  *           IoCreateDevice   (NTOSKRNL.EXE.@)
348  */
349 NTSTATUS WINAPI IoCreateDevice( DRIVER_OBJECT *driver, ULONG ext_size,
350                                 UNICODE_STRING *name, DEVICE_TYPE type,
351                                 ULONG characteristics, BOOLEAN exclusive,
352                                 DEVICE_OBJECT **ret_device )
353 {
354     NTSTATUS status;
355     DEVICE_OBJECT *device;
356     HANDLE handle = 0;
357     HANDLE manager = get_device_manager();
358
359     TRACE( "(%p, %u, %s, %u, %x, %u, %p)\n",
360            driver, ext_size, debugstr_us(name), type, characteristics, exclusive, ret_device );
361
362     if (!(device = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*device) + ext_size )))
363         return STATUS_NO_MEMORY;
364
365     SERVER_START_REQ( create_device )
366     {
367         req->access     = 0;
368         req->attributes = 0;
369         req->rootdir    = 0;
370         req->manager    = manager;
371         req->user_ptr   = device;
372         if (name) wine_server_add_data( req, name->Buffer, name->Length );
373         if (!(status = wine_server_call( req ))) handle = reply->handle;
374     }
375     SERVER_END_REQ;
376
377     if (status == STATUS_SUCCESS)
378     {
379         device->DriverObject    = driver;
380         device->DeviceExtension = device + 1;
381         device->DeviceType      = type;
382         device->Reserved        = handle;
383
384         device->NextDevice   = driver->DeviceObject;
385         driver->DeviceObject = device;
386
387         *ret_device = device;
388     }
389     else HeapFree( GetProcessHeap(), 0, device );
390
391     return status;
392 }
393
394
395 /***********************************************************************
396  *           IoDeleteDevice   (NTOSKRNL.EXE.@)
397  */
398 void WINAPI IoDeleteDevice( DEVICE_OBJECT *device )
399 {
400     NTSTATUS status;
401
402     TRACE( "%p\n", device );
403
404     SERVER_START_REQ( delete_device )
405     {
406         req->handle = device->Reserved;
407         status = wine_server_call( req );
408     }
409     SERVER_END_REQ;
410
411     if (status == STATUS_SUCCESS)
412     {
413         DEVICE_OBJECT **prev = &device->DriverObject->DeviceObject;
414         while (*prev && *prev != device) prev = &(*prev)->NextDevice;
415         if (*prev) *prev = (*prev)->NextDevice;
416         NtClose( device->Reserved );
417         HeapFree( GetProcessHeap(), 0, device );
418     }
419 }
420
421
422 /***********************************************************************
423  *           IoCreateSymbolicLink   (NTOSKRNL.EXE.@)
424  */
425 NTSTATUS WINAPI IoCreateSymbolicLink( UNICODE_STRING *name, UNICODE_STRING *target )
426 {
427     HANDLE handle;
428     OBJECT_ATTRIBUTES attr;
429
430     attr.Length                   = sizeof(attr);
431     attr.RootDirectory            = 0;
432     attr.ObjectName               = name;
433     attr.Attributes               = OBJ_CASE_INSENSITIVE | OBJ_OPENIF;
434     attr.SecurityDescriptor       = NULL;
435     attr.SecurityQualityOfService = NULL;
436
437     TRACE( "%s -> %s\n", debugstr_us(name), debugstr_us(target) );
438     /* FIXME: store handle somewhere */
439     return NtCreateSymbolicLinkObject( &handle, SYMBOLIC_LINK_ALL_ACCESS, &attr, target );
440 }
441
442
443 /***********************************************************************
444  *           IoDeleteSymbolicLink   (NTOSKRNL.EXE.@)
445  */
446 NTSTATUS WINAPI IoDeleteSymbolicLink( UNICODE_STRING *name )
447 {
448     FIXME( "%s\n", debugstr_us(name) );
449     return STATUS_SUCCESS;
450 }
451
452
453 /***********************************************************************
454  *           IoGetDeviceObjectPointer   (NTOSKRNL.EXE.@)
455  */
456 NTSTATUS  WINAPI IoGetDeviceObjectPointer( UNICODE_STRING *name, ACCESS_MASK access, PFILE_OBJECT *file, PDEVICE_OBJECT *device )
457 {
458     FIXME( "stub: %s %x %p %p\n", debugstr_us(name), access, file, device );
459     return STATUS_NOT_IMPLEMENTED;
460 }
461
462 /***********************************************************************
463  *           IofCompleteRequest   (NTOSKRNL.EXE.@)
464  */
465 #ifdef DEFINE_FASTCALL2_ENTRYPOINT
466 DEFINE_FASTCALL2_ENTRYPOINT( IofCompleteRequest )
467 void WINAPI __regs_IofCompleteRequest( IRP *irp, UCHAR priority_boost )
468 #else
469 void WINAPI IofCompleteRequest( IRP *irp, UCHAR priority_boost )
470 #endif
471 {
472     TRACE( "%p %u\n", irp, priority_boost );
473     /* nothing to do for now */
474 }
475
476
477 /***********************************************************************
478  *           InterlockedCompareExchange   (NTOSKRNL.EXE.@)
479  */
480 #ifdef DEFINE_FASTCALL2_ENTRYPOINT
481 DEFINE_FASTCALL2_ENTRYPOINT( NTOSKRNL_InterlockedCompareExchange )
482 LONG WINAPI __regs_NTOSKRNL_InterlockedCompareExchange( LONG volatile *dest, LONG xchg, LONG compare )
483 #else
484 LONG WINAPI NTOSKRNL_InterlockedCompareExchange( LONG volatile *dest, LONG xchg, LONG compare )
485 #endif
486 {
487     return InterlockedCompareExchange( dest, xchg, compare );
488 }
489
490
491 /***********************************************************************
492  *           InterlockedDecrement   (NTOSKRNL.EXE.@)
493  */
494 #ifdef DEFINE_FASTCALL1_ENTRYPOINT
495 DEFINE_FASTCALL1_ENTRYPOINT( NTOSKRNL_InterlockedDecrement )
496 LONG WINAPI __regs_NTOSKRNL_InterlockedDecrement( LONG volatile *dest )
497 #else
498 LONG WINAPI NTOSKRNL_InterlockedDecrement( LONG volatile *dest )
499 #endif
500 {
501     return InterlockedDecrement( dest );
502 }
503
504
505 /***********************************************************************
506  *           InterlockedExchange   (NTOSKRNL.EXE.@)
507  */
508 #ifdef DEFINE_FASTCALL2_ENTRYPOINT
509 DEFINE_FASTCALL2_ENTRYPOINT( NTOSKRNL_InterlockedExchange )
510 LONG WINAPI __regs_NTOSKRNL_InterlockedExchange( LONG volatile *dest, LONG val )
511 #else
512 LONG WINAPI NTOSKRNL_InterlockedExchange( LONG volatile *dest, LONG val )
513 #endif
514 {
515     return InterlockedExchange( dest, val );
516 }
517
518
519 /***********************************************************************
520  *           InterlockedExchangeAdd   (NTOSKRNL.EXE.@)
521  */
522 #ifdef DEFINE_FASTCALL2_ENTRYPOINT
523 DEFINE_FASTCALL2_ENTRYPOINT( NTOSKRNL_InterlockedExchangeAdd )
524 LONG WINAPI __regs_NTOSKRNL_InterlockedExchangeAdd( LONG volatile *dest, LONG incr )
525 #else
526 LONG WINAPI NTOSKRNL_InterlockedExchangeAdd( LONG volatile *dest, LONG incr )
527 #endif
528 {
529     return InterlockedExchangeAdd( dest, incr );
530 }
531
532
533 /***********************************************************************
534  *           InterlockedIncrement   (NTOSKRNL.EXE.@)
535  */
536 #ifdef DEFINE_FASTCALL1_ENTRYPOINT
537 DEFINE_FASTCALL1_ENTRYPOINT( NTOSKRNL_InterlockedIncrement )
538 LONG WINAPI __regs_NTOSKRNL_InterlockedIncrement( LONG volatile *dest )
539 #else
540 LONG WINAPI NTOSKRNL_InterlockedIncrement( LONG volatile *dest )
541 #endif
542 {
543     return InterlockedIncrement( dest );
544 }
545
546
547 /***********************************************************************
548  *           ExAllocatePool   (NTOSKRNL.EXE.@)
549  */
550 PVOID WINAPI ExAllocatePool( POOL_TYPE type, SIZE_T size )
551 {
552     return ExAllocatePoolWithTag( type, size, 0 );
553 }
554
555
556 /***********************************************************************
557  *           ExAllocatePoolWithQuota   (NTOSKRNL.EXE.@)
558  */
559 PVOID WINAPI ExAllocatePoolWithQuota( POOL_TYPE type, SIZE_T size )
560 {
561     return ExAllocatePoolWithTag( type, size, 0 );
562 }
563
564
565 /***********************************************************************
566  *           ExAllocatePoolWithTag   (NTOSKRNL.EXE.@)
567  */
568 PVOID WINAPI ExAllocatePoolWithTag( POOL_TYPE type, SIZE_T size, ULONG tag )
569 {
570     /* FIXME: handle page alignment constraints */
571     void *ret = HeapAlloc( GetProcessHeap(), 0, size );
572     TRACE( "%lu pool %u -> %p\n", size, type, ret );
573     return ret;
574 }
575
576
577 /***********************************************************************
578  *           ExAllocatePoolWithQuotaTag   (NTOSKRNL.EXE.@)
579  */
580 PVOID WINAPI ExAllocatePoolWithQuotaTag( POOL_TYPE type, SIZE_T size, ULONG tag )
581 {
582     return ExAllocatePoolWithTag( type, size, tag );
583 }
584
585
586 /***********************************************************************
587  *           ExFreePool   (NTOSKRNL.EXE.@)
588  */
589 void WINAPI ExFreePool( void *ptr )
590 {
591     ExFreePoolWithTag( ptr, 0 );
592 }
593
594
595 /***********************************************************************
596  *           ExFreePoolWithTag   (NTOSKRNL.EXE.@)
597  */
598 void WINAPI ExFreePoolWithTag( void *ptr, ULONG tag )
599 {
600     TRACE( "%p\n", ptr );
601     HeapFree( GetProcessHeap(), 0, ptr );
602 }
603
604
605 /***********************************************************************
606  *           KeInitializeSpinLock   (NTOSKRNL.EXE.@)
607  */
608 void WINAPI KeInitializeSpinLock( PKSPIN_LOCK SpinLock )
609 {
610     FIXME("%p\n", SpinLock);
611 }
612
613
614 /***********************************************************************
615  *           KeInitializeTimerEx   (NTOSKRNL.EXE.@)
616  */
617 void WINAPI KeInitializeTimerEx( PKTIMER Timer, TIMER_TYPE Type )
618 {
619     FIXME("%p %d\n", Timer, Type);
620 }
621
622
623 /***********************************************************************
624  *           KeInitializeTimer   (NTOSKRNL.EXE.@)
625  */
626 void WINAPI KeInitializeTimer( PKTIMER Timer )
627 {
628     KeInitializeTimerEx(Timer, NotificationTimer);
629 }
630
631
632 /**********************************************************************
633  *           KeQueryActiveProcessors   (NTOSKRNL.EXE.@)
634  *
635  * Return the active Processors as bitmask
636  *
637  * RETURNS
638  *   active Processors as bitmask
639  *
640  */
641 KAFFINITY WINAPI KeQueryActiveProcessors( void )
642 {
643     DWORD_PTR AffinityMask;
644
645     GetProcessAffinityMask( GetCurrentProcess(), &AffinityMask, NULL);
646     return AffinityMask;
647 }
648
649
650 /**********************************************************************
651  *           KeQueryInterruptTime   (NTOSKRNL.EXE.@)
652  *
653  * Return the interrupt time count
654  *
655  */
656 ULONGLONG WINAPI KeQueryInterruptTime( void )
657 {
658     LARGE_INTEGER totaltime;
659
660     KeQueryTickCount(&totaltime);
661     return totaltime.QuadPart;
662 }
663
664
665 /***********************************************************************
666  *           KeQuerySystemTime   (NTOSKRNL.EXE.@)
667  */
668 void WINAPI KeQuerySystemTime( LARGE_INTEGER *time )
669 {
670     NtQuerySystemTime( time );
671 }
672
673
674 /***********************************************************************
675  *           KeQueryTickCount   (NTOSKRNL.EXE.@)
676  */
677 void WINAPI KeQueryTickCount( LARGE_INTEGER *count )
678 {
679     count->QuadPart = NtGetTickCount();
680     /* update the global variable too */
681     KeTickCount.LowPart   = count->u.LowPart;
682     KeTickCount.High1Time = count->u.HighPart;
683     KeTickCount.High2Time = count->u.HighPart;
684 }
685
686
687 /***********************************************************************
688  *           KeQueryTimeIncrement   (NTOSKRNL.EXE.@)
689  */
690 ULONG WINAPI KeQueryTimeIncrement(void)
691 {
692     return 10000;
693 }
694
695
696 /***********************************************************************
697  *           MmAllocateNonCachedMemory   (NTOSKRNL.EXE.@)
698  */
699 PVOID WINAPI MmAllocateNonCachedMemory( SIZE_T size )
700 {
701     TRACE( "%lu\n", size );
702     return VirtualAlloc( NULL, size, MEM_RESERVE|MEM_COMMIT, PAGE_READWRITE|PAGE_NOCACHE );
703 }
704
705
706 /***********************************************************************
707  *           MmFreeNonCachedMemory   (NTOSKRNL.EXE.@)
708  */
709 void WINAPI MmFreeNonCachedMemory( void *addr, SIZE_T size )
710 {
711     TRACE( "%p %lu\n", addr, size );
712     VirtualFree( addr, 0, MEM_RELEASE );
713 }
714
715 /***********************************************************************
716  *           MmIsAddressValid   (NTOSKRNL.EXE.@)
717  *
718  * Check if the process can access the virtual address without a pagefault
719  *
720  * PARAMS
721  *  VirtualAddress [I] Address to check
722  *
723  * RETURNS
724  *  Failure: FALSE
725  *  Success: TRUE  (Accessing the Address works without a Pagefault)
726  *
727  */
728 BOOLEAN WINAPI MmIsAddressValid(PVOID VirtualAddress)
729 {
730     TRACE("(%p)\n", VirtualAddress);
731     return !IsBadWritePtr(VirtualAddress, 1);
732 }
733
734 /***********************************************************************
735  *           MmPageEntireDriver   (NTOSKRNL.EXE.@)
736  */
737 PVOID WINAPI MmPageEntireDriver(PVOID AddrInSection)
738 {
739     TRACE("%p\n", AddrInSection);
740     return AddrInSection;
741 }
742
743 /***********************************************************************
744  *           MmResetDriverPaging   (NTOSKRNL.EXE.@)
745  */
746 void WINAPI MmResetDriverPaging(PVOID AddrInSection)
747 {
748     TRACE("%p\n", AddrInSection);
749 }
750
751 /***********************************************************************
752  *           PsCreateSystemThread   (NTOSKRNL.EXE.@)
753  */
754 NTSTATUS WINAPI PsCreateSystemThread(PHANDLE ThreadHandle, ULONG DesiredAccess,
755                                      POBJECT_ATTRIBUTES ObjectAttributes,
756                                      HANDLE ProcessHandle, PCLIENT_ID ClientId,
757                                      PKSTART_ROUTINE StartRoutine, PVOID StartContext)
758 {
759     if (!ProcessHandle) ProcessHandle = GetCurrentProcess();
760     return RtlCreateUserThread(ProcessHandle, 0, FALSE, 0, 0,
761                                0, StartRoutine, StartContext,
762                                ThreadHandle, ClientId);
763 }
764
765 /***********************************************************************
766  *           PsGetCurrentProcessId   (NTOSKRNL.EXE.@)
767  */
768 HANDLE WINAPI PsGetCurrentProcessId(void)
769 {
770     return (HANDLE)GetCurrentProcessId();  /* FIXME: not quite right... */
771 }
772
773
774 /***********************************************************************
775  *           PsGetCurrentThreadId   (NTOSKRNL.EXE.@)
776  */
777 HANDLE WINAPI PsGetCurrentThreadId(void)
778 {
779     return (HANDLE)GetCurrentThreadId();  /* FIXME: not quite right... */
780 }
781
782
783 /***********************************************************************
784  *           PsGetVersion   (NTOSKRNL.EXE.@)
785  */
786 BOOLEAN WINAPI PsGetVersion(ULONG *major, ULONG *minor, ULONG *build, UNICODE_STRING *version )
787 {
788     RTL_OSVERSIONINFOEXW info;
789
790     RtlGetVersion( &info );
791     if (major) *major = info.dwMajorVersion;
792     if (minor) *minor = info.dwMinorVersion;
793     if (build) *build = info.dwBuildNumber;
794
795     if (version)
796     {
797 #if 0  /* FIXME: GameGuard passes an uninitialized pointer in version->Buffer */
798         size_t len = min( strlenW(info.szCSDVersion)*sizeof(WCHAR), version->MaximumLength );
799         memcpy( version->Buffer, info.szCSDVersion, len );
800         if (len < version->MaximumLength) version->Buffer[len / sizeof(WCHAR)] = 0;
801         version->Length = len;
802 #endif
803     }
804     return TRUE;
805 }
806
807
808 /***********************************************************************
809  *           PsSetCreateProcessNotifyRoutine   (NTOSKRNL.EXE.@)
810  */
811 NTSTATUS WINAPI PsSetCreateProcessNotifyRoutine( PCREATE_PROCESS_NOTIFY_ROUTINE callback, BOOLEAN remove )
812 {
813     FIXME( "stub: %p %d\n", callback, remove );
814     return STATUS_SUCCESS;
815 }
816
817
818 /*****************************************************
819  *           DllMain
820  */
821 BOOL WINAPI DllMain( HINSTANCE inst, DWORD reason, LPVOID reserved )
822 {
823     LARGE_INTEGER count;
824
825     switch(reason)
826     {
827     case DLL_PROCESS_ATTACH:
828         DisableThreadLibraryCalls( inst );
829         RtlAddVectoredExceptionHandler( TRUE, vectored_handler );
830         KeQueryTickCount( &count );  /* initialize the global KeTickCount */
831         break;
832     }
833     return TRUE;
834 }