caddyhttp: Fix host matching when host has a port

2 files changed, 25 insertions, 13 deletions

diff --git a/modules/caddyhttp/matchers.go b/modules/caddyhttp/matchers.go
index 86a2c69..2395e13 100644
--- a/modules/caddyhttp/matchers.go
+++ b/modules/caddyhttp/matchers.go
@@ -113,11 +113,17 @@ func init() {
 
 // Match returns true if r matches m.
 func (m MatchHost) Match(r *http.Request) bool {
+	reqHost, _, err := net.SplitHostPort(r.Host)
+	if err != nil {
+		// OK; probably didn't have a port
+		reqHost = r.Host
+	}
+
 outer:
 	for _, host := range m {
 		if strings.Contains(host, "*") {
 			patternParts := strings.Split(host, ".")
-			incomingParts := strings.Split(r.Host, ".")
+			incomingParts := strings.Split(reqHost, ".")
 			if len(patternParts) != len(incomingParts) {
 				continue
 			}
@@ -130,10 +136,11 @@ outer:
 				}
 			}
 			return true
-		} else if strings.EqualFold(r.Host, host) {
+		} else if strings.EqualFold(reqHost, host) {
 			return true
 		}
 	}
+
 	return false
 }
 
@@ -469,17 +476,17 @@ var wordRE = regexp.MustCompile(`\w+`)
 
 // Interface guards
 var (
-	_ RequestMatcher     = (*MatchHost)(nil)
-	_ RequestMatcher     = (*MatchPath)(nil)
-	_ RequestMatcher     = (*MatchPathRE)(nil)
-	_ RequestMatcher     = (*MatchMethod)(nil)
-	_ RequestMatcher     = (*MatchQuery)(nil)
-	_ RequestMatcher     = (*MatchHeader)(nil)
-	_ RequestMatcher     = (*MatchHeaderRE)(nil)
-	_ RequestMatcher     = (*MatchProtocol)(nil)
-	_ RequestMatcher     = (*MatchRemoteIP)(nil)
+	_ RequestMatcher    = (*MatchHost)(nil)
+	_ RequestMatcher    = (*MatchPath)(nil)
+	_ RequestMatcher    = (*MatchPathRE)(nil)
+	_ RequestMatcher    = (*MatchMethod)(nil)
+	_ RequestMatcher    = (*MatchQuery)(nil)
+	_ RequestMatcher    = (*MatchHeader)(nil)
+	_ RequestMatcher    = (*MatchHeaderRE)(nil)
+	_ RequestMatcher    = (*MatchProtocol)(nil)
+	_ RequestMatcher    = (*MatchRemoteIP)(nil)
 	_ caddy.Provisioner = (*MatchRemoteIP)(nil)
-	_ RequestMatcher     = (*MatchNegate)(nil)
+	_ RequestMatcher    = (*MatchNegate)(nil)
 	_ caddy.Provisioner = (*MatchNegate)(nil)
-	_ RequestMatcher     = (*MatchStarlarkExpr)(nil)
+	_ RequestMatcher    = (*MatchStarlarkExpr)(nil)
 )
diff --git a/modules/caddyhttp/matchers_test.go b/modules/caddyhttp/matchers_test.go
index 08ab156..597d217 100644
--- a/modules/caddyhttp/matchers_test.go
+++ b/modules/caddyhttp/matchers_test.go
@@ -87,6 +87,11 @@ func TestHostMatcher(t *testing.T) {
 			input:  "sub.foo.example.net",
 			expect: false,
 		},
+		{
+			match:  MatchHost{"example.com"},
+			input:  "example.com:5555",
+			expect: true,
+		},
 	} {
 		req := &http.Request{Host: tc.input}
 		actual := tc.match.Match(req)