diff options
Diffstat (limited to 'src')
| -rwxr-xr-x | src/op_mode/traceroute.py | 15 |
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 |