fernflower/src/org/jetbrains/java/decompiler/main/rels/ClassWrapper.java

/*
 *    Fernflower - The Analytical Java Decompiler
 *    http://www.reversed-java.com
 *
 *    (C) 2008 - 2010, Stiver
 *
 *    This software is NEITHER public domain NOR free software 
 *    as per GNU License. See license.txt for more details.
 *
 *    This software is distributed WITHOUT ANY WARRANTY; without 
 *    even the implied warranty of MERCHANTABILITY or FITNESS FOR 
 *    A PARTICULAR PURPOSE. 
 */

package org.jetbrains.java.decompiler.main.rels;

import java.io.IOException;
import java.util.HashSet;

import org.jetbrains.java.decompiler.code.CodeConstants;
import org.jetbrains.java.decompiler.main.DecompilerContext;
import org.jetbrains.java.decompiler.main.collectors.CounterContainer;
import org.jetbrains.java.decompiler.main.collectors.VarNamesCollector;
import org.jetbrains.java.decompiler.main.extern.IFernflowerLogger;
import org.jetbrains.java.decompiler.main.extern.IFernflowerPreferences;
import org.jetbrains.java.decompiler.modules.decompiler.exps.Exprent;
import org.jetbrains.java.decompiler.modules.decompiler.stats.RootStatement;
import org.jetbrains.java.decompiler.modules.decompiler.vars.VarProcessor;
import org.jetbrains.java.decompiler.modules.decompiler.vars.VarVersionPaar;
import org.jetbrains.java.decompiler.struct.StructClass;
import org.jetbrains.java.decompiler.struct.StructField;
import org.jetbrains.java.decompiler.struct.StructMethod;
import org.jetbrains.java.decompiler.struct.attr.StructGeneralAttribute;
import org.jetbrains.java.decompiler.struct.attr.StructLocalVariableTableAttribute;
import org.jetbrains.java.decompiler.struct.gen.MethodDescriptor;
import org.jetbrains.java.decompiler.util.InterpreterUtil;
import org.jetbrains.java.decompiler.util.VBStyleCollection;

public class ClassWrapper {

	private StructClass classStruct;
	
	private HashSet<String> hideMembers = new HashSet<String>(); 
	
	private VBStyleCollection<Exprent, String> staticFieldInitializers = new VBStyleCollection<Exprent, String>();

	private VBStyleCollection<Exprent, String> dynamicFieldInitializers = new VBStyleCollection<Exprent, String>();
	
	private VBStyleCollection<MethodWrapper, String> methods = new VBStyleCollection<MethodWrapper, String>();
	
	
	public ClassWrapper(StructClass classStruct) {
		this.classStruct = classStruct;
	}
	
	@SuppressWarnings("deprecation")
	public void init() throws IOException {
		
		DecompilerContext.setProperty(DecompilerContext.CURRENT_CLASS, classStruct);
		
		DecompilerContext.getLogger().startClass(classStruct.qualifiedName);

		// collect field names
		HashSet<String> setFieldNames = new HashSet<String>();
		for(StructField fd: classStruct.getFields()) {
			setFieldNames.add(fd.getName());
		}
		
		for(StructMethod mt: classStruct.getMethods()) {
			
			DecompilerContext.getLogger().startMethod(mt.getName()+" "+mt.getDescriptor());
			
			VarNamesCollector vc = new VarNamesCollector();
			DecompilerContext.setVarncollector(vc);
			
			CounterContainer counter = new CounterContainer();
			DecompilerContext.setCountercontainer(counter);
			
			DecompilerContext.setProperty(DecompilerContext.CURRENT_METHOD, mt);
			DecompilerContext.setProperty(DecompilerContext.CURRENT_METHOD_DESCRIPTOR, MethodDescriptor.parseDescriptor(mt.getDescriptor()));
			
			VarProcessor varproc = new VarProcessor();
			DecompilerContext.setProperty(DecompilerContext.CURRENT_VAR_PROCESSOR, varproc);
			
			Thread mtthread = null;
			RootStatement root = null;
			
			boolean isError = false;
			
			try {
				if(mt.containsCode()) {
					
					int maxsec = 10 * Integer.parseInt(DecompilerContext.getProperty(IFernflowerPreferences.MAX_PROCESSING_METHOD).toString());
					
					if(maxsec == 0) { // blocking wait
						root = MethodProcessorThread.codeToJava(mt, varproc);
					} else {
						MethodProcessorThread mtproc = new MethodProcessorThread(mt, varproc, DecompilerContext.getCurrentContext());
						mtthread = new Thread(mtproc);

						mtthread.start();

						int sec = 0; 
						while(mtthread.isAlive()) {

							synchronized(mtproc) {
								mtproc.wait(100);
							}

							if(maxsec > 0 && ++sec > maxsec) {
								DecompilerContext.getLogger().writeMessage("Processing time limit ("+maxsec+" sec.) for method " +
										mt.getName()+" "+mt.getDescriptor()+ " exceeded, execution interrupted.", IFernflowerLogger.ERROR);
								mtthread.stop();
								isError = true;
								break;
							}
						}

						if(!isError) {
							if(mtproc.getError() != null) {
								throw mtproc.getError();
							} else {
								root = mtproc.getRoot();
							}
						}
					}
					
				} else {
					boolean thisvar = (mt.getAccessFlags() & CodeConstants.ACC_STATIC) == 0;
					MethodDescriptor md = MethodDescriptor.parseDescriptor(mt.getDescriptor());
					
					int paramcount = 0;
					if(thisvar) {
						varproc.getThisvars().put(new VarVersionPaar(0,0), classStruct.qualifiedName);
						paramcount = 1;
					}
					paramcount += md.params.length;
					
					int varindex = 0;
					for(int i=0;i<paramcount;i++) {
						varproc.setVarName(new VarVersionPaar(varindex, 0), vc.getFreeName(varindex));
	
						if(thisvar) {
							if(i==0) {
								varindex++;
							} else {
								varindex+=md.params[i-1].stack_size;
							}
						} else {
							varindex+=md.params[i].stack_size;
						}
					}
				}

			} catch(ThreadDeath ex) {
				try {
					if(mtthread != null) {
						mtthread.stop();
					}
				} catch(Throwable ignored) { }
				
				throw ex;
			} catch(Throwable ex) {
				DecompilerContext.getLogger().writeMessage("Method "+mt.getName()+" "+mt.getDescriptor()+" couldn't be decompiled.", ex);
				isError = true;
			}
			
			MethodWrapper meth = new MethodWrapper(root, varproc, mt, counter);
			meth.decompiledWithErrors = isError;
			
			methods.addWithKey(meth, InterpreterUtil.makeUniqueKey(mt.getName(), mt.getDescriptor()));

			// rename vars so that no one has the same name as a field
			varproc.refreshVarNames(new VarNamesCollector(setFieldNames));

			// if debug information present and should be used
			if(DecompilerContext.getOption(IFernflowerPreferences.USE_DEBUG_VARNAMES)) {
				StructLocalVariableTableAttribute attr = (StructLocalVariableTableAttribute)mt.getAttributes().getWithKey(
						StructGeneralAttribute.ATTRIBUTE_LOCAL_VARIABLE_TABLE);
				
				if(attr != null) {
					varproc.setDebugVarNames(attr.getMapVarNames());
				}
			}
			
			DecompilerContext.getLogger().endMethod();
		}

		DecompilerContext.getLogger().endClass();
	}
	
	public MethodWrapper getMethodWrapper(String name, String descriptor) {
		return methods.getWithKey(InterpreterUtil.makeUniqueKey(name, descriptor));
	}
	
	public StructClass getClassStruct() {
		return classStruct;
	}

	public VBStyleCollection<MethodWrapper, String> getMethods() {
		return methods;
	}

	public HashSet<String> getHideMembers() {
		return hideMembers;
	}

	public VBStyleCollection<Exprent, String> getStaticFieldInitializers() {
		return staticFieldInitializers;
	}

	public VBStyleCollection<Exprent, String> getDynamicFieldInitializers() {
		return dynamicFieldInitializers;
	}

}