1/*===---- mm_malloc.h - Allocating and Freeing Aligned Memory Blocks -------=== 2 * 3 * Permission is hereby granted, free of charge, to any person obtaining a copy 4 * of this software and associated documentation files (the "Software"), to deal 5 * in the Software without restriction, including without limitation the rights 6 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 7 * copies of the Software, and to permit persons to whom the Software is 8 * furnished to do so, subject to the following conditions: 9 * 10 * The above copyright notice and this permission notice shall be included in 11 * all copies or substantial portions of the Software. 12 * 13 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 14 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 15 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 16 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 17 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 18 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 19 * THE SOFTWARE. 20 * 21 *===-----------------------------------------------------------------------=== 22 */ 23 24#ifndef __MM_MALLOC_H 25#define __MM_MALLOC_H 26
| 1/*===---- mm_malloc.h - Allocating and Freeing Aligned Memory Blocks -------=== 2 * 3 * Permission is hereby granted, free of charge, to any person obtaining a copy 4 * of this software and associated documentation files (the "Software"), to deal 5 * in the Software without restriction, including without limitation the rights 6 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 7 * copies of the Software, and to permit persons to whom the Software is 8 * furnished to do so, subject to the following conditions: 9 * 10 * The above copyright notice and this permission notice shall be included in 11 * all copies or substantial portions of the Software. 12 * 13 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 14 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 15 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 16 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 17 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 18 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 19 * THE SOFTWARE. 20 * 21 *===-----------------------------------------------------------------------=== 22 */ 23 24#ifndef __MM_MALLOC_H 25#define __MM_MALLOC_H 26
|
27#include <errno.h>
| |
28#include <stdlib.h> 29
| 27#include <stdlib.h> 28
|
30static __inline__ void *__attribute__((__always_inline__, __nodebug__))
| 29#ifdef _WIN32 30#include <malloc.h> 31#else 32#ifndef __cplusplus 33extern int posix_memalign(void **memptr, size_t alignment, size_t size); 34#else 35// Some systems (e.g. those with GNU libc) declare posix_memalign with an 36// exception specifier. Via an "egregious workaround" in 37// Sema::CheckEquivalentExceptionSpec, Clang accepts the following as a valid 38// redeclaration of glibc's declaration. 39extern "C" int posix_memalign(void **memptr, size_t alignment, size_t size); 40#endif 41#endif 42 43static __inline__ void *__attribute__((__always_inline__, __nodebug__, 44 __malloc__))
|
31_mm_malloc(size_t size, size_t align) 32{
| 45_mm_malloc(size_t size, size_t align) 46{
|
33 if (align & (align - 1)) { 34 errno = EINVAL; 35 return 0;
| 47 if (align == 1) { 48 return malloc(size);
|
36 } 37
| 49 } 50
|
38 if (!size) 39 return 0;
| 51 if (!(align & (align - 1)) && align < sizeof(void *)) 52 align = sizeof(void *);
|
40
| 53
|
41 if (align < 2 * sizeof(void *)) 42 align = 2 * sizeof(void *); 43 44 void *mallocedMemory = malloc(size + align); 45 if (!mallocedMemory)
| 54 void *mallocedMemory; 55#ifdef _WIN32 56 mallocedMemory = _aligned_malloc(size, align); 57#else 58 if (posix_memalign(&mallocedMemory, align, size))
|
46 return 0;
| 59 return 0;
|
| 60#endif
|
47
| 61
|
48 void *alignedMemory = 49 (void *)(((size_t)mallocedMemory + align) & ~((size_t)align - 1)); 50 ((void **)alignedMemory)[-1] = mallocedMemory; 51 52 return alignedMemory;
| 62 return mallocedMemory;
|
53} 54 55static __inline__ void __attribute__((__always_inline__, __nodebug__)) 56_mm_free(void *p) 57{
| 63} 64 65static __inline__ void __attribute__((__always_inline__, __nodebug__)) 66_mm_free(void *p) 67{
|
58 if (p) 59 free(((void **)p)[-1]);
| 68 free(p);
|
60} 61 62#endif /* __MM_MALLOC_H */
| 69} 70 71#endif /* __MM_MALLOC_H */
|