extensions/cookie/nsPopupWindowManager.cpp

   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 #include "nsPopupWindowManager.h"
   7
   8 #include "nsCRT.h"
   9 #include "nsIServiceManager.h"
  10 #include "nsIPrefService.h"
  11 #include "nsIPrefBranch.h"
  12 #include "nsIPrincipal.h"
  13 #include "nsIURI.h"
  14 #include "mozilla/Services.h"
  15
  16 /**
  17  * The Popup Window Manager maintains popup window permissions by website.
  18  */
  19
  20 static const char kPopupDisablePref[] = "dom.disable_open_during_load";
  21
  22 //*****************************************************************************
  23 //*** nsPopupWindowManager object management and nsISupports
  24 //*****************************************************************************
  25
  26 nsPopupWindowManager::nsPopupWindowManager() :
  27   mPolicy(ALLOW_POPUP)
  28 {
  29 }
  30
  31 nsPopupWindowManager::~nsPopupWindowManager()
  32 {
  33 }
  34
  35 NS_IMPL_ISUPPORTS(nsPopupWindowManager,
  36                   nsIPopupWindowManager,
  37                   nsIObserver,
  38                   nsISupportsWeakReference)
  39
  40 nsresult
  41 nsPopupWindowManager::Init()
  42 {
  43   nsresult rv;
  44   mPermissionManager = mozilla::services::GetPermissionManager();
  45
  46   nsCOMPtr<nsIPrefBranch> prefBranch =
  47     do_GetService(NS_PREFSERVICE_CONTRACTID, &rv);
  48   if (NS_SUCCEEDED(rv)) {
  49     bool permission;
  50     rv = prefBranch->GetBoolPref(kPopupDisablePref, &permission);
  51     if (NS_FAILED(rv)) {
  52       permission = true;
  53     }
  54     mPolicy = permission ? (uint32_t) DENY_POPUP : (uint32_t) ALLOW_POPUP;
  55
  56     prefBranch->AddObserver(kPopupDisablePref, this, true);
  57   }
  58
  59   return NS_OK;
  60 }
  61
  62 //*****************************************************************************
  63 //*** nsPopupWindowManager::nsIPopupWindowManager
  64 //*****************************************************************************
  65
  66 NS_IMETHODIMP
  67 nsPopupWindowManager::TestPermission(nsIPrincipal* aPrincipal,
  68                                      uint32_t *aPermission)
  69 {
  70   NS_ENSURE_ARG_POINTER(aPrincipal);
  71   NS_ENSURE_ARG_POINTER(aPermission);
  72
  73   uint32_t permit;
  74   *aPermission = mPolicy;
  75
  76   if (mPermissionManager) {
  77     if (NS_SUCCEEDED(mPermissionManager->TestPermissionFromPrincipal(aPrincipal, "popup", &permit))) {
  78       // Share some constants between interfaces?
  79       if (permit == nsIPermissionManager::ALLOW_ACTION) {
  80         *aPermission = ALLOW_POPUP;
  81       } else if (permit == nsIPermissionManager::DENY_ACTION) {
  82         *aPermission = DENY_POPUP;
  83       }
  84     }
  85   }
  86
  87   return NS_OK;
  88 }
  89
  90 //*****************************************************************************
  91 //*** nsPopupWindowManager::nsIObserver
  92 //*****************************************************************************
  93 NS_IMETHODIMP
  94 nsPopupWindowManager::Observe(nsISupports *aSubject,
  95                               const char *aTopic,
  96                               const char16_t *aData)
  97 {
  98   nsCOMPtr<nsIPrefBranch> prefBranch = do_QueryInterface(aSubject);
  99   NS_ASSERTION(!nsCRT::strcmp(NS_PREFBRANCH_PREFCHANGE_TOPIC_ID, aTopic),
 100                "unexpected topic - we only deal with pref changes!");
 101
 102   if (prefBranch) {
 103     // refresh our local copy of the "disable popups" pref
 104     bool permission = true;
 105     prefBranch->GetBoolPref(kPopupDisablePref, &permission);
 106
 107     mPolicy = permission ? (uint32_t) DENY_POPUP : (uint32_t) ALLOW_POPUP;
 108   }
 109
 110   return NS_OK;
 111 }