js-confuser 2.1.0 → 2.1.2

package/dist/transforms/controlFlowFlattening.js

@@ -15,6 +15,7 @@ var _IntGen = require("../utils/IntGen");
 var _assert = require("assert");
 var _NameGen = require("../utils/NameGen");
 var _constants = require("../constants");
+var _xorStringTemplate = require("../templates/xorStringTemplate");
 function _interopRequireDefault(e) { return e && e.__esModule ? e : { "default": e }; }
 function _interopRequireWildcard(e, t) { if ("function" == typeof WeakMap) var r = new WeakMap(), n = new WeakMap(); return (_interopRequireWildcard = function _interopRequireWildcard(e, t) { if (!t && e && e.__esModule) return e; var o, i, f = { __proto__: null, "default": e }; if (null === e || "object" != _typeof(e) && "function" != typeof e) return f; if (o = t ? n : r) { if (o.has(e)) return o.get(e); o.set(e, f); } for (var _t in e) "default" !== _t && {}.hasOwnProperty.call(e, _t) && ((i = (o = Object.defineProperty) && Object.getOwnPropertyDescriptor(e, _t)) && (i.get || i.set) ? o(f, _t, i) : f[_t] = e[_t]); return f; })(e, t); }
 function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
@@ -74,9 +75,9 @@ var _default = exports["default"] = function _default(_ref) {
   var addPredicateTests = true; // case scope.A + 10: ...
   var mangleNumericalLiterals = true; // 50 => state + X
   var mangleBooleanLiterals = true; // true => state == X
-  var addWithStatement = true; // Disabling not supported yet
-  var addGeneratorFunction = true; // Wrap in generator function?
+  var mangleStringLiterals = true; // "hi" => xor("fOq", state + X)
 
+  var stateVarsRange = [3, 10];
   var cffPrefix = me.getPlaceholder();
 
   // Amount of blocks changed by Control Flow Flattening
@@ -86,1170 +87,1170 @@ var _default = exports["default"] = function _default(_ref) {
     var fnOrProgram = (0, _astUtils.getParentFunctionOrProgram)(path);
     fnOrProgram.node[CFF_UNSAFE] = reason;
   }
-  return {
-    post: function post() {
-      for (var _i = 0, _functionsModified = functionsModified; _i < _functionsModified.length; _i++) {
-        var node = _functionsModified[_i];
-        node[_constants.UNSAFE] = true;
+  var needsSumFunction = false;
+  var sumFnName = me.getPlaceholder() + "_cff_sum";
+  var xorFnName = me.getPlaceholder() + "_cff_xor";
+  function cffMain(_path) {
+    var programOrFunctionPath = _path;
+
+    // if (cffOptions.excludeLoops) {
+    //   // Exclude loops
+    //   if (programOrFunctionPath.find((p) => p.isForStatement() || p.isWhile()))
+    //     return;
+    // }
+    // var debugName = (programOrFunctionPath?.node as any)?.id?.name;
+
+    // Exclude 'CFF_UNSAFE' functions
+    if (programOrFunctionPath.node[CFF_UNSAFE]) {
+      // console.log(`Skipping ${debugName} because ${programOrFunctionPath.node[CFF_UNSAFE]}`);
+      return;
+    }
+    var programPath = _path.isProgram() ? _path : null;
+    var functionPath = _path.isFunction() ? _path : null;
+    var blockPath;
+    if (programPath) {
+      blockPath = programPath;
+    } else {
+      var fnBlockPath = functionPath.get("body");
+      if (!fnBlockPath.isBlock()) return;
+      blockPath = fnBlockPath;
+    }
+
+    // Don't apply to strict mode blocks
+    var strictModeEnforcingBlock = programOrFunctionPath.find(function (path) {
+      return (0, _astUtils.isStrictMode)(path);
+    });
+    if (strictModeEnforcingBlock) return;
+
+    // Must be at least 3 statements or more
+    if (blockPath.node.body.length < 3) return;
+
+    // Check user's threshold setting
+    if (!me.computeProbabilityMap(me.options.controlFlowFlattening)) {
+      return;
+    }
+    if (functionPath) {
+      // Avoid unsafe functions
+      if (functionPath.node[_constants.UNSAFE]) {
+        // console.log(`Skipping ${debugName} because of UNSAFE flag`);
+        return;
       }
-    },
-    visitor: {
-      // Unsafe detection
-      ThisExpression: function ThisExpression(path) {
-        flagFunctionToAvoid(path, "this");
-      },
-      VariableDeclaration: function VariableDeclaration(path) {
-        if (path.node.declarations.length !== 1) {
-          path.getAncestry().forEach(function (p) {
-            p.node[CFF_UNSAFE] = "multipleDeclarations";
-          });
-        }
-      },
+      if (functionPath.node.async || functionPath.node.generator) return;
+    }
+    programOrFunctionPath.scope.crawl();
+    var hasIllegalNode = false;
+    var bindingNames = new Set();
+    blockPath.traverse({
       Identifier: function Identifier(path) {
-        if (path.node.name === _constants.variableFunctionName || path.node.name === "arguments") {
-          flagFunctionToAvoid(path, "arguments");
+        if (!path.isBindingIdentifier()) return;
+        var binding = path.scope.getBinding(path.node.name);
+        if (!binding) return;
+        var fnParent = path.getFunctionParent();
+        if (path.key === "id" && path.parentPath.isFunctionDeclaration()) {
+          fnParent = path.parentPath.getFunctionParent();
         }
-      },
-      "Super|MetaProperty|AwaitExpression|YieldExpression": function SuperMetaPropertyAwaitExpressionYieldExpression(path) {
-        flagFunctionToAvoid(path, "functionSpecific");
-      },
-      // Main CFF transformation
-      "Program|Function": {
-        exit: function exit(_path) {
-          var programOrFunctionPath = _path;
-
-          // Exclude loops
-          if (programOrFunctionPath.find(function (p) {
-            return p.isForStatement() || p.isWhile();
-          })) return;
-
-          // Exclude 'CFF_UNSAFE' functions
-          if (programOrFunctionPath.node[CFF_UNSAFE]) return;
-          var programPath = _path.isProgram() ? _path : null;
-          var functionPath = _path.isFunction() ? _path : null;
-          var blockPath;
-          if (programPath) {
-            blockPath = programPath;
-          } else {
-            var fnBlockPath = functionPath.get("body");
-            if (!fnBlockPath.isBlock()) return;
-            blockPath = fnBlockPath;
+        if (fnParent !== functionPath) return;
+        if (!(0, _astUtils.isDefiningIdentifier)(path)) {
+          return;
+        }
+        if (bindingNames.has(path.node.name)) {
+          hasIllegalNode = true;
+          path.stop();
+          return;
+        }
+        bindingNames.add(path.node.name);
+      }
+    });
+    if (hasIllegalNode) {
+      return;
+    }
+    me.changeData.blocks++;
+
+    // Limit how many numbers get entangled
+    var mangledLiteralsCreated = 0;
+    var cffIndex = ++cffCounter; // Start from 1
+    var prefix = cffPrefix + "_" + cffIndex;
+    var identifier = function identifier(suffix) {
+      var name;
+      if (isDebug) {
+        name = prefix + "_" + suffix;
+      } else {
+        name = me.obfuscator.nameGen.generate(false);
+      }
+      var id = t.identifier(name);
+      return id;
+    };
+    var mainFnName = identifier("main");
+    var scopeVar = identifier("scope");
+    var statesVar = identifier("states");
+    var runtimeVar = identifier("runtime");
+    var stateVars = new Array(isDebug ? 1 : (0, _randomUtils.getRandomInteger)(stateVarsRange[0], stateVarsRange[1])).fill("").map(function (_, i) {
+      return t.memberExpression((0, _node.deepClone)(statesVar), t.numericLiteral(i), true);
+    });
+    var argVar = identifier("_arg");
+    var usedArgVar = false;
+    var _didReturnVar = identifier("return");
+    var basicBlocks = new Map();
+
+    // Map labels to states
+    var stateIntGen = new _IntGen.IntGen();
+    var defaultBlockPath = blockPath;
+    var scopeCounter = 0;
+    var scopeNameGen = new _NameGen.NameGen(me.options.identifierGenerator);
+    if (!isDebug) {
+      scopeNameGen = me.obfuscator.nameGen;
+    }
+    var ScopeManager = /*#__PURE__*/function () {
+      function ScopeManager(scope, initializingBasicBlock) {
+        _classCallCheck(this, ScopeManager);
+        _defineProperty(this, "isNotUsed", true);
+        _defineProperty(this, "requiresInitializing", true);
+        _defineProperty(this, "nameMap", new Map());
+        _defineProperty(this, "nameGen", new _NameGen.NameGen(me.options.identifierGenerator));
+        this.scope = scope;
+        this.initializingBasicBlock = initializingBasicBlock;
+        this.propertyName = isDebug ? "_" + cffIndex + "_" + scopeCounter++ : scopeNameGen.generate();
+      }
+      return _createClass(ScopeManager, [{
+        key: "getNewName",
+        value: function getNewName(name, originalNode) {
+          if (!this.nameMap.has(name)) {
+            var newName = this.nameGen.generate(false);
+            if (isDebug) {
+              newName = "_" + name;
+            }
+
+            // console.log(name, newName);
+
+            this.nameMap.set(name, newName);
+            me.changeData.variables++;
+
+            // console.log(
+            //   `Renaming ${name} to ${newName}: ${this.scope.path.type} (scope ${this.propertyName})`,
+            // );
+
+            return newName;
           }
+          return this.nameMap.get(name);
+        }
+      }, {
+        key: "getScopeObject",
+        value: function getScopeObject() {
+          return t.memberExpression((0, _node.deepClone)(scopeVar), t.stringLiteral(this.propertyName), true);
+        }
+      }, {
+        key: "getInitializingStatement",
+        value: function getInitializingStatement() {
+          return t.expressionStatement(t.assignmentExpression("=", this.getScopeObject(), this.getInitializingObjectExpression()));
+        }
+      }, {
+        key: "getInitializingObjectExpression",
+        value: function getInitializingObjectExpression() {
+          return isDebug ? new _template["default"]("\n                  ({\n                    identity: \"".concat(this.propertyName, "\"\n                  })\n                    ")).expression() : new _template["default"]("({})").expression();
+        }
+      }, {
+        key: "getMemberExpression",
+        value: function getMemberExpression(name) {
+          var object = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : this.getScopeObject();
+          var memberExpression = t.memberExpression(object, t.stringLiteral(name), true);
+          return memberExpression;
+        }
+      }, {
+        key: "parent",
+        get: function get() {
+          return scopeToScopeManager.get(this.scope.parent);
+        }
+      }, {
+        key: "getObjectExpression",
+        value: function getObjectExpression(refreshLabel) {
+          var refreshScope = basicBlocks.get(refreshLabel).scopeManager;
+          var propertyMap = {};
+          var cursor = this.scope;
+          while (cursor) {
+            var parentScopeManager = scopeToScopeManager.get(cursor);
+            if (parentScopeManager) {
+              propertyMap[parentScopeManager.propertyName] = t.memberExpression((0, _node.deepClone)(scopeVar), t.stringLiteral(parentScopeManager.propertyName), true);
+            }
+            cursor = cursor.parent;
+          }
+          propertyMap[refreshScope.propertyName] = refreshScope.getInitializingObjectExpression();
+          var properties = [];
+          for (var key in propertyMap) {
+            properties.push(t.objectProperty(t.stringLiteral(key), propertyMap[key], true));
+          }
+          return t.objectExpression(properties);
+        }
+      }, {
+        key: "hasOwnName",
+        value: function hasOwnName(name) {
+          return this.nameMap.has(name);
+        }
+      }]);
+    }();
+    var getImpossibleBasicBlocks = function getImpossibleBasicBlocks() {
+      return Array.from(basicBlocks.values()).filter(function (block) {
+        return block.options.impossible;
+      });
+    };
+    var scopeToScopeManager = new Map();
+    /**
+     * A Basic Block is a sequence of instructions with no diversion except at the entry and exit points.
+     */
+    var BasicBlock = /*#__PURE__*/function () {
+      function BasicBlock(label, parentPath) {
+        var _this = this;
+        var options = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : {};
+        _classCallCheck(this, BasicBlock);
+        this.label = label;
+        this.parentPath = parentPath;
+        this.options = options;
+        this.createPath();
+        if (isDebug) {
+          // States in debug mode are just 1, 2, 3, ...
+          this.totalState = basicBlocks.size + 1;
+        } else {
+          this.totalState = stateIntGen.generate();
+        }
 
-          // Don't apply to strict mode blocks
-          var strictModeEnforcingBlock = programOrFunctionPath.find(function (path) {
-            return (0, _astUtils.isStrictMode)(path);
+        // Correct state values
+        // Start with random numbers
+        this.stateValues = stateVars.map(function () {
+          return (0, _randomUtils.getRandomInteger)(-250, 250);
+        });
+
+        // Try to re-use old state values to make diffs smaller
+        if (basicBlocks.size > 1) {
+          var lastBlock = _toConsumableArray(basicBlocks.values()).at(-1);
+          this.stateValues = lastBlock.stateValues.map(function (oldValue, i) {
+            // Increase chance for re-using old values to make state transitions less drastic
+            return (0, _randomUtils.chance)(90) ? oldValue : _this.stateValues[i];
           });
-          if (strictModeEnforcingBlock) return;
+        }
+
+        // Correct one of the values so that the accumulated sum is equal to the state
+        var correctIndex = (0, _randomUtils.getRandomInteger)(0, this.stateValues.length);
+        var getCurrentState = function getCurrentState() {
+          return _this.stateValues.reduce(function (a, b) {
+            return a + b;
+          }, 0);
+        };
 
-          // Must be at least 3 statements or more
-          if (blockPath.node.body.length < 3) return;
+        // Correct the value
+        this.stateValues[correctIndex] = this.totalState - (getCurrentState() - this.stateValues[correctIndex]);
+        (0, _assert.ok)(getCurrentState() === this.totalState);
 
-          // Check user's threshold setting
-          if (!me.computeProbabilityMap(me.options.controlFlowFlattening)) {
-            return;
+        // Store basic block
+        basicBlocks.set(label, this);
+
+        // Create a new scope manager if it doesn't exist
+        if (!scopeToScopeManager.has(this.scope)) {
+          scopeToScopeManager.set(this.scope, new ScopeManager(this.scope, this));
+        }
+        this.initializedScope = this.scopeManager;
+      }
+      return _createClass(BasicBlock, [{
+        key: "createPath",
+        value: function createPath() {
+          var newPath = _traverse.NodePath.get({
+            hub: this.parentPath.hub,
+            parentPath: this.parentPath,
+            parent: this.parentPath.node,
+            container: this.parentPath.node.body,
+            listKey: "body",
+            // Set the correct list key
+            key: "virtual" // Set the index of the new node
+          });
+          newPath.scope = this.parentPath.scope;
+          newPath.parentPath = this.parentPath;
+          newPath.node = t.blockStatement([]);
+          this.thisPath = newPath;
+          this.thisNode = newPath.node;
+        }
+      }, {
+        key: "insertAfter",
+        value: function insertAfter(newNode) {
+          this.body.push(newNode);
+        }
+      }, {
+        key: "scope",
+        get: function get() {
+          return this.parentPath.scope;
+        }
+      }, {
+        key: "scopeManager",
+        get: function get() {
+          return scopeToScopeManager.get(this.scope);
+        }
+      }, {
+        key: "body",
+        get: function get() {
+          return this.thisPath.node.body;
+        }
+      }, {
+        key: "createFalsePredicate",
+        value: function createFalsePredicate() {
+          var predicate = this.createPredicate();
+          if (predicate.value) {
+            // Make predicate false
+            return t.unaryExpression("!", predicate.node);
           }
-          if (functionPath) {
-            // Avoid unsafe functions
-            if (functionPath.node[_constants.UNSAFE]) return;
-            if (functionPath.node.async || functionPath.node.generator) return;
+          return predicate.node;
+        }
+      }, {
+        key: "createTruePredicate",
+        value: function createTruePredicate() {
+          var predicate = this.createPredicate();
+          if (!predicate.value) {
+            // Make predicate true
+            return t.unaryExpression("!", predicate.node);
           }
-          programOrFunctionPath.scope.crawl();
-          var hasIllegalNode = false;
-          var bindingNames = new Set();
-          blockPath.traverse({
-            Identifier: function Identifier(path) {
-              if (!path.isBindingIdentifier()) return;
-              var binding = path.scope.getBinding(path.node.name);
-              if (!binding) return;
-              var fnParent = path.getFunctionParent();
-              if (path.key === "id" && path.parentPath.isFunctionDeclaration()) {
-                fnParent = path.parentPath.getFunctionParent();
-              }
-              if (fnParent !== functionPath) return;
-              if (!(0, _astUtils.isDefiningIdentifier)(path)) {
-                return;
-              }
-              if (bindingNames.has(path.node.name)) {
-                hasIllegalNode = true;
-                path.stop();
-                return;
-              }
-              bindingNames.add(path.node.name);
-            }
-          });
-          if (hasIllegalNode) {
-            return;
+          return predicate.node;
+        }
+      }, {
+        key: "createPredicate",
+        value: function createPredicate() {
+          var stateVarIndex = (0, _randomUtils.getRandomInteger)(0, stateVars.length);
+          var stateValue = this.stateValues[stateVarIndex];
+          var compareValue = (0, _randomUtils.choice)([stateValue, (0, _randomUtils.getRandomInteger)(-250, 250)]);
+          var operator = (0, _randomUtils.choice)(["==", "!=", "<", ">"]);
+          var compareResult;
+          switch (operator) {
+            case "==":
+              compareResult = stateValue === compareValue;
+              break;
+            case "!=":
+              compareResult = stateValue !== compareValue;
+              break;
+            case "<":
+              compareResult = stateValue < compareValue;
+              break;
+            case ">":
+              compareResult = stateValue > compareValue;
+              break;
           }
-          me.changeData.blocks++;
-
-          // Limit how many numbers get entangled
-          var mangledLiteralsCreated = 0;
-          var cffIndex = ++cffCounter; // Start from 1
-          var prefix = cffPrefix + "_" + cffIndex;
-          var withIdentifier = function withIdentifier(suffix) {
-            var name;
-            if (isDebug) {
-              name = prefix + "_" + suffix;
-            } else {
-              name = me.obfuscator.nameGen.generate(false);
-            }
-            var id = t.identifier(name);
-            id[_constants.NO_RENAME] = cffIndex;
-            return id;
+          return {
+            node: t.binaryExpression(operator, (0, _node.deepClone)(stateVars[stateVarIndex]), (0, _node.numericLiteral)(compareValue)),
+            value: compareResult
           };
-          var mainFnName = withIdentifier("main");
-          var scopeVar = withIdentifier("scope");
-          var stateVars = new Array(isDebug ? 1 : (0, _randomUtils.getRandomInteger)(2, 5)).fill("").map(function (_, i) {
-            return withIdentifier("state_".concat(i));
-          });
-          var argVar = withIdentifier("_arg");
-          var usedArgVar = false;
-          var _didReturnVar = withIdentifier("return");
-          var basicBlocks = new Map();
-
-          // Map labels to states
-          var stateIntGen = new _IntGen.IntGen();
-          var defaultBlockPath = blockPath;
-          var scopeCounter = 0;
-          var scopeNameGen = new _NameGen.NameGen(me.options.identifierGenerator);
-          if (!isDebug) {
-            scopeNameGen = me.obfuscator.nameGen;
-          }
+        }
+      }, {
+        key: "identifier",
+        value: function identifier(identifierName, scopeManager) {
+          return scopeManager.getMemberExpression(identifierName);
+        }
+      }]);
+    }();
+    /**
+     * Stage 1: Flatten the code into Basic Blocks
+     *
+     * This involves transforming the Control Flow / Scopes into blocks with 'goto' statements
+     *
+     * - A block is simply a sequence of statements
+     * - A block can have a 'goto' statement to another block
+     * - A block original scope is preserved
+     *
+     * 'goto' & Scopes are transformed in Stage 2
+     */
+    var switchLabel = me.getPlaceholder();
+    var breakStatement = function breakStatement() {
+      return t.breakStatement(t.identifier(switchLabel));
+    };
+    var startLabel = me.getPlaceholder();
+    var endLabel = me.getPlaceholder();
+    var currentBasicBlock = new BasicBlock(startLabel, blockPath);
+    var gotoFunctionName = "GOTO__" + me.getPlaceholder() + "__IF_YOU_CAN_READ_THIS_THERE_IS_A_BUG";
+    function GotoControlStatement(label) {
+      return new _template["default"]("\n              ".concat(gotoFunctionName, "(\"").concat(label, "\");\n              ")).single();
+    }
 
-          // Create 'with' object - Determines which scope gets top-level variable access
-          var withProperty = isDebug ? "with" : scopeNameGen.generate(false);
-          var withMemberExpression = new _template["default"]("".concat(scopeVar.name, "[\"").concat(withProperty, "\"]")).expression();
-          withMemberExpression.object[_constants.NO_RENAME] = cffIndex;
-          var ScopeManager = /*#__PURE__*/function () {
-            function ScopeManager(scope, initializingBasicBlock) {
-              _classCallCheck(this, ScopeManager);
-              _defineProperty(this, "isNotUsed", true);
-              _defineProperty(this, "requiresInitializing", true);
-              _defineProperty(this, "nameMap", new Map());
-              _defineProperty(this, "nameGen", addWithStatement ? me.obfuscator.nameGen : new _NameGen.NameGen(me.options.identifierGenerator));
-              this.scope = scope;
-              this.initializingBasicBlock = initializingBasicBlock;
-              this.propertyName = isDebug ? "_" + cffIndex + "_" + scopeCounter++ : scopeNameGen.generate();
+    // Ends the current block and starts a new one
+    function endCurrentBasicBlock() {
+      var _ref2 = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {},
+        _ref2$jumpToNext = _ref2.jumpToNext,
+        jumpToNext = _ref2$jumpToNext === void 0 ? true : _ref2$jumpToNext,
+        _ref2$nextLabel = _ref2.nextLabel,
+        nextLabel = _ref2$nextLabel === void 0 ? me.getPlaceholder() : _ref2$nextLabel,
+        _ref2$prevJumpTo = _ref2.prevJumpTo,
+        prevJumpTo = _ref2$prevJumpTo === void 0 ? null : _ref2$prevJumpTo,
+        _ref2$nextBlockPath = _ref2.nextBlockPath,
+        nextBlockPath = _ref2$nextBlockPath === void 0 ? null : _ref2$nextBlockPath;
+      (0, _assert.ok)(nextBlockPath);
+      if (prevJumpTo) {
+        currentBasicBlock.insertAfter(GotoControlStatement(prevJumpTo));
+      } else if (jumpToNext) {
+        currentBasicBlock.insertAfter(GotoControlStatement(nextLabel));
+      }
+      currentBasicBlock = new BasicBlock(nextLabel, nextBlockPath);
+    }
+    var prependNodes = [];
+    var functionExpressions = [];
+    function flattenIntoBasicBlocks(bodyIn) {
+      // if (!Array.isArray(bodyIn) && bodyIn.isBlock()) {
+      //   currentBasicBlock.parentPath = bodyIn;
+      // }
+      var body = Array.isArray(bodyIn) ? bodyIn : bodyIn.get("body");
+      var nextBlockPath = Array.isArray(bodyIn) ? currentBasicBlock.parentPath : bodyIn;
+      var _loop = function _loop() {
+          var statement = body[index];
+
+          // Keep Imports before everything else
+          if (statement.isImportDeclaration()) {
+            prependNodes.push(statement.node);
+            return 0; // continue
+          }
+          if (statement.isFunctionDeclaration()) {
+            var fnName = statement.node.id.name;
+            var isIllegal = false;
+            if (!flattenFunctionDeclarations || statement.node.async || statement.node.generator || statement.node[_constants.UNSAFE] || statement.node[CFF_UNSAFE] || (0, _astUtils.isStrictMode)(statement)) {
+              isIllegal = true;
             }
-            return _createClass(ScopeManager, [{
-              key: "findBestWithDiscriminant",
-              value: function findBestWithDiscriminant(basicBlock) {
-                var _this$parent;
-                // This initializing block is forbidden to have a with discriminant
-                // (As no previous code is able to prepare the with discriminant)
-                if (basicBlock !== this.initializingBasicBlock) {
-                  // If no variables were defined in this scope, don't use it
-                  if (Object.keys(this.scope.bindings).length > 0) return this;
-                }
-                return (_this$parent = this.parent) === null || _this$parent === void 0 ? void 0 : _this$parent.findBestWithDiscriminant(basicBlock);
-              }
-            }, {
-              key: "getNewName",
-              value: function getNewName(name, originalNode) {
-                if (!this.nameMap.has(name)) {
-                  var newName = this.nameGen.generate(false);
-                  if (isDebug) {
-                    newName = "_" + name;
-                  }
+            var oldBasicBlock = currentBasicBlock;
+            var _fnLabel = me.getPlaceholder();
+            var sm = currentBasicBlock.scopeManager;
+            var rename = sm.getNewName(fnName);
+            var hoistedBasicBlock = Array.from(basicBlocks.values()).find(function (block) {
+              return block.parentPath === currentBasicBlock.parentPath;
+            });
 
-                  // console.log(name, newName);
+            // Added into scope object due to function flattening causes redeclaration (Test #42 and #44)
+            hoistedBasicBlock.scope.bindings[fnName].kind = "var";
+            if (isIllegal) {
+              // Function must be converted to function expression as hoisted function declaration
+              // gets redeclared (and losing prototype methods) each iteration (Test #43)
+              var asFunctionExpression = statement.node;
+              asFunctionExpression.type = "FunctionExpression";
+              hoistedBasicBlock.body.unshift(t.variableDeclaration("var", [t.variableDeclarator(t.identifier(fnName), asFunctionExpression)]));
+              return 0; // continue
+            }
+            me.changeData.functions++;
+            var functionExpression = t.functionExpression(null, [], t.blockStatement([]));
+            functionExpressions.push([fnName, _fnLabel, currentBasicBlock, functionExpression]);
+
+            // Change the function declaration to a variable declaration
+            hoistedBasicBlock.body.unshift(t.variableDeclaration("var", [t.variableDeclarator(t.identifier(fnName), functionExpression)]));
+            var blockStatement = statement.get("body");
+            endCurrentBasicBlock({
+              nextLabel: _fnLabel,
+              nextBlockPath: blockStatement,
+              jumpToNext: false
+            });
+            var fnTopBlock = currentBasicBlock;
 
-                  this.nameMap.set(name, newName);
-                  me.changeData.variables++;
+            // Implicit return
+            blockStatement.node.body.push(t.returnStatement(t.identifier("undefined")));
+            flattenIntoBasicBlocks(blockStatement);
+            scopeToScopeManager.get(statement.scope).requiresInitializing = false;
 
-                  // console.log(
-                  //   "Renaming " +
-                  //     name +
-                  //     " to " +
-                  //     newName +
-                  //     " : " +
-                  //     this.scope.path.type
-                  // );
+            // Debug label
+            if (isDebug) {
+              fnTopBlock.body.unshift(t.expressionStatement(t.stringLiteral("Function " + statement.node.id.name + " -> Renamed to " + rename)));
+            }
 
-                  return newName;
-                }
-                return this.nameMap.get(name);
-              }
-            }, {
-              key: "getScopeObject",
-              value: function getScopeObject() {
-                return t.memberExpression((0, _node.deepClone)(scopeVar), t.stringLiteral(this.propertyName), true);
-              }
-            }, {
-              key: "getInitializingStatement",
-              value: function getInitializingStatement() {
-                return t.expressionStatement(t.assignmentExpression("=", this.getScopeObject(), this.getInitializingObjectExpression()));
-              }
-            }, {
-              key: "getInitializingObjectExpression",
-              value: function getInitializingObjectExpression() {
-                return isDebug ? new _template["default"]("\n                  ({\n                    identity: \"".concat(this.propertyName, "\"\n                  })\n                    ")).expression() : new _template["default"]("({})").expression();
-              }
-            }, {
-              key: "getMemberExpression",
-              value: function getMemberExpression(name) {
-                var object = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : this.getScopeObject();
-                var memberExpression = t.memberExpression(object, t.stringLiteral(name), true);
-                return memberExpression;
-              }
-            }, {
-              key: "parent",
-              get: function get() {
-                return scopeToScopeManager.get(this.scope.parent);
-              }
-            }, {
-              key: "getObjectExpression",
-              value: function getObjectExpression(refreshLabel) {
-                var refreshScope = basicBlocks.get(refreshLabel).scopeManager;
-                var propertyMap = {};
-                var cursor = this.scope;
-                while (cursor) {
-                  var parentScopeManager = scopeToScopeManager.get(cursor);
-                  if (parentScopeManager) {
-                    propertyMap[parentScopeManager.propertyName] = t.memberExpression((0, _node.deepClone)(scopeVar), t.stringLiteral(parentScopeManager.propertyName), true);
+            // Unpack parameters from the parameter 'argVar'
+            if (statement.node.params.length > 0) {
+              usedArgVar = true;
+              fnTopBlock.body.unshift(t.variableDeclaration("var", [t.variableDeclarator(t.arrayPattern(statement.node.params), (0, _node.deepClone)(argVar))]));
+
+              // Change bindings from 'param' to 'var'
+              statement.get("params").forEach(function (param) {
+                var ids = param.getBindingIdentifierPaths();
+                // Loop over the record of binding identifiers
+                for (var identifierName in ids) {
+                  var identifierPath = ids[identifierName];
+                  if (identifierPath.getFunctionParent() === statement) {
+                    var binding = statement.scope.getBinding(identifierName);
+                    if (binding) {
+                      binding.kind = "var";
+                    }
                   }
-                  cursor = cursor.parent;
-                }
-                propertyMap[refreshScope.propertyName] = refreshScope.getInitializingObjectExpression();
-                var properties = [];
-                for (var key in propertyMap) {
-                  properties.push(t.objectProperty(t.stringLiteral(key), propertyMap[key], true));
                 }
-                return t.objectExpression(properties);
-              }
-            }, {
-              key: "hasOwnName",
-              value: function hasOwnName(name) {
-                return this.nameMap.has(name);
-              }
-            }]);
-          }();
-          var getImpossibleBasicBlocks = function getImpossibleBasicBlocks() {
-            return Array.from(basicBlocks.values()).filter(function (block) {
-              return block.options.impossible;
-            });
-          };
-          var scopeToScopeManager = new Map();
-          /**
-           * A Basic Block is a sequence of instructions with no diversion except at the entry and exit points.
-           */
-          var BasicBlock = /*#__PURE__*/function () {
-            function BasicBlock(label, parentPath) {
-              var _this = this;
-              var options = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : {};
-              _classCallCheck(this, BasicBlock);
-              _defineProperty(this, "allowWithDiscriminant", true);
-              this.label = label;
-              this.parentPath = parentPath;
-              this.options = options;
-              this.createPath();
-              if (isDebug) {
-                // States in debug mode are just 1, 2, 3, ...
-                this.totalState = basicBlocks.size + 1;
-              } else {
-                this.totalState = stateIntGen.generate();
-              }
-
-              // Correct state values
-              // Start with random numbers
-              this.stateValues = stateVars.map(function () {
-                return (0, _randomUtils.getRandomInteger)(-250, 250);
               });
+            }
+            currentBasicBlock = oldBasicBlock;
+            return 0; // continue
+          }
 
-              // Try to re-use old state values to make diffs smaller
-              if (basicBlocks.size > 1) {
-                var lastBlock = _toConsumableArray(basicBlocks.values()).at(-1);
-                this.stateValues = lastBlock.stateValues.map(function (oldValue, i) {
-                  return (0, _randomUtils.choice)([oldValue, _this.stateValues[i]]);
+          // Convert IF statements into Basic Blocks
+          if (statement.isIfStatement() && flattenIfStatements) {
+            var test = statement.get("test");
+            var consequent = statement.get("consequent");
+            var alternate = statement.get("alternate");
+
+            // Both consequent and alternate are blocks
+            if (consequent.isBlockStatement() && (!alternate.node || alternate.isBlockStatement())) {
+              me.changeData.ifStatements++;
+              var consequentLabel = me.getPlaceholder();
+              var alternateLabel = alternate.node ? me.getPlaceholder() : null;
+              var afterPath = me.getPlaceholder();
+              currentBasicBlock.insertAfter(t.ifStatement(test.node, GotoControlStatement(consequentLabel), alternateLabel ? GotoControlStatement(alternateLabel) : GotoControlStatement(afterPath)));
+              var _oldBasicBlock = currentBasicBlock;
+              endCurrentBasicBlock({
+                jumpToNext: false,
+                nextLabel: consequentLabel,
+                nextBlockPath: consequent
+              });
+              flattenIntoBasicBlocks(consequent);
+              currentBasicBlock.initializedScope = _oldBasicBlock.scopeManager;
+              if (alternate.isBlockStatement()) {
+                endCurrentBasicBlock({
+                  prevJumpTo: afterPath,
+                  nextLabel: alternateLabel,
+                  nextBlockPath: alternate
                 });
+                flattenIntoBasicBlocks(alternate);
               }
+              endCurrentBasicBlock({
+                prevJumpTo: afterPath,
+                nextLabel: afterPath,
+                nextBlockPath: _oldBasicBlock.parentPath
+              });
+              return 0; // continue
+            }
+          }
+          if (Number(index) === body.length - 1 && statement.isExpressionStatement() && statement.findParent(function (p) {
+            return p.isBlock();
+          }) === blockPath && programPath // <- ONLY APPLY TO TOP-LEVEL NEVER FUNCTIONS!
+          ) {
+            // Return the result of the last expression for eval() purposes
+            currentBasicBlock.insertAfter(t.returnStatement(statement.get("expression").node));
+            return 0; // continue
+          }
 
-              // Correct one of the values so that the accumulated sum is equal to the state
-              var correctIndex = (0, _randomUtils.getRandomInteger)(0, this.stateValues.length);
-              var getCurrentState = function getCurrentState() {
-                return _this.stateValues.reduce(function (a, b) {
-                  return a + b;
-                }, 0);
-              };
+          // 3 or more statements should be split more
+          if (currentBasicBlock.body.length > 1 && (0, _randomUtils.chance)(50 + currentBasicBlock.body.length)) {
+            endCurrentBasicBlock({
+              nextBlockPath: nextBlockPath
+            });
+          }
 
-              // Correct the value
-              this.stateValues[correctIndex] = this.totalState - (getCurrentState() - this.stateValues[correctIndex]);
-              (0, _assert.ok)(getCurrentState() === this.totalState);
+          // console.log(currentBasicBlock.thisPath.type);
+          // console.log(currentBasicBlock.body);
+          currentBasicBlock.body.push(statement.node);
+        },
+        _ret;
+      for (var index in body) {
+        _ret = _loop();
+        if (_ret === 0) continue;
+      }
+    }
 
-              // Store basic block
-              basicBlocks.set(label, this);
+    // Convert our code into Basic Blocks
+    flattenIntoBasicBlocks(blockPath.get("body"));
+
+    // Ensure always jumped to the Program end
+    endCurrentBasicBlock({
+      jumpToNext: true,
+      nextLabel: endLabel,
+      nextBlockPath: defaultBlockPath
+    });
+    if (!isDebug && addDeadCode) {
+      // DEAD CODE 1/3: Add fake chunks that are never reached
+      var fakeChunkCount = (0, _randomUtils.getRandomInteger)(1, 5);
+      for (var i = 0; i < fakeChunkCount; i++) {
+        // These chunks just jump somewhere random, they are never executed
+        // so it could contain any code
+        var fakeBlock = new BasicBlock(me.getPlaceholder(), blockPath, {
+          impossible: true
+        });
+        var fakeJump = void 0;
+        do {
+          fakeJump = (0, _randomUtils.choice)(Array.from(basicBlocks.keys()));
+        } while (fakeJump === fakeBlock.label);
+        fakeBlock.insertAfter(GotoControlStatement(fakeJump));
+        me.changeData.deadCode++;
+      }
 
-              // Create a new scope manager if it doesn't exist
-              if (!scopeToScopeManager.has(this.scope)) {
-                scopeToScopeManager.set(this.scope, new ScopeManager(this.scope, this));
-              }
-              this.initializedScope = this.scopeManager;
-            }
-            return _createClass(BasicBlock, [{
-              key: "withDiscriminant",
-              get: function get() {
-                if (!this.allowWithDiscriminant) return;
-                return this.bestWithDiscriminant;
-              }
-            }, {
-              key: "createPath",
-              value: function createPath() {
-                var newPath = _traverse.NodePath.get({
-                  hub: this.parentPath.hub,
-                  parentPath: this.parentPath,
-                  parent: this.parentPath.node,
-                  container: this.parentPath.node.body,
-                  listKey: "body",
-                  // Set the correct list key
-                  key: "virtual" // Set the index of the new node
+      // DEAD CODE 2/3: Add fake jumps to really mess with deobfuscators
+      // "irreducible control flow"
+      basicBlocks.forEach(function (basicBlock) {
+        if ((0, _randomUtils.chance)(30 - basicBlocks.size)) {
+          var randomLabel = (0, _randomUtils.choice)(Array.from(basicBlocks.keys()));
+          basicBlock.insertAfter(new _template["default"]("\n                    if({predicate}){\n                      {goto}\n                    }\n                    ").single({
+            "goto": GotoControlStatement(randomLabel),
+            predicate: basicBlock.createFalsePredicate()
+          }));
+          me.changeData.deadCode++;
+        }
+      });
+      // DEAD CODE 3/3: Clone chunks but these chunks are never ran
+      var cloneChunkCount = (0, _randomUtils.getRandomInteger)(1, 5);
+      var _loop2 = function _loop2() {
+        var randomChunk = (0, _randomUtils.choice)(Array.from(basicBlocks.values()));
+
+        // Don't double define functions
+        var hasDeclaration = randomChunk.body.find(function (stmt) {
+          return t.isDeclaration(stmt);
+        });
+        if (!hasDeclaration) {
+          var clonedChunk = new BasicBlock(me.getPlaceholder(), randomChunk.parentPath, {
+            impossible: true
+          });
+          randomChunk.thisNode.body.map(function (x) {
+            return (0, _node.deepClone)(x);
+          }).forEach(function (node) {
+            if (node.type === "EmptyStatement") return;
+            clonedChunk.insertAfter(node);
+          });
+          me.changeData.deadCode++;
+        }
+      };
+      for (var _i = 0; _i < cloneChunkCount; _i++) {
+        _loop2();
+      }
+    }
+
+    /**
+     * Stage 2: Transform 'goto' statements into valid JavaScript
+     *
+     * - 'goto' is replaced with equivalent state updates and break statements
+     * - Original identifiers are converted into member expressions
+     */
+
+    // Remap 'GotoStatement' to actual state assignments and Break statements
+    var _iterator = _createForOfIteratorHelper(basicBlocks.values()),
+      _step;
+    try {
+      var _loop4 = function _loop4() {
+        var basicBlock = _step.value;
+        var currentStateValues = basicBlock.stateValues;
+        // Wrap the statement in a Babel path to allow traversal
+
+        var outerFn = (0, _astUtils.getParentFunctionOrProgram)(basicBlock.parentPath);
+        function isWithinSameFunction(path) {
+          var fn = (0, _astUtils.getParentFunctionOrProgram)(path);
+          return fn.node === outerFn.node;
+        }
+        var visitor = {
+          SequenceExpression: {
+            enter: function enter(path) {
+              var _path$parentPath;
+              var hasComment = function hasComment(node) {
+                var _node$leadingComments;
+                return node === null || node === void 0 || (_node$leadingComments = node.leadingComments) === null || _node$leadingComments === void 0 ? void 0 : _node$leadingComments.some(function (p) {
+                  return p.value.includes("@js-confuser-assert");
                 });
-                newPath.scope = this.parentPath.scope;
-                newPath.parentPath = this.parentPath;
-                newPath.node = t.blockStatement([]);
-                this.thisPath = newPath;
-                this.thisNode = newPath.node;
-              }
-            }, {
-              key: "insertAfter",
-              value: function insertAfter(newNode) {
-                this.body.push(newNode);
-              }
-            }, {
-              key: "scope",
-              get: function get() {
-                return this.parentPath.scope;
-              }
-            }, {
-              key: "scopeManager",
-              get: function get() {
-                return scopeToScopeManager.get(this.scope);
-              }
-            }, {
-              key: "body",
-              get: function get() {
-                return this.thisPath.node.body;
-              }
-            }, {
-              key: "createFalsePredicate",
-              value: function createFalsePredicate() {
-                var predicate = this.createPredicate();
-                if (predicate.value) {
-                  // Make predicate false
-                  return t.unaryExpression("!", predicate.node);
-                }
-                return predicate.node;
+              };
+              var asserted = hasComment(path.node) || hasComment((_path$parentPath = path.parentPath) === null || _path$parentPath === void 0 ? void 0 : _path$parentPath.node);
+              if (!asserted) {
+                return;
               }
-            }, {
-              key: "createTruePredicate",
-              value: function createTruePredicate() {
-                var predicate = this.createPredicate();
-                if (!predicate.value) {
-                  // Make predicate true
-                  return t.unaryExpression("!", predicate.node);
-                }
-                return predicate.node;
+              var exprs = path.get("expressions");
+              if (exprs.length !== 2) {
+                throw new Error("Asserted sequence expressions must have exactly 2 expressions");
               }
-            }, {
-              key: "createPredicate",
-              value: function createPredicate() {
-                var stateVarIndex = (0, _randomUtils.getRandomInteger)(0, stateVars.length);
-                var stateValue = this.stateValues[stateVarIndex];
-                var compareValue = (0, _randomUtils.choice)([stateValue, (0, _randomUtils.getRandomInteger)(-250, 250)]);
-                var operator = (0, _randomUtils.choice)(["==", "!=", "<", ">"]);
-                var compareResult;
-                switch (operator) {
-                  case "==":
-                    compareResult = stateValue === compareValue;
-                    break;
-                  case "!=":
-                    compareResult = stateValue !== compareValue;
-                    break;
-                  case "<":
-                    compareResult = stateValue < compareValue;
-                    break;
-                  case ">":
-                    compareResult = stateValue > compareValue;
-                    break;
-                }
-                return {
-                  node: t.binaryExpression(operator, (0, _node.deepClone)(stateVars[stateVarIndex]), (0, _node.numericLiteral)(compareValue)),
-                  value: compareResult
-                };
-              }
-            }, {
-              key: "identifier",
-              value: function identifier(identifierName, scopeManager) {
-                if (this.withDiscriminant && this.withDiscriminant === scopeManager) {
-                  var id = t.identifier(identifierName);
-                  id[_constants.NO_RENAME] = cffIndex;
-                  id[_constants.WITH_STATEMENT] = true;
-                  return id;
-                }
-                return scopeManager.getMemberExpression(identifierName);
+              var assertedValue = exprs[1].evaluate();
+              if (!assertedValue.confident) {
+                throw new Error("Asserted sequence expression must be statically evaluable");
               }
-            }]);
-          }();
-          /**
-           * Stage 1: Flatten the code into Basic Blocks
-           *
-           * This involves transforming the Control Flow / Scopes into blocks with 'goto' statements
-           *
-           * - A block is simply a sequence of statements
-           * - A block can have a 'goto' statement to another block
-           * - A block original scope is preserved
-           *
-           * 'goto' & Scopes are transformed in Stage 2
-           */
-          var switchLabel = me.getPlaceholder();
-          var breakStatement = function breakStatement() {
-            return t.breakStatement(t.identifier(switchLabel));
-          };
-          var startLabel = me.getPlaceholder();
-          var endLabel = me.getPlaceholder();
-          var currentBasicBlock = new BasicBlock(startLabel, blockPath);
-          currentBasicBlock.allowWithDiscriminant = false;
-          var gotoFunctionName = "GOTO__" + me.getPlaceholder() + "__IF_YOU_CAN_READ_THIS_THERE_IS_A_BUG";
-          function GotoControlStatement(label) {
-            return new _template["default"]("\n              ".concat(gotoFunctionName, "(\"").concat(label, "\");\n              ")).single();
-          }
-
-          // Ends the current block and starts a new one
-          function endCurrentBasicBlock() {
-            var _ref2 = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {},
-              _ref2$jumpToNext = _ref2.jumpToNext,
-              jumpToNext = _ref2$jumpToNext === void 0 ? true : _ref2$jumpToNext,
-              _ref2$nextLabel = _ref2.nextLabel,
-              nextLabel = _ref2$nextLabel === void 0 ? me.getPlaceholder() : _ref2$nextLabel,
-              _ref2$prevJumpTo = _ref2.prevJumpTo,
-              prevJumpTo = _ref2$prevJumpTo === void 0 ? null : _ref2$prevJumpTo,
-              _ref2$nextBlockPath = _ref2.nextBlockPath,
-              nextBlockPath = _ref2$nextBlockPath === void 0 ? null : _ref2$nextBlockPath;
-            (0, _assert.ok)(nextBlockPath);
-            if (prevJumpTo) {
-              currentBasicBlock.insertAfter(GotoControlStatement(prevJumpTo));
-            } else if (jumpToNext) {
-              currentBasicBlock.insertAfter(GotoControlStatement(nextLabel));
+              (0, _assert.ok)(typeof assertedValue.value === "number", "Asserted sequence expression must evaluate to a number");
+              path.replaceWith(t.assignmentExpression("=", (0, _node.deepClone)(runtimeVar), exprs[0].node));
             }
-            currentBasicBlock = new BasicBlock(nextLabel, nextBlockPath);
-          }
-          var prependNodes = [];
-          var functionExpressions = [];
-          function flattenIntoBasicBlocks(bodyIn) {
-            // if (!Array.isArray(bodyIn) && bodyIn.isBlock()) {
-            //   currentBasicBlock.parentPath = bodyIn;
-            // }
-            var body = Array.isArray(bodyIn) ? bodyIn : bodyIn.get("body");
-            var nextBlockPath = Array.isArray(bodyIn) ? currentBasicBlock.parentPath : bodyIn;
-            var _loop = function _loop() {
-                var statement = body[index];
-
-                // Keep Imports before everything else
-                if (statement.isImportDeclaration()) {
-                  prependNodes.push(statement.node);
-                  return 0; // continue
-                }
-                if (statement.isFunctionDeclaration()) {
-                  var fnName = statement.node.id.name;
-                  var isIllegal = false;
-                  if (!flattenFunctionDeclarations || statement.node.async || statement.node.generator || statement.node[_constants.UNSAFE] || statement.node[CFF_UNSAFE] || (0, _astUtils.isStrictMode)(statement)) {
-                    isIllegal = true;
-                  }
-                  var oldBasicBlock = currentBasicBlock;
-                  var _fnLabel = me.getPlaceholder();
-                  var sm = currentBasicBlock.scopeManager;
-                  var rename = sm.getNewName(fnName);
-                  sm.scope.bindings[fnName].kind = "var";
-                  var hoistedBasicBlock = Array.from(basicBlocks.values()).find(function (block) {
-                    return block.parentPath === currentBasicBlock.parentPath;
-                  });
-                  if (isIllegal) {
-                    hoistedBasicBlock.body.unshift(statement.node);
-                    return 0; // continue
-                  }
-                  me.changeData.functions++;
-                  var functionExpression = t.functionExpression(null, [], t.blockStatement([]));
-                  functionExpressions.push([fnName, _fnLabel, currentBasicBlock, functionExpression]);
-
-                  // Change the function declaration to a variable declaration
-                  hoistedBasicBlock.body.unshift(t.variableDeclaration("var", [t.variableDeclarator(t.identifier(fnName), functionExpression)]));
-                  var blockStatement = statement.get("body");
-                  endCurrentBasicBlock({
-                    nextLabel: _fnLabel,
-                    nextBlockPath: blockStatement,
-                    jumpToNext: false
-                  });
-                  var fnTopBlock = currentBasicBlock;
-
-                  // Implicit return
-                  blockStatement.node.body.push(t.returnStatement(t.identifier("undefined")));
-                  flattenIntoBasicBlocks(blockStatement);
-                  scopeToScopeManager.get(statement.scope).requiresInitializing = false;
-                  basicBlocks.get(_fnLabel).allowWithDiscriminant = false;
-
-                  // Debug label
-                  if (isDebug) {
-                    fnTopBlock.body.unshift(t.expressionStatement(t.stringLiteral("Function " + statement.node.id.name + " -> Renamed to " + rename)));
-                  }
-
-                  // Unpack parameters from the parameter 'argVar'
-                  if (statement.node.params.length > 0) {
-                    usedArgVar = true;
-                    fnTopBlock.body.unshift(t.variableDeclaration("var", [t.variableDeclarator(t.arrayPattern(statement.node.params), (0, _node.deepClone)(argVar))]));
-
-                    // Change bindings from 'param' to 'var'
-                    statement.get("params").forEach(function (param) {
-                      var ids = param.getBindingIdentifierPaths();
-                      // Loop over the record of binding identifiers
-                      for (var identifierName in ids) {
-                        var identifierPath = ids[identifierName];
-                        if (identifierPath.getFunctionParent() === statement) {
-                          var binding = statement.scope.getBinding(identifierName);
-                          if (binding) {
-                            binding.kind = "var";
-                          }
-                        }
-                      }
-                    });
-                  }
-                  currentBasicBlock = oldBasicBlock;
-                  return 0; // continue
-                }
-
-                // Convert IF statements into Basic Blocks
-                if (statement.isIfStatement() && flattenIfStatements) {
-                  var test = statement.get("test");
-                  var consequent = statement.get("consequent");
-                  var alternate = statement.get("alternate");
-
-                  // Both consequent and alternate are blocks
-                  if (consequent.isBlockStatement() && (!alternate.node || alternate.isBlockStatement())) {
-                    me.changeData.ifStatements++;
-                    var consequentLabel = me.getPlaceholder();
-                    var alternateLabel = alternate.node ? me.getPlaceholder() : null;
-                    var afterPath = me.getPlaceholder();
-                    currentBasicBlock.insertAfter(t.ifStatement(test.node, GotoControlStatement(consequentLabel), alternateLabel ? GotoControlStatement(alternateLabel) : GotoControlStatement(afterPath)));
-                    var _oldBasicBlock = currentBasicBlock;
-                    endCurrentBasicBlock({
-                      jumpToNext: false,
-                      nextLabel: consequentLabel,
-                      nextBlockPath: consequent
-                    });
-                    flattenIntoBasicBlocks(consequent);
-                    currentBasicBlock.initializedScope = _oldBasicBlock.scopeManager;
-                    if (alternate.isBlockStatement()) {
-                      endCurrentBasicBlock({
-                        prevJumpTo: afterPath,
-                        nextLabel: alternateLabel,
-                        nextBlockPath: alternate
-                      });
-                      flattenIntoBasicBlocks(alternate);
-                    }
-                    endCurrentBasicBlock({
-                      prevJumpTo: afterPath,
-                      nextLabel: afterPath,
-                      nextBlockPath: _oldBasicBlock.parentPath
-                    });
-                    return 0; // continue
-                  }
-                }
-                if (Number(index) === body.length - 1 && statement.isExpressionStatement() && statement.findParent(function (p) {
-                  return p.isBlock();
-                }) === blockPath && programPath // <- ONLY APPLY TO TOP-LEVEL NEVER FUNCTIONS!
-                ) {
-                  // Return the result of the last expression for eval() purposes
-                  currentBasicBlock.insertAfter(t.returnStatement(statement.get("expression").node));
-                  return 0; // continue
-                }
-
-                // 3 or more statements should be split more
-                if (currentBasicBlock.body.length > 1 && (0, _randomUtils.chance)(50 + currentBasicBlock.body.length)) {
-                  endCurrentBasicBlock({
-                    nextBlockPath: nextBlockPath
-                  });
-                }
-
-                // console.log(currentBasicBlock.thisPath.type);
-                // console.log(currentBasicBlock.body);
-                currentBasicBlock.body.push(statement.node);
-              },
-              _ret;
-            for (var index in body) {
-              _ret = _loop();
-              if (_ret === 0) continue;
+          },
+          BooleanLiteral: {
+            exit: function exit(boolPath) {
+              // Don't mangle booleans in debug mode
+              if (isDebug || !mangleBooleanLiterals || me.isSkipped(boolPath)) return;
+              if (!isWithinSameFunction(boolPath)) return;
+              mangledLiteralsCreated++;
+              var index = (0, _randomUtils.getRandomInteger)(0, stateVars.length - 1);
+              var stateVar = stateVars[index];
+              var stateVarValue = currentStateValues[index];
+              var compareValue = (0, _randomUtils.choice)([(0, _randomUtils.getRandomInteger)(-250, 250), stateVarValue]);
+              var compareResult = stateVarValue === compareValue;
+              var newExpression = t.binaryExpression(boolPath.node.value === compareResult ? "==" : "!=", (0, _node.deepClone)(stateVar), (0, _node.numericLiteral)(compareValue));
+              (0, _astUtils.ensureComputedExpression)(boolPath);
+              boolPath.replaceWith(newExpression);
             }
-          }
+          },
+          // Mangle numbers with the state values
+          NumericLiteral: {
+            exit: function exit(numPath) {
+              // Don't mangle numbers in debug mode
+              if (isDebug || !mangleNumericalLiterals || me.isSkipped(numPath)) return;
+              var num = numPath.node.value;
+              if (Math.floor(num) !== num || Math.abs(num) > 100000 || !Number.isFinite(num) || Number.isNaN(num)) return;
+              if (!isWithinSameFunction(numPath)) return;
+              if ((0, _randomUtils.chance)(50)) return;
+              mangledLiteralsCreated++;
+              var index = (0, _randomUtils.getRandomInteger)(0, stateVars.length - 1);
+              var stateVar = stateVars[index];
+
+              // num = 50
+              // stateVar = 30
+              // stateVar + 30
+
+              var diff = t.binaryExpression("+", (0, _node.deepClone)(stateVar), me.skip((0, _node.numericLiteral)(num - currentStateValues[index])));
+              (0, _astUtils.ensureComputedExpression)(numPath);
+              numPath.replaceWith(diff);
+              numPath.skip();
+            }
+          },
+          // Xor encrypt string literals
+          StringLiteral: {
+            exit: function exit(strPath) {
+              // Don't mangle strings in debug mode
+              if (isDebug || !mangleStringLiterals || me.isSkipped(strPath)) return;
+
+              // Don't accidentally break goto call expressions
+              var p = strPath.parentPath;
+              if (p !== null && p !== void 0 && p.isCallExpression() && t.isIdentifier(p.node.callee) && p.node.callee.name === gotoFunctionName) {
+                return;
+              }
+              var str = strPath.node.value;
+              if (str.length > 5000) return;
+              if (!isWithinSameFunction(strPath)) return;
+              if ((0, _randomUtils.chance)(50)) return;
+              var num = (0, _randomUtils.getRandomInteger)(-255, 255); // XOR string encrypt key
 
-          // Convert our code into Basic Blocks
-          flattenIntoBasicBlocks(blockPath.get("body"));
+              var encryptedStr = (0, _xorStringTemplate.xorEncodeString)(str, num);
+              var decryptedStr = (0, _xorStringTemplate.xorDecodeString)(encryptedStr, num);
 
-          // Ensure always jumped to the Program end
-          endCurrentBasicBlock({
-            jumpToNext: true,
-            nextLabel: endLabel,
-            nextBlockPath: defaultBlockPath
-          });
-          basicBlocks.get(endLabel).allowWithDiscriminant = false;
-          if (!isDebug && addDeadCode) {
-            // DEAD CODE 1/3: Add fake chunks that are never reached
-            var fakeChunkCount = (0, _randomUtils.getRandomInteger)(1, 5);
-            for (var i = 0; i < fakeChunkCount; i++) {
-              // These chunks just jump somewhere random, they are never executed
-              // so it could contain any code
-              var fakeBlock = new BasicBlock(me.getPlaceholder(), blockPath, {
-                impossible: true
-              });
-              var fakeJump = void 0;
-              do {
-                fakeJump = (0, _randomUtils.choice)(Array.from(basicBlocks.keys()));
-              } while (fakeJump === fakeBlock.label);
-              fakeBlock.insertAfter(GotoControlStatement(fakeJump));
-              me.changeData.deadCode++;
+              // If decryption fails for some reason, skip
+              if (decryptedStr !== str) {
+                return;
+              }
+              mangledLiteralsCreated++;
+              var index = (0, _randomUtils.getRandomInteger)(0, stateVars.length - 1);
+              var stateVar = stateVars[index];
+
+              // Equals 'num' but entangled with current state var
+              var diff = t.binaryExpression("+", (0, _node.deepClone)(stateVar), me.skip((0, _node.numericLiteral)(num - currentStateValues[index])));
+              var newExpr = t.callExpression(t.identifier(xorFnName), [t.stringLiteral(encryptedStr), diff]);
+              (0, _astUtils.ensureComputedExpression)(strPath);
+              strPath.replaceWith(newExpr);
+              strPath.skip();
             }
-
-            // DEAD CODE 2/3: Add fake jumps to really mess with deobfuscators
-            // "irreducible control flow"
-            basicBlocks.forEach(function (basicBlock) {
-              if ((0, _randomUtils.chance)(30 - basicBlocks.size)) {
-                var randomLabel = (0, _randomUtils.choice)(Array.from(basicBlocks.keys()));
-
-                // The `false` literal will be mangled
-                basicBlock.insertAfter(new _template["default"]("\n                    if({predicate}){\n                      {goto}\n                    }\n                    ").single({
-                  "goto": GotoControlStatement(randomLabel),
-                  predicate: basicBlock.createFalsePredicate()
-                }));
-                me.changeData.deadCode++;
+          },
+          Identifier: {
+            exit: function exit(path) {
+              if (!(0, _astUtils.isVariableIdentifier)(path)) return;
+              if (me.isSkipped(path)) return;
+              if (path.node[_constants.NO_RENAME] === cffIndex) return;
+              // For identifiers using implicit with discriminant, skip
+              if (path.node[_constants.WITH_STATEMENT]) return;
+              var identifierName = path.node.name;
+              if (identifierName === gotoFunctionName) return;
+              var binding = path.scope.getBinding(identifierName);
+              if (!binding) {
+                return;
               }
-            });
-            // DEAD CODE 3/3: Clone chunks but these chunks are never ran
-            var cloneChunkCount = (0, _randomUtils.getRandomInteger)(1, 5);
-            var _loop2 = function _loop2() {
-              var randomChunk = (0, _randomUtils.choice)(Array.from(basicBlocks.values()));
-
-              // Don't double define functions
-              var hasDeclaration = randomChunk.body.find(function (stmt) {
-                return t.isDeclaration(stmt);
-              });
-              if (!hasDeclaration) {
-                var clonedChunk = new BasicBlock(me.getPlaceholder(), randomChunk.parentPath, {
-                  impossible: true
-                });
-                randomChunk.thisNode.body.map(function (x) {
-                  return (0, _node.deepClone)(x);
-                }).forEach(function (node) {
-                  if (node.type === "EmptyStatement") return;
-                  clonedChunk.insertAfter(node);
-                });
-                me.changeData.deadCode++;
+              if (binding.kind === "var" || binding.kind === "let" || binding.kind === "const") {} else {
+                return;
               }
-            };
-            for (var _i2 = 0; _i2 < cloneChunkCount; _i2++) {
-              _loop2();
-            }
-          }
-
-          // Select scope managers for the with statement
-          var _iterator = _createForOfIteratorHelper(basicBlocks.values()),
-            _step;
-          try {
-            for (_iterator.s(); !(_step = _iterator.n()).done;) {
-              var _basicBlock2$initiali;
-              var _basicBlock2 = _step.value;
-              _basicBlock2.bestWithDiscriminant = (_basicBlock2$initiali = _basicBlock2.initializedScope) === null || _basicBlock2$initiali === void 0 ? void 0 : _basicBlock2$initiali.findBestWithDiscriminant(_basicBlock2);
-              if (isDebug && _basicBlock2.withDiscriminant) {
-                _basicBlock2.body.unshift(t.expressionStatement(t.stringLiteral("With " + _basicBlock2.withDiscriminant.propertyName)));
+              var scopeManager = scopeToScopeManager.get(binding.scope);
+              if (!scopeManager) return;
+              var newName = scopeManager.getNewName(identifierName, path.node);
+              var memberExpression = scopeManager.getMemberExpression(newName);
+              scopeManager.isNotUsed = false;
+              if ((0, _astUtils.isDefiningIdentifier)(path)) {
+                (0, _astUtils.replaceDefiningIdentifierToMemberExpression)(path, memberExpression);
+                return;
               }
-            }
-
-            /**
-             * Stage 2: Transform 'goto' statements into valid JavaScript
-             *
-             * - 'goto' is replaced with equivalent state updates and break statements
-             * - Original identifiers are converted into member expressions
-             */
-
-            // Remap 'GotoStatement' to actual state assignments and Break statements
-          } catch (err) {
-            _iterator.e(err);
-          } finally {
-            _iterator.f();
-          }
-          var _iterator2 = _createForOfIteratorHelper(basicBlocks.values()),
-            _step2;
-          try {
-            var _loop4 = function _loop4() {
-              var basicBlock = _step2.value;
-              var currentStateValues = basicBlock.stateValues;
-              // Wrap the statement in a Babel path to allow traversal
-
-              var outerFn = (0, _astUtils.getParentFunctionOrProgram)(basicBlock.parentPath);
-              function isWithinSameFunction(path) {
-                var fn = (0, _astUtils.getParentFunctionOrProgram)(path);
-                return fn.node === outerFn.node;
+              if (!path.container) return;
+              me.skip(memberExpression);
+              path.replaceWith(memberExpression);
+              path.skip();
+
+              // Preserve proper 'this' context when directly calling functions
+              // X.Y.Z() -> (1, X.Y.Z)()
+              if (path.parentPath.isCallExpression() && path.key === "callee") {
+                path.replaceWith(t.sequenceExpression([t.numericLiteral(1), path.node]));
               }
-              var visitor = {
-                BooleanLiteral: {
-                  exit: function exit(boolPath) {
-                    // Don't mangle booleans in debug mode
-                    if (isDebug || !mangleBooleanLiterals || me.isSkipped(boolPath)) return;
-                    if (!isWithinSameFunction(boolPath)) return;
-                    if ((0, _randomUtils.chance)(50 + mangledLiteralsCreated)) return;
-                    mangledLiteralsCreated++;
-                    var index = (0, _randomUtils.getRandomInteger)(0, stateVars.length - 1);
-                    var stateVar = stateVars[index];
-                    var stateVarValue = currentStateValues[index];
-                    var compareValue = (0, _randomUtils.choice)([(0, _randomUtils.getRandomInteger)(-250, 250), stateVarValue]);
-                    var compareResult = stateVarValue === compareValue;
-                    var newExpression = t.binaryExpression(boolPath.node.value === compareResult ? "==" : "!=", (0, _node.deepClone)(stateVar), (0, _node.numericLiteral)(compareValue));
-                    (0, _astUtils.ensureComputedExpression)(boolPath);
-                    boolPath.replaceWith(newExpression);
-                  }
-                },
-                // Mangle numbers with the state values
-                NumericLiteral: {
-                  exit: function exit(numPath) {
-                    // Don't mangle numbers in debug mode
-                    if (isDebug || !mangleNumericalLiterals || me.isSkipped(numPath)) return;
-                    var num = numPath.node.value;
-                    if (Math.floor(num) !== num || Math.abs(num) > 100000 || !Number.isFinite(num) || Number.isNaN(num)) return;
-                    if (!isWithinSameFunction(numPath)) return;
-                    if ((0, _randomUtils.chance)(50 + mangledLiteralsCreated)) return;
-                    mangledLiteralsCreated++;
-                    var index = (0, _randomUtils.getRandomInteger)(0, stateVars.length - 1);
-                    var stateVar = stateVars[index];
-
-                    // num = 50
-                    // stateVar = 30
-                    // stateVar + 30
-
-                    var diff = t.binaryExpression("+", (0, _node.deepClone)(stateVar), me.skip((0, _node.numericLiteral)(num - currentStateValues[index])));
-                    (0, _astUtils.ensureComputedExpression)(numPath);
-                    numPath.replaceWith(diff);
-                    numPath.skip();
-                  }
-                },
-                Identifier: {
-                  exit: function exit(path) {
-                    if (!(0, _astUtils.isVariableIdentifier)(path)) return;
-                    if (me.isSkipped(path)) return;
-                    if (path.node[_constants.NO_RENAME] === cffIndex) return;
-                    // For identifiers using implicit with discriminant, skip
-                    if (path.node[_constants.WITH_STATEMENT]) return;
-                    var identifierName = path.node.name;
-                    if (identifierName === gotoFunctionName) return;
-                    var binding = path.scope.getBinding(identifierName);
-                    if (!binding) {
-                      return;
-                    }
-                    if (binding.kind === "var" || binding.kind === "let" || binding.kind === "const") {} else {
-                      return;
-                    }
-                    var scopeManager = scopeToScopeManager.get(binding.scope);
-                    if (!scopeManager) return;
-                    var newName = scopeManager.getNewName(identifierName, path.node);
-                    var memberExpression = scopeManager.getMemberExpression(newName);
-                    scopeManager.isNotUsed = false;
-
-                    // Scope object as with discriminant? Use identifier
-                    if (typeof basicBlock.withDiscriminant === "undefined") {
-                      var id = t.identifier(scopeManager.propertyName);
-                      id[_constants.WITH_STATEMENT] = true;
-                      id[_constants.NO_RENAME] = cffIndex;
-                      memberExpression = scopeManager.getMemberExpression(newName, id);
-                    }
-                    if ((0, _astUtils.isDefiningIdentifier)(path)) {
-                      (0, _astUtils.replaceDefiningIdentifierToMemberExpression)(path, memberExpression);
-                      return;
-                    }
-                    if (!path.container) return;
-                    if (basicBlock.withDiscriminant && basicBlock.withDiscriminant === scopeManager && basicBlock.withDiscriminant.hasOwnName(identifierName)) {
-                      // The defining mode must directly append to the scope object
-                      // Subsequent uses can use the identifier
-                      var isDefiningNode = path.node === binding.identifier;
-                      if (!isDefiningNode) {
-                        memberExpression = basicBlock.identifier(newName, scopeManager);
-                      }
-                    }
-                    me.skip(memberExpression);
-                    path.replaceWith(memberExpression);
-                    path.skip();
-
-                    // Preserve proper 'this' context when directly calling functions
-                    // X.Y.Z() -> (1, X.Y.Z)()
-                    if (path.parentPath.isCallExpression() && path.key === "callee") {
-                      path.replaceWith(t.sequenceExpression([t.numericLiteral(1), path.node]));
-                    }
-                  }
-                },
-                // Top-level returns set additional flag to indicate that the function has returned
-                ReturnStatement: {
-                  exit: function exit(path) {
-                    var functionParent = path.getFunctionParent();
-                    if (!functionParent || functionParent.get("body") !== blockPath) return;
-                    var returnArgument = path.node.argument || t.identifier("undefined");
-                    path.node.argument = new _template["default"]("\n                ({didReturnVar} = true, {returnArgument})\n                  ").expression({
-                      returnArgument: returnArgument,
-                      didReturnVar: (0, _node.deepClone)(_didReturnVar)
-                    });
-                  }
-                },
-                // goto() calls are replaced with state updates and break statements
-                CallExpression: {
-                  exit: function exit(path) {
-                    if (t.isIdentifier(path.node.callee) && path.node.callee.name === gotoFunctionName) {
-                      var _path$node$arguments = _slicedToArray(path.node.arguments, 1),
-                        labelNode = _path$node$arguments[0];
-                      (0, _assert.ok)(t.isStringLiteral(labelNode));
-                      var _label = labelNode.value;
-                      var jumpBlock = basicBlocks.get(_label);
-                      (0, _assert.ok)(jumpBlock, "Label not found: " + _label);
-                      var newStateValues = jumpBlock.stateValues,
-                        newTotalState = jumpBlock.totalState;
-                      var assignments = [];
-                      if (jumpBlock.withDiscriminant) {
-                        assignments.push(t.assignmentExpression("=", (0, _node.deepClone)(withMemberExpression), jumpBlock.withDiscriminant.getScopeObject()));
-                      } else if (basicBlock.withDiscriminant) {
-                        // Reset the with discriminant to undefined using fake property
-                        // scope["fake"] -> undefined
-
-                        var fakeProperty = scopeNameGen.generate();
-                        assignments.push(t.assignmentExpression("=", (0, _node.deepClone)(withMemberExpression), t.memberExpression((0, _node.deepClone)(scopeVar), t.stringLiteral(fakeProperty), true)));
-                      }
-                      var mutatingStateValues = _toConsumableArray(currentStateValues);
-                      var _loop5 = function _loop5(_i9) {
-                        var oldValue = currentStateValues[_i9];
-                        var newValue = newStateValues[_i9];
-
-                        // console.log(oldValue, newValue);
-                        if (oldValue === newValue) return 1; // continue
-                        // No diff needed if the value doesn't change
-
-                        var assignment = t.assignmentExpression("=", (0, _node.deepClone)(stateVars[_i9]), (0, _node.numericLiteral)(newValue));
-                        if (!isDebug && addRelativeAssignments) {
-                          // Use two level of diffs to create confusing code:
-                          // stateVar += anotherStateVar - diff
-                          // has the effect of:
-                          // stateVar = newState
-                          var mangledNumericLiteral = function mangledNumericLiteral(value) {
-                            var stateVarIndex;
-                            do {
-                              stateVarIndex = (0, _randomUtils.getRandomInteger)(0, stateVars.length);
-                            } while (stateVarIndex === _i9 && stateVars.length > 1);
-                            var stateVarId = stateVars[stateVarIndex];
-                            var stateVarValue = mutatingStateValues[stateVarIndex];
-                            var diff = stateVarValue - value;
-                            return t.binaryExpression("-", (0, _node.deepClone)(stateVarId), (0, _node.numericLiteral)(diff));
-                          };
-                          assignment = t.assignmentExpression("+=", (0, _node.deepClone)(stateVars[_i9]), mangledNumericLiteral(newValue - oldValue));
-                        }
-                        mutatingStateValues[_i9] = newValue;
-                        assignments.push(assignment);
-                      };
-                      for (var _i9 = 0; _i9 < stateVars.length; _i9++) {
-                        if (_loop5(_i9)) continue;
-                      }
-
-                      // Add debug label
-                      if (isDebug) {
-                        assignments.unshift(t.stringLiteral("Goto " + newTotalState));
-                      }
-                      path.parentPath.replaceWith(t.expressionStatement(t.sequenceExpression(assignments)))[0].skip();
-
-                      // Add break after updating state variables
-                      path.insertAfter(breakStatement());
-                    }
-                  }
-                }
-              };
-              basicBlock.thisPath.traverse(visitor);
-            };
-            for (_iterator2.s(); !(_step2 = _iterator2.n()).done;) {
-              _loop4();
             }
-
-            /**
-             * Stage 3: Create a switch statement to handle the control flow
-             *
-             * - Add fake / impossible blocks
-             * - Add fake / predicates to the switch cases tests
-             */
-
-            // Create global numbers for predicates
-          } catch (err) {
-            _iterator2.e(err);
-          } finally {
-            _iterator2.f();
-          }
-          var mainScope = basicBlocks.get(startLabel).scopeManager;
-          var predicateNumbers = new Map();
-          var predicateNumberCount = isDebug || !addPredicateTests ? 0 : (0, _randomUtils.getRandomInteger)(1, 4);
-          for (var _i3 = 0; _i3 < predicateNumberCount; _i3++) {
-            var name = mainScope.getNewName(me.getPlaceholder("predicate_" + _i3));
-            var number = (0, _randomUtils.getRandomInteger)(-250, 250);
-            predicateNumbers.set(name, number);
-          }
-          var predicateSymbol = Symbol("predicate");
-          var createAssignment = function createAssignment(values) {
-            var exprStmt = new _template["default"]("\n              ({predicateVariables} = {values})\n              ").single({
-              predicateVariables: t.arrayPattern(Array.from(predicateNumbers.keys()).map(function (name) {
-                return mainScope.getMemberExpression(name);
-              })),
-              values: t.arrayExpression(values.map(function (value) {
-                return (0, _node.numericLiteral)(value);
-              }))
-            });
-            exprStmt[predicateSymbol] = true;
-            return exprStmt;
-          };
-          basicBlocks.get(startLabel).body.unshift(createAssignment(Array.from(predicateNumbers.values())));
-
-          // Add random assignments to impossible blocks
-          var fakeAssignmentCount = (0, _randomUtils.getRandomInteger)(1, 3);
-          for (var _i4 = 0; _i4 < fakeAssignmentCount; _i4++) {
-            var impossibleBlock = (0, _randomUtils.choice)(getImpossibleBasicBlocks());
-            if (impossibleBlock) {
-              var _impossibleBlock$body;
-              if ((_impossibleBlock$body = impossibleBlock.body[0]) !== null && _impossibleBlock$body !== void 0 && _impossibleBlock$body[predicateSymbol]) continue;
-              var fakeValues = new Array(predicateNumberCount).fill(0).map(function () {
-                return (0, _randomUtils.getRandomInteger)(-250, 250);
+          },
+          // Top-level returns set additional flag to indicate that the function has returned
+          ReturnStatement: {
+            exit: function exit(path) {
+              var functionParent = path.getFunctionParent();
+              if (!functionParent || functionParent.get("body") !== blockPath) return;
+              var returnArgument = path.node.argument || t.identifier("undefined");
+              path.node.argument = new _template["default"]("\n                ({didReturnVar} = true, {returnArgument})\n                  ").expression({
+                returnArgument: returnArgument,
+                didReturnVar: (0, _node.deepClone)(_didReturnVar)
               });
-              impossibleBlock.body.unshift(createAssignment(fakeValues));
-            }
-          }
-
-          // Add scope initializations: scope["_0"] = {identity: "_0"}
-          var _iterator3 = _createForOfIteratorHelper(scopeToScopeManager.values()),
-            _step3;
-          try {
-            for (_iterator3.s(); !(_step3 = _iterator3.n()).done;) {
-              var _scopeManager = _step3.value;
-              if (_scopeManager.isNotUsed) continue;
-              if (!_scopeManager.requiresInitializing) continue;
-              if (_scopeManager.initializingBasicBlock.label === startLabel) continue;
-              _scopeManager.initializingBasicBlock.body.unshift(_scopeManager.getInitializingStatement());
-            }
-          } catch (err) {
-            _iterator3.e(err);
-          } finally {
-            _iterator3.f();
-          }
-          var switchCases = [];
-          var blocks = Array.from(basicBlocks.values());
-          if (!isDebug && addFakeTests) {
-            (0, _randomUtils.shuffle)(blocks);
-          }
-          var _loop3 = function _loop3() {
-            var block = _blocks[_i5];
-            if (block.label === endLabel) {
-              // ok(block.body.length === 0);
-              return 1; // continue
-            }
-            var test = (0, _node.numericLiteral)(block.totalState);
-
-            // Predicate tests cannot apply to the start label
-            // As that's when the numbers are initialized
-            if (!isDebug && addPredicateTests && block.label !== startLabel && (0, _randomUtils.chance)(50)) {
-              var predicateName = (0, _randomUtils.choice)(Array.from(predicateNumbers.keys()));
-              if (predicateName) {
-                var _number = predicateNumbers.get(predicateName);
-                var diff = block.totalState - _number;
-                test = t.binaryExpression("+", mainScope.getMemberExpression(predicateName), (0, _node.numericLiteral)(diff));
-              }
             }
-
-            // Add complex tests
-            if (!isDebug && addComplexTests && (0, _randomUtils.chance)(50)) {
-              // Create complex test expressions for each switch case
-
-              // case STATE+X:
-              var stateVarIndex = (0, _randomUtils.getRandomInteger)(0, stateVars.length);
-              var stateValues = block.stateValues;
-              var difference = stateValues[stateVarIndex] - block.totalState;
-              var conditionNodes = [];
-              var alreadyConditionedItems = new Set();
-
-              // This code finds clash conditions and adds them to 'conditionNodes' array
-              Array.from(basicBlocks.keys()).forEach(function (label) {
-                if (label !== block.label) {
-                  var labelStates = basicBlocks.get(label).stateValues;
-                  var totalState = labelStates.reduce(function (a, b) {
-                    return a + b;
-                  }, 0);
-                  if (totalState === labelStates[stateVarIndex] - difference) {
-                    var differentIndex = labelStates.findIndex(function (v, i) {
-                      return v !== stateValues[i];
-                    });
-                    if (differentIndex !== -1) {
-                      var expressionAsString = stateVars[differentIndex].name + "!=" + labelStates[differentIndex];
-                      if (!alreadyConditionedItems.has(expressionAsString)) {
-                        alreadyConditionedItems.add(expressionAsString);
-                        conditionNodes.push(t.binaryExpression("!=", (0, _node.deepClone)(stateVars[differentIndex]), (0, _node.numericLiteral)(labelStates[differentIndex])));
-                      }
-                    } else {
-                      conditionNodes.push(t.binaryExpression("!=", (0, _node.deepClone)(discriminant), (0, _node.numericLiteral)(totalState)));
-                    }
+          },
+          // goto() calls are replaced with state updates and break statements
+          CallExpression: {
+            exit: function exit(path) {
+              if (t.isIdentifier(path.node.callee) && path.node.callee.name === gotoFunctionName) {
+                var _path$node$arguments = _slicedToArray(path.node.arguments, 1),
+                  labelNode = _path$node$arguments[0];
+                (0, _assert.ok)(t.isStringLiteral(labelNode));
+                var _label = labelNode.value;
+                var jumpBlock = basicBlocks.get(_label);
+                (0, _assert.ok)(jumpBlock, "Label not found: " + _label);
+                var newStateValues = jumpBlock.stateValues,
+                  newTotalState = jumpBlock.totalState;
+                var assignments = [];
+                var mutatingStateValues = _toConsumableArray(currentStateValues);
+                var _loop5 = function _loop5(_i8) {
+                  var oldValue = currentStateValues[_i8];
+                  var newValue = newStateValues[_i8];
+
+                  // console.log(oldValue, newValue);
+                  if (oldValue === newValue) return 1; // continue
+                  // No diff needed if the value doesn't change
+
+                  var assignment = t.assignmentExpression("=", (0, _node.deepClone)(stateVars[_i8]), (0, _node.numericLiteral)(newValue));
+                  if (!isDebug && addRelativeAssignments) {
+                    // Use two level of diffs to create confusing code:
+                    // stateVar += anotherStateVar - diff
+                    // has the effect of:
+                    // stateVar = newState
+                    var mangledNumericLiteral = function mangledNumericLiteral(value) {
+                      var stateVarIndex;
+                      do {
+                        stateVarIndex = (0, _randomUtils.getRandomInteger)(0, stateVars.length);
+                      } while (stateVarIndex === _i8 && stateVars.length > 1);
+                      var stateVarId = stateVars[stateVarIndex];
+                      var stateVarValue = mutatingStateValues[stateVarIndex];
+                      var diff = stateVarValue - value;
+                      return t.binaryExpression("-", (0, _node.deepClone)(stateVarId), (0, _node.numericLiteral)(diff));
+                    };
+                    assignment = t.assignmentExpression("+=", (0, _node.deepClone)(stateVars[_i8]), mangledNumericLiteral(newValue - oldValue));
                   }
+                  mutatingStateValues[_i8] = newValue;
+                  assignments.push(assignment);
+                };
+                for (var _i8 = 0; _i8 < stateVars.length; _i8++) {
+                  if (_loop5(_i8)) continue;
                 }
-              });
 
-              // case STATE!=Y && STATE+X
-              test = t.binaryExpression("-", (0, _node.deepClone)(stateVars[stateVarIndex]), (0, _node.numericLiteral)(difference));
+                // Add debug label
+                if (isDebug) {
+                  assignments.unshift(t.stringLiteral("Goto " + newTotalState));
+                }
+                path.parentPath.replaceWith(t.expressionStatement(t.sequenceExpression(assignments)))[0].skip();
 
-              // Use the 'conditionNodes' to not cause state clashing issues
-              conditionNodes.forEach(function (conditionNode) {
-                test = t.logicalExpression("&&", conditionNode, test);
-              });
-            }
-            var tests = [test];
-            if (!isDebug && addFakeTests && (0, _randomUtils.chance)(50)) {
-              // Add fake tests
-              var fakeTestCount = (0, _randomUtils.getRandomInteger)(1, 3);
-              for (var _i6 = 0; _i6 < fakeTestCount; _i6++) {
-                tests.push((0, _node.numericLiteral)(stateIntGen.generate()));
+                // Add break after updating state variables
+                path.insertAfter(breakStatement());
               }
-              (0, _randomUtils.shuffle)(tests);
             }
-            var lastTest = tests.pop();
-            for (var _i7 = 0, _tests = tests; _i7 < _tests.length; _i7++) {
-              var _test = _tests[_i7];
-              switchCases.push(t.switchCase(_test, []));
-            }
-            switchCases.push(t.switchCase(lastTest, block.thisPath.node.body));
-          };
-          for (var _i5 = 0, _blocks = blocks; _i5 < _blocks.length; _i5++) {
-            if (_loop3()) continue;
           }
-          if (!isDebug && addFakeTests) {
-            // A random test can be 'default'
-            (0, _randomUtils.choice)(switchCases).test = null;
-          }
-          var discriminant = new _template["default"]("\n            ".concat(stateVars.map(function (x) {
-            return x.name;
-          }).join(" + "), "\n          ")).expression();
-          (0, _traverse["default"])(t.program([t.expressionStatement(discriminant)]), {
-            Identifier: function Identifier(path) {
-              path.node[_constants.NO_RENAME] = cffIndex;
-            }
-          });
+        };
+        basicBlock.thisPath.traverse(visitor);
+      };
+      for (_iterator.s(); !(_step = _iterator.n()).done;) {
+        _loop4();
+      }
 
-          // Create a new SwitchStatement
-          var switchStatement = t.labeledStatement(t.identifier(switchLabel), t.switchStatement(discriminant, switchCases));
-          var startStateValues = basicBlocks.get(startLabel).stateValues;
-          var endTotalState = basicBlocks.get(endLabel).totalState;
-          var whileStatement = t.whileStatement(t.binaryExpression("!==", (0, _node.deepClone)(discriminant), (0, _node.numericLiteral)(endTotalState)), t.blockStatement([t.withStatement(new _template["default"]("{withDiscriminant} || {scopeVar}").expression({
-            withDiscriminant: (0, _node.deepClone)(withMemberExpression),
-            scopeVar: (0, _node.deepClone)(scopeVar)
-          }), t.blockStatement([switchStatement]))]));
-          var parameters = [].concat(_toConsumableArray(stateVars), [scopeVar, argVar]).map(function (id) {
-            return (0, _node.deepClone)(id);
-          });
-          var parametersNames = parameters.map(function (id) {
-            return id.name;
-          });
-          for (var _i8 = 0, _functionExpressions = functionExpressions; _i8 < _functionExpressions.length; _i8++) {
-            var _functionExpressions$ = _slicedToArray(_functionExpressions[_i8], 4),
-              originalFnName = _functionExpressions$[0],
-              fnLabel = _functionExpressions$[1],
-              basicBlock = _functionExpressions$[2],
-              fn = _functionExpressions$[3];
-            var _basicBlock = basicBlock,
-              scopeManager = _basicBlock.scopeManager;
-            var _basicBlocks$get = basicBlocks.get(fnLabel),
-              stateValues = _basicBlocks$get.stateValues;
-            var argumentsRestName = me.getPlaceholder();
-            var argumentsNodes = [];
-            var _iterator4 = _createForOfIteratorHelper(parametersNames),
-              _step4;
-            try {
-              for (_iterator4.s(); !(_step4 = _iterator4.n()).done;) {
-                var parameterName = _step4.value;
-                var stateIndex = stateVars.map(function (x) {
-                  return x.name;
-                }).indexOf(parameterName);
-                if (stateIndex !== -1) {
-                  argumentsNodes.push((0, _node.numericLiteral)(stateValues[stateIndex]));
-                } else if (parameterName === argVar.name) {
-                  argumentsNodes.push(t.identifier(argumentsRestName));
-                } else if (parameterName === scopeVar.name) {
-                  argumentsNodes.push(scopeManager.getObjectExpression(fnLabel));
-                } else {
-                  (0, _assert.ok)(false);
+      /**
+       * Stage 3: Create a switch statement to handle the control flow
+       *
+       * - Add fake / impossible blocks
+       * - Add fake / predicates to the switch cases tests
+       */
+
+      // Create global numbers for predicates
+    } catch (err) {
+      _iterator.e(err);
+    } finally {
+      _iterator.f();
+    }
+    var mainScope = basicBlocks.get(startLabel).scopeManager;
+    var predicateNumbers = new Map();
+    var predicateNumberCount = isDebug || !addPredicateTests ? 0 : (0, _randomUtils.getRandomInteger)(1, 4);
+    for (var _i2 = 0; _i2 < predicateNumberCount; _i2++) {
+      var name = mainScope.getNewName(me.getPlaceholder("predicate_" + _i2));
+      var number = (0, _randomUtils.getRandomInteger)(-250, 250);
+      predicateNumbers.set(name, number);
+    }
+    var predicateSymbol = Symbol("predicate");
+    var createAssignment = function createAssignment(values) {
+      var exprStmt = new _template["default"]("\n              ({predicateVariables} = {values})\n              ").single({
+        predicateVariables: t.arrayPattern(Array.from(predicateNumbers.keys()).map(function (name) {
+          return mainScope.getMemberExpression(name);
+        })),
+        values: t.arrayExpression(values.map(function (value) {
+          return (0, _node.numericLiteral)(value);
+        }))
+      });
+      exprStmt[predicateSymbol] = true;
+      return exprStmt;
+    };
+    basicBlocks.get(startLabel).body.unshift(createAssignment(Array.from(predicateNumbers.values())));
+
+    // Add random assignments to impossible blocks
+    var fakeAssignmentCount = (0, _randomUtils.getRandomInteger)(1, 3);
+    for (var _i3 = 0; _i3 < fakeAssignmentCount; _i3++) {
+      var impossibleBlock = (0, _randomUtils.choice)(getImpossibleBasicBlocks());
+      if (impossibleBlock) {
+        var _impossibleBlock$body;
+        if ((_impossibleBlock$body = impossibleBlock.body[0]) !== null && _impossibleBlock$body !== void 0 && _impossibleBlock$body[predicateSymbol]) continue;
+        var fakeValues = new Array(predicateNumberCount).fill(0).map(function () {
+          return (0, _randomUtils.getRandomInteger)(-250, 250);
+        });
+        impossibleBlock.body.unshift(createAssignment(fakeValues));
+      }
+    }
+
+    // Add scope initializations: scope["_0"] = {identity: "_0"}
+    var _iterator2 = _createForOfIteratorHelper(scopeToScopeManager.values()),
+      _step2;
+    try {
+      for (_iterator2.s(); !(_step2 = _iterator2.n()).done;) {
+        var _scopeManager = _step2.value;
+        if (_scopeManager.isNotUsed) continue;
+        if (!_scopeManager.requiresInitializing) continue;
+        if (_scopeManager.initializingBasicBlock.label === startLabel) continue;
+        _scopeManager.initializingBasicBlock.body.unshift(_scopeManager.getInitializingStatement());
+      }
+    } catch (err) {
+      _iterator2.e(err);
+    } finally {
+      _iterator2.f();
+    }
+    var switchCases = [];
+    var blocks = Array.from(basicBlocks.values());
+    if (!isDebug && addFakeTests) {
+      (0, _randomUtils.shuffle)(blocks);
+    }
+    var _loop3 = function _loop3() {
+      var block = _blocks[_i4];
+      if (block.label === endLabel) {
+        // ok(block.body.length === 0);
+        return 1; // continue
+      }
+      var test = (0, _node.numericLiteral)(block.totalState);
+
+      // Add complex tests
+      if (!isDebug && addComplexTests && (0, _randomUtils.chance)(50)) {
+        // Create complex test expressions for each switch case
+
+        // case STATE+X:
+        var stateVarIndex = (0, _randomUtils.getRandomInteger)(0, stateVars.length);
+        var stateValues = block.stateValues;
+        var difference = stateValues[stateVarIndex] - block.totalState;
+        var conditionNodes = [];
+        var alreadyConditionedItems = new Set();
+
+        // This code finds clash conditions and adds them to 'conditionNodes' array
+        Array.from(basicBlocks.keys()).forEach(function (label) {
+          if (label !== block.label) {
+            var labelStates = basicBlocks.get(label).stateValues;
+            var totalState = labelStates.reduce(function (a, b) {
+              return a + b;
+            }, 0);
+            if (totalState === labelStates[stateVarIndex] - difference) {
+              var differentIndex = labelStates.findIndex(function (v, i) {
+                return v !== stateValues[i];
+              });
+              if (differentIndex !== -1) {
+                var expressionAsString = differentIndex + "!=" + labelStates[differentIndex];
+                if (!alreadyConditionedItems.has(expressionAsString)) {
+                  alreadyConditionedItems.add(expressionAsString);
+                  conditionNodes.push(t.binaryExpression("!=", (0, _node.deepClone)(stateVars[differentIndex]), (0, _node.numericLiteral)(labelStates[differentIndex])));
                 }
+              } else {
+                conditionNodes.push(t.binaryExpression("!=", (0, _node.deepClone)(discriminant), (0, _node.numericLiteral)(totalState)));
               }
-
-              // Ensure parameter is added (No effect if not added in this case)
-            } catch (err) {
-              _iterator4.e(err);
-            } finally {
-              _iterator4.f();
             }
-            usedArgVar = true;
-            Object.assign(fn, new _template["default"]("\n              (function (...".concat(argumentsRestName, "){\n                ").concat(isDebug ? "\"Calling ".concat(originalFnName, ", Label: ").concat(fnLabel, "\";") : "", "\n                return {callExpression}\n              })\n              \n              ")).expression({
-              callExpression: createCallExpression(argumentsNodes)
-            }));
-          }
-          var startProgramObjectExpression = basicBlocks.get(startLabel).scopeManager.getObjectExpression(startLabel);
-          var mainParameters = parameters;
-
-          // First state values use the default parameter for initialization
-          // function main(..., scope = { mainScope: {} }, ...){...}
-          mainParameters.splice(mainParameters.findIndex(function (p) {
-            return p.name === scopeVar.name;
-          }), 1, t.assignmentPattern((0, _node.deepClone)(scopeVar), startProgramObjectExpression));
-
-          // Remove parameter 'argVar' if never used (No function calls obfuscated)
-          if (!usedArgVar) {
-            mainParameters.pop();
           }
-          var mainFnDeclaration = t.functionDeclaration((0, _node.deepClone)(mainFnName), parameters, t.blockStatement([whileStatement]), addGeneratorFunction);
-
-          // The main function is always called with same number of arguments
-          mainFnDeclaration[_constants.PREDICTABLE] = true;
-          function createCallExpression(argumentNodes) {
-            var callExpression = t.callExpression((0, _node.deepClone)(mainFnName), argumentNodes);
-            if (!addGeneratorFunction) {
-              return callExpression;
-            }
-            return new _template["default"]("\n              ({callExpression})[\"next\"]()[\"value\"];\n              ").expression({
-              callExpression: callExpression
-            });
+        });
+
+        // case STATE!=Y && STATE+X
+        test = t.binaryExpression("-", (0, _node.deepClone)(stateVars[stateVarIndex]), (0, _node.numericLiteral)(difference));
+
+        // Use the 'conditionNodes' to not cause state clashing issues
+        conditionNodes.forEach(function (conditionNode) {
+          test = t.logicalExpression("&&", conditionNode, test);
+        });
+      }
+      var tests = [test];
+      if (!isDebug && addFakeTests && (0, _randomUtils.chance)(50)) {
+        // Add fake tests
+        var fakeTestCount = (0, _randomUtils.getRandomInteger)(1, 3);
+        for (var _i5 = 0; _i5 < fakeTestCount; _i5++) {
+          tests.push((0, _node.numericLiteral)(stateIntGen.generate()));
+        }
+        (0, _randomUtils.shuffle)(tests);
+      }
+      var lastTest = tests.pop();
+      for (var _i6 = 0, _tests = tests; _i6 < _tests.length; _i6++) {
+        var _test = _tests[_i6];
+        switchCases.push(t.switchCase(_test, []));
+      }
+
+      // Preserve source location for source maps
+      var sup = t.switchCase(lastTest, block.thisPath.node.body);
+      var firstNode = block.thisPath.node.body[0];
+      if (firstNode) {
+        t.inherits(sup, firstNode);
+      }
+      switchCases.push(sup);
+    };
+    for (var _i4 = 0, _blocks = blocks; _i4 < _blocks.length; _i4++) {
+      if (_loop3()) continue;
+    }
+    if (!isDebug && addFakeTests) {
+      // A random test can be 'default'
+      // choice(switchCases).test = null;
+    }
+    needsSumFunction = true;
+    var discriminant = new _template["default"]("\n           ".concat(sumFnName, "({statesVar})\n          ")).expression({
+      statesVar: (0, _node.deepClone)(statesVar)
+    });
+    (0, _traverse["default"])(t.program([t.expressionStatement(discriminant)]), {
+      Identifier: function Identifier(path) {
+        path.node[_constants.NO_RENAME] = cffIndex;
+      }
+    });
+
+    // Create a new SwitchStatement
+    var switchStatement = t.inherits(t.labeledStatement(t.identifier(switchLabel), t.inherits(t.switchStatement((0, _node.deepClone)(discriminant), switchCases), programOrFunctionPath.node)), programOrFunctionPath.node);
+    var startStateValues = basicBlocks.get(startLabel).stateValues;
+    var endTotalState = basicBlocks.get(endLabel).totalState;
+    var whileStatement = t.whileStatement(t.binaryExpression("!==", (0, _node.deepClone)(discriminant), (0, _node.numericLiteral)(endTotalState)), t.inherits(t.blockStatement([switchStatement]), programOrFunctionPath.node));
+    t.inherits(whileStatement, programOrFunctionPath.node);
+    var parameters = [statesVar, scopeVar, runtimeVar, argVar].map(function (id) {
+      return (0, _node.deepClone)(id);
+    });
+    var parametersNames = parameters.map(function (id) {
+      return id.name;
+    });
+    for (var _i7 = 0, _functionExpressions = functionExpressions; _i7 < _functionExpressions.length; _i7++) {
+      var _functionExpressions$ = _slicedToArray(_functionExpressions[_i7], 4),
+        originalFnName = _functionExpressions$[0],
+        fnLabel = _functionExpressions$[1],
+        basicBlock = _functionExpressions$[2],
+        fn = _functionExpressions$[3];
+      var _basicBlock = basicBlock,
+        scopeManager = _basicBlock.scopeManager;
+      var _basicBlocks$get = basicBlocks.get(fnLabel),
+        stateValues = _basicBlocks$get.stateValues;
+      var argumentsRestName = me.getPlaceholder();
+      var argumentsNodes = [];
+      var _iterator3 = _createForOfIteratorHelper(parametersNames),
+        _step3;
+      try {
+        for (_iterator3.s(); !(_step3 = _iterator3.n()).done;) {
+          var parameterName = _step3.value;
+          if (parameterName === statesVar.name) {
+            continue;
+          } else if (parameterName === runtimeVar.name) {
+            argumentsNodes.push((0, _node.deepClone)(runtimeVar));
+          } else if (parameterName === argVar.name) {
+            argumentsNodes.push(t.identifier(argumentsRestName));
+          } else if (parameterName === scopeVar.name) {
+            argumentsNodes.push(scopeManager.getObjectExpression(fnLabel));
+          } else {
+            (0, _assert.ok)(false);
           }
-          var startProgramExpression = createCallExpression(startStateValues.map(function (stateValue) {
-            return (0, _node.numericLiteral)(stateValue);
-          }));
-          var _resultVar = withIdentifier("result");
-          var isTopLevel = blockPath.isProgram();
-          var allowReturns = !isTopLevel && blockPath.find(function (p) {
-            return p.isFunction();
-          });
-          var startPrefix = allowReturns ? "var {resultVar} = " : "";
-          var startProgramStatements = new _template["default"]("\n            ".concat(allowReturns ? "var {didReturnVar};" : "", "\n            ").concat(startPrefix, "{startProgramExpression};\n            ").concat(allowReturns ? "\n            if({didReturnVar}){\n              return {resultVar};\n            }" : "", "\n          ")).compile({
-            startProgramExpression: startProgramExpression,
-            didReturnVar: function didReturnVar() {
-              return (0, _node.deepClone)(_didReturnVar);
-            },
-            resultVar: function resultVar() {
-              return (0, _node.deepClone)(_resultVar);
-            }
+        }
+
+        // Ensure parameter is added (No effect if not added in this case)
+      } catch (err) {
+        _iterator3.e(err);
+      } finally {
+        _iterator3.f();
+      }
+      usedArgVar = true;
+      Object.assign(fn, new _template["default"]("\n              (function (...".concat(argumentsRestName, "){\n                ").concat(isDebug ? "\"Calling ".concat(originalFnName, ", Label: ").concat(fnLabel, "\";") : "", "\n                return {callExpression}\n              })\n              \n              ")).expression({
+        callExpression: createCallExpression(stateValues, argumentsNodes)
+      }));
+    }
+    var startProgramObjectExpression = basicBlocks.get(startLabel).scopeManager.getObjectExpression(startLabel);
+    var mainParameters = parameters;
+
+    // First state values use the default parameter for initialization
+    // function main(..., scope = { mainScope: {} }, ...){...}
+    mainParameters.splice(mainParameters.findIndex(function (p) {
+      return p.name === scopeVar.name;
+    }), 1, t.assignmentPattern((0, _node.deepClone)(scopeVar), startProgramObjectExpression));
+
+    // Remove parameter 'argVar' if never used (No function calls obfuscated)
+    if (!usedArgVar) {
+      mainParameters.pop();
+    }
+    var mainFnDeclaration = t.functionDeclaration((0, _node.deepClone)(mainFnName), parameters, t.inherits(t.blockStatement([whileStatement]), programOrFunctionPath.node), false);
+    t.inherits(mainFnDeclaration, programOrFunctionPath.node);
+
+    // The main function is always called with same number of arguments
+    mainFnDeclaration[_constants.PREDICTABLE] = true;
+    function createCallExpression(stateValues) {
+      var argumentNodes = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : [];
+      var callExpression = t.callExpression((0, _node.deepClone)(mainFnName), [t.arrayExpression(stateValues.map(function (value) {
+        return (0, _node.numericLiteral)(value);
+      }))].concat(_toConsumableArray(argumentNodes)));
+      return callExpression;
+    }
+    var startProgramExpression = createCallExpression(startStateValues);
+    var _resultVar = identifier("result");
+    var isTopLevel = blockPath.isProgram();
+    var allowReturns = !isTopLevel && blockPath.find(function (p) {
+      return p.isFunction();
+    });
+    var startPrefix = allowReturns ? "var {resultVar} = " : "";
+    var startProgramStatements = new _template["default"]("\n            ".concat(allowReturns ? "var {didReturnVar};" : "", "\n            ").concat(startPrefix, "{startProgramExpression};\n            ").concat(allowReturns ? "\n            if({didReturnVar}){\n              return {resultVar};\n            }" : "", "\n          ")).compile({
+      startProgramExpression: startProgramExpression,
+      didReturnVar: function didReturnVar() {
+        return (0, _node.deepClone)(_didReturnVar);
+      },
+      resultVar: function resultVar() {
+        return (0, _node.deepClone)(_resultVar);
+      }
+    });
+    blockPath.node.body = [].concat(prependNodes, [mainFnDeclaration], _toConsumableArray(startProgramStatements));
+    functionsModified.push(programOrFunctionPath.node);
+
+    // Reset all bindings here
+    blockPath.scope.bindings = Object.create(null);
+
+    // Register new declarations
+    var _iterator4 = _createForOfIteratorHelper(blockPath.get("body")),
+      _step4;
+    try {
+      for (_iterator4.s(); !(_step4 = _iterator4.n()).done;) {
+        var node = _step4.value;
+        blockPath.scope.registerDeclaration(node);
+      }
+    } catch (err) {
+      _iterator4.e(err);
+    } finally {
+      _iterator4.f();
+    }
+  }
+  return {
+    post: function post() {
+      for (var _i9 = 0, _functionsModified = functionsModified; _i9 < _functionsModified.length; _i9++) {
+        var node = _functionsModified[_i9];
+        node[_constants.UNSAFE] = true;
+      }
+    },
+    visitor: {
+      // Unsafe detection
+      ThisExpression: function ThisExpression(path) {
+        flagFunctionToAvoid(path, "this");
+      },
+      VariableDeclaration: function VariableDeclaration(path) {
+        if (path.node.declarations.length !== 1) {
+          path.getAncestry().forEach(function (p) {
+            p.node[CFF_UNSAFE] = "multipleDeclarations";
           });
-          blockPath.node.body = [].concat(prependNodes, [mainFnDeclaration], _toConsumableArray(startProgramStatements));
-          functionsModified.push(programOrFunctionPath.node);
-
-          // Reset all bindings here
-          blockPath.scope.bindings = Object.create(null);
-
-          // Register new declarations
-          var _iterator5 = _createForOfIteratorHelper(blockPath.get("body")),
-            _step5;
-          try {
-            for (_iterator5.s(); !(_step5 = _iterator5.n()).done;) {
-              var node = _step5.value;
-              blockPath.scope.registerDeclaration(node);
+        }
+      },
+      Identifier: function Identifier(path) {
+        if (path.node.name === _constants.variableFunctionName || path.node.name === "arguments") {
+          flagFunctionToAvoid(path, "arguments");
+        }
+      },
+      "Super|MetaProperty|AwaitExpression|YieldExpression": function SuperMetaPropertyAwaitExpressionYieldExpression(path) {
+        flagFunctionToAvoid(path, "functionSpecific");
+      },
+      // Main CFF transformation
+      "Program|Function": {
+        exit: function exit(_path) {
+          cffMain(_path);
+          if (_path.isProgram()) {
+            _path.stop(); // This is necessary because we insert a new function and it will get traversed again here
+
+            if (needsSumFunction) {
+              (0, _astUtils.prepend)(_path, new _template["default"]("\n              function ".concat(sumFnName, "(array){\n                for (var sum = 0, i = 0; i < array[\"length\"]; i++) {\n                  sum += array[i]\n                }\n                return sum;\n              }\n              ")).compile());
+              (0, _astUtils.prepend)(_path, _xorStringTemplate.xorDecodeStringTemplate.compile({
+                fnName: xorFnName
+              }));
             }
-          } catch (err) {
-            _iterator5.e(err);
-          } finally {
-            _iterator5.f();
           }
         }
       }