js-confuser 1.7.3 → 2.0.0-alpha.0

package/dist/transforms/extraction/duplicateLiteralsRemoval.js

@@ -1,204 +1,103 @@
 "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 _transform = _interopRequireDefault(require("../transform"));
-var _gen = require("../../util/gen");
-var _insert = require("../../util/insert");
-var _compare = require("../../util/compare");
-var _order = require("../../order");
-var _probability = require("../../probability");
+exports["default"] = void 0;
+var t = _interopRequireWildcard(require("@babel/types"));
 var _assert = require("assert");
-var _random = require("../../util/random");
-var _traverse = require("../../traverse");
-var _identifiers = require("../../util/identifiers");
-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); }
-/**
- * [Duplicate Literals Removal](https://docs.jscrambler.com/code-integrity/documentation/transformations/duplicate-literals-removal) replaces duplicate literals with a variable name.
- *
- * - Potency Medium
- * - Resilience Medium
- * - Cost Medium
- *
- * ```js
- * // Input
- * var foo = "http://www.example.xyz";
- * bar("http://www.example.xyz");
- *
- * // Output
- * var a = "http://www.example.xyz";
- * var foo = a;
- * bar(a);
- * ```
- */
-class DuplicateLiteralsRemoval extends _transform.default {
-  constructor(o) {
-    super(o, _order.ObfuscateOrder.DuplicateLiteralsRemoval);
-    // The array holding all the duplicate literals
-    _defineProperty(this, "arrayName", void 0);
-    // The array expression node to be inserted into the program
-    _defineProperty(this, "arrayExpression", void 0);
-    /**
-     * Literals in the array
-     */
-    _defineProperty(this, "map", void 0);
-    /**
-     * Literals are saved here the first time they are seen.
-     */
-    _defineProperty(this, "first", void 0);
-    /**
-     * Block -> { functionName, indexShift }
-     */
-    _defineProperty(this, "functions", void 0);
-    this.map = new Map();
-    this.first = new Map();
-    this.functions = new Map();
-  }
-  apply(tree) {
-    super.apply(tree);
-    if (this.arrayName && this.arrayExpression.elements.length > 0) {
-      // This function simply returns the array
-      var getArrayFn = this.getPlaceholder();
-      (0, _insert.append)(tree, (0, _gen.FunctionDeclaration)(getArrayFn, [], [(0, _gen.ReturnStatement)(this.arrayExpression)]));
-
-      // This variable holds the array
-      (0, _insert.prepend)(tree, (0, _gen.VariableDeclaration)((0, _gen.VariableDeclarator)(this.arrayName, (0, _gen.CallExpression)((0, _gen.Identifier)(getArrayFn), []))));
-
-      // Create all the functions needed
-      for (var blockNode of this.functions.keys()) {
-        var {
-          functionName,
-          indexShift
-        } = this.functions.get(blockNode);
-        var propertyNode = (0, _gen.BinaryExpression)("-", (0, _gen.Identifier)("index_param"), (0, _gen.Literal)(indexShift));
-        var indexRangeInclusive = [0 + indexShift - 1, this.map.size + indexShift];
-
-        // The function uses mangling to hide the index being accessed
-        var mangleCount = (0, _random.getRandomInteger)(1, 5);
-        for (var i = 0; i < mangleCount; i++) {
-          var operator = (0, _random.choice)([">", "<"]);
-          var compareValue = (0, _random.choice)(indexRangeInclusive);
-          var test = (0, _gen.BinaryExpression)(operator, (0, _gen.Identifier)("index_param"), (0, _gen.Literal)(compareValue));
-          var alternate = (0, _gen.BinaryExpression)("-", (0, _gen.Identifier)("index_param"), (0, _gen.Literal)((0, _random.getRandomInteger)(-100, 100)));
-          var testValue = operator === ">" && compareValue === indexRangeInclusive[0] || operator === "<" && compareValue === indexRangeInclusive[1];
-          propertyNode = (0, _gen.ConditionalExpression)(test, testValue ? propertyNode : alternate, !testValue ? propertyNode : alternate);
-        }
-        var returnArgument = (0, _gen.MemberExpression)((0, _gen.Identifier)(this.arrayName), propertyNode, true);
-        (0, _insert.prepend)(blockNode, (0, _gen.FunctionDeclaration)(functionName, [(0, _gen.Identifier)("index_param")], [(0, _gen.ReturnStatement)(returnArgument)]));
-      }
+var _order = require("../../order");
+var _astUtils = require("../../utils/ast-utils");
+var _node = require("../../utils/node");
+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 fail() {
+  throw new Error("Assertion failed");
+}
+var _default = exports["default"] = function _default(_ref) {
+  var Plugin = _ref.Plugin;
+  var me = Plugin(_order.Order.DuplicateLiteralsRemoval, {
+    changeData: {
+      literals: 0
     }
-  }
-  match(object, parents) {
-    return (0, _compare.isPrimitive)(object) && !(0, _compare.isDirective)(object, parents) && !(0, _compare.isModuleSource)(object, parents) && !parents.find(x => x.$multiTransformSkip);
-  }
+  });
+  return {
+    visitor: {
+      Program: {
+        enter: function enter(programPath) {
+          var arrayName = me.getPlaceholder() + "_dlrArray";
 
-  /**
-   * Converts ordinary literal to go through a getter function.
-   * @param object
-   * @param parents
-   * @param index
-   */
-  transformLiteral(object, parents, index) {
-    var blockNode = (0, _random.choice)(parents.filter(x => this.functions.has(x)));
+          // Collect all literals
+          var literalsMap = new Map();
+          var firstTimeMap = new Map();
+          var arrayExpression = t.arrayExpression([]);
+          var createMemberExpression = function createMemberExpression(index) {
+            return t.memberExpression(t.identifier(arrayName), (0, _node.numericLiteral)(index), true);
+          };
 
-    // Create initial function if none exist
-    if (this.functions.size === 0) {
-      var root = parents[parents.length - 1];
-      var rootFunctionName = this.getPlaceholder() + "_dLR_0";
-      this.functions.set(root, {
-        functionName: rootFunctionName + _constants.predictableFunctionTag,
-        indexShift: (0, _random.getRandomInteger)(-100, 100)
-      });
-      blockNode = root;
-    }
+          // Traverse through all nodes to find literals
+          programPath.traverse({
+            "StringLiteral|BooleanLiteral|NumericLiteral|NullLiteral|Identifier": function StringLiteralBooleanLiteralNumericLiteralNullLiteralIdentifier(_path) {
+              var literalPath = _path;
+              var node = literalPath.node;
+              var isUndefined = false;
+              if (literalPath.isIdentifier()) {
+                // Only referenced variable names
+                if (!literalPath.isReferencedIdentifier()) return;
 
-    // If no function here exist, possibly create new chained function
-    var block = (0, _traverse.getBlock)(object, parents);
-    if (!this.functions.has(block) && (0, _random.chance)(50 - this.functions.size)) {
-      var newFunctionName = this.getPlaceholder() + "_dLR_" + this.functions.size + _constants.predictableFunctionTag;
-      this.functions.set(block, {
-        functionName: newFunctionName,
-        indexShift: (0, _random.getRandomInteger)(-100, 100)
-      });
-      blockNode = block;
-    }
+                // undefined = true; // Skip
+                if (literalPath.isBindingIdentifier()) return;
 
-    // Derive the function to call from the selected blockNode
-    var {
-      functionName,
-      indexShift
-    } = this.functions.get(blockNode);
+                // Allow 'undefined' to be redefined
+                if (literalPath.scope.hasBinding(literalPath.node.name, {
+                  noGlobals: true
+                })) return;
+                if (literalPath.node.name === "undefined") {
+                  isUndefined = true;
+                } else {
+                  return;
+                }
+              }
+              if (t.isRegExpLiteral(node) || t.isTemplateLiteral(node) || t.isDirectiveLiteral(node)) return;
+              var value = isUndefined ? undefined : t.isNullLiteral(node) ? null : t.isLiteral(node) ? node.value : fail();
+              if (typeof value !== "string" && typeof value !== "number" && typeof value !== "boolean" && value !== null && value !== undefined) {
+                return;
+              }
 
-    // Call the function given it's indexShift
-    var callExpression = (0, _gen.CallExpression)((0, _gen.Identifier)(functionName), [(0, _gen.Literal)(index + indexShift)]);
-    this.replaceIdentifierOrLiteral(object, callExpression, parents);
-  }
-  transform(object, parents) {
-    return () => {
-      if (object.type === "Identifier") {
-        var info = (0, _identifiers.getIdentifierInfo)(object, parents);
-        if (info.isLabel || info.spec.isDefined || info.spec.isModified) return;
-      }
-      if (object.regex) {
-        return;
-      }
-      if (!(0, _probability.ComputeProbabilityMap)(this.options.duplicateLiteralsRemoval)) {
-        return;
-      }
-      if (this.arrayName && parents[0].object && parents[0].object.name == this.arrayName) {
-        return;
-      }
-      var stringValue;
-      if (object.type == "Literal") {
-        stringValue = typeof object.value + ":" + object.value;
-        if (object.value === null) {
-          stringValue = "null:null";
-        } else {
-          // Skip empty strings
-          if (typeof object.value === "string" && !object.value) {
-            return;
-          }
-        }
-      } else if (object.type == "Identifier") {
-        stringValue = "identifier:" + object.name;
-      } else {
-        throw new Error("Unsupported primitive type: " + object.type);
-      }
-      (0, _assert.ok)(stringValue);
-      if (this.map.has(stringValue) || this.first.has(stringValue)) {
-        // Create the array if not already made
-        if (!this.arrayName) {
-          this.arrayName = this.getPlaceholder();
-          this.arrayExpression = (0, _gen.ArrayExpression)([]);
-        }
+              // Skip empty strings
+              if (typeof value === "string" && value.length === 0) return;
+              var index = -1;
+              if (literalsMap.has(value)) {
+                index = literalsMap.get(value);
+              } else if (firstTimeMap.has(value)) {
+                // Create new index
+
+                index = literalsMap.size;
+                literalsMap.set(value, index);
+                var firstPath = firstTimeMap.get(value);
+                me.changeData.literals++;
+                (0, _astUtils.ensureComputedExpression)(firstPath);
+                firstPath.replaceWith(createMemberExpression(index));
+                arrayExpression.elements.push((0, _node.createLiteral)(value));
+              } else {
+                firstTimeMap.set(value, literalPath);
+                return;
+              }
+              (0, _assert.ok)(index !== -1);
+              me.changeData.literals++;
+              (0, _astUtils.ensureComputedExpression)(literalPath);
+              literalPath.replaceWith(createMemberExpression(index));
+              literalPath.skip();
+            }
+          });
+          if (arrayExpression.elements.length === 0) return;
 
-        // Delete with first location
-        var firstLocation = this.first.get(stringValue);
-        if (firstLocation) {
-          var index = this.map.size;
-          (0, _assert.ok)(!this.map.has(stringValue));
-          this.map.set(stringValue, index);
-          this.first.delete(stringValue);
-          var pushing = (0, _insert.clone)(object);
-          this.arrayExpression.elements.push(pushing);
-          (0, _assert.ok)(this.arrayExpression.elements[index] === pushing);
-          this.transformLiteral(firstLocation[0], firstLocation[1], index);
+          // Create the literals array declaration
+          var itemsArrayDeclaration = t.variableDeclaration("const", [t.variableDeclarator(t.identifier(arrayName), arrayExpression)]);
+          (0, _astUtils.prepend)(programPath, itemsArrayDeclaration);
         }
-        var index = this.map.get(stringValue);
-        (0, _assert.ok)(typeof index === "number");
-        this.transformLiteral(object, parents, index);
-        return;
       }
-
-      // Save this, maybe a duplicate will be found.
-      this.first.set(stringValue, [object, parents]);
-    };
-  }
-}
-exports.default = DuplicateLiteralsRemoval;
+    }
+  };
+};