Kotlin: Identify data classes during extraction

Author: tamasvajk

java/kotlin-extractor/src/main/kotlin/KotlinFileExtractor.kt

@@ -434,6 +434,10 @@ open class KotlinFileExtractor(
                     } else if (kind != ClassKind.CLASS && kind != ClassKind.OBJECT) {
                         logger.warnElement("Unrecognised class kind $kind", c)
                     }
+
+                    if (c.isData) {
+                        tw.writeKtDataClasses(classId)
+                    }
                 }
 
                 val locId = tw.getLocation(c)

java/ql/lib/config/semmlecode.dbscheme

@@ -1234,3 +1234,7 @@ ktFunctionOriginalNames(
   unique int id: @method ref,
   string name: string ref
 )
+
+ktDataClasses(
+  unique int id: @class ref
+)

java/ql/lib/semmle/code/java/Type.qll

@@ -726,6 +726,13 @@ class CompanionObject extends Class {
   Field getInstance() { type_companion_object(_, result, this) }
 }
 
+/**
+ * A Kotlin data class declaration.
+ */
+class DataClass extends Class {
+  DataClass() { ktDataClasses(this) }
+}
+
 /**
  * A record declaration.
  */

java/ql/test/kotlin/library-tests/data-classes/data_classes.expected

@@ -0,0 +1 @@
+| dc.kt:1:1:1:71 | ProtoMapValue |

java/ql/test/kotlin/library-tests/data-classes/data_classes.ql

@@ -0,0 +1,5 @@
+import java
+
+from DataClass c
+where c.fromSource()
+select c