tftpd/misc.c

   1 /* ----------------------------------------------------------------------- *
   2  *
   3  *   Copyright 2001-2007 H. Peter Anvin - All Rights Reserved
   4  *
   5  *   This program is free software available under the same license
   6  *   as the "OpenBSD" operating system, distributed at
   7  *   http://www.openbsd.org/.
   8  *
   9  * ----------------------------------------------------------------------- */
  10
  11 /*
  12  * misc.c
  13  *
  14  * Minor help routines.
  15  */
  16
  17 #include "config.h"             /* Must be included first! */
  18 #include <syslog.h>
  19 #include "tftpd.h"
  20
  21 /*
  22  * Set the signal handler and flags.  Basically a user-friendly
  23  * wrapper around sigaction().
  24  */
  25 void set_signal(int signum, void (*handler) (int), int flags)
  26 {
  27     struct sigaction sa;
  28
  29     memset(&sa, 0, sizeof sa);
  30     sa.sa_handler = handler;
  31     sigemptyset(&sa.sa_mask);
  32     sa.sa_flags = flags;
  33
  34     if (sigaction(signum, &sa, NULL)) {
  35         syslog(LOG_ERR, "sigaction: %m");
  36         exit(EX_OSERR);
  37     }
  38 }
  39
  40 /*
  41  * malloc() that syslogs an error message and bails if it fails.
  42  */
  43 void *tfmalloc(size_t size)
  44 {
  45     void *p = malloc(size);
  46
  47     if (!p) {
  48         syslog(LOG_ERR, "malloc: %m");
  49         exit(EX_OSERR);
  50     }
  51
  52     return p;
  53 }
  54
  55 /*
  56  * strdup() that does the equivalent
  57  */
  58 char *tfstrdup(const char *str)
  59 {
  60     char *p = strdup(str);
  61
  62     if (!p) {
  63         syslog(LOG_ERR, "strdup: %m");
  64         exit(EX_OSERR);
  65     }
  66
  67     return p;
  68 }