From e8623d439723162f4e30e131f66ed28bc6fa6c82 Mon Sep 17 00:00:00 2001
From: Jake Wharton <jw@squareup.com>
Date: Sun, 10 Jan 2016 19:38:22 -0800
Subject: [PATCH] Remove NEW, DUP instructions when changing NEWINVOKESPECIAL
 to INVOKESTATIC.

---
 .../lambdas/BackportLambdaClass.java          | 57 ++++++++++++-------
 1 file changed, 38 insertions(+), 19 deletions(-)

diff --git a/retrolambda/src/main/java/net/orfjackal/retrolambda/lambdas/BackportLambdaClass.java b/retrolambda/src/main/java/net/orfjackal/retrolambda/lambdas/BackportLambdaClass.java
index 4522c13e..3fe16845 100644
--- a/retrolambda/src/main/java/net/orfjackal/retrolambda/lambdas/BackportLambdaClass.java
+++ b/retrolambda/src/main/java/net/orfjackal/retrolambda/lambdas/BackportLambdaClass.java
@@ -5,6 +5,7 @@
 package net.orfjackal.retrolambda.lambdas;
 
 import org.objectweb.asm.*;
+import org.objectweb.asm.tree.*;
 
 import static org.objectweb.asm.Opcodes.*;
 
@@ -50,7 +51,7 @@ public MethodVisitor visitMethod(int access, String name, String desc, String si
         }
         MethodVisitor next = super.visitMethod(access, name, desc, signature, exceptions);
         next = new RemoveMagicLambdaConstructorCall(next);
-        next = new CallPrivateImplMethodsViaAccessMethods(next);
+        next = new CallPrivateImplMethodsViaAccessMethods(access, name, desc, signature, exceptions, next);
         return next;
     }
 
@@ -131,10 +132,13 @@ public void visitMethodInsn(int opcode, String owner, String name, String desc,
         }
     }
 
-    private class CallPrivateImplMethodsViaAccessMethods extends MethodVisitor {
+    private class CallPrivateImplMethodsViaAccessMethods extends MethodNode {
+        private final MethodVisitor next;
 
-        public CallPrivateImplMethodsViaAccessMethods(MethodVisitor next) {
-            super(ASM5, next);
+        public CallPrivateImplMethodsViaAccessMethods(int access, String name, String desc, String signature,
+                                                      String[] exceptions, MethodVisitor next) {
+            super(ASM5, access, name, desc, signature, exceptions);
+            this.next = next;
         }
 
         @Override
@@ -146,30 +150,45 @@ public void visitMethodInsn(int opcode, String owner, String name, String desc,
             if (owner.equals(implMethod.getOwner())
                     && name.equals(implMethod.getName())
                     && desc.equals(implMethod.getDesc())) {
+
+                if (implMethod.getTag() == H_NEWINVOKESPECIAL
+                        && accessMethod.getTag() == H_INVOKESTATIC) {
+                    // The impl is a private constructor which is called through an access method.
+                    // The current method already did NEW an instance, but we won't use it because
+                    // the access method will also instantiate it. The JVM would be OK with a non-empty
+                    // stack on ARETURN, but it causes a VerifyError on Android, so here we remove the
+                    // unused instance from the stack.
+                    boolean found = false;
+                    for (int i = instructions.size() - 1; i >= 1; i--) {
+                        AbstractInsnNode maybeNew = instructions.get(i - 1);
+                        AbstractInsnNode maybeDup = instructions.get(i);
+                        if (maybeNew.getOpcode() == NEW && maybeDup.getOpcode() == DUP) {
+                            instructions.remove(maybeNew);
+                            instructions.remove(maybeDup);
+                            found = true;
+                            break;
+                        }
+                    }
+                    if (!found) {
+                        throw new IllegalStateException(
+                                "Expected to find NEW, DUP instructions preceding NEWINVOKESPECIAL. Please file this as a bug.");
+                    }
+                }
+
                 super.visitMethodInsn(
                         Handles.getOpcode(accessMethod),
                         accessMethod.getOwner(),
                         accessMethod.getName(),
                         accessMethod.getDesc(),
                         accessMethod.getTag() == H_INVOKEINTERFACE);
-
-                if (implMethod.getTag() == H_NEWINVOKESPECIAL
-                        && accessMethod.getTag() == H_INVOKESTATIC) {
-                    // The impl is a private constructor which is called through an access method.
-                    // XXX: The current method already did NEW an instance, but we won't use it because
-                    // the access method will also instantiate it.
-                    // - The JVM would be OK with a non-empty stack on ARETURN, but it causes a VerifyError
-                    //   on Android, so here we remove the unused instance from the stack.
-                    // - We could improve this backporter so that it would remove the unnecessary
-                    //   "NEW, DUP" instructions, but that would be complicated.
-                    super.visitVarInsn(ASTORE, 1);
-                    super.visitInsn(POP);
-                    super.visitInsn(POP);
-                    super.visitVarInsn(ALOAD, 1);
-                }
             } else {
                 super.visitMethodInsn(opcode, owner, name, desc, itf);
             }
         }
+
+        @Override
+        public void visitEnd() {
+            accept(next);
+        }
     }
 }