lib/libnetbsd/efun.c

   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 (*
  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
  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
  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 *
  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 *
  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 *
 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 *
 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 *
 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 *
 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
 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
 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 }