client: fix dialer after vendor update

It now sends a dial address with unix prefix. And we should bailout when timeout in commonDialer. Signed-off-by: Peng Tao <[email protected]>
kata-containers · Sep 6, 2018 · a6e27d6 · a6e27d6
1 parent cd03e0c
commit a6e27d6
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/protocols/client/client.go b/protocols/client/client.go
@@ -217,6 +217,10 @@ func agentDialer(addr *url.URL, enableYamux bool) dialer {
 }
 
 func unixDialer(sock string, timeout time.Duration) (net.Conn, error) {
+	if strings.HasPrefix(sock, "unix:") {
+		sock = strings.Trim(sock, "unix:")
+	}
+
 	dialFunc := func() (net.Conn, error) {
 		return net.DialTimeout("unix", sock, timeout)
 	}
@@ -285,6 +289,9 @@ func commonDialer(timeout time.Duration, dialFunc func() (net.Conn, error), time
 		if !ok {
 			return nil, timeoutErrMsg
 		}
+	case <-t.C:
+		cancel <- true
+		return nil, timeoutErrMsg
 	}
 
 	return conn, nil