From 34c1e8e64ec4575527a19fb4cb0640c57da784db Mon Sep 17 00:00:00 2001
From: jeff
Date: Mon, 18 Aug 2014 00:55:30 -0400
Subject: crap-ton of bug fixes for inner classes

---
 src/cuchaz/enigma/bytecode/BytecodeTools.java | 57 +++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)

(limited to 'src/cuchaz/enigma/bytecode/BytecodeTools.java')

diff --git a/src/cuchaz/enigma/bytecode/BytecodeTools.java b/src/cuchaz/enigma/bytecode/BytecodeTools.java
index 664350e..0de9bd6 100644
--- a/src/cuchaz/enigma/bytecode/BytecodeTools.java
+++ b/src/cuchaz/enigma/bytecode/BytecodeTools.java
@@ -15,6 +15,7 @@ import java.io.ByteArrayOutputStream;
 import java.io.DataInputStream;
 import java.io.DataOutputStream;
 import java.io.IOException;
+import java.util.List;
 import java.util.Map;
 import java.util.Set;
 
@@ -25,6 +26,7 @@ import javassist.bytecode.CodeAttribute;
 import javassist.bytecode.ConstPool;
 import javassist.bytecode.ExceptionTable;
 
+import com.beust.jcommander.internal.Lists;
 import com.google.common.collect.Maps;
 import com.google.common.collect.Sets;
 
@@ -266,4 +268,59 @@ public class BytecodeTools
 			);
 		}
 	}
+	
+	public static List<String> getParameterTypes( String signature )
+	{
+		List<String> types = Lists.newArrayList();
+		for( int i=0; i<signature.length(); )
+		{
+			char c = signature.charAt( i );
+			
+			// handle parens
+			if( c == '(' )
+			{
+				i++;
+				c = signature.charAt( i );
+			}
+			if( c == ')' )
+			{
+				break;
+			}
+			
+			// find a type
+			String type = null;
+			
+			int arrayDim = 0;
+			while( c == '[' )
+			{
+				// advance to array type
+				arrayDim++;
+				i++;
+				c = signature.charAt( i );
+			}
+			
+			if( c == 'L' )
+			{
+				// read class type
+				int pos = signature.indexOf( ';', i + 1 );
+				String className = signature.substring( i + 1, pos );
+				type = "L" + className + ";";
+				i = pos + 1;
+			}
+			else
+			{
+				// read primitive type
+				type = signature.substring( i, i + 1 );
+				i++;
+			}
+			
+			// was it an array?
+			while( arrayDim-- > 0 )
+			{
+				type = "[" + type;
+			}
+			types.add( type );
+		}
+		return types;
+	}
 }
-- 
cgit v1.2.3