From b63b98ee2766321e2ca6f3b159c2bfb303870105 Mon Sep 17 00:00:00 2001
From: Solomon Hykes <solomon@docker.com>
Date: Fri, 25 Apr 2014 16:47:03 -0700
Subject: [PATCH] engine.Sender and engine.Receiver support stdin

Docker-DCO-1.1-Signed-off-by: Solomon Hykes <solomon@docker.com> (github: shykes)
---
 engine/remote.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/engine/remote.go b/engine/remote.go
index 1e8777a4b7..48638e4383 100644
--- a/engine/remote.go
+++ b/engine/remote.go
@@ -46,6 +46,12 @@ func (s *Sender) Handle(job *Job) Status {
 		tasks.Done()
 		return nil
 	})
+	r.NewRoute().KeyStartsWith("cmd", "log", "stdin").HasAttachment().Handler(func(p []byte, stdin *os.File) error {
+		tasks.Add(1)
+		io.Copy(stdin, job.Stdin)
+		tasks.Done()
+		return nil
+	})
 	var status int
 	r.NewRoute().KeyStartsWith("cmd", "status").Handler(func(p []byte, f *os.File) error {
 		cmd := data.Message(p).Get("cmd")
@@ -96,6 +102,11 @@ func (rcv *Receiver) Run() error {
 			return err
 		}
 		job.Stderr.Add(stderr)
+		stdin, err := beam.SendPipe(peer, data.Empty().Set("cmd", "log", "stdin").Bytes())
+		if err != nil {
+			return err
+		}
+		job.Stdin.Add(stdin)
 		// ignore error because we pass the raw status
 		job.Run()
 		err = peer.Send(data.Empty().Set("cmd", "status", fmt.Sprintf("%d", job.status)).Bytes(), nil)