ignorefreelib.c File Reference

#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);
}

Referenced by START_TEST(), and test_IgnoreFreeLibrary().

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(), ShellBrowserChildT< MainFrameBase >::Notify(), 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.

Referenced by FreeLibrary(), and InitIgnoreFreeLibrary().