Handle shadowBlocks and outerMethod during become

src/de.hpi.swa.trufflesqueak/src/de/hpi/swa/trufflesqueak/model/CompiledCodeObject.java

@@ -367,9 +367,15 @@ public void become(final CompiledCodeObject other) {
         CompilerDirectives.transferToInterpreterAndInvalidate();
         final Object[] literals2 = other.literals;
         final byte[] bytes2 = other.bytes;
+        final EconomicMap<Integer, CompiledCodeObject> shadowBlocks2 = other.shadowBlocks;
+        final CompiledCodeObject outerMethod2 = other.outerMethod;
         other.setLiteralsAndBytes(literals, bytes);
-        setLiteralsAndBytes(literals2, bytes2);
+        other.shadowBlocks = shadowBlocks;
+        other.outerMethod = outerMethod;
         other.callTargetStable.invalidate();
+        setLiteralsAndBytes(literals2, bytes2);
+        shadowBlocks = shadowBlocks2;
+        outerMethod = outerMethod2;
         callTargetStable.invalidate();
     }
 
@@ -516,6 +522,15 @@ public void pointersBecomeOneWay(final Object[] from, final Object[] to) {
                     setLiteral(j, toPointer);
                 }
             }
+            if (fromPointer == outerMethod && to[i] instanceof CompiledCodeObject) {
+                outerMethod = (CompiledCodeObject) to[i];
+            }
+        }
+        // Migrate all shadow blocks
+        if (shadowBlocks != null) {
+            for (final CompiledCodeObject shadowBlock : shadowBlocks.getValues()) {
+                shadowBlock.pointersBecomeOneWay(from, to);
+            }
         }
     }