449 lines
15 KiB
C++
449 lines
15 KiB
C++
// Copyright (c) 2009 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 "browser_appcache_system.h"
|
|
#include "browser_resource_loader_bridge.h"
|
|
|
|
#include "base/callback.h"
|
|
#include "base/lock.h"
|
|
#include "base/task.h"
|
|
#include "base/waitable_event.h"
|
|
#include "webkit/appcache/appcache_interceptor.h"
|
|
#include "webkit/appcache/web_application_cache_host_impl.h"
|
|
|
|
using WebKit::WebApplicationCacheHost;
|
|
using WebKit::WebApplicationCacheHostClient;
|
|
using appcache::WebApplicationCacheHostImpl;
|
|
using appcache::AppCacheBackendImpl;
|
|
using appcache::AppCacheInterceptor;
|
|
using appcache::AppCacheThread;
|
|
|
|
namespace appcache {
|
|
|
|
// An impl of AppCacheThread we need to provide to the appcache lib.
|
|
|
|
bool AppCacheThread::PostTask(
|
|
int id,
|
|
const tracked_objects::Location& from_here,
|
|
Task* task) {
|
|
if (BrowserAppCacheSystem::thread_provider()) {
|
|
return BrowserAppCacheSystem::thread_provider()->PostTask(
|
|
id, from_here, task);
|
|
}
|
|
scoped_ptr<Task> task_ptr(task);
|
|
MessageLoop* loop = BrowserAppCacheSystem::GetMessageLoop(id);
|
|
if (loop)
|
|
loop->PostTask(from_here, task_ptr.release());
|
|
return loop ? true : false;
|
|
}
|
|
|
|
bool AppCacheThread::CurrentlyOn(int id) {
|
|
if (BrowserAppCacheSystem::thread_provider())
|
|
return BrowserAppCacheSystem::thread_provider()->CurrentlyOn(id);
|
|
return MessageLoop::current() == BrowserAppCacheSystem::GetMessageLoop(id);
|
|
}
|
|
|
|
} // namespace appcache
|
|
|
|
// BrowserFrontendProxy --------------------------------------------------------
|
|
// Proxies method calls from the backend IO thread to the frontend UI thread.
|
|
|
|
class BrowserFrontendProxy
|
|
: public base::RefCountedThreadSafe<BrowserFrontendProxy>,
|
|
public appcache::AppCacheFrontend {
|
|
public:
|
|
explicit BrowserFrontendProxy(BrowserAppCacheSystem* appcache_system)
|
|
: system_(appcache_system) {
|
|
}
|
|
|
|
void clear_appcache_system() { system_ = NULL; }
|
|
|
|
virtual void OnCacheSelected(int host_id, int64 cache_id ,
|
|
appcache::Status status) {
|
|
if (!system_)
|
|
return;
|
|
if (system_->is_io_thread())
|
|
system_->ui_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserFrontendProxy::OnCacheSelected,
|
|
host_id, cache_id, status));
|
|
else if (system_->is_ui_thread())
|
|
system_->frontend_impl_.OnCacheSelected(host_id, cache_id, status);
|
|
else
|
|
NOTREACHED();
|
|
}
|
|
|
|
virtual void OnStatusChanged(const std::vector<int>& host_ids,
|
|
appcache::Status status) {
|
|
if (!system_)
|
|
return;
|
|
if (system_->is_io_thread())
|
|
system_->ui_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserFrontendProxy::OnStatusChanged, host_ids, status));
|
|
else if (system_->is_ui_thread())
|
|
system_->frontend_impl_.OnStatusChanged(host_ids, status);
|
|
else
|
|
NOTREACHED();
|
|
}
|
|
|
|
virtual void OnEventRaised(const std::vector<int>& host_ids,
|
|
appcache::EventID event_id) {
|
|
if (!system_)
|
|
return;
|
|
if (system_->is_io_thread())
|
|
system_->ui_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserFrontendProxy::OnEventRaised, host_ids, event_id));
|
|
else if (system_->is_ui_thread())
|
|
system_->frontend_impl_.OnEventRaised(host_ids, event_id);
|
|
else
|
|
NOTREACHED();
|
|
}
|
|
|
|
virtual void OnProgressEventRaised(const std::vector<int>& host_ids,
|
|
const GURL& url,
|
|
int num_total, int num_complete) {
|
|
if (!system_)
|
|
return;
|
|
if (system_->is_io_thread())
|
|
system_->ui_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserFrontendProxy::OnProgressEventRaised, host_ids, url, num_total, num_complete));
|
|
else if (system_->is_ui_thread())
|
|
system_->frontend_impl_.OnProgressEventRaised(host_ids, url, num_total, num_complete);
|
|
else
|
|
NOTREACHED();
|
|
}
|
|
|
|
virtual void OnContentBlocked(int host_id){
|
|
if (!system_)
|
|
return;
|
|
if (system_->is_io_thread())
|
|
system_->ui_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserFrontendProxy::OnContentBlocked, host_id));
|
|
else if (system_->is_ui_thread())
|
|
system_->frontend_impl_.OnContentBlocked(host_id);
|
|
else
|
|
NOTREACHED();
|
|
|
|
}
|
|
|
|
virtual void OnLogMessage(int host_id, appcache::LogLevel log_level,
|
|
const std::string& message) {
|
|
|
|
if (!system_)
|
|
return;
|
|
if (system_->is_io_thread())
|
|
system_->ui_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserFrontendProxy::OnLogMessage, host_id, log_level, message));
|
|
else if (system_->is_ui_thread())
|
|
system_->frontend_impl_.OnLogMessage(host_id, log_level, message);
|
|
else
|
|
NOTREACHED();
|
|
}
|
|
|
|
private:
|
|
friend class base::RefCountedThreadSafe<BrowserFrontendProxy>;
|
|
|
|
~BrowserFrontendProxy() {}
|
|
|
|
BrowserAppCacheSystem* system_;
|
|
};
|
|
|
|
|
|
// BrowserBackendProxy --------------------------------------------------------
|
|
// Proxies method calls from the frontend UI thread to the backend IO thread.
|
|
|
|
class BrowserBackendProxy
|
|
: public base::RefCountedThreadSafe<BrowserBackendProxy>,
|
|
public appcache::AppCacheBackend {
|
|
public:
|
|
explicit BrowserBackendProxy(BrowserAppCacheSystem* appcache_system)
|
|
: system_(appcache_system), event_(true, false) {
|
|
get_status_callback_.reset(
|
|
NewCallback(this, &BrowserBackendProxy::GetStatusCallback));
|
|
start_update_callback_.reset(
|
|
NewCallback(this, &BrowserBackendProxy::StartUpdateCallback));
|
|
swap_cache_callback_.reset(
|
|
NewCallback(this, &BrowserBackendProxy::SwapCacheCallback));
|
|
}
|
|
|
|
virtual void RegisterHost(int host_id) {
|
|
if (system_->is_ui_thread()) {
|
|
system_->io_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserBackendProxy::RegisterHost, host_id));
|
|
} else if (system_->is_io_thread()) {
|
|
system_->backend_impl_->RegisterHost(host_id);
|
|
} else {
|
|
NOTREACHED();
|
|
}
|
|
}
|
|
|
|
virtual void UnregisterHost(int host_id) {
|
|
if (system_->is_ui_thread()) {
|
|
system_->io_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserBackendProxy::UnregisterHost, host_id));
|
|
} else if (system_->is_io_thread()) {
|
|
system_->backend_impl_->UnregisterHost(host_id);
|
|
} else {
|
|
NOTREACHED();
|
|
}
|
|
}
|
|
|
|
virtual void SelectCache(int host_id,
|
|
const GURL& document_url,
|
|
const int64 cache_document_was_loaded_from,
|
|
const GURL& manifest_url) {
|
|
if (system_->is_ui_thread()) {
|
|
system_->io_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserBackendProxy::SelectCache, host_id, document_url,
|
|
cache_document_was_loaded_from, manifest_url));
|
|
} else if (system_->is_io_thread()) {
|
|
system_->backend_impl_->SelectCache(host_id, document_url,
|
|
cache_document_was_loaded_from,
|
|
manifest_url);
|
|
} else {
|
|
NOTREACHED();
|
|
}
|
|
}
|
|
|
|
virtual void SelectCacheForWorker(
|
|
int host_id,
|
|
int parent_process_id,
|
|
int parent_host_id) {
|
|
|
|
if (system_->is_ui_thread()) {
|
|
system_->io_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserBackendProxy::SelectCacheForWorker, host_id, parent_process_id,
|
|
parent_host_id));
|
|
} else if (system_->is_io_thread()) {
|
|
system_->backend_impl_->SelectCacheForWorker(host_id, parent_process_id,
|
|
parent_host_id);
|
|
} else {
|
|
NOTREACHED();
|
|
}
|
|
}
|
|
|
|
virtual void SelectCacheForSharedWorker(
|
|
int host_id,
|
|
int64 appcache_id){
|
|
if (system_->is_ui_thread()) {
|
|
system_->io_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserBackendProxy::SelectCacheForSharedWorker, host_id, appcache_id));
|
|
} else if (system_->is_io_thread()) {
|
|
system_->backend_impl_->SelectCacheForSharedWorker(host_id, appcache_id);
|
|
} else {
|
|
NOTREACHED();
|
|
}
|
|
}
|
|
|
|
virtual void MarkAsForeignEntry(int host_id, const GURL& document_url,
|
|
int64 cache_document_was_loaded_from) {
|
|
if (system_->is_ui_thread()) {
|
|
system_->io_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserBackendProxy::MarkAsForeignEntry, host_id, document_url,
|
|
cache_document_was_loaded_from));
|
|
} else if (system_->is_io_thread()) {
|
|
system_->backend_impl_->MarkAsForeignEntry(
|
|
host_id, document_url,
|
|
cache_document_was_loaded_from);
|
|
} else {
|
|
NOTREACHED();
|
|
}
|
|
}
|
|
|
|
virtual appcache::Status GetStatus(int host_id) {
|
|
if (system_->is_ui_thread()) {
|
|
status_result_ = appcache::UNCACHED;
|
|
event_.Reset();
|
|
system_->io_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserBackendProxy::GetStatus, host_id));
|
|
event_.Wait();
|
|
} else if (system_->is_io_thread()) {
|
|
system_->backend_impl_->GetStatusWithCallback(
|
|
host_id, get_status_callback_.get(), NULL);
|
|
} else {
|
|
NOTREACHED();
|
|
}
|
|
return status_result_;
|
|
}
|
|
|
|
virtual bool StartUpdate(int host_id) {
|
|
if (system_->is_ui_thread()) {
|
|
bool_result_ = false;
|
|
event_.Reset();
|
|
system_->io_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserBackendProxy::StartUpdate, host_id));
|
|
event_.Wait();
|
|
} else if (system_->is_io_thread()) {
|
|
system_->backend_impl_->StartUpdateWithCallback(
|
|
host_id, start_update_callback_.get(), NULL);
|
|
} else {
|
|
NOTREACHED();
|
|
}
|
|
return bool_result_;
|
|
}
|
|
|
|
virtual bool SwapCache(int host_id) {
|
|
if (system_->is_ui_thread()) {
|
|
bool_result_ = false;
|
|
event_.Reset();
|
|
system_->io_message_loop()->PostTask(FROM_HERE, NewRunnableMethod(
|
|
this, &BrowserBackendProxy::SwapCache, host_id));
|
|
event_.Wait();
|
|
} else if (system_->is_io_thread()) {
|
|
system_->backend_impl_->SwapCacheWithCallback(
|
|
host_id, swap_cache_callback_.get(), NULL);
|
|
} else {
|
|
NOTREACHED();
|
|
}
|
|
return bool_result_;
|
|
}
|
|
|
|
void GetStatusCallback(appcache::Status status, void* param) {
|
|
status_result_ = status;
|
|
event_.Signal();
|
|
}
|
|
|
|
void StartUpdateCallback(bool result, void* param) {
|
|
bool_result_ = result;
|
|
event_.Signal();
|
|
}
|
|
|
|
void SwapCacheCallback(bool result, void* param) {
|
|
bool_result_ = result;
|
|
event_.Signal();
|
|
}
|
|
|
|
void SignalEvent() {
|
|
event_.Signal();
|
|
}
|
|
|
|
private:
|
|
friend class base::RefCountedThreadSafe<BrowserBackendProxy>;
|
|
|
|
~BrowserBackendProxy() {}
|
|
|
|
BrowserAppCacheSystem* system_;
|
|
base::WaitableEvent event_;
|
|
bool bool_result_;
|
|
appcache::Status status_result_;
|
|
scoped_ptr<appcache::GetStatusCallback> get_status_callback_;
|
|
scoped_ptr<appcache::StartUpdateCallback> start_update_callback_;
|
|
scoped_ptr<appcache::SwapCacheCallback> swap_cache_callback_;
|
|
};
|
|
|
|
|
|
// BrowserAppCacheSystem --------------------------------------------------------
|
|
|
|
// This class only works for a single process browser.
|
|
static const int kSingleProcessId = 1;
|
|
|
|
// A not so thread safe singleton, but should work for test_shell.
|
|
BrowserAppCacheSystem* BrowserAppCacheSystem::instance_ = NULL;
|
|
|
|
BrowserAppCacheSystem::BrowserAppCacheSystem()
|
|
: io_message_loop_(NULL), ui_message_loop_(NULL),
|
|
ALLOW_THIS_IN_INITIALIZER_LIST(
|
|
backend_proxy_(new BrowserBackendProxy(this))),
|
|
ALLOW_THIS_IN_INITIALIZER_LIST(
|
|
frontend_proxy_(new BrowserFrontendProxy(this))),
|
|
backend_impl_(NULL), service_(NULL), db_thread_("AppCacheDBThread"),
|
|
thread_provider_(NULL) {
|
|
DCHECK(!instance_);
|
|
instance_ = this;
|
|
}
|
|
|
|
static void SignalEvent(base::WaitableEvent* event) {
|
|
event->Signal();
|
|
}
|
|
|
|
BrowserAppCacheSystem::~BrowserAppCacheSystem() {
|
|
DCHECK(!io_message_loop_ && !backend_impl_ && !service_);
|
|
frontend_proxy_->clear_appcache_system(); // in case a task is in transit
|
|
instance_ = NULL;
|
|
|
|
if (db_thread_.IsRunning()) {
|
|
// We pump a task thru the db thread to ensure any tasks previously
|
|
// scheduled on that thread have been performed prior to return.
|
|
base::WaitableEvent event(false, false);
|
|
db_thread_.message_loop()->PostTask(FROM_HERE,
|
|
NewRunnableFunction(&SignalEvent, &event));
|
|
event.Wait();
|
|
}
|
|
}
|
|
|
|
void BrowserAppCacheSystem::InitOnUIThread(
|
|
const FilePath& cache_directory) {
|
|
DCHECK(!ui_message_loop_);
|
|
AppCacheThread::Init(DB_THREAD_ID, IO_THREAD_ID);
|
|
ui_message_loop_ = MessageLoop::current();
|
|
cache_directory_ = cache_directory;
|
|
}
|
|
|
|
void BrowserAppCacheSystem::InitOnIOThread(URLRequestContext* request_context) {
|
|
if (!is_initailized_on_ui_thread())
|
|
return;
|
|
|
|
DCHECK(!io_message_loop_);
|
|
io_message_loop_ = MessageLoop::current();
|
|
io_message_loop_->AddDestructionObserver(this);
|
|
|
|
if (!db_thread_.IsRunning())
|
|
db_thread_.Start();
|
|
|
|
// Recreate and initialize per each IO thread.
|
|
service_ = new appcache::AppCacheService();
|
|
backend_impl_ = new appcache::AppCacheBackendImpl();
|
|
service_->Initialize(cache_directory_,
|
|
BrowserResourceLoaderBridge::GetCacheThread());
|
|
service_->set_request_context(request_context);
|
|
backend_impl_->Initialize(service_, frontend_proxy_.get(), kSingleProcessId);
|
|
|
|
AppCacheInterceptor::EnsureRegistered();
|
|
}
|
|
|
|
WebApplicationCacheHost* BrowserAppCacheSystem::CreateCacheHostForWebKit(
|
|
WebApplicationCacheHostClient* client) {
|
|
if (!is_initailized_on_ui_thread())
|
|
return NULL;
|
|
|
|
DCHECK(is_ui_thread());
|
|
|
|
// The IO thread needs to be running for this system to work.
|
|
BrowserResourceLoaderBridge::EnsureIOThread();
|
|
|
|
if (!is_initialized())
|
|
return NULL;
|
|
return new WebApplicationCacheHostImpl(client, backend_proxy_.get());
|
|
}
|
|
|
|
void BrowserAppCacheSystem::SetExtraRequestBits(
|
|
URLRequest* request, int host_id, ResourceType::Type resource_type) {
|
|
if (is_initialized()) {
|
|
DCHECK(is_io_thread());
|
|
AppCacheInterceptor::SetExtraRequestInfo(
|
|
request, service_, kSingleProcessId, host_id, resource_type);
|
|
}
|
|
}
|
|
|
|
void BrowserAppCacheSystem::GetExtraResponseBits(
|
|
URLRequest* request, int64* cache_id, GURL* manifest_url) {
|
|
if (is_initialized()) {
|
|
DCHECK(is_io_thread());
|
|
AppCacheInterceptor::GetExtraResponseInfo(
|
|
request, cache_id, manifest_url);
|
|
}
|
|
}
|
|
|
|
void BrowserAppCacheSystem::WillDestroyCurrentMessageLoop() {
|
|
DCHECK(is_io_thread());
|
|
|
|
delete backend_impl_;
|
|
delete service_;
|
|
backend_impl_ = NULL;
|
|
service_ = NULL;
|
|
io_message_loop_ = NULL;
|
|
|
|
// Just in case the main thread is waiting on it.
|
|
backend_proxy_->SignalEvent();
|
|
}
|