codegod100/abra
0
0
Fork 0
forked from toolshed/abra
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: add ssh conn time outs

Browse Source
This commit is contained in:
decentral1se
2021-10-26 00:33:18 +02:00
parent 9a513a0700
commit aec11bda28
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pkg/ssh/ssh.go
View File

@ -2,6 +2,7 @@ package ssh
import (
"os/user"
"time"
"github.com/AlecAivazis/survey/v2"
"github.com/kevinburke/ssh_config"
@ -71,7 +72,7 @@ func New(domainName, sshAuth, username, port string) (*simplessh.Client, error)
if sshAuth == "identity-file" {
var err error
client, err = simplessh.ConnectWithAgent(hostConfig.Host, hostConfig.User)
client, err = simplessh.ConnectWithAgentTimeout(hostConfig.Host, hostConfig.User, 5*time.Second)
if err != nil {
return client, err
}
@ -85,7 +86,7 @@ func New(domainName, sshAuth, username, port string) (*simplessh.Client, error)
}
var err error
client, err = simplessh.ConnectWithPassword(hostConfig.Host, hostConfig.User, password)
client, err = simplessh.ConnectWithPasswordTimeout(hostConfig.Host, hostConfig.User, password, 5*time.Second)
if err != nil {
return client, err
}