diff options
| -rw-r--r-- | git-compat-util.h | 4 | ||||
| -rw-r--r-- | usage.c | 46 |
2 files changed, 46 insertions, 4 deletions
diff --git a/git-compat-util.h b/git-compat-util.h index 5d543d29f..b3d4cf532 100644 --- a/git-compat-util.h +++ b/git-compat-util.h @@ -40,6 +40,10 @@ extern void usage(const char *err) NORETURN; extern void die(const char *err, ...) NORETURN __attribute__((format (printf, 1, 2))); extern int error(const char *err, ...) __attribute__((format (printf, 1, 2))); +extern void set_usage_routine(void (*routine)(const char *err) NORETURN); +extern void set_die_routine(void (*routine)(const char *err, va_list params) NORETURN); +extern void set_error_routine(void (*routine)(const char *err, va_list params)); + #ifdef NO_MMAP #ifndef PROT_READ @@ -12,20 +12,58 @@ static void report(const char *prefix, const char *err, va_list params) fputs("\n", stderr); } -void usage(const char *err) +void usage_builtin(const char *err) { fprintf(stderr, "usage: %s\n", err); exit(129); } +void die_builtin(const char *err, va_list params) +{ + report("fatal: ", err, params); + exit(128); +} + +void error_builtin(const char *err, va_list params) +{ + report("error: ", err, params); +} + + +/* If we are in a dlopen()ed .so write to a global variable would segfault + * (ugh), so keep things static. */ +static void (*usage_routine)(const char *err) NORETURN = usage_builtin; +static void (*die_routine)(const char *err, va_list params) NORETURN = die_builtin; +static void (*error_routine)(const char *err, va_list params) = error_builtin; + +void set_usage_routine(void (*routine)(const char *err) NORETURN) +{ + usage_routine = routine; +} + +void set_die_routine(void (*routine)(const char *err, va_list params) NORETURN) +{ + die_routine = routine; +} + +void set_error_routine(void (*routine)(const char *err, va_list params)) +{ + error_routine = routine; +} + + +void usage(const char *err) +{ + usage_routine(err); +} + void die(const char *err, ...) { va_list params; va_start(params, err); - report("fatal: ", err, params); + die_routine(err, params); va_end(params); - exit(128); } int error(const char *err, ...) @@ -33,7 +71,7 @@ int error(const char *err, ...) va_list params; va_start(params, err); - report("error: ", err, params); + error_routine(err, params); va_end(params); return -1; } |