1mod agents;
4pub mod components;
5mod config;
6
7use std::{
8 collections::{HashMap, VecDeque},
9 io::Read as _,
10 path::{Path, PathBuf},
11 sync::{
12 Arc,
13 atomic::{AtomicU64, AtomicUsize, Ordering},
14 },
15 time::Duration,
16};
17
18use anyhow::{Context, Result, anyhow};
19use async_trait::async_trait;
20use config::Config;
21pub use config::{ApprovalPolicy, ConfigOverrides, user_home_path};
22use sha2::{Digest, Sha256};
23pub fn init_user_config() -> anyhow::Result<std::path::PathBuf> {
24 config::Config::init_user_config()
25}
26pub fn update_index_url(workspace: &std::path::Path) -> anyhow::Result<Option<String>> {
27 Ok(config::Config::load(workspace, ConfigOverrides::default())?
28 .update
29 .index_url)
30}
31
32pub use agents::{Endpoint, PI_PROVIDER, WireApi, read_secret};
33pub use scv_tools::{adapters, delegation};
34
35pub fn agent_command(agent: &str) -> Result<std::process::Command> {
40 let config = Config::load_user(ConfigOverrides::default())?;
41 config.prepare_adapter_homes()?;
42 let adapter = config
43 .adapters()
44 .remove(&format!("agent_{agent}"))
45 .ok_or_else(|| anyhow!("unknown agent {agent}"))?;
46 let executable =
47 scv_tools::adapters::resolve_agent_executable(&adapter.command, &adapter.search_dirs)
48 .ok_or_else(|| {
49 anyhow!(
50 "{agent} is not installed: {:?} was not found on PATH or in ~/.local/bin",
51 adapter.command
52 )
53 })?;
54 let mut command = std::process::Command::new(executable);
55 command.current_dir(config.instance_home.join("adapters").join(agent));
56 scv_tools::apply_agent_environment(&mut command, &adapter.environment);
57 Ok(command)
58}
59
60pub fn agent_executable(agent: &str) -> Result<Option<PathBuf>> {
62 let config = Config::load_user(ConfigOverrides::default())?;
63 let adapter = config
64 .adapters()
65 .remove(&format!("agent_{agent}"))
66 .ok_or_else(|| anyhow!("unknown agent {agent}"))?;
67 Ok(scv_tools::adapters::resolve_agent_executable(
68 &adapter.command,
69 &adapter.search_dirs,
70 ))
71}
72
73pub fn agent_home(agent: &str) -> Result<PathBuf> {
75 let config = Config::load_user(ConfigOverrides::default())?;
76 config.prepare_adapter_homes()?;
77 let home = config.instance_home.join("adapters").join(agent);
78 if !home.is_dir() {
79 return Err(anyhow!("unknown agent {agent}"));
80 }
81 Ok(home)
82}
83
84pub fn collect_agent_garbage(
88 agent: Option<&str>,
89 older_than: std::time::Duration,
90 dry_run: bool,
91) -> Result<Vec<(&'static str, scv_tools::conversation::GcReport)>> {
92 let config = Config::load_user(ConfigOverrides::default())?;
93 let markers = config.instance_home.join("run").join("conversations");
94 let mut reports = Vec::new();
95 for adapter in adapters::ADAPTERS {
96 if agent.is_some_and(|agent| agent != adapter.name) {
97 continue;
98 }
99 let Some(files) = adapter.conversation_files else {
100 continue;
101 };
102 let home = config.instance_home.join("adapters").join(adapter.name);
103 if !home.is_dir() {
104 continue;
105 }
106 let report =
107 scv_tools::conversation::collect_garbage(&home, files, &markers, older_than, dry_run)
108 .with_context(|| format!("clean {} transcripts", adapter.name))?;
109 reports.push((adapter.name, report));
110 }
111 Ok(reports)
112}
113
114pub fn conversation_age(value: &str) -> std::result::Result<std::time::Duration, String> {
116 scv_tools::conversation::parse_age(value)
117}
118
119fn key_store_home(agent: &str) -> Result<PathBuf> {
120 scv_tools::adapters::adapter(agent).ok_or_else(|| anyhow!("unknown agent {agent}"))?;
121 agent_home(agent)
122}
123
124pub fn store_agent_key(agent: &str, store: adapters::KeyStore, key: &str) -> Result<Vec<String>> {
126 agents::store_key(store, &key_store_home(agent)?, key)
127}
128
129pub fn agent_stored_status(agent: &str, store: adapters::KeyStore) -> Result<(bool, Vec<String>)> {
132 agents::stored_status(store, &key_store_home(agent)?)
133}
134
135pub fn remove_agent_credentials(agent: &str, store: adapters::KeyStore) -> Result<Vec<String>> {
137 agents::remove_stored(store, &key_store_home(agent)?)
138}
139
140pub fn configure_pi_endpoint(endpoint: &Endpoint, key: &str) -> Result<Vec<String>> {
142 agents::configure_pi_endpoint(&pi_agent_dir()?, endpoint, key)
143}
144
145pub fn import_pi_from_scv_provider() -> Result<Vec<String>> {
149 let config = Config::load_user(ConfigOverrides::default())?;
150 let provider = &config.provider;
151 if provider.kind != "openai-compatible" {
152 return Err(anyhow!("SCV's provider is not openai-compatible"));
153 }
154 let key = match (&provider.api_key, &provider.api_key_env) {
155 (Some(key), _) if !key.trim().is_empty() => key.trim().to_owned(),
156 (_, Some(variable)) => std::env::var(variable)
157 .ok()
158 .filter(|key| !key.trim().is_empty())
159 .ok_or_else(|| {
160 anyhow!("SCV's provider reads its key from ${variable}, which is not set here")
161 })?,
162 _ => return Err(anyhow!("SCV's provider has no API key configured")),
163 };
164 let endpoint = Endpoint {
165 base_url: provider.base_url.clone(),
166 api: WireApi::Responses,
167 model: provider.model.clone(),
168 };
169 let mut notes = agents::configure_pi_endpoint(&pi_agent_dir()?, &endpoint, &key)?;
170 if !provider.headers.is_empty() {
171 notes.push(
172 "Note: SCV's provider sends extra headers, which were not copied; add them to \
173 pi's models.json if the endpoint needs them"
174 .into(),
175 );
176 }
177 Ok(notes)
178}
179
180fn pi_agent_dir() -> Result<PathBuf> {
181 let descriptor =
182 scv_tools::adapters::adapter("pi").ok_or_else(|| anyhow!("unknown agent pi"))?;
183 let scv_tools::adapters::Status::Stored(scv_tools::adapters::KeyStore::Pi { dir }) =
184 descriptor.status
185 else {
186 return Err(anyhow!("pi has no SCV-managed store"));
187 };
188 Ok(agent_home("pi")?.join(dir))
189}
190
191pub fn import_codex(source: &Path) -> Result<Vec<String>> {
195 let config = Config::load_user(ConfigOverrides::default())?;
196 config.prepare_adapter_homes()?;
197 agents::import_codex(source, &config.instance_home.join("adapters").join("codex"))
198}
199
200pub fn import_grok(source: &Path) -> Result<Vec<String>> {
204 let config = Config::load_user(ConfigOverrides::default())?;
205 config.prepare_adapter_homes()?;
206 let descriptor =
207 scv_tools::adapters::adapter("grok").ok_or_else(|| anyhow!("unknown agent grok"))?;
208 let grok_home = descriptor
209 .home_environment
210 .iter()
211 .find(|(variable, _)| *variable == "GROK_HOME")
212 .map(|(_, relative)| *relative)
213 .ok_or_else(|| anyhow!("grok has no GROK_HOME in its adapter home"))?;
214 agents::import_grok(
215 source,
216 &config
217 .instance_home
218 .join("adapters")
219 .join("grok")
220 .join(grok_home),
221 )
222}
223
224pub fn service_name() -> anyhow::Result<String> {
226 if std::env::var_os("SCV_HOME").is_none() {
227 return Ok("scv.service".into());
228 }
229 let home = user_home_path().ok_or_else(|| anyhow!("cannot determine SCV instance home"))?;
230 let digest = Sha256::digest(home.to_string_lossy().as_bytes());
231 let suffix = digest[..8]
232 .iter()
233 .map(|byte| format!("{byte:02x}"))
234 .collect::<String>();
235 Ok(format!("scv-{suffix}.service"))
236}
237
238pub fn service_unit_path() -> anyhow::Result<std::path::PathBuf> {
239 let config =
240 dirs::config_dir().ok_or_else(|| anyhow!("cannot determine XDG config directory"))?;
241 Ok(config.join("systemd/user").join(service_name()?))
242}
243use scv_core::{
244 AgentError, AgentRuntime, ApprovalGate, ApprovalRequest, BudgetContextPolicy, CoreEvent,
245 EventSink, Message, ToolRegistry, ToolRisk,
246};
247use scv_protocol::{
248 ClientMessage, DaemonCommand, DaemonStatus, DelegationInfo, DelegationSummary,
249 PROTOCOL_VERSION, PeerInfo, QueueEntry, ServerEvent, Usage,
250};
251use scv_provider_openai::OpenAiProvider;
252use scv_tools::{
253 DelegationContext, SkillMap, builtin_registry,
254 delegation::{self as delegations, DelegationRegistry},
255};
256use tokio::{
257 io::{AsyncBufRead, AsyncBufReadExt, AsyncWriteExt, BufReader},
258 net::{UnixListener, UnixStream},
259 sync::{Mutex, OwnedSemaphorePermit, Semaphore, mpsc, oneshot},
260 task::JoinHandle,
261};
262use tokio_util::sync::CancellationToken;
263use tokio_util::task::TaskTracker;
264use uuid::Uuid;
265
266const PROMPT_LIMIT_BYTES: usize = 256 * 1024;
267const OUTPUT_QUEUE_CAPACITY: usize = 256;
268const OUTPUT_QUEUE_MIN_BYTES: usize = 16 * 1024 * 1024;
269const SHUTDOWN_GRACE: Duration = Duration::from_secs(3);
270const MAX_QUEUE_ITEMS: usize = 64;
271const MAX_QUEUE_BYTES: usize = 4 * 1024 * 1024;
272
273pub async fn run_stdio(overrides: ConfigOverrides) -> Result<()> {
274 let stdin = tokio::io::stdin();
275 let stdout = tokio::io::stdout();
276 let tasks = TaskTracker::new();
277 let registry = instance_delegations()?;
278 tokio::spawn(reconcile_delegations(Arc::clone(®istry)));
281 let result = run_managed(
282 stdin,
283 stdout,
284 overrides,
285 None,
286 registry,
287 CancellationToken::new(),
288 tasks.clone(),
289 )
290 .await;
291 tasks.close();
292 tasks.wait().await;
293 result
294}
295
296pub fn default_socket_path() -> Result<PathBuf> {
298 scv_client::default_socket_path()
299}
300
301pub async fn run_socket(path: &Path, overrides: ConfigOverrides) -> Result<()> {
303 if let Some(parent) = path.parent() {
304 tokio::fs::create_dir_all(parent)
305 .await
306 .context("create SCV socket directory")?;
307 }
308 let _lock = SocketLock::acquire(path)?;
309 if path.exists() {
310 if UnixStream::connect(path).await.is_ok() {
311 return Err(anyhow!(
312 "SCV server is already running at {}",
313 path.display()
314 ));
315 }
316 use std::os::unix::fs::FileTypeExt;
317 if !std::fs::symlink_metadata(path)?.file_type().is_socket() {
318 return Err(anyhow!(
319 "refusing to remove a non-socket at SCV socket path"
320 ));
321 }
322 tokio::fs::remove_file(path)
323 .await
324 .with_context(|| format!("remove stale SCV socket {}", path.display()))?;
325 }
326 let listener = UnixListener::bind(path)
327 .with_context(|| format!("bind SCV server socket {}", path.display()))?;
328 #[cfg(unix)]
329 {
330 use std::os::unix::fs::PermissionsExt;
331 std::fs::set_permissions(path, std::fs::Permissions::from_mode(0o600))
332 .context("secure SCV socket")?;
333 }
334 let components = Arc::new(Mutex::new(components::Components::new(
335 path.to_owned(),
336 std::env::current_dir()?,
337 )));
338 let registry = instance_delegations()?;
339 if !delegations::become_child_subreaper() {
341 tracing::debug!("SCV daemon is not a child subreaper on this platform");
342 }
343 let cancellation = CancellationToken::new();
344 let delegation_registry = Arc::clone(®istry);
345 let delegation_cancel = cancellation.clone();
346 let mut delegation_task = tokio::spawn(async move {
347 let mut interval = tokio::time::interval(DELEGATION_RECONCILE_INTERVAL);
349 loop {
350 tokio::select! {
351 biased;
352 _ = delegation_cancel.cancelled() => break,
353 _ = interval.tick() => {
354 reconcile_delegations(Arc::clone(&delegation_registry)).await;
355 let zombies = delegations::reap_orphaned_zombies();
356 if zombies > 0 {
357 tracing::debug!("Reaped {zombies} exited orphan processes");
358 }
359 }
360 }
361 }
362 });
363 let _delegation_abort = AbortGuard(delegation_task.abort_handle());
364 let tasks = TaskTracker::new();
365 let mut clients = tokio::task::JoinSet::new();
366 let refresh_components = components.clone();
367 let refresh_cancel = cancellation.clone();
368 let mut refresh_task = tokio::spawn(async move {
369 let mut refresh = tokio::time::interval(Duration::from_secs(2));
370 loop {
371 tokio::select! {
372 biased;
373 _ = refresh_cancel.cancelled() => break,
374 _ = refresh.tick() => {
375 tokio::select! {
376 biased;
377 _ = refresh_cancel.cancelled() => break,
378 result = async { refresh_components.lock().await.reconcile().await } => {
379 if result.is_err() { tracing::warn!("Component account discovery failed"); }
380 }
381 }
382 }
383 }
384 }
385 });
386 let _refresh_abort = AbortGuard(refresh_task.abort_handle());
387 let mut terminate = tokio::signal::unix::signal(tokio::signal::unix::SignalKind::terminate())?;
388 let result = loop {
389 tokio::select! {
390 accepted = listener.accept() => {
391 let (stream, _) = match accepted { Ok(value) => value, Err(error) => break Err(error.into()) };
392 let child_overrides = overrides.clone();
393 let components = components.clone();
394 let registry = Arc::clone(®istry);
395 let cancellation = cancellation.clone();
396 let tasks = tasks.clone();
397 clients.spawn(async move {
398 let (reader, writer) = stream.into_split();
399 if run_managed(reader, writer, child_overrides, Some(components), registry, cancellation, tasks).await.is_err() {
400 tracing::warn!("SCV socket client stopped");
401 }
402 });
403 }
404 _ = clients.join_next(), if !clients.is_empty() => {},
405 _ = tokio::signal::ctrl_c() => break Ok(()),
406 _ = terminate.recv() => break Ok(()),
407 }
408 };
409 drop(listener);
410 cancellation.cancel();
411 let _ = (&mut refresh_task).await;
412 let _ = (&mut delegation_task).await;
413 components.lock().await.shutdown().await;
414 if tokio::time::timeout(Duration::from_secs(8), async {
415 while clients.join_next().await.is_some() {}
416 })
417 .await
418 .is_err()
419 {
420 clients.abort_all();
421 while clients.join_next().await.is_some() {}
422 }
423 tasks.close();
424 tasks.wait().await;
425 let _ = tokio::fs::remove_file(path).await;
426 result
427}
428
429const DELEGATION_RECONCILE_INTERVAL: Duration = Duration::from_secs(60);
430
431fn instance_delegations() -> Result<Arc<DelegationRegistry>> {
433 let home =
434 config::user_home_path().ok_or_else(|| anyhow!("cannot determine SCV instance home"))?;
435 Ok(Arc::new(DelegationRegistry::new(&home)))
436}
437
438async fn reconcile_delegations(registry: Arc<DelegationRegistry>) {
440 let report = registry.reconcile().await;
441 if !report.reaped.is_empty() {
442 tracing::info!(
443 "Reaped {} orphaned delegations: {}",
444 report.reaped.len(),
445 report.reaped.join(", ")
446 );
447 }
448 if report.removed > 0 {
449 tracing::debug!(
450 "Removed {} delegation records whose processes had exited",
451 report.removed
452 );
453 }
454 if report.stale_markers > 0 {
455 tracing::debug!(
456 "Removed {} conversation markers whose SCV process had exited",
457 report.stale_markers
458 );
459 }
460}
461
462enum ControlFailure {
464 Delegation(String),
466 Component,
467}
468
469async fn daemon_control(
471 components: &Arc<Mutex<components::Components>>,
472 registry: &DelegationRegistry,
473 command: DaemonCommand,
474) -> std::result::Result<DaemonStatus, ControlFailure> {
475 let mut killed = Vec::new();
476 let listing = match &command {
477 DaemonCommand::Delegations { all } => Some(*all),
478 DaemonCommand::DelegationKill { handle, orphans } => {
479 if handle.is_none() && !orphans {
480 return Err(ControlFailure::Delegation(
481 "name a delegation handle or ask for orphans".into(),
482 ));
483 }
484 if *orphans {
485 let report = registry.reconcile().await;
486 killed.extend(report.reaped);
487 }
488 if let Some(handle) = handle {
489 registry
490 .kill(handle)
491 .await
492 .map_err(ControlFailure::Delegation)?;
493 killed.push(handle.clone());
494 }
495 Some(true)
496 }
497 _ => None,
498 };
499 let mut status = components
500 .lock()
501 .await
502 .control(command)
503 .await
504 .map_err(|_| ControlFailure::Component)?;
505 let running = registry.list(false);
506 status.delegations = DelegationSummary {
507 active: running.len() as u64,
508 reaped: registry.reaped_total(),
509 entries: match listing {
510 Some(true) => registry.list(true),
511 Some(false) => running,
512 None => Vec::new(),
513 }
514 .into_iter()
515 .map(|entry| DelegationInfo {
516 handle: entry.record.handle,
517 agent: entry.record.agent,
518 session: entry.record.session,
519 depth: entry.record.depth,
520 pid: entry.record.process.pid,
521 owner_pid: entry.record.owner.pid,
522 processes: u32::try_from(entry.processes).unwrap_or(u32::MAX),
523 cwd: entry.record.cwd.display().to_string(),
524 started_unix_seconds: entry.record.started_unix,
525 orphaned: entry.orphaned,
526 conversation: entry.record.conversation,
527 turn: entry.record.turn,
528 })
529 .collect(),
530 killed,
531 };
532 Ok(status)
533}
534
535async fn run_managed<R, W>(
536 reader: R,
537 writer: W,
538 overrides: ConfigOverrides,
539 components: Option<Arc<Mutex<components::Components>>>,
540 registry: Arc<DelegationRegistry>,
541 cancellation: CancellationToken,
542 tasks: TaskTracker,
543) -> Result<()>
544where
545 R: tokio::io::AsyncRead + Unpin,
546 W: tokio::io::AsyncWrite + Unpin + Send + 'static,
547{
548 let initial_output_bytes =
549 output_queue_bytes(Config::default().protocol.max_server_frame_bytes)?;
550 let (output_tx, mut output_rx) = outbound_channel(initial_output_bytes);
551 let mut writer_task = tasks.spawn(async move {
552 let mut writer = writer;
553 while let Some(frame) = output_rx.recv().await {
554 writer.write_all(&frame.bytes).await?;
555 writer.write_all(b"\n").await?;
556 writer.flush().await?;
557 }
558 Ok::<(), std::io::Error>(())
559 });
560 let _writer_abort = AbortGuard(writer_task.abort_handle());
561 let (done_tx, mut done_rx) = mpsc::channel::<TurnDone>(4);
562 let approvals = Arc::new(ApprovalBroker::default());
563 let mut reader = BufReader::new(reader);
564 let mut frames = FrameBuffer::default();
565 let mut initialized = false;
566 let mut session: Option<Session> = None;
567 let mut active: Option<ActiveTurn> = None;
568 let mut fatal = false;
569 let mut writer_finished = false;
570
571 let loop_result: Result<()> = async {
572 loop {
573 let frame_limit = session.as_ref().map_or_else(
574 || Config::default().protocol.max_client_frame_bytes,
575 |value| value.config.protocol.max_client_frame_bytes,
576 );
577 tokio::select! {
578 _ = cancellation.cancelled() => break,
579 read = frames.read(&mut reader, frame_limit) => {
580 let frame = match read.context("read protocol input")? {
581 FrameRead::Eof => {
582 if let Some(active) = &active { active.cancellation.cancel(); }
583 break;
584 }
585 FrameRead::TooLarge => {
586 send_error(&output_tx, "", "invalid_request", "client frame exceeds configured limit", false, server_frame_limit(&session)).await?;
587 continue;
588 }
589 FrameRead::Frame(frame) => frame,
590 };
591 if frame.is_empty() {
592 send_error(&output_tx, "", "invalid_json", "protocol frame is empty", false, server_frame_limit(&session)).await?;
593 continue;
594 }
595 let message = match serde_json::from_slice::<ClientMessage>(&frame) {
596 Ok(message) => message,
597 Err(error) => {
598 send_error(&output_tx, "", "invalid_json", &format!("invalid protocol JSON: {error}"), false, server_frame_limit(&session)).await?;
599 continue;
600 }
601 };
602 match message {
603 ClientMessage::Initialize { request_id, protocol_version, .. } => {
604 if initialized {
605 send_error(&output_tx, &request_id, "invalid_request", "connection is already initialized", false, server_frame_limit(&session)).await?;
606 continue;
607 }
608 if protocol_version != PROTOCOL_VERSION {
609 send_error(&output_tx, &request_id, "version_mismatch", &format!("server supports protocol {PROTOCOL_VERSION}"), true, server_frame_limit(&session)).await?;
610 fatal = true;
611 break;
612 }
613 initialized = true;
614 send_event(&output_tx, ServerEvent::Initialized {
615 request_id,
616 protocol_version: PROTOCOL_VERSION,
617 server: PeerInfo { name: "scv-server".into(), version: env!("CARGO_PKG_VERSION").into() },
618 }, Config::default().protocol.max_server_frame_bytes).await?;
619 }
620 other if !initialized => {
621 send_error(&output_tx, other.request_id(), "not_initialized", "initialize must be the first message", false, server_frame_limit(&session)).await?;
622 }
623 ClientMessage::DaemonControl { request_id, command } => {
624 if let Some(components) = &components {
625 let result = tokio::select! {
626 biased;
627 _ = cancellation.cancelled() => break,
628 result = daemon_control(components, ®istry, command) => result,
629 };
630 match result {
631 Ok(status) => send_event(&output_tx, ServerEvent::DaemonStatus { request_id, status }, server_frame_limit(&session)).await?,
632 Err(ControlFailure::Delegation(message)) => send_error(&output_tx, &request_id, "delegation_error", &message, false, server_frame_limit(&session)).await?,
633 Err(ControlFailure::Component) => send_error(&output_tx, &request_id, "component_error", "Component operation failed; check account credentials, private file permissions and absolute workspace", false, server_frame_limit(&session)).await?,
634 }
635 } else {
636 send_error(&output_tx, &request_id, "unsupported", "Component management requires the daemon socket", false, server_frame_limit(&session)).await?;
637 }
638 }
639 ClientMessage::SessionStart { request_id, cwd, provider, model, base_url, no_tools } => {
640 if session.is_some() {
641 send_error(&output_tx, &request_id, "invalid_request", "this connection already has a session", false, server_frame_limit(&session)).await?;
642 continue;
643 }
644 let session_overrides = ConfigOverrides {
645 provider: provider.or_else(|| overrides.provider.clone()),
646 model: model.or_else(|| overrides.model.clone()),
647 base_url: base_url.or_else(|| overrides.base_url.clone()),
648 approval_policy: overrides.approval_policy,
649 no_tools: no_tools.unwrap_or(overrides.no_tools),
650 };
651 match build_session(&cwd, session_overrides, ®istry).await {
652 Ok(new_session) => {
653 output_tx.ensure_capacity(output_queue_bytes(
654 new_session.config.protocol.max_server_frame_bytes,
655 )?)?;
656 let event = ServerEvent::SessionStarted {
657 request_id,
658 session_id: new_session.id.clone(),
659 cwd: new_session.workspace.display().to_string(),
660 model: new_session.runtime.model().to_owned(),
661 context_max_tokens: new_session.config.context.max_tokens,
662 max_server_frame_bytes: new_session.config.protocol.max_server_frame_bytes,
663 max_transcript_bytes: new_session.config.tui.max_transcript_bytes,
664 max_transcript_items: new_session.config.tui.max_transcript_items,
665 max_prompt_history_bytes: new_session.config.tui.max_prompt_history_bytes,
666 max_prompt_history_items: new_session.config.tui.max_prompt_history_items,
667 };
668 send_event(&output_tx, event, new_session.config.protocol.max_server_frame_bytes).await?;
669 send_event(&output_tx, ServerEvent::QueueSnapshot {
670 request_id: None,
671 session_id: new_session.id.clone(),
672 seq: next_seq(&new_session.seq),
673 entries: new_session.queue.lock().await.iter().cloned().collect(),
674 paused: new_session.paused.load(Ordering::Acquire),
675 }, new_session.config.protocol.max_server_frame_bytes).await?;
676 session = Some(new_session);
677 }
678 Err(error) => {
679 send_error(&output_tx, &request_id, "invalid_request", &error.to_string(), false, server_frame_limit(&session)).await?;
680 }
681 }
682 }
683 ClientMessage::SessionAttach { request_id, .. } => {
684 send_error(&output_tx, &request_id, "unsupported", "session attach requires the shared socket server", false, server_frame_limit(&session)).await?;
685 }
686 ClientMessage::TurnStart { request_id, session_id, prompt } => {
687 let Some(current) = session.as_ref() else {
688 send_error(&output_tx, &request_id, "session_not_found", "start a session first", false, server_frame_limit(&session)).await?;
689 continue;
690 };
691 if current.id != session_id {
692 send_error(&output_tx, &request_id, "session_not_found", "session id does not match", false, server_frame_limit(&session)).await?;
693 continue;
694 }
695 if prompt.trim().is_empty() || prompt.len() > PROMPT_LIMIT_BYTES {
696 send_error(&output_tx, &request_id, "invalid_request", "prompt must be non-empty and no larger than 256 KiB", false, server_frame_limit(&session)).await?;
697 continue;
698 }
699 if active.is_some() {
700 let entry = match current.enqueue(prompt, request_id.clone()).await {
701 Ok(entry) => entry,
702 Err(code) => { send_error(&output_tx, &request_id, code, "session queue limit reached", false, server_frame_limit(&session)).await?; continue; }
703 };
704 let position = current.queue.lock().await.len().saturating_sub(1);
705 send_event(&output_tx, ServerEvent::QueueEnqueued {
706 request_id, session_id: current.id.clone(), seq: next_seq(¤t.seq), entry, position,
707 }, current.config.protocol.max_server_frame_bytes).await?;
708 continue;
709 }
710 let turn_id = Uuid::new_v4().to_string();
711 let cancellation = cancellation.child_token();
712 let meta = TurnMeta {
713 request_id: request_id.clone(),
714 session_id: current.id.clone(),
715 turn_id: turn_id.clone(),
716 seq: Arc::clone(¤t.seq),
717 max_server_frame: current.config.protocol.max_server_frame_bytes,
718 };
719 send_event(&output_tx, ServerEvent::TurnStarted {
720 request_id: request_id.clone(),
721 session_id: current.id.clone(),
722 turn_id: turn_id.clone(),
723 seq: next_seq(¤t.seq),
724 }, current.config.protocol.max_server_frame_bytes).await?;
725 let runtime = Arc::clone(¤t.runtime);
726 let history = Arc::clone(¤t.history);
727 let sink: Arc<dyn EventSink> = Arc::new(ProtocolSink {
728 meta: meta.clone(),
729 output: output_tx.clone(),
730 cancellation: cancellation.clone(),
731 });
732 let gate: Arc<dyn ApprovalGate> = Arc::new(ProtocolApprovalGate {
733 policy: current.config.tools.approval_policy,
734 broker: Arc::clone(&approvals),
735 meta,
736 output: output_tx.clone(),
737 });
738 let task_cancel = cancellation.clone();
739 let task_done = done_tx.clone();
740 let task_request = request_id.clone();
741 let task_session = current.id.clone();
742 let task_turn = turn_id.clone();
743 let task = tasks.spawn(async move {
744 let mut history = history.lock().await;
745 let result = runtime.run_turn(&mut history, prompt, sink, gate, task_cancel).await;
746 let _ = task_done.send(TurnDone {
747 request_id: task_request,
748 session_id: task_session,
749 turn_id: task_turn,
750 result,
751 }).await;
752 });
753 active = Some(ActiveTurn { turn_id, cancellation, task });
754 }
755 ClientMessage::QueueUpdate { request_id, session_id, queue_id, revision, prompt } => {
756 let Some(current) = session.as_ref() else { send_error(&output_tx, &request_id, "session_not_found", "start a session first", false, server_frame_limit(&session)).await?; continue; };
757 if current.id != session_id { send_error(&output_tx, &request_id, "session_not_found", "session id does not match", false, server_frame_limit(&session)).await?; continue; }
758 if prompt.trim().is_empty() || prompt.len() > PROMPT_LIMIT_BYTES { send_error(&output_tx, &request_id, "invalid_request", "prompt must be non-empty and no larger than 256 KiB", false, server_frame_limit(&session)).await?; continue; }
759 match current.update_queue(&queue_id, revision, prompt).await {
760 Ok(entry) => send_event(&output_tx, ServerEvent::QueueUpdated { request_id, session_id: current.id.clone(), seq: next_seq(¤t.seq), entry }, current.config.protocol.max_server_frame_bytes).await?,
761 Err(code) => send_error(&output_tx, &request_id, code, "queue entry was not found or revision is stale", false, server_frame_limit(&session)).await?,
762 }
763 }
764 ClientMessage::QueueMove { request_id, session_id, queue_id, revision, before_queue_id } => {
765 let Some(current) = session.as_ref() else { send_error(&output_tx, &request_id, "session_not_found", "start a session first", false, server_frame_limit(&session)).await?; continue; };
766 match current.move_queue(&session_id, &queue_id, revision, before_queue_id).await {
767 Ok((id, rev, pos)) => send_event(&output_tx, ServerEvent::QueueMoved { request_id, session_id: current.id.clone(), seq: next_seq(¤t.seq), queue_id: id, position: pos, revision: rev }, current.config.protocol.max_server_frame_bytes).await?,
768 Err(code) => send_error(&output_tx, &request_id, code, "queue entry was not found or revision is stale", false, server_frame_limit(&session)).await?,
769 }
770 }
771 ClientMessage::QueueRemove { request_id, session_id, queue_id, revision } => {
772 let Some(current) = session.as_ref() else { send_error(&output_tx, &request_id, "session_not_found", "start a session first", false, server_frame_limit(&session)).await?; continue; };
773 match current.remove_queue(&session_id, &queue_id, revision).await {
774 Ok((id, rev)) => send_event(&output_tx, ServerEvent::QueueRemoved { request_id, session_id: current.id.clone(), seq: next_seq(¤t.seq), queue_id: id, revision: rev }, current.config.protocol.max_server_frame_bytes).await?,
775 Err(code) => send_error(&output_tx, &request_id, code, "queue entry was not found or revision is stale", false, server_frame_limit(&session)).await?,
776 }
777 }
778 ClientMessage::SessionPause { request_id, session_id, paused } => {
779 let Some(current) = session.as_ref() else { send_error(&output_tx, &request_id, "session_not_found", "start a session first", false, server_frame_limit(&session)).await?; continue; };
780 if current.id != session_id { send_error(&output_tx, &request_id, "session_not_found", "session id does not match", false, server_frame_limit(&session)).await?; continue; }
781 current.paused.store(paused, Ordering::Release);
782 send_event(&output_tx, ServerEvent::SessionPaused { request_id, session_id: current.id.clone(), seq: next_seq(¤t.seq), paused }, current.config.protocol.max_server_frame_bytes).await?;
783 }
784 ClientMessage::TurnCancel { request_id, session_id, turn_id } => {
785 match (&session, &active) {
786 (Some(current), Some(running)) if current.id == session_id && running.turn_id == turn_id => running.cancellation.cancel(),
787 _ => send_error(&output_tx, &request_id, "turn_not_found", "active turn was not found", false, server_frame_limit(&session)).await?,
788 }
789 }
790 ClientMessage::ApprovalResolve { request_id, session_id, approval_id, approved } => {
791 if session.as_ref().is_none_or(|current| current.id != session_id) {
792 send_error(&output_tx, &request_id, "session_not_found", "session id does not match", false, server_frame_limit(&session)).await?;
793 } else if !approvals.resolve(&approval_id, approved).await {
794 send_error(&output_tx, &request_id, "approval_not_found", "approval was not found or already resolved", false, server_frame_limit(&session)).await?;
795 }
796 }
797 ClientMessage::SessionClear { request_id, session_id } => {
798 let Some(current) = session.as_ref() else {
799 send_error(&output_tx, &request_id, "session_not_found", "session was not found", false, server_frame_limit(&session)).await?;
800 continue;
801 };
802 if current.id != session_id {
803 send_error(&output_tx, &request_id, "session_not_found", "session id does not match", false, server_frame_limit(&session)).await?;
804 } else if active.is_some() {
805 send_error(&output_tx, &request_id, "turn_active", "cancel the active turn before clearing", false, server_frame_limit(&session)).await?;
806 } else {
807 current.history.lock().await.clear();
808 current.queue.lock().await.clear();
809 send_event(&output_tx, ServerEvent::SessionCleared {
810 request_id,
811 session_id: current.id.clone(),
812 seq: next_seq(¤t.seq),
813 }, current.config.protocol.max_server_frame_bytes).await?;
814 send_event(&output_tx, ServerEvent::QueueSnapshot { request_id: None, session_id: current.id.clone(), seq: next_seq(¤t.seq), entries: Vec::new(), paused: current.paused.load(Ordering::Acquire) }, current.config.protocol.max_server_frame_bytes).await?;
815 }
816 }
817 }
818 }
819 writer = &mut writer_task => {
820 writer_finished = true;
821 writer.context("join protocol writer")??;
822 break;
823 }
824 done = done_rx.recv(), if active.is_some() => {
825 if let Some(done) = done {
826 if let Some(current) = session.as_ref() {
827 let seq = next_seq(¤t.seq);
828 let event = match done.result {
829 Ok(outcome) => ServerEvent::TurnCompleted {
830 request_id: done.request_id,
831 session_id: done.session_id,
832 turn_id: done.turn_id,
833 seq,
834 steps: outcome.steps,
835 usage: Usage { input_tokens: outcome.usage.input_tokens, output_tokens: outcome.usage.output_tokens },
836 },
837 Err(AgentError::Cancelled) => ServerEvent::TurnCancelled {
838 request_id: done.request_id,
839 session_id: done.session_id,
840 turn_id: done.turn_id,
841 seq,
842 },
843 Err(error) => ServerEvent::TurnFailed {
844 request_id: done.request_id,
845 session_id: done.session_id,
846 turn_id: done.turn_id,
847 seq,
848 code: error.code().into(),
849 message: error.to_string(),
850 },
851 };
852 send_event(&output_tx, event, current.config.protocol.max_server_frame_bytes).await?;
853 }
854 if let Some(mut active) = active.take() {
855 let _ = (&mut active.task).await;
856 }
857 if let Some(current) = session.as_ref()
858 && !current.paused.load(Ordering::Acquire)
859 && let Some(entry) = current.queue.lock().await.pop_front()
860 {
861 let turn_id = Uuid::new_v4().to_string();
862 let cancellation = cancellation.child_token();
863 send_event(&output_tx, ServerEvent::QueueDequeued {
864 request_id: entry.submitter.clone(),
865 session_id: current.id.clone(),
866 seq: next_seq(¤t.seq),
867 queue_id: entry.queue_id,
868 turn_id: turn_id.clone(),
869 }, current.config.protocol.max_server_frame_bytes).await?;
870 send_event(&output_tx, ServerEvent::TurnStarted {
871 request_id: entry.submitter.clone(),
872 session_id: current.id.clone(),
873 turn_id: turn_id.clone(),
874 seq: next_seq(¤t.seq),
875 }, current.config.protocol.max_server_frame_bytes).await?;
876 let meta = TurnMeta {
877 request_id: entry.submitter.clone(), session_id: current.id.clone(), turn_id: turn_id.clone(),
878 seq: Arc::clone(¤t.seq), max_server_frame: current.config.protocol.max_server_frame_bytes,
879 };
880 let sink: Arc<dyn EventSink> = Arc::new(ProtocolSink { meta: meta.clone(), output: output_tx.clone(), cancellation: cancellation.clone() });
881 let gate: Arc<dyn ApprovalGate> = Arc::new(ProtocolApprovalGate { policy: current.config.tools.approval_policy, broker: Arc::clone(&approvals), meta, output: output_tx.clone() });
882 let runtime = Arc::clone(¤t.runtime);
883 let history = Arc::clone(¤t.history);
884 let task_done = done_tx.clone();
885 let task_request = entry.submitter;
886 let task_session = current.id.clone();
887 let task_turn = turn_id.clone();
888 let task_cancel = cancellation.clone();
889 let task = tasks.spawn(async move {
890 let mut history = history.lock().await;
891 let result = runtime.run_turn(&mut history, entry.prompt, sink, gate, task_cancel).await;
892 let _ = task_done.send(TurnDone { request_id: task_request, session_id: task_session, turn_id: task_turn, result }).await;
893 });
894 active = Some(ActiveTurn { turn_id, cancellation, task });
895 }
896 }
897 }
898 }
899 }
900 Ok(())
901 }
902 .await;
903
904 if let Some(active) = active.take() {
905 shutdown_active_turn(active, SHUTDOWN_GRACE).await;
906 }
907 drop(output_tx);
908 let writer_result = if writer_finished {
909 Ok(())
910 } else {
911 shutdown_writer(writer_task, SHUTDOWN_GRACE).await
912 };
913 loop_result?;
914 writer_result?;
915 if fatal {
916 return Err(anyhow!("protocol version mismatch"));
917 }
918 Ok(())
919}
920
921enum FrameRead {
922 Eof,
923 Frame(Vec<u8>),
924 TooLarge,
925}
926
927struct OutboundFrame {
928 bytes: Vec<u8>,
929 _byte_permit: OwnedSemaphorePermit,
930}
931
932#[derive(Clone)]
933struct OutboundSender {
934 frames: mpsc::Sender<OutboundFrame>,
935 budget: Arc<Semaphore>,
936 capacity: Arc<AtomicUsize>,
937}
938
939#[derive(Debug, PartialEq, Eq)]
940enum OutboundSendError {
941 Cancelled,
942 Closed,
943 TimedOut,
944 FrameExceedsQueue { frame_bytes: usize, capacity: usize },
945}
946
947impl std::fmt::Display for OutboundSendError {
948 fn fmt(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
949 match self {
950 Self::Cancelled => formatter.write_str("outbound send cancelled"),
951 Self::Closed => formatter.write_str("protocol client disconnected"),
952 Self::TimedOut => formatter.write_str("outbound send timed out under backpressure"),
953 Self::FrameExceedsQueue {
954 frame_bytes,
955 capacity,
956 } => write!(
957 formatter,
958 "outbound frame uses {frame_bytes} bytes but queue capacity is {capacity} bytes"
959 ),
960 }
961 }
962}
963
964impl std::error::Error for OutboundSendError {}
965
966fn outbound_channel(capacity: usize) -> (OutboundSender, mpsc::Receiver<OutboundFrame>) {
967 let (frames, receiver) = mpsc::channel(OUTPUT_QUEUE_CAPACITY);
968 (
969 OutboundSender {
970 frames,
971 budget: Arc::new(Semaphore::new(capacity)),
972 capacity: Arc::new(AtomicUsize::new(capacity)),
973 },
974 receiver,
975 )
976}
977
978impl OutboundSender {
979 fn ensure_capacity(&self, required: usize) -> Result<()> {
980 if required > Semaphore::MAX_PERMITS {
981 return Err(anyhow!(
982 "outbound queue capacity {required} exceeds runtime limit {}",
983 Semaphore::MAX_PERMITS
984 ));
985 }
986 let current = self.capacity.load(Ordering::Acquire);
987 if required > current {
988 self.budget.add_permits(required - current);
989 self.capacity.store(required, Ordering::Release);
990 }
991 Ok(())
992 }
993
994 async fn send(
995 &self,
996 bytes: Vec<u8>,
997 cancellation: Option<&CancellationToken>,
998 ) -> std::result::Result<(), OutboundSendError> {
999 self.send_with_timeout(bytes, cancellation, SHUTDOWN_GRACE)
1000 .await
1001 }
1002
1003 async fn send_with_timeout(
1004 &self,
1005 bytes: Vec<u8>,
1006 cancellation: Option<&CancellationToken>,
1007 control_timeout: Duration,
1008 ) -> std::result::Result<(), OutboundSendError> {
1009 let frame_bytes =
1010 bytes
1011 .len()
1012 .checked_add(1)
1013 .ok_or(OutboundSendError::FrameExceedsQueue {
1014 frame_bytes: usize::MAX,
1015 capacity: self.capacity.load(Ordering::Acquire),
1016 })?;
1017 let capacity = self.capacity.load(Ordering::Acquire);
1018 let permits =
1019 u32::try_from(frame_bytes).map_err(|_| OutboundSendError::FrameExceedsQueue {
1020 frame_bytes,
1021 capacity,
1022 })?;
1023 if frame_bytes > capacity {
1024 return Err(OutboundSendError::FrameExceedsQueue {
1025 frame_bytes,
1026 capacity,
1027 });
1028 }
1029
1030 let control_deadline = tokio::time::Instant::now() + control_timeout;
1031 let acquire = Arc::clone(&self.budget).acquire_many_owned(permits);
1032 let permit = if let Some(cancellation) = cancellation {
1033 tokio::select! {
1034 biased;
1035 _ = cancellation.cancelled() => return Err(OutboundSendError::Cancelled),
1036 permit = acquire => permit.map_err(|_| OutboundSendError::Closed)?,
1037 }
1038 } else {
1039 tokio::time::timeout_at(control_deadline, acquire)
1040 .await
1041 .map_err(|_| OutboundSendError::TimedOut)?
1042 .map_err(|_| OutboundSendError::Closed)?
1043 };
1044 let frame = OutboundFrame {
1045 bytes,
1046 _byte_permit: permit,
1047 };
1048 if let Some(cancellation) = cancellation {
1049 tokio::select! {
1050 biased;
1051 _ = cancellation.cancelled() => Err(OutboundSendError::Cancelled),
1052 result = self.frames.send(frame) => result.map_err(|_| OutboundSendError::Closed),
1053 }
1054 } else {
1055 tokio::time::timeout_at(control_deadline, self.frames.send(frame))
1056 .await
1057 .map_err(|_| OutboundSendError::TimedOut)?
1058 .map_err(|_| OutboundSendError::Closed)
1059 }
1060 }
1061}
1062
1063fn output_queue_bytes(max_frame_bytes: usize) -> Result<usize> {
1064 let required = max_frame_bytes
1065 .checked_add(1)
1066 .and_then(|bytes| bytes.checked_mul(2))
1067 .ok_or_else(|| anyhow!("configured server frame limit is too large"))?
1068 .max(OUTPUT_QUEUE_MIN_BYTES);
1069 if required > Semaphore::MAX_PERMITS {
1070 return Err(anyhow!(
1071 "configured server frame limit requires an outbound queue larger than the runtime supports"
1072 ));
1073 }
1074 Ok(required)
1075}
1076
1077#[derive(Default)]
1079struct FrameBuffer {
1080 bytes: Vec<u8>,
1081 oversized: bool,
1082}
1083
1084impl FrameBuffer {
1085 async fn read<R>(&mut self, reader: &mut R, max_bytes: usize) -> std::io::Result<FrameRead>
1086 where
1087 R: AsyncBufRead + Unpin,
1088 {
1089 loop {
1090 let available = reader.fill_buf().await?;
1091 let eof = available.is_empty();
1092 let end = available.iter().position(|b| *b == b'\n');
1093 let take = end.map_or(available.len(), |n| n + 1);
1094 if !self.oversized {
1095 if self.bytes.len().saturating_add(take) > max_bytes.saturating_add(2) {
1096 self.oversized = true;
1097 self.bytes.clear();
1098 } else {
1099 self.bytes.extend_from_slice(&available[..take]);
1100 }
1101 }
1102 reader.consume(take);
1103 if end.is_some() || eof {
1104 if std::mem::take(&mut self.oversized) {
1105 return Ok(FrameRead::TooLarge);
1106 }
1107 if eof && self.bytes.is_empty() {
1108 return Ok(FrameRead::Eof);
1109 }
1110 let mut bytes = std::mem::take(&mut self.bytes);
1111 while matches!(bytes.last(), Some(b'\n' | b'\r')) {
1112 bytes.pop();
1113 }
1114 return Ok(if bytes.len() > max_bytes {
1115 FrameRead::TooLarge
1116 } else {
1117 FrameRead::Frame(bytes)
1118 });
1119 }
1120 }
1121 }
1122}
1123
1124#[cfg(test)]
1125async fn read_bounded_frame<R: AsyncBufRead + Unpin>(
1126 reader: &mut R,
1127 max_bytes: usize,
1128) -> std::io::Result<FrameRead> {
1129 FrameBuffer::default().read(reader, max_bytes).await
1130}
1131
1132struct SocketLock(std::fs::File);
1134impl SocketLock {
1135 fn acquire(socket: &Path) -> Result<Self> {
1136 use std::os::unix::{fs::OpenOptionsExt, io::AsRawFd};
1137 let file = std::fs::OpenOptions::new()
1138 .read(true)
1139 .write(true)
1140 .create(true)
1141 .truncate(false)
1142 .mode(0o600)
1143 .custom_flags(libc::O_NOFOLLOW)
1144 .open(socket.with_extension("lock"))?;
1145 if unsafe { libc::flock(file.as_raw_fd(), libc::LOCK_EX | libc::LOCK_NB) } != 0 {
1147 return Err(anyhow!("SCV daemon already owns this socket"));
1148 }
1149 Ok(Self(file))
1150 }
1151}
1152impl Drop for SocketLock {
1153 fn drop(&mut self) {
1154 use std::os::unix::io::AsRawFd;
1155 unsafe {
1157 libc::flock(self.0.as_raw_fd(), libc::LOCK_UN);
1158 }
1159 }
1160}
1161
1162fn server_frame_limit(session: &Option<Session>) -> usize {
1163 session.as_ref().map_or_else(
1164 || Config::default().protocol.max_server_frame_bytes,
1165 |value| value.config.protocol.max_server_frame_bytes,
1166 )
1167}
1168
1169struct Session {
1170 id: String,
1171 workspace: PathBuf,
1172 config: Config,
1173 runtime: Arc<AgentRuntime>,
1174 history: Arc<Mutex<Vec<Message>>>,
1175 seq: Arc<AtomicU64>,
1176 queue: Arc<Mutex<VecDeque<QueueEntry>>>,
1177 paused: Arc<std::sync::atomic::AtomicBool>,
1178}
1179
1180impl Session {
1181 async fn enqueue(
1182 &self,
1183 prompt: String,
1184 submitter: String,
1185 ) -> std::result::Result<QueueEntry, &'static str> {
1186 let entry = QueueEntry {
1187 queue_id: Uuid::new_v4().to_string(),
1188 revision: 1,
1189 prompt,
1190 submitter,
1191 };
1192 let mut queue = self.queue.lock().await;
1193 let bytes: usize = queue.iter().map(|item| item.prompt.len()).sum();
1194 if queue.len() >= MAX_QUEUE_ITEMS
1195 || bytes.saturating_add(entry.prompt.len()) > MAX_QUEUE_BYTES
1196 {
1197 return Err("queue_limit");
1198 }
1199 queue.push_back(entry.clone());
1200 Ok(entry)
1201 }
1202
1203 async fn update_queue(
1204 &self,
1205 id: &str,
1206 revision: u64,
1207 prompt: String,
1208 ) -> std::result::Result<QueueEntry, &'static str> {
1209 let mut queue = self.queue.lock().await;
1210 let bytes: usize = queue.iter().map(|item| item.prompt.len()).sum();
1211 let entry = queue
1212 .iter_mut()
1213 .find(|entry| entry.queue_id == id)
1214 .ok_or("queue_not_found")?;
1215 if entry.revision != revision {
1216 return Err("queue_conflict");
1217 }
1218 if bytes
1219 .saturating_sub(entry.prompt.len())
1220 .saturating_add(prompt.len())
1221 > MAX_QUEUE_BYTES
1222 {
1223 return Err("queue_limit");
1224 }
1225 entry.prompt = prompt;
1226 entry.revision += 1;
1227 Ok(entry.clone())
1228 }
1229
1230 async fn move_queue(
1231 &self,
1232 session_id: &str,
1233 id: &str,
1234 revision: u64,
1235 before: Option<String>,
1236 ) -> std::result::Result<(String, u64, usize), &'static str> {
1237 if self.id != session_id {
1238 return Err("session_not_found");
1239 }
1240 let mut queue = self.queue.lock().await;
1241 let index = queue
1242 .iter()
1243 .position(|entry| entry.queue_id == id)
1244 .ok_or("queue_not_found")?;
1245 if queue[index].revision != revision {
1246 return Err("queue_conflict");
1247 }
1248 let target_index = match before.as_deref() {
1251 Some(target) if target == id => return Ok((id.to_string(), revision, index)),
1252 Some(target) => Some(
1253 queue
1254 .iter()
1255 .position(|item| item.queue_id == target)
1256 .ok_or("queue_not_found")?,
1257 ),
1258 None => None,
1259 };
1260 let mut entry = queue.remove(index).expect("queue index exists");
1261 let target = target_index.map_or(queue.len(), |target| {
1262 target.saturating_sub(usize::from(target > index))
1263 });
1264 let pos = target.min(queue.len());
1265 let id = entry.queue_id.clone();
1266 let rev = entry.revision + 1;
1267 entry.revision = rev;
1268 queue.insert(pos, entry);
1269 Ok((id, rev, pos))
1270 }
1271
1272 async fn remove_queue(
1273 &self,
1274 session_id: &str,
1275 id: &str,
1276 revision: u64,
1277 ) -> std::result::Result<(String, u64), &'static str> {
1278 if self.id != session_id {
1279 return Err("session_not_found");
1280 }
1281 let mut queue = self.queue.lock().await;
1282 let index = queue
1283 .iter()
1284 .position(|entry| entry.queue_id == id)
1285 .ok_or("queue_not_found")?;
1286 if queue[index].revision != revision {
1287 return Err("queue_conflict");
1288 }
1289 let entry = queue.remove(index).expect("queue index exists");
1290 Ok((entry.queue_id, entry.revision))
1291 }
1292}
1293
1294struct ActiveTurn {
1295 turn_id: String,
1296 cancellation: CancellationToken,
1297 task: JoinHandle<()>,
1298}
1299
1300impl Drop for ActiveTurn {
1301 fn drop(&mut self) {
1302 self.cancellation.cancel();
1303 self.task.abort();
1304 }
1305}
1306
1307struct AbortGuard(tokio::task::AbortHandle);
1308impl Drop for AbortGuard {
1309 fn drop(&mut self) {
1310 self.0.abort();
1311 }
1312}
1313
1314async fn shutdown_active_turn(mut active: ActiveTurn, grace: Duration) -> bool {
1315 active.cancellation.cancel();
1316 if tokio::time::timeout(grace, &mut active.task).await.is_ok() {
1317 true
1318 } else {
1319 active.task.abort();
1320 let _ = (&mut active.task).await;
1321 false
1322 }
1323}
1324
1325async fn shutdown_writer(
1326 mut writer: JoinHandle<std::io::Result<()>>,
1327 grace: Duration,
1328) -> Result<()> {
1329 match tokio::time::timeout(grace, &mut writer).await {
1330 Ok(result) => {
1331 result.context("join protocol writer")??;
1332 Ok(())
1333 }
1334 Err(_) => {
1335 writer.abort();
1336 let _ = writer.await;
1337 Err(anyhow!("protocol writer shutdown timed out"))
1338 }
1339 }
1340}
1341
1342struct TurnDone {
1343 request_id: String,
1344 session_id: String,
1345 turn_id: String,
1346 result: Result<scv_core::TurnOutcome, AgentError>,
1347}
1348
1349#[derive(Clone)]
1350struct TurnMeta {
1351 request_id: String,
1352 session_id: String,
1353 turn_id: String,
1354 seq: Arc<AtomicU64>,
1355 max_server_frame: usize,
1356}
1357
1358fn next_seq(sequence: &AtomicU64) -> u64 {
1359 sequence.fetch_add(1, Ordering::Relaxed) + 1
1360}
1361
1362async fn build_session(
1363 cwd: &str,
1364 overrides: ConfigOverrides,
1365 registry: &Arc<DelegationRegistry>,
1366) -> Result<Session> {
1367 let id = Uuid::new_v4().to_string();
1368 let workspace = std::fs::canonicalize(cwd).with_context(|| format!("resolve cwd {cwd}"))?;
1369 if !workspace.is_dir() {
1370 return Err(anyhow!("cwd is not a directory"));
1371 }
1372 let no_tools = overrides.no_tools;
1373 let config = Config::load(&workspace, overrides)?;
1374 if !no_tools {
1375 config.prepare_adapter_homes()?;
1376 }
1377 let provider_config = config.provider.clone();
1378 let api_key = provider_config.api_key.clone().or_else(|| {
1379 provider_config.api_key_env.as_deref().and_then(|name| std::env::var(name).ok())
1380 }).filter(|key| !key.trim().is_empty()).ok_or_else(|| anyhow!("provider credential is not configured; set provider.api_key or provider.api_key_env"))?;
1381 let skills = discover_skills(&workspace, &config, !no_tools)?;
1382 let system_prompt = build_system_prompt(&workspace, &config, &skills)?;
1383 let mut provider = OpenAiProvider::new(
1384 provider_config.model.clone(),
1385 provider_config.base_url.clone(),
1386 api_key,
1387 Duration::from_secs(provider_config.timeout_seconds),
1388 config.provider_limits(),
1389 provider_config.headers.clone(),
1390 )?;
1391 if !no_tools && config.hosted_web_search() {
1392 provider = provider.with_web_search();
1393 }
1394 let provider = Arc::new(provider);
1395 let tools = if no_tools {
1396 Arc::new(ToolRegistry::default())
1397 } else {
1398 let mut tools = config.tools();
1399 tools.delegation = Some(DelegationContext {
1400 registry: Arc::clone(registry),
1401 session: id.clone(),
1402 });
1403 let mut registry = builtin_registry(
1404 tools,
1405 skills.map,
1406 skills.roots,
1407 config.skills.max_skill_bytes,
1408 config.adapters(),
1409 )?;
1410 if let Some(web) = config.web_tools() {
1411 scv_tools::web::register(&mut registry, web)?;
1412 }
1413 Arc::new(registry)
1414 };
1415 let context = Arc::new(BudgetContextPolicy::new((&config.context).into())?);
1416 let runtime = Arc::new(AgentRuntime::new(
1417 provider,
1418 tools,
1419 context,
1420 config.core_agent(system_prompt),
1421 workspace.clone(),
1422 ));
1423 Ok(Session {
1424 id,
1425 workspace,
1426 config,
1427 runtime,
1428 history: Arc::new(Mutex::new(Vec::new())),
1429 seq: Arc::new(AtomicU64::new(0)),
1430 queue: Arc::new(Mutex::new(VecDeque::new())),
1431 paused: Arc::new(std::sync::atomic::AtomicBool::new(false)),
1432 })
1433}
1434
1435fn build_system_prompt(
1436 workspace: &Path,
1437 config: &Config,
1438 skills: &DiscoveredSkills,
1439) -> Result<String> {
1440 let mut prompt = config.agent.system_prompt.clone();
1441 prompt.push_str(&format!(
1442 "\nCurrent working directory: {}\n",
1443 workspace.display()
1444 ));
1445 let agents_path = workspace.join("AGENTS.md");
1446 if agents_path.is_file() {
1447 let canonical = std::fs::canonicalize(&agents_path).context("resolve project AGENTS.md")?;
1448 if !canonical.starts_with(workspace) {
1449 return Err(anyhow!("project AGENTS.md escaped workspace"));
1450 }
1451 let (bytes, truncated) = read_prefix(&canonical, config.tools.max_read_bytes)
1452 .context("read project AGENTS.md")?;
1453 let instructions = std::str::from_utf8(&bytes).context("project AGENTS.md is not UTF-8")?;
1454 prompt.push_str("\n# Project instructions\n");
1455 prompt.push_str(instructions);
1456 if truncated {
1457 prompt.push_str("\n[AGENTS.md truncated by configured read limit]\n");
1458 }
1459 }
1460 if !skills.listing.is_empty() {
1461 prompt.push_str("\n# Available skills\n");
1462 prompt.push_str(&skills.listing);
1463 prompt.push_str("\nUse read_skill with a skill name when its workflow applies.\n");
1464 }
1465 if !skills.project_listing.is_empty() {
1466 prompt.push_str("\n# Project skills\n");
1467 prompt.push_str(
1468 "Projects in this workspace provide these skills to agents working in them:\n",
1469 );
1470 prompt.push_str(&skills.project_listing);
1471 prompt.push_str(
1472 "\nTo use one, delegate with an agent_* tool such as agent_codex or agent_claude, \
1473 set its cwd to the skill's project, and name the skill in the prompt: that agent \
1474 then loads the project's instructions and skills itself. read_skill loads a \
1475 skill for reference.\n",
1476 );
1477 }
1478 Ok(prompt)
1479}
1480
1481struct DiscoveredSkills {
1484 map: SkillMap,
1485 roots: Vec<PathBuf>,
1486 listing: String,
1487 project_listing: String,
1488}
1489
1490const PROJECT_SKILL_DIRS: [&str; 2] = [".agents/skills", ".claude/skills"];
1493const MAX_WORKSPACE_ENTRIES: usize = 4096;
1496const MAX_SKILL_PROJECTS: usize = 256;
1497const PROJECT_SKILL_HEADER_BYTES: usize = 16 * 1024;
1499const MAX_PROJECT_SKILL_DESCRIPTION: usize = 400;
1500
1501fn discover_skills(workspace: &Path, config: &Config, tools: bool) -> Result<DiscoveredSkills> {
1502 let mut skills = SkillMap::new();
1503 let mut roots = Vec::new();
1504 let project_root = workspace.join(&config.skills.project_dir);
1505 for (root, must_be_workspace) in [(&project_root, true), (&config.skills.user_dir, false)] {
1506 if !root.is_dir() {
1507 continue;
1508 }
1509 let canonical = std::fs::canonicalize(root)
1510 .with_context(|| format!("resolve skill root {}", root.display()))?;
1511 if must_be_workspace && !canonical.starts_with(workspace) {
1512 return Err(anyhow!("project skill root escaped workspace"));
1513 }
1514 roots.push(canonical.clone());
1515 let mut entries: Vec<_> = std::fs::read_dir(&canonical)
1516 .with_context(|| format!("read skill root {}", canonical.display()))?
1517 .filter_map(Result::ok)
1518 .collect();
1519 entries.sort_by_key(|entry| entry.file_name());
1520 for entry in entries {
1521 if skills.len() >= config.skills.max_skills {
1522 break;
1523 }
1524 let path = entry.path().join("SKILL.md");
1525 if !path.is_file() {
1526 continue;
1527 }
1528 let canonical_file = std::fs::canonicalize(&path)
1529 .with_context(|| format!("resolve skill {}", path.display()))?;
1530 if !canonical_file.starts_with(&canonical) {
1531 continue;
1532 }
1533 let name = entry.file_name().to_string_lossy().to_string();
1534 skills.entry(name).or_insert(canonical_file);
1535 }
1536 }
1537 let mut names: Vec<_> = skills.keys().cloned().collect();
1538 names.sort();
1539 let mut listing = String::new();
1540 for name in names {
1541 let path = &skills[&name];
1542 let bytes = read_prefix(path, config.skills.max_skill_bytes)
1543 .map(|(bytes, _)| bytes)
1544 .unwrap_or_default();
1545 let content = String::from_utf8_lossy(&bytes);
1546 let description = skill_description(&content);
1547 listing.push_str(&format!("- {name}: {description}\n"));
1548 }
1549 let project_listing = if tools && config.skills.scan_projects {
1552 discover_project_skills(workspace, config, &mut skills, &mut roots)
1553 } else {
1554 String::new()
1555 };
1556 Ok(DiscoveredSkills {
1557 map: skills,
1558 roots,
1559 listing,
1560 project_listing,
1561 })
1562}
1563
1564fn discover_project_skills(
1570 workspace: &Path,
1571 config: &Config,
1572 skills: &mut SkillMap,
1573 roots: &mut Vec<PathBuf>,
1574) -> String {
1575 let mut projects = vec![(None, workspace.to_path_buf())];
1576 let mut names: Vec<_> = std::fs::read_dir(workspace)
1577 .into_iter()
1578 .flatten()
1579 .filter_map(Result::ok)
1580 .take(MAX_WORKSPACE_ENTRIES)
1581 .map(|entry| entry.file_name().to_string_lossy().into_owned())
1582 .filter(|name| !name.starts_with('.'))
1583 .collect();
1584 names.sort();
1585 for name in names {
1586 if projects.len() > MAX_SKILL_PROJECTS {
1587 break;
1588 }
1589 let Ok(directory) = std::fs::canonicalize(workspace.join(&name)) else {
1590 continue;
1591 };
1592 if directory.is_dir()
1593 && directory.starts_with(workspace)
1594 && !projects.iter().any(|(_, seen)| seen == &directory)
1595 {
1596 projects.push((Some(name), directory));
1597 }
1598 }
1599 let mut seen_files = std::collections::HashSet::new();
1600 let mut listing = String::new();
1601 'projects: for (project, directory) in projects {
1602 let project_roots: Vec<PathBuf> = PROJECT_SKILL_DIRS
1603 .iter()
1604 .filter_map(|relative| std::fs::canonicalize(directory.join(relative)).ok())
1605 .filter(|root| root.is_dir() && root.starts_with(workspace))
1606 .collect();
1607 for root in &project_roots {
1608 if !roots.contains(root) {
1609 roots.push(root.clone());
1610 }
1611 }
1612 for root in &project_roots {
1613 let mut entries: Vec<_> = std::fs::read_dir(root)
1614 .into_iter()
1615 .flatten()
1616 .filter_map(Result::ok)
1617 .take(MAX_WORKSPACE_ENTRIES)
1618 .collect();
1619 entries.sort_by_key(|entry| entry.file_name());
1620 for entry in entries {
1621 if skills.len() >= config.skills.max_skills {
1622 break 'projects;
1623 }
1624 let Ok(file) = std::fs::canonicalize(entry.path().join("SKILL.md")) else {
1625 continue;
1626 };
1627 if !file.is_file()
1628 || !project_roots.iter().any(|root| file.starts_with(root))
1629 || !seen_files.insert(file.clone())
1630 {
1631 continue;
1632 }
1633 let skill = entry.file_name().to_string_lossy().into_owned();
1634 let (name, location) = match &project {
1635 Some(project) => (format!("{project}:{skill}"), format!("project {project}")),
1636 None => (skill, "workspace root".to_owned()),
1637 };
1638 if skills.contains_key(&name) {
1640 continue;
1641 }
1642 let header = read_prefix(
1643 &file,
1644 config
1645 .skills
1646 .max_skill_bytes
1647 .min(PROJECT_SKILL_HEADER_BYTES),
1648 )
1649 .map(|(bytes, _)| bytes)
1650 .unwrap_or_default();
1651 let description: String = skill_description(&String::from_utf8_lossy(&header))
1652 .chars()
1653 .take(MAX_PROJECT_SKILL_DESCRIPTION)
1654 .collect();
1655 listing.push_str(&format!("- {name} ({location}): {description}\n"));
1656 skills.insert(name, file);
1657 }
1658 }
1659 }
1660 listing
1661}
1662
1663fn read_prefix(path: &Path, max_bytes: usize) -> std::io::Result<(Vec<u8>, bool)> {
1664 let file = std::fs::File::open(path)?;
1665 let mut bytes = Vec::with_capacity(max_bytes.min(8192));
1666 file.take(
1667 u64::try_from(max_bytes)
1668 .unwrap_or(u64::MAX)
1669 .saturating_add(1),
1670 )
1671 .read_to_end(&mut bytes)?;
1672 let truncated = bytes.len() > max_bytes;
1673 bytes.truncate(max_bytes);
1674 Ok((bytes, truncated))
1675}
1676
1677fn skill_description(content: &str) -> String {
1678 if let Some(frontmatter) = content.strip_prefix("---\n")
1679 && let Some((header, _)) = frontmatter.split_once("\n---")
1680 {
1681 for line in header.lines() {
1682 if let Some(description) = line.strip_prefix("description:") {
1683 return description.trim().trim_matches('"').to_owned();
1684 }
1685 }
1686 }
1687 content
1688 .lines()
1689 .map(str::trim)
1690 .find(|line| !line.is_empty() && !line.starts_with('#'))
1691 .unwrap_or("No description provided")
1692 .chars()
1693 .take(240)
1694 .collect()
1695}
1696
1697struct ProtocolSink {
1698 meta: TurnMeta,
1699 output: OutboundSender,
1700 cancellation: CancellationToken,
1701}
1702
1703#[async_trait]
1704impl EventSink for ProtocolSink {
1705 async fn emit(&self, event: CoreEvent) -> Result<(), AgentError> {
1706 let seq = next_seq(&self.meta.seq);
1707 let event = match event {
1708 CoreEvent::AssistantDelta { content } => ServerEvent::AssistantDelta {
1709 request_id: self.meta.request_id.clone(),
1710 session_id: self.meta.session_id.clone(),
1711 turn_id: self.meta.turn_id.clone(),
1712 seq,
1713 content,
1714 },
1715 CoreEvent::AssistantCompleted { content } => ServerEvent::AssistantCompleted {
1716 request_id: self.meta.request_id.clone(),
1717 session_id: self.meta.session_id.clone(),
1718 turn_id: self.meta.turn_id.clone(),
1719 seq,
1720 content,
1721 },
1722 CoreEvent::ToolProposed {
1723 call_id,
1724 name,
1725 arguments,
1726 } => ServerEvent::ToolProposed {
1727 request_id: self.meta.request_id.clone(),
1728 session_id: self.meta.session_id.clone(),
1729 turn_id: self.meta.turn_id.clone(),
1730 seq,
1731 call_id,
1732 name,
1733 arguments,
1734 },
1735 CoreEvent::ToolStarted { call_id, name } => ServerEvent::ToolStarted {
1736 request_id: self.meta.request_id.clone(),
1737 session_id: self.meta.session_id.clone(),
1738 turn_id: self.meta.turn_id.clone(),
1739 seq,
1740 call_id,
1741 name,
1742 },
1743 CoreEvent::ToolCompleted {
1744 call_id,
1745 name,
1746 output,
1747 } => ServerEvent::ToolCompleted {
1748 request_id: self.meta.request_id.clone(),
1749 session_id: self.meta.session_id.clone(),
1750 turn_id: self.meta.turn_id.clone(),
1751 seq,
1752 call_id,
1753 name,
1754 success: !output.is_error,
1755 output: output.content,
1756 truncated: output.truncated,
1757 },
1758 CoreEvent::ContextCompacted {
1759 before_tokens,
1760 after_tokens,
1761 removed_messages,
1762 } => ServerEvent::ContextCompacted {
1763 request_id: self.meta.request_id.clone(),
1764 session_id: self.meta.session_id.clone(),
1765 turn_id: self.meta.turn_id.clone(),
1766 seq,
1767 before_tokens,
1768 after_tokens,
1769 removed_messages,
1770 },
1771 CoreEvent::SessionTrimmed {
1772 removed_messages,
1773 history_bytes,
1774 } => ServerEvent::SessionTrimmed {
1775 request_id: self.meta.request_id.clone(),
1776 session_id: self.meta.session_id.clone(),
1777 seq,
1778 removed_messages,
1779 history_bytes,
1780 },
1781 };
1782 send_turn_event(
1783 &self.output,
1784 event,
1785 self.meta.max_server_frame,
1786 &self.cancellation,
1787 )
1788 .await
1789 }
1790}
1791
1792#[derive(Default)]
1793struct ApprovalBroker {
1794 pending: Mutex<HashMap<String, oneshot::Sender<bool>>>,
1795}
1796
1797impl ApprovalBroker {
1798 async fn insert(&self, id: String, sender: oneshot::Sender<bool>) {
1799 self.pending.lock().await.insert(id, sender);
1800 }
1801
1802 async fn remove(&self, id: &str) {
1803 self.pending.lock().await.remove(id);
1804 }
1805
1806 async fn resolve(&self, id: &str, approved: bool) -> bool {
1807 let sender = self.pending.lock().await.remove(id);
1808 sender.is_some_and(|sender| sender.send(approved).is_ok())
1809 }
1810}
1811
1812struct ProtocolApprovalGate {
1813 policy: ApprovalPolicy,
1814 broker: Arc<ApprovalBroker>,
1815 meta: TurnMeta,
1816 output: OutboundSender,
1817}
1818
1819#[async_trait]
1820impl ApprovalGate for ProtocolApprovalGate {
1821 async fn approve(
1822 &self,
1823 request: ApprovalRequest,
1824 cancellation: CancellationToken,
1825 ) -> Result<bool, AgentError> {
1826 match self.policy {
1827 ApprovalPolicy::OnRisk if request.risk == ToolRisk::ReadOnly => return Ok(true),
1828 ApprovalPolicy::Never => return Ok(request.risk == ToolRisk::ReadOnly),
1829 ApprovalPolicy::Always | ApprovalPolicy::OnRisk => {}
1830 }
1831 let approval_id = Uuid::new_v4().to_string();
1832 let (sender, receiver) = oneshot::channel();
1833 self.broker.insert(approval_id.clone(), sender).await;
1834 let event = ServerEvent::ApprovalRequested {
1835 request_id: self.meta.request_id.clone(),
1836 session_id: self.meta.session_id.clone(),
1837 turn_id: self.meta.turn_id.clone(),
1838 seq: next_seq(&self.meta.seq),
1839 approval_id: approval_id.clone(),
1840 call_id: request.call_id,
1841 name: request.name,
1842 risk: request.risk.as_str().into(),
1843 cwd: request.cwd.display().to_string(),
1844 summary: request.summary,
1845 };
1846 if let Err(error) = send_turn_event(
1847 &self.output,
1848 event,
1849 self.meta.max_server_frame,
1850 &cancellation,
1851 )
1852 .await
1853 {
1854 self.broker.remove(&approval_id).await;
1855 return Err(error);
1856 }
1857 tokio::select! {
1858 result = receiver => result.map_err(|_| AgentError::Cancelled),
1859 _ = cancellation.cancelled() => {
1860 self.broker.remove(&approval_id).await;
1861 Err(AgentError::Cancelled)
1862 }
1863 }
1864 }
1865}
1866
1867async fn send_event(output: &OutboundSender, event: ServerEvent, max_bytes: usize) -> Result<()> {
1868 let bytes = encode_event(&event, max_bytes)?;
1869 output.send(bytes, None).await.map_err(anyhow::Error::new)
1870}
1871
1872async fn send_turn_event(
1873 output: &OutboundSender,
1874 event: ServerEvent,
1875 max_bytes: usize,
1876 cancellation: &CancellationToken,
1877) -> Result<(), AgentError> {
1878 let bytes = encode_event(&event, max_bytes)
1879 .map_err(|error| AgentError::ResponseLimit(error.to_string()))?;
1880 match output.send(bytes, Some(cancellation)).await {
1881 Ok(()) => Ok(()),
1882 Err(OutboundSendError::Cancelled) => Err(AgentError::Cancelled),
1883 Err(error) => Err(AgentError::Internal(error.to_string())),
1884 }
1885}
1886
1887fn encode_event(event: &ServerEvent, max_bytes: usize) -> Result<Vec<u8>> {
1888 let bytes = serde_json::to_vec(event).context("serialize protocol event")?;
1889 if bytes.len() > max_bytes {
1890 return Err(anyhow!("server event exceeds configured frame limit"));
1891 }
1892 Ok(bytes)
1893}
1894
1895async fn send_error(
1896 output: &OutboundSender,
1897 request_id: &str,
1898 code: &str,
1899 message: &str,
1900 fatal: bool,
1901 max_bytes: usize,
1902) -> Result<()> {
1903 send_event(
1904 output,
1905 ServerEvent::Error {
1906 request_id: (!request_id.is_empty()).then(|| request_id.to_owned()),
1907 code: code.into(),
1908 message: message.into(),
1909 fatal,
1910 },
1911 max_bytes,
1912 )
1913 .await
1914}
1915
1916#[cfg(test)]
1917mod tests {
1918 use std::{
1919 future::pending,
1920 io::Cursor,
1921 sync::atomic::{AtomicBool, Ordering},
1922 };
1923
1924 fn test_registry() -> Arc<DelegationRegistry> {
1926 let home = tempfile::tempdir().unwrap().keep();
1927 Arc::new(DelegationRegistry::new(&home))
1928 }
1929
1930 use super::*;
1931
1932 struct DropSignal(Arc<AtomicBool>);
1933
1934 impl Drop for DropSignal {
1935 fn drop(&mut self) {
1936 self.0.store(true, Ordering::Release);
1937 }
1938 }
1939
1940 #[tokio::test]
1941 async fn nonreading_management_client_does_not_hold_component_lock() {
1942 let (mut input, server_input) = tokio::io::duplex(65536);
1943 let (server_output, _blocked_output) = tokio::io::duplex(1);
1944 let tasks = TaskTracker::new();
1945 let components = Arc::new(Mutex::new(components::Components::new(
1946 PathBuf::from("/unused.sock"),
1947 PathBuf::from("/"),
1948 )));
1949 let cancel = CancellationToken::new();
1950 let handler = tokio::spawn(run_managed(
1951 server_input,
1952 server_output,
1953 ConfigOverrides::default(),
1954 Some(components.clone()),
1955 test_registry(),
1956 cancel.clone(),
1957 tasks.clone(),
1958 ));
1959 input.write_all(b"{\"type\":\"initialize\",\"request_id\":\"init\",\"protocol_version\":2,\"client\":{\"name\":\"test\",\"version\":\"0\"}}\n").await.unwrap();
1960 for _ in 0..300 {
1961 input.write_all(b"{\"type\":\"daemon.control\",\"request_id\":\"s\",\"command\":{\"action\":\"status\"}}\n").await.unwrap();
1962 }
1963 tokio::time::sleep(Duration::from_millis(50)).await;
1964 let status = tokio::time::timeout(Duration::from_millis(100), async {
1965 components.lock().await.status()
1966 })
1967 .await
1968 .unwrap();
1969 assert_eq!(status.pid, std::process::id());
1970 cancel.cancel();
1971 handler.abort();
1972 let _ = handler.await;
1973 tasks.close();
1974 tokio::time::timeout(Duration::from_secs(1), tasks.wait())
1975 .await
1976 .unwrap();
1977 }
1978
1979 #[tokio::test]
1980 async fn forced_connection_abort_drops_and_joins_writer_descendants() {
1981 let (mut input, server_input) = tokio::io::duplex(512);
1982 let (server_output, _blocked_output) = tokio::io::duplex(1);
1983 let tasks = TaskTracker::new();
1984 let handler = tokio::spawn(run_managed(
1985 server_input,
1986 server_output,
1987 ConfigOverrides::default(),
1988 None,
1989 test_registry(),
1990 CancellationToken::new(),
1991 tasks.clone(),
1992 ));
1993 input.write_all(b"{\"type\":\"initialize\",\"request_id\":\"init\",\"protocol_version\":2,\"client\":{\"name\":\"test\",\"version\":\"0\"}}\n").await.unwrap();
1994 tokio::time::timeout(Duration::from_secs(1), async {
1995 while tasks.is_empty() {
1996 tokio::task::yield_now().await;
1997 }
1998 })
1999 .await
2000 .unwrap();
2001 handler.abort();
2002 let _ = handler.await;
2003 tasks.close();
2004 tokio::time::timeout(Duration::from_secs(1), tasks.wait())
2005 .await
2006 .unwrap();
2007 assert!(tasks.is_empty());
2008 }
2009
2010 #[tokio::test]
2011 async fn forced_handler_abort_cancels_and_joins_active_turn() {
2012 let tasks = TaskTracker::new();
2013 let cancellation = CancellationToken::new();
2014 let child_cancel = cancellation.child_token();
2015 let observed_cancel = child_cancel.clone();
2016 let dropped = Arc::new(AtomicBool::new(false));
2017 let (ready_tx, ready_rx) = oneshot::channel();
2018 let task = tasks.spawn({
2019 let dropped = dropped.clone();
2020 async move {
2021 let _guard = DropSignal(dropped);
2022 let _ = ready_tx.send(());
2023 pending::<()>().await;
2024 }
2025 });
2026 ready_rx.await.unwrap();
2027 let (owned_tx, owned_rx) = oneshot::channel();
2028 let handler = tokio::spawn(async move {
2029 let _active = ActiveTurn {
2030 turn_id: "test".into(),
2031 cancellation: child_cancel,
2032 task,
2033 };
2034 let _ = owned_tx.send(());
2035 pending::<()>().await;
2036 });
2037 owned_rx.await.unwrap();
2038 handler.abort();
2039 let _ = handler.await;
2040 tasks.close();
2041 tokio::time::timeout(Duration::from_secs(1), tasks.wait())
2042 .await
2043 .unwrap();
2044 assert!(observed_cancel.is_cancelled());
2045 assert!(dropped.load(Ordering::Acquire));
2046 }
2047
2048 #[tokio::test]
2049 async fn frame_buffer_preserves_partial_and_discard_state_across_cancellation() {
2050 let (mut input, output) = tokio::io::duplex(64);
2051 let mut reader = BufReader::new(output);
2052 let mut frames = FrameBuffer::default();
2053 input.write_all(b"12").await.unwrap();
2054 assert!(
2055 tokio::time::timeout(Duration::from_millis(10), frames.read(&mut reader, 4))
2056 .await
2057 .is_err()
2058 );
2059 input.write_all(b"34\n").await.unwrap();
2060 assert!(
2061 matches!(frames.read(&mut reader, 4).await.unwrap(), FrameRead::Frame(value) if value == b"1234")
2062 );
2063 input.write_all(b"123456789").await.unwrap();
2064 assert!(
2065 tokio::time::timeout(Duration::from_millis(10), frames.read(&mut reader, 4))
2066 .await
2067 .is_err()
2068 );
2069 input.write_all(b"\n{}\n").await.unwrap();
2070 assert!(matches!(
2071 frames.read(&mut reader, 4).await.unwrap(),
2072 FrameRead::TooLarge
2073 ));
2074 assert!(
2075 matches!(frames.read(&mut reader, 4).await.unwrap(), FrameRead::Frame(value) if value == b"{}")
2076 );
2077 }
2078
2079 #[tokio::test]
2080 async fn bounded_reader_discards_an_oversized_line() {
2081 let input = format!("{}\n{{}}\n", "x".repeat(10));
2082 let mut reader = BufReader::new(Cursor::new(input.into_bytes()));
2083 assert!(matches!(
2084 read_bounded_frame(&mut reader, 4).await.unwrap(),
2085 FrameRead::TooLarge
2086 ));
2087 match read_bounded_frame(&mut reader, 4).await.unwrap() {
2088 FrameRead::Frame(frame) => assert_eq!(frame, b"{}"),
2089 _ => panic!("expected the frame following the oversized line"),
2090 }
2091 }
2092
2093 #[tokio::test]
2094 async fn bounded_reader_accepts_exact_crlf_limit() {
2095 let mut reader = BufReader::new(Cursor::new(b"1234\r\n".to_vec()));
2096 match read_bounded_frame(&mut reader, 4).await.unwrap() {
2097 FrameRead::Frame(frame) => assert_eq!(frame, b"1234"),
2098 _ => panic!("expected an exact-limit frame"),
2099 }
2100 }
2101
2102 #[tokio::test]
2103 async fn outbound_byte_backpressure_is_cancellation_aware() {
2104 let (output, mut receiver) = outbound_channel(5);
2105 output.send(vec![0; 4], None).await.unwrap();
2106
2107 let cancellation = CancellationToken::new();
2108 let blocked = tokio::spawn({
2109 let output = output.clone();
2110 let cancellation = cancellation.clone();
2111 async move { output.send(vec![1; 4], Some(&cancellation)).await }
2112 });
2113 tokio::task::yield_now().await;
2114 assert!(!blocked.is_finished());
2115
2116 cancellation.cancel();
2117 assert_eq!(blocked.await.unwrap(), Err(OutboundSendError::Cancelled));
2118
2119 drop(receiver.recv().await.unwrap());
2120 output.send(vec![2; 4], None).await.unwrap();
2121 }
2122
2123 #[tokio::test]
2124 async fn outbound_control_send_times_out_under_byte_backpressure() {
2125 let (output, _receiver) = outbound_channel(5);
2126 output.send(vec![0; 4], None).await.unwrap();
2127 let result = output
2128 .send_with_timeout(vec![1; 4], None, Duration::from_millis(10))
2129 .await;
2130 assert_eq!(result, Err(OutboundSendError::TimedOut));
2131 }
2132
2133 #[tokio::test]
2134 async fn active_turn_shutdown_aborts_after_grace_period() {
2135 let cancellation = CancellationToken::new();
2136 let dropped = Arc::new(AtomicBool::new(false));
2137 let (started_tx, started_rx) = oneshot::channel();
2138 let task = tokio::spawn({
2139 let dropped = Arc::clone(&dropped);
2140 async move {
2141 let _signal = DropSignal(dropped);
2142 let _ = started_tx.send(());
2143 pending::<()>().await;
2144 }
2145 });
2146 started_rx.await.unwrap();
2147
2148 let graceful = shutdown_active_turn(
2149 ActiveTurn {
2150 turn_id: "turn".into(),
2151 cancellation,
2152 task,
2153 },
2154 Duration::from_millis(10),
2155 )
2156 .await;
2157
2158 assert!(!graceful);
2159 assert!(dropped.load(Ordering::Acquire));
2160 }
2161
2162 #[tokio::test]
2163 async fn writer_shutdown_aborts_after_grace_period() {
2164 let dropped = Arc::new(AtomicBool::new(false));
2165 let (started_tx, started_rx) = oneshot::channel();
2166 let writer = tokio::spawn({
2167 let dropped = Arc::clone(&dropped);
2168 async move {
2169 let _signal = DropSignal(dropped);
2170 let _ = started_tx.send(());
2171 pending::<std::io::Result<()>>().await
2172 }
2173 });
2174 started_rx.await.unwrap();
2175
2176 let result = shutdown_writer(writer, Duration::from_millis(10)).await;
2177
2178 assert!(result.is_err());
2179 assert!(dropped.load(Ordering::Acquire));
2180 }
2181
2182 #[tokio::test]
2183 async fn workspace_projects_list_their_agent_skills_for_delegation() {
2184 use std::os::unix::fs::symlink;
2185 let temporary = tempfile::tempdir().unwrap();
2186 let workspace = temporary.path().canonicalize().unwrap();
2187 let outside = tempfile::tempdir().unwrap();
2188 let outside = outside.path().canonicalize().unwrap();
2189 let write_skill = |directory: &Path, description: &str| {
2190 std::fs::create_dir_all(directory).unwrap();
2191 std::fs::write(
2192 directory.join("SKILL.md"),
2193 format!(
2194 "---\nname: skill\ndescription: {description}\n---\nBody of {description}\n"
2195 ),
2196 )
2197 .unwrap();
2198 };
2199 write_skill(
2200 &workspace.join("scv/.agents/skills/feature-flow"),
2201 "Land SCV",
2202 );
2203 std::fs::create_dir_all(workspace.join("scv/.claude/skills")).unwrap();
2204 symlink(
2205 "../../.agents/skills/feature-flow",
2206 workspace.join("scv/.claude/skills/feature-flow"),
2207 )
2208 .unwrap();
2209 write_skill(
2210 &workspace.join("web/.claude/skills/deploy"),
2211 "Deploy the site",
2212 );
2213 write_skill(&workspace.join(".agents/skills/triage"), "Root triage");
2214 write_skill(&workspace.join(".agents/skills/notes"), "Root notes");
2215 write_skill(&workspace.join(".scv/skills/triage"), "SCV triage");
2216 write_skill(&workspace.join(".hidden/.agents/skills/secret"), "Hidden");
2217 write_skill(&outside.join(".agents/skills/evil"), "Outside");
2218 symlink(&outside, workspace.join("escape")).unwrap();
2219 std::fs::create_dir_all(workspace.join("rogue/.agents")).unwrap();
2220 symlink(
2221 outside.join(".agents/skills"),
2222 workspace.join("rogue/.agents/skills"),
2223 )
2224 .unwrap();
2225 std::fs::create_dir_all(workspace.join("sneaky/.agents/skills/leak")).unwrap();
2226 symlink(
2227 outside.join(".agents/skills/evil/SKILL.md"),
2228 workspace.join("sneaky/.agents/skills/leak/SKILL.md"),
2229 )
2230 .unwrap();
2231 std::fs::write(workspace.join("file"), "not a project").unwrap();
2232 let mut config = Config::default();
2233 config.skills.user_dir = workspace.join("no-user-skills");
2234
2235 let skills = discover_skills(&workspace, &config, true).unwrap();
2236 let mut names: Vec<_> = skills.map.keys().cloned().collect();
2237 names.sort();
2238 assert_eq!(names, ["notes", "scv:feature-flow", "triage", "web:deploy"]);
2239 assert_eq!(
2240 skills.map["triage"],
2241 workspace.join(".scv/skills/triage/SKILL.md")
2242 );
2243 assert_eq!(
2244 skills.project_listing,
2245 "- notes (workspace root): Root notes\n\
2246 - scv:feature-flow (project scv): Land SCV\n\
2247 - web:deploy (project web): Deploy the site\n"
2248 );
2249 let prompt = build_system_prompt(&workspace, &config, &skills).unwrap();
2250 assert!(prompt.contains("# Project skills"));
2251 assert!(prompt.contains("set its cwd to the skill's project"));
2252
2253 let registry = builtin_registry(
2254 config.tools(),
2255 skills.map,
2256 skills.roots,
2257 config.skills.max_skill_bytes,
2258 HashMap::new(),
2259 )
2260 .unwrap();
2261 let read_skill = registry.get("read_skill").unwrap();
2262 let loaded = read_skill
2263 .execute(
2264 serde_json::json!({"name":"scv:feature-flow"}),
2265 scv_core::ToolContext {
2266 workspace: workspace.clone(),
2267 cancellation: CancellationToken::new(),
2268 },
2269 )
2270 .await
2271 .unwrap();
2272 assert!(loaded.content.contains("Body of Land SCV"));
2273
2274 let tool_free = discover_skills(&workspace, &config, false).unwrap();
2275 assert!(tool_free.project_listing.is_empty());
2276 assert!(!tool_free.map.contains_key("scv:feature-flow"));
2277 config.skills.scan_projects = false;
2278 let disabled = discover_skills(&workspace, &config, true).unwrap();
2279 assert!(disabled.project_listing.is_empty());
2280 config.skills.scan_projects = true;
2281 config.skills.max_skills = 3;
2282 let capped = discover_skills(&workspace, &config, true).unwrap();
2283 assert_eq!(capped.map.len(), 3);
2284 assert!(capped.map.contains_key("scv:feature-flow"));
2285 assert!(!capped.map.contains_key("web:deploy"));
2286 }
2287
2288 #[tokio::test]
2289 async fn daemon_control_lists_and_stops_delegations() {
2290 use std::os::unix::process::CommandExt as _;
2291 let home = tempfile::tempdir().unwrap();
2292 let registry = DelegationRegistry::new(home.path());
2293 let components = Arc::new(Mutex::new(components::Components::new(
2294 PathBuf::from("/unused.sock"),
2295 PathBuf::from("/"),
2296 )));
2297 let mut owner = std::process::Command::new("sleep")
2299 .arg("30")
2300 .spawn()
2301 .unwrap();
2302 let mut agent = std::process::Command::new("sleep")
2303 .arg("30")
2304 .process_group(0)
2305 .spawn()
2306 .unwrap();
2307 let identity = |pid| delegations::ProcessIdentity::of(pid).unwrap();
2308 let record = delegations::DelegationRecord {
2309 handle: "codex-a1b2c3".into(),
2310 agent: "codex".into(),
2311 instance: registry.instance().into(),
2312 session: "session".into(),
2313 owner: identity(owner.id()),
2314 process: identity(agent.id()),
2315 pgid: agent.id(),
2316 cwd: "/work/project\u{7}".into(),
2317 started_unix: 1,
2318 depth: 1,
2319 conversation: Some("codex-2".into()),
2320 turn: Some(3),
2321 };
2322 std::fs::create_dir_all(registry.record_dir()).unwrap();
2323 std::fs::write(
2324 registry.record_dir().join("codex-a1b2c3.json"),
2325 serde_json::to_vec(&record).unwrap(),
2326 )
2327 .unwrap();
2328 let control = |command| daemon_control(&components, ®istry, command);
2329 let Ok(status) = control(DaemonCommand::Status).await else {
2330 panic!("status failed");
2331 };
2332 assert_eq!(status.delegations.active, 1);
2333 assert!(status.delegations.entries.is_empty());
2334 let Ok(status) = control(DaemonCommand::Delegations { all: false }).await else {
2335 panic!("listing failed");
2336 };
2337 let [entry] = status.delegations.entries.as_slice() else {
2338 panic!("{:?}", status.delegations);
2339 };
2340 assert_eq!(entry.handle, "codex-a1b2c3");
2341 assert_eq!(entry.conversation.as_deref(), Some("codex-2"));
2342 assert_eq!(entry.turn, Some(3));
2343 assert_eq!(entry.pid, agent.id());
2344 assert_eq!(entry.owner_pid, owner.id());
2345 assert!(!entry.orphaned);
2346 assert_eq!(entry.processes, 1);
2347 for command in [
2348 DaemonCommand::DelegationKill {
2349 handle: Some("codex-nosuch".into()),
2350 orphans: false,
2351 },
2352 DaemonCommand::DelegationKill {
2353 handle: None,
2354 orphans: false,
2355 },
2356 ] {
2357 assert!(matches!(
2358 control(command).await,
2359 Err(ControlFailure::Delegation(_))
2360 ));
2361 }
2362 let Ok(status) = control(DaemonCommand::DelegationKill {
2363 handle: Some("codex-a1b2c3".into()),
2364 orphans: false,
2365 })
2366 .await
2367 else {
2368 panic!("kill failed");
2369 };
2370 assert_eq!(status.delegations.killed, ["codex-a1b2c3"]);
2371 assert!(agent.wait().unwrap().code().is_none());
2372 owner.kill().unwrap();
2375 owner.wait().unwrap();
2376 let Ok(status) = control(DaemonCommand::Delegations { all: true }).await else {
2377 panic!("listing failed");
2378 };
2379 assert!(status.delegations.entries[0].orphaned);
2380 assert_eq!(status.delegations.active, 0);
2381 let Ok(_) = control(DaemonCommand::DelegationKill {
2382 handle: None,
2383 orphans: true,
2384 })
2385 .await
2386 else {
2387 panic!("orphan sweep failed");
2388 };
2389 assert!(registry.list(true).is_empty());
2390 }
2391}