mirror of
https://github.com/goharbor/harbor.git
synced 2025-01-01 13:37:47 +01:00
114 lines
2.8 KiB
Go
114 lines
2.8 KiB
Go
/*
|
|
Copyright (c) 2016 VMware, Inc. All Rights Reserved.
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package registry
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/vmware/harbor/utils/log"
|
|
"github.com/vmware/harbor/utils/registry/auth"
|
|
)
|
|
|
|
// NewHTTPClientAuthHandlersEmbeded return a http.Client which will authorize the request
|
|
// and send it again when encounters a 401 error
|
|
func NewClientStandardAuthHandlersEmbeded(credential *auth.Credential) *http.Client {
|
|
handlers := []auth.Handler{}
|
|
|
|
tokenHandler := auth.NewStandardTokenHandler(credential)
|
|
|
|
handlers = append(handlers, tokenHandler)
|
|
|
|
transport := NewAuthTransport(http.DefaultTransport, handlers)
|
|
|
|
return &http.Client{
|
|
Transport: transport,
|
|
}
|
|
}
|
|
|
|
func NewClientSessionAuthHandlersEmbeded(sessionID string) *http.Client {
|
|
handlers := []auth.Handler{}
|
|
|
|
tokenHandler := auth.NewSessionTokenHandler(sessionID)
|
|
|
|
handlers = append(handlers, tokenHandler)
|
|
|
|
transport := NewAuthTransport(http.DefaultTransport, handlers)
|
|
|
|
return &http.Client{
|
|
Transport: transport,
|
|
}
|
|
}
|
|
|
|
type authTransport struct {
|
|
transport http.RoundTripper
|
|
handlers []auth.Handler
|
|
}
|
|
|
|
// NewAuthTransport wraps the AuthHandlers to be http.RounTripper
|
|
func NewAuthTransport(transport http.RoundTripper, handlers []auth.Handler) http.RoundTripper {
|
|
return &authTransport{
|
|
transport: transport,
|
|
handlers: handlers,
|
|
}
|
|
}
|
|
|
|
// RoundTrip ...
|
|
func (a *authTransport) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
originResp, originErr := a.transport.RoundTrip(req)
|
|
|
|
if originErr != nil {
|
|
return originResp, originErr
|
|
}
|
|
|
|
log.Debugf("%d | %s %s", originResp.StatusCode, req.Method, req.URL)
|
|
|
|
if originResp.StatusCode != http.StatusUnauthorized {
|
|
return originResp, nil
|
|
}
|
|
|
|
challenges := auth.ParseChallengeFromResponse(originResp)
|
|
|
|
reqChanged := false
|
|
for _, challenge := range challenges {
|
|
|
|
scheme := challenge.Scheme
|
|
|
|
for _, handler := range a.handlers {
|
|
if scheme != handler.Schema() {
|
|
log.Debugf("scheme not match: %s %s, skip", scheme, handler.Schema())
|
|
continue
|
|
}
|
|
|
|
if err := handler.AuthorizeRequest(req, challenge.Parameters); err != nil {
|
|
return nil, err
|
|
}
|
|
reqChanged = true
|
|
}
|
|
}
|
|
|
|
if !reqChanged {
|
|
log.Warning("no handler match scheme")
|
|
return originResp, nil
|
|
}
|
|
|
|
resp, err := a.transport.RoundTrip(req)
|
|
if err == nil {
|
|
log.Debugf("%d | %s %s", resp.StatusCode, req.Method, req.URL)
|
|
}
|
|
|
|
return resp, err
|
|
}
|