usage.c

   1 /*
   2  * GIT - The information manager from hell
   3  *
   4  * Copyright (C) Linus Torvalds, 2005
   5  */
   6 #include "git-compat-util.h"
   7 #include "cache.h"
   8
   9 void vreportf(const char *prefix, const char *err, va_list params)
  10 {
  11         char msg[4096];
  12         char *p;
  13
  14         vsnprintf(msg, sizeof(msg), err, params);
  15         for (p = msg; *p; p++) {
  16                 if (iscntrl(*p) && *p != '\t' && *p != '\n')
  17                         *p = '?';
  18         }
  19         fprintf(stderr, "%s%s\n", prefix, msg);
  20 }
  21
  22 static NORETURN void usage_builtin(const char *err, va_list params)
  23 {
  24         vreportf("usage: ", err, params);
  25         exit(129);
  26 }
  27
  28 static NORETURN void die_builtin(const char *err, va_list params)
  29 {
  30         vreportf("fatal: ", err, params);
  31         exit(128);
  32 }
  33
  34 static void error_builtin(const char *err, va_list params)
  35 {
  36         vreportf("error: ", err, params);
  37 }
  38
  39 static void warn_builtin(const char *warn, va_list params)
  40 {
  41         vreportf("warning: ", warn, params);
  42 }
  43
  44 static int die_is_recursing_builtin(void)
  45 {
  46         static int dying;
  47         /*
  48          * Just an arbitrary number X where "a < x < b" where "a" is
  49          * "maximum number of pthreads we'll ever plausibly spawn" and
  50          * "b" is "something less than Inf", since the point is to
  51          * prevent infinite recursion.
  52          */
  53         static const int recursion_limit = 1024;
  54
  55         dying++;
  56         if (dying > recursion_limit) {
  57                 return 1;
  58         } else if (dying == 2) {
  59                 warning("die() called many times. Recursion error or racy threaded death!");
  60                 return 0;
  61         } else {
  62                 return 0;
  63         }
  64 }
  65
  66 /* If we are in a dlopen()ed .so write to a global variable would segfault
  67  * (ugh), so keep things static. */
  68 static NORETURN_PTR void (*usage_routine)(const char *err, va_list params) = usage_builtin;
  69 static NORETURN_PTR void (*die_routine)(const char *err, va_list params) = die_builtin;
  70 static void (*error_routine)(const char *err, va_list params) = error_builtin;
  71 static void (*warn_routine)(const char *err, va_list params) = warn_builtin;
  72 static int (*die_is_recursing)(void) = die_is_recursing_builtin;
  73
  74 void set_die_routine(NORETURN_PTR void (*routine)(const char *err, va_list params))
  75 {
  76         die_routine = routine;
  77 }
  78
  79 void set_error_routine(void (*routine)(const char *err, va_list params))
  80 {
  81         error_routine = routine;
  82 }
  83
  84 void (*get_error_routine(void))(const char *err, va_list params)
  85 {
  86         return error_routine;
  87 }
  88
  89 void set_warn_routine(void (*routine)(const char *warn, va_list params))
  90 {
  91         warn_routine = routine;
  92 }
  93
  94 void (*get_warn_routine(void))(const char *warn, va_list params)
  95 {
  96         return warn_routine;
  97 }
  98
  99 void set_die_is_recursing_routine(int (*routine)(void))
 100 {
 101         die_is_recursing = routine;
 102 }
 103
 104 void NORETURN usagef(const char *err, ...)
 105 {
 106         va_list params;
 107
 108         va_start(params, err);
 109         usage_routine(err, params);
 110         va_end(params);
 111 }
 112
 113 void NORETURN usage(const char *err)
 114 {
 115         usagef("%s", err);
 116 }
 117
 118 void NORETURN die(const char *err, ...)
 119 {
 120         va_list params;
 121
 122         if (die_is_recursing()) {
 123                 fputs("fatal: recursion detected in die handler\n", stderr);
 124                 exit(128);
 125         }
 126
 127         va_start(params, err);
 128         die_routine(err, params);
 129         va_end(params);
 130 }
 131
 132 static const char *fmt_with_err(char *buf, int n, const char *fmt)
 133 {
 134         char str_error[256], *err;
 135         int i, j;
 136
 137         err = strerror(errno);
 138         for (i = j = 0; err[i] && j < sizeof(str_error) - 1; ) {
 139                 if ((str_error[j++] = err[i++]) != '%')
 140                         continue;
 141                 if (j < sizeof(str_error) - 1) {
 142                         str_error[j++] = '%';
 143                 } else {
 144                         /* No room to double the '%', so we overwrite it with
 145                          * '\0' below */
 146                         j--;
 147                         break;
 148                 }
 149         }
 150         str_error[j] = 0;
 151         snprintf(buf, n, "%s: %s", fmt, str_error);
 152         return buf;
 153 }
 154
 155 void NORETURN die_errno(const char *fmt, ...)
 156 {
 157         char buf[1024];
 158         va_list params;
 159
 160         if (die_is_recursing()) {
 161                 fputs("fatal: recursion detected in die_errno handler\n",
 162                         stderr);
 163                 exit(128);
 164         }
 165
 166         va_start(params, fmt);
 167         die_routine(fmt_with_err(buf, sizeof(buf), fmt), params);
 168         va_end(params);
 169 }
 170
 171 #undef error_errno
 172 int error_errno(const char *fmt, ...)
 173 {
 174         char buf[1024];
 175         va_list params;
 176
 177         va_start(params, fmt);
 178         error_routine(fmt_with_err(buf, sizeof(buf), fmt), params);
 179         va_end(params);
 180         return -1;
 181 }
 182
 183 #undef error
 184 int error(const char *err, ...)
 185 {
 186         va_list params;
 187
 188         va_start(params, err);
 189         error_routine(err, params);
 190         va_end(params);
 191         return -1;
 192 }
 193
 194 void warning_errno(const char *warn, ...)
 195 {
 196         char buf[1024];
 197         va_list params;
 198
 199         va_start(params, warn);
 200         warn_routine(fmt_with_err(buf, sizeof(buf), warn), params);
 201         va_end(params);
 202 }
 203
 204 void warning(const char *warn, ...)
 205 {
 206         va_list params;
 207
 208         va_start(params, warn);
 209         warn_routine(warn, params);
 210         va_end(params);
 211 }
 212
 213 static NORETURN void BUG_vfl(const char *file, int line, const char *fmt, va_list params)
 214 {
 215         char prefix[256];
 216
 217         /* truncation via snprintf is OK here */
 218         if (file)
 219                 snprintf(prefix, sizeof(prefix), "BUG: %s:%d: ", file, line);
 220         else
 221                 snprintf(prefix, sizeof(prefix), "BUG: ");
 222
 223         vreportf(prefix, fmt, params);
 224         abort();
 225 }
 226
 227 #ifdef HAVE_VARIADIC_MACROS
 228 NORETURN void BUG_fl(const char *file, int line, const char *fmt, ...)
 229 {
 230         va_list ap;
 231         va_start(ap, fmt);
 232         BUG_vfl(file, line, fmt, ap);
 233         va_end(ap);
 234 }
 235 #else
 236 NORETURN void BUG(const char *fmt, ...)
 237 {
 238         va_list ap;
 239         va_start(ap, fmt);
 240         BUG_vfl(NULL, 0, fmt, ap);
 241         va_end(ap);
 242 }
 243 #endif
 244
 245 #ifdef SUPPRESS_ANNOTATED_LEAKS
 246 void unleak_memory(const void *ptr, size_t len)
 247 {
 248         static struct suppressed_leak_root {
 249                 struct suppressed_leak_root *next;
 250                 char data[FLEX_ARRAY];
 251         } *suppressed_leaks;
 252         struct suppressed_leak_root *root;
 253
 254         FLEX_ALLOC_MEM(root, data, ptr, len);
 255         root->next = suppressed_leaks;
 256         suppressed_leaks = root;
 257 }
 258 #endif