Python: ObjectAPI to ValueAPI: WrongNumberArgumentsInCall: ObjectAPI.qll: Adds getAFunctionCall and getAMethodCall predicates to FunctionValue

BekaValentine · BekaValentine · commit 2ad0f5af2df1 · 2020-04-03T20:04:32.000-07:00
diff --git a/python/ql/src/semmle/python/objects/ObjectAPI.qll b/python/ql/src/semmle/python/objects/ObjectAPI.qll
@@ -594,6 +594,17 @@ abstract class FunctionValue extends CallableValue {
         or
         result = TBuiltinClassObject(this.(BuiltinMethodObjectInternal).getReturnType())
     }
+
+    /** Gets a call-site from where this function is called as a function */
+    CallNode getAFunctionCall() { result.getFunction().pointsTo() = this }
+
+    /** Gets a call-site from where this function is called as a method */
+    CallNode getAMethodCall() {
+        exists(BoundMethodObjectInternal bm |
+            result.getFunction().pointsTo() = bm and
+            bm.getFunction() = this
+        )
+    }
 }
 
 /** Class representing Python functions */
