hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2026-05-04 05:05:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

Python points-to: Fix up support for six.add_metaclass()

Browse Source
This commit is contained in:
Mark Shannon
2019-05-09 15:15:40 +01:00
parent 92d0aef6f4
commit 93f0b8f1b7
1 changed files with 20 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
python/ql/src/semmle/python/pointsto/PointsTo.qll
View File

@@ -1693,9 +1693,27 @@ cached module Types {
result = has_six_add_metaclass(cls).booleanOr(has_metaclass_var_metaclass(cls))
}
private ControlFlowNode decorator_call_callee(PythonClassObjectInternal cls) {
result = cls.getScope().getADecorator().getAFlowNode().(CallNode).getFunction()
}
private boolean has_six_add_metaclass(PythonClassObjectInternal cls) {
// TO DO...
result = false
exists(ControlFlowNode callee, ObjectInternal func |
callee = decorator_call_callee(cls) and
PointsToInternal::pointsTo(callee, _, func, _)
|
func = six_add_metaclass_function() and result = true
or
func != six_add_metaclass_function() and result = false
)
or
not exists(Module m | m.getName() = "six") and result = false
or
exists(Class pycls |
pycls = cls.getScope() and
not exists(pycls.getADecorator()) and
result = false
)
}
private boolean has_metaclass_var_metaclass(PythonClassObjectInternal cls) {