0
Fork 0
mirror of https://github.com/willnorris/imageproxy.git synced 2024-12-16 21:56:43 -05:00

Switch to Hostname() for checking whether a host is allowed or not (#238)

Using .Host allows you to get around an allowHosts or denyHosts entry by adding a port
This commit is contained in:
Blake Stoddard 2020-06-21 00:44:01 -04:00 committed by GitHub
parent f91e9cb508
commit 0da684b81e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 3 deletions

View file

@ -319,14 +319,14 @@ func contentTypeMatches(patterns []string, contentType string) bool {
// hostMatches returns whether the host in u matches one of hosts. // hostMatches returns whether the host in u matches one of hosts.
func hostMatches(hosts []string, u *url.URL) bool { func hostMatches(hosts []string, u *url.URL) bool {
for _, host := range hosts { for _, host := range hosts {
if u.Host == host { if u.Hostname() == host {
return true return true
} }
if strings.HasPrefix(host, "*.") && strings.HasSuffix(u.Host, host[2:]) { if strings.HasPrefix(host, "*.") && strings.HasSuffix(u.Hostname(), host[2:]) {
return true return true
} }
// Checks whether the host in u is an IP // Checks whether the host in u is an IP
if ip := net.ParseIP(u.Host); ip != nil { if ip := net.ParseIP(u.Hostname()); ip != nil {
// Checks whether our current host is a CIDR // Checks whether our current host is a CIDR
if _, ipnet, err := net.ParseCIDR(host); err == nil { if _, ipnet, err := net.ParseCIDR(host); err == nil {
// Checks if our host contains the IP in u // Checks if our host contains the IP in u

View file

@ -170,9 +170,11 @@ func TestAllowed(t *testing.T) {
// deny requests that match denyHosts, even if signature is valid or also matches allowHosts // deny requests that match denyHosts, even if signature is valid or also matches allowHosts
{"http://test/image", emptyOptions, nil, []string{"test"}, nil, nil, nil, false}, {"http://test/image", emptyOptions, nil, []string{"test"}, nil, nil, nil, false},
{"http://test:3000/image", emptyOptions, nil, []string{"test"}, nil, nil, nil, false},
{"http://test/image", emptyOptions, []string{"test"}, []string{"test"}, nil, nil, nil, false}, {"http://test/image", emptyOptions, []string{"test"}, []string{"test"}, nil, nil, nil, false},
{"http://test/image", Options{Signature: "NDx5zZHx7QfE8E-ijowRreq6CJJBZjwiRfOVk_mkfQQ="}, nil, []string{"test"}, nil, key, nil, false}, {"http://test/image", Options{Signature: "NDx5zZHx7QfE8E-ijowRreq6CJJBZjwiRfOVk_mkfQQ="}, nil, []string{"test"}, nil, key, nil, false},
{"http://127.0.0.1/image", emptyOptions, nil, []string{"127.0.0.0/8"}, nil, nil, nil, false}, {"http://127.0.0.1/image", emptyOptions, nil, []string{"127.0.0.0/8"}, nil, nil, nil, false},
{"http://127.0.0.1:3000/image", emptyOptions, nil, []string{"127.0.0.0/8"}, nil, nil, nil, false},
} }
for _, tt := range tests { for _, tt := range tests {