chiark / gitweb /
Prep v233: Unmask now needed functions in src/basic
[elogind.git] / src / basic / alloc-util.c
1 /***
2   This file is part of systemd.
3
4   Copyright 2010 Lennart Poettering
5
6   systemd is free software; you can redistribute it and/or modify it
7   under the terms of the GNU Lesser General Public License as published by
8   the Free Software Foundation; either version 2.1 of the License, or
9   (at your option) any later version.
10
11   systemd is distributed in the hope that it will be useful, but
12   WITHOUT ANY WARRANTY; without even the implied warranty of
13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14   Lesser General Public License for more details.
15
16   You should have received a copy of the GNU Lesser General Public License
17   along with systemd; If not, see <http://www.gnu.org/licenses/>.
18 ***/
19
20 #include <stdint.h>
21 #include <string.h>
22
23 #include "alloc-util.h"
24 #include "macro.h"
25 #include "util.h"
26
27 void* memdup(const void *p, size_t l) {
28         void *r;
29
30         assert(p);
31
32         r = malloc(l);
33         if (!r)
34                 return NULL;
35
36         memcpy(r, p, l);
37         return r;
38 }
39
40 void* greedy_realloc(void **p, size_t *allocated, size_t need, size_t size) {
41         size_t a, newalloc;
42         void *q;
43
44         assert(p);
45         assert(allocated);
46
47         if (*allocated >= need)
48                 return *p;
49
50         newalloc = MAX(need * 2, 64u / size);
51         a = newalloc * size;
52
53         /* check for overflows */
54         if (a < size * need)
55                 return NULL;
56
57         q = realloc(*p, a);
58         if (!q)
59                 return NULL;
60
61         *p = q;
62         *allocated = newalloc;
63         return q;
64 }
65
66 void* greedy_realloc0(void **p, size_t *allocated, size_t need, size_t size) {
67         size_t prev;
68         uint8_t *q;
69
70         assert(p);
71         assert(allocated);
72
73         prev = *allocated;
74
75         q = greedy_realloc(p, allocated, need, size);
76         if (!q)
77                 return NULL;
78
79         if (*allocated > prev)
80                 memzero(q + prev * size, (*allocated - prev) * size);
81
82         return q;
83 }