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;
|
||
}
|