src/die.h

   1 /*
   2  * netyack
   3  * By Daniel Borkmann <daniel@netyack.org>
   4  * Copyright 2009, 2010 Daniel Borkmann.
   5  * Subject to the GPL.
   6  */
   7
   8 #ifndef DIE_H
   9 #define DIE_H
  10
  11 #include <stdio.h>
  12 #include <stdlib.h>
  13 #include <stdarg.h>
  14 #include <string.h>
  15 #include <errno.h>
  16
  17 #include "tty.h"
  18
  19 static inline void error_and_die(int status, char *msg, ...)
  20 {
  21         va_list vl;
  22         va_start(vl, msg);
  23         vfprintf(stderr, msg, vl);
  24         va_end(vl);
  25
  26         exit(status);
  27 }
  28
  29 static inline void die(void)
  30 {
  31         exit(EXIT_FAILURE);
  32 }
  33
  34 static inline void panic(char *msg, ...)
  35 {
  36         va_list vl;
  37         va_start(vl, msg);
  38         vfprintf(stderr, msg, vl);
  39         va_end(vl);
  40
  41         die();
  42 }
  43
  44 static inline void whine(char *msg, ...)
  45 {
  46         va_list vl;
  47         va_start(vl, msg);
  48         vfprintf(stderr, msg, vl);
  49         va_end(vl);
  50 }
  51
  52 static inline void info(char *msg, ...)
  53 {
  54         va_list vl;
  55         va_start(vl, msg);
  56         vfprintf(stdout, msg, vl);
  57         va_end(vl);
  58 }
  59
  60 #ifdef _DEBUG_
  61 static inline void debug(char *msg, ...)
  62 {
  63         va_list vl;
  64         va_start(vl, msg);
  65         vfprintf(stderr, msg, vl);
  66         va_end(vl);
  67
  68         fflush(stderr);
  69 }
  70
  71 static inline void debug_blue(char *msg, ...)
  72 {
  73         va_list vl;
  74
  75         fprintf(stderr, "%s", colorize_start_full(white, blue));
  76         va_start(vl, msg);
  77         vfprintf(stderr, msg, vl);
  78         va_end(vl);
  79         fprintf(stderr, "%s\n", colorize_end());
  80
  81         fflush(stderr);
  82 }
  83
  84 static inline void debug_red(char *msg, ...)
  85 {
  86         va_list vl;
  87
  88         fprintf(stderr, "%s", colorize_start_full(white, red));
  89         va_start(vl, msg);
  90         vfprintf(stderr, msg, vl);
  91         va_end(vl);
  92         fprintf(stderr, "%s\n", colorize_end());
  93
  94         fflush(stderr);
  95 }
  96
  97 static inline void debug_green(char *msg, ...)
  98 {
  99         va_list vl;
 100
 101         fprintf(stderr, "%s", colorize_start_full(white, green));
 102         va_start(vl, msg);
 103         vfprintf(stderr, msg, vl);
 104         va_end(vl);
 105         fprintf(stderr, "%s\n", colorize_end());
 106
 107         fflush(stderr);
 108 }
 109 #else
 110 static inline void debug(char *msg, ...)
 111 {
 112         /* NOP */
 113 }
 114
 115 static inline void debug_blue(char *msg, ...)
 116 {
 117         /* NOP */
 118 }
 119
 120 static inline void debug_red(char *msg, ...)
 121 {
 122         /* NOP */
 123 }
 124
 125 static inline void debug_green(char *msg, ...)
 126 {
 127         /* NOP */
 128 }
 129 #endif /* _DEBUG_ */
 130
 131 static inline void print_blue(char *msg, ...)
 132 {
 133         va_list vl;
 134
 135         fprintf(stdout, "%s", colorize_start_full(white, blue));
 136         va_start(vl, msg);
 137         vfprintf(stdout, msg, vl);
 138         va_end(vl);
 139         fprintf(stdout, "%s\n", colorize_end());
 140
 141         fflush(stdout);
 142 }
 143
 144 static inline void print_red(char *msg, ...)
 145 {
 146         va_list vl;
 147
 148         fprintf(stdout, "%s", colorize_start_full(white, red));
 149         va_start(vl, msg);
 150         vfprintf(stdout, msg, vl);
 151         va_end(vl);
 152         fprintf(stdout, "%s\n", colorize_end());
 153
 154         fflush(stdout);
 155 }
 156
 157 static inline void print_green(char *msg, ...)
 158 {
 159         va_list vl;
 160
 161         fprintf(stdout, "%s", colorize_start_full(white, green));
 162         va_start(vl, msg);
 163         vfprintf(stdout, msg, vl);
 164         va_end(vl);
 165         fprintf(stdout, "%s\n", colorize_end());
 166
 167         fflush(stdout);
 168 }
 169
 170 static inline void puke_and_die_num(int status, int num, char *msg, ...)
 171 {
 172         va_list vl;
 173         va_start(vl, msg);
 174         vfprintf(stderr, msg, vl);
 175         va_end(vl);
 176
 177         fprintf(stderr, ": %s\n", strerror(num));
 178
 179         exit(status);
 180 }
 181
 182 static inline void puke_and_die(int status, char *msg, ...)
 183 {
 184         va_list vl;
 185         va_start(vl, msg);
 186         vfprintf(stderr, msg, vl);
 187         va_end(vl);
 188
 189         fprintf(stderr, ": %s\n", strerror(errno));
 190
 191         exit(status);
 192 }
 193
 194 #endif /* DIE_H */