improve trusted location detection

Docker-DCO-1.1-Signed-off-by: Victor Vieux <vieux@docker.com> (github: vieux)
This commit is contained in:
Victor Vieux 2014-06-07 21:17:56 +00:00
parent df9d5752ec
commit 5cef006c5a
2 changed files with 2 additions and 2 deletions

View file

@ -766,7 +766,7 @@ func trustedLocation(req *http.Request) bool {
} }
for _, trusted := range trusteds { for _, trusted := range trusteds {
if strings.HasSuffix(hostname, trusted) { if hostname == trusted || strings.HasSuffix(hostname, "."+trusted) {
return true return true
} }
} }

View file

@ -235,7 +235,7 @@ func TestValidRepositoryName(t *testing.T) {
} }
func TestTrustedLocation(t *testing.T) { func TestTrustedLocation(t *testing.T) {
for _, url := range []string{"http://example.com", "https://example.com:7777", "http://docker.io", "http://test.docker.io"} { for _, url := range []string{"http://example.com", "https://example.com:7777", "http://docker.io", "http://test.docker.io", "https://fakedocker.com"} {
req, _ := http.NewRequest("GET", url, nil) req, _ := http.NewRequest("GET", url, nil)
if trustedLocation(req) == true { if trustedLocation(req) == true {
t.Fatalf("'%s' shouldn't be detected as a trusted location", url) t.Fatalf("'%s' shouldn't be detected as a trusted location", url)