mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-10-28 00:53:45 +08:00
firestar5683
ec8d7eb98b
Define persistent settings, typed schema fields and storage migrations before feature owners consume them. Keep calibration and parameter readers aligned with the shared data contracts. Co-authored-by: Adeeb Shihadeh <adeebshihadeh@gmail.com> Co-authored-by: Danny <dannyirwin31@yahoo.com> Co-authored-by: Harald Schäfer <harald.the.engineer@gmail.com> Co-authored-by: Prabhaav Pillai <prabhaav@msu.edu> Co-authored-by: dirwin31 <83434411+dirwin31@users.noreply.github.com> Co-authored-by: firestarsdog <229254897+firestarsdog@users.noreply.github.com> Co-authored-by: haraschax <6804392+haraschax@users.noreply.github.com> Co-authored-by: Dom <168790843+firestar5683@users.noreply.github.com> Co-authored-by: prabhaavp <prabhaav@msu.edu>
82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
"""One exact-source Params document replacement under the native Params lock."""
|
|
|
|
from collections.abc import Callable
|
|
from dataclasses import dataclass
|
|
import fcntl
|
|
import os
|
|
from pathlib import Path
|
|
import tempfile
|
|
import time
|
|
|
|
from openpilot.starpilot.saved_source import read_saved
|
|
|
|
|
|
LOCK_WAIT_SECONDS = .25
|
|
|
|
|
|
def acquire_native_lock(fd: int) -> None:
|
|
deadline = time.monotonic() + LOCK_WAIT_SECONDS
|
|
while True:
|
|
try:
|
|
fcntl.flock(fd, fcntl.LOCK_EX | fcntl.LOCK_NB)
|
|
return
|
|
except BlockingIOError:
|
|
remaining = deadline - time.monotonic()
|
|
if remaining <= 0:
|
|
raise
|
|
time.sleep(min(.01, remaining))
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class WriteResult:
|
|
committed: bool
|
|
verified: bool
|
|
|
|
|
|
def commit_exact(params, *, key: str, max_bytes: int, raw: bytes, expected: bytes | None,
|
|
authorized: Callable[[], bool], temp_prefix: str) -> WriteResult:
|
|
"""Stage outside the lock; recheck authority and source before replacement."""
|
|
if not authorized():
|
|
return WriteResult(False, False)
|
|
source, readable = read_saved(params, key, max_bytes)
|
|
if not readable or source != expected:
|
|
return WriteResult(False, False)
|
|
temporary = None
|
|
lock_fd = None
|
|
committed = False
|
|
try:
|
|
destination = Path(params.get_param_path(key))
|
|
root = destination.parent.parent
|
|
with tempfile.NamedTemporaryFile(prefix=temp_prefix, dir=root, delete=False) as stage:
|
|
temporary = stage.name
|
|
stage.write(raw)
|
|
stage.flush()
|
|
os.fsync(stage.fileno())
|
|
lock_fd = os.open(root / ".lock", os.O_CREAT | os.O_RDONLY, 0o775)
|
|
acquire_native_lock(lock_fd)
|
|
if not authorized():
|
|
return WriteResult(False, False)
|
|
source, readable = read_saved(params, key, max_bytes)
|
|
if not readable or source != expected or not authorized():
|
|
return WriteResult(False, False)
|
|
os.replace(temporary, destination)
|
|
temporary = None
|
|
committed = True
|
|
directory = os.open(destination.parent, os.O_RDONLY)
|
|
try:
|
|
os.fsync(directory)
|
|
finally:
|
|
os.close(directory)
|
|
observed, readable = read_saved(params, key, max_bytes)
|
|
return WriteResult(True, readable and observed == raw)
|
|
except (OSError, ValueError, TypeError):
|
|
return WriteResult(committed, False)
|
|
finally:
|
|
if lock_fd is not None:
|
|
os.close(lock_fd)
|
|
if temporary is not None:
|
|
try:
|
|
os.unlink(temporary)
|
|
except OSError:
|
|
pass
|