1 // Copyright 2017 The Abseil Authors.
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // https://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14 //
15 // Thread-safe logging routines that do not allocate any memory or
16 // acquire any locks, and can therefore be used by low-level memory
17 // allocation, synchronization, and signal-handling code.
18
19 #ifndef ABSL_BASE_INTERNAL_RAW_LOGGING_H_
20 #define ABSL_BASE_INTERNAL_RAW_LOGGING_H_
21
22 #include <string>
23
24 #include "absl/base/attributes.h"
25 #include "absl/base/config.h"
26 #include "absl/base/internal/atomic_hook.h"
27 #include "absl/base/log_severity.h"
28 #include "absl/base/macros.h"
29 #include "absl/base/optimization.h"
30 #include "absl/base/port.h"
31
32 // This is similar to LOG(severity) << format..., but
33 // * it is to be used ONLY by low-level modules that can't use normal LOG()
34 // * it is designed to be a low-level logger that does not allocate any
35 // memory and does not need any locks, hence:
36 // * it logs straight and ONLY to STDERR w/o buffering
37 // * it uses an explicit printf-format and arguments list
38 // * it will silently chop off really long message strings
39 // Usage example:
40 // ABSL_RAW_LOG(ERROR, "Failed foo with %i: %s", status, error);
41 // This will print an almost standard log line like this to stderr only:
42 // E0821 211317 file.cc:123] RAW: Failed foo with 22: bad_file
43
44 #define ABSL_RAW_LOG(severity, ...) \
45 do { \
46 constexpr const char* absl_raw_log_internal_basename = \
47 ::absl::raw_log_internal::Basename(__FILE__, sizeof(__FILE__) - 1); \
48 ::absl::raw_log_internal::RawLog(ABSL_RAW_LOG_INTERNAL_##severity, \
49 absl_raw_log_internal_basename, __LINE__, \
50 __VA_ARGS__); \
51 } while (0)
52
53 // Similar to CHECK(condition) << message, but for low-level modules:
54 // we use only ABSL_RAW_LOG that does not allocate memory.
55 // We do not want to provide args list here to encourage this usage:
56 // if (!cond) ABSL_RAW_LOG(FATAL, "foo ...", hard_to_compute_args);
57 // so that the args are not computed when not needed.
58 #define ABSL_RAW_CHECK(condition, message) \
59 do { \
60 if (ABSL_PREDICT_FALSE(!(condition))) { \
61 ABSL_RAW_LOG(FATAL, "Check %s failed: %s", #condition, message); \
62 } \
63 } while (0)
64
65 // ABSL_INTERNAL_LOG and ABSL_INTERNAL_CHECK work like the RAW variants above,
66 // except that if the richer log library is linked into the binary, we dispatch
67 // to that instead. This is potentially useful for internal logging and
68 // assertions, where we are using RAW_LOG neither for its async-signal-safety
69 // nor for its non-allocating nature, but rather because raw logging has very
70 // few other dependencies.
71 //
72 // The API is a subset of the above: each macro only takes two arguments. Use
73 // StrCat if you need to build a richer message.
74 #define ABSL_INTERNAL_LOG(severity, message) \
75 do { \
76 constexpr const char* absl_raw_log_internal_filename = __FILE__; \
77 ::absl::raw_log_internal::internal_log_function( \
78 ABSL_RAW_LOG_INTERNAL_##severity, absl_raw_log_internal_filename, \
79 __LINE__, message); \
80 if (ABSL_RAW_LOG_INTERNAL_##severity == ::absl::LogSeverity::kFatal) \
81 ABSL_UNREACHABLE(); \
82 } while (0)
83
84 #define ABSL_INTERNAL_CHECK(condition, message) \
85 do { \
86 if (ABSL_PREDICT_FALSE(!(condition))) { \
87 std::string death_message = "Check " #condition " failed: "; \
88 death_message += std::string(message); \
89 ABSL_INTERNAL_LOG(FATAL, death_message); \
90 } \
91 } while (0)
92
93 #define ABSL_RAW_LOG_INTERNAL_INFO ::absl::LogSeverity::kInfo
94 #define ABSL_RAW_LOG_INTERNAL_WARNING ::absl::LogSeverity::kWarning
95 #define ABSL_RAW_LOG_INTERNAL_ERROR ::absl::LogSeverity::kError
96 #define ABSL_RAW_LOG_INTERNAL_FATAL ::absl::LogSeverity::kFatal
97 #define ABSL_RAW_LOG_INTERNAL_LEVEL(severity) \
98 ::absl::NormalizeLogSeverity(severity)
99
100 namespace absl {
101 ABSL_NAMESPACE_BEGIN
102 namespace raw_log_internal {
103
104 // Helper function to implement ABSL_RAW_LOG
105 // Logs format... at "severity" level, reporting it
106 // as called from file:line.
107 // This does not allocate memory or acquire locks.
108 void RawLog(absl::LogSeverity severity, const char* file, int line,
109 const char* format, ...) ABSL_PRINTF_ATTRIBUTE(4, 5);
110
111 // Writes the provided buffer directly to stderr, in a signal-safe, low-level
112 // manner.
113 void AsyncSignalSafeWriteToStderr(const char* s, size_t len);
114
115 // compile-time function to get the "base" filename, that is, the part of
116 // a filename after the last "/" or "\" path separator. The search starts at
117 // the end of the string; the second parameter is the length of the string.
Basename(const char * fname,int offset)118 constexpr const char* Basename(const char* fname, int offset) {
119 return offset == 0 || fname[offset - 1] == '/' || fname[offset - 1] == '\\'
120 ? fname + offset
121 : Basename(fname, offset - 1);
122 }
123
124 // For testing only.
125 // Returns true if raw logging is fully supported. When it is not
126 // fully supported, no messages will be emitted, but a log at FATAL
127 // severity will cause an abort.
128 //
129 // TODO(gfalcon): Come up with a better name for this method.
130 bool RawLoggingFullySupported();
131
132 // Function type for a raw_log customization hook for suppressing messages
133 // by severity, and for writing custom prefixes on non-suppressed messages.
134 //
135 // The installed hook is called for every raw log invocation. The message will
136 // be logged to stderr only if the hook returns true. FATAL errors will cause
137 // the process to abort, even if writing to stderr is suppressed. The hook is
138 // also provided with an output buffer, where it can write a custom log message
139 // prefix.
140 //
141 // The raw_log system does not allocate memory or grab locks. User-provided
142 // hooks must avoid these operations, and must not throw exceptions.
143 //
144 // 'severity' is the severity level of the message being written.
145 // 'file' and 'line' are the file and line number where the ABSL_RAW_LOG macro
146 // was located.
147 // 'buf' and 'buf_size' are pointers to the buffer and buffer size. If the
148 // hook writes a prefix, it must increment *buf and decrement *buf_size
149 // accordingly.
150 using LogFilterAndPrefixHook = bool (*)(absl::LogSeverity severity,
151 const char* file, int line, char** buf,
152 int* buf_size);
153
154 // Function type for a raw_log customization hook called to abort a process
155 // when a FATAL message is logged. If the provided AbortHook() returns, the
156 // logging system will call abort().
157 //
158 // 'file' and 'line' are the file and line number where the ABSL_RAW_LOG macro
159 // was located.
160 // The NUL-terminated logged message lives in the buffer between 'buf_start'
161 // and 'buf_end'. 'prefix_end' points to the first non-prefix character of the
162 // buffer (as written by the LogFilterAndPrefixHook.)
163 //
164 // The lifetime of the filename and message buffers will not end while the
165 // process remains alive.
166 using AbortHook = void (*)(const char* file, int line, const char* buf_start,
167 const char* prefix_end, const char* buf_end);
168
169 // Internal logging function for ABSL_INTERNAL_LOG to dispatch to.
170 //
171 // TODO(gfalcon): When string_view no longer depends on base, change this
172 // interface to take its message as a string_view instead.
173 using InternalLogFunction = void (*)(absl::LogSeverity severity,
174 const char* file, int line,
175 const std::string& message);
176
177 ABSL_INTERNAL_ATOMIC_HOOK_ATTRIBUTES ABSL_DLL extern base_internal::AtomicHook<
178 InternalLogFunction>
179 internal_log_function;
180
181 // Registers hooks of the above types. Only a single hook of each type may be
182 // registered. It is an error to call these functions multiple times with
183 // different input arguments.
184 //
185 // These functions are safe to call at any point during initialization; they do
186 // not block or malloc, and are async-signal safe.
187 void RegisterLogFilterAndPrefixHook(LogFilterAndPrefixHook func);
188 void RegisterAbortHook(AbortHook func);
189 void RegisterInternalLogFunction(InternalLogFunction func);
190
191 } // namespace raw_log_internal
192 ABSL_NAMESPACE_END
193 } // namespace absl
194
195 #endif // ABSL_BASE_INTERNAL_RAW_LOGGING_H_
196