// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/crx_file/id_util.h"

#include <stdint.h>

#include <string_view>

#include "base/files/file_path.h"
#include "base/hash/sha1.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/string_util.h"
#include "build/build_config.h"
#include "crypto/hash.h"
#include "third_party/abseil-cpp/absl/strings/ascii.h"

namespace {

// Converts a normal hexadecimal string into the alphabet used by extensions.
// We use the characters 'a'-'p' instead of '0'-'f' to avoid ever having a
// completely numeric host, since some software interprets that as an IP
// address.
static void ConvertHexadecimalToIDAlphabet(std::string* id) {
  for (auto& ch : *id) {
    int val;
    if (base::HexStringToInt(std::string_view(&ch, 1), &val)) {
      ch = 'a' + val;
    } else {
      ch = 'a';
    }
  }
}

}  // namespace

namespace crx_file::id_util {

// First 16 bytes of SHA256 hashed public key.
const size_t kIdSize = 16;

std::string GenerateId(std::string_view input) {
  return GenerateId(base::as_byte_span(input));
}

std::string GenerateId(base::span<const uint8_t> input) {
  return GenerateIdFromHash(crypto::hash::Sha256(input));
}

std::string GenerateIdFromHash(base::span<const uint8_t> hash) {
  std::string result = base::HexEncode(hash.first(kIdSize));
  ConvertHexadecimalToIDAlphabet(&result);
  return result;
}

std::string GenerateIdFromHex(const std::string& input) {
  std::string output = input;
  ConvertHexadecimalToIDAlphabet(&output);
  return output;
}

std::string GenerateIdForPath(const base::FilePath& path) {
  base::FilePath new_path = MaybeNormalizePath(path);
  return GenerateId(base::as_byte_span(new_path.value()));
}

std::string HashedIdInHex(const std::string& id) {
  return base::HexEncode(base::SHA1Hash(base::as_byte_span(id)));
}

base::FilePath MaybeNormalizePath(const base::FilePath& path) {
#if BUILDFLAG(IS_WIN)
  // Normalize any drive letter to upper-case. We do this for consistency with
  // net_utils::FilePathToFileURL(), which does the same thing, to make string
  // comparisons simpler.
  base::FilePath::StringType path_str = path.value();
  if (path_str.size() >= 2 && path_str[0] >= L'a' && path_str[0] <= L'z' &&
      path_str[1] == L':') {
    path_str[0] = absl::ascii_toupper(static_cast<unsigned char>(path_str[0]));
  }

  return base::FilePath(path_str);
#else
  return path;
#endif
}

bool IdIsValid(std::string_view id) {
  // Verify that the id is legal.
  if (id.size() != (crx_file::id_util::kIdSize * 2)) {
    return false;
  }

  for (char ch : id) {
    ch = base::ToLowerASCII(ch);
    if (ch < 'a' || ch > 'p') {
      return false;
    }
  }

  return true;
}

}  // namespace crx_file::id_util