// Copyright 2011-2019 Molecular Matters GmbH, all rights reserved.

#include "LC_SymbolResolution.h"
#include "LC_PointerUtil.h"
#include "LC_Logging.h"
#include "Windows/WindowsHWrapper.h"

#pragma push_macro("OPTIONAL")
#ifndef OPTIONAL
#define OPTIONAL
#endif
#include <dbghelp.h>
#pragma pop_macro("OPTIONAL")

namespace symbolResolution
{
	void Startup(void)
	{
		::SymSetOptions(SYMOPT_DEFERRED_LOADS | SYMOPT_LOAD_LINES | SYMOPT_UNDNAME | SYMOPT_INCLUDE_32BIT_MODULES);
		const BOOL success = ::SymInitialize(::GetCurrentProcess(), "", true);
		if (success == Windows::FALSE)
		{
			const DWORD error = ::GetLastError();
			LC_ERROR_DEV("SymInitialize failed with error 0x%X", error);
		}
	}


	void Shutdown(void)
	{
		const BOOL success = ::SymCleanup(::GetCurrentProcess());
		if (success == Windows::FALSE)
		{
			const DWORD error = ::GetLastError();
			LC_ERROR_DEV("SymCleanup failed with error 0x%X", error);
		}
	}


	SymbolInfo ResolveSymbolsForAddress(const void* const address)
	{
		// retrieve function
		DWORD64 displacement64 = 0;
		ULONG64 buffer[(sizeof(SYMBOL_INFO) + MAX_SYM_NAME * sizeof(TCHAR) + sizeof(ULONG64) - 1) / sizeof(ULONG64)] = {};
		SYMBOL_INFO* symbolInfo = reinterpret_cast<SYMBOL_INFO*>(buffer);
		symbolInfo->SizeOfStruct = sizeof(SYMBOL_INFO);
		symbolInfo->MaxNameLen = MAX_SYM_NAME;

		const HANDLE process = ::GetCurrentProcess();
		const DWORD64 intAddress = pointer::AsInteger<DWORD64>(address);

		if (::SymFromAddr(process, intAddress, &displacement64, symbolInfo))
		{
			// retrieve filename and line number
			DWORD displacement = 0;
			IMAGEHLP_LINE64 line = {};
			line.SizeOfStruct = sizeof(IMAGEHLP_LINE64);

			if (::SymGetLineFromAddr64(process, intAddress, &displacement, &line))
				return SymbolInfo(symbolInfo->Name, line.FileName, line.LineNumber);
		}

		return SymbolInfo("unknown", "unknown", 0);
	}
}