main.c File Reference

#include <windef.h>
#include <winbase.h>
#include <ndk/rtlfuncs.h>

Include dependency graph for main.c:

Go to the source code of this file.

Macros
#define	WIN32_NO_STATUS
#define	NTOS_MODE_USER

Functions
NTSTATUS NTAPI	LdrDisableThreadCalloutsForDll (_In_ PVOID BaseAddress)
BOOL NTAPI	DllMain (_In_ HINSTANCE hDll, _In_ ULONG dwReason, _In_opt_ PVOID pReserved)

Variables
HANDLE	ProcessHeap

Macro Definition Documentation

NTOS_MODE_USER

#define NTOS_MODE_USER

Definition at line 15 of file main.c.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 11 of file main.c.

Function Documentation

DllMain()

BOOL NTAPI DllMain	(	_In_ HINSTANCE	hDll,
		_In_ ULONG	dwReason,
		_In_opt_ PVOID	pReserved
	)

Definition at line 33 of file main.c.

{
    UNREFERENCED_PARAMETER(pReserved);
 
    if (dwReason == DLL_PROCESS_ATTACH)
    {
        LdrDisableThreadCalloutsForDll(hDll);
        ProcessHeap = RtlGetProcessHeap();
    }
 
    return TRUE;
}

LdrDisableThreadCalloutsForDll()

NTSTATUS NTAPI LdrDisableThreadCalloutsForDll

(

_In_ PVOID

BaseAddress

)

Definition at line 1154 of file ldrapi.c.

{
    PLDR_DATA_TABLE_ENTRY LdrEntry;
    NTSTATUS Status;
    BOOLEAN LockHeld;
    ULONG_PTR Cookie;
    DPRINT("LdrDisableThreadCalloutsForDll (BaseAddress %p)\n", BaseAddress);
 
    /* Don't do it during shutdown */
    if (LdrpShutdownInProgress) return STATUS_SUCCESS;
 
    /* Check if we should grab the lock */
    LockHeld = FALSE;
    if (!LdrpInLdrInit)
    {
        /* Grab the lock */
        Status = LdrLockLoaderLock(0, NULL, &Cookie);
        if (!NT_SUCCESS(Status)) return Status;
        LockHeld = TRUE;
    }
 
    /* Make sure the DLL is valid and get its entry */
    Status = STATUS_DLL_NOT_FOUND;
    if (LdrpCheckForLoadedDllHandle(BaseAddress, &LdrEntry))
    {
        /* Get if it has a TLS slot */
        if (!LdrEntry->TlsIndex)
        {
            /* It doesn't, so you're allowed to call this */
            LdrEntry->Flags |= LDRP_DONT_CALL_FOR_THREADS;
            Status = STATUS_SUCCESS;
        }
    }
 
    /* Check if the lock was held */
    if (LockHeld)
    {
        /* Release it */
        LdrUnlockLoaderLock(LDR_UNLOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS, Cookie);
    }
 
    /* Return the status */
    return Status;
}

Referenced by DisableThreadLibraryCalls(), and DllMain().

Variable Documentation

ProcessHeap

HANDLE ProcessHeap

extern

Definition at line 15 of file servman.c.

Referenced by DllMain().