打印函数堆栈调用情况

windows环境下C++代码打印函数堆栈调用情况_AlbertS的博客-CSDN博客_windows打印堆栈

linux环境下C++代码打印函数堆栈调用情况_L7256的博客-CSDN博客

windows:

#include <windows.h>
#include <dbghelp.h>
#include <stdio.h>

#if _MSC_VER
#define snprintf _snprintf
#endif

#define STACK_INFO_LEN  1024

void ShowTraceStack(char* szBriefInfo)
{
    static const int MAX_STACK_FRAMES = 12;
    void *pStack[MAX_STACK_FRAMES];
    static char szStackInfo[STACK_INFO_LEN * MAX_STACK_FRAMES];
    static char szFrameInfo[STACK_INFO_LEN];

    HANDLE process = GetCurrentProcess();
    SymInitialize(process, NULL, TRUE);
    WORD frames = CaptureStackBackTrace(0, MAX_STACK_FRAMES, pStack, NULL);
    strcpy(szStackInfo, szBriefInfo == NULL ? "stack traceback:\n" : szBriefInfo);

    for (WORD i = 0; i < frames; ++i) {
        DWORD64 address = (DWORD64)(pStack[i]);

        DWORD64 displacementSym = 0;
        char buffer[sizeof(SYMBOL_INFO)+MAX_SYM_NAME * sizeof(TCHAR)];
        PSYMBOL_INFO pSymbol = (PSYMBOL_INFO)buffer;
        pSymbol->SizeOfStruct = sizeof(SYMBOL_INFO);
        pSymbol->MaxNameLen = MAX_SYM_NAME;

        DWORD displacementLine = 0;
        IMAGEHLP_LINE64 line;
        line.SizeOfStruct = sizeof(IMAGEHLP_LINE64);

        if (SymFromAddr(process, address, &displacementSym, pSymbol) && 
            SymGetLineFromAddr64(process, address, &displacementLine, &line))
        {
            snprintf(szFrameInfo, sizeof(szFrameInfo), "\t%s() at %s:%d(0x%x)\n", 
                pSymbol->Name, line.FileName, line.LineNumber, pSymbol->Address);
        }
        else
        {
            snprintf(szFrameInfo, sizeof(szFrameInfo), "\terror: %d\n", GetLastError());
        }
        strcat(szStackInfo, szFrameInfo);
    }

    printf("%s", szStackInfo); // 输出到控制台，也可以打印到日志文件中
}

void func2()
{
    bool isError = true;
    if (isError)
    {
        ShowTraceStack("error in func2\n");
    }
    else
    {
        printf("this is func2\n");
    }
}

void func1()
{
    int sum = 0;
    for (int i = 0; i < 100; ++i)
        sum += i;

    func2();
}

int main(int argc, char* argv[])
{
    printf("hello world\n");
    func1();

    return 0;
}
 

linux:

#include <execinfo.h>
#include <unistd.h>
#include <string.h>
#include <stdio.h>
 
#define STACK_INFO_LEN  1024  //每个栈的最大长度
 
void ShowTraceStack(const char* szBriefInfo) //放在被调用的最后一个函数，可看到程序调用到这里的所有调用
{
    static const int MAX_STACK_FRAMES = 12; //最大的堆调用数
    void *pStack[MAX_STACK_FRAMES];
    static char szStackInfo[STACK_INFO_LEN * MAX_STACK_FRAMES];
 
    char ** pStackList = NULL;
    int frames_len = backtrace(pStack, MAX_STACK_FRAMES);  //传入一个数组pStack[12]，并将调用信息放入pStack[12]，并返回调用的深度
    printf("ftrames = %d\n",frames_len); //6
    pStackList = backtrace_symbols(pStack, frames_len);  //返回调用数组指针，配合frames_len可打印信息
    if (NULL == pStackList)
        return;
 
    strcpy(szStackInfo, szBriefInfo == NULL ? "stack traceback:\n" : szBriefInfo);  //szStackInfo放入"stack traceback:\n"
    for (int i = 0; i < frames_len; ++i)
    {
        if (NULL == pStackList[i])
            break;
 
        strncat(szStackInfo, pStackList[i], STACK_INFO_LEN);//拷贝堆栈信息到szStackInfo字符串数组
        strcat(szStackInfo, "\n");
    }
 
    printf("%s", szStackInfo); // 输出到控制台。
}
 
void func2()
{
    bool isError = true;
    if (isError)
    {
        ShowTraceStack("error in func2\n");
    }
    else
    {
        printf("this is func2\n");
    }
}
 
void func1()
{
    int sum = 0;
    for (int i = 0; i < 100; ++i)
        sum += i;
 
    func2();
}
 
 
int main(int argc, char* argv[])
{
    printf("hello world\n");
    func1();
 
    return 0;
}