ldrnotify.c File Reference

#include "ntdll_vista.h"

Include dependency graph for ldrnotify.c:

Go to the source code of this file.

Classes
struct	_LDR_DLL_NOTIFICATION_ENTRY

Macros
#define	LdrpAssertDllNotificationDataMember(x)

Typedefs
typedef struct _LDR_DLL_NOTIFICATION_ENTRY	LDR_DLL_NOTIFICATION_ENTRY
typedef struct _LDR_DLL_NOTIFICATION_ENTRY *	PLDR_DLL_NOTIFICATION_ENTRY

Functions
static	RTL_STATIC_LIST_HEAD (LdrpDllNotificationList)
NTSTATUS NTAPI	LdrRegisterDllNotification (_In_ ULONG Flags, _In_ PLDR_DLL_NOTIFICATION_FUNCTION NotificationFunction, _In_opt_ PVOID Context, _Out_ PVOID *Cookie)
NTSTATUS NTAPI	LdrUnregisterDllNotification (_In_ PVOID Cookie)
VOID NTAPI	LdrpSendDllNotifications (_In_ PLDR_DATA_TABLE_ENTRY DllEntry, _In_ ULONG NotificationReason)

Variables
static RTL_CRITICAL_SECTION	LdrpDllNotificationLock
static RTL_CRITICAL_SECTION_DEBUG	LdrpDllNotificationLockDebug

Macro Definition Documentation

LdrpAssertDllNotificationDataMember

#define LdrpAssertDllNotificationDataMember

(

x

)

Value:

    C_ASSERT(FIELD_OFFSET(LDR_DLL_NOTIFICATION_DATA, Loaded.x) ==\
             FIELD_OFFSET(LDR_DLL_NOTIFICATION_DATA, Unloaded.x))

Typedef Documentation

LDR_DLL_NOTIFICATION_ENTRY

typedef struct _LDR_DLL_NOTIFICATION_ENTRY LDR_DLL_NOTIFICATION_ENTRY

PLDR_DLL_NOTIFICATION_ENTRY

typedef struct _LDR_DLL_NOTIFICATION_ENTRY * PLDR_DLL_NOTIFICATION_ENTRY

Function Documentation

LdrpSendDllNotifications()

VOID NTAPI LdrpSendDllNotifications	(	_In_ PLDR_DATA_TABLE_ENTRY	DllEntry,
		_In_ ULONG	NotificationReason
	)

Definition at line 104 of file ldrnotify.c.

{
    PLIST_ENTRY Entry;
    PLDR_DLL_NOTIFICATION_ENTRY NotificationEntry;
    LDR_DLL_NOTIFICATION_DATA NotificationData;
 
    /*
     * LDR_DLL_LOADED_NOTIFICATION_DATA and LDR_DLL_UNLOADED_NOTIFICATION_DATA
     * currently are the same. Use C_ASSERT to ensure it, then fill either of them.
     */
#define LdrpAssertDllNotificationDataMember(x)\
    C_ASSERT(FIELD_OFFSET(LDR_DLL_NOTIFICATION_DATA, Loaded.x) ==\
             FIELD_OFFSET(LDR_DLL_NOTIFICATION_DATA, Unloaded.x))
 
    C_ASSERT(sizeof(NotificationData.Loaded) == sizeof(NotificationData.Unloaded));
    LdrpAssertDllNotificationDataMember(Flags);
    LdrpAssertDllNotificationDataMember(FullDllName);
    LdrpAssertDllNotificationDataMember(BaseDllName);
    LdrpAssertDllNotificationDataMember(DllBase);
    LdrpAssertDllNotificationDataMember(SizeOfImage);
 
#undef LdrpAssertDllNotificationDataMember
 
    NotificationData.Loaded.Flags = 0; /* Reserved and always 0, not DllEntry->Flags */
    NotificationData.Loaded.FullDllName = &DllEntry->FullDllName;
    NotificationData.Loaded.BaseDllName = &DllEntry->BaseDllName;
    NotificationData.Loaded.DllBase = DllEntry->DllBase;
    NotificationData.Loaded.SizeOfImage = DllEntry->SizeOfImage;
 
    /* Send notification to all registered callbacks */
    RtlEnterCriticalSection(&LdrpDllNotificationLock);
    _SEH2_TRY
    {
        for (Entry = LdrpDllNotificationList.Flink;
             Entry != &LdrpDllNotificationList;
             Entry = Entry->Flink)
        {
            NotificationEntry = CONTAINING_RECORD(Entry, LDR_DLL_NOTIFICATION_ENTRY, List);
            NotificationEntry->Callback(NotificationReason,
                                        &NotificationData,
                                        NotificationEntry->Context);
        }
    }
    _SEH2_FINALLY
    {
        RtlLeaveCriticalSection(&LdrpDllNotificationLock);
    }
    _SEH2_END;
}

Referenced by LdrpMapDll(), and LdrUnloadDll().

LdrRegisterDllNotification()

NTSTATUS NTAPI LdrRegisterDllNotification	(	_In_ ULONG	Flags,
		_In_ PLDR_DLL_NOTIFICATION_FUNCTION	NotificationFunction,
		_In_opt_ PVOID	Context,
		_Out_ PVOID *	Cookie
	)

Definition at line 39 of file ldrnotify.c.

{
    PLDR_DLL_NOTIFICATION_ENTRY NewEntry;
 
    /* Check input parameters */
    if (Flags != 0 || NotificationFunction == NULL || Cookie == NULL)
    {
        return STATUS_INVALID_PARAMETER;
    }
 
    /* Allocate new entry and assign input values */
    NewEntry = RtlAllocateHeap(LdrpHeap, 0, sizeof(*NewEntry));
    if (NewEntry == NULL)
    {
        return STATUS_NO_MEMORY;
    }
    NewEntry->Callback = NotificationFunction;
    NewEntry->Context = Context;
 
    /* Add node to the end of global list */
    RtlEnterCriticalSection(&LdrpDllNotificationLock);
    InsertTailList(&LdrpDllNotificationList, &NewEntry->List);
    RtlLeaveCriticalSection(&LdrpDllNotificationLock);
 
    /* Cookie is address of the new entry */
    *Cookie = NewEntry;
    return STATUS_SUCCESS;
}

LdrUnregisterDllNotification()

NTSTATUS NTAPI LdrUnregisterDllNotification

(

_In_ PVOID

Cookie

)

Definition at line 74 of file ldrnotify.c.

{
    NTSTATUS Status = STATUS_DLL_NOT_FOUND;
    PLIST_ENTRY Entry;
 
    /* Find entry to remove */
    RtlEnterCriticalSection(&LdrpDllNotificationLock);
    for (Entry = LdrpDllNotificationList.Flink;
         Entry != &LdrpDllNotificationList;
         Entry = Entry->Flink)
    {
        if (Entry == Cookie)
        {
            RemoveEntryList(Entry);
            Status = STATUS_SUCCESS;
            break;
        }
    }
    RtlLeaveCriticalSection(&LdrpDllNotificationLock);
 
    if (NT_SUCCESS(Status))
    {
        RtlFreeHeap(LdrpHeap, 0, Entry);
    }
    return Status;
}

RTL_STATIC_LIST_HEAD()

static RTL_STATIC_LIST_HEAD ( LdrpDllNotificationList  )

static

Variable Documentation

LdrpDllNotificationLock

static RTL_CRITICAL_SECTION LdrpDllNotificationLock

static

Initial value:

= {
    &LdrpDllNotificationLockDebug,
    -1,
    0,
    0,
    0,
    0
}

Definition at line 22 of file ldrnotify.c.

Referenced by LdrpSendDllNotifications(), LdrRegisterDllNotification(), and LdrUnregisterDllNotification().

LdrpDllNotificationLockDebug

RTL_CRITICAL_SECTION_DEBUG LdrpDllNotificationLockDebug

static

Initial value:

= {
    .CriticalSection = &LdrpDllNotificationLock
}

Definition at line 23 of file ldrnotify.c.