From 36d22d82aa202bb199967e9512281e9a53db42c9 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 7 Apr 2024 21:33:14 +0200 Subject: Adding upstream version 115.7.0esr. Signed-off-by: Daniel Baumann --- dom/cache/QuotaClientImpl.h | 136 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 136 insertions(+) create mode 100644 dom/cache/QuotaClientImpl.h (limited to 'dom/cache/QuotaClientImpl.h') diff --git a/dom/cache/QuotaClientImpl.h b/dom/cache/QuotaClientImpl.h new file mode 100644 index 0000000000..62001ce8d8 --- /dev/null +++ b/dom/cache/QuotaClientImpl.h @@ -0,0 +1,136 @@ +/* -*- 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/. */ + +#ifndef mozilla_dom_cache_QuotaClientImpl_h +#define mozilla_dom_cache_QuotaClientImpl_h + +#include "mozilla/dom/QMResult.h" +#include "mozilla/dom/cache/QuotaClient.h" +#include "mozilla/dom/cache/FileUtils.h" +#include "mozilla/dom/quota/ResultExtensions.h" + +namespace mozilla::dom::cache { + +class CacheQuotaClient final : public quota::Client { + static CacheQuotaClient* sInstance; + + public: + using OriginMetadata = quota::OriginMetadata; + using PersistenceType = quota::PersistenceType; + using UsageInfo = quota::UsageInfo; + + CacheQuotaClient(); + + static CacheQuotaClient* Get(); + + virtual Type GetType() override; + + virtual Result InitOrigin( + PersistenceType aPersistenceType, const OriginMetadata& aOriginMetadata, + const AtomicBool& aCanceled) override; + + virtual nsresult InitOriginWithoutTracking( + PersistenceType aPersistenceType, const OriginMetadata& aOriginMetadata, + const AtomicBool& aCanceled) override; + + virtual Result GetUsageForOrigin( + PersistenceType aPersistenceType, const OriginMetadata& aOriginMetadata, + const AtomicBool& aCanceled) override; + + virtual void OnOriginClearCompleted(PersistenceType aPersistenceType, + const nsACString& aOrigin) override; + + void OnRepositoryClearCompleted(PersistenceType aPersistenceType) override; + + virtual void ReleaseIOThreadObjects() override; + + void AbortOperationsForLocks( + const DirectoryLockIdTable& aDirectoryLockIds) override; + + virtual void AbortOperationsForProcess( + ContentParentId aContentParentId) override; + + virtual void AbortAllOperations() override; + + virtual void StartIdleMaintenance() override; + + virtual void StopIdleMaintenance() override; + + nsresult UpgradeStorageFrom2_0To2_1(nsIFile* aDirectory) override; + + template + nsresult MaybeUpdatePaddingFileInternal(nsIFile& aBaseDir, + mozIStorageConnection& aConn, + const int64_t aIncreaseSize, + const int64_t aDecreaseSize, + Callable&& aCommitHook) { + MOZ_ASSERT(!NS_IsMainThread()); + MOZ_DIAGNOSTIC_ASSERT(aIncreaseSize >= 0); + MOZ_DIAGNOSTIC_ASSERT(aDecreaseSize >= 0); + + // Temporary should be removed at the end of each action. If not, it means + // the failure happened. + const bool temporaryPaddingFileExist = + DirectoryPaddingFileExists(aBaseDir, DirPaddingFile::TMP_FILE); + + if (aIncreaseSize == aDecreaseSize && !temporaryPaddingFileExist) { + // Early return here, since most cache actions won't modify padding size. + QM_TRY(MOZ_TO_RESULT(aCommitHook())); + + return NS_OK; + } + + // Don't delete the temporary padding file in case of an error to force the + // next action recalculate the padding size. + QM_TRY(MOZ_TO_RESULT( + UpdateDirectoryPaddingFile(aBaseDir, aConn, aIncreaseSize, + aDecreaseSize, temporaryPaddingFileExist))); + + // Don't delete the temporary padding file in case of an error to force the + // next action recalculate the padding size. + QM_TRY(MOZ_TO_RESULT(aCommitHook())); + + QM_WARNONLY_TRY(MOZ_TO_RESULT(DirectoryPaddingFinalizeWrite(aBaseDir)), + ([&aBaseDir](const nsresult) { + // Force restore file next time. + QM_WARNONLY_TRY(QM_TO_RESULT(DirectoryPaddingDeleteFile( + aBaseDir, DirPaddingFile::FILE))); + + // Ensure that we are able to force the padding file to + // be restored. + MOZ_ASSERT(DirectoryPaddingFileExists( + aBaseDir, DirPaddingFile::TMP_FILE)); + + // Since both the body file and header have been stored + // in the file-system, just make the action be resolve + // and let the padding file be restored in the next + // action. + })); + + return NS_OK; + } + + nsresult RestorePaddingFileInternal(nsIFile* aBaseDir, + mozIStorageConnection* aConn); + + nsresult WipePaddingFileInternal( + const CacheDirectoryMetadata& aDirectoryMetadata, nsIFile* aBaseDir); + + private: + ~CacheQuotaClient(); + + void InitiateShutdown() override; + bool IsShutdownCompleted() const override; + nsCString GetShutdownStatus() const override; + void ForceKillActors() override; + void FinalizeShutdown() override; + + NS_INLINE_DECL_THREADSAFE_REFCOUNTING(CacheQuotaClient, override) +}; + +} // namespace mozilla::dom::cache + +#endif // mozilla_dom_cache_QuotaClientImpl_h -- cgit v1.2.3