Kotlin: Recognize generated files

Author: tamasvajk

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

@@ -51,9 +51,9 @@ library class MarkerCommentGeneratedFile extends GeneratedFile {
 /**
  * A marker comment that indicates that it is in a generated file.
  */
-private class GeneratedFileMarker extends Top instanceof JavadocElement {
+private class GeneratedFileMarker extends Top {
   GeneratedFileMarker() {
-    exists(string msg | msg = this.getText() |
+    exists(string msg | msg = this.(JavadocElement).getText() or msg = this.(KtComment).getText() |
       msg.regexpMatch("(?i).*\\bGenerated By\\b.*\\bDo not edit\\b.*") or
       msg.regexpMatch("(?i).*\\bThis (file|class|interface|art[ei]fact) (was|is|(has been)) (?:auto[ -]?)?gener(e?)ated.*") or
       msg.regexpMatch("(?i).*\\bAny modifications to this file will be lost\\b.*") or
@@ -65,3 +65,10 @@ private class GeneratedFileMarker extends Top instanceof JavadocElement {
     )
   }
 }
+
+/**
+ * A file detected as generated by the Kotlin Annotation Processing Tool (kapt). Detection is based on file path.
+ */
+private class KaptFile extends GeneratedFile {
+  KaptFile() { this.getRelativePath().matches("%/generated/source/kapt%") }
+}

java/ql/test/kotlin/library-tests/GeneratedFiles/Generated.expected

@@ -0,0 +1,2 @@
+| Generated.kt:0:0:0:0 | Generated |
+| generated/source/kapt/Generated.kt:0:0:0:0 | Generated |

java/ql/test/kotlin/library-tests/GeneratedFiles/Generated.kt

@@ -0,0 +1,3 @@
+// This file was auto generated by me
+
+class B

java/ql/test/kotlin/library-tests/GeneratedFiles/Generated.ql

@@ -0,0 +1,4 @@
+import java
+
+from GeneratedFile f
+select f

java/ql/test/kotlin/library-tests/GeneratedFiles/NonGenerated.kt

@@ -0,0 +1,3 @@
+// This file was not generated
+
+class A

java/ql/test/kotlin/library-tests/GeneratedFiles/generated/source/NonGenerated.kt

@@ -0,0 +1 @@
+class D

java/ql/test/kotlin/library-tests/GeneratedFiles/generated/source/kapt/Generated.kt

@@ -0,0 +1 @@
+class C