mirror of
				https://github.com/python/cpython.git
				synced 2025-10-31 13:41:24 +00:00 
			
		
		
		
	gh-122029: Log call events in sys.setprofile when it's a method with c function (GH-122072)
Log call events in sys.setprofile when it is a method with a C function.
This commit is contained in:
		
							parent
							
								
									41a91bd67f
								
							
						
					
					
						commit
						e91ef13861
					
				
					 3 changed files with 28 additions and 0 deletions
				
			
		|  | @ -479,6 +479,20 @@ def f(): | |||
|         sys.setprofile(lambda *args: None) | ||||
|         f() | ||||
| 
 | ||||
|     def test_method_with_c_function(self): | ||||
|         # gh-122029 | ||||
|         # When we have a PyMethodObject whose im_func is a C function, we | ||||
|         # should record both the call and the return. f = classmethod(repr) | ||||
|         # is just a way to create a PyMethodObject with a C function. | ||||
|         class A: | ||||
|             f = classmethod(repr) | ||||
|         events = [] | ||||
|         sys.setprofile(lambda frame, event, args: events.append(event)) | ||||
|         A().f() | ||||
|         sys.setprofile(None) | ||||
|         # The last c_call is the call to sys.setprofile | ||||
|         self.assertEqual(events, ['c_call', 'c_return', 'c_call']) | ||||
| 
 | ||||
| 
 | ||||
| if __name__ == "__main__": | ||||
|     unittest.main() | ||||
|  |  | |||
|  | @ -0,0 +1 @@ | |||
| Emit ``c_call`` events in :func:`sys.setprofile` when a ``PyMethodObject`` pointing to a ``PyCFunction`` is called. | ||||
|  | @ -121,6 +121,19 @@ sys_profile_call_or_return( | |||
|         Py_DECREF(meth); | ||||
|         return res; | ||||
|     } | ||||
|     else if (Py_TYPE(callable) == &PyMethod_Type) { | ||||
|         // CALL instruction will grab the function from the method,
 | ||||
|         // so if the function is a C function, the return event will
 | ||||
|         // be emitted. However, CALL event happens before CALL
 | ||||
|         // instruction, so we need to handle this case here.
 | ||||
|         PyObject* func = PyMethod_GET_FUNCTION(callable); | ||||
|         if (func == NULL) { | ||||
|             return NULL; | ||||
|         } | ||||
|         if (PyCFunction_Check(func)) { | ||||
|             return call_profile_func(self, func); | ||||
|         } | ||||
|     } | ||||
|     Py_RETURN_NONE; | ||||
| } | ||||
| 
 | ||||
|  |  | |||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue
	
	 Tian Gao
						Tian Gao