lib/libutil/efun.c

   1 /*      $NetBSD: efun.c,v 1.10 2015/07/26 02:20:30 kamil 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.10 2015/07/26 02:20:30 kamil Exp $");
  39 #endif
  40
  41 #include <err.h>
  42 #include <errno.h>
  43 #include <inttypes.h>
  44 #include <string.h>
  45 #include <stdlib.h>
  46 #include <stdio.h>
  47 #include <stdarg.h>
  48 #include <util.h>
  49
  50 static void (*efunc)(int, const char *, ...) = err;
  51
  52 void (*
  53 esetfunc(void (*ef)(int, const char *, ...)))(int, const char *, ...)
  54 {
  55         void (*of)(int, const char *, ...) = efunc;
  56         efunc = ef == NULL ? (void (*)(int, const char *, ...))exit : ef;
  57         return of;
  58 }
  59
  60 size_t
  61 estrlcpy(char *dst, const char *src, size_t len)
  62 {
  63         size_t rv;
  64         if ((rv = strlcpy(dst, src, len)) >= len) {
  65                 errno = ENAMETOOLONG;
  66                 (*efunc)(1,
  67                     "Cannot copy string; %zu chars needed %zu provided",
  68                     rv, len);
  69         }
  70         return rv;
  71 }
  72
  73 size_t
  74 estrlcat(char *dst, const char *src, size_t len)
  75 {
  76         size_t rv;
  77         if ((rv = strlcat(dst, src, len)) >= len) {
  78                 errno = ENAMETOOLONG;
  79                 (*efunc)(1,
  80                     "Cannot append to string; %zu chars needed %zu provided",
  81                     rv, len);
  82         }
  83         return rv;
  84 }
  85
  86 char *
  87 estrdup(const char *s)
  88 {
  89         char *d = strdup(s);
  90         if (d == NULL)
  91                 (*efunc)(1, "Cannot copy string");
  92         return d;
  93 }
  94
  95 char *
  96 estrndup(const char *s, size_t len)
  97 {
  98         char *d = strndup(s, len);
  99         if (d == NULL)
 100                 (*efunc)(1, "Cannot copy string");
 101         return d;
 102 }
 103
 104 void *
 105 emalloc(size_t n)
 106 {
 107         void *p = malloc(n);
 108         if (p == NULL && n != 0)
 109                 (*efunc)(1, "Cannot allocate %zu bytes", n);
 110         return p;
 111 }
 112
 113 void *
 114 ecalloc(size_t n, size_t s)
 115 {
 116         void *p = calloc(n, s);
 117         if (p == NULL && n != 0 && s != 0)
 118                 (*efunc)(1, "Cannot allocate %zu blocks of size %zu", n, s);
 119         return p;
 120 }
 121
 122 void *
 123 erealloc(void *p, size_t n)
 124 {
 125         void *q = realloc(p, n);
 126         if (q == NULL && n != 0)
 127                 (*efunc)(1, "Cannot re-allocate %zu bytes", n);
 128         return q;
 129 }
 130
 131 void
 132 ereallocarr(void *p, size_t n, size_t s)
 133 {
 134         int rv = reallocarr(p, n, s);
 135         if (rv != 0) {
 136                 errno = rv;
 137                 (*efunc)(1, "Cannot re-allocate %zu * %zu bytes", n, s);
 138         }
 139 }
 140
 141 FILE *
 142 efopen(const char *p, const char *m)
 143 {
 144         FILE *fp = fopen(p, m);
 145         if (fp == NULL)
 146                 (*efunc)(1, "Cannot open `%s'", p);
 147         return fp;
 148 }
 149
 150 int
 151 easprintf(char ** __restrict ret, const char * __restrict format, ...)
 152 {
 153         int rv;
 154         va_list ap;
 155         va_start(ap, format);
 156         if ((rv = vasprintf(ret, format, ap)) == -1)
 157                 (*efunc)(1, "Cannot format string");
 158         va_end(ap);
 159         return rv;
 160 }
 161
 162 int
 163 evasprintf(char ** __restrict ret, const char * __restrict format, va_list ap)
 164 {
 165         int rv;
 166         if ((rv = vasprintf(ret, format, ap)) == -1)
 167                 (*efunc)(1, "Cannot format string");
 168         return rv;
 169 }
 170
 171 intmax_t
 172 estrtoi(const char * nptr, int base, intmax_t lo, intmax_t hi)
 173 {
 174         int e;
 175         intmax_t rv = strtoi(nptr, NULL, base, lo, hi, &e);
 176         if (e != 0) {
 177                 errno = e;
 178                 (*efunc)(1,
 179                     "Cannot convert string value '%s' with base %d to a number in range [%jd .. %jd]",
 180                     nptr, base, lo, hi);
 181         }
 182         return rv;
 183 }
 184
 185 uintmax_t
 186 estrtou(const char * nptr, int base, uintmax_t lo, uintmax_t hi)
 187 {
 188         int e;
 189         uintmax_t rv = strtou(nptr, NULL, base, lo, hi, &e);
 190         if (e != 0) {
 191                 errno = e;
 192                 (*efunc)(1,
 193                     "Cannot convert string value '%s' with base %d to a number in range [%ju .. %ju]",
 194                     nptr, base, lo, hi);
 195         }
 196         return rv;
 197 }