Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
Newer
Older
100644 155 lines (135 sloc) 4.573 kb
ed9b544 @antirez first commit
authored
1 /* zmalloc - total amount of allocated memory aware version of malloc()
2 *
12d090d @antirez A problem with replication with multiple slaves connectiong to a sing…
authored
3 * Copyright (c) 2009-2010, Salvatore Sanfilippo <antirez at gmail dot com>
ed9b544 @antirez first commit
authored
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
6b47e12 @antirez initial skiplist implementation. Most memory checks removed and zmall…
authored
31 #include <stdio.h>
ed9b544 @antirez first commit
authored
32 #include <stdlib.h>
33 #include <string.h>
a69a0c9 @antirez More threaded I/O VM work + Redis init script
authored
34 #include <pthread.h>
dde65f3 @antirez added config.h for #ifdef business isolation, added fstat64 for Mac OS X
authored
35 #include "config.h"
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
36
d8b5f18 @antirez zmalloc Solaris fixes thanks to Alan Harder
authored
37 #if defined(__sun)
38 #define PREFIX_SIZE sizeof(long long)
39 #else
40 #define PREFIX_SIZE sizeof(size_t)
41 #endif
42
4ad3748 @antirez thread safe zmalloc used memory counter
authored
43 #define increment_used_memory(_n) do { \
44 if (zmalloc_thread_safe) { \
45 pthread_mutex_lock(&used_memory_mutex); \
46 used_memory += _n; \
47 pthread_mutex_unlock(&used_memory_mutex); \
48 } else { \
49 used_memory += _n; \
50 } \
51 } while(0)
52
53 #define decrement_used_memory(_n) do { \
54 if (zmalloc_thread_safe) { \
55 pthread_mutex_lock(&used_memory_mutex); \
56 used_memory -= _n; \
57 pthread_mutex_unlock(&used_memory_mutex); \
58 } else { \
59 used_memory -= _n; \
60 } \
61 } while(0)
62
ed9b544 @antirez first commit
authored
63 static size_t used_memory = 0;
4ad3748 @antirez thread safe zmalloc used memory counter
authored
64 static int zmalloc_thread_safe = 0;
a69a0c9 @antirez More threaded I/O VM work + Redis init script
authored
65 pthread_mutex_t used_memory_mutex = PTHREAD_MUTEX_INITIALIZER;
ed9b544 @antirez first commit
authored
66
6b47e12 @antirez initial skiplist implementation. Most memory checks removed and zmall…
authored
67 static void zmalloc_oom(size_t size) {
412a8bc @antirez Fixed Issue 83:Using TYPE on a zset results in a malformed response f…
authored
68 fprintf(stderr, "zmalloc: Out of memory trying to allocate %zu bytes\n",
6b47e12 @antirez initial skiplist implementation. Most memory checks removed and zmall…
authored
69 size);
70 fflush(stderr);
71 abort();
72 }
73
ed9b544 @antirez first commit
authored
74 void *zmalloc(size_t size) {
d8b5f18 @antirez zmalloc Solaris fixes thanks to Alan Harder
authored
75 void *ptr = malloc(size+PREFIX_SIZE);
ed9b544 @antirez first commit
authored
76
6b47e12 @antirez initial skiplist implementation. Most memory checks removed and zmall…
authored
77 if (!ptr) zmalloc_oom(size);
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
78 #ifdef HAVE_MALLOC_SIZE
4ad3748 @antirez thread safe zmalloc used memory counter
authored
79 increment_used_memory(redis_malloc_size(ptr));
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
80 return ptr;
81 #else
ed9b544 @antirez first commit
authored
82 *((size_t*)ptr) = size;
4ad3748 @antirez thread safe zmalloc used memory counter
authored
83 increment_used_memory(size+PREFIX_SIZE);
d8b5f18 @antirez zmalloc Solaris fixes thanks to Alan Harder
authored
84 return (char*)ptr+PREFIX_SIZE;
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
85 #endif
ed9b544 @antirez first commit
authored
86 }
87
88 void *zrealloc(void *ptr, size_t size) {
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
89 #ifndef HAVE_MALLOC_SIZE
ed9b544 @antirez first commit
authored
90 void *realptr;
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
91 #endif
ed9b544 @antirez first commit
authored
92 size_t oldsize;
93 void *newptr;
94
95 if (ptr == NULL) return zmalloc(size);
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
96 #ifdef HAVE_MALLOC_SIZE
97 oldsize = redis_malloc_size(ptr);
98 newptr = realloc(ptr,size);
6b47e12 @antirez initial skiplist implementation. Most memory checks removed and zmall…
authored
99 if (!newptr) zmalloc_oom(size);
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
100
4ad3748 @antirez thread safe zmalloc used memory counter
authored
101 decrement_used_memory(oldsize);
102 increment_used_memory(redis_malloc_size(newptr));
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
103 return newptr;
104 #else
d8b5f18 @antirez zmalloc Solaris fixes thanks to Alan Harder
authored
105 realptr = (char*)ptr-PREFIX_SIZE;
ed9b544 @antirez first commit
authored
106 oldsize = *((size_t*)realptr);
d8b5f18 @antirez zmalloc Solaris fixes thanks to Alan Harder
authored
107 newptr = realloc(realptr,size+PREFIX_SIZE);
6b47e12 @antirez initial skiplist implementation. Most memory checks removed and zmall…
authored
108 if (!newptr) zmalloc_oom(size);
ed9b544 @antirez first commit
authored
109
110 *((size_t*)newptr) = size;
4ad3748 @antirez thread safe zmalloc used memory counter
authored
111 decrement_used_memory(oldsize);
112 increment_used_memory(size);
d8b5f18 @antirez zmalloc Solaris fixes thanks to Alan Harder
authored
113 return (char*)newptr+PREFIX_SIZE;
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
114 #endif
ed9b544 @antirez first commit
authored
115 }
116
117 void zfree(void *ptr) {
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
118 #ifndef HAVE_MALLOC_SIZE
ed9b544 @antirez first commit
authored
119 void *realptr;
120 size_t oldsize;
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
121 #endif
ed9b544 @antirez first commit
authored
122
123 if (ptr == NULL) return;
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
124 #ifdef HAVE_MALLOC_SIZE
4ad3748 @antirez thread safe zmalloc used memory counter
authored
125 decrement_used_memory(redis_malloc_size(ptr));
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
126 free(ptr);
127 #else
d8b5f18 @antirez zmalloc Solaris fixes thanks to Alan Harder
authored
128 realptr = (char*)ptr-PREFIX_SIZE;
ed9b544 @antirez first commit
authored
129 oldsize = *((size_t*)realptr);
4ad3748 @antirez thread safe zmalloc used memory counter
authored
130 decrement_used_memory(oldsize+PREFIX_SIZE);
ed9b544 @antirez first commit
authored
131 free(realptr);
ec93bba @antirez macosx specific zmalloc.c, uses malloc_size function in order to avoi…
authored
132 #endif
ed9b544 @antirez first commit
authored
133 }
134
135 char *zstrdup(const char *s) {
136 size_t l = strlen(s)+1;
137 char *p = zmalloc(l);
138
139 memcpy(p,s,l);
140 return p;
141 }
142
143 size_t zmalloc_used_memory(void) {
4ad3748 @antirez thread safe zmalloc used memory counter
authored
144 size_t um;
145
146 if (zmalloc_thread_safe) pthread_mutex_lock(&used_memory_mutex);
147 um = used_memory;
148 if (zmalloc_thread_safe) pthread_mutex_unlock(&used_memory_mutex);
149 return um;
150 }
151
152 void zmalloc_enable_thread_safeness(void) {
153 zmalloc_thread_safe = 1;
ed9b544 @antirez first commit
authored
154 }
Something went wrong with that request. Please try again.