netwerk/cache2/CacheStorage.h

   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 #ifndef CacheStorage__h__
   6 #define CacheStorage__h__
   7
   8 #include "nsICacheStorage.h"
   9 #include "CacheEntry.h"
  10 #include "LoadContextInfo.h"
  11
  12 #include "nsRefPtrHashtable.h"
  13 #include "nsThreadUtils.h"
  14 #include "nsCOMPtr.h"
  15 #include "nsILoadContextInfo.h"
  16
  17 class nsIURI;
  18
  19 namespace mozilla {
  20 namespace net {
  21
  22 // This dance is needed to make CacheEntryTable declarable-only in headers
  23 // w/o exporting CacheEntry.h file to make nsNetModule.cpp compilable.
  24 using TCacheEntryTable = nsRefPtrHashtable<nsCStringHashKey, CacheEntry>;
  25 class CacheEntryTable : public TCacheEntryTable {
  26  public:
  27   enum EType { MEMORY_ONLY, ALL_ENTRIES };
  28
  29   explicit CacheEntryTable(EType aType) : mType(aType) {}
  30   EType Type() const { return mType; }
  31
  32  private:
  33   EType const mType;
  34   CacheEntryTable() = delete;
  35 };
  36
  37 class CacheStorage : public nsICacheStorage {
  38   NS_DECL_THREADSAFE_ISUPPORTS
  39   NS_DECL_NSICACHESTORAGE
  40
  41  public:
  42   CacheStorage(nsILoadContextInfo* aInfo, bool aAllowDisk, bool aSkipSizeCheck,
  43                bool aPinning);
  44
  45  protected:
  46   virtual ~CacheStorage() = default;
  47
  48   RefPtr<LoadContextInfo> mLoadContextInfo;
  49   bool mWriteToDisk : 1;
  50   bool mSkipSizeCheck : 1;
  51   bool mPinning : 1;
  52
  53  public:
  54   nsILoadContextInfo* LoadInfo() const { return mLoadContextInfo; }
  55   bool WriteToDisk() const {
  56     return mWriteToDisk &&
  57            (!mLoadContextInfo || !mLoadContextInfo->IsPrivate());
  58   }
  59   bool SkipSizeCheck() const { return mSkipSizeCheck; }
  60   bool Pinning() const { return mPinning; }
  61 };
  62
  63 }  // namespace net
  64 }  // namespace mozilla
  65
  66 #endif