1 /* zmalloc - total amount of allocated memory aware version of malloc() 2 * 3 * Copyright (c) 2009-2010, Salvatore Sanfilippo <antirez at gmail dot com> 4 * All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions are met: 8 * 9 * * Redistributions of source code must retain the above copyright notice, 10 * this list of conditions and the following disclaimer. 11 * * Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * * Neither the name of Redis nor the names of its contributors may be used 15 * to endorse or promote products derived from this software without 16 * specific prior written permission. 17 * 18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 19 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 21 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE 22 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 23 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 24 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 25 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 26 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 27 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 28 * POSSIBILITY OF SUCH DAMAGE. 29 */ 30 31 #ifndef __ZMALLOC_H 32 #define __ZMALLOC_H 33 34 /* Double expansion needed for stringification of macro values. */ 35 #define __xstr(s) __str(s) 36 #define __str(s) #s 37 38 #if defined(USE_TCMALLOC) 39 #define ZMALLOC_LIB ("tcmalloc-" __xstr(TC_VERSION_MAJOR) "." __xstr(TC_VERSION_MINOR)) 40 #include <google/tcmalloc.h> 41 #if (TC_VERSION_MAJOR == 1 && TC_VERSION_MINOR >= 6) || (TC_VERSION_MAJOR > 1) 42 #define HAVE_MALLOC_SIZE 1 43 #define zmalloc_size(p) tc_malloc_size(p) 44 #else 45 #error "Newer version of tcmalloc required" 46 #endif 47 48 #elif defined(USE_JEMALLOC) 49 #define ZMALLOC_LIB ("jemalloc-" __xstr(JEMALLOC_VERSION_MAJOR) "." __xstr(JEMALLOC_VERSION_MINOR) "." __xstr(JEMALLOC_VERSION_BUGFIX)) 50 #include <jemalloc/jemalloc.h> 51 #if (JEMALLOC_VERSION_MAJOR == 2 && JEMALLOC_VERSION_MINOR >= 1) || (JEMALLOC_VERSION_MAJOR > 2) 52 #define HAVE_MALLOC_SIZE 1 53 #define zmalloc_size(p) je_malloc_usable_size(p) 54 #else 55 #error "Newer version of jemalloc required" 56 #endif 57 58 #elif defined(__APPLE__) 59 #include <malloc/malloc.h> 60 #define HAVE_MALLOC_SIZE 1 61 #define zmalloc_size(p) malloc_size(p) 62 #endif 63 64 #ifndef ZMALLOC_LIB 65 #define ZMALLOC_LIB "libc" 66 #endif 67 68 void *zmalloc(size_t size); 69 void *zcalloc(size_t size); 70 void *zrealloc(void *ptr, size_t size); 71 void zfree(void *ptr); 72 char *zstrdup(const char *s); 73 size_t zmalloc_used_memory(void); 74 void zmalloc_enable_thread_safeness(void); 75 void zmalloc_set_oom_handler(void (*oom_handler)(size_t)); 76 float zmalloc_get_fragmentation_ratio(size_t rss); 77 size_t zmalloc_get_rss(void); 78 size_t zmalloc_get_private_dirty(void); 79 size_t zmalloc_get_smap_bytes_by_field(char *field); 80 size_t zmalloc_get_memory_size(void); 81 void zlibc_free(void *ptr); 82 83 #ifndef HAVE_MALLOC_SIZE 84 size_t zmalloc_size(void *ptr); 85 #endif 86 87 #endif /* __ZMALLOC_H */ 88