// Copyright 2018 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 "chromeos-config/libcros_config/identity_arm.h"

#include <arpa/inet.h>
#include <string>

#include <base/logging.h>
#include <base/files/file_path.h>
#include <base/files/file_util.h>
#include <base/strings/stringprintf.h>
#include <base/values.h>
#include "chromeos-config/libcros_config/cros_config.h"
#include "chromeos-config/libcros_config/cros_config_interface.h"

namespace {
constexpr int kSkuIdLength = 4;
}  // namespace

namespace brillo {

CrosConfigIdentityArm::CrosConfigIdentityArm() {}

CrosConfigIdentityArm::~CrosConfigIdentityArm() {}

bool CrosConfigIdentityArm::FakeProductFilesForTesting(
    const std::string& device_name,
    const int sku_id,
    base::FilePath* dt_compatible_file_out,
    base::FilePath* sku_id_file_out) {
  *dt_compatible_file_out = base::FilePath("dt_compatible");
  if (base::WriteFile(*dt_compatible_file_out, device_name.c_str(),
                      device_name.length()) != device_name.length()) {
    CROS_CONFIG_LOG(ERROR) << "Failed to write device-tree compatible file";
    return false;
  }
  char sku_id_char[kSkuIdLength];
  uint32_t sku_id_fdt = htonl(sku_id);
  *sku_id_file_out = base::FilePath("sku-id");

  std::memcpy(sku_id_char, &sku_id_fdt, kSkuIdLength);

  if (base::WriteFile(*sku_id_file_out, sku_id_char,
                      kSkuIdLength) != kSkuIdLength) {
    CROS_CONFIG_LOG(ERROR) << "Failed to write sku-id file";
    return false;
  }

  return true;
}

bool CrosConfigIdentityArm::ReadInfo(const base::FilePath& dt_compatible_file,
                                     const base::FilePath& sku_id_file) {
  if (!base::ReadFileToString(dt_compatible_file, &compatible_devices_)) {
    CROS_CONFIG_LOG(ERROR) << "Failed to read device-tree compatible file: "
                           << dt_compatible_file.MaybeAsASCII();
    return false;
  }

  char sku_id_char[kSkuIdLength];
  if (base::ReadFile(sku_id_file, sku_id_char, kSkuIdLength) != kSkuIdLength) {
    CROS_CONFIG_LOG(WARNING) << "Cannot read product_sku file ";
    sku_id_ = kDefaultSkuId;
  } else {
    std::memcpy(&sku_id_, sku_id_char, kSkuIdLength);
    sku_id_ = ntohl(sku_id_);
  }

  CROS_CONFIG_LOG(INFO) << "Read device-tree compatible list: "
                        << compatible_devices_
                        << ", sku_id: " << sku_id_;
  return true;
}

bool CrosConfigIdentityArm::IsCompatible(const std::string& device_name) const {
  return compatible_devices_.find(device_name) != std::string::npos;
}

bool CrosConfigIdentityArm::PlatformIdentityMatch(
    const base::Value& identity_dict) const {
  const std::string* dt_compatible_match =
      identity_dict.FindStringKey("device-tree-compatible-match");
  if (!dt_compatible_match)
    return false;
  return IsCompatible(*dt_compatible_match);
}

std::string CrosConfigIdentityArm::DebugString() const {
  return base::StringPrintf(
      "arm-identity[dt-compatible=\"%s\", sku=%d, vpd=\"%s\"]",
      compatible_devices_.c_str(), sku_id_, GetVpdId().c_str());
}

}  // namespace brillo