46 строки
2.1 KiB
Python
46 строки
2.1 KiB
Python
#!/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}')
|
|
|
|
changed = []
|
|
|
|
text = app_path.read_text()
|
|
old_signature = 'def network_forward_add(request: Request, vm_name: str = Form(...), guest_ip: str = Form(...), external_port: int = Form(...), guest_port: int = Form(...), protocol: str = Form("tcp"), note: str = Form("")):'
|
|
new_signature = 'def network_forward_add(request: Request, vm_name: str = Form(...), guest_ip: str = Form(...), external_port: str = Form(...), guest_port: str = Form(...), protocol: str = Form("tcp"), note: str = Form("")):'
|
|
if old_signature in text:
|
|
text = text.replace(old_signature, new_signature, 1)
|
|
app_path.write_text(text)
|
|
changed.append('app.py forward/add accepts port ranges')
|
|
elif new_signature in text:
|
|
changed.append('app.py forward/add already accepts port ranges')
|
|
else:
|
|
raise SystemExit('network_forward_add signature marker not found')
|
|
|
|
core_path = app_path.with_name('network_core.py')
|
|
if core_path.exists():
|
|
core_text = core_path.read_text()
|
|
before = core_text
|
|
core_text = core_text.replace(
|
|
'guest_to = f"{guest_ip}:{iptables_port_value(item[\'guest_port_start\'], item[\'guest_port_end\'])}"',
|
|
'guest_to_port = nft_port_value(item[\'guest_port_start\'], item[\'guest_port_end\'])\n guest_to = f"{guest_ip}:{guest_to_port}"',
|
|
)
|
|
core_text = core_text.replace(
|
|
'f"{vm_ip}:{matching_forward[\'guest_port_label\'].replace(\'-\', \':\')}" in ipt_nat_text',
|
|
'(f"{vm_ip}:{matching_forward[\'guest_port_label\']}" in ipt_nat_text or f"{vm_ip}:{matching_forward[\'guest_port_label\'].replace(\'-\', \':\')}" in ipt_nat_text)',
|
|
)
|
|
if core_text != before:
|
|
core_path.write_text(core_text)
|
|
changed.append('network_core.py iptables DNAT range syntax patched')
|
|
else:
|
|
changed.append('network_core.py range syntax already ok')
|
|
else:
|
|
changed.append(f'network_core.py not found near {app_path}')
|
|
|
|
print('network ranges patch applied:')
|
|
for item in changed:
|
|
print(f'- {item}')
|
© 2026 StackWorks