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