Home | History | Annotate | Download | only in util
      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 // Simplified version of Google's logging.
      6 
      7 #ifndef RE2_UTIL_LOGGING_H__
      8 #define RE2_UTIL_LOGGING_H__
      9 
     10 #ifndef WIN32
     11 #include <unistd.h>  /* for write */
     12 #endif
     13 #include <sstream>
     14 #ifdef WIN32
     15 #include <io.h>
     16 #endif
     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_ERROR LOG_INFO
     38 #define LOG_WARNING LOG_INFO
     39 #define LOG_FATAL LogMessageFatal(__FILE__, __LINE__)
     40 #define LOG_QFATAL LOG_FATAL
     41 
     42 #define VLOG(x) if((x)>0){}else LOG_INFO.stream()
     43 
     44 #ifdef NDEBUG
     45 #define DEBUG_MODE 0
     46 #define LOG_DFATAL LOG_ERROR
     47 #else
     48 #define DEBUG_MODE 1
     49 #define LOG_DFATAL LOG_FATAL
     50 #endif
     51 
     52 #define LOG(severity) LOG_ ## severity.stream()
     53 
     54 class LogMessage {
     55  public:
     56   LogMessage(const char* file, int line) : flushed_(false) {
     57     stream() << file << ":" << line << ": ";
     58   }
     59   void Flush() {
     60     stream() << "\n";
     61     string s = str_.str();
     62     int n = (int)s.size(); // shut up msvc
     63     if(write(2, s.data(), n) < 0) {}  // shut up gcc
     64     flushed_ = true;
     65   }
     66   ~LogMessage() {
     67     if (!flushed_) {
     68       Flush();
     69     }
     70   }
     71   ostream& stream() { return str_; }
     72 
     73  private:
     74   bool flushed_;
     75   std::ostringstream str_;
     76   DISALLOW_EVIL_CONSTRUCTORS(LogMessage);
     77 };
     78 
     79 class LogMessageFatal : public LogMessage {
     80  public:
     81   LogMessageFatal(const char* file, int line)
     82     : LogMessage(file, line) { }
     83   ~LogMessageFatal() {
     84     Flush();
     85     abort();
     86   }
     87  private:
     88   DISALLOW_EVIL_CONSTRUCTORS(LogMessageFatal);
     89 };
     90 
     91 #endif  // RE2_UTIL_LOGGING_H__
     92