1 files changed, 134 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/mapping/Ancestries.java b/src/cuchaz/enigma/mapping/Ancestries.java
new file mode 100644
index 0000000..b7a5e24
--- /dev/null
+++ b/src/cuchaz/enigma/mapping/Ancestries.java
@@ -0,0 +1,134 @@
+/*******************************************************************************
+ * Copyright (c) 2014 Jeff Martin.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the GNU Public License v3.0
+ * which accompanies this distribution, and is available at
+ * http://www.gnu.org/licenses/gpl.html
+ * 
+ * Contributors:
+ *     Jeff Martin - initial API and implementation
+ ******************************************************************************/
+package cuchaz.enigma.mapping;
+
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipInputStream;
+
+import javassist.ByteArrayClassPath;
+import javassist.ClassPool;
+import javassist.CtClass;
+import javassist.NotFoundException;
+import javassist.bytecode.Descriptor;
+
+import com.google.common.collect.Maps;
+
+import cuchaz.enigma.Constants;
+
+public class Ancestries implements Serializable
+{
+        private static final long serialVersionUID = 738687982126844179L;
+        
+        private Map<String,String> m_superclasses;
+        
+        public Ancestries( )
+        {
+                m_superclasses = Maps.newHashMap();
+        }
+        
+        public void readFromJar( InputStream in )
+        throws IOException
+        {
+                ClassPool classPool = new ClassPool();
+                
+                ZipInputStream zin = new ZipInputStream( in );
+                ZipEntry entry;
+                while( ( entry = zin.getNextEntry() ) != null )
+                {
+                        // filter out non-classes
+                        if( entry.isDirectory() || !entry.getName().endsWith( ".class" ) )
+                        {
+                                continue;
+                        }
+                        
+                        // read the class into a buffer
+                        ByteArrayOutputStream bos = new ByteArrayOutputStream();
+                        byte[] buf = new byte[Constants.KiB];
+                        int totalNumBytesRead = 0;
+                        while( zin.available() > 0 )
+                        {
+                                int numBytesRead = zin.read( buf );
+                                if( numBytesRead < 0 )
+                                {
+                                        break;
+                                }
+                                bos.write( buf, 0, numBytesRead );
+                                
+                                // sanity checking
+                                totalNumBytesRead += numBytesRead;
+                                if( totalNumBytesRead > Constants.MiB )
+                                {
+                                        throw new Error( "Class file " + entry.getName() + " larger than 1 MiB! Something is wrong!" );
+                                }
+                        }
+                        
+                        // determine the class name (ie chop off the ".class")
+                        String className = Descriptor.toJavaName( entry.getName().substring( 0, entry.getName().length() - ".class".length() ) );
+                        
+                        // get a javassist handle for the class
+                        classPool.insertClassPath( new ByteArrayClassPath( className, bos.toByteArray() ) );
+                        try
+                        {
+                                CtClass c = classPool.get( className );
+                                addSuperclass( c.getName(), c.getClassFile().getSuperclass() );
+                        }
+                        catch( NotFoundException ex )
+                        {
+                                throw new Error( "Unable to load class: " + className );
+                        }
+                }
+        }
+        
+        public void addSuperclass( String className, String superclassName )
+        {
+                className = Descriptor.toJvmName( className );
+                superclassName = Descriptor.toJvmName( superclassName );
+                
+                if( className.equals( superclassName ) )
+                {
+                        throw new IllegalArgumentException( "Class cannot be its own superclass! " + className );
+                }
+                
+                if( !isJre( className ) && !isJre( superclassName ) )
+                {
+                        m_superclasses.put( className, superclassName );
+                }
+        }
+        
+        public String getSuperclassName( String className )
+        {
+                return m_superclasses.get( className );
+        }
+        
+        public List<String> getAncestry( String className )
+        {
+                List<String> ancestors = new ArrayList<String>();
+                while( className != null )
+                {
+                        className = getSuperclassName( className );
+                        ancestors.add( className );
+                }
+                return ancestors;
+        }
+        
+        private boolean isJre( String className )
+        {
+                return className.startsWith( "java/" )
+                        || className.startsWith( "javax/" );
+        }
+}