我正在编写一个用于在许多不同主机上并行执行命令的小工具。所有主机都可以使用相同的私钥登录。因此，我想使用SSHAgent获取身份验证进行登录。当我将其用于单个主机时，它运行良好。但是，当我在许多goroutine中使用它时，它将无法工作。我很困惑，有人可以帮助我吗？非常感谢。共享使用的代码如下：
func ExcuteRemote(uname,host,cmd string) (bool,error) {
ip,err := GetIp(host)
if err != nil {
fmt.Println(err)
return false,err
}
auths := []ssh.AuthMethod{}
if sshAgent, err := net.Dial("unix", os.Getenv("SSH_AUTH_SOCK")); err == nil {
fmt.Println("get sock")
auths = append(auths, ssh.PublicKeysCallback(agent.NewClient(sshAgent).Signers))
defer sshAgent.Close()
}
config := &ssh.ClientConfig{
User: uname,
Auth: auths,
}
fmt.Println("after config")
client, err := ssh.Dial("tcp", ip+":"+"22", config)
if err != nil {
fmt.Println("something wrong when dial")
fmt.Println(err)
return false,err
}
session, err := client.NewSession()
if err != nil {
return false,err
}
defer session.Close()
var b bytes.Buffer
session.Stdout = &b
err = session.Run(cmd)
if err != nil {
return false,err
}
fmt.Println(b.String())
return true,nil
}
并且使用的通道代码如下：
func testchannel(ch chan int, uname,host,cmd string) error{
result,err := ExcuteRemote(uname,host,cmd)
if result {
fmt.Println(host + " ok")
} else {
fmt.Println(err)
fmt.Println(host + " failed")
}
ch <- 1
return nil
}
在主要方面：
if *list != "" {
runtime.GOMAXPROCS(runtime.NumCPU())
hosts,err := ReadLine(*list)
if nil != err {
panic(err.Error())
}
chs := make([]chan int, len(hosts))
for index := 0; index < len(hosts); index++ {
chs[index] = make(chan int)
go testchannel(chs[index],uname,hosts[index],*cmd)
}
for _, ch := range(chs) {
<-ch
}
}
像这样的恐慌：
panic: unreachable
goroutine 5 [running]:
golang.org/x/crypto/ssh/agent.(*client).Sign(0xc20801e440, 0x7f053381a9f8, 0xc208094100, 0xc208121b00, 0x201, 0x240, 0xc2080a6009, 0x0, 0x0)
/usr/local/gosrc/src/golang.org/x/crypto/ssh/agent/client.go:342 +0x472
golang.org/x/crypto/ssh/agent.(*agentKeyringSigner).Sign(0xc20801e560, 0x7f053380fb38, 0xc20803c2d0, 0xc208121b00, 0x201, 0x240, 0xe, 0x0, 0x0)
/usr/local/gosrc/src/golang.org/x/crypto/ssh/agent/client.go:562 +0x71
golang.org/x/crypto/ssh.publicKeyCallback.auth(0xc20801e480, 0xc2080a4040, 0x14, 0x20, 0xc20802a954, 0x4, 0x7f053381a998, 0xc2080ca000, 0x7f053380fb38, 0xc20803c2d0, ...)
/usr/local/gosrc/src/golang.org/x/crypto/ssh/client_auth.go:210 +0x5ee
golang.org/x/crypto/ssh.(*connection).clientAuthenticate(0xc2080b8080, 0xc2080b2000, 0x0, 0x0)
/usr/local/gosrc/src/golang.org/x/crypto/ssh/client_auth.go:34 +0x4e3
golang.org/x/crypto/ssh.(*connection).clientHandshake(0xc2080b8080, 0xc20801e4c0, 0x11, 0xc2080b2000, 0x0, 0x0)
/usr/local/gosrc/src/golang.org/x/crypto/ssh/client.go:112 +0x62e
golang.org/x/crypto/ssh.NewClientConn(0x7f05338111e8, 0xc2080aa000, 0xc20801e4c0, 0x11, 0xc2080420a0, 0x0, 0x0, 0x0, 0x41be71, 0x0, ...)
/usr/local/gosrc/src/golang.org/x/crypto/ssh/client.go:74 +0x140
golang.org/x/crypto/ssh.Dial(0x785a50, 0x3, 0xc20801e4c0, 0x11, 0xc2080420a0, 0x11, 0x0, 0x0)
/usr/local/gosrc/src/golang.org/x/crypto/ssh/client.go:176 +0xf9
main.ExcuteRemote(0xc20802a954, 0x4, 0xc20802a9a0, 0xe, 0x7fffa171b7fd, 0x6, 0xc200000000, 0x0, 0x0)
/home/myname/easyssh/r.go:126 +0x83f
main.testchannel(0xc208036120, 0xc20802a954, 0x4, 0xc20802a9a0, 0xe, 0x7fffa171b7fd, 0x6, 0x0, 0x0)
/home/myname/easyssh/r.go:217 +0x8c
created by main.main
/home/myname/gocode/easyssh/r.go:262 +0x638
我自己解决了这个问题。据我了解，这种恐慌是由多个goroutine想要使用sshagent与本地服务器的连接引起的。现在，我只设置一次连接，并保持连接直到程序结束。函数main中的代码：
//get config
var config *ssh.ClientConfig
if sshAgent, err := net.Dial("unix", os.Getenv("SSH_AUTH_SOCK")); err == nil {
config = GetConfig(uname,sshAgent)
defer sshAgent.Close()
} else {
panic(err)
}
当我需要在goroutine中进行配置时，我会将配置作为参数传递给它，它运行良好！
我在goroutine中的功能如下：
func RunInChannel(ch chan int, host,cmd string,config *ssh.ClientConfig) error{
result,_ := ExcuteRemote(host,cmd,config)
if result {
fmt.Println("\n" + host + "\t\tok\n")
} else {
ret,err := ExcuteRemote(host,cmd,config)
if ret {
fmt.Println("\n" + host + "\t\tok\n")
} else {
fmt.Println(err)
fmt.Println("\n" + host + "\t\tfailed\n")
}
}
ch <- 1
return nil
}
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句