Adding customization for firefox

sources/firefox/requirements.txt

@@ -0,0 +1 @@
+R2Log

sources/firefox/setup.py

@@ -0,0 +1,212 @@
+#!/usr/bin/env python3
+# -- coding: utf-8 --
+# File name          : setup.py
+# Author             : Wlayzz (@wlayzz - Lucien Doustaly) and Skilo (@askilow - Alexis Marquois)
+# Date created       : 27 February 2023
+# Python Version     : 3.*
+
+import glob
+import json
+import os
+import re
+import shutil
+import subprocess
+import time
+import zipfile
+import sqlite3
+from pathlib import Path
+
+import requests
+from R2Log import logger
+
+PATHNAME = "/root/.mozilla/firefox/**.Exegol/"
+
+# pip3 install R2Log
+# Define addons urls
+urls = [
+    "https://addons.mozilla.org/fr/firefox/addon/foxyproxy-standard/",
+    "https://addons.mozilla.org/fr/firefox/addon/darkreader/",
+    "https://addons.mozilla.org/fr/firefox/addon/uaswitcher/",
+    "https://addons.mozilla.org/fr/firefox/addon/cookie-editor/",
+    "https://addons.mozilla.org/fr/firefox/addon/wappalyzer/"
+]
+
+# Define regex
+reurl = r"(https:\/\/addons\.mozilla\.org\/firefox\/downloads\/file\/[0-9]+\/)([a-zA-Z0-9\-\_\.]+\.xpi)"
+reid = r'"id": "([^"]+)"'
+
+def get_link(url):
+    logger.info(f"Extracting download link from {url}")
+    response = requests.get(url)
+    # Extract download link and addon name from the response text using regex
+    dlextract = re.search(reurl, response.text)
+    # Concat link (group 1) and addon name (group 2)
+    link = ''.join(dlextract.groups())
+    # Extract xpi filename
+    addon_name = dlextract.group(2)
+    return link, addon_name
+
+
+def download_addon(link, addon_name):
+    logger.info(f"Downloading addon {addon_name}")
+    addon_dl = requests.get(link)
+    # Save xpi addon on filesystem
+    with open(addon_name, 'wb') as addon_file:
+        addon_file.write(addon_dl.content)
+
+
+def read_manifest(addon_name):
+    archive = zipfile.ZipFile(addon_name, 'r')
+    manifest = archive.read('manifest.json').decode()
+    # Read the id in the manifest
+    addon_id = re.search(reid, manifest).group(1)
+    return addon_id
+
+
+def install_addons(addon_name, addon_id):
+    logger.info(f"Installing addon {addon_name} with id {addon_id}")
+    # Get the path of the Exegol profile
+    try:
+        dest = glob.glob("%s" % PATHNAME)[0]
+    except:
+        logger.error("Firefox profile Exegol does not exist")
+        raise
+    # Create the extensions folder
+    Path(dest + "/extensions").mkdir(parents=True, exist_ok=True)
+    # Move the addon to the extensions folder
+    shutil.move(addon_name, dest + "/extensions/" + addon_id + ".xpi")
+
+
+def activate_addons(addon_list):
+    for addons in addon_list:
+        addon_id, addon_name, disable = addons
+        if disable:
+            logger.info(f"Disabling {addon_name}")
+        else:
+            logger.info(f"Enabling {addon_name}")
+        try:
+            with open(Path(glob.glob("%s" % PATHNAME)[0] + "/extensions.json"), 'r+') as extensions_file:
+                extensions_config = json.load(extensions_file)
+                for addon in extensions_config["addons"]:
+                    if addon["id"] == addon_id:
+                        addon["active"] = not disable
+                        addon["userDisabled"] = disable
+                        addon["seen"] = not disable
+                extensions_file.seek(0)  # <--- should reset file position to the beginning.
+                json.dump(extensions_config, extensions_file)
+                extensions_file.truncate()  # remove remaining part
+            if disable:
+                logger.success(f"{addon_name} sucessfully disabled\n")
+            else:
+                logger.success(f"{addon_name} sucessfully enabled\n")
+        except:
+            if disable:
+                logger.error(f"Could not disable {addon_name}\n")
+            else:
+                logger.error(f"Could not enable {addon_name}\n")
+            pass
+
+def adjust_ui():
+    with open(Path(glob.glob("%s" % PATHNAME)[0] + "/prefs.js"), 'r+') as pref_js:
+        new_pref = re.sub(r'\\"import-button\\",', '', pref_js.read())
+        new_pref = re.sub(r'\\"save-to-pocket-button\\",', '', new_pref)
+        new_pref = re.sub('"extensions.activeThemeID", "default-theme@mozilla.org"', '"extensions.activeThemeID", "firefox-compact-dark@mozilla.org"', new_pref)
+        pref_js.seek(0)
+        pref_js.write(new_pref)
+        pref_js.truncate()
+
+def import_bookmarks():
+    dirname = os.path.dirname(__file__)
+    filename = os.path.join(dirname, './places.sqlite')
+    src = sqlite3.connect(filename)
+    dst = sqlite3.connect(glob.glob("%s" % PATHNAME)[0] + "places.sqlite")
+    with dst:
+        src.backup(dst)
+    dst.close()
+    src.close()
+
+if __name__ == "__main__":
+
+    # Create firefox profile Exegol
+    logger.info("Creating Firefox profile")
+    try:
+        subprocess.run(["firefox", "-CreateProfile", "Exegol", "-headless"], stderr=subprocess.DEVNULL, stdout=subprocess.DEVNULL)
+        assert(Path(glob.glob("%s" % PATHNAME)[0]).is_dir())
+        logger.success("Firefox profile Exegol created\n")
+    except:
+        logger.error("Could not create Firefox profile Exegol")
+        raise
+
+    # Define a list containing all addons names and ids
+    addon_list = []
+
+    # Iterate through addons
+    for url in urls:
+        # Make a request to the URL
+        link, addon_name = get_link(url)
+        # Download the addon
+        download_addon(link, addon_name)
+        # Read manifest.json in the archive
+        addon_id = read_manifest(addon_name)
+        install_addons(addon_name, addon_id)
+        logger.success(f"{addon_name} installed sucessfully\n")
+        addon_list.append((addon_id, addon_name[0:-4], False))
+
+    logger.success("All addons were installed sucessfully\n")
+
+    # Run firefox to initialise profile
+    logger.info("Initialising Firefox profile")
+    try:
+        p_firefox = subprocess.Popen(["firefox", "-P", "Exegol", "-headless"], stderr=subprocess.DEVNULL, stdout=subprocess.DEVNULL)
+        time.sleep(5)
+        p_firefox.kill()
+        assert(Path(glob.glob("%s" % PATHNAME)[0] + "/extensions.json").is_file())
+        logger.success("Firefox profile initialised sucessfully\n")
+    except:
+        logger.error("Could not initialise Firefox profile")
+        raise
+
+    # Enable dark mode
+    addon_list.append(("firefox-compact-dark@mozilla.org", "Dark mode", False))
+    # Disable default theme
+    addon_list.append(("default-theme@mozilla.org", "Default theme", True))
+    # Activate all addons
+    activate_addons(addon_list)
+
+    # Update UI
+    logger.info("Updating user interface")
+    try:
+        adjust_ui()
+        # Remove existing sessions
+        shutil.rmtree(glob.glob("%s" % PATHNAME)[0] + "sessionstore-backups")
+        logger.success("User interface successfully updated\n")
+    except:
+        logger.error("An error has occurred while trying to update user interface\n")
+
+    # Restore bookmarks
+    logger.info("Setting up profile's bookmarks")
+    try:
+        import_bookmarks()
+        logger.success("Bookmarks successfully setup\n")
+    except:
+        logger.error("Could not setup profile's bookmarks")
+        raise
+
+    # Remove backup file interfering with addons activation
+    logger.info("Removing backup file interfering with addons activation")
+    try:
+        Path(glob.glob("%s" % PATHNAME)[0] + "/addonStartup.json.lz4").unlink()
+        logger.success("Backup file successfully removed\n")
+    except:
+        logger.error("Could not remove the backup file")
+        raise
+
+    # Restart firefox to apply modifications
+    logger.info("Restarting firefox to apply modifications")
+    try:
+        p_firefox = subprocess.Popen(["firefox", "-headless"], stderr=subprocess.DEVNULL, stdout=subprocess.DEVNULL)
+        time.sleep(5)
+        p_firefox.kill()
+        logger.success("Modifications successfully applied")
+    except:
+        logger.error("Could not restart firefox")

sources/install.sh

@@ -3249,6 +3249,15 @@ function install_ctf-party() {
   add-test-command "ctf-party --help"
 }
 
+function install_firefox() {
+  colorecho "Installing firefox"
+  mkdir /opt/tools/firefox
+  mv /root/sources/firefox/* /opt/tools/firefox/
+  python3 -m pip install -r /opt/tools/firefox/requirements.txt
+  python3 /opt/tools/firefox/setup.py
+  add-test-command "firefox --version"
+}
+
 # Package dedicated to the basic things the env needs
 function package_base() {
   update || exit
@@ -3359,6 +3368,7 @@ function package_base() {
   install_logrotate
   fapt openjdk-17-jre
   install_chromium
+  install_firefox
 }
 
 # Package dedicated to offensive miscellaneous tools