netwerk/url-classifier/ChannelClassifierService.h

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim:set expandtab ts=2 sw=2 sts=2 cin: */
   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 #ifndef mozilla_net_ChannelClassifierService_h
   8 #define mozilla_net_ChannelClassifierService_h
   9
  10 #include "nsIChannelClassifierService.h"
  11 #include "mozilla/net/UrlClassifierCommon.h"
  12 #include "nsCOMPtr.h"
  13 #include "nsTArray.h"
  14
  15 namespace mozilla {
  16 namespace net {
  17
  18 enum class ChannelBlockDecision {
  19   Blocked,
  20   Replaced,
  21   Allowed,
  22 };
  23
  24 class UrlClassifierBlockedChannel final
  25     : public nsIUrlClassifierBlockedChannel {
  26  public:
  27   NS_DECL_ISUPPORTS
  28   NS_DECL_NSIURLCLASSIFIERBLOCKEDCHANNEL
  29
  30   explicit UrlClassifierBlockedChannel(nsIChannel* aChannel);
  31
  32   bool IsUnblocked() const {
  33     return mDecision != ChannelBlockDecision::Blocked;
  34   }
  35
  36   ChannelBlockDecision GetDecision() { return mDecision; };
  37
  38   void SetReason(const nsACString& aFeatureName, const nsACString& aTableName);
  39
  40  protected:
  41   ~UrlClassifierBlockedChannel() = default;
  42
  43  private:
  44   nsCOMPtr<nsIChannel> mChannel;
  45   ChannelBlockDecision mDecision;
  46   uint8_t mReason;
  47   nsCString mTables;
  48 };
  49
  50 class ChannelClassifierService final : public nsIChannelClassifierService {
  51  public:
  52   NS_DECL_ISUPPORTS
  53   NS_DECL_NSICHANNELCLASSIFIERSERVICE
  54
  55   friend class UrlClassifierBlockedChannel;
  56
  57   static already_AddRefed<nsIChannelClassifierService> GetSingleton();
  58
  59   static ChannelBlockDecision OnBeforeBlockChannel(
  60       nsIChannel* aChannel, const nsACString& aFeatureName,
  61       const nsACString& aTableName);
  62
  63   nsresult OnBeforeBlockChannel(nsIChannel* aChannel,
  64                                 const nsACString& aFeatureName,
  65                                 const nsACString& aTableName,
  66                                 ChannelBlockDecision& aDecision);
  67
  68   bool HasListener() const { return !mListeners.IsEmpty(); }
  69
  70  private:
  71   ChannelClassifierService();
  72   ~ChannelClassifierService() = default;
  73
  74   nsTArray<nsCOMPtr<nsIObserver>> mListeners;
  75 };
  76
  77 }  // namespace net
  78 }  // namespace mozilla
  79
  80 #endif  // mozilla_net_ChannelClassifierService_h