dom/filesystem/GetDirectoryListingTask.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
   5  * You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "GetDirectoryListingTask.h"
   8
   9 #include "HTMLSplitOnSpacesTokenizer.h"
  10 #include "js/Value.h"
  11 #include "mozilla/dom/FileBlobImpl.h"
  12 #include "mozilla/dom/FileSystemBase.h"
  13 #include "mozilla/dom/FileSystemUtils.h"
  14 #include "mozilla/dom/IPCBlobUtils.h"
  15 #include "mozilla/dom/PFileSystemParams.h"
  16 #include "mozilla/dom/Promise.h"
  17 #include "mozilla/dom/UnionTypes.h"
  18 #include "mozilla/ipc/BackgroundParent.h"
  19 #include "nsIFile.h"
  20 #include "nsString.h"
  21
  22 namespace mozilla::dom {
  23
  24 /**
  25  * GetDirectoryListingTaskChild
  26  */
  27
  28 /* static */
  29 already_AddRefed<GetDirectoryListingTaskChild>
  30 GetDirectoryListingTaskChild::Create(FileSystemBase* aFileSystem,
  31                                      Directory* aDirectory,
  32                                      nsIFile* aTargetPath,
  33                                      const nsAString& aFilters,
  34                                      ErrorResult& aRv) {
  35   MOZ_ASSERT(aFileSystem);
  36   MOZ_ASSERT(aDirectory);
  37   aFileSystem->AssertIsOnOwningThread();
  38
  39   nsCOMPtr<nsIGlobalObject> globalObject = aFileSystem->GetParentObject();
  40   MOZ_ASSERT(globalObject);
  41
  42   RefPtr<GetDirectoryListingTaskChild> task = new GetDirectoryListingTaskChild(
  43       globalObject, aFileSystem, aDirectory, aTargetPath, aFilters);
  44
  45   // aTargetPath can be null. In this case SetError will be called.
  46
  47   task->mPromise = Promise::Create(globalObject, aRv);
  48   if (NS_WARN_IF(aRv.Failed())) {
  49     return nullptr;
  50   }
  51
  52   return task.forget();
  53 }
  54
  55 GetDirectoryListingTaskChild::GetDirectoryListingTaskChild(
  56     nsIGlobalObject* aGlobalObject, FileSystemBase* aFileSystem,
  57     Directory* aDirectory, nsIFile* aTargetPath, const nsAString& aFilters)
  58     : FileSystemTaskChildBase(aGlobalObject, aFileSystem),
  59       mDirectory(aDirectory),
  60       mTargetPath(aTargetPath),
  61       mFilters(aFilters) {
  62   MOZ_ASSERT(aFileSystem);
  63   aFileSystem->AssertIsOnOwningThread();
  64 }
  65
  66 GetDirectoryListingTaskChild::~GetDirectoryListingTaskChild() {
  67   mFileSystem->AssertIsOnOwningThread();
  68 }
  69
  70 already_AddRefed<Promise> GetDirectoryListingTaskChild::GetPromise() {
  71   mFileSystem->AssertIsOnOwningThread();
  72   return RefPtr<Promise>(mPromise).forget();
  73 }
  74
  75 FileSystemParams GetDirectoryListingTaskChild::GetRequestParams(
  76     const nsString& aSerializedDOMPath, ErrorResult& aRv) const {
  77   mFileSystem->AssertIsOnOwningThread();
  78
  79   // this is the real path.
  80   nsAutoString path;
  81   aRv = mTargetPath->GetPath(path);
  82   if (NS_WARN_IF(aRv.Failed())) {
  83     return FileSystemGetDirectoryListingParams();
  84   }
  85
  86   // this is the dom path.
  87   nsAutoString directoryPath;
  88   mDirectory->GetPath(directoryPath, aRv);
  89   if (NS_WARN_IF(aRv.Failed())) {
  90     return FileSystemGetDirectoryListingParams();
  91   }
  92
  93   return FileSystemGetDirectoryListingParams(aSerializedDOMPath, path,
  94                                              directoryPath, mFilters);
  95 }
  96
  97 void GetDirectoryListingTaskChild::SetSuccessRequestResult(
  98     const FileSystemResponseValue& aValue, ErrorResult& aRv) {
  99   mFileSystem->AssertIsOnOwningThread();
 100   MOZ_ASSERT(aValue.type() ==
 101              FileSystemResponseValue::TFileSystemDirectoryListingResponse);
 102
 103   FileSystemDirectoryListingResponse r = aValue;
 104   for (uint32_t i = 0; i < r.data().Length(); ++i) {
 105     const FileSystemDirectoryListingResponseData& data = r.data()[i];
 106
 107     OwningFileOrDirectory* ofd = mTargetData.AppendElement(fallible);
 108     if (!ofd) {
 109       aRv.Throw(NS_ERROR_OUT_OF_MEMORY);
 110       return;
 111     }
 112
 113     if (data.type() == FileSystemDirectoryListingResponseData::
 114                            TFileSystemDirectoryListingResponseFile) {
 115       const FileSystemDirectoryListingResponseFile& d =
 116           data.get_FileSystemDirectoryListingResponseFile();
 117
 118       RefPtr<BlobImpl> blobImpl = IPCBlobUtils::Deserialize(d.blob());
 119       MOZ_ASSERT(blobImpl);
 120
 121       nsCOMPtr<nsIGlobalObject> globalObject = mFileSystem->GetParentObject();
 122       MOZ_ASSERT(globalObject);
 123
 124       RefPtr<File> file = File::Create(globalObject, blobImpl);
 125       MOZ_ASSERT(file);
 126
 127       ofd->SetAsFile() = file;
 128     } else {
 129       MOZ_ASSERT(data.type() ==
 130                  FileSystemDirectoryListingResponseData::
 131                      TFileSystemDirectoryListingResponseDirectory);
 132       const FileSystemDirectoryListingResponseDirectory& d =
 133           data.get_FileSystemDirectoryListingResponseDirectory();
 134
 135       nsCOMPtr<nsIFile> path;
 136       aRv = NS_NewLocalFile(d.directoryRealPath(), true, getter_AddRefs(path));
 137       if (NS_WARN_IF(aRv.Failed())) {
 138         return;
 139       }
 140
 141       RefPtr<Directory> directory =
 142           Directory::Create(mFileSystem->GetParentObject(), path, mFileSystem);
 143       MOZ_ASSERT(directory);
 144
 145       ofd->SetAsDirectory() = directory;
 146     }
 147   }
 148 }
 149
 150 void GetDirectoryListingTaskChild::HandlerCallback() {
 151   mFileSystem->AssertIsOnOwningThread();
 152
 153   if (mFileSystem->IsShutdown()) {
 154     mPromise = nullptr;
 155     return;
 156   }
 157
 158   if (HasError()) {
 159     mPromise->MaybeReject(mErrorValue);
 160     mPromise = nullptr;
 161     return;
 162   }
 163
 164   mPromise->MaybeResolve(mTargetData);
 165   mPromise = nullptr;
 166 }
 167
 168 /**
 169  * GetDirectoryListingTaskParent
 170  */
 171
 172 /* static */
 173 already_AddRefed<GetDirectoryListingTaskParent>
 174 GetDirectoryListingTaskParent::Create(
 175     FileSystemBase* aFileSystem,
 176     const FileSystemGetDirectoryListingParams& aParam,
 177     FileSystemRequestParent* aParent, ErrorResult& aRv) {
 178   MOZ_ASSERT(XRE_IsParentProcess(), "Only call from parent process!");
 179   mozilla::ipc::AssertIsOnBackgroundThread();
 180   MOZ_ASSERT(aFileSystem);
 181
 182   RefPtr<GetDirectoryListingTaskParent> task =
 183       new GetDirectoryListingTaskParent(aFileSystem, aParam, aParent);
 184
 185   aRv = NS_NewLocalFile(aParam.realPath(), true,
 186                         getter_AddRefs(task->mTargetPath));
 187   if (NS_WARN_IF(aRv.Failed())) {
 188     return nullptr;
 189   }
 190
 191   return task.forget();
 192 }
 193
 194 GetDirectoryListingTaskParent::GetDirectoryListingTaskParent(
 195     FileSystemBase* aFileSystem,
 196     const FileSystemGetDirectoryListingParams& aParam,
 197     FileSystemRequestParent* aParent)
 198     : FileSystemTaskParentBase(aFileSystem, aParam, aParent),
 199       mDOMPath(aParam.domPath()),
 200       mFilters(aParam.filters()) {
 201   MOZ_ASSERT(XRE_IsParentProcess(), "Only call from parent process!");
 202   mozilla::ipc::AssertIsOnBackgroundThread();
 203   MOZ_ASSERT(aFileSystem);
 204 }
 205
 206 FileSystemResponseValue GetDirectoryListingTaskParent::GetSuccessRequestResult(
 207     ErrorResult& aRv) const {
 208   mozilla::ipc::AssertIsOnBackgroundThread();
 209
 210   nsTArray<FileSystemDirectoryListingResponseData> inputs;
 211
 212   for (unsigned i = 0; i < mTargetData.Length(); i++) {
 213     if (mTargetData[i].mType == FileOrDirectoryPath::eFilePath) {
 214       nsCOMPtr<nsIFile> path;
 215       nsresult rv =
 216           NS_NewLocalFile(mTargetData[i].mPath, true, getter_AddRefs(path));
 217       if (NS_WARN_IF(NS_FAILED(rv))) {
 218         continue;
 219       }
 220
 221       FileSystemDirectoryListingResponseFile fileData;
 222       RefPtr<BlobImpl> blobImpl = new FileBlobImpl(path);
 223
 224       nsAutoString filePath;
 225       filePath.Assign(mDOMPath);
 226
 227       // This is specific for unix root filesystem.
 228       if (!mDOMPath.EqualsLiteral(FILESYSTEM_DOM_PATH_SEPARATOR_LITERAL)) {
 229         filePath.AppendLiteral(FILESYSTEM_DOM_PATH_SEPARATOR_LITERAL);
 230       }
 231
 232       nsAutoString name;
 233       blobImpl->GetName(name);
 234       filePath.Append(name);
 235       blobImpl->SetDOMPath(filePath);
 236
 237       IPCBlob ipcBlob;
 238       rv = IPCBlobUtils::Serialize(blobImpl, ipcBlob);
 239       if (NS_WARN_IF(NS_FAILED(rv))) {
 240         continue;
 241       }
 242
 243       fileData.blob() = ipcBlob;
 244       inputs.AppendElement(fileData);
 245     } else {
 246       MOZ_ASSERT(mTargetData[i].mType == FileOrDirectoryPath::eDirectoryPath);
 247       FileSystemDirectoryListingResponseDirectory directoryData;
 248       directoryData.directoryRealPath() = mTargetData[i].mPath;
 249       inputs.AppendElement(directoryData);
 250     }
 251   }
 252
 253   FileSystemDirectoryListingResponse response;
 254   response.data() = std::move(inputs);
 255   return response;
 256 }
 257
 258 nsresult GetDirectoryListingTaskParent::IOWork() {
 259   MOZ_ASSERT(XRE_IsParentProcess(), "Only call from parent process!");
 260   MOZ_ASSERT(!NS_IsMainThread(), "Only call on worker thread!");
 261
 262   if (mFileSystem->IsShutdown()) {
 263     return NS_ERROR_FAILURE;
 264   }
 265
 266   bool exists;
 267   nsresult rv = mTargetPath->Exists(&exists);
 268   if (NS_WARN_IF(NS_FAILED(rv))) {
 269     return rv;
 270   }
 271
 272   if (!exists) {
 273     if (!mFileSystem->ShouldCreateDirectory()) {
 274       return NS_ERROR_DOM_FILE_NOT_FOUND_ERR;
 275     }
 276
 277     rv = mTargetPath->Create(nsIFile::DIRECTORY_TYPE, 0777);
 278     if (NS_WARN_IF(NS_FAILED(rv))) {
 279       return rv;
 280     }
 281   }
 282
 283   // Get isDirectory.
 284   bool isDir;
 285   rv = mTargetPath->IsDirectory(&isDir);
 286   if (NS_WARN_IF(NS_FAILED(rv))) {
 287     return rv;
 288   }
 289
 290   if (!isDir) {
 291     return NS_ERROR_DOM_FILESYSTEM_TYPE_MISMATCH_ERR;
 292   }
 293
 294   nsCOMPtr<nsIDirectoryEnumerator> entries;
 295   rv = mTargetPath->GetDirectoryEntries(getter_AddRefs(entries));
 296   if (NS_WARN_IF(NS_FAILED(rv))) {
 297     return rv;
 298   }
 299
 300   bool filterOutSensitive = false;
 301   {
 302     HTMLSplitOnSpacesTokenizer tokenizer(mFilters, ';');
 303     nsAutoString token;
 304     while (tokenizer.hasMoreTokens()) {
 305       token = tokenizer.nextToken();
 306       if (token.EqualsLiteral("filter-out-sensitive")) {
 307         filterOutSensitive = true;
 308       } else {
 309         MOZ_CRASH("Unrecognized filter");
 310       }
 311     }
 312   }
 313
 314   for (;;) {
 315     nsCOMPtr<nsIFile> currFile;
 316     if (NS_WARN_IF(NS_FAILED(entries->GetNextFile(getter_AddRefs(currFile)))) ||
 317         !currFile) {
 318       break;
 319     }
 320     bool isSpecial, isFile;
 321     if (NS_WARN_IF(NS_FAILED(currFile->IsSpecial(&isSpecial))) || isSpecial) {
 322       continue;
 323     }
 324     if (NS_WARN_IF(NS_FAILED(currFile->IsFile(&isFile)) ||
 325                    NS_FAILED(currFile->IsDirectory(&isDir))) ||
 326         !(isFile || isDir)) {
 327       continue;
 328     }
 329
 330     if (filterOutSensitive) {
 331       bool isHidden;
 332       if (NS_WARN_IF(NS_FAILED(currFile->IsHidden(&isHidden))) || isHidden) {
 333         continue;
 334       }
 335       nsAutoString leafName;
 336       if (NS_WARN_IF(NS_FAILED(currFile->GetLeafName(leafName)))) {
 337         continue;
 338       }
 339       if (leafName[0] == char16_t('.')) {
 340         continue;
 341       }
 342     }
 343
 344     nsAutoString path;
 345     if (NS_WARN_IF(NS_FAILED(currFile->GetPath(path)))) {
 346       continue;
 347     }
 348
 349     FileOrDirectoryPath element;
 350     element.mPath = path;
 351     element.mType = isDir ? FileOrDirectoryPath::eDirectoryPath
 352                           : FileOrDirectoryPath::eFilePath;
 353
 354     if (!mTargetData.AppendElement(element, fallible)) {
 355       return NS_ERROR_OUT_OF_MEMORY;
 356     }
 357   }
 358   return NS_OK;
 359 }
 360
 361 nsresult GetDirectoryListingTaskParent::GetTargetPath(nsAString& aPath) const {
 362   return mTargetPath->GetPath(aPath);
 363 }
 364
 365 }  // namespace mozilla::dom