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 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 char *sstrndup(const char *str, size_t size) {
  52     char *result = strndup(str, size);
  53     if (result == NULL)
  54         err(EXIT_FAILURE, "strndup()");
  55     return result;
  56 }
  57
  58 int sasprintf(char **strp, const char *fmt, ...) {
  59     va_list args;
  60     int result;
  61
  62     va_start(args, fmt);
  63     if ((result = vasprintf(strp, fmt, args)) == -1)
  64         err(EXIT_FAILURE, "asprintf(%s)", fmt);
  65     va_end(args);
  66     return result;
  67 }
  68
  69 ssize_t writeall(int fd, const void *buf, size_t count) {
  70     size_t written = 0;
  71     ssize_t n = 0;
  72
  73     while (written < count) {
  74         n = write(fd, buf + written, count - written);
  75         if (n == -1) {
  76             if (errno == EINTR || errno == EAGAIN)
  77                 continue;
  78             return n;
  79         }
  80         written += (size_t)n;
  81     }
  82
  83     return written;
  84 }
  85
  86 ssize_t swrite(int fd, const void *buf, size_t count) {
  87     ssize_t n;
  88
  89     n = writeall(fd, buf, count);
  90     if (n == -1)
  91         err(EXIT_FAILURE, "Failed to write %d", fd);
  92     else
  93         return n;
  94 }