[Asterisk-code-review] dns_core: Create new API ast_dns_resolve_ipv6_and_ipv4 (...asterisk[16])

[George Joseph]

George Joseph has submitted this change and it was merged. ( https://gerrit.asterisk.org/c/asterisk/+/12766 )

Change subject: dns_core:  Create new API ast_dns_resolve_ipv6_and_ipv4
......................................................................

dns_core:  Create new API ast_dns_resolve_ipv6_and_ipv4

The new function takes in a pointer to an ast_sockaddr structure,
a hostname and an optional port and then dispatches parallel
"AAAA" and "A" record queries.  If an "AAAA" record is returned,
it's parsed into the ast_sockaddr structure along with the port
if it was supplied.  If no "AAAA" record was returned, the
first "A" record returned (if any) is parsed instead.

This is a synchronous call.  If you need asynchronous lookups,
use ast_dns_query_set_resolve_async and roll your own.

Change-Id: I194b0b0e73da94b35cc35263a868ffac3a8d0a95
---
M include/asterisk/dns_core.h
M main/dns_core.c
2 files changed, 94 insertions(+), 0 deletions(-)

Approvals:
  Joshua Colp: Looks good to me, but someone else must approve
  Benjamin Keith Ford: Looks good to me, but someone else must approve
  George Joseph: Looks good to me, approved; Approved for Submit



diff --git a/include/asterisk/dns_core.h b/include/asterisk/dns_core.h
index fe67e34..d74b012 100644
--- a/include/asterisk/dns_core.h
+++ b/include/asterisk/dns_core.h
@@ -28,6 +28,8 @@
 extern "C" {
 #endif
 
+#include "asterisk/netsock2.h"
+
 /*! \brief Opaque structure for an active DNS query */
 struct ast_dns_query_active;
 
@@ -269,6 +271,26 @@
  */
 int ast_dns_resolve(const char *name, int rr_type, int rr_class, struct ast_dns_result **result);
 
+/*!
+ * \brief Synchronously resolves host to  an AAAA or A record
+ * \since 16.6.0
+ *
+ * \param address A pointer to an ast_sockaddr structure to receive the IPv6 or IPv4 address
+ * \param host The hostname to resolve
+ * \param port (optional) A port to parse into the final ast_sockaddr structure
+ *
+ * \retval 0 success - query was completed and result is available
+ * \retval -1 failure
+ *
+ * \note This function makes parallel queries for both AAAA and A records for the host.
+ * 		 The first returned AAAA record (if any) is used and if not found, the first A record
+ * 		 is used.
+ *
+ * \warning This function is synchronous and will block until records are returned or an error
+ *          occurrs.
+ */
+int ast_dns_resolve_ipv6_and_ipv4(struct ast_sockaddr *address, const char *host, const char *port);
+
 #if defined(__cplusplus) || defined(c_plusplus)
 }
 #endif
diff --git a/main/dns_core.c b/main/dns_core.c
index 6f37a5d..f262d7d 100644
--- a/main/dns_core.c
+++ b/main/dns_core.c
@@ -40,6 +40,7 @@
 #include "asterisk/dns_recurring.h"
 #include "asterisk/dns_resolver.h"
 #include "asterisk/dns_internal.h"
+#include "asterisk/netsock2.h"
 
 #include <netinet/in.h>
 #include <arpa/nameser.h>
@@ -365,6 +366,77 @@
 	return *result ? 0 : -1;
 }
 
+int ast_dns_resolve_ipv6_and_ipv4(struct ast_sockaddr *address, const char *host, const char *port)
+{
+	RAII_VAR(struct ast_dns_query_set *, queries, ast_dns_query_set_create(), ao2_cleanup);
+	int i;
+	int rc;
+
+	if (!queries) {
+		ast_log(LOG_ERROR, "Couldn't allocate DNS query structure\n");
+		return -1;
+	}
+	rc = ast_dns_query_set_add(queries, host, ns_t_aaaa, ns_c_in);
+	if (rc != 0) {
+		ast_log(LOG_ERROR, "Couldn't add 'AAAA' DNS query for '%s'\n", host);
+		return -1;
+	}
+	rc = ast_dns_query_set_add(queries, host, ns_t_a, ns_c_in);
+	if (rc != 0) {
+		ast_log(LOG_ERROR, "Couldn't add 'A' DNS query for '%s'\n", host);
+		return -1;
+	}
+	rc = ast_query_set_resolve(queries);
+	if (rc != 0) {
+		ast_log(LOG_ERROR, "Query set resolve failure for '%s'\n", host);
+		return -1;
+	}
+	for (i = 0; i < ast_dns_query_set_num_queries(queries); ++i) {
+		struct ast_dns_query *query = ast_dns_query_set_get(queries, i);
+		struct ast_dns_result *result = ast_dns_query_get_result(query);
+		const struct ast_dns_record *record;
+		in_port_t in_port = 0;
+
+		if (!ast_strlen_zero(port)) {
+			in_port = htons(atoi(port));
+		}
+
+		for (record = ast_dns_result_get_records(result); record; record = ast_dns_record_get_next(record)) {
+			size_t data_size = ast_dns_record_get_data_size(record);
+			const unsigned char *data = (unsigned char *)ast_dns_record_get_data(record);
+			int rr_type = ast_dns_record_get_rr_type(record);
+
+			if (rr_type == ns_t_aaaa && data_size == 16) {
+				struct sockaddr_in6 sin6 = { 0, };
+
+				sin6.sin6_port = in_port;
+				memcpy(&sin6.sin6_addr, data, data_size);
+				sin6.sin6_family = AF_INET6;
+				memcpy(&address->ss, &sin6, sizeof(sin6));
+				address->len = sizeof(sin6);
+
+				return 0;
+			} else if (rr_type == ns_t_a && data_size == 4) {
+				struct sockaddr_in sin4 = { 0, };
+
+				sin4.sin_port = in_port;
+				memcpy(&sin4.sin_addr, data, data_size);
+				sin4.sin_family = AF_INET;
+				memcpy(&address->ss, &sin4, sizeof(sin4));
+				address->len = sizeof(sin4);
+
+				return 0;
+			} else {
+				ast_debug(3, "Unrecognized rr_type '%u' or data_size '%zu' from DNS query for host '%s'\n",
+					rr_type, data_size, host);
+				continue;
+			}
+		}
+	}
+
+	return -1;
+}
+
 int ast_dns_resolver_set_data(struct ast_dns_query *query, void *data)
 {
 	if (query->resolver_data) {

-- 
To view, visit https://gerrit.asterisk.org/c/asterisk/+/12766
To unsubscribe, or for help writing mail filters, visit https://gerrit.asterisk.org/settings

Gerrit-Project: asterisk
Gerrit-Branch: 16
Gerrit-Change-Id: I194b0b0e73da94b35cc35263a868ffac3a8d0a95
Gerrit-Change-Number: 12766
Gerrit-PatchSet: 5
Gerrit-Owner: George Joseph <gjoseph at digium.com>
Gerrit-Reviewer: Benjamin Keith Ford <bford at digium.com>
Gerrit-Reviewer: Friendly Automation
Gerrit-Reviewer: George Joseph <gjoseph at digium.com>
Gerrit-Reviewer: Joshua Colp <jcolp at digium.com>
Gerrit-MessageType: merged
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://lists.digium.com/pipermail/asterisk-code-review/attachments/20190822/d0f8fb97/attachment-0001.html>