dom/base/nsContentPermissionHelper.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 "nsContentPermissionHelper.h"
   6 #include "nsIContentPermissionPrompt.h"
   7 #include "nsCOMPtr.h"
   8 #include "nsIDOMElement.h"
   9 #include "nsIPrincipal.h"
  10 #include "mozilla/dom/Element.h"
  11 #include "mozilla/unused.h"
  12
  13 using mozilla::unused;          // <snicker>
  14 using namespace mozilla::dom;
  15
  16 nsContentPermissionRequestProxy::nsContentPermissionRequestProxy()
  17 {
  18   MOZ_COUNT_CTOR(nsContentPermissionRequestProxy);
  19 }
  20
  21 nsContentPermissionRequestProxy::~nsContentPermissionRequestProxy()
  22 {
  23   MOZ_COUNT_DTOR(nsContentPermissionRequestProxy);
  24 }
  25
  26 nsresult
  27 nsContentPermissionRequestProxy::Init(const nsACString & type,
  28                                       const nsACString & access,
  29                                       ContentPermissionRequestParent* parent)
  30 {
  31   NS_ASSERTION(parent, "null parent");
  32   mParent = parent;
  33   mType   = type;
  34   mAccess = access;
  35
  36   nsCOMPtr<nsIContentPermissionPrompt> prompt = do_CreateInstance(NS_CONTENT_PERMISSION_PROMPT_CONTRACTID);
  37   if (!prompt) {
  38     return NS_ERROR_FAILURE;
  39   }
  40
  41   prompt->Prompt(this);
  42   return NS_OK;
  43 }
  44
  45 void
  46 nsContentPermissionRequestProxy::OnParentDestroyed()
  47 {
  48   mParent = nullptr;
  49 }
  50
  51 NS_IMPL_ISUPPORTS1(nsContentPermissionRequestProxy, nsIContentPermissionRequest)
  52
  53 NS_IMETHODIMP
  54 nsContentPermissionRequestProxy::GetType(nsACString & aType)
  55 {
  56   aType = mType;
  57   return NS_OK;
  58 }
  59
  60 NS_IMETHODIMP
  61 nsContentPermissionRequestProxy::GetAccess(nsACString & aAccess)
  62 {
  63   aAccess = mAccess;
  64   return NS_OK;
  65 }
  66
  67 NS_IMETHODIMP
  68 nsContentPermissionRequestProxy::GetWindow(nsIDOMWindow * *aRequestingWindow)
  69 {
  70   NS_ENSURE_ARG_POINTER(aRequestingWindow);
  71   *aRequestingWindow = nullptr; // ipc doesn't have a window
  72   return NS_OK;
  73 }
  74
  75 NS_IMETHODIMP
  76 nsContentPermissionRequestProxy::GetPrincipal(nsIPrincipal * *aRequestingPrincipal)
  77 {
  78   NS_ENSURE_ARG_POINTER(aRequestingPrincipal);
  79   if (mParent == nullptr) {
  80     return NS_ERROR_FAILURE;
  81   }
  82
  83   NS_ADDREF(*aRequestingPrincipal = mParent->mPrincipal);
  84   return NS_OK;
  85 }
  86
  87 NS_IMETHODIMP
  88 nsContentPermissionRequestProxy::GetElement(nsIDOMElement * *aRequestingElement)
  89 {
  90   NS_ENSURE_ARG_POINTER(aRequestingElement);
  91   if (mParent == nullptr) {
  92     return NS_ERROR_FAILURE;
  93   }
  94
  95   nsCOMPtr<nsIDOMElement> elem = do_QueryInterface(mParent->mElement);
  96   elem.forget(aRequestingElement);
  97   return NS_OK;
  98 }
  99
 100 NS_IMETHODIMP
 101 nsContentPermissionRequestProxy::Cancel()
 102 {
 103   if (mParent == nullptr) {
 104     return NS_ERROR_FAILURE;
 105   }
 106
 107   unused << ContentPermissionRequestParent::Send__delete__(mParent, false);
 108   mParent = nullptr;
 109   return NS_OK;
 110 }
 111
 112 NS_IMETHODIMP
 113 nsContentPermissionRequestProxy::Allow()
 114 {
 115   if (mParent == nullptr) {
 116     return NS_ERROR_FAILURE;
 117   }
 118   unused << ContentPermissionRequestParent::Send__delete__(mParent, true);
 119   mParent = nullptr;
 120   return NS_OK;
 121 }
 122
 123 namespace mozilla {
 124 namespace dom {
 125
 126 ContentPermissionRequestParent::ContentPermissionRequestParent(const nsACString& aType,
 127                                                                const nsACString& aAccess,
 128                                                                Element* aElement,
 129                                                                const IPC::Principal& aPrincipal)
 130 {
 131   MOZ_COUNT_CTOR(ContentPermissionRequestParent);
 132
 133   mPrincipal = aPrincipal;
 134   mElement   = aElement;
 135   mType      = aType;
 136   mAccess    = aAccess;
 137 }
 138
 139 ContentPermissionRequestParent::~ContentPermissionRequestParent()
 140 {
 141   MOZ_COUNT_DTOR(ContentPermissionRequestParent);
 142 }
 143
 144 bool
 145 ContentPermissionRequestParent::Recvprompt()
 146 {
 147   mProxy = new nsContentPermissionRequestProxy();
 148   NS_ASSERTION(mProxy, "Alloc of request proxy failed");
 149   if (NS_FAILED(mProxy->Init(mType, mAccess, this))) {
 150     mProxy->Cancel();
 151   }
 152   return true;
 153 }
 154
 155 void
 156 ContentPermissionRequestParent::ActorDestroy(ActorDestroyReason why)
 157 {
 158   if (mProxy) {
 159     mProxy->OnParentDestroyed();
 160   }
 161 }
 162
 163 } // namespace dom
 164 } // namespace mozilla