mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-06-21 03:41:55 +03:00
Compare commits
10 Commits
aeaecaf404
...
eaa0c92e78
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
eaa0c92e78 | ||
|
|
652ea2454f | ||
|
|
c575fb223b | ||
|
|
9fa345f192 | ||
|
|
7b55c4b758 | ||
|
|
570e2e3f76 | ||
|
|
39fba9b0a7 | ||
|
|
328a26b169 | ||
|
|
311fb1bd19 | ||
|
|
48b1426891 |
@ -12,6 +12,7 @@ import time
|
|||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import List, Optional
|
from typing import List, Optional
|
||||||
|
from urllib.parse import parse_qs, urlparse
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, HTTPException, Request, Response
|
from fastapi import APIRouter, Depends, HTTPException, Request, Response
|
||||||
from fastapi.responses import JSONResponse, RedirectResponse
|
from fastapi.responses import JSONResponse, RedirectResponse
|
||||||
@ -26,7 +27,11 @@ from frigate.api.defs.request.app_body import (
|
|||||||
AppPutRoleBody,
|
AppPutRoleBody,
|
||||||
)
|
)
|
||||||
from frigate.api.defs.tags import Tags
|
from frigate.api.defs.tags import Tags
|
||||||
from frigate.api.media_auth import check_camera_access, deny_response_for_media_uri
|
from frigate.api.media_auth import (
|
||||||
|
check_camera_access,
|
||||||
|
deny_response_for_media_uri,
|
||||||
|
is_role_restricted,
|
||||||
|
)
|
||||||
from frigate.config import AuthConfig, NetworkingConfig, ProxyConfig
|
from frigate.config import AuthConfig, NetworkingConfig, ProxyConfig
|
||||||
from frigate.const import CONFIG_DIR, JWT_SECRET_ENV_VAR, PASSWORD_HASH_ALGORITHM
|
from frigate.const import CONFIG_DIR, JWT_SECRET_ENV_VAR, PASSWORD_HASH_ALGORITHM
|
||||||
from frigate.models import User
|
from frigate.models import User
|
||||||
@ -658,6 +663,10 @@ def auth(request: Request):
|
|||||||
if deny_status is not None:
|
if deny_status is not None:
|
||||||
return Response("", status_code=deny_status)
|
return Response("", status_code=deny_status)
|
||||||
|
|
||||||
|
deny_status = deny_response_for_go2rtc_stream(original_url, role, request)
|
||||||
|
if deny_status is not None:
|
||||||
|
return Response("", status_code=deny_status)
|
||||||
|
|
||||||
return success_response
|
return success_response
|
||||||
|
|
||||||
# now apply authentication
|
# now apply authentication
|
||||||
@ -757,6 +766,10 @@ def auth(request: Request):
|
|||||||
if deny_status is not None:
|
if deny_status is not None:
|
||||||
return Response("", status_code=deny_status)
|
return Response("", status_code=deny_status)
|
||||||
|
|
||||||
|
deny_status = deny_response_for_go2rtc_stream(original_url, role, request)
|
||||||
|
if deny_status is not None:
|
||||||
|
return Response("", status_code=deny_status)
|
||||||
|
|
||||||
return success_response
|
return success_response
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"Error parsing jwt: {e}")
|
logger.error(f"Error parsing jwt: {e}")
|
||||||
@ -1112,6 +1125,66 @@ def _get_stream_owner_cameras(request: Request, stream_name: str) -> set[str]:
|
|||||||
return owner_cameras
|
return owner_cameras
|
||||||
|
|
||||||
|
|
||||||
|
# nginx proxies these paths straight to go2rtc with authentication-only checks
|
||||||
|
# (see auth_request.conf). Each names the desired stream via the `src` query
|
||||||
|
# param, so the camera-level check must happen here in the `/auth` subrequest —
|
||||||
|
# `require_go2rtc_stream_access` only guards the REST `/go2rtc/streams/{name}`
|
||||||
|
# endpoint, not these proxied live-stream paths.
|
||||||
|
GO2RTC_STREAM_PROXY_PATHS = frozenset(
|
||||||
|
{
|
||||||
|
"/live/mse/api/ws",
|
||||||
|
"/live/webrtc/api/ws",
|
||||||
|
"/api/go2rtc/webrtc",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def deny_response_for_go2rtc_stream(
|
||||||
|
original_url: Optional[str], role: Optional[str], request: Request
|
||||||
|
) -> Optional[int]:
|
||||||
|
"""Block role-restricted users from go2rtc live streams they cannot access.
|
||||||
|
|
||||||
|
Returns 403 when any `src` stream named in `original_url` resolves to a
|
||||||
|
camera outside the role's allow-list (or when no `src` is provided on a
|
||||||
|
stream-proxy path), otherwise None. Mirrors the resolution logic in
|
||||||
|
`require_go2rtc_stream_access` so substream names map to their owning
|
||||||
|
camera correctly.
|
||||||
|
"""
|
||||||
|
if not original_url:
|
||||||
|
return None
|
||||||
|
|
||||||
|
parsed = urlparse(original_url)
|
||||||
|
if parsed.path not in GO2RTC_STREAM_PROXY_PATHS:
|
||||||
|
return None
|
||||||
|
|
||||||
|
frigate_config = request.app.frigate_config
|
||||||
|
|
||||||
|
# admin and full-access roles (no allow-list) bypass the camera check
|
||||||
|
if not role or not is_role_restricted(role, frigate_config):
|
||||||
|
return None
|
||||||
|
|
||||||
|
sources = parse_qs(parsed.query).get("src", [])
|
||||||
|
if not sources:
|
||||||
|
# a stream-proxy request naming no stream has nothing legitimate to
|
||||||
|
# show a restricted user
|
||||||
|
return 403
|
||||||
|
|
||||||
|
allowed_cameras = set(
|
||||||
|
User.get_allowed_cameras(
|
||||||
|
role,
|
||||||
|
frigate_config.auth.roles,
|
||||||
|
set(frigate_config.cameras.keys()),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
# deny if any requested source resolves outside the allow-list
|
||||||
|
for src in sources:
|
||||||
|
if not (_get_stream_owner_cameras(request, src) & allowed_cameras):
|
||||||
|
return 403
|
||||||
|
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
async def require_go2rtc_stream_access(
|
async def require_go2rtc_stream_access(
|
||||||
stream_name: Optional[str] = None,
|
stream_name: Optional[str] = None,
|
||||||
request: Request = None,
|
request: Request = None,
|
||||||
|
|||||||
@ -389,82 +389,106 @@ def events_explore(
|
|||||||
limit: int = 10,
|
limit: int = 10,
|
||||||
allowed_cameras: List[str] = Depends(get_allowed_cameras_for_filter),
|
allowed_cameras: List[str] = Depends(get_allowed_cameras_for_filter),
|
||||||
):
|
):
|
||||||
# get distinct labels for all events
|
if not allowed_cameras:
|
||||||
distinct_labels = (
|
return JSONResponse(content=[])
|
||||||
Event.select(Event.label)
|
|
||||||
.where(Event.camera << allowed_cameras)
|
explore_columns = (
|
||||||
.distinct()
|
Event.id,
|
||||||
.order_by(Event.label)
|
Event.camera,
|
||||||
|
Event.label,
|
||||||
|
Event.sub_label,
|
||||||
|
Event.zones,
|
||||||
|
Event.start_time,
|
||||||
|
Event.end_time,
|
||||||
|
Event.has_clip,
|
||||||
|
Event.has_snapshot,
|
||||||
|
Event.plus_id,
|
||||||
|
Event.retain_indefinitely,
|
||||||
|
Event.top_score,
|
||||||
|
Event.false_positive,
|
||||||
|
Event.box,
|
||||||
|
Event.data,
|
||||||
)
|
)
|
||||||
|
|
||||||
label_counts = {}
|
# Single query: per-label COUNT and top-N ranking by start_time computed
|
||||||
|
# via window functions in a CTE, then filtered to rn <= limit
|
||||||
def event_generator():
|
event_count = (
|
||||||
for label_obj in distinct_labels.iterator():
|
fn.COUNT(Event.id).over(partition_by=[Event.label]).alias("event_count")
|
||||||
label = label_obj.label
|
|
||||||
|
|
||||||
# get most recent events for this label
|
|
||||||
label_events = (
|
|
||||||
Event.select()
|
|
||||||
.where((Event.label == label) & (Event.camera << allowed_cameras))
|
|
||||||
.order_by(Event.start_time.desc())
|
|
||||||
.limit(limit)
|
|
||||||
.iterator()
|
|
||||||
)
|
|
||||||
|
|
||||||
# count total events for this label
|
|
||||||
label_counts[label] = (
|
|
||||||
Event.select()
|
|
||||||
.where((Event.label == label) & (Event.camera << allowed_cameras))
|
|
||||||
.count()
|
|
||||||
)
|
|
||||||
|
|
||||||
yield from label_events
|
|
||||||
|
|
||||||
def process_events():
|
|
||||||
for event in event_generator():
|
|
||||||
processed_event = {
|
|
||||||
"id": event.id,
|
|
||||||
"camera": event.camera,
|
|
||||||
"label": event.label,
|
|
||||||
"zones": event.zones,
|
|
||||||
"start_time": event.start_time,
|
|
||||||
"end_time": event.end_time,
|
|
||||||
"has_clip": event.has_clip,
|
|
||||||
"has_snapshot": event.has_snapshot,
|
|
||||||
"plus_id": event.plus_id,
|
|
||||||
"retain_indefinitely": event.retain_indefinitely,
|
|
||||||
"sub_label": event.sub_label,
|
|
||||||
"top_score": event.top_score,
|
|
||||||
"false_positive": event.false_positive,
|
|
||||||
"box": event.box,
|
|
||||||
"data": {
|
|
||||||
k: v
|
|
||||||
for k, v in event.data.items()
|
|
||||||
if k
|
|
||||||
in [
|
|
||||||
"type",
|
|
||||||
"score",
|
|
||||||
"top_score",
|
|
||||||
"description",
|
|
||||||
"sub_label_score",
|
|
||||||
"average_estimated_speed",
|
|
||||||
"velocity_angle",
|
|
||||||
"path_data",
|
|
||||||
"recognized_license_plate",
|
|
||||||
"recognized_license_plate_score",
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"event_count": label_counts[event.label],
|
|
||||||
}
|
|
||||||
yield processed_event
|
|
||||||
|
|
||||||
# convert iterator to list and sort
|
|
||||||
processed_events = sorted(
|
|
||||||
process_events(),
|
|
||||||
key=lambda x: (x["event_count"], x["start_time"]),
|
|
||||||
reverse=True,
|
|
||||||
)
|
)
|
||||||
|
rn = (
|
||||||
|
fn.ROW_NUMBER()
|
||||||
|
.over(partition_by=[Event.label], order_by=[Event.start_time.desc()])
|
||||||
|
.alias("rn")
|
||||||
|
)
|
||||||
|
|
||||||
|
base_query = Event.select(
|
||||||
|
*explore_columns,
|
||||||
|
event_count,
|
||||||
|
rn,
|
||||||
|
).where(Event.camera << allowed_cameras)
|
||||||
|
ranked = base_query.cte("ranked")
|
||||||
|
query = (
|
||||||
|
Event.select(
|
||||||
|
ranked.c.id,
|
||||||
|
ranked.c.camera,
|
||||||
|
ranked.c.label,
|
||||||
|
ranked.c.sub_label,
|
||||||
|
ranked.c.zones,
|
||||||
|
ranked.c.start_time,
|
||||||
|
ranked.c.end_time,
|
||||||
|
ranked.c.has_clip,
|
||||||
|
ranked.c.has_snapshot,
|
||||||
|
ranked.c.plus_id,
|
||||||
|
ranked.c.retain_indefinitely,
|
||||||
|
ranked.c.top_score,
|
||||||
|
ranked.c.false_positive,
|
||||||
|
ranked.c.box,
|
||||||
|
ranked.c.data,
|
||||||
|
ranked.c.event_count,
|
||||||
|
)
|
||||||
|
.from_(ranked)
|
||||||
|
.with_cte(ranked)
|
||||||
|
.where(ranked.c.rn <= limit)
|
||||||
|
.order_by(ranked.c.event_count.desc(), ranked.c.start_time.desc())
|
||||||
|
.objects()
|
||||||
|
)
|
||||||
|
|
||||||
|
allowed_data_keys = {
|
||||||
|
"type",
|
||||||
|
"score",
|
||||||
|
"top_score",
|
||||||
|
"description",
|
||||||
|
"sub_label_score",
|
||||||
|
"average_estimated_speed",
|
||||||
|
"velocity_angle",
|
||||||
|
"path_data",
|
||||||
|
"recognized_license_plate",
|
||||||
|
"recognized_license_plate_score",
|
||||||
|
}
|
||||||
|
|
||||||
|
processed_events = [
|
||||||
|
{
|
||||||
|
"id": event.id,
|
||||||
|
"camera": event.camera,
|
||||||
|
"label": event.label,
|
||||||
|
"zones": event.zones,
|
||||||
|
"start_time": event.start_time,
|
||||||
|
"end_time": event.end_time,
|
||||||
|
"has_clip": event.has_clip,
|
||||||
|
"has_snapshot": event.has_snapshot,
|
||||||
|
"plus_id": event.plus_id,
|
||||||
|
"retain_indefinitely": event.retain_indefinitely,
|
||||||
|
"sub_label": event.sub_label,
|
||||||
|
"top_score": event.top_score,
|
||||||
|
"false_positive": event.false_positive,
|
||||||
|
"box": event.box,
|
||||||
|
"data": {
|
||||||
|
k: v for k, v in (event.data or {}).items() if k in allowed_data_keys
|
||||||
|
},
|
||||||
|
"event_count": event.event_count,
|
||||||
|
}
|
||||||
|
for event in query
|
||||||
|
]
|
||||||
|
|
||||||
return JSONResponse(content=processed_events)
|
return JSONResponse(content=processed_events)
|
||||||
|
|
||||||
@ -487,22 +511,18 @@ async def event_ids(ids: str, request: Request):
|
|||||||
status_code=400,
|
status_code=400,
|
||||||
)
|
)
|
||||||
|
|
||||||
for event_id in ids:
|
|
||||||
try:
|
|
||||||
event = Event.get(Event.id == event_id)
|
|
||||||
await require_camera_access(event.camera, request=request)
|
|
||||||
except DoesNotExist:
|
|
||||||
# we should not fail the entire request if an event is not found
|
|
||||||
continue
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
events = Event.select().where(Event.id << ids).dicts().iterator()
|
events = list(Event.select().where(Event.id << ids).dicts().iterator())
|
||||||
return JSONResponse(list(events))
|
|
||||||
except Exception:
|
except Exception:
|
||||||
return JSONResponse(
|
return JSONResponse(
|
||||||
content=({"success": False, "message": "Events not found"}), status_code=400
|
content=({"success": False, "message": "Events not found"}), status_code=400
|
||||||
)
|
)
|
||||||
|
|
||||||
|
for event in events:
|
||||||
|
await require_camera_access(event["camera"], request=request)
|
||||||
|
|
||||||
|
return JSONResponse(events)
|
||||||
|
|
||||||
|
|
||||||
@router.get(
|
@router.get(
|
||||||
"/events/search",
|
"/events/search",
|
||||||
|
|||||||
@ -10,7 +10,7 @@ import pandas as pd
|
|||||||
from fastapi import APIRouter, Request
|
from fastapi import APIRouter, Request
|
||||||
from fastapi.params import Depends
|
from fastapi.params import Depends
|
||||||
from fastapi.responses import JSONResponse
|
from fastapi.responses import JSONResponse
|
||||||
from peewee import Case, DoesNotExist, IntegrityError, fn, operator
|
from peewee import Case, DoesNotExist, fn, operator
|
||||||
from playhouse.shortcuts import model_to_dict
|
from playhouse.shortcuts import model_to_dict
|
||||||
|
|
||||||
from frigate.api.auth import (
|
from frigate.api.auth import (
|
||||||
@ -172,11 +172,19 @@ async def review_ids(request: Request, ids: str):
|
|||||||
status_code=400,
|
status_code=400,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
reviews = list(
|
||||||
|
ReviewSegment.select().where(ReviewSegment.id << ids).dicts().iterator()
|
||||||
|
)
|
||||||
|
except Exception:
|
||||||
|
return JSONResponse(
|
||||||
|
content=({"success": False, "message": "Review segments not found"}),
|
||||||
|
status_code=400,
|
||||||
|
)
|
||||||
|
|
||||||
|
found_ids = {r["id"] for r in reviews}
|
||||||
for review_id in ids:
|
for review_id in ids:
|
||||||
try:
|
if review_id not in found_ids:
|
||||||
review = ReviewSegment.get(ReviewSegment.id == review_id)
|
|
||||||
await require_camera_access(review.camera, request=request)
|
|
||||||
except DoesNotExist:
|
|
||||||
return JSONResponse(
|
return JSONResponse(
|
||||||
content=(
|
content=(
|
||||||
{"success": False, "message": f"Review {review_id} not found"}
|
{"success": False, "message": f"Review {review_id} not found"}
|
||||||
@ -184,16 +192,10 @@ async def review_ids(request: Request, ids: str):
|
|||||||
status_code=404,
|
status_code=404,
|
||||||
)
|
)
|
||||||
|
|
||||||
try:
|
for review in reviews:
|
||||||
reviews = (
|
await require_camera_access(review["camera"], request=request)
|
||||||
ReviewSegment.select().where(ReviewSegment.id << ids).dicts().iterator()
|
|
||||||
)
|
return JSONResponse(reviews)
|
||||||
return JSONResponse(list(reviews))
|
|
||||||
except Exception:
|
|
||||||
return JSONResponse(
|
|
||||||
content=({"success": False, "message": "Review segments not found"}),
|
|
||||||
status_code=400,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@router.get(
|
@router.get(
|
||||||
@ -490,27 +492,52 @@ async def set_multiple_reviewed(
|
|||||||
|
|
||||||
user_id = current_user["username"]
|
user_id = current_user["username"]
|
||||||
|
|
||||||
for review_id in body.ids:
|
reviews = list(
|
||||||
try:
|
ReviewSegment.select(ReviewSegment.id, ReviewSegment.camera).where(
|
||||||
review = ReviewSegment.get(ReviewSegment.id == review_id)
|
ReviewSegment.id << body.ids
|
||||||
await require_camera_access(review.camera, request=request)
|
)
|
||||||
review_status = UserReviewStatus.get(
|
)
|
||||||
UserReviewStatus.user_id == user_id,
|
|
||||||
UserReviewStatus.review_segment == review_id,
|
for review in reviews:
|
||||||
|
await require_camera_access(review.camera, request=request)
|
||||||
|
|
||||||
|
found_ids = [r.id for r in reviews]
|
||||||
|
|
||||||
|
if found_ids:
|
||||||
|
existing_statuses = list(
|
||||||
|
UserReviewStatus.select().where(
|
||||||
|
(UserReviewStatus.user_id == user_id)
|
||||||
|
& (UserReviewStatus.review_segment << found_ids)
|
||||||
)
|
)
|
||||||
# Update based on the reviewed parameter
|
)
|
||||||
if review_status.has_been_reviewed != body.reviewed:
|
|
||||||
review_status.has_been_reviewed = body.reviewed
|
status_by_review = {s.review_segment_id: s for s in existing_statuses}
|
||||||
review_status.save()
|
|
||||||
except DoesNotExist:
|
to_update = []
|
||||||
try:
|
to_create = []
|
||||||
UserReviewStatus.create(
|
|
||||||
user_id=user_id,
|
for review_id in found_ids:
|
||||||
review_segment=ReviewSegment.get(id=review_id),
|
if review_id in status_by_review:
|
||||||
has_been_reviewed=body.reviewed,
|
status = status_by_review[review_id]
|
||||||
|
if status.has_been_reviewed != body.reviewed:
|
||||||
|
status.has_been_reviewed = body.reviewed
|
||||||
|
to_update.append(status)
|
||||||
|
else:
|
||||||
|
to_create.append(
|
||||||
|
{
|
||||||
|
"user_id": user_id,
|
||||||
|
"review_segment_id": review_id,
|
||||||
|
"has_been_reviewed": body.reviewed,
|
||||||
|
}
|
||||||
)
|
)
|
||||||
except (DoesNotExist, IntegrityError):
|
|
||||||
pass
|
if to_update:
|
||||||
|
UserReviewStatus.bulk_update(
|
||||||
|
to_update, fields=[UserReviewStatus.has_been_reviewed], batch_size=100
|
||||||
|
)
|
||||||
|
|
||||||
|
if to_create:
|
||||||
|
UserReviewStatus.insert_many(to_create).execute()
|
||||||
|
|
||||||
return JSONResponse(
|
return JSONResponse(
|
||||||
content=(
|
content=(
|
||||||
|
|||||||
175
frigate/test/test_go2rtc_stream_auth.py
Normal file
175
frigate/test/test_go2rtc_stream_auth.py
Normal file
@ -0,0 +1,175 @@
|
|||||||
|
"""Unit tests for `deny_response_for_go2rtc_stream`.
|
||||||
|
|
||||||
|
Covers the camera-level authorization enforced in the `/auth` subrequest for
|
||||||
|
the nginx-proxied go2rtc live-stream paths (MSE/WebRTC WebSockets and the
|
||||||
|
WebRTC signaling endpoint). These paths name the stream via the `src` query
|
||||||
|
param, which the static-media auth in `media_auth` does not inspect.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import types
|
||||||
|
import unittest
|
||||||
|
|
||||||
|
from frigate.api.auth import deny_response_for_go2rtc_stream
|
||||||
|
from frigate.config import FrigateConfig
|
||||||
|
|
||||||
|
_CONFIG = {
|
||||||
|
"mqtt": {"host": "mqtt"},
|
||||||
|
"auth": {
|
||||||
|
"roles": {
|
||||||
|
"limited_user": ["front_door"],
|
||||||
|
"dual_user": ["front_door", "back_door"],
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"cameras": {
|
||||||
|
"front_door": {
|
||||||
|
"ffmpeg": {
|
||||||
|
"inputs": [{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}]
|
||||||
|
},
|
||||||
|
"detect": {"height": 1080, "width": 1920, "fps": 5},
|
||||||
|
# go2rtc stream name differs from the camera name (substream)
|
||||||
|
"live": {"streams": {"Main Stream": "front_door_sub"}},
|
||||||
|
},
|
||||||
|
"back_door": {
|
||||||
|
"ffmpeg": {
|
||||||
|
"inputs": [{"path": "rtsp://10.0.0.2:554/video", "roles": ["detect"]}]
|
||||||
|
},
|
||||||
|
"detect": {"height": 1080, "width": 1920, "fps": 5},
|
||||||
|
},
|
||||||
|
"garage": {
|
||||||
|
"ffmpeg": {
|
||||||
|
"inputs": [{"path": "rtsp://10.0.0.3:554/video", "roles": ["detect"]}]
|
||||||
|
},
|
||||||
|
"detect": {"height": 1080, "width": 1920, "fps": 5},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def _request(config: FrigateConfig) -> types.SimpleNamespace:
|
||||||
|
return types.SimpleNamespace(app=types.SimpleNamespace(frigate_config=config))
|
||||||
|
|
||||||
|
|
||||||
|
class TestDenyResponseForGo2rtcStream(unittest.TestCase):
|
||||||
|
def setUp(self) -> None:
|
||||||
|
self.config = FrigateConfig(**_CONFIG)
|
||||||
|
self.request = _request(self.config)
|
||||||
|
|
||||||
|
def _deny(self, url: str, role: str):
|
||||||
|
return deny_response_for_go2rtc_stream(url, role, self.request)
|
||||||
|
|
||||||
|
# --- non-stream paths pass through ---
|
||||||
|
|
||||||
|
def test_non_stream_path_passes_through(self):
|
||||||
|
self.assertIsNone(
|
||||||
|
self._deny("http://host/clips/back_door-1.jpg", "limited_user")
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_empty_url_passes_through(self):
|
||||||
|
self.assertIsNone(self._deny("", "limited_user"))
|
||||||
|
|
||||||
|
def test_jsmpeg_path_not_handled_here(self):
|
||||||
|
# jsmpeg is authorized per-frame in the output pipeline, not here
|
||||||
|
self.assertIsNone(
|
||||||
|
self._deny("http://host/live/jsmpeg/back_door", "limited_user")
|
||||||
|
)
|
||||||
|
|
||||||
|
# --- restricted role: allowed vs forbidden cameras ---
|
||||||
|
|
||||||
|
def test_mse_allowed_camera(self):
|
||||||
|
self.assertIsNone(
|
||||||
|
self._deny("http://host/live/mse/api/ws?src=front_door", "limited_user")
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_mse_forbidden_camera_denied(self):
|
||||||
|
self.assertEqual(
|
||||||
|
self._deny("http://host/live/mse/api/ws?src=back_door", "limited_user"),
|
||||||
|
403,
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_webrtc_ws_forbidden_camera_denied(self):
|
||||||
|
self.assertEqual(
|
||||||
|
self._deny("http://host/live/webrtc/api/ws?src=back_door", "limited_user"),
|
||||||
|
403,
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_webrtc_signaling_forbidden_camera_denied(self):
|
||||||
|
self.assertEqual(
|
||||||
|
self._deny("http://host/api/go2rtc/webrtc?src=back_door", "limited_user"),
|
||||||
|
403,
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_unknown_camera_denied(self):
|
||||||
|
self.assertEqual(
|
||||||
|
self._deny("http://host/live/mse/api/ws?src=nonexistent", "limited_user"),
|
||||||
|
403,
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_missing_src_denied(self):
|
||||||
|
self.assertEqual(self._deny("http://host/live/mse/api/ws", "limited_user"), 403)
|
||||||
|
|
||||||
|
# --- multi-camera role: each assigned camera allowed, others denied ---
|
||||||
|
|
||||||
|
def test_multi_camera_role_allows_first_assigned(self):
|
||||||
|
self.assertIsNone(
|
||||||
|
self._deny("http://host/live/mse/api/ws?src=front_door", "dual_user")
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_multi_camera_role_allows_second_assigned(self):
|
||||||
|
self.assertIsNone(
|
||||||
|
self._deny("http://host/live/mse/api/ws?src=back_door", "dual_user")
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_multi_camera_role_denies_unassigned(self):
|
||||||
|
# garage is configured but not in dual_user's allow-list
|
||||||
|
self.assertEqual(
|
||||||
|
self._deny("http://host/live/mse/api/ws?src=garage", "dual_user"),
|
||||||
|
403,
|
||||||
|
)
|
||||||
|
|
||||||
|
# --- substream names resolve to their owning camera ---
|
||||||
|
|
||||||
|
def test_allowed_substream_resolves_to_owning_camera(self):
|
||||||
|
# front_door_sub is owned by front_door, which limited_user may access
|
||||||
|
self.assertIsNone(
|
||||||
|
self._deny("http://host/live/mse/api/ws?src=front_door_sub", "limited_user")
|
||||||
|
)
|
||||||
|
|
||||||
|
# --- multiple src values: deny if any is forbidden ---
|
||||||
|
|
||||||
|
def test_multiple_src_one_forbidden_denied(self):
|
||||||
|
self.assertEqual(
|
||||||
|
self._deny(
|
||||||
|
"http://host/live/mse/api/ws?src=front_door&src=back_door",
|
||||||
|
"limited_user",
|
||||||
|
),
|
||||||
|
403,
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_multiple_src_all_allowed(self):
|
||||||
|
self.assertIsNone(
|
||||||
|
self._deny(
|
||||||
|
"http://host/live/mse/api/ws?src=front_door&src=front_door_sub",
|
||||||
|
"limited_user",
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
# --- privileged roles bypass the check ---
|
||||||
|
|
||||||
|
def test_admin_bypasses(self):
|
||||||
|
self.assertIsNone(
|
||||||
|
self._deny("http://host/live/mse/api/ws?src=back_door", "admin")
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_builtin_viewer_role_bypasses(self):
|
||||||
|
# the built-in viewer role is not in the config allow-list map, so it
|
||||||
|
# is treated as full access
|
||||||
|
self.assertIsNone(
|
||||||
|
self._deny("http://host/live/mse/api/ws?src=back_door", "viewer")
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_missing_role_bypasses(self):
|
||||||
|
self.assertIsNone(self._deny("http://host/live/mse/api/ws?src=back_door", None))
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
unittest.main()
|
||||||
27
migrations/036_add_perf_indexes.py
Normal file
27
migrations/036_add_perf_indexes.py
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
"""Peewee migrations -- 036_add_perf_indexes.py.
|
||||||
|
|
||||||
|
Adds composite/single-column indexes to speed up the most common queries
|
||||||
|
issued by the web UI on initial page load:
|
||||||
|
|
||||||
|
- event(camera, start_time DESC): /events list filtered by camera + time range
|
||||||
|
- reviewsegment(camera, start_time DESC): /api/review filtered by camera + time range
|
||||||
|
- reviewsegment(end_time): supports the end_time > after half of /api/review's range
|
||||||
|
|
||||||
|
The existing event(label, start_time DESC) index from migration 027 already
|
||||||
|
covers /events/explore, so it is intentionally not duplicated here.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import peewee as pw
|
||||||
|
|
||||||
|
SQL = pw.SQL
|
||||||
|
|
||||||
|
|
||||||
|
def migrate(migrator, database, fake=False, **kwargs):
|
||||||
|
migrator.sql(
|
||||||
|
'CREATE INDEX IF NOT EXISTS "event_camera_start_time" '
|
||||||
|
'ON "event" ("camera", "start_time" DESC)'
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def rollback(migrator, database, fake=False, **kwargs):
|
||||||
|
migrator.sql('DROP INDEX IF EXISTS "event_camera_start_time"')
|
||||||
@ -243,12 +243,7 @@ export default function CameraReviewClassification({
|
|||||||
handleZoneToggle("alerts.required_zones", zone.name)
|
handleZoneToggle("alerts.required_zones", zone.name)
|
||||||
}
|
}
|
||||||
/>
|
/>
|
||||||
<Label
|
<Label className="font-normal">
|
||||||
className={cn(
|
|
||||||
"font-normal",
|
|
||||||
!zone.friendly_name && "smart-capitalize",
|
|
||||||
)}
|
|
||||||
>
|
|
||||||
{zone.friendly_name || zone.name}
|
{zone.friendly_name || zone.name}
|
||||||
</Label>
|
</Label>
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@ -29,8 +29,8 @@ function getZoneDisplayName(zoneName: string, context?: FormContext): string {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// Fallback to cleaning up the zone name
|
// Fallback to the raw zone id verbatim (no friendly_name available)
|
||||||
return String(zoneName).replace(/_/g, " ");
|
return String(zoneName);
|
||||||
}
|
}
|
||||||
|
|
||||||
export function ZoneSwitchesWidget(props: WidgetProps) {
|
export function ZoneSwitchesWidget(props: WidgetProps) {
|
||||||
|
|||||||
@ -1197,14 +1197,7 @@ function LifecycleIconRow({
|
|||||||
backgroundColor: `rgb(${color})`,
|
backgroundColor: `rgb(${color})`,
|
||||||
}}
|
}}
|
||||||
/>
|
/>
|
||||||
<span
|
<span>{item.data?.zones_friendly_names?.[zidx]}</span>
|
||||||
className={cn(
|
|
||||||
item.data?.zones_friendly_names?.[zidx] === zone &&
|
|
||||||
"smart-capitalize",
|
|
||||||
)}
|
|
||||||
>
|
|
||||||
{item.data?.zones_friendly_names?.[zidx]}
|
|
||||||
</span>
|
|
||||||
</Badge>
|
</Badge>
|
||||||
);
|
);
|
||||||
})}
|
})}
|
||||||
|
|||||||
@ -7,12 +7,12 @@ export function resolveZoneName(
|
|||||||
zoneId: string,
|
zoneId: string,
|
||||||
cameraId?: string,
|
cameraId?: string,
|
||||||
) {
|
) {
|
||||||
if (!config) return String(zoneId).replace(/_/g, " ");
|
if (!config) return String(zoneId);
|
||||||
|
|
||||||
if (cameraId) {
|
if (cameraId) {
|
||||||
const camera = config.cameras?.[String(cameraId)];
|
const camera = config.cameras?.[String(cameraId)];
|
||||||
const zone = camera?.zones?.[zoneId];
|
const zone = camera?.zones?.[zoneId];
|
||||||
return zone?.friendly_name || String(zoneId).replace(/_/g, " ");
|
return zone?.friendly_name || String(zoneId);
|
||||||
}
|
}
|
||||||
|
|
||||||
for (const camKey in config.cameras) {
|
for (const camKey in config.cameras) {
|
||||||
@ -21,12 +21,12 @@ export function resolveZoneName(
|
|||||||
if (!cam?.zones) continue;
|
if (!cam?.zones) continue;
|
||||||
if (Object.prototype.hasOwnProperty.call(cam.zones, zoneId)) {
|
if (Object.prototype.hasOwnProperty.call(cam.zones, zoneId)) {
|
||||||
const zone = cam.zones[zoneId];
|
const zone = cam.zones[zoneId];
|
||||||
return zone?.friendly_name || String(zoneId).replace(/_/g, " ");
|
return zone?.friendly_name || String(zoneId);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Fallback: return a cleaned-up zoneId string
|
// Fallback: display the raw zone id verbatim (no friendly_name available)
|
||||||
return String(zoneId).replace(/_/g, " ");
|
return String(zoneId);
|
||||||
}
|
}
|
||||||
|
|
||||||
export function useZoneFriendlyName(zoneId: string, cameraId?: string): string {
|
export function useZoneFriendlyName(zoneId: string, cameraId?: string): string {
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user