Undo interception in top windows

Fixes #431

lib/extension.js

@@ -90,5 +90,46 @@
 		}
 	};
 	
+	const changedPropertiesByWindow = new WeakMap();
+	scope.changeProperty = function(window, group, {object, name, type, changed}){
+		let changedProperties = changedPropertiesByWindow.get(scope.getWrapped(window));
+		if (!changedProperties){
+			changedProperties = [];
+			changedPropertiesByWindow.set(scope.getWrapped(window), changedProperties);
+		}
+		const descriptor = Object.getOwnPropertyDescriptor(object, name);
+		const original = descriptor[type];
+		descriptor[type] = changed;
+		Object.defineProperty(object, name, descriptor);
+		changedProperties.push({group, object, name, type, original});
+	};
+	scope.revertProperties = function(window, group){
+		window = scope.getWrapped(window);
+		let changedProperties = changedPropertiesByWindow.get(window);
+		if (!changedProperties){
+			return;
+		}
+		if (group){
+			const remainingProperties = changedProperties.filter(function({group}){
+				return group !== group;
+			});
+			changedPropertiesByWindow.set(window, remainingProperties);
+			changedProperties = changedProperties.filter(function({group}){
+				return group === group;
+			});
+		}
+		else {
+			changedPropertiesByWindow.delete(window);
+		}
+		
+		for (let i = changedProperties.length - 1; i >= 0; i -= 1){
+			const {object, name, type, original} = changedProperties[i];
+			logging.verbose("reverting", name, "on", object);
+			const descriptor = Object.getOwnPropertyDescriptor(object, name);
+			descriptor[type] = original;
+			Object.defineProperty(object, name, descriptor);
+		}
+	};
+	
 	Object.seal(scope);
 }());

lib/frame.js

@@ -125,7 +125,6 @@
 		return settings.get(...args);
 	}
 	
-	
 	const interceptedWindows = new WeakMap();
 	function interceptWindow(window){
 		let wrappedTry;
@@ -144,10 +143,16 @@
 		if (!enabled || interceptedWindows.get(wrappedWindow)){
 			return false;
 		}
-		if (wrappedWindow.matchMedia(extensionSecret[0]) === extensionSecret[1]){
+		const canvasBlockerData = wrappedWindow.matchMedia(extensionSecret[0]);
+		if (canvasBlockerData.secret === extensionSecret[1]){
+			if (wrappedWindow.top === wrappedWindow){
+				canvasBlockerData.undoIntercept(extension.extensionID);
+			}
+			else {
 				interceptedWindows.set(wrappedWindow, true);
 				return false;
 			}
+		}
 		
 		logging.message("intercepting window", window);
 		intercept(
@@ -168,17 +173,28 @@
 		
 		const matchMediaDescriptor = Object.getOwnPropertyDescriptor(wrappedWindow, "matchMedia");
 		const originalMatchMedia = matchMediaDescriptor.value;
-		matchMediaDescriptor.value = extension.exportFunctionWithName(function matchMedia(query){
+		extension.changeProperty(window, "matchMedia", {
+			object: wrappedWindow,
+			name: "matchMedia",
+			type: "value",
+			changed: extension.exportFunctionWithName(function matchMedia(query){
 				if (query === extensionSecret[0]){
-				return extensionSecret[1];
+					return {
+						secret: extensionSecret[1],
+						undoIntercept: function(token){
+							if (token === extension.extensionID){
+								extension.revertProperties(window);
+							}
+						}
+					};
 				}
 				else {
 					return arguments.length > 1?
 						originalMatchMedia.call(this, ...arguments):
 						originalMatchMedia.call(this, query);
 				}
-		}, window, originalMatchMedia.name);
-		Object.defineProperty(wrappedWindow, "matchMedia", matchMediaDescriptor);
+			}, window, originalMatchMedia.name)
+		});
 		
 		interceptedWindows.set(wrappedWindow, true);
 		return true;

lib/iframeProtection.js

@@ -27,23 +27,12 @@
 			if ((typeof changed) === "function"){
 				changed = extension.exportFunctionWithName(changed, window, original.name);
 			}
-			descriptor[type] = changed;
-			Object.defineProperty(object, name, descriptor);
-			registerChangedProperty(object, name, descriptor, type, original);
-		}
-		const changedProperties = [];
-		// eslint-disable-next-line max-params
-		function registerChangedProperty(object, name, descriptor, type, original){
-			changedProperties.push({object, name, descriptor, type, original});
+			extension.changeProperty(window, "iframeProtection", {object, name, type, changed});
 		}
 		if (settings.isStillDefault){
 			settings.onloaded(function(){
 				if (isWhitelisted(window.location)){
-					changedProperties.forEach(function({object, name, descriptor, type, original}){
-						descriptor[type] = original;
-						Object.defineProperty(object, name, descriptor);
-					});
-					changedProperties.length = 0;
+					extension.revertProperties(window, "iframeProtection");
 				}
 			});
 		}

lib/intercept.js

@@ -331,21 +331,22 @@
 				});
 				const descriptor = Object.getOwnPropertyDescriptor(object, name);
 				if (!descriptor) return;
-				
-				if (descriptor.hasOwnProperty("value")){
+				const type = descriptor.hasOwnProperty("value")? "value": "get";
+				let changed;
+				if (type ==="value"){
 					if (changedFunction.fakeGenerator){
-						descriptor.value = extension.exportFunctionWithName(
+						changed = extension.exportFunctionWithName(
 							changedFunction.fakeGenerator(checker, original, windowToProcess),
 							windowToProcess,
 							original.name
 						);
 					}
 					else {
-						descriptor.value = null;
+						changed = null;
 					}
 				}
 				else {
-					descriptor.get = extension.exportFunctionWithName(function(){
+					changed = extension.exportFunctionWithName(function(){
 						return extension.exportFunctionWithName(
 							changedFunction.fakeGenerator(checker),
 							windowToProcess,
@@ -353,7 +354,9 @@
 						);
 					}, windowToProcess, descriptor.get.name);
 				}
-				Object.defineProperty(object, name, descriptor);
+				extension.changeProperty(windowToProcess, changedFunction.api, {
+					object, name, type, changed
+				});
 			});
 		});
 	}
@@ -378,7 +381,12 @@
 						window: windowToProcess, prefs, checkStack, ask, notify
 					});
 					const getter = changedGetter.getterGenerator(checker, original, windowToProcess);
-					descriptor.get = extension.exportFunctionWithName(getter, windowToProcess, original.name);
+					extension.changeProperty(windowToProcess, changedGetter.api,
+						{
+							object, name, type: "get",
+							changed: extension.exportFunctionWithName(getter, windowToProcess, original.name)
+						}
+					);
 					
 					if (descriptor.hasOwnProperty("set") && descriptor.set && changedGetter.setterGenerator){
 						const original = descriptor.set;
@@ -387,10 +395,14 @@
 							original,
 							prefs
 						);
-						descriptor.set = extension.exportFunctionWithName(setter, windowToProcess, original.name);
+						extension.changeProperty(windowToProcess, changedGetter.api,
+							{
+								object, name, type: "set",
+								changed: extension.exportFunctionWithName(setter, windowToProcess, original.name)
+							}
+						);
 					}
 					
-					Object.defineProperty(object, name, descriptor);
 				}
 				else if (
 					changedGetter.valueGenerator &&
@@ -405,7 +417,9 @@
 					}
 					switch (functionStatus.mode){
 						case "ask": case "block": case "fake":
-							descriptor.value = changedGetter.valueGenerator({
+							extension.changeProperty(windowToProcess, changedGetter.api, {
+								object, name, type: "value",
+								changed: changedGetter.valueGenerator({
 									mode: functionStatus.mode,
 									original: descriptor.value,
 									notify: function notifyCallback(messageId){
@@ -418,8 +432,8 @@
 											api: changedGetter.api
 										});
 									}
+								})
 							});
-							Object.defineProperty(object, name, descriptor);
 							break;
 					}
 				}

versions/updates.json

@@ -109,6 +109,10 @@
 				{
 					"version": "1.1Alpha20192001",
 					"update_link": "https://canvasblocker.kkapsner.de/versions/canvasblocker_beta-1.1Alpha20192001-an+fx.xpi"
+				},
+				{
+					"version": "1.1Alpha20192301",
+					"update_link": "https://canvasblocker.kkapsner.de/versions/canvasblocker_beta-1.1Alpha20192301-an+fx.xpi"
 				}
 			]
 		}