add parameter to allow ignore cert
Signed-off-by: shaowenchen <mail@chenshaowen.com>
This commit is contained in:
@@ -23,7 +23,7 @@ import (
|
||||
func TestDigestFromDockerHub(t *testing.T) {
|
||||
|
||||
testImage := Image{Domain: "docker.io", Path: "library/alpine", Tag: "latest"}
|
||||
r, err := CreateRegistryClient("", "", "docker.io", true)
|
||||
r, err := CreateRegistryClient("", "", "docker.io", true, false)
|
||||
if err != nil {
|
||||
t.Fatalf("Could not get client: %s", err)
|
||||
}
|
||||
|
||||
@@ -21,6 +21,8 @@ import (
|
||||
"encoding/base64"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/docker/docker/api/types"
|
||||
"github.com/docker/docker/client"
|
||||
"github.com/emicklei/go-restful"
|
||||
@@ -28,7 +30,6 @@ import (
|
||||
"k8s.io/client-go/informers"
|
||||
"k8s.io/klog"
|
||||
"kubesphere.io/kubesphere/pkg/api"
|
||||
"strings"
|
||||
)
|
||||
|
||||
const (
|
||||
@@ -55,7 +56,7 @@ type DockerConfigEntry struct {
|
||||
|
||||
type RegistryGetter interface {
|
||||
VerifyRegistryCredential(credential api.RegistryCredential) error
|
||||
GetEntry(namespace, secretName, imageName string) (ImageDetails, error)
|
||||
GetEntry(namespace, secretName, imageName string, insecure bool) (ImageDetails, error)
|
||||
}
|
||||
|
||||
type registryGetter struct {
|
||||
@@ -96,8 +97,8 @@ func (c *registryGetter) VerifyRegistryCredential(credential api.RegistryCredent
|
||||
}
|
||||
}
|
||||
|
||||
func (c *registryGetter) GetEntry(namespace, secretName, imageName string) (ImageDetails, error) {
|
||||
imageDetails, err := c.getEntryBySecret(namespace, secretName, imageName)
|
||||
func (c *registryGetter) GetEntry(namespace, secretName, imageName string, insecure bool) (ImageDetails, error) {
|
||||
imageDetails, err := c.getEntryBySecret(namespace, secretName, imageName, insecure)
|
||||
if imageDetails.Status == StatusFailed {
|
||||
imageDetails.Message = err.Error()
|
||||
}
|
||||
@@ -105,7 +106,7 @@ func (c *registryGetter) GetEntry(namespace, secretName, imageName string) (Imag
|
||||
return imageDetails, err
|
||||
}
|
||||
|
||||
func (c *registryGetter) getEntryBySecret(namespace, secretName, imageName string) (ImageDetails, error) {
|
||||
func (c *registryGetter) getEntryBySecret(namespace, secretName, imageName string, insecure bool) (ImageDetails, error) {
|
||||
failedImageDetails := ImageDetails{
|
||||
Status: StatusFailed,
|
||||
Message: "",
|
||||
@@ -152,7 +153,7 @@ func (c *registryGetter) getEntryBySecret(namespace, secretName, imageName strin
|
||||
useSSL := checkSSl(config.ServerAddress)
|
||||
|
||||
// Create the registry client.
|
||||
r, err := CreateRegistryClient(config.Username, config.Password, image.Domain, useSSL)
|
||||
r, err := CreateRegistryClient(config.Username, config.Password, image.Domain, useSSL, insecure)
|
||||
if err != nil {
|
||||
return failedImageDetails, err
|
||||
}
|
||||
|
||||
@@ -18,17 +18,19 @@ package registries
|
||||
|
||||
import (
|
||||
"compress/gzip"
|
||||
"crypto/tls"
|
||||
"errors"
|
||||
"fmt"
|
||||
"github.com/docker/docker/api/types"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
log "k8s.io/klog"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"regexp"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/docker/docker/api/types"
|
||||
log "k8s.io/klog"
|
||||
)
|
||||
|
||||
const (
|
||||
@@ -63,10 +65,11 @@ type Registry struct {
|
||||
|
||||
// Opt holds the options for a new registry.
|
||||
type RegistryOpt struct {
|
||||
Domain string
|
||||
Timeout time.Duration
|
||||
Headers map[string]string
|
||||
UseSSL bool
|
||||
Domain string
|
||||
Timeout time.Duration
|
||||
Headers map[string]string
|
||||
UseSSL bool
|
||||
Insecure bool
|
||||
}
|
||||
|
||||
type authToken struct {
|
||||
@@ -80,7 +83,7 @@ type authService struct {
|
||||
Scope []string
|
||||
}
|
||||
|
||||
func CreateRegistryClient(username, password, domain string, useSSL bool) (*Registry, error) {
|
||||
func CreateRegistryClient(username, password, domain string, useSSL bool, insecure bool) (*Registry, error) {
|
||||
authDomain := domain
|
||||
auth, err := GetAuthConfig(username, password, authDomain)
|
||||
if err != nil {
|
||||
@@ -90,8 +93,9 @@ func CreateRegistryClient(username, password, domain string, useSSL bool) (*Regi
|
||||
|
||||
// Create the registry client.
|
||||
return New(auth, RegistryOpt{
|
||||
Domain: domain,
|
||||
UseSSL: useSSL,
|
||||
Domain: domain,
|
||||
UseSSL: useSSL,
|
||||
Insecure: insecure,
|
||||
})
|
||||
}
|
||||
|
||||
@@ -135,11 +139,16 @@ func newFromTransport(auth types.AuthConfig, opt RegistryOpt) (*Registry, error)
|
||||
}
|
||||
|
||||
registryURL, _ := url.Parse(registryUrl)
|
||||
|
||||
tr := &http.Transport{
|
||||
TLSClientConfig: &tls.Config{InsecureSkipVerify: opt.Insecure},
|
||||
}
|
||||
registry := &Registry{
|
||||
URL: registryURL.String(),
|
||||
Domain: registryURL.Host,
|
||||
Client: &http.Client{
|
||||
Timeout: DefaultTimeout,
|
||||
Timeout: DefaultTimeout,
|
||||
Transport: tr,
|
||||
},
|
||||
Username: auth.Username,
|
||||
Password: auth.Password,
|
||||
|
||||
@@ -41,7 +41,7 @@ func TestCreateRegistryClient(t *testing.T) {
|
||||
}
|
||||
|
||||
for _, testImage := range testImages {
|
||||
reg, err := CreateRegistryClient(testImage.Username, testImage.Password, testImage.Domain, testImage.UseSSL)
|
||||
reg, err := CreateRegistryClient(testImage.Username, testImage.Password, testImage.Domain, testImage.UseSSL, false)
|
||||
if err != nil {
|
||||
t.Fatalf("Get err %s", err)
|
||||
}
|
||||
@@ -57,7 +57,7 @@ func TestCreateRegistryClient(t *testing.T) {
|
||||
}
|
||||
|
||||
testImage := Image{Domain: DockerHub, Path: "library/alpine", Tag: "latest"}
|
||||
r, err := CreateRegistryClient("", "", DockerHub, true)
|
||||
r, err := CreateRegistryClient("", "", DockerHub, true, false)
|
||||
if err != nil {
|
||||
t.Fatalf("Could not get client: %s", err)
|
||||
}
|
||||
|
||||
@@ -48,7 +48,7 @@ func (asm authServiceMock) equalTo(v *authService) bool {
|
||||
|
||||
func TestToken(t *testing.T) {
|
||||
testImage := Image{Domain: "docker.io", Path: "library/alpine", Tag: "latest"}
|
||||
r, err := CreateRegistryClient("", "", "docker.io", true)
|
||||
r, err := CreateRegistryClient("", "", "docker.io", true, false)
|
||||
if err != nil {
|
||||
t.Fatalf("Could not get registry client: %s", err)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user