c api check callback func

HXSecurity · Jan 20, 2022 · cdb3179 · cdb3179
1 parent c2c7080
commit cdb3179
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/dongtai_agent_python/assess_ext/patch.c b/dongtai_agent_python/assess_ext/patch.c
@@ -91,6 +91,10 @@ PyObject *install(PyObject *self, PyObject *arg) {
 }
 
 void patch_string_callback(char *prop_method_name, PyObject *source, PyObject *target, PyObject *hook_args, PyObject *hook_kwargs) {
+    if (!PyObject_HasAttrString(patch_module, prop_method_name)) {
+        return;
+    }
+
     PyObject *result;
     PyObject *prop_hook_args;
     int free_hook_args = 0;