OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "remoting/host/native_messaging/log_message_handler.h" | 5 #include "remoting/host/native_messaging/log_message_handler.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/lazy_instance.h" | 8 #include "base/lazy_instance.h" |
9 #include "base/single_thread_task_runner.h" | 9 #include "base/single_thread_task_runner.h" |
10 #include "base/strings/string_util.h" | 10 #include "base/strings/string_util.h" |
(...skipping 19 matching lines...) Expand all Loading... |
30 const Delegate& delegate) | 30 const Delegate& delegate) |
31 : delegate_(delegate), | 31 : delegate_(delegate), |
32 suppress_logging_(false), | 32 suppress_logging_(false), |
33 caller_task_runner_(base::ThreadTaskRunnerHandle::Get()), | 33 caller_task_runner_(base::ThreadTaskRunnerHandle::Get()), |
34 weak_ptr_factory_(this) { | 34 weak_ptr_factory_(this) { |
35 base::AutoLock lock(g_log_message_handler_lock.Get()); | 35 base::AutoLock lock(g_log_message_handler_lock.Get()); |
36 if (g_log_message_handler) { | 36 if (g_log_message_handler) { |
37 LOG(FATAL) << "LogMessageHandler is already registered. Only one instance " | 37 LOG(FATAL) << "LogMessageHandler is already registered. Only one instance " |
38 << "per process is allowed."; | 38 << "per process is allowed."; |
39 } | 39 } |
40 previous_log_message_handler_ = logging::GetLogMessageHandler(); | 40 logging::AddLogMessageHandler(&LogMessageHandler::OnLogMessage); |
41 logging::SetLogMessageHandler(&LogMessageHandler::OnLogMessage); | |
42 g_log_message_handler = this; | 41 g_log_message_handler = this; |
43 } | 42 } |
44 | 43 |
45 LogMessageHandler::~LogMessageHandler() { | 44 LogMessageHandler::~LogMessageHandler() { |
46 base::AutoLock lock(g_log_message_handler_lock.Get()); | 45 base::AutoLock lock(g_log_message_handler_lock.Get()); |
47 if (logging::GetLogMessageHandler() != &LogMessageHandler::OnLogMessage) { | 46 logging::RemoveLogMessageHandler(&LogMessageHandler::OnLogMessage); |
48 LOG(FATAL) << "LogMessageHandler is not the top-most message handler. " | |
49 << "Cannot unregister."; | |
50 } | |
51 logging::SetLogMessageHandler(previous_log_message_handler_); | |
52 g_log_message_handler = nullptr; | 47 g_log_message_handler = nullptr; |
53 } | 48 } |
54 | 49 |
55 // static | 50 // static |
56 const char* LogMessageHandler::kDebugMessageTypeName = "_debug_log"; | 51 const char* LogMessageHandler::kDebugMessageTypeName = "_debug_log"; |
57 | 52 |
58 // static | 53 // static |
59 bool LogMessageHandler::OnLogMessage( | 54 bool LogMessageHandler::OnLogMessage( |
60 logging::LogSeverity severity, | 55 logging::LogSeverity severity, |
61 const char* file, | 56 const std::string& file, |
62 int line, | 57 int line, |
63 size_t message_start, | |
64 const std::string& str) { | 58 const std::string& str) { |
65 base::AutoLock lock(g_log_message_handler_lock.Get()); | 59 base::AutoLock lock(g_log_message_handler_lock.Get()); |
66 if (g_log_message_handler) { | 60 if (g_log_message_handler) { |
67 g_log_message_handler->PostLogMessageToCorrectThread( | 61 g_log_message_handler->PostLogMessageToCorrectThread( |
68 severity, file, line, message_start, str); | 62 severity, file, line, str); |
69 } | 63 } |
70 return false; | 64 return false; |
71 } | 65 } |
72 | 66 |
73 void LogMessageHandler::PostLogMessageToCorrectThread( | 67 void LogMessageHandler::PostLogMessageToCorrectThread( |
74 logging::LogSeverity severity, | 68 logging::LogSeverity severity, |
75 const char* file, | 69 const std::string& file, |
76 int line, | 70 int line, |
77 size_t message_start, | |
78 const std::string& str) { | 71 const std::string& str) { |
79 // Don't process this message if we're already logging and on the caller | 72 // Don't process this message if we're already logging and on the caller |
80 // thread. This guards against an infinite loop if any code called by this | 73 // thread. This guards against an infinite loop if any code called by this |
81 // class logs something. | 74 // class logs something. |
82 if (suppress_logging_ && caller_task_runner_->BelongsToCurrentThread()) { | 75 if (suppress_logging_ && caller_task_runner_->BelongsToCurrentThread()) { |
83 return; | 76 return; |
84 } | 77 } |
85 | 78 |
86 // This method is always called under the global lock, so post a task to | 79 // This method is always called under the global lock, so post a task to |
87 // handle the log message, even if we're already on the correct thread. | 80 // handle the log message, even if we're already on the correct thread. |
88 // This alows the lock to be released quickly. | 81 // This alows the lock to be released quickly. |
89 // | 82 // |
90 // Note that this means that LOG(FATAL) messages will be lost because the | 83 // Note that this means that LOG(FATAL) messages will be lost because the |
91 // process will exit before the message is sent to the client. | 84 // process will exit before the message is sent to the client. |
92 caller_task_runner_->PostTask( | 85 caller_task_runner_->PostTask( |
93 FROM_HERE, | 86 FROM_HERE, |
94 base::Bind(&LogMessageHandler::SendLogMessageToClient, | 87 base::Bind(&LogMessageHandler::SendLogMessageToClient, |
95 weak_ptr_factory_.GetWeakPtr(), | 88 weak_ptr_factory_.GetWeakPtr(), |
96 severity, file, line, message_start, str)); | 89 severity, file, line, str)); |
97 } | 90 } |
98 | 91 |
99 void LogMessageHandler::SendLogMessageToClient( | 92 void LogMessageHandler::SendLogMessageToClient( |
100 logging::LogSeverity severity, | 93 logging::LogSeverity severity, |
101 const char* file, | 94 const std::string& file, |
102 int line, | 95 int line, |
103 size_t message_start, | |
104 const std::string& str) { | 96 const std::string& str) { |
105 suppress_logging_ = true; | 97 suppress_logging_ = true; |
106 | 98 |
107 std::string severity_string = "log"; | 99 std::string severity_string = "log"; |
108 switch (severity) { | 100 switch (severity) { |
109 case logging::LOG_WARNING: | 101 case logging::LOG_WARNING: |
110 severity_string = "warn"; | 102 severity_string = "warn"; |
111 break; | 103 break; |
112 case logging::LOG_FATAL: | 104 case logging::LOG_FATAL: |
113 case logging::LOG_ERROR: | 105 case logging::LOG_ERROR: |
(...skipping 10 matching lines...) Expand all Loading... |
124 dictionary->SetString("message", message); | 116 dictionary->SetString("message", message); |
125 dictionary->SetString("file", file); | 117 dictionary->SetString("file", file); |
126 dictionary->SetInteger("line", line); | 118 dictionary->SetInteger("line", line); |
127 | 119 |
128 delegate_.Run(std::move(dictionary)); | 120 delegate_.Run(std::move(dictionary)); |
129 | 121 |
130 suppress_logging_ = false; | 122 suppress_logging_ = false; |
131 } | 123 } |
132 | 124 |
133 } // namespace remoting | 125 } // namespace remoting |
OLD | NEW |