mem.h 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166
  1. /*
  2. * Copyright (c) 2001-2004 Swedish Institute of Computer Science.
  3. * All rights reserved.
  4. *
  5. * Redistribution and use in source and binary forms, with or without modification,
  6. * are permitted provided that the following conditions are met:
  7. *
  8. * 1. Redistributions of source code must retain the above copyright notice,
  9. * this list of conditions and the following disclaimer.
  10. * 2. Redistributions in binary form must reproduce the above copyright notice,
  11. * this list of conditions and the following disclaimer in the documentation
  12. * and/or other materials provided with the distribution.
  13. * 3. The name of the author may not be used to endorse or promote products
  14. * derived from this software without specific prior written permission.
  15. *
  16. * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
  17. * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  18. * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT
  19. * SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  20. * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
  21. * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  22. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  23. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
  24. * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
  25. * OF SUCH DAMAGE.
  26. *
  27. * This file is part of the lwIP TCP/IP stack.
  28. *
  29. * Author: Adam Dunkels <adam@sics.se>
  30. *
  31. */
  32. #ifndef __LWIP_MEM_H__
  33. #define __LWIP_MEM_H__
  34. #include "lwip/opt.h"
  35. //#include "mem_manager.h"
  36. #ifdef __cplusplus
  37. extern "C" {
  38. #endif
  39. #if MEM_LIBC_MALLOC
  40. #include <stddef.h> /* for size_t */
  41. typedef size_t mem_size_t;
  42. /* aliases for C library malloc() */
  43. #define mem_init()
  44. /* in case C library malloc() needs extra protection,
  45. * allow these defines to be overridden.
  46. */
  47. #ifndef MEMLEAK_DEBUG
  48. #ifndef mem_free
  49. #define mem_free vPortFree
  50. #endif
  51. #ifndef mem_malloc
  52. #define mem_malloc pvPortMalloc
  53. #endif
  54. #ifndef mem_calloc
  55. #define mem_calloc pvPortCalloc
  56. #endif
  57. #ifndef mem_realloc
  58. #define mem_realloc pvPortRealloc
  59. #endif
  60. #ifndef mem_zalloc
  61. #define mem_zalloc pvPortZalloc
  62. #endif
  63. #else
  64. #ifndef mem_free
  65. #define mem_free(s) \
  66. do{\
  67. const char *file = mem_debug_file;\
  68. vPortFree(s, file, __LINE__);\
  69. }while(0)
  70. #endif
  71. #ifndef mem_malloc
  72. #define mem_malloc(s) ({const char *file = mem_debug_file; pvPortMalloc(s, file, __LINE__);})
  73. #endif
  74. #ifndef mem_calloc
  75. #define mem_calloc(l, s) ({const char *file = mem_debug_file; pvPortCalloc(l, s, file, __LINE__);})
  76. #endif
  77. #ifndef mem_realloc
  78. #define mem_realloc(p, s) ({const char *file = mem_debug_file; pvPortRealloc(p, s, file, __LINE__);})
  79. #endif
  80. #ifndef mem_zalloc
  81. #define mem_zalloc(s) ({const char *file = mem_debug_file; pvPortZalloc(s, file, __LINE__);})
  82. #endif
  83. #endif
  84. #ifndef os_malloc
  85. #define os_malloc(s) mem_malloc((s))
  86. #endif
  87. #ifndef os_realloc
  88. #define os_realloc(p, s) mem_realloc((p), (s))
  89. #endif
  90. #ifndef os_zalloc
  91. #define os_zalloc(s) mem_zalloc((s))
  92. #endif
  93. #ifndef os_free
  94. #define os_free(p) mem_free((p))
  95. #endif
  96. /* Since there is no C library allocation function to shrink memory without
  97. moving it, define this to nothing. */
  98. #ifndef mem_trim
  99. #define mem_trim(mem, size) (mem)
  100. #endif
  101. #else /* MEM_LIBC_MALLOC */
  102. /* MEM_SIZE would have to be aligned, but using 64000 here instead of
  103. * 65535 leaves some room for alignment...
  104. */
  105. #if MEM_SIZE > 64000l
  106. typedef u32_t mem_size_t;
  107. #define MEM_SIZE_F U32_F
  108. #else
  109. typedef u16_t mem_size_t;
  110. #define MEM_SIZE_F U16_F
  111. #endif /* MEM_SIZE > 64000 */
  112. #if MEM_USE_POOLS
  113. /** mem_init is not used when using pools instead of a heap */
  114. #define mem_init()
  115. /** mem_trim is not used when using pools instead of a heap:
  116. we can't free part of a pool element and don't want to copy the rest */
  117. #define mem_trim(mem, size) (mem)
  118. #else /* MEM_USE_POOLS */
  119. /* lwIP alternative malloc */
  120. void mem_init(void)ICACHE_FLASH_ATTR;
  121. void *mem_trim(void *mem, mem_size_t size)ICACHE_FLASH_ATTR;
  122. #endif /* MEM_USE_POOLS */
  123. void *mem_malloc(mem_size_t size)ICACHE_FLASH_ATTR;
  124. void *mem_calloc(mem_size_t count, mem_size_t size)ICACHE_FLASH_ATTR;
  125. void mem_free(void *mem)ICACHE_FLASH_ATTR;
  126. #endif /* MEM_LIBC_MALLOC */
  127. /** Calculate memory size for an aligned buffer - returns the next highest
  128. * multiple of MEM_ALIGNMENT (e.g. LWIP_MEM_ALIGN_SIZE(3) and
  129. * LWIP_MEM_ALIGN_SIZE(4) will both yield 4 for MEM_ALIGNMENT == 4).
  130. */
  131. #ifndef LWIP_MEM_ALIGN_SIZE
  132. #define LWIP_MEM_ALIGN_SIZE(size) (((size) + MEM_ALIGNMENT - 1) & ~(MEM_ALIGNMENT-1))
  133. #endif
  134. /** Calculate safe memory size for an aligned buffer when using an unaligned
  135. * type as storage. This includes a safety-margin on (MEM_ALIGNMENT - 1) at the
  136. * start (e.g. if buffer is u8_t[] and actual data will be u32_t*)
  137. */
  138. #ifndef LWIP_MEM_ALIGN_BUFFER
  139. #define LWIP_MEM_ALIGN_BUFFER(size) (((size) + MEM_ALIGNMENT - 1))
  140. #endif
  141. /** Align a memory pointer to the alignment defined by MEM_ALIGNMENT
  142. * so that ADDR % MEM_ALIGNMENT == 0
  143. */
  144. #ifndef LWIP_MEM_ALIGN
  145. #define LWIP_MEM_ALIGN(addr) ((void *)(((mem_ptr_t)(addr) + MEM_ALIGNMENT - 1) & ~(mem_ptr_t)(MEM_ALIGNMENT-1)))
  146. #endif
  147. #ifdef __cplusplus
  148. }
  149. #endif
  150. #endif /* __LWIP_MEM_H__ */