diff --git a/src/agent/rustjail/src/pipestream.rs b/src/agent/rustjail/src/pipestream.rs index 4215190ace..1312a4b6a4 100644 --- a/src/agent/rustjail/src/pipestream.rs +++ b/src/agent/rustjail/src/pipestream.rs @@ -77,10 +77,6 @@ impl PipeStream { Ok(Self(AsyncFd::new(StreamFd(fd))?)) } - pub fn shutdown(&mut self) -> io::Result<()> { - self.0.get_mut().close() - } - pub fn from_fd(fd: RawFd) -> Self { unsafe { Self::from_raw_fd(fd) } } @@ -164,7 +160,10 @@ impl AsyncWrite for PipeStream { } fn poll_shutdown(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll> { - self.get_mut().shutdown()?; + // Do nothing in shutdown is very important + // The only right way to shutdown pipe is drop it + // Otherwise PipeStream will conflict with its twins + // Because they both have same fd, and both registered. Poll::Ready(Ok(())) } } diff --git a/src/agent/rustjail/src/sync_with_async.rs b/src/agent/rustjail/src/sync_with_async.rs index 7f44ab2819..4626801158 100644 --- a/src/agent/rustjail/src/sync_with_async.rs +++ b/src/agent/rustjail/src/sync_with_async.rs @@ -117,28 +117,20 @@ pub async fn write_async(pipe_w: &mut PipeStream, msg_type: i32, data_str: &str) } match msg_type { - SYNC_FAILED => match write_count(pipe_w, data_str.as_bytes(), data_str.len()).await { - Ok(_) => pipe_w.shutdown()?, - Err(e) => { - pipe_w.shutdown()?; + SYNC_FAILED => { + if let Err(e) = write_count(pipe_w, data_str.as_bytes(), data_str.len()).await { return Err(anyhow!(e).context("error in send message to process")); } - }, + } SYNC_DATA => { let length: i32 = data_str.len() as i32; write_count(pipe_w, &length.to_be_bytes(), MSG_SIZE) .await - .or_else(|e| { - pipe_w.shutdown()?; - Err(anyhow!(e).context("error in send message to process")) - })?; + .map_err(|e| anyhow!(e).context("error in send message to process"))?; write_count(pipe_w, data_str.as_bytes(), data_str.len()) .await - .or_else(|e| { - pipe_w.shutdown()?; - Err(anyhow!(e).context("error in send message to process")) - })?; + .map_err(|e| anyhow!(e).context("error in send message to process"))?; } _ => (),