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 #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.6 2008/04/28 20:23:02 martin 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)
 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)
 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)
 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 }