forcedxsetupsuccess.c File Reference

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

Include dependency graph for forcedxsetupsuccess.c:

Go to the source code of this file.

Macros
#define	WIN32_NO_STATUS
#define	SHIM_NS   ForceDXSetupSuccess
#define	DSETUPERR_SUCCESS   0
#define	SHIM_NUM_HOOKS   6
#define	SHIM_SETUP_HOOKS

Typedefs
typedef HMODULE(WINAPI *	LOADLIBRARYAPROC) (LPCSTR lpLibFileName)
typedef HMODULE(WINAPI *	LOADLIBRARYWPROC) (LPCWSTR lpLibFileName)
typedef HMODULE(WINAPI *	LOADLIBRARYEXAPROC) (LPCSTR lpLibFileName, HANDLE hFile, DWORD dwFlags)
typedef HMODULE(WINAPI *	LOADLIBRARYEXWPROC) (LPCWSTR lpLibFileName, HANDLE hFile, DWORD dwFlags)
typedef FARPROC(WINAPI *	GETPROCADDRESSPROC) (HMODULE hModule, LPCSTR lpProcName)
typedef BOOL(WINAPI *	FREELIBRARYPROC) (HINSTANCE hLibModule)

Functions
INT_PTR WINAPI	DirectXSetup (HWND hWnd, LPSTR lpszRootPath, DWORD dwFlags)
INT_PTR WINAPI	DirectXSetupA (HWND hWnd, LPSTR lpszRootPath, DWORD dwFlags)
INT_PTR WINAPI	DirectXSetupW (HWND hWnd, LPWSTR lpszRootPath, DWORD dwFlags)
INT_PTR WINAPI	DirectXSetupGetVersion (DWORD *lpdwVersion, DWORD *lpdwMinorVersion)
static BOOL	IsDxSetupW (PCUNICODE_STRING LibraryPath)
static BOOL	IsDxSetupA (PCANSI_STRING LibraryPath)
HMODULE WINAPI SHIM_OBJ_NAME()	APIHook_LoadLibraryA (LPCSTR lpLibFileName)
HMODULE WINAPI SHIM_OBJ_NAME()	APIHook_LoadLibraryW (LPCWSTR lpLibFileName)
HMODULE WINAPI SHIM_OBJ_NAME()	APIHook_LoadLibraryExA (LPCSTR lpLibFileName, HANDLE hFile, DWORD dwFlags)
HMODULE WINAPI SHIM_OBJ_NAME()	APIHook_LoadLibraryExW (LPCWSTR lpLibFileName, HANDLE hFile, DWORD dwFlags)
FARPROC WINAPI SHIM_OBJ_NAME()	APIHook_GetProcAddress (HMODULE hModule, LPCSTR lpProcName)
BOOL WINAPI SHIM_OBJ_NAME()	APIHook_FreeLibrary (HINSTANCE hLibModule)

Macro Definition Documentation

DSETUPERR_SUCCESS

#define DSETUPERR_SUCCESS   0

Definition at line 26 of file forcedxsetupsuccess.c.

SHIM_NS

#define SHIM_NS   ForceDXSetupSuccess

Definition at line 22 of file forcedxsetupsuccess.c.

SHIM_NUM_HOOKS

#define SHIM_NUM_HOOKS   6

Definition at line 192 of file forcedxsetupsuccess.c.

SHIM_SETUP_HOOKS

#define SHIM_SETUP_HOOKS

Value:

    SHIM_HOOK(0, "KERNEL32.DLL", "LoadLibraryA", SHIM_OBJ_NAME(APIHook_LoadLibraryA)) \
    SHIM_HOOK(1, "KERNEL32.DLL", "LoadLibraryW", SHIM_OBJ_NAME(APIHook_LoadLibraryW)) \
    SHIM_HOOK(2, "KERNEL32.DLL", "LoadLibraryExA", SHIM_OBJ_NAME(APIHook_LoadLibraryExA)) \
    SHIM_HOOK(3, "KERNEL32.DLL", "LoadLibraryExW", SHIM_OBJ_NAME(APIHook_LoadLibraryExW)) \
    SHIM_HOOK(4, "KERNEL32.DLL", "GetProcAddress", SHIM_OBJ_NAME(APIHook_GetProcAddress)) \
    SHIM_HOOK(5, "KERNEL32.DLL", "FreeLibrary", SHIM_OBJ_NAME(APIHook_FreeLibrary))

Definition at line 193 of file forcedxsetupsuccess.c.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 8 of file forcedxsetupsuccess.c.

Typedef Documentation

FREELIBRARYPROC

typedef BOOL(WINAPI * FREELIBRARYPROC) (HINSTANCE hLibModule)

Definition at line 19 of file forcedxsetupsuccess.c.

GETPROCADDRESSPROC

typedef FARPROC(WINAPI * GETPROCADDRESSPROC) (HMODULE hModule, LPCSTR lpProcName)

Definition at line 18 of file forcedxsetupsuccess.c.

LOADLIBRARYAPROC

typedef HMODULE(WINAPI * LOADLIBRARYAPROC) (LPCSTR lpLibFileName)

Definition at line 14 of file forcedxsetupsuccess.c.

LOADLIBRARYEXAPROC

typedef HMODULE(WINAPI * LOADLIBRARYEXAPROC) (LPCSTR lpLibFileName, HANDLE hFile, DWORD dwFlags)

Definition at line 16 of file forcedxsetupsuccess.c.

LOADLIBRARYEXWPROC

typedef HMODULE(WINAPI * LOADLIBRARYEXWPROC) (LPCWSTR lpLibFileName, HANDLE hFile, DWORD dwFlags)

Definition at line 17 of file forcedxsetupsuccess.c.

LOADLIBRARYWPROC

typedef HMODULE(WINAPI * LOADLIBRARYWPROC) (LPCWSTR lpLibFileName)

Definition at line 15 of file forcedxsetupsuccess.c.

Function Documentation

APIHook_FreeLibrary()

BOOL WINAPI SHIM_OBJ_NAME() APIHook_FreeLibrary

(

HINSTANCE

hLibModule

)

Definition at line 180 of file forcedxsetupsuccess.c.

{
    if (hLibModule == ShimLib_Instance())
    {
        SHIM_MSG("Intercepted\n");
        return TRUE;
    }
 
    return CALL_SHIM(5, FREELIBRARYPROC)(hLibModule);
}

APIHook_GetProcAddress()

FARPROC WINAPI SHIM_OBJ_NAME() APIHook_GetProcAddress	(	HMODULE	hModule,
		LPCSTR	lpProcName
	)

Definition at line 148 of file forcedxsetupsuccess.c.

{
    static const STRING DxSetupFunctions[] = {
        RTL_CONSTANT_STRING("DirectXSetup"),
        RTL_CONSTANT_STRING("DirectXSetupA"),
        RTL_CONSTANT_STRING("DirectXSetupW"),
        RTL_CONSTANT_STRING("DirectXSetupGetVersion"),
    };
    static const FARPROC DxSetupRedirections[] = {
        DirectXSetup,
        DirectXSetupA,
        DirectXSetupW,
        DirectXSetupGetVersion,
    };
    DWORD n;
 
    if (hModule == ShimLib_Instance() && ((ULONG_PTR)lpProcName > MAXUSHORT))
    {
        STRING ProcName;
        RtlInitAnsiString(&ProcName, lpProcName);
        for (n = 0; n < ARRAYSIZE(DxSetupFunctions); ++n)
        {
            if (RtlEqualString(&ProcName, DxSetupFunctions + n, TRUE))
            {
                SHIM_MSG("Intercepted %Z\n", DxSetupFunctions + n);
                return DxSetupRedirections[n];
            }
        }
    }
    return CALL_SHIM(4, GETPROCADDRESSPROC)(hModule, lpProcName);
}

APIHook_LoadLibraryA()

HMODULE WINAPI SHIM_OBJ_NAME() APIHook_LoadLibraryA

(

LPCSTR

lpLibFileName

)

Definition at line 104 of file forcedxsetupsuccess.c.

{
    ANSI_STRING Library;
 
    RtlInitAnsiString(&Library, lpLibFileName);
    if (IsDxSetupA(&Library))
        return ShimLib_Instance();
 
    return CALL_SHIM(0, LOADLIBRARYAPROC)(lpLibFileName);
}

APIHook_LoadLibraryExA()

HMODULE WINAPI SHIM_OBJ_NAME() APIHook_LoadLibraryExA	(	LPCSTR	lpLibFileName,
		HANDLE	hFile,
		DWORD	dwFlags
	)

Definition at line 126 of file forcedxsetupsuccess.c.

{
    ANSI_STRING Library;
 
    RtlInitAnsiString(&Library, lpLibFileName);
    if (IsDxSetupA(&Library))
        return ShimLib_Instance();
 
    return CALL_SHIM(2, LOADLIBRARYEXAPROC)(lpLibFileName, hFile, dwFlags);
}

APIHook_LoadLibraryExW()

HMODULE WINAPI SHIM_OBJ_NAME() APIHook_LoadLibraryExW	(	LPCWSTR	lpLibFileName,
		HANDLE	hFile,
		DWORD	dwFlags
	)

Definition at line 137 of file forcedxsetupsuccess.c.

{
    UNICODE_STRING Library;
 
    RtlInitUnicodeString(&Library, lpLibFileName);
    if (IsDxSetupW(&Library))
        return ShimLib_Instance();
 
    return CALL_SHIM(3, LOADLIBRARYEXWPROC)(lpLibFileName, hFile, dwFlags);
}

APIHook_LoadLibraryW()

HMODULE WINAPI SHIM_OBJ_NAME() APIHook_LoadLibraryW

(

LPCWSTR

lpLibFileName

)

Definition at line 115 of file forcedxsetupsuccess.c.

{
    UNICODE_STRING Library;
 
    RtlInitUnicodeString(&Library, lpLibFileName);
    if (IsDxSetupW(&Library))
        return ShimLib_Instance();
 
    return CALL_SHIM(1, LOADLIBRARYWPROC)(lpLibFileName);
}

DirectXSetup()

INT_PTR WINAPI DirectXSetup	(	HWND	hWnd,
		LPSTR	lpszRootPath,
		DWORD	dwFlags
	)

Definition at line 28 of file forcedxsetupsuccess.c.

{
    SHIM_MSG("Returning DSETUPERR_SUCCESS\n");
    return DSETUPERR_SUCCESS;
}

Referenced by APIHook_GetProcAddress().

DirectXSetupA()

INT_PTR WINAPI DirectXSetupA	(	HWND	hWnd,
		LPSTR	lpszRootPath,
		DWORD	dwFlags
	)

Definition at line 34 of file forcedxsetupsuccess.c.

{
    SHIM_MSG("Returning DSETUPERR_SUCCESS\n");
    return DSETUPERR_SUCCESS;
}

Referenced by APIHook_GetProcAddress().

DirectXSetupGetVersion()

INT_PTR WINAPI DirectXSetupGetVersion	(	DWORD *	lpdwVersion,
		DWORD *	lpdwMinorVersion
	)

Definition at line 46 of file forcedxsetupsuccess.c.

{
    if (lpdwVersion)
        *lpdwVersion = MAKELONG(7, 4);     // DirectX 7.0
    if (lpdwMinorVersion)
        *lpdwMinorVersion = MAKELONG(1792, 0);
 
    return TRUE;
}

Referenced by APIHook_GetProcAddress(), START_TEST(), and test_GetProcAddress().

DirectXSetupW()

INT_PTR WINAPI DirectXSetupW	(	HWND	hWnd,
		LPWSTR	lpszRootPath,
		DWORD	dwFlags
	)

Definition at line 40 of file forcedxsetupsuccess.c.

{
    SHIM_MSG("Returning DSETUPERR_SUCCESS\n");
    return DSETUPERR_SUCCESS;
}

Referenced by APIHook_GetProcAddress().

IsDxSetupA()

static BOOL IsDxSetupA ( PCANSI_STRING  LibraryPath )

static

Definition at line 90 of file forcedxsetupsuccess.c.

{
    BOOL bIsDxSetup;
    UNICODE_STRING LibraryPathW;
 
    if (NT_SUCCESS(RtlAnsiStringToUnicodeString(&LibraryPathW, LibraryPath, TRUE)))
    {
        bIsDxSetup = IsDxSetupW(&LibraryPathW);
        RtlFreeUnicodeString(&LibraryPathW);
        return bIsDxSetup;
    }
    return FALSE;
}

Referenced by APIHook_LoadLibraryA(), and APIHook_LoadLibraryExA().

IsDxSetupW()

static BOOL IsDxSetupW ( PCUNICODE_STRING  LibraryPath )

static

Definition at line 56 of file forcedxsetupsuccess.c.

{
    static const UNICODE_STRING DxSetupDlls[] = {
        RTL_CONSTANT_STRING(L"dsetup.dll"),
        RTL_CONSTANT_STRING(L"dsetup32.dll"),
        RTL_CONSTANT_STRING(L"dsetup"),
        RTL_CONSTANT_STRING(L"dsetup32"),
    };
    static const UNICODE_STRING PathDividerFind = RTL_CONSTANT_STRING(L"\\/");
    UNICODE_STRING LibraryName;
    USHORT PathDivider;
    DWORD n;
 
    if (!NT_SUCCESS(RtlFindCharInUnicodeString(RTL_FIND_CHAR_IN_UNICODE_STRING_START_AT_END, LibraryPath, &PathDividerFind, &PathDivider)))
        PathDivider = 0;
 
    if (PathDivider)
        PathDivider += sizeof(WCHAR);
 
    LibraryName.Buffer = LibraryPath->Buffer + PathDivider / sizeof(WCHAR);
    LibraryName.Length = LibraryPath->Length - PathDivider;
    LibraryName.MaximumLength = LibraryPath->MaximumLength - PathDivider;
 
    for (n = 0; n < ARRAYSIZE(DxSetupDlls); ++n)
    {
        if (RtlEqualUnicodeString(&LibraryName, DxSetupDlls + n, TRUE))
        {
            SHIM_MSG("Found %wZ\n", DxSetupDlls + n);
            return TRUE;
        }
    }
    return FALSE;
}

Referenced by APIHook_LoadLibraryExW(), APIHook_LoadLibraryW(), and IsDxSetupA().