Home | History | Annotate | Line # | Download | only in libutil
efun.c revision 1.7 
      1 /*	$NetBSD: efun.c,v 1.7 2012/12/30 17:36:00 dholland 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 #if HAVE_NBTOOL_CONFIG_H
     33 #include "nbtool_config.h"
     34 #endif
     35 
     36 #include <sys/cdefs.h>
     37 #ifdef __RCSID
     38 __RCSID("$NetBSD: efun.c,v 1.7 2012/12/30 17:36:00 dholland Exp $");
     39 #endif
     40 
     41 #include <err.h>
     42 #include <errno.h>
     43 #include <string.h>
     44 #include <stdlib.h>
     45 #include <stdio.h>
     46 #include <stdarg.h>
     47 #include <util.h>
     48 
     49 static void (*efunc)(int, const char *, ...) = err;
     50 
     51 void (*
     52 esetfunc(void (*ef)(int, const char *, ...)))(int, const char *, ...)
     53 {
     54 	void (*of)(int, const char *, ...) = efunc;
     55 	efunc = ef == NULL ? (void (*)(int, const char *, ...))exit : ef;
     56 	return of;
     57 }
     58 
     59 size_t
     60 estrlcpy(char *dst, const char *src, size_t len)
     61 {
     62 	size_t rv;
     63 	if ((rv = strlcpy(dst, src, len)) >= len) {
     64 		errno = ENAMETOOLONG;
     65 		(*efunc)(1,
     66 		    "Cannot copy string; %zu chars needed %zu provided",
     67 		    rv, len);
     68 	}
     69 	return rv;
     70 }
     71 
     72 size_t
     73 estrlcat(char *dst, const char *src, size_t len)
     74 {
     75 	size_t rv;
     76 	if ((rv = strlcat(dst, src, len)) >= len) {
     77 		errno = ENAMETOOLONG;
     78 		(*efunc)(1,
     79 		    "Cannot append to string; %zu chars needed %zu provided",
     80 		    rv, len);
     81 	}
     82 	return rv;
     83 }
     84 
     85 char *
     86 estrdup(const char *s)
     87 {
     88 	char *d = strdup(s);
     89 	if (d == NULL)
     90 		(*efunc)(1, "Cannot copy string");
     91 	return d;
     92 }
     93 
     94 char *
     95 estrndup(const char *s, size_t len)
     96 {
     97 	char *d = strndup(s, len);
     98 	if (d == NULL)
     99 		(*efunc)(1, "Cannot copy string");
    100 	return d;
    101 }
    102 
    103 void *
    104 emalloc(size_t n)
    105 {
    106 	void *p = malloc(n);
    107 	if (p == NULL && n != 0)
    108 		(*efunc)(1, "Cannot allocate %zu bytes", n);
    109 	return p;
    110 }
    111 
    112 void *
    113 ecalloc(size_t n, size_t s)
    114 {
    115 	void *p = calloc(n, s);
    116 	if (p == NULL && n != 0 && s != 0)
    117 		(*efunc)(1, "Cannot allocate %zu bytes", n);
    118 	return p;
    119 }
    120 
    121 void *
    122 erealloc(void *p, size_t n)
    123 {
    124 	void *q = realloc(p, n);
    125 	if (q == NULL && n != 0)
    126 		(*efunc)(1, "Cannot re-allocate %zu bytes", n);
    127 	return q;
    128 }
    129 
    130 FILE *
    131 efopen(const char *p, const char *m)
    132 {
    133 	FILE *fp = fopen(p, m);
    134 	if (fp == NULL)
    135 		(*efunc)(1, "Cannot open `%s'", p);
    136 	return fp;
    137 }
    138 
    139 int
    140 easprintf(char ** __restrict ret, const char * __restrict format, ...)
    141 {
    142 	int rv;
    143 	va_list ap;
    144 	va_start(ap, format);
    145 	if ((rv = vasprintf(ret, format, ap)) == -1)
    146 		(*efunc)(1, "Cannot format string");
    147 	va_end(ap);
    148 	return rv;
    149 }
    150 
    151 int
    152 evasprintf(char ** __restrict ret, const char * __restrict format, va_list ap)
    153 {
    154 	int rv;
    155 	if ((rv = vasprintf(ret, format, ap)) == -1)
    156 		(*efunc)(1, "Cannot format string");
    157 	return rv;
    158 }
    159