libutil -- Import error functions from NetBSD.
[dragonfly.git] / lib / libutil / efun.c
1 /*      $NetBSD: efun.c,v 1.6 2008/04/28 20:23:02 martin Exp $  */
2
3 /*-
4  * Copyright (c) 2006 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by Christos Zoulas.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29  * POSSIBILITY OF SUCH DAMAGE.
30  */
31
32 #include <sys/cdefs.h>
33 #include <err.h>
34 #include <errno.h>
35 #include <string.h>
36 #include <stdlib.h>
37 #include <stdio.h>
38 #include <stdarg.h>
39 #include <libutil.h>
40
41 static void (*efunc)(int, const char *, ...) = err;
42
43 void (*
44 esetfunc(void (*ef)(int, const char *, ...)))(int, const char *, ...)
45 {
46         void (*of)(int, const char *, ...) = efunc;
47         efunc = ef == NULL ? (void (*)(int, const char *, ...))exit : ef;
48         return of;
49 }
50
51 size_t
52 estrlcpy(char *dst, const char *src, size_t len)
53 {
54         size_t rv;
55         if ((rv = strlcpy(dst, src, len)) >= len) {
56                 errno = ENAMETOOLONG;
57                 (*efunc)(1,
58                     "Cannot copy string; %zu chars needed %zu provided",
59                     rv, len);
60         }
61         return rv;
62 }
63
64 size_t
65 estrlcat(char *dst, const char *src, size_t len)
66 {
67         size_t rv;
68         if ((rv = strlcat(dst, src, len)) >= len) {
69                 errno = ENAMETOOLONG;
70                 (*efunc)(1,
71                     "Cannot append to string; %zu chars needed %zu provided",
72                     rv, len);
73         }
74         return rv;
75 }
76
77 char *
78 estrdup(const char *s)
79 {
80         char *d = strdup(s);
81         if (d == NULL)
82                 (*efunc)(1, "Cannot copy string");
83         return d;
84 }
85
86 char *
87 estrndup(const char *s, size_t len)
88 {
89         char *d = strndup(s, len);
90         if (d == NULL)
91                 (*efunc)(1, "Cannot copy string");
92         return d;
93 }
94
95 void *
96 emalloc(size_t n)
97 {
98         void *p = malloc(n);
99         if (p == NULL)
100                 (*efunc)(1, "Cannot allocate %zu bytes", n);
101         return p;
102 }
103
104 void *
105 ecalloc(size_t n, size_t s)
106 {
107         void *p = calloc(n, s);
108         if (p == NULL)
109                 (*efunc)(1, "Cannot allocate %zu bytes", n);
110         return p;
111 }
112
113 void *
114 erealloc(void *p, size_t n)
115 {
116         void *q = realloc(p, n);
117         if (q == NULL)
118                 (*efunc)(1, "Cannot re-allocate %zu bytes", n);
119         return q;
120 }
121
122 int
123 easprintf(char ** __restrict ret, const char * __restrict format, ...)
124 {
125         int rv;
126         va_list ap;
127         va_start(ap, format);
128         if ((rv = vasprintf(ret, format, ap)) == -1)
129                 (*efunc)(1, "Cannot format string");
130         va_end(ap);
131         return rv;
132 }
133