ensure stream is closed in localhost provider (moby#36)

Signed-off-by: Alex Couture-Beil <alex@earthly.dev>
alexcb · Apr 1, 2021 · d20d543 · d20d543
1 parent 5b54467
commit d20d543
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/session/localhost/localhost.go b/session/localhost/localhost.go
@@ -41,6 +41,7 @@ func LocalhostExec(ctx context.Context, c session.Caller, args []string, dir str
 	if err != nil {
 		return errors.WithStack(err)
 	}
+	defer stream.CloseSend()
 
 	req := InputMessage{
 		Command: args,
@@ -91,6 +92,7 @@ func LocalhostGet(ctx context.Context, c session.Caller, src, dest string, mount
 	if err != nil {
 		return errors.WithStack(err)
 	}
+	defer stream.CloseSend()
 
 	req := BytesMessage{
 		Data: []byte(src),
@@ -213,6 +215,7 @@ func LocalhostPut(ctx context.Context, c session.Caller, src, dst string) error
 	if err != nil {
 		return errors.WithStack(err)
 	}
+	defer stream.CloseSend()
 
 	fi, err := os.Stat(src)
 	if err != nil {