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 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 #define JEMALLOC_MANGLE 51 #include <jemalloc/jemalloc.h> 52 #if JEMALLOC_VERSION_MAJOR >= 2 && JEMALLOC_VERSION_MINOR >= 1 53 #define HAVE_MALLOC_SIZE 1 54 #define zmalloc_size(p) JEMALLOC_P(malloc_usable_size)(p) 55 #else 56 #error "Newer version of jemalloc required" 57 #endif 58 59 #elif defined(__APPLE__) 60 #include <malloc/malloc.h> 61 #define HAVE_MALLOC_SIZE 1 62 #define zmalloc_size(p) malloc_size(p) 63 #endif 64 65 #ifndef ZMALLOC_LIB 66 #define ZMALLOC_LIB "libc" 67 #endif 68 69 void *zmalloc(size_t size); 70 void *zcalloc(size_t size); 71 void *zrealloc(void *ptr, size_t size); 72 void zfree(void *ptr); 73 char *zstrdup(const char *s); 74 size_t zmalloc_used_memory(void); 75 void zmalloc_enable_thread_safeness(void); 76 float zmalloc_get_fragmentation_ratio(void); 77 size_t zmalloc_get_rss(void); 78 size_t zmalloc_allocations_for_size(size_t size); 79 80 #define ZMALLOC_MAX_ALLOC_STAT 256 81 82 #endif /* __ZMALLOC_H */ 83