zema1 · xXxsomebodyoncetoldmexXx · Jul 9, 2024
diff --git a/ctrl/config.go b/ctrl/config.go
@@ -28,6 +28,7 @@ type Suo5Config struct {
 
 	TestExit                string                               `json:"-"`
 	ExcludeDomainMap        map[string]bool                      `json:"-"`
+	ExcludeDomainWCMap      map[string]bool                      `json:"-"`
 	Offset                  int                                  `json:"-"`
 	Header                  http.Header                          `json:"-"`
 	OnRemoteConnected       func(e *ConnectedEvent)              `json:"-"`
@@ -43,8 +44,14 @@ func (s *Suo5Config) Parse() error {
 
 func (s *Suo5Config) parseExcludeDomain() {
 	s.ExcludeDomainMap = make(map[string]bool)
+	s.ExcludeDomainWCMap = make(map[string]bool)
 	for _, domain := range s.ExcludeDomain {
-		s.ExcludeDomainMap[strings.ToLower(strings.TrimSpace(domain))] = true
+		domain = strings.ToLower(strings.TrimSpace(domain))
+		if domain[0] == '*' {
+			s.ExcludeDomainWCMap[domain[2:]] = true
+		} else {
+			s.ExcludeDomainMap[domain] = true
+		}
 	}
 }
 

diff --git a/ctrl/handler.go b/ctrl/handler.go
@@ -9,6 +9,7 @@
 	"net"
 	"net/http"
 	"strconv"
+	"strings"
 	"sync"
 	"time"
 
@@ -44,6 +45,24 @@
 	selector        gosocks5.Selector
 }
 
+func IsExcludeDomain(m *socks5Handler, d string) bool {
+	// Default Exclude
+	if m.config.ExcludeDomainMap[d] {
+		return true
+	}
+
+	// WildCard Exclude
+	ntld := strings.Count(d, ".")
+	for ntld > 0 && !m.config.ExcludeDomainWCMap[d] {
+		d = strings.SplitN(d, ".", 2)[1]
+		ntld--
+	}
+	if ntld != 0 {
+		return true
+	}
+	return false
+}
+
 func (m *socks5Handler) Handle(conn net.Conn) error {
 	defer conn.Close()
 
@@ -54,7 +73,7 @@
 		return err
 	}
 
-	if m.config.ExcludeDomainMap[req.Addr.Host] {
+	if IsExcludeDomain(m, req.Addr.Host) {
 		log.Infof("drop connection to %s", req.Addr.Host)
 		return nil
 	}

diff --git a/suo5 b/suo5