From f31b113e7e85bf7057cb88a70d88c5ce9325e208 Mon Sep 17 00:00:00 2001
From: Aaron Marcher <me@drkhsh.at>
Date: Sat, 19 May 2018 13:34:18 +0200
Subject: Simplify ipv* functions

---
 components/ip.c | 45 ++++++++++++---------------------------------
 1 file changed, 12 insertions(+), 33 deletions(-)

(limited to 'components')

diff --git a/components/ip.c b/components/ip.c
index fce2b66..0b6293e 100644
--- a/components/ip.c
+++ b/components/ip.c
@@ -11,8 +11,8 @@
 
 #include "../util.h"
 
-const char *
-ipv4(const char *iface)
+static const char *
+ip(const char *iface, unsigned short sa_family)
 {
 	struct ifaddrs *ifaddr, *ifa;
 	int s;
@@ -27,10 +27,10 @@ ipv4(const char *iface)
 		if (!ifa->ifa_addr) {
 			continue;
 		}
-		s = getnameinfo(ifa->ifa_addr, sizeof(struct sockaddr_in), host,
-		                NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
+		s = getnameinfo(ifa->ifa_addr, sizeof(struct sockaddr_in6),
+		                host, NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
 		if (!strcmp(ifa->ifa_name, iface) &&
-		    (ifa->ifa_addr->sa_family == AF_INET)) {
+		    (ifa->ifa_addr->sa_family == sa_family)) {
 			if (s != 0) {
 				warn("getnameinfo: %s", gai_strerror(s));
 				return NULL;
@@ -45,34 +45,13 @@ ipv4(const char *iface)
 }
 
 const char *
-ipv6(const char *iface)
+ipv4(const char *iface)
 {
-	struct ifaddrs *ifaddr, *ifa;
-	int s;
-	char host[NI_MAXHOST];
-
-	if (getifaddrs(&ifaddr) < 0) {
-		warn("getifaddrs:");
-		return NULL;
-	}
-
-	for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
-		if (!ifa->ifa_addr) {
-			continue;
-		}
-		s = getnameinfo(ifa->ifa_addr, sizeof(struct sockaddr_in6), host,
-		                NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
-		if (!strcmp(ifa->ifa_name, iface) &&
-		    (ifa->ifa_addr->sa_family == AF_INET6)) {
-			if (s != 0) {
-				warn("getnameinfo: %s", gai_strerror(s));
-				return NULL;
-			}
-			return bprintf("%s", host);
-		}
-	}
-
-	freeifaddrs(ifaddr);
+	return ip(iface, AF_INET);
+}
 
-	return NULL;
+const char *
+ipv6(const char *iface)
+{
+	return ip(iface, AF_INET6);
 }
-- 
cgit v1.2.3