mirror of
https://github.com/NLnetLabs/unbound.git
synced 2024-09-21 14:47:09 +00:00
108 lines
2.6 KiB
C
108 lines
2.6 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#include <stdint.h>
|
|
#include "config.h"
|
|
#include "sldns/str2wire.h"
|
|
#include "sldns/wire2str.h"
|
|
|
|
int print_usage(FILE *out, const char *progname)
|
|
{
|
|
fprintf(out, "usage: %s [ -u ] <zonefile> [<origin>]\n", progname);
|
|
fprintf(out, "\t-u\tprint in unknown type (RFC3597) format\n");
|
|
return out == stdout ? EXIT_SUCCESS : EXIT_FAILURE;
|
|
}
|
|
|
|
int main(int argc, char *const *argv)
|
|
{
|
|
char *progname = argv[0];
|
|
uint8_t rr[LDNS_RR_BUF_SIZE];
|
|
char *str = malloc(1024 * 1024);
|
|
size_t str_len = sizeof(str);
|
|
struct sldns_file_parse_state state;
|
|
FILE *in = NULL;
|
|
int s = -1;
|
|
int opt;
|
|
int print_in_unknown_type_format = 0;
|
|
|
|
while ((opt = getopt(argc, argv, "hu")) != -1) {
|
|
switch (opt) {
|
|
case 'h':
|
|
return print_usage(stdout, progname);
|
|
case 'u':
|
|
print_in_unknown_type_format = 1;
|
|
break;
|
|
default:
|
|
return print_usage(stderr, progname);
|
|
}
|
|
}
|
|
argc -= optind;
|
|
argv += optind;
|
|
|
|
memset(&state, 0, sizeof(state));
|
|
state.default_ttl = 3600;
|
|
state.lineno = 1;
|
|
if (argc == 2) {
|
|
state.origin_len = sizeof(state.origin);
|
|
s = sldns_str2wire_dname_buf(argv[1], state.origin
|
|
, &state.origin_len);
|
|
if (s) {
|
|
fprintf(stderr, "Error parsing origin: %s\n"
|
|
, sldns_get_errorstr_parse(s));
|
|
return EXIT_FAILURE;
|
|
}
|
|
s = -1;
|
|
}
|
|
if (!str)
|
|
fprintf(stderr, "Memory allocation error: %s\n"
|
|
, strerror(errno));
|
|
|
|
else if (argc != 1 && argc != 2)
|
|
return print_usage(stderr, progname);
|
|
|
|
else if (!(in = fopen(argv[0], "r")))
|
|
fprintf(stderr, "Error opening \"%s\": %s\n"
|
|
, argv[0], strerror(errno));
|
|
else while (!feof(in)) {
|
|
size_t rr_len = sizeof(rr), dname_len = 0;
|
|
size_t written;
|
|
|
|
s = sldns_fp2wire_rr_buf(in, rr, &rr_len, &dname_len, &state);
|
|
if (s) {
|
|
fprintf( stderr, "parse error %d:%d: %s"
|
|
, state.lineno, LDNS_WIREPARSE_OFFSET(s)
|
|
, sldns_get_errorstr_parse(s));
|
|
break;
|
|
}
|
|
if (rr_len == 0)
|
|
continue;
|
|
|
|
if (print_in_unknown_type_format)
|
|
written = sldns_wire2str_rr_unknown_buf(
|
|
rr, rr_len, str, str_len);
|
|
else
|
|
written = sldns_wire2str_rr_buf(
|
|
rr, rr_len, str, str_len);
|
|
|
|
if (written > str_len) {
|
|
while (written > str_len)
|
|
str_len *= 2;
|
|
free(str);
|
|
if (!(str = malloc(str_len))) {
|
|
fprintf(stderr, "Memory allocation error: %s\n"
|
|
, strerror(errno));
|
|
s = -1;
|
|
break;
|
|
}
|
|
(void) sldns_wire2str_rr_buf(rr, rr_len, str, str_len);
|
|
}
|
|
fprintf(stdout, "%s", str);
|
|
}
|
|
if (in)
|
|
fclose(in);
|
|
return !in || s ? EXIT_FAILURE : EXIT_SUCCESS;
|
|
}
|