From 746189188dad6426ee4cd5e4dfe69429d8cee459 Mon Sep 17 00:00:00 2001
From: Corey Arthur <1339555+CoreyD97@users.noreply.github.com>
Date: Thu, 25 May 2023 07:32:36 +0100
Subject: [PATCH] Update features.js

Add container header for custom-named containers
---
 firefox/src/features.js | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/firefox/src/features.js b/firefox/src/features.js
index 9a9bdae..3bf2e26 100644
--- a/firefox/src/features.js
+++ b/firefox/src/features.js
@@ -110,6 +110,9 @@ async function colorHeaderHandler(e) {
         const name = "X-PwnFox-Color"
         const value = colorMap[identity.color]
         e.requestHeaders.push({ name, value })
+    }else{
+        //If it isn't a pwnfox container, add the container's name as a header instead.
+        e.requestHeaders.push({"name": "X-FF-Container", "value": identity.name})
     }
     return { requestHeaders: e.requestHeaders }
 }
@@ -295,4 +298,4 @@ class ContentScriptFeatures extends FeaturesGroup {
         ]
         super(config, features)
     }
-}
\ No newline at end of file
+}