fix(v1.0): support file upload md5 validate

cmd/root.go

@@ -51,12 +51,12 @@ var rootCmd = &cobra.Command{
 				case "ssh":
 					sshType.Cmd(host, command)
 				case "scp":
-					sshType.Copy(host, localFilePath, remoteFilePath)
+					sshType.CopyForMD5(host, localFilePath, remoteFilePath, "")
 				case "ssh|scp":
 					sshType.Cmd(host, command)
-					sshType.Copy(host, localFilePath, remoteFilePath)
+					sshType.CopyForMD5(host, localFilePath, remoteFilePath, "")
 				case "scp|ssh":
-					sshType.Copy(host, localFilePath, remoteFilePath)
+					sshType.CopyForMD5(host, localFilePath, remoteFilePath, "")
 					sshType.Cmd(host, command)
 				default:
 					sshType.Cmd(host, command)

pkg/sshutil/scp.go

@@ -1,14 +1,42 @@
 package sshutil
 
 import (
+	"fmt"
 	"github.com/pkg/sftp"
 	"github.com/wonderivan/logger"
 	"golang.org/x/crypto/ssh"
 	"net"
 	"os"
+	"os/exec"
 	"time"
 )
 
+//Copy is
+func (ss *SSH) CopyForMD5(host, localFilePath, remoteFilePath, md5 string) bool {
+	//如果有md5则可以验证
+	//如果没有md5则拿到本地数据后验证
+	if md5 == "" {
+		cmd := fmt.Sprintf("md5sum %s | cut -d\" \" -f1", localFilePath)
+		c := exec.Command("sh", "-c", cmd)
+		out, err := c.CombinedOutput()
+		if err != nil {
+			logger.Error(err)
+		}
+		md5 = string(out)
+	}
+	logger.Debug("source file md5 value is %s", md5)
+	ss.Copy(host, localFilePath, remoteFilePath)
+	cmd := fmt.Sprintf("md5sum %s | cut -d\" \" -f1", remoteFilePath)
+	remoteMD5 := ss.CmdToString(host, cmd)
+	logger.Debug("host: %s , remote md5: %s", host, remoteMD5)
+	if remoteMD5 == md5 {
+		logger.Info("md5 validate true")
+		return true
+	}
+	logger.Error("md5 validate false")
+	return false
+}
+
 //Copy is
 func (ss *SSH) Copy(host, localFilePath, remoteFilePath string) {
 	sftpClient, err := ss.sftpConnect(host)