backend: add utils for interop support

backend.native/compiler/ir/backend.native/src/org/jetbrains/kotlin/backend/konan/Context.kt

@@ -3,6 +3,7 @@ package org.jetbrains.kotlin.backend.konan
 import llvm.LLVMDumpModule
 import llvm.LLVMModuleRef
 import org.jetbrains.kotlin.backend.jvm.descriptors.initialize
+import org.jetbrains.kotlin.backend.konan.InteropBuiltIns
 import org.jetbrains.kotlin.backend.konan.descriptors.deepPrint
 import org.jetbrains.kotlin.backend.konan.descriptors.synthesizedName
 import org.jetbrains.kotlin.backend.konan.ir.Ir
@@ -57,6 +58,10 @@ internal final class Context(val config: KonanConfig) : KonanBackendContext() {
     override val irBuiltIns
         get() = ir.irModule.irBuiltins
 
+    val interopBuiltIns by lazy {
+        InteropBuiltIns(this.builtIns)
+    }
+
     var llvmModule: LLVMModuleRef? = null
         set(module: LLVMModuleRef?) {
             if (field != null) {

backend.native/compiler/ir/backend.native/src/org/jetbrains/kotlin/backend/konan/InteropUtils.kt

@@ -0,0 +1,103 @@
+package org.jetbrains.kotlin.backend.konan
+
+import org.jetbrains.kotlin.builtins.KotlinBuiltIns
+import org.jetbrains.kotlin.descriptors.ClassDescriptor
+import org.jetbrains.kotlin.incremental.components.NoLookupLocation
+import org.jetbrains.kotlin.name.FqName
+import org.jetbrains.kotlin.name.Name
+import org.jetbrains.kotlin.resolve.scopes.MemberScope
+
+private val cPointerName = "CPointer"
+private val nativePointedName = "NativePointed"
+private val nativePtrName = "NativePtr"
+
+internal class InteropBuiltIns(builtIns: KonanBuiltIns) {
+
+    object FqNames {
+        val packageName = FqName("kotlinx.cinterop")
+
+        val nativePtr = packageName.child(Name.identifier(nativePtrName)).toUnsafe()
+        val cPointer = packageName.child(Name.identifier(cPointerName)).toUnsafe()
+        val nativePointed = packageName.child(Name.identifier(nativePointedName)).toUnsafe()
+    }
+
+    private val packageScope = builtIns.builtInsModule.getPackage(FqNames.packageName).memberScope
+
+    val getNativeNullPtr = packageScope.getContributedFunctions("getNativeNullPtr").single()
+
+    val getPointerSize = packageScope.getContributedFunctions("getPointerSize").single()
+
+    private val nativePtr = packageScope.getContributedClassifier(nativePtrName) as ClassDescriptor
+
+    private val nativePointed = packageScope.getContributedClassifier(nativePointedName) as ClassDescriptor
+
+    private val cPointer = this.packageScope.getContributedClassifier(cPointerName) as ClassDescriptor
+
+    val cPointerRawValue = cPointer.unsubstitutedMemberScope.getContributedVariables("rawValue").single()
+
+    val nativePointedRawPtrGetter =
+            nativePointed.unsubstitutedMemberScope.getContributedVariables("rawPtr").single().getter!!
+
+    val memberAt = packageScope.getContributedFunctions("memberAt").single()
+
+    val interpretPointed = packageScope.getContributedFunctions("interpretPointed").single()
+
+    val arrayGetByIntIndex = packageScope.getContributedFunctions("get").single {
+        KotlinBuiltIns.isInt(it.valueParameters.single().type)
+    }
+
+    val arrayGetByLongIndex = packageScope.getContributedFunctions("get").single {
+        KotlinBuiltIns.isLong(it.valueParameters.single().type)
+    }
+
+    val allocUninitializedArrayWithIntLength = packageScope.getContributedFunctions("allocArray").single {
+        it.valueParameters.size == 1 && KotlinBuiltIns.isInt(it.valueParameters[0].type)
+    }
+
+    val allocUninitializedArrayWithLongLength = packageScope.getContributedFunctions("allocArray").single {
+        it.valueParameters.size == 1 && KotlinBuiltIns.isLong(it.valueParameters[0].type)
+    }
+
+    val allocVariable = packageScope.getContributedFunctions("alloc").single {
+        it.valueParameters.size == 0
+    }
+
+    val typeOf = packageScope.getContributedFunctions("typeOf").single()
+
+    val variableClass = packageScope.getContributedClassifier("CVariable") as ClassDescriptor
+
+    val variableTypeClass =
+            variableClass.unsubstitutedInnerClassesScope.getContributedClassifier("Type") as ClassDescriptor
+
+    val variableTypeSize = variableTypeClass.unsubstitutedMemberScope.getContributedVariables("size").single()
+
+    val variableTypeAlign = variableTypeClass.unsubstitutedMemberScope.getContributedVariables("align").single()
+
+    val nativeMemUtils = packageScope.getContributedClassifier("nativeMemUtils") as ClassDescriptor
+
+    private val primitives = listOf(
+            builtIns.byte, builtIns.short, builtIns.int, builtIns.long,
+            builtIns.float, builtIns.double,
+            nativePtr
+    )
+
+    val readPrimitive = primitives.map {
+        nativeMemUtils.unsubstitutedMemberScope.getContributedFunctions("get" + it.name).single()
+    }.toSet()
+
+    val writePrimitive = primitives.map {
+        nativeMemUtils.unsubstitutedMemberScope.getContributedFunctions("put" + it.name).single()
+    }.toSet()
+
+    val nativePtrPlusLong = nativePtr.unsubstitutedMemberScope.getContributedFunctions("plus").single()
+
+}
+
+private fun MemberScope.getContributedVariables(name: String) =
+        this.getContributedVariables(Name.identifier(name), NoLookupLocation.FROM_BUILTINS)
+
+private fun MemberScope.getContributedClassifier(name: String) =
+        this.getContributedClassifier(Name.identifier(name), NoLookupLocation.FROM_BUILTINS)
+
+private fun MemberScope.getContributedFunctions(name: String) =
+        this.getContributedFunctions(Name.identifier(name), NoLookupLocation.FROM_BUILTINS)