More efficient window.name persistence for tab-scoped permissions.

src/bg/ChildPolicies.js

@@ -102,7 +102,7 @@
         let preamble = info ? `${marker} + ${JSON.stringify(JSON.stringify([info]))} + ${marker} + "," + ` : "";
         await browser.tabs.executeScript(tabId, {
           code: `window.name = ${preamble}window.name.split(${marker} + ",").pop();`,
-          allFrames: false,
+          allFrames: true,
           matchAboutBlank: true,
           runAt: "document_start",
         });

src/bg/RequestGuard.js

@@ -317,6 +317,10 @@ var RequestGuard = (() => {
             await ChildPolicies.update(policy);
           }
           capabilities = perms.capabilities;
+        } else {
+          if (!isMainFrame && type === "sub_frame") {
+            await Settings.enforceTabRestrictions(tabId);
+          }
         }
         if (isMainFrame && !TabStatus.map.has(tabId)) {
           debug("No TabStatus data yet for noscriptFrame", tabId);

src/bg/Settings.js

@@ -90,7 +90,7 @@ var Settings = {
 
     if (typeof unrestrictedTab === "boolean") {
       ns.unrestrictedTabs[unrestrictedTab ? "add" : "delete"](tabId);
-      ChildPolicies.storeTabInfo(tabId, unrestrictedTab && {unrestricted: true});
+      this.enforceTabRestrictions(tabId, unrestrictedTab);
     }
     if (reloadAffected) {
       browser.tabs.reload(tabId);
@@ -123,4 +123,8 @@ var Settings = {
     }, null, 2);
   },
 
+  async enforceTabRestrictions(tabId, unrestricted = ns.unrestrictedTabs.has(tabId)) {
+    await ChildPolicies.storeTabInfo(tabId, unrestricted && {unrestricted: true});
+    return unrestricted;
+  }
 }

src/content/staticNS.js

@@ -38,33 +38,40 @@
       if(!this.config.CURRENT) this.config.CURRENT = DEFAULT;
 
       // ugly hack: since now we use registerContentScript instead of the
-      // filterRequest dynamic script injection hack, we use top.name
+      // filterRequest dynamic script injection hack, we use window.name
       // to store per-tab information. We don't want web content to
       // mess with it, though, so we wrap it around auto-hiding accessors
-      this.config.MARKER = MARKER;
+
-      let eraseTabInfoRx = new RegExp(`[^]*${MARKER},?`);
+      if (this.config.MARKER = MARKER) {
-      if (eraseTabInfoRx.test(top.name)) {
+
-        let _name = top.name;
         let tabInfoRx = new RegExp(`^${MARKER}\\[([^]*?)\\]${MARKER},`);
-        if (top === window) { // wrap to hide
+        let name = window.name;
-          Reflect.defineProperty(top.wrappedJSObject, "name", {
+        try {
-            get: exportFunction(() => top.name.replace(eraseTabInfoRx, ""), top.wrappedJSObject),
+          name = top.name;
-            set: exportFunction(value => {
+        } catch(e) {
-              let preamble = top.name.match(tabInfoRx);
+          // won't work cross-origin
-              top.name = `${preamble && preamble[0] || ""}${value}`;
-              return value;
-            }, top.wrappedJSObject)
-          });
         }
-        let tabInfoMatch = _name.match(tabInfoRx);
+        let tabInfoMatch = name.match(tabInfoRx);
-        if (tabInfoMatch) try {
+        if (tabInfoMatch) {
+          try {
             this.config.tabInfo = JSON.parse(tabInfoMatch[1]);
           } catch (e) {
             error(e);
           }
         }
-      
+        let splitter = `${MARKER},`;
+        this.getWindowName = () => window.name.split(splitter).pop();
+        Reflect.defineProperty(window.wrappedJSObject, "name", {
+          get: exportFunction(() => this.getWindowName(), window.wrappedJSObject),
+          set: exportFunction(value => {
+            let tabInfoMatch = window.name.match(tabInfoRx);
+            window.name = tabInfoMatch ? `${tabInfoMatch[0]}${value}` : value;
+            return value;
+          }, window.wrappedJSObject)
+        });
+      }
       if (!this.config.DEFAULT || this.config.tabInfo.unrestricted) {
+        debug("%s is loading unrestricted by user's choice (%o).", document.URL, this.config);
         this.allows = () => true;
         this.capabilities =  Object.assign(
           new Set(["script"]), { has() { return true; } });
@@ -84,10 +91,7 @@
     },
 
     getWindowName() {
-      let marker = this.config.MARKER;
+      return window.name;
-      return (top === window && marker) ? 
-          window.name.split(`${marker},`).pop()
-          : window.name;
     }
   };