Merge pull request #107 from betarixm/fix/address

Ensure non-loopback address
betarixm · Dec 7, 2023 · 686981e · 686981e
2 parents 925405d + 5b83507
commit 686981e
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/utils/src/main/scala/Hooks.scala b/utils/src/main/scala/Hooks.scala
@@ -1,12 +1,22 @@
 package kr.ac.postech.paranode.utils
 
-import java.net.InetAddress
+import java.net.Inet4Address
+import java.net.NetworkInterface
 import java.net.ServerSocket
+import scala.jdk.CollectionConverters._
 import scala.reflect.io.Directory
 import scala.util.Using
 
 object Hooks {
-  def useLocalHostAddress: String = InetAddress.getLocalHost.getHostAddress
+  def useLocalHostAddress: String =
+    NetworkInterface.getNetworkInterfaces.asScala
+      .flatMap(_.getInetAddresses.asScala.toList)
+      .find(address =>
+        address.isInstanceOf[Inet4Address] && !address.isLoopbackAddress
+      )
+      .get
+      .getHostAddress
+
   def useAvailablePort: Int = Using(new ServerSocket(0))(_.getLocalPort).get
   def useTemporaryDirectory: Directory = Directory.makeTemp("paranode")
 }