From b62e2bb84c896bc8f9f926f1084872031bf76140 Mon Sep 17 00:00:00 2001
From: Willem Toorop <willem@nlnetlabs.nl>
Date: Thu, 16 Oct 2014 11:28:32 +0200
Subject: [PATCH] Prepare datastructs for tcp stub resolving

---
 src/context.c        | 31 ++++++++++++++++++++++---------
 src/context.h        |  7 +++++--
 src/types-internal.h |  4 ++++
 3 files changed, 31 insertions(+), 11 deletions(-)

diff --git a/src/context.c b/src/context.c
index bce53f5e..726a596b 100755
--- a/src/context.c
+++ b/src/context.c
@@ -338,6 +338,26 @@ upstream_dict(getdns_context *context, struct getdns_upstream *upstream)
 	return r;
 }
 
+static int
+net_req_query_id_cmp(const void *id1, const void *id2)
+{
+	return (int)((struct getdns_network_req *)id1)->query_id -
+	       (int)((struct getdns_network_req *)id1)->query_id;
+}
+
+static void
+upstream_init(struct getdns_upstream *upstream, struct addrinfo *ai)
+{
+	assert(upstream && ai);
+	upstream->addr_len = ai->ai_addrlen;
+	(void) memcpy(&upstream->addr, ai->ai_addr, ai->ai_addrlen);
+	upstream->to_retry =  2;
+	upstream->tcp_fd   = -1;
+	(void) memset(&upstream->tcp_event, 0, sizeof(upstream->tcp_event));
+	getdns_rbtree_init(&upstream->netreq_by_query_id,
+	    net_req_query_id_cmp);
+}
+
 /*---------------------------------------- set_os_defaults
   we use ldns to read the resolv.conf file - the ldns resolver is
   destroyed once the file is read
@@ -440,11 +460,7 @@ set_os_defaults(struct getdns_context *context)
 
 		upstream =   &context->upstreams->
 		    upstreams[context->upstreams->count++];
-		upstream->rtt = 1;
-		upstream->tcp_fd = -1;
-		upstream->addr_len = result->ai_addrlen;
-		(void) memcpy(&upstream->addr,
-		    result->ai_addr, result->ai_addrlen);
+		upstream_init(upstream, result);
 		freeaddrinfo(result);
 	}
 	fclose(in);
@@ -1305,10 +1321,7 @@ getdns_context_set_upstream_recursive_servers(struct getdns_context *context,
 		if (getaddrinfo(addrstr, portstr, &hints, &ai))
 			goto invalid_parameter;
 
-		upstream->rtt = 1;
-		upstream->tcp_fd = -1;
-		upstream->addr_len = ai->ai_addrlen;
-		(void) memcpy(&upstream->addr, ai->ai_addr, ai->ai_addrlen);
+		upstream_init(upstream, ai);
 		upstreams->count++;
 		freeaddrinfo(ai);
 	}
diff --git a/src/context.h b/src/context.h
index 151e774b..31cd3244 100755
--- a/src/context.h
+++ b/src/context.h
@@ -42,6 +42,7 @@
 #include "config.h"
 #include "types-internal.h"
 #include "extension/libmini_event.h"
+#include "util/rbtree.h"
 
 struct getdns_dns_req;
 struct ldns_rbtree_t;
@@ -71,8 +72,10 @@ struct filechg {
 struct getdns_upstream {
 	socklen_t               addr_len;
 	struct sockaddr_storage addr;
-	int tcp_fd;
-	int rtt;
+	int                     to_retry;
+	int                     tcp_fd;
+	getdns_eventloop_event  tcp_event;
+	getdns_rbtree_t         netreq_by_query_id;
 };
 
 typedef struct getdns_upstreams {
diff --git a/src/types-internal.h b/src/types-internal.h
index 415998d6..285eb29c 100644
--- a/src/types-internal.h
+++ b/src/types-internal.h
@@ -40,6 +40,8 @@
 #include <ldns/ldns.h>
 #include "getdns/getdns.h"
 #include "getdns/getdns_extra.h"
+#include "util/rbtree.h"
+
 struct getdns_context;
 struct getdns_upstreams;
 struct getdns_upstream;
@@ -153,6 +155,8 @@ typedef struct getdns_extension_format
  **/
 typedef struct getdns_network_req
 {
+	/* For storage in upstream->netreq_by_query_id */
+	getdns_rbnode_t node;
 	/* the async_id from unbound */
 	int unbound_id;
 	/* state var */