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: Local variables

Browse Source
This commit is contained in:
Ian Lynagh
2021-08-11 18:38:00 +01:00
parent 46add88bb5
commit 799cf64fd2
4 changed files with 22 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -17,6 +17,7 @@ import org.jetbrains.kotlin.ir.declarations.IrFile
import org.jetbrains.kotlin.ir.declarations.IrFunction
import org.jetbrains.kotlin.ir.declarations.IrValueParameter
import org.jetbrains.kotlin.ir.declarations.IrProperty
import org.jetbrains.kotlin.ir.declarations.IrVariable
import org.jetbrains.kotlin.ir.util.dump
import org.jetbrains.kotlin.ir.util.IdSignature
import org.jetbrains.kotlin.ir.util.packageFqName
@@ -305,11 +306,25 @@ class KotlinFileExtractor(val tw: TrapWriter) {
}
}
fun extractVariable(v: IrVariable, parent: Label<out DbStmtparent>, idx: Int) {
val id = tw.getFreshIdLabel<DbLocalvariabledeclexpr>()
val locId = tw.getLocation(v.startOffset, v.endOffset)
val typeId = useType(v.type)
tw.writeExprs_localvariabledeclexpr(id, typeId, parent, idx)
tw.writeHasLocation(id, locId)
val varId = tw.getFreshIdLabel<DbLocalvar>()
tw.writeLocalvars(varId, v.name.asString(), typeId, id)
tw.writeHasLocation(varId, locId)
}
fun extractStatement(s: IrStatement, callable: Label<out DbCallable>, parent: Label<out DbStmtparent>, idx: Int) {
when(s) {
is IrExpression -> {
extractExpression(s, callable, parent, idx)
}
is IrVariable -> {
extractVariable(s, parent, idx)
}
else -> {
extractorBug("Unrecognised IrStatement: " + s.javaClass)
}