common/telemetry_info.cc

//
// Copyright (C) 2021 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//

#include "update_engine/common/telemetry_info.h"

#include <base/strings/string_number_conversions.h>
#include <base/strings/string_util.h>
#include <base/strings/stringprintf.h>

namespace chromeos_update_engine {

std::string TelemetryInfo::GetWirelessDrivers() const {
  return GetBusDeviceDrivers(BusDevice::BusDeviceClass::kWirelessController);
}

std::string TelemetryInfo::GetWirelessIds() const {
  return GetBusDeviceIds(BusDevice::BusDeviceClass::kWirelessController);
}

std::string TelemetryInfo::GetGpuDrivers() const {
  return GetBusDeviceDrivers(BusDevice::BusDeviceClass::kDisplayController);
}

std::string TelemetryInfo::GetGpuIds() const {
  return GetBusDeviceIds(BusDevice::BusDeviceClass::kDisplayController);
}

std::string TelemetryInfo::GetBusDeviceDrivers(
    BusDevice::BusDeviceClass bus_device_class) const {
  std::vector<std::string> drivers;
  for (const auto& bus_device : bus_devices) {
    if (bus_device.device_class != bus_device_class)
      continue;
    if (const auto* pci_bus_info =
            std::get_if<BusDevice::PciBusInfo>(&bus_device.bus_type_info)) {
      const auto& driver = pci_bus_info->driver;
      if (!driver.empty())
        drivers.push_back(driver);
    }
  }
  return base::JoinString(drivers, " ");
}

std::string TelemetryInfo::GetBusDeviceIds(
    BusDevice::BusDeviceClass bus_device_class) const {
  std::vector<std::string> ids;
  for (const auto& bus_device : bus_devices) {
    if (bus_device.device_class != bus_device_class)
      continue;
    if (const auto* pci_bus_info =
            std::get_if<BusDevice::PciBusInfo>(&bus_device.bus_type_info)) {
      ids.push_back(base::JoinString(
          {
              base::StringPrintf("%04X", pci_bus_info->vendor_id),
              base::StringPrintf("%04X", pci_bus_info->device_id),
          },
          ":"));
    } else if (const auto* usb_bus_info = std::get_if<BusDevice::UsbBusInfo>(
                   &bus_device.bus_type_info)) {
      ids.push_back(base::JoinString(
          {
              base::StringPrintf("%04X", usb_bus_info->vendor_id),
              base::StringPrintf("%04X", usb_bus_info->product_id),
          },
          ":"));
    }
  }
  return base::JoinString(ids, " ");
}

}  // namespace chromeos_update_engine