Source code
Revision control
Copy as Markdown
Other Tools
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
#include "URLDecorationStripper.h"
#include "mozilla/Preferences.h"
#include "nsCharSeparatedTokenizer.h"
#include "nsEffectiveTLDService.h"
#include "nsIURI.h"
#include "nsIURIMutator.h"
#include "nsURLHelper.h"
namespace {
static const char* kPrefName =
"privacy.restrict3rdpartystorage.url_decorations";
} // namespace
namespace mozilla {
nsresult URLDecorationStripper::StripTrackingIdentifiers(nsIURI* aURI,
nsACString& aOutSpec) {
nsAutoCString tokenList;
nsresult rv = Preferences::GetCString(kPrefName, tokenList);
ToLowerCase(tokenList);
nsAutoCString path;
rv = aURI->GetPathQueryRef(path);
NS_ENSURE_SUCCESS(rv, rv);
ToLowerCase(path);
int32_t queryBegins = path.FindChar('?');
// Only positive values are valid since the path must begin with a '/'.
if (queryBegins > 0) {
for (const nsACString& token : tokenList.Split(' ')) {
if (token.IsEmpty()) {
continue;
}
nsAutoCString value;
if (URLParams::Extract(Substring(path, queryBegins + 1), token, value) &&
!value.IsVoid()) {
// Tracking identifier found in the URL!
return StripToRegistrableDomain(aURI, aOutSpec);
}
}
}
return aURI->GetSpec(aOutSpec);
}
nsresult URLDecorationStripper::StripToRegistrableDomain(nsIURI* aURI,
nsACString& aOutSpec) {
NS_MutateURI mutator(aURI);
mutator.SetPathQueryRef(""_ns).SetUserPass(""_ns);
RefPtr<nsEffectiveTLDService> etldService =
nsEffectiveTLDService::GetInstance();
NS_ENSURE_TRUE(etldService, NS_ERROR_FAILURE);
nsAutoCString baseDomain;
nsresult rv = etldService->GetBaseDomain(aURI, 0, baseDomain);
if (NS_SUCCEEDED(rv)) {
mutator.SetHost(baseDomain);
} else {
// If this is an IP address or something like "localhost", ignore the error.
if (rv != NS_ERROR_HOST_IS_IP_ADDRESS &&
rv != NS_ERROR_INSUFFICIENT_DOMAIN_LEVELS) {
return rv;
}
}
nsCOMPtr<nsIURI> uri;
rv = mutator.Finalize(uri);
NS_ENSURE_SUCCESS(rv, rv);
return uri->GetSpec(aOutSpec);
}
} // namespace mozilla