X-Git-Url: https://git.xinqibao.xyz/slstatus.git/blobdiff_plain/ebf5a35052add4298e43e0f261927b7ab2a2e7b9..9ac721c23fb640de2a6d1f84c84a79b2ccc26691:/components/ip.c

diff --git a/components/ip.c b/components/ip.c
index 8a40cfe..70724eb 100644
--- a/components/ip.c
+++ b/components/ip.c
@@ -1,33 +1,41 @@
 /* See LICENSE file for copyright and license details. */
-#ifdef __linux__
-#include <err.h>
 #include <ifaddrs.h>
 #include <netdb.h>
 #include <stdio.h>
 #include <string.h>
+#if defined(__OpenBSD__)
+	#include <sys/types.h>
+	#include <sys/socket.h>
+#elif defined(__FreeBSD__)
+	#include <netinet/in.h>
+	#include <sys/socket.h>
+#endif
 
 #include "../util.h"
 
-const char *
-ipv4(const char *iface)
+static const char *
+ip(const char *interface, unsigned short sa_family)
 {
 	struct ifaddrs *ifaddr, *ifa;
 	int s;
 	char host[NI_MAXHOST];
 
-	if (getifaddrs(&ifaddr) == -1) {
-		warn("Failed to get IPv4 address for interface %s", iface);
+	if (getifaddrs(&ifaddr) < 0) {
+		warn("getifaddrs:");
 		return NULL;
 	}
 
 	for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
-		if (ifa->ifa_addr == NULL) {
+		if (!ifa->ifa_addr) {
 			continue;
 		}
-		s = getnameinfo(ifa->ifa_addr, sizeof(struct sockaddr_in), host, NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
-		if ((strcmp(ifa->ifa_name, iface) == 0) && (ifa->ifa_addr->sa_family == AF_INET)) {
+		s = getnameinfo(ifa->ifa_addr, sizeof(struct sockaddr_in6),
+		                host, NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
+		if (!strcmp(ifa->ifa_name, interface) &&
+		    (ifa->ifa_addr->sa_family == sa_family)) {
+			freeifaddrs(ifaddr);
 			if (s != 0) {
-				warnx("Failed to get IPv4 address for interface %s", iface);
+				warn("getnameinfo: %s", gai_strerror(s));
 				return NULL;
 			}
 			return bprintf("%s", host);
@@ -40,33 +48,13 @@ ipv4(const char *iface)
 }
 
 const char *
-ipv6(const char *iface)
+ipv4(const char *interface)
 {
-	struct ifaddrs *ifaddr, *ifa;
-	int s;
-	char host[NI_MAXHOST];
-
-	if (getifaddrs(&ifaddr) == -1) {
-		warn("Failed to get IPv6 address for interface %s", iface);
-		return NULL;
-	}
-
-	for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
-		if (ifa->ifa_addr == NULL) {
-			continue;
-		}
-		s = getnameinfo(ifa->ifa_addr, sizeof(struct sockaddr_in6), host, NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
-		if ((strcmp(ifa->ifa_name, iface) == 0) && (ifa->ifa_addr->sa_family == AF_INET6)) {
-			if (s != 0) {
-				warnx("Failed to get IPv6 address for interface %s", iface);
-				return NULL;
-			}
-			return bprintf("%s", host);
-		}
-	}
-
-	freeifaddrs(ifaddr);
+	return ip(interface, AF_INET);
+}
 
-	return NULL;
+const char *
+ipv6(const char *interface)
+{
+	return ip(interface, AF_INET6);
 }
-#endif