Factor out get_default_ip

(cherry picked from commit 66e913b)

Author: NickCao

packages/jumpstarter-driver-http/jumpstarter_driver_http/driver.py

@@ -6,6 +6,7 @@
 from aiohttp import web
 from jumpstarter_driver_opendal.driver import Opendal
 
+from jumpstarter.common.ipaddress import get_default_ip
 from jumpstarter.driver import Driver, export
 
 
@@ -41,18 +42,7 @@ def __post_init__(self):
             ]
         )
         if self.host is None:
-            self.host = self.get_default_ip()
-
-    def get_default_ip(self):
-        try:
-            import socket
-
-            with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
-                s.connect(("8.8.8.8", 80))
-                return s.getsockname()[0]
-        except Exception:
-            self.logger.warning("Could not determine default IP address, falling back to 0.0.0.0")
-            return "0.0.0.0"
+            self.host = get_default_ip(logger=self.logger)
 
     @classmethod
     def client(cls) -> str:

packages/jumpstarter-driver-tftp/jumpstarter_driver_tftp/driver.py

@@ -9,6 +9,7 @@
 
 from jumpstarter_driver_tftp.server import TftpServer
 
+from jumpstarter.common.ipaddress import get_default_ip
 from jumpstarter.driver import Driver, export
 
 
@@ -55,17 +56,7 @@ def __post_init__(self):
         self.storage = self.children["storage"]
 
         if self.host == "":
-            self.host = self.get_default_ip()
-
-    def get_default_ip(self):
-        """Get the IP address of the default route interface"""
-        try:
-            with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
-                s.connect(("8.8.8.8", 80))
-                return s.getsockname()[0]
-        except Exception:
-            self.logger.warning("Could not determine default IP address, falling back to 0.0.0.0")
-            return "0.0.0.0"
+            self.host = get_default_ip(logger=self.logger)
 
     @classmethod
     def client(cls) -> str:

packages/jumpstarter/jumpstarter/common/ipaddress.py

@@ -1,3 +1,4 @@
+import logging
 import socket
 
 
@@ -12,3 +13,15 @@ def get_ip_address() -> str:
         s.connect(("1.1.1.1", 0))
         address = s.getsockname()[0]
     return address
+
+
+def get_default_ip(logger: logging.Logger | None = None):
+    """Get the IP address of the default route interface"""
+    try:
+        with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
+            s.connect(("8.8.8.8", 80))
+            return s.getsockname()[0]
+    except Exception:
+        if logger:
+            logger.warning("Could not determine default IP address, falling back to 0.0.0.0")
+        return "0.0.0.0"