diff --git a/rust/chg/src/locator.rs b/rust/chg/src/locator.rs --- a/rust/chg/src/locator.rs +++ b/rust/chg/src/locator.rs @@ -20,8 +20,11 @@ use tokio_process::{Child, CommandExt}; use tokio_timer; +use super::message::ServerSpec; use super::procutil; +const REQUIRED_SERVER_CAPABILITIES: &[&str] = &["attachio", "chdir", "runcommand"]; + /// Helper to connect to and spawn a server process. #[derive(Clone, Debug)] pub struct Locator { @@ -68,10 +71,15 @@ /// Tries to connect to the existing server, or spawns new if not running. fn try_connect(self) -> impl Future { debug!("try connect to {}", self.base_sock_path.display()); - UnixClient::connect(self.base_sock_path.clone()).then(|res| match res { - Ok(client) => Either::A(future::ok((self, client))), - Err(_) => Either::B(self.spawn_connect()), - }) + UnixClient::connect(self.base_sock_path.clone()) + .then(|res| match res { + Ok(client) => Either::A(future::ok((self, client))), + Err(_) => Either::B(self.spawn_connect()), + }) + .and_then(|(loc, client)| { + check_server_capabilities(client.server_spec())?; + Ok((loc, client)) + }) } /// Spawns new server process and connects to it. @@ -239,3 +247,20 @@ Err(io::Error::new(io::ErrorKind::Other, "insecure directory")) } } + +fn check_server_capabilities(spec: &ServerSpec) -> io::Result<()> { + let unsupported: Vec<_> = REQUIRED_SERVER_CAPABILITIES + .iter() + .cloned() + .filter(|&s| !spec.capabilities.contains(s)) + .collect(); + if unsupported.is_empty() { + Ok(()) + } else { + let msg = format!( + "insufficient server capabilities: {}", + unsupported.join(", ") + ); + Err(io::Error::new(io::ErrorKind::Other, msg)) + } +}