Copy sam wrappers during inline

#KT-21671 Fixed

compiler/backend/src/org/jetbrains/kotlin/codegen/SamWrapperCodegen.java

@@ -185,6 +185,8 @@ public class SamWrapperCodegen {
     private FqName getWrapperName(@NotNull KtFile containingFile) {
         FqName fileClassFqName = JvmFileClassUtil.getFileClassInfoNoResolve(containingFile).getFileClassFqName();
         JavaClassDescriptor descriptor = samType.getJavaClassDescriptor();
+        //Change sam wrapper name template carefully cause it's used in inliner:
+        // see isSamWrapper/isSamWrapperConstructorCall in inlineCodegenUtils.kt
         int hash = PackagePartClassUtils.getPathHashCode(containingFile.getVirtualFile()) * 31 +
                 DescriptorUtils.getFqNameSafe(descriptor).hashCode();
         String shortName = String.format(

compiler/backend/src/org/jetbrains/kotlin/codegen/inline/MethodInliner.kt

@@ -180,7 +180,7 @@ class MethodInliner(
             }
 
             override fun anew(type: Type) {
-                if (isAnonymousClass(type.internalName)) {
+                if (isSamWrapper(type.internalName) || isAnonymousClass(type.internalName)) {
                     handleAnonymousObjectRegeneration()
                 }
 
@@ -430,12 +430,15 @@ class MethodInliner(
                         val lambdaInfo = getLambdaIfExistsAndMarkInstructions(sourceValue, true, instructions, sources, toDelete)
                         invokeCalls.add(InvokeCall(lambdaInfo, currentFinallyDeep))
                     }
+                    else if (isSamWrapperConstructorCall(owner, name)) {
+                        transformations.add(SamWrapperTransformationInfo(owner, inliningContext, isAlreadyRegenerated(owner)))
+                    }
                     else if (isAnonymousConstructorCall(owner, name)) {
                         val lambdaMapping = HashMap<Int, LambdaInfo>()
 
                         var offset = 0
                         var capturesAnonymousObjectThatMustBeRegenerated = false
-                        for (i in 0..paramCount - 1) {
+                        for (i in 0 until paramCount) {
                             val sourceValue = frame.getStack(firstParameterIndex + i)
                             val lambdaInfo = getLambdaIfExistsAndMarkInstructions(sourceValue, false, instructions, sources, toDelete
                             )

compiler/backend/src/org/jetbrains/kotlin/codegen/inline/SamWrapperTransformer.kt

@@ -0,0 +1,67 @@
+/*
+ * Copyright 2010-2017 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.kotlin.codegen.inline
+
+import org.jetbrains.org.objectweb.asm.ClassReader
+import org.jetbrains.org.objectweb.asm.ClassVisitor
+
+class SamWrapperTransformationInfo(override val oldClassName: String, private val inliningContext: InliningContext, private val alreadyRegenerated: Boolean): TransformationInfo {
+    override val nameGenerator: NameGenerator
+        get() = object: NameGenerator("stub") {
+            override fun getGeneratorClass(): String {
+                error ("Shouldn't be called on $oldClassName transformation")
+            }
+
+            override fun subGenerator(inliningMethod: String?): NameGenerator {
+                error ("Shouldn't be called on $oldClassName transformation")
+            }
+
+            override fun subGenerator(lambdaNoWhen: Boolean, nameSuffix: String?): NameGenerator {
+                error ("Shouldn't be called on $oldClassName transformation")
+            }
+        }
+
+    //TODO consider to use package class instead of inliningContext.root.callSiteInfo.ownerClassName
+    override val newClassName: String
+        get() = inliningContext.root.callSiteInfo.ownerClassName + "\$inlined" + "\$sam$".run { this + oldClassName.substringAfter(this) }
+
+    override fun shouldRegenerate(sameModule: Boolean) = !sameModule && !alreadyRegenerated
+
+    override fun canRemoveAfterTransformation() = false
+
+    override fun createTransformer(inliningContext: InliningContext, sameModule: Boolean) =
+            SamWrapperTransformer(this, inliningContext)
+}
+
+class SamWrapperTransformer(transformationInfo: SamWrapperTransformationInfo, private val inliningContext: InliningContext) :
+        ObjectTransformer<SamWrapperTransformationInfo>(transformationInfo, inliningContext.state) {
+
+    override fun doTransform(parentRemapper: FieldRemapper): InlineResult {
+        val classReader = createClassReader()
+        val classBuilder = createRemappingClassBuilderViaFactory(inliningContext)
+
+        classReader.accept(object : ClassVisitor(API, classBuilder.visitor) {
+            override fun visit(version: Int, access: Int, name: String, signature: String?, superName: String, interfaces: Array<String>) {
+                classBuilder.defineClass(null, version, access, name, signature, superName, interfaces)
+            }
+
+        }, ClassReader.SKIP_FRAMES)
+        classBuilder.done()
+
+        return transformationResult
+    }
+}

compiler/backend/src/org/jetbrains/kotlin/codegen/inline/inlineCodegenUtils.kt

@@ -219,9 +219,15 @@ internal fun isWhenMappingAccess(internalName: String, fieldName: String): Boole
 internal fun isAnonymousSingletonLoad(internalName: String, fieldName: String): Boolean =
         JvmAbi.INSTANCE_FIELD == fieldName && isAnonymousClass(internalName)
 
-internal fun isAnonymousClass(internalName: String): Boolean {
-    return internalName.substringAfterLast('/').substringAfterLast("$", "").isInteger()
-}
+internal fun isSamWrapper(internalName: String) =
+        internalName.contains("\$sam$") && internalName.substringAfter("\$i$", "").run { length == 8 && toLongOrNull(16) != null }
+
+internal fun isSamWrapperConstructorCall(internalName: String, methodName: String) =
+        isConstructor(methodName) && isSamWrapper(internalName)
+
+internal fun isAnonymousClass(internalName: String) =
+        !isSamWrapper(internalName) &&
+        internalName.substringAfterLast('/').substringAfterLast("$", "").isInteger()
 
 fun wrapWithMaxLocalCalc(methodNode: MethodNode) =
         MaxStackFrameSizeAndLocalsCalculator(API, methodNode.access, methodNode.desc, methodNode)