diff --git a/LICENSE b/LICENSE new file mode 100644 index 0000000..e8e87ec --- /dev/null +++ b/LICENSE @@ -0,0 +1,202 @@ + + Apache License + Version 2.0, January 2004 + http://www.apache.org/licenses/ + + TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION + + 1. Definitions. + + "License" shall mean the terms and conditions for use, reproduction, + and distribution as defined by Sections 1 through 9 of this document. + + "Licensor" shall mean the copyright owner or entity authorized by + the copyright owner that is granting the License. + + "Legal Entity" shall mean the union of the acting entity and all + other entities that control, are controlled by, or are under common + control with that entity. For the purposes of this definition, + "control" means (i) the power, direct or indirect, to cause the + direction or management of such entity, whether by contract or + otherwise, or (ii) ownership of fifty percent (50%) or more of the + outstanding shares, or (iii) beneficial ownership of such entity. + + "You" (or "Your") shall mean an individual or Legal Entity + exercising permissions granted by this License. + + "Source" form shall mean the preferred form for making modifications, + including but not limited to software source code, documentation + source, and configuration files. + + "Object" form shall mean any form resulting from mechanical + transformation or translation of a Source form, including but + not limited to compiled object code, generated documentation, + and conversions to other media types. + + "Work" shall mean the work of authorship, whether in Source or + Object form, made available under the License, as indicated by a + copyright notice that is included in or attached to the work + (an example is provided in the Appendix below). + + "Derivative Works" shall mean any work, whether in Source or Object + form, that is based on (or derived from) the Work and for which the + editorial revisions, annotations, elaborations, or other modifications + represent, as a whole, an original work of authorship. For the purposes + of this License, Derivative Works shall not include works that remain + separable from, or merely link (or bind by name) to the interfaces of, + the Work and Derivative Works thereof. + + "Contribution" shall mean any work of authorship, including + the original version of the Work and any modifications or additions + to that Work or Derivative Works thereof, that is intentionally + submitted to Licensor for inclusion in the Work by the copyright owner + or by an individual or Legal Entity authorized to submit on behalf of + the copyright owner. For the purposes of this definition, "submitted" + means any form of electronic, verbal, or written communication sent + to the Licensor or its representatives, including but not limited to + communication on electronic mailing lists, source code control systems, + and issue tracking systems that are managed by, or on behalf of, the + Licensor for the purpose of discussing and improving the Work, but + excluding communication that is conspicuously marked or otherwise + designated in writing by the copyright owner as "Not a Contribution." + + "Contributor" shall mean Licensor and any individual or Legal Entity + on behalf of whom a Contribution has been received by Licensor and + subsequently incorporated within the Work. + + 2. Grant of Copyright License. Subject to the terms and conditions of + this License, each Contributor hereby grants to You a perpetual, + worldwide, non-exclusive, no-charge, royalty-free, irrevocable + copyright license to reproduce, prepare Derivative Works of, + publicly display, publicly perform, sublicense, and distribute the + Work and such Derivative Works in Source or Object form. + + 3. Grant of Patent License. Subject to the terms and conditions of + this License, each Contributor hereby grants to You a perpetual, + worldwide, non-exclusive, no-charge, royalty-free, irrevocable + (except as stated in this section) patent license to make, have made, + use, offer to sell, sell, import, and otherwise transfer the Work, + where such license applies only to those patent claims licensable + by such Contributor that are necessarily infringed by their + Contribution(s) alone or by combination of their Contribution(s) + with the Work to which such Contribution(s) was submitted. If You + institute patent litigation against any entity (including a + cross-claim or counterclaim in a lawsuit) alleging that the Work + or a Contribution incorporated within the Work constitutes direct + or contributory patent infringement, then any patent licenses + granted to You under this License for that Work shall terminate + as of the date such litigation is filed. + + 4. Redistribution. You may reproduce and distribute copies of the + Work or Derivative Works thereof in any medium, with or without + modifications, and in Source or Object form, provided that You + meet the following conditions: + + (a) You must give any other recipients of the Work or + Derivative Works a copy of this License; and + + (b) You must cause any modified files to carry prominent notices + stating that You changed the files; and + + (c) You must retain, in the Source form of any Derivative Works + that You distribute, all copyright, patent, trademark, and + attribution notices from the Source form of the Work, + excluding those notices that do not pertain to any part of + the Derivative Works; and + + (d) If the Work includes a "NOTICE" text file as part of its + distribution, then any Derivative Works that You distribute must + include a readable copy of the attribution notices contained + within such NOTICE file, excluding those notices that do not + pertain to any part of the Derivative Works, in at least one + of the following places: within a NOTICE text file distributed + as part of the Derivative Works; within the Source form or + documentation, if provided along with the Derivative Works; or, + within a display generated by the Derivative Works, if and + wherever such third-party notices normally appear. The contents + of the NOTICE file are for informational purposes only and + do not modify the License. You may add Your own attribution + notices within Derivative Works that You distribute, alongside + or as an addendum to the NOTICE text from the Work, provided + that such additional attribution notices cannot be construed + as modifying the License. + + You may add Your own copyright statement to Your modifications and + may provide additional or different license terms and conditions + for use, reproduction, or distribution of Your modifications, or + for any such Derivative Works as a whole, provided Your use, + reproduction, and distribution of the Work otherwise complies with + the conditions stated in this License. + + 5. Submission of Contributions. Unless You explicitly state otherwise, + any Contribution intentionally submitted for inclusion in the Work + by You to the Licensor shall be under the terms and conditions of + this License, without any additional terms or conditions. + Notwithstanding the above, nothing herein shall supersede or modify + the terms of any separate license agreement you may have executed + with Licensor regarding such Contributions. + + 6. Trademarks. This License does not grant permission to use the trade + names, trademarks, service marks, or product names of the Licensor, + except as required for reasonable and customary use in describing the + origin of the Work and reproducing the content of the NOTICE file. + + 7. Disclaimer of Warranty. Unless required by applicable law or + agreed to in writing, Licensor provides the Work (and each + Contributor provides its Contributions) on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or + implied, including, without limitation, any warranties or conditions + of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A + PARTICULAR PURPOSE. You are solely responsible for determining the + appropriateness of using or redistributing the Work and assume any + risks associated with Your exercise of permissions under this License. + + 8. Limitation of Liability. In no event and under no legal theory, + whether in tort (including negligence), contract, or otherwise, + unless required by applicable law (such as deliberate and grossly + negligent acts) or agreed to in writing, shall any Contributor be + liable to You for damages, including any direct, indirect, special, + incidental, or consequential damages of any character arising as a + result of this License or out of the use or inability to use the + Work (including but not limited to damages for loss of goodwill, + work stoppage, computer failure or malfunction, or any and all + other commercial damages or losses), even if such Contributor + has been advised of the possibility of such damages. + + 9. Accepting Warranty or Additional Liability. While redistributing + the Work or Derivative Works thereof, You may choose to offer, + and charge a fee for, acceptance of support, warranty, indemnity, + or other liability obligations and/or rights consistent with this + License. However, in accepting such obligations, You may act only + on Your own behalf and on Your sole responsibility, not on behalf + of any other Contributor, and only if You agree to indemnify, + defend, and hold each Contributor harmless for any liability + incurred by, or claims asserted against, such Contributor by reason + of your accepting any such warranty or additional liability. + + END OF TERMS AND CONDITIONS + + APPENDIX: How to apply the Apache License to your work. + + To apply the Apache License to your work, attach the following + boilerplate notice, with the fields enclosed by brackets "[]" + replaced with your own identifying information. (Don't include + the brackets!) The text should be enclosed in the appropriate + comment syntax for the file format. We also recommend that a + file or class name and description of purpose be included on the + same "printed page" as the copyright notice for easier + identification within third-party archives. + + Copyright [2017] [shanghai-edu] + + Licensed under the Apache License, Version 2.0 (the "License"); + you may not use this file except in compliance with the License. + You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + + Unless required by applicable law or agreed to in writing, software + distributed under the License is distributed on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + See the License for the specific language governing permissions and + limitations under the License. \ No newline at end of file diff --git a/README.MD b/README.MD index bc46240..4e8033e 100644 --- a/README.MD +++ b/README.MD @@ -31,9 +31,9 @@ go build -j string jsonFile //保存大量主机，包括主机地址，SSH用户名，SSH密码，SSH端口，所需执行的cmd指令文件地址 -outTxt bool - outTxt (default false) //是否允许把结果保存到文件中，true为允许 false为默认值 - -t duration - timeLimit (default 30) //最大并发访问时间 默认为30s + outTxt (default false) //是否允许把结果保存到文件中，文件名为 ssh 连接的主机名（host 或 ip)，true为允许 false为默认值 + -t int + timeLimit (default 30) //单个 ssh 会话的最大时间，超过时间命令未执行完则超时 默认为30s -n int numLimit (default 20) //最大并发访问量 默认为20 @@ -45,86 +45,109 @@ exit ``` **hostfile 示例** ``` -10.10.15.101 -10.10.15.102 +192.168.15.101 +192.168.15.102 ``` **ipfile 示例** ``` -10.10.15.101-10.10.15.102 +192.168.15.101-192.168.15.102 ``` +**ssh.json 示例** +`` +{ + "SshHosts": [ + { + "Host": "192.168.15.101", + "Port": 22, + "Username": "admin", + "Password": "admin", + "CmdFile": "cmd1.txt.example" + }, + { + "Host": "192.168.83.40", + "Port": 22, + "Username": "root", + "Password": "root", + "CmdFile": "cmd2.txt.example" + } + ] +} +`` + ## 用法 #### cmd string & host string ``` -./multissh -cmd "show clock;exit" -hosts "10.10.15.101;10.10.15.102" -u admin -p admin +./multissh -cmd "show clock;exit" -hosts "192.168.15.101;192.168.15.102" -u admin -p admin -10.10.15.101 ssh start +192.168.15.101 ssh start sw-1#show clock 05:26:40.649 UTC Tue Jun 6 2017 sw-1#exit -10.10.15.101 ssh end +192.168.15.101 ssh end -10.10.15.102 ssh start +192.168.15.102 ssh start sw-2#show clock 05:24:38.708 UTC Tue Jun 6 2017 sw-2#exit -10.10.15.102 ssh end +192.168.15.102 ssh end ``` #### cmdfile & hostfile ``` ./multissh -cmdfile cmd.txt -hostfile host.txt -u admin -p admin -10.10.15.101 ssh start +192.168.15.101 ssh start sw-1#show clock 05:29:43.269 UTC Tue Jun 6 2017 sw-1#exit -10.10.15.101 ssh end +192.168.15.101 ssh end -10.10.15.102 ssh start +192.168.15.102 ssh start sw-2#show clock 05:27:41.332 UTC Tue Jun 6 2017 sw-2#exit -10.10.15.102 ssh end +192.168.15.102 ssh end ``` #### ipfile ``` ./multissh -cmdfile cmd.txt -ipfile ip.txt -u admin -p admin -10.10.15.101 ssh start +192.168.15.101 ssh start sw-1#show clock 05:29:43.269 UTC Tue Jun 6 2017 sw-1#exit -10.10.15.101 ssh end +192.168.15.101 ssh end -10.10.15.102 ssh start +192.168.15.102 ssh start sw-2#show clock 05:27:41.332 UTC Tue Jun 6 2017 sw-2#exit -10.10.15.102 ssh end +192.168.15.102 ssh end ``` -#### ipfile +#### ssh.json ``` -./multissh -j jsonSample.json -t 30 -n 20 -outTxt true -10.10.15.101 ssh start +./multissh -j ssh.json -t 30 -n 20 -outTxt true +192.168.15.101 ssh start sw-1#show clock 05:29:43.269 UTC Tue Jun 6 2017 sw-1#exit -10.10.15.101 ssh end +192.168.15.101 ssh end -10.10.15.102 ssh start -sw-2#show clock -05:27:41.332 UTC Tue Jun 6 2017 -sw-2#exit +192.168.83.40 ssh start +2017年 06月 09日 星期五 09:33:11 CST +2017年 06月 09日 星期五 09:33:14 CST +192.168.83.40 ssh end +``` -10.10.15.102 ssh end -``` \ No newline at end of file +#### LICENSE +Apache License 2.0 \ No newline at end of file diff --git a/cmd1.txt.example b/cmd1.txt.example new file mode 100644 index 0000000..f61b2a7 --- /dev/null +++ b/cmd1.txt.example @@ -0,0 +1,2 @@ +show clock +exit \ No newline at end of file diff --git a/cmdSample.txt b/cmd2.txt.example similarity index 60% rename from cmdSample.txt rename to cmd2.txt.example index 332fea8..12a5f15 100644 --- a/cmdSample.txt +++ b/cmd2.txt.example @@ -1,4 +1,4 @@ date -sleep 10 +sleep 3 date exit \ No newline at end of file diff --git a/host.txt.example b/host.txt.example new file mode 100644 index 0000000..024af64 --- /dev/null +++ b/host.txt.example @@ -0,0 +1,2 @@ +192.168.15.101 +192.168.15.110 \ No newline at end of file diff --git a/ip.txt.example b/ip.txt.example new file mode 100644 index 0000000..a99ff44 --- /dev/null +++ b/ip.txt.example @@ -0,0 +1 @@ +192.168.15.101-192.168.15.110 \ No newline at end of file diff --git a/jsonSample.json b/jsonSample.json deleted file mode 100644 index 0d17385..0000000 --- a/jsonSample.json +++ /dev/null @@ -1 +0,0 @@ -{"SshHosts":[{"Host":"1.1.1.1","Port":11,"Username":"xxx","Password":"xxx","CmdFile":"cmd1.txt"},{"Host":"1.1.1.1","Port":11,"Username":"yyy","Password":"yyy","CmdFile":"cmd2.txt"}]} diff --git a/main.go b/main.go index 73bfe28..0d0dd79 100644 --- a/main.go +++ b/main.go @@ -4,13 +4,17 @@ import ( "flag" "fmt" "log" + "os" "strconv" "strings" "time" - // "github.com/bitly/go-simplejson" ) +const ( + VERSION = "0.1" +) + type SSHHost struct { Host string Port int @@ -25,6 +29,7 @@ type HostJson struct { } func main() { + version := flag.Bool("v", false, "show version") hosts := flag.String("hosts", "", "host address list") cmd := flag.String("cmd", "", "cmds") username := flag.String("u", "", "username") @@ -33,11 +38,10 @@ func main() { 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", "", "Json File Path") outTxt := flag.Bool("outTxt", false, "write result into txt") - timeLimit := flag.Duration("t", 30, "max timeout") + timeLimit := flag.Int("t", 30, "max timeout") numLimit := flag.Int("n", 20, "max execute number") flag.Parse() @@ -47,6 +51,12 @@ func main() { sshHosts := []SSHHost{} var host_Struct SSHHost + timeout := time.Duration(*timeLimit) * time.Second + + if *version { + fmt.Println(VERSION) + os.Exit(0) + } if *ipFile != "" { hostList, err = GetIpList(*ipFile) @@ -77,7 +87,7 @@ func main() { if *cmd != "" { cmdList = strings.Split(*cmd, ";") } - if *cfg == "" { + if *jsonFile == "" { for _, host := range hostList { host_Struct.Host = host host_Struct.Username = *username @@ -100,30 +110,10 @@ func main() { log.Println("load cmdFile error: ", err) return } - //fmt.Println(cmdList) sshHosts[i].Cmd = cmdList } - //为什么不能用for range } - /* - else { - cfgjson, err := GetfileAll(*cfg) - if err != nil { - log.Println("load cfg error: ", err) - return - } - - js, js_err := simplejson.NewJson(cfgjson) - if js_err != nil { - log.Println("json format error: ", js_err) - return - } - - - } - */ - //fmt.Println(sshhosts) chLimit := make(chan bool, *numLimit) //控制并发访问量 chs := make([]chan string, len(sshHosts)) limitFunc := func(chLimit chan bool, ch chan string, host SSHHost) { @@ -143,7 +133,7 @@ func main() { fmt.Println(res) sshHosts[i].Result += res } - case <-time.After(*timeLimit * 1000 * 1000 * 1000): + case <-time.After(timeout): log.Println("SSH run timeout") sshHosts[i].Result += ("SSH run timeout：" + strconv.Itoa(int(*timeLimit)) + "second.") } @@ -152,7 +142,7 @@ func main() { } //gu - if !*outTxt { + if *outTxt { for i := 0; i < len(sshHosts); i++ { err = WriteIntoTxt(sshHosts[i]) if err != nil { diff --git a/ssh.json.example b/ssh.json.example new file mode 100644 index 0000000..f18a812 --- /dev/null +++ b/ssh.json.example @@ -0,0 +1,18 @@ +{ + "SshHosts": [ + { + "Host": "192.168.15.101", + "Port": 22, + "Username": "admin", + "Password": "admin", + "CmdFile": "cmd1.txt.example" + }, + { + "Host": "192.168.83.40", + "Port": 22, + "Username": "root", + "Password": "root", + "CmdFile": "cmd2.txt.example" + } + ] +} \ No newline at end of file