使用ATLTRACE打造轻量级DEBUG LOG

来源：互联网发布：淘宝网儿童服装女装编辑：程序博客网时间：2024/06/07 20:21

原文转载自：http://0cch.net/wordpress/?p=457

作者:occh 1363142043-59

众所周知，Debug Log是非常好的调试手段。所以我经常也尝试各种各样的第三方Log库。Log库分很多类型，例如可以给服务器使用的功能完备Log，也有轻量级的Log库，只是为Debug所设计。作为客户端开发，我还是比较喜欢后者这种Log库。不过使用第三方库有一个这样的麻烦事，走到哪你都得下一个，然后添加到自己的代码里。对于Log这样的功能，几乎所有程序都是需要的，使用的极其频繁。所以我就想找到一种方法，它可以使用SDK现有功能，来完成一个轻量级Log的功能。对我来说，不需要这个Log有多么高效，完备，唯一需要的就是方便，拿来就可以用。

结合这些目的，我第一个想到的就是ATL的ATLTRACE。但是，ATLTRACE输出的日志都是显示在Debug Output窗口。如果想将信息输出到文件或者控制台上，这就够呛了。那么，接下来就要想办法改变ATLTRACE的输出设备了。由于ATL是有代码的，所以很容易的可以看到代码运行的脉络。看完这份代码的第一个收获就是知道了ATLTRACE运行效率不会很高，不过这个对我来说并不重要。另外一个就是，找到了改变输出设备的方法。

在没有定义_ATL_NO_DEBUG_CRT的情况下，ATLTRACE最终的输出是通过_CrtDbgReport实现的，而如果定义了这个宏，那么输出是直接调用OutputDebugString。但一般程序都不会使用_ATL_NO_DEBUG_CRT这个宏，所以大部分情况下ATLTRACE都是调用的_CrtDbgReport。那么办法就来了_CrtDbgReport输出的数据，是可以通过_CrtSetReportMode和_CrtSetReportFile来改变输出设备的。例如我们想输出到控制台，我们只需要这样：
_CrtSetReportMode(_CRT_WARN, _CRTDBG_MODE_FILE);
_CrtSetReportFile(_CRT_WARN, _CRTDBG_FILE_STDOUT);

如果要输出到文件也只需要这样：
HANDLE log_file;
log_file = CreateFile(“c:\\log.txt”, GENERIC_WRITE,
FILE_SHARE_WRITE, NULL, CREATE_ALWAYS,
FILE_ATTRIBUTE_NORMAL, NULL);
_CrtSetReportFile(_CRT_WARN, log_file);
CloseHandle(log_file);
或者
freopen( “c:\\log2.txt”, “w”, stdout);
_CrtSetReportMode(_CRT_ERROR, _CRTDBG_MODE_FILE);
_CrtSetReportFile(_CRT_ERROR, _CRTDBG_FILE_STDOUT);

好了，这样就能解决输出设备的问题。既然已经说到这里，继续介绍下ATLTRACE很少人知道的其他优点吧。
1.可以通过ATL/MFT TRACE Tool 随时设定Log的输出Filter，并且可以保持配置（工具用法很简单，具体直接用用就知道了）。
2.通过AtlDebugAPI的接口，可以给自己的代码中添加读取配置文件的函数。这样每次修改配置文件就能改变ATLTRACE的行为。
3.通过AtlDebugAPI的接口，可以直接制定输出内容，不用配置文件也可以。
这三条涉及到的接口有：
AtlTraceOpenProcess
AtlTraceModifyProcess
AtlTraceCloseProcess
AtlTraceLoadSettings

为了更方便使用，我这写了几个宏代码如下：

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
#include <atlbase.h>
#include <atltrace.h>
#include <atldebugapi.h>
#include <atlpath.h>
 
#define TRACEHELPA(fmt, ...)    \
do                             \
{                               \
    SYSTEMTIMEtm;              \
    GetLocalTime(&tm);          \
    ATLTRACE("%s: [%02d-%02d-%02d %02d:%02d:%02d:%03d] "fmt, __FUNCTION__,                          \
    tm.wYear,tm.wMonth,tm.wDay,tm.wHour,tm.wMinute,tm.wSecond,tm.wMilliseconds, __VA_ARGS__); \
}while(0)
 
#define TRACEHELPW(fmt, ...)    \
do                             \
{                               \
    SYSTEMTIMEtm;              \
    GetLocalTime(&tm);          \
    ATLTRACE(L"%S: [%02d-%02d-%02d %02d:%02d:%02d:%03d] "fmt, __FUNCTION__,                         \
    tm.wYear,tm.wMonth,tm.wDay,tm.wHour,tm.wMinute,tm.wSecond,tm.wMilliseconds, __VA_ARGS__); \
}while(0)
 
#define TRACEHELPEXA(category, level, fmt, ...) \
do                             \
{                               \
    SYSTEMTIMEtm;              \
    GetLocalTime(&tm);          \
    ATLTRACE(category, level, "%s: [%02d-%02d-%02d %02d:%02d:%02d:%03d] "fmt, __FUNCTION__,         \
    tm.wYear,tm.wMonth,tm.wDay,tm.wHour,tm.wMinute,tm.wSecond,tm.wMilliseconds, __VA_ARGS__); \
}while(0)
 
#define TRACEHELPEXW(category, level, fmt, ...) \
do                             \
{                               \
    SYSTEMTIMEtm;              \
    GetLocalTime(&tm);          \
    ATLTRACE(category, level, L"%S: [%02d-%02d-%02d %02d:%02d:%02d:%03d] "fmt, __FUNCTION__,        \
    tm.wYear,tm.wMonth,tm.wDay,tm.wHour,tm.wMinute,tm.wSecond,tm.wMilliseconds, __VA_ARGS__); \
}while(0)
 
#define SetAtlTraceOpt(level, enable, category, filename_lineno, report_type, report_file)  \
do                                                                                         \
{                                                                                           \
    DWORD_PTRtrace_process = AtlTraceOpenProcess(GetCurrentProcessId());                   \
    AtlTraceModifyProcess(trace_process, level, enable, category, filename_lineno);         \
    AtlTraceCloseProcess(trace_process);                                                    \
    _CrtSetReportMode(level, report_type);                                                  \
    if(report_type == _CRTDBG_MODE_FILE) {                                                 \
        _CrtSetReportFile(level, report_file);                                              \
    }                                                                                       \
}while(0)
 
#define LoadAtlDebugCfgExA(path)                                                            \
do                                                                                         \
{                                                                                           \
    DWORD_PTRtrace_process = AtlTraceOpenProcess(GetCurrentProcessId());                   \
    AtlTraceLoadSettingsA(path, trace_process);                                             \
    AtlTraceCloseProcess(trace_process);                                                    \
}while(0)
 
#define LoadAtlDebugCfgExW(path)                                                            \
do                                                                                         \
{                                                                                           \
    DWORD_PTRtrace_process = AtlTraceOpenProcess(GetCurrentProcessId());                   \
    AtlTraceLoadSettingsU(path, trace_process);                                             \
    AtlTraceCloseProcess(trace_process);                                                    \
}while(0)
 
#define LoadAtlDebugCfgA()                                                                  \
do                                                                                         \
{                                                                                           \
    CHARdebug_cfg_path[MAX_PATH] = {0};                                                    \
    GetModuleFileNameA(NULL, debug_cfg_path, MAX_PATH);                                     \
    CPathA debug_cfg_path_obj(debug_cfg_path);                                              \
    debug_cfg_path_obj.RemoveExtension();                                                   \
    debug_cfg_path_obj.AddExtension(".trc");                                                \
    LoadAtlDebugCfgExA(debug_cfg_path_obj.m_strPath.GetString());                           \
}while(0)
 
#define LoadAtlDebugCfgW()                                                                  \
do                                                                                         \
{                                                                                           \
    WCHARdebug_cfg_path[MAX_PATH] = {0};                                                   \
    GetModuleFileNameW(NULL, debug_cfg_path, MAX_PATH);                                     \
    CPathW debug_cfg_path_obj(debug_cfg_path);                                              \
    debug_cfg_path_obj.RemoveExtension();                                                   \
    debug_cfg_path_obj.AddExtension(L".trc");                                               \
    LoadAtlDebugCfgExW(debug_cfg_path_obj.m_strPath.GetString());                           \
}while(0)
 
int_tmain(intargc, _TCHAR* argv[])
{
 
    SetAtlTraceOpt(_CRT_WARN,true,true,true, _CRTDBG_MODE_FILE, _CRTDBG_FILE_STDOUT);
    LoadAtlDebugCfgW();
    CTraceCategory MY_CATEGORY(_T("MyCategoryName"));
    TRACEHELPEXA(MY_CATEGORY, 0, "test test test\n");
    TRACEHELPEXW(MY_CATEGORY, 0, L"test test test\n");
    return0;
}

0 0