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 ( import (
"os/user" "os/user"
"time"
"github.com/AlecAivazis/survey/v2" "github.com/AlecAivazis/survey/v2"
"github.com/kevinburke/ssh_config" "github.com/kevinburke/ssh_config"
@ -71,7 +72,7 @@ func New(domainName, sshAuth, username, port string) (*simplessh.Client, error)
if sshAuth == "identity-file" { if sshAuth == "identity-file" {
var err error 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 { if err != nil {
return client, err return client, err
} }
@ -85,7 +86,7 @@ func New(domainName, sshAuth, username, port string) (*simplessh.Client, error)
} }
var err 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 { if err != nil {
return client, err return client, err
} }