[Wasm] Wasi stdlib implementation
KT-56608
This commit is contained in:
Igor Yakovlev
committed by
Zalim Bashorov
Zalim Bashorov
parent
090f393f97
commit
8cc0660693
@@ -0,0 +1,8 @@
|
||||
/*
|
||||
* Copyright 2010-2022 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package kotlin.wasm.internal
|
||||
|
||||
internal var isNotFirstWasmExportCall: Boolean = false
|
||||
@@ -0,0 +1,21 @@
|
||||
/*
|
||||
* Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package kotlin.wasm.internal
|
||||
|
||||
@Suppress("UNUSED_PARAMETER")
|
||||
@ExcludedFromCodegen
|
||||
/*
|
||||
* Compiler generates inplace next code:
|
||||
* ```
|
||||
* block (result anyref) {
|
||||
* local.get 0
|
||||
* extern.internalize
|
||||
* br_on_cast_fail 0000_0011b 0 (type any) (type $kotlin.Any)
|
||||
* return
|
||||
* }
|
||||
* ```
|
||||
*/
|
||||
internal fun returnArgumentIfItIsKotlinAny(): Unit = implementedAsIntrinsic
|
||||
@@ -1,23 +0,0 @@
|
||||
/*
|
||||
* Copyright 2010-2022 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package kotlin.wasm.internal
|
||||
|
||||
@Suppress("UNUSED_PARAMETER") // TODO: Remove after bootstrap update
|
||||
private fun throwJsError(message: String?, wasmTypeName: String?, stack: ExternalInterfaceType): Nothing {
|
||||
js("""
|
||||
const error = new Error();
|
||||
error.message = message;
|
||||
error.name = wasmTypeName;
|
||||
error.stack = stack;
|
||||
throw error;
|
||||
""")
|
||||
}
|
||||
|
||||
internal fun throwAsJsException(t: Throwable): Nothing {
|
||||
throwJsError(t.message, getSimpleName(t.typeInfo), t.jsStack)
|
||||
}
|
||||
|
||||
internal var isNotFirstWasmExportCall: Boolean = false
|
||||
@@ -1,376 +0,0 @@
|
||||
/*
|
||||
* Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
@file:Suppress("UNUSED_PARAMETER") // TODO: Remove after bootstrap update
|
||||
|
||||
package kotlin.wasm.internal
|
||||
|
||||
import kotlin.wasm.internal.reftypes.anyref
|
||||
import kotlin.wasm.unsafe.withScopedMemoryAllocator
|
||||
import kotlin.wasm.unsafe.UnsafeWasmMemoryApi
|
||||
|
||||
internal typealias ExternalInterfaceType = JsAny
|
||||
|
||||
internal class JsExternalBox @WasmPrimitiveConstructor constructor(val ref: ExternalInterfaceType) {
|
||||
override fun toString(): String =
|
||||
externrefToString(ref)
|
||||
|
||||
override fun equals(other: Any?): Boolean =
|
||||
if (other is JsExternalBox) {
|
||||
externrefEquals(ref, other.ref)
|
||||
} else {
|
||||
false
|
||||
}
|
||||
|
||||
override fun hashCode(): Int {
|
||||
var hashCode = _hashCode
|
||||
if (hashCode != 0) return hashCode
|
||||
hashCode = externrefHashCode(ref)
|
||||
_hashCode = hashCode
|
||||
return hashCode
|
||||
}
|
||||
}
|
||||
|
||||
@Suppress("DEPRECATION")
|
||||
//language=js
|
||||
@JsFun("""
|
||||
(() => {
|
||||
const dataView = new DataView(new ArrayBuffer(8));
|
||||
function numberHashCode(obj) {
|
||||
if ((obj | 0) === obj) {
|
||||
return obj | 0;
|
||||
} else {
|
||||
dataView.setFloat64(0, obj, true);
|
||||
return (dataView.getInt32(0, true) * 31 | 0) + dataView.getInt32(4, true) | 0;
|
||||
}
|
||||
}
|
||||
|
||||
const hashCodes = new WeakMap();
|
||||
function getObjectHashCode(obj) {
|
||||
const res = hashCodes.get(obj);
|
||||
if (res === undefined) {
|
||||
const POW_2_32 = 4294967296;
|
||||
const hash = (Math.random() * POW_2_32) | 0;
|
||||
hashCodes.set(obj, hash);
|
||||
return hash;
|
||||
}
|
||||
return res;
|
||||
}
|
||||
|
||||
function getStringHashCode(str) {
|
||||
var hash = 0;
|
||||
for (var i = 0; i < str.length; i++) {
|
||||
var code = str.charCodeAt(i);
|
||||
hash = (hash * 31 + code) | 0;
|
||||
}
|
||||
return hash;
|
||||
}
|
||||
|
||||
return (obj) => {
|
||||
if (obj == null) {
|
||||
return 0;
|
||||
}
|
||||
switch (typeof obj) {
|
||||
case "object":
|
||||
case "function":
|
||||
return getObjectHashCode(obj);
|
||||
case "number":
|
||||
return numberHashCode(obj);
|
||||
case "boolean":
|
||||
return obj ? 1231 : 1237;
|
||||
default:
|
||||
return getStringHashCode(String(obj));
|
||||
}
|
||||
}
|
||||
})()"""
|
||||
)
|
||||
private external fun externrefHashCode(ref: ExternalInterfaceType): Int
|
||||
|
||||
private fun externrefToString(ref: ExternalInterfaceType): String =
|
||||
js("String(ref)")
|
||||
|
||||
private fun externrefToInt(ref: ExternalInterfaceType): Int =
|
||||
js("Number(ref)")
|
||||
|
||||
private fun externrefToLong(ref: ExternalInterfaceType): Long =
|
||||
js("Number(ref)")
|
||||
|
||||
private fun externrefToBoolean(ref: ExternalInterfaceType): Boolean =
|
||||
js("Boolean(ref)")
|
||||
|
||||
private fun externrefToFloat(ref: ExternalInterfaceType): Float =
|
||||
js("Number(ref)")
|
||||
|
||||
private fun externrefToDouble(ref: ExternalInterfaceType): Double =
|
||||
js("Number(ref)")
|
||||
|
||||
private fun intToExternref(x: Int): JsNumber =
|
||||
js("x")
|
||||
|
||||
private fun longToExternref(x: Long): ExternalInterfaceType =
|
||||
js("x")
|
||||
|
||||
private fun booleanToExternref(x: Boolean): ExternalInterfaceType =
|
||||
js("x")
|
||||
|
||||
private fun floatToExternref(x: Float): ExternalInterfaceType =
|
||||
js("x")
|
||||
|
||||
private fun doubleToExternref(x: Double): JsNumber =
|
||||
js("x")
|
||||
|
||||
private fun externrefEquals(lhs: ExternalInterfaceType, rhs: ExternalInterfaceType): Boolean =
|
||||
js("lhs === rhs")
|
||||
|
||||
private external fun tryGetOrSetExternrefBox(
|
||||
ref: ExternalInterfaceType,
|
||||
ifNotCached: JsReference<JsExternalBox>
|
||||
): JsReference<JsExternalBox>?
|
||||
|
||||
@WasmNoOpCast
|
||||
@Suppress("unused")
|
||||
private fun Any?.asWasmAnyref(): anyref =
|
||||
implementedAsIntrinsic
|
||||
|
||||
@WasmOp(WasmOp.EXTERN_INTERNALIZE)
|
||||
private fun ExternalInterfaceType.externAsWasmAnyref(): anyref =
|
||||
implementedAsIntrinsic
|
||||
|
||||
@WasmOp(WasmOp.EXTERN_EXTERNALIZE)
|
||||
private fun Any.asWasmExternRef(): ExternalInterfaceType =
|
||||
implementedAsIntrinsic
|
||||
|
||||
internal fun isNullish(ref: ExternalInterfaceType?): Boolean =
|
||||
js("ref == null")
|
||||
|
||||
@Suppress("UNUSED_PARAMETER")
|
||||
@ExcludedFromCodegen
|
||||
/*
|
||||
* Compiler generates inplace next code:
|
||||
* ```
|
||||
* block (result anyref) {
|
||||
* local.get 0
|
||||
* extern.internalize
|
||||
* br_on_cast_fail 0000_0011b 0 (type any) (type $kotlin.Any)
|
||||
* return
|
||||
* }
|
||||
* ```
|
||||
*/
|
||||
internal fun returnArgumentIfItIsKotlinAny(): Unit = implementedAsIntrinsic
|
||||
|
||||
internal fun externRefToAny(ref: ExternalInterfaceType): Any? {
|
||||
// TODO rewrite it so to get something like:
|
||||
// block {
|
||||
// refAsAnyref
|
||||
// br_on_cast_fail null 0 $kotlin.Any
|
||||
// return
|
||||
// }
|
||||
// If ref is an instance of kotlin class -- return it casted to Any
|
||||
returnArgumentIfItIsKotlinAny()
|
||||
|
||||
// If we have Null in notNullRef -- return null
|
||||
// If we already have a box -- return it,
|
||||
// otherwise -- remember new box and return it.
|
||||
return tryGetOrSetExternrefBox(ref, JsExternalBox(ref).toJsReference())
|
||||
}
|
||||
|
||||
|
||||
internal fun anyToExternRef(x: Any): ExternalInterfaceType {
|
||||
return if (x is JsExternalBox)
|
||||
x.ref
|
||||
else
|
||||
x.asWasmExternRef()
|
||||
}
|
||||
|
||||
internal fun stringLength(x: ExternalInterfaceType): Int =
|
||||
js("x.length")
|
||||
|
||||
// kotlin string to js string export
|
||||
// TODO Uint16Array may work with byte endian different with Wasm (i.e. little endian)
|
||||
internal fun importStringFromWasm(address: Int, length: Int, prefix: ExternalInterfaceType?): JsString {
|
||||
js("""
|
||||
const mem16 = new Uint16Array(wasmExports.memory.buffer, address, length);
|
||||
const str = String.fromCharCode.apply(null, mem16);
|
||||
return (prefix == null) ? str : prefix + str;
|
||||
""")
|
||||
}
|
||||
|
||||
internal fun kotlinToJsStringAdapter(x: String?): JsString? {
|
||||
// Using nullable String to represent default value
|
||||
// for parameters with default values
|
||||
if (x == null) return null
|
||||
if (x.isEmpty()) return jsEmptyString
|
||||
|
||||
val srcArray = x.chars
|
||||
val stringLength = srcArray.len()
|
||||
val maxStringLength = STRING_INTEROP_MEM_BUFFER_SIZE / CHAR_SIZE_BYTES
|
||||
|
||||
@OptIn(UnsafeWasmMemoryApi::class)
|
||||
withScopedMemoryAllocator { allocator ->
|
||||
val memBuffer = allocator.allocate(stringLength.coerceAtMost(maxStringLength) * CHAR_SIZE_BYTES).address.toInt()
|
||||
|
||||
var result: ExternalInterfaceType? = null
|
||||
var srcStartIndex = 0
|
||||
while (srcStartIndex < stringLength - maxStringLength) {
|
||||
unsafeWasmCharArrayToRawMemory(srcArray, srcStartIndex, maxStringLength, memBuffer)
|
||||
result = importStringFromWasm(memBuffer, maxStringLength, result)
|
||||
srcStartIndex += maxStringLength
|
||||
}
|
||||
|
||||
unsafeWasmCharArrayToRawMemory(srcArray, srcStartIndex, stringLength - srcStartIndex, memBuffer)
|
||||
return importStringFromWasm(memBuffer, stringLength - srcStartIndex, result)
|
||||
}
|
||||
}
|
||||
|
||||
internal fun jsCheckIsNullOrUndefinedAdapter(x: ExternalInterfaceType?): ExternalInterfaceType? =
|
||||
x.takeIf { !isNullish(it) }
|
||||
|
||||
// js string to kotlin string import
|
||||
// TODO Uint16Array may work with byte endian different with Wasm (i.e. little endian)
|
||||
internal fun jsExportStringToWasm(src: ExternalInterfaceType, srcOffset: Int, srcLength: Int, dstAddr: Int) {
|
||||
js("""
|
||||
const mem16 = new Uint16Array(wasmExports.memory.buffer, dstAddr, srcLength);
|
||||
let arrayIndex = 0;
|
||||
let srcIndex = srcOffset;
|
||||
while (arrayIndex < srcLength) {
|
||||
mem16.set([src.charCodeAt(srcIndex)], arrayIndex);
|
||||
srcIndex++;
|
||||
arrayIndex++;
|
||||
}
|
||||
""")
|
||||
}
|
||||
|
||||
private const val STRING_INTEROP_MEM_BUFFER_SIZE = 65_536 // 1 page 4KiB
|
||||
|
||||
internal fun jsToKotlinStringAdapter(x: ExternalInterfaceType): String {
|
||||
val stringLength = stringLength(x)
|
||||
val dstArray = WasmCharArray(stringLength)
|
||||
if (stringLength == 0) {
|
||||
return dstArray.createString()
|
||||
}
|
||||
|
||||
@OptIn(UnsafeWasmMemoryApi::class)
|
||||
withScopedMemoryAllocator { allocator ->
|
||||
val maxStringLength = STRING_INTEROP_MEM_BUFFER_SIZE / CHAR_SIZE_BYTES
|
||||
val memBuffer = allocator.allocate(stringLength.coerceAtMost(maxStringLength) * CHAR_SIZE_BYTES).address.toInt()
|
||||
|
||||
var srcStartIndex = 0
|
||||
while (srcStartIndex < stringLength - maxStringLength) {
|
||||
jsExportStringToWasm(x, srcStartIndex, maxStringLength, memBuffer)
|
||||
unsafeRawMemoryToWasmCharArray(memBuffer, srcStartIndex, maxStringLength, dstArray)
|
||||
srcStartIndex += maxStringLength
|
||||
}
|
||||
|
||||
jsExportStringToWasm(x, srcStartIndex, stringLength - srcStartIndex, memBuffer)
|
||||
unsafeRawMemoryToWasmCharArray(memBuffer, srcStartIndex, stringLength - srcStartIndex, dstArray)
|
||||
}
|
||||
|
||||
return dstArray.createString()
|
||||
}
|
||||
|
||||
|
||||
private fun getJsEmptyString(): JsString =
|
||||
js("''")
|
||||
|
||||
private fun getJsTrue(): JsBoolean =
|
||||
js("true")
|
||||
|
||||
private fun getJsFalse(): JsBoolean =
|
||||
js("false")
|
||||
|
||||
private var _jsEmptyString: JsString? = null
|
||||
private val jsEmptyString: JsString
|
||||
get() {
|
||||
var value = _jsEmptyString
|
||||
if (value == null) {
|
||||
value = getJsEmptyString()
|
||||
_jsEmptyString = value
|
||||
}
|
||||
|
||||
return value
|
||||
}
|
||||
|
||||
private var _jsTrue: JsBoolean? = null
|
||||
private val jsTrue: JsBoolean
|
||||
get() {
|
||||
var value = _jsTrue
|
||||
if (value == null) {
|
||||
value = getJsTrue()
|
||||
_jsTrue = value
|
||||
}
|
||||
|
||||
return value
|
||||
}
|
||||
|
||||
private var _jsFalse: JsBoolean? = null
|
||||
private val jsFalse: JsBoolean
|
||||
get() {
|
||||
var value = _jsFalse
|
||||
if (value == null) {
|
||||
value = getJsFalse()
|
||||
_jsFalse = value
|
||||
}
|
||||
|
||||
return value
|
||||
}
|
||||
|
||||
internal fun numberToDoubleAdapter(x: Number): Double =
|
||||
x.toDouble()
|
||||
|
||||
internal fun kotlinToJsAnyAdapter(x: Any?): ExternalInterfaceType? =
|
||||
if (x == null) null else anyToExternRef(x)
|
||||
|
||||
internal fun jsToKotlinAnyAdapter(x: ExternalInterfaceType?): Any? =
|
||||
if (x == null) null else externRefToAny(x)
|
||||
|
||||
internal fun jsToKotlinByteAdapter(x: Int): Byte = x.toByte()
|
||||
internal fun jsToKotlinShortAdapter(x: Int): Short = x.toShort()
|
||||
internal fun jsToKotlinCharAdapter(x: Int): Char = x.toChar()
|
||||
|
||||
internal fun externRefToKotlinIntAdapter(x: ExternalInterfaceType): Int =
|
||||
externrefToInt(x)
|
||||
|
||||
internal fun externRefToKotlinBooleanAdapter(x: ExternalInterfaceType): Boolean =
|
||||
externrefToBoolean(x)
|
||||
|
||||
internal fun externRefToKotlinLongAdapter(x: ExternalInterfaceType): Long =
|
||||
externrefToLong(x)
|
||||
|
||||
internal fun externRefToKotlinFloatAdapter(x: ExternalInterfaceType): Float =
|
||||
externrefToFloat(x)
|
||||
|
||||
internal fun externRefToKotlinDoubleAdapter(x: ExternalInterfaceType): Double =
|
||||
externrefToDouble(x)
|
||||
|
||||
internal fun kotlinIntToExternRefAdapter(x: Int): JsNumber =
|
||||
intToExternref(x)
|
||||
|
||||
internal fun kotlinBooleanToExternRefAdapter(x: Boolean): JsBoolean =
|
||||
if (x) jsTrue else jsFalse
|
||||
|
||||
internal fun kotlinLongToExternRefAdapter(x: Long): ExternalInterfaceType =
|
||||
longToExternref(x)
|
||||
|
||||
internal fun kotlinFloatToExternRefAdapter(x: Float): ExternalInterfaceType =
|
||||
floatToExternref(x)
|
||||
|
||||
internal fun kotlinDoubleToExternRefAdapter(x: Double): JsNumber =
|
||||
doubleToExternref(x)
|
||||
|
||||
internal fun kotlinByteToExternRefAdapter(x: Byte): ExternalInterfaceType =
|
||||
intToExternref(x.toInt())
|
||||
|
||||
internal fun kotlinShortToExternRefAdapter(x: Short): ExternalInterfaceType =
|
||||
intToExternref(x.toInt())
|
||||
|
||||
internal fun kotlinCharToExternRefAdapter(x: Char): ExternalInterfaceType =
|
||||
intToExternref(x.toInt())
|
||||
|
||||
internal fun newJsArray(): ExternalInterfaceType =
|
||||
js("[]")
|
||||
|
||||
internal fun jsArrayPush(array: ExternalInterfaceType, element: ExternalInterfaceType) {
|
||||
js("array.push(element);")
|
||||
}
|
||||
@@ -1,5 +1,5 @@
|
||||
/*
|
||||
* Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Copyright 2010-2023 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
@@ -9,8 +9,6 @@
|
||||
|
||||
package kotlin.wasm.internal
|
||||
|
||||
import kotlin.wasm.internal.ExternalInterfaceType
|
||||
|
||||
@WasmOp(WasmOp.UNREACHABLE)
|
||||
internal fun wasm_unreachable(): Nothing =
|
||||
implementedAsIntrinsic
|
||||
@@ -495,10 +493,4 @@ internal fun wasm_i64_popcnt(a: Long): Long =
|
||||
@PublishedApi
|
||||
@WasmOp(WasmOp.I64_CTZ)
|
||||
internal fun wasm_i64_ctz(a: Long): Long =
|
||||
implementedAsIntrinsic
|
||||
|
||||
// Reference type operators
|
||||
|
||||
@WasmOp(WasmOp.REF_IS_NULL)
|
||||
internal external fun wasm_externref_is_null(a: ExternalInterfaceType?): Boolean =
|
||||
implementedAsIntrinsic
|
||||
Reference in New Issue
Block a user