From e926d64cf529c7053348799973ab7ef63c1eeece Mon Sep 17 00:00:00 2001
From: Lev Walkin <vlm@lionet.info>
Date: Fri, 24 Mar 2017 17:20:02 -0700
Subject: [PATCH] Fix memory leak

---
 ipaddr.c | 44 +++++++++++++++++++++++++++++++++++---------
 1 file changed, 35 insertions(+), 9 deletions(-)

diff --git a/ipaddr.c b/ipaddr.c
index bc960cb1..ec8956bd 100644
--- a/ipaddr.c
+++ b/ipaddr.c
@@ -296,33 +296,55 @@ int ipaddr_remote(struct ipaddr *addr, const char *name, int port, int mode,
             int err = errno;
             fdclean(fd);
             errno = err;
-            if(dill_slow(rc < 0)) return -1;
+            if(dill_slow(rc < 0)) {
+                dns_ai_close(ai);
+                return -1;
+            }
             continue;
         }
         if(rc == ENOENT || (rc >= DNS_EBASE && rc <= DNS_ELAST))
             break;
-        if(!ipv4 && it && it->ai_family == AF_INET)
+        if(!ipv4 && it && it->ai_family == AF_INET) {
             ipv4 = it;
-        if(!ipv6 && it && it->ai_family == AF_INET6)
+            it = NULL;
+        }
+        if(!ipv6 && it && it->ai_family == AF_INET6) {
             ipv6 = it;
+            it = NULL;
+        }
+        if(it) freeaddrinfo(it); /* Ended up useless */
         if(ipv4 && ipv6)
             break;
     }
     switch(mode) {
     case IPADDR_IPV4:
-        ipv6 = NULL;
+        if(ipv6) {
+            freeaddrinfo(ipv6);
+            ipv6 = NULL;
+        }
         break;
     case IPADDR_IPV6:
-        ipv4 = NULL;
+        if(ipv4) {
+            freeaddrinfo(ipv4);
+            ipv4 = NULL;
+        }
         break;
     case 0:
     case IPADDR_PREF_IPV4:
-        if(ipv4)
-           ipv6 = NULL;
+        if(ipv4) {
+            if(ipv6) {
+                freeaddrinfo(ipv6);
+                ipv6 = NULL;
+            }
+        }
         break;
     case IPADDR_PREF_IPV6:
-        if(ipv6)
-           ipv4 = NULL;
+        if(ipv6) {
+            if(ipv4) {
+                freeaddrinfo(ipv4);
+                ipv4 = NULL;
+            }
+        }
         break;
     default:
         dill_assert(0);
@@ -331,6 +353,8 @@ int ipaddr_remote(struct ipaddr *addr, const char *name, int port, int mode,
         struct sockaddr_in *inaddr = (struct sockaddr_in*)addr;
         memcpy(inaddr, ipv4->ai_addr, sizeof (struct sockaddr_in));
         inaddr->sin_port = htons(port);
+        if(ipv4) freeaddrinfo(ipv4);
+        if(ipv6) freeaddrinfo(ipv6);
         dns_ai_close(ai);
         return 0;
     }
@@ -338,6 +362,8 @@ int ipaddr_remote(struct ipaddr *addr, const char *name, int port, int mode,
         struct sockaddr_in6 *inaddr = (struct sockaddr_in6*)addr;
         memcpy(inaddr, ipv6->ai_addr, sizeof (struct sockaddr_in6));
         inaddr->sin6_port = htons(port);
+        if(ipv4) freeaddrinfo(ipv4);
+        if(ipv6) freeaddrinfo(ipv6);
         dns_ai_close(ai);
         return 0;
     }