codeql/java/ql/test/library-tests/dataflow/taintsources/local.ql

import java
import semmle.code.java.dataflow.FlowSources
import utils.test.InlineExpectationsTest

predicate isTestSink(DataFlow::Node n) {
  exists(MethodCall ma | ma.getMethod().hasName("sink") | n.asExpr() = ma.getAnArgument())
}

module LocalValueConfig implements DataFlow::ConfigSig {
  predicate isSource(DataFlow::Node n) { n instanceof LocalUserInput }

  predicate isSink(DataFlow::Node n) { isTestSink(n) }
}

module LocalValueFlow = DataFlow::Global<LocalValueConfig>;

module LocalTaintConfig implements DataFlow::ConfigSig {
  predicate isSource(DataFlow::Node n) { n instanceof LocalUserInput }

  predicate isSink(DataFlow::Node n) { isTestSink(n) }
}

module LocalTaintFlow = TaintTracking::Global<LocalTaintConfig>;

module LocalFlowTest implements TestSig {
  string getARelevantTag() { result = ["hasLocalValueFlow", "hasLocalTaintFlow"] }

  predicate hasActualResult(Location location, string element, string tag, string value) {
    tag = "hasLocalValueFlow" and
    exists(DataFlow::Node sink | LocalValueFlow::flowTo(sink) |
      sink.getLocation() = location and
      element = sink.toString() and
      value = ""
    )
    or
    tag = "hasLocalTaintFlow" and
    exists(DataFlow::Node src, DataFlow::Node sink |
      LocalTaintFlow::flow(src, sink) and not LocalValueFlow::flow(src, sink)
    |
      sink.getLocation() = location and
      element = sink.toString() and
      value = ""
    )
  }
}

import MakeTest<LocalFlowTest>