communicator/ssh: more error handlingin Upload

043264ca · Mitchell Hashimoto · a46c7afc · 043264ca
Commit 043264ca authored Aug 23, 2013 by Mitchell Hashimoto
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

communicator/ssh/communicator.go communicator/ssh/communicator.go +4 -1

No files found.
--- a/communicator/ssh/communicator.go
+++ b/communicator/ssh/communicator.go
@@ -160,7 +160,10 @@ func (c *comm) Upload(path string, input io.Reader) error {
 		return err
 	}

-	io.Copy(w, input_memory)
+	if _, err := io.Copy(w, input_memory); err != nil{
+		return err
+	}
+
 	fmt.Fprint(w, "\x00")
 	err = checkSCPStatus(stdoutR)
 	if err != nil {