httpserver spec support wildcards

pkg/object/httpserver/routers/spec.go

@@ -22,6 +22,7 @@ import (
  "net/http"
  "net/url"
  "regexp"
+ "strings"
 
  "github.com/megaease/easegress/v2/pkg/logger"
  "github.com/megaease/easegress/v2/pkg/util/ipfilter"
@@ -38,6 +39,8 @@ type Paths []*Path
 type Host struct {
  IsRegexp bool `json:"isRegexp" jsonschema:"omitempty"`
  Value string `json:"value" jsonschema:"required"`
+ prefix string `json:"-"`
+ suffix string `json:"-"`
  re *regexp.Regexp
 }
 
@@ -124,6 +127,20 @@ func (rule *Rule) Init() {
  for i := range rule.Hosts {
  h := &rule.Hosts[i]
  if !h.IsRegexp {
+ if h.Value != "" {
+ count := strings.Count(h.Value, "*")
+ if count > 1 {
+ logger.Errorf("invalid host %s, only one wildcard is allowed", h.Value)
+ continue
+ }
+ if h.Value[0] == '*' {
+ h.suffix = h.Value[1:]
+ } else if h.Value[len(h.Value)-1] == '*' {
+ h.prefix = h.Value[:len(h.Value)-1]
+ } else {
+ logger.Errorf("invalid host %s, only wildcard prefix or suffix is allowed", h.Value)
+ }
+ }
  continue
  }
  if re, err := regexp.Compile(h.Value); err != nil {
@@ -154,6 +171,10 @@ func (rule *Rule) MatchHost(ctx *RouteContext) bool {
  }
  } else if host == h.Value {
  return true
+ } else if h.prefix != "" && strings.HasPrefix(host, h.prefix) {
+ return true
+ } else if h.suffix != "" && strings.HasSuffix(host, h.suffix) {
+ return true
  }
  }
 

pkg/object/httpserver/routers/spec_test.go

@@ -126,6 +126,29 @@ func TestRuleMatch(t *testing.T) {
  rule.Init()
  assert.NotNil(rule)
  assert.False(rule.MatchHost(ctx))
+
+ testCases := []struct {
+ request string
+ value string
+ result bool
+ }{
+ {request: "http:https://www.megaease.com:8080", value: "www.megaease.com", result: true},
+ {request: "http:https://www.megaease.com:8080", value: "*.megaease.com", result: true},
+ {request: "http:https://www.sub.megaease.com:8080", value: "*.megaease.com", result: true},
+ {request: "http:https://www.example.megaease.com:8080", value: "*.megaease.com", result: true},
+ {request: "http:https://www.megaease.com:8080", value: "www.megaease.*", result: true},
+ {request: "http:https://www.megaease.cn:8080", value: "www.megaease.*", result: true},
+ {request: "http:https://www.google.com:8080", value: "*.megaease.com", result: false},
+ }
+ for _, tc := range testCases {
+ stdr, _ := http.NewRequest(http.MethodGet, tc.request, nil)
+ req, _ := httpprot.NewRequest(stdr)
+ ctx := NewContext(req)
+
+ rule = &Rule{Hosts: []Host{{Value: tc.value}}}
+ rule.Init()
+ assert.Equal(tc.result, rule.MatchHost(ctx))
+ }
 }
 
 func TestRuleAllowIP(t *testing.T) {