Propagate uncertainty information to Evaluators

codyze-backends/cpg/src/main/kotlin/de/fraunhofer/aisec/codyze/backends/cpg/coko/CokoCpgBackend.kt

@@ -81,7 +81,7 @@ class CokoCpgBackend(config: BackendConfiguration) :
      */
     override fun order(baseNodes: Op, block: Order.() -> Unit): OrderEvaluator =
         OrderEvaluator(
-            baseNodes = baseNodes.cpgGetNodes(),
+            baseNodes = baseNodes.cpgGetNodes().keys,
             order = Order().apply(block)
         )
 

codyze-backends/cpg/src/main/kotlin/de/fraunhofer/aisec/codyze/backends/cpg/coko/dsl/DataItemCpgDsl.kt

@@ -42,9 +42,9 @@ fun DataItem<*>.cpgGetAllNodes(): Nodes =
 context(CokoBackend)
 fun DataItem<*>.cpgGetNodes(): Nodes {
     return when (this@DataItem) {
-        is ReturnValueItem -> op.cpgGetNodes().flatMap { it.getVariableInNextDFGOrThis() }
+        is ReturnValueItem -> op.cpgGetNodes().flatMap { it.key.getVariableInNextDFGOrThis() }
         is Value -> [email protected]()
-        is ArgumentItem -> op.cpgGetNodes().map { it.arguments[index] } // TODO: Do we count starting at 0 or 1?
+        is ArgumentItem -> op.cpgGetNodes().map { it.key.arguments[index] } // TODO: Do we count starting at 0 or 1?
     }
 }
 

codyze-backends/cpg/src/main/kotlin/de/fraunhofer/aisec/codyze/backends/cpg/coko/dsl/ImplementationDsl.kt

@@ -18,16 +18,22 @@
 package de.fraunhofer.aisec.codyze.backends.cpg.coko.dsl
 
 import de.fraunhofer.aisec.codyze.backends.cpg.coko.Nodes
+import de.fraunhofer.aisec.codyze.backends.cpg.coko.dsl.Result.*
 import de.fraunhofer.aisec.codyze.specificationLanguages.coko.core.CokoBackend
 import de.fraunhofer.aisec.codyze.specificationLanguages.coko.core.CokoMarker
 import de.fraunhofer.aisec.codyze.specificationLanguages.coko.core.dsl.*
-import de.fraunhofer.aisec.codyze.specificationLanguages.coko.core.modelling.*
+import de.fraunhofer.aisec.codyze.specificationLanguages.coko.core.modelling.DataItem
+import de.fraunhofer.aisec.codyze.specificationLanguages.coko.core.modelling.Definition
+import de.fraunhofer.aisec.codyze.specificationLanguages.coko.core.modelling.ParameterGroup
+import de.fraunhofer.aisec.codyze.specificationLanguages.coko.core.modelling.Signature
 import de.fraunhofer.aisec.cpg.TranslationResult
 import de.fraunhofer.aisec.cpg.graph.*
 import de.fraunhofer.aisec.cpg.graph.declarations.ValueDeclaration
-import de.fraunhofer.aisec.cpg.graph.statements.expressions.*
-import de.fraunhofer.aisec.cpg.query.dataFlow
-import de.fraunhofer.aisec.cpg.query.executionPath
+import de.fraunhofer.aisec.cpg.graph.statements.expressions.CallExpression
+import de.fraunhofer.aisec.cpg.graph.statements.expressions.ConstructExpression
+import de.fraunhofer.aisec.cpg.graph.statements.expressions.Expression
+import de.fraunhofer.aisec.cpg.graph.statements.expressions.MemberExpression
+import de.fraunhofer.aisec.cpg.query.*
 
 //
 // all functions/properties defined here must use CokoBackend
@@ -60,34 +66,50 @@
  * [Definition]s.
  */
 context(CokoBackend)
-fun Op.cpgGetNodes(): Collection<CallExpression> =
+fun Op.cpgGetNodes(): Map<CallExpression, Result> =
     when (this@Op) {
-        is FunctionOp ->
-            [email protected]
-                .flatMap { def ->
-                    [email protected](def.fqn) {
-                        def.signatures.any { sig ->
-                            cpgSignature(*sig.parameters.toTypedArray()) &&
-                                sig.unorderedParameters.all { it?.cpgFlowsTo(arguments) ?: false }
+        is FunctionOp -> {
+            val results = mutableListOf<Result>()
+            val fqn = [email protected] { def ->
+                [email protected](def.fqn) {
+                    def.signatures.any { sig ->
+                        // We consider a result when both the signature and the flow are not invalid
+                        // However, if at least one of them is OPEN we propagate this information to the caller
+                        val signature = cpgSignature(*sig.parameters.toTypedArray())
+                        val flow = sig.unorderedParameters.allResult { it?.cpgFlowsTo(arguments) }
+                        if (signature != INVALID && flow != INVALID) {
+                            results.add(signature.and(flow))
+                        } else {
+                            false
                         }
                     }
                 }
-        is ConstructorOp ->
-            [email protected]
-                .flatMap { sig ->
-                    [email protected]([email protected]) {
-                        cpgSignature(*sig.parameters.toTypedArray()) &&
-                            sig.unorderedParameters.all { it?.cpgFlowsTo(arguments) ?: false }
+            }
+            fqn.zip(results).toMap()
+        }
+        is ConstructorOp -> {
+            val results = mutableListOf<Result>()
+            val fqn = [email protected] { sig ->
+                [email protected]([email protected]) {
+                    val signature = cpgSignature(*sig.parameters.toTypedArray())
+                    val flow = sig.unorderedParameters.allResult { it?.cpgFlowsTo(arguments) }
+                    if (signature != INVALID && flow != INVALID) {
+                        results.add(signature.and(flow))
+                    } else {
+                        false
                     }
                 }
-        is GroupingOp -> [email protected] { it.cpgGetNodes() }
+            }
+            fqn.zip(results).toMap()
+        }
+        is GroupingOp -> [email protected] { it.cpgGetNodes().entries }.associate { it.toPair() }
         is ConditionalOp -> {
             val resultNodes = resultOp.cpgGetNodes()
             val conditionNodes = conditionOp.cpgGetNodes()
             resultNodes.filter { resultNode ->
                 conditionNodes.any { conditionNode ->
                     // TODO: Is it correct to use the EOG relationship here?
-                    val result = executionPath(conditionNode, resultNode)
+                    val result = executionPath(conditionNode.key, resultNode.key)
                     result.value
                 }
             }
@@ -148,7 +170,7 @@
  * - If this is a Collection, we check if at least one of the elements flows to [that]
  * - If this is a [Node], we use the DFG of the CPG.
  */
-infix fun Any.cpgFlowsTo(that: Node): Boolean =
+infix fun Any.cpgFlowsTo(that: Node): Result =
     this.cpgFlowsTo(listOf(that))
 
 // it should only be available in the context of a CallExpression
@@ -159,22 +181,45 @@
  * - If this is a Collection, we check if at least one of the elements flows to [that]
  * - If this is a [Node], we use the DFG of the CPG.
  */
-infix fun Any.cpgFlowsTo(that: Collection<Node>): Boolean =
-    if (this is Wildcard) {
-        true
-    } else {
+infix fun Any.cpgFlowsTo(that: Collection<Node>): Result =
+    Result.convert(
         when (this) {
+            is Wildcard -> true
             is String -> that.any {
                 val regex = Regex(this)
                 regex.matches((it as? Expression)?.evaluate()?.toString().orEmpty()) || regex.matches(it.code.orEmpty())
             }
-            is Iterable<*> -> this.any { it?.cpgFlowsTo(that) ?: false }
-            is Array<*> -> this.any { it?.cpgFlowsTo(that) ?: false }
+            // Separate cases for IntRange and LongRange result in a huge performance boost for large ranges
+            is LongRange, is IntRange -> checkRange(that)
+            is Iterable<*> -> this.anyResult { it?.cpgFlowsTo(that) }
+            is Array<*> -> this.anyResult { it?.cpgFlowsTo(that) }
             is Node -> that.any { dataFlow(this, it).value }
-            is ParameterGroup -> this.parameters.all { it?.cpgFlowsTo(that) ?: false }
+            is ParameterGroup -> this.parameters.allResult { it?.cpgFlowsTo(that) }
             else -> this in that.map { (it as Expression).evaluate() }
         }
+    )
+
+private fun Any.checkRange(that: Collection<Node>): Boolean {
+    when (this) {
+        // I would love to combine the following two cases, but any implementation loses the benefit of
+        // quickly reading the last value of the range, therefore making the whole distinction useless.
+        is IntRange -> {
+            return that.all {
+                val minValue = min(it).value.toInt()
+                val maxValue = max(it).value.toInt()
+                minValue > this.first && maxValue < this.last
+            }
+        }
+        is LongRange -> {
+            return that.all {
+                val minValue = min(it).value.toInt()
+                val maxValue = max(it).value.toInt()
+                minValue > this.first && maxValue < this.last
+            }
+        }
+        else -> throw IllegalArgumentException("Unexpected type")
     }
+}
 
 context(CokoBackend)
 // TODO: better description
@@ -195,21 +240,20 @@
  * are not important to the analysis
  */
 @Suppress("UnsafeCallOnNullableType")
-fun CallExpression.cpgSignature(vararg parameters: Any?, hasVarargs: Boolean = false): Boolean {
+fun CallExpression.cpgSignature(vararg parameters: Any?, hasVarargs: Boolean = false): Result {
     // checks if amount of parameters is the same as amount of arguments of this CallExpression
-    return cpgCheckArgsSize(parameters, hasVarargs) &&
+    if (cpgCheckArgsSize(parameters, hasVarargs)) {
         // checks if the CallExpression matches with the parameters
-        parameters.withIndex().all { (i: Int, parameter: Any?) ->
+        return parameters.withIndex().allResult { (i: Int, parameter: Any?) ->
             when (parameter) {
                 // if any parameter is null, signature returns false
-                null -> false
+                null -> INVALID
                 is ParamWithType ->
                     // if `parameter` is a `ParamWithType` object we want to check the type and
                     // if there is dataflow
-                    cpgCheckType(parameter.type, i) &&
-                        parameter.param cpgFlowsTo arguments[i]
+                    if (cpgCheckType(parameter.type, i)) parameter.param cpgFlowsTo arguments[i] else INVALID
                 // checks if the type of the argument is the same
-                is Type -> cpgCheckType(parameter, i)
+                is Type -> Result.convert(cpgCheckType(parameter, i))
                 // check if any of the Nodes of the Op flow to the argument
                 is Op -> parameter.cpgGetNodes() cpgFlowsTo arguments[i]
                 // check if any of the Nodes of the DataItem flow to the argument
@@ -218,6 +262,8 @@
                 else -> parameter cpgFlowsTo arguments[i]
             }
         }
+    }
+    return INVALID
 }
 
 /** Checks the [type] against the type of the argument at [index] for the Call Expression */

codyze-backends/cpg/src/main/kotlin/de/fraunhofer/aisec/codyze/backends/cpg/coko/dsl/Result.kt

@@ -0,0 +1,86 @@
+/*
+ * Copyright (c) 2024, Fraunhofer AISEC. All rights reserved.
+ *
+ * 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.
+ */
+package de.fraunhofer.aisec.codyze.backends.cpg.coko.dsl
+
+import de.fraunhofer.aisec.codyze.backends.cpg.coko.dsl.Result.*
+
+/**
+ * A data class that serves as a ternary value for the analysis result.
+ *
+ * OPEN is used where we cannot deduce either VALID or INVALID results because of lack of information.
+ */
+enum class Result {
+    VALID,
+    INVALID,
+    OPEN;
+
+    companion object {
+        fun convert(from: Any?): Result {
+            return when (from) {
+                is Result -> from
+                is Boolean -> if (from) VALID else INVALID
+                else -> OPEN
+            }
+        }
+    }
+}
+
+/** returns VALID if all Results are VALID, otherwise returns OPEN if any result is OPEN, otherwise returns INVALID */
+fun <T> Iterable<T>.allResult(predicate: (T) -> Result?): Result {
+    var invalidFlag = false
+    for (element in this) {
+        if (predicate(element) == OPEN) {
+            return OPEN
+        } else if (predicate(element) == INVALID) {
+            invalidFlag = true
+        }
+    }
+    return if (invalidFlag) INVALID else VALID
+}
+
+/** returns VALID if any Result is VALID, otherwise returns OPEN if any result is OPEN, otherwise returns INVALID */
+fun <T> Iterable<T>.anyResult(predicate: (T) -> Result?): Result {
+    var openFlag = false
+    for (element in this) {
+        if (predicate(element) == VALID) {
+            return VALID
+        } else if (predicate(element) == OPEN) {
+            openFlag = true
+        }
+    }
+    return if (openFlag) OPEN else INVALID
+}
+
+/** returns VALID if all Results are VALID, otherwise returns OPEN if any result is OPEN, otherwise returns INVALID */
+fun <T> Array<T>.allResult(predicate: (T) -> Result?): Result {
+    return this.asIterable().allResult(predicate)
+}
+
+/** returns VALID if any Result is VALID, otherwise returns OPEN if any result is OPEN, otherwise returns INVALID */
+fun <T> Array<T>.anyResult(predicate: (T) -> Result?): Result {
+    return this.asIterable().anyResult(predicate)
+}
+
+/** precedence order for ternary and: OPEN > INVALID > VALID */
+fun Result.and(other: Result): Result {
+    return if (this == OPEN || other == OPEN) {
+        OPEN
+    } else if (this == INVALID || other == INVALID) {
+        INVALID
+    } else {
+        VALID
+    }
+}

codyze-backends/cpg/src/main/kotlin/de/fraunhofer/aisec/codyze/backends/cpg/coko/evaluators/CpgWheneverEvaluator.kt

@@ -331,7 +331,7 @@ class CpgWheneverEvaluator(premise: ConditionComponent) : WheneverEvaluator(prem
         callConditionComponent: CallConditionComponent,
         premiseNode: Node? = null
     ): EvaluationResult {
-        val callNodes = callConditionComponent.op.cpgGetNodes().filterWithDistanceToPremise(premiseNode)
+        val callNodes = callConditionComponent.op.cpgGetNodes().keys.filterWithDistanceToPremise(premiseNode)
         return EvaluationResult(callNodes, emptyList(), Problems())
     }
 

codyze-backends/cpg/src/main/kotlin/de/fraunhofer/aisec/codyze/backends/cpg/coko/evaluators/FollowsEvaluator.kt

@@ -38,10 +38,10 @@ class FollowsEvaluator(val ifOp: Op, val thenOp: Op) : Evaluator {
 
     override fun evaluate(context: EvaluationContext): List<CpgFinding> {
         val (unreachableThisNodes, thisNodes) =
-            with(this@CokoCpgBackend) { ifOp.cpgGetNodes().toSet() }
+            with(this@CokoCpgBackend) { ifOp.cpgGetNodes().keys }
                 .partition { it.isUnreachable() }
 
-        val thatNodes = with(this@CokoCpgBackend) { thenOp.cpgGetNodes().toSet() }
+        val thatNodes = with(this@CokoCpgBackend) { thenOp.cpgGetNodes().keys }
 
         val findings = mutableListOf<CpgFinding>()