/*
* 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.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.LocalDataArea;
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.ActionParseException;
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.flash.types.annotations.Reserved;
import com.jpexs.decompiler.flash.types.annotations.SWFVersion;
import com.jpexs.decompiler.graph.GraphSourceItem;
import com.jpexs.decompiler.graph.GraphSourceItemContainer;
import com.jpexs.decompiler.graph.GraphTargetItem;
import com.jpexs.decompiler.graph.TranslateStack;
import com.jpexs.helpers.Helper;
import com.jpexs.helpers.utf8.Utf8Helper;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Set;
/**
*
* @author JPEXS
*/
@SWFVersion(from = 7)
public class ActionTry extends Action implements GraphSourceItemContainer {
@Reserved
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;
@Override
public boolean execute(LocalDataArea lda) {
//FIXME!!!
return false;
}
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
protected void getContentBytes(SWFOutputStream sos) throws IOException {
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);
}
}
/**
* Gets the length of action converted to bytes
*
* @return Length
*/
@Override
protected int getContentBytesLength() {
int res = 8;
if (!catchInRegisterFlag) {
res += Utf8Helper.getBytesLength(catchName == null ? "" : catchName);
}
return res;
}
public ActionTry(FlasmLexer lexer, int version) throws IOException, ActionParseException {
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 ActionParseException("Unknown symbol after Try", lexer.yyline());
}
lexBlockOpen(lexer);
}
@Override
public String getASMSourceBetween(int pos) {
if (pos == 0) {
if (catchBlockFlag) {
return "Catch {\r\n";
}
if (finallyBlockFlag) {
return "Finally {\r\n";
}
}
if (pos == 1) {
if (catchBlockFlag && finallyBlockFlag) {
return "Finally {\r\n";
}
}
return "";
}
@Override
public String getASMSource(ActionList container, Set<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.escapeActionScriptString(catchName)).append("\"");
}
ret.append(" ");
}
ret.append("{");
return ret.toString();
}
@Override
public long getHeaderSize() {
return getBytesLength();
}
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 | ActionParseException 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, GraphSourceItem lineStartItem, 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, lineStartItem, -1, new RegisterNumber(this.catchRegister), new ArrayList<>());
} else {
catchName = new DirectValueActionItem(this, lineStartItem, -1, this.catchName, new ArrayList<>());
}
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;
}
}