C#: Add CaptureSourceModel query.

This commit is contained in:
Michael Nebel
2022-03-18 10:54:15 +01:00
parent 45234b1631
commit 4ae5dc323f
3 changed files with 48 additions and 0 deletions

View File

@@ -0,0 +1,12 @@
/**
* @name Capture source models.
* @description Finds APIs that act as sources as they expose already known sources.
* @id csharp/utils/model-generator/sink-models
*/
private import ModelGeneratorUtils
private import CaptureSourceModels
from TargetApi api, string sink
where sink = captureSource(api)
select sink order by sink

View File

@@ -0,0 +1,32 @@
private import CaptureSourceModelsSpecific
private import ModelGeneratorUtils
class FromSourceConfiguration extends TaintTracking::Configuration {
FromSourceConfiguration() { this = "FromSourceConfiguration" }
override predicate isSource(DataFlow::Node source) { sourceNode(source, _) }
override predicate isSink(DataFlow::Node sink) {
exists(TargetApi c |
sink instanceof ReturnNodeExt and
sink.getEnclosingCallable() = c
)
}
override DataFlow::FlowFeature getAFeature() {
result instanceof DataFlow::FeatureHasSinkCallContext
}
override predicate isAdditionalTaintStep(DataFlow::Node node1, DataFlow::Node node2) {
isRelevantTaintStep(node1, node2)
}
}
string captureSource(TargetApi api) {
exists(DataFlow::Node source, DataFlow::Node sink, FromSourceConfiguration config, string kind |
config.hasFlow(source, sink) and
sourceNode(source, kind) and
api = sink.getEnclosingCallable() and
result = asSourceModel(api, returnNodeAsOutput(sink), kind)
)
}

View File

@@ -0,0 +1,4 @@
import csharp
import semmle.code.csharp.dataflow.TaintTracking
import semmle.code.csharp.dataflow.ExternalFlow
import semmle.code.csharp.dataflow.internal.DataFlowImplCommon