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

原创 2016年05月31日 10:46:44
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端口,必须和-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 [][]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)
		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
}

版权声明:本文为博主原创文章,交流邮箱czxichen@163.com,交流群:259693140(此群不允许闲聊)

Go语言使用Scp传输文件

package main import ( "fmt" "go-ssh/ssh" "io" "os" ) const ( user = "user" ip_port = "i...
  • fyxichen
  • fyxichen
  • 2015年05月05日 09:26
  • 1114

使用GoLang实现文件远程传输

This tool(file_scp) likes scp(secure copy, remote file copy program) but uses golang to copies files...
  • delphiwcdj
  • delphiwcdj
  • 2015年08月23日 13:34
  • 2565

Golang优秀开源项目汇总

原文链接:https://studygolang.com/articles/7708 Golang优秀开源项目汇总(持续更新。。。) 我把这个汇总放在github上了, 后面更新也会在githu...
  • qq_21794823
  • qq_21794823
  • 2017年09月20日 10:03
  • 2115

go语言资源大全

转载地址:https://github.com/avelino/awesome-go Contents Awesome Go Audio & MusicAuthenticatio...
  • yang889999888
  • yang889999888
  • 2017年07月19日 13:05
  • 1111

批量SCP文件的脚本

最近在集群上部署Hadoop时发现在服务器之间传递文件比较麻烦,原因是使用scp每次只能指定一台目标服务器,并且必须指定用户名和目标路径。当集群中存在很多服务器时,分发Hadoop的配置文件就会耗费很...
  • sky_walker85
  • sky_walker85
  • 2014年02月27日 16:19
  • 3427

ubuntu下scp批量拷贝文件不需要输入密码

写在前面的话 总有一些时候,我总是对我的人生充满恐惧。比如说现在,总是不知所措,总是自卑,总是悲观。 唯一使我平静下来的事情只有一件,就是想到那个总是一个人十万分专注的静静的坐在5教自习教室的109的...
  • Grace_0642
  • Grace_0642
  • 2014年12月31日 11:26
  • 1880

golang 远程传输文件

概述之前有一篇介绍如何使用 golang 通过SSH协议来执行远程命令:golang 执行远程命令 同样,通过SSH协议也可以使用 golang 来远程传输文件。除了 SSH 的库,为了传输文件,还需...
  • u012798391
  • u012798391
  • 2016年10月20日 14:37
  • 211

go安装

环境: go1.9.2 centos 7 下载: https://golang.org/dl/ 选择最新版本 go1.9.2.linux-amd64.tar.gz 安装 在us...
  • a1368783069
  • a1368783069
  • 2018年01月06日 18:46
  • 59

通过java程序实现scp上传和下载文件

我们对ssh服务器进行文件上传和下载操作以及通过客户端直接发送命令来实现对服务器的操作该怎么实现呢?目前主流的还是直接在客户端使用shell脚本来实现,关于linux中ssh使用rsa免密码登录和下载...
  • wyc_cs
  • wyc_cs
  • 2016年06月24日 09:33
  • 3116

批量执行linux环境下的SCP命令

expect用于自动化地执行linux环境下的命令行交互任务,例如scp、ssh之类需要用户手动输入密码然后确认的任务。有了这个工具,定义在scp过程中可能遇到的情况,然后编写相应的处理语句,就可以自...
  • bluetropic
  • bluetropic
  • 2013年09月24日 10:10
  • 1310
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:golang实现跨平台scp批量群发文件
举报原因:
原因补充:

(最多只允许输入30个字)