Fixed #123 and #125, added tests for debug protection template

src/declarations/threads.d.ts

@@ -0,0 +1,15 @@
+declare module 'threads' {
+    type PostMessage <U> = (data: U) => void;
+    type SpawnCallback <T, U> = (data: T, postMessage: PostMessage <U>) => void;
+    type ResponseCallback <U> = (response: U) => void;
+
+    class Thread <T, U> {
+        public killed: boolean;
+
+        public send (data: T): Thread <T, U>;
+        public on (eventType: string, responseCallback: ResponseCallback<U>): Thread <T, U>;
+        public kill (): void;
+    }
+
+    export function spawn <T, U> (spawnCallback: SpawnCallback <T, U>): Thread <T, U>;
+}

src/enums/node/NodeType.ts

@@ -3,6 +3,7 @@ export enum NodeType {
     ArrowFunctionExpression = 'ArrowFunctionExpression',
     AssignmentExpression = 'AssignmentExpression',
     AssignmentPattern = 'AssignmentPattern',
+    AwaitExpression = 'AwaitExpression',
     BinaryExpression = 'BinaryExpression',
     BlockStatement = 'BlockStatement',
     BreakStatement = 'BreakStatement',

src/node-transformers/dead-code-injection-transformers/DeadCodeInjectionTransformer.ts

@@ -174,7 +174,8 @@ export class DeadCodeInjectionTransformer extends AbstractNodeTransformer {
                 if (
                     nestedBlockStatementsCount > DeadCodeInjectionTransformer.maxNestedBlockStatementsCount ||
                     NodeGuards.isBreakStatementNode(node) ||
-                    NodeGuards.isContinueStatementNode(node)
+                    NodeGuards.isContinueStatementNode(node) ||
+                    NodeGuards.isAwaitExpressionNode(node)
                 ) {
                     isValidBlockStatementNode = false;
 

src/node/NodeGuards.ts

@@ -21,6 +21,14 @@ export class NodeGuards {
         return node.type === NodeType.AssignmentPattern;
     }
 
+    /**
+     * @param {Node} node
+     * @returns {boolean}
+     */
+    public static isAwaitExpressionNode (node: ESTree.Node): node is ESTree.AwaitExpression {
+        return node.type === NodeType.AwaitExpression;
+    }
+
     /**
      * @param {Node} node
      * @returns {boolean}

src/templates/debug-protection-nodes/debug-protection-function-call-node/DebufProtectionFunctionCallTemplate.ts

@@ -5,7 +5,7 @@ export function DebugProtectionFunctionCallTemplate (): string {
     return `
         (function () {
             var regExp1 = new RegExp('function *\\\\( *\\\\)');
-            var regExp2 = new RegExp('\\\\+\\\\+ *_0x([a-f0-9]){4,6}');
+            var regExp2 = new RegExp('\\\\+\\\\+ *\\(?:_0x([a-f0-9]){4,6}|\\\\b[a-zA-Z]{1,2}\\\\b\\)');
             var result = {debugProtectionFunctionName}('init');
             
             if (!regExp1.test(result + 'chain') || !regExp2.test(result + 'input')) {