mem.h 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226
  1. /*
  2. * copyright (c) 2006 Michael Niedermayer <michaelni@gmx.at>
  3. *
  4. * This file is part of FFmpeg.
  5. *
  6. * FFmpeg is free software; you can redistribute it and/or
  7. * modify it under the terms of the GNU Lesser General Public
  8. * License as published by the Free Software Foundation; either
  9. * version 2.1 of the License, or (at your option) any later version.
  10. *
  11. * FFmpeg is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14. * Lesser General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU Lesser General Public
  17. * License along with FFmpeg; if not, write to the Free Software
  18. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19. */
  20. /**
  21. * @file
  22. * memory handling functions
  23. */
  24. #ifndef AVUTIL_MEM_H
  25. #define AVUTIL_MEM_H
  26. #include <limits.h>
  27. #include "attributes.h"
  28. #include "error.h"
  29. #include "avutil.h"
  30. /**
  31. * @addtogroup lavu_mem
  32. * @{
  33. */
  34. #if defined(__INTEL_COMPILER) && __INTEL_COMPILER < 1110 || defined(__SUNPRO_C)
  35. #define DECLARE_ALIGNED(n,t,v) t __attribute__ ((aligned (n))) v
  36. #define DECLARE_ASM_CONST(n,t,v) const t __attribute__ ((aligned (n))) v
  37. #elif defined(__TI_COMPILER_VERSION__)
  38. #define DECLARE_ALIGNED(n,t,v) \
  39. AV_PRAGMA(DATA_ALIGN(v,n)) \
  40. t __attribute__((aligned(n))) v
  41. #define DECLARE_ASM_CONST(n,t,v) \
  42. AV_PRAGMA(DATA_ALIGN(v,n)) \
  43. static const t __attribute__((aligned(n))) v
  44. #elif defined(__GNUC__)
  45. #define DECLARE_ALIGNED(n,t,v) t __attribute__ ((aligned (n))) v
  46. #define DECLARE_ASM_CONST(n,t,v) static const t av_used __attribute__ ((aligned (n))) v
  47. #elif defined(_MSC_VER)
  48. #define DECLARE_ALIGNED(n,t,v) __declspec(align(n)) t v
  49. #define DECLARE_ASM_CONST(n,t,v) __declspec(align(n)) static const t v
  50. #else
  51. #define DECLARE_ALIGNED(n,t,v) t v
  52. #define DECLARE_ASM_CONST(n,t,v) static const t v
  53. #endif
  54. #if AV_GCC_VERSION_AT_LEAST(3,1)
  55. #define av_malloc_attrib __attribute__((__malloc__))
  56. #else
  57. #define av_malloc_attrib
  58. #endif
  59. #if AV_GCC_VERSION_AT_LEAST(4,3)
  60. #define av_alloc_size(...) __attribute__((alloc_size(__VA_ARGS__)))
  61. #else
  62. #define av_alloc_size(...)
  63. #endif
  64. /**
  65. * Allocate a block of size bytes with alignment suitable for all
  66. * memory accesses (including vectors if available on the CPU).
  67. * @param size Size in bytes for the memory block to be allocated.
  68. * @return Pointer to the allocated block, NULL if the block cannot
  69. * be allocated.
  70. * @see av_mallocz()
  71. */
  72. void *av_malloc(size_t size) av_malloc_attrib av_alloc_size(1);
  73. /**
  74. * Helper function to allocate a block of size * nmemb bytes with
  75. * using av_malloc()
  76. * @param nmemb Number of elements
  77. * @param size Size of the single element
  78. * @return Pointer to the allocated block, NULL if the block cannot
  79. * be allocated.
  80. * @see av_malloc()
  81. */
  82. av_alloc_size(1,2) static inline void *av_malloc_array(size_t nmemb, size_t size)
  83. {
  84. if (size <= 0 || nmemb >= INT_MAX / size) {
  85. return NULL;
  86. }
  87. return av_malloc(nmemb * size);
  88. }
  89. /**
  90. * Allocate or reallocate a block of memory.
  91. * If ptr is NULL and size > 0, allocate a new block. If
  92. * size is zero, free the memory block pointed to by ptr.
  93. * @param ptr Pointer to a memory block already allocated with
  94. * av_malloc(z)() or av_realloc() or NULL.
  95. * @param size Size in bytes for the memory block to be allocated or
  96. * reallocated.
  97. * @return Pointer to a newly reallocated block or NULL if the block
  98. * cannot be reallocated or the function is used to free the memory block.
  99. * @see av_fast_realloc()
  100. */
  101. void *av_realloc(void *ptr, size_t size) av_alloc_size(2);
  102. /**
  103. * Allocate or reallocate a block of memory.
  104. * This function does the same thing as av_realloc, except:
  105. * - It takes two arguments and checks the result of the multiplication for
  106. * integer overflow.
  107. * - It frees the input block in case of failure, thus avoiding the memory
  108. * leak with the classic "buf = realloc(buf); if (!buf) return -1;".
  109. */
  110. void *av_realloc_f(void *ptr, size_t nelem, size_t elsize);
  111. /**
  112. * Free a memory block which has been allocated with av_malloc(z)() or
  113. * av_realloc().
  114. * @param ptr Pointer to the memory block which should be freed.
  115. * @note ptr = NULL is explicitly allowed.
  116. * @note It is recommended that you use av_freep() instead.
  117. * @see av_freep()
  118. */
  119. void av_free(void *ptr);
  120. /**
  121. * Allocate a block of size bytes with alignment suitable for all
  122. * memory accesses (including vectors if available on the CPU) and
  123. * zero all the bytes of the block.
  124. * @param size Size in bytes for the memory block to be allocated.
  125. * @return Pointer to the allocated block, NULL if it cannot be allocated.
  126. * @see av_malloc()
  127. */
  128. void *av_mallocz(size_t size) av_malloc_attrib av_alloc_size(1);
  129. /**
  130. * Allocate a block of nmemb * size bytes with alignment suitable for all
  131. * memory accesses (including vectors if available on the CPU) and
  132. * zero all the bytes of the block.
  133. * The allocation will fail if nmemb * size is greater than or equal
  134. * to INT_MAX.
  135. * @param nmemb
  136. * @param size
  137. * @return Pointer to the allocated block, NULL if it cannot be allocated.
  138. */
  139. void *av_calloc(size_t nmemb, size_t size) av_malloc_attrib;
  140. /**
  141. * Helper function to allocate a block of size * nmemb bytes with
  142. * using av_mallocz()
  143. * @param nmemb Number of elements
  144. * @param size Size of the single element
  145. * @return Pointer to the allocated block, NULL if the block cannot
  146. * be allocated.
  147. * @see av_mallocz()
  148. * @see av_malloc_array()
  149. */
  150. av_alloc_size(1,2) static inline void *av_mallocz_array(size_t nmemb, size_t size)
  151. {
  152. if (size <= 0 || nmemb >= INT_MAX / size) {
  153. return NULL;
  154. }
  155. return av_mallocz(nmemb * size);
  156. }
  157. /**
  158. * Duplicate the string s.
  159. * @param s string to be duplicated
  160. * @return Pointer to a newly allocated string containing a
  161. * copy of s or NULL if the string cannot be allocated.
  162. */
  163. char *av_strdup(const char *s) av_malloc_attrib;
  164. /**
  165. * Free a memory block which has been allocated with av_malloc(z)() or
  166. * av_realloc() and set the pointer pointing to it to NULL.
  167. * @param ptr Pointer to the pointer to the memory block which should
  168. * be freed.
  169. * @see av_free()
  170. */
  171. void av_freep(void *ptr);
  172. /**
  173. * Add an element to a dynamic array.
  174. *
  175. * @param tab_ptr Pointer to the array.
  176. * @param nb_ptr Pointer to the number of elements in the array.
  177. * @param elem Element to be added.
  178. */
  179. void av_dynarray_add(void *tab_ptr, int *nb_ptr, void *elem);
  180. /**
  181. * Multiply two size_t values checking for overflow.
  182. * @return 0 if success, AVERROR(EINVAL) if overflow.
  183. */
  184. static inline int av_size_mult(size_t a, size_t b, size_t *r)
  185. {
  186. size_t t = a * b;
  187. /* Hack inspired from glibc: only try the division if nelem and elsize
  188. * are both greater than sqrt(SIZE_MAX). */
  189. if ((a | b) >= ((size_t)1 << (sizeof(size_t) * 4)) && a && t / a != b) {
  190. return AVERROR(EINVAL);
  191. }
  192. *r = t;
  193. return 0;
  194. }
  195. /**
  196. * Set the maximum size that may me allocated in one block.
  197. */
  198. void av_max_alloc(size_t max);
  199. /**
  200. * @}
  201. */
  202. #endif /* AVUTIL_MEM_H */