feat(exec_utils): Add sshpass support for scp #19

sikalabs · Jan 27, 2024 · b986118 · b986118
1 parent 9f89bc3
commit b986118
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/pkg/utils/exec_utils/exec_utils.go b/pkg/utils/exec_utils/exec_utils.go
@@ -74,15 +74,23 @@ func rawSCP(taskInput libtask.TaskInput, localPath string, remotePath string) er
 	if taskInput.NoStrictHostKeyChecking {
 		args = append([]string{"-o", "StrictHostKeyChecking=no"}, args...)
 	}
-	return Exec(taskInput, "scp", args...)
+	if taskInput.SSHPassword != "" {
+		return Exec(taskInput, "sshpass", append([]string{"-p", taskInput.SSHPassword, "scp"}, args...)...)
+	} else {
+		return Exec(taskInput, "scp", args...)
+	}
 }
 
 func rawSCPRemoteToLocal(taskInput libtask.TaskInput, remotePath string, localPath string) error {
 	args := []string{taskInput.SSHTarget + ":" + remotePath, localPath}
 	if taskInput.NoStrictHostKeyChecking {
 		args = append([]string{"-o", "StrictHostKeyChecking=no"}, args...)
 	}
-	return Exec(taskInput, "scp", args...)
+	if taskInput.SSHPassword != "" {
+		return Exec(taskInput, "sshpass", append([]string{"-p", taskInput.SSHPassword, "scp"}, args...)...)
+	} else {
+		return Exec(taskInput, "scp", args...)
+	}
 }
 
 func SCP(taskInput libtask.TaskInput, localSrc string, remoteDst string) error {