generic: ssh timeout

Signed-off-by: Evan Hazlett <ejhazlett@gmail.com>
This commit is contained in:
Evan Hazlett 2015-04-30 17:44:34 -04:00
parent 2d525d0abd
commit afb9cb39e2
1 changed files with 6 additions and 1 deletions

View File

@ -5,6 +5,7 @@ import (
"net"
"os"
"path/filepath"
"time"
"github.com/codegangsta/cli"
"github.com/docker/machine/drivers"
@ -29,6 +30,10 @@ type Driver struct {
storePath string
}
const (
defaultTimeout = 5 * time.Second
)
func init() {
drivers.Register("generic", &drivers.RegisteredDriver{
New: NewDriver,
@ -165,7 +170,7 @@ func (d *Driver) GetIP() (string, error) {
func (d *Driver) GetState() (state.State, error) {
addr := fmt.Sprintf("%s:%d", d.IPAddress, d.SSHPort)
_, err := net.Dial("tcp", addr)
_, err := net.DialTimeout("tcp", addr, defaultTimeout)
var st state.State
if err != nil {
st = state.Stopped