jpexs-decompiler/libsrc/ffdec_lib/test/com/jpexs/decompiler/flash/RecompileTest.java

/*
 *  Copyright (C) 2010-2014 JPEXS, All rights reserved.
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3.0 of the License, or (at your option) any later version.
 * 
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library.
 */
package com.jpexs.decompiler.flash;

import com.jpexs.decompiler.flash.abc.ABC;
import com.jpexs.decompiler.flash.abc.ClassPath;
import com.jpexs.decompiler.flash.abc.NotSameException;
import com.jpexs.decompiler.flash.abc.ScriptPack;
import com.jpexs.decompiler.flash.abc.avm2.parser.AVM2ParseException;
import com.jpexs.decompiler.flash.action.Action;
import com.jpexs.decompiler.flash.action.parser.ActionParseException;
import com.jpexs.decompiler.flash.action.parser.script.ActionScriptParser;
import com.jpexs.decompiler.flash.configuration.Configuration;
import com.jpexs.decompiler.flash.exporters.modes.ScriptExportMode;
import com.jpexs.decompiler.flash.helpers.CodeFormatting;
import com.jpexs.decompiler.flash.helpers.HighlightedTextWriter;
import com.jpexs.decompiler.flash.helpers.collections.MyEntry;
import com.jpexs.decompiler.flash.tags.ABCContainerTag;
import com.jpexs.decompiler.flash.tags.base.ASMSource;
import com.jpexs.decompiler.graph.CompilationException;
import com.jpexs.decompiler.graph.TranslateException;
import com.jpexs.helpers.Helper;
import java.io.BufferedInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FilenameFilter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import static org.testng.Assert.fail;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

/**
 *
 * @author JPEXS
 */
public class RecompileTest {

    @BeforeClass
    public void init() {
        //Main.initLogging(false);
    }

    public static final String TESTDATADIR = "testdata/recompile";

    @Test(dataProvider = "provideFiles")
    public void testRecompile(String filename) {
        try {
            try (FileInputStream fis = new FileInputStream(TESTDATADIR + File.separator + filename)) {
                Configuration.debugCopy.set(true);
                SWF swf = new SWF(new BufferedInputStream(fis), false);
                swf.saveTo(new ByteArrayOutputStream());
            }
        } catch (IOException | InterruptedException ex) {
            fail();
        } catch (NotSameException ex) {
            fail("File is different after recompiling: " + filename);
        }
    }

    @Test(dataProvider = "provideFiles")
    public void testDirectEditing(String filename) throws IOException, InterruptedException, AVM2ParseException, CompilationException {
        Configuration.autoDeobfuscate.set(false);
        try {
            SWF swf = new SWF(new BufferedInputStream(new FileInputStream(TESTDATADIR + File.separator + filename)), false);
            if (swf.isAS3) {
                boolean dotest = false;
                List<ABC> allAbcs = new ArrayList<>();
                for (ABCContainerTag ct : swf.abcList) {
                    allAbcs.add(ct.getABC());
                }
                for (ABC abc : allAbcs) {                   
                    for (int s = 0; s < abc.script_info.size(); s++) {
                        String startAfter = null;
                        HighlightedTextWriter htw = new HighlightedTextWriter(new CodeFormatting(), false);
                        MyEntry<ClassPath, ScriptPack> en = abc.script_info.get(s).getPacks(abc, s).get(0);
                        if (startAfter == null || en.getKey().toString().equals(startAfter)) {
                            dotest = true;
                        }
                        if (!dotest) {
                            System.out.println("Skipped:" + en.getKey().toString());
                            continue;
                        }

                        System.out.println("Recompiling:" + en.getKey().toString() + "...");
                        en.getValue().toSource(htw, swf.abcList, abc.script_info.get(s).traits.traits, ScriptExportMode.AS, false);
                        String original = htw.toString();
                        ABC nabc = abc; //new ABC(swf);
                        com.jpexs.decompiler.flash.abc.avm2.parser.script.ActionScriptParser.compile(original, nabc, allAbcs, false, en.getKey().className + ".as", abc.instance_info.size());
                        //remove last compiled script:
                        abc.script_info.remove(abc.script_info.size()-1);
                    }
                }
            } else {
                Map<String, ASMSource> asms = swf.getASMs();

                for (ASMSource asm : asms.values()) {
                    try {
                        HighlightedTextWriter writer = new HighlightedTextWriter(new CodeFormatting(), false);
                        Action.actionsToSource(asm, asm.getActions(), asm.toString()/*FIXME?*/, writer);
                        String as = writer.toString();
                        as = asm.removePrefixAndSuffix(as);
                        ActionScriptParser par = new ActionScriptParser(swf.version);
                        try {
                            asm.setActions(par.actionsFromString(as));
                        } catch (ActionParseException | CompilationException ex) {
                            fail("Unable to parse: " + asm.getSwf().getShortFileName() + "/" + asm.toString());
                        }
                        writer = new HighlightedTextWriter(new CodeFormatting(), false);
                        Action.actionsToSource(asm, asm.getActions(), asm.toString()/*FIXME?*/, writer);
                        String as2 = writer.toString();
                        as2 = asm.removePrefixAndSuffix(as2);
                        try {
                            asm.setActions(par.actionsFromString(as2));
                        } catch (ActionParseException | CompilationException ex) {
                            fail("Unable to parse: " + asm.getSwf().getShortFileName() + "/" + asm.toString());
                        }
                        writer = new HighlightedTextWriter(new CodeFormatting(), false);
                        Action.actionsToSource(asm, asm.getActions(), asm.toString()/*FIXME?*/, writer);
                        String as3 = writer.toString();
                        as3 = asm.removePrefixAndSuffix(as3);
                        if (!as3.equals(as2)) {
                            fail("ActionScript is different: " + asm.getSwf().getShortFileName() + "/" + asm.toString());
                        }
                    } catch (InterruptedException | IOException | OutOfMemoryError | TranslateException | StackOverflowError ex) {
                    }
                }
            }
        } catch (Exception ex) {
            System.out.println("FAIL");
            throw ex;
        }
    }

    @DataProvider(name = "provideFiles")
    public Object[][] provideFiles() {
        File dir = new File(TESTDATADIR);
        File[] files = new File[0];
        if (dir.exists()) {
            files = dir.listFiles(new FilenameFilter() {
                @Override
                public boolean accept(File dir, String name) {
                    return name.toLowerCase().endsWith(".swf");
                }
            });
        }
        Object[][] ret = new Object[files.length + 2][1];
        ret[0][0] = "..\\as2\\as2.swf";
        ret[1][0] = "..\\as3\\as3.swf";
        for (int f = 0; f < files.length; f++) {
            ret[f + 2][0] = files[f].getName();
        }
        return ret;
    }
}