1 //
2 // Copyright (C) 2022 The Android Open Source Project
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 // http://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15
16 #include "host/commands/cvd/logger.h"
17
18 #include <shared_mutex>
19 #include <thread>
20 #include <unordered_map>
21
22 #include <android-base/logging.h>
23 #include <android-base/threads.h>
24
25 #include "common/libs/fs/shared_buf.h"
26 #include "common/libs/utils/tee_logging.h"
27 #include "host/commands/cvd/server_client.h"
28
29 namespace cuttlefish {
30
ServerLogger()31 ServerLogger::ServerLogger() {
32 auto log_callback = [this](android::base::LogId log_buffer_id,
33 android::base::LogSeverity severity,
34 const char* tag, const char* file,
35 unsigned int line, const char* message) {
36 auto thread_id = std::this_thread::get_id();
37 std::shared_lock lock(thread_loggers_lock_);
38 auto logger_it = thread_loggers_.find(thread_id);
39 if (logger_it == thread_loggers_.end()) {
40 return;
41 }
42 logger_it->second->LogMessage(log_buffer_id, severity, tag, file, line,
43 message);
44 };
45 android::base::SetLogger(log_callback);
46 }
47
~ServerLogger()48 ServerLogger::~ServerLogger() {
49 android::base::SetLogger(android::base::StderrLogger);
50 }
51
LogThreadToFd(SharedFD target)52 ServerLogger::ScopedLogger ServerLogger::LogThreadToFd(SharedFD target) {
53 return ScopedLogger(*this, std::move(target));
54 }
55
ScopedLogger(ServerLogger & server_logger,SharedFD target)56 ServerLogger::ScopedLogger::ScopedLogger(ServerLogger& server_logger,
57 SharedFD target)
58 : server_logger_(server_logger), target_(std::move(target)) {
59 auto thread_id = std::this_thread::get_id();
60 std::unique_lock lock(server_logger_.thread_loggers_lock_);
61 server_logger_.thread_loggers_[thread_id] = this;
62 }
63
ScopedLogger(ServerLogger::ScopedLogger && other)64 ServerLogger::ScopedLogger::ScopedLogger(
65 ServerLogger::ScopedLogger&& other) noexcept
66 : server_logger_(other.server_logger_), target_(std::move(other.target_)) {
67 auto thread_id = std::this_thread::get_id();
68 std::unique_lock lock(server_logger_.thread_loggers_lock_);
69 server_logger_.thread_loggers_[thread_id] = this;
70 }
71
~ScopedLogger()72 ServerLogger::ScopedLogger::~ScopedLogger() {
73 auto thread_id = std::this_thread::get_id();
74 std::unique_lock lock(server_logger_.thread_loggers_lock_);
75 auto logger_it = server_logger_.thread_loggers_.find(thread_id);
76 if (logger_it == server_logger_.thread_loggers_.end()) {
77 return;
78 }
79 if (logger_it->second == this) {
80 server_logger_.thread_loggers_.erase(logger_it);
81 }
82 }
83
LogMessage(android::base::LogId,android::base::LogSeverity severity,const char * tag,const char * file,unsigned int line,const char * message)84 void ServerLogger::ScopedLogger::LogMessage(
85 android::base::LogId /* log_buffer_id */,
86 android::base::LogSeverity severity, const char* tag, const char* file,
87 unsigned int line, const char* message) {
88 time_t t = time(nullptr);
89 struct tm now;
90 localtime_r(&t, &now);
91 auto output_string =
92 StderrOutputGenerator(now, getpid(), android::base::GetThreadId(),
93 severity, tag, file, line, message);
94 WriteAll(target_, output_string);
95 }
96
97 } // namespace cuttlefish
98