一个支持自动抛出异常的宏

freeeyes

写了一个简单的错误捕捉的宏。用于简单的错误排查，这个类会自动输出当前出错的文件名，函数名，感觉在DEBUG的时候挺有用的。在这里记录一下，很简单的一个.h，使用很简单。windows和linux下都能运行。
建立一个define.h

1. #ifndef _DEFIINE_H
2. #define _DEFIINE_H
4. #include <stdio.h>
5. #include <stdarg.h>
6. #include <string.h>
7. #include <errno.h>
8. #include <time.h>
10. #define SHOW_PRINT    1
11. #define SHOW_WRITELOG 2
13. //nType = 1为显示在桌面上，2为记录为日志
14. void __show__(const char* pMsg, int nType)
15. {
16.         if(nType == 1)
17.         {
18.                 printf("[Error]%s", pMsg);
19.         }
21.         if(nType == 2)
22.         {
23.                 FILE* pFile = fopen("assert.log", "a");
24.                 if(pFile != NULL)
25.                 {
26.                         fwrite(pMsg, 1, strlen(pMsg), pFile);
27.                         fwrite("\r\n", 1, 2, pFile);
28.                         fclose(pFile);
29.                 }
30.                 else
31.                 {
32.                         printf("[Error]file open Error(%s).\n", strerror(errno));
33.                 }
34.         }
35. }
37. void sprintf_safe(char* szText, int nLen, char* fmt ...)
38. {
39.         if(szText == NULL)
40.         {
41.                 return;
42.         }
44.         va_list ap;
45.         va_start(ap, fmt);
46.         vsnprintf(szText, nLen, fmt, ap);
47.         va_end(ap);
48. }
50. void __assertspecial__(const char* pFile, int nLen, const char* pFuncName, const char* pExpr, const char* pMsg)
51. {
52.         char szTemp[1024] = {'\0'};
53.         char szDate[50] = {'\0'};
55.         struct tm *local;
56.         time_t t;
57.         t = time(NULL);
58.         local = localtime(&t);
59.         sprintf_safe(szDate, 50, "%04d-%02d-%02d %02d:%02d:%02d", local->tm_year + 1900, local->tm_mon + 1, local->tm_mday, local->tm_hour, local->tm_min, local->tm_sec);
61.         sprintf_safe(szTemp, 1024, "[%s](%s)(%d)(%s)(%s)(%s)\n", szDate, pFile, nLen, pFuncName, pExpr, pMsg);
63.         __show__(szTemp, SHOW_WRITELOG);
64. }
66. #ifdef WIN32
67.         #define AssertSpecial(type, msg) {if(type == false) { __assertspecial__(__FILE__, __LINE__, __FUNCTION__, #type, msg);} }
68. #else
69.         #define AssertSpecial(type, msg) {if(type == false) { __assertspecial__(__FILE__, __LINE__, __PRETTY_FUNCTION__, #type, msg);} }
70. #endif
72. #define _ENTER_FUNCTION {try{
74. #ifdef WIN32
75.         #define _LEAVE_FUNCTION }catch(...) { AssertSpecial(false, __FUNCTION__); }}
76. #else
77.         #define _LEAVE_FUNCTION }catch(...) { AssertSpecial(false, __PRETTY_FUNCTION__); }}
78. #endif
80. #endif

复制代码

测试一下：

2. int main(int argc, char* argv[])
3. {
5. printf("[main]Begin.\n");
7. _ENTER_FUNCTION
10. throw 1;
13. _LEAVE_FUNCTION
15. return 0;
16. }

复制代码

输出日志:
[2011-02-10 15:32:49](testerror.cpp)(14)(main)(false)(main)

其实这个可以稍微再改一下，支持一下msg，不过对于不复杂的错误检测，差不多够了，有助于查错。