Apply annotation feature to modules

Author: corona10

Modules/_ctypes/malloc_closure.c

@@ -14,6 +14,7 @@
 #  endif
 #endif
 #include "ctypes.h"
+#include "pycore_obmalloc.h"      // _PyMem_Annotate_Mmap()
 
 /* BLOCKSIZE can be adjusted.  Larger blocksize will take a larger memory
    overhead, but allocate less blocks from the system.  It may be that some
@@ -82,6 +83,7 @@ static void more_core(void)
                         0);
     if (item == (void *)MAP_FAILED)
         return;
+    _PyMem_Annotate_Mmap(item, count * sizeof(ITEM), "Python:more_core");
 #endif
 
 #ifdef MALLOC_CLOSURE_DEBUG

Modules/mmapmodule.c

@@ -26,6 +26,7 @@
 #include "pycore_abstract.h"      // _Py_convert_optional_to_ssize_t()
 #include "pycore_bytesobject.h"   // _PyBytes_Find()
 #include "pycore_fileutils.h"     // _Py_stat_struct
+#include "pycore_obmalloc.h"      // _PyMem_Annotate_Mmap()
 #include "pycore_weakref.h"       // FT_CLEAR_WEAKREFS()
 
 #include <stddef.h>               // offsetof()
@@ -1951,6 +1952,7 @@ new_mmap_object(PyTypeObject *type, PyObject *args, PyObject *kwdict)
         PyErr_SetFromErrno(PyExc_OSError);
         return NULL;
     }
+    _PyMem_Annotate_Mmap(m_obj->data, map_size, "Python:new_mmap_object");
     m_obj->access = (access_mode)access;
     return (PyObject *)m_obj;
 }

Objects/obmalloc.c

@@ -467,6 +467,7 @@ _PyMem_ArenaAlloc(void *Py_UNUSED(ctx), size_t size)
     if (ptr == MAP_FAILED)
         return NULL;
     assert(ptr != NULL);
+    _PyMem_Annotate_Mmap(ptr, size, "Python:PyMem_ArenaAlloc");
     return ptr;
 #else
     return malloc(size);

Python/jit.c

@@ -71,6 +71,9 @@ jit_alloc(size_t size)
     int prot = PROT_READ | PROT_WRITE;
     unsigned char *memory = mmap(NULL, size, prot, flags, -1, 0);
     int failed = memory == MAP_FAILED;
+    if (!failed) {
+        _PyMem_Annotate_Mmap(memory, size, "Python:jit_alloc");
+    }
 #endif
     if (failed) {
         jit_error("unable to allocate memory");

Python/perf_jit_trampoline.c

@@ -61,6 +61,7 @@
 #include "pycore_ceval.h"         // _PyPerf_Callbacks
 #include "pycore_frame.h"
 #include "pycore_interp.h"
+#include "pycore_obmalloc.h"      // _PyMem_Annotate_Mmap()
 #include "pycore_runtime.h"       // _PyRuntime
 
 #ifdef PY_HAVE_PERF_TRAMPOLINE
@@ -1085,6 +1086,7 @@ static void* perf_map_jit_init(void) {
         close(fd);
         return NULL;  // Memory mapping failed
     }
+    _PyMem_Annotate_Mmap(perf_jit_map_state.mapped_buffer, page_size, "Python:perf_map_jit_init");
 #endif
 
     perf_jit_map_state.mapped_size = page_size;

Python/perf_trampoline.c

@@ -132,6 +132,7 @@ any DWARF information available for them).
 #include "Python.h"
 #include "pycore_ceval.h"         // _PyPerf_Callbacks
 #include "pycore_interpframe.h"   // _PyFrame_GetCode()
+#include "pycore_obmalloc.h"      // _PyMem_Annotate_Mmap()
 #include "pycore_runtime.h"       // _PyRuntime
 
 
@@ -290,6 +291,7 @@ new_code_arena(void)
         perf_status = PERF_STATUS_FAILED;
         return -1;
     }
+    _PyMem_Annotate_Mmap(memory, mem_size, "Python:new_code_arena");
     void *start = &_Py_trampoline_func_start;
     void *end = &_Py_trampoline_func_end;
     size_t code_size = end - start;