From 304c8fbfbe69f939565e38320fa27c112a1bf63c Mon Sep 17 00:00:00 2001 From: Rafal Szczesniak Date: Sun, 9 Jul 2006 12:57:40 +0000 Subject: r16896: Allow to use both binding and unc string to specify host name being looked up. rafal (This used to be commit e597e9867f50f495a338bed05cee76ff84fd38b6) --- source4/torture/libnet/libnet_lookup.c | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'source4/torture/libnet/libnet_lookup.c') diff --git a/source4/torture/libnet/libnet_lookup.c b/source4/torture/libnet/libnet_lookup.c index 5d347cd049..12f3863afb 100644 --- a/source4/torture/libnet/libnet_lookup.c +++ b/source4/torture/libnet/libnet_lookup.c @@ -23,6 +23,7 @@ #include "lib/cmdline/popt_common.h" #include "libnet/libnet.h" #include "librpc/gen_ndr/nbt.h" +#include "librpc/rpc/dcerpc.h" #include "torture/torture.h" @@ -33,6 +34,8 @@ BOOL torture_lookup(struct torture_context *torture) TALLOC_CTX *mem_ctx; struct libnet_context *ctx; struct libnet_Lookup lookup; + struct dcerpc_binding *bind; + const char *bindstr; const char *address; mem_ctx = talloc_init("test_lookup"); @@ -43,6 +46,14 @@ BOOL torture_lookup(struct torture_context *torture) address = talloc_array(ctx, const char, 16); lookup.in.hostname = lp_parm_string(-1, "torture", "host"); + if (lookup.in.hostname == NULL) { + bindstr = lp_parm_string(-1, "torture", "binding"); + status = dcerpc_parse_binding(mem_ctx, bindstr, &bind); + if (NT_STATUS_IS_OK(status)) { + lookup.in.hostname = bind->host; + } + } + lookup.in.type = NBT_NAME_CLIENT; lookup.in.methods = NULL; lookup.out.address = &address; @@ -70,6 +81,8 @@ BOOL torture_lookup_host(struct torture_context *torture) TALLOC_CTX *mem_ctx; struct libnet_context *ctx; struct libnet_Lookup lookup; + struct dcerpc_binding *bind; + const char *bindstr; const char *address; mem_ctx = talloc_init("test_lookup_host"); @@ -80,6 +93,14 @@ BOOL torture_lookup_host(struct torture_context *torture) address = talloc_array(mem_ctx, const char, 16); lookup.in.hostname = lp_parm_string(-1, "torture", "host"); + if (lookup.in.hostname == NULL) { + bindstr = lp_parm_string(-1, "torture", "binding"); + status = dcerpc_parse_binding(mem_ctx, bindstr, &bind); + if (NT_STATUS_IS_OK(status)) { + lookup.in.hostname = bind->host; + } + } + lookup.in.methods = NULL; lookup.out.address = &address; -- cgit