shimlib.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	tagHOOKAPI
struct	tagSHIMREG

Macros
#define	SHIM_REASON_INIT   100
#define	SHIM_REASON_DEINIT   101
#define	SHIM_REASON_DLL_LOAD   102 /* Arg: PLDR_DATA_TABLE_ENTRY */
#define	SHIM_REASON_DLL_UNLOAD   103 /* Arg: PLDR_DATA_TABLE_ENTRY */
#define	SHIM_NOTIFY_ATTACH   1
#define	SHIM_NOTIFY_DETACH   2
#define	SHIM_NOTIFY_DLL_LOAD   3 /* Arg: PLDR_DATA_TABLE_ENTRY */
#define	SHIM_NOTIFY_DLL_UNLOAD   4 /* Arg: PLDR_DATA_TABLE_ENTRY */
#define	SHIM_MSG(fmt, ...)   do { if (g_ShimEngDebugLevel) SeiDbgPrint(SEI_MSG, SHIM_OBJ_NAME(g_szModuleName), fmt, ##__VA_ARGS__ ); } while (0)
#define	SHIM_FAIL(fmt, ...)   do { if (g_ShimEngDebugLevel) SeiDbgPrint(SEI_FAIL, SHIM_OBJ_NAME(g_szModuleName), fmt, ##__VA_ARGS__ ); } while (0)
#define	SHIM_WARN(fmt, ...)   do { if (g_ShimEngDebugLevel) SeiDbgPrint(SEI_WARN, SHIM_OBJ_NAME(g_szModuleName), fmt, ##__VA_ARGS__ ); } while (0)
#define	SHIM_INFO(fmt, ...)   do { if (g_ShimEngDebugLevel) SeiDbgPrint(SEI_INFO, SHIM_OBJ_NAME(g_szModuleName), fmt, ##__VA_ARGS__ ); } while (0)

Typedefs
typedef struct tagHOOKAPI	HOOKAPI
typedef struct tagHOOKAPI *	PHOOKAPI
typedef enum _SEI_LOG_LEVEL	SEI_LOG_LEVEL
typedef PHOOKAPI(WINAPI *	_PVGetHookAPIs) (DWORD, PCSTR, PDWORD)
typedef BOOL(WINAPI *	_PVNotify) (DWORD, PVOID)
typedef struct tagSHIMREG	SHIMREG
typedef struct tagSHIMREG *	PSHIMREG

Enumerations
enum	_SEI_LOG_LEVEL { SEI_MSG = 1 , SEI_FAIL = 2 , SEI_WARN = 3 , SEI_INFO = 4 }

Functions
PVOID	ShimLib_ShimMalloc (SIZE_T dwSize)
VOID	ShimLib_ShimFree (PVOID pData)
PCSTR	ShimLib_StringDuplicateA (PCSTR szString)
PCSTR	ShimLib_StringNDuplicateA (PCSTR szString, SIZE_T stringLength)
BOOL	ShimLib_StrAEqualsWNC (PCSTR szString, PCWSTR wszString)
HINSTANCE	ShimLib_Instance (VOID)
VOID	ShimLib_Init (HINSTANCE hInstance)
VOID	ShimLib_Deinit (VOID)
PHOOKAPI WINAPI	ShimLib_GetHookAPIs (LPCSTR szCommandLine, LPCWSTR wszShimName, PDWORD pdwHookCount)
BOOL WINAPI	ShimLib_NotifyShims (DWORD fdwReason, PVOID ptr)
BOOL WINAPIV	SeiDbgPrint (SEI_LOG_LEVEL Level, PCSTR Function, PCSTR Format,...)

Variables
ULONG	g_ShimEngDebugLevel

Macro Definition Documentation

SHIM_FAIL

#define SHIM_FAIL	(		fmt,
			...
	)		do { if (g_ShimEngDebugLevel) SeiDbgPrint(SEI_FAIL, SHIM_OBJ_NAME(g_szModuleName), fmt, ##__VA_ARGS__ ); } while (0)

Definition at line 76 of file shimlib.h.

SHIM_INFO

#define SHIM_INFO	(		fmt,
			...
	)		do { if (g_ShimEngDebugLevel) SeiDbgPrint(SEI_INFO, SHIM_OBJ_NAME(g_szModuleName), fmt, ##__VA_ARGS__ ); } while (0)

Definition at line 78 of file shimlib.h.

SHIM_MSG

#define SHIM_MSG	(		fmt,
			...
	)		do { if (g_ShimEngDebugLevel) SeiDbgPrint(SEI_MSG, SHIM_OBJ_NAME(g_szModuleName), fmt, ##__VA_ARGS__ ); } while (0)

Definition at line 75 of file shimlib.h.

SHIM_NOTIFY_ATTACH

#define SHIM_NOTIFY_ATTACH   1

Definition at line 50 of file shimlib.h.

SHIM_NOTIFY_DETACH

#define SHIM_NOTIFY_DETACH   2

Definition at line 51 of file shimlib.h.

SHIM_NOTIFY_DLL_LOAD

#define SHIM_NOTIFY_DLL_LOAD   3 /* Arg: PLDR_DATA_TABLE_ENTRY */

Definition at line 52 of file shimlib.h.

SHIM_NOTIFY_DLL_UNLOAD

#define SHIM_NOTIFY_DLL_UNLOAD   4 /* Arg: PLDR_DATA_TABLE_ENTRY */

Definition at line 53 of file shimlib.h.

SHIM_REASON_DEINIT

#define SHIM_REASON_DEINIT   101

Definition at line 46 of file shimlib.h.

SHIM_REASON_DLL_LOAD

#define SHIM_REASON_DLL_LOAD   102 /* Arg: PLDR_DATA_TABLE_ENTRY */

Definition at line 47 of file shimlib.h.

SHIM_REASON_DLL_UNLOAD

#define SHIM_REASON_DLL_UNLOAD   103 /* Arg: PLDR_DATA_TABLE_ENTRY */

Definition at line 48 of file shimlib.h.

SHIM_REASON_INIT

#define SHIM_REASON_INIT   100

Definition at line 45 of file shimlib.h.

SHIM_WARN

#define SHIM_WARN	(		fmt,
			...
	)		do { if (g_ShimEngDebugLevel) SeiDbgPrint(SEI_WARN, SHIM_OBJ_NAME(g_szModuleName), fmt, ##__VA_ARGS__ ); } while (0)

Definition at line 77 of file shimlib.h.

Typedef Documentation

_PVGetHookAPIs

typedef PHOOKAPI(WINAPI * _PVGetHookAPIs) (DWORD, PCSTR, PDWORD)

Definition at line 81 of file shimlib.h.

_PVNotify

typedef BOOL(WINAPI * _PVNotify) (DWORD, PVOID)

Definition at line 82 of file shimlib.h.

HOOKAPI

typedef struct tagHOOKAPI HOOKAPI

PHOOKAPI

typedef struct tagHOOKAPI * PHOOKAPI

PSHIMREG

typedef struct tagSHIMREG * PSHIMREG

SEI_LOG_LEVEL

typedef enum _SEI_LOG_LEVEL SEI_LOG_LEVEL

SHIMREG

typedef struct tagSHIMREG SHIMREG

Enumeration Type Documentation

_SEI_LOG_LEVEL

enum _SEI_LOG_LEVEL

Enumerator
SEI_MSG
SEI_FAIL
SEI_WARN
SEI_INFO

Definition at line 57 of file shimlib.h.

                            {
    SEI_MSG = 1,
    SEI_FAIL = 2,
    SEI_WARN = 3,
    SEI_INFO = 4,
} SEI_LOG_LEVEL;

Function Documentation

SeiDbgPrint()

BOOL WINAPIV SeiDbgPrint	(	SEI_LOG_LEVEL	Level,
		PCSTR	Function,
		PCSTR	Format,
			...
	)

Outputs diagnostic info.

Parameters

[in]	Level	The level to log this message with, choose any of [SHIM_ERR, SHIM_WARN, SHIM_INFO].
[in]	FunctionName	The function this log should be attributed to.
[in]	Format	The format string.
	...	Variable arguments providing additional information.

Returns

Success: TRUE Failure: FALSE.

Definition at line 160 of file shimeng.c.

{
    char Buffer[512];
    char* Current = Buffer;
    const char* LevelStr;
    size_t Length = sizeof(Buffer);
    va_list ArgList;
    HRESULT hr;
 
    if (g_ShimEngDebugLevel == 0xffffffff)
        SeiInitDebugSupport();
 
    if (Level > g_ShimEngDebugLevel)
        return FALSE;
 
    switch (Level)
    {
    case SEI_MSG:
        LevelStr = "MSG ";
        break;
    case SEI_FAIL:
        LevelStr = "FAIL";
        break;
    case SEI_WARN:
        LevelStr = "WARN";
        break;
    case SEI_INFO:
        LevelStr = "INFO";
        break;
    default:
        LevelStr = "USER";
        break;
    }
 
    if (Function)
        hr = StringCchPrintfExA(Current, Length, &Current, &Length, STRSAFE_NULL_ON_FAILURE, "[%s] [%s] ", LevelStr, Function);
    else
        hr = StringCchPrintfExA(Current, Length, &Current, &Length, STRSAFE_NULL_ON_FAILURE, "[%s] ", LevelStr);
 
    if (!SUCCEEDED(hr))
        return FALSE;
 
    va_start(ArgList, Format);
    hr = StringCchVPrintfExA(Current, Length, &Current, &Length, STRSAFE_NULL_ON_FAILURE, Format, ArgList);
    va_end(ArgList);
    if (!SUCCEEDED(hr))
        return FALSE;
 
    DbgPrint("%s", Buffer);
    return TRUE;
}

Referenced by SeiBuildShimRefArray(), and SeiInit().

ShimLib_Deinit()

VOID ShimLib_Deinit

(

VOID

)

Definition at line 39 of file shimlib.c.

{
    // Is this a good idea?
    HeapDestroy(g_ShimLib_Heap);
}

Referenced by DllMain().

ShimLib_GetHookAPIs()

PHOOKAPI WINAPI ShimLib_GetHookAPIs	(	LPCSTR	szCommandLine,
		LPCWSTR	wszShimName,
		PDWORD	pdwHookCount
	)

ShimLib_Init()

VOID ShimLib_Init

(

HINSTANCE

hInstance

)

Definition at line 30 of file shimlib.c.

{
    g_ShimLib_hInstance = hInstance;
    g_ShimLib_Heap = HeapCreate(0, 0x10000, 0);
 
    g_UsedShims = (PSLIST_HEADER)ShimLib_ShimMalloc(sizeof(SLIST_HEADER));
    RtlInitializeSListHead(g_UsedShims);
}

Referenced by DllMain().

ShimLib_Instance()

HINSTANCE ShimLib_Instance

(

VOID

)

Definition at line 55 of file shimlib.c.

{
    return g_ShimLib_hInstance;
}

Referenced by APIHook_FreeLibrary(), APIHook_GetProcAddress(), APIHook_LoadLibraryA(), APIHook_LoadLibraryExA(), APIHook_LoadLibraryExW(), and APIHook_LoadLibraryW().

ShimLib_NotifyShims()

BOOL WINAPI ShimLib_NotifyShims	(	DWORD	fdwReason,
		PVOID	ptr
	)

Definition at line 136 of file shimlib.c.

{
    PSLIST_ENTRY pEntry = RtlFirstEntrySList(g_UsedShims);
 
    if (fdwReason < SHIM_REASON_INIT)
        fdwReason += (SHIM_REASON_INIT - SHIM_NOTIFY_ATTACH);
 
    while (pEntry)
    {
        pUsedShim pUsed = CONTAINING_RECORD(pEntry, UsedShim, Entry);
        _PVNotify Notify = pUsed->pShim->Notify;
#if (WINVER > _WIN32_WINNT_WS03)
        if (pUsed->bInitCalled && fdwReason == SHIM_REASON_INIT)
            Notify = NULL;
#endif
        if (Notify)
            Notify(fdwReason, ptr);
#if (WINVER > _WIN32_WINNT_WS03)
        if (fdwReason == SHIM_REASON_INIT)
            pUsed->bInitCalled = TRUE;
#endif
 
        pEntry = pEntry->Next;
    }
 
    return TRUE;
}

Referenced by NotifyShims().

ShimLib_ShimFree()

VOID ShimLib_ShimFree

(

PVOID

pData

)

Definition at line 50 of file shimlib.c.

{
    HeapFree(g_ShimLib_Heap, 0, pData);
}

Referenced by FreeLibrary(), and InitIgnoreFreeLibrary().

ShimLib_ShimMalloc()

PVOID ShimLib_ShimMalloc

(

SIZE_T

dwSize

)

Definition at line 45 of file shimlib.c.

{
    return HeapAlloc(g_ShimLib_Heap, 0, dwSize);
}

Referenced by FreeLibrary(), InitIgnoreFreeLibrary(), ShimLib_GetHookAPIs(), ShimLib_Init(), and ShimLib_StringNDuplicateA().

ShimLib_StrAEqualsWNC()

BOOL ShimLib_StrAEqualsWNC	(	PCSTR	szString,
		PCWSTR	wszString
	)

Definition at line 72 of file shimlib.c.

{
    while (toupper(*szString) == towupper(*wszString))
    {
        if (!*szString)
            return TRUE;
 
        szString++; wszString++;
    }
    return FALSE;
}

Referenced by ShimLib_GetHookAPIs().

ShimLib_StringDuplicateA()

PCSTR ShimLib_StringDuplicateA

(

PCSTR

szString

)

Definition at line 67 of file shimlib.c.

{
    return ShimLib_StringNDuplicateA(szString, lstrlenA(szString) + 1);
}

Referenced by InitIgnoreFreeLibrary().

ShimLib_StringNDuplicateA()

PCSTR ShimLib_StringNDuplicateA	(	PCSTR	szString,
		SIZE_T	stringLength
	)

Definition at line 60 of file shimlib.c.

{
    PSTR NewString = ShimLib_ShimMalloc(stringLengthIncludingNullTerm);
    StringCchCopyA(NewString, stringLengthIncludingNullTerm, szString);
    return NewString;
}

Referenced by InitIgnoreFreeLibrary(), and ShimLib_StringDuplicateA().

Variable Documentation

g_ShimEngDebugLevel

ULONG g_ShimEngDebugLevel

extern

Definition at line 32 of file shimeng.c.

Referenced by SeiDbgPrint(), and SeiInitDebugSupport().