Flytter post processor.

raphw · Jun 2, 2020 · 3bca4ce · 3bca4ce
1 parent bb85163
commit 3bca4ce
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/byte-buddy-dep/src/main/java/net/bytebuddy/asm/Advice.java b/byte-buddy-dep/src/main/java/net/bytebuddy/asm/Advice.java
@@ -8716,11 +8716,11 @@ public void visitEnd() {
                         stackMapFrameHandler.injectReturnFrame(methodVisitor);
                         methodVisitor.visitVarInsn(Opcodes.ASTORE, exit ? argumentHandler.exit() : argumentHandler.enter());
                     }
-                    methodSizeHandler.requireStackSize(relocationHandler.apply(methodVisitor, exit ? argumentHandler.exit() : argumentHandler.enter()));
-                    stackMapFrameHandler.injectCompletionFrame(methodVisitor);
                     methodSizeHandler.recordMaxima(postProcessor
                             .resolve(instrumentedType, instrumentedMethod, assigner, argumentHandler)
                             .apply(mv, implementationContext).getMaximalSize(), EMPTY);
+                    methodSizeHandler.requireStackSize(relocationHandler.apply(methodVisitor, exit ? argumentHandler.exit() : argumentHandler.enter()));
+                    stackMapFrameHandler.injectCompletionFrame(methodVisitor);
                 }
 
                 @Override
@@ -9068,12 +9068,12 @@ public void apply() {
                         } else if (!adviceMethod.getReturnType().represents(void.class)) {
                             methodVisitor.visitVarInsn(Opcodes.ASTORE, isExitAdvice() ? argumentHandler.exit() : argumentHandler.enter());
                         }
-                        methodSizeHandler.requireStackSize(relocationHandler.apply(methodVisitor, isExitAdvice() ? argumentHandler.exit() : argumentHandler.enter()));
-                        stackMapFrameHandler.injectCompletionFrame(methodVisitor);
-                        methodSizeHandler.recordMaxima(Math.max(maximumStackSize, adviceMethod.getReturnType().getStackSize().getSize()), EMPTY);
                         methodSizeHandler.recordMaxima(postProcessor
                                 .resolve(instrumentedType, instrumentedMethod, assigner, argumentHandler)
                                 .apply(methodVisitor, implementationContext).getMaximalSize(), EMPTY);
+                        methodSizeHandler.requireStackSize(relocationHandler.apply(methodVisitor, isExitAdvice() ? argumentHandler.exit() : argumentHandler.enter()));
+                        stackMapFrameHandler.injectCompletionFrame(methodVisitor);
+                        methodSizeHandler.recordMaxima(Math.max(maximumStackSize, adviceMethod.getReturnType().getStackSize().getSize()), EMPTY);
                     }
 
                     /**