golang实现跨平台scp批量群发文件

package main
import (
	"bufio"
	"flag"
	"fmt"
	"io"
	"os"
	"path/filepath"
	"strings"
	"github.com/czxichen/AutoWork/tools/split"
	"golang.org/x/crypto/ssh"
)
var (
	passwd = flag.String("p", "", "-p passwd 指定密码.")
	user = flag.String("u", "root", "-u root 指定登录用户.")
	cfg = flag.String("c", "serverlist", "-c serverlist 指定serverlist")
	ip_port = flag.String("i", "", "-i ip:port 指定目标机器的IP端口,不能和-c同时使用.")
	dpath = flag.String("d", "", "-d /tmp/20160531.zip 指定发送到的路径,不能为空.")
	spath = flag.String("s", "", "-s 20160531.zip 指定要发送文件的路径,不能为空.")
)
func main() {
	flag.Parse()
	if *dpath == "" || *spath == "" {
		flag.PrintDefaults()
		os.Exit(1)
	}
	File, err := os.Open(*spath)
	if err != nil {
		fmt.Println("打开文件失败:", err)
		os.Exit(1)
	}
	info, _ := File.Stat()
	defer File.Close()
	if *ip_port != "" && *passwd != "" {
		Client, err := dail(*user, *passwd, *ip_port)
		if err != nil {
			fmt.Printf("连接%s失败.\n", err)
			os.Exit(1)
		}
		scp(Client, File, info.Size(), *dpath)
		return
	}
	var list [][]string
	ok := (*passwd != "" && *ip_port == "")
	list = config(*cfg, ok)
	if len(list) <= 0 {
		fmt.Println("serverlist 不能为空.")
		os.Exit(1)
	}
	for _, v := range list {
		if ok {
			*ip_port = v[0]
		} else {
			*user = v[0]
			*passwd = v[1]
			*ip_port = v[2]
		}
		Client, err := dail(*user, *passwd, *ip_port)
		if err != nil {
			fmt.Printf("连接%s失败.\n", err)
			continue
		}
		scp(Client, File, info.Size(), *dpath)
	}
}
func dail(user, password, ip_port string) (*ssh.Client, error) {
	PassWd := []ssh.AuthMethod{ssh.Password(password)}
	Conf := ssh.ClientConfig{User: user, Auth: PassWd}
	return ssh.Dial("tcp", ip_port, &Conf)
}
func scp(Client *ssh.Client, File io.Reader, size int64, path string) {
	filename := filepath.Base(path)
	dirname := strings.Replace(filepath.Dir(path), "\\", "/", -1)
	defer Client.Close()
	session, err := Client.NewSession()
	if err != nil {
		fmt.Println("创建Session失败:", err)
		return
	}
	go func() {
		w, _ := session.StdinPipe()
		fmt.Fprintln(w, "C0644", size, filename)
		io.CopyN(w, File, size)
		fmt.Fprint(w, "\x00")
		w.Close()
	}()
	if err := session.Run(fmt.Sprintf("/usr/bin/scp -qrt %s", dirname)); err != nil {
		fmt.Println("执行scp命令失败:", err)
		return
	} else {
		fmt.Printf("%s 发送成功.\n", Client.RemoteAddr())
		session.Close()
	}
	if session, err = Client.NewSession(); err == nil {
		defer session.Close()
		buf, err := session.Output(fmt.Sprintf("/usr/bin/md5sum %s", path))
		if err != nil {
			fmt.Println("检查md5失败:", err)
			return
		}
		fmt.Printf("%s 的MD5:\n%s\n", Client.RemoteAddr(), string(buf))
	}
}
func config(path string, ok bool) (list [][]string) {
	File, err := os.Open(path)
	if err != nil {
		fmt.Printf("打开配置文件失败:%s\n", err)
		os.Exit(1)
	}
	defer File.Close()
	buf := bufio.NewReader(File)
	for {
		line, _, err := buf.ReadLine()
		if err != nil {
			break
		}
		str := strings.TrimSpace(string(line))
		strs := split.Split(str)
		if ok {
			if len(strs) != 1 {
				continue
			}
		} else {
			if len(strs) != 3 {
				continue
			}
		}
		list = append(list, strs)
	}
	return
}

本文来自:CSDN博客

感谢作者:fyxichen

查看原文:golang实现跨平台scp批量群发文件

• 编辑
• 预览