javascript-obfuscator/test/functional-tests/custom-code-helpers/self-defending/templates/SelfDefendingTemplate.spec.ts

import { assert } from 'chai';

import { TInputOptions } from '../../../../../src/types/options/TInputOptions';

import { IdentifierNamesGenerator } from '../../../../../src/enums/generators/identifier-names-generators/IdentifierNamesGenerator';

import { evaluateInWorker } from '../../../../helpers/evaluateInWorker';
import { beautifyCode } from '../../../../helpers/beautifyCode';
import { readFileAsString } from '../../../../helpers/readFileAsString';

import { JavaScriptObfuscator } from '../../../../../src/JavaScriptObfuscatorFacade';

describe('SelfDefendingTemplate', function () {
    const correctEvaluationTimeout: number = 100;
    const redosEvaluationTimeout: number = 10000;

    const baseOptions: TInputOptions = {
        optionsPreset: 'high-obfuscation',
        debugProtection: false
    };

    this.timeout(30000);

    describe('Variant #1: correctly obfuscate code with `HexadecimalIdentifierNamesGenerator``', () => {
        const expectedEvaluationResult: number = 1;

        let obfuscatedCode: string,
            evaluationResult: number = 0;

        before(() => {
            const code: string = readFileAsString(__dirname + '/fixtures/input.js');

            obfuscatedCode = JavaScriptObfuscator.obfuscate(
                code,
                {
                    ...baseOptions,
                    selfDefending: true,
                    identifierNamesGenerator: IdentifierNamesGenerator.HexadecimalIdentifierNamesGenerator
                }
            ).getObfuscatedCode();

            return evaluateInWorker(obfuscatedCode, correctEvaluationTimeout)
                .then((result: string | null) => {
                    if (!result) {
                        return;
                    }

                    evaluationResult = parseInt(result, 10);
                });
        });

        it('should correctly evaluate code with enabled self defending', () => {
            assert.equal(evaluationResult, expectedEvaluationResult);
        });
    });

    describe('Variant #2: correctly obfuscate code with `MangledIdentifierNamesGenerator` option', () => {
        const expectedEvaluationResult: number = 1;

        let obfuscatedCode: string,
            evaluationResult: number = 0;

        before(() => {
            const code: string = readFileAsString(__dirname + '/fixtures/input.js');

            obfuscatedCode = JavaScriptObfuscator.obfuscate(
                code,
                {
                    baseOptions,
                    selfDefending: true,
                    identifierNamesGenerator: IdentifierNamesGenerator.MangledIdentifierNamesGenerator
                }
            ).getObfuscatedCode();

            return evaluateInWorker(obfuscatedCode, correctEvaluationTimeout)
                .then((result: string | null) => {
                    if (!result) {
                        return;
                    }

                    evaluationResult = parseInt(result, 10);
                });
        });

        it('should correctly evaluate code with enabled self defending', () => {
            assert.equal(evaluationResult, expectedEvaluationResult);
        });
    });

    describe('Variant #3: correctly obfuscate code with `DictionaryIdentifierNamesGenerator` option', () => {
        const expectedEvaluationResult: number = 1;

        let obfuscatedCode: string,
            evaluationResult: number = 0;

        before(() => {
            const code: string = readFileAsString(__dirname + '/fixtures/input.js');

            obfuscatedCode = JavaScriptObfuscator.obfuscate(
                code,
                {
                    baseOptions,
                    selfDefending: true,
                    identifierNamesGenerator: IdentifierNamesGenerator.DictionaryIdentifierNamesGenerator,
                    identifiersDictionary: ['foo', 'bar', 'baz', 'bark', 'hawk', 'eagle']
                }
            ).getObfuscatedCode();

            return evaluateInWorker(obfuscatedCode, correctEvaluationTimeout)
                .then((result: string | null) => {
                    if (!result) {
                        return;
                    }

                    evaluationResult = parseInt(result, 10);
                });
        });

        it('should correctly evaluate code with enabled self defending', () => {
            assert.equal(evaluationResult, expectedEvaluationResult);
        });
    });

    describe('Variant #4: obfuscated code with beautified self defending code', () => {
        describe('Variant #1: beautify with spaces', () => {
            const expectedEvaluationResult: number = 0;

            let obfuscatedCode: string,
                evaluationResult: number = 0;

            before(() => {
                const code: string = readFileAsString(__dirname + '/fixtures/input.js');

                obfuscatedCode = JavaScriptObfuscator.obfuscate(
                    code,
                    {
                        ...baseOptions,
                        selfDefending: true
                    }
                ).getObfuscatedCode();
                obfuscatedCode = beautifyCode(obfuscatedCode, 'space');

                return evaluateInWorker(obfuscatedCode, redosEvaluationTimeout)
                    .then((result: string | null) => {
                        if (!result) {
                            return;
                        }

                        evaluationResult = parseInt(result, 10);
                    });
            });

            it('should enter code in infinity loop', () => {
                assert.equal(evaluationResult, expectedEvaluationResult);
            });
        });

        describe('Variant #2: beautify with tabs', () => {
            const expectedEvaluationResult: number = 0;

            let obfuscatedCode: string,
                evaluationResult: number = 0;

            before(() => {
                const code: string = readFileAsString(__dirname + '/fixtures/input.js');

                obfuscatedCode = JavaScriptObfuscator.obfuscate(
                    code,
                    {
                        ...baseOptions,
                        selfDefending: true
                    }
                ).getObfuscatedCode();
                obfuscatedCode = beautifyCode(obfuscatedCode, 'tab');

                return evaluateInWorker(obfuscatedCode, redosEvaluationTimeout)
                    .then((result: string | null) => {
                        if (!result) {
                            return;
                        }

                        evaluationResult = parseInt(result, 10);
                    });
            });

            it('should enter code in infinity loop', () => {
                assert.equal(evaluationResult, expectedEvaluationResult);
            });
        });
    });

    describe('Variant #5: JavaScript obfuscator code', () => {
        describe('Variant #1: correct evaluation', () => {
            const evaluationTimeout: number = 5000;
            const expectedEvaluationResult: string = 'var foo=0x1;';

            let obfuscatedCode: string,
                evaluationResult: string = '';

            before(() => {
                const code: string = readFileAsString(process.cwd() + '/dist/index.js');

                obfuscatedCode = JavaScriptObfuscator.obfuscate(
                    `
                        ${code}
                        module.exports.obfuscate('var foo = 1;').getObfuscatedCode();
                    `,
                    {
                        selfDefending: true,
                        identifierNamesGenerator: IdentifierNamesGenerator.HexadecimalIdentifierNamesGenerator
                    }
                ).getObfuscatedCode();

                return evaluateInWorker(obfuscatedCode, evaluationTimeout)
                    .then((result: string | null) => {
                        if (!result) {
                            return;
                        }

                        evaluationResult = result;
                    });
            });

            it('should correctly evaluate code with enabled self defending', () => {
                assert.equal(evaluationResult, expectedEvaluationResult);
            });
        });

        describe('Variant #2: beautify with spaces', () => {
            const expectedEvaluationResult: string = '';

            let obfuscatedCode: string,
                evaluationResult: string = '';

            before(() => {
                const code: string = readFileAsString(process.cwd() + '/dist/index.js');

                obfuscatedCode = JavaScriptObfuscator.obfuscate(
                    `
                        ${code}
                        module.exports.obfuscate('var foo = 1;').getObfuscatedCode();
                    `,
                    {
                        selfDefending: true,
                        identifierNamesGenerator: IdentifierNamesGenerator.HexadecimalIdentifierNamesGenerator
                    }
                ).getObfuscatedCode();
                obfuscatedCode = beautifyCode(obfuscatedCode, 'space');

                return evaluateInWorker(obfuscatedCode, redosEvaluationTimeout)
                    .then((result: string | null) => {
                        if (!result) {
                            return;
                        }

                        evaluationResult = result;
                    });
            });

            it('should enter code in infinity loop', () => {
                assert.equal(evaluationResult, expectedEvaluationResult);
            });
        });
    });
});