Suppport IPv6 and dual-stack ADB server connections in ppadb

ppadb/client.py

@@ -5,7 +5,7 @@
 logger = AdbLogging.get_logger(__name__)
 
 class Client(Host):
-    def __init__(self, host='127.0.0.1', port=5037):
+    def __init__(self, host='localhost', port=5037):
         self.host = host
         self.port = port
 

ppadb/connection.py

@@ -23,23 +23,46 @@ def __exit__(self, type, value, traceback):
     def connect(self):
         logger.debug("Connect to adb server - {}:{}".format(self.host, self.port))
 
-        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-
-        l_onoff = 1
-        l_linger = 0
-
-        self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_LINGER, struct.pack('ii', l_onoff, l_linger))
-        if self.timeout:
-            self.socket.settimeout(self.timeout)
-
+        err = None
+        connected = False
         try:
-            self.socket.connect((self.host, self.port))
+            addr_list = socket.getaddrinfo(self.host, self.port, 0, socket.SOCK_STREAM)
         except socket.error as e:
-            self.close()
             raise RuntimeError("ERROR: connecting to {}:{} {}.\nIs adb running on your computer?".format(
                 self.host,
                 self.port,
                 e
+            )) from e
+
+        for af, socktype, proto, _, sa in addr_list:
+            try:
+                self.socket = socket.socket(af, socktype, proto)
+            except socket.error as e:
+                err = e
+                self.socket = None
+                continue
+
+            l_onoff = 1
+            l_linger = 0
+            self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_LINGER, struct.pack('ii', l_onoff, l_linger))
+            if self.timeout:
+                self.socket.settimeout(self.timeout)
+
+            try:
+                self.socket.connect(sa)
+                connected = True
+                break
+            except socket.error as e:
+                err = e
+                self.close()
+
+        if not connected:
+            if self.socket is None and err is not None:
+                raise err
+            raise RuntimeError("ERROR: connecting to {}:{} {}.\nIs adb running on your computer?".format(
+                self.host,
+                self.port,
+                err
             ))
 
         return self.socket