#include <windef.h>
#include <winbase.h>
#include <shimlib.h>
#include <strsafe.h>
#include <setup_shim.inl>
#include <implement_shim.inl>

Include dependency graph for ignorefreelib.c:

Go to the source code of this file.

Macros
#define	WIN32_NO_STATUS

#define	SHIM_NS IgnoreFreeLibrary

#define	SHIM_NOTIFY_FN SHIM_OBJ_NAME(Notify)

#define	SHIM_NUM_HOOKS 1

#define	SHIM_SETUP_HOOKS SHIM_HOOK(0, "KERNEL32.DLL", "FreeLibrary", SHIM_OBJ_NAME(FreeLibrary))

Typedefs
typedef BOOL(WINAPI *	FREELIBRARYPROC) (HMODULE)

Functions
BOOL WINAPI SHIM_OBJ_NAME()	FreeLibrary (HMODULE hModule)

static VOID	InitIgnoreFreeLibrary (PCSTR CommandLine)

BOOL WINAPI SHIM_OBJ_NAME()	Notify (DWORD fdwReason, PVOID ptr)

Variables
const char **	g_Names

static int	g_NameCount

Macro Definition Documentation

◆ SHIM_NOTIFY_FN

#define SHIM_NOTIFY_FN SHIM_OBJ_NAME(Notify)

Definition at line 139 of file ignorefreelib.c.

◆ SHIM_NS

#define SHIM_NS IgnoreFreeLibrary

Definition at line 15 of file ignorefreelib.c.

◆ SHIM_NUM_HOOKS

#define SHIM_NUM_HOOKS 1

Definition at line 140 of file ignorefreelib.c.

◆ SHIM_SETUP_HOOKS

#define SHIM_SETUP_HOOKS SHIM_HOOK(0, "KERNEL32.DLL", "FreeLibrary", SHIM_OBJ_NAME(FreeLibrary))

Definition at line 141 of file ignorefreelib.c.

◆ WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 8 of file ignorefreelib.c.

Typedef Documentation

◆ FREELIBRARYPROC

typedef BOOL(WINAPI * FREELIBRARYPROC) (HMODULE)

Definition at line 18 of file ignorefreelib.c.

Function Documentation

◆ FreeLibrary()

BOOL WINAPI SHIM_OBJ_NAME() FreeLibrary ( HMODULE hModule )

Definition at line 23 of file ignorefreelib.c.

{
    char Buffer[MAX_PATH], *Ptr = Buffer;
    DWORD len, wanted = ARRAYSIZE(Buffer);
    for (;;)
    {
        len = GetModuleFileNameA(hModule, Ptr, wanted);
        if (len < wanted)
            break;
 
        wanted *= 2;
        if (Ptr != Buffer)
            ShimLib_ShimFree(Ptr);
 
        Ptr = ShimLib_ShimMalloc(wanted);
        if (!Ptr)
            break;
    }
 
    if (Ptr && len)
    {
        char* ModuleName = NULL;
        int n;
        for (; len; len--)
        {
            ModuleName = Ptr + len;
            if (ModuleName[-1] == '/' || ModuleName[-1] == '\\')
                break;
        }
        for (n = 0; n < g_NameCount; ++n)
        {
            if (!stricmp(g_Names[n], ModuleName))
            {
                SHIM_INFO("Prevented unload of %s\n", ModuleName);
                if (Ptr && Ptr != Buffer)
                    ShimLib_ShimFree(Ptr);
                return TRUE;
            }
        }
    }
 
    if (Ptr && Ptr != Buffer)
        ShimLib_ShimFree(Ptr);
 
    return CALL_SHIM(0, FREELIBRARYPROC)(hModule);
}

◆ InitIgnoreFreeLibrary()

static VOID InitIgnoreFreeLibrary ( PCSTR CommandLine )

static

Definition at line 70 of file ignorefreelib.c.

{
    PCSTR prev, cur;
    int count = 1, n = 0;
    const char** names;
 
    if (!CommandLine || !*CommandLine)
        return;
 
    prev = CommandLine;
    while ((cur = strchr(prev, ';')))
    {
        count++;
        prev = cur + 1;
    }
 
    names = ShimLib_ShimMalloc(sizeof(char*) * count);
    if (!names)
    {
        SHIM_WARN("Unable to allocate %u bytes\n", sizeof(char*) * count);
        return;
    }
 
    prev = CommandLine;
    while ((cur = strchr(prev, ';')))
    {
        names[n] = ShimLib_StringNDuplicateA(prev, cur - prev + 1);
        if (!names[n])
        {
            SHIM_WARN("Unable to allocate %u bytes\n", cur - prev + 2);
            goto fail;
        }
        n++;
        prev = cur + 1;
    }
    names[n] = ShimLib_StringDuplicateA(prev);
    if (!names[n])
    {
        SHIM_WARN("Unable to allocate last string\n");
        goto fail;
    }
 
    g_Names = names;
    g_NameCount = count;
    return;
 
fail:
    --n;
    while (n >= 0)
    {
        if (names[n])
            ShimLib_ShimFree((PVOID)names[n]);
 
        --n;
    }
    ShimLib_ShimFree((PVOID)names);
}

Referenced by Notify().

◆ Notify()

BOOL WINAPI SHIM_OBJ_NAME() Notify	(	DWORD	fdwReason,
		PVOID	ptr
	)

Definition at line 128 of file ignorefreelib.c.

{
    if (fdwReason == SHIM_NOTIFY_ATTACH)
    {
        SHIM_MSG("IgnoreFreeLibrary(%s)\n", SHIM_OBJ_NAME(g_szCommandLine));
        InitIgnoreFreeLibrary(SHIM_OBJ_NAME(g_szCommandLine));
    }
    return TRUE;
}

Referenced by AcpiEvAsynchExecuteGpeMethod(), AcpiEvDeleteGpeHandlers(), AcpiSetupGpeForWake(), DECLARE_INTERFACE_(), DoNotifyPositionEvents(), IDirectSoundNotify_fnSetNotificationPositions(), NewDirectSoundNotify(), NotifyGPEvents(), RegisterGPNotification(), ShimLib_NotifyShims(), and UnregisterGPNotification().

Variable Documentation

◆ g_NameCount

int g_NameCount

static

Definition at line 21 of file ignorefreelib.c.

Referenced by FreeLibrary(), and InitIgnoreFreeLibrary().

◆ g_Names

const char** g_Names

Definition at line 20 of file ignorefreelib.c.