projects / apple / network_cmds.git / blame
| Commit | Line | Data |
|---|---|---|
| 89c4ed63 A |
1 | Index: smallapp/unbound-host.c |
| 2 | =================================================================== | |
| 3 | --- smallapp/unbound-host.c (revision 2115) | |
| 4 | +++ smallapp/unbound-host.c (working copy) | |
| 5 | @@ -62,9 +62,18 @@ | |
| 6 | #include "libunbound/unbound.h" | |
| 7 | #include <ldns/ldns.h> | |
| 8 | ||
| 9 | +/** status variable ala nagios */ | |
| 10 | +#define FINAL_STATUS_OK 0 | |
| 11 | +#define FINAL_STATUS_WARNING 1 | |
| 12 | +#define FINAL_STATUS_CRITICAL 2 | |
| 13 | +#define FINAL_STATUS_UNKNOWN 3 | |
| 14 | + | |
| 15 | /** verbosity for unbound-host app */ | |
| 16 | static int verb = 0; | |
| 17 | ||
| 18 | +/** variable to determine final output */ | |
| 19 | +static int final_status = FINAL_STATUS_UNKNOWN; | |
| 20 | + | |
| 21 | /** Give unbound-host usage, and exit (1). */ | |
| 22 | static void | |
| 23 | usage() | |
| 24 | @@ -93,7 +102,7 @@ | |
| 25 | printf("Version %s\n", PACKAGE_VERSION); | |
| 26 | printf("BSD licensed, see LICENSE in source package for details.\n"); | |
| 27 | printf("Report bugs to %s\n", PACKAGE_BUGREPORT); | |
| 28 | - exit(1); | |
| 29 | + exit(FINAL_STATUS_UNKNOWN); | |
| 30 | } | |
| 31 | ||
| 32 | /** determine if str is ip4 and put into reverse lookup format */ | |
| 33 | @@ -138,7 +147,7 @@ | |
| 34 | *res = strdup(buf); | |
| 35 | if(!*res) { | |
| 36 | fprintf(stderr, "error: out of memory\n"); | |
| 37 | - exit(1); | |
| 38 | + exit(FINAL_STATUS_UNKNOWN); | |
| 39 | } | |
| 40 | return 1; | |
| 41 | } | |
| 42 | @@ -158,7 +167,7 @@ | |
| 43 | } | |
| 44 | if(!res) { | |
| 45 | fprintf(stderr, "error: out of memory\n"); | |
| 46 | - exit(1); | |
| 47 | + exit(FINAL_STATUS_UNKNOWN); | |
| 48 | } | |
| 49 | return res; | |
| 50 | } | |
| 51 | @@ -172,7 +181,7 @@ | |
| 52 | if(r == 0 && strcasecmp(t, "TYPE0") != 0 && | |
| 53 | strcmp(t, "") != 0) { | |
| 54 | fprintf(stderr, "error unknown type %s\n", t); | |
| 55 | - exit(1); | |
| 56 | + exit(FINAL_STATUS_UNKNOWN); | |
| 57 | } | |
| 58 | return r; | |
| 59 | } | |
| 60 | @@ -191,7 +200,7 @@ | |
| 61 | if(r == 0 && strcasecmp(c, "CLASS0") != 0 && | |
| 62 | strcmp(c, "") != 0) { | |
| 63 | fprintf(stderr, "error unknown class %s\n", c); | |
| 64 | - exit(1); | |
| 65 | + exit(FINAL_STATUS_UNKNOWN); | |
| 66 | } | |
| 67 | return r; | |
| 68 | } | |
| 69 | @@ -207,6 +216,19 @@ | |
| 70 | return "(insecure)"; | |
| 71 | } | |
| 72 | ||
| 73 | +/** update the final status for the exit code */ | |
| 74 | +void | |
| 75 | +update_final_status(struct ub_result* result) | |
| 76 | +{ | |
| 77 | + if (final_status == FINAL_STATUS_UNKNOWN || final_status == FINAL_STATUS_OK) { | |
| 78 | + if (result->secure) final_status = FINAL_STATUS_OK; | |
| 79 | + else if (result->bogus) final_status = FINAL_STATUS_CRITICAL; | |
| 80 | + else final_status = FINAL_STATUS_WARNING; | |
| 81 | + } | |
| 82 | + else if (final_status == FINAL_STATUS_WARNING && result->bogus) | |
| 83 | + final_status = FINAL_STATUS_CRITICAL; | |
| 84 | +} | |
| 85 | + | |
| 86 | /** nice string for type */ | |
| 87 | static void | |
| 88 | pretty_type(char* s, size_t len, int t) | |
| 89 | @@ -353,7 +375,7 @@ | |
| 90 | } else { | |
| 91 | fprintf(stderr, "could not parse " | |
| 92 | "reply packet to ANY query\n"); | |
| 93 | - exit(1); | |
| 94 | + exit(FINAL_STATUS_UNKNOWN); | |
| 95 | } | |
| 96 | ldns_pkt_free(p); | |
| 97 | ||
| 98 | @@ -388,9 +410,10 @@ | |
| 99 | ret = ub_resolve(ctx, q, t, c, &result); | |
| 100 | if(ret != 0) { | |
| 101 | fprintf(stderr, "resolve error: %s\n", ub_strerror(ret)); | |
| 102 | - exit(1); | |
| 103 | + exit(FINAL_STATUS_UNKNOWN); | |
| 104 | } | |
| 105 | pretty_output(q, t, c, result, docname); | |
| 106 | + update_final_status(result); | |
| 107 | ret = result->nxdomain; | |
| 108 | ub_resolve_free(result); | |
| 109 | return ret; | |
| 110 | @@ -427,7 +450,7 @@ | |
| 111 | { | |
| 112 | if(r != 0) { | |
| 113 | fprintf(stderr, "error: %s\n", ub_strerror(r)); | |
| 114 | - exit(1); | |
| 115 | + exit(FINAL_STATUS_UNKNOWN); | |
| 116 | } | |
| 117 | } | |
| 118 | ||
| 119 | @@ -448,7 +471,7 @@ | |
| 120 | ctx = ub_ctx_create(); | |
| 121 | if(!ctx) { | |
| 122 | fprintf(stderr, "error: out of memory\n"); | |
| 123 | - exit(1); | |
| 124 | + exit(FINAL_STATUS_UNKNOWN); | |
| 125 | } | |
| 126 | ||
| 127 | /* parse the options */ | |
| 128 | @@ -509,5 +532,5 @@ | |
| 129 | usage(); | |
| 130 | ||
| 131 | lookup(ctx, argv[0], qtype, qclass); | |
| 132 | - return 0; | |
| 133 | + return final_status; | |
| 134 | } |