fixed bugs related to stack map tables.

git-svn-id: http://anonsvn.jboss.org/repos/javassist/trunk@378 30ef5769-5b8d-40dd-aea6-55b5d6557bb3

1 files changed, 10 insertions, 1 deletions

diff --git a/src/main/javassist/expr/ExprEditor.java b/src/main/javassist/expr/ExprEditor.java
index 9bba13b7..db9cb437 100644
--- a/src/main/javassist/expr/ExprEditor.java
+++ b/src/main/javassist/expr/ExprEditor.java
@@ -101,12 +101,21 @@ public class ExprEditor {
             }
         }
 
-        // codeAttr might be modified by other partys
+        // codeAttr might be modified by other partiess
         // so I check the current value of max-locals.
         if (codeAttr.getMaxLocals() < context.maxLocals)
             codeAttr.setMaxLocals(context.maxLocals);
 
         codeAttr.setMaxStack(codeAttr.getMaxStack() + context.maxStack);
+        try {
+            if (edited)
+                minfo.rebuildStackMapIf6(clazz.getClassPool(),
+                                         clazz.getClassFile2());
+        }
+        catch (BadBytecode b) {
+            throw new CannotCompileException(b.getMessage(), b);
+        }
+
         return edited;
     }