Do not assume GetManifest is called before other dockerImageSource methods

Call dockerImageSource.ping() in .makeRequest() if needed, instead of
expecting a caller to do it (which only happened in GetManifest).

This required splitting the URLs into the baseURL (dependent on .ping()
result) and the suffix (independent of it), which was a simplification
anyway.

Also rename WWWAuthenticate to wwwAuthenticate, it is a private cache
field.
This commit is contained in:
Miloslav Trmač
2016-03-30 20:42:22 +02:00
parent 14bc664b48
commit 60fbdd3988

View File

@@ -31,9 +31,9 @@ const (
dockerCfgObsolete = ".dockercfg" dockerCfgObsolete = ".dockercfg"
baseURL = "%s://%s/v2/" baseURL = "%s://%s/v2/"
tagsURL = baseURL + "%s/tags/list" tagsURL = "%s/tags/list"
manifestURL = baseURL + "%s/manifests/%s" manifestURL = "%s/manifests/%s"
blobsURL = baseURL + "%s/blobs/%s" blobsURL = "%s/blobs/%s"
) )
var ( var (
@@ -86,7 +86,7 @@ func (i *dockerImage) Manifest() (types.ImageManifest, error) {
func (i *dockerImage) getTags() ([]string, error) { func (i *dockerImage) getTags() ([]string, error) {
// FIXME? Breaking the abstraction. // FIXME? Breaking the abstraction.
url := fmt.Sprintf(tagsURL, i.src.scheme, i.src.registry, i.src.ref.RemoteName()) url := fmt.Sprintf(tagsURL, i.src.ref.RemoteName())
res, err := i.src.makeRequest("GET", url, nil) res, err := i.src.makeRequest("GET", url, nil)
if err != nil { if err != nil {
return nil, err return nil, err
@@ -188,19 +188,13 @@ type dockerImageSource struct {
registry string registry string
username string username string
password string password string
WWWAuthenticate string // Obtained by s.ping() wwwAuthenticate string // Cache of a value set by ping() if scheme is not empty
scheme string // Obtained by s.ping() scheme string // Cache of a value returned by a successful ping() if not empty
transport *http.Transport transport *http.Transport
} }
func (s *dockerImageSource) GetManifest() (manifest []byte, unverifiedCanonicalDigest string, err error) { func (s *dockerImageSource) GetManifest() (manifest []byte, unverifiedCanonicalDigest string, err error) {
pr, err := s.ping() url := fmt.Sprintf(manifestURL, s.ref.RemoteName(), s.tag)
if err != nil {
return nil, "", err
}
s.WWWAuthenticate = pr.WWWAuthenticate
s.scheme = pr.scheme
url := fmt.Sprintf(manifestURL, s.scheme, s.registry, s.ref.RemoteName(), s.tag)
// TODO(runcom) set manifest version header! schema1 for now - then schema2 etc etc and v1 // TODO(runcom) set manifest version header! schema1 for now - then schema2 etc etc and v1
// TODO(runcom) NO, switch on the resulter manifest like Docker is doing // TODO(runcom) NO, switch on the resulter manifest like Docker is doing
res, err := s.makeRequest("GET", url, nil) res, err := s.makeRequest("GET", url, nil)
@@ -219,7 +213,7 @@ func (s *dockerImageSource) GetManifest() (manifest []byte, unverifiedCanonicalD
} }
func (s *dockerImageSource) GetLayer(digest string) (io.ReadCloser, error) { func (s *dockerImageSource) GetLayer(digest string) (io.ReadCloser, error) {
url := fmt.Sprintf(blobsURL, s.scheme, s.registry, s.ref.RemoteName(), digest) url := fmt.Sprintf(blobsURL, s.ref.RemoteName(), digest)
logrus.Infof("Downloading %s", url) logrus.Infof("Downloading %s", url)
res, err := s.makeRequest("GET", url, nil) res, err := s.makeRequest("GET", url, nil)
if err != nil { if err != nil {
@@ -237,6 +231,16 @@ func (s *dockerImageSource) GetSignatures() ([][]byte, error) {
} }
func (s *dockerImageSource) makeRequest(method, url string, headers map[string]string) (*http.Response, error) { func (s *dockerImageSource) makeRequest(method, url string, headers map[string]string) (*http.Response, error) {
if s.scheme == "" {
pr, err := s.ping()
if err != nil {
return nil, err
}
s.wwwAuthenticate = pr.WWWAuthenticate
s.scheme = pr.scheme
}
url = fmt.Sprintf(baseURL, s.scheme, s.registry) + url
req, err := http.NewRequest("GET", url, nil) req, err := http.NewRequest("GET", url, nil)
if err != nil { if err != nil {
return nil, err return nil, err
@@ -245,7 +249,7 @@ func (s *dockerImageSource) makeRequest(method, url string, headers map[string]s
for n, h := range headers { for n, h := range headers {
req.Header.Add(n, h) req.Header.Add(n, h)
} }
if s.WWWAuthenticate != "" { if s.wwwAuthenticate != "" {
if err := s.setupRequestAuth(req); err != nil { if err := s.setupRequestAuth(req); err != nil {
return nil, err return nil, err
} }
@@ -262,9 +266,9 @@ func (s *dockerImageSource) makeRequest(method, url string, headers map[string]s
} }
func (s *dockerImageSource) setupRequestAuth(req *http.Request) error { func (s *dockerImageSource) setupRequestAuth(req *http.Request) error {
tokens := strings.SplitN(strings.TrimSpace(s.WWWAuthenticate), " ", 2) tokens := strings.SplitN(strings.TrimSpace(s.wwwAuthenticate), " ", 2)
if len(tokens) != 2 { if len(tokens) != 2 {
return fmt.Errorf("expected 2 tokens in WWW-Authenticate: %d, %s", len(tokens), s.WWWAuthenticate) return fmt.Errorf("expected 2 tokens in WWW-Authenticate: %d, %s", len(tokens), s.wwwAuthenticate)
} }
switch tokens[0] { switch tokens[0] {
case "Basic": case "Basic":