busybox/libbb/xfuncs.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Utility routines.
   4  *
   5  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15  * General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  20  */
  21
  22 #include <sys/types.h>
  23 #include <sys/stat.h>
  24 #include <stdio.h>
  25 #include <string.h>
  26 #include <stdlib.h>
  27 #include <unistd.h>
  28 #include <fcntl.h>
  29 #include "libbb.h"
  30
  31
  32 #ifndef DMALLOC
  33 #ifdef L_xmalloc
  34 extern void *xmalloc(size_t size)
  35 {
  36         void *ptr = malloc(size);
  37         if (ptr == NULL && size != 0)
  38                 bb_error_msg_and_die(bb_msg_memory_exhausted);
  39         return ptr;
  40 }
  41 #endif
  42
  43 #ifdef L_xrealloc
  44 extern void *xrealloc(void *ptr, size_t size)
  45 {
  46         ptr = realloc(ptr, size);
  47         if (ptr == NULL && size != 0)
  48                 bb_error_msg_and_die(bb_msg_memory_exhausted);
  49         return ptr;
  50 }
  51 #endif
  52
  53 #ifdef L_xcalloc
  54 extern void *xcalloc(size_t nmemb, size_t size)
  55 {
  56         void *ptr = calloc(nmemb, size);
  57         if (ptr == NULL && nmemb != 0 && size != 0)
  58                 bb_error_msg_and_die(bb_msg_memory_exhausted);
  59         return ptr;
  60 }
  61 #endif
  62 #endif /* DMALLOC */
  63
  64 #ifdef L_xstrdup
  65 extern char * bb_xstrdup (const char *s) {
  66         char *t;
  67
  68         if (s == NULL)
  69                 return NULL;
  70
  71         t = strdup (s);
  72
  73         if (t == NULL)
  74                 bb_error_msg_and_die(bb_msg_memory_exhausted);
  75
  76         return t;
  77 }
  78 #endif
  79
  80 #ifdef L_xstrndup
  81 extern char * bb_xstrndup (const char *s, int n) {
  82         char *t;
  83
  84         if (s == NULL)
  85                 bb_error_msg_and_die("bb_xstrndup bug");
  86
  87         t = xmalloc(++n);
  88
  89         return safe_strncpy(t,s,n);
  90 }
  91 #endif
  92
  93 #ifdef L_xfopen
  94 FILE *bb_xfopen(const char *path, const char *mode)
  95 {
  96         FILE *fp;
  97         if ((fp = fopen(path, mode)) == NULL)
  98                 bb_perror_msg_and_die("%s", path);
  99         return fp;
 100 }
 101 #endif
 102
 103 #ifdef L_xopen
 104 extern int bb_xopen(const char *pathname, int flags)
 105 {
 106         int ret;
 107
 108         ret = open(pathname, flags, 0777);
 109         if (ret == -1) {
 110                 bb_perror_msg_and_die("%s", pathname);
 111         }
 112         return ret;
 113 }
 114 #endif
 115
 116 #ifdef L_xread
 117 extern ssize_t bb_xread(int fd, void *buf, size_t count)
 118 {
 119         ssize_t size;
 120
 121         size = read(fd, buf, count);
 122         if (size == -1) {
 123                 bb_perror_msg_and_die("Read error");
 124         }
 125         return(size);
 126 }
 127 #endif
 128
 129 #ifdef L_xread_all
 130 extern void bb_xread_all(int fd, void *buf, size_t count)
 131 {
 132         ssize_t size;
 133
 134         while (count) {
 135                 if ((size = bb_xread(fd, buf, count)) == 0) {   /* EOF */
 136                         bb_error_msg_and_die("Short read");
 137                 }
 138                 count -= size;
 139                 buf = ((char *) buf) + size;
 140         }
 141         return;
 142 }
 143 #endif
 144
 145 #ifdef L_xread_char
 146 extern unsigned char bb_xread_char(int fd)
 147 {
 148         char tmp;
 149
 150         bb_xread_all(fd, &tmp, 1);
 151
 152         return(tmp);
 153 }
 154 #endif
 155
 156 #ifdef L_xferror
 157 extern void bb_xferror(FILE *fp, const char *fn)
 158 {
 159         if (ferror(fp)) {
 160                 bb_error_msg_and_die("%s", fn);
 161         }
 162 }
 163 #endif
 164
 165 #ifdef L_xferror_stdout
 166 extern void bb_xferror_stdout(void)
 167 {
 168         bb_xferror(stdout, bb_msg_standard_output);
 169 }
 170 #endif
 171
 172 #ifdef L_xfflush_stdout
 173 extern void bb_xfflush_stdout(void)
 174 {
 175         if (fflush(stdout)) {
 176                 bb_perror_msg_and_die(bb_msg_standard_output);
 177         }
 178 }
 179 #endif
 180
 181 #ifdef L_strlen
 182 /* Stupid gcc always includes its own builtin strlen()... */
 183 #undef strlen
 184 size_t bb_strlen(const char *string)
 185 {
 186             return(strlen(string));
 187 }
 188 #endif
 189
 190 /* END CODE */
 191 /*
 192 Local Variables:
 193 c-file-style: "linux"
 194 c-basic-offset: 4
 195 tab-width: 4
 196 End:
 197 */