
This is step one for cross-region ECR support and has no visible effects yet. I'm not crazy about the name LazyProvide. Perhaps the interface method could remain like that and the package method of the same name could become LateBind(). I still don't understand why the credential provider has a DockerConfigEntry that has the same fields but is distinct from docker.AuthConfiguration. I had to write a converter now that we do that in more than one place. In step two, I'll add another intermediate, lazy provider for each AWS region, whose empty LazyAuthConfiguration will have a refresh time of months or years. Behind the scenes, it'll use an actual ecrProvider with the usual ~12 hour credentials, that will get created (and later refreshed) only when kubelet is attempting to pull an image. If we simply turned ecrProvider directly into a lazy provider, we would bypass all the caching and get new credentials for each image pulled.
120 lines
3.4 KiB
Go
120 lines
3.4 KiB
Go
/*
|
|
Copyright 2014 The Kubernetes Authors 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 credentialprovider
|
|
|
|
import (
|
|
"os"
|
|
"reflect"
|
|
"sync"
|
|
"time"
|
|
|
|
docker "github.com/fsouza/go-dockerclient"
|
|
"github.com/golang/glog"
|
|
)
|
|
|
|
// DockerConfigProvider is the interface that registered extensions implement
|
|
// to materialize 'dockercfg' credentials.
|
|
type DockerConfigProvider interface {
|
|
Enabled() bool
|
|
Provide() DockerConfig
|
|
// LazyProvide() gets called after URL matches have been performed, so the
|
|
// location used as the key in DockerConfig would be redundant.
|
|
LazyProvide() *DockerConfigEntry
|
|
}
|
|
|
|
func LazyProvide(creds LazyAuthConfiguration) docker.AuthConfiguration {
|
|
if creds.Provider != nil {
|
|
entry := *creds.Provider.LazyProvide()
|
|
return DockerConfigEntryToLazyAuthConfiguration(entry).AuthConfiguration
|
|
} else {
|
|
return creds.AuthConfiguration
|
|
}
|
|
|
|
}
|
|
|
|
// A DockerConfigProvider that simply reads the .dockercfg file
|
|
type defaultDockerConfigProvider struct{}
|
|
|
|
// init registers our default provider, which simply reads the .dockercfg file.
|
|
func init() {
|
|
RegisterCredentialProvider(".dockercfg",
|
|
&CachingDockerConfigProvider{
|
|
Provider: &defaultDockerConfigProvider{},
|
|
Lifetime: 5 * time.Minute,
|
|
})
|
|
}
|
|
|
|
// CachingDockerConfigProvider implements DockerConfigProvider by composing
|
|
// with another DockerConfigProvider and caching the DockerConfig it provides
|
|
// for a pre-specified lifetime.
|
|
type CachingDockerConfigProvider struct {
|
|
Provider DockerConfigProvider
|
|
Lifetime time.Duration
|
|
|
|
// cache fields
|
|
cacheDockerConfig DockerConfig
|
|
expiration time.Time
|
|
mu sync.Mutex
|
|
}
|
|
|
|
// Enabled implements dockerConfigProvider
|
|
func (d *defaultDockerConfigProvider) Enabled() bool {
|
|
return true
|
|
}
|
|
|
|
// Provide implements dockerConfigProvider
|
|
func (d *defaultDockerConfigProvider) Provide() DockerConfig {
|
|
// Read the standard Docker credentials from .dockercfg
|
|
if cfg, err := ReadDockerConfigFile(); err == nil {
|
|
return cfg
|
|
} else if !os.IsNotExist(err) {
|
|
glog.V(4).Infof("Unable to parse Docker config file: %v", err)
|
|
}
|
|
return DockerConfig{}
|
|
}
|
|
|
|
// LazyProvide implements dockerConfigProvider. Should never be called.
|
|
func (d *defaultDockerConfigProvider) LazyProvide() *DockerConfigEntry {
|
|
return nil
|
|
}
|
|
|
|
// Enabled implements dockerConfigProvider
|
|
func (d *CachingDockerConfigProvider) Enabled() bool {
|
|
return d.Provider.Enabled()
|
|
}
|
|
|
|
// LazyProvide implements dockerConfigProvider. Should never be called.
|
|
func (d *CachingDockerConfigProvider) LazyProvide() *DockerConfigEntry {
|
|
return nil
|
|
}
|
|
|
|
// Provide implements dockerConfigProvider
|
|
func (d *CachingDockerConfigProvider) Provide() DockerConfig {
|
|
d.mu.Lock()
|
|
defer d.mu.Unlock()
|
|
|
|
// If the cache hasn't expired, return our cache
|
|
if time.Now().Before(d.expiration) {
|
|
return d.cacheDockerConfig
|
|
}
|
|
|
|
glog.V(2).Infof("Refreshing cache for provider: %v", reflect.TypeOf(d.Provider).String())
|
|
d.cacheDockerConfig = d.Provider.Provide()
|
|
d.expiration = time.Now().Add(d.Lifetime)
|
|
return d.cacheDockerConfig
|
|
}
|