Skip to content

Commit 98923ce

Browse files
committed
ATM: update missing .qll
1 parent ae0d82e commit 98923ce

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

javascript/ql/experimental/adaptivethreatmodeling/modelbuilding/extraction/ExtractEndpointDataTraining.qll

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ private import experimental.adaptivethreatmodeling.NosqlInjectionATM as NosqlInj
1414
private import experimental.adaptivethreatmodeling.SqlInjectionATM as SqlInjectionAtm
1515
private import experimental.adaptivethreatmodeling.TaintedPathATM as TaintedPathAtm
1616
private import experimental.adaptivethreatmodeling.XssATM as XssAtm
17+
private import experimental.adaptivethreatmodeling.XssThroughDomATM as XssThroughDomAtm
1718

1819
/**
1920
* Gets the set of featureName-featureValue pairs for each endpoint in the training set.
@@ -214,6 +215,8 @@ DataFlow::Configuration getDataFlowCfg(Query query) {
214215
query instanceof TaintedPathQuery and result instanceof TaintedPathAtm::TaintedPathAtmConfig
215216
or
216217
query instanceof XssQuery and result instanceof XssAtm::DomBasedXssAtmConfig
218+
or
219+
query instanceof XssThroughDomQuery and result instanceof XssThroughDomAtm::XssThroughDomAtmConfig
217220
}
218221

219222
// TODO: Delete this once we are no longer surfacing `hasFlowFromSource`.

0 commit comments

Comments
 (0)