Merge pull request #990 from JasonRuonanWang/helper-ipaddress

added AvailableIpAddresses in helper
ornladios · Nov 7, 2018 · fb90b51 · fb90b51
2 parents 1f43693 + 3e8e09b
commit fb90b51
Show file tree

Hide file tree

Showing 2 changed files with 54 additions and 1 deletion.
diff --git a/source/adios2/helper/adiosSystem.cpp b/source/adios2/helper/adiosSystem.cpp
@@ -8,7 +8,13 @@
  *      Author: William F Godoy godoywf@ornl.gov
  */
 #include "adiosSystem.h"
-
+#ifndef _WIN32
+#include <arpa/inet.h> //AvailableIpAddresses() inet_ntoa
+#include <net/if.h>    //AvailableIpAddresses() struct if_nameindex
+#include <string.h>    //AvailableIpAddresses() strncp
+#include <sys/ioctl.h> //AvailableIpAddresses() ioctl
+#include <unistd.h>    //AvailableIpAddresses() close
+#endif
 #include <chrono> //system_clock, now
 #include <ctime>
 #include <iostream>  //std::cerr
@@ -74,6 +80,46 @@ bool IsZeroIndexed(const std::string hostLanguage) noexcept
     return isZeroIndexed;
 }
 
+#ifndef _WIN32
+std::vector<std::string> AvailableIpAddresses() noexcept
+{
+    std::vector<std::string> ips;
+    int socket_handler = -1;
+    struct if_nameindex *p = 0;
+    struct if_nameindex *head = 0;
+    if ((socket_handler = socket(PF_INET, SOCK_DGRAM, 0)) < 0)
+    {
+        return ips;
+    }
+    head = if_nameindex();
+    p = if_nameindex();
+    while ((p != NULL) && (p->if_name != NULL))
+    {
+        struct ifreq req;
+        strncpy(req.ifr_name, p->if_name, IFNAMSIZ);
+        if (ioctl(socket_handler, SIOCGIFADDR, &req) < 0)
+        {
+            if (errno == EADDRNOTAVAIL)
+            {
+                ++p;
+                continue;
+            }
+            close(socket_handler);
+            return ips;
+        }
+        const std::string ip =
+            inet_ntoa(((struct sockaddr_in *)&req.ifr_addr)->sin_addr);
+        if (ip != "127.0.0.1")
+        {
+            ips.emplace_back(ip);
+        }
+        ++p;
+    }
+    if_freenameindex(head);
+    close(socket_handler);
+    return ips;
+}
+#endif
 int ExceptionToError(const std::string &function)
 {
     try

diff --git a/source/adios2/helper/adiosSystem.h b/source/adios2/helper/adiosSystem.h
@@ -57,6 +57,13 @@ bool IsRowMajor(const std::string hostLanguage) noexcept;
  */
 bool IsZeroIndexed(const std::string hostLanguage) noexcept;
 
+/**
+ * returns a vector of strings with all available IP addresses on the node
+ * @return vector of strings
+ */
+#ifndef _WIN32
+std::vector<std::string> AvailableIpAddresses() noexcept;
+#endif
 /**
  * Function to be called inside a catch(...) block to convert C++ exceptions to
  * error integers. Used by C, Fortran bindings.