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
nsPopupWindowManager()26 nsPopupWindowManager::nsPopupWindowManager() :
27 mPolicy(ALLOW_POPUP)
28 {
29 }
30
~nsPopupWindowManager()31 nsPopupWindowManager::~nsPopupWindowManager()
32 {
33 }
34
NS_IMPL_ISUPPORTS(nsPopupWindowManager,nsIPopupWindowManager,nsIObserver,nsISupportsWeakReference)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
TestPermission(nsIPrincipal * aPrincipal,uint32_t * aPermission)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
Observe(nsISupports * aSubject,const char * aTopic,const char16_t * aData)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 }
112