Fix registry verification failed

This commit is contained in:
hongming
2022-02-17 14:19:38 +08:00
parent 794129d056
commit fecc331697
2 changed files with 90 additions and 8 deletions

View File

@@ -2,6 +2,7 @@ package v2
import (
"context"
"crypto/tls"
"encoding/json"
"fmt"
"net/http"
@@ -93,9 +94,16 @@ func (s *secretAuthenticator) AuthRegistry(reg string) (bool, error) {
}
options := make([]name.Option, 0)
if url.Scheme == "http" || s.insecure {
if url.Scheme == "http" {
// allows image references to be fetched without TLS
// transport.NewWithContext will auto-select the right scheme
options = append(options, name.Insecure)
}
tr := http.DefaultTransport.(*http.Transport).Clone()
// skip tls verify
if s.insecure {
tr.TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
}
registry, err := name.NewRegistry(url.Host, options...)
if err != nil {
@@ -103,7 +111,7 @@ func (s *secretAuthenticator) AuthRegistry(reg string) (bool, error) {
}
ctx := context.TODO()
_, err = transport.NewWithContext(ctx, registry, s, http.DefaultTransport, []string{})
_, err = transport.NewWithContext(ctx, registry, s, tr, []string{})
if err != nil {
return false, err
}
@@ -113,11 +121,24 @@ func (s *secretAuthenticator) AuthRegistry(reg string) (bool, error) {
func (s *secretAuthenticator) Options() []Option {
options := make([]Option, 0)
options = append(options, WithAuth(s))
if s.insecure {
if s.registryScheme() == "http" {
options = append(options, Insecure)
}
if s.insecure {
tr := http.DefaultTransport.(*http.Transport).Clone()
tr.TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
options = append(options, WithTransport(tr))
}
return options
}
func (s *secretAuthenticator) registryScheme() string {
for registry := range s.auths {
u, err := url.Parse(registry)
if err == nil {
return u.Scheme
}
}
return "https"
}

View File

@@ -1,10 +1,16 @@
package v2
import (
"fmt"
"testing"
"encoding/base64"
"fmt"
"net/http"
"net/http/httptest"
"testing"
"time"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/onsi/gomega/gexec"
"github.com/google/go-cmp/cmp"
"github.com/google/go-containerregistry/pkg/authn"
@@ -109,3 +115,58 @@ func TestAuthn(t *testing.T) {
})
}
}
var (
registryServer *httptest.Server
tlsRegistryServer *httptest.Server
)
func TestRegistry(t *testing.T) {
RegisterFailHandler(Fail)
RunSpecs(t, "Registry Test Suite")
}
var _ = BeforeSuite(func(done Done) {
// anonymous registry
fakeHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusOK)
w.Write([]byte("ok"))
})
tlsRegistryServer = httptest.NewTLSServer(fakeHandler)
registryServer = httptest.NewServer(fakeHandler)
close(done)
}, 30)
var _ = AfterSuite(func() {
By("tearing down the test environment")
gexec.KillAndWait(5 * time.Second)
registryServer.Close()
tlsRegistryServer.Close()
})
var _ = Describe("Registry", func() {
Context("Registry", func() {
It("skip TLS certification checks", func() {
secret := buildSecret(tlsRegistryServer.URL, "", "", true)
secretAuthenticator, err := NewSecretAuthenticator(secret)
Expect(err).Should(BeNil())
_, err = secretAuthenticator.Auth()
Expect(err).Should(BeNil())
})
It("self-signed certs are not trusted", func() {
secret := buildSecret(tlsRegistryServer.URL, "", "", false)
secretAuthenticator, err := NewSecretAuthenticator(secret)
Expect(err).Should(BeNil())
_, err = secretAuthenticator.Auth()
Expect(err).ShouldNot(BeNil())
})
It("insecure registry", func() {
// Loopback addr always be insecure, http scheme will be used.
secret := buildSecret(registryServer.URL, "", "", false)
secretAuthenticator, err := NewSecretAuthenticator(secret)
Expect(err).Should(BeNil())
_, err = secretAuthenticator.Auth()
Expect(err).Should(BeNil())
})
})
})