mirror of
https://git.huckle.dev/Huckles-Minecraft-Archive/jpexs-decompiler.git
synced 2026-09-24 21:01:20 +00:00
910 lines
39 KiB
Java
910 lines
39 KiB
Java
/*
|
|
* Copyright (C) 2010-2011 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 2
|
|
* 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, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
*/
|
|
package com.jpexs.asdec.abc;
|
|
|
|
import com.jpexs.asdec.Main;
|
|
import com.jpexs.asdec.abc.avm2.AVM2Code;
|
|
import com.jpexs.asdec.abc.avm2.ConstantPool;
|
|
import com.jpexs.asdec.abc.avm2.UnknownInstructionCode;
|
|
import com.jpexs.asdec.abc.avm2.instructions.AVM2Instruction;
|
|
import com.jpexs.asdec.abc.avm2.treemodel.InitPropertyTreeItem;
|
|
import com.jpexs.asdec.abc.avm2.treemodel.SetPropertyTreeItem;
|
|
import com.jpexs.asdec.abc.avm2.treemodel.TreeItem;
|
|
import com.jpexs.asdec.abc.types.*;
|
|
import com.jpexs.asdec.abc.types.traits.Trait;
|
|
import com.jpexs.asdec.abc.types.traits.TraitMethodGetterSetter;
|
|
import com.jpexs.asdec.abc.types.traits.TraitSlotConst;
|
|
import com.jpexs.asdec.abc.types.traits.Traits;
|
|
import com.jpexs.asdec.abc.usages.ClassNameMultinameUsage;
|
|
import com.jpexs.asdec.abc.usages.ConstVarNameMultinameUsage;
|
|
import com.jpexs.asdec.abc.usages.ConstVarTypeMultinameUsage;
|
|
import com.jpexs.asdec.abc.usages.ExtendsMultinameUsage;
|
|
import com.jpexs.asdec.abc.usages.ImplementsMultinameUsage;
|
|
import com.jpexs.asdec.abc.usages.MethodBodyMultinameUsage;
|
|
import com.jpexs.asdec.abc.usages.MethodNameMultinameUsage;
|
|
import com.jpexs.asdec.abc.usages.MethodParamsMultinameUsage;
|
|
import com.jpexs.asdec.abc.usages.MethodReturnTypeMultinameUsage;
|
|
import com.jpexs.asdec.abc.usages.InsideClassMultinameUsage;
|
|
import com.jpexs.asdec.abc.usages.MultinameUsage;
|
|
import com.jpexs.asdec.abc.usages.TypeNameMultinameUsage;
|
|
import com.jpexs.asdec.helpers.Highlighting;
|
|
|
|
import java.io.*;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import javax.swing.JOptionPane;
|
|
|
|
public class ABC {
|
|
|
|
public int major_version = 0;
|
|
public int minor_version = 0;
|
|
public ConstantPool constants;
|
|
public MethodInfo method_info[];
|
|
public MetadataInfo metadata_info[];
|
|
public InstanceInfo instance_info[];
|
|
public ClassInfo class_info[];
|
|
public ScriptInfo script_info[];
|
|
public MethodBody bodies[];
|
|
public long stringOffsets[];
|
|
public static String IDENT_STRING = " ";
|
|
public static final int MINORwithDECIMAL = 17;
|
|
private int fixNamesStrategy = 0;
|
|
private final int STRATEGY_FIX_NAMES = 1;
|
|
private final int STRATEGY_IGNORE = 2;
|
|
public static final boolean AUTOINIT_STATIC_VARIABLES = false;
|
|
|
|
private void fixNameWithStrategy(int index) {
|
|
if (fixNamesStrategy == 0) {
|
|
if (!isValidName(index, false)) {
|
|
int val = JOptionPane.showOptionDialog(null, "Decompiler found unusual name '" + constants.constant_string[index] + "' for a variable/class which can cause problems when decompiling.\r\nDo you want the decompiler to fix it?", "Decompilation", 0, JOptionPane.INFORMATION_MESSAGE, null, new String[]{"Yes", "Yes to all", "No", "No to all"}, "Yes to all");
|
|
if (val == JOptionPane.CLOSED_OPTION) {
|
|
val = 2; //NO
|
|
}
|
|
if ((val == 0) || (val == 1)) //YES,YES TO ALL
|
|
{
|
|
isValidName(index, true);
|
|
}
|
|
if (val == 1) //YES TO ALL
|
|
{
|
|
fixNamesStrategy = STRATEGY_FIX_NAMES;
|
|
}
|
|
if (val == 3) { //NO TO ALL
|
|
fixNamesStrategy = STRATEGY_IGNORE;
|
|
}
|
|
}
|
|
} else if (fixNamesStrategy == STRATEGY_FIX_NAMES) {
|
|
isValidName(index, true);
|
|
} else if (fixNamesStrategy == STRATEGY_IGNORE) {
|
|
}
|
|
}
|
|
|
|
public ABC(InputStream is) throws IOException {
|
|
ABCInputStream ais = new ABCInputStream(is);
|
|
minor_version = ais.readU16();
|
|
major_version = ais.readU16();
|
|
constants = new ConstantPool();
|
|
//constant integers
|
|
int constant_int_pool_count = ais.readU30();
|
|
constants.constant_int = new long[constant_int_pool_count];
|
|
for (int i = 1; i < constant_int_pool_count; i++) { //index 0 not used. Values 1..n-1
|
|
constants.constant_int[i] = ais.readS32();
|
|
}
|
|
|
|
//constant unsigned integers
|
|
int constant_uint_pool_count = ais.readU30();
|
|
constants.constant_uint = new long[constant_uint_pool_count];
|
|
for (int i = 1; i < constant_uint_pool_count; i++) { //index 0 not used. Values 1..n-1
|
|
constants.constant_uint[i] = ais.readU32();
|
|
}
|
|
|
|
//constant double
|
|
int constant_double_pool_count = ais.readU30();
|
|
constants.constant_double = new double[constant_double_pool_count];
|
|
for (int i = 1; i < constant_double_pool_count; i++) { //index 0 not used. Values 1..n-1
|
|
constants.constant_double[i] = ais.readDouble();
|
|
}
|
|
|
|
|
|
//constant decimal
|
|
if (minor_version >= MINORwithDECIMAL) {
|
|
int constant_decimal_pool_count = ais.readU30();
|
|
constants.constant_decimal = new Decimal[constant_decimal_pool_count];
|
|
for (int i = 1; i < constant_decimal_pool_count; i++) { //index 0 not used. Values 1..n-1
|
|
constants.constant_decimal[i] = ais.readDecimal();
|
|
}
|
|
} else {
|
|
constants.constant_decimal = new Decimal[0];
|
|
}
|
|
|
|
//constant string
|
|
int constant_string_pool_count = ais.readU30();
|
|
constants.constant_string = new String[constant_string_pool_count];
|
|
stringOffsets = new long[constant_string_pool_count];
|
|
constants.constant_string[0] = "";
|
|
for (int i = 1; i < constant_string_pool_count; i++) { //index 0 not used. Values 1..n-1
|
|
long pos = ais.getPosition();
|
|
constants.constant_string[i] = ais.readString();
|
|
stringOffsets[i] = pos;
|
|
}
|
|
|
|
//constant namespace
|
|
int constant_namespace_pool_count = ais.readU30();
|
|
constants.constant_namespace = new Namespace[constant_namespace_pool_count];
|
|
for (int i = 1; i < constant_namespace_pool_count; i++) { //index 0 not used. Values 1..n-1
|
|
constants.constant_namespace[i] = ais.readNamespace();
|
|
fixNameWithStrategy(constants.constant_namespace[i].name_index);
|
|
}
|
|
|
|
//constant namespace set
|
|
int constant_namespace_set_pool_count = ais.readU30();
|
|
constants.constant_namespace_set = new NamespaceSet[constant_namespace_set_pool_count];
|
|
for (int i = 1; i < constant_namespace_set_pool_count; i++) { //index 0 not used. Values 1..n-1
|
|
constants.constant_namespace_set[i] = new NamespaceSet();
|
|
int namespace_count = ais.readU30();
|
|
constants.constant_namespace_set[i].namespaces = new int[namespace_count];
|
|
for (int j = 0; j < namespace_count; j++) {
|
|
constants.constant_namespace_set[i].namespaces[j] = ais.readU30();
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
//constant multiname
|
|
int constant_multiname_pool_count = ais.readU30();
|
|
constants.constant_multiname = new Multiname[constant_multiname_pool_count];
|
|
for (int i = 1; i < constant_multiname_pool_count; i++) { //index 0 not used. Values 1..n-1
|
|
constants.constant_multiname[i] = ais.readMultiname();
|
|
fixNameWithStrategy(constants.constant_multiname[i].name_index);
|
|
}
|
|
|
|
|
|
//method info
|
|
int methods_count = ais.readU30();
|
|
method_info = new MethodInfo[methods_count];
|
|
for (int i = 0; i < methods_count; i++) {
|
|
method_info[i] = ais.readMethodInfo();
|
|
}
|
|
|
|
//metadata info
|
|
int metadata_count = ais.readU30();
|
|
metadata_info = new MetadataInfo[metadata_count];
|
|
for (int i = 0; i < metadata_count; i++) {
|
|
int name_index = ais.readU30();
|
|
int values_count = ais.readU30();
|
|
int keys[] = new int[values_count];
|
|
for (int v = 0; v < values_count; v++) {
|
|
keys[v] = ais.readU30();
|
|
}
|
|
int values[] = new int[values_count];
|
|
for (int v = 0; v < values_count; v++) {
|
|
values[v] = ais.readU30();
|
|
}
|
|
metadata_info[i] = new MetadataInfo(name_index, keys, values);
|
|
}
|
|
|
|
int class_count = ais.readU30();
|
|
instance_info = new InstanceInfo[class_count];
|
|
for (int i = 0; i < class_count; i++) {
|
|
instance_info[i] = ais.readInstanceInfo();
|
|
}
|
|
class_info = new ClassInfo[class_count];
|
|
for (int i = 0; i < class_count; i++) {
|
|
class_info[i] = new ClassInfo();
|
|
class_info[i].cinit_index = ais.readU30();
|
|
class_info[i].static_traits = ais.readTraits();
|
|
}
|
|
int script_count = ais.readU30();
|
|
script_info = new ScriptInfo[script_count];
|
|
for (int i = 0; i < script_count; i++) {
|
|
script_info[i] = new ScriptInfo();
|
|
script_info[i].init_index = ais.readU30();
|
|
script_info[i].traits = ais.readTraits();
|
|
}
|
|
|
|
int bodies_count = ais.readU30();
|
|
bodies = new MethodBody[bodies_count];
|
|
for (int i = 0; i < bodies_count; i++) {
|
|
bodies[i] = new MethodBody();
|
|
bodies[i].method_info = ais.readU30();
|
|
bodies[i].max_stack = ais.readU30();
|
|
bodies[i].max_regs = ais.readU30();
|
|
bodies[i].init_scope_depth = ais.readU30();
|
|
bodies[i].max_scope_depth = ais.readU30();
|
|
int code_length = ais.readU30();
|
|
bodies[i].codeBytes = new byte[code_length];
|
|
for (int j = 0; j < code_length; j++) {
|
|
bodies[i].codeBytes[j] = (byte) ais.read();
|
|
}
|
|
try {
|
|
bodies[i].code = new AVM2Code(new ByteArrayInputStream(bodies[i].codeBytes));
|
|
} catch (UnknownInstructionCode re) {
|
|
bodies[i].code = new AVM2Code();
|
|
System.err.println(re.toString());
|
|
}
|
|
int ex_count = ais.readU30();
|
|
bodies[i].exceptions = new ABCException[ex_count];
|
|
for (int j = 0; j < ex_count; j++) {
|
|
bodies[i].exceptions[j] = new ABCException();
|
|
bodies[i].exceptions[j].start = ais.readU30();
|
|
bodies[i].exceptions[j].end = ais.readU30();
|
|
bodies[i].exceptions[j].target = ais.readU30();
|
|
bodies[i].exceptions[j].type_index = ais.readU30();
|
|
bodies[i].exceptions[j].name_index = ais.readU30();
|
|
}
|
|
bodies[i].traits = ais.readTraits();
|
|
}
|
|
}
|
|
|
|
public void saveToStream(OutputStream os) throws IOException {
|
|
ABCOutputStream aos = new ABCOutputStream(os);
|
|
aos.writeU16(minor_version);
|
|
aos.writeU16(major_version);
|
|
|
|
aos.writeU30(constants.constant_int.length);
|
|
for (int i = 1; i < constants.constant_int.length; i++) {
|
|
aos.writeS32(constants.constant_int[i]);
|
|
}
|
|
aos.writeU30(constants.constant_uint.length);
|
|
for (int i = 1; i < constants.constant_uint.length; i++) {
|
|
aos.writeU32(constants.constant_uint[i]);
|
|
}
|
|
|
|
aos.writeU30(constants.constant_double.length);
|
|
for (int i = 1; i < constants.constant_double.length; i++) {
|
|
aos.writeDouble(constants.constant_double[i]);
|
|
}
|
|
|
|
if (minor_version >= MINORwithDECIMAL) {
|
|
aos.writeU30(constants.constant_decimal.length);
|
|
for (int i = 1; i < constants.constant_decimal.length; i++) {
|
|
aos.writeDecimal(constants.constant_decimal[i]);
|
|
}
|
|
}
|
|
|
|
aos.writeU30(constants.constant_string.length);
|
|
for (int i = 1; i < constants.constant_string.length; i++) {
|
|
aos.writeString(constants.constant_string[i]);
|
|
}
|
|
|
|
aos.writeU30(constants.constant_namespace.length);
|
|
for (int i = 1; i < constants.constant_namespace.length; i++) {
|
|
aos.writeNamespace(constants.constant_namespace[i]);
|
|
}
|
|
|
|
aos.writeU30(constants.constant_namespace_set.length);
|
|
for (int i = 1; i < constants.constant_namespace_set.length; i++) {
|
|
aos.writeU30(constants.constant_namespace_set[i].namespaces.length);
|
|
for (int j = 0; j < constants.constant_namespace_set[i].namespaces.length; j++) {
|
|
aos.writeU30(constants.constant_namespace_set[i].namespaces[j]);
|
|
}
|
|
}
|
|
|
|
aos.writeU30(constants.constant_multiname.length);
|
|
//System.out.println("Writing "+constants.constant_multiname.length+" multinames");
|
|
for (int i = 1; i < constants.constant_multiname.length; i++) {
|
|
aos.writeMultiname(constants.constant_multiname[i]);
|
|
}
|
|
|
|
aos.writeU30(method_info.length);
|
|
for (int i = 0; i < method_info.length; i++) {
|
|
aos.writeMethodInfo(method_info[i]);
|
|
}
|
|
|
|
aos.writeU30(metadata_info.length);
|
|
for (int i = 0; i < metadata_info.length; i++) {
|
|
aos.writeU30(metadata_info[i].name_index);
|
|
aos.writeU30(metadata_info[i].values.length);
|
|
for (int j = 0; j < metadata_info[i].values.length; j++) {
|
|
aos.writeU30(metadata_info[i].keys[j]);
|
|
}
|
|
for (int j = 0; j < metadata_info[i].values.length; j++) {
|
|
aos.writeU30(metadata_info[i].values[j]);
|
|
}
|
|
}
|
|
|
|
aos.writeU30(class_info.length);
|
|
for (int i = 0; i < instance_info.length; i++) {
|
|
aos.writeInstanceInfo(instance_info[i]);
|
|
}
|
|
for (int i = 0; i < class_info.length; i++) {
|
|
aos.writeU30(class_info[i].cinit_index);
|
|
aos.writeTraits(class_info[i].static_traits);
|
|
}
|
|
aos.writeU30(script_info.length);
|
|
for (int i = 0; i < script_info.length; i++) {
|
|
aos.writeU30(script_info[i].init_index);
|
|
aos.writeTraits(script_info[i].traits);
|
|
}
|
|
|
|
aos.writeU30(bodies.length);
|
|
for (int i = 0; i < bodies.length; i++) {
|
|
aos.writeU30(bodies[i].method_info);
|
|
aos.writeU30(bodies[i].max_stack);
|
|
aos.writeU30(bodies[i].max_regs);
|
|
aos.writeU30(bodies[i].init_scope_depth);
|
|
aos.writeU30(bodies[i].max_scope_depth);
|
|
byte codeBytes[] = bodies[i].code.getBytes();
|
|
aos.writeU30(codeBytes.length);
|
|
try {
|
|
aos.write(codeBytes);
|
|
} catch (NotSameException ex) {
|
|
System.out.println(bodies[i].code.toString(constants));
|
|
System.exit(0);
|
|
return;
|
|
}
|
|
aos.writeU30(bodies[i].exceptions.length);
|
|
for (int j = 0; j < bodies[i].exceptions.length; j++) {
|
|
aos.writeU30(bodies[i].exceptions[j].start);
|
|
aos.writeU30(bodies[i].exceptions[j].end);
|
|
aos.writeU30(bodies[i].exceptions[j].target);
|
|
aos.writeU30(bodies[i].exceptions[j].type_index);
|
|
aos.writeU30(bodies[i].exceptions[j].name_index);
|
|
}
|
|
aos.writeTraits(bodies[i].traits);
|
|
}
|
|
}
|
|
|
|
private void parseImportFromMultiname(List imports, Multiname m) {
|
|
if (m != null) {
|
|
Namespace ns = m.getNamespace(constants);
|
|
String name = m.getName(constants);
|
|
if (ns != null) {
|
|
String newimport = ns.getName(constants);
|
|
if (!newimport.equals("")) {
|
|
newimport += "." + name;
|
|
if (newimport.contains(":")) {
|
|
return;
|
|
}
|
|
if (!imports.contains(newimport)) {
|
|
imports.add(newimport);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private List getImports(int instanceIndex) {
|
|
List<String> imports = new ArrayList<String>();
|
|
|
|
//constructor
|
|
|
|
//parseImportFromMultiname(imports, constants.constant_multiname[instance_info[instanceIndex].name_index]);
|
|
|
|
if (instance_info[instanceIndex].super_index > 0) {
|
|
parseImportFromMultiname(imports, constants.constant_multiname[instance_info[instanceIndex].super_index]);
|
|
}
|
|
for (int i : instance_info[instanceIndex].interfaces) {
|
|
parseImportFromMultiname(imports, constants.constant_multiname[i]);
|
|
}
|
|
//static
|
|
for (Trait t : class_info[instanceIndex].static_traits.traits) {
|
|
//parseImportFromMultiname(imports, t.getMultiName(constants));
|
|
if (t instanceof TraitMethodGetterSetter) {
|
|
TraitMethodGetterSetter tm = (TraitMethodGetterSetter) t;
|
|
if (tm.method_info != 0) {
|
|
MethodBody body = findBody(tm.method_info);
|
|
if (body != null) {
|
|
for (AVM2Instruction ins : body.code.code) {
|
|
for (int k = 0; k < ins.definition.operands.length; k++) {
|
|
if (ins.definition.operands[k] == AVM2Code.DAT_MULTINAME_INDEX) {
|
|
int multinameIndex = ins.operands[k];
|
|
parseImportFromMultiname(imports, constants.constant_multiname[multinameIndex]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
for (int p = 0; p < method_info[tm.method_info].param_types.length; p++) {
|
|
if (method_info[tm.method_info].param_types[p] != 0) {
|
|
parseImportFromMultiname(imports, constants.constant_multiname[method_info[tm.method_info].param_types[p]]);
|
|
}
|
|
if (method_info[tm.method_info].ret_type != 0) {
|
|
parseImportFromMultiname(imports, constants.constant_multiname[method_info[tm.method_info].ret_type]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
//instance
|
|
for (Trait t : instance_info[instanceIndex].instance_traits.traits) {
|
|
//parseImportFromMultiname(imports, t.getMultiName(constants));
|
|
if (t instanceof TraitMethodGetterSetter) {
|
|
TraitMethodGetterSetter tm = (TraitMethodGetterSetter) t;
|
|
if (tm.method_info != 0) {
|
|
MethodBody body = findBody(tm.method_info);
|
|
if (body != null) {
|
|
for (AVM2Instruction ins : body.code.code) {
|
|
for (int k = 0; k < ins.definition.operands.length; k++) {
|
|
if (ins.definition.operands[k] == AVM2Code.DAT_MULTINAME_INDEX) {
|
|
int multinameIndex = ins.operands[k];
|
|
parseImportFromMultiname(imports, constants.constant_multiname[multinameIndex]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
for (int p = 0; p < method_info[tm.method_info].param_types.length; p++) {
|
|
if (method_info[tm.method_info].param_types[p] != 0) {
|
|
parseImportFromMultiname(imports, constants.constant_multiname[method_info[tm.method_info].param_types[p]]);
|
|
}
|
|
if (method_info[tm.method_info].ret_type != 0) {
|
|
parseImportFromMultiname(imports, constants.constant_multiname[method_info[tm.method_info].ret_type]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return imports;
|
|
}
|
|
|
|
public MethodBody findBody(int methodInfo) {
|
|
int pos = findBodyIndex(methodInfo);
|
|
if (pos == -1) {
|
|
return null;
|
|
} else {
|
|
return bodies[pos];
|
|
}
|
|
}
|
|
|
|
public int findBodyIndex(int methodInfo) {
|
|
if (methodInfo == -1) {
|
|
return -1;
|
|
}
|
|
for (int b = 0; b < bodies.length; b++) {
|
|
if (bodies[b].method_info == methodInfo) {
|
|
return b;
|
|
}
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
public MethodBody findBodyByClassAndName(String className, String methodName) {
|
|
for (int i = 0; i < instance_info.length; i++) {
|
|
if (className.equals(constants.constant_multiname[instance_info[i].name_index].getName(constants))) {
|
|
for (Trait t : instance_info[i].instance_traits.traits) {
|
|
if (t instanceof TraitMethodGetterSetter) {
|
|
TraitMethodGetterSetter t2 = (TraitMethodGetterSetter) t;
|
|
if (methodName.equals(t2.getMethodName(constants))) {
|
|
for (MethodBody body : bodies) {
|
|
if (body.method_info == t2.method_info) {
|
|
return body;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
//break;
|
|
}
|
|
}
|
|
for (int i = 0; i < class_info.length; i++) {
|
|
if (className.equals(constants.constant_multiname[instance_info[i].name_index].getName(constants))) {
|
|
for (Trait t : class_info[i].static_traits.traits) {
|
|
if (t instanceof TraitMethodGetterSetter) {
|
|
TraitMethodGetterSetter t2 = (TraitMethodGetterSetter) t;
|
|
if (methodName.equals(t2.getMethodName(constants))) {
|
|
for (MethodBody body : bodies) {
|
|
if (body.method_info == t2.method_info) {
|
|
return body;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
//break;
|
|
}
|
|
}
|
|
|
|
|
|
return null;
|
|
}
|
|
|
|
private String addTabs(String s, int tabs) {
|
|
String parts[] = s.split("\r\n");
|
|
if (!s.contains("\r\n")) {
|
|
parts = s.split("\n");
|
|
}
|
|
String ret = "";
|
|
for (int i = 0; i < parts.length; i++) {
|
|
for (int t = 0; t < tabs; t++) {
|
|
ret += IDENT_STRING;
|
|
}
|
|
ret += parts[i];
|
|
if (i < parts.length - 1) {
|
|
ret += "\r\n";
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public Trait findTraitByTraitId(int classIndex, int traitId) {
|
|
if (traitId < class_info[classIndex].static_traits.traits.length) {
|
|
return class_info[classIndex].static_traits.traits[traitId];
|
|
} else if (traitId < class_info[classIndex].static_traits.traits.length + instance_info[classIndex].instance_traits.traits.length) {
|
|
traitId -= class_info[classIndex].static_traits.traits.length;
|
|
return instance_info[classIndex].instance_traits.traits[traitId];
|
|
} else {
|
|
return null; //Can be class or instance initializer
|
|
}
|
|
}
|
|
|
|
public int findMethodIdByTraitId(int classIndex, int traitId) {
|
|
if (traitId < class_info[classIndex].static_traits.traits.length) {
|
|
if (class_info[classIndex].static_traits.traits[traitId] instanceof TraitMethodGetterSetter) {
|
|
return ((TraitMethodGetterSetter) class_info[classIndex].static_traits.traits[traitId]).method_info;
|
|
} else {
|
|
return -1;
|
|
}
|
|
} else if (traitId < class_info[classIndex].static_traits.traits.length + instance_info[classIndex].instance_traits.traits.length) {
|
|
traitId -= class_info[classIndex].static_traits.traits.length;
|
|
if (instance_info[classIndex].instance_traits.traits[traitId] instanceof TraitMethodGetterSetter) {
|
|
return ((TraitMethodGetterSetter) instance_info[classIndex].instance_traits.traits[traitId]).method_info;
|
|
} else {
|
|
return -1;
|
|
}
|
|
} else {
|
|
traitId -= class_info[classIndex].static_traits.traits.length + instance_info[classIndex].instance_traits.traits.length;
|
|
if (traitId == 0) {
|
|
return instance_info[classIndex].iinit_index;
|
|
} else if (traitId == 1) {
|
|
return class_info[classIndex].cinit_index;
|
|
} else {
|
|
return -1;
|
|
}
|
|
}
|
|
}
|
|
|
|
public String classToString(int i, boolean highlight, boolean pcode) {
|
|
String ret = "";
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
PrintStream out = new PrintStream(baos);
|
|
String packageName = instance_info[i].getName(constants).getNamespace(constants).getName(constants);
|
|
out.println("package " + packageName);
|
|
out.println("{");
|
|
|
|
//imports
|
|
List<String> imports = getImports(i);
|
|
for (String imp : imports) {
|
|
out.println(IDENT_STRING + "import " + imp + ";");
|
|
}
|
|
out.println();
|
|
|
|
//class header
|
|
String classHeader = instance_info[i].getClassHeaderStr(constants);
|
|
if (classHeader.startsWith("private ")) {
|
|
classHeader = "public " + classHeader.substring("private ".length());
|
|
}
|
|
out.println(IDENT_STRING + classHeader);
|
|
out.println(IDENT_STRING + "{");
|
|
|
|
String toPrint = "";
|
|
|
|
//if (class_info[i].cinit_index != 0) {
|
|
if (AUTOINIT_STATIC_VARIABLES) {
|
|
int bodyIndex = findBodyIndex(class_info[i].cinit_index);
|
|
List<TreeItem> initializer = bodies[bodyIndex].code.toTree(true, i, this, constants, method_info, bodies[bodyIndex]);
|
|
for (TreeItem ti : initializer) {
|
|
if (ti instanceof SetPropertyTreeItem) {
|
|
int multinameIndex = ((SetPropertyTreeItem) ti).propertyName.multinameIndex;
|
|
TreeItem value = ((SetPropertyTreeItem) ti).value;
|
|
for (Trait t : class_info[i].static_traits.traits) {
|
|
if (t.name_index == multinameIndex) {
|
|
if (t instanceof TraitSlotConst) {
|
|
((TraitSlotConst) t).assignedValue = value;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if (ti instanceof InitPropertyTreeItem) {
|
|
int multinameIndex = ((InitPropertyTreeItem) ti).propertyName.multinameIndex;
|
|
TreeItem value = ((InitPropertyTreeItem) ti).value;
|
|
for (Trait t : class_info[i].static_traits.traits) {
|
|
if (t.name_index == multinameIndex) {
|
|
if (t instanceof TraitSlotConst) {
|
|
((TraitSlotConst) t).assignedValue = value;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
String bodyStr = "";
|
|
int bodyIndex = findBodyIndex(class_info[i].cinit_index);
|
|
if (bodyIndex != -1) {
|
|
bodyStr = addTabs(bodies[bodyIndex].toString(pcode, true, i, this, constants, method_info, highlight), 3);
|
|
}
|
|
// if (!bodyStr.equals("")) {
|
|
toPrint = IDENT_STRING + IDENT_STRING + "{\r\n" + bodyStr + "\r\n" + IDENT_STRING + IDENT_STRING + "}";
|
|
if (highlight) {
|
|
toPrint = Highlighting.hilighTrait(toPrint, class_info[i].static_traits.traits.length + instance_info[i].instance_traits.traits.length + 1);
|
|
}
|
|
out.println(toPrint);
|
|
//}
|
|
//}
|
|
|
|
//constructor
|
|
//if (instance_info[i].iinit_index != 0) {
|
|
String modifier = "";
|
|
Multiname m = constants.constant_multiname[instance_info[i].name_index];
|
|
if (m != null) {
|
|
Namespace ns = m.getNamespace(constants);
|
|
if (ns != null) {
|
|
modifier = ns.getPrefix(constants) + " ";
|
|
if (modifier.equals(" ")) {
|
|
modifier = "";
|
|
}
|
|
}
|
|
}
|
|
String constructorParams = "";
|
|
|
|
bodyStr = "";
|
|
bodyIndex = findBodyIndex(instance_info[i].iinit_index);
|
|
if (bodyIndex != -1) {
|
|
bodyStr = addTabs(bodies[bodyIndex].toString(pcode, false, i, this, constants, method_info, highlight), 3);
|
|
constructorParams = method_info[instance_info[i].iinit_index].getParamStr(constants, bodies[bodyIndex], this);
|
|
} else {
|
|
constructorParams = method_info[instance_info[i].iinit_index].getParamStr(constants, null, this);
|
|
}
|
|
toPrint = IDENT_STRING + IDENT_STRING + modifier + "function " + constants.constant_multiname[instance_info[i].name_index].getName(constants) + "(" + constructorParams + ") {\r\n" + bodyStr + "\r\n" + IDENT_STRING + IDENT_STRING + "}";
|
|
if (highlight) {
|
|
toPrint = Highlighting.hilighTrait(toPrint, class_info[i].static_traits.traits.length + instance_info[i].instance_traits.traits.length);
|
|
}
|
|
out.println(toPrint);
|
|
//}
|
|
|
|
//static variables,constants & methods
|
|
for (int ti = 0; ti < class_info[i].static_traits.traits.length; ti++) {
|
|
Trait t = class_info[i].static_traits.traits[ti];
|
|
toPrint = "";
|
|
if (t instanceof TraitMethodGetterSetter) {
|
|
TraitMethodGetterSetter tm = (TraitMethodGetterSetter) t;
|
|
bodyStr = "";
|
|
bodyIndex = findBodyIndex(tm.method_info);
|
|
if (bodyIndex != -1) {
|
|
bodyStr = addTabs(bodies[bodyIndex].toString(pcode, true, i, this, constants, method_info, highlight), 3);
|
|
}
|
|
toPrint = IDENT_STRING + IDENT_STRING + tm.convert(constants, method_info, this, true) + " {\r\n" + bodyStr + "\r\n" + IDENT_STRING + IDENT_STRING + "}";
|
|
}
|
|
if (t instanceof TraitSlotConst) {
|
|
TraitSlotConst ts = (TraitSlotConst) t;
|
|
|
|
toPrint = IDENT_STRING + IDENT_STRING + ts.convert(constants, method_info, this, true) + ";";
|
|
}
|
|
if (highlight) {
|
|
toPrint = Highlighting.hilighTrait(toPrint, ti);
|
|
} else {
|
|
toPrint = Highlighting.stripHilights(toPrint);
|
|
}
|
|
out.println(toPrint);
|
|
}
|
|
for (int ti = 0; ti < instance_info[i].instance_traits.traits.length; ti++) {
|
|
Trait t = instance_info[i].instance_traits.traits[ti];
|
|
toPrint = "";
|
|
if (t instanceof TraitSlotConst) {
|
|
TraitSlotConst ts = (TraitSlotConst) t;
|
|
toPrint = IDENT_STRING + IDENT_STRING + ts.convert(constants, method_info, this, false) + ";";
|
|
}
|
|
|
|
if (t instanceof TraitMethodGetterSetter) {
|
|
TraitMethodGetterSetter tm = (TraitMethodGetterSetter) t;
|
|
bodyStr = "";
|
|
bodyIndex = findBodyIndex(tm.method_info);
|
|
if (bodyIndex != -1) {
|
|
bodyStr = addTabs(bodies[bodyIndex].toString(pcode, false, i, this, constants, method_info, highlight), 3);
|
|
}
|
|
toPrint = IDENT_STRING + IDENT_STRING + tm.convert(constants, method_info, this, false) + " {\r\n" + bodyStr + "\r\n" + IDENT_STRING + IDENT_STRING + "}";
|
|
}
|
|
if (highlight) {
|
|
toPrint = Highlighting.hilighTrait(toPrint, class_info[i].static_traits.traits.length + ti);
|
|
} else {
|
|
toPrint = Highlighting.stripHilights(toPrint);
|
|
}
|
|
out.println(toPrint);
|
|
}
|
|
|
|
|
|
out.println(IDENT_STRING + "}");//class
|
|
out.println("}");//package
|
|
out.flush();
|
|
|
|
return baos.toString();
|
|
}
|
|
|
|
public void export(String directory, boolean pcode) throws IOException {
|
|
for (int i = 0; i < instance_info.length; i++) {
|
|
String packageName = instance_info[i].getName(constants).getNamespace(constants).getName(constants);
|
|
String className = instance_info[i].getName(constants).getName(constants);
|
|
Main.startWork("Exporting " + (i + 1) + "/" + instance_info.length + " " + packageName + "." + className + "...");
|
|
File outDir = new File(directory + File.separatorChar + packageName.replace('.', File.separatorChar));
|
|
if (!outDir.exists()) {
|
|
outDir.mkdirs();
|
|
}
|
|
String fileName = outDir.toString() + File.separator + className + ".as";
|
|
FileOutputStream fos = new FileOutputStream(fileName);
|
|
fos.write(classToString(i, false, pcode).getBytes());
|
|
fos.close();
|
|
}
|
|
|
|
}
|
|
|
|
public void dump(OutputStream os) {
|
|
PrintStream output = new PrintStream(os);
|
|
constants.dump(output);
|
|
for (int i = 0; i < method_info.length; i++) {
|
|
output.println("MethodInfo[" + i + "]:" + method_info[i].toString(constants));
|
|
}
|
|
for (int i = 0; i < metadata_info.length; i++) {
|
|
output.println("MetadataInfo[" + i + "]:" + metadata_info[i].toString(constants));
|
|
}
|
|
for (int i = 0; i < instance_info.length; i++) {
|
|
output.println("InstanceInfo[" + i + "]:" + instance_info[i].toString(this));
|
|
}
|
|
for (int i = 0; i < class_info.length; i++) {
|
|
output.println("ClassInfo[" + i + "]:" + class_info[i].toString(this));
|
|
}
|
|
for (int i = 0; i < script_info.length; i++) {
|
|
output.println("ScriptInfo[" + i + "]:" + script_info[i].toString(this));
|
|
}
|
|
for (int i = 0; i < bodies.length; i++) {
|
|
output.println("MethodBody[" + i + "]:"); //+ bodies[i].toString(this, constants, method_info));
|
|
}
|
|
}
|
|
public static final String[] reservedWords = {
|
|
"as", "break", "case", "catch", "class", "const", "continue", "default", "delete", "do", "each", "else",
|
|
"extends", "false", "finally", "for", "function", "if", "implements", "import", "in", "instanceof",
|
|
"interface", "internal", "is", "native", "new", "null", "package", "private", "protected", "public",
|
|
"return", "super", "switch", "this", "throw", "true", "try", "typeof", "use", "var", /*"void",*/ "while",
|
|
"with", "dynamic", "default", "final", "in"};
|
|
public int unknownCount = 0;
|
|
|
|
public boolean isValidName(int index, boolean autoFix) {
|
|
if (index <= 0) {
|
|
return true;
|
|
}
|
|
String s = constants.constant_string[index];
|
|
boolean isValid = true;
|
|
boolean isReserved = false;
|
|
for (String rw : reservedWords) {
|
|
if (rw.equals(s.trim())) {
|
|
isValid = false;
|
|
isReserved = true;
|
|
break;
|
|
}
|
|
}
|
|
if (isValid) {
|
|
for (int i = 0; i < s.length(); i++) {
|
|
if (s.charAt(i) > 127) {
|
|
isValid = false;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!isValid) {
|
|
if (autoFix) {
|
|
if (isReserved) {
|
|
constants.constant_string[index] = "name_" + s.replace(" ", "_");
|
|
} else {
|
|
unknownCount++;
|
|
constants.constant_string[index] = "_name" + unknownCount;
|
|
}
|
|
}
|
|
}
|
|
return isValid;
|
|
}
|
|
|
|
private void checkMultinameUsedInMethod(int multinameIndex, int methodInfo, List<MultinameUsage> ret, int classIndex, int traitIndex, boolean isStatic, boolean isInitializer,Traits traits,int parentTraitIndex) {
|
|
for (int p = 0; p < method_info[methodInfo].param_types.length; p++) {
|
|
if (method_info[methodInfo].param_types[p] == multinameIndex) {
|
|
ret.add(new MethodParamsMultinameUsage(multinameIndex, classIndex, traitIndex, isStatic, isInitializer,traits,parentTraitIndex));
|
|
break;
|
|
}
|
|
}
|
|
if (method_info[methodInfo].ret_type == multinameIndex) {
|
|
ret.add(new MethodReturnTypeMultinameUsage(multinameIndex, classIndex, traitIndex, isStatic, isInitializer,traits,parentTraitIndex));
|
|
}
|
|
MethodBody body = findBody(methodInfo);
|
|
if (body != null) {
|
|
findMultinameUsageInTraits(body.traits,multinameIndex,isStatic,classIndex,ret,traitIndex);
|
|
for(ABCException e:body.exceptions){
|
|
if((e.name_index==multinameIndex)||(e.type_index==multinameIndex)){
|
|
ret.add(new MethodBodyMultinameUsage(multinameIndex, classIndex, traitIndex, isStatic, isInitializer,traits,parentTraitIndex));
|
|
return;
|
|
}
|
|
}
|
|
for (AVM2Instruction ins : body.code.code) {
|
|
for (int o = 0; o < ins.definition.operands.length; o++) {
|
|
if (ins.definition.operands[o] == AVM2Code.DAT_MULTINAME_INDEX) {
|
|
if (ins.operands[o] == multinameIndex) {
|
|
ret.add(new MethodBodyMultinameUsage(multinameIndex, classIndex, traitIndex, isStatic, isInitializer,traits,parentTraitIndex));
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private void findMultinameUsageInTraits(Traits traits, int multinameIndex, boolean isStatic, int classIndex, List<MultinameUsage> ret,int parentTraitIndex) {
|
|
for (int t = 0; t < traits.traits.length; t++) {
|
|
if (traits.traits[t] instanceof TraitSlotConst) {
|
|
TraitSlotConst tsc = (TraitSlotConst) traits.traits[t];
|
|
if (tsc.name_index == multinameIndex) {
|
|
ret.add(new ConstVarNameMultinameUsage(multinameIndex, classIndex, t, isStatic,traits,parentTraitIndex));
|
|
}
|
|
if (tsc.type_index == multinameIndex) {
|
|
ret.add(new ConstVarTypeMultinameUsage(multinameIndex, classIndex, t, isStatic,traits,parentTraitIndex));
|
|
}
|
|
}
|
|
if (traits.traits[t] instanceof TraitMethodGetterSetter) {
|
|
TraitMethodGetterSetter tmgs = (TraitMethodGetterSetter) traits.traits[t];
|
|
if (tmgs.name_index == multinameIndex) {
|
|
ret.add(new MethodNameMultinameUsage(multinameIndex, classIndex, t, isStatic, false,traits,parentTraitIndex));
|
|
}
|
|
checkMultinameUsedInMethod(multinameIndex, tmgs.method_info, ret, classIndex, t, isStatic, false,traits,parentTraitIndex);
|
|
}
|
|
}
|
|
}
|
|
|
|
public List<MultinameUsage> findMultinameUsage(int multinameIndex) {
|
|
List<MultinameUsage> ret = new ArrayList<MultinameUsage>();
|
|
if (multinameIndex == 0) {
|
|
return ret;
|
|
}
|
|
for (int c = 0; c < instance_info.length; c++) {
|
|
if (instance_info[c].name_index == multinameIndex) {
|
|
ret.add(new ClassNameMultinameUsage(multinameIndex, c));
|
|
}
|
|
if (instance_info[c].super_index == multinameIndex) {
|
|
ret.add(new ExtendsMultinameUsage(multinameIndex, c));
|
|
}
|
|
for (int i = 0; i < instance_info[c].interfaces.length; i++) {
|
|
if (instance_info[c].interfaces[i] == multinameIndex) {
|
|
ret.add(new ImplementsMultinameUsage(multinameIndex, c));
|
|
}
|
|
}
|
|
checkMultinameUsedInMethod(multinameIndex, instance_info[c].iinit_index, ret, c, 0, false, true,null,-1);
|
|
checkMultinameUsedInMethod(multinameIndex, class_info[c].cinit_index, ret, c, 0, true, true,null,-1);
|
|
findMultinameUsageInTraits(instance_info[c].instance_traits, multinameIndex, false, c, ret,-1);
|
|
findMultinameUsageInTraits(class_info[c].static_traits, multinameIndex, true, c, ret,-1);
|
|
}
|
|
loopm:
|
|
for (int m = 1; m < constants.constant_multiname.length; m++) {
|
|
if (constants.constant_multiname[m].kind == Multiname.TYPENAME) {
|
|
if (constants.constant_multiname[m].qname_index == multinameIndex) {
|
|
ret.add(new TypeNameMultinameUsage(m));
|
|
continue;
|
|
}
|
|
for (int mp : constants.constant_multiname[m].params) {
|
|
if (mp == multinameIndex) {
|
|
ret.add(new TypeNameMultinameUsage(m));
|
|
continue loopm;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public void autoFillAllBodyParams() {
|
|
for (int i = 0; i < bodies.length; i++) {
|
|
bodies[i].autoFillStats(this);
|
|
}
|
|
}
|
|
}
|