reactos/reactos/subsys/win32k/main/dllmain.c

/*
 *  ReactOS W32 Subsystem
 *  Copyright (C) 1998, 1999, 2000, 2001, 2002, 2003, 2004, 2005 ReactOS Team
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */
/* $Id$
 *
 *  Entry Point for win32k.sys
 */

#include <w32k.h>
#include <win32k/ntddraw.h>
#include <include/napi.h>

#define NDEBUG
#include <debug.h>

BOOL INTERNAL_CALL GDI_CleanupForProcess (struct _EPROCESS *Process);

extern SSDT Win32kSSDT[];
extern SSPT Win32kSSPT[];
extern ULONG Win32kNumberOfSysCalls;

PSHARED_SECTION_POOL SessionSharedSectionPool = NULL;

NTSTATUS 
STDCALL
Win32kProcessCallback(struct _EPROCESS *Process,
                      BOOLEAN Create)
{
    PW32PROCESS Win32Process;
    
    /* Get the Win32 Process */
    Win32Process = PsGetProcessWin32Process(Process);
    
    /* Allocate one if needed */
    if (!Win32Process)
    {
        /* FIXME - lock the process */
        Win32Process = ExAllocatePoolWithTag(NonPagedPool,
                                             sizeof(W32PROCESS),
                                             TAG('W', '3', '2', 'p'));

        if (Win32Process == NULL) return STATUS_NO_MEMORY;

        RtlZeroMemory(Win32Process, sizeof(W32PROCESS));
        
        PsSetProcessWin32Process(Process, Win32Process);
        /* FIXME - unlock the process */
    }

  if (Create)
    {
      DPRINT("Creating W32 process PID:%d at IRQ level: %lu\n", Process->UniqueProcessId, KeGetCurrentIrql());

      InitializeListHead(&Win32Process->ClassListHead);
      ExInitializeFastMutex(&Win32Process->ClassListLock);

      InitializeListHead(&Win32Process->MenuListHead);
      ExInitializeFastMutex(&Win32Process->MenuListLock);

      InitializeListHead(&Win32Process->PrivateFontListHead);
      ExInitializeFastMutex(&Win32Process->PrivateFontListLock);

      InitializeListHead(&Win32Process->DriverObjListHead);
      ExInitializeFastMutex(&Win32Process->DriverObjListLock);

      Win32Process->KeyboardLayout = W32kGetDefaultKeyLayout();

      if(Process->Peb != NULL)
      {
        /* map the gdi handle table to user land */
        Process->Peb->GdiSharedHandleTable = GDI_MapHandleTable(Process);
      }

      /* setup process flags */
      Win32Process->Flags = 0;
    }
  else
    {
      DPRINT("Destroying W32 process PID:%d at IRQ level: %lu\n", Process->UniqueProcessId, KeGetCurrentIrql());
      IntRemoveProcessWndProcHandles((HANDLE)Process->UniqueProcessId);
      IntCleanupMenus(Process, Win32Process);
      IntCleanupCurIcons(Process, Win32Process);
      IntEngCleanupDriverObjs(Process, Win32Process);
      CleanupMonitorImpl();


      GDI_CleanupForProcess(Process);

      IntGraphicsCheck(FALSE);

      /*
       * Deregister logon application automatically
       */
      if(LogonProcess == Win32Process)
      {
        LogonProcess = NULL;
      }
    }

  return STATUS_SUCCESS;
}


NTSTATUS 
STDCALL
Win32kThreadCallback(struct _ETHREAD *Thread,
                     BOOLEAN Create)
{
    struct _EPROCESS *Process;
    PW32THREAD Win32Thread;

    Process = Thread->ThreadsProcess;
    
    /* Get the Win32 Thread */
    Win32Thread = PsGetThreadWin32Thread(Thread);
    
    /* Allocate one if needed */
    if (!Win32Thread)
    {
        /* FIXME - lock the process */
        Win32Thread = ExAllocatePoolWithTag(NonPagedPool,
                                            sizeof(W32THREAD),
                                            TAG('W', '3', '2', 't'));

        if (Win32Thread == NULL) return STATUS_NO_MEMORY;

        RtlZeroMemory(Win32Thread, sizeof(W32THREAD));
        
        PsSetThreadWin32Thread(Thread, Win32Thread);
        /* FIXME - unlock the process */
    }
  if (Create)
    {
      HWINSTA hWinSta = NULL;
      HDESK hDesk = NULL;
      NTSTATUS Status;
      PUNICODE_STRING DesktopPath;
      PRTL_USER_PROCESS_PARAMETERS ProcessParams = (Process->Peb ? Process->Peb->ProcessParameters : NULL);

      DPRINT("Creating W32 thread TID:%d at IRQ level: %lu\n", Thread->Cid.UniqueThread, KeGetCurrentIrql());

      /*
       * inherit the thread desktop and process window station (if not yet inherited) from the process startup
       * info structure. See documentation of CreateProcess()
       */
      DesktopPath = (ProcessParams ? ((ProcessParams->DesktopInfo.Length > 0) ? &ProcessParams->DesktopInfo : NULL) : NULL);
      Status = IntParseDesktopPath(Process,
                                   DesktopPath,
                                   &hWinSta,
                                   &hDesk);
      if(NT_SUCCESS(Status))
      {
        if(hWinSta != NULL)
        {
          if(Process != CsrProcess)
          {
            HWINSTA hProcessWinSta = (HWINSTA)InterlockedCompareExchangePointer((PVOID)&Process->Win32WindowStation, (PVOID)hWinSta, NULL);
            if(hProcessWinSta != NULL)
            {
              /* our process is already assigned to a different window station, we don't need the handle anymore */
              NtClose(hWinSta);
            }
          }
          else
          {
            NtClose(hWinSta);
          }
        }

        if (hDesk != NULL)
        {
          Status = ObReferenceObjectByHandle(hDesk,
                                             0,
                                             ExDesktopObjectType,
                                             KernelMode,
                                             (PVOID*)&Win32Thread->Desktop,
                                             NULL);
          NtClose(hDesk);
          if(!NT_SUCCESS(Status))
          {
            DPRINT1("Unable to reference thread desktop handle 0x%x\n", hDesk);
            Win32Thread->Desktop = NULL;
          }
        }
      }
      Win32Thread->IsExiting = FALSE;
      IntDestroyCaret(Win32Thread);
      Win32Thread->MessageQueue = MsqCreateMessageQueue(Thread);
      Win32Thread->KeyboardLayout = W32kGetDefaultKeyLayout();
      Win32Thread->MessagePumpHookValue = 0;
      InitializeListHead(&Win32Thread->WindowListHead);
      ExInitializeFastMutex(&Win32Thread->WindowListLock);
      InitializeListHead(&Win32Thread->W32CallbackListHead);
    }
  else
    {
      DPRINT("Destroying W32 thread TID:%d at IRQ level: %lu\n", Thread->Cid.UniqueThread, KeGetCurrentIrql());

      Win32Thread->IsExiting = TRUE;
      HOOK_DestroyThreadHooks(Thread);
      UnregisterThreadHotKeys(Thread);
      DestroyThreadWindows(Thread);
      IntBlockInput(Win32Thread, FALSE);
      MsqDestroyMessageQueue(Win32Thread->MessageQueue);
      IntCleanupThreadCallbacks(Win32Thread);
      if(Win32Thread->Desktop != NULL)
      {
        ObDereferenceObject(Win32Thread->Desktop);
      }
    }

  return STATUS_SUCCESS;
}

/* Only used in ntuser/input.c KeyboardThreadMain(). If it's
   not called there anymore, please delete */
NTSTATUS
Win32kInitWin32Thread(PETHREAD Thread)
{
  PEPROCESS Process;

  Process = Thread->ThreadsProcess;

  if (Process->Win32Process == NULL)
    {
      /* FIXME - lock the process */
      Process->Win32Process = ExAllocatePool(NonPagedPool, sizeof(W32PROCESS));

      if (Process->Win32Process == NULL)
	return STATUS_NO_MEMORY;

      RtlZeroMemory(Process->Win32Process, sizeof(W32PROCESS));
      /* FIXME - unlock the process */

      Win32kProcessCallback(Process, TRUE);
    }

  if (Thread->Tcb.Win32Thread == NULL)
    {
      Thread->Tcb.Win32Thread = ExAllocatePool (NonPagedPool, sizeof(W32THREAD));
      if (Thread->Tcb.Win32Thread == NULL)
	return STATUS_NO_MEMORY;

      RtlZeroMemory(Thread->Tcb.Win32Thread, sizeof(W32THREAD));

      Win32kThreadCallback(Thread, TRUE);
    }

  return(STATUS_SUCCESS);
}


/*
 * This definition doesn't work
 */
NTSTATUS STDCALL
DriverEntry (
  IN	PDRIVER_OBJECT	DriverObject,
  IN	PUNICODE_STRING	RegistryPath)
{
  NTSTATUS Status;
  BOOLEAN Result;
  W32_CALLOUT_DATA CalloutData;

  /*
   * Register user mode call interface
   * (system service table index = 1)
   */
  Result = KeAddSystemServiceTable (Win32kSSDT,
				    NULL,
				    Win32kNumberOfSysCalls,
				    Win32kSSPT,
				    1);
  if (Result == FALSE)
    {
      DPRINT1("Adding system services failed!\n");
      return STATUS_UNSUCCESSFUL;
    }

    /*
     * Register Object Manager Callbacks
     */
    CalloutData.WinStaCreate = IntWinStaObjectOpen;
    CalloutData.WinStaParse = IntWinStaObjectParse;
    CalloutData.WinStaDelete = IntWinStaObjectDelete;
    CalloutData.WinStaFind = IntWinStaObjectFind;
    CalloutData.DesktopCreate = IntDesktopObjectCreate;
    CalloutData.DesktopDelete = IntDesktopObjectDelete;
    CalloutData.W32ProcessCallout = Win32kProcessCallback;
    CalloutData.W32ThreadCallout = Win32kThreadCallback;
    
    /*
     * Register our per-process and per-thread structures.
     */
    PsEstablishWin32Callouts(&CalloutData);

  Status = IntUserCreateSharedSectionPool(48 * 1024 * 1024, /* 48 MB by default */
                                          &SessionSharedSectionPool);
  if (!NT_SUCCESS(Status))
  {
    DPRINT1("Failed to initialize the shared section pool: Status 0x%x\n", Status);
  }

  Status = InitWindowStationImpl();
  if (!NT_SUCCESS(Status))
  {
    DPRINT1("Failed to initialize window station implementation!\n");
    return STATUS_UNSUCCESSFUL;
  }

  Status = InitClassImpl();
  if (!NT_SUCCESS(Status))
  {
    DPRINT1("Failed to initialize window class implementation!\n");
    return STATUS_UNSUCCESSFUL;
  }

  Status = InitDesktopImpl();
  if (!NT_SUCCESS(Status))
  {
    DPRINT1("Failed to initialize desktop implementation!\n");
    return STATUS_UNSUCCESSFUL;
  }

  Status = InitWindowImpl();
  if (!NT_SUCCESS(Status))
  {
    DPRINT1("Failed to initialize window implementation!\n");
    return STATUS_UNSUCCESSFUL;
  }

  Status = InitMenuImpl();
  if (!NT_SUCCESS(Status))
  {
    DPRINT1("Failed to initialize menu implementation!\n");
    return STATUS_UNSUCCESSFUL;
  }

  Status = InitInputImpl();
  if (!NT_SUCCESS(Status))
    {
      DPRINT1("Failed to initialize input implementation.\n");
      return(Status);
    }

  Status = InitKeyboardImpl();
  if (!NT_SUCCESS(Status))
    {
      DPRINT1("Failed to initialize keyboard implementation.\n");
      return(Status);
    }

  Status = InitMonitorImpl();
  if (!NT_SUCCESS(Status))
    {
      DbgPrint("Failed to initialize monitor implementation!\n");
      return STATUS_UNSUCCESSFUL;
    }

  Status = MsqInitializeImpl();
  if (!NT_SUCCESS(Status))
    {
      DPRINT1("Failed to initialize message queue implementation.\n");
      return(Status);
    }

  Status = InitTimerImpl();
  if (!NT_SUCCESS(Status))
    {
      DPRINT1("Failed to initialize timer implementation.\n");
      return(Status);
    }

  Status = InitAcceleratorImpl();
  if (!NT_SUCCESS(Status))
    {
      DPRINT1("Failed to initialize accelerator implementation.\n");
      return(Status);
    }

  Status = InitGuiCheckImpl();
  if (!NT_SUCCESS(Status))
    {
      DPRINT1("Failed to initialize GUI check implementation.\n");
      return(Status);
    }

  InitGdiObjectHandleTable ();

  /* Initialize FreeType library */
  if (! InitFontSupport())
    {
      DPRINT1("Unable to initialize font support\n");
      return STATUS_UNSUCCESSFUL;
    }

  /* Create stock objects, ie. precreated objects commonly
     used by win32 applications */
  CreateStockObjects();
  CreateSysColorObjects();

  return STATUS_SUCCESS;
}


BOOLEAN STDCALL
Win32kInitialize (VOID)
{
  return TRUE;
}

/* EOF */