Update token response handling

Registry authorization token is now taken from the response body rather than
the repsonse header.

Docker-DCO-1.1-Signed-off-by: Josh Hawn <josh.hawn@docker.com> (github: jlhawn)
This commit is contained in:
Josh Hawn 2014-12-12 13:30:12 -08:00 committed by Derek McGowan
parent 24895820bd
commit e256a0e0bc

View file

@ -1,6 +1,7 @@
package registry package registry
import ( import (
"encoding/json"
"errors" "errors"
"fmt" "fmt"
"net/http" "net/http"
@ -10,6 +11,10 @@ import (
"github.com/docker/docker/utils" "github.com/docker/docker/utils"
) )
type tokenResponse struct {
Token string `json:"token"`
}
func getToken(username, password string, params map[string]string, registryEndpoint *Endpoint, client *http.Client, factory *utils.HTTPRequestFactory) (token string, err error) { func getToken(username, password string, params map[string]string, registryEndpoint *Endpoint, client *http.Client, factory *utils.HTTPRequestFactory) (token string, err error) {
realm, ok := params["realm"] realm, ok := params["realm"]
if !ok { if !ok {
@ -57,14 +62,20 @@ func getToken(username, password string, params map[string]string, registryEndpo
} }
defer resp.Body.Close() defer resp.Body.Close()
if !(resp.StatusCode == http.StatusOK || resp.StatusCode == http.StatusNoContent) { if resp.StatusCode != http.StatusOK {
return "", fmt.Errorf("token auth attempt for registry %s: %s request failed with status: %d %s", registryEndpoint, req.URL, resp.StatusCode, http.StatusText(resp.StatusCode)) return "", fmt.Errorf("token auth attempt for registry %s: %s request failed with status: %d %s", registryEndpoint, req.URL, resp.StatusCode, http.StatusText(resp.StatusCode))
} }
token = resp.Header.Get("X-Auth-Token") decoder := json.NewDecoder(resp.Body)
if token == "" {
return "", errors.New("token server did not include a token in the response header") tr := new(tokenResponse)
if err = decoder.Decode(tr); err != nil {
return "", fmt.Errorf("unable to decode token response: %s", err)
} }
return token, nil if tr.Token == "" {
return "", errors.New("authorization server did not include a token in the response")
}
return tr.Token, nil
} }