dllmain.c File Reference

#include <k32.h>
#include <debug.h>

Include dependency graph for dllmain.c:

Go to the source code of this file.

Macros
#define	NDEBUG
#define	WIN_OBJ_DIR   L"\\Windows"
#define	SESSION_DIR   L"\\Sessions"

Functions
BOOL FASTCALL	NlsInit (VOID)
VOID FASTCALL	NlsUninit (VOID)
NTSTATUS NTAPI	BaseCreateThreadPoolThread (IN PTHREAD_START_ROUTINE Function, IN PVOID Parameter, OUT PHANDLE ThreadHandle)
NTSTATUS NTAPI	BaseExitThreadPoolThread (IN NTSTATUS ExitStatus)
BOOL WINAPI	DllMain (HANDLE hDll, DWORD dwReason, LPVOID lpReserved)

Variables
PBASE_STATIC_SERVER_DATA	BaseStaticServerData
BOOLEAN	BaseRunningInServerProcess = FALSE
WCHAR	BaseDefaultPathBuffer [6140]
HANDLE	BaseNamedObjectDirectory
HMODULE	hCurrentModule = NULL
HMODULE	kernel32_handle = NULL
PPEB	Peb
ULONG	SessionId
static BOOL	DllInitialized = FALSE
RTL_CRITICAL_SECTION	BaseDllDirectoryLock

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 14 of file dllmain.c.

SESSION_DIR

#define SESSION_DIR   L"\\Sessions"

Definition at line 38 of file dllmain.c.

WIN_OBJ_DIR

#define WIN_OBJ_DIR   L"\\Windows"

Definition at line 37 of file dllmain.c.

Function Documentation

BaseCreateThreadPoolThread()

NTSTATUS NTAPI BaseCreateThreadPoolThread	(	IN PTHREAD_START_ROUTINE	Function,
		IN PVOID	Parameter,
		OUT PHANDLE	ThreadHandle
	)

Definition at line 44 of file dllmain.c.

{
    NTSTATUS Status;
 
    /* Create a Win32 thread */
    *ThreadHandle = CreateRemoteThread(NtCurrentProcess(),
                                       NULL,
                                       0,
                                       Function,
                                       Parameter,
                                       CREATE_SUSPENDED,
                                       NULL);
    if (!(*ThreadHandle))
    {
        /* Get the status value if we couldn't get a handle */
        Status = NtCurrentTeb()->LastStatusValue;
        if (NT_SUCCESS(Status)) Status = STATUS_UNSUCCESSFUL;
    }
    else
    {
        /* Set success code */
        Status = STATUS_SUCCESS;
    }
 
    /* All done */
    return Status;
}

Referenced by DllMain().

BaseExitThreadPoolThread()

NTSTATUS NTAPI BaseExitThreadPoolThread

(

IN NTSTATUS

ExitStatus

)

Definition at line 76 of file dllmain.c.

{
    /* Exit the thread */
    ExitThread(ExitStatus);
    return STATUS_SUCCESS;
}

Referenced by DllMain().

DllMain()

BOOL WINAPI DllMain	(	HANDLE	hDll,
		DWORD	dwReason,
		LPVOID	lpReserved
	)

Definition at line 85 of file dllmain.c.

{
    NTSTATUS Status;
    BASESRV_API_CONNECTINFO ConnectInfo;
    ULONG ConnectInfoSize = sizeof(ConnectInfo);
    WCHAR SessionDir[256];
 
    DPRINT("DllMain(hInst %p, dwReason %lu)\n",
           hDll, dwReason);
 
    Basep8BitStringToUnicodeString = RtlAnsiStringToUnicodeString;
 
    /* Cache the PEB and Session ID */
    Peb = NtCurrentPeb();
    SessionId = Peb->SessionId;
 
    switch (dwReason)
    {
        case DLL_PROCESS_ATTACH:
        {
            /* Set no filter initially */
            GlobalTopLevelExceptionFilter = RtlEncodePointer(NULL);
 
            /* Enable the Rtl thread pool and timer queue to use proper Win32 thread */
            RtlSetThreadPoolStartFunc(BaseCreateThreadPoolThread, BaseExitThreadPoolThread);
 
            /* Register the manifest prober routine */
            LdrSetDllManifestProber(BasepProbeForDllManifest);
 
            /* Don't bother us for each thread */
            LdrDisableThreadCalloutsForDll((PVOID)hDll);
 
            /* Initialize default path to NULL */
            RtlInitUnicodeString(&BaseDefaultPath, NULL);
 
            /* Setup the Object Directory path */
            if (!SessionId)
            {
                /* Use the raw path */
                wcscpy(SessionDir, WIN_OBJ_DIR);
            }
            else
            {
                /* Use the session path */
                swprintf(SessionDir,
                         L"%ws\\%ld%ws",
                         SESSION_DIR,
                         SessionId,
                         WIN_OBJ_DIR);
            }
 
            /* Connect to the Base Server */
            Status = CsrClientConnectToServer(SessionDir,
                                              BASESRV_SERVERDLL_INDEX,
                                              &ConnectInfo,
                                              &ConnectInfoSize,
                                              &BaseRunningInServerProcess);
            if (!NT_SUCCESS(Status))
            {
                DPRINT1("Failed to connect to CSR (Status %lx)\n", Status);
                NtTerminateProcess(NtCurrentProcess(), Status);
                return FALSE;
            }
 
            /* Get the server data */
            ASSERT(Peb->ReadOnlyStaticServerData);
            BaseStaticServerData = Peb->ReadOnlyStaticServerData[BASESRV_SERVERDLL_INDEX];
            ASSERT(BaseStaticServerData);
 
            /* Check if we are running a CSR Server */
            if (!BaseRunningInServerProcess)
            {
                /* Set the termination port for the thread */
                DPRINT("Creating new thread for CSR\n");
                CsrNewThread();
            }
 
            /* Initialize heap handle table */
            BaseDllInitializeMemoryManager();
 
            /* Set HMODULE for our DLL */
            kernel32_handle = hCurrentModule = hDll;
 
            /* Set the directories */
            BaseWindowsDirectory = BaseStaticServerData->WindowsDirectory;
            BaseWindowsSystemDirectory = BaseStaticServerData->WindowsSystemDirectory;
 
            /* Construct the default path (using the static buffer) */
            Status = RtlStringCbPrintfW(BaseDefaultPathBuffer,
                                     sizeof(BaseDefaultPathBuffer),
                                     L"%wZ;%wZ\\system;%wZ;",
                                     &BaseWindowsSystemDirectory,
                                     &BaseWindowsDirectory,
                                     &BaseWindowsDirectory);
            if (!NT_SUCCESS(Status))
            {
                DPRINT1("NLS Init failed\n");
                return FALSE;
            }
 
            BaseDefaultPath.Buffer = BaseDefaultPathBuffer;
            BaseDefaultPath.Length = (USHORT)wcslen(BaseDefaultPathBuffer) * sizeof(WCHAR);
            BaseDefaultPath.MaximumLength = sizeof(BaseDefaultPathBuffer);
 
            /* Use remaining part of the default path buffer for the append path */
            BaseDefaultPathAppend.Buffer = (PWSTR)((ULONG_PTR)BaseDefaultPathBuffer + BaseDefaultPath.Length);
            BaseDefaultPathAppend.Length = 0;
            BaseDefaultPathAppend.MaximumLength = BaseDefaultPath.MaximumLength - BaseDefaultPath.Length;
 
            /* Initialize command line */
            InitCommandLines();
 
            /* Initialize the DLL critical section */
            RtlInitializeCriticalSection(&BaseDllDirectoryLock);
 
            /* Initialize the National Language Support routines */
            if (!NlsInit())
            {
                DPRINT1("NLS Init failed\n");
                return FALSE;
            }
 
            /* Initialize Console Support */
            if (!ConDllInitialize(dwReason, SessionDir))
            {
                DPRINT1("Failed to set up console\n");
                return FALSE;
            }
 
            /* Initialize application certification globals */
            InitializeListHead(&BasepAppCertDllsList);
            RtlInitializeCriticalSection(&gcsAppCert);
 
            /* Insert more dll attach stuff here! */
            DllInitialized = TRUE;
            break;
        }
 
        case DLL_PROCESS_DETACH:
        {
            if (DllInitialized != FALSE)
            {
                /* Uninitialize console support */
                ConDllInitialize(dwReason, NULL);
 
                /* Insert more dll detach stuff here! */
                NlsUninit();
 
                /* Delete DLL critical section */
                RtlDeleteCriticalSection(&BaseDllDirectoryLock);
            }
            break;
        }
 
        case DLL_THREAD_ATTACH:
        {
            /* ConDllInitialize sets the current console locale for the new thread */
            return ConDllInitialize(dwReason, NULL);
        }
 
        default:
            break;
    }
 
    return TRUE;
}

NlsInit()

BOOL FASTCALL NlsInit

(

VOID

)

Definition at line 151 of file nls.c.

{
    NTSTATUS Status;
    UNICODE_STRING DirName;
    OBJECT_ATTRIBUTES ObjectAttributes;
    HANDLE Handle;
    UCHAR SecurityDescriptor[NLS_SECTION_SECURITY_DESCRIPTOR_SIZE +
                             NLS_SIZEOF_ACE_AND_SIDS(2)];
 
    InitializeListHead(&CodePageListHead);
    RtlInitializeCriticalSection(&CodePageListLock);
 
    /*
     * FIXME: Eventually this should be done only for the NLS Server
     * process, but since we don't have anything like that (yet?) we
     * always try to create the "\NLS" directory here.
     */
    RtlInitUnicodeString(&DirName, L"\\NLS");
 
    /* Create a security descriptor for the NLS directory */
    Status = CreateNlsDirectorySecurity(&SecurityDescriptor,
                                        sizeof(SecurityDescriptor));
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("Failed to create NLS directory security (Status 0x%08x)\n", Status);
        return FALSE;
    }
 
    InitializeObjectAttributes(&ObjectAttributes,
                               &DirName,
                               OBJ_CASE_INSENSITIVE | OBJ_PERMANENT,
                               NULL,
                               &SecurityDescriptor);
 
    Status = NtCreateDirectoryObject(&Handle,
                                     DIRECTORY_TRAVERSE | DIRECTORY_CREATE_OBJECT,
                                     &ObjectAttributes);
    if (NT_SUCCESS(Status))
    {
        NtClose(Handle);
    }
 
    /* Setup ANSI code page. */
    AnsiCodePage.SectionHandle = NULL;
    AnsiCodePage.SectionMapping = NtCurrentTeb()->ProcessEnvironmentBlock->AnsiCodePageData;
 
    RtlInitCodePageTable((PUSHORT)AnsiCodePage.SectionMapping,
                         &AnsiCodePage.CodePageTable);
    AnsiCodePage.CodePage = AnsiCodePage.CodePageTable.CodePage;
 
    InsertTailList(&CodePageListHead, &AnsiCodePage.Entry);
 
    /* Setup OEM code page. */
    OemCodePage.SectionHandle = NULL;
    OemCodePage.SectionMapping = NtCurrentTeb()->ProcessEnvironmentBlock->OemCodePageData;
 
    RtlInitCodePageTable((PUSHORT)OemCodePage.SectionMapping,
                         &OemCodePage.CodePageTable);
    OemCodePage.CodePage = OemCodePage.CodePageTable.CodePage;
    InsertTailList(&CodePageListHead, &OemCodePage.Entry);
 
    return TRUE;
}

Referenced by DllMain().

NlsUninit()

VOID FASTCALL NlsUninit

(

VOID

)

Definition at line 223 of file nls.c.

{
    PCODEPAGE_ENTRY Current;
 
    /* Delete the code page list. */
    while (!IsListEmpty(&CodePageListHead))
    {
        Current = CONTAINING_RECORD(CodePageListHead.Flink, CODEPAGE_ENTRY, Entry);
        if (Current->SectionHandle != NULL)
        {
            UnmapViewOfFile(Current->SectionMapping);
            NtClose(Current->SectionHandle);
        }
        RemoveHeadList(&CodePageListHead);
    }
    RtlDeleteCriticalSection(&CodePageListLock);
}

Referenced by DllMain().

Variable Documentation

BaseDefaultPathBuffer

WCHAR BaseDefaultPathBuffer[6140]

Definition at line 22 of file dllmain.c.

Referenced by DllMain().

BaseDllDirectoryLock

RTL_CRITICAL_SECTION BaseDllDirectoryLock

Definition at line 32 of file dllmain.c.

Referenced by BaseComputeProcessDllPath(), DllMain(), GetDllDirectoryA(), GetDllDirectoryW(), SetDllDirectoryA(), and SetDllDirectoryW().

BaseNamedObjectDirectory

HANDLE BaseNamedObjectDirectory

Definition at line 24 of file dllmain.c.

Referenced by BaseGetNamedObjectDirectory().

BaseRunningInServerProcess

BOOLEAN BaseRunningInServerProcess = FALSE

Definition at line 20 of file dllmain.c.

Referenced by BaseProcessStartup(), BaseThreadStartup(), CreateRemoteThread(), DllMain(), ExitProcess(), and UnhandledExceptionFilter().

BaseStaticServerData

PBASE_STATIC_SERVER_DATA BaseStaticServerData

Definition at line 19 of file dllmain.c.

Referenced by AddBSMRequest(), BaseCreateStack(), BaseGetNamedObjectDirectory(), CreateBSMThread(), CreateProcessInternalW(), CSR_API(), DefineDosDeviceW(), DllMain(), FileTimeToLocalFileTime(), GetLocalTime(), GetProcessAffinityMask(), GetSystemTimes(), GlobalMemoryStatusEx(), IsBadReadPtr(), IsBadWritePtr(), IsTimeZoneRedirectionEnabled(), LocalFileTimeToFileTime(), QueryDosDeviceW(), SetLocalTime(), and VirtualAllocEx().

DllInitialized

BOOL DllInitialized = FALSE

static

Definition at line 29 of file dllmain.c.

Referenced by DllMain().

hCurrentModule

HMODULE hCurrentModule = NULL

Definition at line 25 of file dllmain.c.

Referenced by DllMain(), and GetLocalisedText().

kernel32_handle

HMODULE kernel32_handle = NULL

Definition at line 26 of file dllmain.c.

Referenced by DllMain(), EnumSystemLocalesA(), EnumSystemLocalesEx(), EnumSystemLocalesW(), EnumUILanguagesA(), EnumUILanguagesW(), GetLocaleInfoW(), IsValidLocale(), NLS_GetLanguageGroupName(), and search_message().

Peb

PPEB Peb

Definition at line 27 of file dllmain.c.

Referenced by _IRQL_requires_max_(), BasePushProcessParameters(), check_loadconfig(), CreateProcessInternalW(), CsrSrvCreateSharedSection(), DbgkpPostFakeModuleMessages(), DllMain(), FlsAlloc(), FlsFree(), get_base_dir(), GetModuleFileNameW(), GetProcessVersion(), GetVersion(), hash_basename(), InitExeName(), KsecGatherEntropyData(), LdrpHandleOneNewFormatImportDescriptor(), LdrpHandleOneOldFormatImportDescriptor(), LdrpInit(), LdrpInitFailure(), LdrpInitializeApplicationVerifierPackage(), LdrpInitializeExecutionOptions(), LdrpInitializeProcess(), LdrpInitializeProcessCompat(), LdrpInitializeThread(), LdrpLoadDll(), LdrpLoadImportModule(), LdrpMapDll(), LdrpRunInitializeRoutines(), LdrpWalkImportDescriptor(), LdrShutdownProcess(), LdrShutdownThread(), LdrUnloadDll(), MiCheckForUserStackOverflow(), MmCreatePeb(), NtProcessStartup(), RtlAcquirePebLock(), RtlGetProcessHeaps(), RtlGetVersion(), RtlInitializeHeapManager(), RtlpAddHeapToProcessList(), RtlpInitEnvironment(), RtlpRemoveHeapFromProcessList(), RtlpSetHeapParameters(), RtlQueryEnvironmentVariable_U(), RtlReleasePebLock(), SeiInit(), START_TEST(), test_query_object(), test_RemoveDirectoryA(), test_RemoveDirectoryW(), TlsAlloc(), TlsFree(), UserCreateWinstaDirectory(), UserpGetClientFileName(), and wined3d_cs_create().

SessionId

ULONG SessionId

Definition at line 28 of file dllmain.c.

Referenced by _main(), AddPortW(), ConfigurePortW(), CSR_API(), CsrAllocateNtSession(), CsrUnhandledExceptionFilter(), DllMain(), GetNextDosSesId(), MiDereferenceSession(), MiReleaseProcessReferenceToSessionDataPage(), MiSessionCreateInternal(), MmGetSessionById(), MmSessionCreate(), NtQueryInformationToken(), NtSetInformationToken(), PerfDataGetSessionId(), QueryTokenSessionIdTests(), RxFindOrConstructVirtualNetRoot(), RxFirstCanonicalize(), RxGetSessionId(), RxInitializeVNetRootParameters(), SeSubProcessToken(), SetTokenSessionIdTests(), SmpAllocateSessionId(), SmpDeleteSession(), SmpGetProcessMuSessionId(), SmpHandleConnectionRequest(), SmpSbCreateSession(), SmpSessionIdToSession(), SmpSetProcessMuSessionId(), SmSessionComplete(), SSI_DEF(), WTSDisconnectSession(), WTSQuerySessionInformationA(), WTSQuerySessionInformationW(), WTSSendMessageA(), WTSSendMessageW(), WTSVirtualChannelOpen(), and WTSVirtualChannelOpenEx().