diff --git a/sshutil/ssh.go b/sshutil/ssh.go index b9af2ec..7b87c1e 100644 --- a/sshutil/ssh.go +++ b/sshutil/ssh.go @@ -99,7 +99,9 @@ func NewSSHSessionWithDelayedTimeout(ctx context.Context, timeout time.Duration, go func() { select { case <-time.After(timeout): - preTimeout() + if preTimeout != nil { + preTimeout() + } timedOut = true _ = sc.Close() case <-done: