java-decompiler: post-import cleanup (classes moved)

1 files changed, 321 insertions, 0 deletions

diff --git a/src/org/jetbrains/java/decompiler/main/InitializerProcessor.java b/src/org/jetbrains/java/decompiler/main/InitializerProcessor.java
new file mode 100644
index 0000000..ff6233c
--- /dev/null
+++ b/src/org/jetbrains/java/decompiler/main/InitializerProcessor.java
@@ -0,0 +1,321 @@
+/*
+ *    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;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.jetbrains.java.decompiler.code.CodeConstants;
+import org.jetbrains.java.decompiler.main.ClassesProcessor.ClassNode;
+import org.jetbrains.java.decompiler.main.extern.IFernflowerPreferences;
+import org.jetbrains.java.decompiler.main.rels.ClassWrapper;
+import org.jetbrains.java.decompiler.main.rels.MethodWrapper;
+import org.jetbrains.java.decompiler.modules.decompiler.exps.AssignmentExprent;
+import org.jetbrains.java.decompiler.modules.decompiler.exps.Exprent;
+import org.jetbrains.java.decompiler.modules.decompiler.exps.FieldExprent;
+import org.jetbrains.java.decompiler.modules.decompiler.exps.InvocationExprent;
+import org.jetbrains.java.decompiler.modules.decompiler.exps.VarExprent;
+import org.jetbrains.java.decompiler.modules.decompiler.stats.RootStatement;
+import org.jetbrains.java.decompiler.modules.decompiler.stats.Statement;
+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.util.InterpreterUtil;
+
+
+public class InitializerProcessor {
+
+	public static void extractInitializers(ClassWrapper wrapper) {
+		
+		MethodWrapper meth = wrapper.getMethodWrapper("<clinit>", "()V");
+		if(meth != null && meth.root != null) {  // successfully decompiled static constructor
+			extractStaticInitializers(wrapper, meth);
+		}
+		
+		extractDynamicInitializers(wrapper);
+		
+		// required e.g. if anonymous class is being decompiled as a standard one.
+		// This can happen if InnerClasses attributes are erased
+		liftConstructor(wrapper);
+		
+		if(DecompilerContext.getOption(IFernflowerPreferences.HIDE_EMPTY_SUPER)) {
+			hideEmptySuper(wrapper);
+		}
+	}
+	
+	
+	private static void liftConstructor(ClassWrapper wrapper) {  
+		
+		for(MethodWrapper meth : wrapper.getMethods()) {
+			if("<init>".equals(meth.methodStruct.getName()) && meth.root != null) {
+				Statement firstdata = findFirstData(meth.root);
+				if(firstdata == null) {
+					return;
+				}
+
+				
+				int index = 0;
+				List<Exprent> lstExprents = firstdata.getExprents();
+				
+				for(Exprent exprent : lstExprents) {
+
+					int action = 0;
+					
+					if(exprent.type == Exprent.EXPRENT_ASSIGNMENT) {
+						AssignmentExprent asexpr = (AssignmentExprent)exprent;
+						if(asexpr.getLeft().type == Exprent.EXPRENT_FIELD && asexpr.getRight().type == Exprent.EXPRENT_VAR) {
+							FieldExprent fexpr = (FieldExprent)asexpr.getLeft();
+							if(fexpr.getClassname().equals(wrapper.getClassStruct().qualifiedName)) {
+								StructField structField = wrapper.getClassStruct().getField(fexpr.getName(), fexpr.getDescriptor().descriptorString);
+								if(structField != null && (structField.access_flags & CodeConstants.ACC_FINAL) != 0) {
+									action = 1;
+								}
+							}
+						}
+					} else if(index > 0 && exprent.type == Exprent.EXPRENT_INVOCATION && 
+							isInvocationInitConstructor((InvocationExprent)exprent, meth, wrapper, true)) {
+						// this() or super()
+						lstExprents.add(0, lstExprents.remove(index));
+						action = 2; 
+					}
+					
+					if(action != 1) {
+						break;
+					}
+					
+					index++;
+				}
+			}
+		}
+	}
+	
+	
+	private static void hideEmptySuper(ClassWrapper wrapper) {
+		
+		for(MethodWrapper meth : wrapper.getMethods()) {
+			if("<init>".equals(meth.methodStruct.getName()) && meth.root != null) {
+				Statement firstdata = findFirstData(meth.root);
+				if(firstdata == null || firstdata.getExprents().isEmpty()) {
+					return;
+				}
+				
+				Exprent exprent = firstdata.getExprents().get(0);
+				if(exprent.type == Exprent.EXPRENT_INVOCATION) {
+					InvocationExprent invexpr = (InvocationExprent)exprent;
+					if(isInvocationInitConstructor(invexpr, meth, wrapper, false) && invexpr.getLstParameters().isEmpty()) {
+						firstdata.getExprents().remove(0);
+					}
+				}
+			}
+		}
+	}
+	
+	private static void extractStaticInitializers(ClassWrapper wrapper, MethodWrapper meth) {
+		
+		RootStatement root = meth.root;
+		StructClass cl = wrapper.getClassStruct();
+		
+		Statement firstdata = findFirstData(root);
+		if(firstdata != null) {
+			while(!firstdata.getExprents().isEmpty()) {
+				Exprent exprent = firstdata.getExprents().get(0);
+				
+				boolean found = false;
+				
+				if(exprent.type == Exprent.EXPRENT_ASSIGNMENT) {
+					AssignmentExprent asexpr = (AssignmentExprent)exprent;
+					if(asexpr.getLeft().type == Exprent.EXPRENT_FIELD) {
+						FieldExprent fexpr = (FieldExprent)asexpr.getLeft();
+						if(fexpr.isStatic() && fexpr.getClassname().equals(cl.qualifiedName) &&
+										cl.hasField(fexpr.getName(), fexpr.getDescriptor().descriptorString)) {
+							
+							if(isExprentIndependent(asexpr.getRight(), meth)) {
+
+								String keyField = InterpreterUtil.makeUniqueKey(fexpr.getName(), fexpr.getDescriptor().descriptorString);
+								if(!wrapper.getStaticFieldInitializers().containsKey(keyField)) {
+    								wrapper.getStaticFieldInitializers().addWithKey(asexpr.getRight(), keyField);
+    								firstdata.getExprents().remove(0);
+    								found = true;
+    							}
+							}
+						}
+					}
+				}
+				
+				if(!found) {
+					break;
+				}
+			}
+		}
+	}
+	
+	private static void extractDynamicInitializers(ClassWrapper wrapper) {
+		
+		StructClass cl = wrapper.getClassStruct();
+
+		boolean isAnonymous = DecompilerContext.getClassprocessor().getMapRootClasses().get(cl.qualifiedName).type == ClassNode.CLASS_ANONYMOUS; 
+		
+		List<List<Exprent>> lstFirst = new ArrayList<List<Exprent>>();
+		List<MethodWrapper> lstMethWrappers = new ArrayList<MethodWrapper>();
+		
+		for(MethodWrapper meth : wrapper.getMethods()) {
+			if("<init>".equals(meth.methodStruct.getName()) && meth.root != null) { // successfully decompiled constructor
+				Statement firstdata = findFirstData(meth.root);
+				if(firstdata == null || firstdata.getExprents().isEmpty()) {
+					return;
+				}
+				lstFirst.add(firstdata.getExprents());
+				lstMethWrappers.add(meth);
+				
+				Exprent exprent = firstdata.getExprents().get(0);
+				if(!isAnonymous) { // FIXME: doesn't make sense 
+					if(exprent.type != Exprent.EXPRENT_INVOCATION || !isInvocationInitConstructor((InvocationExprent)exprent, meth, wrapper, false)) {
+						return;
+					}
+				}
+			}
+		}
+		
+		if(lstFirst.isEmpty()) {
+			return;
+		}
+		
+		for(;;) {
+			
+			String fieldWithDescr = null;
+			Exprent value = null;
+			
+			for(int i=0; i<lstFirst.size(); i++) {
+				
+				List<Exprent> lst = lstFirst.get(i);
+				
+				if(lst.size() < (isAnonymous?1:2)) {
+					return;
+				}
+				
+				Exprent exprent = lst.get(isAnonymous?0:1);
+				
+				boolean found = false;
+				
+				if(exprent.type == Exprent.EXPRENT_ASSIGNMENT) {
+					AssignmentExprent asexpr = (AssignmentExprent)exprent;
+					if(asexpr.getLeft().type == Exprent.EXPRENT_FIELD) {
+						FieldExprent fexpr = (FieldExprent)asexpr.getLeft();
+						if(!fexpr.isStatic() && fexpr.getClassname().equals(cl.qualifiedName) &&
+								cl.hasField(fexpr.getName(), fexpr.getDescriptor().descriptorString)) { // check for the physical existence of the field. Could be defined in a superclass.
+							
+							if(isExprentIndependent(asexpr.getRight(), lstMethWrappers.get(i))) {
+								String fieldKey = InterpreterUtil.makeUniqueKey(fexpr.getName(), fexpr.getDescriptor().descriptorString);
+								if(fieldWithDescr == null) {
+									fieldWithDescr = fieldKey;
+									value = asexpr.getRight();
+								} else {
+									if(!fieldWithDescr.equals(fieldKey) ||
+											!value.equals(asexpr.getRight())) {
+										return; 
+									}
+								}
+								found = true;
+							}
+						}
+					}
+				}
+				
+				if(!found) {
+					return; 
+				}
+			}
+			
+			if(!wrapper.getDynamicFieldInitializers().containsKey(fieldWithDescr)) {
+				wrapper.getDynamicFieldInitializers().addWithKey(value, fieldWithDescr);
+				
+				for(List<Exprent> lst : lstFirst) {
+					lst.remove(isAnonymous?0:1);
+				}
+			} else {
+				return;
+			}
+		}
+		
+	}
+	
+	private static boolean isExprentIndependent(Exprent exprent, MethodWrapper meth) {
+		
+		List<Exprent> lst = exprent.getAllExprents(true);
+		lst.add(exprent);
+		
+		for(Exprent expr : lst) {
+			switch(expr.type) {
+			case Exprent.EXPRENT_VAR:
+				VarVersionPaar varpaar = new VarVersionPaar((VarExprent)expr);
+				if(!meth.varproc.getExternvars().contains(varpaar)) {
+					String varname = meth.varproc.getVarName(varpaar);
+					
+					if(!varname.equals("this") && !varname.endsWith(".this")) { // FIXME: remove direct comparison with strings
+						return false;
+					}
+				}
+				break;
+			case Exprent.EXPRENT_FIELD:
+				return false;
+			}
+		}
+
+		return true;
+	}
+	
+	
+	private static Statement findFirstData(Statement stat) {
+
+		if(stat.getExprents() != null) {
+			return stat;
+		} else {
+			if(stat.isLabeled()) { // FIXME: Why??
+				return null;
+			}
+			
+			switch(stat.type) {
+			case Statement.TYPE_SEQUENCE:
+			case Statement.TYPE_IF:
+			case Statement.TYPE_ROOT:
+			case Statement.TYPE_SWITCH:
+			case Statement.TYPE_SYNCRONIZED:
+				return findFirstData(stat.getFirst());
+			default:
+				return null;
+			}
+		}
+	}
+	
+	private static boolean isInvocationInitConstructor(InvocationExprent inv, MethodWrapper meth, ClassWrapper wrapper, boolean withThis) {
+		
+		if(inv.getFunctype() == InvocationExprent.TYP_INIT) {
+			if(inv.getInstance().type == Exprent.EXPRENT_VAR) {
+				VarExprent instvar = (VarExprent)inv.getInstance();
+				VarVersionPaar varpaar = new VarVersionPaar(instvar);
+				
+				String classname = meth.varproc.getThisvars().get(varpaar);
+				
+				if(classname!=null) { // any this instance. TODO: Restrict to current class? 
+					if(withThis || !wrapper.getClassStruct().qualifiedName.equals(inv.getClassname())) {
+						return true;
+					}
+				}
+			}
+		}
+		
+		return false;
+	}
+}