mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-06 20:43:46 +08:00
firestarsdog
89 lines
3.6 KiB
Python
89 lines
3.6 KiB
Python
"""Authenticated recording responses with range and live-session checks."""
|
|
import os
|
|
from urllib.parse import unquote
|
|
|
|
from openpilot.starpilot.galaxy.recording_media import (RecordingMediaBusy, RecordingMediaChanged, RecordingMediaMissing,
|
|
RecordingMediaNotPrepared, RecordingMediaUnavailable,
|
|
RecordingMediaUnsupported, byte_range)
|
|
|
|
|
|
def camera_video(self, encoded_name: str, *, media, combined=False):
|
|
if not self.require_session():
|
|
return
|
|
name, separator, camera = unquote(encoded_name).partition('/')
|
|
camera = camera if separator else "qcamera"
|
|
if not name or '/' in camera or '?' in name or camera not in {"qcamera", "fcamera", "dcamera", "ecamera"}:
|
|
self.json(400, {'error': 'Invalid recording identity'})
|
|
return
|
|
try:
|
|
lease = media().open_route(name, camera=camera, prepare=self.command != 'HEAD') if combined else \
|
|
media().open(name, prepare=self.command != 'HEAD', **({'camera': camera} if camera != 'qcamera' else {}))
|
|
except ValueError:
|
|
if self.require_session():
|
|
self.json(400, {'error': 'Invalid recording identity'})
|
|
return
|
|
except RecordingMediaMissing:
|
|
if self.require_session():
|
|
self.json(404, {'error': 'Closed camera recording unavailable'})
|
|
return
|
|
except RecordingMediaNotPrepared:
|
|
if self.require_session():
|
|
self.json(202, {'status': 'Camera video not prepared; use GET to prepare it'})
|
|
return
|
|
except RecordingMediaChanged:
|
|
if self.require_session():
|
|
self.json(409, {'error': 'Recording changed; refresh the inventory'})
|
|
return
|
|
except RecordingMediaUnsupported:
|
|
if self.require_session():
|
|
self.json(415, {'error': 'Camera recording format is unsupported'})
|
|
return
|
|
except (RecordingMediaBusy, RecordingMediaUnavailable, OSError):
|
|
if self.require_session():
|
|
self.json(503, {'error': 'Camera video unavailable'})
|
|
return
|
|
self.send_recording(lease)
|
|
|
|
|
|
def send_recording(self, lease, content_type='video/mp4', filename=None):
|
|
try:
|
|
if not self.require_session():
|
|
return
|
|
if not lease.source.current():
|
|
self.json(409, {'error': 'Recording changed; refresh the inventory'})
|
|
return
|
|
selected = byte_range(self.headers.get('Range'), lease.size)
|
|
if selected is None:
|
|
self.send_response(416)
|
|
self.send_header('Content-Range', f'bytes */{lease.size}')
|
|
self.send_header('Content-Length', '0')
|
|
self.send_header('Cache-Control', 'no-store')
|
|
self.end_headers()
|
|
return
|
|
start, end = selected
|
|
self.send_response(206 if self.headers.get('Range') is not None else 200)
|
|
self.send_header('Content-Type', content_type)
|
|
if filename is not None:
|
|
self.send_header('Content-Disposition', f'attachment; filename="{filename}"')
|
|
self.send_header('Content-Length', str(end - start + 1))
|
|
self.send_header('Accept-Ranges', 'bytes')
|
|
if self.headers.get('Range') is not None:
|
|
self.send_header('Content-Range', f'bytes {start}-{end}/{lease.size}')
|
|
self.send_header('Cache-Control', 'no-store')
|
|
self.send_header('X-Content-Type-Options', 'nosniff')
|
|
self.send_header('Referrer-Policy', 'no-referrer')
|
|
self.end_headers()
|
|
if self.command == 'HEAD':
|
|
return
|
|
cursor = start
|
|
while cursor <= end and self.authenticated() and lease.source.current():
|
|
chunk = os.pread(lease.video_fd, min(64 * 1024, end - cursor + 1), cursor)
|
|
if not chunk:
|
|
break
|
|
self.wfile.write(chunk)
|
|
cursor += len(chunk)
|
|
if cursor <= end:
|
|
self.close_connection = True
|
|
finally:
|
|
lease.close()
|