Some more IPv6 support.

ployground · Feb 7, 2024 · 8bc64df · 8bc64df
1 parent c0fea68
commit 8bc64df
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/ploy/common.py b/ploy/common.py
@@ -663,9 +663,13 @@ def parse_ssh_keygen(text):
 
 
 def wait_for_ssh(host, port, timeout=5):
-    with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as s:
+    addrinfos = socket.getaddrinfo(host, port, 0, socket.SOCK_STREAM)
+    if not addrinfos:
+        raise socket.gaierror
+    addrinfo = addrinfos[0]
+    with closing(socket.socket(*addrinfo[:3])) as s:
         s.settimeout(timeout)
-        if s.connect_ex((host, port)) == 0:
+        if s.connect_ex(addrinfo[4]) == 0:
             if s.recv(5).startswith(b'SSH-2'):
                 return
 

diff --git a/ploy/plain.py b/ploy/plain.py
@@ -64,6 +64,8 @@ class Instance(BaseInstance):
 
     def get_host(self):
         if 'host' not in self.config:
+            if 'ip' not in self.config:
+                return self.config['ipv6']
             return self.config['ip']
         return self.config['host']