mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-08 20:47:44 +00:00
214 lines
6.6 KiB
C++
214 lines
6.6 KiB
C++
/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* ***** BEGIN LICENSE BLOCK *****
|
|
* Version: NPL 1.1/GPL 2.0/LGPL 2.1
|
|
*
|
|
* The contents of this file are subject to the Netscape Public License
|
|
* Version 1.1 (the "License"); you may not use this file except in
|
|
* compliance with the License. You may obtain a copy of the License at
|
|
* http://www.mozilla.org/NPL/
|
|
*
|
|
* Software distributed under the License is distributed on an "AS IS" basis,
|
|
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
|
* for the specific language governing rights and limitations under the
|
|
* License.
|
|
*
|
|
* The Original Code is mozilla.org code.
|
|
*
|
|
* The Initial Developer of the Original Code is
|
|
* Netscape Communications Corporation.
|
|
* Portions created by the Initial Developer are Copyright (C) 1998
|
|
* the Initial Developer. All Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
*
|
|
*
|
|
* Alternatively, the contents of this file may be used under the terms of
|
|
* either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
* in which case the provisions of the GPL or the LGPL are applicable instead
|
|
* of those above. If you wish to allow use of your version of this file only
|
|
* under the terms of either the GPL or the LGPL, and not to allow others to
|
|
* use your version of this file under the terms of the NPL, indicate your
|
|
* decision by deleting the provisions above and replace them with the notice
|
|
* and other provisions required by the GPL or the LGPL. If you do not delete
|
|
* the provisions above, a recipient may use your version of this file under
|
|
* the terms of any one of the NPL, the GPL or the LGPL.
|
|
*
|
|
* ***** END LICENSE BLOCK ***** */
|
|
|
|
#include "nsPermissions.h"
|
|
#include "nsUtils.h"
|
|
|
|
#include "nsVoidArray.h"
|
|
#include "prmem.h"
|
|
#include "nsIPref.h"
|
|
#include "nsTextFormatter.h"
|
|
#include "nsIServiceManager.h"
|
|
#include "nsIIOService.h"
|
|
|
|
#define image_behaviorPref "network.image.imageBehavior"
|
|
#define image_warningPref "network.image.warnAboutImages"
|
|
|
|
typedef struct _permission_HostStruct {
|
|
char * host;
|
|
nsVoidArray * permissionList;
|
|
} permission_HostStruct;
|
|
|
|
typedef struct _permission_TypeStruct {
|
|
PRInt32 type;
|
|
PRBool permission;
|
|
} permission_TypeStruct;
|
|
|
|
PRIVATE PERMISSION_BehaviorEnum image_behavior = PERMISSION_Accept;
|
|
PRIVATE PRBool image_warning = PR_FALSE;
|
|
|
|
PRIVATE void
|
|
image_SetBehaviorPref(PERMISSION_BehaviorEnum x) {
|
|
image_behavior = x;
|
|
}
|
|
|
|
PRIVATE void
|
|
image_SetWarningPref(PRBool x) {
|
|
image_warning = x;
|
|
}
|
|
|
|
PRIVATE PERMISSION_BehaviorEnum
|
|
image_GetBehaviorPref() {
|
|
return image_behavior;
|
|
}
|
|
|
|
PRIVATE PRBool
|
|
image_GetWarningPref() {
|
|
return image_warning;
|
|
}
|
|
|
|
MODULE_PRIVATE int PR_CALLBACK
|
|
image_BehaviorPrefChanged(const char * newpref, void * data) {
|
|
PRInt32 n;
|
|
nsresult rv;
|
|
nsCOMPtr<nsIPref> prefs(do_GetService(NS_PREF_CONTRACTID, &rv));
|
|
if (NS_FAILED(prefs->GetIntPref(image_behaviorPref, &n))) {
|
|
image_SetBehaviorPref(PERMISSION_Accept);
|
|
} else {
|
|
image_SetBehaviorPref((PERMISSION_BehaviorEnum)n);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
MODULE_PRIVATE int PR_CALLBACK
|
|
image_WarningPrefChanged(const char * newpref, void * data) {
|
|
PRBool x;
|
|
nsresult rv;
|
|
nsCOMPtr<nsIPref> prefs(do_GetService(NS_PREF_CONTRACTID, &rv));
|
|
if (NS_FAILED(prefs->GetBoolPref(image_warningPref, &x))) {
|
|
x = PR_FALSE;
|
|
}
|
|
image_SetWarningPref(x);
|
|
return 0;
|
|
}
|
|
|
|
PUBLIC void
|
|
IMAGE_RegisterPrefCallbacks(void) {
|
|
PRInt32 n;
|
|
PRBool x;
|
|
nsresult rv;
|
|
nsCOMPtr<nsIPref> prefs(do_GetService(NS_PREF_CONTRACTID, &rv));
|
|
|
|
// Initialize for image_behaviorPref
|
|
if (NS_FAILED(prefs->GetIntPref(image_behaviorPref, &n))) {
|
|
n = PERMISSION_Accept;
|
|
}
|
|
image_SetBehaviorPref((PERMISSION_BehaviorEnum)n);
|
|
prefs->RegisterCallback(image_behaviorPref, image_BehaviorPrefChanged, NULL);
|
|
|
|
// Initialize for image_warningPref
|
|
if (NS_FAILED(prefs->GetBoolPref(image_warningPref, &x))) {
|
|
x = PR_FALSE;
|
|
}
|
|
image_SetWarningPref(x);
|
|
prefs->RegisterCallback(image_warningPref, image_WarningPrefChanged, NULL);
|
|
}
|
|
|
|
PUBLIC nsresult
|
|
IMAGE_CheckForPermission
|
|
(const char * hostname, const char * firstHostname, PRBool *permission) {
|
|
|
|
/* exit if imageblocker is not enabled */
|
|
nsresult rv;
|
|
PRBool prefvalue = PR_FALSE;
|
|
nsCOMPtr<nsIPref> prefs(do_GetService(NS_PREF_CONTRACTID, &rv));
|
|
if (NS_FAILED(rv) ||
|
|
NS_FAILED(prefs->GetBoolPref("imageblocker.enabled", &prefvalue)) ||
|
|
!prefvalue) {
|
|
*permission = (image_GetBehaviorPref() != PERMISSION_DontUse);
|
|
return NS_OK;
|
|
}
|
|
|
|
/* try to make a decision based on pref settings */
|
|
if ((image_GetBehaviorPref() == PERMISSION_DontUse)) {
|
|
*permission = PR_FALSE;
|
|
return NS_OK;
|
|
}
|
|
if (image_GetBehaviorPref() == PERMISSION_DontAcceptForeign) {
|
|
/* compare tails of names checking to see if they have a common domain */
|
|
/* we do this by comparing the tails of both names where each tail includes at least one dot */
|
|
PRInt32 dotcount = 0;
|
|
const char * tailHostname = hostname + PL_strlen(hostname) - 1;
|
|
while (tailHostname > hostname) {
|
|
if (*tailHostname == '.') {
|
|
dotcount++;
|
|
}
|
|
if (dotcount == 2) {
|
|
tailHostname++;
|
|
break;
|
|
}
|
|
tailHostname--;
|
|
}
|
|
dotcount = 0;
|
|
const char * tailFirstHostname = firstHostname + PL_strlen(firstHostname) - 1;
|
|
while (tailFirstHostname > firstHostname) {
|
|
if (*tailFirstHostname == '.') {
|
|
dotcount++;
|
|
}
|
|
if (dotcount == 2) {
|
|
tailFirstHostname++;
|
|
break;
|
|
}
|
|
tailFirstHostname--;
|
|
}
|
|
if (PL_strcmp(tailFirstHostname, tailHostname)) {
|
|
*permission = PR_FALSE;
|
|
return NS_OK;
|
|
}
|
|
}
|
|
|
|
/* use common routine to make decision */
|
|
PRUnichar * message = CKutil_Localize(NS_LITERAL_STRING("PermissionToAcceptImage").get());
|
|
PRUnichar * new_string = nsTextFormatter::smprintf(message, hostname ? hostname : "");
|
|
if (NS_SUCCEEDED(PERMISSION_Read())) {
|
|
*permission = Permission_Check(0, hostname, IMAGEPERMISSION,
|
|
image_GetWarningPref(), new_string);
|
|
} else {
|
|
*permission = PR_TRUE;
|
|
}
|
|
PR_FREEIF(new_string);
|
|
Recycle(message);
|
|
return NS_OK;
|
|
}
|
|
|
|
PUBLIC nsresult
|
|
IMAGE_Block(const char* imageURL,
|
|
nsIIOService* ioService) {
|
|
if (!imageURL || !(*imageURL)) {
|
|
return NS_ERROR_NULL_POINTER;
|
|
}
|
|
nsresult rv = NS_OK;
|
|
nsCAutoString host;
|
|
NS_ASSERTION(ioService, "IOService not available");
|
|
rv = ioService->ExtractUrlPart(nsDependentCString(imageURL),
|
|
nsIIOService::url_Host |
|
|
nsIIOService::url_Port, host);
|
|
Permission_AddHost((char*)host.get(), PR_FALSE, IMAGEPERMISSION, PR_TRUE);
|
|
return NS_OK;
|
|
}
|