hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2025-12-17 09:13:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Kotlin: Identify the int type better

Browse Source
This commit is contained in:
Ian Lynagh
2021-08-10 11:49:36 +01:00
parent f608384085
commit 9a75ca7f62
1 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
java/kotlin-extractor/src/main/kotlin/KotlinExtractorExtension.kt
View File

@@ -20,6 +20,7 @@ import org.jetbrains.kotlin.ir.util.dump
import org.jetbrains.kotlin.ir.util.packageFqName
import org.jetbrains.kotlin.ir.visitors.IrElementVisitor
import org.jetbrains.kotlin.ir.IrFileEntry
import org.jetbrains.kotlin.ir.types.isInt
import org.jetbrains.kotlin.ir.types.IrType
import org.jetbrains.kotlin.ir.types.IrSimpleType
import org.jetbrains.kotlin.ir.expressions.IrBody
@@ -140,13 +141,20 @@ class KotlinFileExtractor(val tw: TrapWriter) {
}
@Suppress("UNUSED_PARAMETER")
fun useSimpleType(c: IrSimpleType): Label<out DbPrimitive> {
// TODO: This shouldn't assume all SimpleType's are Int
val label = "@\"type;int\""
val id: Label<DbPrimitive> = tw.getLabelFor(label, {
tw.writePrimitives(it, "int")
})
return id
fun useSimpleType(s: IrSimpleType): Label<out DbPrimitive> {
when {
s.isInt() -> {
val label = "@\"type;int\""
val id: Label<DbPrimitive> = tw.getLabelFor(label, {
tw.writePrimitives(it, "int")
})
return id
}
else -> {
extractorBug("Unrecognised IrSimpleType: " + s.javaClass)
return Label(0)
}
}
}
fun useClass(c: IrClass): Label<out DbClass> {