C++ - 从结束地址获取函数的起始地址/获取函数的大小

Question

我正在使用 Visual Studio 的 /Gh 和 /GH 编译器选项来分析一堆代码。 使用的两种方法是 _penter 和 _pexit，它们在被分析的代码中进入或退出函数时调用。 由于我需要对特定函数进行分析/调试，因此我使用了一个已定义的数组FuncTable ，其中包含我需要使用其名称作为字符串进行检测的函数的地址。 因此，当输入函数时，基本上包含寄存器内容的pStack[0]包含正在执行的代码的当前行的地址。 类似地，当函数退出时， pStack[0]包含代码最后一行的地址。

问题：当一个函数被输入时（_penter 被调用），我得到了 pStack[0] 中函数第一行的地址，因此我可以通过减去一个常量（-5）来得到函数的地址并将它保存到我的稍后在 _pexit 函数中检索的列表。 但是由于在 _pexit 中我得到了函数最后一行的地址，我需要找到函数的大小，以便从 pStack[0] 中的地址中减去该大小以到达函数的起始地址，然后进行比较该地址指向我列表中保存的地址。 下面贴上代码。

void _stdcall EnterFunc0(unsigned * pStack)
{
    void      * pCaller;
    pCaller = (void *)(pStack[0] - 5); // pStack[0] is first line, -5 for function address
    Signature * funct = FuncTable;

    while (funct->function)
    {
        const BYTE * func = (const BYTE *)funct->function;
        if ((func == (const BYTE *)pCaller) || ((*func == 0xE9) && ((func + *(DWORD *)(func + 1) + 5) == (const BYTE *)pCaller)))
        {
            Stack_Push(funct->name, funct->returnType, true, pCaller);          
        }
        funct++;
    }

}

extern "C" __declspec(naked) void __cdecl _penter()
{
    _asm
    {
        pushad              // save all general purpose registers
        mov    eax, esp     // current stack pointer
        add    eax, 32      // stack pointer before pushad
        push   eax          // push pointer to return address as parameter to EnterFunc0

        call   EnterFunc0

        popad               // restore general purpose registers
        ret                 // start executing original function
    }
}
void _stdcall ExitFunc0(unsigned * pStack)
{
    if (startRecording) 
    {
        StackEntry * start = top;
        while (start != NULL)
        {
            //**HERE I NEED TO COMPARE THE ADDRESS OF THE FUNCTION WITH THE ONE ALREADY IN MY STACK**
                            if ((void *)(pStack[0] - sizeOfTheFunction) == start->Address)
            {
                OutputDebugString("Function Found\n");
            }
            start = start->next;
        }
    }

}
extern "C" __declspec(naked) void __cdecl _pexit()
{
    _asm
    {
        pushad              // save all general purpose registers
        mov    eax, esp     // current stack pointer
        add    eax, 32      // stack pointer before pushad
        push   eax          // push pointer to return address as parameter to EnterFunc0

        call   ExitFunc0

        popad               // restore general purpose registers
        ret                 // start executing original function
    }
}

Answer 1

您已经知道 _pexit() 函数中的地址，它是在 _penter() 函数中传递给您的。 您所要做的就是支持嵌套函数调用。 std::stack<> 对此有好处。 使用 push() 将地址保存在 _penter 中，使用 _pexit 函数中的 top() 检索它并调用 pop()。

根本不需要知道函数体的大小。

Answer 2

由于编译器确保在每个函数的开始和结束时调用_penter和_pexit ，您可以确定在调用_pexit ，由_penter创建的堆栈顶部的函数指针始终指向当前函数。 没有必要搜索它。

（这应该是正确的，除非您手动调用其中一个函数，您不应该这样做，或者有一个多线程程序。在后一种情况下，您应该为每个线程创建一个私有堆栈。当然，您也将拥有将Stack_Pop调用添加到_pexit ，但我假设您无论如何都打算这样做。）

Answer 3

事实证明，我以错误的方式看待问题。 我通过使用SymFromAddress方法从dbghelp.lib获取符号信息解决了这个问题。 一旦我得到了方法的名称，我就能够将它与我存储在 _penter 中的信息进行比较。

SYMBOL_INFO  * mysymbol;
HANDLE         process;
char           temp[MAX_TEMP_LENGTH] = "                                                                       ";

process = GetCurrentProcess();
SymInitialize(process, NULL, TRUE);

mysymbol = (SYMBOL_INFO *)calloc(sizeof(SYMBOL_INFO) + 256 * sizeof(char), 1);
mysymbol->MaxNameLen = 255;
mysymbol->SizeOfStruct = sizeof(SYMBOL_INFO);
SymFromAddr(process, (DWORD64)((void *)pStack[0]), 0, mysymbol);
OutputDebugString(mysymbol->Name);
OutputDebugString("\n");

Answer 4

Configuration Properties > C/C++ > Command Line

将编译器选项添加到Additional Options框

像这样

为 _penter 钩子添加标志 /Gh
为 _pexit 钩子添加标志 /GH

我用于跟踪/记录的代码

#include <intrin.h>

extern "C"  void __declspec(naked) __cdecl _penter(void) {
    __asm {
        push ebp;               // standard prolog
        mov ebp, esp;
        sub esp, __LOCAL_SIZE
        pushad;                 // save registers
    }
    // _ReturnAddress always returns the address directly after the call, but that is not the start of the function!
    PBYTE addr;
    addr = (PBYTE)_ReturnAddress() - 5;

    SYMBOL_INFO* mysymbol;
    HANDLE       process;
    process = GetCurrentProcess();
    SymInitialize(process, NULL, TRUE);
    mysymbol = (SYMBOL_INFO*)calloc(sizeof(SYMBOL_INFO) + 256 * sizeof(char), 1);
    mysymbol->MaxNameLen = 255;
    mysymbol->SizeOfStruct = sizeof(SYMBOL_INFO);
    SymFromAddr(process, (DWORD64)((void*)addr), 0, mysymbol);
    myprintf("Entered Function: %s [0x%X]\n", mysymbol->Name, addr);

    _asm {
        popad;              // restore regs
        mov esp, ebp;       // standard epilog
        pop ebp;
        ret;
    }
}

extern "C"  void __declspec(naked) __cdecl _pexit(void) {
    __asm {
        push ebp;               // standard prolog
        mov ebp, esp;
        sub esp, __LOCAL_SIZE
        pushad;                 // save registers
    }
    // _ReturnAddress always returns the address directly after the call, but that is not the start of the function!
    PBYTE addr;
    addr = (PBYTE)_ReturnAddress() - 5;

    SYMBOL_INFO* mysymbol;
    HANDLE       process;
    process = GetCurrentProcess();
    SymInitialize(process, NULL, TRUE);
    mysymbol = (SYMBOL_INFO*)calloc(sizeof(SYMBOL_INFO) + 256 * sizeof(char), 1);
    mysymbol->MaxNameLen = 255;
    mysymbol->SizeOfStruct = sizeof(SYMBOL_INFO);
    SymFromAddr(process, (DWORD64)((void*)addr), 0, mysymbol);
    myprintf("Exit Function: %s [0x%X]\n", mysymbol->Name, addr);

    _asm {
        popad;              // restore regs
        mov esp, ebp;       // standard epilog
        pop ebp;
        ret;
    }
}