Protect document.open too

snow.js

@@ -636,7 +636,7 @@ function natively(win, cb) {
 function natives(win) {
   const {
     EventTarget
-  } = win;
+  } = win; // PR#62
   return natively(win, function (win) {
     const {
       console,
@@ -970,6 +970,7 @@ function hook(win, native) {
 function hookOpen(win) {
   hookMessageEvent(win);
   win.open = hook(win, win.open);
+  win.document.open = hook(win, win.document.open);
 }
 module.exports = hookOpen;
 

src/open.js

@@ -83,6 +83,7 @@ function hook(win, native) {
 function hookOpen(win) {
     hookMessageEvent(win);
     win.open = hook(win, win.open);
+    win.document.open = hook(win, win.document.open);
 }
 
 module.exports = hookOpen;

test/open.js

@@ -93,4 +93,99 @@ describe('window.open API', () => {
         });
         expect(result).toBe('V');
     });
+});
+
+describe('document.open API', () => {
+    beforeEach(setup);
+
+    it('should fail to use atob of a window that was created via document.open API', async () => {
+        const result = await browser.executeAsync(function(done) {
+            const bypass = (wins) => done(wins.map(win => (win && win.atob ? win : top).atob('WA==')).join(','));
+            (function(){
+                const win = document.open('', '', '');
+                bypass([win]);
+            }());
+        });
+        expect(result).toBe('V');
+    });
+
+    it('should fail to use atob of a window that was created via document.open API to cross origin and then changed to same origin', async () => {
+        const result = await browser.executeAsync(function(done) {
+            const bypass = (wins) => done(wins.map(win => (win && win.atob ? win : top).atob('WA==')).join(','));
+            (function(){
+                const win = document.open('https://example1.com', '', '');
+                setTimeout(() => {
+                    if (!win || !win.location) {
+                        return bypass([top]); // give up
+                    }
+                    win.location.href = 'about:blank';
+                    setTimeout(() => {
+                        bypass([win]);
+                    }, 1000)
+                }, 1000);
+            }());
+        });
+        expect(result).toBe('V');
+    });
+
+    it('should fail to use atob of a window that was created via document.open API to cross origin and then changed to same origin and leaked it via postMessage (onmessage)', async () => {
+        const result = await browser.executeAsync(function(done) {
+            const bypass = (wins) => done(wins.map(win => (win && win.atob ? win : top).atob('WA==')).join(','));
+            (function(){
+                onmessage = (a) => {
+                    const x = a.source;
+                    if (!x || !x.location) {
+                        return bypass([top]); // give up
+                    }
+                    x.location.href = 'https://example.com/';
+                    setTimeout(() => {
+                        bypass([x]);
+                    }, 1000);
+                };
+                document.open('https://lavamoat.github.io/snow/test/test-util.html', '', '');
+            }());
+        });
+        expect(result).toBe('V');
+    });
+
+    it('should fail to use atob of a window that was created via document.open API to cross origin and then changed to same origin and leaked it via postMessage (message)', async () => {
+        const result = await browser.executeAsync(function(done) {
+            const bypass = (wins) => done(wins.map(win => (win && win.atob ? win : top).atob('WA==')).join(','));
+            (function(){
+                addEventListener('message', a => {
+                    const x = a.source;
+                    if (!x || !x.location) {
+                        return bypass([top]); // give up
+                    }
+                    x.location.href = 'https://example.com/';
+                    setTimeout(() => {
+                        bypass([x]);
+                    }, 1000);
+                });
+                const x = {};
+                x.toString = () => 'https://lavamoat.github.io/snow/test/test-util.html';
+                document.open(x, '', '');
+            }());
+        });
+        expect(result).toBe('V');
+    });
+
+    it('should fail to use atob of a window that was created via document.open API to javascript: scheme, leaked to opener and then changed to cross origin and back to same origin', async () => {
+        const result = await browser.executeAsync(function(done) {
+            const bypass = (wins) => done(wins.map(win => (win && win.atob ? win : top).atob('WA==')).join(','));
+            (function(){
+                document.open('javAscRipt\:opener.win=window;location.href="data:1"', '', '');
+                setTimeout(() => {
+                    if (!top.win) {
+                        return bypass([top]); // give up
+                    }
+                    top.win.location.href = 'https://example.com';
+                    setTimeout(() => {
+                        bypass([top.win]);
+                    }, 500);
+                }, 500);
+            }());
+        });
+        expect(result).toBe('V');
+    });
 });