mirror of
https://github.com/classilla/tenfourfox.git
synced 2024-09-10 11:54:40 +00:00
242 lines
6.5 KiB
C++
242 lines
6.5 KiB
C++
|
/* -*- 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 "mozilla/dom/cache/QuotaClient.h"
|
||
|
|
||
|
#include "mozilla/DebugOnly.h"
|
||
|
#include "mozilla/dom/cache/Manager.h"
|
||
|
#include "mozilla/dom/quota/QuotaManager.h"
|
||
|
#include "mozilla/dom/quota/UsageInfo.h"
|
||
|
#include "mozilla/ipc/BackgroundParent.h"
|
||
|
#include "nsIFile.h"
|
||
|
#include "nsISimpleEnumerator.h"
|
||
|
#include "nsThreadUtils.h"
|
||
|
|
||
|
namespace {
|
||
|
|
||
|
using mozilla::DebugOnly;
|
||
|
using mozilla::dom::ContentParentId;
|
||
|
using mozilla::dom::cache::Manager;
|
||
|
using mozilla::dom::quota::Client;
|
||
|
using mozilla::dom::quota::PersistenceType;
|
||
|
using mozilla::dom::quota::QuotaManager;
|
||
|
using mozilla::dom::quota::UsageInfo;
|
||
|
using mozilla::ipc::AssertIsOnBackgroundThread;
|
||
|
|
||
|
static nsresult
|
||
|
GetBodyUsage(nsIFile* aDir, UsageInfo* aUsageInfo)
|
||
|
{
|
||
|
nsCOMPtr<nsISimpleEnumerator> entries;
|
||
|
nsresult rv = aDir->GetDirectoryEntries(getter_AddRefs(entries));
|
||
|
if (NS_WARN_IF(NS_FAILED(rv))) { return rv; }
|
||
|
|
||
|
bool hasMore;
|
||
|
while (NS_SUCCEEDED(rv = entries->HasMoreElements(&hasMore)) && hasMore &&
|
||
|
!aUsageInfo->Canceled()) {
|
||
|
nsCOMPtr<nsISupports> entry;
|
||
|
rv = entries->GetNext(getter_AddRefs(entry));
|
||
|
if (NS_WARN_IF(NS_FAILED(rv))) { return rv; }
|
||
|
|
||
|
nsCOMPtr<nsIFile> file = do_QueryInterface(entry);
|
||
|
|
||
|
bool isDir;
|
||
|
rv = file->IsDirectory(&isDir);
|
||
|
if (NS_WARN_IF(NS_FAILED(rv))) { return rv; }
|
||
|
|
||
|
if (isDir) {
|
||
|
rv = GetBodyUsage(file, aUsageInfo);
|
||
|
if (NS_WARN_IF(NS_FAILED(rv))) { return rv; }
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
int64_t fileSize;
|
||
|
rv = file->GetFileSize(&fileSize);
|
||
|
if (NS_WARN_IF(NS_FAILED(rv))) { return rv; }
|
||
|
MOZ_ASSERT(fileSize >= 0);
|
||
|
|
||
|
aUsageInfo->AppendToFileUsage(fileSize);
|
||
|
}
|
||
|
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
class CacheQuotaClient final : public Client
|
||
|
{
|
||
|
public:
|
||
|
virtual Type
|
||
|
GetType() override
|
||
|
{
|
||
|
return DOMCACHE;
|
||
|
}
|
||
|
|
||
|
virtual nsresult
|
||
|
InitOrigin(PersistenceType aPersistenceType, const nsACString& aGroup,
|
||
|
const nsACString& aOrigin, UsageInfo* aUsageInfo) override
|
||
|
{
|
||
|
// The QuotaManager passes a nullptr UsageInfo if there is no quota being
|
||
|
// enforced against the origin.
|
||
|
if (!aUsageInfo) {
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
return GetUsageForOrigin(aPersistenceType, aGroup, aOrigin, aUsageInfo);
|
||
|
}
|
||
|
|
||
|
virtual nsresult
|
||
|
GetUsageForOrigin(PersistenceType aPersistenceType, const nsACString& aGroup,
|
||
|
const nsACString& aOrigin,
|
||
|
UsageInfo* aUsageInfo) override
|
||
|
{
|
||
|
MOZ_ASSERT(aUsageInfo);
|
||
|
|
||
|
QuotaManager* qm = QuotaManager::Get();
|
||
|
MOZ_ASSERT(qm);
|
||
|
|
||
|
nsCOMPtr<nsIFile> dir;
|
||
|
nsresult rv = qm->GetDirectoryForOrigin(aPersistenceType, aOrigin,
|
||
|
getter_AddRefs(dir));
|
||
|
if (NS_WARN_IF(NS_FAILED(rv))) { return rv; }
|
||
|
|
||
|
rv = dir->Append(NS_LITERAL_STRING(DOMCACHE_DIRECTORY_NAME));
|
||
|
if (NS_WARN_IF(NS_FAILED(rv))) { return rv; }
|
||
|
|
||
|
nsCOMPtr<nsISimpleEnumerator> entries;
|
||
|
rv = dir->GetDirectoryEntries(getter_AddRefs(entries));
|
||
|
if (NS_WARN_IF(NS_FAILED(rv))) { return rv; }
|
||
|
|
||
|
bool hasMore;
|
||
|
while (NS_SUCCEEDED(rv = entries->HasMoreElements(&hasMore)) && hasMore &&
|
||
|
!aUsageInfo->Canceled()) {
|
||
|
nsCOMPtr<nsISupports> entry;
|
||
|
rv = entries->GetNext(getter_AddRefs(entry));
|
||
|
if (NS_WARN_IF(NS_FAILED(rv))) { return rv; }
|
||
|
|
||
|
nsCOMPtr<nsIFile> file = do_QueryInterface(entry);
|
||
|
|
||
|
nsAutoString leafName;
|
||
|
rv = file->GetLeafName(leafName);
|
||
|
if (NS_WARN_IF(NS_FAILED(rv))) { return rv; }
|
||
|
|
||
|
bool isDir;
|
||
|
rv = file->IsDirectory(&isDir);
|
||
|
if (NS_WARN_IF(NS_FAILED(rv))) { return rv; }
|
||
|
|
||
|
if (isDir) {
|
||
|
if (leafName.EqualsLiteral("morgue")) {
|
||
|
rv = GetBodyUsage(file, aUsageInfo);
|
||
|
if (NS_WARN_IF(NS_FAILED(rv))) { return rv; }
|
||
|
} else {
|
||
|
NS_WARNING("Unknown Cache directory found!");
|
||
|
}
|
||
|
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
// Ignore transient sqlite files and marker files
|
||
|
if (leafName.EqualsLiteral("caches.sqlite-journal") ||
|
||
|
leafName.EqualsLiteral("caches.sqlite-shm") ||
|
||
|
leafName.Find(NS_LITERAL_CSTRING("caches.sqlite-mj"), false, 0, 0) == 0 ||
|
||
|
leafName.EqualsLiteral("context_open.marker")) {
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
if (leafName.EqualsLiteral("caches.sqlite") ||
|
||
|
leafName.EqualsLiteral("caches.sqlite-wal")) {
|
||
|
int64_t fileSize;
|
||
|
rv = file->GetFileSize(&fileSize);
|
||
|
if (NS_WARN_IF(NS_FAILED(rv))) { return rv; }
|
||
|
MOZ_ASSERT(fileSize >= 0);
|
||
|
|
||
|
aUsageInfo->AppendToDatabaseUsage(fileSize);
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
NS_WARNING("Unknown Cache file found!");
|
||
|
}
|
||
|
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
virtual void
|
||
|
OnOriginClearCompleted(PersistenceType aPersistenceType,
|
||
|
const nsACString& aOrigin) override
|
||
|
{
|
||
|
// Nothing to do here.
|
||
|
}
|
||
|
|
||
|
virtual void
|
||
|
ReleaseIOThreadObjects() override
|
||
|
{
|
||
|
// Nothing to do here as the Context handles cleaning everything up
|
||
|
// automatically.
|
||
|
}
|
||
|
|
||
|
virtual void
|
||
|
AbortOperations(const nsACString& aOrigin) override
|
||
|
{
|
||
|
AssertIsOnBackgroundThread();
|
||
|
|
||
|
Manager::Abort(aOrigin);
|
||
|
}
|
||
|
|
||
|
virtual void
|
||
|
AbortOperationsForProcess(ContentParentId aContentParentId) override
|
||
|
{
|
||
|
// The Cache and Context can be shared by multiple client processes. They
|
||
|
// are not exclusively owned by a single process.
|
||
|
//
|
||
|
// As far as I can tell this is used by QuotaManager to abort operations
|
||
|
// when a particular process goes away. We definitely don't want this
|
||
|
// since we are shared. Also, the Cache actor code already properly
|
||
|
// handles asynchronous actor destruction when the child process dies.
|
||
|
//
|
||
|
// Therefore, do nothing here.
|
||
|
}
|
||
|
|
||
|
virtual void
|
||
|
StartIdleMaintenance() override
|
||
|
{ }
|
||
|
|
||
|
virtual void
|
||
|
StopIdleMaintenance() override
|
||
|
{ }
|
||
|
|
||
|
virtual void
|
||
|
ShutdownWorkThreads() override
|
||
|
{
|
||
|
AssertIsOnBackgroundThread();
|
||
|
|
||
|
// spins the event loop and synchronously shuts down all Managers
|
||
|
Manager::ShutdownAll();
|
||
|
}
|
||
|
|
||
|
private:
|
||
|
~CacheQuotaClient()
|
||
|
{
|
||
|
AssertIsOnBackgroundThread();
|
||
|
}
|
||
|
|
||
|
NS_INLINE_DECL_REFCOUNTING(CacheQuotaClient, override)
|
||
|
};
|
||
|
|
||
|
} // namespace
|
||
|
|
||
|
namespace mozilla {
|
||
|
namespace dom {
|
||
|
namespace cache {
|
||
|
|
||
|
already_AddRefed<quota::Client> CreateQuotaClient()
|
||
|
{
|
||
|
AssertIsOnBackgroundThread();
|
||
|
|
||
|
RefPtr<CacheQuotaClient> ref = new CacheQuotaClient();
|
||
|
return ref.forget();
|
||
|
}
|
||
|
|
||
|
} // namespace cache
|
||
|
} // namespace dom
|
||
|
} // namespace mozilla
|