Allow connection to ws console over IPv6. Fixes GNS3/gns3-web-ui#1400

GNS3 · Aug 2, 2023 · 658bfb7 · 658bfb7
1 parent 88a1cef
commit 658bfb7
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/gns3server/handlers/api/controller/node_handler.py b/gns3server/handlers/api/controller/node_handler.py
@@ -17,6 +17,7 @@
 
 import aiohttp
 import asyncio
+import ipaddress
 
 from gns3server.web.route import Route
 from gns3server.controller import Controller
@@ -477,7 +478,16 @@ async def ws_console(request, response):
         await ws.prepare(request)
         request.app['websockets'].add(ws)
 
-        ws_console_compute_url = "ws://{compute_host}:{compute_port}/v2/compute/projects/{project_id}/{node_type}/nodes/{node_id}/console/ws".format(compute_host=compute.host,
+        compute_host = compute.host
+        try:
+            # handle IPv6 address
+            ip = ipaddress.ip_address(compute_host)
+            if isinstance(ip, ipaddress.IPv6Address):
+                compute_host = '[' + compute_host + ']'
+        except ValueError:
+            pass
+
+        ws_console_compute_url = "ws://{compute_host}:{compute_port}/v2/compute/projects/{project_id}/{node_type}/nodes/{node_id}/console/ws".format(compute_host=compute_host,
                                                                                                                                                      compute_port=compute.port,
                                                                                                                                                      project_id=project.id,
                                                                                                                                                      node_type=node.node_type,