diff options
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/samba-dig.c | 160 | ||||
| -rw-r--r-- | utils/wscript_build | 7 |
2 files changed, 167 insertions, 0 deletions
diff --git a/utils/samba-dig.c b/utils/samba-dig.c new file mode 100644 index 0000000000..3fd2ecffc4 --- /dev/null +++ b/utils/samba-dig.c @@ -0,0 +1,160 @@ +/* + Unix SMB/CIFS implementation. + + DNS query too for Samba with socketwrapper support + + Copyright (C) 2012 Kai Blin <kai@samba.org> + + 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 "includes.h" +#include <talloc.h> +#include <tevent.h> +#include "lib/util/samba_util.h" +#include "librpc/ndr/libndr.h" +#include "librpc/gen_ndr/ndr_dns.h" +#include "libcli/dns/libdns.h" + +static void usage(void) +{ + printf("Usage: samba-dig <dns-server-ip> <data> <record-type>\n\n"); +} + +static struct dns_name_packet *make_name_packet(TALLOC_CTX *mem_ctx, + uint16_t operation) +{ + struct dns_name_packet *packet = talloc_zero(mem_ctx, + struct dns_name_packet); + if (packet == NULL) { + return NULL; + } + + packet->id = random(); + packet->operation |= operation; + + return packet; +} + +#define QTYPE_MAP(type) if (strncmp(type_string, #type , strlen( #type )) == 0) \ + return DNS_QTYPE_ ## type ; + +static enum dns_qtype parse_qtype(const char *type_string) +{ + QTYPE_MAP(AAAA); + QTYPE_MAP(A); + QTYPE_MAP(SOA); + QTYPE_MAP(PTR); + return -1; +} +#undef QTYPE_MAP + +static struct dns_name_question *make_question(TALLOC_CTX *mem_ctx, + const char *name, + enum dns_qtype type) +{ + struct dns_name_question *question = talloc(mem_ctx, + struct dns_name_question); + if (question == NULL) { + return NULL; + } + + question->name = talloc_strdup(question, name); + question->question_type = type; + question->question_class = DNS_QCLASS_IN; + + return question; +} + +int main(int argc, char **argv) +{ + TALLOC_CTX *mem_ctx = talloc_init("samba-dig"); + struct tevent_context *ev; + struct dns_name_packet *dns_packet, *in_packet; + struct dns_name_question *question; + enum dns_qtype type; + enum ndr_err_code ndr_err; + struct tevent_req *req; + WERROR w_err; + DATA_BLOB out, in; + int ret = 0; + + if (argc < 4) { + usage(); + exit(1); + } + + ev = tevent_context_init(mem_ctx); + setup_logging("samba-dig", DEBUG_STDERR); + debug_parse_levels("1"); + + DEBUG(1,("Querying %s for %s %s\n", argv[1], argv[2], argv[3])); + + dns_packet = make_name_packet(mem_ctx, DNS_OPCODE_QUERY); + + type = parse_qtype(argv[3]); + if (type == -1) { + DEBUG(0, ("Invalid DNS_QTYPE %s\n", argv[3])); + ret = 1; + goto error; + } + + question = make_question(dns_packet, argv[2], type); + + dns_packet->qdcount = 1; + dns_packet->questions = question; + NDR_PRINT_DEBUG(dns_name_packet, dns_packet); + + ndr_err = ndr_push_struct_blob(&out, mem_ctx, dns_packet, + (ndr_push_flags_fn_t)ndr_push_dns_name_packet); + if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) { + DEBUG(0, ("Failed to marshall dns_name_packet: %d\n", ndr_err)); + ret = 1; + goto error; + } + + req = dns_udp_request_send(mem_ctx, ev, argv[1], out.data, out.length); + if (req == NULL) { + DEBUG(0, ("Failed to allocate memory for tevent_req\n")); + ret = 1; + goto error; + } + if (!tevent_req_poll(req, ev)) { + DEBUG(0, ("Error sending dns request\n")); + ret = 1; + goto error; + } + w_err = dns_udp_request_recv(req, mem_ctx, &in.data, &in.length); + if (!W_ERROR_IS_OK(w_err)) { + DEBUG(0, ("Error receiving dns request: %s\n", win_errstr(w_err))); + ret = 1; + goto error; + } + + in_packet = talloc(mem_ctx, struct dns_name_packet); + + ndr_err = ndr_pull_struct_blob(&in, in_packet, in_packet, + (ndr_pull_flags_fn_t)ndr_pull_dns_name_packet); + if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) { + DEBUG(0, ("Failed to unmarshall dns_name_packet: %d\n", ndr_err)); + ret = 1; + goto error; + } + + NDR_PRINT_DEBUG(dns_name_packet, in_packet); + +error: + talloc_free(mem_ctx); + return ret; +} diff --git a/utils/wscript_build b/utils/wscript_build new file mode 100644 index 0000000000..3c22ad27ae --- /dev/null +++ b/utils/wscript_build @@ -0,0 +1,7 @@ +#!/usr/bin/env python + +bld.SAMBA_BINARY('samba-dig', + source='samba-dig.c', + deps='samba-util NDR_DNS tevent LIBCLI_DNS' + ) + |