netwerk/cookie/CookieServiceChild.h

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* This Source Code Form is subject to the terms of the Mozilla Public
   3  * License, v. 2.0. If a copy of the MPL was not distributed with this
   4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   5
   6 #ifndef mozilla_net_CookieServiceChild_h__
   7 #define mozilla_net_CookieServiceChild_h__
   8
   9 #include "CookieKey.h"
  10 #include "mozilla/net/PCookieServiceChild.h"
  11 #include "nsClassHashtable.h"
  12 #include "nsICookieService.h"
  13 #include "mozIThirdPartyUtil.h"
  14 #include "nsWeakReference.h"
  15 #include "nsThreadUtils.h"
  16
  17 class nsIEffectiveTLDService;
  18 class nsILoadInfo;
  19
  20 namespace mozilla {
  21 namespace net {
  22
  23 class Cookie;
  24 class CookieStruct;
  25
  26 class CookieServiceChild final : public PCookieServiceChild,
  27                                  public nsICookieService,
  28                                  public nsSupportsWeakReference {
  29   friend class PCookieServiceChild;
  30
  31  public:
  32   NS_DECL_ISUPPORTS
  33   NS_DECL_NSICOOKIESERVICE
  34
  35   typedef nsTArray<RefPtr<Cookie>> CookiesList;
  36   typedef nsClassHashtable<CookieKey, CookiesList> CookiesMap;
  37
  38   CookieServiceChild();
  39
  40   void Init();
  41
  42   static already_AddRefed<CookieServiceChild> GetSingleton();
  43
  44   RefPtr<GenericPromise> TrackCookieLoad(nsIChannel* aChannel);
  45
  46  private:
  47   ~CookieServiceChild();
  48
  49   void RecordDocumentCookie(Cookie* aCookie, const OriginAttributes& aAttrs);
  50
  51   uint32_t CountCookiesFromHashTable(const nsACString& aBaseDomain,
  52                                      const OriginAttributes& aOriginAttrs);
  53
  54   static bool RequireThirdPartyCheck(nsILoadInfo* aLoadInfo);
  55
  56   mozilla::ipc::IPCResult RecvTrackCookiesLoad(
  57       nsTArray<CookieStructTable>&& aCookiesListTable);
  58
  59   mozilla::ipc::IPCResult RecvRemoveAll();
  60
  61   mozilla::ipc::IPCResult RecvRemoveBatchDeletedCookies(
  62       nsTArray<CookieStruct>&& aCookiesList,
  63       nsTArray<OriginAttributes>&& aAttrsList);
  64
  65   mozilla::ipc::IPCResult RecvRemoveCookie(const CookieStruct& aCookie,
  66                                            const OriginAttributes& aAttrs);
  67
  68   mozilla::ipc::IPCResult RecvAddCookie(const CookieStruct& aCookie,
  69                                         const OriginAttributes& aAttrs);
  70
  71   void RemoveSingleCookie(const CookieStruct& aCookie,
  72                           const OriginAttributes& aAttrs);
  73
  74   CookiesMap mCookiesMap;
  75   nsCOMPtr<mozIThirdPartyUtil> mThirdPartyUtil;
  76   nsCOMPtr<nsIEffectiveTLDService> mTLDService;
  77 };
  78
  79 }  // namespace net
  80 }  // namespace mozilla
  81
  82 #endif  // mozilla_net_CookieServiceChild_h__