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

#include "content/browser/indexed_db/file_path_util.h"

#include <inttypes.h>
#include <stddef.h>
#include <stdint.h>

#include <string>
#include <string_view>

#include "base/containers/span.h"
#include "base/files/file_enumerator.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/functional/function_ref.h"
#include "base/logging.h"
#include "base/metrics/histogram_functions.h"
#include "base/strings/string_util.h"
#include "base/strings/stringprintf.h"
#include "build/buildflag.h"
#include "components/base32/base32.h"
#include "components/services/storage/public/cpp/buckets/bucket_locator.h"
#include "crypto/hash.h"
#include "storage/common/database/database_identifier.h"
#include "third_party/blink/public/common/storage_key/storage_key.h"

namespace content::indexed_db {

namespace {
constexpr base::FilePath::CharType kBlobExtension[] =
    FILE_PATH_LITERAL(".blob");

// The file name used for databases that have an empty name.
constexpr char kSqliteEmptyDatabaseNameFileName[] = "0";
}  // namespace

bool ShouldUseLegacyFilePath(const storage::BucketLocator& bucket_locator) {
  return bucket_locator.storage_key.IsFirstPartyContext() &&
         bucket_locator.is_default;
}

base::FilePath GetBlobStoreFileName(
    const storage::BucketLocator& bucket_locator) {
  if (ShouldUseLegacyFilePath(bucket_locator)) {
    // First-party blob files, for legacy reasons, are stored at:
    // {{first_party_data_path}}/{{serialized_origin}}.indexeddb.blob
    return base::FilePath()
        .AppendASCII(storage::GetIdentifierFromOrigin(
            bucket_locator.storage_key.origin()))
        .AddExtension(kIndexedDBExtension)
        .AddExtension(kBlobExtension);
  }

  // Third-party blob files are stored at:
  // {{third_party_data_path}}/{{bucket_id}}/IndexedDB/indexeddb.blob
  return base::FilePath(kIndexedDBFile).AddExtension(kBlobExtension);
}

base::FilePath GetLevelDBFileName(
    const storage::BucketLocator& bucket_locator) {
  if (ShouldUseLegacyFilePath(bucket_locator)) {
    // First-party leveldb files, for legacy reasons, are stored at:
    // {{first_party_data_path}}/{{serialized_origin}}.indexeddb.leveldb
    // TODO(crbug.com/40855748): Migrate all first party buckets to the new
    // path.
    return base::FilePath()
        .AppendASCII(storage::GetIdentifierFromOrigin(
            bucket_locator.storage_key.origin()))
        .AddExtension(kIndexedDBExtension)
        .AddExtension(kLevelDBExtension);
  }

  // Third-party leveldb files are stored at:
  // {{third_party_data_path}}/{{bucket_id}}/IndexedDB/indexeddb.leveldb
  return base::FilePath(kIndexedDBFile).AddExtension(kLevelDBExtension);
}

base::FilePath GetBlobDirectoryName(const base::FilePath& path_base,
                                    int64_t database_id) {
  return path_base.AppendASCII(base::StringPrintf("%" PRIx64, database_id));
}

base::FilePath GetBlobDirectoryNameForKey(const base::FilePath& path_base,
                                          int64_t database_id,
                                          int64_t blob_number) {
  base::FilePath path = GetBlobDirectoryName(path_base, database_id);
  path = path.AppendASCII(base::StringPrintf(
      "%02x", static_cast<int>(blob_number & 0x000000000000ff00) >> 8));
  return path;
}

base::FilePath GetBlobFileNameForKey(const base::FilePath& path_base,
                                     int64_t database_id,
                                     int64_t blob_number) {
  base::FilePath path =
      GetBlobDirectoryNameForKey(path_base, database_id, blob_number);
  path = path.AppendASCII(base::StringPrintf("%" PRIx64, blob_number));
  return path;
}

bool IsPathTooLong(const base::FilePath& path) {
  int limit = base::GetMaximumPathComponentLength(path.DirName());
  if (limit < 0) {
    DPLOG(WARNING) << "GetMaximumPathComponentLength returned -1 for "
                   << path.DirName();
// In limited testing, ChromeOS returns 143, other OSes 255.
#if BUILDFLAG(IS_CHROMEOS)
    limit = 143;
#else
    limit = 255;
#endif
  }
  return path.BaseName().value().length() > static_cast<uint32_t>(limit);
}

base::FilePath GetSqliteDbDirectory(
    const storage::BucketLocator& bucket_locator) {
  if (ShouldUseLegacyFilePath(bucket_locator)) {
    // All sites share a single data path for their default bucket. Append a
    // directory for this specific site.
    return base::FilePath().AppendASCII(
        storage::GetIdentifierFromOrigin(bucket_locator.storage_key.origin()));
  }

  // The base data path is already specific to the site and bucket. The SQLite
  // DB will be stored within it.
  return base::FilePath();
}

base::FilePath DatabaseNameToFileName(std::u16string_view db_name) {
  // The goal is to create a deterministic mapping from DB name to file name.
  // There are essentially no constraints on `db_name`, in terms of length or
  // contents. File names have to conform to a certain character set and length,
  // (which depends on the file system). Thus, the space of all file names is
  // smaller than the space of all database names, and we can't simply use the
  // db name as the file name.
  //
  // To address this, we first hash the db name using SHA256, which ensures a
  // negligible probability of collisions. Then we encode using Base32, because
  // it uses only a character set that is safe for all file systems, including
  // case-insensitive ones.
  return db_name.empty()
             ? base::FilePath::FromASCII(kSqliteEmptyDatabaseNameFileName)
             : base::FilePath::FromASCII(base32::Base32Encode(
                   crypto::hash::Sha256(base::as_byte_span(db_name)),
                   base32::Base32EncodePolicy::OMIT_PADDING));
}

void EnumerateDatabasesInDirectory(
    const base::FilePath& directory,
    base::FunctionRef<void(const base::FilePath& path)> ref) {
  base::FileEnumerator enumerator(directory, /*recursive=*/false,
                                  base::FileEnumerator::FILES);
  enumerator.ForEach([&](const base::FilePath& path) {
    if (path.BaseName() ==
        base::FilePath::FromASCII(kSqliteEmptyDatabaseNameFileName)) {
      ref(path);
      return;
    }

    std::string ascii_name = path.BaseName().MaybeAsASCII();
    if (ascii_name.empty()) {
      return;
    }

    if (base32::Base32Decode(ascii_name).size() !=
        crypto::hash::DigestSizeForHashKind(crypto::hash::HashKind::kSha256)) {
      return;
    }

    ref(path);
  });
}

}  // namespace content::indexed_db