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 #include <sys/cdefs.h>
  33 #include <err.h>
  34 #include <errno.h>
  35 #include <string.h>
  36 #include <stdlib.h>
  37 #include <stdio.h>
  38 #include <stdarg.h>
  39 #include <libutil.h>
  40
  41 static void (*efunc)(int, const char *, ...) = err;
  42
  43 void (*
  44 esetfunc(void (*ef)(int, const char *, ...)))(int, const char *, ...)
  45 {
  46         void (*of)(int, const char *, ...) = efunc;
  47         efunc = ef == NULL ? (void (*)(int, const char *, ...))exit : ef;
  48         return of;
  49 }
  50
  51 size_t
  52 estrlcpy(char *dst, const char *src, size_t len)
  53 {
  54         size_t rv;
  55         if ((rv = strlcpy(dst, src, len)) >= len) {
  56                 errno = ENAMETOOLONG;
  57                 (*efunc)(1,
  58                     "Cannot copy string; %zu chars needed %zu provided",
  59                     rv, len);
  60         }
  61         return rv;
  62 }
  63
  64 size_t
  65 estrlcat(char *dst, const char *src, size_t len)
  66 {
  67         size_t rv;
  68         if ((rv = strlcat(dst, src, len)) >= len) {
  69                 errno = ENAMETOOLONG;
  70                 (*efunc)(1,
  71                     "Cannot append to string; %zu chars needed %zu provided",
  72                     rv, len);
  73         }
  74         return rv;
  75 }
  76
  77 char *
  78 estrdup(const char *s)
  79 {
  80         char *d = strdup(s);
  81         if (d == NULL)
  82                 (*efunc)(1, "Cannot copy string");
  83         return d;
  84 }
  85
  86 char *
  87 estrndup(const char *s, size_t len)
  88 {
  89         char *d = strndup(s, len);
  90         if (d == NULL)
  91                 (*efunc)(1, "Cannot copy string");
  92         return d;
  93 }
  94
  95 void *
  96 emalloc(size_t n)
  97 {
  98         void *p = malloc(n);
  99         if (p == NULL)
 100                 (*efunc)(1, "Cannot allocate %zu bytes", n);
 101         return p;
 102 }
 103
 104 void *
 105 ecalloc(size_t n, size_t s)
 106 {
 107         void *p = calloc(n, s);
 108         if (p == NULL)
 109                 (*efunc)(1, "Cannot allocate %zu bytes", n);
 110         return p;
 111 }
 112
 113 void *
 114 erealloc(void *p, size_t n)
 115 {
 116         void *q = realloc(p, n);
 117         if (q == NULL)
 118                 (*efunc)(1, "Cannot re-allocate %zu bytes", n);
 119         return q;
 120 }
 121
 122 int
 123 easprintf(char ** __restrict ret, const char * __restrict format, ...)
 124 {
 125         int rv;
 126         va_list ap;
 127         va_start(ap, format);
 128         if ((rv = vasprintf(ret, format, ap)) == -1)
 129                 (*efunc)(1, "Cannot format string");
 130         va_end(ap);
 131         return rv;
 132 }
 133