js-confuser 1.7.1 → 1.7.3

package/dist/transforms/string/stringCompression.js

@@ -4,160 +4,167 @@ Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports.default = void 0;
-
 var _assert = require("assert");
-
 var _order = require("../../order");
-
 var _probability = require("../../probability");
-
 var _template = _interopRequireDefault(require("../../templates/template"));
-
 var _compare = require("../../util/compare");
-
 var _gen = require("../../util/gen");
-
 var _insert = require("../../util/insert");
-
 var _transform = _interopRequireDefault(require("../transform"));
-
-var _stringConcealing = require("./stringConcealing");
-
+var _constants = require("../../constants");
 var _random = require("../../util/random");
-
-function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
-
-function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
-
+function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
+function _defineProperty(e, r, t) { return (r = _toPropertyKey(r)) in e ? Object.defineProperty(e, r, { value: t, enumerable: !0, configurable: !0, writable: !0 }) : e[r] = t, e; }
+function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == typeof i ? i : i + ""; }
+function _toPrimitive(t, r) { if ("object" != typeof t || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != typeof i) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
 function LZ_encode(c) {
   (0, _assert.ok)(c);
   var x = "charCodeAt",
-      b,
-      e = {},
-      f = c.split(""),
-      d = [],
-      a = f[0],
-      g = 256;
-
+    b,
+    e = {},
+    f = c.split(""),
+    d = [],
+    a = f[0],
+    g = 256;
   for (b = 1; b < f.length; b++) c = f[b], null != e[a + c] ? a += c : (d.push(1 < a.length ? e[a] : a[x](0)), e[a + c] = g, g++, a = c);
-
   d.push(1 < a.length ? e[a] : a[x](0));
-
   for (b = 0; b < d.length; b++) d[b] = String.fromCharCode(d[b]);
-
   return d.join("");
 }
-
 function LZ_decode(b) {
   (0, _assert.ok)(b);
   var o,
-      f,
-      a,
-      e = {},
-      d = b.split(""),
-      c = f = d[0],
-      g = [c],
-      h = o = 256;
-
+    f,
+    a,
+    e = {},
+    d = b.split(""),
+    c = f = d[0],
+    g = [c],
+    h = o = 256;
   for (var i = 1; i < d.length; i++) a = d[i].charCodeAt(0), a = h > a ? d[i] : e[a] ? e[a] : f + c, g.push(a), c = a.charAt(0), e[o] = f + c, o++, f = a;
-
   return g.join("");
 }
-
-const DecodeTemplate = (0, _template.default)("function {name}(b){\n    var o,\n    f,\n    a,\n    e = {},\n    d = b.split(\"\"),\n    c = (f = d[0]),\n    g = [c],\n    h = (o = 256);\n  for (b = 1; b < d.length; b++)\n    (a = d[b].charCodeAt(0)),\n      (a = h > a ? d[b] : e[a] ? e[a] : f + c),\n      g.push(a),\n      (c = a.charAt(0)),\n      (e[o] = f + c),\n      o++,\n      (f = a);\n  return g.join(\"\").split(\"{delimiter}\");\n  }");
-
+const DecodeTemplate = new _template.default(`function {name}(b){
+    var o,
+    f,
+    a,
+    e = {},
+    d = b.split(""),
+    c = (f = d[0]),
+    g = [c],
+    h = (o = 256);
+  for (b = 1; b < d.length; b++)
+    (a = d[b].charCodeAt(0)),
+      (a = h > a ? d[b] : e[a] ? e[a] : f + c),
+      g.push(a),
+      (c = a.charAt(0)),
+      (e[o] = f + c),
+      o++,
+      (f = a);
+  return g.join("").split("{delimiter}");
+  }`);
 class StringCompression extends _transform.default {
   constructor(o) {
     super(o, _order.ObfuscateOrder.StringCompression);
-
     _defineProperty(this, "map", void 0);
-
     _defineProperty(this, "ignore", void 0);
-
     _defineProperty(this, "string", void 0);
-
     _defineProperty(this, "delimiter", "|");
-
     _defineProperty(this, "fnName", void 0);
-
     this.map = new Map();
     this.ignore = new Set();
     this.string = "";
-    this.fnName = this.getPlaceholder();
+    this.fnName = this.getPlaceholder() + _constants.predictableFunctionTag;
   }
-
   apply(tree) {
     super.apply(tree);
     this.string = this.string.slice(0, this.string.length - 1);
-
     if (!this.string.length) {
       return;
     }
-
     var split = this.getPlaceholder();
     var decoder = this.getPlaceholder();
-    var getStringName = this.getPlaceholder();
-    var encoded = LZ_encode(this.string);
+    var getStringName = this.getPlaceholder() + _constants.predictableFunctionTag; // Returns the string payload
 
+    var encoded = LZ_encode(this.string);
     if (LZ_decode(encoded) !== this.string) {
       this.error(new Error("String failed to be decoded. Try disabling the 'stringCompression' option."));
     }
-
     var getStringParamName = this.getPlaceholder();
     var decoderParamName = this.getPlaceholder();
     var callExpression = (0, _gen.CallExpression)((0, _gen.Identifier)(decoderParamName), [(0, _gen.CallExpression)((0, _gen.Identifier)(getStringParamName), [])]);
     (0, _insert.prepend)(tree, (0, _gen.VariableDeclaration)((0, _gen.VariableDeclarator)(split, (0, _gen.CallExpression)((0, _gen.FunctionExpression)([(0, _gen.Identifier)(getStringParamName), (0, _gen.Identifier)(decoderParamName)], [(0, _gen.ReturnStatement)(callExpression)]), [(0, _gen.Identifier)(getStringName), (0, _gen.Identifier)(decoder)]))));
-    (0, _insert.append)(tree, (0, _gen.FunctionDeclaration)(getStringName, [], [(0, _gen.ReturnStatement)((0, _gen.Literal)(encoded))]));
+    var keys = new Set();
+    var keysToMake = (0, _random.getRandomInteger)(4, 14);
+    for (var i = 0; i < keysToMake; i++) {
+      keys.add((0, _random.getRandomString)((0, _random.getRandomInteger)(4, 14)));
+    }
+    var objectExpression = (0, _gen.ObjectExpression)(Array.from(keys).map(key => {
+      return (0, _gen.Property)((0, _gen.Literal)(key), (0, _random.getRandomFalseExpression)(), true);
+    }));
+
+    // Get string function
+    var getStringBody = [];
+    var splits = (0, _random.splitIntoChunks)(encoded, Math.floor(encoded.length / (0, _random.getRandomInteger)(3, 6)));
+    getStringBody.push((0, _gen.VariableDeclaration)((0, _gen.VariableDeclarator)("str", (0, _gen.Literal)(splits.shift()))));
+    getStringBody.push((0, _gen.VariableDeclaration)((0, _gen.VariableDeclarator)("objectToTest", objectExpression)));
+    const addIfStatement = (testingFor, literalValueToBeAppended) => {
+      getStringBody.push((0, _gen.IfStatement)((0, _gen.BinaryExpression)("in", (0, _gen.Literal)(testingFor), (0, _gen.Identifier)("objectToTest")), [(0, _gen.ExpressionStatement)((0, _gen.AssignmentExpression)("+=", (0, _gen.Identifier)("str"), (0, _gen.Literal)(literalValueToBeAppended)))]));
+    };
+    for (const split of splits) {
+      if ((0, _random.chance)(50)) {
+        var fakeKey;
+        do {
+          fakeKey = (0, _random.getRandomString)((0, _random.getRandomInteger)(4, 14));
+        } while (keys.has(fakeKey) || fakeKey in {});
+        addIfStatement(fakeKey, (0, _random.getRandomString)(split.length));
+      }
+      addIfStatement((0, _random.choice)(Array.from(keys)), split);
+    }
+
+    // Return computed string
+    getStringBody.push((0, _gen.ReturnStatement)((0, _gen.Identifier)("str")));
+    (0, _insert.append)(tree, (0, _gen.FunctionDeclaration)(getStringName, [], getStringBody));
     (0, _insert.append)(tree, (0, _gen.FunctionDeclaration)(this.fnName, [(0, _gen.Identifier)("index")], [(0, _gen.ReturnStatement)((0, _gen.MemberExpression)((0, _gen.Identifier)(split), (0, _gen.Identifier)("index"), true))]));
     (0, _insert.append)(tree, DecodeTemplate.single({
       name: decoder,
       delimiter: this.delimiter
     }));
   }
-
   match(object, parents) {
-    return object.type == "Literal" && typeof object.value === "string" && object.value && object.value.length > 3 && !(0, _compare.isDirective)(object, parents) && !(0, _stringConcealing.isModuleSource)(object, parents);
+    return object.type == "Literal" && typeof object.value === "string" && object.value && object.value.length > 3 && !(0, _compare.isDirective)(object, parents) && !(0, _compare.isModuleSource)(object, parents) && !parents.find(x => x.$multiTransformSkip);
   }
-
   transform(object, parents) {
     if (!object.value) {
       return;
     }
-
     if (this.ignore.has(object.value) || object.value.includes(this.delimiter)) {
       return;
     }
-
     if (!parents[0] || parents[0].type == "CallExpression" && parents[0].callee.type == "Identifier" && parents[0].callee.name == this.fnName) {
       return;
     }
-
     if (!(0, _probability.ComputeProbabilityMap)(this.options.stringCompression, x => x, object.value)) {
       return;
-    } // HARD CODED LIMIT of 10,000 (after 1,000 elements)
-
-
-    if (this.map.size > 1000 && !(0, _random.chance)(this.map.size / 100)) return;
-    var index = this.map.get(object.value); // New string, add it!
+    }
+    var index = this.map.get(object.value);
 
+    // New string, add it!
     if (typeof index !== "number") {
       // Ensure the string gets properly decoded
       if (LZ_decode(LZ_encode(object.value)) !== object.value) {
         this.ignore.add(object.value);
         return;
       }
-
       index = this.map.size;
       this.map.set(object.value, index);
       this.string += object.value + this.delimiter;
     }
-
     (0, _assert.ok)(typeof index === "number");
     return () => {
       this.replaceIdentifierOrLiteral(object, (0, _gen.CallExpression)((0, _gen.Identifier)(this.fnName), [(0, _gen.Literal)(index)]), parents);
     };
   }
-
 }
-
 exports.default = StringCompression;

package/dist/transforms/string/stringConcealing.js

@@ -4,236 +4,231 @@ Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports.default = void 0;
-exports.isModuleSource = isModuleSource;
-
 var _assert = require("assert");
-
 var _order = require("../../order");
-
 var _template = _interopRequireDefault(require("../../templates/template"));
-
 var _traverse = require("../../traverse");
-
 var _compare = require("../../util/compare");
-
 var _gen = require("../../util/gen");
-
 var _insert = require("../../util/insert");
-
 var _random = require("../../util/random");
-
 var _transform = _interopRequireDefault(require("../transform"));
-
-var _encoding = _interopRequireDefault(require("./encoding"));
-
+var _encoding = require("./encoding");
 var _probability = require("../../probability");
-
 var _bufferToString = require("../../templates/bufferToString");
-
-function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
-
-function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
-
-function isModuleSource(object, parents) {
-  if (!parents[0]) {
-    return false;
-  }
-
-  if (parents[0].type == "ImportDeclaration" && parents[0].source == object) {
-    return true;
-  }
-
-  if (parents[0].type == "ImportExpression" && parents[0].source == object) {
-    return true;
-  }
-
-  if (parents[1] && parents[1].type == "CallExpression" && parents[1].arguments[0] === object && parents[1].callee.type == "Identifier") {
-    if (parents[1].callee.name == "require" || parents[1].callee.name == "import") {
-      return true;
-    }
-  }
-
-  return false;
-}
-
+var _constants = require("../../constants");
+function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
+function _defineProperty(e, r, t) { return (r = _toPropertyKey(r)) in e ? Object.defineProperty(e, r, { value: t, enumerable: !0, configurable: !0, writable: !0 }) : e[r] = t, e; }
+function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == typeof i ? i : i + ""; }
+function _toPrimitive(t, r) { if ("object" != typeof t || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != typeof i) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
 class StringConcealing extends _transform.default {
   constructor(o) {
     super(o, _order.ObfuscateOrder.StringConcealing);
-
     _defineProperty(this, "arrayExpression", void 0);
-
     _defineProperty(this, "set", void 0);
-
     _defineProperty(this, "index", void 0);
-
+    // index, fnName, block
     _defineProperty(this, "arrayName", this.getPlaceholder());
-
     _defineProperty(this, "ignore", new Set());
-
     _defineProperty(this, "variablesMade", 1);
-
-    _defineProperty(this, "encoding", Object.create(null));
-
     _defineProperty(this, "gen", void 0);
-
-    _defineProperty(this, "hasAllEncodings", void 0);
-
+    _defineProperty(this, "functionObjects", []);
     this.set = new Set();
     this.index = Object.create(null);
     this.arrayExpression = (0, _gen.ArrayExpression)([]);
-    this.hasAllEncodings = false;
-    this.gen = this.getGenerator(); // Pad array with useless strings
-
-    var dead = (0, _random.getRandomInteger)(5, 15);
+    this.gen = this.getGenerator();
+  }
+  apply(tree) {
+    super.apply(tree);
 
+    // Pad array with useless strings
+    var dead = (0, _random.getRandomInteger)(50, 200);
     for (var i = 0; i < dead; i++) {
       var str = (0, _random.getRandomString)((0, _random.getRandomInteger)(5, 40));
-      var fn = this.transform((0, _gen.Literal)(str), []);
-
+      var fn = this.transform((0, _gen.Literal)(str), [tree]);
       if (fn) {
         fn();
       }
     }
-  }
-
-  apply(tree) {
-    super.apply(tree);
     var cacheName = this.getPlaceholder();
-    var bufferToStringName = this.getPlaceholder(); // This helper functions convert UInt8 Array to UTf-string
+    var bufferToStringName = this.getPlaceholder() + _constants.predictableFunctionTag;
 
+    // This helper functions convert UInt8 Array to UTf-string
     (0, _insert.prepend)(tree, ..._bufferToString.BufferToStringTemplate.compile({
-      name: bufferToStringName
+      name: bufferToStringName,
+      getGlobalFnName: this.getPlaceholder() + _constants.predictableFunctionTag,
+      GetGlobalTemplate: (0, _bufferToString.createGetGlobalTemplate)(this, tree, [])
     }));
-    Object.keys(this.encoding).forEach(type => {
+    for (var functionObject of this.functionObjects) {
       var {
-        template
-      } = _encoding.default[type];
-      var decodeFn = this.getPlaceholder();
-      var getterFn = this.encoding[type];
-      (0, _insert.append)(tree, template.single({
-        name: decodeFn,
-        bufferToString: bufferToStringName
+        block,
+        fnName: getterFnName,
+        encodingImplementation
+      } = functionObject;
+      var decodeFn = this.getPlaceholder() + _constants.predictableFunctionTag + _constants.criticalFunctionTag;
+      (0, _insert.append)(block, encodingImplementation.template.single({
+        __fnName__: decodeFn,
+        __bufferToString__: bufferToStringName
       }));
-      (0, _insert.append)(tree, (0, _template.default)("\n            \n            function ".concat(getterFn, "(x, y, z, a = ").concat(decodeFn, ", b = ").concat(cacheName, "){\n              if ( z ) {\n                return y[").concat(cacheName, "[z]] = ").concat(getterFn, "(x, y);\n              } else if ( y ) {\n                [b, y] = [a(b), x || z]\n              }\n            \n              return y ? x[b[y]] : ").concat(cacheName, "[x] || (z=(b[x], a), ").concat(cacheName, "[x] = z(").concat(this.arrayName, "[x]))\n            }\n  \n            ")).single());
-    });
-    var flowIntegrity = this.getPlaceholder();
-    (0, _insert.prepend)(tree, (0, _gen.VariableDeclaration)([(0, _gen.VariableDeclarator)(cacheName, (0, _gen.ArrayExpression)([])), (0, _gen.VariableDeclarator)(flowIntegrity, (0, _gen.Literal)(0)), (0, _gen.VariableDeclarator)(this.arrayName, (0, _gen.CallExpression)((0, _gen.FunctionExpression)([], [(0, _gen.VariableDeclaration)((0, _gen.VariableDeclarator)("a", this.arrayExpression)), (0, _template.default)("return (".concat(flowIntegrity, " ? a[\"pop\"]() : ").concat(flowIntegrity, "++, a)")).single()]), []))]));
-  }
+      // All these are fake and never ran
+      var ifStatements = new _template.default(`if ( z == x ) {
+          return y[${cacheName}[z]] = ${getterFnName}(x, y);
+        }
+        if ( y ) {
+          [b, y] = [a(b), x || z]
+          return ${getterFnName}(x, b, z)
+        }
+        if ( z && a !== ${decodeFn} ) {
+          ${getterFnName} = ${decodeFn}
+          return ${getterFnName}(x, -1, z, a, b)
+        }
+        if ( a === ${getterFnName} ) {
+          ${decodeFn} = y
+          return ${decodeFn}(z)
+        }
+        if( a === undefined ) {
+          ${getterFnName} = b
+        }
+        if( z == a ) {
+          return y ? x[b[y]] : ${cacheName}[x] || (z=(b[x] || a), ${cacheName}[x] = z(${this.arrayName}[x]))
+        }
+        `).compile();
 
-  match(object, parents) {
-    return object.type == "Literal" && typeof object.value === "string" && object.value.length >= 3 && !isModuleSource(object, parents) && !(0, _compare.isDirective)(object, parents) //&&
+      // Not all fake if-statements are needed
+      ifStatements = ifStatements.filter(() => (0, _random.chance)(50));
 
-    /*!parents.find((x) => x.$dispatcherSkip)*/
-    ;
+      // This one is always used
+      ifStatements.push(new _template.default(`
+      if ( x !== y ) {
+        return b[x] || (b[x] = a(${this.arrayName}[x]))
+      }
+      `).single());
+      (0, _random.shuffle)(ifStatements);
+      var varDeclaration = new _template.default(`
+      var ${getterFnName} = (x, y, z, a, b)=>{
+        if(typeof a === "undefined") {
+          a = ${decodeFn}
+        }
+        if(typeof b === "undefined") {
+          b = ${cacheName}
+        }
+      }
+      `).single();
+      varDeclaration.declarations[0].init.body.body.push(...ifStatements);
+      (0, _insert.prepend)(block, varDeclaration);
+    }
+    (0, _insert.prepend)(tree, (0, _gen.VariableDeclaration)([(0, _gen.VariableDeclarator)(cacheName, (0, _gen.ArrayExpression)([])), (0, _gen.VariableDeclarator)(this.arrayName, this.arrayExpression)]));
+  }
+  match(object, parents) {
+    return object.type == "Literal" && typeof object.value === "string" && object.value.length >= 3 && !(0, _compare.isModuleSource)(object, parents) && !(0, _compare.isDirective)(object, parents) //&&
+    /*!parents.find((x) => x.$multiTransformSkip)*/;
   }
-
   transform(object, parents) {
     return () => {
       // Empty strings are discarded
       if (!object.value || this.ignore.has(object.value) || object.value.length == 0) {
         return;
-      } // Allow user to choose which strings get changed
-
+      }
 
+      // Allow user to choose which strings get changed
       if (!(0, _probability.ComputeProbabilityMap)(this.options.stringConcealing, x => x, object.value)) {
         return;
-      } // HARD CODED LIMIT of 10,000 (after 1,000 elements)
-
-
-      if (this.set.size > 1000 && !(0, _random.chance)(this.set.size / 100)) return;
-      var types = Object.keys(this.encoding);
-      var type = (0, _random.choice)(types);
-
-      if (!type || !this.hasAllEncodings && (0, _random.chance)(10)) {
-        var allowed = Object.keys(_encoding.default).filter(type => !this.encoding[type]);
-
-        if (!allowed.length) {
-          this.hasAllEncodings = true;
-        } else {
-          var random = (0, _random.choice)(allowed);
-          type = random;
-          this.encoding[random] = this.getPlaceholder();
-        }
       }
-
-      var fnName = this.encoding[type];
-      var encoder = _encoding.default[type]; // The decode function must return correct result
-
-      var encoded = encoder.encode(object.value);
-
-      if (encoder.decode(encoded) != object.value) {
-        this.ignore.add(object.value);
-        this.warn(type, object.value.slice(0, 100));
-        return;
+      var currentBlock = (0, _traverse.getBlock)(object, parents);
+
+      // Find created functions
+      var functionObjects = parents.filter(node => node.$stringConcealingFunctionObject).map(item => item.$stringConcealingFunctionObject);
+
+      // Choose random functionObject to use
+      var functionObject = (0, _random.choice)(functionObjects);
+      if (!functionObject || !(0, _encoding.hasAllEncodings)() && (0, _random.chance)(25 / this.functionObjects.length) && !currentBlock.$stringConcealingFunctionObject) {
+        // No functions, create one
+
+        var newFunctionObject = {
+          block: currentBlock,
+          encodingImplementation: (0, _encoding.createEncodingImplementation)(),
+          fnName: this.getPlaceholder() + _constants.predictableFunctionTag
+        };
+        this.functionObjects.push(newFunctionObject);
+        currentBlock.$stringConcealingFunctionObject = newFunctionObject;
+        functionObject = newFunctionObject;
       }
-
+      var {
+        fnName,
+        encodingImplementation
+      } = functionObject;
       var index = -1;
 
-      if (!this.set.has(object.value)) {
+      // String already decoded?
+      if (this.set.has(object.value)) {
+        var row = this.index[object.value];
+        if (parents.includes(row[2])) {
+          [index, fnName] = row;
+          (0, _assert.ok)(typeof index === "number");
+        }
+      }
+      if (index == -1) {
+        // The decode function must return correct result
+        var encoded = encodingImplementation.encode(object.value);
+        if (encodingImplementation.decode(encoded) !== object.value) {
+          this.ignore.add(object.value);
+          this.warn(encodingImplementation.identity, object.value.slice(0, 100));
+          delete _encoding.EncodingImplementations[encodingImplementation.identity];
+          return;
+        }
         this.arrayExpression.elements.push((0, _gen.Literal)(encoded));
         index = this.arrayExpression.elements.length - 1;
-        this.index[object.value] = [index, fnName];
+        this.index[object.value] = [index, fnName, currentBlock];
         this.set.add(object.value);
-      } else {
-        [index, fnName] = this.index[object.value];
-        (0, _assert.ok)(typeof index === "number");
       }
-
       (0, _assert.ok)(index != -1, "index == -1");
-      var callExpr = (0, _gen.CallExpression)((0, _gen.Identifier)(fnName), [(0, _gen.Literal)(index)]); // use `.apply` to fool automated de-obfuscators
+      var callExpr = (0, _gen.CallExpression)((0, _gen.Identifier)(fnName), [(0, _gen.Literal)(index)]);
 
+      // use `.apply` to fool automated de-obfuscators
       if ((0, _random.chance)(10)) {
         callExpr = (0, _gen.CallExpression)((0, _gen.MemberExpression)((0, _gen.Identifier)(fnName), (0, _gen.Literal)("apply"), true), [(0, _gen.Identifier)("undefined"), (0, _gen.ArrayExpression)([(0, _gen.Literal)(index)])]);
-      } // use `.call`
+      }
+
+      // use `.call`
       else if ((0, _random.chance)(10)) {
         callExpr = (0, _gen.CallExpression)((0, _gen.MemberExpression)((0, _gen.Identifier)(fnName), (0, _gen.Literal)("call"), true), [(0, _gen.Identifier)("undefined"), (0, _gen.Literal)(index)]);
       }
-
       var referenceType = "call";
-
       if (parents.length && (0, _random.chance)(50 - this.variablesMade)) {
         referenceType = "constantReference";
       }
-
       var newExpr = callExpr;
-
       if (referenceType === "constantReference") {
         // Define the string earlier, reference the name here
         this.variablesMade++;
         var constantReferenceType = (0, _random.choice)(["variable", "array", "object"]);
-        var place = (0, _random.choice)(parents.filter(node => (0, _traverse.isBlock)(node)));
-
+        var place = currentBlock;
         if (!place) {
           this.error(new Error("No lexical block to insert code"));
         }
-
         switch (constantReferenceType) {
           case "variable":
             var name = this.getPlaceholder();
             (0, _insert.prepend)(place, (0, _gen.VariableDeclaration)((0, _gen.VariableDeclarator)(name, callExpr)));
             newExpr = (0, _gen.Identifier)(name);
             break;
-
           case "array":
             if (!place.$stringConcealingArray) {
               place.$stringConcealingArray = (0, _gen.ArrayExpression)([]);
               place.$stringConcealingArrayName = this.getPlaceholder();
               (0, _insert.prepend)(place, (0, _gen.VariableDeclaration)((0, _gen.VariableDeclarator)(place.$stringConcealingArrayName, place.$stringConcealingArray)));
             }
-
             var arrayIndex = place.$stringConcealingArray.elements.length;
             place.$stringConcealingArray.elements.push(callExpr);
             var memberExpression = (0, _gen.MemberExpression)((0, _gen.Identifier)(place.$stringConcealingArrayName), (0, _gen.Literal)(arrayIndex), true);
             newExpr = memberExpression;
             break;
-
           case "object":
             if (!place.$stringConcealingObject) {
               place.$stringConcealingObject = (0, _gen.ObjectExpression)([]);
               place.$stringConcealingObjectName = this.getPlaceholder();
               (0, _insert.prepend)(place, (0, _gen.VariableDeclaration)((0, _gen.VariableDeclarator)(place.$stringConcealingObjectName, place.$stringConcealingObject)));
             }
-
             var propName = this.gen.generate();
             var property = (0, _gen.Property)((0, _gen.Literal)(propName), callExpr, true);
             place.$stringConcealingObject.properties.push(property);
@@ -242,11 +237,8 @@ class StringConcealing extends _transform.default {
             break;
         }
       }
-
       this.replaceIdentifierOrLiteral(object, newExpr, parents);
     };
   }
-
 }
-
 exports.default = StringConcealing;