From 1505907698bba62dc95d56a6c7da70506e6f484b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=92=D0=B8=D0=BA=D1=82=D0=BE=D1=80?= <78488229+viktor138irk@users.noreply.github.com> Date: Thu, 7 May 2026 00:03:04 +0900 Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5?= =?UTF-8?q?=D0=BD=20=D0=BF=D0=B0=D1=82=D1=87=20=D0=BE=D0=B1=D1=80=D0=B0?= =?UTF-8?q?=D0=B1=D0=BE=D1=82=D1=87=D0=B8=D0=BA=D0=B0=20=D0=B4=D0=B8=D0=B0?= =?UTF-8?q?=D0=B3=D0=BD=D0=BE=D1=81=D1=82=D0=B8=D0=BA=D0=B8=20=D1=81=D0=B5?= =?UTF-8?q?=D1=82=D0=B8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- scripts/patch_network_diagnostics.py | 43 ++++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 scripts/patch_network_diagnostics.py diff --git a/scripts/patch_network_diagnostics.py b/scripts/patch_network_diagnostics.py new file mode 100644 index 0000000..cc1fd8f --- /dev/null +++ b/scripts/patch_network_diagnostics.py @@ -0,0 +1,43 @@ +#!/usr/bin/env python3 +from pathlib import Path +import sys + +app_path = Path(sys.argv[1]) if len(sys.argv) > 1 else Path('/opt/virtuality/web/app.py') +if not app_path.exists(): + raise SystemExit(f'app.py not found: {app_path}') + +text = app_path.read_text() +if 'def network_diagnose(' in text: + print('network diagnostics handler already applied') + raise SystemExit(0) + +handler = ''' + +@app.post("/network/diagnose", response_class=HTMLResponse) +def network_diagnose(request: Request, vm_name: str = Form(...), external_port: int = Form(...), guest_port: int = Form(...), protocol: str = Form("tcp")): + auth_redirect = require_auth(request) + if auth_redirect: + return auth_redirect + diagnostics = None + error = None + try: + diagnostics = network_core.diagnose_public_access(vm_name, external_port, guest_port, protocol) + except NetworkError as exc: + error = str(exc) + return templates.TemplateResponse("network.html", { + "request": request, + "app_name": APP_NAME, + "user": AUTH_USER, + "vms": parse_virsh_list(), + "ctx": network_core.network_context(), + "error": error, + "diagnostics": diagnostics, + }, status_code=400 if error else 200) +''' + +marker = '\n\n@app.get("/operations", response_class=HTMLResponse)' +if marker not in text: + raise SystemExit('operations marker not found in app.py') +text = text.replace(marker, handler + marker, 1) +app_path.write_text(text) +print(f'network diagnostics handler applied: {app_path}')