[TS] Support await (#303)

Author: Lipen

buildSrc/src/main/kotlin/Dependencies.kt

@@ -6,7 +6,7 @@ object Versions {
     const val clikt = "5.0.0"
     const val detekt = "1.23.7"
     const val ini4j = "0.5.4"
-    const val jacodb = "5889d3c784"
+    const val jacodb = "da338ffc83"
     const val juliet = "1.3.2"
     const val junit = "5.9.3"
     const val kotlin = "2.1.0"

usvm-ts-dataflow/src/main/kotlin/org/usvm/dataflow/ts/infer/annotation/ExprTypeAnnotator.kt

@@ -72,6 +72,8 @@ class ExprTypeAnnotator(
     private val valueAnnotator: ValueTypeAnnotator,
 ) : EtsExpr.Visitor<EtsExpr> {
 
+    private fun annotate(local: EtsLocal) = valueAnnotator.visit(local)
+
     private fun annotate(value: EtsValue) = value.accept(valueAnnotator)
 
     private fun annotate(expr: EtsExpr) = expr.accept(this)
@@ -267,8 +269,8 @@ class ExprTypeAnnotator(
     )
 
     override fun visit(expr: EtsInstanceCallExpr): EtsExpr {
-        val baseInferred = annotate(expr.instance) as EtsLocal // safe cast
-        val argsInferred = expr.args.map { annotate(it) as EtsLocal } // safe cast
+        val baseInferred = annotate(expr.instance)
+        val argsInferred = expr.args.map { annotate(it) }
         val methodInferred = when (val baseType = baseInferred.type) {
             is EtsClassType -> {
                 val etsClass = scene.projectAndSdkClasses.find { it.signature == baseType.signature }
@@ -284,13 +286,13 @@ class ExprTypeAnnotator(
     }
 
     override fun visit(expr: EtsStaticCallExpr): EtsExpr {
-        val argsInferred = expr.args.map { annotate(it) as EtsLocal } // safe cast
+        val argsInferred = expr.args.map { annotate(it) }
         return expr.copy(args = argsInferred)
     }
 
     override fun visit(expr: EtsPtrCallExpr): EtsExpr {
-        val ptrInferred = annotate(expr.ptr) as EtsLocal // safe cast
-        val argsInferred = expr.args.map { annotate(it) as EtsLocal } // safe cast
+        val ptrInferred = annotate(expr.ptr)
+        val argsInferred = expr.args.map { annotate(it) }
         return expr.copy(ptr = ptrInferred, args = argsInferred)
     }
 }

usvm-ts/src/main/kotlin/org/usvm/machine/TsContext.kt

@@ -36,7 +36,6 @@ import org.usvm.machine.expr.TsUndefinedSort
 import org.usvm.machine.expr.TsUnresolvedSort
 import org.usvm.machine.expr.TsVoidSort
 import org.usvm.machine.expr.TsVoidValue
-import org.usvm.machine.expr.tctx
 import org.usvm.machine.interpreter.TsStepScope
 import org.usvm.machine.types.EtsFakeType
 import org.usvm.memory.UReadOnlyMemory
@@ -173,13 +172,12 @@ class TsContext(
         return this
     }
 
-    fun UHeapRef.unwrapRefWithPathConstraint(scope: TsStepScope): UHeapRef = with(tctx) {
-        if (this@unwrapRefWithPathConstraint.isFakeObject()) {
+    fun UHeapRef.unwrapRefWithPathConstraint(scope: TsStepScope): UHeapRef {
+        if (isFakeObject()) {
             scope.assert(getFakeType(scope).refTypeExpr)
-            extractRef(scope)
-        } else {
-            asExpr(addressSort)
+            return extractRef(scope)
         }
+        return this
     }
 
     fun createFakeObjectRef(): UConcreteHeapRef {
@@ -274,7 +272,6 @@ class Constants {
     }
 }
 
-
 enum class IntermediateLValueField {
     BOOL, FP, REF
 }