diff --git a/test/com/jpexs/decompiler/flash/gui/AdobeFlashExecutor.java b/test/com/jpexs/decompiler/flash/gui/AdobeFlashExecutor.java index 4a68fd922..a0543e673 100644 --- a/test/com/jpexs/decompiler/flash/gui/AdobeFlashExecutor.java +++ b/test/com/jpexs/decompiler/flash/gui/AdobeFlashExecutor.java @@ -1,300 +1,299 @@ -/* - * Copyright (C) 2010-2015 JPEXS - * - * This program is free software: you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation, either version 3 of the License, or - * (at your option) any later version. - * - * This program 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 General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program. If not, see . - */ -package com.jpexs.decompiler.flash.gui; - -import com.jpexs.decompiler.flash.SWF; -import com.jpexs.decompiler.flash.abc.ABC; -import com.jpexs.decompiler.flash.abc.avm2.AVM2Code; -import com.jpexs.decompiler.flash.abc.avm2.instructions.AVM2Instruction; -import com.jpexs.decompiler.flash.abc.avm2.instructions.AVM2Instructions; -import com.jpexs.decompiler.flash.abc.types.MethodBody; -import com.jpexs.decompiler.flash.abc.types.MethodInfo; -import com.jpexs.decompiler.flash.abc.types.Multiname; -import com.jpexs.decompiler.flash.abc.types.Namespace; -import com.jpexs.decompiler.flash.abc.types.traits.TraitMethodGetterSetter; -import com.jpexs.decompiler.flash.action.Action; -import com.jpexs.decompiler.flash.action.ActionList; -import com.jpexs.decompiler.flash.tags.DoABC2Tag; -import com.jpexs.decompiler.flash.tags.Tag; -import com.jpexs.decompiler.flash.tags.base.ASMSource; -import com.jpexs.javactivex.ActiveX; -import com.jpexs.javactivex.ActiveXEvent; -import com.jpexs.javactivex.ActiveXEventListener; -import com.jpexs.javactivex.Reference; -import com.jpexs.javactivex.example.controls.flash.ShockwaveFlash; -import java.awt.Panel; -import java.io.BufferedInputStream; -import java.io.BufferedOutputStream; -import java.io.File; -import java.io.FileInputStream; -import java.io.FileOutputStream; -import java.io.IOException; -import java.io.InputStream; -import java.io.OutputStream; -import java.util.ArrayList; -import java.util.Date; -import java.util.List; -import java.util.Map; -import java.util.concurrent.atomic.AtomicInteger; -import java.util.logging.Level; -import java.util.logging.Logger; - -/** - * - * @author JPEXS - */ -public class AdobeFlashExecutor { - - private static final AtomicInteger id = new AtomicInteger(); - - private final Object lockObj = new Object(); - - private final Reference resultRef = new Reference<>(null); - - private final File runFileAs2; - - private final File runFileAs3; - - private final ShockwaveFlash flash; - - public DoABC2Tag as3TestSwfAbcTag; - - public AdobeFlashExecutor() { - flash = ActiveX.createObject(ShockwaveFlash.class, new Panel()); - flash.setAllowScriptAccess("always"); - flash.setAllowNetworking("all"); - flash.addFSCommandListener(new ActiveXEventListener() { - - @Override - public void onEvent(ActiveXEvent axe) { - resultRef.setVal((String) axe.args.get("args")); - synchronized (lockObj) { - lockObj.notify(); - } - } - }); - - runFileAs2 = new File("libsrc/ffdec_lib/testdata/run_as2/run_as2.swf"); - runFileAs3 = new File("libsrc/ffdec_lib/testdata/run_as3/run.swf"); - } - - public String executeActionList(List actionsToExecute) { - try { - File f2 = new File("run_test_" + new Date().getTime() + "_" + id.getAndIncrement() + ".swf"); - f2.deleteOnExit(); - - SWF swf; - try (InputStream is = new BufferedInputStream(new FileInputStream(runFileAs2))) { - swf = new SWF(is, false); - } - - Map asms = swf.getASMs(true); - ASMSource asm = asms.get("\\frame_1\\DoAction"); - ActionList actions = asm.getActions(); - actions.removeAction(2, 4); - - actions.addActions(2, actionsToExecute); - - asm.setActions(actions); - asm.setModified(); - try (OutputStream fos = new BufferedOutputStream(new FileOutputStream(f2))) { - swf.saveTo(fos); - } - - flash.setMovie(f2.getAbsolutePath()); - - synchronized (lockObj) { - lockObj.wait(); - } - - //String str = flash.GetVariable("myText.text"); - f2.delete(); - - String flashResult = resultRef.getVal(); - return flashResult; - } catch (IOException | InterruptedException ex) { - Logger.getLogger(AdobeFlashExecutor.class.getName()).log(Level.SEVERE, null, ex); - } - - return null; - } - - public void loadTestSwf() throws IOException, InterruptedException { - SWF swf = new SWF(new BufferedInputStream(new FileInputStream(runFileAs3)), false); - swf.version = SWF.MAX_VERSION; - DoABC2Tag abcTag = null; - for (Tag t : swf.tags) { - if (t instanceof DoABC2Tag) { - abcTag = ((DoABC2Tag) t); - break; - } - } - - as3TestSwfAbcTag = abcTag; - } - - public String executeAvm2(AVM2Code code) { - try { - File f2 = new File("run_test_" + new Date().getTime() + "_" + id.getAndIncrement() + ".swf"); - f2.deleteOnExit(); - - if (as3TestSwfAbcTag == null) { - loadTestSwf(); - } - - DoABC2Tag abcTag = as3TestSwfAbcTag; - ABC abc = abcTag.getABC(); - MethodBody body = abc.findBodyByClassAndName("Run", "run"); - body.max_stack = 20; - body.max_regs = 10; - - body.setCode(code); - body.markOffsets(); - - abcTag.setModified(true); - - try (OutputStream fos = new BufferedOutputStream(new FileOutputStream(f2))) { - abcTag.getSwf().saveTo(fos); - } - - flash.setMovie(f2.getAbsolutePath()); - - synchronized (lockObj) { - lockObj.wait(); - } - - f2.delete(); - - String flashResult = resultRef.getVal(); - return flashResult; - - /*int cnt = 0; - while (flash.getReadyState() != 4) { - Thread.sleep(50); - if (cnt > 100) { - fail("Flash init timeout"); - } - - cnt++; - }*/ - /*try { - String res = flash.CallFunction("something"); - //String str = flash.GetVariable("_root.myText.text"); - throw new Error(res + " " + body.getCode().toString() + ""); - } catch (Exception ex) { - int a = 1; - }*/ - } catch (IOException | InterruptedException ex) { - Logger.getLogger(AdobeFlashExecutor.class.getName()).log(Level.SEVERE, null, ex); - } - - return null; - } - - public void executeAvm2(List tasks) { - try { - File f2 = new File("run_test_" + new Date().getTime() + "_" + id.getAndIncrement() + ".swf"); - f2.deleteOnExit(); - - if (as3TestSwfAbcTag == null) { - loadTestSwf(); - } - - DoABC2Tag abcTag = as3TestSwfAbcTag; - ABC abc = abcTag.getABC(); - int classId = abc.findClassByName("Run"); - MethodBody body = abc.findBodyByClassAndName("Run", "run"); - body.max_stack = 20; - body.max_regs = 10; - - Multiname multiname = new Multiname(); - multiname.kind = Multiname.QNAME; - multiname.name_index = abc.constants.getStringId("executeStaticMethod", true); - multiname.namespace_index = abc.constants.getNamespaceId(Namespace.KIND_PACKAGE, "", 0, true); - int multinameId = abc.constants.getMultinameId(multiname, true); - - AVM2Code ccode = new AVM2Code(); - ccode.code = new ArrayList<>(); - List code = ccode.code; - code.add(new AVM2Instruction(0, AVM2Instructions.GetLocal0, null)); - code.add(new AVM2Instruction(0, AVM2Instructions.PushScope, null)); - code.add(new AVM2Instruction(0, AVM2Instructions.PushString, new int[]{abc.constants.getStringId("", true)})); - - int i = 0; - for (AS3ExecuteTask task : tasks) { - String testName = "test_" + i; - code.add(new AVM2Instruction(0, AVM2Instructions.FindPropertyStrict, new int[]{multinameId})); - code.add(new AVM2Instruction(0, AVM2Instructions.PushString, new int[]{abc.constants.getStringId(testName, true)})); - code.add(new AVM2Instruction(0, AVM2Instructions.CallProperty, new int[]{multinameId, 1})); - code.add(new AVM2Instruction(0, AVM2Instructions.Add, null)); - - addMethod(abc, classId, testName, true, task.code); - i++; - } - - code.add(new AVM2Instruction(0, AVM2Instructions.ReturnValue, null)); - body.setCode(ccode); - body.markOffsets(); - - abcTag.setModified(true); - - try (OutputStream fos = new BufferedOutputStream(new FileOutputStream(f2))) { - abcTag.getSwf().saveTo(fos); - } - - flash.setMovie(f2.getAbsolutePath()); - - synchronized (lockObj) { - lockObj.wait(); - } - - f2.delete(); - - //String str = flash.GetVariable("myText.text"); - String flashResult = resultRef.getVal(); - if (!flashResult.startsWith("Result:") || !flashResult.endsWith(" Type:string")) { - return; - } - - flashResult = flashResult.substring(7); - flashResult = flashResult.substring(0, flashResult.length() - 12); - String[] lines = flashResult.split("(\r\n|\r|\n)"); - if (lines.length == tasks.size()) { - for (int j = 0; j < tasks.size(); j++) { - tasks.get(j).flashResult = lines[j]; - } - } - } catch (IOException | InterruptedException ex) { - Logger.getLogger(AdobeFlashExecutor.class.getName()).log(Level.SEVERE, null, ex); - } - } - - private int addMethod(ABC abc, int classId, String name, boolean isStatic, AVM2Code code) { - TraitMethodGetterSetter methodTrait = abc.addMethod(classId, name, isStatic); - MethodInfo methodInfo = abc.method_info.get(methodTrait.method_info); - MethodBody methodBody = methodInfo.getBody(); - methodBody.max_stack = 10; - methodBody.max_regs = 10; - methodBody.init_scope_depth = 3; - methodBody.max_scope_depth = 10; - - methodBody.setCode(code); - methodBody.markOffsets(); - - return methodTrait.name_index; - } -} +/* + * Copyright (C) 2010-2015 JPEXS + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program 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 General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ +package com.jpexs.decompiler.flash.gui; + +import com.jpexs.decompiler.flash.SWF; +import com.jpexs.decompiler.flash.abc.ABC; +import com.jpexs.decompiler.flash.abc.avm2.AVM2Code; +import com.jpexs.decompiler.flash.abc.avm2.instructions.AVM2Instruction; +import com.jpexs.decompiler.flash.abc.avm2.instructions.AVM2Instructions; +import com.jpexs.decompiler.flash.abc.types.MethodBody; +import com.jpexs.decompiler.flash.abc.types.MethodInfo; +import com.jpexs.decompiler.flash.abc.types.Multiname; +import com.jpexs.decompiler.flash.abc.types.Namespace; +import com.jpexs.decompiler.flash.abc.types.traits.TraitMethodGetterSetter; +import com.jpexs.decompiler.flash.action.Action; +import com.jpexs.decompiler.flash.action.ActionList; +import com.jpexs.decompiler.flash.tags.DoABC2Tag; +import com.jpexs.decompiler.flash.tags.Tag; +import com.jpexs.decompiler.flash.tags.base.ASMSource; +import com.jpexs.javactivex.ActiveX; +import com.jpexs.javactivex.ActiveXEvent; +import com.jpexs.javactivex.ActiveXEventListener; +import com.jpexs.javactivex.Reference; +import com.jpexs.javactivex.example.controls.flash.ShockwaveFlash; +import java.awt.Panel; +import java.io.BufferedInputStream; +import java.io.BufferedOutputStream; +import java.io.File; +import java.io.FileInputStream; +import java.io.FileOutputStream; +import java.io.IOException; +import java.io.InputStream; +import java.io.OutputStream; +import java.util.ArrayList; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.atomic.AtomicInteger; +import java.util.logging.Level; +import java.util.logging.Logger; + +/** + * + * @author JPEXS + */ +public class AdobeFlashExecutor { + + private static final AtomicInteger id = new AtomicInteger(); + + private final Object lockObj = new Object(); + + private final Reference resultRef = new Reference<>(null); + + private final File runFileAs2; + + private final File runFileAs3; + + private final ShockwaveFlash flash; + + public DoABC2Tag as3TestSwfAbcTag; + + public AdobeFlashExecutor() { + flash = ActiveX.createObject(ShockwaveFlash.class, new Panel()); + flash.setAllowScriptAccess("always"); + flash.setAllowNetworking("all"); + flash.addFSCommandListener(new ActiveXEventListener() { + + @Override + public void onEvent(ActiveXEvent axe) { + resultRef.setVal((String) axe.args.get("args")); + synchronized (lockObj) { + lockObj.notify(); + } + } + }); + + runFileAs2 = new File("libsrc/ffdec_lib/testdata/run_as2/run_as2.swf"); + runFileAs3 = new File("libsrc/ffdec_lib/testdata/run_as3/run.swf"); + } + + public String executeActionList(List actionsToExecute) { + try { + File f2 = new File("run_test_" + new Date().getTime() + "_" + id.getAndIncrement() + ".swf"); + f2.deleteOnExit(); + + SWF swf; + try (InputStream is = new BufferedInputStream(new FileInputStream(runFileAs2))) { + swf = new SWF(is, false); + } + + Map asms = swf.getASMs(true); + ASMSource asm = asms.get("\\frame_1\\DoAction"); + ActionList actions = asm.getActions(); + actions.removeAction(2, 4); + + actions.addActions(2, actionsToExecute); + + asm.setActions(actions); + asm.setModified(); + try (OutputStream fos = new BufferedOutputStream(new FileOutputStream(f2))) { + swf.saveTo(fos); + } + + flash.setMovie(f2.getAbsolutePath()); + + synchronized (lockObj) { + lockObj.wait(); + } + + //String str = flash.GetVariable("myText.text"); + f2.delete(); + + String flashResult = resultRef.getVal(); + return flashResult; + } catch (IOException | InterruptedException ex) { + Logger.getLogger(AdobeFlashExecutor.class.getName()).log(Level.SEVERE, null, ex); + } + + return null; + } + + public void loadTestSwf() throws IOException, InterruptedException { + SWF swf = new SWF(new BufferedInputStream(new FileInputStream(runFileAs3)), false); + swf.version = SWF.MAX_VERSION; + DoABC2Tag abcTag = null; + for (Tag t : swf.tags) { + if (t instanceof DoABC2Tag) { + abcTag = ((DoABC2Tag) t); + break; + } + } + + as3TestSwfAbcTag = abcTag; + } + + public String executeAvm2(AVM2Code code) { + try { + File f2 = new File("run_test_" + new Date().getTime() + "_" + id.getAndIncrement() + ".swf"); + f2.deleteOnExit(); + + if (as3TestSwfAbcTag == null) { + loadTestSwf(); + } + + DoABC2Tag abcTag = as3TestSwfAbcTag; + ABC abc = abcTag.getABC(); + MethodBody body = abc.findBodyByClassAndName("Run", "run"); + body.max_stack = 20; + body.max_regs = 10; + + body.setCode(code); + body.markOffsets(); + + abcTag.setModified(true); + + try (OutputStream fos = new BufferedOutputStream(new FileOutputStream(f2))) { + abcTag.getSwf().saveTo(fos); + } + + flash.setMovie(f2.getAbsolutePath()); + + synchronized (lockObj) { + lockObj.wait(); + } + + f2.delete(); + + String flashResult = resultRef.getVal(); + return flashResult; + + /*int cnt = 0; + while (flash.getReadyState() != 4) { + Thread.sleep(50); + if (cnt > 100) { + fail("Flash init timeout"); + } + + cnt++; + }*/ + /*try { + String res = flash.CallFunction("something"); + //String str = flash.GetVariable("_root.myText.text"); + throw new Error(res + " " + body.getCode().toString() + ""); + } catch (Exception ex) { + int a = 1; + }*/ + } catch (IOException | InterruptedException ex) { + Logger.getLogger(AdobeFlashExecutor.class.getName()).log(Level.SEVERE, null, ex); + } + + return null; + } + + public void executeAvm2(List tasks) { + try { + File f2 = new File("run_test_" + new Date().getTime() + "_" + id.getAndIncrement() + ".swf"); + f2.deleteOnExit(); + + if (as3TestSwfAbcTag == null) { + loadTestSwf(); + } + + DoABC2Tag abcTag = as3TestSwfAbcTag; + ABC abc = abcTag.getABC(); + int classId = abc.findClassByName("Run"); + MethodBody body = abc.findBodyByClassAndName("Run", "run"); + body.max_stack = 20; + body.max_regs = 10; + + Multiname multiname = new Multiname(); + multiname.kind = Multiname.QNAME; + multiname.name_index = abc.constants.getStringId("executeStaticMethod", true); + multiname.namespace_index = abc.constants.getNamespaceId(Namespace.KIND_PACKAGE, "", 0, true); + int multinameId = abc.constants.getMultinameId(multiname, true); + + AVM2Code ccode = new AVM2Code(); + ccode.code = new ArrayList<>(); + List code = ccode.code; + code.add(new AVM2Instruction(0, AVM2Instructions.GetLocal0, null)); + code.add(new AVM2Instruction(0, AVM2Instructions.PushScope, null)); + code.add(new AVM2Instruction(0, AVM2Instructions.PushString, new int[]{abc.constants.getStringId("", true)})); + + int i = 0; + for (AS3ExecuteTask task : tasks) { + String testName = "test_" + i; + code.add(new AVM2Instruction(0, AVM2Instructions.FindPropertyStrict, new int[]{multinameId})); + code.add(new AVM2Instruction(0, AVM2Instructions.PushString, new int[]{abc.constants.getStringId(testName, true)})); + code.add(new AVM2Instruction(0, AVM2Instructions.CallProperty, new int[]{multinameId, 1})); + code.add(new AVM2Instruction(0, AVM2Instructions.Add, null)); + + addMethod(abc, classId, testName, true, task.code); + i++; + } + + code.add(new AVM2Instruction(0, AVM2Instructions.ReturnValue, null)); + body.setCode(ccode); + body.markOffsets(); + + abcTag.setModified(true); + + try (OutputStream fos = new BufferedOutputStream(new FileOutputStream(f2))) { + abcTag.getSwf().saveTo(fos); + } + + flash.setMovie(f2.getAbsolutePath()); + + synchronized (lockObj) { + lockObj.wait(); + } + + f2.delete(); + + String flashResult = resultRef.getVal(); + if (!flashResult.startsWith("Result:") || !flashResult.endsWith(" Type:string")) { + return; + } + + flashResult = flashResult.substring(7); + flashResult = flashResult.substring(0, flashResult.length() - 12); + String[] lines = flashResult.split("(\r\n|\r|\n)"); + if (lines.length == tasks.size()) { + for (int j = 0; j < tasks.size(); j++) { + tasks.get(j).flashResult = lines[j]; + } + } + } catch (IOException | InterruptedException ex) { + Logger.getLogger(AdobeFlashExecutor.class.getName()).log(Level.SEVERE, null, ex); + } + } + + private int addMethod(ABC abc, int classId, String name, boolean isStatic, AVM2Code code) { + TraitMethodGetterSetter methodTrait = abc.addMethod(classId, name, isStatic); + MethodInfo methodInfo = abc.method_info.get(methodTrait.method_info); + MethodBody methodBody = methodInfo.getBody(); + methodBody.max_stack = 10; + methodBody.max_regs = 10; + methodBody.init_scope_depth = 3; + methodBody.max_scope_depth = 10; + + methodBody.setCode(code); + methodBody.markOffsets(); + + return methodTrait.name_index; + } +}