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;
 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  *           IoCreateDevice   (NTOSKRNL.EXE.@)
 253  */
 254 NTSTATUS WINAPI IoCreateDevice( DRIVER_OBJECT *driver, ULONG ext_size,
 255                                 UNICODE_STRING *name, DEVICE_TYPE type,
 256                                 ULONG characteristics, BOOLEAN exclusive,
 257                                 DEVICE_OBJECT **ret_device )
 258 {
 259     NTSTATUS status;
 260     DEVICE_OBJECT *device;
 261     HANDLE handle = 0;
 262     HANDLE manager = get_device_manager();
 263
 264     TRACE( "(%p, %u, %s, %u, %x, %u, %p)\n",
 265            driver, ext_size, debugstr_us(name), type, characteristics, exclusive, ret_device );
 266
 267     if (!(device = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*device) + ext_size )))
 268         return STATUS_NO_MEMORY;
 269
 270     SERVER_START_REQ( create_device )
 271     {
 272         req->access     = 0;
 273         req->attributes = 0;
 274         req->rootdir    = 0;
 275         req->manager    = manager;
 276         req->user_ptr   = device;
 277         if (name) wine_server_add_data( req, name->Buffer, name->Length );
 278         if (!(status = wine_server_call( req ))) handle = reply->handle;
 279     }
 280     SERVER_END_REQ;
 281
 282     if (status == STATUS_SUCCESS)
 283     {
 284         device->DriverObject    = driver;
 285         device->DeviceExtension = device + 1;
 286         device->DeviceType      = type;
 287         device->Reserved        = handle;
 288
 289         device->NextDevice   = driver->DeviceObject;
 290         driver->DeviceObject = device;
 291
 292         *ret_device = device;
 293     }
 294     else HeapFree( GetProcessHeap(), 0, device );
 295
 296     return status;
 297 }
 298
 299
 300 /***********************************************************************
 301  *           IoDeleteDevice   (NTOSKRNL.EXE.@)
 302  */
 303 void WINAPI IoDeleteDevice( DEVICE_OBJECT *device )
 304 {
 305     NTSTATUS status;
 306
 307     TRACE( "%p\n", device );
 308
 309     SERVER_START_REQ( delete_device )
 310     {
 311         req->handle = device->Reserved;
 312         status = wine_server_call( req );
 313     }
 314     SERVER_END_REQ;
 315
 316     if (status == STATUS_SUCCESS)
 317     {
 318         DEVICE_OBJECT **prev = &device->DriverObject->DeviceObject;
 319         while (*prev && *prev != device) prev = &(*prev)->NextDevice;
 320         if (*prev) *prev = (*prev)->NextDevice;
 321         NtClose( device->Reserved );
 322         HeapFree( GetProcessHeap(), 0, device );
 323     }
 324 }
 325
 326
 327 /***********************************************************************
 328  *           IoCreateSymbolicLink   (NTOSKRNL.EXE.@)
 329  */
 330 NTSTATUS WINAPI IoCreateSymbolicLink( UNICODE_STRING *name, UNICODE_STRING *target )
 331 {
 332     HANDLE handle;
 333     OBJECT_ATTRIBUTES attr;
 334
 335     attr.Length                   = sizeof(attr);
 336     attr.RootDirectory            = 0;
 337     attr.ObjectName               = name;
 338     attr.Attributes               = OBJ_CASE_INSENSITIVE | OBJ_OPENIF;
 339     attr.SecurityDescriptor       = NULL;
 340     attr.SecurityQualityOfService = NULL;
 341
 342     TRACE( "%s -> %s\n", debugstr_us(name), debugstr_us(target) );
 343     /* FIXME: store handle somewhere */
 344     return NtCreateSymbolicLinkObject( &handle, SYMBOLIC_LINK_ALL_ACCESS, &attr, target );
 345 }
 346
 347
 348 /***********************************************************************
 349  *           IofCompleteRequest   (NTOSKRNL.EXE.@)
 350  */
 351 #ifdef DEFINE_FASTCALL2_ENTRYPOINT
 352 DEFINE_FASTCALL2_ENTRYPOINT( IofCompleteRequest )
 353 void WINAPI __regs_IofCompleteRequest( IRP *irp, UCHAR priority_boost )
 354 #else
 355 void WINAPI IofCompleteRequest( IRP *irp, UCHAR priority_boost )
 356 #endif
 357 {
 358     TRACE( "%p %u\n", irp, priority_boost );
 359     /* nothing to do for now */
 360 }
 361
 362
 363 /***********************************************************************
 364  *           ExAllocatePool   (NTOSKRNL.EXE.@)
 365  */
 366 PVOID WINAPI ExAllocatePool( POOL_TYPE type, SIZE_T size )
 367 {
 368     return ExAllocatePoolWithTag( type, size, 0 );
 369 }
 370
 371
 372 /***********************************************************************
 373  *           ExAllocatePoolWithQuota   (NTOSKRNL.EXE.@)
 374  */
 375 PVOID WINAPI ExAllocatePoolWithQuota( POOL_TYPE type, SIZE_T size )
 376 {
 377     return ExAllocatePoolWithTag( type, size, 0 );
 378 }
 379
 380
 381 /***********************************************************************
 382  *           ExAllocatePoolWithTag   (NTOSKRNL.EXE.@)
 383  */
 384 PVOID WINAPI ExAllocatePoolWithTag( POOL_TYPE type, SIZE_T size, ULONG tag )
 385 {
 386     /* FIXME: handle page alignment constraints */
 387     void *ret = HeapAlloc( GetProcessHeap(), 0, size );
 388     TRACE( "%lu pool %u -> %p\n", size, type, ret );
 389     return ret;
 390 }
 391
 392
 393 /***********************************************************************
 394  *           ExAllocatePoolWithQuotaTag   (NTOSKRNL.EXE.@)
 395  */
 396 PVOID WINAPI ExAllocatePoolWithQuotaTag( POOL_TYPE type, SIZE_T size, ULONG tag )
 397 {
 398     return ExAllocatePoolWithTag( type, size, tag );
 399 }
 400
 401
 402 /***********************************************************************
 403  *           ExFreePool   (NTOSKRNL.EXE.@)
 404  */
 405 void WINAPI ExFreePool( void *ptr )
 406 {
 407     ExFreePoolWithTag( ptr, 0 );
 408 }
 409
 410
 411 /***********************************************************************
 412  *           ExFreePoolWithTag   (NTOSKRNL.EXE.@)
 413  */
 414 void WINAPI ExFreePoolWithTag( void *ptr, ULONG tag )
 415 {
 416     TRACE( "%p\n", ptr );
 417     HeapFree( GetProcessHeap(), 0, ptr );
 418 }
 419
 420
 421 /***********************************************************************
 422  *           KeQuerySystemTime   (NTOSKRNL.EXE.@)
 423  */
 424 void WINAPI KeQuerySystemTime( LARGE_INTEGER *time )
 425 {
 426     NtQuerySystemTime( time );
 427 }
 428
 429
 430 /***********************************************************************
 431  *           KeQueryTickCount   (NTOSKRNL.EXE.@)
 432  */
 433 void WINAPI KeQueryTickCount( LARGE_INTEGER *count )
 434 {
 435     count->QuadPart = NtGetTickCount();
 436     /* update the global variable too */
 437     KeTickCount.LowPart   = count->u.LowPart;
 438     KeTickCount.High1Time = count->u.HighPart;
 439     KeTickCount.High2Time = count->u.HighPart;
 440 }
 441
 442
 443 /***********************************************************************
 444  *           KeQueryTimeIncrement   (NTOSKRNL.EXE.@)
 445  */
 446 ULONG WINAPI KeQueryTimeIncrement(void)
 447 {
 448     return 10000;
 449 }
 450
 451
 452 /***********************************************************************
 453  *           MmAllocateNonCachedMemory   (NTOSKRNL.EXE.@)
 454  */
 455 LPVOID WINAPI MmAllocateNonCachedMemory( SIZE_T size )
 456 {
 457     TRACE( "%lu\n", size );
 458     return VirtualAlloc( NULL, size, MEM_RESERVE|MEM_COMMIT, PAGE_READWRITE|PAGE_NOCACHE );
 459 }
 460
 461
 462 /***********************************************************************
 463  *           MmFreeNonCachedMemory   (NTOSKRNL.EXE.@)
 464  */
 465 void WINAPI MmFreeNonCachedMemory( void *addr, SIZE_T size )
 466 {
 467     TRACE( "%p %lu\n", addr, size );
 468     VirtualFree( addr, 0, MEM_RELEASE );
 469 }
 470
 471
 472 /***********************************************************************
 473  *           PsGetCurrentProcessId   (NTOSKRNL.EXE.@)
 474  */
 475 HANDLE WINAPI PsGetCurrentProcessId(void)
 476 {
 477     return (HANDLE)GetCurrentProcessId();  /* FIXME: not quite right... */
 478 }
 479
 480
 481 /***********************************************************************
 482  *           PsGetCurrentThreadId   (NTOSKRNL.EXE.@)
 483  */
 484 HANDLE WINAPI PsGetCurrentThreadId(void)
 485 {
 486     return (HANDLE)GetCurrentThreadId();  /* FIXME: not quite right... */
 487 }
 488
 489
 490 /***********************************************************************
 491  *           PsGetVersion   (NTOSKRNL.EXE.@)
 492  */
 493 BOOLEAN WINAPI PsGetVersion(ULONG *major, ULONG *minor, ULONG *build, UNICODE_STRING *version )
 494 {
 495     RTL_OSVERSIONINFOEXW info;
 496
 497     RtlGetVersion( &info );
 498     if (major) *major = info.dwMajorVersion;
 499     if (minor) *minor = info.dwMinorVersion;
 500     if (build) *build = info.dwBuildNumber;
 501
 502     if (version)
 503     {
 504 #if 0  /* FIXME: GameGuard passes an uninitialized pointer in version->Buffer */
 505         size_t len = min( strlenW(info.szCSDVersion)*sizeof(WCHAR), version->MaximumLength );
 506         memcpy( version->Buffer, info.szCSDVersion, len );
 507         if (len < version->MaximumLength) version->Buffer[len / sizeof(WCHAR)] = 0;
 508         version->Length = len;
 509 #endif
 510     }
 511     return TRUE;
 512 }
 513
 514
 515 /***********************************************************************
 516  *           PsSetCreateProcessNotifyRoutine   (NTOSKRNL.EXE.@)
 517  */
 518 NTSTATUS WINAPI PsSetCreateProcessNotifyRoutine( PCREATE_PROCESS_NOTIFY_ROUTINE callback, BOOLEAN remove )
 519 {
 520     FIXME( "stub: %p %d\n", callback, remove );
 521     return STATUS_SUCCESS;
 522 }
 523
 524
 525 /*****************************************************
 526  *           DllMain
 527  */
 528 BOOL WINAPI DllMain( HINSTANCE inst, DWORD reason, LPVOID reserved )
 529 {
 530     LARGE_INTEGER count;
 531
 532     switch(reason)
 533     {
 534     case DLL_PROCESS_ATTACH:
 535         DisableThreadLibraryCalls( inst );
 536         RtlAddVectoredExceptionHandler( TRUE, vectored_handler );
 537         KeQueryTickCount( &count );  /* initialize the global KeTickCount */
 538         break;
 539     }
 540     return TRUE;
 541 }