crown-engine/engine/core/stacktrace_windows.cpp

/*
 * Copyright (c) 2012-2014 Daniele Bartolini and individual contributors.
 * License: https://github.com/taylor001/crown/blob/master/LICENSE
 */

#include "config.h"

#if CROWN_PLATFORM_WINDOWS

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

namespace crown
{

void stacktrace()
{
	SymInitialize(GetCurrentProcess(), NULL, TRUE);
	SymSetOptions(SYMOPT_LOAD_LINES | SYMOPT_UNDNAME);
	DWORD mtype;
	CONTEXT ctx;
	ZeroMemory(&ctx, sizeof(CONTEXT));
	ctx.ContextFlags = CONTEXT_CONTROL;
	RtlCaptureContext(&ctx);

	STACKFRAME64 stack;
	ZeroMemory(&stack, sizeof(STACKFRAME64));
#ifdef _M_IX86
	mtype = IMAGE_FILE_MACHINE_I386;
	stack.AddrPC.Offset = ctx.Eip;
	stack.AddrPC.Mode = AddrModeFlat;
	stack.AddrFrame.Offset = ctx.Ebp;
	stack.AddrFrame.Mode = AddrModeFlat;
	stack.AddrStack.Offset = ctx.Esp;
	stack.AddrStack.Mode = AddrModeFlat;
#elif _M_X64
	mtype = IMAGE_FILE_MACHINE_AMD64;
	stack.AddrPC.Offset = ctx.Rip;
	stack.AddrPC.Mode = AddrModeFlat;
	stack.AddrFrame.Offset = ctx.Rsp;
	stack.AddrFrame.Mode = AddrModeFlat;
	stack.AddrStack.Offset = ctx.Rsp;
	stack.AddrStack.Mode = AddrModeFlat;
#endif

	DWORD ldsp = 0;
	IMAGEHLP_LINE64 line;
	ZeroMemory(&line, sizeof(IMAGEHLP_LINE64));
	line.SizeOfStruct = sizeof(IMAGEHLP_LINE64);

	char buf[sizeof(SYMBOL_INFO) + MAX_SYM_NAME * sizeof(TCHAR)];
	PSYMBOL_INFO sym = (PSYMBOL_INFO)buf;
	sym->SizeOfStruct = sizeof(SYMBOL_INFO);
	sym->MaxNameLen = MAX_SYM_NAME;

	UINT num = 0;
	while (StackWalk64(mtype,
			GetCurrentProcess(),
			GetCurrentThread(),
			&stack,
			&ctx,
			NULL,
			SymFunctionTableAccess64,
			SymGetModuleBase64,
			NULL))
	{

		if (stack.AddrPC.Offset == 0)
			break;

		++num;
		BOOL res = SymGetLineFromAddr64(GetCurrentProcess(), stack.AddrPC.Offset, &ldsp, &line) &&
			SymFromAddr(GetCurrentProcess(), stack.AddrPC.Offset, 0, sym);

		if (res == TRUE)
			printf("\t[%i] %s (%s:%d)\n", num, sym->Name, line.FileName, line.LineNumber);
		else
			printf("\t[%i] 0x%p\n", num, stack.AddrPC.Offset);
	}

	SymCleanup(GetCurrentProcess());
}
} // namespace crown

#endif // CROWN_PLATFORM_WINDOWS