From b57c0dbe305ea1cb2f3908bcc3529dfa5882c03c Mon Sep 17 00:00:00 2001 From: Stefan Reiter Date: Tue, 16 Feb 2021 18:07:01 +0100 Subject: [PATCH] rest: implement tower service for UnixStream This allows anything that can be represented as a UnixStream to be used as transport for an API server (e.g. virtio sockets). A tower service expects an IP address as it's peer, which we can't reliably provide for unix socket based transports, so just fake one. Signed-off-by: Stefan Reiter --- src/server/rest.rs | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/src/server/rest.rs b/src/server/rest.rs index fc59be9a..9bf494fd 100644 --- a/src/server/rest.rs +++ b/src/server/rest.rs @@ -107,6 +107,26 @@ impl tower_service::Service<&tokio::net::TcpStream> for RestServer { } } +impl tower_service::Service<&tokio::net::UnixStream> for RestServer { + type Response = ApiService; + type Error = Error; + type Future = Pin> + Send>>; + + fn poll_ready(&mut self, _cx: &mut Context) -> Poll> { + Poll::Ready(Ok(())) + } + + fn call(&mut self, _ctx: &tokio::net::UnixStream) -> Self::Future { + // TODO: Find a way to actually represent the vsock peer in the ApiService struct - for now + // it doesn't really matter, so just use a fake IP address + let fake_peer = "0.0.0.0:807".parse().unwrap(); + future::ok(ApiService { + peer: fake_peer, + api_config: self.api_config.clone() + }).boxed() + } +} + pub struct ApiService { pub peer: std::net::SocketAddr, pub api_config: Arc,