1 files changed, 235 insertions, 0 deletions
diff --git a/dom/fs/parent/FileSystemQuotaClient.cpp b/dom/fs/parent/FileSystemQuotaClient.cpp
new file mode 100644
index 0000000000..6c0271060f
--- /dev/null
+++ b/dom/fs/parent/FileSystemQuotaClient.cpp
@@ -0,0 +1,235 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "FileSystemQuotaClient.h"
+
+#include "ResultStatement.h"
+#include "datamodel/FileSystemDatabaseManager.h"
+#include "datamodel/FileSystemFileManager.h"
+#include "mozIStorageService.h"
+#include "mozStorageCID.h"
+#include "mozilla/dom/FileSystemDataManager.h"
+#include "mozilla/dom/quota/Assertions.h"
+#include "mozilla/dom/quota/Client.h"
+#include "mozilla/dom/quota/QuotaCommon.h"
+#include "mozilla/dom/quota/QuotaManager.h"
+#include "mozilla/dom/quota/ResultExtensions.h"
+#include "mozilla/dom/quota/UsageInfo.h"
+#include "mozilla/ipc/BackgroundParent.h"
+#include "nsIFile.h"
+
+namespace mozilla::dom::fs {
+
+namespace {
+
+auto toNSResult = [](const auto& aRv) { return ToNSResult(aRv); };
+
+class QuotaClient final : public mozilla::dom::quota::Client {
+ public:
+  QuotaClient();
+
+  NS_INLINE_DECL_THREADSAFE_REFCOUNTING(mozilla::dom::fs::QuotaClient, override)
+
+  Type GetType() override;
+
+  Result<quota::UsageInfo, nsresult> InitOrigin(
+      quota::PersistenceType aPersistenceType,
+      const quota::OriginMetadata& aOriginMetadata,
+      const AtomicBool& aCanceled) override;
+
+  nsresult InitOriginWithoutTracking(
+      quota::PersistenceType aPersistenceType,
+      const quota::OriginMetadata& aOriginMetadata,
+      const AtomicBool& aCanceled) override;
+
+  Result<quota::UsageInfo, nsresult> GetUsageForOrigin(
+      quota::PersistenceType aPersistenceType,
+      const quota::OriginMetadata& aOriginMetadata,
+      const AtomicBool& aCanceled) override;
+
+  void OnOriginClearCompleted(quota::PersistenceType aPersistenceType,
+                              const nsACString& aOrigin) override;
+
+  void ReleaseIOThreadObjects() override;
+
+  void AbortOperationsForLocks(
+      const DirectoryLockIdTable& aDirectoryLockIds) override;
+
+  void AbortOperationsForProcess(ContentParentId aContentParentId) override;
+
+  void AbortAllOperations() override;
+
+  void StartIdleMaintenance() override;
+
+  void StopIdleMaintenance() override;
+
+ private:
+  ~QuotaClient() = default;
+
+  void InitiateShutdown() override;
+  bool IsShutdownCompleted() const override;
+  nsCString GetShutdownStatus() const override;
+  void ForceKillActors() override;
+  void FinalizeShutdown() override;
+};
+
+Result<ResultConnection, QMResult> GetStorageConnection(const Origin& aOrigin) {
+  QM_TRY_INSPECT(const nsCOMPtr<nsIFile>& databaseFile,
+                 data::GetDatabaseFile(aOrigin));
+
+  QM_TRY_INSPECT(
+      const auto& storageService,
+      QM_TO_RESULT_TRANSFORM(MOZ_TO_RESULT_GET_TYPED(
+          nsCOMPtr<mozIStorageService>, MOZ_SELECT_OVERLOAD(do_GetService),
+          MOZ_STORAGE_SERVICE_CONTRACTID)));
+
+  QM_TRY_UNWRAP(
+      auto connection,
+      QM_TO_RESULT_TRANSFORM(MOZ_TO_RESULT_INVOKE_MEMBER_TYPED(
+          nsCOMPtr<mozIStorageConnection>, storageService, OpenDatabase,
+          databaseFile, mozIStorageService::CONNECTION_DEFAULT)));
+
+  ResultConnection result(connection);
+
+  return result;
+}
+
+}  // namespace
+
+QuotaClient::QuotaClient() { ::mozilla::ipc::AssertIsOnBackgroundThread(); }
+
+mozilla::dom::quota::Client::Type QuotaClient::GetType() {
+  return quota::Client::Type::FILESYSTEM;
+}
+
+Result<quota::UsageInfo, nsresult> QuotaClient::InitOrigin(
+    quota::PersistenceType aPersistenceType,
+    const quota::OriginMetadata& aOriginMetadata, const AtomicBool& aCanceled) {
+  quota::AssertIsOnIOThread();
+
+  const Origin& origin = aOriginMetadata.mOrigin;
+
+  {
+    QM_TRY_INSPECT(const nsCOMPtr<nsIFile>& databaseFile,
+                   data::GetDatabaseFile(origin).mapErr(toNSResult));
+
+    bool exists = false;
+    QM_TRY(MOZ_TO_RESULT(databaseFile->Exists(&exists)));
+    // If database doesn't already exist, we do not create it
+    if (!exists) {
+      return quota::UsageInfo();
+    }
+  }
+
+  QM_TRY_INSPECT(const ResultConnection& conn,
+                 GetStorageConnection(origin).mapErr(toNSResult));
+
+  return data::FileSystemDatabaseManager::GetUsage(conn, origin)
+      .mapErr(toNSResult);
+}
+
+nsresult QuotaClient::InitOriginWithoutTracking(
+    quota::PersistenceType /* aPersistenceType */,
+    const quota::OriginMetadata& /* aOriginMetadata */,
+    const AtomicBool& /* aCanceled */) {
+  quota::AssertIsOnIOThread();
+
+  // This is called when a storage/permanent/${origin}/fs directory exists. Even
+  // though this shouldn't happen with a "good" profile, we shouldn't return an
+  // error here, since that would cause origin initialization to fail. We just
+  // warn and otherwise ignore that.
+  UNKNOWN_FILE_WARNING(
+      NS_LITERAL_STRING_FROM_CSTRING(FILESYSTEM_DIRECTORY_NAME));
+
+  return NS_OK;
+}
+
+Result<quota::UsageInfo, nsresult> QuotaClient::GetUsageForOrigin(
+    quota::PersistenceType aPersistenceType,
+    const quota::OriginMetadata& aOriginMetadata,
+    const AtomicBool& /* aCanceled */) {
+  quota::AssertIsOnIOThread();
+
+  MOZ_ASSERT(aPersistenceType ==
+             quota::PersistenceType::PERSISTENCE_TYPE_DEFAULT);
+
+  quota::QuotaManager* quotaManager = quota::QuotaManager::Get();
+  MOZ_ASSERT(quotaManager);
+
+  // We can't open the database at this point because the quota manager may not
+  // allow it. Use the cached value instead.
+  return quotaManager->GetUsageForClient(aPersistenceType, aOriginMetadata,
+                                         quota::Client::FILESYSTEM);
+}
+
+void QuotaClient::OnOriginClearCompleted(
+    quota::PersistenceType aPersistenceType, const nsACString& aOrigin) {
+  quota::AssertIsOnIOThread();
+}
+
+void QuotaClient::ReleaseIOThreadObjects() { quota::AssertIsOnIOThread(); }
+
+void QuotaClient::AbortOperationsForLocks(
+    const DirectoryLockIdTable& aDirectoryLockIds) {
+  ::mozilla::ipc::AssertIsOnBackgroundThread();
+
+  data::FileSystemDataManager::AbortOperationsForLocks(aDirectoryLockIds);
+}
+
+void QuotaClient::AbortOperationsForProcess(ContentParentId aContentParentId) {
+  ::mozilla::ipc::AssertIsOnBackgroundThread();
+}
+
+void QuotaClient::AbortAllOperations() {
+  ::mozilla::ipc::AssertIsOnBackgroundThread();
+}
+
+void QuotaClient::StartIdleMaintenance() {
+  ::mozilla::ipc::AssertIsOnBackgroundThread();
+}
+
+void QuotaClient::StopIdleMaintenance() {
+  ::mozilla::ipc::AssertIsOnBackgroundThread();
+}
+
+void QuotaClient::InitiateShutdown() {
+  ::mozilla::ipc::AssertIsOnBackgroundThread();
+
+  data::FileSystemDataManager::InitiateShutdown();
+}
+
+bool QuotaClient::IsShutdownCompleted() const {
+  ::mozilla::ipc::AssertIsOnBackgroundThread();
+
+  return data::FileSystemDataManager::IsShutdownCompleted();
+}
+
+nsCString QuotaClient::GetShutdownStatus() const {
+  ::mozilla::ipc::AssertIsOnBackgroundThread();
+
+  return "Not implemented"_ns;
+}
+
+void QuotaClient::ForceKillActors() {
+  ::mozilla::ipc::AssertIsOnBackgroundThread();
+
+  // Hopefully not needed.
+}
+
+void QuotaClient::FinalizeShutdown() {
+  ::mozilla::ipc::AssertIsOnBackgroundThread();
+
+  // Empty for now.
+}
+
+already_AddRefed<mozilla::dom::quota::Client> CreateQuotaClient() {
+  ::mozilla::ipc::AssertIsOnBackgroundThread();
+
+  RefPtr<QuotaClient> client = new fs::QuotaClient();
+  return client.forget();
+}
+
+}  // namespace mozilla::dom::fs