From f59019d0d4faa04380c40229dad3bd9886decade Mon Sep 17 00:00:00 2001 From: Jean Helie Date: Fri, 25 Mar 2022 18:41:01 +0100 Subject: [PATCH] also extract Unknown endpoints in training --- .../modelbuilding/extraction/ExtractEndpointDataTraining.ql | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/javascript/ql/experimental/adaptivethreatmodeling/modelbuilding/extraction/ExtractEndpointDataTraining.ql b/javascript/ql/experimental/adaptivethreatmodeling/modelbuilding/extraction/ExtractEndpointDataTraining.ql index 20ece497585..265ff1adcea 100644 --- a/javascript/ql/experimental/adaptivethreatmodeling/modelbuilding/extraction/ExtractEndpointDataTraining.ql +++ b/javascript/ql/experimental/adaptivethreatmodeling/modelbuilding/extraction/ExtractEndpointDataTraining.ql @@ -11,8 +11,9 @@ query predicate endpoints( DataFlow::Node endpoint, string queryName, string key, string value, string valueType ) { ExtractEndpointData::endpoints(endpoint, queryName, key, value, valueType) and - // only select endpoints that are either Sink or NotASink - ExtractEndpointData::endpoints(endpoint, queryName, "sinkLabel", ["Sink", "NotASink"], "string") and + // only select endpoints that are either Sink, NotASink or Unknown + ExtractEndpointData::endpoints(endpoint, queryName, "sinkLabel", ["Sink", "NotASink", "Unknown"], + "string") and // do not select endpoints filtered out by end-to-end evaluation ExtractEndpointData::endpoints(endpoint, queryName, "isExcludedFromEndToEndEvaluation", "false", "boolean") and