314 lines
11 KiB
C++
314 lines
11 KiB
C++
|
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
|
||
|
// Use of this source code is governed by a BSD-style license that can be
|
||
|
// found in the LICENSE file.
|
||
|
|
||
|
#include "libcef/browser_database_system.h"
|
||
|
#include "libcef/cef_context.h"
|
||
|
|
||
|
#include "base/auto_reset.h"
|
||
|
#include "base/bind.h"
|
||
|
#include "base/bind_helpers.h"
|
||
|
#include "base/file_util.h"
|
||
|
#include "base/message_loop.h"
|
||
|
#include "base/message_loop_proxy.h"
|
||
|
#include "base/synchronization/waitable_event.h"
|
||
|
#include "base/threading/platform_thread.h"
|
||
|
#include "base/utf_string_conversions.h"
|
||
|
#include "third_party/sqlite/sqlite3.h"
|
||
|
#include "third_party/WebKit/Source/WebKit/chromium/public/WebDatabase.h"
|
||
|
#include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebString.h"
|
||
|
#include "webkit/database/database_util.h"
|
||
|
#include "webkit/database/vfs_backend.h"
|
||
|
|
||
|
using webkit_database::DatabaseTracker;
|
||
|
using webkit_database::DatabaseUtil;
|
||
|
using webkit_database::OriginInfo;
|
||
|
using webkit_database::VfsBackend;
|
||
|
|
||
|
BrowserDatabaseSystem* BrowserDatabaseSystem::instance_ = NULL;
|
||
|
|
||
|
BrowserDatabaseSystem* BrowserDatabaseSystem::GetInstance() {
|
||
|
DCHECK(instance_);
|
||
|
return instance_;
|
||
|
}
|
||
|
|
||
|
BrowserDatabaseSystem::BrowserDatabaseSystem()
|
||
|
: db_thread_("BrowserDBThread"),
|
||
|
quota_per_origin_(5 * 1024 * 1024),
|
||
|
open_connections_(new webkit_database::DatabaseConnectionsWrapper) {
|
||
|
DCHECK(!instance_);
|
||
|
instance_ = this;
|
||
|
db_tracker_ =
|
||
|
new DatabaseTracker(_Context->cache_path(), false, NULL, NULL, NULL);
|
||
|
db_tracker_->AddObserver(this);
|
||
|
db_thread_.Start();
|
||
|
db_thread_proxy_ = db_thread_.message_loop_proxy();
|
||
|
}
|
||
|
|
||
|
BrowserDatabaseSystem::~BrowserDatabaseSystem() {
|
||
|
base::WaitableEvent done_event(false, false);
|
||
|
db_thread_proxy_->PostTask(
|
||
|
FROM_HERE,
|
||
|
base::Bind(&BrowserDatabaseSystem::ThreadCleanup,
|
||
|
base::Unretained(this), &done_event));
|
||
|
done_event.Wait();
|
||
|
instance_ = NULL;
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::databaseOpened(
|
||
|
const WebKit::WebDatabase& database) {
|
||
|
string16 origin_identifier = database.securityOrigin().databaseIdentifier();
|
||
|
string16 database_name = database.name();
|
||
|
open_connections_->AddOpenConnection(origin_identifier, database_name);
|
||
|
db_thread_proxy_->PostTask(
|
||
|
FROM_HERE,
|
||
|
base::Bind(&BrowserDatabaseSystem::DatabaseOpened,
|
||
|
base::Unretained(this),
|
||
|
origin_identifier,
|
||
|
database_name, database.displayName(),
|
||
|
database.estimatedSize()));
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::databaseModified(
|
||
|
const WebKit::WebDatabase& database) {
|
||
|
db_thread_proxy_->PostTask(
|
||
|
FROM_HERE,
|
||
|
base::Bind(&BrowserDatabaseSystem::DatabaseModified,
|
||
|
base::Unretained(this),
|
||
|
database.securityOrigin().databaseIdentifier(),
|
||
|
database.name()));
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::databaseClosed(
|
||
|
const WebKit::WebDatabase& database) {
|
||
|
string16 origin_identifier = database.securityOrigin().databaseIdentifier();
|
||
|
string16 database_name = database.name();
|
||
|
db_thread_proxy_->PostTask(
|
||
|
FROM_HERE,
|
||
|
base::Bind(&BrowserDatabaseSystem::DatabaseClosed,
|
||
|
base::Unretained(this), origin_identifier, database_name));
|
||
|
}
|
||
|
|
||
|
base::PlatformFile BrowserDatabaseSystem::OpenFile(
|
||
|
const string16& vfs_file_name, int desired_flags) {
|
||
|
base::PlatformFile result = base::kInvalidPlatformFileValue;
|
||
|
base::WaitableEvent done_event(false, false);
|
||
|
db_thread_proxy_->PostTask(
|
||
|
FROM_HERE,
|
||
|
base::Bind(&BrowserDatabaseSystem::VfsOpenFile,
|
||
|
base::Unretained(this),
|
||
|
vfs_file_name, desired_flags,
|
||
|
&result, &done_event));
|
||
|
done_event.Wait();
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
int BrowserDatabaseSystem::DeleteFile(
|
||
|
const string16& vfs_file_name, bool sync_dir) {
|
||
|
int result = SQLITE_OK;
|
||
|
base::WaitableEvent done_event(false, false);
|
||
|
db_thread_proxy_->PostTask(
|
||
|
FROM_HERE,
|
||
|
base::Bind(&BrowserDatabaseSystem::VfsDeleteFile,
|
||
|
base::Unretained(this),
|
||
|
vfs_file_name, sync_dir,
|
||
|
&result, &done_event));
|
||
|
done_event.Wait();
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
uint32 BrowserDatabaseSystem::GetFileAttributes(const string16& vfs_file_name) {
|
||
|
uint32 result = 0;
|
||
|
base::WaitableEvent done_event(false, false);
|
||
|
db_thread_proxy_->PostTask(
|
||
|
FROM_HERE,
|
||
|
base::Bind(&BrowserDatabaseSystem::VfsGetFileAttributes,
|
||
|
base::Unretained(this), vfs_file_name, &result, &done_event));
|
||
|
done_event.Wait();
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
int64 BrowserDatabaseSystem::GetFileSize(const string16& vfs_file_name) {
|
||
|
int64 result = 0;
|
||
|
base::WaitableEvent done_event(false, false);
|
||
|
db_thread_proxy_->PostTask(
|
||
|
FROM_HERE,
|
||
|
base::Bind(&BrowserDatabaseSystem::VfsGetFileSize,
|
||
|
base::Unretained(this), vfs_file_name, &result, &done_event));
|
||
|
done_event.Wait();
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
int64 BrowserDatabaseSystem::GetSpaceAvailable(
|
||
|
const string16& origin_identifier) {
|
||
|
int64 result = 0;
|
||
|
base::WaitableEvent done_event(false, false);
|
||
|
db_thread_proxy_->PostTask(
|
||
|
FROM_HERE,
|
||
|
base::Bind(&BrowserDatabaseSystem::VfsGetSpaceAvailable,
|
||
|
base::Unretained(this), origin_identifier,
|
||
|
&result, &done_event));
|
||
|
done_event.Wait();
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::ClearAllDatabases() {
|
||
|
open_connections_->WaitForAllDatabasesToClose();
|
||
|
db_thread_proxy_->PostTask(
|
||
|
FROM_HERE,
|
||
|
base::Bind(&BrowserDatabaseSystem::ResetTracker, base::Unretained(this)));
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::SetDatabaseQuota(int64 quota) {
|
||
|
if (!db_thread_proxy_->BelongsToCurrentThread()) {
|
||
|
db_thread_proxy_->PostTask(
|
||
|
FROM_HERE,
|
||
|
base::Bind(&BrowserDatabaseSystem::SetDatabaseQuota,
|
||
|
base::Unretained(this), quota));
|
||
|
return;
|
||
|
}
|
||
|
quota_per_origin_ = quota;
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::DatabaseOpened(const string16& origin_identifier,
|
||
|
const string16& database_name,
|
||
|
const string16& description,
|
||
|
int64 estimated_size) {
|
||
|
DCHECK(db_thread_proxy_->BelongsToCurrentThread());
|
||
|
int64 database_size = 0;
|
||
|
db_tracker_->DatabaseOpened(
|
||
|
origin_identifier, database_name, description,
|
||
|
estimated_size, &database_size);
|
||
|
OnDatabaseSizeChanged(origin_identifier, database_name,
|
||
|
database_size);
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::DatabaseModified(const string16& origin_identifier,
|
||
|
const string16& database_name) {
|
||
|
DCHECK(db_thread_proxy_->BelongsToCurrentThread());
|
||
|
db_tracker_->DatabaseModified(origin_identifier, database_name);
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::DatabaseClosed(const string16& origin_identifier,
|
||
|
const string16& database_name) {
|
||
|
DCHECK(db_thread_proxy_->BelongsToCurrentThread());
|
||
|
db_tracker_->DatabaseClosed(origin_identifier, database_name);
|
||
|
open_connections_->RemoveOpenConnection(origin_identifier, database_name);
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::OnDatabaseSizeChanged(
|
||
|
const string16& origin_identifier,
|
||
|
const string16& database_name,
|
||
|
int64 database_size) {
|
||
|
DCHECK(db_thread_proxy_->BelongsToCurrentThread());
|
||
|
// We intentionally call into webkit on our background db_thread_
|
||
|
// to better emulate what happens in chrome where this method is
|
||
|
// invoked on the background ipc thread.
|
||
|
WebKit::WebDatabase::updateDatabaseSize(
|
||
|
origin_identifier, database_name, database_size);
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::OnDatabaseScheduledForDeletion(
|
||
|
const string16& origin_identifier,
|
||
|
const string16& database_name) {
|
||
|
DCHECK(db_thread_proxy_->BelongsToCurrentThread());
|
||
|
// We intentionally call into webkit on our background db_thread_
|
||
|
// to better emulate what happens in chrome where this method is
|
||
|
// invoked on the background ipc thread.
|
||
|
WebKit::WebDatabase::closeDatabaseImmediately(
|
||
|
origin_identifier, database_name);
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::VfsOpenFile(
|
||
|
const string16& vfs_file_name, int desired_flags,
|
||
|
base::PlatformFile* file_handle, base::WaitableEvent* done_event ) {
|
||
|
DCHECK(db_thread_proxy_->BelongsToCurrentThread());
|
||
|
FilePath file_name = GetFullFilePathForVfsFile(vfs_file_name);
|
||
|
if (file_name.empty()) {
|
||
|
VfsBackend::OpenTempFileInDirectory(
|
||
|
db_tracker_->DatabaseDirectory(), desired_flags, file_handle);
|
||
|
} else {
|
||
|
VfsBackend::OpenFile(file_name, desired_flags, file_handle);
|
||
|
}
|
||
|
done_event->Signal();
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::VfsDeleteFile(
|
||
|
const string16& vfs_file_name, bool sync_dir,
|
||
|
int* result, base::WaitableEvent* done_event) {
|
||
|
DCHECK(db_thread_proxy_->BelongsToCurrentThread());
|
||
|
// We try to delete the file multiple times, because that's what the default
|
||
|
// VFS does (apparently deleting a file can sometimes fail on Windows).
|
||
|
// We sleep for 10ms between retries for the same reason.
|
||
|
const int kNumDeleteRetries = 3;
|
||
|
int num_retries = 0;
|
||
|
*result = SQLITE_OK;
|
||
|
FilePath file_name = GetFullFilePathForVfsFile(vfs_file_name);
|
||
|
do {
|
||
|
*result = VfsBackend::DeleteFile(file_name, sync_dir);
|
||
|
} while ((++num_retries < kNumDeleteRetries) &&
|
||
|
(*result == SQLITE_IOERR_DELETE) &&
|
||
|
(base::PlatformThread::Sleep(
|
||
|
base::TimeDelta::FromMilliseconds(10)),
|
||
|
1));
|
||
|
|
||
|
done_event->Signal();
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::VfsGetFileAttributes(
|
||
|
const string16& vfs_file_name,
|
||
|
uint32* result, base::WaitableEvent* done_event) {
|
||
|
DCHECK(db_thread_proxy_->BelongsToCurrentThread());
|
||
|
*result = VfsBackend::GetFileAttributes(
|
||
|
GetFullFilePathForVfsFile(vfs_file_name));
|
||
|
done_event->Signal();
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::VfsGetFileSize(
|
||
|
const string16& vfs_file_name,
|
||
|
int64* result, base::WaitableEvent* done_event) {
|
||
|
DCHECK(db_thread_proxy_->BelongsToCurrentThread());
|
||
|
*result = VfsBackend::GetFileSize(GetFullFilePathForVfsFile(vfs_file_name));
|
||
|
done_event->Signal();
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::VfsGetSpaceAvailable(
|
||
|
const string16& origin_identifier,
|
||
|
int64* result, base::WaitableEvent* done_event) {
|
||
|
DCHECK(db_thread_proxy_->BelongsToCurrentThread());
|
||
|
// This method isn't actually part of the "vfs" interface, but it is
|
||
|
// used from within webcore and handled here in the same fashion.
|
||
|
OriginInfo info;
|
||
|
if (db_tracker_->GetOriginInfo(origin_identifier, &info)) {
|
||
|
int64 space_available = quota_per_origin_ - info.TotalSize();
|
||
|
*result = space_available < 0 ? 0 : space_available;
|
||
|
} else {
|
||
|
NOTREACHED();
|
||
|
*result = 0;
|
||
|
}
|
||
|
done_event->Signal();
|
||
|
}
|
||
|
|
||
|
FilePath BrowserDatabaseSystem::GetFullFilePathForVfsFile(
|
||
|
const string16& vfs_file_name) {
|
||
|
DCHECK(db_thread_proxy_->BelongsToCurrentThread());
|
||
|
if (vfs_file_name.empty()) // temp file, used for vacuuming
|
||
|
return FilePath();
|
||
|
return DatabaseUtil::GetFullFilePathForVfsFile(
|
||
|
db_tracker_.get(), vfs_file_name);
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::ResetTracker() {
|
||
|
DCHECK(db_thread_proxy_->BelongsToCurrentThread());
|
||
|
db_tracker_->CloseTrackerDatabaseAndClearCaches();
|
||
|
file_util::Delete(db_tracker_->DatabaseDirectory(), true);
|
||
|
}
|
||
|
|
||
|
void BrowserDatabaseSystem::ThreadCleanup(base::WaitableEvent* done_event) {
|
||
|
ResetTracker();
|
||
|
db_tracker_->RemoveObserver(this);
|
||
|
db_tracker_ = NULL;
|
||
|
done_event->Signal();
|
||
|
}
|
||
|
|