feat: upgrade goVirtualHost to support prefix hostname match

Author: marjune163

README.md

@@ -256,4 +256,13 @@ server [options]
     If request host name does not match any virtual host,
     server will try to use first virtual host that has no hostname,
     otherwise use the first virtual host.
+
+    If hostname option starts with ".",
+    treat it as a suffix,
+    to match all levels of sub domains,
+    e.g. ".example.com".
+
+    If hostname option ends with ".",
+    treat it as a prefix,
+    to match all levels of suffix domains.
 ```

README.zh-CN.md

@@ -253,4 +253,13 @@ server [选项]
     如果请求的主机名不匹配任何虚拟主机，
     服务器尝试使用第一个没有指定主机名的虚拟主机，
     如果失败则使用第一个虚拟主机。
+
+    如果主机名选项以“.”开头，
+    则将其当作后缀，
+    匹配所有层级的子域名，
+    例如“.example.com”。
+
+    如果主机名以“.”结尾，
+    则将其当作前缀，
+    匹配所有域名后缀。
 ```

src/goVirtualHost/util.go

@@ -14,10 +14,10 @@ func extractHostName(host string) string {
 		maxIndex := hostLen - 1
 		closeIndex := strings.IndexByte(host, ']')
 		if closeIndex == maxIndex {
-			return host
+			return host[1:closeIndex]
 		}
 		if closeIndex > 1 && closeIndex < maxIndex && host[closeIndex+1] == ':' {
-			return host[:closeIndex+1]
+			return host[1:closeIndex]
 		}
 	}
 

src/goVirtualHost/util_test.go

@@ -19,13 +19,13 @@ func TestExtractHostname(t *testing.T) {
 
 	host = "[fe80::1]"
 	hostname = extractHostName(host)
-	if hostname != "[fe80::1]" {
+	if hostname != "fe80::1" {
 		t.Error(hostname)
 	}
 
 	host = "[fe80::1]:8080"
 	hostname = extractHostName(host)
-	if hostname != "[fe80::1]" {
+	if hostname != "fe80::1" {
 		t.Error(hostname)
 	}
 }

src/goVirtualHost/vhost.go

@@ -22,8 +22,12 @@ func (v *vhost) matchHostName(name string) bool {
 		if hostname == reqHostName {
 			return true
 		}
-		if len(hostname) > 0 && hostname[0] == '.' && strings.HasSuffix(reqHostName, hostname) {
-			return true
+		if len(hostname) > 1 {
+			if hostname[0] == '.' && strings.HasSuffix(reqHostName, hostname) {
+				return true
+			} else if hostname[len(hostname)-1] == '.' && strings.HasPrefix(reqHostName, hostname) {
+				return true
+			}
 		}
 	}
 	return false

src/goVirtualHost/vhost_test.go

@@ -0,0 +1,40 @@
+package goVirtualHost
+
+import "testing"
+
+func TestMatchHostName(t *testing.T) {
+
+	var vh *vhost
+
+	vh = newVhost(nil, []string{"www.example.com"}, nil)
+	if !vh.matchHostName("www.example.com") {
+		t.Error()
+	}
+	if vh.matchHostName("example.com") {
+		t.Error()
+	}
+
+	vh = newVhost(nil, []string{".example.com"}, nil)
+	if !vh.matchHostName("www.example.com") {
+		t.Error()
+	}
+	if vh.matchHostName("example.com") {
+		t.Error()
+	}
+
+	vh = newVhost(nil, []string{".example.com", "example.com"}, nil)
+	if !vh.matchHostName("www.example.com") {
+		t.Error()
+	}
+	if !vh.matchHostName("example.com") {
+		t.Error()
+	}
+
+	vh = newVhost(nil, []string{"example."}, nil)
+	if !vh.matchHostName("example.com") {
+		t.Error()
+	}
+	if !vh.matchHostName("example.net") {
+		t.Error()
+	}
+}