diff options
Diffstat (limited to 'src/utils/nsupdate/nsupdate_params.c')
| -rw-r--r-- | src/utils/nsupdate/nsupdate_params.c | 267 |
1 files changed, 267 insertions, 0 deletions
diff --git a/src/utils/nsupdate/nsupdate_params.c b/src/utils/nsupdate/nsupdate_params.c new file mode 100644 index 0000000..4d1f054 --- /dev/null +++ b/src/utils/nsupdate/nsupdate_params.c @@ -0,0 +1,267 @@ +/* Copyright (C) 2011 CZ.NIC, z.s.p.o. <knot-dns@labs.nic.cz> + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see <http://www.gnu.org/licenses/>. +*/ + +#include <config.h> +#include <string.h> +#include <stdio.h> +#include <getopt.h> +#include <stdlib.h> +#include <stdarg.h> + +#include "utils/nsupdate/nsupdate_params.h" +#include "utils/common/msg.h" +#include "utils/common/netio.h" +#include "common/errcode.h" +#include "common/descriptor.h" +#include "libknot/libknot.h" + +#define DEFAULT_RETRIES_NSUPDATE 3 +#define DEFAULT_TIMEOUT_NSUPDATE 12 + +static const style_t DEFAULT_STYLE_NSUPDATE = { + .format = FORMAT_NSUPDATE, + .style = { .wrap = false, .show_class = true, .show_ttl = true, + .verbose = false, .reduce = false, .human_ttl = false, + .human_tmstamp = true }, + .show_query = false, + .show_header = true, + .show_edns = false, + .show_question = true, + .show_answer = true, + .show_authority = true, + .show_additional = true, + .show_footer = false, +}; + +static void parse_rr(const scanner_t *s) +{ + UNUSED(s); +} + +static void parse_err(const scanner_t *s) { + ERR("failed to parse RR, %s\n", knot_strerror(s->error_code)); +} + +static int parser_set_default(scanner_t *s, const char *fmt, ...) +{ + /* Format string. */ + char buf[512]; /* Must suffice for domain name and TTL. */ + va_list ap; + va_start(ap, fmt); + int n = vsnprintf(buf, sizeof(buf), fmt, ap); + va_end(ap); + + if (n < 0 || (size_t)n >= sizeof(buf)) { + return KNOT_ESPACE; + } + + /* fmt must contain newline */ + if (scanner_process(buf, buf + n, 0, s) < 0) { + return KNOT_EPARSEFAIL; + } + + return KNOT_EOK; +} + +static int nsupdate_init(nsupdate_params_t *params) +{ + memset(params, 0, sizeof(nsupdate_params_t)); + + params->stop = false; + + /* Initialize list. */ + init_list(¶ms->qfiles); + + /* Default server. */ + params->server = server_create(DEFAULT_IPV4_NAME, DEFAULT_DNS_PORT); + if (!params->server) return KNOT_ENOMEM; + + /* Default settings. */ + params->ip = IP_ALL; + params->protocol = PROTO_ALL; + params->retries = DEFAULT_RETRIES_NSUPDATE; + params->wait = DEFAULT_TIMEOUT_NSUPDATE; + params->class_num = KNOT_CLASS_IN; + params->type_num = KNOT_RRTYPE_SOA; + + /* Default style. */ + params->style = DEFAULT_STYLE_NSUPDATE; + + /* Initialize RR parser. */ + params->rrp = scanner_create("."); + if (!params->rrp) return KNOT_ENOMEM; + params->rrp->process_record = parse_rr; + params->rrp->process_error = parse_err; + params->rrp->default_class = params->class_num; + params->zone = strdup("."); + nsupdate_set_ttl(params, 0); + nsupdate_set_origin(params, "."); + + return KNOT_EOK; +} + +void nsupdate_clean(nsupdate_params_t *params) +{ + strnode_t *n = NULL, *nxt = NULL; + + if (params == NULL) { + return; + } + + server_free(params->server); + server_free(params->srcif); + free(params->zone); + scanner_free(params->rrp); + knot_packet_free(¶ms->pkt); + knot_packet_free(¶ms->resp); + + /* Free qfiles. */ + WALK_LIST_DELSAFE(n, nxt, params->qfiles) { + free(n); + } + + knot_free_key_params(¶ms->key_params); + + /* Clean up the structure. */ + memset(params, 0, sizeof(*params)); +} + +static void nsupdate_help(void) +{ + printf("Usage: knsupdate [-d] [-v] [-k keyfile | -y [hmac:]name:key]\n" + " [-p port] [-t timeout] [-r retries] [filename]\n"); +} + +int nsupdate_parse(nsupdate_params_t *params, int argc, char *argv[]) +{ + int opt = 0, li = 0; + int ret = KNOT_EOK; + + if (params == NULL || argv == NULL) { + return KNOT_EINVAL; + } + + ret = nsupdate_init(params); + if (ret != KNOT_EOK) { + return ret; + } + + // Long options. + struct option opts[] = { + { "version", no_argument, 0, 'V' }, + { "help", no_argument, 0, 'h' }, + { 0, 0, 0, 0 } + }; + + /* Command line options processing. */ + while ((opt = getopt_long(argc, argv, "dhDvVp:t:r:y:k:", opts, &li)) + != -1) { + switch (opt) { + case 'd': + case 'D': /* Extra debugging. */ + msg_enable_debug(1); + break; + case 'h': + nsupdate_help(); + params->stop = true; + return KNOT_EOK; + case 'v': + params->protocol = PROTO_TCP; + break; + case 'V': + printf(KNSUPDATE_VERSION); + params->stop = true; + return KNOT_EOK; + case 'p': + free(params->server->service); + params->server->service = strdup(optarg); + if (!params->server->service) { + ERR("failed to set default port '%s'\n", optarg); + return KNOT_ENOMEM; + } + break; + case 'r': + ret = params_parse_num(optarg, ¶ms->retries); + if (ret != KNOT_EOK) return ret; + break; + case 't': + ret = params_parse_wait(optarg, ¶ms->wait); + if (ret != KNOT_EOK) return ret; + break; + case 'y': + ret = params_parse_tsig(optarg, ¶ms->key_params); + if (ret != KNOT_EOK) return ret; + break; + case 'k': + ret = params_parse_keyfile(optarg, ¶ms->key_params); + if (ret != KNOT_EOK) return ret; + break; + default: + nsupdate_help(); + return KNOT_ENOTSUP; + } + } + + /* No retries for TCP. */ + if (params->protocol == PROTO_TCP) { + params->retries = 0; + } else { + /* If wait/tries < 1 s, set 1 second for each try. */ + if (params->wait > 0 && + (uint32_t)params->wait < ( 1 + params->retries)) { + params->wait = 1; + } else { + params->wait /= (1 + params->retries); + } + } + + /* Process non-option parameters. */ + for (; optind < argc; ++optind) { + strnode_t *n = malloc(sizeof(strnode_t)); + if (!n) { /* Params will be cleaned on exit. */ + return KNOT_ENOMEM; + } + n->str = argv[optind]; + add_tail(¶ms->qfiles, &n->n); + } + + return ret; +} + +int nsupdate_set_ttl(nsupdate_params_t *params, const uint32_t ttl) +{ + int ret = parser_set_default(params->rrp, "$TTL %u\n", ttl); + if (ret == KNOT_EOK) { + params->ttl = ttl; + } else { + ERR("failed to set default TTL, %s\n", knot_strerror(ret)); + } + return ret; +} + +int nsupdate_set_origin(nsupdate_params_t *params, const char *origin) +{ + char *fqdn = get_fqd_name(origin); + + int ret = parser_set_default(params->rrp, "$ORIGIN %s\n", fqdn); + + free(fqdn); + + if (ret != KNOT_EOK) { + ERR("failed to set default origin, %s\n", knot_strerror(ret)); + } + return ret; +} |