traceroute: T8539: fix source-address AF lookup

1 files changed, 13 insertions, 2 deletions

diff --git a/src/op_mode/traceroute.py b/src/op_mode/traceroute.py
index 6d6ef27aa..222b2d326 100755
--- a/src/op_mode/traceroute.py
+++ b/src/op_mode/traceroute.py
@@ -222,11 +222,22 @@ if __name__ == '__main__':
             args.append(name)
             args.append(option['dflt'])
 
+    af = socket.AF_UNSPEC
+    for i in range(len(args)):
+        matched = complete(args[i])
+        if len(matched) == 1 and matched[0] == 'source-address' and i + 1 < len(args):
+            try:
+                src_version = ipaddress.ip_address(args[i + 1]).version
+                af = socket.AF_INET6 if src_version == 6 else socket.AF_INET
+            except ValueError:
+                pass
+            break
+
     try:
-        info = socket.getaddrinfo(host, None, socket.AF_UNSPEC, socket.SOCK_STREAM)
+        info = socket.getaddrinfo(host, None, af, socket.SOCK_STREAM)
         ip = info[0][4][0]
     except UnicodeError:
-        sys.exit(f'tracroute: Unknown host: {host}')
+        sys.exit(f'traceroute: Unknown host: {host}')
     except socket.gaierror:
         ip = host