route windows vnc correctly

pwncollege · Oct 1, 2024 · 407f99a · 407f99a
1 parent cc99572
commit 407f99a
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/dojo_plugin/api/v1/workspace.py b/dojo_plugin/api/v1/workspace.py
@@ -56,6 +56,17 @@ def get(self):
  "password": password,
  }
  iframe_src = url_for("pwncollege_workspace.forward_workspace", service=service_param, service_path="vnc.html", **vnc_params)
+ elif service == "desktop-windows":
+ service_param = "~".join(("desktop-windows", str(user.id), container_password(container, "desktop-windows")))
+ vnc_params = {
+ "autoconnect": 1,
+ "reconnect": 1,
+ "reconnect_delay": 200,
+ "resize": "local",
+ "path": url_for("pwncollege_workspace.forward_workspace", service=service_param, service_path="websockify"),
+ "password": "password", # Not really used for auth!
+ }
+ iframe_src = url_for("pwncollege_workspace.forward_workspace", service=service_param, service_path="vnc.html", **vnc_params)
  else:
  iframe_src = f"/workspace/{service}/"