update scripts to handle remote hosts

oresat · Jan 4, 2024 · 830c0a1 · 830c0a1
1 parent fba0312
commit 830c0a1
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/scripts/edl_cmd_shell.py b/scripts/edl_cmd_shell.py
@@ -31,6 +31,9 @@ def __init__(self, host: str, uplink_port: int, downlink_port: int):
         self._uplink_address = (host, uplink_port)
         self._uplink_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
 
+        if host not in ["localhost", "127.0.0.1"]:
+            host = ""
+
         self._downlink_address = (host, downlink_port)
         self._downlink_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
         self._downlink_socket.bind(self._downlink_address)

diff --git a/scripts/edl_file_upload.py b/scripts/edl_file_upload.py
@@ -254,7 +254,9 @@ def main():
         file_data = bytes([random.randint(0, 255) for _ in range(args.random_data)])
 
     uplink_address = (args.host, args.uplink_port)
-    downlink_address = (args.host, args.downlink_port)
+
+    downlink_host = args.host if args.host in ["localhost", "127.0.0.1"] else ""
+    downlink_address = (downlink_host, args.downlink_port)
 
     t = Thread(target=recv_thread, args=(downlink_address, args.bad_connection), daemon=True)
     t.start()

diff --git a/scripts/edl_ping_loop.py b/scripts/edl_ping_loop.py
@@ -88,7 +88,8 @@ def main():
 
     uplink_address = (args.host, args.uplink_port)
 
-    downlink_address = (args.host, args.downlink_port)
+    downlink_host = args.host if args.host in ["localhost", "127.0.0.1"] else ""
+    downlink_address = (downlink_host, args.downlink_port)
     downlink_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
     downlink_socket.bind(downlink_address)
     downlink_socket.settimeout(1)