/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ /* vim: set ts=8 sts=2 et sw=2 tw=80: */ /* This Source Code Form is subject to the terms of the Mozilla Public * License, v. 2.0. If a copy of the MPL was not distributed with this file, * You can obtain one at http://mozilla.org/MPL/2.0/. */ #include "GetDirectoryListingTask.h" #include "HTMLSplitOnSpacesTokenizer.h" #include "js/Value.h" #include "mozilla/dom/File.h" #include "mozilla/dom/FileSystemBase.h" #include "mozilla/dom/FileSystemUtils.h" #include "mozilla/dom/PFileSystemParams.h" #include "mozilla/dom/Promise.h" #include "mozilla/dom/UnionTypes.h" #include "mozilla/dom/ipc/BlobChild.h" #include "mozilla/dom/ipc/BlobParent.h" #include "nsIFile.h" #include "nsISimpleEnumerator.h" #include "nsStringGlue.h" namespace mozilla { namespace dom { /** * GetDirectoryListingTaskChild */ /* static */ already_AddRefed<GetDirectoryListingTaskChild> GetDirectoryListingTaskChild::Create(FileSystemBase* aFileSystem, Directory* aDirectory, nsIFile* aTargetPath, const nsAString& aFilters, ErrorResult& aRv) { MOZ_ASSERT(aFileSystem); MOZ_ASSERT(aDirectory); aFileSystem->AssertIsOnOwningThread(); RefPtr<GetDirectoryListingTaskChild> task = new GetDirectoryListingTaskChild(aFileSystem, aDirectory, aTargetPath, aFilters); // aTargetPath can be null. In this case SetError will be called. nsCOMPtr<nsIGlobalObject> globalObject = do_QueryInterface(aFileSystem->GetParentObject()); if (NS_WARN_IF(!globalObject)) { aRv.Throw(NS_ERROR_FAILURE); return nullptr; } task->mPromise = Promise::Create(globalObject, aRv); if (NS_WARN_IF(aRv.Failed())) { return nullptr; } return task.forget(); } GetDirectoryListingTaskChild::GetDirectoryListingTaskChild(FileSystemBase* aFileSystem, Directory* aDirectory, nsIFile* aTargetPath, const nsAString& aFilters) : FileSystemTaskChildBase(aFileSystem) , mDirectory(aDirectory) , mTargetPath(aTargetPath) , mFilters(aFilters) { MOZ_ASSERT(aFileSystem); aFileSystem->AssertIsOnOwningThread(); } GetDirectoryListingTaskChild::~GetDirectoryListingTaskChild() { mFileSystem->AssertIsOnOwningThread(); } already_AddRefed<Promise> GetDirectoryListingTaskChild::GetPromise() { mFileSystem->AssertIsOnOwningThread(); return RefPtr<Promise>(mPromise).forget(); } FileSystemParams GetDirectoryListingTaskChild::GetRequestParams(const nsString& aSerializedDOMPath, ErrorResult& aRv) const { mFileSystem->AssertIsOnOwningThread(); // this is the real path. nsAutoString path; aRv = mTargetPath->GetPath(path); if (NS_WARN_IF(aRv.Failed())) { return FileSystemGetDirectoryListingParams(); } // this is the dom path. nsAutoString directoryPath; mDirectory->GetPath(directoryPath, aRv); if (NS_WARN_IF(aRv.Failed())) { return FileSystemGetDirectoryListingParams(); } return FileSystemGetDirectoryListingParams(aSerializedDOMPath, path, directoryPath, mFilters); } void GetDirectoryListingTaskChild::SetSuccessRequestResult(const FileSystemResponseValue& aValue, ErrorResult& aRv) { mFileSystem->AssertIsOnOwningThread(); MOZ_ASSERT(aValue.type() == FileSystemResponseValue::TFileSystemDirectoryListingResponse); FileSystemDirectoryListingResponse r = aValue; for (uint32_t i = 0; i < r.data().Length(); ++i) { const FileSystemDirectoryListingResponseData& data = r.data()[i]; OwningFileOrDirectory* ofd = mTargetData.AppendElement(fallible); if (!ofd) { aRv.Throw(NS_ERROR_OUT_OF_MEMORY); return; } if (data.type() == FileSystemDirectoryListingResponseData::TFileSystemDirectoryListingResponseFile) { const FileSystemDirectoryListingResponseFile& d = data.get_FileSystemDirectoryListingResponseFile(); RefPtr<BlobImpl> blobImpl = static_cast<BlobChild*>(d.blobChild())->GetBlobImpl(); MOZ_ASSERT(blobImpl); RefPtr<File> file = File::Create(mFileSystem->GetParentObject(), blobImpl); MOZ_ASSERT(file); ofd->SetAsFile() = file; } else { MOZ_ASSERT(data.type() == FileSystemDirectoryListingResponseData::TFileSystemDirectoryListingResponseDirectory); const FileSystemDirectoryListingResponseDirectory& d = data.get_FileSystemDirectoryListingResponseDirectory(); nsCOMPtr<nsIFile> path; aRv = NS_NewLocalFile(d.directoryRealPath(), true, getter_AddRefs(path)); if (NS_WARN_IF(aRv.Failed())) { return; } RefPtr<Directory> directory = Directory::Create(mFileSystem->GetParentObject(), path, mFileSystem); MOZ_ASSERT(directory); ofd->SetAsDirectory() = directory; } } } void GetDirectoryListingTaskChild::HandlerCallback() { mFileSystem->AssertIsOnOwningThread(); if (mFileSystem->IsShutdown()) { mPromise = nullptr; return; } if (HasError()) { mPromise->MaybeReject(mErrorValue); mPromise = nullptr; return; } mPromise->MaybeResolve(mTargetData); mPromise = nullptr; } /** * GetDirectoryListingTaskParent */ /* static */ already_AddRefed<GetDirectoryListingTaskParent> GetDirectoryListingTaskParent::Create(FileSystemBase* aFileSystem, const FileSystemGetDirectoryListingParams& aParam, FileSystemRequestParent* aParent, ErrorResult& aRv) { MOZ_ASSERT(XRE_IsParentProcess(), "Only call from parent process!"); AssertIsOnBackgroundThread(); MOZ_ASSERT(aFileSystem); RefPtr<GetDirectoryListingTaskParent> task = new GetDirectoryListingTaskParent(aFileSystem, aParam, aParent); aRv = NS_NewLocalFile(aParam.realPath(), true, getter_AddRefs(task->mTargetPath)); if (NS_WARN_IF(aRv.Failed())) { return nullptr; } return task.forget(); } GetDirectoryListingTaskParent::GetDirectoryListingTaskParent(FileSystemBase* aFileSystem, const FileSystemGetDirectoryListingParams& aParam, FileSystemRequestParent* aParent) : FileSystemTaskParentBase(aFileSystem, aParam, aParent) , mDOMPath(aParam.domPath()) , mFilters(aParam.filters()) { MOZ_ASSERT(XRE_IsParentProcess(), "Only call from parent process!"); AssertIsOnBackgroundThread(); MOZ_ASSERT(aFileSystem); } FileSystemResponseValue GetDirectoryListingTaskParent::GetSuccessRequestResult(ErrorResult& aRv) const { AssertIsOnBackgroundThread(); InfallibleTArray<PBlobParent*> blobs; nsTArray<FileSystemDirectoryListingResponseData> inputs; for (unsigned i = 0; i < mTargetData.Length(); i++) { if (mTargetData[i].mType == FileOrDirectoryPath::eFilePath) { nsCOMPtr<nsIFile> path; nsresult rv = NS_NewLocalFile(mTargetData[i].mPath, true, getter_AddRefs(path)); if (NS_WARN_IF(NS_FAILED(rv))) { return FileSystemErrorResponse(rv); } FileSystemDirectoryListingResponseFile fileData; RefPtr<BlobImpl> blobImpl = new BlobImplFile(path); nsAutoString filePath; filePath.Assign(mDOMPath); // This is specific for unix root filesystem. if (!mDOMPath.EqualsLiteral(FILESYSTEM_DOM_PATH_SEPARATOR_LITERAL)) { filePath.AppendLiteral(FILESYSTEM_DOM_PATH_SEPARATOR_LITERAL); } nsAutoString name; blobImpl->GetName(name); filePath.Append(name); blobImpl->SetDOMPath(filePath); fileData.blobParent() = BlobParent::GetOrCreate(mRequestParent->Manager(), blobImpl); inputs.AppendElement(fileData); } else { MOZ_ASSERT(mTargetData[i].mType == FileOrDirectoryPath::eDirectoryPath); FileSystemDirectoryListingResponseDirectory directoryData; directoryData.directoryRealPath() = mTargetData[i].mPath; inputs.AppendElement(directoryData); } } FileSystemDirectoryListingResponse response; response.data().SwapElements(inputs); return response; } nsresult GetDirectoryListingTaskParent::IOWork() { MOZ_ASSERT(XRE_IsParentProcess(), "Only call from parent process!"); MOZ_ASSERT(!NS_IsMainThread(), "Only call on worker thread!"); if (mFileSystem->IsShutdown()) { return NS_ERROR_FAILURE; } bool exists; nsresult rv = mTargetPath->Exists(&exists); if (NS_WARN_IF(NS_FAILED(rv))) { return rv; } if (!exists) { if (!mFileSystem->ShouldCreateDirectory()) { return NS_ERROR_DOM_FILE_NOT_FOUND_ERR; } rv = mTargetPath->Create(nsIFile::DIRECTORY_TYPE, 0777); if (NS_WARN_IF(NS_FAILED(rv))) { return rv; } } // Get isDirectory. bool isDir; rv = mTargetPath->IsDirectory(&isDir); if (NS_WARN_IF(NS_FAILED(rv))) { return rv; } if (!isDir) { return NS_ERROR_DOM_FILESYSTEM_TYPE_MISMATCH_ERR; } nsCOMPtr<nsISimpleEnumerator> entries; rv = mTargetPath->GetDirectoryEntries(getter_AddRefs(entries)); if (NS_WARN_IF(NS_FAILED(rv))) { return rv; } bool filterOutSensitive = false; { HTMLSplitOnSpacesTokenizer tokenizer(mFilters, ';'); nsAutoString token; while (tokenizer.hasMoreTokens()) { token = tokenizer.nextToken(); if (token.EqualsLiteral("filter-out-sensitive")) { filterOutSensitive = true; } else { MOZ_CRASH("Unrecognized filter"); } } } for (;;) { bool hasMore = false; if (NS_WARN_IF(NS_FAILED(entries->HasMoreElements(&hasMore))) || !hasMore) { break; } nsCOMPtr<nsISupports> supp; if (NS_WARN_IF(NS_FAILED(entries->GetNext(getter_AddRefs(supp))))) { break; } nsCOMPtr<nsIFile> currFile = do_QueryInterface(supp); MOZ_ASSERT(currFile); bool isSpecial, isFile; if (NS_WARN_IF(NS_FAILED(currFile->IsSpecial(&isSpecial))) || isSpecial) { continue; } if (NS_WARN_IF(NS_FAILED(currFile->IsFile(&isFile)) || NS_FAILED(currFile->IsDirectory(&isDir))) || !(isFile || isDir)) { continue; } if (filterOutSensitive) { bool isHidden; if (NS_WARN_IF(NS_FAILED(currFile->IsHidden(&isHidden))) || isHidden) { continue; } nsAutoString leafName; if (NS_WARN_IF(NS_FAILED(currFile->GetLeafName(leafName)))) { continue; } if (leafName[0] == char16_t('.')) { continue; } } nsAutoString path; if (NS_WARN_IF(NS_FAILED(currFile->GetPath(path)))) { continue; } FileOrDirectoryPath element; element.mPath = path; element.mType = isDir ? FileOrDirectoryPath::eDirectoryPath : FileOrDirectoryPath::eFilePath; if (!mTargetData.AppendElement(element, fallible)) { return NS_ERROR_OUT_OF_MEMORY; } } return NS_OK; } nsresult GetDirectoryListingTaskParent::GetTargetPath(nsAString& aPath) const { return mTargetPath->GetPath(aPath); } } // namespace dom } // namespace mozilla