1 files changed, 80 insertions, 0 deletions
diff --git a/toolkit/components/antitracking/URLDecorationStripper.cpp b/toolkit/components/antitracking/URLDecorationStripper.cpp
new file mode 100644
index 0000000000..38af391945
--- /dev/null
+++ b/toolkit/components/antitracking/URLDecorationStripper.cpp
@@ -0,0 +1,80 @@
+/* -*- 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,
+ * You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#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) {
+  nsAutoString tokenList;
+  nsresult rv = Preferences::GetString(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 nsAString& token : tokenList.Split(' ')) {
+      if (token.IsEmpty()) {
+        continue;
+      }
+
+      nsAutoString 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