Files
jpexs-decompiler/src/com/jpexs/decompiler/flash/action/swf7/ActionTry.java
2014-08-23 23:04:02 +02:00

283 lines
10 KiB
Java

/*
* Copyright (C) 2010-2014 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 <http://www.gnu.org/licenses/>.
*/
package com.jpexs.decompiler.flash.action.swf7;
import com.jpexs.decompiler.flash.SWFInputStream;
import com.jpexs.decompiler.flash.SWFOutputStream;
import com.jpexs.decompiler.flash.action.Action;
import com.jpexs.decompiler.flash.action.ActionList;
import com.jpexs.decompiler.flash.action.model.ActionItem;
import com.jpexs.decompiler.flash.action.model.DirectValueActionItem;
import com.jpexs.decompiler.flash.action.model.clauses.TryActionItem;
import com.jpexs.decompiler.flash.action.parser.ParseException;
import com.jpexs.decompiler.flash.action.parser.pcode.ASMParsedSymbol;
import com.jpexs.decompiler.flash.action.parser.pcode.FlasmLexer;
import com.jpexs.decompiler.flash.action.swf4.RegisterNumber;
import com.jpexs.decompiler.flash.exporters.modes.ScriptExportMode;
import com.jpexs.decompiler.graph.GraphSourceItemContainer;
import com.jpexs.decompiler.graph.GraphTargetItem;
import com.jpexs.decompiler.graph.TranslateStack;
import com.jpexs.helpers.Helper;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
public class ActionTry extends Action implements GraphSourceItemContainer {
public int reserved;
public boolean catchInRegisterFlag;
public boolean finallyBlockFlag;
public boolean catchBlockFlag;
public String catchName;
public int catchRegister;
long trySize;
long catchSize;
long finallySize;
private final int version;
public ActionTry(boolean catchInRegisterFlag, boolean finallyBlockFlag, boolean catchBlockFlag, String catchName, int catchRegister, long trySize, long catchSize, long finallySize, int version) {
super(0x8F, 0);
this.catchInRegisterFlag = catchInRegisterFlag;
this.finallyBlockFlag = finallyBlockFlag;
this.catchBlockFlag = catchBlockFlag;
this.catchName = catchName;
this.catchRegister = catchRegister;
this.trySize = trySize;
this.catchSize = catchSize;
this.finallySize = finallySize;
this.version = version;
}
public ActionTry(int actionLength, SWFInputStream sis, int version) throws IOException {
super(0x8F, actionLength);
long startPos = sis.getPos();
this.version = version;
reserved = (int) sis.readUB(5, "reserved");
catchInRegisterFlag = sis.readUB(1, "catchInRegisterFlag") == 1;
finallyBlockFlag = sis.readUB(1, "finallyBlockFlag") == 1;
catchBlockFlag = sis.readUB(1, "catchBlockFlag") == 1;
trySize = sis.readUI16("trySize");
catchSize = sis.readUI16("catchSize");
finallySize = sis.readUI16("finallySize");
if (catchInRegisterFlag) {
catchRegister = sis.readUI8("catchRegister");
} else {
catchName = sis.readString("catchName");
}
}
@Override
public byte[] getBytes(int version) {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
SWFOutputStream sos = new SWFOutputStream(baos, version);
try {
sos.writeUB(5, reserved);
sos.writeUB(1, catchInRegisterFlag ? 1 : 0);
sos.writeUB(1, finallyBlockFlag ? 1 : 0);
sos.writeUB(1, catchBlockFlag ? 1 : 0);
sos.writeUI16((int) trySize);
sos.writeUI16((int) catchSize);
sos.writeUI16((int) finallySize);
if (catchInRegisterFlag) {
sos.writeUI8(catchRegister);
} else {
sos.writeString(catchName == null ? "" : catchName);
}
sos.close();
} catch (IOException e) {
}
return surroundWithAction(baos.toByteArray(), version);
}
public ActionTry(FlasmLexer lexer, int version) throws IOException, ParseException {
super(0x8F, 0);
this.version = version;
ASMParsedSymbol symb = lexer.yylex();
if (symb.type == ASMParsedSymbol.TYPE_STRING) {
catchInRegisterFlag = false;
catchName = (String) symb.value;
} else if (symb.type == ASMParsedSymbol.TYPE_REGISTER) {
catchRegister = ((RegisterNumber) symb.value).number;
catchInRegisterFlag = true;
} else if (symb.type == ASMParsedSymbol.TYPE_BLOCK_START) {
return;
} else {
throw new ParseException("Unknown symbol after Try", lexer.yyline());
}
lexBlockOpen(lexer);
}
@Override
public String getASMSourceBetween(int pos) {
String ret = "";
if (pos == 0) {
if (catchBlockFlag) {
ret += "Catch";
ret += " {\r\n";
return ret;
}
if (finallyBlockFlag) {
ret += "Finally {\r\n";
return ret;
}
}
if (pos == 1) {
if (catchBlockFlag && finallyBlockFlag) {
ret += "Finally {\r\n";
return ret;
}
}
return ret;
}
@Override
public String getASMSource(ActionList container, List<Long> knownAddreses, ScriptExportMode exportMode) {
StringBuilder ret = new StringBuilder();
ret.append("Try ");
if (catchBlockFlag) {
if (catchInRegisterFlag) {
ret.append("register").append(catchRegister);
} else {
ret.append("\"").append(Helper.escapeString(catchName)).append("\"");
}
ret.append(" ");
}
ret.append("{");
return ret.toString();
}
@Override
public long getHeaderSize() {
return getBytes(version).length;
}
public long getTrySize() {
return trySize;
}
@Override
public List<Long> getContainerSizes() {
List<Long> ret = new ArrayList<>();
ret.add(trySize);
ret.add(catchSize);
ret.add(finallySize);
return ret;
}
@Override
public void setContainerSize(int index, long size) {
switch (index) {
case 0:
trySize = size;
break;
case 1:
catchSize = size;
break;
case 2:
finallySize = size;
break;
default:
throw new IllegalArgumentException("Valid indexes are 0, 1, and 2.");
}
}
@Override
public boolean parseDivision(long size, FlasmLexer lexer) {
try {
ASMParsedSymbol symb = lexer.yylex();
//catchBlockFlag = false;
if (symb.type == ASMParsedSymbol.TYPE_INSTRUCTION_NAME) {
if (((String) symb.value).toLowerCase().equals("catch")) {
trySize = size - getHeaderSize();
catchBlockFlag = true;
lexBlockOpen(lexer);
return true;
}
if (symb.type == ASMParsedSymbol.TYPE_INSTRUCTION_NAME) {
if (((String) symb.value).toLowerCase().equals("finally")) {
if (catchBlockFlag) {
catchSize = size - getHeaderSize() - trySize;
} else {
trySize = size - getHeaderSize();
}
finallyBlockFlag = true;
lexBlockOpen(lexer);
return true;
} else {
//finallyBlockFlag = false;
lexer.yypushback(lexer.yylength());
}
} else {
//finallyBlockFlag = false;
lexer.yypushback(lexer.yylength());
}
} else {
lexer.yypushback(lexer.yylength());
}
} catch (IOException | ParseException ex) {
}
if (finallyBlockFlag) {
finallySize = size - getHeaderSize() - trySize - catchSize;
} else if (catchBlockFlag) {
catchSize = size - getHeaderSize() - trySize;
}
lexer.yybegin(0);
return false;
}
@Override
public void translateContainer(List<List<GraphTargetItem>> contents, TranslateStack stack, List<GraphTargetItem> output, HashMap<Integer, String> regNames, HashMap<String, GraphTargetItem> variables, HashMap<String, GraphTargetItem> functions) {
List<GraphTargetItem> tryCommands = contents.get(0);
ActionItem catchName;
if (catchInRegisterFlag) {
catchName = new DirectValueActionItem(this, -1, new RegisterNumber(this.catchRegister), new ArrayList<String>());
} else {
catchName = new DirectValueActionItem(this, -1, this.catchName, new ArrayList<String>());
}
List<GraphTargetItem> catchExceptions = new ArrayList<>();
if (catchBlockFlag) {
catchExceptions.add(catchName);
}
List<List<GraphTargetItem>> catchCommands = new ArrayList<>();
if (catchBlockFlag) {
catchCommands.add(contents.get(1));
}
List<GraphTargetItem> finallyCommands = contents.get(2);
output.add(new TryActionItem(tryCommands, catchExceptions, catchCommands, finallyCommands));
}
@Override
public String toString() {
return "Try";
}
@Override
public HashMap<Integer, String> getRegNames() {
return new HashMap<>();
}
@Override
public String getName() {
return null;
}
}