Kotlin: Add support for IrSyntheticBody

igfoo · igfoo · commit 2b973fa3b10a · 2022-05-10T19:51:06.000+01:00
I'm not sure it's worth adding an entity to the database for them,
although that would allow us to use a 'case' in the dbscheme for
the different kinds.

There's no QLL support for this info yet.
diff --git a/java/kotlin-extractor/src/main/kotlin/KotlinFileExtractor.kt b/java/kotlin-extractor/src/main/kotlin/KotlinFileExtractor.kt
@@ -459,21 +459,31 @@ open class KotlinFileExtractor(
 
     fun extractBody(b: IrBody, callable: Label<out DbCallable>) {
         when(b) {
-            is IrBlockBody -> extractBlockBody(b, callable, callable, 0)
-            else -> logger.warnElement(Severity.ErrorSevere, "Unrecognised IrBody: " + b.javaClass, b)
+            is IrBlockBody -> extractBlockBody(b, callable)
+            is IrSyntheticBody -> extractSyntheticBody(b, callable)
+            else -> {
+                logger.warnElement(Severity.ErrorSevere, "Unrecognised IrBody: " + b.javaClass, b)
+            }
         }
     }
 
-    fun extractBlockBody(b: IrBlockBody, callable: Label<out DbCallable>, parent: Label<out DbStmtparent>, idx: Int) {
+    fun extractBlockBody(b: IrBlockBody, callable: Label<out DbCallable>) {
         val id = tw.getFreshIdLabel<DbBlock>()
         val locId = tw.getLocation(b)
-        tw.writeStmts_block(id, parent, idx, callable)
+        tw.writeStmts_block(id, callable, 0, callable)
         tw.writeHasLocation(id, locId)
         for((sIdx, stmt) in b.statements.withIndex()) {
             extractStatement(stmt, callable, id, sIdx)
         }
     }
 
+    fun extractSyntheticBody(b: IrSyntheticBody, callable: Label<out DbCallable>) {
+        when (b.kind) {
+            IrSyntheticBodyKind.ENUM_VALUES -> tw.writeKtSyntheticBody(callable, 1)
+            IrSyntheticBodyKind.ENUM_VALUEOF -> tw.writeKtSyntheticBody(callable, 2)
+        }
+    }
+
     fun extractVariable(v: IrVariable, callable: Label<out DbCallable>, parent: Label<out DbStmtparent>, idx: Int) {
         val stmtId = tw.getFreshIdLabel<DbLocalvariabledeclstmt>()
         val locId = tw.getLocation(v)
diff --git a/java/ql/lib/config/semmlecode.dbscheme b/java/ql/lib/config/semmlecode.dbscheme
@@ -1137,3 +1137,10 @@ ktAnonymousClassDeclarationStmts(
   unique int id: @anonymousclassdeclstmt ref,
   int classId: @class ref
 )
+
+ktSyntheticBody(
+  unique int id: @callable ref,
+  int kind: int ref
+  // 1: ENUM_VALUES
+  // 2: ENUM_VALUEOF
+)
