tanner/protocast
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Track last cast IP and validate stop requests

Browse Source 
Co-authored-by: aider (gemini/gemini-2.5-pro) <aider@aider.chat>
This commit is contained in:
Tanner Collin
2026-03-10 21:48:24 -06:00
parent f06e728513
commit 38f4d5c5e2
1 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
main.py
View File

@@ -21,11 +21,12 @@ def cast_spell():
logging.warning(f"Invalid or missing ip_address parameter: {ip_address}") logging.warning(f"Invalid or missing ip_address parameter: {ip_address}")
return f"Invalid or missing 'ip_address' parameter.", 400 return f"Invalid or missing 'ip_address' parameter.", 400
global auto_stop_timer global auto_stop_timer, LAST_CAST_ADDRESS
logging.info(f"Casting to {ip_address}.") logging.info(f"Casting to {ip_address}.")
kill_vnc() # Kill any existing VNC session first kill_vnc() # Kill any existing VNC session first
cast_machine(ip_address) cast_machine(ip_address)
LAST_CAST_ADDRESS = ip_address
# Restart auto-stop timer # Restart auto-stop timer
if auto_stop_timer is not None and auto_stop_timer.is_alive(): if auto_stop_timer is not None and auto_stop_timer.is_alive():
auto_stop_timer.cancel() auto_stop_timer.cancel()
@@ -37,8 +38,17 @@ def cast_spell():
@app.route('/stop', methods=['POST']) @app.route('/stop', methods=['POST'])
def stop_cast(): def stop_cast():
global auto_stop_timer global auto_stop_timer, LAST_CAST_ADDRESS
logging.info("Received POST request on /stop.") ip_address = request.form.get('ip_address')
logging.info(f"Received POST request on /stop for ip_address: {ip_address}")
if not ip_address:
logging.warning(f"Invalid or missing ip_address parameter: {ip_address}")
return f"Invalid or missing 'ip_address' parameter.", 400
if ip_address != LAST_CAST_ADDRESS:
logging.warning(f"Stop request for {ip_address} does not match last cast address {LAST_CAST_ADDRESS}.")
return f"Stop request for {ip_address} does not match last cast address.", 403
if auto_stop_timer is not None and auto_stop_timer.is_alive(): if auto_stop_timer is not None and auto_stop_timer.is_alive():
auto_stop_timer.cancel() auto_stop_timer.cancel()
@@ -46,6 +56,7 @@ def stop_cast():
logging.info("Manual stop: auto-stop timer cancelled.") logging.info("Manual stop: auto-stop timer cancelled.")
kill_vnc() kill_vnc()
LAST_CAST_ADDRESS = None
return "Attempted to stop VNC viewers.", 200 return "Attempted to stop VNC viewers.", 200
def _auto_stop_vnc(): def _auto_stop_vnc():