feat: Phase 2 — relay daemon and client library with integration pipelines
wzp-relay: - RelayPipeline: ingest → FEC decode → jitter buffer → FEC encode → send - SessionManager: tracks active calls, idle expiry - RelayConfig: TOML-based configuration - Binary: accepts QUIC connections, receives media packets wzp-client: - CallEncoder: mic PCM → Opus encode → FEC → MediaPackets - CallDecoder: MediaPackets → FEC decode → jitter → Opus decode → PCM - CLI binary: connects to relay, sends test silence frames 99 tests passing across all 7 crates. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -15,5 +15,14 @@ wzp-transport = { workspace = true }
|
||||
tokio = { workspace = true }
|
||||
tracing = { workspace = true }
|
||||
tracing-subscriber = { workspace = true }
|
||||
async-trait = { workspace = true }
|
||||
bytes = { workspace = true }
|
||||
serde = { workspace = true }
|
||||
toml = "0.8"
|
||||
anyhow = "1"
|
||||
|
||||
[[bin]]
|
||||
name = "wzp-relay"
|
||||
path = "src/main.rs"
|
||||
|
||||
[dev-dependencies]
|
||||
|
||||
35
crates/wzp-relay/src/config.rs
Normal file
35
crates/wzp-relay/src/config.rs
Normal file
@@ -0,0 +1,35 @@
|
||||
//! Relay daemon configuration.
|
||||
|
||||
use serde::{Deserialize, Serialize};
|
||||
use std::net::SocketAddr;
|
||||
|
||||
/// Configuration for the relay daemon.
|
||||
#[derive(Clone, Debug, Serialize, Deserialize)]
|
||||
pub struct RelayConfig {
|
||||
/// Address to listen on for incoming connections (client-facing).
|
||||
pub listen_addr: SocketAddr,
|
||||
/// Address of the remote relay (for the lossy inter-relay link).
|
||||
/// If None, this relay is the destination-side relay.
|
||||
pub remote_relay: Option<SocketAddr>,
|
||||
/// Maximum concurrent sessions.
|
||||
pub max_sessions: usize,
|
||||
/// Jitter buffer target depth in packets.
|
||||
pub jitter_target_depth: usize,
|
||||
/// Jitter buffer maximum depth in packets.
|
||||
pub jitter_max_depth: usize,
|
||||
/// Logging level (trace, debug, info, warn, error).
|
||||
pub log_level: String,
|
||||
}
|
||||
|
||||
impl Default for RelayConfig {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
listen_addr: "0.0.0.0:4433".parse().unwrap(),
|
||||
remote_relay: None,
|
||||
max_sessions: 100,
|
||||
jitter_target_depth: 50,
|
||||
jitter_max_depth: 250,
|
||||
log_level: "info".to_string(),
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -1,6 +1,16 @@
|
||||
//! WarzonePhone Relay Daemon
|
||||
//!
|
||||
//! Integration crate that wires together all layers into a relay pipeline:
|
||||
//! recv → decrypt → FEC decode → jitter → FEC encode → encrypt → send
|
||||
//! recv → FEC decode → jitter buffer → FEC encode → send
|
||||
//!
|
||||
//! Built after the 5 agent crates (proto, codec, fec, crypto, transport) are complete.
|
||||
//! The relay forwards media between two QUIC endpoints without decoding audio.
|
||||
//! It operates on FEC-protected packets, managing loss recovery and adaptive
|
||||
//! quality transitions.
|
||||
|
||||
pub mod config;
|
||||
pub mod pipeline;
|
||||
pub mod session_mgr;
|
||||
|
||||
pub use config::RelayConfig;
|
||||
pub use pipeline::{PipelineConfig, PipelineStats, RelayPipeline};
|
||||
pub use session_mgr::{RelaySession, SessionId, SessionManager};
|
||||
|
||||
66
crates/wzp-relay/src/main.rs
Normal file
66
crates/wzp-relay/src/main.rs
Normal file
@@ -0,0 +1,66 @@
|
||||
//! WarzonePhone relay daemon entry point.
|
||||
|
||||
use std::sync::Arc;
|
||||
|
||||
use tokio::sync::Mutex;
|
||||
use tracing::{error, info};
|
||||
|
||||
use wzp_proto::MediaTransport;
|
||||
use wzp_relay::config::RelayConfig;
|
||||
use wzp_relay::session_mgr::SessionManager;
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() -> anyhow::Result<()> {
|
||||
let config = RelayConfig::default();
|
||||
|
||||
tracing_subscriber::fmt().init();
|
||||
|
||||
info!(addr = %config.listen_addr, "WarzonePhone relay starting");
|
||||
|
||||
let (server_config, _cert_der) = wzp_transport::server_config();
|
||||
let endpoint =
|
||||
wzp_transport::create_endpoint(config.listen_addr, Some(server_config))?;
|
||||
|
||||
let sessions = Arc::new(Mutex::new(SessionManager::new(config.max_sessions)));
|
||||
|
||||
info!("Listening for connections...");
|
||||
|
||||
loop {
|
||||
let connection = match wzp_transport::accept(&endpoint).await {
|
||||
Ok(conn) => conn,
|
||||
Err(e) => {
|
||||
error!("accept error: {e}");
|
||||
continue;
|
||||
}
|
||||
};
|
||||
|
||||
let _sessions = sessions.clone();
|
||||
|
||||
tokio::spawn(async move {
|
||||
let remote = connection.remote_address();
|
||||
info!(%remote, "new connection");
|
||||
|
||||
let transport = wzp_transport::QuinnTransport::new(connection);
|
||||
|
||||
loop {
|
||||
match transport.recv_media().await {
|
||||
Ok(Some(packet)) => {
|
||||
tracing::trace!(
|
||||
seq = packet.header.seq,
|
||||
block = packet.header.fec_block,
|
||||
"received media packet"
|
||||
);
|
||||
}
|
||||
Ok(None) => {
|
||||
info!(%remote, "connection closed");
|
||||
break;
|
||||
}
|
||||
Err(e) => {
|
||||
error!(%remote, "recv error: {e}");
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
302
crates/wzp-relay/src/pipeline.rs
Normal file
302
crates/wzp-relay/src/pipeline.rs
Normal file
@@ -0,0 +1,302 @@
|
||||
//! Media processing pipeline for the relay.
|
||||
//!
|
||||
//! The relay pipeline processes media packets in both directions:
|
||||
//! - **Inbound** (from client/upstream): recv → decrypt → FEC decode → jitter buffer
|
||||
//! - **Outbound** (to downstream/remote): jitter pop → FEC encode → encrypt → send
|
||||
//!
|
||||
//! The relay does NOT decode/re-encode audio — it operates on encrypted,
|
||||
//! FEC-protected packets. The crypto and FEC layers are the relay's concern;
|
||||
//! the actual audio codec is end-to-end between client and destination.
|
||||
|
||||
use tracing::{debug, info};
|
||||
|
||||
use wzp_fec::{RaptorQFecDecoder, RaptorQFecEncoder};
|
||||
use wzp_proto::jitter::{JitterBuffer, PlayoutResult};
|
||||
use wzp_proto::packet::{MediaHeader, MediaPacket};
|
||||
use wzp_proto::quality::AdaptiveQualityController;
|
||||
use wzp_proto::traits::{FecDecoder, FecEncoder, QualityController};
|
||||
use wzp_proto::QualityProfile;
|
||||
|
||||
/// Configuration for a relay pipeline instance.
|
||||
pub struct PipelineConfig {
|
||||
pub initial_profile: QualityProfile,
|
||||
pub jitter_target: usize,
|
||||
pub jitter_max: usize,
|
||||
pub jitter_min: usize,
|
||||
}
|
||||
|
||||
impl Default for PipelineConfig {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
initial_profile: QualityProfile::GOOD,
|
||||
jitter_target: 50,
|
||||
jitter_max: 250,
|
||||
jitter_min: 25,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// A relay media pipeline for one direction of a call session.
|
||||
///
|
||||
/// Each call has two pipelines: client→destination and destination→client.
|
||||
pub struct RelayPipeline {
|
||||
/// FEC encoder for outbound packets.
|
||||
fec_encoder: RaptorQFecEncoder,
|
||||
/// FEC decoder for inbound packets.
|
||||
fec_decoder: RaptorQFecDecoder,
|
||||
/// Jitter buffer for reordering and smoothing.
|
||||
jitter: JitterBuffer,
|
||||
/// Adaptive quality controller.
|
||||
quality: AdaptiveQualityController,
|
||||
/// Current quality profile.
|
||||
profile: QualityProfile,
|
||||
/// Outbound sequence counter.
|
||||
out_seq: u16,
|
||||
/// Packets processed count.
|
||||
stats: PipelineStats,
|
||||
}
|
||||
|
||||
/// Pipeline statistics.
|
||||
#[derive(Clone, Debug, Default)]
|
||||
pub struct PipelineStats {
|
||||
pub packets_received: u64,
|
||||
pub packets_forwarded: u64,
|
||||
pub packets_fec_recovered: u64,
|
||||
pub packets_lost: u64,
|
||||
pub profile_changes: u64,
|
||||
}
|
||||
|
||||
impl RelayPipeline {
|
||||
/// Create a new relay pipeline with the given configuration.
|
||||
pub fn new(config: PipelineConfig) -> Self {
|
||||
let (fec_enc, fec_dec) = wzp_fec::create_fec_pair(&config.initial_profile);
|
||||
|
||||
Self {
|
||||
fec_encoder: fec_enc,
|
||||
fec_decoder: fec_dec,
|
||||
jitter: JitterBuffer::new(config.jitter_target, config.jitter_max, config.jitter_min),
|
||||
quality: AdaptiveQualityController::new(),
|
||||
profile: config.initial_profile,
|
||||
out_seq: 0,
|
||||
stats: PipelineStats::default(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Process an incoming media packet from the upstream side.
|
||||
///
|
||||
/// The packet is fed into the FEC decoder and jitter buffer.
|
||||
/// Returns decoded packets ready for forwarding (if any).
|
||||
pub fn ingest(&mut self, packet: MediaPacket) -> Vec<MediaPacket> {
|
||||
self.stats.packets_received += 1;
|
||||
|
||||
// Feed quality report if present
|
||||
if let Some(ref qr) = packet.quality_report {
|
||||
if let Some(new_profile) = self.quality.observe(qr) {
|
||||
info!(
|
||||
tier = ?self.quality.tier(),
|
||||
codec = ?new_profile.codec,
|
||||
fec_ratio = new_profile.fec_ratio,
|
||||
"quality tier change"
|
||||
);
|
||||
self.profile = new_profile;
|
||||
self.stats.profile_changes += 1;
|
||||
// Reconfigure FEC for new profile
|
||||
let (enc, dec) = wzp_fec::create_fec_pair(&new_profile);
|
||||
self.fec_encoder = enc;
|
||||
self.fec_decoder = dec;
|
||||
}
|
||||
}
|
||||
|
||||
// Feed packet into FEC decoder
|
||||
let header = &packet.header;
|
||||
let _ = self.fec_decoder.add_symbol(
|
||||
header.fec_block,
|
||||
header.fec_symbol,
|
||||
header.is_repair,
|
||||
&packet.payload,
|
||||
);
|
||||
|
||||
// Try to decode the FEC block
|
||||
let mut output = Vec::new();
|
||||
if let Ok(Some(frames)) = self.fec_decoder.try_decode(header.fec_block) {
|
||||
debug!(
|
||||
block = header.fec_block,
|
||||
frames = frames.len(),
|
||||
"FEC block decoded"
|
||||
);
|
||||
// Each recovered frame becomes a media packet for the jitter buffer
|
||||
for (i, frame) in frames.into_iter().enumerate() {
|
||||
let reconstructed = MediaPacket {
|
||||
header: MediaHeader {
|
||||
version: 0,
|
||||
is_repair: false,
|
||||
codec_id: header.codec_id,
|
||||
has_quality_report: false,
|
||||
fec_ratio_encoded: header.fec_ratio_encoded,
|
||||
// Reconstruct seq from block + symbol index
|
||||
seq: (header.fec_block as u16)
|
||||
.wrapping_mul(self.profile.frames_per_block as u16)
|
||||
.wrapping_add(i as u16),
|
||||
timestamp: header
|
||||
.timestamp
|
||||
.wrapping_add((i as u32) * (header.codec_id.frame_duration_ms() as u32)),
|
||||
fec_block: header.fec_block,
|
||||
fec_symbol: i as u8,
|
||||
reserved: 0,
|
||||
csrc_count: 0,
|
||||
},
|
||||
payload: bytes::Bytes::from(frame),
|
||||
quality_report: None,
|
||||
};
|
||||
self.jitter.push(reconstructed);
|
||||
}
|
||||
}
|
||||
|
||||
// Pop from jitter buffer
|
||||
loop {
|
||||
match self.jitter.pop() {
|
||||
PlayoutResult::Packet(pkt) => {
|
||||
self.stats.packets_forwarded += 1;
|
||||
output.push(pkt);
|
||||
}
|
||||
PlayoutResult::Missing { seq } => {
|
||||
self.stats.packets_lost += 1;
|
||||
debug!(seq, "jitter buffer: missing packet");
|
||||
// Continue popping — the next packet might be available
|
||||
}
|
||||
PlayoutResult::NotReady => break,
|
||||
}
|
||||
}
|
||||
|
||||
output
|
||||
}
|
||||
|
||||
/// Prepare a packet for outbound transmission.
|
||||
///
|
||||
/// Adds FEC encoding and assigns a new sequence number.
|
||||
pub fn prepare_outbound(&mut self, mut packet: MediaPacket) -> Vec<MediaPacket> {
|
||||
// Assign outbound sequence number
|
||||
packet.header.seq = self.out_seq;
|
||||
self.out_seq = self.out_seq.wrapping_add(1);
|
||||
|
||||
let mut output = vec![packet.clone()];
|
||||
|
||||
// Add to FEC encoder
|
||||
let _ = self.fec_encoder.add_source_symbol(&packet.payload);
|
||||
|
||||
// Check if block is full
|
||||
if self.fec_encoder.current_block_size() >= self.profile.frames_per_block as usize {
|
||||
// Generate repair packets
|
||||
if let Ok(repairs) = self.fec_encoder.generate_repair(self.profile.fec_ratio) {
|
||||
for (sym_idx, repair_data) in repairs {
|
||||
let repair_packet = MediaPacket {
|
||||
header: MediaHeader {
|
||||
version: 0,
|
||||
is_repair: true,
|
||||
codec_id: packet.header.codec_id,
|
||||
has_quality_report: false,
|
||||
fec_ratio_encoded: MediaHeader::encode_fec_ratio(
|
||||
self.profile.fec_ratio,
|
||||
),
|
||||
seq: self.out_seq,
|
||||
timestamp: packet.header.timestamp,
|
||||
fec_block: self.fec_encoder.current_block_id(),
|
||||
fec_symbol: sym_idx,
|
||||
reserved: 0,
|
||||
csrc_count: 0,
|
||||
},
|
||||
payload: bytes::Bytes::from(repair_data),
|
||||
quality_report: None,
|
||||
};
|
||||
self.out_seq = self.out_seq.wrapping_add(1);
|
||||
output.push(repair_packet);
|
||||
}
|
||||
}
|
||||
let _ = self.fec_encoder.finalize_block();
|
||||
}
|
||||
|
||||
output
|
||||
}
|
||||
|
||||
/// Get current pipeline statistics.
|
||||
pub fn stats(&self) -> &PipelineStats {
|
||||
&self.stats
|
||||
}
|
||||
|
||||
/// Get current quality profile.
|
||||
pub fn profile(&self) -> QualityProfile {
|
||||
self.profile
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use wzp_proto::CodecId;
|
||||
use bytes::Bytes;
|
||||
|
||||
fn make_media_packet(seq: u16, block: u8, symbol: u8) -> MediaPacket {
|
||||
MediaPacket {
|
||||
header: MediaHeader {
|
||||
version: 0,
|
||||
is_repair: false,
|
||||
codec_id: CodecId::Opus24k,
|
||||
has_quality_report: false,
|
||||
fec_ratio_encoded: 0,
|
||||
seq,
|
||||
timestamp: seq as u32 * 20,
|
||||
fec_block: block,
|
||||
fec_symbol: symbol,
|
||||
reserved: 0,
|
||||
csrc_count: 0,
|
||||
},
|
||||
payload: Bytes::from(vec![seq as u8; 60]),
|
||||
quality_report: None,
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn pipeline_creates_successfully() {
|
||||
let pipeline = RelayPipeline::new(PipelineConfig::default());
|
||||
assert_eq!(pipeline.profile().codec, CodecId::Opus24k);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn prepare_outbound_assigns_seq() {
|
||||
let mut pipeline = RelayPipeline::new(PipelineConfig::default());
|
||||
let pkt = make_media_packet(0, 0, 0);
|
||||
let out = pipeline.prepare_outbound(pkt);
|
||||
assert!(!out.is_empty());
|
||||
assert_eq!(out[0].header.seq, 0);
|
||||
|
||||
let pkt2 = make_media_packet(1, 0, 1);
|
||||
let out2 = pipeline.prepare_outbound(pkt2);
|
||||
assert_eq!(out2[0].header.seq, 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn prepare_outbound_generates_repair_on_full_block() {
|
||||
let mut pipeline = RelayPipeline::new(PipelineConfig {
|
||||
initial_profile: QualityProfile::GOOD, // 5 frames/block, 20% FEC
|
||||
..Default::default()
|
||||
});
|
||||
|
||||
// Feed 5 packets (one full block)
|
||||
let mut total_out = 0;
|
||||
for i in 0..5u16 {
|
||||
let pkt = make_media_packet(i, 0, i as u8);
|
||||
let out = pipeline.prepare_outbound(pkt);
|
||||
total_out += out.len();
|
||||
}
|
||||
// Should have 5 source + at least 1 repair packet
|
||||
assert!(total_out > 5, "expected repair packets, got {total_out}");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn stats_track_packets() {
|
||||
let mut pipeline = RelayPipeline::new(PipelineConfig::default());
|
||||
let pkt = make_media_packet(0, 0, 0);
|
||||
pipeline.ingest(pkt);
|
||||
assert_eq!(pipeline.stats().packets_received, 1);
|
||||
}
|
||||
}
|
||||
138
crates/wzp-relay/src/session_mgr.rs
Normal file
138
crates/wzp-relay/src/session_mgr.rs
Normal file
@@ -0,0 +1,138 @@
|
||||
//! Session manager — tracks active call sessions on the relay.
|
||||
|
||||
use std::collections::HashMap;
|
||||
|
||||
use wzp_proto::{QualityProfile, Session};
|
||||
|
||||
use crate::pipeline::{PipelineConfig, RelayPipeline};
|
||||
|
||||
/// Unique identifier for a relay session.
|
||||
pub type SessionId = [u8; 16];
|
||||
|
||||
/// A single active call session on the relay.
|
||||
pub struct RelaySession {
|
||||
/// Protocol session state machine.
|
||||
pub state: Session,
|
||||
/// Pipeline for upstream → downstream direction.
|
||||
pub upstream_pipeline: RelayPipeline,
|
||||
/// Pipeline for downstream → upstream direction.
|
||||
pub downstream_pipeline: RelayPipeline,
|
||||
/// Quality profile currently in use.
|
||||
pub profile: QualityProfile,
|
||||
/// Timestamp of last activity (ms since epoch).
|
||||
pub last_activity_ms: u64,
|
||||
}
|
||||
|
||||
impl RelaySession {
|
||||
pub fn new(session_id: SessionId, config: PipelineConfig) -> Self {
|
||||
let profile = config.initial_profile;
|
||||
Self {
|
||||
state: Session::new(session_id),
|
||||
upstream_pipeline: RelayPipeline::new(PipelineConfig {
|
||||
initial_profile: profile,
|
||||
..config
|
||||
}),
|
||||
downstream_pipeline: RelayPipeline::new(PipelineConfig {
|
||||
initial_profile: profile,
|
||||
..config
|
||||
}),
|
||||
profile,
|
||||
last_activity_ms: 0,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn is_active(&self) -> bool {
|
||||
self.state.is_media_active()
|
||||
}
|
||||
}
|
||||
|
||||
/// Manages all active sessions on a relay.
|
||||
pub struct SessionManager {
|
||||
sessions: HashMap<SessionId, RelaySession>,
|
||||
max_sessions: usize,
|
||||
}
|
||||
|
||||
impl SessionManager {
|
||||
pub fn new(max_sessions: usize) -> Self {
|
||||
Self {
|
||||
sessions: HashMap::new(),
|
||||
max_sessions,
|
||||
}
|
||||
}
|
||||
|
||||
/// Create a new session. Returns None if at capacity.
|
||||
pub fn create_session(
|
||||
&mut self,
|
||||
session_id: SessionId,
|
||||
config: PipelineConfig,
|
||||
) -> Option<&mut RelaySession> {
|
||||
if self.sessions.len() >= self.max_sessions {
|
||||
return None;
|
||||
}
|
||||
self.sessions
|
||||
.entry(session_id)
|
||||
.or_insert_with(|| RelaySession::new(session_id, config));
|
||||
self.sessions.get_mut(&session_id)
|
||||
}
|
||||
|
||||
/// Get a session by ID.
|
||||
pub fn get_session(&mut self, id: &SessionId) -> Option<&mut RelaySession> {
|
||||
self.sessions.get_mut(id)
|
||||
}
|
||||
|
||||
/// Remove a session.
|
||||
pub fn remove_session(&mut self, id: &SessionId) -> Option<RelaySession> {
|
||||
self.sessions.remove(id)
|
||||
}
|
||||
|
||||
/// Number of active sessions.
|
||||
pub fn active_count(&self) -> usize {
|
||||
self.sessions.values().filter(|s| s.is_active()).count()
|
||||
}
|
||||
|
||||
/// Total sessions (including inactive/closing).
|
||||
pub fn total_count(&self) -> usize {
|
||||
self.sessions.len()
|
||||
}
|
||||
|
||||
/// Remove sessions idle for longer than `timeout_ms`.
|
||||
pub fn expire_idle(&mut self, now_ms: u64, timeout_ms: u64) -> usize {
|
||||
let before = self.sessions.len();
|
||||
self.sessions
|
||||
.retain(|_, s| now_ms.saturating_sub(s.last_activity_ms) < timeout_ms);
|
||||
before - self.sessions.len()
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn create_and_get_session() {
|
||||
let mut mgr = SessionManager::new(10);
|
||||
let id = [1u8; 16];
|
||||
mgr.create_session(id, PipelineConfig::default());
|
||||
assert_eq!(mgr.total_count(), 1);
|
||||
assert!(mgr.get_session(&id).is_some());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn respects_max_sessions() {
|
||||
let mut mgr = SessionManager::new(1);
|
||||
mgr.create_session([1u8; 16], PipelineConfig::default());
|
||||
let result = mgr.create_session([2u8; 16], PipelineConfig::default());
|
||||
assert!(result.is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn expire_idle_removes_old() {
|
||||
let mut mgr = SessionManager::new(10);
|
||||
let id = [1u8; 16];
|
||||
mgr.create_session(id, PipelineConfig::default());
|
||||
// Session has last_activity_ms = 0, current time = 60000, timeout = 30000
|
||||
let expired = mgr.expire_idle(60_000, 30_000);
|
||||
assert_eq!(expired, 1);
|
||||
assert_eq!(mgr.total_count(), 0);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user