This commit is contained in:
Rasmus Lerchedahl Petersen 2023-09-28 13:35:29 +02:00
Родитель eb1be08bce
Коммит 2a7b593285
2 изменённых файлов: 24 добавлений и 22 удалений

Просмотреть файл

@ -7,7 +7,7 @@ private import semmle.python.dataflow.new.DataFlow
private import semmle.python.Concepts
private import semmle.python.ApiGraphs
module BSon {
private module BSon {
/**
* ObjectId returns a string representing an id.
* If at any time ObjectId can't parse it's input (like when a tainted dict in passed in),

Просмотреть файл

@ -144,7 +144,7 @@ private module PyMongo {
private class MongoMapReduceQuery extends API::CallNode, NoSqlExecution::Range {
MongoMapReduceQuery() { this = mongoCollection().getMember("map_reduce").getACall() }
override DataFlow::Node getQuery() { result in [this.getArgByName("query")] }
override DataFlow::Node getQuery() { result = this.getArgByName("query") }
override predicate interpretsDict() { any() }
@ -178,10 +178,10 @@ private module PyMongo {
* See https://www.mongodb.com/docs/manual/reference/operator/aggregation/function/#mongodb-expression-exp.-function
*/
private class FunctionQueryOperator extends DataFlow::Node, Decoding::Range {
API::Node dictionary;
DataFlow::Node query;
FunctionQueryOperator() {
exists(API::Node dictionary |
dictionary =
mongoCollection()
.getMember(mongoCollectionMethodName())
@ -191,6 +191,7 @@ private module PyMongo {
.getSubscript("$function") and
query = dictionary.getSubscript("body").asSink() and
this = dictionary.asSink()
)
}
override DataFlow::Node getAnInput() { result = query }
@ -208,10 +209,10 @@ private module PyMongo {
* See https://www.mongodb.com/docs/manual/reference/operator/aggregation/accumulator/#mongodb-group-grp.-accumulator
*/
private class AccumulatorQueryOperator extends DataFlow::Node, Decoding::Range {
API::Node dictionary;
DataFlow::Node query;
AccumulatorQueryOperator() {
exists(API::Node dictionary |
dictionary =
mongoCollection()
.getMember("aggregate")
@ -221,6 +222,7 @@ private module PyMongo {
.getSubscript("$accumulator") and
query = dictionary.getSubscript(["init", "accumulate", "merge", "finalize"]).asSink() and
this = dictionary.asSink()
)
}
override DataFlow::Node getAnInput() { result = query }