containerd-shim-kata-v2: add the service CloseIO support

Add the CloseIO api support to close a process's input stream. Signed-off-by: fupan <[email protected]>
kata-containers · Nov 28, 2018 · ec4f27b · ec4f27b
1 parent 8c95b75
commit ec4f27b
Showing 1 changed file with 24 additions and 1 deletion.
diff --git a/containerd-shim-v2/service.go b/containerd-shim-v2/service.go
@@ -501,7 +501,30 @@ func (s *service) Pids(ctx context.Context, r *taskAPI.PidsRequest) (*taskAPI.Pi
 
 // CloseIO of a process
 func (s *service) CloseIO(ctx context.Context, r *taskAPI.CloseIORequest) (*ptypes.Empty, error) {
-	return nil, errdefs.ErrNotImplemented
+	s.Lock()
+	defer s.Unlock()
+
+	c, err := s.getContainer(r.ID)
+	if err != nil {
+		return nil, err
+	}
+
+	tty := c.ttyio
+	if r.ExecID != "" {
+		execs, err := c.getExec(r.ExecID)
+		if err != nil {
+			return nil, err
+		}
+		tty = execs.ttyio
+	}
+
+	if tty != nil && tty.Stdin != nil {
+		if err := tty.Stdin.Close(); err != nil {
+			return nil, errors.Wrap(err, "close stdin")
+		}
+	}
+
+	return empty, nil
 }
 
 // Checkpoint the container