|
@@ -1,31 +1,63 @@
|
|
|
+import { injectable, inject } from 'inversify';
|
|
|
+import { ServiceIdentifiers } from './container/ServiceIdentifiers';
|
|
|
+
|
|
|
import * as estraverse from 'estraverse';
|
|
|
import * as ESTree from 'estree';
|
|
|
|
|
|
+import { TNodeTransformersFactory } from './types/container/TNodeTransformersFactory';
|
|
|
import { TVisitorDirection } from './types/TVisitorDirection';
|
|
|
|
|
|
import { ICustomNode } from './interfaces/custom-nodes/ICustomNode';
|
|
|
-import { IObfuscationEventEmitter } from './interfaces/IObfuscationEventEmitter';
|
|
|
+import { IObfuscationEventEmitter } from './interfaces/event-emitters/IObfuscationEventEmitter';
|
|
|
import { IObfuscator } from './interfaces/IObfuscator';
|
|
|
-import { IOptions } from './interfaces/IOptions';
|
|
|
-import { INodeTransformer } from './interfaces/INodeTransformer';
|
|
|
-import { INodeTransformersFactory } from './interfaces/INodeTransformersFactory';
|
|
|
+import { IOptions } from './interfaces/options/IOptions';
|
|
|
+import { INodeTransformer } from './interfaces/node-transformers/INodeTransformer';
|
|
|
import { IStackTraceAnalyzer } from './interfaces/stack-trace-analyzer/IStackTraceAnalyzer';
|
|
|
-import { IStorage } from './interfaces/IStorage';
|
|
|
+import { IStorage } from './interfaces/storages/IStorage';
|
|
|
|
|
|
+import { NodeTransformers } from './enums/container/NodeTransformers';
|
|
|
+import { NodeType } from './enums/NodeType';
|
|
|
import { ObfuscationEvents } from './enums/ObfuscationEvents';
|
|
|
import { VisitorDirection } from './enums/VisitorDirection';
|
|
|
|
|
|
-import { NodeControlFlowTransformersFactory } from './node-transformers/node-control-flow-transformers/factory/NodeControlFlowTransformersFactory';
|
|
|
-import { NodeObfuscatorsFactory } from './node-transformers/node-obfuscators/factory/NodeObfuscatorsFactory';
|
|
|
-
|
|
|
import { Node } from './node/Node';
|
|
|
import { NodeUtils } from './node/NodeUtils';
|
|
|
+import { IStackTraceData } from './interfaces/stack-trace-analyzer/IStackTraceData';
|
|
|
|
|
|
+@injectable()
|
|
|
export class Obfuscator implements IObfuscator {
|
|
|
/**
|
|
|
- * @type {IStorage<ICustomNode>}
|
|
|
+ * @type {Map<string, NodeTransformers[]>}
|
|
|
+ */
|
|
|
+ private static readonly nodeControlFlowTransformersMap: Map <string, NodeTransformers[]> = new Map <string, NodeTransformers[]> ([
|
|
|
+ [NodeType.FunctionDeclaration, [NodeTransformers.FunctionControlFlowTransformer]],
|
|
|
+ [NodeType.FunctionExpression, [NodeTransformers.FunctionControlFlowTransformer]]
|
|
|
+ ]);
|
|
|
+
|
|
|
+ /**
|
|
|
+ * @type {Map<string, NodeTransformers[]>}
|
|
|
*/
|
|
|
- private readonly customNodesStorage: IStorage<ICustomNode>;
|
|
|
+ private static readonly nodeObfuscatorsMap: Map <string, NodeTransformers[]> = new Map <string, NodeTransformers[]> ([
|
|
|
+ [NodeType.ArrowFunctionExpression, [NodeTransformers.FunctionObfuscator]],
|
|
|
+ [NodeType.ClassDeclaration, [NodeTransformers.FunctionDeclarationObfuscator]],
|
|
|
+ [NodeType.CatchClause, [NodeTransformers.CatchClauseObfuscator]],
|
|
|
+ [NodeType.FunctionDeclaration, [
|
|
|
+ NodeTransformers.FunctionDeclarationObfuscator,
|
|
|
+ NodeTransformers.FunctionObfuscator
|
|
|
+ ]],
|
|
|
+ [NodeType.FunctionExpression, [NodeTransformers.FunctionObfuscator]],
|
|
|
+ [NodeType.MemberExpression, [NodeTransformers.MemberExpressionObfuscator]],
|
|
|
+ [NodeType.MethodDefinition, [NodeTransformers.MethodDefinitionObfuscator]],
|
|
|
+ [NodeType.ObjectExpression, [NodeTransformers.ObjectExpressionObfuscator]],
|
|
|
+ [NodeType.VariableDeclaration, [NodeTransformers.VariableDeclarationObfuscator]],
|
|
|
+ [NodeType.LabeledStatement, [NodeTransformers.LabeledStatementObfuscator]],
|
|
|
+ [NodeType.Literal, [NodeTransformers.LiteralObfuscator]]
|
|
|
+ ]);
|
|
|
+
|
|
|
+ /**
|
|
|
+ * @type {TNodeTransformersFactory}
|
|
|
+ */
|
|
|
+ private readonly nodeTransformersFactory: TNodeTransformersFactory;
|
|
|
|
|
|
/**
|
|
|
* @type {IObfuscationEventEmitter}
|
|
@@ -43,60 +75,64 @@ export class Obfuscator implements IObfuscator {
|
|
|
private readonly stackTraceAnalyzer: IStackTraceAnalyzer;
|
|
|
|
|
|
/**
|
|
|
- * @param obfuscationEventEmitter
|
|
|
* @param stackTraceAnalyzer
|
|
|
- * @param customNodesStorage
|
|
|
+ * @param obfuscationEventEmitter
|
|
|
+ * @param nodeTransformersFactory
|
|
|
* @param options
|
|
|
*/
|
|
|
constructor (
|
|
|
- obfuscationEventEmitter: IObfuscationEventEmitter,
|
|
|
- stackTraceAnalyzer: IStackTraceAnalyzer,
|
|
|
- customNodesStorage: IStorage<ICustomNode>,
|
|
|
- options: IOptions
|
|
|
+ @inject(ServiceIdentifiers.IStackTraceAnalyzer) stackTraceAnalyzer: IStackTraceAnalyzer,
|
|
|
+ @inject(ServiceIdentifiers.IObfuscationEventEmitter) obfuscationEventEmitter: IObfuscationEventEmitter,
|
|
|
+ @inject(ServiceIdentifiers['Factory<INodeTransformer[]>']) nodeTransformersFactory: TNodeTransformersFactory,
|
|
|
+ @inject(ServiceIdentifiers.IOptions) options: IOptions
|
|
|
) {
|
|
|
- this.obfuscationEventEmitter = obfuscationEventEmitter;
|
|
|
this.stackTraceAnalyzer = stackTraceAnalyzer;
|
|
|
- this.customNodesStorage = customNodesStorage;
|
|
|
+ this.obfuscationEventEmitter = obfuscationEventEmitter;
|
|
|
+ this.nodeTransformersFactory = nodeTransformersFactory;
|
|
|
this.options = options;
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
* @param astTree
|
|
|
+ * @param customNodesStorage
|
|
|
* @returns {ESTree.Program}
|
|
|
*/
|
|
|
- public obfuscateAstTree (astTree: ESTree.Program): ESTree.Program {
|
|
|
+ public obfuscateAstTree (astTree: ESTree.Program, customNodesStorage: IStorage<ICustomNode>): ESTree.Program {
|
|
|
if (Node.isProgramNode(astTree) && !astTree.body.length) {
|
|
|
return astTree;
|
|
|
}
|
|
|
|
|
|
- // zero pass: parentize all nodes
|
|
|
NodeUtils.parentize(astTree);
|
|
|
|
|
|
+ const stackTraceData: IStackTraceData[] = this.stackTraceAnalyzer.analyze(astTree.body);
|
|
|
+
|
|
|
// prepare custom nodes
|
|
|
- this.customNodesStorage.initialize(this.stackTraceAnalyzer.analyze(astTree.body));
|
|
|
- this.customNodesStorage
|
|
|
+ customNodesStorage.initialize(stackTraceData);
|
|
|
+ customNodesStorage
|
|
|
.getStorage()
|
|
|
.forEach((customNode: ICustomNode) => {
|
|
|
this.obfuscationEventEmitter.once(customNode.getAppendEvent(), customNode.appendNode.bind(customNode));
|
|
|
});
|
|
|
|
|
|
- this.obfuscationEventEmitter.emit(ObfuscationEvents.BeforeObfuscation, astTree);
|
|
|
+ this.obfuscationEventEmitter.emit(ObfuscationEvents.BeforeObfuscation, astTree, stackTraceData);
|
|
|
|
|
|
// first pass: control flow flattening
|
|
|
if (this.options.controlFlowFlattening) {
|
|
|
- this.transformAstTree(astTree, VisitorDirection.leave, new NodeControlFlowTransformersFactory(
|
|
|
- this.customNodesStorage,
|
|
|
- this.options
|
|
|
- ));
|
|
|
+ this.transformAstTree(
|
|
|
+ astTree,
|
|
|
+ VisitorDirection.leave,
|
|
|
+ this.nodeTransformersFactory(Obfuscator.nodeControlFlowTransformersMap)
|
|
|
+ );
|
|
|
}
|
|
|
|
|
|
// second pass: nodes obfuscation
|
|
|
- this.transformAstTree(astTree, VisitorDirection.enter, new NodeObfuscatorsFactory(
|
|
|
- this.customNodesStorage,
|
|
|
- this.options
|
|
|
- ));
|
|
|
+ this.transformAstTree(
|
|
|
+ astTree,
|
|
|
+ VisitorDirection.enter,
|
|
|
+ this.nodeTransformersFactory(Obfuscator.nodeObfuscatorsMap)
|
|
|
+ );
|
|
|
|
|
|
- this.obfuscationEventEmitter.emit(ObfuscationEvents.AfterObfuscation, astTree);
|
|
|
+ this.obfuscationEventEmitter.emit(ObfuscationEvents.AfterObfuscation, astTree, stackTraceData);
|
|
|
|
|
|
return astTree;
|
|
|
}
|
|
@@ -104,16 +140,16 @@ export class Obfuscator implements IObfuscator {
|
|
|
/**
|
|
|
* @param astTree
|
|
|
* @param direction
|
|
|
- * @param nodeTransformersFactory
|
|
|
+ * @param nodeTransformersConcreteFactory
|
|
|
*/
|
|
|
private transformAstTree (
|
|
|
astTree: ESTree.Program,
|
|
|
direction: TVisitorDirection,
|
|
|
- nodeTransformersFactory: INodeTransformersFactory
|
|
|
+ nodeTransformersConcreteFactory: (nodeType: string) => INodeTransformer[]
|
|
|
): void {
|
|
|
estraverse.traverse(astTree, {
|
|
|
[direction]: (node: ESTree.Node, parentNode: ESTree.Node): void => {
|
|
|
- const nodeTransformers: INodeTransformer[] = nodeTransformersFactory.initializeNodeTransformers(node.type);
|
|
|
+ const nodeTransformers: INodeTransformer[] = nodeTransformersConcreteFactory(node.type);
|
|
|
|
|
|
nodeTransformers.forEach((nodeTransformer: INodeTransformer) => {
|
|
|
nodeTransformer.transformNode(node, parentNode);
|