X-Git-Url: https://jfr.im/git/irc/rqf/shadowircd.git/blobdiff_plain/1728a2acd0cc89dddb27ea92c4ff304e2f6b7a09..40c6b59b6340a6fc459020c64f5ee35a842a02c7:/src/res.c

diff --git a/src/res.c b/src/res.c
index 81e23c0..06585ab 100644
--- a/src/res.c
+++ b/src/res.c
@@ -60,14 +60,14 @@ struct reslist
 {
 	rb_dlink_node node;
 	int id;
-	int sent;		/* number of requests sent */
 	time_t ttl;
 	char type;
-	char queryname[128];	/* name currently being queried */
+	char queryname[IRCD_RES_HOSTLEN + 1]; /* name currently being queried */
 	char retries;		/* retry counter */
 	char sends;		/* number of sends (>1 means resent) */
 	time_t sentat;
 	time_t timeout;
+	unsigned int lastns;	/* index of last server sent to */
 	struct rb_sockaddr_storage addr;
 	char *name;
 	struct DNSQuery *query;	/* query callback for this request */
@@ -75,6 +75,7 @@ struct reslist
 
 static rb_fde_t *res_fd;
 static rb_dlink_list request_list = { NULL, NULL, 0 };
+static int ns_timeout_count[IRCD_MAXNS];
 
 static void rem_request(struct reslist *request);
 static struct reslist *make_request(struct DNSQuery *query);
@@ -131,7 +132,10 @@ static int res_ourserver(const struct rb_sockaddr_storage *inp)
 						sizeof(struct in6_addr)) == 0) ||
 					      (memcmp(&v6->sin6_addr.s6_addr, &in6addr_any,
 						sizeof(struct in6_addr)) == 0))
+					  {
+						  ns_timeout_count[ns] = 0;
 						  return 1;
+					  }
 			  break;
 #endif
 		  case AF_INET:
@@ -139,7 +143,10 @@ static int res_ourserver(const struct rb_sockaddr_storage *inp)
 				  if (v4->sin_port == v4in->sin_port)
 					  if ((v4->sin_addr.s_addr == INADDR_ANY)
 					      || (v4->sin_addr.s_addr == v4in->sin_addr.s_addr))
+					  {
+						  ns_timeout_count[ns] = 0;
 						  return 1;
+					  }
 			  break;
 		  default:
 			  break;
@@ -176,6 +183,7 @@ static time_t timeout_query_list(time_t now)
 			}
 			else
 			{
+				ns_timeout_count[request->lastns]++;
 				request->sentat = now;
 				request->timeout += request->timeout;
 				resend_query(request);
@@ -207,7 +215,11 @@ static struct ev_entry *timeout_resolver_ev = NULL;
  */
 static void start_resolver(void)
 {
+	int i;
+
 	irc_res_init();
+	for (i = 0; i < irc_nscount; i++)
+		ns_timeout_count[i] = 0;
 
 	if (res_fd == NULL)
 	{
@@ -315,33 +327,67 @@ void delete_resolver_queries(const struct DNSQuery *query)
 }
 
 /*
- * send_res_msg - sends msg to all nameservers found in the "_res" structure.
- * This should reflect /etc/resolv.conf. We will get responses
- * which arent needed but is easier than checking to see if nameserver
- * isnt present. Returns number of messages successfully sent to 
- * nameservers or -1 if no successful sends.
+ * retryfreq - determine how many queries to wait before resending
+ * if there have been that many consecutive timeouts
+ */
+static int retryfreq(int timeouts)
+{
+	switch (timeouts)
+	{
+		case 1:
+			return 3;
+		case 2:
+			return 9;
+		case 3:
+			return 27;
+		case 4:
+			return 81;
+		default:
+			return 243;
+	}
+}
+
+/*
+ * send_res_msg - sends msg to a nameserver.
+ * This should reflect /etc/resolv.conf.
+ * Returns number of nameserver successfully sent to 
+ * or -1 if no successful sends.
  */
 static int send_res_msg(const char *msg, int len, int rcount)
 {
 	int i;
-	int sent = 0;
-	int max_queries = IRCD_MIN(irc_nscount, rcount);
+	int ns;
+	static int retrycnt;
 
-	/* RES_PRIMARY option is not implemented
-	 * if (res.options & RES_PRIMARY || 0 == max_queries)
+	retrycnt++;
+	/* First try a nameserver that seems to work.
+	 * Every once in a while, try a possibly broken one to check
+	 * if it is working again.
 	 */
-	if (max_queries == 0)
-		max_queries = 1;
+	for (i = 0; i < irc_nscount; i++)
+	{
+		ns = (i + rcount - 1) % irc_nscount;
+		if (ns_timeout_count[ns] && retrycnt % retryfreq(ns_timeout_count[ns]))
+			continue;
+		if (sendto(rb_get_fd(res_fd), msg, len, 0,
+		     (struct sockaddr *)&(irc_nsaddr_list[ns]), 
+				GET_SS_LEN(&irc_nsaddr_list[ns])) == len)
+			return ns;
+	}
 
-	for (i = 0; sent < max_queries && i < irc_nscount; i++)
+	/* No known working nameservers, try some broken one. */
+	for (i = 0; i < irc_nscount; i++)
 	{
+		ns = (i + rcount - 1) % irc_nscount;
+		if (!ns_timeout_count[ns])
+			continue;
 		if (sendto(rb_get_fd(res_fd), msg, len, 0,
-		     (struct sockaddr *)&(irc_nsaddr_list[i]), 
-				GET_SS_LEN(&irc_nsaddr_list[i])) == len)
-			++sent;
+		     (struct sockaddr *)&(irc_nsaddr_list[ns]), 
+				GET_SS_LEN(&irc_nsaddr_list[ns])) == len)
+			return ns;
 	}
 
-	return (sent);
+	return -1;
 }
 
 /*
@@ -387,10 +433,10 @@ void gethost_byaddr(const struct rb_sockaddr_storage *addr, struct DNSQuery *que
 static void do_query_name(struct DNSQuery *query, const char *name, struct reslist *request,
 			  int type)
 {
-	char host_name[HOSTLEN + 1];
+	char host_name[IRCD_RES_HOSTLEN + 1];
 
-	rb_strlcpy(host_name, name, HOSTLEN + 1);
-	add_local_domain(host_name, HOSTLEN);
+	rb_strlcpy(host_name, name, IRCD_RES_HOSTLEN + 1);
+	add_local_domain(host_name, IRCD_RES_HOSTLEN);
 
 	if (request == NULL)
 	{
@@ -416,7 +462,7 @@ static void do_query_number(struct DNSQuery *query, const struct rb_sockaddr_sto
 	{
 		request = make_request(query);
 		memcpy(&request->addr, addr, sizeof(struct rb_sockaddr_storage));
-		request->name = (char *)rb_malloc(HOSTLEN + 1);
+		request->name = (char *)rb_malloc(IRCD_RES_HOSTLEN + 1);
 	}
 
 	if (addr->ss_family == AF_INET)
@@ -465,6 +511,7 @@ static void query_name(struct reslist *request)
 {
 	char buf[MAXPACKET];
 	int request_len = 0;
+	int ns;
 
 	memset(buf, 0, sizeof(buf));
 
@@ -498,7 +545,9 @@ static void query_name(struct reslist *request)
 		request->id = header->id;
 		++request->sends;
 
-		request->sent += send_res_msg(buf, request_len, request->sends);
+		ns = send_res_msg(buf, request_len, request->sends);
+		if (ns != -1)
+			request->lastns = ns;
 	}
 }
 
@@ -527,7 +576,7 @@ static void resend_query(struct reslist *request)
  */
 static int check_question(struct reslist *request, HEADER * header, char *buf, char *eob)
 {
-	char hostbuf[128];	/* working buffer */
+	char hostbuf[IRCD_RES_HOSTLEN + 1];	/* working buffer */
 	unsigned char *current;	/* current position in buf */
 	int n;			/* temp count */
 
@@ -548,7 +597,7 @@ static int check_question(struct reslist *request, HEADER * header, char *buf, c
  */
 static int proc_answer(struct reslist *request, HEADER * header, char *buf, char *eob)
 {
-	char hostbuf[HOSTLEN + 100];	/* working buffer */
+	char hostbuf[IRCD_RES_HOSTLEN + 100];	/* working buffer */
 	unsigned char *current;	/* current position in buf */
 	int query_class;	/* answer class */
 	int type;		/* answer type */
@@ -593,7 +642,7 @@ static int proc_answer(struct reslist *request, HEADER * header, char *buf, char
 			return (0);
 		}
 
-		hostbuf[HOSTLEN] = '\0';
+		hostbuf[IRCD_RES_HOSTLEN] = '\0';
 
 		/* With Address arithmetic you have to be very anal
 		 * this code was not working on alpha due to that
@@ -659,7 +708,7 @@ static int proc_answer(struct reslist *request, HEADER * header, char *buf, char
 			  else if (n == 0)
 				  return (0);	/* no more answers left */
 
-			  rb_strlcpy(request->name, hostbuf, HOSTLEN + 1);
+			  rb_strlcpy(request->name, hostbuf, IRCD_RES_HOSTLEN + 1);
 
 			  return (1);
 			  break;
@@ -682,9 +731,10 @@ static int proc_answer(struct reslist *request, HEADER * header, char *buf, char
 }
 
 /*
- * res_readreply - read a dns reply from the nameserver and process it.
+ * res_read_single_reply - read a dns reply from the nameserver and process it.
+ * Return value: 1 if a packet was read, 0 otherwise
  */
-static void res_readreply(rb_fde_t *F, void *data)
+static int res_read_single_reply(rb_fde_t *F, void *data)
 {
 	char buf[sizeof(HEADER) + MAXPACKET]
 		/* Sparc and alpha need 16bit-alignment for accessing header->id 
@@ -705,13 +755,13 @@ static void res_readreply(rb_fde_t *F, void *data)
 
 	rc = recvfrom(rb_get_fd(F), buf, sizeof(buf), 0, (struct sockaddr *)&lsin, &len);
 
-	/* Re-schedule a read *after* recvfrom, or we'll be registering
-	 * interest where it'll instantly be ready for read :-) -- adrian
-	 */
-	rb_setselect(F, RB_SELECT_READ, res_readreply, NULL);
-	/* Better to cast the sizeof instead of rc */
+	/* No packet */
+	if (rc == 0 || rc == -1)
+		return 0;
+
+	/* Too small */
 	if (rc <= (int)(sizeof(HEADER)))
-		return;
+		return 1;
 
 	/*
 	 * convert DNS reply reader from Network byte order to CPU byte order.
@@ -727,16 +777,16 @@ static void res_readreply(rb_fde_t *F, void *data)
 	 * just ignore this response.
 	 */
 	if (0 == (request = find_id(header->id)))
-		return;
+		return 1;
 
 	/*
 	 * check against possibly fake replies
 	 */
 	if (!res_ourserver(&lsin))
-		return;
+		return 1;
 
 	if (!check_question(request, header, buf, buf + rc))
-		return;
+		return 1;
 
 	if ((header->rcode != NO_ERRORS) || (header->ancount == 0))
 	{
@@ -754,7 +804,7 @@ static void res_readreply(rb_fde_t *F, void *data)
 			(*request->query->callback) (request->query->ptr, NULL);
 			rem_request(request);
 		}
-		return;
+		return 1;
 	}
 	/*
 	 * If this fails there was an error decoding the received packet, 
@@ -774,7 +824,7 @@ static void res_readreply(rb_fde_t *F, void *data)
 				 */
 				(*request->query->callback) (request->query->ptr, reply);
 				rem_request(request);
-				return;
+				return 1;
 			}
 
 			/*
@@ -807,6 +857,14 @@ static void res_readreply(rb_fde_t *F, void *data)
 		(*request->query->callback) (request->query->ptr, NULL);
 		rem_request(request);
 	}
+	return 1;
+}
+
+static void res_readreply(rb_fde_t *F, void *data)
+{
+	while (res_read_single_reply(F, data))
+		;
+	rb_setselect(F, RB_SELECT_READ, res_readreply, NULL);
 }
 
 static struct DNSReply *make_dnsreply(struct reslist *request)
@@ -832,6 +890,6 @@ void report_dns_servers(struct Client *source_p)
 				ipaddr, sizeof ipaddr))
 			rb_strlcpy(ipaddr, "?", sizeof ipaddr);
 		sendto_one_numeric(source_p, RPL_STATSDEBUG,
-				"A %s", ipaddr);
+				"A %s %d", ipaddr, ns_timeout_count[i]);
 	}
 }