git.sur5r.net Git - i3/i3/blob - libi3/safewrappers.c

   1 /*
   2  * vim:ts=4:sw=4:expandtab
   3  *
   4  * i3 - an improved dynamic tiling window manager
   5  * © 2009-2011 Michael Stapelberg and contributors (see also: LICENSE)
   6  *
   7  */
   8 #include <string.h>
   9 #include <stdlib.h>
  10 #include <stdarg.h>
  11 #include <unistd.h>
  12 #include <stdio.h>
  13 #include <err.h>
  14 #include <errno.h>
  15
  16 #include "libi3.h"
  17
  18 /*
  19  * The s* functions (safe) are wrappers around malloc, strdup, …, which exits if one of
  20  * the called functions returns NULL, meaning that there is no more memory available
  21  *
  22  */
  23 void *smalloc(size_t size) {
  24     void *result = malloc(size);
  25     if (result == NULL)
  26         err(EXIT_FAILURE, "malloc(%zd)", size);
  27     return result;
  28 }
  29
  30 void *scalloc(size_t size) {
  31     void *result = calloc(size, 1);
  32     if (result == NULL)
  33         err(EXIT_FAILURE, "calloc(%zd)", size);
  34     return result;
  35 }
  36
  37 void *srealloc(void *ptr, size_t size) {
  38     void *result = realloc(ptr, size);
  39     if (result == NULL && size > 0)
  40         err(EXIT_FAILURE, "realloc(%zd)", size);
  41     return result;
  42 }
  43
  44 char *sstrdup(const char *str) {
  45     char *result = strdup(str);
  46     if (result == NULL)
  47         err(EXIT_FAILURE, "strdup()");
  48     return result;
  49 }
  50
  51 int sasprintf(char **strp, const char *fmt, ...) {
  52     va_list args;
  53     int result;
  54
  55     va_start(args, fmt);
  56     if ((result = vasprintf(strp, fmt, args)) == -1)
  57         err(EXIT_FAILURE, "asprintf(%s)", fmt);
  58     va_end(args);
  59     return result;
  60 }
  61
  62 ssize_t writeall(int fd, const void *buf, size_t count) {
  63     int written = 0;
  64     ssize_t n = 0;
  65
  66     while (written < count) {
  67         n = write(fd, buf + written, count - written);
  68         if (n == -1) {
  69             if (errno == EINTR || errno == EAGAIN)
  70                 continue;
  71             return n;
  72         }
  73         written += n;
  74     }
  75
  76     return written;
  77 }
  78
  79 ssize_t swrite(int fd, const void *buf, size_t count) {
  80     ssize_t n;
  81
  82     n = writeall(fd, buf, count);
  83     if (n == -1)
  84         err(EXIT_FAILURE, "Failed to write %d", fd);
  85     else
  86         return n;
  87 }