ClangFormat: apply to source, most of intern
[blender.git] / intern / guardedalloc / intern / mallocn.c
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  */
16
17 /** \file
18  * \ingroup MEM
19  *
20  * Guarded memory allocation, and boundary-write detection.
21  */
22
23 #include "MEM_guardedalloc.h"
24
25 /* to ensure strict conversions */
26 #include "../../source/blender/blenlib/BLI_strict_flags.h"
27
28 #include <assert.h>
29
30 #include "mallocn_intern.h"
31
32 #ifdef WITH_JEMALLOC_CONF
33 /* If jemalloc is used, it reads this global variable and enables background
34  * threads to purge dirty pages. Otherwise we release memory too slowly or not
35  * at all if the thread that did the allocation stays inactive. */
36 const char *malloc_conf = "background_thread:true,dirty_decay_ms:4000";
37 #endif
38
39 size_t (*MEM_allocN_len)(const void *vmemh) = MEM_lockfree_allocN_len;
40 void (*MEM_freeN)(void *vmemh) = MEM_lockfree_freeN;
41 void *(*MEM_dupallocN)(const void *vmemh) = MEM_lockfree_dupallocN;
42 void *(*MEM_reallocN_id)(void *vmemh, size_t len, const char *str) = MEM_lockfree_reallocN_id;
43 void *(*MEM_recallocN_id)(void *vmemh, size_t len, const char *str) = MEM_lockfree_recallocN_id;
44 void *(*MEM_callocN)(size_t len, const char *str) = MEM_lockfree_callocN;
45 void *(*MEM_calloc_arrayN)(size_t len, size_t size, const char *str) = MEM_lockfree_calloc_arrayN;
46 void *(*MEM_mallocN)(size_t len, const char *str) = MEM_lockfree_mallocN;
47 void *(*MEM_malloc_arrayN)(size_t len, size_t size, const char *str) = MEM_lockfree_malloc_arrayN;
48 void *(*MEM_mallocN_aligned)(size_t len,
49                              size_t alignment,
50                              const char *str) = MEM_lockfree_mallocN_aligned;
51 void *(*MEM_mapallocN)(size_t len, const char *str) = MEM_lockfree_mapallocN;
52 void (*MEM_printmemlist_pydict)(void) = MEM_lockfree_printmemlist_pydict;
53 void (*MEM_printmemlist)(void) = MEM_lockfree_printmemlist;
54 void (*MEM_callbackmemlist)(void (*func)(void *)) = MEM_lockfree_callbackmemlist;
55 void (*MEM_printmemlist_stats)(void) = MEM_lockfree_printmemlist_stats;
56 void (*MEM_set_error_callback)(void (*func)(const char *)) = MEM_lockfree_set_error_callback;
57 bool (*MEM_consistency_check)(void) = MEM_lockfree_consistency_check;
58 void (*MEM_set_lock_callback)(void (*lock)(void),
59                               void (*unlock)(void)) = MEM_lockfree_set_lock_callback;
60 void (*MEM_set_memory_debug)(void) = MEM_lockfree_set_memory_debug;
61 size_t (*MEM_get_memory_in_use)(void) = MEM_lockfree_get_memory_in_use;
62 size_t (*MEM_get_mapped_memory_in_use)(void) = MEM_lockfree_get_mapped_memory_in_use;
63 unsigned int (*MEM_get_memory_blocks_in_use)(void) = MEM_lockfree_get_memory_blocks_in_use;
64 void (*MEM_reset_peak_memory)(void) = MEM_lockfree_reset_peak_memory;
65 size_t (*MEM_get_peak_memory)(void) = MEM_lockfree_get_peak_memory;
66
67 #ifndef NDEBUG
68 const char *(*MEM_name_ptr)(void *vmemh) = MEM_lockfree_name_ptr;
69 #endif
70
71 void *aligned_malloc(size_t size, size_t alignment)
72 {
73 #ifdef _WIN32
74   return _aligned_malloc(size, alignment);
75 #elif defined(__APPLE__)
76   /* On Mac OS X, both the heap and the stack are guaranteed 16-byte aligned so
77    * they work natively with SSE types with no further work.
78    */
79   assert(alignment == 16);
80   (void)alignment;
81   return malloc(size);
82 #elif defined(__FreeBSD__) || defined(__NetBSD__)
83   void *result;
84
85   if (posix_memalign(&result, alignment, size)) {
86     /* non-zero means allocation error
87      * either no allocation or bad alignment value
88      */
89     return NULL;
90   }
91   return result;
92 #else /* This is for Linux. */
93   return memalign(alignment, size);
94 #endif
95 }
96
97 void aligned_free(void *ptr)
98 {
99 #ifdef _WIN32
100   _aligned_free(ptr);
101 #else
102   free(ptr);
103 #endif
104 }
105
106 void MEM_use_guarded_allocator(void)
107 {
108   MEM_allocN_len = MEM_guarded_allocN_len;
109   MEM_freeN = MEM_guarded_freeN;
110   MEM_dupallocN = MEM_guarded_dupallocN;
111   MEM_reallocN_id = MEM_guarded_reallocN_id;
112   MEM_recallocN_id = MEM_guarded_recallocN_id;
113   MEM_callocN = MEM_guarded_callocN;
114   MEM_calloc_arrayN = MEM_guarded_calloc_arrayN;
115   MEM_mallocN = MEM_guarded_mallocN;
116   MEM_malloc_arrayN = MEM_guarded_malloc_arrayN;
117   MEM_mallocN_aligned = MEM_guarded_mallocN_aligned;
118   MEM_mapallocN = MEM_guarded_mapallocN;
119   MEM_printmemlist_pydict = MEM_guarded_printmemlist_pydict;
120   MEM_printmemlist = MEM_guarded_printmemlist;
121   MEM_callbackmemlist = MEM_guarded_callbackmemlist;
122   MEM_printmemlist_stats = MEM_guarded_printmemlist_stats;
123   MEM_set_error_callback = MEM_guarded_set_error_callback;
124   MEM_consistency_check = MEM_guarded_consistency_check;
125   MEM_set_lock_callback = MEM_guarded_set_lock_callback;
126   MEM_set_memory_debug = MEM_guarded_set_memory_debug;
127   MEM_get_memory_in_use = MEM_guarded_get_memory_in_use;
128   MEM_get_mapped_memory_in_use = MEM_guarded_get_mapped_memory_in_use;
129   MEM_get_memory_blocks_in_use = MEM_guarded_get_memory_blocks_in_use;
130   MEM_reset_peak_memory = MEM_guarded_reset_peak_memory;
131   MEM_get_peak_memory = MEM_guarded_get_peak_memory;
132
133 #ifndef NDEBUG
134   MEM_name_ptr = MEM_guarded_name_ptr;
135 #endif
136 }