mirror of
				https://github.com/SoftFever/OrcaSlicer.git
				synced 2025-11-02 20:51:23 -07:00 
			
		
		
		
	
		
			
				
	
	
		
			284 lines
		
	
	
		
			No EOL
		
	
	
		
			6.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			284 lines
		
	
	
		
			No EOL
		
	
	
		
			6.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
#pragma once
 | 
						|
#include <Windows.h>
 | 
						|
#include <tchar.h>
 | 
						|
#include <vector>
 | 
						|
 | 
						|
namespace textconv_helper
 | 
						|
{
 | 
						|
	// Forward declarations of our classes. They are defined later.
 | 
						|
	class CA2A_;
 | 
						|
	class CA2W_;
 | 
						|
	class CW2A_;
 | 
						|
	class CW2W_;
 | 
						|
	class CA2BSTR_;
 | 
						|
	class CW2BSTR_;
 | 
						|
 | 
						|
	// typedefs for the well known text conversions
 | 
						|
	typedef CA2W_ A2W_;
 | 
						|
	typedef CW2A_ W2A_;
 | 
						|
	//typedef CW2BSTR_ W2BSTR_;
 | 
						|
	//typedef CA2BSTR_ A2BSTR_;
 | 
						|
	typedef CW2A_ BSTR2A_;
 | 
						|
	typedef CW2W_ BSTR2W_;
 | 
						|
 | 
						|
#ifdef _UNICODE
 | 
						|
	typedef CA2W_ A2T_;
 | 
						|
	typedef CW2A_ T2A_;
 | 
						|
	typedef CW2W_ T2W_;
 | 
						|
	typedef CW2W_ W2T_;
 | 
						|
	//typedef CW2BSTR_ T2BSTR_;
 | 
						|
	//typedef BSTR2W_ BSTR2T_;
 | 
						|
#else
 | 
						|
	typedef CA2A_ A2T_;
 | 
						|
	typedef CA2A_ T2A_;
 | 
						|
	typedef CA2W_ T2W_;
 | 
						|
	typedef CW2A_ W2T_;
 | 
						|
	typedef CA2BSTR_ T2BSTR_;
 | 
						|
	typedef BSTR2A_ BSTR2T_;
 | 
						|
#endif
 | 
						|
 | 
						|
	typedef A2W_  A2OLE_;
 | 
						|
	typedef T2W_  T2OLE_;
 | 
						|
	typedef CW2W_ W2OLE_;
 | 
						|
	typedef W2A_  OLE2A_;
 | 
						|
	typedef W2T_  OLE2T_;
 | 
						|
	typedef CW2W_ OLE2W_;
 | 
						|
 | 
						|
	class CA2W_
 | 
						|
	{
 | 
						|
	public:
 | 
						|
		CA2W_(LPCSTR pStr, UINT codePage = CP_ACP) : m_pStr(pStr)
 | 
						|
		{
 | 
						|
			if (pStr)
 | 
						|
			{
 | 
						|
				// Resize the vector and assign null WCHAR to each element
 | 
						|
				int length = MultiByteToWideChar(codePage, 0, pStr, -1, NULL, 0) + 1;
 | 
						|
				m_vWideArray.assign(length, L'\0');
 | 
						|
 | 
						|
				// Fill our vector with the converted WCHAR array
 | 
						|
				MultiByteToWideChar(codePage, 0, pStr, -1, &m_vWideArray[0], length);
 | 
						|
			}
 | 
						|
		}
 | 
						|
		~CA2W_() {}
 | 
						|
		operator LPCWSTR() { return m_pStr ? &m_vWideArray[0] : NULL; }
 | 
						|
		//operator LPOLESTR() { return m_pStr ? (LPOLESTR)&m_vWideArray[0] : (LPOLESTR)NULL; }
 | 
						|
 | 
						|
	private:
 | 
						|
		CA2W_(const CA2W_&);
 | 
						|
		CA2W_& operator= (const CA2W_&);
 | 
						|
		std::vector<wchar_t> m_vWideArray;
 | 
						|
		LPCSTR m_pStr;
 | 
						|
	};
 | 
						|
 | 
						|
	class CW2A_
 | 
						|
	{
 | 
						|
	public:
 | 
						|
		CW2A_(LPCWSTR pWStr, UINT codePage = CP_ACP) : m_pWStr(pWStr)
 | 
						|
			// Usage:
 | 
						|
			//   CW2A_ ansiString(L"Some Text");
 | 
						|
			//   CW2A_ utf8String(L"Some Text", CP_UTF8);
 | 
						|
			//
 | 
						|
			// or
 | 
						|
			//   SetWindowTextA( W2A(L"Some Text") ); The ANSI version of SetWindowText
 | 
						|
		{
 | 
						|
			// Resize the vector and assign null char to each element
 | 
						|
			int length = WideCharToMultiByte(codePage, 0, pWStr, -1, NULL, 0, NULL, NULL) + 1;
 | 
						|
			m_vAnsiArray.assign(length, '\0');
 | 
						|
 | 
						|
			// Fill our vector with the converted char array
 | 
						|
			WideCharToMultiByte(codePage, 0, pWStr, -1, &m_vAnsiArray[0], length, NULL, NULL);
 | 
						|
		}
 | 
						|
 | 
						|
		~CW2A_()
 | 
						|
		{
 | 
						|
			m_pWStr = 0;
 | 
						|
		}
 | 
						|
		operator LPCSTR() { return m_pWStr ? &m_vAnsiArray[0] : NULL; }
 | 
						|
 | 
						|
	private:
 | 
						|
		CW2A_(const CW2A_&);
 | 
						|
		CW2A_& operator= (const CW2A_&);
 | 
						|
		std::vector<char> m_vAnsiArray;
 | 
						|
		LPCWSTR m_pWStr;
 | 
						|
	};
 | 
						|
 | 
						|
	class CW2W_
 | 
						|
	{
 | 
						|
	public:
 | 
						|
		CW2W_(LPCWSTR pWStr) : m_pWStr(pWStr) {}
 | 
						|
		operator LPCWSTR() { return const_cast<LPWSTR>(m_pWStr); }
 | 
						|
		//operator LPOLESTR() { return const_cast<LPOLESTR>(m_pWStr); }
 | 
						|
 | 
						|
	private:
 | 
						|
		CW2W_(const CW2W_&);
 | 
						|
		CW2W_& operator= (const CW2W_&);
 | 
						|
 | 
						|
		LPCWSTR m_pWStr;
 | 
						|
	};
 | 
						|
 | 
						|
	class CA2A_
 | 
						|
	{
 | 
						|
	public:
 | 
						|
		CA2A_(LPCSTR pStr) : m_pStr(pStr) {}
 | 
						|
		operator LPCSTR() { return (LPSTR)m_pStr; }
 | 
						|
 | 
						|
	private:
 | 
						|
		CA2A_(const CA2A_&);
 | 
						|
		CA2A_& operator= (const CA2A_&);
 | 
						|
 | 
						|
		LPCSTR m_pStr;
 | 
						|
	};
 | 
						|
 | 
						|
	/*class CW2BSTR_
 | 
						|
	{
 | 
						|
	public:
 | 
						|
		CW2BSTR_(LPCWSTR pWStr) { m_bstrString = ::SysAllocString(pWStr); }
 | 
						|
		~CW2BSTR_() { ::SysFreeString(m_bstrString); }
 | 
						|
		operator BSTR() { return m_bstrString; }
 | 
						|
 | 
						|
	private:
 | 
						|
		CW2BSTR_(const CW2BSTR_&);
 | 
						|
		CW2BSTR_& operator= (const CW2BSTR_&);
 | 
						|
		BSTR m_bstrString;
 | 
						|
	};
 | 
						|
 | 
						|
	class CA2BSTR_
 | 
						|
	{
 | 
						|
	public:
 | 
						|
		CA2BSTR_(LPCSTR pStr) { m_bstrString = ::SysAllocString(textconv_helper::CA2W_(pStr)); }
 | 
						|
		~CA2BSTR_() { ::SysFreeString(m_bstrString); }
 | 
						|
		operator BSTR() { return m_bstrString; }
 | 
						|
 | 
						|
	private:
 | 
						|
		CA2BSTR_(const CA2BSTR_&);
 | 
						|
		CA2BSTR_& operator= (const CA2BSTR_&);
 | 
						|
		BSTR m_bstrString;
 | 
						|
	};*/
 | 
						|
}
 | 
						|
 | 
						|
#define MAX_SYMBOL_PATH 1024
 | 
						|
#define MAX_MODULE_NAME32 255
 | 
						|
#define TH32CS_SNAPMODULE 0x00000008
 | 
						|
#define MAX_VERSION_LENGTH 512
 | 
						|
#define STACKWALK_MAX_NAMELEN 1024
 | 
						|
 | 
						|
#define ASSERT(judge)\
 | 
						|
	{\
 | 
						|
		if(!(judge))\
 | 
						|
		{\
 | 
						|
			DebugBreak();\
 | 
						|
		}\
 | 
						|
	}
 | 
						|
 | 
						|
typedef struct tagMODULEENTRY32
 | 
						|
{
 | 
						|
	DWORD   dwSize;
 | 
						|
	DWORD   th32ModuleID;  // This module
 | 
						|
	DWORD   th32ProcessID; // owning process
 | 
						|
	DWORD   GlblcntUsage;  // Global usage count on the module
 | 
						|
	DWORD   ProccntUsage;  // Module usage count in th32ProcessID's context
 | 
						|
	BYTE*   modBaseAddr;   // Base address of module in th32ProcessID's context
 | 
						|
	DWORD   modBaseSize;   // Size in bytes of module starting at modBaseAddr
 | 
						|
	HMODULE hModule;       // The hModule of this module in th32ProcessID's context
 | 
						|
	TCHAR   szModule[MAX_MODULE_NAME32 + 1];
 | 
						|
	TCHAR   szExePath[MAX_PATH];
 | 
						|
} MODULEENTRY32;
 | 
						|
 | 
						|
typedef struct _MODULEINFO
 | 
						|
{
 | 
						|
	LPVOID lpBaseOfDll;
 | 
						|
	DWORD  SizeOfImage;
 | 
						|
	LPVOID EntryPoint;
 | 
						|
} MODULEINFO, *LPMODULEINFO;
 | 
						|
 | 
						|
typedef MODULEENTRY32* PMODULEENTRY32;
 | 
						|
typedef MODULEENTRY32* LPMODULEENTRY32;
 | 
						|
 | 
						|
typedef struct _tag_MODULE_INFO 
 | 
						|
{
 | 
						|
	DWORD64 ModuleAddress;
 | 
						|
	DWORD dwModSize;
 | 
						|
	TCHAR szModuleName[MAX_MODULE_NAME32 + 1];
 | 
						|
	TCHAR szModulePath[MAX_PATH];
 | 
						|
	TCHAR szSymbolPath[MAX_PATH];
 | 
						|
	TCHAR szVersion[MAX_VERSION_LENGTH];
 | 
						|
	struct _tag_MODULE_INFO* pNext;
 | 
						|
}MODULE_INFO, *LPMODULE_INFO;
 | 
						|
 | 
						|
typedef struct tagSTACKINFO
 | 
						|
{
 | 
						|
	DWORD64 szFncAddr;
 | 
						|
	TCHAR szFileName[MAX_PATH];
 | 
						|
	TCHAR szFncName[MAX_PATH];
 | 
						|
	unsigned long uFileNum;
 | 
						|
	TCHAR    undName[STACKWALK_MAX_NAMELEN];
 | 
						|
	TCHAR    undFullName[STACKWALK_MAX_NAMELEN];
 | 
						|
	tagSTACKINFO *pNext;
 | 
						|
}STACKINFO, *LPSTACKINFO;
 | 
						|
 | 
						|
class CStackWalker
 | 
						|
{
 | 
						|
public:
 | 
						|
	CStackWalker(HANDLE hProcess = GetCurrentProcess(), WORD wPID = GetCurrentProcessId(), LPCTSTR lpSymbolPath = NULL);
 | 
						|
	~CStackWalker(void);
 | 
						|
	BOOL LoadSymbol();
 | 
						|
	LPMODULE_INFO GetLoadModules();
 | 
						|
	void GetModuleInformation(LPMODULE_INFO pmi);
 | 
						|
 | 
						|
	void FreeModuleInformations(LPMODULE_INFO pmi);
 | 
						|
	virtual void OutputString(LPCTSTR lpszFormat, ...);
 | 
						|
 | 
						|
	LPSTACKINFO StackWalker(HANDLE hThread = GetCurrentThread(), const CONTEXT* context = NULL);
 | 
						|
	void FreeStackInformations(LPSTACKINFO psi);
 | 
						|
 | 
						|
protected:
 | 
						|
	LPMODULE_INFO GetModulesTH32();
 | 
						|
	LPMODULE_INFO GetModulesPSAPI();
 | 
						|
protected:
 | 
						|
	HANDLE m_hProcess;
 | 
						|
	WORD m_wPID;
 | 
						|
	LPTSTR m_lpszSymbolPath;
 | 
						|
	BOOL m_bSymbolLoaded;
 | 
						|
};
 | 
						|
 | 
						|
#if defined(_M_IX86)
 | 
						|
#ifdef CURRENT_THREAD_VIA_EXCEPTION
 | 
						|
#define GET_CURRENT_THREAD_CONTEXT(c, contextFlags)\
 | 
						|
do\
 | 
						|
{\
 | 
						|
	memset(&c, 0, sizeof(CONTEXT));\
 | 
						|
	EXCEPTION_POINTERS* pExp = NULL;\
 | 
						|
	__try\
 | 
						|
	{\
 | 
						|
		throw 0;\
 | 
						|
	}\
 | 
						|
	__except (((pExp = GetExceptionInformation()) ? EXCEPTION_EXECUTE_HANDLER:EXCEPTION_EXECUTE_HANDLER))\
 | 
						|
	{\
 | 
						|
	}\
 | 
						|
	if (pExp != NULL)\
 | 
						|
		memcpy(&c, pExp->ContextRecord, sizeof(CONTEXT));\
 | 
						|
	c.ContextFlags = contextFlags;\
 | 
						|
} while (0);
 | 
						|
#else
 | 
						|
#define GET_CURRENT_THREAD_CONTEXT(c, contextFlags) \
 | 
						|
	do\
 | 
						|
	{\
 | 
						|
		memset(&c, 0, sizeof(CONTEXT));\
 | 
						|
		c.ContextFlags = contextFlags;\
 | 
						|
		__asm    call $+5\
 | 
						|
		__asm    pop eax\
 | 
						|
		__asm    mov c.Eip, eax\
 | 
						|
		__asm    mov c.Ebp, ebp\
 | 
						|
		__asm    mov c.Esp, esp\
 | 
						|
} while (0)
 | 
						|
#endif
 | 
						|
 | 
						|
#else
 | 
						|
#define GET_CURRENT_THREAD_CONTEXT(c, contextFlags) \
 | 
						|
	do\
 | 
						|
	{ \
 | 
						|
		memset(&c, 0, sizeof(CONTEXT));\
 | 
						|
		c.ContextFlags = contextFlags;\
 | 
						|
		RtlCaptureContext(&c);\
 | 
						|
} while (0);
 | 
						|
#endif |