]> git.saurik.com Git - redis.git/blame_incremental - zmalloc.c
Fixed compilation on Linux
[redis.git] / zmalloc.c
... / ...
CommitLineData
1/* zmalloc - total amount of allocated memory aware version of malloc()
2 *
3 * Copyright (c) 2006-2009, 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#include <stdio.h>
32#include <stdlib.h>
33#include <string.h>
34#include "config.h"
35
36#if defined(__sun)
37#define PREFIX_SIZE sizeof(long long)
38#else
39#define PREFIX_SIZE sizeof(size_t)
40#endif
41
42static size_t used_memory = 0;
43
44static void zmalloc_oom(size_t size) {
45 fprintf(stderr, "zmalloc: Out of memory trying to allocate %lu bytes\n",
46 size);
47 fflush(stderr);
48 abort();
49}
50
51void *zmalloc(size_t size) {
52 void *ptr = malloc(size+PREFIX_SIZE);
53
54 if (!ptr) zmalloc_oom(size);
55#ifdef HAVE_MALLOC_SIZE
56 used_memory += redis_malloc_size(ptr);
57 return ptr;
58#else
59 *((size_t*)ptr) = size;
60 used_memory += size+PREFIX_SIZE;
61 return (char*)ptr+PREFIX_SIZE;
62#endif
63}
64
65void *zrealloc(void *ptr, size_t size) {
66#ifndef HAVE_MALLOC_SIZE
67 void *realptr;
68#endif
69 size_t oldsize;
70 void *newptr;
71
72 if (ptr == NULL) return zmalloc(size);
73#ifdef HAVE_MALLOC_SIZE
74 oldsize = redis_malloc_size(ptr);
75 newptr = realloc(ptr,size);
76 if (!newptr) zmalloc_oom(size);
77
78 used_memory -= oldsize;
79 used_memory += redis_malloc_size(newptr);
80 return newptr;
81#else
82 realptr = (char*)ptr-PREFIX_SIZE;
83 oldsize = *((size_t*)realptr);
84 newptr = realloc(realptr,size+PREFIX_SIZE);
85 if (!newptr) zmalloc_oom(size);
86
87 *((size_t*)newptr) = size;
88 used_memory -= oldsize;
89 used_memory += size;
90 return (char*)newptr+PREFIX_SIZE;
91#endif
92}
93
94void zfree(void *ptr) {
95#ifndef HAVE_MALLOC_SIZE
96 void *realptr;
97 size_t oldsize;
98#endif
99
100 if (ptr == NULL) return;
101#ifdef HAVE_MALLOC_SIZE
102 used_memory -= redis_malloc_size(ptr);
103 free(ptr);
104#else
105 realptr = (char*)ptr-PREFIX_SIZE;
106 oldsize = *((size_t*)realptr);
107 used_memory -= oldsize+PREFIX_SIZE;
108 free(realptr);
109#endif
110}
111
112char *zstrdup(const char *s) {
113 size_t l = strlen(s)+1;
114 char *p = zmalloc(l);
115
116 memcpy(p,s,l);
117 return p;
118}
119
120size_t zmalloc_used_memory(void) {
121 return used_memory;
122}