Issue 7236 - Handle sub properties in abort-on-property snippets

lib/content/snippets.js

Index: lib/content/snippets.js
===================================================================
--- a/lib/content/snippets.js
+++ b/lib/content/snippets.js
@@ -693,23 +693,53 @@
   // 2176782336 is 36^6 which mean 6 chars [a-z0-9]
   // 60466176 is 36^5
   // 2176782336 - 60466176 = 2116316160. This ensure to always have 6
   // chars even if Math.random() returns its minimum value 0.0
   //
   return Math.floor(Math.random() * 2116316160 + 60466176).toString(36);
 }
 
-function wrapPropertyAccess(object, property, descriptor)
+function wrapPropertyAccess(object, property, descriptor, magic)
 {
-  let currentDescriptor = Object.getOwnPropertyDescriptor(object, property);
-  if (currentDescriptor && !currentDescriptor.configurable)
-    return false;
+  let dot = property.indexOf(".");
+  if (dot == -1)
+  {
+    // simple property case.
+    let currentDescriptor = Object.getOwnPropertyDescriptor(object, property);
+    if (currentDescriptor && !currentDescriptor.configurable)
+      return false;
+
+    Object.defineProperty(object, property, descriptor);
+    return true;
+  }
+  let name = property.slice(0, dot);
+  property = property.slice(dot + 1);
+  let value = object[name];
+  if (value && typeof value == "object")
+    return wrapPropertyAccess(value, property, descriptor, magic);
 
-  Object.defineProperty(object, property, descriptor);
+  let currentDescriptor = Object.getOwnPropertyDescriptor(object, name);
+  if (currentDescriptor)
+  {
+    if (currentDescriptor.set && currentDescriptor.set.hasOwnProperty(magic))
+      return true;
+    if (!currentDescriptor.configurable)
+      return false;
+  }
+
+  let v;
+  let setter = a =>
+  {
+    v = a;
+    if (a && typeof a == "object")
+      wrapPropertyAccess(a, property, descriptor, magic);
+  };
+  setter[magic] = undefined;
+  Object.defineProperty(object, name, {get: () => v, set: setter});
   return true;
 }
 
 /**
  * Overrides the <code>onerror</code> handler to discard tagged error messages
  * from our property wrapping.
  *
  * @param {string} magic The magic string that tags the error message.
@@ -744,17 +774,17 @@
 
   let rid = randomId();
 
   function abort()
   {
     throw new ReferenceError(rid);
   }
 
-  if (wrapPropertyAccess(window, property, {get: abort, set() {}}))
+  if (wrapPropertyAccess(window, property, {get: abort, set() {}}, rid))
     overrideOnError(rid);
 }
 
 exports["abort-on-property-read"] = makeInjector(abortOnPropertyRead,
                                                  wrapPropertyAccess,
                                                  overrideOnError,
                                                  randomId);
 
@@ -776,17 +806,17 @@
 
   let rid = randomId();
 
   function abort()
   {
     throw new ReferenceError(rid);
   }
 
-  if (wrapPropertyAccess(window, property, {set: abort}))
+  if (wrapPropertyAccess(window, property, {set: abort}, rid))
     overrideOnError(rid);
 }
 
 exports["abort-on-property-write"] = makeInjector(abortOnPropertyWrite,
                                                   wrapPropertyAccess,
                                                   overrideOnError,
                                                   randomId);
 
@@ -839,15 +869,15 @@
     },
     set(value)
     {
       abort();
       currentValue = value;
     }
   };
 
-  if (wrapPropertyAccess(object, name, descriptor))
+  if (wrapPropertyAccess(object, name, descriptor, rid))
     overrideOnError(rid);
 }
 
 exports["abort-current-inline-script"] =
   makeInjector(abortCurrentInlineScript, wrapPropertyAccess, toRegExp,
                overrideOnError, regexEscape, randomId);