org.jacodb.api.ext.cfg.JcInstructions.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jacodb-api Show documentation
analyse JVM bytecode with pleasure
There is a newer version: 1.4.5
/*
 *  Copyright 2022 UnitTestBot contributors (utbot.org)
 * 
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 * 

 *  http://www.apache.org/licenses/LICENSE-2.0
 * 
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

@file:JvmName("JcInstructions")

package org.jacodb.api.ext.cfg

import org.jacodb.api.cfg.DefaultJcExprVisitor
import org.jacodb.api.cfg.DefaultJcInstVisitor
import org.jacodb.api.cfg.JcArrayAccess
import org.jacodb.api.cfg.JcCallExpr
import org.jacodb.api.cfg.JcExpr
import org.jacodb.api.cfg.JcFieldRef
import org.jacodb.api.cfg.JcInst
import org.jacodb.api.cfg.JcInstList
import org.jacodb.api.cfg.JcLocal
import org.jacodb.api.cfg.JcRawExpr
import org.jacodb.api.cfg.JcRawExprVisitor
import org.jacodb.api.cfg.JcRawInst
import org.jacodb.api.cfg.JcRawInstVisitor
import org.jacodb.api.cfg.JcValue
import org.jacodb.api.cfg.LocalResolver
import org.jacodb.api.cfg.ValueResolver

fun JcInstList.apply(visitor: JcRawInstVisitor): JcInstList {
    instructions.forEach { it.accept(visitor) }
    return this
}

fun > JcInstList.applyAndGet(visitor: T, getter: (T) -> R): R {
    instructions.forEach { it.accept(visitor) }
    return getter(visitor)
}

fun  JcInstList.collect(visitor: JcRawInstVisitor): Collection {
    return instructions.map { it.accept(visitor) }
}


fun > JcRawInst.applyAndGet(visitor: T, getter: (T) -> R): R {
    this.accept(visitor)
    return getter(visitor)
}

fun > JcRawExpr.applyAndGet(visitor: T, getter: (T) -> R): R {
    this.accept(visitor)
    return getter(visitor)
}


object FieldRefVisitor : DefaultJcExprVisitor, DefaultJcInstVisitor {

    override val defaultExprHandler: (JcExpr) -> JcFieldRef?
        get() = { null }

    override val defaultInstHandler: (JcInst) -> JcFieldRef?
        get() = {
            it.operands.map { it.accept(this) }.firstOrNull { it != null }
        }

    override fun visitJcFieldRef(value: JcFieldRef): JcFieldRef {
        return value
    }
}

object ArrayAccessVisitor : DefaultJcExprVisitor, DefaultJcInstVisitor {

    override val defaultExprHandler: (JcExpr) -> JcArrayAccess?
        get() = {
            it.operands.filterIsInstance().firstOrNull()
        }

    override val defaultInstHandler: (JcInst) -> JcArrayAccess?
        get() = {
            it.operands.map { it.accept(this) }.firstOrNull { it != null }
        }

}

object CallExprVisitor : DefaultJcInstVisitor {

    override val defaultInstHandler: (JcInst) -> JcCallExpr?
        get() = {
            it.operands.filterIsInstance().firstOrNull()
        }

}

val JcInst.fieldRef: JcFieldRef?
    get() {
        return accept(FieldRefVisitor)
    }

val JcInst.arrayRef: JcArrayAccess?
    get() {
        return accept(ArrayAccessVisitor)
    }

val JcInst.callExpr: JcCallExpr?
    get() {
        return accept(CallExprVisitor)
    }

val JcInstList.locals: Set
    get() {
        val resolver = LocalResolver().also {res ->
            forEach { it.accept(res) }
        }
        return resolver.result
    }

val JcInstList.values: Set
    get() {
        val resolver = ValueResolver().also {res ->
            forEach { it.accept(res) }
        }
        return resolver.result
    }