/*
* Copyright (C) 2010-2016 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.action.Action;
import com.jpexs.decompiler.flash.action.ActionList;
import com.jpexs.decompiler.flash.action.ActionListReader;
import com.jpexs.decompiler.flash.action.parser.ActionParseException;
import com.jpexs.decompiler.flash.action.parser.pcode.ASMParser;
import com.jpexs.decompiler.flash.action.parser.script.ActionScript2Parser;
import com.jpexs.decompiler.flash.configuration.Configuration;
import com.jpexs.decompiler.flash.helpers.CodeFormatting;
import com.jpexs.decompiler.flash.helpers.HighlightedTextWriter;
import com.jpexs.decompiler.flash.tags.DoActionTag;
import com.jpexs.decompiler.graph.CompilationException;
import java.io.BufferedInputStream;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeoutException;
import static org.testng.Assert.assertTrue;
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 ActionScript2DeobfuscatorTest extends ActionScript2TestBase {
@BeforeClass
public void init() throws IOException, InterruptedException {
//Main.initLogging(false);
Configuration.autoDeobfuscate.set(true);
swf = new SWF(new BufferedInputStream(new FileInputStream("testdata/as2/as2.swf")), false);
}
private String recompile(String str) throws ActionParseException, IOException, CompilationException, InterruptedException, TimeoutException {
ActionScript2Parser par = new ActionScript2Parser(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<>(), new SWFInputStream(swf, hex), SWF.DEFAULT_VERSION, 0, hex.length, "", 1);
Action.actionsToSource(null, list, "", writer);
return writer.toString();
}
@DataProvider(name = "provideBasicTrueExpressions")
public Object[][] provideBasicTrueExpressions() {
return new Object[][]{
{"1!=5"}, {"5==5"}, {"1<4"}, {"5>4"}, {"5*6==30"}
};
}
@DataProvider(name = "provideBasicFalseExpressions")
public Object[][] provideBasicFalseExpressions() {
return new Object[][]{
{"1==5"}, {"5!=5"}, {"1>4"}, {"5<4"}, {"5*7==12"}
};
}
@Test(dataProvider = "provideBasicTrueExpressions")
public void testRemoveBasicTrueExpressions(String expression) throws ActionParseException, IOException, CompilationException, InterruptedException, TimeoutException {
String res = recompile("if(" + expression + "){"
+ "trace(\"OK\");"
+ "} else {"
+ "trace(\"FAIL\");"
+ "}");
if (res.contains("\"FAIL\"")) {
fail("OnFalse clause was not removed: " + res);
}
if (!res.contains("\"OK\"")) {
fail("OnTrue clause was removed: " + res);
}
}
@Test(dataProvider = "provideBasicFalseExpressions")
public void testRemoveBasicFalseExpressions(String expression) throws Exception {
String res = recompile("if(" + expression + "){"
+ "trace(\"FAIL\");"
+ "} else {"
+ "trace(\"OK\");"
+ "}");
if (res.contains("\"FAIL\"")) {
fail("OnTrue clause was not removed:" + res);
}
if (!res.contains("\"OK\"")) {
fail("OnFalse clause was removed:" + res);
}
}
// todo: honfika @Test
public void testRemoveKnownVariables() throws Exception {
String res = recompile("var a = true; var b = false;"
+ "if(a){"
+ "trace(\"OK1\");"
+ "}else{"
+ "trace(\"FAIL1\");"
+ "}"
+ "if(b){"
+ "trace(\"FAIL2\");"
+ "}else{"
+ "trace(\"OK2\");"
+ "}");
if (!res.contains("\"OK1\"")) {
fail("if true OnTrue removed");
}
if (!res.contains("\"OK2\"")) {
fail("if false OnFalse removed");
}
if (res.contains("\"FAIL1\"")) {
fail("if true OnFalse not removed");
}
if (res.contains("\"FAIL2\"")) {
fail("if false OnTrue not removed");
}
if (res.contains("var ")) {
fail("variables for obsucation not removed");
}
if (res.contains("if")) {
fail("if clauses not removed");
}
}
// todo: honfika @Test
public void testNotRemoveParams() throws Exception {
String res = recompile("function tst(p1,p2){"
+ "var a = 2;"
+ "var b = 3 * a;"
+ "if(b>1){"
+ "trace(\"OK1\");"
+ "}else{"
+ "trace(\"FAIL1\");"
+ "}"
+ "var c = p1*5;"
+ "if(c){"
+ "trace(\"OK2\");"
+ "}else{"
+ "trace(\"OK3\");"
+ "}"
+ "}");
if (!res.contains("\"OK1\"")) {
fail("basic if true onTrue removed");
}
if (res.contains("\"FAIL1\"")) {
fail("basic if true onFalse not removed");
}
if (!res.contains("\"OK2\"")) {
fail("if parameter onTrue removed");
}
if (!res.contains("\"OK3\"")) {
fail("if parameter onFalse removed");
}
}
@Test
public void testEvailExpressionAfterWhile() throws Exception {
String res = recompile("var a = 5;"
+ "while(true){"
+ "if(a==73){"
+ "a = 15;"
+ "}"
+ "if(a==1){"
+ "trace(\"FAIL1\");"
+ "}"
+ "if(a==5){"
+ "a=50;"
+ "}"
+ "if(a == 201){"
+ "break;"
+ "}"
+ "a++;"
+ "if(a == 53){"
+ "a = a + 20;"
+ "}"
+ "if(a>500){"
+ "trace(\"FAIL2\");"
+ "}"
+ "if(a==16){"
+ "a = 200;"
+ "}"
+ "}"
+ ""
+ "if(a == 201){"
+ "trace(\"OK\");"
+ "}else{"
+ "trace(\"FAIL3\");"
+ "}");
if (res.contains("\"FAIL1\"")) {
fail("unreachable if onTrue not removed");
}
if (res.contains("\"FAIL2\"")) {
fail("unreachable if onTrue 2 not removed");
}
if (res.contains("\"FAIL3\"")) {
fail("unreachable if onTrue 3 not removed");
}
if (!res.contains("\"OK\"")) {
fail("reachable of onTrue removed");
}
}
@Test
public void testRemoveJumpsToTheNextAction() {
String actionsString = "ConstantPool \"a\" \"b\" \"c\"\n"
+ "Push false register1\n"
+ "StoreRegister 2\n"
+ "Pop\n"
+ "Push register2\n"
+ "StoreRegister 0\n"
+ "Push \"a\"\n"
+ "StrictEquals\n"
+ "If loc005a\n"
+ "Push register0 \"b\"\n"
+ "StrictEquals\n"
+ "If loc0068\n"
+ "Jump loc0048;\n"
+ "loc0048:Push register0 \"c\"\n"
+ "StrictEquals\n"
+ "If loc0076\n"
+ "Jump loc0084\n"
+ "loc005a:Push 1\n"
+ "Trace\n"
+ "Jump loc0084\n"
+ "loc0068:Push 2\n"
+ "Trace\n"
+ "Jump loc0084\n"
+ "loc0076:Push 3\n"
+ "Trace\n"
+ "Jump loc0084\n"
+ "loc0084:";
try {
List<Action> actions = ASMParser.parse(0, true, actionsString, swf.version, false);
DoActionTag doa = getFirstActionTag();
doa.setActionBytes(Action.actionsToBytes(actions, true, swf.version));
HighlightedTextWriter writer = new HighlightedTextWriter(new CodeFormatting(), false);
Action.actionsToSource(doa, doa.getActions(), "", writer);
String actualResult = writer.toString();
assertTrue(actualResult.contains("case \"c\":"));
} catch (IOException | ActionParseException | InterruptedException ex) {
fail();
}
}
@Test
public void testRemoveGetTime() {
String actionsString = "ConstantPool \"a\"\n"
+ "GetTime\n"
+ "If loc1\n"
+ "Push \"FAIL\"\n"
+ "Trace\n"
+ "loc1:Push \"OK\"\n"
+ "Trace\n"
+ "loc2:";
try {
List<Action> actions = ASMParser.parse(0, true, actionsString, swf.version, false);
DoActionTag doa = getFirstActionTag();
doa.setActionBytes(Action.actionsToBytes(actions, true, swf.version));
HighlightedTextWriter writer = new HighlightedTextWriter(new CodeFormatting(), false);
Action.actionsToSource(doa, doa.getActions(), "", writer);
String actualResult = writer.toString();
assertTrue(!actualResult.contains("FAIL"));
assertTrue(actualResult.contains("OK"));
} catch (IOException | ActionParseException | InterruptedException ex) {
fail();
}
}
@Test
public void testRemoveGetTimeAndIncrement() {
String actionsString = "ConstantPool \"a\"\n"
+ "GetTime\n"
+ "Increment\n"
+ "If loc1\n"
+ "Push \"FAIL\"\n"
+ "Trace\n"
+ "loc1:Push \"OK\"\n"
+ "Trace\n"
+ "loc2:";
try {
List<Action> actions = ASMParser.parse(0, true, actionsString, swf.version, false);
DoActionTag doa = getFirstActionTag();
doa.setActionBytes(Action.actionsToBytes(actions, true, swf.version));
HighlightedTextWriter writer = new HighlightedTextWriter(new CodeFormatting(), false);
Action.actionsToSource(doa, doa.getActions(), "", writer);
String actualResult = writer.toString();
assertTrue(!actualResult.contains("FAIL"));
assertTrue(actualResult.contains("OK"));
} catch (IOException | ActionParseException | InterruptedException ex) {
fail();
}
}
}