1 | #ifndef Py_CPYTHON_PYMEM_H |
2 | # error "this header file must not be included directly" |
3 | #endif |
4 | |
5 | PyAPI_FUNC(void *) PyMem_RawMalloc(size_t size); |
6 | PyAPI_FUNC(void *) PyMem_RawCalloc(size_t nelem, size_t elsize); |
7 | PyAPI_FUNC(void *) PyMem_RawRealloc(void *ptr, size_t new_size); |
8 | PyAPI_FUNC(void) PyMem_RawFree(void *ptr); |
9 | |
10 | /* Try to get the allocators name set by _PyMem_SetupAllocators(). */ |
11 | PyAPI_FUNC(const char*) _PyMem_GetCurrentAllocatorName(void); |
12 | |
13 | /* strdup() using PyMem_RawMalloc() */ |
14 | PyAPI_FUNC(char *) _PyMem_RawStrdup(const char *str); |
15 | |
16 | /* strdup() using PyMem_Malloc() */ |
17 | PyAPI_FUNC(char *) _PyMem_Strdup(const char *str); |
18 | |
19 | /* wcsdup() using PyMem_RawMalloc() */ |
20 | PyAPI_FUNC(wchar_t*) _PyMem_RawWcsdup(const wchar_t *str); |
21 | |
22 | |
23 | typedef enum { |
24 | /* PyMem_RawMalloc(), PyMem_RawRealloc() and PyMem_RawFree() */ |
25 | PYMEM_DOMAIN_RAW, |
26 | |
27 | /* PyMem_Malloc(), PyMem_Realloc() and PyMem_Free() */ |
28 | PYMEM_DOMAIN_MEM, |
29 | |
30 | /* PyObject_Malloc(), PyObject_Realloc() and PyObject_Free() */ |
31 | PYMEM_DOMAIN_OBJ |
32 | } PyMemAllocatorDomain; |
33 | |
34 | typedef enum { |
35 | PYMEM_ALLOCATOR_NOT_SET = 0, |
36 | PYMEM_ALLOCATOR_DEFAULT = 1, |
37 | PYMEM_ALLOCATOR_DEBUG = 2, |
38 | PYMEM_ALLOCATOR_MALLOC = 3, |
39 | PYMEM_ALLOCATOR_MALLOC_DEBUG = 4, |
40 | #ifdef WITH_PYMALLOC |
41 | PYMEM_ALLOCATOR_PYMALLOC = 5, |
42 | PYMEM_ALLOCATOR_PYMALLOC_DEBUG = 6, |
43 | #endif |
44 | } PyMemAllocatorName; |
45 | |
46 | |
47 | typedef struct { |
48 | /* user context passed as the first argument to the 4 functions */ |
49 | void *ctx; |
50 | |
51 | /* allocate a memory block */ |
52 | void* (*malloc) (void *ctx, size_t size); |
53 | |
54 | /* allocate a memory block initialized by zeros */ |
55 | void* (*calloc) (void *ctx, size_t nelem, size_t elsize); |
56 | |
57 | /* allocate or resize a memory block */ |
58 | void* (*realloc) (void *ctx, void *ptr, size_t new_size); |
59 | |
60 | /* release a memory block */ |
61 | void (*free) (void *ctx, void *ptr); |
62 | } PyMemAllocatorEx; |
63 | |
64 | /* Get the memory block allocator of the specified domain. */ |
65 | PyAPI_FUNC(void) PyMem_GetAllocator(PyMemAllocatorDomain domain, |
66 | PyMemAllocatorEx *allocator); |
67 | |
68 | /* Set the memory block allocator of the specified domain. |
69 | |
70 | The new allocator must return a distinct non-NULL pointer when requesting |
71 | zero bytes. |
72 | |
73 | For the PYMEM_DOMAIN_RAW domain, the allocator must be thread-safe: the GIL |
74 | is not held when the allocator is called. |
75 | |
76 | If the new allocator is not a hook (don't call the previous allocator), the |
77 | PyMem_SetupDebugHooks() function must be called to reinstall the debug hooks |
78 | on top on the new allocator. */ |
79 | PyAPI_FUNC(void) PyMem_SetAllocator(PyMemAllocatorDomain domain, |
80 | PyMemAllocatorEx *allocator); |
81 | |
82 | /* Setup hooks to detect bugs in the following Python memory allocator |
83 | functions: |
84 | |
85 | - PyMem_RawMalloc(), PyMem_RawRealloc(), PyMem_RawFree() |
86 | - PyMem_Malloc(), PyMem_Realloc(), PyMem_Free() |
87 | - PyObject_Malloc(), PyObject_Realloc() and PyObject_Free() |
88 | |
89 | Newly allocated memory is filled with the byte 0xCB, freed memory is filled |
90 | with the byte 0xDB. Additional checks: |
91 | |
92 | - detect API violations, ex: PyObject_Free() called on a buffer allocated |
93 | by PyMem_Malloc() |
94 | - detect write before the start of the buffer (buffer underflow) |
95 | - detect write after the end of the buffer (buffer overflow) |
96 | |
97 | The function does nothing if Python is not compiled is debug mode. */ |
98 | PyAPI_FUNC(void) PyMem_SetupDebugHooks(void); |
99 | |