| // 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. |
| |
| #include "runtime_probe/functions/input_device.h" |
| |
| #include <string> |
| #include <utility> |
| |
| #include <base/files/file_util.h> |
| #include <base/json/json_reader.h> |
| #include <base/json/json_writer.h> |
| #include <base/strings/string_split.h> |
| #include <base/strings/string_util.h> |
| #include <pcrecpp.h> |
| |
| #include "runtime_probe/utils/value_utils.h" |
| |
| namespace runtime_probe { |
| |
| namespace { |
| constexpr auto kInputDevicesPath = "/proc/bus/input/devices"; |
| const pcrecpp::RE kEventPatternRe(R"(event[\d]+)"); |
| |
| base::Value LoadInputDevices() { |
| base::Value results(base::Value::Type::LIST); |
| std::string input_devices_str; |
| if (!base::ReadFileToString(base::FilePath(kInputDevicesPath), |
| &input_devices_str)) { |
| LOG(ERROR) << "Failed to read " << kInputDevicesPath << "."; |
| return results; |
| } |
| |
| base::Value data(base::Value::Type::DICTIONARY); |
| auto input_devices_lines = base::SplitStringPiece( |
| input_devices_str, "\n", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL); |
| for (const auto& line : input_devices_lines) { |
| if (line.length() > 0) { |
| const auto prefix = line[0]; |
| const auto& content = line.substr(3); |
| if (prefix == 'I') { |
| if (!data.DictEmpty()) { |
| RenameKey(&data, "sysfs", "path"); |
| results.Append(std::move(data)); |
| data = base::Value(base::Value::Type::DICTIONARY); |
| } |
| base::StringPairs keyVals; |
| if (!base::SplitStringIntoKeyValuePairs(content, '=', ' ', &keyVals)) { |
| LOG(ERROR) << "Failed to parse input devices (I)."; |
| return base::Value(base::Value::Type::LIST); |
| } |
| for (const auto& keyVal : keyVals) { |
| base::StringPiece key, value; |
| std::tie(key, value) = keyVal; |
| data.SetStringKey(base::ToLowerASCII(key), value); |
| } |
| } else if (prefix == 'N' || prefix == 'S') { |
| base::StringPairs keyVals; |
| if (!base::SplitStringIntoKeyValuePairs(content, '=', '\n', &keyVals)) { |
| LOG(ERROR) << "Failed to parse input devices (N/S)."; |
| return base::Value(base::Value::Type::LIST); |
| } |
| base::StringPiece key, value; |
| std::tie(key, value) = keyVals[0]; |
| data.SetStringKey(base::ToLowerASCII(key), |
| base::TrimString(value, "\"", base::TRIM_ALL)); |
| } else if (prefix == 'H') { |
| base::StringPairs keyVals; |
| if (!base::SplitStringIntoKeyValuePairs(content, '=', '\n', &keyVals)) { |
| LOG(ERROR) << "Failed to parse input devices (H)."; |
| return base::Value(base::Value::Type::LIST); |
| } |
| base::StringPiece value = keyVals[0].second; |
| const auto& handlers = base::SplitStringPiece( |
| value, " ", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL); |
| for (const auto& handler : handlers) { |
| if (kEventPatternRe.FullMatch(handler.as_string())) { |
| data.SetStringKey("event", handler); |
| break; |
| } |
| } |
| } |
| } |
| } |
| if (!data.DictEmpty()) { |
| RenameKey(&data, "sysfs", "path"); |
| results.Append(std::move(data)); |
| } |
| return results; |
| } |
| |
| } // namespace |
| |
| InputDeviceFunction::DataType InputDeviceFunction::Eval() const { |
| auto json_output = InvokeHelperToJSON(); |
| if (!json_output) { |
| LOG(ERROR) << "Failed to invoke helper to retrieve sysfs results."; |
| return {}; |
| } |
| if (!json_output->is_list()) { |
| LOG(ERROR) << "Failed to parse json output as list."; |
| return {}; |
| } |
| |
| return DataType(json_output->TakeList()); |
| } |
| |
| int InputDeviceFunction::EvalInHelper(std::string* output) const { |
| auto results = LoadInputDevices(); |
| |
| if (!base::JSONWriter::Write(results, output)) { |
| LOG(ERROR) << "Failed to serialize usb probed result to json string"; |
| return -1; |
| } |
| return 0; |
| } |
| |
| } // namespace runtime_probe |