js-confuser 1.7.2 → 2.0.0-alpha.0

package/dist/transforms/identifier/renameVariables.js

@@ -1,215 +1,268 @@
 "use strict";
 
+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); }
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.default = void 0;
-var _assert = require("assert");
+exports["default"] = void 0;
+var t = _interopRequireWildcard(require("@babel/types"));
 var _order = require("../../order");
-var _traverse = require("../../traverse");
-var _identifiers = require("../../util/identifiers");
-var _insert = require("../../util/insert");
-var _transform = _interopRequireDefault(require("../transform"));
 var _constants = require("../../constants");
 var _probability = require("../../probability");
-var _variableAnalysis = _interopRequireDefault(require("./variableAnalysis"));
-function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
+var _astUtils = require("../../utils/ast-utils");
+var _functionUtils = require("../../utils/function-utils");
+function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(e) { return e ? t : r; })(e); }
+function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != _typeof(e) && "function" != typeof e) return { "default": e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && {}.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n["default"] = e, t && t.set(e, n), n; }
+function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
+function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return 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); }
-/**
- * Rename variables to randomly generated names.
- *
- * - 1. First collect data on identifiers in all scope using 'VariableAnalysis'
- * - 2. After 'VariableAnalysis' is finished start applying to each scope (top-down)
- * - 3. Each scope, find the all names used here and exclude those names from being re-named
- * - 4. Now loop through all the defined names in this scope and set it to a random name (or re-use previously generated name)
- * - 5. Update all the Identifiers node's 'name' property to reflect this change
- */
-class RenameVariables extends _transform.default {
-  constructor(o) {
-    super(o, _order.ObfuscateOrder.RenameVariables);
-    // Names already used
-    _defineProperty(this, "generated", void 0);
-    // Map of Context->Object of changes
-    _defineProperty(this, "changed", void 0);
-    // Ref to VariableAnalysis data
-    _defineProperty(this, "variableAnalysis", void 0);
-    // Option to re-use previously generated names
-    _defineProperty(this, "reusePreviousNames", true);
-    this.changed = new Map();
-
-    // 1.
-    this.variableAnalysis = new _variableAnalysis.default(o);
-    this.before.push(this.variableAnalysis);
-    this.generated = [];
-  }
-  match(object, parents) {
-    return (0, _insert.isContext)(object) || object.type === "Identifier";
-  }
-  transformContext(object, parents) {
-    // 2. Notice this is on 'onEnter' (top-down)
-    var isGlobal = object.type == "Program";
-    var type = isGlobal ? "root" : (0, _insert.isVarContext)(object) ? "var" : (0, _insert.isLexContext)(object) ? "lex" : undefined;
-    (0, _assert.ok)(type);
-    var newNames = Object.create(null);
-    var defined = this.variableAnalysis.defined.get(object) || new Set();
-    var references = this.variableAnalysis.references.get(object) || new Set();
-
-    // No changes needed here
-    if (!defined && !this.changed.has(object)) {
-      this.changed.set(object, Object.create(null));
-      return;
+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 _slicedToArray(r, e) { return _arrayWithHoles(r) || _iterableToArrayLimit(r, e) || _unsupportedIterableToArray(r, e) || _nonIterableRest(); }
+function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
+function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t["return"] && (u = t["return"](), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
+function _arrayWithHoles(r) { if (Array.isArray(r)) return r; }
+function _createForOfIteratorHelper(r, e) { var t = "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (!t) { if (Array.isArray(r) || (t = _unsupportedIterableToArray(r)) || e && r && "number" == typeof r.length) { t && (r = t); var _n = 0, F = function F() {}; return { s: F, n: function n() { return _n >= r.length ? { done: !0 } : { done: !1, value: r[_n++] }; }, e: function e(r) { throw r; }, f: F }; } throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } var o, a = !0, u = !1; return { s: function s() { t = t.call(r); }, n: function n() { var r = t.next(); return a = r.done, r; }, e: function e(r) { u = !0, o = r; }, f: function f() { try { a || null == t["return"] || t["return"](); } finally { if (u) throw o; } } }; }
+function _unsupportedIterableToArray(r, a) { if (r) { if ("string" == typeof r) return _arrayLikeToArray(r, a); var t = {}.toString.call(r).slice(8, -1); return "Object" === t && r.constructor && (t = r.constructor.name), "Map" === t || "Set" === t ? Array.from(r) : "Arguments" === t || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(t) ? _arrayLikeToArray(r, a) : void 0; } }
+function _arrayLikeToArray(r, a) { (null == a || a > r.length) && (a = r.length); for (var e = 0, n = Array(a); e < a; e++) n[e] = r[e]; return n; }
+var RENAMED = Symbol("Renamed");
+var reusePreviousNames = true;
+var _default = exports["default"] = function _default(_ref) {
+  var Plugin = _ref.Plugin;
+  var me = Plugin(_order.Order.RenameVariables, {
+    changeData: {
+      variables: 0
     }
+  });
+  var definedMap = new Map();
+  var referencedMap = new Map();
+  var bindingMap = new Map();
+  var renamedVariables = new Map();
+  me.obfuscator.globalState.renamedVariables = renamedVariables;
+  var generated = Array.from(me.obfuscator.nameGen.generatedNames);
+  var VariableAnalysisVisitor = {
+    Program: {
+      enter: function enter(path) {
+        // Analyze all scopes
+        path.traverse({
+          Identifier: function Identifier(path) {
+            if (!(0, _astUtils.isVariableIdentifier)(path)) return;
+            var contextPaths = [(0, _astUtils.getParentFunctionOrProgram)(path)];
+            var isDefined = false;
+            if (path.isBindingIdentifier() && (0, _astUtils.isDefiningIdentifier)(path)) {
+              isDefined = true;
 
-    // Names possible to be re-used here
-    var possible = new Set();
-
-    // 3. Try to re-use names when possible
-    if (this.reusePreviousNames && this.generated.length && !isGlobal) {
-      var allReferences = new Set();
-      var nope = new Set(defined);
-      (0, _traverse.walk)(object, [], (o, p) => {
-        var ref = this.variableAnalysis.references.get(o);
-        if (ref) {
-          ref.forEach(x => allReferences.add(x));
-        }
-        var def = this.variableAnalysis.defined.get(o);
-        if (def) {
-          def.forEach(x => allReferences.add(x));
-        }
-      });
-      var passed = new Set();
-      parents.forEach(p => {
-        var changes = this.changed.get(p);
-        if (changes) {
-          Object.keys(changes).forEach(x => {
-            var name = changes[x];
-            if (!allReferences.has(x) && !references.has(x)) {
-              passed.add(name);
-            } else {
-              nope.add(name);
+              // Function ID is defined in the parent's function declaration
+              if (path.key === "id" && path.parentPath.isFunctionDeclaration()) {
+                contextPaths = [(0, _astUtils.getParentFunctionOrProgram)(path.parentPath)];
+              }
             }
-          });
-        }
-      });
-      nope.forEach(x => passed.delete(x));
-      possible = passed;
-    }
+            contextPaths.forEach(function (contextPath) {
+              // console.log(contextPath.node.type, path.node.name, isDefined);
 
-    // 4. Defined names to new names
-    for (var name of defined) {
-      if (!name.startsWith(_constants.noRenameVariablePrefix) && (
-      // Variables prefixed with '__NO_JS_CONFUSER_RENAME__' are never renamed
-      isGlobal && !name.startsWith(_constants.placeholderVariablePrefix) // Variables prefixed with '__p_' are created by the obfuscator, always renamed
-      ? (0, _probability.ComputeProbabilityMap)(this.options.renameGlobals, x => x, name) : true) && (0, _probability.ComputeProbabilityMap)(
-      // Check the user's option for renaming variables
-      this.options.renameVariables, x => x, name, isGlobal)) {
-        // Create a new name from (1) or (2) methods
-        var newName;
-        do {
-          if (possible.size) {
-            // (1) Re-use previously generated name
-            var first = possible.values().next().value;
-            possible.delete(first);
-            newName = first;
-          } else {
-            // (2) Create a new name with `generateIdentifier` function
-            var generatedName = this.generateIdentifier();
-            newName = generatedName;
-            this.generated.push(generatedName);
+              if (isDefined) {
+                // Add to defined map
+                if (!definedMap.has(contextPath.node)) {
+                  definedMap.set(contextPath.node, new Set());
+                }
+                definedMap.get(contextPath.node).add(path.node.name);
+                if (!bindingMap.has(contextPath.node)) {
+                  bindingMap.set(contextPath.node, new Map());
+                }
+                bindingMap.get(contextPath.node).set(path.node.name, path);
+              } else {
+                // Add to reference map
+                if (!referencedMap.has(contextPath.node)) {
+                  referencedMap.set(contextPath.node, new Set());
+                }
+                referencedMap.get(contextPath.node).add(path.node.name);
+              }
+            });
           }
-        } while (this.variableAnalysis.globals.has(newName)); // Ensure global names aren't overridden
+        });
 
-        newNames[name] = newName;
-      } else {
-        // This variable name was deemed not to be renamed.
-        newNames[name] = name;
+        //
       }
     }
-
-    // console.log(object.type, newNames);
-    this.changed.set(object, newNames);
-  }
-  transformIdentifier(object, parents) {
-    const identifierName = object.name;
-    if (_constants.reservedIdentifiers.has(identifierName) || this.options.globalVariables.has(identifierName)) {
-      return;
-    }
-    if (object.$renamed) {
-      return;
-    }
-    var info = (0, _identifiers.getIdentifierInfo)(object, parents);
-    if (info.spec.isExported) {
-      return;
-    }
-    if (!info.spec.isReferenced) {
-      return;
-    }
-    var contexts = [object, ...parents].filter(x => (0, _insert.isContext)(x));
-    var newName = null;
-
-    // Function default parameter check!
-    var functionIndices = [];
-    for (var i in parents) {
-      if ((0, _insert.isFunction)(parents[i])) {
-        functionIndices.push(i);
+  };
+  var VariableRenamingVisitor = {
+    Identifier: function Identifier(identifierPath) {
+      if (!(0, _astUtils.isVariableIdentifier)(identifierPath)) return;
+      var node = identifierPath.node;
+      var identifierName = node.name;
+      if (node[RENAMED]) {
+        return;
       }
-    }
-    for (var functionIndex of functionIndices) {
-      if (parents[functionIndex].id === object) {
-        // This context is not referenced, so remove it
-        contexts = contexts.filter(context => context != parents[functionIndex]);
-        continue;
+      var contextPaths = identifierPath.getAncestry();
+
+      // A Function ID is not in the same context as it's body
+      if (identifierPath.key === "id" && identifierPath.parentPath.isFunctionDeclaration()) {
+        contextPaths = contextPaths.filter(function (x) {
+          return x !== identifierPath.parentPath;
+        });
       }
-      if (parents[functionIndex].params === parents[functionIndex - 1]) {
-        var isReferencedHere = true;
-        var slicedParents = parents.slice(0, functionIndex);
-        var forIndex = 0;
-        for (var parent of slicedParents) {
-          var childNode = slicedParents[forIndex - 1] || object;
-          if (parent.type === "AssignmentPattern" && parent.right === childNode) {
-            isReferencedHere = false;
-            break;
+      var newName = null;
+      var _iterator = _createForOfIteratorHelper(contextPaths),
+        _step;
+      try {
+        for (_iterator.s(); !(_step = _iterator.n()).done;) {
+          var contextPath = _step.value;
+          var _node = contextPath.node;
+          var defined = definedMap.get(_node);
+          if (defined !== null && defined !== void 0 && defined.has(identifierName)) {
+            var renamed = renamedVariables.get(_node);
+            if (renamed !== null && renamed !== void 0 && renamed.has(identifierName)) {
+              newName = renamed.get(identifierName);
+              break;
+            }
           }
-          forIndex++;
-        }
-        if (!isReferencedHere) {
-          // This context is not referenced, so remove it
-          contexts = contexts.filter(context => context != parents[functionIndex]);
         }
+      } catch (err) {
+        _iterator.e(err);
+      } finally {
+        _iterator.f();
       }
-    }
-    for (var check of contexts) {
-      if (this.variableAnalysis.defined.has(check) && this.variableAnalysis.defined.get(check).has(identifierName)) {
-        if (this.changed.has(check) && this.changed.get(check)[identifierName]) {
-          newName = this.changed.get(check)[identifierName];
-          break;
+      if (newName && typeof newName === "string") {
+        // __JS_CONFUSER_VAR__ function
+        if ((0, _functionUtils.isVariableFunctionIdentifier)(identifierPath)) {
+          identifierPath.parentPath.replaceWith(t.stringLiteral(newName));
+          return;
         }
+
+        // 5. Update Identifier node's 'name' property
+        node.name = newName;
+        node[RENAMED] = true;
       }
-    }
-    if (newName && typeof newName === "string") {
-      // Strange behavior where the `local` and `imported` objects are the same
-      if (info.isImportSpecifier) {
-        var importSpecifierIndex = parents.findIndex(x => x.type === "ImportSpecifier");
-        if (importSpecifierIndex != -1 && parents[importSpecifierIndex].imported === (parents[importSpecifierIndex - 1] || object) && parents[importSpecifierIndex].imported && parents[importSpecifierIndex].imported.type === "Identifier") {
-          parents[importSpecifierIndex].imported = (0, _insert.clone)(parents[importSpecifierIndex - 1] || object);
+    },
+    Scopable: function Scopable(scopePath) {
+      // 2. Notice this is on 'onEnter' (top-down)
+      var isGlobal = scopePath.isProgram();
+      var node = scopePath.scope.path.node;
+      if (renamedVariables.has(node)) return;
+      var defined = definedMap.get(node) || new Set();
+      var references = referencedMap.get(node) || new Set();
+      var bindings = bindingMap.get(node);
+
+      // No changes needed here
+      if (!defined && !renamedVariables.has(node)) {
+        renamedVariables.set(node, Object.create(null));
+        return;
+      }
+      var newNames = new Map();
+
+      // Names possible to be re-used here
+      var possible = new Set();
+
+      // 3. Try to re-use names when possible
+      if (reusePreviousNames && generated.length && !isGlobal) {
+        var allReferences = new Set();
+        var nope = new Set(defined);
+        scopePath.traverse({
+          Scopable: function Scopable(path) {
+            var node = path.scope.path.node;
+            var ref = referencedMap.get(node);
+            if (ref) {
+              ref.forEach(function (x) {
+                return allReferences.add(x);
+              });
+            }
+            var def = definedMap.get(node);
+            if (def) {
+              def.forEach(function (x) {
+                return allReferences.add(x);
+              });
+            }
+          }
+        });
+        var passed = new Set();
+        var parentPaths = scopePath.getAncestry();
+        parentPaths.forEach(function (p) {
+          if (p === scopePath) return;
+          var changes = renamedVariables.get(p.node);
+          if (changes) {
+            var _iterator2 = _createForOfIteratorHelper(changes),
+              _step2;
+            try {
+              for (_iterator2.s(); !(_step2 = _iterator2.n()).done;) {
+                var _step2$value = _slicedToArray(_step2.value, 2),
+                  oldName = _step2$value[0],
+                  newName = _step2$value[1];
+                if (!allReferences.has(oldName) && !references.has(oldName)) {
+                  passed.add(newName);
+                } else {
+                  nope.add(newName);
+                }
+              }
+            } catch (err) {
+              _iterator2.e(err);
+            } finally {
+              _iterator2.f();
+            }
+          }
+        });
+        nope.forEach(function (x) {
+          return passed["delete"](x);
+        });
+        possible = passed;
+      }
+      function shouldRename(name) {
+        // __NO_JS_CONFUSER_RENAME__
+        if (name.startsWith(_constants.noRenameVariablePrefix)) return false;
+
+        // Placeholder variables should always be renamed
+        if (name.startsWith(_constants.placeholderVariablePrefix)) return true;
+
+        // Do not rename exports
+        if ((0, _astUtils.isExportedIdentifier)(bindings === null || bindings === void 0 ? void 0 : bindings.get(name))) return false;
+        if (name === me.obfuscator.getStringCompressionLibraryName()) return false;
+
+        // Global variables are additionally checked against user option
+        if (isGlobal) {
+          if (!(0, _probability.computeProbabilityMap)(me.options.renameGlobals, name)) return false;
         }
+        if (!(0, _probability.computeProbabilityMap)(me.options.renameVariables, name, isGlobal)) return false;
+        return true;
       }
 
-      // console.log(o.name, "->", newName);
-      // 5. Update Identifier node's 'name' property
-      object.name = newName;
-      object.$renamed = true;
-    }
-  }
-  transform(object, parents) {
-    var matchType = object.type === "Identifier" ? "Identifier" : "Context";
-    if (matchType === "Identifier") {
-      this.transformIdentifier(object, parents);
-    } else {
-      this.transformContext(object, parents);
+      // 4. Defined names to new names
+      var _iterator3 = _createForOfIteratorHelper(defined),
+        _step3;
+      try {
+        for (_iterator3.s(); !(_step3 = _iterator3.n()).done;) {
+          var name = _step3.value;
+          var newName = name;
+          if (shouldRename(name)) {
+            me.changeData.variables++;
+
+            // Create a new name from (1) or (2) methods
+            do {
+              if (possible.size) {
+                // (1) Re-use previously generated name
+                var first = possible.values().next().value;
+                possible["delete"](first);
+                newName = first;
+              } else {
+                // (2) Create a new name with global `nameGen`
+                var generatedName = me.obfuscator.nameGen.generate();
+                newName = generatedName;
+                generated.push(generatedName);
+              }
+            } while (scopePath.scope.hasGlobal(newName) || me.obfuscator.nameGen.notSafeForReuseNames.has(newName));
+            // Ensure global names aren't overridden
+          }
+          newNames.set(name, newName);
+        }
+
+        // console.log(node.type, newNames);
+      } catch (err) {
+        _iterator3.e(err);
+      } finally {
+        _iterator3.f();
+      }
+      renamedVariables.set(node, newNames);
     }
-  }
-}
-exports.default = RenameVariables;
+  };
+  return {
+    visitor: _objectSpread(_objectSpread({}, VariableAnalysisVisitor), VariableRenamingVisitor)
+  };
+};