agent: Fix ctr exec stuck problem

Fixes: #9532

Close stdin when write_stdin receives data of length 0.

Stop call notify_term_close() in close_stdin, because it could
discard stdout unexpectedly.

Signed-off-by: Tim Zhang <tim@hyper.sh>

src/agent/rustjail/src/process.rs

@@ -200,15 +200,8 @@ impl Process {
     }
 
     pub async fn close_stdin(&mut self) {
-        // stdin will be closed automatically in passfd-io senario
-        if self.proc_io.is_some() {
-            return;
-        }
-
         close_process_stream!(self, term_master, TermMaster);
         close_process_stream!(self, parent_stdin, ParentStdin);
-
-        self.notify_term_close();
     }
 
     pub fn cleanup_process_stream(&mut self) {

src/agent/src/rpc.rs

@@ -584,6 +584,15 @@ impl AgentService {
         let cid = req.container_id;
         let eid = req.exec_id;
 
+        let mut resp = WriteStreamResponse::new();
+        resp.set_len(req.data.len() as u32);
+
+        // EOF of stdin
+        if req.data.is_empty() {
+            let mut sandbox = self.sandbox.lock().await;
+            let p = sandbox.find_container_process(cid.as_str(), eid.as_str())?;
+            p.close_stdin().await;
+        } else {
             let writer = {
                 let mut sandbox = self.sandbox.lock().await;
                 let p = sandbox.find_container_process(cid.as_str(), eid.as_str())?;
@@ -599,9 +608,7 @@ impl AgentService {
 
             let writer = writer.ok_or_else(|| anyhow!(ERR_CANNOT_GET_WRITER))?;
             writer.lock().await.write_all(req.data.as_slice()).await?;
-
+        }
-        let mut resp = WriteStreamResponse::new();
-        resp.set_len(req.data.len() as u32);
 
         Ok(resp)
     }
@@ -645,6 +652,7 @@ impl AgentService {
             biased;
             v = read_stream(&reader, req.len as usize)  => {
                 let vector = v?;
+
                 let mut resp = ReadStreamResponse::new();
                 resp.set_data(vector);
 
@@ -845,6 +853,9 @@ impl agent_ttrpc::AgentService for AgentService {
         ctx: &TtrpcContext,
         req: protocols::agent::CloseStdinRequest,
     ) -> ttrpc::Result<Empty> {
+        // The stdin will be closed when EOF is got in rpc `write_stdin`[runtime-rs]
+        // so this rpc will not be called anymore by runtime-rs.
+
         trace_rpc_call!(ctx, "close_stdin", req);
         is_allowed(&req).await?;