1 files changed, 144 insertions, 0 deletions
diff --git a/toolkit/components/aboutthirdparty/tests/browser/head.js b/toolkit/components/aboutthirdparty/tests/browser/head.js
new file mode 100644
index 0000000000..eba3948046
--- /dev/null
+++ b/toolkit/components/aboutthirdparty/tests/browser/head.js
@@ -0,0 +1,144 @@
+/* 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/. */
+
+"use strict";
+
+const kClsidTestShellEx = "{10a9521e-0205-4cc7-93a1-62f30a9a54b3}";
+const kFriendlyName = "Minimum Shell Extension for Firefox testing";
+const kExtensionSubkeys = [".zzz\\shellex\\IconHandler"];
+const kExtensionModuleName = "TestShellEx.dll";
+const kUserBlockedModuleName = "TestDllBlocklist_UserBlocked.dll";
+const kFileFilterInDialog = "*.zzz";
+const kATP = Cc["@mozilla.org/about-thirdparty;1"].getService(
+  Ci.nsIAboutThirdParty
+);
+
+function loadShellExtension() {
+  // This method call opens the file dialog and shows the support file
+  // "hello.zzz" in it, which loads TestShellEx.dll to show an icon
+  // for files with the .zzz extension.
+  kATP.openAndCloseFileDialogForTesting(
+    kExtensionModuleName,
+    getTestFilePath(""),
+    kFileFilterInDialog
+  );
+}
+
+async function registerObject() {
+  const reg = Cc["@mozilla.org/windows-registry-key;1"].createInstance(
+    Ci.nsIWindowsRegKey
+  );
+
+  reg.create(
+    Ci.nsIWindowsRegKey.ROOT_KEY_CURRENT_USER,
+    "Software\\Classes\\CLSID\\" + kClsidTestShellEx,
+    Ci.nsIWindowsRegKey.ACCESS_ALL
+  );
+
+  reg.writeStringValue("", kFriendlyName);
+
+  const inprocServer = reg.createChild(
+    "InprocServer32",
+    Ci.nsIWindowsRegKey.ACCESS_ALL
+  );
+
+  const moduleFullPath = getTestFilePath(kExtensionModuleName);
+  Assert.ok(await IOUtils.exists(moduleFullPath), "The module file exists.");
+
+  inprocServer.writeStringValue("", moduleFullPath);
+  inprocServer.writeStringValue("ThreadingModel", "Apartment");
+  reg.close();
+
+  info("registerObject() done - " + moduleFullPath);
+}
+
+function registerExtensions() {
+  for (const subkey of kExtensionSubkeys) {
+    const reg = Cc["@mozilla.org/windows-registry-key;1"].createInstance(
+      Ci.nsIWindowsRegKey
+    );
+    reg.create(
+      Ci.nsIWindowsRegKey.ROOT_KEY_CURRENT_USER,
+      "Software\\Classes\\" + subkey,
+      Ci.nsIWindowsRegKey.ACCESS_ALL
+    );
+
+    let currentExtension = "";
+    try {
+      // If the key was just created above, the default value does not exist,
+      // so readStringValue will throw NS_ERROR_FAILURE.
+      currentExtension = reg.readStringValue("");
+    } catch (e) {}
+
+    try {
+      if (!currentExtension) {
+        reg.writeStringValue("", kClsidTestShellEx);
+      } else if (currentExtension != kClsidTestShellEx) {
+        throw new Error(
+          `Another extension \`${currentExtension}\` has been registered.`
+        );
+      }
+    } catch (e) {
+      throw new Error("Failed to register TestShellEx.dll: " + e);
+    } finally {
+      reg.close();
+    }
+  }
+}
+
+function unregisterAll() {
+  for (const subkey of kExtensionSubkeys) {
+    const reg = Cc["@mozilla.org/windows-registry-key;1"].createInstance(
+      Ci.nsIWindowsRegKey
+    );
+
+    try {
+      reg.open(
+        Ci.nsIWindowsRegKey.ROOT_KEY_CURRENT_USER,
+        "Software\\Classes\\" + subkey,
+        Ci.nsIWindowsRegKey.ACCESS_ALL
+      );
+
+      if (reg.readStringValue("") != kClsidTestShellEx) {
+        // If another extension is registered, don't overwrite it.
+        continue;
+      }
+
+      // Set an empty string instead of deleting the key
+      // not to touch non-default values.
+      reg.writeStringValue("", "");
+    } catch (e) {
+      info(`Failed to unregister \`${subkey}\`: ` + e);
+    } finally {
+      reg.close();
+    }
+  }
+
+  const reg = Cc["@mozilla.org/windows-registry-key;1"].createInstance(
+    Ci.nsIWindowsRegKey
+  );
+  reg.open(
+    Ci.nsIWindowsRegKey.ROOT_KEY_CURRENT_USER,
+    "Software\\Classes\\CLSID",
+    Ci.nsIWindowsRegKey.ACCESS_ALL
+  );
+
+  try {
+    const child = reg.openChild(
+      kClsidTestShellEx,
+      Ci.nsIWindowsRegKey.ACCESS_ALL
+    );
+    try {
+      child.removeChild("InprocServer32");
+    } catch (e) {
+    } finally {
+      child.close();
+    }
+
+    reg.removeChild(kClsidTestShellEx);
+  } catch (e) {
+  } finally {
+    reg.close();
+  }
+}