379 lines
8.0 KiB
C++
379 lines
8.0 KiB
C++
|
#include <stdarg.h>
|
|||
|
|
#include <stdio.h>
|
|||
|
|
|
|||
|
|
#ifdef WIN32
|
|||
|
|
#include <Windows.h>
|
|||
|
|
#include <dbghelp.h>
|
|||
|
|
#include <TCHAR.h>
|
|||
|
|
#include <locale.h>
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
#include "_osdef.h"
|
|||
|
|
#include <Tick.h>
|
|||
|
|
#include "Lock.h"
|
|||
|
|
|
|||
|
|
#include <string.h>
|
|||
|
|
|
|||
|
|
#include "ShareUtil.h"
|
|||
|
|
|
|||
|
|
//RTL_CRITICAL_SECTION g_OutputMsgLock;
|
|||
|
|
wylib::sync::lock::CCSLock g_OutputMsgLock;
|
|||
|
|
|
|||
|
|
SHAREOUTPUTMSGFN g_lpOutputMsg;
|
|||
|
|
static TCHAR sTempMsgBuf[8192 * 4];
|
|||
|
|
#define TMPBUFSIZE (sizeof(sTempMsgBuf)/sizeof(sTempMsgBuf[0]))
|
|||
|
|
|
|||
|
|
|
|||
|
|
#ifdef WIN32
|
|||
|
|
|
|||
|
|
static int wExceptionDumpFlag = -1 ; //这个是生成dump的标记
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
void SetMiniDumpFlag(int nFlag)
|
|||
|
|
{
|
|||
|
|
wExceptionDumpFlag =nFlag;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
LONG WINAPI DefaultUnHandleExceptionFilter(struct _EXCEPTION_POINTERS *ExceptionInfo)
|
|||
|
|
{
|
|||
|
|
TCHAR sDumpFileName[256];
|
|||
|
|
_tcscpy(sDumpFileName,szExceptionDumpFile);
|
|||
|
|
INT_PTR nDotPos = _tcslen(sDumpFileName);
|
|||
|
|
for(; nDotPos>=0; nDotPos--)
|
|||
|
|
{
|
|||
|
|
if(sDumpFileName[nDotPos] =='.')
|
|||
|
|
{
|
|||
|
|
break;;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if(nDotPos <0)
|
|||
|
|
{
|
|||
|
|
nDotPos =0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
SYSTEMTIME Systime;
|
|||
|
|
GetLocalTime( &Systime );
|
|||
|
|
|
|||
|
|
_stprintf(&sDumpFileName[nDotPos], _T("-%d-%d-%d-%d-%d-%d"), Systime.wYear - 2000, Systime.wMonth, Systime.wDay, Systime.wHour, Systime.wMinute, Systime.wSecond );
|
|||
|
|
_tcscat(sDumpFileName, _T(".dmp"));
|
|||
|
|
|
|||
|
|
HANDLE hFile = CreateFile( sDumpFileName, GENERIC_READ | GENERIC_WRITE,
|
|||
|
|
0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0 );
|
|||
|
|
MINIDUMP_TYPE wDumpFlag = MiniDumpWithFullMemory;
|
|||
|
|
if(wExceptionDumpFlag >=0)
|
|||
|
|
{
|
|||
|
|
wDumpFlag = (MINIDUMP_TYPE)wExceptionDumpFlag;
|
|||
|
|
}
|
|||
|
|
if ( hFile != INVALID_HANDLE_VALUE )
|
|||
|
|
{
|
|||
|
|
MINIDUMP_EXCEPTION_INFORMATION ExInfo;
|
|||
|
|
|
|||
|
|
ExInfo.ThreadId = GetCurrentThreadId();
|
|||
|
|
ExInfo.ExceptionPointers = ExceptionInfo;
|
|||
|
|
ExInfo.ClientPointers = NULL;
|
|||
|
|
|
|||
|
|
MiniDumpWriteDump(GetCurrentProcess(), GetCurrentProcessId(),
|
|||
|
|
hFile, wDumpFlag, &ExInfo, NULL, NULL );
|
|||
|
|
CloseHandle(hFile);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
ExitProcess( -1 );
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
LPCTSTR GetSysErrorMessage(const INT_PTR ErrorCode,OUT LPTSTR sBuffer, size_t dwBufferSize, size_t *dwBufferNeeded)
|
|||
|
|
{
|
|||
|
|
DWORD dwLen;
|
|||
|
|
TCHAR Buf[2048];
|
|||
|
|
|
|||
|
|
dwLen = FormatMessage( FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS |
|
|||
|
|
FORMAT_MESSAGE_ARGUMENT_ARRAY, NULL, (DWORD)ErrorCode, 0, Buf, sizeof(Buf), NULL );
|
|||
|
|
while ( dwLen > 0 )
|
|||
|
|
{
|
|||
|
|
if ( (BYTE)Buf[dwLen-1] <= 0x32 )
|
|||
|
|
dwLen--;
|
|||
|
|
else break;
|
|||
|
|
}
|
|||
|
|
Buf[dwLen] = 0;
|
|||
|
|
if ( sBuffer )
|
|||
|
|
{
|
|||
|
|
if ( dwBufferSize > dwLen )
|
|||
|
|
{
|
|||
|
|
memcpy( sBuffer, Buf, dwLen * sizeof(Buf[0]) );
|
|||
|
|
sBuffer[dwLen] = 0;
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
memcpy( sBuffer, Buf, dwBufferSize * sizeof(Buf[0]) );
|
|||
|
|
sBuffer[dwBufferSize-1] = 0;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
*dwBufferNeeded = dwLen;
|
|||
|
|
return sBuffer;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
int SNPRINTFA(char* dst, int len, const char* format, ...)
|
|||
|
|
{
|
|||
|
|
if (NULL == dst || len <= 0 || NULL == format) return -1;
|
|||
|
|
|
|||
|
|
va_list va;
|
|||
|
|
va_start(va, format);
|
|||
|
|
|
|||
|
|
int ret = vsnprintf(dst, len, format, va);
|
|||
|
|
|
|||
|
|
if (ret >= len) ret = -1;
|
|||
|
|
|
|||
|
|
va_end(va);
|
|||
|
|
|
|||
|
|
dst[len - 1] = '\0';
|
|||
|
|
|
|||
|
|
return ret;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int VSNPRINTFA(char* dst, int len, const char* format, va_list args)
|
|||
|
|
{
|
|||
|
|
if (NULL == dst || len <= 0 || NULL == format) return -1;
|
|||
|
|
|
|||
|
|
int ret = vsnprintf(dst, len, format, args);
|
|||
|
|
|
|||
|
|
if (ret >= len) ret = -1;
|
|||
|
|
|
|||
|
|
dst[len - 1] = '\0';
|
|||
|
|
|
|||
|
|
return ret;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 本函数代码出自 openbsd 的 strlcpy.c 文件，用于非bsd并且不提供strlcpy函数的系统
|
|||
|
|
size_t
|
|||
|
|
strlcpy(char *dst, const char *src, size_t siz)
|
|||
|
|
{
|
|||
|
|
char *d = dst;
|
|||
|
|
const char *s = src;
|
|||
|
|
size_t n = siz;
|
|||
|
|
|
|||
|
|
/* Copy as many bytes as will fit */
|
|||
|
|
if (n != 0) {
|
|||
|
|
while (--n != 0) {
|
|||
|
|
if ((*d++ = *s++) == '\0')
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/* Not enough room in dst, add NUL and traverse rest of src */
|
|||
|
|
if (n == 0) {
|
|||
|
|
if (siz != 0)
|
|||
|
|
*d = '\0'; /* NUL-terminate dst */
|
|||
|
|
while (*s++)
|
|||
|
|
;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return(s - src - 1); /* count does not include NUL */
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 本函数代码出自 openbsd 的 strlcat.c 文件，用于非bsd并且不提供strlcat函数的系统
|
|||
|
|
size_t
|
|||
|
|
strlcat(char *dst, const char *src, size_t siz)
|
|||
|
|
{
|
|||
|
|
char *d = dst;
|
|||
|
|
const char *s = src;
|
|||
|
|
size_t n = siz;
|
|||
|
|
size_t dlen;
|
|||
|
|
|
|||
|
|
/* Find the end of dst and adjust bytes left but don't go past end */
|
|||
|
|
while (n-- != 0 && *d != '\0')
|
|||
|
|
d++;
|
|||
|
|
dlen = d - dst;
|
|||
|
|
n = siz - dlen;
|
|||
|
|
|
|||
|
|
if (n == 0)
|
|||
|
|
return(dlen + strlen(s));
|
|||
|
|
while (*s != '\0') {
|
|||
|
|
if (n != 1) {
|
|||
|
|
*d++ = *s;
|
|||
|
|
n--;
|
|||
|
|
}
|
|||
|
|
s++;
|
|||
|
|
}
|
|||
|
|
*d = '\0';
|
|||
|
|
|
|||
|
|
return(dlen + (s - src)); /* count does not include NUL */
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
void GetSystemTime(SystemTime& sysTime)
|
|||
|
|
{
|
|||
|
|
#ifdef _MSC_VER
|
|||
|
|
SYSTEMTIME nowSysTime; // 服务器的开启时间
|
|||
|
|
GetLocalTime(&nowSysTime);
|
|||
|
|
sysTime.sec_ = nowSysTime.wSecond;
|
|||
|
|
sysTime.min_ = nowSysTime.wMinute;
|
|||
|
|
sysTime.hour_ = nowSysTime.wHour;
|
|||
|
|
sysTime.mday_ = nowSysTime.wDay;
|
|||
|
|
sysTime.mon_ = nowSysTime.wMonth;
|
|||
|
|
sysTime.year_ = nowSysTime.wYear;
|
|||
|
|
sysTime.wday_ = nowSysTime.wDayOfWeek;
|
|||
|
|
#else
|
|||
|
|
time_t lcurtime;
|
|||
|
|
time(&lcurtime);
|
|||
|
|
struct tm curTime;
|
|||
|
|
localtime_r(&lcurtime,&curTime);
|
|||
|
|
sysTime.sec_ = curTime.tm_sec;
|
|||
|
|
sysTime.min_ = curTime.tm_min;
|
|||
|
|
sysTime.hour_ = curTime.tm_hour;
|
|||
|
|
sysTime.mday_ = curTime.tm_mday;
|
|||
|
|
sysTime.mon_ = curTime.tm_mon + 1;
|
|||
|
|
sysTime.year_ = curTime.tm_year + 1900;
|
|||
|
|
sysTime.wday_ = curTime.tm_wday;
|
|||
|
|
#endif
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//默认消息输出函数
|
|||
|
|
INT_PTR STDCALL StdDefOutputMsg(SHAREOUTPUTMSGTYPE MsgType, LPCTSTR lpMsg, INT_PTR nMsgLen)
|
|||
|
|
{
|
|||
|
|
SYSTEMTIME Systime;
|
|||
|
|
INT_PTR Result;
|
|||
|
|
|
|||
|
|
GetLocalTime( &Systime );
|
|||
|
|
Result = _tprintf( _T("[%d-%d-%d %d:%d:%d]"), Systime.wYear - 2000, Systime.wMonth, Systime.wDay, Systime.wHour, Systime.wMinute, Systime.wSecond );
|
|||
|
|
switch ( MsgType )
|
|||
|
|
{
|
|||
|
|
case rmWaning: Result += _tprintf( _T("[WRN]:") ); break;
|
|||
|
|
case rmTip: Result += _tprintf( _T("[TIP]:") ); break;
|
|||
|
|
case rmError: Result += _tprintf( _T("[ERR]:") ); break;
|
|||
|
|
default: Result += _tprintf( _T(":") ); break;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Result += (int)fwrite( lpMsg, 1, nMsgLen * sizeof(*lpMsg), stdout );
|
|||
|
|
// Result += (int)fwrite( _T("\n"), 1, 1 * sizeof(*lpMsg), stdout );
|
|||
|
|
_tprintf( _T("%s\n"), lpMsg );
|
|||
|
|
|
|||
|
|
return Result;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
//全局消息输出函数
|
|||
|
|
INT_PTR STDCALL OutputMsg(SHAREOUTPUTMSGTYPE MsgType, LPCTSTR sFormat, ...)
|
|||
|
|
{
|
|||
|
|
va_list args;
|
|||
|
|
INT_PTR Result = 0;
|
|||
|
|
if (g_lpOutputMsg)
|
|||
|
|
{
|
|||
|
|
//EnterCriticalSection( &g_OutputMsgLock );
|
|||
|
|
g_OutputMsgLock.Lock();
|
|||
|
|
va_start(args, sFormat);
|
|||
|
|
Result = _vsntprintf(sTempMsgBuf, TMPBUFSIZE - 1, sFormat, args);
|
|||
|
|
va_end(args);
|
|||
|
|
|
|||
|
|
Result = g_lpOutputMsg( MsgType, sTempMsgBuf, Result );
|
|||
|
|
|
|||
|
|
//LeaveCriticalSection( &g_OutputMsgLock );
|
|||
|
|
g_OutputMsgLock.Unlock();
|
|||
|
|
}
|
|||
|
|
return Result;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
INT _StrToInt(LPCTSTR sText)
|
|||
|
|
{
|
|||
|
|
int nResult = 0;
|
|||
|
|
_stscanf(sText, _T("%d"), &nResult);
|
|||
|
|
return nResult;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
INT64 StrToInt64(LPCTSTR sText)
|
|||
|
|
{
|
|||
|
|
INT64 nResult = 0;
|
|||
|
|
_stscanf(sText, _T("%lld"), &nResult);//Unix中的libc库使用%lld
|
|||
|
|
return nResult;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
void STDCALL UninitDefMsgOut()
|
|||
|
|
{
|
|||
|
|
//DeleteCriticalSection( &g_OutputMsgLock );
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//设置全局消息输出函数,返回当前的消息输出函数
|
|||
|
|
SHAREOUTPUTMSGFN STDCALL SetOutputMsgFn(SHAREOUTPUTMSGFN lpFn)
|
|||
|
|
{
|
|||
|
|
//EnterCriticalSection( &g_OutputMsgLock );
|
|||
|
|
g_OutputMsgLock.Lock();
|
|||
|
|
SHAREOUTPUTMSGFN lpCurFn = g_lpOutputMsg;
|
|||
|
|
g_lpOutputMsg = lpFn;
|
|||
|
|
//LeaveCriticalSection( &g_OutputMsgLock );
|
|||
|
|
g_OutputMsgLock.Unlock();
|
|||
|
|
return lpCurFn;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
void STDCALL InitDefMsgOut()
|
|||
|
|
{
|
|||
|
|
//InitializeCriticalSection( &g_OutputMsgLock );
|
|||
|
|
|
|||
|
|
g_lpOutputMsg = StdDefOutputMsg;
|
|||
|
|
|
|||
|
|
//设置CTYPE为本地代码页，防止出现无法打印UNICODE 255之后的字符的问题。
|
|||
|
|
#ifdef WIN32
|
|||
|
|
char sACP[16];
|
|||
|
|
sprintf(sACP, ".%d", GetACP());
|
|||
|
|
setlocale(LC_CTYPE, sACP);
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
INT_PTR STDCALL OutputError(const INT_PTR nErrorCode, LPCTSTR sFormat, ...)
|
|||
|
|
{
|
|||
|
|
INT_PTR Result = 0;
|
|||
|
|
LPTSTR sptr;
|
|||
|
|
size_t dwLen;
|
|||
|
|
va_list args;
|
|||
|
|
|
|||
|
|
if (g_lpOutputMsg)
|
|||
|
|
{
|
|||
|
|
//EnterCriticalSection( &g_OutputMsgLock );
|
|||
|
|
g_OutputMsgLock.Lock();
|
|||
|
|
sptr = sTempMsgBuf;
|
|||
|
|
|
|||
|
|
va_start(args, sFormat);
|
|||
|
|
sptr += _vsntprintf(sTempMsgBuf, TMPBUFSIZE - 1, sFormat, args);
|
|||
|
|
va_end(args);
|
|||
|
|
|
|||
|
|
dwLen = TMPBUFSIZE - 1 - (sptr - sTempMsgBuf);
|
|||
|
|
sptr += _sntprintf( sptr, dwLen, _T(" %d "), (int)nErrorCode );
|
|||
|
|
|
|||
|
|
Result = TMPBUFSIZE - 1 - (sptr - sTempMsgBuf);
|
|||
|
|
|
|||
|
|
#ifdef WIN32
|
|||
|
|
GetSysErrorMessage( nErrorCode, sptr, Result, &dwLen );
|
|||
|
|
#else
|
|||
|
|
dwLen =0;
|
|||
|
|
#endif
|
|||
|
|
sptr += __min((INT_PTR)dwLen, Result);
|
|||
|
|
|
|||
|
|
Result = DWORD(sptr - sTempMsgBuf);
|
|||
|
|
Result = g_lpOutputMsg( rmError, sTempMsgBuf, Result );
|
|||
|
|
|
|||
|
|
//LeaveCriticalSection( &g_OutputMsgLock );
|
|||
|
|
g_OutputMsgLock.Unlock();
|
|||
|
|
}
|
|||
|
|
return Result;
|
|||
|
|
}
|