庆祝破元老，赠送坚毅人士一个点子 - 全球主机交流论坛

package main
import (
"fmt"
"golang.org/x/crypto/ssh"
//"golang.org/x/crypto/ssh/terminal"
"io/ioutil"
"log"
"os"
"os/user"
"time"
)
func main() {
//for test ,get rsa key file
key, err := getKeyFile()
if err != nil {
panic(err)
} else {
println("keyfile is %d ", key)
}
//get ssh session
session, err := connect("root", "密码", nil, "198.IP", 22)
if err != nil {
log.Fatal(err)
}
defer session.Close()
//defind a cmd to exec
cmd := "ifconfig "
RunOnce(session, cmd)
}
//利用本地的shell 直接连接到远程的shell上去，按tab无法不全，上下键盘无法使用。
func Interact(session *ssh.Session, cmd string) {
var err error
session.Stdout = os.Stdout
session.Stderr = os.Stderr
ce := func(err error, msg string) {
if err != nil {
log.Fatalf("%s error: %v", msg, err)
}
}
session.Stdin = os.Stdin //输入
modes := ssh.TerminalModes{
ssh.ECHO: 0,
ssh.TTY_OP_ISPEED: 14400,
ssh.TTY_OP_OSPEED: 14400,
}
// Request pseudo terminal
if err := session.RequestPty("xterm", 25, 80, modes); err != nil {
log.Fatal(err)
}
//session.Run("ls /; ls /abc")
err = session.Shell()
ce(err, "start shell")
err = session.Wait()
ce(err, "return")
//session.Run("wget -N --no-check-certificate https://raw.githubusercontent.com/91yun/91yuntest/master/test_91yun.sh && bash test_91yun.sh s")
}
//执行一次Shell脚本，获得所有记录后返回
func RunOnce(session *ssh.Session, cmd string) {
session.Stdout = os.Stdout
session.Stderr = os.Stderr
session.Run(cmd)
//print(str)
//print(str2)
//session.Run("wget -N --no-check-certificate https://raw.githubusercontent.com/91yun/91yuntest/master/test_91yun.sh && bash test_91yun.sh s")
}
//---------------------Private Func for SSH connect-------------------------------------
//连接服务端，如果是走rsa file的花，则keyfile存ssh.Signer值，否则nil
func connect(user string, password string, key ssh.Signer, host string, port int) (*ssh.Session, error) {
var (
auth []ssh.AuthMethod
addr string
clientConfig *ssh.ClientConfig
client *ssh.Client
session *ssh.Session
err error
)
// get auth method
auth = make([]ssh.AuthMethod, 0)
if key == nil {
auth = append(auth, ssh.Password(password))
} else {
auth = append(auth, ssh.PublicKeys(key))
}
clientConfig = &ssh.ClientConfig{
User: user,
Auth: auth,
Timeout: 30 * time.Second,
}
// connet to ssh
addr = fmt.Sprintf("%s:%d", host, port)
if client, err = ssh.Dial("tcp", addr, clientConfig); err != nil {
return nil, err
}
// create session
if session, err = client.NewSession(); err != nil {
return nil, err
}
return session, nil
}
//Get RSA private key File ,and then the app can use RSA key for connect
func getKeyFile() (key ssh.Signer, err error) {
usr, _ := user.Current()
file := usr.HomeDir + "/.ssh/id_rsa"
buf, err := ioutil.ReadFile(file)
if err != nil {
return
}
key, err = ssh.ParsePrivateKey(buf)
if err != nil {
return
}
return
}

复制代码