michael@0: /* -*- Mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; tab-width: 40 -*- */
michael@0: /* vim: set ts=2 et sw=2 tw=80: */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
michael@0:  * You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #include "CreateFileTask.h"
michael@0: 
michael@0: #include <algorithm>
michael@0: 
michael@0: #include "DOMError.h"
michael@0: #include "mozilla/Preferences.h"
michael@0: #include "mozilla/dom/FileSystemBase.h"
michael@0: #include "mozilla/dom/FileSystemUtils.h"
michael@0: #include "mozilla/dom/Promise.h"
michael@0: #include "nsDOMFile.h"
michael@0: #include "nsIFile.h"
michael@0: #include "nsNetUtil.h"
michael@0: #include "nsStringGlue.h"
michael@0: 
michael@0: namespace mozilla {
michael@0: namespace dom {
michael@0: 
michael@0: uint32_t CreateFileTask::sOutputBufferSize = 0;
michael@0: 
michael@0: CreateFileTask::CreateFileTask(FileSystemBase* aFileSystem,
michael@0:                                const nsAString& aPath,
michael@0:                                nsIDOMBlob* aBlobData,
michael@0:                                InfallibleTArray<uint8_t>& aArrayData,
michael@0:                                bool replace)
michael@0:   : FileSystemTaskBase(aFileSystem)
michael@0:   , mTargetRealPath(aPath)
michael@0:   , mBlobData(aBlobData)
michael@0:   , mReplace(replace)
michael@0: {
michael@0:   MOZ_ASSERT(NS_IsMainThread(), "Only call on main thread!");
michael@0:   MOZ_ASSERT(aFileSystem);
michael@0:   GetOutputBufferSize();
michael@0:   if (mBlobData) {
michael@0:     nsresult rv = mBlobData->GetInternalStream(getter_AddRefs(mBlobStream));
michael@0:     NS_WARN_IF(NS_FAILED(rv));
michael@0:   }
michael@0:   mArrayData.SwapElements(aArrayData);
michael@0:   nsCOMPtr<nsIGlobalObject> globalObject =
michael@0:     do_QueryInterface(aFileSystem->GetWindow());
michael@0:   if (!globalObject) {
michael@0:     return;
michael@0:   }
michael@0:   mPromise = new Promise(globalObject);
michael@0: }
michael@0: 
michael@0: CreateFileTask::CreateFileTask(FileSystemBase* aFileSystem,
michael@0:                        const FileSystemCreateFileParams& aParam,
michael@0:                        FileSystemRequestParent* aParent)
michael@0:   : FileSystemTaskBase(aFileSystem, aParam, aParent)
michael@0:   , mReplace(false)
michael@0: {
michael@0:   MOZ_ASSERT(FileSystemUtils::IsParentProcess(),
michael@0:              "Only call from parent process!");
michael@0:   MOZ_ASSERT(NS_IsMainThread(), "Only call on main thread!");
michael@0:   MOZ_ASSERT(aFileSystem);
michael@0:   GetOutputBufferSize();
michael@0: 
michael@0:   mTargetRealPath = aParam.realPath();
michael@0: 
michael@0:   mReplace = aParam.replace();
michael@0: 
michael@0:   auto& data = aParam.data();
michael@0: 
michael@0:   if (data.type() == FileSystemFileDataValue::TArrayOfuint8_t) {
michael@0:     mArrayData = data;
michael@0:     return;
michael@0:   }
michael@0: 
michael@0:   BlobParent* bp = static_cast<BlobParent*>(static_cast<PBlobParent*>(data));
michael@0:   mBlobData = bp->GetBlob();
michael@0:   MOZ_ASSERT(mBlobData, "mBlobData should not be null.");
michael@0:   nsresult rv = mBlobData->GetInternalStream(getter_AddRefs(mBlobStream));
michael@0:   NS_WARN_IF(NS_FAILED(rv));
michael@0: }
michael@0: 
michael@0: CreateFileTask::~CreateFileTask()
michael@0: {
michael@0:   MOZ_ASSERT(!mPromise || NS_IsMainThread(),
michael@0:              "mPromise should be released on main thread!");
michael@0:   if (mBlobStream) {
michael@0:     mBlobStream->Close();
michael@0:   }
michael@0: }
michael@0: 
michael@0: already_AddRefed<Promise>
michael@0: CreateFileTask::GetPromise()
michael@0: {
michael@0:   MOZ_ASSERT(NS_IsMainThread(), "Only call on main thread!");
michael@0:   return nsRefPtr<Promise>(mPromise).forget();
michael@0: }
michael@0: 
michael@0: FileSystemParams
michael@0: CreateFileTask::GetRequestParams(const nsString& aFileSystem) const
michael@0: {
michael@0:   MOZ_ASSERT(NS_IsMainThread(), "Only call on main thread!");
michael@0:   FileSystemCreateFileParams param;
michael@0:   param.filesystem() = aFileSystem;
michael@0:   param.realPath() = mTargetRealPath;
michael@0:   param.replace() = mReplace;
michael@0:   if (mBlobData) {
michael@0:     BlobChild* actor
michael@0:       = ContentChild::GetSingleton()->GetOrCreateActorForBlob(mBlobData);
michael@0:     if (actor) {
michael@0:       param.data() = actor;
michael@0:     }
michael@0:   } else {
michael@0:     param.data() = mArrayData;
michael@0:   }
michael@0:   return param;
michael@0: }
michael@0: 
michael@0: FileSystemResponseValue
michael@0: CreateFileTask::GetSuccessRequestResult() const
michael@0: {
michael@0:   MOZ_ASSERT(NS_IsMainThread(), "Only call on main thread!");
michael@0:   BlobParent* actor = GetBlobParent(mTargetFile);
michael@0:   if (!actor) {
michael@0:     return FileSystemErrorResponse(NS_ERROR_DOM_FILESYSTEM_UNKNOWN_ERR);
michael@0:   }
michael@0:   FileSystemFileResponse response;
michael@0:   response.blobParent() = actor;
michael@0:   return response;
michael@0: }
michael@0: 
michael@0: void
michael@0: CreateFileTask::SetSuccessRequestResult(const FileSystemResponseValue& aValue)
michael@0: {
michael@0:   MOZ_ASSERT(NS_IsMainThread(), "Only call on main thread!");
michael@0:   FileSystemFileResponse r = aValue;
michael@0:   BlobChild* actor = static_cast<BlobChild*>(r.blobChild());
michael@0:   nsCOMPtr<nsIDOMBlob> blob = actor->GetBlob();
michael@0:   mTargetFile = do_QueryInterface(blob);
michael@0: }
michael@0: 
michael@0: nsresult
michael@0: CreateFileTask::Work()
michael@0: {
michael@0:   class AutoClose
michael@0:   {
michael@0:   public:
michael@0:     AutoClose(nsIOutputStream* aStream)
michael@0:       : mStream(aStream)
michael@0:     {
michael@0:       MOZ_ASSERT(aStream);
michael@0:     }
michael@0: 
michael@0:     ~AutoClose()
michael@0:     {
michael@0:       mStream->Close();
michael@0:     }
michael@0:   private:
michael@0:     nsCOMPtr<nsIOutputStream> mStream;
michael@0:   };
michael@0: 
michael@0:   MOZ_ASSERT(FileSystemUtils::IsParentProcess(),
michael@0:              "Only call from parent process!");
michael@0:   MOZ_ASSERT(!NS_IsMainThread(), "Only call on worker thread!");
michael@0: 
michael@0:   if (mFileSystem->IsShutdown()) {
michael@0:     return NS_ERROR_FAILURE;
michael@0:   }
michael@0: 
michael@0:   nsCOMPtr<nsIFile> file = mFileSystem->GetLocalFile(mTargetRealPath);
michael@0:   if (!file) {
michael@0:     return NS_ERROR_DOM_FILESYSTEM_INVALID_PATH_ERR;
michael@0:   }
michael@0: 
michael@0:   if (!mFileSystem->IsSafeFile(file)) {
michael@0:     return NS_ERROR_DOM_SECURITY_ERR;
michael@0:   }
michael@0: 
michael@0:   bool exists = false;
michael@0:   nsresult rv = file->Exists(&exists);
michael@0:   if (NS_WARN_IF(NS_FAILED(rv))) {
michael@0:     return rv;
michael@0:   }
michael@0: 
michael@0:   if (exists) {
michael@0:     bool isFile = false;
michael@0:     rv = file->IsFile(&isFile);
michael@0:     if (NS_WARN_IF(NS_FAILED(rv))) {
michael@0:       return rv;
michael@0:     }
michael@0: 
michael@0:     if (!isFile) {
michael@0:       return NS_ERROR_DOM_FILESYSTEM_TYPE_MISMATCH_ERR;
michael@0:     }
michael@0: 
michael@0:     if (!mReplace) {
michael@0:       return NS_ERROR_DOM_FILESYSTEM_PATH_EXISTS_ERR;
michael@0:     }
michael@0: 
michael@0:     // Remove the old file before creating.
michael@0:     rv = file->Remove(false);
michael@0:     if (NS_WARN_IF(NS_FAILED(rv))) {
michael@0:       return rv;
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   rv = file->Create(nsIFile::NORMAL_FILE_TYPE, 0600);
michael@0:   if (NS_WARN_IF(NS_FAILED(rv))) {
michael@0:     return rv;
michael@0:   }
michael@0: 
michael@0:   nsCOMPtr<nsIOutputStream> outputStream;
michael@0:   rv = NS_NewLocalFileOutputStream(getter_AddRefs(outputStream), file);
michael@0:   if (NS_WARN_IF(NS_FAILED(rv))) {
michael@0:     return rv;
michael@0:   }
michael@0: 
michael@0:   AutoClose acOutputStream(outputStream);
michael@0: 
michael@0:   nsCOMPtr<nsIOutputStream> bufferedOutputStream;
michael@0:   rv = NS_NewBufferedOutputStream(getter_AddRefs(bufferedOutputStream),
michael@0:                                   outputStream,
michael@0:                                   sOutputBufferSize);
michael@0:   if (NS_WARN_IF(NS_FAILED(rv))) {
michael@0:     return rv;
michael@0:   }
michael@0: 
michael@0:   AutoClose acBufferedOutputStream(bufferedOutputStream);
michael@0: 
michael@0:   if (mBlobStream) {
michael@0:     // Write the file content from blob data.
michael@0: 
michael@0:     uint64_t bufSize = 0;
michael@0:     rv = mBlobStream->Available(&bufSize);
michael@0:     if (NS_WARN_IF(NS_FAILED(rv))) {
michael@0:       return rv;
michael@0:     }
michael@0: 
michael@0:     while (bufSize && !mFileSystem->IsShutdown()) {
michael@0:       uint32_t written = 0;
michael@0:       uint32_t writeSize = bufSize < UINT32_MAX ? bufSize : UINT32_MAX;
michael@0:       rv = bufferedOutputStream->WriteFrom(mBlobStream, writeSize, &written);
michael@0:       if (NS_WARN_IF(NS_FAILED(rv))) {
michael@0:         return rv;
michael@0:       }
michael@0:       bufSize -= written;
michael@0:     }
michael@0: 
michael@0:     mBlobStream->Close();
michael@0:     mBlobStream = nullptr;
michael@0: 
michael@0:     if (mFileSystem->IsShutdown()) {
michael@0:       return NS_ERROR_FAILURE;
michael@0:     }
michael@0: 
michael@0:     mTargetFile = new nsDOMFileFile(file);
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0:   // Write file content from array data.
michael@0: 
michael@0:   uint32_t written;
michael@0:   rv = bufferedOutputStream->Write(
michael@0:     reinterpret_cast<char*>(mArrayData.Elements()),
michael@0:     mArrayData.Length(),
michael@0:     &written);
michael@0:   if (NS_WARN_IF(NS_FAILED(rv))) {
michael@0:     return rv;
michael@0:   }
michael@0: 
michael@0:   if (mArrayData.Length() != written) {
michael@0:     return NS_ERROR_DOM_FILESYSTEM_UNKNOWN_ERR;
michael@0:   }
michael@0: 
michael@0:   mTargetFile = new nsDOMFileFile(file);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: void
michael@0: CreateFileTask::HandlerCallback()
michael@0: {
michael@0:   MOZ_ASSERT(NS_IsMainThread(), "Only call on main thread!");
michael@0:   if (mFileSystem->IsShutdown()) {
michael@0:     mPromise = nullptr;
michael@0:     return;
michael@0:   }
michael@0: 
michael@0:   if (HasError()) {
michael@0:     nsRefPtr<DOMError> domError = new DOMError(mFileSystem->GetWindow(),
michael@0:       mErrorValue);
michael@0:     mPromise->MaybeReject(domError);
michael@0:     mPromise = nullptr;
michael@0:     return;
michael@0:   }
michael@0: 
michael@0:   mPromise->MaybeResolve(mTargetFile);
michael@0:   mPromise = nullptr;
michael@0: }
michael@0: 
michael@0: void
michael@0: CreateFileTask::GetPermissionAccessType(nsCString& aAccess) const
michael@0: {
michael@0:   if (mReplace) {
michael@0:     aAccess.AssignLiteral("write");
michael@0:     return;
michael@0:   }
michael@0: 
michael@0:   aAccess.AssignLiteral("create");
michael@0: }
michael@0: 
michael@0: void
michael@0: CreateFileTask::GetOutputBufferSize() const
michael@0: {
michael@0:   if (sOutputBufferSize || !FileSystemUtils::IsParentProcess()) {
michael@0:     return;
michael@0:   }
michael@0:   sOutputBufferSize =
michael@0:     mozilla::Preferences::GetUint("dom.filesystem.outputBufferSize", 4096 * 4);
michael@0: }
michael@0: 
michael@0: } // namespace dom
michael@0: } // namespace mozilla