diff options
-rw-r--r-- | Makefile | 4 | ||||
-rw-r--r-- | src/ask_resolver.c (renamed from src/ask_localhost.c) | 18 |
2 files changed, 12 insertions, 10 deletions
@@ -7,7 +7,7 @@ CFLAGS = -std=c99 -Wall -Werror -I include build/%.o : src/%.c | build gcc $(CFLAGS) $^ -c -o $@ -ask_localhost : build/ask_localhost.o +ask_resolver : build/ask_resolver.o $(CC) $^ -lunbound -o $@ build : @@ -16,6 +16,6 @@ build : all : 0tDNS receive_respond clean : - -rm -r build 0tDNS ask_localhost + -rm -r build 0tDNS ask_resolver .PHONY : clean diff --git a/src/ask_localhost.c b/src/ask_resolver.c index 49845c3..dc7c660 100644 --- a/src/ask_localhost.c +++ b/src/ask_resolver.c @@ -74,7 +74,8 @@ void examine_result(const char *query, struct ub_result *result) printf("result has %d data element(s)\n", num); } -struct ub_ctx *create_ub_context(int debuglevel) { +struct ub_ctx *create_ub_context(char *forwarder_addr, int debuglevel) +{ int rc; struct ub_ctx* ctx; char *error_message_format; @@ -86,7 +87,7 @@ struct ub_ctx *create_ub_context(int debuglevel) { } error_message_format = "Couldn't set forwarder: %s\n"; - rc = ub_ctx_set_fwd(ctx, "127.0.0.1"); + rc = ub_ctx_set_fwd(ctx, forwarder_addr); if (rc) goto out; @@ -103,7 +104,8 @@ out: return ctx; } -void ztdns_try_resolve(struct ub_ctx *ctx, const char *name) { +void ztdns_try_resolve(struct ub_ctx *ctx, const char *name) +{ struct ub_result* result; int rc; rc = ub_resolve(ctx, name, @@ -120,17 +122,17 @@ void ztdns_try_resolve(struct ub_ctx *ctx, const char *name) { int main(int argc, char** argv) { struct ub_ctx *ctx; - - if (argc < 2) { - printf("Usage: %s DOMAINNAME\n", argv[0]); + + if (argc != 3) { + printf("Usage: %s DNS_SERVER DOMAINNAME\n", argv[0]); return EXIT_FAILURE; } - ctx = create_ub_context(3); + ctx = create_ub_context(argv[1], 3); if (!ctx) return EXIT_FAILURE; - ztdns_try_resolve(ctx, argv[1]); + ztdns_try_resolve(ctx, argv[2]); ub_ctx_delete(ctx); |