90 lines
1.8 KiB
Go
90 lines
1.8 KiB
Go
package docker
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/coreos/clair/cmd/clairctl/docker/httpclient"
|
|
"github.com/coreos/clair/cmd/clairctl/xerrors"
|
|
)
|
|
|
|
type Authentication struct {
|
|
Username, Password string
|
|
}
|
|
|
|
var User Authentication
|
|
|
|
type token struct {
|
|
Value string `json:"token"`
|
|
}
|
|
|
|
func (tok token) String() string {
|
|
return tok.Value
|
|
}
|
|
|
|
//BearerAuthParams parse Bearer Token on Www-Authenticate header
|
|
func BearerAuthParams(r *http.Response) map[string]string {
|
|
s := strings.Fields(r.Header.Get("Www-Authenticate"))
|
|
if len(s) != 2 || s[0] != "Bearer" {
|
|
return nil
|
|
}
|
|
result := map[string]string{}
|
|
|
|
for _, kv := range strings.Split(s[1], ",") {
|
|
parts := strings.Split(kv, "=")
|
|
if len(parts) != 2 {
|
|
continue
|
|
}
|
|
result[strings.Trim(parts[0], "\" ")] = strings.Trim(parts[1], "\" ")
|
|
}
|
|
return result
|
|
}
|
|
|
|
func AuthenticateResponse(dockerResponse *http.Response, request *http.Request) error {
|
|
bearerToken := BearerAuthParams(dockerResponse)
|
|
url := bearerToken["realm"] + "?service=" + bearerToken["service"]
|
|
if bearerToken["scope"] != "" {
|
|
url += "&scope=" + bearerToken["scope"]
|
|
}
|
|
req, err := http.NewRequest("GET", url, nil)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
req.SetBasicAuth(User.Username, User.Password)
|
|
|
|
response, err := httpclient.Get().Do(req)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if response.StatusCode == http.StatusUnauthorized {
|
|
return xerrors.Unauthorized
|
|
}
|
|
|
|
if response.StatusCode != http.StatusOK {
|
|
return fmt.Errorf("authentication server response: %v - %v", response.StatusCode, response.Status)
|
|
}
|
|
|
|
defer response.Body.Close()
|
|
body, err := ioutil.ReadAll(response.Body)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var tok token
|
|
err = json.Unmarshal(body, &tok)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
request.Header.Set("Authorization", "Bearer "+tok.String())
|
|
|
|
return nil
|
|
}
|