tftpd/misc.c

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