From 47366d7696e22ab13e43053e60d894258d256cb8 Mon Sep 17 00:00:00 2001 From: alen <63181977@qq.com> Date: Thu, 8 Jun 2017 10:06:13 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E6=88=90=E8=AF=BB=E5=8F=96json?= =?UTF-8?q?=E6=96=87=E4=BB=B6=E3=80=81=E7=BB=93=E6=9E=9C=E5=AF=BC=E5=87=BA?= =?UTF-8?q?=E4=B8=BAtxt=E6=96=87=E4=BB=B6=E3=80=81=E8=B6=85=E6=97=B6?= =?UTF-8?q?=E5=A4=84=E7=90=86=EF=BC=88=E6=97=B6=E9=99=90=E5=8F=82=E6=95=B0?= =?UTF-8?q?=E5=8F=AF=E8=BE=93=E5=85=A5=EF=BC=89=20=E6=8E=A7=E5=88=B6?= =?UTF-8?q?=E5=B9=B6=E5=8F=91=E8=AE=BF=E9=97=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cfg.go | 32 ++++++++++++ main.go | 140 ++++++++++++++++++++++++++++++++++++-------------- sshconnect.go | 14 ++--- 3 files changed, 141 insertions(+), 45 deletions(-) diff --git a/cfg.go b/cfg.go index eb9827a..7aa4d88 100644 --- a/cfg.go +++ b/cfg.go @@ -1,10 +1,13 @@ package main import ( + "bufio" "encoding/binary" + "encoding/json" "io/ioutil" "log" "net" + "os" "strconv" "strings" ) @@ -36,6 +39,35 @@ func Getfile(filePath string) ([]string, error) { return result, nil } +//gu +func GetJsonFile(filePath string) ([]SSHHost, error) { + result := []SSHHost{} + b, err := ioutil.ReadFile(filePath) + if err != nil { + log.Println("read file ", filePath, err) + return result, err + } + var m HostJson + json.Unmarshal(b, &m) + result = m.SshHosts + return result, nil +} +func WriteIntoTxt(sshHost SSHHost) error { + outputFile, outputError := os.OpenFile(sshHost.Host+".txt", os.O_WRONLY|os.O_CREATE, 0666) + if outputError != nil { + return outputError + } + defer outputFile.Close() + + outputWriter := bufio.NewWriter(outputFile) + //var outputString string + + outputString := sshHost.Result + outputWriter.WriteString(outputString) + outputWriter.Flush() + return nil +} + func GetIpList(filePath string) ([]string, error) { res, err := Getfile(filePath) if err != nil { diff --git a/main.go b/main.go index 1dea59a..ed05b1d 100644 --- a/main.go +++ b/main.go @@ -4,17 +4,23 @@ import ( "flag" "fmt" "log" + "strconv" "strings" "time" // "github.com/bitly/go-simplejson" ) -type sshhost struct { - host string - port int - username string - password string - cmd []string +type SSHHost struct { + Host string + Port int + Username string + Password string + CmdFile string + Cmd []string + Result string +} +type HostJson struct { + SshHosts []SSHHost } func main() { @@ -22,62 +28,100 @@ func main() { cmd := flag.String("cmd", "", "cmds") username := flag.String("u", "", "username") password := flag.String("p", "", "password") - port := flag.Int("port", 22, "ssh port") - cmdfile := flag.String("cmdfile", "", "cmdfile path") - hostfile := flag.String("hostfile", "", "hostfile path") - ipfile := flag.String("ipfile", "", "hostfile path") + port := flag.String("port", "", "ssh port") + cmdFile := flag.String("cmdfile", "", "cmdfile path") + hostFile := flag.String("hostfile", "", "hostfile path") + ipFile := flag.String("ipfile", "", "hostfile path") cfg := flag.String("cfg", "", "cfg path") + //gu + jsonFile := flag.String("j", "ssh.json", "Json File Path") + outTxt := flag.Bool("outTxt", false, "write result into txt") + timeLimit := flag.Duration("t", 30, "max timeout") + numLimit := flag.Int("n", 20, "max execute number") flag.Parse() - - var cmdlist []string - var hostlist []string + var cmdList []string + var hostList []string var err error + //gu + var usernameList []string + var passwordList []string + var portList []string - sshhosts := []sshhost{} - var host_struct sshhost + sshHosts := []SSHHost{} + var host_Struct SSHHost - if *ipfile != "" { - hostlist, err = GetIpList(*ipfile) + if *ipFile != "" { + hostList, err = GetIpList(*ipFile) if err != nil { log.Println("load hostlist error: ", err) return } } - if *hostfile != "" { - hostlist, err = Getfile(*hostfile) + if *hostFile != "" { + hostList, err = Getfile(*hostFile) if err != nil { log.Println("load hostfile error: ", err) return } } if *hosts != "" { - hostlist = strings.Split(*hosts, ";") - + hostList = strings.Split(*hosts, ";") } - if *cmdfile != "" { - cmdlist, err = Getfile(*cmdfile) + //gu + if *username != "" { + usernameList = strings.Split(*username, ";") + } + + if *password != "" { + passwordList = strings.Split(*password, ";") + } + + if *port != "" { + portList = strings.Split(*port, ";") + } + //// + if *cmdFile != "" { + cmdList, err = Getfile(*cmdFile) if err != nil { log.Println("load cmdfile error: ", err) return } } if *cmd != "" { - cmdlist = strings.Split(*cmd, ";") + cmdList = strings.Split(*cmd, ";") } - if *cfg == "" { - for _, host := range hostlist { - host_struct.host = host - host_struct.username = *username - host_struct.password = *password - host_struct.port = *port - host_struct.cmd = cmdlist - sshhosts = append(sshhosts, host_struct) + for pos, host := range hostList { + host_Struct.Host = host + host_Struct.Username = usernameList[pos] + host_Struct.Password = passwordList[pos] + host_Struct.Port, _ = strconv.Atoi(portList[pos]) + host_Struct.Cmd = cmdList + sshHosts = append(sshHosts, host_Struct) } } + //gu + if *jsonFile != "" { + sshHosts, err = GetJsonFile(*jsonFile) + if err != nil { + log.Println("load jsonFile error: ", err) + return + } + for i := 0; i < len(sshHosts); i++ { + cmdList, err = Getfile(sshHosts[i].CmdFile) + if err != nil { + log.Println("load cmdFile error: ", err) + return + } + //fmt.Println(cmdList) + sshHosts[i].Cmd = cmdList + } + //为什么不能用for range + } + /* else { cfgjson, err := GetfileAll(*cfg) @@ -96,23 +140,41 @@ func main() { } */ //fmt.Println(sshhosts) - - chs := make([]chan string, len(sshhosts)) - for i, host := range sshhosts { + chLimit := make(chan bool, numLimit) + chs := make([]chan string, len(sshHosts)) + limitFunc := func(chLimit chan bool, ch chan string, host SSHHost) { + dossh(host.Username, host.Password, host.Host, host.Cmd, host.Port, ch) + <-chLimit + } + for i, host := range sshHosts { chs[i] = make(chan string, 1) - go dossh(host.username, host.password, host.host, host.cmd, host.port, chs[i]) + chLimit <- true + go limitFunc(chLimit, chs[i], host) } for i, ch := range chs { - fmt.Println(sshhosts[i].host, " ssh start") + fmt.Println(sshHosts[i].Host, " ssh start") select { case res := <-ch: if res != "" { fmt.Println(res) + sshHosts[i].Result += res } - case <-time.After(30 * 1000 * 1000 * 1000): + case <-time.After(*timeLimit * 1000 * 1000 * 1000): log.Println("SSH run timeout") + sshHosts[i].Result += ("SSH run timeout：" + strconv.Itoa(int(*timeLimit)) + "second.") + } + fmt.Println(sshHosts[i].Host, " ssh end") + } + + //gu + if *outTxt { + for i := 0; i < len(sshHosts); i++ { + err = WriteIntoTxt(sshHosts[i]) + if err != nil { + log.Println("write into txt error: ", err) + return + } } - fmt.Println(sshhosts[i].host, " ssh end\n") } } diff --git a/sshconnect.go b/sshconnect.go index ecc9c93..c44ffcb 100644 --- a/sshconnect.go +++ b/sshconnect.go @@ -3,9 +3,9 @@ package main import ( "bytes" "fmt" - "time" - "net" + //"os" + "time" "golang.org/x/crypto/ssh" ) @@ -55,29 +55,31 @@ func connect(user, password, host string, port int) (*ssh.Session, error) { func dossh(username, password, ip string, cmdlist []string, port int, ch chan string) { session, err := connect(username, password, ip, port) + if err != nil { ch <- fmt.Sprintf("<%s>", err.Error()) + //<-chLimit return + } defer session.Close() // cmd := "ls;date;exit" - stdinBuf, _ := session.StdinPipe() - + //fmt.Fprintf(os.Stdout, "%s", stdinBuf) var outbt, errbt bytes.Buffer session.Stdout = &outbt session.Stderr = &errbt - err = session.Shell() for _, c := range cmdlist { c = c + "\n" stdinBuf.Write([]byte(c)) + } session.Wait() ch <- (outbt.String() + errbt.String()) - + //<-chLimit return }