src/org/ibex/classgen/Type.java

   1 package org.ibex.classgen;
   2
   3 import java.util.StringTokenizer;
   4 import java.util.Hashtable;
   5
   6 public abstract class Type implements CGConst {
   7
   8     private static Hashtable instances = new Hashtable();  // this has to appear at the top of the file
   9
  10     // Public API //////////////////////////////////////////////////////////////////////////////
  11
  12     public static final Type VOID = new Primitive("V", "void");
  13     public static final Type INT = new Primitive("I", "int");
  14     public static final Type LONG = new Primitive("J", "long");
  15     public static final Type BOOLEAN = new Primitive("Z", "boolean");
  16     public static final Type DOUBLE = new Primitive("D", "double");
  17     public static final Type FLOAT = new Primitive("F", "float");
  18     public static final Type BYTE = new Primitive("B", "byte");
  19     public static final Type CHAR = new Primitive("C", "char");
  20     public static final Type SHORT = new Primitive("S", "short");
  21
  22     public static final Type.Class OBJECT = Type.Class.instance("java.lang.Object");
  23     public static final Type.Class STRING = Type.Class.instance("java.lang.String");
  24     public static final Type.Class STRINGBUFFER = Type.Class.instance("java.lang.StringBuffer");
  25     public static final Type.Class INTEGER_OBJECT = Type.Class.instance("java.lang.Integer");
  26     public static final Type.Class DOUBLE_OBJECT = Type.Class.instance("java.lang.Double");
  27     public static final Type.Class FLOAT_OBJECT = Type.Class.instance("java.lang.Float");
  28
  29     /** A zero element Type[] array (can be passed as the "args" param when a method takes no arguments */
  30     public static final Type[] NO_ARGS = new Type[0];
  31
  32     /**
  33      *  A "descriptor" is the classfile-mangled text representation of a type (see JLS section 4.3)
  34      *  guarantee: there will only be one instance of Type for a given descriptor ==> equals() and == are interchangeable
  35      */
  36     public static Type fromDescriptor(String d) {
  37         Type ret = (Type)instances.get(d);
  38         if (ret != null) return ret;
  39         if (d.startsWith("[")) return new Type.Array(Type.fromDescriptor(d.substring(1)));
  40         return new Type.Class(d);
  41     }
  42
  43     public final String  toString() { return super.toString(); }
  44     public abstract String debugToString();
  45
  46     public final String  getDescriptor() { return descriptor; }
  47
  48     public Type.Array  makeArray() { return (Type.Array)Type.fromDescriptor("["+descriptor); }
  49     public Type.Array  makeArray(int i) { return i==0 ? (Type.Array)this : makeArray().makeArray(i-1); }
  50
  51     public Type.Ref    asRef()       { throw new RuntimeException("attempted to use "+this+" as a Type.Ref, which it is not"); }
  52     public Type.Class  asClass()     { throw new RuntimeException("attempted to use "+this+" as a Type.Class, which it is not"); }
  53     public Type.Array  asArray()     { throw new RuntimeException("attempted to use "+this+" as a Type.Array, which it is not"); }
  54     public boolean     isPrimitive() { return false; }
  55     public boolean     isRef()       { return false; }
  56     public boolean     isClass()     { return false; }
  57     public boolean     isArray()     { return false; }
  58
  59     // Protected/Private //////////////////////////////////////////////////////////////////////////////
  60
  61     protected final String descriptor;
  62
  63     protected Type(String descriptor) {
  64         this.descriptor = descriptor;
  65         instances.put(descriptor, this);
  66     }
  67
  68     public static class Primitive extends Type {
  69         private String humanReadable;
  70         Primitive(String descriptor, String humanReadable) {
  71             super(descriptor);
  72             this.humanReadable = humanReadable;
  73         }
  74         public String debugToString() { return humanReadable; }
  75         public boolean     isPrimitive() { return true; }
  76     }
  77
  78     public abstract static class Ref extends Type {
  79         protected Ref(String descriptor) { super(descriptor); }
  80         public abstract String debugToString();
  81         public    Type.Ref asRef() { return this; }
  82         public    boolean  isRef() { return true; }
  83     }
  84
  85     public static class Array extends Type.Ref {
  86         public final Type base;
  87         protected Array(Type t) { super("[" + t.getDescriptor()); base = t; }
  88         public Type.Array asArray() { return this; }
  89         public boolean isArray() { return true; }
  90         public String debugToString() { return base.debugToString() + "[]"; }
  91         public Type getElementType() { return Type.fromDescriptor(getDescriptor().substring(0, getDescriptor().length()-1)); }
  92     }
  93
  94     public static class Class extends Type.Ref {
  95         protected Class(String s) { super(_initHelper(s)); }
  96         public Type.Class asClass() { return this; }
  97         public boolean isClass() { return true; }
  98         public static Type.Class instance(String className) {
  99             return (Type.Class)Type.fromDescriptor("L"+className.replace('.', '/')+";"); }
 100         public boolean extendsOrImplements(Type.Class c, Context cx) {
 101             if (this==c) return true;
 102             if (this==OBJECT) return false;
 103             ClassFile cf = cx.resolve(getName());
 104             if (cf==null) {
 105                 System.err.println("warning: could not resolve class " + getName());
 106                 return false;
 107             }
 108             if (cf.superType == c) return true;
 109             for(int i=0; i<cf.interfaces.length; i++) if (cf.interfaces[i].extendsOrImplements(c,cx)) return true;
 110             if (cf.superType == null) return false;
 111             return cf.superType.extendsOrImplements(c, cx);
 112         }
 113         String internalForm() { return descriptor.substring(1, descriptor.length()-1); }
 114         public String debugToString() { return internalForm().replace('/','.'); }
 115         public String getName() { return internalForm().replace('/','.'); }
 116         public String getShortName() {
 117             int p = descriptor.lastIndexOf('/');
 118             return p == -1 ? descriptor.substring(1,descriptor.length()-1) : descriptor.substring(p+1,descriptor.length()-1);
 119         }
 120         private static String _initHelper(String s) {
 121             if (!s.startsWith("L") || !s.endsWith(";")) throw new Error("invalid");
 122             return s;
 123         }
 124         String[] components() {
 125             StringTokenizer st = new StringTokenizer(descriptor.substring(1, descriptor.length()-1), "/");
 126             String[] a = new String[st.countTokens()];
 127             for(int i=0;st.hasMoreTokens();i++) a[i] = st.nextToken();
 128             return a;
 129         }
 130
 131         public Type.Class.Body getBody(Context cx) { return cx.resolve(this.getName()); }
 132         public abstract class Body extends HasAttributes {
 133             public abstract Type.Class.Method.Body[] methods();
 134             public abstract Type.Class.Field.Body addField(Type.Class.Field field, int flags);
 135             public abstract Type.Class.Field.Body[] fields();
 136             public Body(int flags, ClassFile.AttrGen attrs) {
 137                 super(flags, attrs);
 138                 if ((flags & ~(PUBLIC|FINAL|SUPER|INTERFACE|ABSTRACT)) != 0)
 139                     throw new IllegalArgumentException("invalid flags: " + Integer.toString(flags,16));
 140             }
 141         }
 142
 143         public Field field(String name, Type type) { return new Field(name, type); }
 144
 145         public Method method(String name, Type returnType, Type[] argTypes) { return new Method(name, returnType, argTypes); }
 146         public Method method(String leftCrap, String rightCrap) { return method(leftCrap+rightCrap); }
 147
 148         /** see JVM Spec section 2.10.2 */
 149         public Method method(String signature) {
 150             // FEATURE: This parser is ugly but it works (and shouldn't be a problem) might want to clean it up though
 151             String name = signature.substring(0, signature.indexOf('('));
 152             String s = signature.substring(signature.indexOf('('));
 153             if(!s.startsWith("(")) throw new IllegalArgumentException("invalid method type descriptor");
 154             int p = s.indexOf(')');
 155             if(p == -1) throw new IllegalArgumentException("invalid method type descriptor");
 156             String argsDesc = s.substring(1,p);
 157             String retDesc = s.substring(p+1);
 158             Type[] argsBuf = new Type[argsDesc.length()];
 159             int i;
 160             for(i=0,p=0;argsDesc.length() > 0;i++,p=0) {
 161                 while(p < argsDesc.length() && argsDesc.charAt(p) == '[') p++;
 162                 if(p == argsDesc.length())  throw new IllegalArgumentException("invalid method type descriptor");
 163                 if(argsDesc.charAt(p) == 'L') {
 164                     p = argsDesc.indexOf(';');
 165                     if(p == -1) throw new IllegalArgumentException("invalid method type descriptor");
 166                 }
 167                 argsBuf[i] = Type.fromDescriptor(argsDesc.substring(0,p+1));
 168                 argsDesc = argsDesc.substring(p+1);
 169             }
 170             Type args[] = new Type[i];
 171             System.arraycopy(argsBuf,0,args,0,i);
 172             return method(name, Type.fromDescriptor(retDesc), args);
 173         }
 174
 175         public abstract class Member {
 176             public final String name;
 177             private Member(String name) { this.name = name; }
 178             public Type.Class getDeclaringClass() { return Type.Class.this; }
 179             public String getName() { return name; }
 180             public abstract String getTypeDescriptor();
 181             public abstract String debugToString();
 182         }
 183
 184         public class Field extends Member {
 185             public final Type type;
 186             private Field(String name, Type t) { super(name); this.type = t; }
 187             public String getTypeDescriptor() { return type.getDescriptor(); }
 188             public Type getType() { return type; }
 189             public String debugToString() { return getDeclaringClass().debugToString()+"."+name+"["+type.debugToString()+"]"; }
 190             public class Body extends HasAttributes {
 191                 public Field getField() { return Field.this; }
 192                 public Body(int flags, ClassFile.AttrGen attrs) {
 193                     super(flags, attrs);
 194                     if ((flags & ~VALID_FIELD_FLAGS) != 0) throw new IllegalArgumentException("invalid flags");
 195                 }
 196             }
 197         }
 198
 199         public class Method extends Member {
 200             final Type[] argTypes;
 201             public final Type   returnType;
 202             public Type getReturnType()   { return returnType; }
 203             public int  getNumArgs()      { return argTypes.length; }
 204             public Type getArgType(int i) { return argTypes[i]; }
 205             public Type[] getArgTypes()   {
 206                 Type[] ret = new Type[argTypes.length];
 207                 System.arraycopy(argTypes, 0, ret, 0, ret.length);
 208                 return ret;
 209             }
 210             public boolean isConstructor() { return getName().equals("<init>"); }
 211             public boolean isClassInitializer() { return getName().equals("<clinit>"); }
 212             public String debugToString() {
 213                 StringBuffer sb = new StringBuffer();
 214                 if (name.equals("<clinit>")) sb.append("static ");
 215                 else {
 216                     if (name.equals("<init>"))
 217                         sb.append(Class.this.getShortName());
 218                     else
 219                         sb.append(returnType.debugToString()).append(".").append(name);
 220                     sb.append("(");
 221                     for(int i=0; i<argTypes.length; i++)
 222                         sb.append((i==0?"":", ")+argTypes[i].debugToString());
 223                     sb.append(") ");
 224                 }
 225                 return sb.toString();
 226             }
 227             private Method(String name, Type returnType, Type[] argTypes) {
 228                 super(name);
 229                 this.argTypes = argTypes;
 230                 this.returnType = returnType;
 231             }
 232             //public Method.Body getBody(Context cx) { }
 233             public String getTypeDescriptor() {
 234                 StringBuffer sb = new StringBuffer(argTypes.length*4);
 235                 sb.append("(");
 236                 for(int i=0;i<argTypes.length;i++) sb.append(argTypes[i].getDescriptor());
 237                 sb.append(")");
 238                 sb.append(returnType.getDescriptor());
 239                 return sb.toString();
 240             }
 241             public abstract class Body extends HasAttributes {
 242                 public abstract java.util.Hashtable getThrownExceptions();
 243                 public abstract void debugBodyToString(StringBuffer sb);
 244                 public Method getMethod() { return Method.this; }
 245                 public Body(int flags, ClassFile.AttrGen attrs) {
 246                     super(flags, attrs);
 247                     if ((flags & ~VALID_METHOD_FLAGS) != 0) throw new IllegalArgumentException("invalid flags");
 248                 }
 249                 public void debugToString(StringBuffer sb, String constructorName) {
 250                     int flags = getFlags();
 251                     sb.append("  ").append(ClassFile.flagsToString(flags,false));
 252                     sb.append(Method.this.debugToString());
 253                     java.util.Hashtable thrownExceptions = getThrownExceptions();
 254                     if (thrownExceptions.size() > 0) {
 255                         sb.append("throws");
 256                         for(java.util.Enumeration e = thrownExceptions.keys();e.hasMoreElements();)
 257                             sb.append(" ").append(((Type.Class)e.nextElement()).debugToString()).append(",");
 258                         sb.setLength(sb.length()-1);
 259                         sb.append(" ");
 260                     }
 261                     if ((flags & (NATIVE|ABSTRACT))==0) {
 262                         sb.append("{\n");
 263                         debugBodyToString(sb);
 264                         sb.append("  }\n");
 265                     } else {
 266                         sb.append(";");
 267                     }
 268                 }
 269             }
 270         }
 271     }
 272
 273     // FEATURE: This probably isn't the best place for these
 274     static String methodTypeDescriptor(Type[] argTypes, Type returnType) {
 275         StringBuffer sb = new StringBuffer(argTypes.length*4);
 276         sb.append("(");
 277         for(int i=0;i<argTypes.length;i++) sb.append(argTypes[i].getDescriptor());
 278         sb.append(")");
 279         sb.append(returnType.getDescriptor());
 280         return sb.toString();
 281     }
 282
 283 }