]>
Commit | Line | Data |
---|---|---|
1 | /* $OpenBSD: xmalloc.c,v 1.21 2006/03/27 01:21:18 deraadt Exp $ */ | |
2 | /* | |
3 | * Author: Tatu Ylonen <ylo@cs.hut.fi> | |
4 | * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland | |
5 | * All rights reserved | |
6 | * Versions of malloc and friends that check their results, and never return | |
7 | * failure (they call fatal if they encounter an error). | |
8 | * | |
9 | * As far as I am concerned, the code I have written for this software | |
10 | * can be used freely for any purpose. Any derived versions of this | |
11 | * software must be clearly marked as such, and if the derived work is | |
12 | * incompatible with the protocol description in the RFC file, it must be | |
13 | * called by a name other than "ssh" or "Secure Shell". | |
14 | */ | |
15 | ||
16 | #include "includes.h" | |
17 | ||
18 | #include "xmalloc.h" | |
19 | #include "log.h" | |
20 | ||
21 | void * | |
22 | xmalloc(size_t size) | |
23 | { | |
24 | void *ptr; | |
25 | ||
26 | if (size == 0) | |
27 | fatal("xmalloc: zero size"); | |
28 | ptr = malloc(size); | |
29 | if (ptr == NULL) | |
30 | fatal("xmalloc: out of memory (allocating %lu bytes)", (u_long) size); | |
31 | return ptr; | |
32 | } | |
33 | ||
34 | void * | |
35 | xcalloc(size_t nmemb, size_t size) | |
36 | { | |
37 | void *ptr; | |
38 | ||
39 | if (size == 0 || nmemb == 0) | |
40 | fatal("xcalloc: zero size"); | |
41 | if (SIZE_T_MAX / nmemb < size) | |
42 | fatal("xcalloc: nmemb * size > SIZE_T_MAX"); | |
43 | ptr = calloc(nmemb, size); | |
44 | if (ptr == NULL) | |
45 | fatal("xcalloc: out of memory (allocating %lu bytes)", | |
46 | (u_long)(size * nmemb)); | |
47 | return ptr; | |
48 | } | |
49 | ||
50 | void * | |
51 | xrealloc(void *ptr, size_t nmemb, size_t size) | |
52 | { | |
53 | void *new_ptr; | |
54 | size_t new_size = nmemb * size; | |
55 | ||
56 | if (new_size == 0) | |
57 | fatal("xrealloc: zero size"); | |
58 | if (SIZE_T_MAX / nmemb < size) | |
59 | fatal("xrealloc: nmemb * size > SIZE_T_MAX"); | |
60 | if (ptr == NULL) | |
61 | new_ptr = malloc(new_size); | |
62 | else | |
63 | new_ptr = realloc(ptr, new_size); | |
64 | if (new_ptr == NULL) | |
65 | fatal("xrealloc: out of memory (new_size %lu bytes)", | |
66 | (u_long) new_size); | |
67 | return new_ptr; | |
68 | } | |
69 | ||
70 | void | |
71 | xfree(void *ptr) | |
72 | { | |
73 | if (ptr == NULL) | |
74 | fatal("xfree: NULL pointer given as argument"); | |
75 | free(ptr); | |
76 | } | |
77 | ||
78 | char * | |
79 | xstrdup(const char *str) | |
80 | { | |
81 | size_t len; | |
82 | char *cp; | |
83 | ||
84 | len = strlen(str) + 1; | |
85 | cp = xmalloc(len); | |
86 | strlcpy(cp, str, len); | |
87 | return cp; | |
88 | } | |
89 | ||
90 | int | |
91 | xasprintf(char **ret, const char *fmt, ...) | |
92 | { | |
93 | va_list ap; | |
94 | int i; | |
95 | ||
96 | va_start(ap, fmt); | |
97 | i = vasprintf(ret, fmt, ap); | |
98 | va_end(ap); | |
99 | ||
100 | if (i < 0 || *ret == NULL) | |
101 | fatal("xasprintf: could not allocate memory"); | |
102 | ||
103 | return (i); | |
104 | } |