Fix wrongly overridden toString function from Function constructor (c…

…lose DevExpress#999)
LavrovArtem · Dec 26, 2016 · 00d8789 · 00d8789
1 parent b18d347
commit 00d8789
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 6 deletions.
diff --git a/src/client/sandbox/native-methods.js b/src/client/sandbox/native-methods.js
@@ -181,6 +181,7 @@ class NativeMethods {
         this.XMLHttpRequest   = win.XMLHttpRequest;
         this.Image            = win.Image;
         this.Function         = win.Function;
+        this.functionToString = win.Function.toString;
         this.FontFace         = win.FontFace;
         this.StorageEvent     = win.StorageEvent;
         this.MutationObserver = win.MutationObserver;

diff --git a/src/client/sandbox/node/window.js b/src/client/sandbox/node/window.js
@@ -228,7 +228,7 @@ export default class WindowSandbox extends SandboxBase {
         };
         window.Image.prototype = nativeMethods.Image.prototype;
 
-        window.Function = function (...args) {
+        var FunctionWrapper = function (...args) {
             var functionBodyArgIndex = args.length - 1;
 
             if (typeof args[functionBodyArgIndex] === 'string')
@@ -237,12 +237,15 @@ export default class WindowSandbox extends SandboxBase {
             return nativeMethods.Function.apply(this, args);
         };
 
+        window.Function                       = FunctionWrapper;
         window.Function.prototype             = nativeMethods.Function.prototype;
-        window.Function.prototype.constructor = window.Function;
+        window.Function.prototype.constructor = FunctionWrapper;
 
         // NOTE: We need to create function which returns string without calling toString every time
         // because if the Function.prototype.toString is overridden it can be the cause of recursion
-        window.Function.toString = () => nativeFunctionToString;
+        window.Function.toString = function () {
+            return this === FunctionWrapper ? nativeFunctionToString : nativeMethods.functionToString.call(this);
+        };
 
         if (typeof window.history.pushState === 'function' && typeof window.history.replaceState === 'function') {
             window.history.pushState = function () {

diff --git a/test/client/fixtures/sandbox/node/window-test.js b/test/client/fixtures/sandbox/node/window-test.js
@@ -132,7 +132,7 @@ test('window.onerror must be overriden (B238830)', function () {
 });
 
 test('the constructor field of a function should return a wrapped Function object (GH-913)', function () {
-    var f = function () {
+    var f = function f () {
     };
 
     strictEqual(f.constructor, Function);
@@ -143,12 +143,14 @@ test('the constructor field of a function should return a wrapped Function objec
 
     /*eslint-disable no-extend-native*/
     Function.prototype.toString = function () {
-        toString.call(this);
+        var str = toString.call(this);
 
         ok(true);
+
+        return str;
     };
 
-    f.toString();
+    strictEqual(f.toString().replace(/\s+/g, ''), 'functionf(){}');
 
     Function.prototype.toString = nativeToString;
     /*eslint-enable no-extend-native*/