better error for modification attempts after finish()
[org.ibex.classgen.git] / src / org / ibex / classgen / MethodGen.java
index b82567e..6603ebd 100644 (file)
@@ -4,7 +4,10 @@ import java.io.*;
 import java.util.*;
 
 public class MethodGen implements CGConst {
-    private final static boolean EMIT_NOPS = true;
+    private final static boolean EMIT_NOPS = false;
+    
+    private static final int NO_CODE = -1;
+    private static final int FINISHED = -2;
     
     private final CPGen cp;
     private final String name;
@@ -75,7 +78,8 @@ public class MethodGen implements CGConst {
     
     private final void grow() { if(size == capacity) grow(size+1); }
     private final void grow(int newCap) {
-        if(capacity == -1) throw new IllegalStateException("method can't have code");
+        if(capacity == NO_CODE) throw new IllegalStateException("method can't have code");
+        if(capacity == FINISHED) throw new IllegalStateException("method has been finished");
         if(newCap <= capacity) return;
         newCap = Math.max(newCap,capacity == 0 ? 256 : capacity*2);
         
@@ -113,74 +117,84 @@ public class MethodGen implements CGConst {
     
     
     public final void set(int pos, byte op, boolean b) { set(pos,op,b?1:0); }
-    public final void set(int pos, byte op, int n) {
-        if(op == LDC) {
-            switch(n) {
-                case -1: set(pos,ICONST_M1); return;
-                case 0:  set(pos,ICONST_0);  return;
-                case 1:  set(pos,ICONST_1);  return;
-                case 2:  set(pos,ICONST_2);  return; 
-                case 3:  set(pos,ICONST_3);  return;
-                case 4:  set(pos,ICONST_4);  return;
-                case 5:  set(pos,ICONST_5);  return;
-            }
-            Object arg;
-            if(n >= -128 && n <= 127) { op = BIPUSH; arg = N(n); } 
-            else if(n >= -32767 && n <= 32767) { op = SIPUSH; arg = N(n); }
-            else { arg = cp.add(N(n)); }
-            this.op[pos] = op;
-            this.arg[pos] = arg;
-        } else {
-            set(pos,op,N(n));
-        }
-    }
     
-    public void set(int pos, byte op, Object arg) {
-        switch(op) {
+    // This MUST handle x{LOAD,STORE} and LDC with an int arg WITHOUT falling back to set(int,byte,Object)
+    public final void set(int pos, byte op, int n) {
+        Object arg = null;
+        OUTER: switch(op) {
+            case LDC:
+                switch(n) {
+                    case -1: op = ICONST_M1; break OUTER;
+                    case 0:  op = ICONST_0;  break OUTER;
+                    case 1:  op = ICONST_1;  break OUTER;
+                    case 2:  op = ICONST_2;  break OUTER; 
+                    case 3:  op = ICONST_3;  break OUTER;
+                    case 4:  op = ICONST_4;  break OUTER;
+                    case 5:  op = ICONST_5;  break OUTER;
+                }
+                if(n >= -128 && n <= 127) { op = BIPUSH; arg = N(n); } 
+                else if(n >= -32768 && n <= 32767) { op = SIPUSH; arg = N(n); }
+                else { arg = cp.add(N(n)); }
+                break;
             case ILOAD: case ISTORE: case LLOAD: case LSTORE: case FLOAD:
             case FSTORE: case DLOAD: case DSTORE: case ALOAD: case ASTORE:
-            {
-                int iarg = ((Integer)arg).intValue();
-                if(iarg >= 0 && iarg <= 3) {
+                if(n >= 0 && n <= 3) {
                     byte base = 0;
                     switch(op) {
-                        case ILOAD:  base = ILOAD_0; break;
+                        case ILOAD:  base = ILOAD_0;  break;
                         case ISTORE: base = ISTORE_0; break;
-                        case LLOAD:  base = LLOAD_0; break;
+                        case LLOAD:  base = LLOAD_0;  break;
                         case LSTORE: base = LSTORE_0; break; 
-                        case FLOAD:  base = FLOAD_0; break;
+                        case FLOAD:  base = FLOAD_0;  break;
                         case FSTORE: base = FSTORE_0; break;
-                        case DLOAD:  base = DLOAD_0; break;
+                        case DLOAD:  base = DLOAD_0;  break;
                         case DSTORE: base = DSTORE_0; break;
-                        case ALOAD:  base = ALOAD_0; break;
+                        case ALOAD:  base = ALOAD_0;  break;
                         case ASTORE: base = ASTORE_0; break;
                     }
-                    op = (byte)((base&0xff) + iarg);
+                    op = (byte)((base&0xff) + n);
                 } else {
-                    if(iarg >= maxLocals) maxLocals = iarg + 1;
+                    if(n >= maxLocals) maxLocals = n + 1;
+                    arg = N(n);
                 }
                 break;
-            }
+            default:
+                set(pos,op,N(n));
+                return;
+        }            
+        this.op[pos] = op;
+        this.arg[pos] = arg;
+    }
+    
+    public final void set(int pos, byte op, Object arg) {
+        switch(op) {
+            case ILOAD: case ISTORE: case LLOAD: case LSTORE: case FLOAD:
+            case FSTORE: case DLOAD: case DSTORE: case ALOAD: case ASTORE:
+                // set(int,byte,int) always handles these ops itself
+                set(pos,op,((Integer)arg).intValue());
+                return;
             case LDC:
+                // set(int,byte,int) always handles these opts itself
                 if(arg instanceof Integer) { set(pos,op,((Integer)arg).intValue()); return; }
                 if(arg instanceof Boolean) { set(pos,op,((Boolean)arg).booleanValue()); return; }
+                
                 if(arg instanceof Long) {
                     long l = ((Long)arg).longValue();
-                    if(l == 0L) { set(pos,LCONST_0); return; }
-                    if(l == 1L) { set(pos,LCONST_1); return; }
+                    if(l == 0L) { this.op[pos] = LCONST_0; return; }
+                    if(l == 1L) { this.op[pos] = LCONST_1; return; }
                 }
                 
                 if(arg instanceof Long || arg instanceof Double) op = LDC2_W;
-                // fall through
-            default: {
-                int opdata = OP_DATA[op&0xff];
-                if((opdata&OP_CPENT_FLAG) != 0 && !(arg instanceof CPGen.Ent))
-                    arg = cp.add(arg);
-                else if((opdata&OP_VALID_FLAG) == 0)
-                    throw new IllegalArgumentException("unknown bytecode");
                 break;
-            }
+            case INVOKEINTERFACE:
+                if(arg instanceof MethodRef) arg = new MethodRef.I((MethodRef)arg);
+                break;
         }
+        int opdata = OP_DATA[op&0xff];
+        if((opdata&OP_CPENT_FLAG) != 0 && !(arg instanceof CPGen.Ent))
+            arg = cp.add(arg);
+        else if((opdata&OP_VALID_FLAG) == 0)
+            throw new IllegalArgumentException("unknown bytecode");
         this.op[pos] = op;
         this.arg[pos] = arg;
     }
@@ -197,7 +211,7 @@ public class MethodGen implements CGConst {
         public int size() { return targets.length; }
         
         public int getTarget(int pos) { return ((Integer)targets[pos]).intValue(); }
-        public int getDefaultTarget() { return ((Integer)defaultTarget).intValue(); }
+        public int getDefaultTarget() { return ((Integer)defaultTarget).intValue(); }        
     }
     
     public static class TSI extends SI {
@@ -254,7 +268,7 @@ public class MethodGen implements CGConst {
     }
     
     private void _finish() throws IOException {
-        if(size == -1) return;
+        if(size == FINISHED) return;
         
         ByteArrayOutputStream baos = new ByteArrayOutputStream();
         DataOutput o = new DataOutputStream(baos);
@@ -296,6 +310,12 @@ public class MethodGen implements CGConst {
                     p += 1 + 3 + 4; // opcode itself, padding, default
                     if(op == TABLESWITCH) p += 4 + 4 + targets.length * 4; // lo, hi, targets
                     else p += 4 + targets.length * 4 * 2; // count, key,val * targets
+                    if(op == LOOKUPSWITCH) {
+                        int[] vals = ((LSI)si).vals;
+                        for(j=1;j<vals.length;j++)
+                            if(vals[j] <= vals[j-1])
+                                throw new IllegalStateException("out of order/duplicate lookupswitch values");
+                    }
                     break;
                 }
                 case LDC:
@@ -334,7 +354,7 @@ public class MethodGen implements CGConst {
                 case TABLESWITCH:
                 case LOOKUPSWITCH: {
                     SI si = (SI) arg[i];
-                    p++; // opcpde itself
+                    p++; // opcode itself
                     p = (p + 3) & ~3; // padding
                     p += 4; // default
                     if(op == TABLESWITCH) p += 4 + 4 + si.size() * 4; // lo, hi, targets
@@ -397,6 +417,8 @@ public class MethodGen implements CGConst {
                     }
                     break;
                 }
+                case WIDE:
+                    throw new Error("WIDE instruction not yet supported");
                     
                 default:
                     if((opdata & OP_BRANCH_FLAG) != 0) {
@@ -408,7 +430,7 @@ public class MethodGen implements CGConst {
                         if(argLength == 1) o.writeByte(v);
                         else if(argLength == 2) o.writeShort(v);
                         else throw new Error("should never happen");
-                    } else if(argLength == -1) {
+                    } else if(argLength == 7) {
                         throw new Error("should never happen - variable length instruction not explicitly handled");
                     } else {
                         int iarg  = ((Integer)arg).intValue();
@@ -416,7 +438,7 @@ public class MethodGen implements CGConst {
                             if(iarg < -128 || iarg >= 256) throw new ClassGen.Exn("overflow of s/u1 option");
                             o.writeByte(iarg);
                         } else if(argLength == 2) {
-                            if(iarg < -32767 || iarg >= 65536) throw new ClassGen.Exn("overflow of s/u2 option"); 
+                            if(iarg < -32768 || iarg >= 65536) throw new ClassGen.Exn("overflow of s/u2 option"); 
                             o.writeShort(iarg);
                         } else {
                             throw new Error("should never happen");
@@ -446,7 +468,7 @@ public class MethodGen implements CGConst {
             o.writeShort(((CPGen.Ent)thrownExceptions.get(e.nextElement())).getIndex());
         attrs.add("Exceptions",baos.toByteArray());
         
-        size = -1;        
+        size = capacity = FINISHED;        
     }
         
     public void dump(DataOutput o) throws IOException {