diff options
Diffstat (limited to 'lib/dns/ttl.c')
| -rw-r--r-- | lib/dns/ttl.c | 196 |
1 files changed, 196 insertions, 0 deletions
diff --git a/lib/dns/ttl.c b/lib/dns/ttl.c new file mode 100644 index 00000000..ef588ca8 --- /dev/null +++ b/lib/dns/ttl.c @@ -0,0 +1,196 @@ +/* + * Copyright (C) 1999, 2000 Internet Software Consortium. + * + * Permission to use, copy, modify, and distribute this software for any + * purpose with or without fee is hereby granted, provided that the above + * copyright notice and this permission notice appear in all copies. + * + * THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM DISCLAIMS + * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES + * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL INTERNET SOFTWARE + * CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL + * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR + * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS + * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS + * SOFTWARE. + */ + +#include <config.h> + +#include <ctype.h> +#include <string.h> + +#include <isc/assertions.h> +#include <isc/error.h> +#include <isc/boolean.h> +#include <isc/region.h> +#include <isc/buffer.h> +#include <isc/print.h> + +#include <dns/result.h> +#include <dns/ttl.h> + +#define RETERR(x) do { \ + isc_result_t __r = (x); \ + if (__r != DNS_R_SUCCESS) \ + return (__r); \ + } while (0) + + +static isc_result_t bind_ttl(isc_textregion_t *source, isc_uint32_t *ttl); + +/* Helper for dns_ttl_totext(). */ + +static isc_result_t +ttlfmt(unsigned int t, char *s, isc_boolean_t verbose, + isc_boolean_t space, isc_buffer_t *target) +{ + char tmp[60]; + size_t len; + isc_region_t region; + if (verbose) + len = snprintf(tmp, sizeof(tmp), "%s%u %s%s", + space ? " " : "", + t, s, + t == 1 ? "" : "s"); + else + len = snprintf(tmp, sizeof(tmp), "%u%c", t, s[0]); + INSIST(len + 1 <= sizeof tmp); + isc_buffer_available(target, ®ion); + if (len > region.length) + return (DNS_R_NOSPACE); + memcpy(region.base, tmp, len); + isc_buffer_add(target, len); + return (DNS_R_SUCCESS); +} + +/* Derived from bind8 ns_format_ttl(). */ + +isc_result_t +dns_ttl_totext(isc_uint32_t src, isc_boolean_t verbose, + isc_buffer_t *target) +{ + unsigned secs, mins, hours, days, weeks, x; + + secs = src % 60; src /= 60; + mins = src % 60; src /= 60; + hours = src % 24; src /= 24; + days = src % 7; src /= 7; + weeks = src; src = 0; + + x = 0; + if (weeks != 0) { + RETERR(ttlfmt(weeks, "week", verbose, ISC_TF(x > 0), target)); + x++; + } + if (days != 0) { + RETERR(ttlfmt(days, "day", verbose, ISC_TF(x > 0), target)); + x++; + } + if (hours != 0) { + RETERR(ttlfmt(hours, "hour", verbose, ISC_TF(x > 0), target)); + x++; + } + if (mins != 0) { + RETERR(ttlfmt(mins, "minute", verbose, ISC_TF(x > 0), target)); + x++; + } + if (secs != 0 || (weeks == 0 && days == 0 && hours == 0 && mins == 0)) { + RETERR(ttlfmt(secs, "second", verbose, ISC_TF(x > 0), target)); + x++; + } + INSIST (x > 0); + /* + * If only a single unit letter is printed, print it + * in upper case. (Why? Because BIND 8 does that. + * Presumably it has a reason.) + */ + if (x == 1 && !verbose) { + isc_region_t region; + /* + * The unit letter is the last character in the + * used region of the buffer. + */ + isc_buffer_used(target, ®ion); + region.base[region.length - 1] = + toupper(region.base[region.length - 1]); + } + return (DNS_R_SUCCESS); +} + +isc_result_t +dns_counter_fromtext(isc_textregion_t *source, isc_uint32_t *ttl) { + return (bind_ttl(source, ttl)); +} + +isc_result_t +dns_ttl_fromtext(isc_textregion_t *source, isc_uint32_t *ttl) { + isc_result_t result; + + result = bind_ttl(source, ttl); + if (result != DNS_R_SUCCESS) + result = DNS_R_BADTTL; + return (result); +} + +static isc_result_t +bind_ttl(isc_textregion_t *source, isc_uint32_t *ttl) { + isc_uint32_t tmp = 0; + unsigned long n; + char *e, *s; + char buf[64]; + + /* + * Copy the buffer as it may not be NULL terminated. + * No legal counter / ttl is longer that 63 characters. + */ + if (source->length > sizeof(buf) - 1) + return(DNS_R_SYNTAX); + strncpy(buf, source->base, source->length); + buf[source->length] = '\0'; + s = buf; + + do { + n = strtoul(s, &e, 10); + if (s == e) + return (DNS_R_SYNTAX); + switch (*e) { + case 'w': + case 'W': + tmp += n * 7 * 24 * 3600; + s = e + 1; + break; + case 'd': + case 'D': + tmp += n * 24 * 3600; + s = e + 1; + break; + case 'h': + case 'H': + tmp += n * 3600; + s = e + 1; + break; + case 'm': + case 'M': + tmp += n * 60; + s = e + 1; + break; + case 's': + case 'S': + tmp += n; + s = e + 1; + break; + case '\0': + /* Plain number? */ + if (tmp != 0) + return (DNS_R_SYNTAX); + tmp = n; + s = e; + break; + default: + return (DNS_R_SYNTAX); + } + } while (*s != 0); + *ttl = tmp; + return (DNS_R_SUCCESS); +} |