almalloc.c 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. #include "config.h"
  2. #include "almalloc.h"
  3. #include <stdlib.h>
  4. #include <string.h>
  5. #ifdef HAVE_MALLOC_H
  6. #include <malloc.h>
  7. #endif
  8. #ifdef HAVE_WINDOWS_H
  9. #include <windows.h>
  10. #else
  11. #include <unistd.h>
  12. #endif
  13. #ifdef __GNUC__
  14. #define LIKELY(x) __builtin_expect(!!(x), !0)
  15. #define UNLIKELY(x) __builtin_expect(!!(x), 0)
  16. #else
  17. #define LIKELY(x) (!!(x))
  18. #define UNLIKELY(x) (!!(x))
  19. #endif
  20. void *al_malloc(size_t alignment, size_t size)
  21. {
  22. #if defined(HAVE_ALIGNED_ALLOC)
  23. size = (size+(alignment-1))&~(alignment-1);
  24. return aligned_alloc(alignment, size);
  25. #elif defined(HAVE_POSIX_MEMALIGN)
  26. void *ret;
  27. if(posix_memalign(&ret, alignment, size) == 0)
  28. return ret;
  29. return NULL;
  30. #elif defined(HAVE__ALIGNED_MALLOC)
  31. return _aligned_malloc(size, alignment);
  32. #else
  33. char *ret = malloc(size+alignment);
  34. if(ret != NULL)
  35. {
  36. *(ret++) = 0x00;
  37. while(((ptrdiff_t)ret&(alignment-1)) != 0)
  38. *(ret++) = 0x55;
  39. }
  40. return ret;
  41. #endif
  42. }
  43. void *al_calloc(size_t alignment, size_t size)
  44. {
  45. void *ret = al_malloc(alignment, size);
  46. if(ret) memset(ret, 0, size);
  47. return ret;
  48. }
  49. void al_free(void *ptr)
  50. {
  51. #if defined(HAVE_ALIGNED_ALLOC) || defined(HAVE_POSIX_MEMALIGN)
  52. free(ptr);
  53. #elif defined(HAVE__ALIGNED_MALLOC)
  54. _aligned_free(ptr);
  55. #else
  56. if(ptr != NULL)
  57. {
  58. char *finder = ptr;
  59. do {
  60. --finder;
  61. } while(*finder == 0x55);
  62. free(finder);
  63. }
  64. #endif
  65. }
  66. size_t al_get_page_size(void)
  67. {
  68. static size_t psize = 0;
  69. if(UNLIKELY(!psize))
  70. {
  71. #ifdef HAVE_SYSCONF
  72. #if defined(_SC_PAGESIZE)
  73. if(!psize) psize = sysconf(_SC_PAGESIZE);
  74. #elif defined(_SC_PAGE_SIZE)
  75. if(!psize) psize = sysconf(_SC_PAGE_SIZE);
  76. #endif
  77. #endif /* HAVE_SYSCONF */
  78. #ifdef _WIN32
  79. if(!psize)
  80. {
  81. SYSTEM_INFO sysinfo;
  82. memset(&sysinfo, 0, sizeof(sysinfo));
  83. GetSystemInfo(&sysinfo);
  84. psize = sysinfo.dwPageSize;
  85. }
  86. #endif
  87. if(!psize) psize = DEF_ALIGN;
  88. }
  89. return psize;
  90. }
  91. int al_is_sane_alignment_allocator(void)
  92. {
  93. #if defined(HAVE_ALIGNED_ALLOC) || defined(HAVE_POSIX_MEMALIGN) || defined(HAVE__ALIGNED_MALLOC)
  94. return 1;
  95. #else
  96. return 0;
  97. #endif
  98. }