ignorefreelib.c File Reference

#include <ntstatus.h>
#include <windows.h>
#include "wine/test.h"
#include "appshim_apitest.h"

Include dependency graph for ignorefreelib.c:

Go to the source code of this file.

Macros
#define	WIN32_NO_STATUS

Typedefs
typedef BOOL(WINAPI *	FREELIBRARYPROC) (HMODULE)

Functions
static void	test_IgnoreFreeLibrary (VOID)
	START_TEST (ignorefreelib)

Variables
static tGETHOOKAPIS	pGetHookAPIs

Macro Definition Documentation

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 9 of file ignorefreelib.c.

Typedef Documentation

FREELIBRARYPROC

typedef BOOL(WINAPI * FREELIBRARYPROC) (HMODULE)

Definition at line 17 of file ignorefreelib.c.

Function Documentation

START_TEST()

START_TEST

(

ignorefreelib

)

Definition at line 77 of file ignorefreelib.c.

{
    HMODULE avifil32, esent, msi;
    LONG initial;
 
    pGetHookAPIs = LoadShimDLL2(L"acgenral.dll");
 
    if (!pGetHookAPIs)
        return;
 
    /* Ensure that we can freely load / unload avifil32, esent and msi */
 
    initial = winetest_get_failures();
 
    avifil32 = GetModuleHandleA("avifil32.dll");
    ok_ptr(avifil32, NULL);
    esent = GetModuleHandleA("esent.dll");
    ok_ptr(esent, NULL);
    msi = GetModuleHandleA("msi.dll");
    ok_ptr(msi, NULL);
 
    avifil32 = LoadLibraryA("avifil32.dll");
    ok(avifil32 != NULL, "Unable to load avifil32\n");
    esent = GetModuleHandleA("esent.dll");
    ok_ptr(esent, NULL);
    msi = GetModuleHandleA("msi.dll");
    ok_ptr(msi, NULL);
 
    ok(FreeLibrary(avifil32), "Unable to unload avifil32\n");
    avifil32 = GetModuleHandleA("avifil32.dll");
    ok_ptr(avifil32, NULL);
 
    esent = LoadLibraryA("esent.dll");
    ok(esent != NULL, "Unable to load esent\n");
    avifil32 = GetModuleHandleA("avifil32.dll");
    ok_ptr(avifil32, NULL);
    msi = GetModuleHandleA("msi.dll");
    ok_ptr(msi, NULL);
 
    ok(FreeLibrary(esent), "Unable to unload esent\n");
    esent = GetModuleHandleA("esent.dll");
    ok_ptr(esent, NULL);
 
    msi = LoadLibraryA("msi.dll");
    ok(msi != NULL, "Unable to load msi\n");
    avifil32 = GetModuleHandleA("avifil32.dll");
    ok_ptr(avifil32, NULL);
    esent = GetModuleHandleA("esent.dll");
    ok_ptr(esent, NULL);
 
    ok(FreeLibrary(msi), "Unable to unload msi\n");
    msi = GetModuleHandleA("msi.dll");
    ok_ptr(msi, NULL);
 
    if (initial != winetest_get_failures())
    {
        // We cannot continue, one or more of our target dll's does not behave as expected
        return;
    }
 
    test_IgnoreFreeLibrary();
}

test_IgnoreFreeLibrary()

static void test_IgnoreFreeLibrary ( VOID  )

static

Definition at line 20 of file ignorefreelib.c.

{
    DWORD num_shims = 0;
    PHOOKAPI hook;
    HMODULE avifil32, esent, msi;
    FREELIBRARYPROC freeProc;
 
    // Show that:
    // * partial names (both start + end) do not work.
    // * wildcards do not work
    // * comparison is case insensitive
    // * multiple dlls can be specified
    hook = pGetHookAPIs("AvIFiL32.Dll;esent;esent*;ent.dll;msi.dll", L"IgnoreFreeLibrary", &num_shims);
 
    ok(hook != NULL, "Expected hook to be a valid pointer\n");
    ok(num_shims == 1, "Expected num_shims to be 1, was: %u\n", num_shims);
 
    if (!hook || !num_shims)
        return;
 
    ok_str(hook->LibraryName, "KERNEL32.DLL");
    ok_str(hook->FunctionName, "FreeLibrary");
 
    avifil32 = LoadLibraryA("avifil32.dll");
    ok(avifil32 != NULL, "Unable to load avifil32\n");
 
    esent = LoadLibraryA("esent.dll");
    ok(esent != NULL, "Unable to load esent\n");
 
    msi = LoadLibraryA("msi.dll");
    ok(msi != NULL, "Unable to load msi\n");
 
    hook->OriginalFunction = FreeLibrary;
    freeProc = hook->ReplacementFunction;
 
    ok(freeProc != NULL, "\n");
 
    if (!freeProc)
        return;
 
    // Not unloading
    ok(freeProc(avifil32), "Unable to unload avifil32\n");
    avifil32 = GetModuleHandleA("avifil32.dll");
    ok(avifil32 != NULL, "avifil32 should not unload\n");
 
    // Unloading
    ok(freeProc(esent), "Unable to unload esent\n");
    esent = GetModuleHandleA("esent.dll");
    ok(esent == NULL, "esent should be unloaded\n");
 
    // Not unloading
    ok(freeProc(msi), "Unable to unload msi\n");
    msi = GetModuleHandleA("msi.dll");
    ok(msi != NULL, "msi should not unload\n");
}

Referenced by START_TEST().

Variable Documentation

pGetHookAPIs

tGETHOOKAPIS pGetHookAPIs

static

Definition at line 15 of file ignorefreelib.c.

Referenced by START_TEST(), and test_IgnoreFreeLibrary().