mirror of
https://git.huckle.dev/Huckles-Minecraft-Archive/jpexs-decompiler.git
synced 2026-06-01 06:05:25 +00:00
test fixed
This commit is contained in:
honfika@gmail.com
@@ -16,144 +16,52 @@
|
||||
*/
|
||||
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.Tag;
|
||||
import com.jpexs.decompiler.flash.tags.TagStub;
|
||||
import com.jpexs.decompiler.flash.tags.base.ASMSource;
|
||||
import com.jpexs.decompiler.graph.CompilationException;
|
||||
import com.jpexs.decompiler.graph.TranslateException;
|
||||
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 {
|
||||
public class RecompileTest extends FileTestBase {
|
||||
|
||||
@BeforeClass
|
||||
public void init() {
|
||||
//Main.initLogging(false);
|
||||
Configuration.autoDeobfuscate.set(false);
|
||||
}
|
||||
|
||||
public static final String TESTDATADIR = "testdata/recompile";
|
||||
|
||||
@Test(dataProvider = "provideFiles")
|
||||
public void testRecompile(String filename) {
|
||||
public void testRecompile(String fileName) {
|
||||
try {
|
||||
try (FileInputStream fis = new FileInputStream(TESTDATADIR + File.separator + filename)) {
|
||||
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);
|
||||
fail("File is different after recompiling: " + fileName);
|
||||
} catch (IOException | InterruptedException ex) {
|
||||
fail("Exception during decompilation: " + fileName + " " + ex.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
@Test(dataProvider = "provideFiles")
|
||||
public void testDirectEditing(String filename) throws IOException, InterruptedException, AVM2ParseException, CompilationException {
|
||||
Configuration.autoDeobfuscate.set(false);
|
||||
public void testTagEditing(String fileName) throws IOException, InterruptedException {
|
||||
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.getAbcList()) {
|
||||
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, abc.script_info.get(s).traits.traits, ScriptExportMode.AS, false);
|
||||
String original = htw.toString();
|
||||
com.jpexs.decompiler.flash.abc.avm2.parser.script.ActionScriptParser.compile(original, abc, 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(false);
|
||||
|
||||
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;
|
||||
}
|
||||
}
|
||||
|
||||
@Test(dataProvider = "provideFiles")
|
||||
public void testTagEditing(String filename) throws IOException, InterruptedException {
|
||||
try {
|
||||
SWF swf = new SWF(new BufferedInputStream(new FileInputStream(TESTDATADIR + File.separator + filename)), false);
|
||||
Configuration.debugCopy.set(false);
|
||||
SWF swf = new SWF(new BufferedInputStream(new FileInputStream(TESTDATADIR + File.separator + fileName)), false);
|
||||
for (Tag tag : swf.tags) {
|
||||
if (!(tag instanceof TagStub)) {
|
||||
Tag tag2 = tag.cloneTag();
|
||||
@@ -163,28 +71,12 @@ public class RecompileTest {
|
||||
}
|
||||
}
|
||||
} catch (Exception ex) {
|
||||
// ignore
|
||||
fail("Exception during decompilation: " + fileName + " " + ex.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
@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;
|
||||
@Override
|
||||
public String getTestDataDir() {
|
||||
return TESTDATADIR;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user