From: Andrew Tridgell <tridge@samba.org>
Date: Mon, 13 Sep 2010 13:08:28 +0000 (+1000)
Subject: s4-resolve: added resolve_name_multiple_recv()
X-Git-Url: http://git.samba.org/?a=commitdiff_plain;h=dbc9b185afa2fe59f73526b081ef6acff7d61aaa;p=metze%2Fsamba%2Fwip.git

s4-resolve: added resolve_name_multiple_recv()

this allows for multiple replies to a SRV lookup

Pair-Programmed-With: Andrew Bartlett <abartlet@samba.org>
---

diff --git a/source4/libcli/resolve/resolve.c b/source4/libcli/resolve/resolve.c
index d9352b4226e2..c17e93a1f65a 100644
--- a/source4/libcli/resolve/resolve.c
+++ b/source4/libcli/resolve/resolve.c
@@ -266,6 +266,40 @@ NTSTATUS resolve_name_recv(struct composite_context *c,
 	return status;
 }
 
+/*
+  receive multiple responses from resolve_name_send()
+ */
+NTSTATUS resolve_name_multiple_recv(struct composite_context *c,
+				    TALLOC_CTX *mem_ctx,
+				    const char ***reply_addrs)
+{
+	NTSTATUS status;
+	struct socket_address **addrs = NULL;
+	int i;
+
+	status = resolve_name_all_recv(c, mem_ctx, &addrs, NULL);
+	NT_STATUS_NOT_OK_RETURN(status);
+
+	/* count the addresses */
+	for (i=0; addrs[i]; i++) ;
+
+	*reply_addrs = talloc_array(mem_ctx, const char *, i+1);
+	NT_STATUS_HAVE_NO_MEMORY(*reply_addrs);
+
+	for (i=0; addrs[i]; i++) {
+		struct tsocket_address *t_addr = socket_address_to_tsocket_address(addrs, addrs[i]);
+		NT_STATUS_HAVE_NO_MEMORY(t_addr);
+
+		(*reply_addrs)[i] = tsocket_address_inet_addr_string(t_addr, *reply_addrs);
+		NT_STATUS_HAVE_NO_MEMORY((*reply_addrs)[i]);
+	}
+	(*reply_addrs)[i] = NULL;
+
+	talloc_free(addrs);
+
+	return status;
+}
+
 /*
   general name resolution - sync call
  */