-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathControlFlow.js
130 lines (98 loc) · 3.54 KB
/
ControlFlow.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
const fs = require('fs');
const parser = require("@babel/parser");
const traverse = require("@babel/traverse").default;
const types = require("@babel/types");
const generator = require("@babel/generator").default;
//将源代码解析为AST
process.argv.length > 2?encodeFile = process.argv[2]:encodeFile = "./encode.js";
process.argv.length > 3?decodeFile = process.argv[3]:decodeFile = "./decode_result.js";
sourceCode = fs.readFileSync(encodeFile, {encoding: "utf-8"});
ast = parser.parse(sourceCode);
function getRandomArray(length)
{
let puzzleArr = [];
for (var i = 0; i < length; i++) {
puzzleArr.push(i)
}
puzzleArr = puzzleArr.sort(() => {
return Math.random() - 0.5
});
return puzzleArr;
}
function getRandomName(length)
{
let puzzleArr = ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "a", "b", "c", "d", "e", "f","g"];
let randomName = "_0x";
for (var i=0;i<length;i++)
{
randomName += puzzleArr[Math.round(Math.random() * 16)];
}
return randomName;
}
let names = [];
const renameIdentifier =
{
"Identifier"({node,scope})
{
let oldName = node.name,newName;
if (oldName.startsWith("_0x")) return;
do
{
newName = getRandomName(6);
}while (names.includes(newName))
scope.rename(oldName,newName);
},
}
traverse(ast, renameIdentifier);
const codeControlFlow =
{
"FunctionDeclaration|FunctionExpression|ForStatement"({node})
{
let body = node.body.body;
if (!body || body.length == 0) return;
let newArray = getRandomArray(body.length);
let firstVarName = getRandomName(6);
let secondVarName = getRandomName(6);
let firstVarNode = types.VariableDeclarator(types.Identifier(firstVarName),types.valueToNode(newArray));
let secodeVarNode = types.VariableDeclarator(types.Identifier(secondVarName),types.valueToNode(0));
let firstNode = types.VariableDeclaration("var",[firstVarNode,secodeVarNode]);
let Unarynode = types.UnaryExpression("!",types.ArrayExpression(),true);
let testNode = types.UnaryExpression("!",Unarynode,true);
let updateNode = types.UpdateExpression("++",types.Identifier(secondVarName),false);
let discriminNode = types.MemberExpression(types.Identifier(firstVarName),updateNode,true);
let cases = [];
for (let index = 0;index < body.length; index++)
{
let testNode = types.valueToNode(String(newArray[index]));
let consequent = [body[index],types.ContinueStatement()];
let SwitchCaseNode = types.SwitchCase(testNode,consequent);
cases[newArray[index]] = SwitchCaseNode;
}
let switchNode = types.SwitchStatement(discriminNode,cases);
let WhileNode = types.WhileStatement(testNode,types.BlockStatement([switchNode,types.BreakStatement()]));
node.body.body = [firstNode,WhileNode];
},
}
traverse(ast, codeControlFlow);
const arrayToCall =
{
ArrayExpression(path)
{
let elements = path.node.elements;
if (elements.length == 0) return;
if (!elements.every(element => types.isNumericLiteral(element)))
{
return;
}
let arr = [];
elements.forEach((ele,index) =>{arr[index] = ele.value;});
let newString = arr.join("|");
let memberNode = types.MemberExpression(types.valueToNode(newString),types.valueToNode("split"),true);
let argumentNode = types.valueToNode("|");
let callNode = types.CallExpression(memberNode,[argumentNode]);
path.replaceWith(callNode);
}
}
traverse(ast, arrayToCall);
let {code} = generator(ast);
fs.writeFile(decodeFile, code, (err) => {});