diff --git a/apps/desktop/desktop_native/core/src/ssh_agent/named_pipe_listener_stream.rs b/apps/desktop/desktop_native/core/src/ssh_agent/named_pipe_listener_stream.rs index 917d8dd2ecf..e50498b200d 100644 --- a/apps/desktop/desktop_native/core/src/ssh_agent/named_pipe_listener_stream.rs +++ b/apps/desktop/desktop_native/core/src/ssh_agent/named_pipe_listener_stream.rs @@ -11,11 +11,14 @@ use tokio::{ }; use tokio_util::sync::CancellationToken; +use super::peerinfo; +use super::peerinfo::models::PeerInfo; + const PIPE_NAME: &str = r"\\.\pipe\openssh-ssh-agent"; #[pin_project::pin_project] pub struct NamedPipeServerStream { - rx: tokio::sync::mpsc::Receiver, + rx: tokio::sync::mpsc::Receiver<(NamedPipeServer, PeerInfo)>, } impl NamedPipeServerStream { @@ -48,12 +51,12 @@ impl NamedPipeServerStream { } impl Stream for NamedPipeServerStream { - type Item = io::Result; + type Item = io::Result<(NamedPipeServer, PeerInfo)>; fn poll_next( self: Pin<&mut Self>, cx: &mut Context<'_>, - ) -> Poll>> { + ) -> Poll>> { let this = self.project(); this.rx.poll_recv(cx).map(|v| v.map(Ok)) diff --git a/apps/desktop/desktop_native/core/src/ssh_agent/peercred_unix_listener_stream.rs b/apps/desktop/desktop_native/core/src/ssh_agent/peercred_unix_listener_stream.rs index c17eab62fc9..85887df579c 100644 --- a/apps/desktop/desktop_native/core/src/ssh_agent/peercred_unix_listener_stream.rs +++ b/apps/desktop/desktop_native/core/src/ssh_agent/peercred_unix_listener_stream.rs @@ -46,13 +46,8 @@ impl Stream for PeercredUnixListenerStream { }; let peer_info = peerinfo::gather::get_peer_info(pid as u32); match peer_info { - Ok(info) => { - println!("name {:?}", info.process_name()); - println!("uid {:?}", info.uid()); - Poll::Ready(Some(Ok((stream, info)))) - } + Ok(info) => Poll::Ready(Some(Ok((stream, info)))), Err(err) => { - println!("Failed to get peer info: {}", err); Poll::Ready(Some(Err(io::Error::new( io::ErrorKind::Other, format!("Failed to get peer info: {}", err),