From 3a21f339f1637aded4121715555c0e5fc7269f0e Mon Sep 17 00:00:00 2001
From: Derek <crq@kernel.org>
Date: Thu, 22 May 2014 23:58:56 -0700
Subject: [PATCH] Use Timeout Conn wrapper to set read deadline for downloading
 layer

Docker-DCO-1.1-Signed-off-by: Derek <crq@kernel.org> (github: crquan)
---
 docs/registry.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/docs/registry.go b/docs/registry.go
index 2e3e7e03a..3d0a3ed2d 100644
--- a/docs/registry.go
+++ b/docs/registry.go
@@ -726,7 +726,17 @@ type Registry struct {
 }
 
 func NewRegistry(authConfig *AuthConfig, factory *utils.HTTPRequestFactory, indexEndpoint string) (r *Registry, err error) {
+	httpDial := func(proto string, addr string) (net.Conn, error) {
+		conn, err := net.Dial(proto, addr)
+		if err != nil {
+			return nil, err
+		}
+		conn = utils.NewTimeoutConn(conn, time.Duration(1)*time.Minute)
+		return conn, nil
+	}
+
 	httpTransport := &http.Transport{
+		Dial:              httpDial,
 		DisableKeepAlives: true,
 		Proxy:             http.ProxyFromEnvironment,
 	}
@@ -738,6 +748,7 @@ func NewRegistry(authConfig *AuthConfig, factory *utils.HTTPRequestFactory, inde
 		},
 		indexEndpoint: indexEndpoint,
 	}
+
 	r.client.Jar, err = cookiejar.New(nil)
 	if err != nil {
 		return nil, err