communicator/ssh: more error handlingin Upload

13 years ago · 043264ca77
parent a46c7afcf7
commit 043264ca77
1 changed files with 4 additions and 1 deletions
--- 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 {