Simplify injectPolicyScript and computeChildPolicy

Author: aaronkollasch

src/bg/RequestGuard.js

@@ -644,10 +644,9 @@ var RequestGuard = (() => {
     return ABORT;
   }
 
-  async function injectPolicyScript(details) {
-    let {url, tabId, frameId} = details;
-    let tab = await browser.tabs.get(tabId);
-    let policy = ns.computeChildPolicy({url}, {tab, frameId});
+  function injectPolicyScript(details) {
+    let {url, tabId, frameId, cookieStoreId} = details;
+    let policy = ns.computeChildPolicy({url}, {tab: {id: tabId}, frameId, cookieStoreId});
     policy.navigationURL = url;
     let debugStatement = ns.local.debug ? `
       let mark = Date.now() + ":" + Math.random();

src/bg/main.js

@@ -261,18 +261,19 @@
 
     computeChildPolicy({url, contextUrl}, sender) {
       let {tab, frameId} = sender;
-      let cookieStoreId = (tab)? tab.cookieStoreId : "default";
-      var policy = ns.policy;
-      debug("computing policy", cookieStoreId, ns.contextStore);
-      if (
-          cookieStoreId && 
-          !cookieStoreId.toLowerCase().includes("default") &&  //exclude firefox-default
-          ns.contextStore && 
-          ns.contextStore.policies.hasOwnProperty(cookieStoreId)
-      ) {
-        policy = ns.contextStore.policies[cookieStoreId];
+      let tabId = tab ? tab.id : -1;
+      let topUrl;
+      if (frameId === 0) {
+        topUrl = url;
+      } else if (tab) {
+        if (!tab.url) tab = TabCache.get(tabId);
+        if (tab) topUrl = tab.url;
       }
-      debug("chose policy", policy);
+      if (!topUrl) topUrl = url;
+      if (!contextUrl) contextUrl = topUrl;
+
+      let cookieStoreId = sender.cookieStoreId ? sender.cookieStoreId : tab.cookieStoreId;
+      let policy = ns.getPolicy(cookieStoreId);
       let {isTorBrowser} = ns.local;
       if (!policy) {
         console.log("Policy is null, initializing: %o, sending fallback.", ns.initializing);
@@ -285,17 +286,6 @@
         };
       }
 
-      let tabId = tab ? tab.id : -1;
-      let topUrl;
-      if (frameId === 0) {
-        topUrl = url;
-      } else if (tab) {
-        if (!tab.url) tab = TabCache.get(tabId);
-        if (tab) topUrl = tab.url;
-      }
-      if (!topUrl) topUrl = url;
-      if (!contextUrl) contextUrl = topUrl;
-
       if (Sites.isInternal(url) || !ns.isEnforced(tabId)) {
         policy = null;
       }