Fuzzer fails to find combinations with constants from switch-case (#294)

Author: Markoutte

utbot-framework/src/main/kotlin/org/utbot/fuzzer/FuzzerFunctions.kt

@@ -17,14 +17,17 @@ import soot.CharType
 import soot.DoubleType
 import soot.FloatType
 import soot.IntType
+import soot.Local
 import soot.LongType
 import soot.ShortType
 import soot.Unit
 import soot.Value
 import soot.ValueBox
 import soot.jimple.Constant
+import soot.jimple.IntConstant
 import soot.jimple.InvokeExpr
 import soot.jimple.NullConstant
+import soot.jimple.internal.AbstractSwitchStmt
 import soot.jimple.internal.ImmediateBox
 import soot.jimple.internal.JAssignStmt
 import soot.jimple.internal.JCastExpr
@@ -33,8 +36,10 @@ import soot.jimple.internal.JGeExpr
 import soot.jimple.internal.JGtExpr
 import soot.jimple.internal.JIfStmt
 import soot.jimple.internal.JLeExpr
+import soot.jimple.internal.JLookupSwitchStmt
 import soot.jimple.internal.JLtExpr
 import soot.jimple.internal.JNeExpr
+import soot.jimple.internal.JTableSwitchStmt
 import soot.jimple.internal.JVirtualInvokeExpr
 import soot.toolkits.graph.ExceptionalUnitGraph
 
@@ -45,17 +50,18 @@ private val logger = KotlinLogging.logger {}
  */
 fun collectConstantsForFuzzer(graph: ExceptionalUnitGraph): Set<FuzzedConcreteValue> {
     return graph.body.units.reversed().asSequence()
-        .filter { it is JIfStmt || it is JAssignStmt }
+        .filter { it is JIfStmt || it is JAssignStmt || it is AbstractSwitchStmt}
         .flatMap { unit ->
             unit.useBoxes.map { unit to it.value }
         }
         .filter { (_, value) ->
-            value is Constant || value is JCastExpr || value is InvokeExpr
+            value is Constant || value is Local || value is JCastExpr || value is InvokeExpr
         }
         .flatMap { (unit, value) ->
             sequenceOf(
                 ConstantsFromIfStatement,
                 ConstantsFromCast,
+                ConstantsFromSwitchCase,
                 BoundValuesForDoubleChecks,
                 StringConstant,
             ).flatMap { finder ->
@@ -158,6 +164,24 @@ private object ConstantsFromCast: ConstantsFinder {
 
 }
 
+private object ConstantsFromSwitchCase: ConstantsFinder {
+    override fun find(graph: ExceptionalUnitGraph, unit: Unit, value: Value): List<FuzzedConcreteValue> {
+        if (unit !is JTableSwitchStmt && unit !is JLookupSwitchStmt) return emptyList()
+        val result = mutableListOf<FuzzedConcreteValue>()
+        if (unit is JTableSwitchStmt) {
+            for (i in unit.lowIndex..unit.highIndex) {
+                result.add(FuzzedConcreteValue(intClassId, i, FuzzedOp.EQ))
+            }
+        }
+        if (unit is JLookupSwitchStmt) {
+            unit.lookupValues.asSequence().filterIsInstance<IntConstant>().forEach {
+                result.add(FuzzedConcreteValue(intClassId, it.value, FuzzedOp.EQ))
+            }
+        }
+        return result
+    }
+}
+
 private object BoundValuesForDoubleChecks: ConstantsFinder {
     override fun find(graph: ExceptionalUnitGraph, unit: Unit, value: Value): List<FuzzedConcreteValue> {
         if (value !is InvokeExpr) return emptyList()