ram: Use POSIX types

[slstatus.git] / components / ip.c

diff --git a/components/ip.c b/components/ip.c
index f98b2ed..a7b1bfd 100644 (file)
--- a/components/ip.c
+++ b/components/ip.c
@@ -1,32 +1,38 @@
 /* See LICENSE file for copyright and license details. */
-#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>
+#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);
@@ -39,32 +45,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);
 }