shimtest_ros.c File Reference

#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <ntndk.h>
#include <winbase.h>

Include dependency graph for shimtest_ros.c:

Go to the source code of this file.

Macros
#define	DPFLTR_ERROR_LEVEL   0

Functions
NTSYSAPI ULONG NTAPI	vDbgPrintEx (_In_ ULONG ComponentId, _In_ ULONG Level, _In_z_ PCCH Format, _In_ va_list ap)
void	xprintf (const char *fmt,...)
void	test_normal_imports ()
INT WINAPI	SHStringFromGUIDA (REFGUID, LPSTR, INT)
void	test_ordinal_imports ()
int	main (int argc, char *argv[])

Macro Definition Documentation

DPFLTR_ERROR_LEVEL

#define DPFLTR_ERROR_LEVEL   0

Definition at line 15 of file shimtest_ros.c.

Function Documentation

main()

int main	(	int argc	,
		char *	argv[]
	)

Definition at line 112 of file shimtest_ros.c.

{
    xprintf("Normal import (kernel32!GetComputerNameA)\n");
    test_normal_imports();
    xprintf("\nOrdinal import (shlwapi!#23)\n");
    test_ordinal_imports();
 
    return 0;
}

SHStringFromGUIDA()

INT WINAPI SHStringFromGUIDA	(	REFGUID	guid,
		LPSTR	lpszDest,
		INT	cchMax
	)

Definition at line 51 of file shimtest.c.

{
    CHAR Buffer[100] = {0};
 
    if (CALL_SHIM(1, tSHStringFromGUIDA)(guid, Buffer, sizeof(Buffer)-1))
    {
        SHIM_INFO("Original function returned: '%s'\n", Buffer);
    }
    else
    {
        SHIM_INFO("Original function failed\n");
        Buffer[0] = '\0';
    }
 
    StringCchCopyA(lpszDest, cchMax, "ShimTest:");
    StringCchCatA(lpszDest, cchMax, Buffer);
    return 0;
}

Referenced by SHPinDllOfCLSID(), and test_ordinal_imports().

test_normal_imports()

void test_normal_imports

(

)

Definition at line 27 of file shimtest_ros.c.

{
    char buf[100];
    DWORD dwSize;
    HMODULE mod;
    BOOL (WINAPI *p_GetComputerNameA)(LPSTR lpBuffer, LPDWORD lpnSize);
 
    /* Call it directly */
    ZeroMemory(buf, sizeof(buf));
    dwSize = sizeof(buf);
    if (!GetComputerNameA(buf, &dwSize))
    {
        xprintf("GetComputerNameA failed: %d\n", GetLastError());
    }
    else
    {
        xprintf("GetComputerNameA returned: '%s'\n", buf);
    }
 
    /* Call it using GetProcAddress */
    mod = GetModuleHandleA("kernel32.dll");
    p_GetComputerNameA = (void*)GetProcAddress(mod, "GetComputerNameA");
    if (p_GetComputerNameA == NULL)
    {
        xprintf("GetProcAddress failed: %d\n", GetLastError());
    }
    else
    {
        ZeroMemory(buf, sizeof(buf));
        dwSize = sizeof(buf);
        if (!p_GetComputerNameA(buf, &dwSize))
        {
            xprintf("p_GetComputerNameA failed: %d\n", GetLastError());
        }
        else
        {
            xprintf("p_GetComputerNameA returned: '%s'\n", buf);
        }
    }
}

Referenced by main().

test_ordinal_imports()

void test_ordinal_imports

(

)

Definition at line 69 of file shimtest_ros.c.

{
    GUID guid = { 0x11223344, 0x5566, 0x7788 };
    char buf[100];
    int r;
    HMODULE mod;
    INT (WINAPI *p_SHStringFromGUIDA)(REFGUID guid, LPSTR lpszDest, INT cchMax);
 
    /* Call it directly */
    ZeroMemory(buf, sizeof(buf));
    r = SHStringFromGUIDA(&guid, buf, _countof(buf)-1);
    if (r)
    {
        xprintf("SHStringFromGUIDA failed (%d)\n", r);
    }
    else
    {
        xprintf("SHStringFromGUIDA returned: '%s'\n", buf);
    }
 
    /* Call it using GetProcAddress */
    mod = GetModuleHandleA("shlwapi.dll");
    p_SHStringFromGUIDA = (void*)GetProcAddress(mod, (LPCSTR)23);
    if (p_SHStringFromGUIDA == NULL)
    {
        xprintf("GetProcAddress failed: %d\n", GetLastError());
    }
    else
    {
        ZeroMemory(buf, sizeof(buf));
        r = p_SHStringFromGUIDA(&guid, buf, _countof(buf)-1);
        if (r)
        {
            xprintf("p_SHStringFromGUIDA failed (%d)\n", r);
        }
        else
        {
            xprintf("p_SHStringFromGUIDA returned: '%s'\n", buf);
        }
    }
}

Referenced by main().

vDbgPrintEx()

NTSYSAPI ULONG NTAPI vDbgPrintEx	(	_In_ ULONG	ComponentId,
		_In_ ULONG	Level,
		_In_z_ PCCH	Format,
		_In_ va_list	ap
	)

Referenced by xprintf().

xprintf()

void xprintf	(	const char *	fmt,
			...
	)

Definition at line 17 of file shimtest_ros.c.

{
    va_list ap;
 
    va_start(ap, fmt);
    vprintf(fmt, ap);
    vDbgPrintEx(-1, DPFLTR_ERROR_LEVEL, fmt, ap);
    va_end(ap);
}

Referenced by main(), test_normal_imports(), and test_ordinal_imports().