Allow multiple logging::LogMessage{Handler,Listener}s

base/logging_win.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/logging_win.h"
+#include "base/memory/ptr_util.h"
 #include "base/memory/singleton.h"
 #include <initguid.h>  // NOLINT
 
 namespace logging {
 
 using base::win::EtwEventLevel;
 using base::win::EtwMofEvent;
 
 DEFINE_GUID(kLogEventId,
     0x7fe69228, 0x633e, 0x4f06, 0x80, 0xc1, 0x52, 0x7f, 0xea, 0x23, 0xe3, 0xa7);
 
 LogEventProvider::LogEventProvider() : old_log_level_(LOG_NONE) {
 }
 
+LogEventProvider::~LogEventProvider() {}
+
 LogEventProvider* LogEventProvider::GetInstance() {
   return base::Singleton<LogEventProvider, base::StaticMemorySingletonTraits<
                                                LogEventProvider>>::get();
 }
 
-bool LogEventProvider::LogMessage(logging::LogSeverity severity,
-    const char* file, int line, size_t message_start,
+void LogEventProvider::WinLogMessageListener::OnMessage(
+    logging::LogSeverity severity,
+    const char* file,
+    int line,
+    size_t message_start,
     const std::string& message) {
   EtwEventLevel level = TRACE_LEVEL_NONE;
 
   // Convert the log severity to the most appropriate ETW trace level.
   if (severity >= 0) {
     switch (severity) {
       case LOG_INFO:
         level = TRACE_LEVEL_INFORMATION;
         break;
       case LOG_WARNING:
         level = TRACE_LEVEL_WARNING;
         break;
       case LOG_ERROR:
         level = TRACE_LEVEL_ERROR;
         break;
       case LOG_FATAL:
         level = TRACE_LEVEL_FATAL;
         break;
     }
   } else {  // severity < 0 is VLOG verbosity levels.
     level = static_cast<EtwEventLevel>(TRACE_LEVEL_INFORMATION - severity);
   }
 
   // Bail if we're not logging, not at that level,
   // or if we're post-atexit handling.
   LogEventProvider* provider = LogEventProvider::GetInstance();
   if (provider == NULL || level > provider->enable_level())
-    return false;
+    return;
 
   // And now log the event.
   if (provider->enable_flags() & ENABLE_LOG_MESSAGE_ONLY) {
     EtwMofEvent<1> event(kLogEventId, LOG_MESSAGE, level);
     event.SetField(0, message.length() + 1 - message_start,
         message.c_str() + message_start);
 
     provider->Log(event.get());
   } else {
     const size_t kMaxBacktraceDepth = 32;
@@ skipping 16 matching lines @@
     event.SetField(2, sizeof(line), &line);
     // The file.
     event.SetField(3, strlen(file) + 1, file);
     // And finally the message.
     event.SetField(4, message.length() + 1 - message_start,
         message.c_str() + message_start);
 
     provider->Log(event.get());
   }
 
-  // Don't increase verbosity in other log destinations.
-  if (severity < provider->old_log_level_)
-    return true;
-
-  return false;
+  // The log verbosity for other log destinations is increased.
 }
 
 void LogEventProvider::Initialize(const GUID& provider_name) {
   LogEventProvider* provider = LogEventProvider::GetInstance();
 
   provider->set_provider_name(provider_name);
   provider->Register();
 
-  // Register our message handler with logging.
-  SetLogMessageHandler(LogMessage);
+  // Register our message listener with logging.
+  provider->log_listener_ = base::MakeUnique<WinLogMessageListener>();
 }
 
 void LogEventProvider::Uninitialize() {
-  LogEventProvider::GetInstance()->Unregister();
+  LogEventProvider* provider = LogEventProvider::GetInstance();
+  provider->Unregister();
+  provider->log_listener_.reset();
 }
 
 void LogEventProvider::OnEventsEnabled() {
   // Grab the old log level so we can restore it later.
   old_log_level_ = GetMinLogLevel();
 
   // Convert the new trace level to a logging severity
   // and enable logging at that level.
   EtwEventLevel level = enable_level();
   if (level == TRACE_LEVEL_NONE || level == TRACE_LEVEL_FATAL) {
     SetMinLogLevel(LOG_FATAL);
   } else if (level == TRACE_LEVEL_ERROR) {
     SetMinLogLevel(LOG_ERROR);
   } else if (level == TRACE_LEVEL_WARNING) {
     SetMinLogLevel(LOG_WARNING);
   } else if (level == TRACE_LEVEL_INFORMATION) {
     SetMinLogLevel(LOG_INFO);
   } else if (level >= TRACE_LEVEL_VERBOSE) {
     // Above INFO, we enable verbose levels with negative severities.
     SetMinLogLevel(TRACE_LEVEL_INFORMATION - level);
   }
 }
 
 void LogEventProvider::OnEventsDisabled() {
   // Restore the old log level.
   SetMinLogLevel(old_log_level_);
 }
 
 }  // namespace logging