1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/dom/file/FileHelper.cpp Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,231 @@
1.4 +/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
1.5 +/* vim: set ts=2 et sw=2 tw=80: */
1.6 +/* This Source Code Form is subject to the terms of the Mozilla Public
1.7 + * License, v. 2.0. If a copy of the MPL was not distributed with this file,
1.8 + * You can obtain one at http://mozilla.org/MPL/2.0/. */
1.9 +
1.10 +#include "FileHelper.h"
1.11 +
1.12 +#include "nsIFileStorage.h"
1.13 +
1.14 +#include "nsError.h"
1.15 +#include "nsProxyRelease.h"
1.16 +
1.17 +#include "FileHandle.h"
1.18 +#include "FileRequest.h"
1.19 +#include "FileService.h"
1.20 +#include "nsIRequest.h"
1.21 +#include "nsThreadUtils.h"
1.22 +
1.23 +USING_FILE_NAMESPACE
1.24 +
1.25 +namespace {
1.26 +
1.27 +LockedFile* gCurrentLockedFile = nullptr;
1.28 +
1.29 +} // anonymous namespace
1.30 +
1.31 +FileHelper::FileHelper(LockedFile* aLockedFile,
1.32 + FileRequest* aFileRequest)
1.33 +: mFileStorage(aLockedFile->mFileHandle->mFileStorage),
1.34 + mLockedFile(aLockedFile),
1.35 + mFileRequest(aFileRequest),
1.36 + mResultCode(NS_OK),
1.37 + mFinished(false)
1.38 +{
1.39 + NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
1.40 +}
1.41 +
1.42 +FileHelper::~FileHelper()
1.43 +{
1.44 + NS_ASSERTION(!mFileStorage && !mLockedFile && !mFileRequest && !mListener &&
1.45 + !mRequest, "Should have cleared this!");
1.46 +}
1.47 +
1.48 +NS_IMPL_ISUPPORTS(FileHelper, nsIRequestObserver)
1.49 +
1.50 +nsresult
1.51 +FileHelper::Enqueue()
1.52 +{
1.53 + NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
1.54 +
1.55 + FileService* service = FileService::GetOrCreate();
1.56 + NS_ENSURE_TRUE(service, NS_ERROR_FAILURE);
1.57 +
1.58 + nsresult rv = service->Enqueue(mLockedFile, this);
1.59 + NS_ENSURE_SUCCESS(rv, rv);
1.60 +
1.61 + if (mLockedFile) {
1.62 + mLockedFile->OnNewRequest();
1.63 + }
1.64 +
1.65 + return NS_OK;
1.66 +}
1.67 +
1.68 +nsresult
1.69 +FileHelper::AsyncRun(FileHelperListener* aListener)
1.70 +{
1.71 + NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
1.72 +
1.73 + // Assign the listener early, so we can use it synchronously if the code
1.74 + // below fails.
1.75 + mListener = aListener;
1.76 +
1.77 + nsresult rv;
1.78 +
1.79 + nsCOMPtr<nsISupports> stream;
1.80 + if (mLockedFile->mRequestMode == LockedFile::PARALLEL) {
1.81 + rv = mLockedFile->CreateParallelStream(getter_AddRefs(stream));
1.82 + }
1.83 + else {
1.84 + rv = mLockedFile->GetOrCreateStream(getter_AddRefs(stream));
1.85 + }
1.86 +
1.87 + if (NS_SUCCEEDED(rv)) {
1.88 + NS_ASSERTION(stream, "This should never be null!");
1.89 +
1.90 + rv = DoAsyncRun(stream);
1.91 + }
1.92 +
1.93 + if (NS_FAILED(rv)) {
1.94 + mResultCode = NS_ERROR_DOM_FILEHANDLE_UNKNOWN_ERR;
1.95 + Finish();
1.96 + }
1.97 +
1.98 + return NS_OK;
1.99 +}
1.100 +
1.101 +NS_IMETHODIMP
1.102 +FileHelper::OnStartRequest(nsIRequest* aRequest, nsISupports* aCtxt)
1.103 +{
1.104 + NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
1.105 +
1.106 + return NS_OK;
1.107 +}
1.108 +
1.109 +NS_IMETHODIMP
1.110 +FileHelper::OnStopRequest(nsIRequest* aRequest, nsISupports* aCtxt,
1.111 + nsresult aStatus)
1.112 +{
1.113 + NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
1.114 +
1.115 + if (NS_FAILED(aStatus)) {
1.116 + if (aStatus == NS_ERROR_FILE_NO_DEVICE_SPACE) {
1.117 + mResultCode = NS_ERROR_DOM_FILEHANDLE_QUOTA_ERR;
1.118 + }
1.119 + else {
1.120 + mResultCode = NS_ERROR_DOM_FILEHANDLE_UNKNOWN_ERR;
1.121 + }
1.122 + }
1.123 +
1.124 + Finish();
1.125 +
1.126 + return NS_OK;
1.127 +}
1.128 +
1.129 +void
1.130 +FileHelper::OnStreamProgress(uint64_t aProgress, uint64_t aProgressMax)
1.131 +{
1.132 + NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
1.133 +
1.134 + if (mLockedFile->IsAborted()) {
1.135 + NS_ASSERTION(mRequest, "Should have a request!\n");
1.136 +
1.137 + nsresult rv = mRequest->Cancel(NS_BINDING_ABORTED);
1.138 + if (NS_FAILED(rv)) {
1.139 + NS_WARNING("Failed to cancel the request!");
1.140 + }
1.141 +
1.142 + return;
1.143 + }
1.144 +
1.145 + if (mFileRequest) {
1.146 + mFileRequest->OnProgress(aProgress, aProgressMax);
1.147 + }
1.148 +}
1.149 +
1.150 +// static
1.151 +LockedFile*
1.152 +FileHelper::GetCurrentLockedFile()
1.153 +{
1.154 + NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
1.155 +
1.156 + return gCurrentLockedFile;
1.157 +}
1.158 +
1.159 +nsresult
1.160 +FileHelper::GetSuccessResult(JSContext* aCx,
1.161 + JS::MutableHandle<JS::Value> aVal)
1.162 +{
1.163 + NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
1.164 +
1.165 + aVal.setUndefined();
1.166 + return NS_OK;
1.167 +}
1.168 +
1.169 +void
1.170 +FileHelper::ReleaseObjects()
1.171 +{
1.172 + NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
1.173 +
1.174 + mFileStorage = nullptr;
1.175 + mLockedFile = nullptr;
1.176 + mFileRequest = nullptr;
1.177 + mListener = nullptr;
1.178 + mRequest = nullptr;
1.179 +}
1.180 +
1.181 +void
1.182 +FileHelper::Finish()
1.183 +{
1.184 + NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
1.185 +
1.186 + if (mFinished) {
1.187 + return;
1.188 + }
1.189 +
1.190 + mFinished = true;
1.191 +
1.192 + if (mLockedFile->IsAborted()) {
1.193 + // Always fire a "error" event with ABORT_ERR if the transaction was
1.194 + // aborted, even if the request succeeded or failed with another error.
1.195 + mResultCode = NS_ERROR_DOM_FILEHANDLE_ABORT_ERR;
1.196 + }
1.197 +
1.198 + LockedFile* oldLockedFile = gCurrentLockedFile;
1.199 + gCurrentLockedFile = mLockedFile;
1.200 +
1.201 + if (mFileRequest) {
1.202 + nsresult rv = mFileRequest->NotifyHelperCompleted(this);
1.203 + if (NS_SUCCEEDED(mResultCode) && NS_FAILED(rv)) {
1.204 + mResultCode = rv;
1.205 + }
1.206 + }
1.207 +
1.208 + NS_ASSERTION(gCurrentLockedFile == mLockedFile, "Should be unchanged!");
1.209 + gCurrentLockedFile = oldLockedFile;
1.210 +
1.211 + mLockedFile->OnRequestFinished();
1.212 +
1.213 + mListener->OnFileHelperComplete(this);
1.214 +
1.215 + ReleaseObjects();
1.216 +
1.217 + NS_ASSERTION(!(mFileStorage || mLockedFile || mFileRequest || mListener ||
1.218 + mRequest), "Subclass didn't call FileHelper::ReleaseObjects!");
1.219 +
1.220 +}
1.221 +
1.222 +void
1.223 +FileHelper::OnStreamClose()
1.224 +{
1.225 + NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
1.226 + Finish();
1.227 +}
1.228 +
1.229 +void
1.230 +FileHelper::OnStreamDestroy()
1.231 +{
1.232 + NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
1.233 + Finish();
1.234 +}
