Merge pull request #443 from Priyanka-Middha/v4.1.0

NP-7067 | Added V6 networking support in Transport Node module
vmware · Sep 23, 2022 · 9245e16 · 9245e16
2 parents 76cc298 + ee98e51
commit 9245e16
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/plugins/modules/nsxt_transport_nodes.py b/plugins/modules/nsxt_transport_nodes.py
@@ -582,6 +582,7 @@
 import socket
 import hashlib
 import ssl
+import ipaddress
 
 FAILED_STATES = ["failed"]
 IN_PROGRESS_STATES = ["pending", "in_progress"]
@@ -813,7 +814,11 @@ def check_for_update(module, manager_url, mgr_username, mgr_password, validate_c
     return False
 
 def get_api_cert_thumbprint(ip_address, module):
-    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    ip = ipaddress.ip_address(ip_address)
+    if isinstance(ip, ipaddress.IPv4Address):
+        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    elif isinstance(ip, ipaddress.IPv6Address):
+        sock = socket.socket(socket.AF_INET6, socket.SOCK_STREAM)
     sock.settimeout(1)
     wrappedSocket = ssl.wrap_socket(sock)
     try: