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