#include "stdafx.h"
|
#include "Context.h"
|
#include <dbghelp.h>
|
|
#pragma comment(lib, "dbghelp.lib")
|
|
|
void PrintStackTrace()
|
{
|
void* stack[62]; // ×î¶à×¥62²ã
|
USHORT frames = CaptureStackBackTrace(0, 62, stack, NULL);
|
|
HANDLE process = GetCurrentProcess();
|
SymInitialize(process, NULL, TRUE);
|
for (USHORT i = 0; i < frames; ++i) {
|
DWORD64 address = (DWORD64)(stack[i]);
|
char buffer[sizeof(SYMBOL_INFO) + MAX_SYM_NAME * sizeof(TCHAR)];
|
PSYMBOL_INFO symbol = (PSYMBOL_INFO)buffer;
|
symbol->SizeOfStruct = sizeof(SYMBOL_INFO);
|
symbol->MaxNameLen = MAX_SYM_NAME;
|
|
if (SymFromAddr(process, address, 0, symbol)) {
|
TRACE(">>>> name:%s\n", symbol->Name);
|
}
|
else {
|
TRACE(">>>> name:%s\n", "SymFromAddr failed");
|
}
|
}
|
|
SymCleanup(process);
|
}
|
CContext::CContext()
|
{
|
m_onReleaseCallback = nullptr;
|
m_nRef = 0;
|
m_nRetCode = 0;
|
m_hEvent = ::CreateEvent(NULL, TRUE, FALSE, NULL);
|
InitializeCriticalSection(&m_criticalSection);
|
}
|
|
|
CContext::~CContext()
|
{
|
if (m_hEvent != NULL) {
|
CloseHandle(m_hEvent);
|
m_hEvent = NULL;
|
}
|
DeleteCriticalSection(&m_criticalSection);
|
}
|
|
void CContext::setOnRelease(ONRELEASECALLBACK fOnRelease)
|
{
|
m_onReleaseCallback = fOnRelease;
|
}
|
|
int CContext::addRef()
|
{
|
Lock();
|
m_nRef++;
|
Unlock();
|
|
return m_nRef;
|
}
|
|
int CContext::release()
|
{
|
Lock();
|
if (m_nRef > 0) {
|
m_nRef--;
|
BOOL bRefCount0 = m_nRef == 0;
|
Unlock();
|
|
if (bRefCount0 && m_onReleaseCallback != nullptr) {
|
m_onReleaseCallback(this);
|
}
|
}
|
else {
|
Unlock();
|
}
|
|
return m_nRef;
|
}
|
|
void CContext::setRetCode(int code)
|
{
|
m_nRetCode = code;
|
}
|
|
int CContext::getRetCode()
|
{
|
return m_nRetCode;
|
}
|
|
void CContext::setRetMsg(const char* pszMsg)
|
{
|
m_strRetMsg = pszMsg;
|
}
|
|
std::string& CContext::getRetMsg()
|
{
|
return m_strRetMsg;
|
}
|
|
HANDLE CContext::getEventHandle()
|
{
|
return m_hEvent;
|
}
|
|
void CContext::setEvent()
|
{
|
if (m_hEvent != NULL) {
|
SetEvent(m_hEvent);
|
}
|
}
|
|
void CContext::resetEvent()
|
{
|
if (m_hEvent != NULL) {
|
ResetEvent(m_hEvent);
|
}
|
}
|
|
void CContext::ReadString(CArchive& ar, std::string& string)
|
{
|
CString strTemp;
|
ar >> strTemp;
|
string = (LPTSTR)(LPCTSTR)strTemp;
|
}
|
|
void CContext::WriteString(CArchive& ar, std::string& string)
|
{
|
CString strTemp = string.c_str();
|
ar << strTemp;
|
}
|
|
void CContext::createDir(CString strDir)
|
{
|
int lastIndex = 0;
|
int index = strDir.Find(_T("\\"), lastIndex);
|
while (index > 0) {
|
CString strTempDir = strDir.Left(index);
|
CreateDirectory(strTempDir, NULL);
|
|
lastIndex = index + 1;
|
index = strDir.Find(_T("\\"), lastIndex);
|
}
|
CreateDirectory(strDir, NULL);
|
}
|
|
BOOL CContext::IsBoxId(CString& strId)
|
{
|
//'W20'¿ªÍ·£¬µ¹ÊýµÚÎåλΪ'Z'µÄ×Ö·û´®
|
if (strId.GetLength() < 8) {
|
return FALSE;
|
}
|
|
if (strId.Left(3).Compare(_T("W20")) != 0) {
|
return FALSE;
|
}
|
|
if (strId.Right(5).GetAt(0) != (char)'Z') {
|
return FALSE;
|
}
|
|
return TRUE;
|
}
|
|
BOOL CContext::IsLotId(CString& strId)
|
{
|
//'N20'¿ªÍ·£¬µ¹ÊýµÚÎåλΪ'Z'µÄ×Ö·û´®
|
if (strId.GetLength() < 8) {
|
return FALSE;
|
}
|
|
if (strId.Left(3).Compare(_T("N20")) != 0) {
|
return FALSE;
|
}
|
|
if (strId.Right(5).GetAt(0) != (char)'Z') {
|
return FALSE;
|
}
|
|
return TRUE;
|
}
|
