github/containerd
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 2 Wiki Activity

Call CloseIO when stdin closes in ctr

Browse Source 
Fixes #2439

Signed-off-by: Michael Crosby <crosbymichael@gmail.com>
This commit is contained in:
Michael Crosby 2019-07-26 19:52:03 +00:00
parent ab78270e03
commit f543f2fbcd
2 changed files with 40 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
cmd/ctr/commands/tasks/exec.go
View File

@ -18,9 +18,12 @@ package tasks
import (
"errors"
"io"
"net/url"
"os"
"github.com/containerd/console"
"github.com/containerd/containerd"
"github.com/containerd/containerd/cio"
"github.com/containerd/containerd/cmd/ctr/commands"
"github.com/sirupsen/logrus"
@ -91,7 +94,12 @@ var execCommand = cli.Command{
pspec.Terminal = tty
pspec.Args = args
var ioCreator cio.Creator
var (
ioCreator cio.Creator
stdinC = &stdinCloser{
stdin: os.Stdin,
}
)
if logURI := context.String("log-uri"); logURI != "" {
uri, err := url.Parse(logURI)
@ -109,7 +117,7 @@ var execCommand = cli.Command{
ioCreator = cio.LogURI(uri)
} else {
cioOpts := []cio.Opt{cio.WithStdio, cio.WithFIFODir(context.String("fifo-dir"))}
cioOpts := []cio.Opt{cio.WithStreams(stdinC, os.Stdout, os.Stderr), cio.WithFIFODir(context.String("fifo-dir"))}
if tty {
cioOpts = append(cioOpts, cio.WithTerminal)
}
@ -120,6 +128,9 @@ var execCommand = cli.Command{
if err != nil {
return err
}
stdinC.closer = func() {
process.CloseIO(ctx, containerd.WithStdinCloser)
}
// if detach, we should not call this defer
if !detach {
defer process.Delete(ctx)
@ -166,3 +177,18 @@ var execCommand = cli.Command{
return nil
},
}
type stdinCloser struct {
stdin *os.File
closer func()
}
func (s *stdinCloser) Read(p []byte) (int, error) {
n, err := s.stdin.Read(p)
if err == io.EOF {
if s.closer != nil {
s.closer()
}
}
return n, err
}

14
cmd/ctr/commands/tasks/tasks_unix.go
View File

@ -69,7 +69,10 @@ func HandleConsoleResize(ctx gocontext.Context, task resizer, con console.Consol
// NewTask creates a new task
func NewTask(ctx gocontext.Context, client *containerd.Client, container containerd.Container, checkpoint string, con console.Console, nullIO bool, logURI string, ioOpts []cio.Opt, opts ...containerd.NewTaskOpts) (containerd.Task, error) {
stdio := cio.NewCreator(append([]cio.Opt{cio.WithStdio}, ioOpts...)...)
stdinC := &stdinCloser{
stdin: os.Stdin,
}
stdio := cio.NewCreator(append([]cio.Opt{cio.WithStreams(stdinC, os.Stdout, os.Stderr)}, ioOpts...)...)
if checkpoint != "" {
im, err := client.GetImage(ctx, checkpoint)
if err != nil {
@ -94,7 +97,14 @@ func NewTask(ctx gocontext.Context, client *containerd.Client, container contain
}
ioCreator = cio.LogURI(u)
}
return container.NewTask(ctx, ioCreator, opts...)
t, err := container.NewTask(ctx, ioCreator, opts...)
if err != nil {
return nil, err
}
stdinC.closer = func() {
t.CloseIO(ctx, containerd.WithStdinCloser)
}
return t, nil
}
func getNewTaskOpts(context *cli.Context) []containerd.NewTaskOpts {