From b1fcb1745f71ea841cf362d9907968aad11f26ed Mon Sep 17 00:00:00 2001
From: Michael Crosby <crosbymichael@gmail.com>
Date: Fri, 24 Mar 2017 14:03:26 -0700
Subject: [PATCH] Open stdin write side in shim

Signed-off-by: Michael Crosby <crosbymichael@gmail.com>
---
 cmd/ctr/run.go     |  2 ++
 linux/shim/exec.go | 14 ++++++++++++++
 linux/shim/init.go | 13 +++++++++++++
 linux/shim/io.go   |  3 +++
 4 files changed, 32 insertions(+)

diff --git a/cmd/ctr/run.go b/cmd/ctr/run.go
index ce2a959..49063c6 100644
--- a/cmd/ctr/run.go
+++ b/cmd/ctr/run.go
@@ -4,6 +4,7 @@ import (
 	gocontext "context"
 	"encoding/json"
 	"fmt"
+	"os"
 	"path/filepath"
 	"runtime"
 	"strconv"
@@ -254,6 +255,7 @@ var runCommand = cli.Command{
 		if err != nil {
 			return err
 		}
+		defer os.RemoveAll(tmpDir)
 		events, err := containers.Events(ctx, &execution.EventsRequest{})
 		if err != nil {
 			return err
diff --git a/linux/shim/exec.go b/linux/shim/exec.go
index 4b45273..5328a6e 100644
--- a/linux/shim/exec.go
+++ b/linux/shim/exec.go
@@ -4,14 +4,17 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"io"
 	"os"
 	"path/filepath"
 	"sync"
+	"syscall"
 
 	"github.com/crosbymichael/console"
 	runc "github.com/crosbymichael/go-runc"
 	shimapi "github.com/docker/containerd/api/services/shim"
 	specs "github.com/opencontainers/runtime-spec/specs-go"
+	"github.com/tonistiigi/fifo"
 )
 
 type execProcess struct {
@@ -22,6 +25,7 @@ type execProcess struct {
 	io      runc.IO
 	status  int
 	pid     int
+	closers []io.Closer
 
 	parent *initProcess
 }
@@ -66,6 +70,13 @@ func newExecProcess(context context.Context, path string, r *shimapi.ExecRequest
 	if err := parent.runc.Exec(context, parent.id, spec, opts); err != nil {
 		return nil, err
 	}
+	if r.Stdin != "" {
+		sc, err := fifo.OpenFifo(context, r.Stdin, syscall.O_WRONLY|syscall.O_NONBLOCK, 0)
+		if err != nil {
+			return nil, err
+		}
+		e.closers = append(e.closers, sc)
+	}
 	if socket != nil {
 		console, err := socket.ReceiveMaster()
 		if err != nil {
@@ -111,6 +122,9 @@ func (e *execProcess) Exited(status int) {
 	e.status = status
 	e.Wait()
 	if e.io != nil {
+		for _, c := range e.closers {
+			c.Close()
+		}
 		e.io.Close()
 	}
 }
diff --git a/linux/shim/init.go b/linux/shim/init.go
index aba5c60..d11669b 100644
--- a/linux/shim/init.go
+++ b/linux/shim/init.go
@@ -2,6 +2,7 @@ package shim
 
 import (
 	"context"
+	"io"
 	"os"
 	"path/filepath"
 	"sync"
@@ -11,6 +12,7 @@ import (
 	runc "github.com/crosbymichael/go-runc"
 	"github.com/docker/containerd"
 	shimapi "github.com/docker/containerd/api/services/shim"
+	"github.com/tonistiigi/fifo"
 )
 
 type initProcess struct {
@@ -23,6 +25,7 @@ type initProcess struct {
 	runc    *runc.Runc
 	status  int
 	pid     int
+	closers []io.Closer
 }
 
 func newInitProcess(context context.Context, path string, r *shimapi.CreateRequest) (*initProcess, error) {
@@ -73,6 +76,13 @@ func newInitProcess(context context.Context, path string, r *shimapi.CreateReque
 	if err := p.runc.Create(context, r.ID, r.Bundle, opts); err != nil {
 		return nil, err
 	}
+	if r.Stdin != "" {
+		sc, err := fifo.OpenFifo(context, r.Stdin, syscall.O_WRONLY|syscall.O_NONBLOCK, 0)
+		if err != nil {
+			return nil, err
+		}
+		p.closers = append(p.closers, sc)
+	}
 	if socket != nil {
 		console, err := socket.ReceiveMaster()
 		if err != nil {
@@ -125,6 +135,9 @@ func (p *initProcess) Delete(context context.Context) error {
 	p.Wait()
 	err := p.runc.Delete(context, p.id)
 	if p.io != nil {
+		for _, c := range p.closers {
+			c.Close()
+		}
 		p.io.Close()
 	}
 	return err
diff --git a/linux/shim/io.go b/linux/shim/io.go
index 8fbd647..8387c04 100644
--- a/linux/shim/io.go
+++ b/linux/shim/io.go
@@ -68,6 +68,9 @@ func copyPipes(ctx context.Context, rio runc.IO, stdin, stdout, stderr string, w
 		}
 		dest(fw, fr)
 	}
+	if stdin == "" {
+		return nil
+	}
 	f, err := fifo.OpenFifo(ctx, stdin, syscall.O_RDONLY, 0)
 	if err != nil {
 		return fmt.Errorf("containerd-shim: opening %s failed: %s", stdin, err)