LocalFileSystemChromium.cpp [plain text]
#include "config.h"
#include "LocalFileSystem.h"
#if ENABLE(FILE_SYSTEM)
#include "AsyncFileSystem.h"
#include "CrossThreadTask.h"
#include "Document.h"
#include "ErrorCallback.h"
#include "FileSystemCallback.h"
#include "FileSystemCallbacks.h"
#include "PlatformString.h"
#include "WebFileError.h"
#include "platform/WebFileSystem.h"
#include "WebFileSystemCallbacksImpl.h"
#include "WebFrameClient.h"
#include "WebFrameImpl.h"
#include "WebPermissionClient.h"
#include "WebViewImpl.h"
#include "WebWorkerBase.h"
#include "WorkerContext.h"
#include "WorkerFileSystemCallbacksBridge.h"
#include "WorkerThread.h"
#include <wtf/Threading.h>
using namespace WebKit;
namespace WebCore {
LocalFileSystem& LocalFileSystem::localFileSystem()
{
AtomicallyInitializedStatic(LocalFileSystem*, localFileSystem = new LocalFileSystem(""));
return *localFileSystem;
}
namespace {
enum CreationFlag {
OpenExisting,
CreateIfNotPresent
};
#if ENABLE(WORKERS)
static const char allowFileSystemMode[] = "allowFileSystemMode";
static const char openFileSystemMode[] = "openFileSystemMode";
class AllowFileSystemMainThreadBridge : public ThreadSafeRefCounted<AllowFileSystemMainThreadBridge> {
public:
static PassRefPtr<AllowFileSystemMainThreadBridge> create(WebCore::WorkerLoaderProxy* workerLoaderProxy, const String& mode, WebCommonWorkerClient* commonClient)
{
return adoptRef(new AllowFileSystemMainThreadBridge(workerLoaderProxy, mode, commonClient));
}
void cancel()
{
MutexLocker locker(m_mutex);
m_workerLoaderProxy = 0;
}
bool result()
{
return m_result;
}
void signalCompleted(const String& mode, bool result)
{
MutexLocker locker(m_mutex);
if (!m_workerLoaderProxy)
return;
m_workerLoaderProxy->postTaskForModeToWorkerContext(
createCallbackTask(&didComplete, AllowCrossThreadAccess(this), result), mode);
}
private:
AllowFileSystemMainThreadBridge(WebCore::WorkerLoaderProxy* workerLoaderProxy, const String& mode, WebCommonWorkerClient* commonClient)
: m_result(false)
, m_workerLoaderProxy(workerLoaderProxy)
{
WebWorkerBase::dispatchTaskToMainThread(createCallbackTask(&allowFileSystemTask, mode, AllowCrossThreadAccess(commonClient), this));
}
static void allowFileSystemTask(WebCore::ScriptExecutionContext* context, const String& mode, WebCommonWorkerClient* commonClient, PassRefPtr<AllowFileSystemMainThreadBridge> bridge)
{
bool allowFileSystem = commonClient ? commonClient->allowFileSystem() : false;
bridge->signalCompleted(mode, allowFileSystem);
}
static void didComplete(WebCore::ScriptExecutionContext* context, PassRefPtr<AllowFileSystemMainThreadBridge> bridge, bool result)
{
bridge->m_result = result;
}
bool m_result;
Mutex m_mutex;
WebCore::WorkerLoaderProxy* m_workerLoaderProxy;
};
bool allowFileSystemForWorker(WebCommonWorkerClient* commonClient)
{
WorkerScriptController* controller = WorkerScriptController::controllerForContext();
WorkerContext* workerContext = controller->workerContext();
WebCore::WorkerThread* workerThread = workerContext->thread();
WorkerRunLoop& runLoop = workerThread->runLoop();
WebCore::WorkerLoaderProxy* workerLoaderProxy = &workerThread->workerLoaderProxy();
String mode = allowFileSystemMode;
mode.append(String::number(runLoop.createUniqueId()));
RefPtr<AllowFileSystemMainThreadBridge> bridge = AllowFileSystemMainThreadBridge::create(workerLoaderProxy, mode, commonClient);
if (runLoop.runInMode(workerContext, mode) == MessageQueueTerminated) {
bridge->cancel();
return false;
}
return bridge->result();
}
void openFileSystemForWorker(WebCommonWorkerClient* commonClient, WebFileSystem::Type type, long long size, bool create, WebFileSystemCallbacks* callbacks, bool synchronous)
{
WorkerScriptController* controller = WorkerScriptController::controllerForContext();
WorkerContext* workerContext = controller->workerContext();
WebCore::WorkerThread* workerThread = workerContext->thread();
WorkerRunLoop& runLoop = workerThread->runLoop();
WebCore::WorkerLoaderProxy* workerLoaderProxy = &workerThread->workerLoaderProxy();
String mode = openFileSystemMode;
mode.append(String::number(runLoop.createUniqueId()));
RefPtr<WorkerFileSystemCallbacksBridge> bridge = WorkerFileSystemCallbacksBridge::create(workerLoaderProxy, workerContext, callbacks);
bridge->postOpenFileSystemToMainThread(commonClient, type, size, create, mode);
if (synchronous) {
if (runLoop.runInMode(workerContext, mode) == MessageQueueTerminated)
bridge->stop();
}
}
#endif // ENABLE(WORKERS)
}
static void openFileSystemNotAllowed(ScriptExecutionContext*, PassOwnPtr<AsyncFileSystemCallbacks> callbacks)
{
callbacks->didFail(WebKit::WebFileErrorAbort);
}
static void openFileSystemHelper(ScriptExecutionContext* context, AsyncFileSystem::Type type, PassOwnPtr<AsyncFileSystemCallbacks> callbacks, bool synchronous, long long size, CreationFlag create)
{
bool allowed = true;
ASSERT(context);
if (context->isDocument()) {
Document* document = static_cast<Document*>(context);
WebFrameImpl* webFrame = WebFrameImpl::fromFrame(document->frame());
WebKit::WebViewImpl* webView = webFrame->viewImpl();
if (webView->permissionClient() && !webView->permissionClient()->allowFileSystem(webFrame))
allowed = false;
else
webFrame->client()->openFileSystem(webFrame, static_cast<WebFileSystem::Type>(type), size, create == CreateIfNotPresent, new WebFileSystemCallbacksImpl(callbacks, type));
} else {
#if ENABLE(WORKERS)
WorkerContext* workerContext = static_cast<WorkerContext*>(context);
WorkerLoaderProxy* workerLoaderProxy = &workerContext->thread()->workerLoaderProxy();
WebWorkerBase* webWorker = static_cast<WebWorkerBase*>(workerLoaderProxy);
if (!allowFileSystemForWorker(webWorker->commonClient()))
allowed = false;
else
openFileSystemForWorker(webWorker->commonClient(), static_cast<WebFileSystem::Type>(type), size, create == CreateIfNotPresent, new WebFileSystemCallbacksImpl(callbacks, type, context, synchronous), synchronous);
#else
ASSERT_NOT_REACHED();
#endif
}
if (!allowed) {
context->postTask(createCallbackTask(&openFileSystemNotAllowed, callbacks));
}
}
void LocalFileSystem::readFileSystem(ScriptExecutionContext* context, AsyncFileSystem::Type type, PassOwnPtr<AsyncFileSystemCallbacks> callbacks, bool synchronous)
{
openFileSystemHelper(context, type, callbacks, synchronous, 0, OpenExisting);
}
void LocalFileSystem::requestFileSystem(ScriptExecutionContext* context, AsyncFileSystem::Type type, long long size, PassOwnPtr<AsyncFileSystemCallbacks> callbacks, bool synchronous)
{
openFileSystemHelper(context, type, callbacks, synchronous, size, CreateIfNotPresent);
}
}
#endif // ENABLE(FILE_SYSTEM)