| // Copyright 2020 The Chromium OS Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| // |
| // This tool is used for getting dmesg information through debugd. |
| |
| #include "debugd/src/dmesg_tool.h" |
| |
| #include <vector> |
| |
| #include <base/containers/span.h> |
| #include <base/strings/strcat.h> |
| #include <base/strings/string_piece.h> |
| #include <base/strings/string_split.h> |
| #include <base/strings/string_util.h> |
| |
| #include "debugd/src/error_utils.h" |
| #include "debugd/src/process_with_output.h" |
| #include "debugd/src/variant_utils.h" |
| #include "linux/capability.h" |
| |
| namespace { |
| |
| constexpr char kDmesgPath[] = "/bin/dmesg"; |
| |
| } // namespace |
| |
| namespace debugd { |
| // static |
| void DmesgTool::Tail(uint32_t lines, std::string& output) { |
| if (lines <= 0) { |
| output.clear(); |
| return; |
| } |
| |
| if (output.empty()) { |
| return; |
| } |
| |
| std::vector<base::StringPiece> split = base::SplitStringPiece( |
| output, "\n", base::KEEP_WHITESPACE, base::SPLIT_WANT_ALL); |
| |
| // When the last character is \n, we get an extra blank line at the end which |
| // we don't actually want. (We want other blank lines, but not the extra blank |
| // at the end.) |
| bool had_ending_newline = false; |
| if (output.back() == '\n') { |
| split.pop_back(); |
| had_ending_newline = true; |
| } |
| |
| if (split.size() <= lines) { |
| return; |
| } |
| |
| base::span<base::StringPiece> desired_lines( |
| split.begin() + (split.size() - lines), lines); |
| |
| output = base::StrCat({base::JoinString(desired_lines, "\n"), |
| (had_ending_newline ? "\n" : "")}); |
| } |
| |
| bool DmesgTool::CallDmesg(const brillo::VariantDictionary& options, |
| brillo::ErrorPtr* error, |
| std::string* output) { |
| ProcessWithOutput process; |
| |
| // WARNING: CAP_TO_MASK() produces bad results when used with values >=32. |
| process.SetCapabilities(1ULL << CAP_SYSLOG); |
| if (!process.Init()) { |
| *output = "<process init failed>"; |
| return false; |
| } |
| |
| process.AddArg(kDmesgPath); |
| |
| if (!AddBoolOption(&process, options, "show-delta", "-d", error) || |
| !AddBoolOption(&process, options, "human", "--human", error) || |
| !AddBoolOption(&process, options, "kernel", "-k", error) || |
| !AddBoolOption(&process, options, "color", "--color=always", error) || |
| !AddBoolOption(&process, options, "force-prefix", "-p", error) || |
| !AddBoolOption(&process, options, "raw", "-r", error) || |
| !AddBoolOption(&process, options, "ctime", "-T", error) || |
| !AddBoolOption(&process, options, "notime", "-t", error) || |
| !AddBoolOption(&process, options, "userspace", "-u", error) || |
| !AddBoolOption(&process, options, "decode", "-x", error)) { |
| *output = "<invalid option>"; |
| return false; |
| } |
| |
| if (process.Run() != 0) { |
| *output = "<process exited with nonzero status>"; |
| return false; |
| } |
| |
| process.GetOutput(output); |
| |
| uint32_t lines = 0; |
| ParseResult result = GetOption(options, "tail", &lines, error); |
| if (result == ParseResult::PARSE_ERROR) { |
| *output = "<invalid option to tail>"; |
| return false; |
| } |
| if (result == ParseResult::PARSED) { |
| Tail(lines, *output); |
| } |
| |
| return true; |
| } |
| |
| } // namespace debugd |