refactor(host/W1): split native.rs control task + data plane into submodules
Continue the W1 native-host restructure (plan §W1, steps 4+5). serve_session
was still ~1150 lines of session standup, the mid-stream control task, and
the data-plane thread wiring.
- native/control.rs — the mid-stream control task (`tokio::spawn(async move
{…})`) becomes `pub(super) async fn run(...)`: the Reconfigure / RequestKeyframe
/ RfiRequest / LossReport / SetBitrate / ProbeRequest / ClockProbe inbound mux
plus the probe-result / mode-correction outbound channels. Call site is now
`tokio::spawn(control::run(...))`.
- native/stream.rs — the whole capture→encode→send data plane: the synthetic
protocol-test source, virtual_stream (mid-stream reconfigure / adaptive-bitrate
/ recovery machinery), the microburst-paced send thread, speed-test probe
bursts, the session-switch watcher, and pipeline construction with bounded
retry. Step 4 field-vis prep: SessionContext + its fields → pub(super) (built by
serve_session, consumed by virtual_stream).
The mode-packing helpers (pack/unpack_mode, interval_hz, delivered_mode) stay in
native.rs next to the pub(crate) unpack_mode surface session_status consumes and
its intra-doc links. native.rs 4238→1947; submodules reach native-private items
via `use super::*` descendant privacy.
Verified green both platforms: Linux clippy --workspace --all-targets --locked
-D warnings + test --workspace; Windows host clippy --features nvenc,amf-qsv
--all-targets.
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
+30
-2321
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,206 @@
|
|||||||
|
//! The native `punktfunk/1` mid-stream control task (plan §W1 — carved out of [`super`]'s
|
||||||
|
//! `serve_session`). After the handshake the control stream stays open for renegotiation and
|
||||||
|
//! speed tests; this task multiplexes the inbound client requests (`Reconfigure` /
|
||||||
|
//! `RequestKeyframe` / `RfiRequest` / `LossReport` / `SetBitrate` / `ProbeRequest` / `ClockProbe`)
|
||||||
|
//! with the outbound probe-result and mode-correction channels, handing every validated change to
|
||||||
|
//! the data-plane thread over the session's mpsc bridges.
|
||||||
|
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
/// Run the control task for one live session. Owns the control streams (`serve_session` hands them
|
||||||
|
/// off after negotiation) plus every channel end that bridges to the data-plane thread. Returns
|
||||||
|
/// when the control stream closes or a data-plane channel drops.
|
||||||
|
#[allow(clippy::too_many_arguments)]
|
||||||
|
pub(super) async fn run(
|
||||||
|
mut ctrl_send: quinn::SendStream,
|
||||||
|
mut ctrl_recv: quinn::RecvStream,
|
||||||
|
initial_mode: punktfunk_core::Mode,
|
||||||
|
codec: crate::encode::Codec,
|
||||||
|
live_reconfig_ok: bool,
|
||||||
|
adaptive_fec: bool,
|
||||||
|
session_bitrate_kbps: u32,
|
||||||
|
fec_target_ctl: Arc<AtomicU8>,
|
||||||
|
reconfig_tx: std::sync::mpsc::Sender<punktfunk_core::Mode>,
|
||||||
|
keyframe_tx: std::sync::mpsc::Sender<()>,
|
||||||
|
rfi_tx: std::sync::mpsc::Sender<(u32, u32)>,
|
||||||
|
bitrate_tx: std::sync::mpsc::Sender<u32>,
|
||||||
|
probe_tx: std::sync::mpsc::Sender<ProbeRequest>,
|
||||||
|
mut probe_result_rx: tokio::sync::mpsc::UnboundedReceiver<ProbeResult>,
|
||||||
|
mut reconfig_result_rx: tokio::sync::mpsc::UnboundedReceiver<Reconfigured>,
|
||||||
|
) {
|
||||||
|
let mut active = initial_mode;
|
||||||
|
// Host-side switch rate limit (a backstop against a hostile/broken client spamming
|
||||||
|
// Reconfigure into pipeline-rebuild churn — the drain-to-newest in the data plane already
|
||||||
|
// coalesces a well-behaved resize drag; compliant clients self-limit to ≥ 1 s).
|
||||||
|
const MIN_SWITCH_INTERVAL: std::time::Duration = std::time::Duration::from_millis(500);
|
||||||
|
let mut last_accepted_switch: Option<std::time::Instant> = None;
|
||||||
|
loop {
|
||||||
|
tokio::select! {
|
||||||
|
msg = io::read_msg(&mut ctrl_recv) => {
|
||||||
|
let Ok(msg) = msg else { break }; // stream closed
|
||||||
|
if let Ok(req) = Reconfigure::decode(&msg) {
|
||||||
|
let now = std::time::Instant::now();
|
||||||
|
let valid = req.mode.refresh_hz > 0
|
||||||
|
&& crate::encode::validate_dimensions(
|
||||||
|
codec,
|
||||||
|
req.mode.width,
|
||||||
|
req.mode.height,
|
||||||
|
)
|
||||||
|
.is_ok();
|
||||||
|
let too_soon = last_accepted_switch
|
||||||
|
.is_some_and(|t| now.duration_since(t) < MIN_SWITCH_INTERVAL);
|
||||||
|
let ok = if !live_reconfig_ok {
|
||||||
|
// Backend can't live-reconfigure (gamescope / synthetic /
|
||||||
|
// per-client-mode identity — see the gate above): honest downgrade,
|
||||||
|
// the client keeps scaling client-side.
|
||||||
|
tracing::info!(mode = ?req.mode,
|
||||||
|
"mode switch rejected (backend cannot live-reconfigure)");
|
||||||
|
false
|
||||||
|
} else if !valid {
|
||||||
|
tracing::warn!(mode = ?req.mode, "mode switch rejected (invalid dimensions)");
|
||||||
|
false
|
||||||
|
} else if too_soon {
|
||||||
|
tracing::warn!(mode = ?req.mode, "mode switch rejected (rate-limited)");
|
||||||
|
false
|
||||||
|
} else {
|
||||||
|
true
|
||||||
|
};
|
||||||
|
if ok {
|
||||||
|
active = req.mode;
|
||||||
|
last_accepted_switch = Some(now);
|
||||||
|
tracing::info!(mode = ?req.mode, "mode switch accepted");
|
||||||
|
}
|
||||||
|
let ack = Reconfigured { accepted: ok, mode: active };
|
||||||
|
if io::write_msg(&mut ctrl_send, &ack.encode()).await.is_err() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if ok && reconfig_tx.send(req.mode).is_err() {
|
||||||
|
break; // data plane gone
|
||||||
|
}
|
||||||
|
} else if RequestKeyframe::decode(&msg).is_ok() {
|
||||||
|
// Client recovery: its decoder wedged — force the next encoded frame to
|
||||||
|
// be an IDR. Coalesced in the encode loop (a wedge fires several before
|
||||||
|
// the IDR lands); a send error just means the data plane is gone.
|
||||||
|
tracing::debug!("client requested keyframe (decode recovery)");
|
||||||
|
if keyframe_tx.send(()).is_err() {
|
||||||
|
break; // data plane gone
|
||||||
|
}
|
||||||
|
} else if let Ok(req) = RfiRequest::decode(&msg) {
|
||||||
|
// Client LTR-RFI recovery: it lost the frame range `[first, last]` and asks
|
||||||
|
// the encoder to re-reference a known-good older frame instead of paying for
|
||||||
|
// a full IDR. The encode loop attempts `invalidate_ref_frames`, falling back
|
||||||
|
// to a coalesced keyframe when the encoder can't (range too old / no RFI).
|
||||||
|
tracing::debug!(
|
||||||
|
first = req.first_frame,
|
||||||
|
last = req.last_frame,
|
||||||
|
"client requested reference-frame invalidation (loss recovery)"
|
||||||
|
);
|
||||||
|
if rfi_tx.send((req.first_frame, req.last_frame)).is_err() {
|
||||||
|
break; // data plane gone
|
||||||
|
}
|
||||||
|
} else if let Ok(rep) = LossReport::decode(&msg) {
|
||||||
|
// Adaptive FEC: size recovery to the loss the client is seeing. The data-plane
|
||||||
|
// send loop reads `fec_target_ctl` and applies it per frame. Ignored when FEC
|
||||||
|
// is pinned via PUNKTFUNK_FEC_PCT.
|
||||||
|
if adaptive_fec {
|
||||||
|
// Fast attack, slow decay: jump straight to what the reported loss
|
||||||
|
// needs, but come DOWN only one point per clean report (~750 ms). The
|
||||||
|
// memoryless controller ping-ponged on periodic burst loss (Wi-Fi
|
||||||
|
// scans / BT coexistence, a burst every few seconds): a single clean
|
||||||
|
// window dropped FEC back to the floor, so every next burst hit an
|
||||||
|
// unprotected stream — an unrecoverable frame, a freeze, and a
|
||||||
|
// recovery-IDR burst, once per cycle. Decaying over ~10 windows keeps
|
||||||
|
// the stream covered across the gap while still converging to FEC_MIN
|
||||||
|
// on a genuinely clean link.
|
||||||
|
let prev = fec_target_ctl.load(Ordering::Relaxed);
|
||||||
|
let target = adapt_fec(rep.loss_ppm).max(prev.saturating_sub(1));
|
||||||
|
fec_target_ctl.store(target, Ordering::Relaxed);
|
||||||
|
if prev != target {
|
||||||
|
tracing::debug!(
|
||||||
|
loss_ppm = rep.loss_ppm,
|
||||||
|
fec_pct = target,
|
||||||
|
prev_fec_pct = prev,
|
||||||
|
"adaptive FEC adjusted"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if let Ok(req) = SetBitrate::decode(&msg) {
|
||||||
|
// Mid-stream bitrate renegotiation (adaptive bitrate): clamp exactly like
|
||||||
|
// the Hello request, ack the resolved value, then hand it to the data-plane
|
||||||
|
// thread, which rebuilds the encoder in place at the same mode — the fresh
|
||||||
|
// encoder's first frame is an IDR with in-band parameter sets, so the
|
||||||
|
// client's decoder follows without a reconnect.
|
||||||
|
// PyroWave: the rate is PINNED (§4.6 — quality collapses under rate
|
||||||
|
// descent; recovery pressure is answered by codec fallback, not AIMD).
|
||||||
|
// Our client controller is off for this codec; this guards older or
|
||||||
|
// foreign clients by acking the unchanged session rate.
|
||||||
|
let resolved = if codec == crate::encode::Codec::PyroWave {
|
||||||
|
tracing::info!(
|
||||||
|
requested_kbps = req.bitrate_kbps,
|
||||||
|
pinned_kbps = session_bitrate_kbps,
|
||||||
|
"PyroWave session: mid-stream bitrate retarget refused (pinned)"
|
||||||
|
);
|
||||||
|
session_bitrate_kbps
|
||||||
|
} else {
|
||||||
|
resolve_bitrate_kbps(req.bitrate_kbps)
|
||||||
|
};
|
||||||
|
tracing::debug!(
|
||||||
|
requested_kbps = req.bitrate_kbps,
|
||||||
|
resolved_kbps = resolved,
|
||||||
|
"mid-stream bitrate change requested"
|
||||||
|
);
|
||||||
|
let ack = BitrateChanged {
|
||||||
|
bitrate_kbps: resolved,
|
||||||
|
};
|
||||||
|
if io::write_msg(&mut ctrl_send, &ack.encode()).await.is_err() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if bitrate_tx.send(resolved).is_err() {
|
||||||
|
break; // data plane gone
|
||||||
|
}
|
||||||
|
} else if let Ok(req) = ProbeRequest::decode(&msg) {
|
||||||
|
tracing::info!(
|
||||||
|
target_kbps = req.target_kbps,
|
||||||
|
duration_ms = req.duration_ms,
|
||||||
|
"speed-test probe requested"
|
||||||
|
);
|
||||||
|
if probe_tx.send(req).is_err() {
|
||||||
|
break; // data plane gone
|
||||||
|
}
|
||||||
|
} else if let Ok(probe) = ClockProbe::decode(&msg) {
|
||||||
|
// Wall-clock skew handshake: echo the client's t1 with our receive (t2) and
|
||||||
|
// send (t3) stamps, both in the host clock the AU pts_ns uses. Answered
|
||||||
|
// inline on the control stream — cheap, no data-plane involvement.
|
||||||
|
let t2_ns = now_ns();
|
||||||
|
let echo = ClockEcho {
|
||||||
|
t1_ns: probe.t1_ns,
|
||||||
|
t2_ns,
|
||||||
|
t3_ns: now_ns(),
|
||||||
|
};
|
||||||
|
if io::write_msg(&mut ctrl_send, &echo.encode()).await.is_err() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
tracing::warn!("unknown control message — ignoring");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
result = probe_result_rx.recv() => {
|
||||||
|
let Some(result) = result else { break }; // data plane gone
|
||||||
|
if io::write_msg(&mut ctrl_send, &result.encode()).await.is_err() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
correction = reconfig_result_rx.recv() => {
|
||||||
|
// H2 rollback/correction ack: the data plane reports the mode ACTUALLY live
|
||||||
|
// after a rebuild that failed (stayed at the old mode) or that the backend
|
||||||
|
// honored at a different refresh. Track it so a later rejection's
|
||||||
|
// `mode: active` echo is truthful too.
|
||||||
|
let Some(ack) = correction else { break }; // data plane gone
|
||||||
|
active = ack.mode;
|
||||||
|
if io::write_msg(&mut ctrl_send, &ack.encode()).await.is_err() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user