dom/indexedDB/FileInfo.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=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
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "FileInfo.h"
   8 #include "nsThreadUtils.h"
   9 #include "mozilla/dom/quota/QuotaManager.h"
  10
  11 USING_INDEXEDDB_NAMESPACE
  12
  13 namespace {
  14
  15 class CleanupFileRunnable MOZ_FINAL : public nsIRunnable
  16 {
  17   ~CleanupFileRunnable() {}
  18
  19 public:
  20   NS_DECL_THREADSAFE_ISUPPORTS
  21   NS_DECL_NSIRUNNABLE
  22
  23   CleanupFileRunnable(FileManager* aFileManager, int64_t aFileId);
  24
  25 private:
  26   nsRefPtr<FileManager> mFileManager;
  27   int64_t mFileId;
  28 };
  29
  30 } // anonymous namespace
  31
  32 // static
  33 FileInfo*
  34 FileInfo::Create(FileManager* aFileManager, int64_t aId)
  35 {
  36   MOZ_ASSERT(aId > 0, "Wrong id!");
  37
  38   if (aId <= INT16_MAX) {
  39     return new FileInfo16(aFileManager, aId);
  40   }
  41
  42   if (aId <= INT32_MAX) {
  43     return new FileInfo32(aFileManager, aId);
  44   }
  45
  46   return new FileInfo64(aFileManager, aId);
  47 }
  48
  49 void
  50 FileInfo::GetReferences(int32_t* aRefCnt, int32_t* aDBRefCnt,
  51                         int32_t* aSliceRefCnt)
  52 {
  53   if (IndexedDatabaseManager::IsClosed()) {
  54     NS_ERROR("Shouldn't be called after shutdown!");
  55
  56     if (aRefCnt) {
  57       *aRefCnt = -1;
  58     }
  59
  60     if (aDBRefCnt) {
  61       *aDBRefCnt = -1;
  62     }
  63
  64     if (aSliceRefCnt) {
  65       *aSliceRefCnt = -1;
  66     }
  67
  68     return;
  69   }
  70
  71   MutexAutoLock lock(IndexedDatabaseManager::FileMutex());
  72
  73   if (aRefCnt) {
  74     *aRefCnt = mRefCnt;
  75   }
  76
  77   if (aDBRefCnt) {
  78     *aDBRefCnt = mDBRefCnt;
  79   }
  80
  81   if (aSliceRefCnt) {
  82     *aSliceRefCnt = mSliceRefCnt;
  83   }
  84 }
  85
  86 void
  87 FileInfo::UpdateReferences(mozilla::ThreadSafeAutoRefCnt& aRefCount,
  88                            int32_t aDelta, bool aClear)
  89 {
  90   if (IndexedDatabaseManager::IsClosed()) {
  91     NS_ERROR("Shouldn't be called after shutdown!");
  92     return;
  93   }
  94
  95   bool needsCleanup;
  96   {
  97     MutexAutoLock lock(IndexedDatabaseManager::FileMutex());
  98
  99     aRefCount = aClear ? 0 : aRefCount + aDelta;
 100
 101     if (mRefCnt + mDBRefCnt + mSliceRefCnt > 0) {
 102       return;
 103     }
 104
 105     mFileManager->mFileInfos.Remove(Id());
 106
 107     needsCleanup = !mFileManager->Invalidated();
 108   }
 109
 110   if (needsCleanup) {
 111     Cleanup();
 112   }
 113
 114   delete this;
 115 }
 116
 117 void
 118 FileInfo::Cleanup()
 119 {
 120   nsRefPtr<CleanupFileRunnable> cleaner =
 121     new CleanupFileRunnable(mFileManager, Id());
 122
 123   // IndexedDatabaseManager is main-thread only.
 124   if (!NS_IsMainThread()) {
 125     NS_DispatchToMainThread(cleaner);
 126     return;
 127   }
 128
 129   cleaner->Run();
 130 }
 131
 132 CleanupFileRunnable::CleanupFileRunnable(FileManager* aFileManager,
 133                                          int64_t aFileId)
 134 : mFileManager(aFileManager), mFileId(aFileId)
 135 {
 136 }
 137
 138 NS_IMPL_ISUPPORTS(CleanupFileRunnable,
 139                   nsIRunnable)
 140
 141 NS_IMETHODIMP
 142 CleanupFileRunnable::Run()
 143 {
 144   if (mozilla::dom::quota::QuotaManager::IsShuttingDown()) {
 145     return NS_OK;
 146   }
 147
 148   nsRefPtr<IndexedDatabaseManager> mgr = IndexedDatabaseManager::Get();
 149   MOZ_ASSERT(mgr);
 150
 151   if (NS_FAILED(mgr->AsyncDeleteFile(mFileManager, mFileId))) {
 152     NS_WARNING("Failed to delete file asynchronously!");
 153   }
 154
 155   return NS_OK;
 156 }