private String recompile(String str)
throws ActionParseException, IOException, CompilationException, InterruptedException,
TimeoutException {
ActionScriptParser par = new ActionScriptParser(SWF.DEFAULT_VERSION);
HighlightedTextWriter writer = new HighlightedTextWriter(new CodeFormatting(), false);
List<Action> actions = par.actionsFromString(str);
byte[] hex = Action.actionsToBytes(actions, true, SWF.DEFAULT_VERSION);
ActionList list =
ActionListReader.readActionListTimeout(
new ArrayList<DisassemblyListener>(),
new SWFInputStream(swf, hex),
SWF.DEFAULT_VERSION,
0,
hex.length,
"");
Action.actionsToSource(null, list, "", writer);
return writer.toString();
}
@Test(dataProvider = "provideFiles")
public void testDirectEditing(String filePath)
throws IOException, InterruptedException, AVM2ParseException, CompilationException {
File playerSWC = Configuration.getPlayerSWC();
if (playerSWC == null) {
throw new IOException(
"Player SWC library not found, please place it to " + Configuration.getFlashLibPath());
}
try {
SWF swf = new SWF(new BufferedInputStream(new FileInputStream(filePath)), 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);
ScriptPack en = abc.script_info.get(s).getPacks(abc, s, null, allAbcs).get(0);
String classPathString = en.getClassPath().toString();
if (startAfter == null || classPathString.equals(startAfter)) {
dotest = true;
}
if (!dotest) {
System.out.println("Skipped:" + classPathString);
continue;
}
System.out.println("Recompiling:" + classPathString + "...");
en.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.getClassPath().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: " + as + "/" + asm.toString(), ex);
}
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(), ex);
}
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) {
fail("Exception during decompilation: " + filePath, ex);
}
}