mm_malloc.h (208954) | mm_malloc.h (218893) |
---|---|
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: --- 10 unchanged lines hidden (view full) --- 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: --- 10 unchanged lines hidden (view full) --- 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 */ |