netwerk/cache2/CacheIndexIterator.cpp

   1 /* This Source Code Form is subject to the terms of the Mozilla Public
   2  * License, v. 2.0. If a copy of the MPL was not distributed with this
   3  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   4
   5 #include "CacheLog.h"
   6 #include "CacheIndexIterator.h"
   7 #include "CacheIndex.h"
   8 #include "nsString.h"
   9 #include "mozilla/DebugOnly.h"
  10
  11
  12 namespace mozilla {
  13 namespace net {
  14
  15 CacheIndexIterator::CacheIndexIterator(CacheIndex *aIndex, bool aAddNew)
  16   : mStatus(NS_OK)
  17   , mIndex(aIndex)
  18   , mAddNew(aAddNew)
  19 {
  20   LOG(("CacheIndexIterator::CacheIndexIterator() [this=%p]", this));
  21 }
  22
  23 CacheIndexIterator::~CacheIndexIterator()
  24 {
  25   LOG(("CacheIndexIterator::~CacheIndexIterator() [this=%p]", this));
  26
  27   Close();
  28 }
  29
  30 nsresult
  31 CacheIndexIterator::GetNextHash(SHA1Sum::Hash *aHash)
  32 {
  33   LOG(("CacheIndexIterator::GetNextHash() [this=%p]", this));
  34
  35   StaticMutexAutoLock lock(CacheIndex::sLock);
  36
  37   if (NS_FAILED(mStatus)) {
  38     return mStatus;
  39   }
  40
  41   if (!mRecords.Length()) {
  42     CloseInternal(NS_ERROR_NOT_AVAILABLE);
  43     return mStatus;
  44   }
  45
  46   memcpy(aHash, mRecords[mRecords.Length() - 1]->mHash, sizeof(SHA1Sum::Hash));
  47   mRecords.RemoveLastElement();
  48
  49   return NS_OK;
  50 }
  51
  52 nsresult
  53 CacheIndexIterator::Close()
  54 {
  55   LOG(("CacheIndexIterator::Close() [this=%p]", this));
  56
  57   StaticMutexAutoLock lock(CacheIndex::sLock);
  58
  59   return CloseInternal(NS_ERROR_NOT_AVAILABLE);
  60 }
  61
  62 nsresult
  63 CacheIndexIterator::CloseInternal(nsresult aStatus)
  64 {
  65   LOG(("CacheIndexIterator::CloseInternal() [this=%p, status=0x%08" PRIx32 "]", this,
  66        static_cast<uint32_t>(aStatus)));
  67
  68   // Make sure status will be a failure
  69   MOZ_ASSERT(NS_FAILED(aStatus));
  70   if (NS_SUCCEEDED(aStatus)) {
  71     aStatus = NS_ERROR_UNEXPECTED;
  72   }
  73
  74   if (NS_FAILED(mStatus)) {
  75     return NS_ERROR_NOT_AVAILABLE;
  76   }
  77
  78   DebugOnly<bool> removed = mIndex->mIterators.RemoveElement(this);
  79   MOZ_ASSERT(removed);
  80   mStatus = aStatus;
  81
  82   return NS_OK;
  83 }
  84
  85 void
  86 CacheIndexIterator::AddRecord(CacheIndexRecord *aRecord)
  87 {
  88   LOG(("CacheIndexIterator::AddRecord() [this=%p, record=%p]", this, aRecord));
  89
  90   mRecords.AppendElement(aRecord);
  91 }
  92
  93 bool
  94 CacheIndexIterator::RemoveRecord(CacheIndexRecord *aRecord)
  95 {
  96   LOG(("CacheIndexIterator::RemoveRecord() [this=%p, record=%p]", this,
  97        aRecord));
  98
  99   return mRecords.RemoveElement(aRecord);
 100 }
 101
 102 bool
 103 CacheIndexIterator::ReplaceRecord(CacheIndexRecord *aOldRecord,
 104                                   CacheIndexRecord *aNewRecord)
 105 {
 106   LOG(("CacheIndexIterator::ReplaceRecord() [this=%p, oldRecord=%p, "
 107        "newRecord=%p]", this, aOldRecord, aNewRecord));
 108
 109   if (RemoveRecord(aOldRecord)) {
 110     AddRecord(aNewRecord);
 111     return true;
 112   }
 113
 114   return false;
 115 }
 116
 117 } // namespace net
 118 } // namespace mozilla