From 8d74433d9b8da1fc508f03ad2eb27fd8ea3ba504 Mon Sep 17 00:00:00 2001
From: PiBa-NL <pba_2k3@yahoo.com>
Date: Wed, 24 Jun 2015 21:12:47 +0200
Subject: haproxy-devel, fix dns resolving dns_get_record does not properly
 resolve if not all requested types are present..

---
 config/haproxy-devel/pkg/haproxy_utils.inc | 41 +++++++++++++++++-------------
 1 file changed, 23 insertions(+), 18 deletions(-)

(limited to 'config/haproxy-devel/pkg')

diff --git a/config/haproxy-devel/pkg/haproxy_utils.inc b/config/haproxy-devel/pkg/haproxy_utils.inc
index 3d841a25..ec72b986 100644
--- a/config/haproxy-devel/pkg/haproxy_utils.inc
+++ b/config/haproxy-devel/pkg/haproxy_utils.inc
@@ -39,32 +39,37 @@ class haproxy_utils {
 	public function query_dns($host, $querytype="A,AAAA") {
 		$result = array();
 		$types = explode(',',$querytype);
-		$recordtypes = 0;
+		$recordtype = 0;
 		foreach($types as $type){
 			switch ($type) {
 				case 'A': 
-					$recordtypes += DNS_A; 
+					$recordtype = DNS_A; 
 					break;
 				case 'AAAA': 
-					$recordtypes += DNS_AAAA; 
+					$recordtype = DNS_AAAA; 
 					break;
 			}
-		}
-		if ($recordtypes == 0)
-			return $result;
-		
-		$dnsresult = dns_get_record($host, $recordtypes);
-		foreach($dnsresult as $item) {
-			$newitem["typeid"] = $item['type'];
-			switch ($item['type']) {
-				case 'A':
-					$newitem["data"] = $item['ip']; 
-					break;
-				case 'AAAA':
-					$newitem["data"] = $item['ipv6']; 
-					break;
+			if ($recordtype != 0) {
+				//query one type at a time, querying multiple types in one call dns_get_record fails if one is not present..
+				$errreporting = error_reporting();
+				error_reporting($errreporting & ~E_WARNING);// dns_get_record throws a warning if nothing is resolved..
+				$dnsresult = dns_get_record($host, $recordtype);
+				error_reporting($errreporting);
+				if (is_array($dnsresult)) {
+					foreach($dnsresult as $item) {
+						$newitem["typeid"] = $item['type'];
+						switch ($item['type']) {
+							case 'A':
+								$newitem["data"] = $item['ip']; 
+								break;
+							case 'AAAA':
+								$newitem["data"] = $item['ipv6']; 
+								break;
+						}
+						$result[] = $newitem;
+					}
+				}
 			}
-			$result[] = $newitem;
 		}
 		return $result;
 	}
-- 
cgit v1.2.3