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

package mainimport ("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端口,必须和-p结合使用否则不生效.")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 [][]stringok := (*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)session.Close()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}