extensions/cookie/nsPopupWindowManager.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* ***** BEGIN LICENSE BLOCK *****
   3  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   4  *
   5  * The contents of this file are subject to the Mozilla Public License Version
   6  * 1.1 (the "License"); you may not use this file except in compliance with
   7  * the License. You may obtain a copy of the License at
   8  * http://www.mozilla.org/MPL/
   9  *
  10  * Software distributed under the License is distributed on an "AS IS" basis,
  11  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  12  * for the specific language governing rights and limitations under the
  13  * License.
  14  *
  15  * The Original Code is mozilla.org code.
  16  *
  17  * The Initial Developer of the Original Code is
  18  * Netscape Communications Corporation.
  19  * Portions created by the Initial Developer are Copyright (C) 2002
  20  * the Initial Developer. All Rights Reserved.
  21  *
  22  * Contributor(s):
  23  *
  24  * Alternatively, the contents of this file may be used under the terms of
  25  * either the GNU General Public License Version 2 or later (the "GPL"), or
  26  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  27  * in which case the provisions of the GPL or the LGPL are applicable instead
  28  * of those above. If you wish to allow use of your version of this file only
  29  * under the terms of either the GPL or the LGPL, and not to allow others to
  30  * use your version of this file under the terms of the MPL, indicate your
  31  * decision by deleting the provisions above and replace them with the notice
  32  * and other provisions required by the GPL or the LGPL. If you do not delete
  33  * the provisions above, a recipient may use your version of this file under
  34  * the terms of any one of the MPL, the GPL or the LGPL.
  35  *
  36  * ***** END LICENSE BLOCK ***** */
  37
  38 #include "nsPopupWindowManager.h"
  39
  40 #include "nsCRT.h"
  41 #include "nsIServiceManager.h"
  42 #include "nsIPrefService.h"
  43 #include "nsIPrefBranch.h"
  44 #include "nsIPrefBranch2.h"
  45
  46 /**
  47  * The Popup Window Manager maintains popup window permissions by website.
  48  */
  49
  50 static const char kPopupDisablePref[] = "dom.disable_open_during_load";
  51
  52 //*****************************************************************************
  53 //*** nsPopupWindowManager object management and nsISupports
  54 //*****************************************************************************
  55
  56 nsPopupWindowManager::nsPopupWindowManager() :
  57   mPolicy(ALLOW_POPUP)
  58 {
  59 }
  60
  61 nsPopupWindowManager::~nsPopupWindowManager()
  62 {
  63 }
  64
  65 NS_IMPL_ISUPPORTS3(nsPopupWindowManager,
  66                    nsIPopupWindowManager,
  67                    nsIObserver,
  68                    nsSupportsWeakReference)
  69
  70 nsresult
  71 nsPopupWindowManager::Init()
  72 {
  73   nsresult rv;
  74   mPermissionManager = do_GetService(NS_PERMISSIONMANAGER_CONTRACTID);
  75
  76   nsCOMPtr<nsIPrefBranch2> prefBranch =
  77     do_GetService(NS_PREFSERVICE_CONTRACTID, &rv);
  78   if (NS_SUCCEEDED(rv)) {
  79     PRBool permission;
  80     rv = prefBranch->GetBoolPref(kPopupDisablePref, &permission);
  81     if (NS_FAILED(rv)) {
  82       permission = PR_TRUE;
  83     }
  84     mPolicy = permission ? (PRUint32) DENY_POPUP : (PRUint32) ALLOW_POPUP;
  85
  86     prefBranch->AddObserver(kPopupDisablePref, this, PR_TRUE);
  87   }
  88
  89   return NS_OK;
  90 }
  91
  92 //*****************************************************************************
  93 //*** nsPopupWindowManager::nsIPopupWindowManager
  94 //*****************************************************************************
  95
  96 NS_IMETHODIMP
  97 nsPopupWindowManager::TestPermission(nsIURI *aURI, PRUint32 *aPermission)
  98 {
  99   NS_ENSURE_ARG_POINTER(aURI);
 100   NS_ENSURE_ARG_POINTER(aPermission);
 101
 102   nsresult rv;
 103   PRUint32 permit;
 104
 105   *aPermission = mPolicy;
 106
 107   if (mPermissionManager) {
 108     rv = mPermissionManager->TestPermission(aURI, "popup", &permit);
 109
 110     if (NS_SUCCEEDED(rv)) {
 111       // Share some constants between interfaces?
 112       if (permit == nsIPermissionManager::ALLOW_ACTION) {
 113         *aPermission = ALLOW_POPUP;
 114       } else if (permit == nsIPermissionManager::DENY_ACTION) {
 115         *aPermission = DENY_POPUP;
 116       }
 117     }
 118   }
 119
 120   return NS_OK;
 121 }
 122
 123 //*****************************************************************************
 124 //*** nsPopupWindowManager::nsIObserver
 125 //*****************************************************************************
 126 NS_IMETHODIMP
 127 nsPopupWindowManager::Observe(nsISupports *aSubject,
 128                               const char *aTopic,
 129                               const PRUnichar *aData)
 130 {
 131   nsCOMPtr<nsIPrefBranch> prefBranch = do_QueryInterface(aSubject);
 132   NS_ASSERTION(!nsCRT::strcmp(NS_PREFBRANCH_PREFCHANGE_TOPIC_ID, aTopic),
 133                "unexpected topic - we only deal with pref changes!");
 134
 135   if (prefBranch) {
 136     // refresh our local copy of the "disable popups" pref
 137     PRBool permission = PR_TRUE;
 138     prefBranch->GetBoolPref(kPopupDisablePref, &permission);
 139
 140     mPolicy = permission ? (PRUint32) DENY_POPUP : (PRUint32) ALLOW_POPUP;
 141   }
 142
 143   return NS_OK;
 144 }