reactos/subsystems/win32/win32k/main/dllmain.c

   1 /*
   2  * COPYRIGHT:        See COPYING in the top level directory
   3  * PROJECT:          ReactOS kernel
   4  * PURPOSE:          Driver entry and initialization of win32k
   5  * FILE:             subsystems/win32/win32k/main/main.c
   6  * PROGRAMER:
   7  */
   8
   9 #include <win32k.h>
  10 #include <include/napi.h>
  11
  12 #define NDEBUG
  13 #include <debug.h>
  14
  15 HANDLE hModuleWin;
  16
  17 PGDI_HANDLE_TABLE INTERNAL_CALL GDIOBJ_iAllocHandleTable(OUT PSECTION_OBJECT *SectionObject);
  18 BOOL INTERNAL_CALL GDI_CleanupForProcess (struct _EPROCESS *Process);
  19
  20 HANDLE GlobalUserHeap = NULL;
  21 PSECTION_OBJECT GlobalUserHeapSection = NULL;
  22
  23 PSERVERINFO gpsi = NULL; // Global User Server Information.
  24
  25 SHORT gusLanguageID;
  26
  27 extern ULONG_PTR Win32kSSDT[];
  28 extern UCHAR Win32kSSPT[];
  29 extern ULONG Win32kNumberOfSysCalls;
  30
  31 NTSTATUS
  32 APIENTRY
  33 Win32kProcessCallback(struct _EPROCESS *Process,
  34                       BOOLEAN Create)
  35 {
  36     PPROCESSINFO Win32Process;
  37     DECLARE_RETURN(NTSTATUS);
  38
  39     DPRINT("Enter Win32kProcessCallback\n");
  40     UserEnterExclusive();
  41
  42     /* Get the Win32 Process */
  43     Win32Process = PsGetProcessWin32Process(Process);
  44
  45     /* Allocate one if needed */
  46     if (!Win32Process)
  47     {
  48         /* FIXME - lock the process */
  49         Win32Process = ExAllocatePoolWithTag(NonPagedPool,
  50                                              sizeof(PROCESSINFO),
  51                                              'p23W');
  52
  53         if (Win32Process == NULL) RETURN( STATUS_NO_MEMORY);
  54
  55         RtlZeroMemory(Win32Process, sizeof(PROCESSINFO));
  56
  57         PsSetProcessWin32Process(Process, Win32Process);
  58         /* FIXME - unlock the process */
  59     }
  60
  61     if (Create)
  62     {
  63         SIZE_T ViewSize = 0;
  64         LARGE_INTEGER Offset;
  65         PVOID UserBase = NULL;
  66         NTSTATUS Status;
  67         extern PSECTION_OBJECT GlobalUserHeapSection;
  68         DPRINT("Creating W32 process PID:%d at IRQ level: %lu\n", Process->UniqueProcessId, KeGetCurrentIrql());
  69
  70         /* map the global heap into the process */
  71         Offset.QuadPart = 0;
  72         Status = MmMapViewOfSection(GlobalUserHeapSection,
  73                                     PsGetCurrentProcess(),
  74                                     &UserBase,
  75                                     0,
  76                                     0,
  77                                     &Offset,
  78                                     &ViewSize,
  79                                     ViewUnmap,
  80                                     SEC_NO_CHANGE,
  81                                     PAGE_EXECUTE_READ); /* would prefer PAGE_READONLY, but thanks to RTL heaps... */
  82         if (!NT_SUCCESS(Status))
  83         {
  84             DPRINT1("Failed to map the global heap! 0x%x\n", Status);
  85             RETURN(Status);
  86         }
  87         Win32Process->HeapMappings.Next = NULL;
  88         Win32Process->HeapMappings.KernelMapping = (PVOID)GlobalUserHeap;
  89         Win32Process->HeapMappings.UserMapping = UserBase;
  90         Win32Process->HeapMappings.Count = 1;
  91
  92         InitializeListHead(&Win32Process->MenuListHead);
  93
  94         InitializeListHead(&Win32Process->GDIBrushAttrFreeList);
  95         InitializeListHead(&Win32Process->GDIDcAttrFreeList);
  96
  97         InitializeListHead(&Win32Process->PrivateFontListHead);
  98         ExInitializeFastMutex(&Win32Process->PrivateFontListLock);
  99
 100         InitializeListHead(&Win32Process->DriverObjListHead);
 101         ExInitializeFastMutex(&Win32Process->DriverObjListLock);
 102
 103         Win32Process->KeyboardLayout = W32kGetDefaultKeyLayout();
 104         EngCreateEvent((PEVENT *)&Win32Process->InputIdleEvent);
 105         KeInitializeEvent(Win32Process->InputIdleEvent, NotificationEvent, FALSE);
 106
 107         if(Process->Peb != NULL)
 108         {
 109             /* map the gdi handle table to user land */
 110             Process->Peb->GdiSharedHandleTable = GDI_MapHandleTable(GdiTableSection, Process);
 111             Process->Peb->GdiDCAttributeList = GDI_BATCH_LIMIT;
 112         }
 113
 114         Win32Process->peProcess = Process;
 115         /* setup process flags */
 116         Win32Process->W32PF_flags = 0;
 117     }
 118     else
 119     {
 120         DPRINT("Destroying W32 process PID:%d at IRQ level: %lu\n", Process->UniqueProcessId, KeGetCurrentIrql());
 121         Win32Process->W32PF_flags |= W32PF_TERMINATED;
 122         if (Win32Process->InputIdleEvent)
 123         {
 124            EngFreeMem((PVOID)Win32Process->InputIdleEvent);
 125            Win32Process->InputIdleEvent = NULL;
 126         }
 127
 128         IntCleanupMenus(Process, Win32Process);
 129         IntCleanupCurIcons(Process, Win32Process);
 130         CleanupMonitorImpl();
 131
 132         /* no process windows should exist at this point, or the function will assert! */
 133         DestroyProcessClasses(Win32Process);
 134         Win32Process->W32PF_flags &= ~W32PF_CLASSESREGISTERED;
 135
 136         GDI_CleanupForProcess(Process);
 137
 138         co_IntGraphicsCheck(FALSE);
 139
 140         /*
 141          * Deregister logon application automatically
 142          */
 143         if(LogonProcess == Win32Process)
 144         {
 145             LogonProcess = NULL;
 146         }
 147     }
 148
 149     RETURN( STATUS_SUCCESS);
 150
 151 CLEANUP:
 152     UserLeave();
 153     DPRINT("Leave Win32kProcessCallback, ret=%i\n",_ret_);
 154     END_CLEANUP;
 155 }
 156
 157
 158 NTSTATUS
 159 APIENTRY
 160 Win32kThreadCallback(struct _ETHREAD *Thread,
 161                      PSW32THREADCALLOUTTYPE Type)
 162 {
 163     struct _EPROCESS *Process;
 164     PTHREADINFO Win32Thread;
 165     int i;
 166     DECLARE_RETURN(NTSTATUS);
 167
 168     DPRINT("Enter Win32kThreadCallback\n");
 169     UserEnterExclusive();
 170
 171     Process = Thread->ThreadsProcess;
 172
 173     /* Get the Win32 Thread */
 174     Win32Thread = PsGetThreadWin32Thread(Thread);
 175
 176     /* Allocate one if needed */
 177     if (!Win32Thread)
 178     {
 179         /* FIXME - lock the process */
 180         Win32Thread = ExAllocatePoolWithTag(NonPagedPool,
 181                                             sizeof(THREADINFO),
 182                                             't23W');
 183
 184         if (Win32Thread == NULL) RETURN( STATUS_NO_MEMORY);
 185
 186         RtlZeroMemory(Win32Thread, sizeof(THREADINFO));
 187
 188         PsSetThreadWin32Thread(Thread, Win32Thread);
 189         /* FIXME - unlock the process */
 190     }
 191     if (Type == PsW32ThreadCalloutInitialize)
 192     {
 193         HWINSTA hWinSta = NULL;
 194         PTEB pTeb;
 195         HDESK hDesk = NULL;
 196         NTSTATUS Status;
 197         PUNICODE_STRING DesktopPath;
 198         PRTL_USER_PROCESS_PARAMETERS ProcessParams = (Process->Peb ? Process->Peb->ProcessParameters : NULL);
 199
 200         DPRINT("Creating W32 thread TID:%d at IRQ level: %lu\n", Thread->Cid.UniqueThread, KeGetCurrentIrql());
 201
 202         InitializeListHead(&Win32Thread->WindowListHead);
 203         InitializeListHead(&Win32Thread->W32CallbackListHead);
 204         InitializeListHead(&Win32Thread->PtiLink);
 205         for (i = 0; i < NB_HOOKS; i++)
 206         {
 207             InitializeListHead(&Win32Thread->aphkStart[i]);
 208         }
 209
 210         /*
 211          * inherit the thread desktop and process window station (if not yet inherited) from the process startup
 212          * info structure. See documentation of CreateProcess()
 213          */
 214         DesktopPath = (ProcessParams ? ((ProcessParams->DesktopInfo.Length > 0) ? &ProcessParams->DesktopInfo : NULL) : NULL);
 215         Status = IntParseDesktopPath(Process,
 216                                      DesktopPath,
 217                                      &hWinSta,
 218                                      &hDesk);
 219         if(NT_SUCCESS(Status))
 220         {
 221             if(hWinSta != NULL)
 222             {
 223                 if(Process != CsrProcess)
 224                 {
 225                     HWINSTA hProcessWinSta = (HWINSTA)InterlockedCompareExchangePointer((PVOID)&Process->Win32WindowStation, (PVOID)hWinSta, NULL);
 226                     if(hProcessWinSta != NULL)
 227                     {
 228                         /* our process is already assigned to a different window station, we don't need the handle anymore */
 229                         NtClose(hWinSta);
 230                     }
 231                 }
 232                 else
 233                 {
 234                     NtClose(hWinSta);
 235                 }
 236             }
 237
 238             if (hDesk != NULL)
 239             {
 240                 Win32Thread->rpdesk = NULL;
 241                 Win32Thread->hdesk = NULL;
 242                 if (!IntSetThreadDesktop(hDesk, FALSE))
 243                 {
 244                         DPRINT1("Unable to set thread desktop\n");
 245                 }
 246             }
 247         }
 248         else
 249         {
 250            DPRINT1("No Desktop handle for this Thread!\n");
 251         }
 252         Win32Thread->TIF_flags &= ~TIF_INCLEANUP;
 253         co_IntDestroyCaret(Win32Thread);
 254         Win32Thread->ppi = PsGetCurrentProcessWin32Process();
 255         Win32Thread->ptiSibling = Win32Thread->ppi->ptiList;
 256         Win32Thread->ppi->ptiList = Win32Thread;
 257         Win32Thread->ppi->cThreads++;
 258         if (Win32Thread->rpdesk && !Win32Thread->pDeskInfo)
 259         {
 260            Win32Thread->pDeskInfo = Win32Thread->rpdesk->pDeskInfo;
 261         }
 262         Win32Thread->MessageQueue = MsqCreateMessageQueue(Thread);
 263         Win32Thread->KeyboardLayout = W32kGetDefaultKeyLayout();
 264         pTeb = NtCurrentTeb();
 265         if (pTeb)
 266         { /* Attempt to startup client support which should have been initialized in IntSetThreadDesktop. */
 267            PCLIENTINFO pci = (PCLIENTINFO)pTeb->Win32ClientInfo;
 268            Win32Thread->pClientInfo = pci;
 269            pci->ppi = Win32Thread->ppi;
 270            pci->fsHooks = Win32Thread->fsHooks;
 271            if (Win32Thread->KeyboardLayout) pci->hKL = Win32Thread->KeyboardLayout->hkl;
 272            pci->dwTIFlags = Win32Thread->TIF_flags;
 273            /* CI may not have been initialized. */
 274            if (!pci->pDeskInfo && Win32Thread->pDeskInfo)
 275            {
 276               if (!pci->ulClientDelta) pci->ulClientDelta = DesktopHeapGetUserDelta();
 277
 278               pci->pDeskInfo = (PVOID)((ULONG_PTR)Win32Thread->pDeskInfo - pci->ulClientDelta);
 279            }
 280            if (Win32Thread->pcti && pci->pDeskInfo)
 281               pci->pClientThreadInfo = (PVOID)((ULONG_PTR)Win32Thread->pcti - pci->ulClientDelta);
 282            else
 283               pci->pClientThreadInfo = NULL;
 284         }
 285         else
 286         {
 287            DPRINT1("No TEB for this Thread!\n");
 288            // System thread running! Now SendMessage should be okay.
 289            Win32Thread->pcti = &Win32Thread->cti;
 290         }
 291         Win32Thread->pEThread = Thread;
 292     }
 293     else
 294     {
 295         PTHREADINFO pti;
 296         PSINGLE_LIST_ENTRY e;
 297
 298         DPRINT("Destroying W32 thread TID:%d at IRQ level: %lu\n", Thread->Cid.UniqueThread, KeGetCurrentIrql());
 299
 300         Win32Thread->TIF_flags |= TIF_INCLEANUP;
 301         pti = Win32Thread->ppi->ptiList;
 302         if (pti == Win32Thread)
 303         {
 304            Win32Thread->ppi->ptiList = Win32Thread->ptiSibling;
 305            Win32Thread->ppi->cThreads--;
 306         }
 307         else
 308         {
 309            do
 310            {
 311                if (pti->ptiSibling == Win32Thread)
 312                {
 313                   pti->ptiSibling = Win32Thread->ptiSibling;
 314                   Win32Thread->ppi->cThreads--;
 315                   break;
 316                }
 317                pti = pti->ptiSibling;
 318            }
 319            while (pti);
 320         }
 321         DceFreeThreadDCE(Win32Thread);
 322         HOOK_DestroyThreadHooks(Thread);
 323         EVENT_DestroyThreadEvents(Thread);
 324         /* Cleanup timers */
 325         DestroyTimersForThread(Win32Thread);
 326         KeSetEvent(Win32Thread->MessageQueue->NewMessages, IO_NO_INCREMENT, FALSE);
 327         UnregisterThreadHotKeys(Thread);
 328         /* what if this co_ func crash in umode? what will clean us up then? */
 329         co_DestroyThreadWindows(Thread);
 330         IntBlockInput(Win32Thread, FALSE);
 331         MsqDestroyMessageQueue(Win32Thread->MessageQueue);
 332         IntCleanupThreadCallbacks(Win32Thread);
 333
 334         /* cleanup user object references stack */
 335         e = PopEntryList(&Win32Thread->ReferencesList);
 336         while (e)
 337         {
 338             PUSER_REFERENCE_ENTRY ref = CONTAINING_RECORD(e, USER_REFERENCE_ENTRY, Entry);
 339             DPRINT("thread clean: remove reference obj 0x%x\n",ref->obj);
 340             UserDereferenceObject(ref->obj);
 341
 342             e = PopEntryList(&Win32Thread->ReferencesList);
 343         }
 344
 345         IntSetThreadDesktop(NULL, TRUE);
 346
 347         PsSetThreadWin32Thread(Thread, NULL);
 348     }
 349
 350     RETURN( STATUS_SUCCESS);
 351
 352 CLEANUP:
 353     UserLeave();
 354     DPRINT("Leave Win32kThreadCallback, ret=%i\n",_ret_);
 355     END_CLEANUP;
 356 }
 357
 358 NTSTATUS
 359 Win32kInitWin32Thread(PETHREAD Thread)
 360 {
 361     PEPROCESS Process;
 362
 363     Process = Thread->ThreadsProcess;
 364
 365     if (Process->Win32Process == NULL)
 366     {
 367         /* FIXME - lock the process */
 368         Process->Win32Process = ExAllocatePoolWithTag(NonPagedPool, sizeof(PROCESSINFO), USERTAG_PROCESSINFO);
 369
 370         if (Process->Win32Process == NULL)
 371             return STATUS_NO_MEMORY;
 372
 373         RtlZeroMemory(Process->Win32Process, sizeof(PROCESSINFO));
 374         /* FIXME - unlock the process */
 375
 376         Win32kProcessCallback(Process, TRUE);
 377     }
 378
 379     if (Thread->Tcb.Win32Thread == NULL)
 380     {
 381         Thread->Tcb.Win32Thread = ExAllocatePoolWithTag(NonPagedPool, sizeof(THREADINFO), USERTAG_THREADINFO);
 382         if (Thread->Tcb.Win32Thread == NULL)
 383             return STATUS_NO_MEMORY;
 384
 385         RtlZeroMemory(Thread->Tcb.Win32Thread, sizeof(THREADINFO));
 386
 387         Win32kThreadCallback(Thread, PsW32ThreadCalloutInitialize);
 388     }
 389
 390     return(STATUS_SUCCESS);
 391 }
 392
 393 C_ASSERT(sizeof(SERVERINFO) <= PAGE_SIZE);
 394
 395 // Return on failure
 396 #define NT_ROF(x) \
 397     Status = (x); \
 398     if (!NT_SUCCESS(Status)) \
 399     { \
 400         DPRINT1("Failed '%s' (0x%lx)\n", #x, Status); \
 401         return Status; \
 402     }
 403
 404 /*
 405  * This definition doesn't work
 406  */
 407 INIT_FUNCTION
 408 NTSTATUS
 409 APIENTRY
 410 DriverEntry(
 411     IN  PDRIVER_OBJECT  DriverObject,
 412     IN  PUNICODE_STRING RegistryPath)
 413 {
 414     NTSTATUS Status;
 415     BOOLEAN Result;
 416     WIN32_CALLOUTS_FPNS CalloutData = {0};
 417     PVOID GlobalUserHeapBase = NULL;
 418
 419     /*
 420      * Register user mode call interface
 421      * (system service table index = 1)
 422      */
 423     Result = KeAddSystemServiceTable(Win32kSSDT,
 424                                      NULL,
 425                                      Win32kNumberOfSysCalls,
 426                                      Win32kSSPT,
 427                                      1);
 428     if (Result == FALSE)
 429     {
 430         DPRINT1("Adding system services failed!\n");
 431         return STATUS_UNSUCCESSFUL;
 432     }
 433
 434     hModuleWin = MmPageEntireDriver(DriverEntry);
 435     DPRINT("Win32k hInstance 0x%x!\n",hModuleWin);
 436
 437     /* Register Object Manager Callbacks */
 438     CalloutData.WindowStationParseProcedure = IntWinStaObjectParse;
 439     CalloutData.WindowStationDeleteProcedure = IntWinStaObjectDelete;
 440     CalloutData.DesktopDeleteProcedure = IntDesktopObjectDelete;
 441     CalloutData.ProcessCallout = Win32kProcessCallback;
 442     CalloutData.ThreadCallout = Win32kThreadCallback;
 443     CalloutData.BatchFlushRoutine = NtGdiFlushUserBatch;
 444
 445     /* Register our per-process and per-thread structures. */
 446     PsEstablishWin32Callouts((PWIN32_CALLOUTS_FPNS)&CalloutData);
 447
 448 #if 0 // DBG
 449     /* Register service hook callbacks */
 450     KdSystemDebugControl('CsoR', DbgPreServiceHook, ID_Win32PreServiceHook, 0, 0, 0, 0);
 451     KdSystemDebugControl('CsoR', DbgPostServiceHook, ID_Win32PostServiceHook, 0, 0, 0, 0);
 452 #endif
 453
 454     /* Create the global USER heap */
 455     GlobalUserHeap = UserCreateHeap(&GlobalUserHeapSection,
 456                                     &GlobalUserHeapBase,
 457                                     1 * 1024 * 1024); /* FIXME - 1 MB for now... */
 458     if (GlobalUserHeap == NULL)
 459     {
 460         DPRINT1("Failed to initialize the global heap!\n");
 461         return STATUS_UNSUCCESSFUL;
 462     }
 463
 464     /* Allocate global server info structure */
 465     gpsi = UserHeapAlloc(sizeof(SERVERINFO));
 466     if (!gpsi)
 467     {
 468         DPRINT1("Failed allocate server info structure!\n");
 469         return STATUS_UNSUCCESSFUL;
 470     }
 471
 472     RtlZeroMemory(gpsi, sizeof(SERVERINFO));
 473     DPRINT("Global Server Data -> %x\n", gpsi);
 474
 475     NT_ROF(InitGdiHandleTable());
 476     NT_ROF(InitPaletteImpl());
 477
 478     /* Create stock objects, ie. precreated objects commonly
 479        used by win32 applications */
 480     CreateStockObjects();
 481     CreateSysColorObjects();
 482
 483     NT_ROF(InitXlateImpl());
 484     NT_ROF(InitPDEVImpl());
 485     NT_ROF(InitLDEVImpl());
 486     NT_ROF(InitDeviceImpl());
 487     NT_ROF(InitDcImpl());
 488     NT_ROF(InitUserImpl());
 489     NT_ROF(InitHotkeyImpl());
 490     NT_ROF(InitWindowStationImpl());
 491     NT_ROF(InitDesktopImpl());
 492     NT_ROF(InitWindowImpl());
 493     NT_ROF(InitMenuImpl());
 494     NT_ROF(InitInputImpl());
 495     NT_ROF(InitKeyboardImpl());
 496     NT_ROF(InitMonitorImpl());
 497     NT_ROF(MsqInitializeImpl());
 498     NT_ROF(InitTimerImpl());
 499     NT_ROF(InitAcceleratorImpl());
 500     NT_ROF(InitGuiCheckImpl());
 501
 502     /* Initialize FreeType library */
 503     if (!InitFontSupport())
 504     {
 505         DPRINT1("Unable to initialize font support\n");
 506         return Status;
 507     }
 508
 509     gusLanguageID = IntGdiGetLanguageID();
 510
 511     return STATUS_SUCCESS;
 512 }
 513
 514 /* EOF */