rsym.c File Reference

#include <ntstatus.h>
#include <windows.h>
#include <dbghelp.h>
#include <cvconst.h>
#include <stdio.h>
#include "wine/test.h"

Include dependency graph for rsym.c:

Go to the source code of this file.

Classes
struct	_test_context
struct	_test_data

Macros
#define	WIN32_NO_STATUS
#define	ok_ulonglong(expression, result)
#define	ok_ulonglong_(file, line, expression, result)
#define	init_sym(fInvadeProcess)   init_sym_imp(fInvadeProcess, __FILE__, __LINE__)
#define	INIT_PSYM(buff)

Typedefs
typedef struct _test_context	test_context

Functions
void	dump_rsym (const char *filename)
int	extract_gcc_dll (char szFile[MAX_PATH])
void	cleanup_gcc_dll ()
static HANDLE	proc ()
static BOOL	init_sym_imp (BOOL fInvadeProcess, const char *file, int line)
static void	deinit_sym ()
static BOOL	supports_rsym (HANDLE hProc, DWORD64 BaseAddress)
static void	test_SymFromName (HANDLE hProc, DWORD64 BaseAddress)
static void	test_SymFromAddr (HANDLE hProc, DWORD64 BaseAddress)
static BOOL CALLBACK	EnumSymProc (PSYMBOL_INFO pSymInfo, ULONG SymbolSize, PVOID UserContext)
static void	test_SymEnumSymbols (HANDLE hProc, DWORD64 BaseAddress)
	START_TEST (rsym)

Variables
static struct _test_data	test_data []

Macro Definition Documentation

INIT_PSYM

#define INIT_PSYM

(

buff

)

Value:

    do { \
    memset((buff), 0, sizeof((buff))); \
    ((PSYMBOL_INFO)(buff))->SizeOfStruct = sizeof(SYMBOL_INFO); \
    ((PSYMBOL_INFO)(buff))->MaxNameLen = MAX_SYM_NAME; \
} while (0)

Definition at line 64 of file rsym.c.

init_sym

#define init_sym

(

fInvadeProcess

)

init_sym_imp(fInvadeProcess, __FILE__, __LINE__)

Definition at line 62 of file rsym.c.

ok_ulonglong

#define ok_ulonglong	(		expression,
			result
	)

Value:

    do { \
        ULONG64 _value = (expression); \
        ULONG64 _result = (result); \
        ok(_value == _result, "Wrong value for '%s', expected: " #result " (%s), got: %s\n", \
           #expression, wine_dbgstr_longlong(_result), wine_dbgstr_longlong(_value)); \
    } while (0)

Definition at line 19 of file rsym.c.

ok_ulonglong_

#define ok_ulonglong_	(		file,
			line,
			expression,
			result
	)

Value:

    do { \
        ULONG64 _value = (expression); \
        ULONG64 _result = (result); \
        ok_(file, line)(_value == _result, "Wrong value for '%s', expected: " #result " (%s), got: %s\n", \
           #expression, wine_dbgstr_longlong(_result), wine_dbgstr_longlong(_value)); \
    } while (0)

Definition at line 27 of file rsym.c.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 11 of file rsym.c.

Typedef Documentation

test_context

typedef struct _test_context test_context

Function Documentation

cleanup_gcc_dll()

void cleanup_gcc_dll

(

)

Definition at line 120 of file data.c.

{
    char szFile[MAX_PATH];
    BOOL ret;
    const char* dir = tmpdir();
 
    sprintf(szFile, "%s\\uffs.dll", dir);
    ret = DeleteFileA(szFile);
    ok(ret, "DeleteFileA failed(%d)\n", GetLastError());
    ret = RemoveDirectoryA(dir);
    ok(ret, "RemoveDirectoryA failed(%d)\n", GetLastError());
}

Referenced by START_TEST().

deinit_sym()

static void deinit_sym ( )

static

Definition at line 57 of file rsym.c.

{
    SymCleanup(proc());
}

Referenced by START_TEST().

dump_rsym()

void dump_rsym

(

const char *

filename

)

EnumSymProc()

static BOOL CALLBACK EnumSymProc ( PSYMBOL_INFO  pSymInfo, ULONG  SymbolSize, PVOID  UserContext  )

static

Definition at line 252 of file rsym.c.

{
    test_context* ctx = UserContext;
 
    if (ctx->Index < ARRAYSIZE(test_data))
    {
        ok_ulonglong_(__FILE__, test_data[ctx->Index].Line, pSymInfo->ModBase, ctx->BaseAddress);
        if (test_data[ctx->Index].AddressOffset > 0x100000)
            ok_ulonglong_(__FILE__, test_data[ctx->Index].Line, pSymInfo->Address, test_data[ctx->Index].AddressOffset);
        else
            ok_ulonglong_(__FILE__, test_data[ctx->Index].Line, pSymInfo->Address, ctx->BaseAddress + test_data[ctx->Index].AddressOffset);
        ok_hex_(__FILE__, test_data[ctx->Index].Line, pSymInfo->Tag, test_data[ctx->Index].Tag);
        ok_str_(__FILE__, test_data[ctx->Index].Line, pSymInfo->Name, test_data[ctx->Index].Name);
 
        ctx->Index++;
    }
    else
    {
        ok(0, "Out of bounds (%lu), max is: %i!\n", ctx->Index, ARRAYSIZE(test_data));
    }
 
    return TRUE;
}

Referenced by test_SymEnumSymbols().

extract_gcc_dll()

int extract_gcc_dll

(

char

szFile[MAX_PATH]

)

Definition at line 107 of file data.c.

{
    const char* dir = tmpdir();
    BOOL ret = CreateDirectoryA(dir, NULL);
    ok(ret, "CreateDirectoryA failed(%d)\n", GetLastError());
 
    sprintf(szFile, "%s\\uffs.dll", dir);
    if (!extract_one(szFile, "gcc_uffs.dll"))
        return 0;
 
    return 1;
}

Referenced by START_TEST().

init_sym_imp()

static BOOL init_sym_imp ( BOOL  fInvadeProcess, const char *  file, int  line  )

static

Definition at line 46 of file rsym.c.

{
    if (!SymInitialize(proc(), NULL, fInvadeProcess))
    {
        DWORD err = GetLastError();
        ok_(file, line)(0, "Failed to init: 0x%x\n", err);
        return FALSE;
    }
    return TRUE;
}

proc()

static HANDLE proc ( )

static

Definition at line 41 of file rsym.c.

{
    return GetCurrentProcess();
}

Referenced by deinit_sym(), init_sym_imp(), and START_TEST().

START_TEST()

START_TEST

(

rsym

)

Definition at line 297 of file rsym.c.

{
    char szDllName[MAX_PATH];
#ifdef _M_IX86
    HMODULE hMod;
#endif
    DWORD64 BaseAddress;
    DWORD dwErr, Options;
 
    Options = SymGetOptions();
    Options &= ~(SYMOPT_UNDNAME);
    //Options |= SYMOPT_DEBUG;
    SymSetOptions(Options);
 
    if (!extract_gcc_dll(szDllName))
    {
        ok(0, "Failed extracting files\n");
        return;
    }
 
    if (init_sym(FALSE))
    {
        SetLastError(ERROR_SUCCESS);
        BaseAddress = SymLoadModule64(proc(), NULL, szDllName, NULL, 0x600000, 0);
        dwErr = GetLastError();
 
        ok_ulonglong(BaseAddress, 0x600000);
        ok(dwErr == ERROR_SUCCESS || dwErr == ERROR_FILE_NOT_FOUND, "Got 0x%x\n", dwErr);
 
        if (BaseAddress == 0x600000)
        {
            trace("Module loaded by SymLoadModule64\n");
            test_SymFromName(proc(), BaseAddress);
            test_SymFromAddr(proc(), BaseAddress);
            test_SymEnumSymbols(proc(), BaseAddress);
        }
 
        deinit_sym();
    }
 
#ifdef _M_IX86
    hMod = LoadLibraryA(szDllName);
    if (hMod)
    {
        BaseAddress = (DWORD64)(DWORD_PTR)hMod;
        /* Invade process */
        if (init_sym(TRUE))
        {
            trace("Module loaded by LoadLibraryA\n");
            test_SymFromName(proc(), BaseAddress);
            test_SymFromAddr(proc(), BaseAddress);
            test_SymEnumSymbols(proc(), BaseAddress);
 
            deinit_sym();
        }
 
        FreeLibrary(hMod);
    }
#endif
    cleanup_gcc_dll();
}

supports_rsym()

static BOOL supports_rsym ( HANDLE  hProc, DWORD64  BaseAddress  )

static

Definition at line 70 of file rsym.c.

{
    IMAGEHLP_MODULE64 ModuleInfo;
    BOOL Ret;
 
    memset(&ModuleInfo, 0, sizeof(ModuleInfo));
    ModuleInfo.SizeOfStruct = sizeof(ModuleInfo);
    Ret = SymGetModuleInfo64(hProc, BaseAddress, &ModuleInfo);
 
    return Ret &&
        ModuleInfo.SymType == SymDia &&
        ModuleInfo.CVSig == ('R' | ('S' << 8) | ('Y' << 16) | ('M' << 24));
}

Referenced by test_SymEnumSymbols(), test_SymFromAddr(), and test_SymFromName().

test_SymEnumSymbols()

static void test_SymEnumSymbols ( HANDLE  hProc, DWORD64  BaseAddress  )

static

Definition at line 276 of file rsym.c.

{
    BOOL Ret;
    test_context ctx;
 
    ctx.Index = 0;
    ctx.BaseAddress = BaseAddress;
 
    if (!supports_rsym(hProc, ctx.BaseAddress))
    {
        skip("dbghelp.dll cannot parse rsym\n");
    }
    else
    {
        Ret = SymEnumSymbols(hProc, ctx.BaseAddress, NULL, EnumSymProc, &ctx);
        ok_int(Ret, TRUE);
        ok_int(ctx.Index, ARRAYSIZE(test_data));
    }
}

Referenced by START_TEST().

test_SymFromAddr()

static void test_SymFromAddr ( HANDLE  hProc, DWORD64  BaseAddress  )

static

Definition at line 126 of file rsym.c.

{
    BOOL Ret;
    char buffer[sizeof(SYMBOL_INFO) + MAX_SYM_NAME * sizeof(TCHAR)];
    PSYMBOL_INFO pSymbol = (PSYMBOL_INFO)buffer;
 
    DWORD64 Displacement;
    DWORD dwErr;
 
    if (!supports_rsym(hProc, BaseAddress))
    {
        skip("dbghelp.dll cannot parse rsym\n");
    }
    else
    {
        /* No address found before load address of module */
        Displacement = 0;
        INIT_PSYM(buffer);
        Ret = SymFromAddr(hProc, BaseAddress -1, &Displacement, pSymbol);
        dwErr = GetLastError();
        ok_int(Ret, FALSE);
        ok_hex(dwErr, ERROR_MOD_NOT_FOUND);
 
        /* Right at the start of the module is recognized as the first symbol found */
        Displacement = 0;
        INIT_PSYM(buffer);
        Ret = SymFromAddr(hProc, BaseAddress, &Displacement, pSymbol);
        /* Our dbghelp.dll does not recognize this yet */
        todo_if(!Ret)
        {
            ok_int(Ret, TRUE);
            ok_ulonglong(Displacement, 0xffffffffffffffff);
            ok_ulonglong(pSymbol->ModBase, BaseAddress);
            ok_hex(pSymbol->Flags, SYMFLAG_FUNCTION);
            ok_ulonglong(pSymbol->Address, BaseAddress + 0x1000);
            ok_hex(pSymbol->Tag, SymTagFunction);
            ok_str(pSymbol->Name, "DllMain");
        }
 
        /* The actual first instruction of the function */
        Displacement = 0;
        INIT_PSYM(buffer);
        Ret = SymFromAddr(hProc, BaseAddress + 0x1000, &Displacement, pSymbol);
        ok_int(Ret, TRUE);
        ok_ulonglong(Displacement, 0);
        ok_ulonglong(pSymbol->ModBase, BaseAddress);
        ok_hex(pSymbol->Flags, SYMFLAG_FUNCTION);
        ok_ulonglong(pSymbol->Address, BaseAddress + 0x1000);
        ok_hex(pSymbol->Tag, SymTagFunction);
        ok_str(pSymbol->Name, "DllMain");
 
        /* The last instruction in the function */
        Displacement = 0;
        INIT_PSYM(buffer);
        Ret = SymFromAddr(hProc, BaseAddress + 0x1009, &Displacement, pSymbol);
        ok_int(Ret, TRUE);
        ok_ulonglong(Displacement, 0x9);
        ok_ulonglong(pSymbol->ModBase, BaseAddress);
        ok_hex(pSymbol->Flags, SYMFLAG_FUNCTION);
        ok_ulonglong(pSymbol->Address, BaseAddress + 0x1000);
        ok_hex(pSymbol->Tag, SymTagFunction);
        ok_str(pSymbol->Name, "DllMain");
 
        /* First byte of the next function */
        Displacement = 0;
        INIT_PSYM(buffer);
        Ret = SymFromAddr(hProc, BaseAddress + 0x103F, &Displacement, pSymbol);
        ok_int(Ret, TRUE);
        ok_ulonglong(Displacement, 0);
        ok_ulonglong(pSymbol->ModBase, BaseAddress);
        ok_hex(pSymbol->Flags, SYMFLAG_FUNCTION);
        ok_ulonglong(pSymbol->Address, BaseAddress + 0x103F);
        ok_hex(pSymbol->Tag, SymTagFunction);
        ok_str(pSymbol->Name, "FfsChkdsk");
 
        /* .idata */
        Displacement = 0;
        INIT_PSYM(buffer);
        Ret = SymFromAddr(hProc, BaseAddress + 0x4000, &Displacement, pSymbol);
        ok_int(Ret, TRUE);
        ok_ulonglong(Displacement, 0);
        ok_ulonglong(pSymbol->ModBase, BaseAddress);
        ok_hex(pSymbol->Flags, SYMFLAG_EXPORT);
        ok_ulonglong(pSymbol->Address, BaseAddress + 0x4000);
        ok_hex(pSymbol->Tag, SymTagPublicSymbol);
        ok_str(pSymbol->Name, "_head_dll_ntdll_libntdll_a");
    }
}

Referenced by START_TEST().

test_SymFromName()

static void test_SymFromName ( HANDLE  hProc, DWORD64  BaseAddress  )

static

Definition at line 85 of file rsym.c.

{
    BOOL Ret;
    char buffer[sizeof(SYMBOL_INFO) + MAX_SYM_NAME * sizeof(TCHAR)];
    PSYMBOL_INFO pSymbol = (PSYMBOL_INFO)buffer;
 
    if (!supports_rsym(hProc, BaseAddress))
    {
        skip("dbghelp.dll cannot parse rsym\n");
    }
    else
    {
        INIT_PSYM(buffer);
        Ret = SymFromName(hProc, "DllMain", pSymbol);
        ok_int(Ret, TRUE);
        ok_ulonglong(pSymbol->ModBase, BaseAddress);
        ok_hex(pSymbol->Flags, SYMFLAG_FUNCTION);
        ok_ulonglong(pSymbol->Address, BaseAddress + 0x1000);
        ok_hex(pSymbol->Tag, SymTagFunction);
        ok_str(pSymbol->Name, "DllMain");
 
        INIT_PSYM(buffer);
        Ret = SymFromName(hProc, "FfsChkdsk", pSymbol);
        ok_int(Ret, TRUE);
        ok_ulonglong(pSymbol->ModBase, BaseAddress);
        ok_hex(pSymbol->Flags, SYMFLAG_FUNCTION);
        ok_ulonglong(pSymbol->Address, BaseAddress + 0x103F);
        ok_hex(pSymbol->Tag, SymTagFunction);
        ok_str(pSymbol->Name, "FfsChkdsk");
 
        INIT_PSYM(buffer);
        Ret = SymFromName(hProc, "FfsFormat", pSymbol);
        ok_int(Ret, TRUE);
        ok_ulonglong(pSymbol->ModBase, BaseAddress);
        ok_hex(pSymbol->Flags, SYMFLAG_FUNCTION);
        ok_ulonglong(pSymbol->Address, BaseAddress + 0x100C);
        ok_hex(pSymbol->Tag, SymTagFunction);
        ok_str(pSymbol->Name, "FfsFormat");
    }
}

Referenced by START_TEST().

Variable Documentation

test_data

struct _test_data test_data[]

static

Initial value:

= {
 
    
    { 0x107c, 0, SymTagPublicSymbol, "__CTOR_LIST__", __LINE__ },
    { 0x2074, 0, SymTagPublicSymbol, "__RUNTIME_PSEUDO_RELOC_LIST_END__", __LINE__ },
    { 0x1000, 12, SymTagPublicSymbol, "EntryPoint", __LINE__ },
    { 0x100c, 51, SymTagFunction, "FfsFormat", __LINE__ },
    { 0x4030, 0, SymTagPublicSymbol, "_imp__DbgPrint", __LINE__ },
    { 0x1084, 0, SymTagPublicSymbol, "__DTOR_LIST__", __LINE__ },
    { 0x103f, 53, SymTagFunction, "FfsChkdsk", __LINE__ },
    { 0x2074, 0, SymTagPublicSymbol, "_rt_psrelocs_end", __LINE__ },
    { 0x103f, 53, SymTagPublicSymbol, "ChkdskEx", __LINE__ },
    { 0x4048, 0, SymTagPublicSymbol, "_dll_ntdll_libntdll_a_iname", __LINE__ },
 
 
 
    { 0x2074, 0, SymTagPublicSymbol, "_rt_psrelocs_start", __LINE__ },
    { 0x1000, 12, SymTagFunction, "DllMain", __LINE__ },
    { 0x100c, 0, SymTagPublicSymbol, "FormatEx", __LINE__ },
    { 0x1074, 0, SymTagPublicSymbol, "DbgPrint", __LINE__ },
    { 0x68900000, 0, SymTagPublicSymbol, "__ImageBase", __LINE__ },
    { 0x68902074, 0, SymTagPublicSymbol, "__RUNTIME_PSEUDO_RELOC_LIST__", __LINE__ },
    { 0x4000, 0, SymTagPublicSymbol, "_head_dll_ntdll_libntdll_a", __LINE__ },
}