subsystems/win32/csrss/win32csr/dllmain.c

   1 /*
   2  * COPYRIGHT:       See COPYING in the top level directory
   3  * PROJECT:         ReactOS system libraries
   4  * FILE:            subsys/csrss/win32csr/dllmain.c
   5  * PURPOSE:         Initialization
   6  * PROGRAMMERS:     Dmitry Philippov (shedon@mail.ru)
   7  */
   8
   9 /* INCLUDES ******************************************************************/
  10 #define NDEBUG
  11 #include "w32csr.h"
  12 #include "file.h"
  13 #include <debug.h>
  14
  15 /* Not defined in any header file */
  16 extern VOID WINAPI PrivateCsrssManualGuiCheck(LONG Check);
  17 extern VOID WINAPI InitializeAppSwitchHook();
  18 extern LIST_ENTRY DosDeviceHistory;
  19 extern RTL_CRITICAL_SECTION Win32CsrDefineDosDeviceCritSec;
  20
  21 /* GLOBALS *******************************************************************/
  22
  23 HANDLE Win32CsrApiHeap;
  24 HINSTANCE Win32CsrDllHandle = NULL;
  25 static CSRSS_EXPORTED_FUNCS CsrExports;
  26
  27 static CSRSS_API_DEFINITION Win32CsrApiDefinitions[] =
  28 {
  29     CSRSS_DEFINE_API(GET_INPUT_HANDLE,             CsrGetHandle),
  30     CSRSS_DEFINE_API(GET_OUTPUT_HANDLE,            CsrGetHandle),
  31     CSRSS_DEFINE_API(CLOSE_HANDLE,                 CsrCloseHandle),
  32     CSRSS_DEFINE_API(VERIFY_HANDLE,                CsrVerifyHandle),
  33     CSRSS_DEFINE_API(DUPLICATE_HANDLE,             CsrDuplicateHandle),
  34     CSRSS_DEFINE_API(GET_INPUT_WAIT_HANDLE,        CsrGetInputWaitHandle),
  35     CSRSS_DEFINE_API(WRITE_CONSOLE,                CsrWriteConsole),
  36     CSRSS_DEFINE_API(READ_CONSOLE,                 CsrReadConsole),
  37     CSRSS_DEFINE_API(ALLOC_CONSOLE,                CsrAllocConsole),
  38     CSRSS_DEFINE_API(FREE_CONSOLE,                 CsrFreeConsole),
  39     CSRSS_DEFINE_API(SCREEN_BUFFER_INFO,           CsrGetScreenBufferInfo),
  40     CSRSS_DEFINE_API(SET_CURSOR,                   CsrSetCursor),
  41     CSRSS_DEFINE_API(FILL_OUTPUT,                  CsrFillOutputChar),
  42     CSRSS_DEFINE_API(READ_INPUT,                   CsrReadInputEvent),
  43     CSRSS_DEFINE_API(WRITE_CONSOLE_OUTPUT_CHAR,    CsrWriteConsoleOutputChar),
  44     CSRSS_DEFINE_API(WRITE_CONSOLE_OUTPUT_ATTRIB,  CsrWriteConsoleOutputAttrib),
  45     CSRSS_DEFINE_API(FILL_OUTPUT_ATTRIB,           CsrFillOutputAttrib),
  46     CSRSS_DEFINE_API(GET_CURSOR_INFO,              CsrGetCursorInfo),
  47     CSRSS_DEFINE_API(SET_CURSOR_INFO,              CsrSetCursorInfo),
  48     CSRSS_DEFINE_API(SET_ATTRIB,                   CsrSetTextAttrib),
  49     CSRSS_DEFINE_API(GET_CONSOLE_MODE,             CsrGetConsoleMode),
  50     CSRSS_DEFINE_API(SET_CONSOLE_MODE,             CsrSetConsoleMode),
  51     CSRSS_DEFINE_API(CREATE_SCREEN_BUFFER,         CsrCreateScreenBuffer),
  52     CSRSS_DEFINE_API(SET_SCREEN_BUFFER,            CsrSetScreenBuffer),
  53     CSRSS_DEFINE_API(SET_TITLE,                    CsrSetTitle),
  54     CSRSS_DEFINE_API(GET_TITLE,                    CsrGetTitle),
  55     CSRSS_DEFINE_API(WRITE_CONSOLE_OUTPUT,         CsrWriteConsoleOutput),
  56     CSRSS_DEFINE_API(FLUSH_INPUT_BUFFER,           CsrFlushInputBuffer),
  57     CSRSS_DEFINE_API(SCROLL_CONSOLE_SCREEN_BUFFER, CsrScrollConsoleScreenBuffer),
  58     CSRSS_DEFINE_API(READ_CONSOLE_OUTPUT_CHAR,     CsrReadConsoleOutputChar),
  59     CSRSS_DEFINE_API(READ_CONSOLE_OUTPUT_ATTRIB,   CsrReadConsoleOutputAttrib),
  60     CSRSS_DEFINE_API(GET_NUM_INPUT_EVENTS,         CsrGetNumberOfConsoleInputEvents),
  61     CSRSS_DEFINE_API(EXIT_REACTOS,                 CsrExitReactos),
  62     CSRSS_DEFINE_API(PEEK_CONSOLE_INPUT,           CsrPeekConsoleInput),
  63     CSRSS_DEFINE_API(READ_CONSOLE_OUTPUT,          CsrReadConsoleOutput),
  64     CSRSS_DEFINE_API(WRITE_CONSOLE_INPUT,          CsrWriteConsoleInput),
  65     CSRSS_DEFINE_API(SETGET_CONSOLE_HW_STATE,      CsrHardwareStateProperty),
  66     CSRSS_DEFINE_API(GET_CONSOLE_WINDOW,           CsrGetConsoleWindow),
  67     CSRSS_DEFINE_API(CREATE_DESKTOP,               CsrCreateDesktop),
  68     CSRSS_DEFINE_API(SHOW_DESKTOP,                 CsrShowDesktop),
  69     CSRSS_DEFINE_API(HIDE_DESKTOP,                 CsrHideDesktop),
  70     CSRSS_DEFINE_API(SET_CONSOLE_ICON,             CsrSetConsoleIcon),
  71     CSRSS_DEFINE_API(SET_LOGON_NOTIFY_WINDOW,      CsrSetLogonNotifyWindow),
  72     CSRSS_DEFINE_API(REGISTER_LOGON_PROCESS,       CsrRegisterLogonProcess),
  73     CSRSS_DEFINE_API(GET_CONSOLE_CP,               CsrGetConsoleCodePage),
  74     CSRSS_DEFINE_API(SET_CONSOLE_CP,               CsrSetConsoleCodePage),
  75     CSRSS_DEFINE_API(GET_CONSOLE_OUTPUT_CP,        CsrGetConsoleOutputCodePage),
  76     CSRSS_DEFINE_API(SET_CONSOLE_OUTPUT_CP,        CsrSetConsoleOutputCodePage),
  77     CSRSS_DEFINE_API(GET_PROCESS_LIST,             CsrGetProcessList),
  78     CSRSS_DEFINE_API(ADD_CONSOLE_ALIAS,      CsrAddConsoleAlias),
  79     CSRSS_DEFINE_API(GET_CONSOLE_ALIAS,      CsrGetConsoleAlias),
  80     CSRSS_DEFINE_API(GET_ALL_CONSOLE_ALIASES,         CsrGetAllConsoleAliases),
  81     CSRSS_DEFINE_API(GET_ALL_CONSOLE_ALIASES_LENGTH,  CsrGetAllConsoleAliasesLength),
  82     CSRSS_DEFINE_API(GET_CONSOLE_ALIASES_EXES,        CsrGetConsoleAliasesExes),
  83     CSRSS_DEFINE_API(GET_CONSOLE_ALIASES_EXES_LENGTH, CsrGetConsoleAliasesExesLength),
  84     CSRSS_DEFINE_API(GENERATE_CTRL_EVENT,          CsrGenerateCtrlEvent),
  85     CSRSS_DEFINE_API(SET_SCREEN_BUFFER_SIZE,       CsrSetScreenBufferSize),
  86     CSRSS_DEFINE_API(GET_CONSOLE_SELECTION_INFO,   CsrGetConsoleSelectionInfo),
  87     CSRSS_DEFINE_API(GET_COMMAND_HISTORY_LENGTH,   CsrGetCommandHistoryLength),
  88     CSRSS_DEFINE_API(GET_COMMAND_HISTORY,          CsrGetCommandHistory),
  89     CSRSS_DEFINE_API(EXPUNGE_COMMAND_HISTORY,      CsrExpungeCommandHistory),
  90     CSRSS_DEFINE_API(SET_HISTORY_NUMBER_COMMANDS,  CsrSetHistoryNumberCommands),
  91     CSRSS_DEFINE_API(GET_HISTORY_INFO,             CsrGetHistoryInfo),
  92     CSRSS_DEFINE_API(SET_HISTORY_INFO,             CsrSetHistoryInfo),
  93     CSRSS_DEFINE_API(GET_TEMP_FILE,                CsrGetTempFile),
  94     CSRSS_DEFINE_API(DEFINE_DOS_DEVICE,            CsrDefineDosDevice),
  95     { 0, 0, NULL }
  96 };
  97
  98 /* FUNCTIONS *****************************************************************/
  99
 100 BOOL WINAPI
 101 DllMain(HANDLE hDll,
 102         DWORD dwReason,
 103         LPVOID lpReserved)
 104 {
 105     if (DLL_PROCESS_ATTACH == dwReason)
 106     {
 107         Win32CsrDllHandle = hDll;
 108         InitializeAppSwitchHook();
 109     }
 110
 111     if (DLL_PROCESS_DETACH == dwReason)
 112     {
 113         CsrCleanupDefineDosDevice();
 114     }
 115     return TRUE;
 116 }
 117
 118 /* Ensure that a captured buffer is safe to access */
 119 BOOL FASTCALL
 120 Win32CsrValidateBuffer(PCSRSS_PROCESS_DATA ProcessData, PVOID Buffer,
 121                        SIZE_T NumElements, SIZE_T ElementSize)
 122 {
 123     /* Check that the following conditions are true:
 124      * 1. The start of the buffer is somewhere within the process's
 125      *    shared memory section view.
 126      * 2. The remaining space in the view is at least as large as the buffer.
 127      *    (NB: Please don't try to "optimize" this by using multiplication
 128      *    instead of division; remember that 2147483648 * 2 = 0.)
 129      * 3. The buffer is DWORD-aligned.
 130      */
 131     ULONG_PTR Offset = (BYTE *)Buffer - (BYTE *)ProcessData->CsrSectionViewBase;
 132     if (Offset >= ProcessData->CsrSectionViewSize
 133             || NumElements > (ProcessData->CsrSectionViewSize - Offset) / ElementSize
 134             || (Offset & (sizeof(DWORD) - 1)) != 0)
 135     {
 136         DPRINT1("Invalid buffer %p(%u*%u); section view is %p(%u)\n",
 137                 Buffer, NumElements, ElementSize,
 138                 ProcessData->CsrSectionViewBase, ProcessData->CsrSectionViewSize);
 139         return FALSE;
 140     }
 141     return TRUE;
 142 }
 143
 144 NTSTATUS FASTCALL
 145 Win32CsrEnumProcesses(CSRSS_ENUM_PROCESS_PROC EnumProc,
 146                       PVOID Context)
 147 {
 148     return (CsrExports.CsrEnumProcessesProc)(EnumProc, Context);
 149 }
 150
 151 static BOOL WINAPI
 152 Win32CsrInitComplete(void)
 153 {
 154     return TRUE;
 155 }
 156
 157 VOID
 158 WINAPI
 159 PrivateCsrssManualGuiCheck(LONG Check)
 160 {
 161   NtUserCallOneParam(Check, ONEPARAM_ROUTINE_CSRSS_GUICHECK);
 162 }
 163
 164 BOOL WINAPI
 165 Win32CsrInitialization(PCSRSS_API_DEFINITION *ApiDefinitions,
 166                        PCSRPLUGIN_SERVER_PROCS ServerProcs,
 167                        PCSRSS_EXPORTED_FUNCS Exports,
 168                        HANDLE CsrssApiHeap)
 169 {
 170     NTSTATUS Status;
 171     CsrExports = *Exports;
 172     Win32CsrApiHeap = CsrssApiHeap;
 173
 174     Status = NtUserInitialize(0, NULL, NULL);
 175
 176     PrivateCsrssManualGuiCheck(0);
 177     CsrInitConsoleSupport();
 178
 179     *ApiDefinitions = Win32CsrApiDefinitions;
 180     ServerProcs->InitCompleteProc = Win32CsrInitComplete;
 181     ServerProcs->HardErrorProc = Win32CsrHardError;
 182     ServerProcs->ProcessInheritProc = Win32CsrDuplicateHandleTable;
 183     ServerProcs->ProcessDeletedProc = Win32CsrReleaseConsole;
 184
 185     Status = RtlInitializeCriticalSection(&Win32CsrDefineDosDeviceCritSec);
 186     InitializeListHead(&DosDeviceHistory);
 187     return TRUE;
 188 }
 189
 190 /* EOF */