1 | // Copyright 2009 The RE2 Authors. All Rights Reserved. |
2 | // Use of this source code is governed by a BSD-style |
3 | // license that can be found in the LICENSE file. |
4 | |
5 | #ifndef UTIL_LOGGING_H_ |
6 | #define UTIL_LOGGING_H_ |
7 | |
8 | // Simplified version of Google's logging. |
9 | |
10 | #include <assert.h> |
11 | #include <stdio.h> |
12 | #include <stdlib.h> |
13 | #include <ostream> |
14 | #include <sstream> |
15 | |
16 | #include "util/util.h" |
17 | |
18 | // Debug-only checking. |
19 | #define DCHECK(condition) assert(condition) |
20 | #define DCHECK_EQ(val1, val2) assert((val1) == (val2)) |
21 | #define DCHECK_NE(val1, val2) assert((val1) != (val2)) |
22 | #define DCHECK_LE(val1, val2) assert((val1) <= (val2)) |
23 | #define DCHECK_LT(val1, val2) assert((val1) < (val2)) |
24 | #define DCHECK_GE(val1, val2) assert((val1) >= (val2)) |
25 | #define DCHECK_GT(val1, val2) assert((val1) > (val2)) |
26 | |
27 | // Always-on checking |
28 | #define CHECK(x) if(x){}else LogMessageFatal(__FILE__, __LINE__).stream() << "Check failed: " #x |
29 | #define CHECK_LT(x, y) CHECK((x) < (y)) |
30 | #define CHECK_GT(x, y) CHECK((x) > (y)) |
31 | #define CHECK_LE(x, y) CHECK((x) <= (y)) |
32 | #define CHECK_GE(x, y) CHECK((x) >= (y)) |
33 | #define CHECK_EQ(x, y) CHECK((x) == (y)) |
34 | #define CHECK_NE(x, y) CHECK((x) != (y)) |
35 | |
36 | #define LOG_INFO LogMessage(__FILE__, __LINE__) |
37 | #define LOG_WARNING LogMessage(__FILE__, __LINE__) |
38 | #define LOG_ERROR LogMessage(__FILE__, __LINE__) |
39 | #define LOG_FATAL LogMessageFatal(__FILE__, __LINE__) |
40 | #define LOG_QFATAL LOG_FATAL |
41 | |
42 | // It seems that one of the Windows header files defines ERROR as 0. |
43 | #ifdef _WIN32 |
44 | #define LOG_0 LOG_INFO |
45 | #endif |
46 | |
47 | #ifdef NDEBUG |
48 | #define LOG_DFATAL LOG_ERROR |
49 | #else |
50 | #define LOG_DFATAL LOG_FATAL |
51 | #endif |
52 | |
53 | #define LOG(severity) LOG_ ## severity.stream() |
54 | |
55 | #define VLOG(x) if((x)>0){}else LOG_INFO.stream() |
56 | |
57 | class LogMessage { |
58 | public: |
59 | LogMessage(const char* file, int line) |
60 | : flushed_(false) { |
61 | stream() << file << ":" << line << ": " ; |
62 | } |
63 | void Flush() { |
64 | stream() << "\n" ; |
65 | std::string s = str_.str(); |
66 | size_t n = s.size(); |
67 | if (fwrite(s.data(), 1, n, stderr) < n) {} // shut up gcc |
68 | flushed_ = true; |
69 | } |
70 | ~LogMessage() { |
71 | if (!flushed_) { |
72 | Flush(); |
73 | } |
74 | } |
75 | std::ostream& stream() { return str_; } |
76 | |
77 | private: |
78 | bool flushed_; |
79 | std::ostringstream str_; |
80 | |
81 | LogMessage(const LogMessage&) = delete; |
82 | LogMessage& operator=(const LogMessage&) = delete; |
83 | }; |
84 | |
85 | // Silence "destructor never returns" warning for ~LogMessageFatal(). |
86 | // Since this is a header file, push and then pop to limit the scope. |
87 | #ifdef _MSC_VER |
88 | #pragma warning(push) |
89 | #pragma warning(disable: 4722) |
90 | #endif |
91 | |
92 | class LogMessageFatal : public LogMessage { |
93 | public: |
94 | LogMessageFatal(const char* file, int line) |
95 | : LogMessage(file, line) {} |
96 | ATTRIBUTE_NORETURN ~LogMessageFatal() { |
97 | Flush(); |
98 | abort(); |
99 | } |
100 | private: |
101 | LogMessageFatal(const LogMessageFatal&) = delete; |
102 | LogMessageFatal& operator=(const LogMessageFatal&) = delete; |
103 | }; |
104 | |
105 | #ifdef _MSC_VER |
106 | #pragma warning(pop) |
107 | #endif |
108 | |
109 | #endif // UTIL_LOGGING_H_ |
110 | |