Support routing

Author: dima424658

config.go

@@ -1,15 +1,19 @@
 package awgproxy
 
 import (
+	"bufio"
 	"encoding/base64"
 	"encoding/hex"
 	"errors"
+	"io"
+	"log"
 	"net"
 	"os"
 	"strings"
 
 	"github.com/go-ini/ini"
 
+	"net/http"
 	"net/netip"
 )
 
@@ -68,6 +72,7 @@ type HTTPConfig struct {
 	Password    string
 	CertFile    string
 	KeyFile     string
+	RouteHosts  map[string]bool
 }
 
 type Configuration struct {
@@ -531,6 +536,43 @@ func parseHTTPConfig(section *ini.Section) (RoutineSpawner, error) {
 	keyFile, _ := parseString(section, "KeyFile")
 	config.KeyFile = keyFile
 
+	config.RouteHosts = make(map[string]bool)
+
+	domainsUrl, err := parseString(section, "DomainsUrl")
+	if err == nil && domainsUrl != "" {
+		resp, err := http.Get(domainsUrl)
+		if err != nil {
+			return nil, err
+		}
+
+		rd := bufio.NewReader(resp.Body)
+		for {
+			str, err := rd.ReadString('\n')
+			if err == io.EOF {
+				break
+			} else if err != nil {
+				log.Printf("erra %s", err)
+				return nil, err
+			}
+
+			config.RouteHosts[strings.Trim(str, " \n\r")] = true
+		}
+	}
+
+	includeStr, err := parseString(section, "RouteInclude")
+	if err == nil {
+		for _, str := range strings.Split(includeStr, ",") {
+			config.RouteHosts[strings.Trim(str, " \n\r")] = true
+		}
+	}
+
+	excludeStr, err := parseString(section, "RouteExclude")
+	if err == nil {
+		for _, str := range strings.Split(excludeStr, ",") {
+			delete(config.RouteHosts, strings.Trim(str, " \n\r"))
+		}
+	}
+
 	return config, nil
 }
 

http.go

@@ -10,6 +10,7 @@ import (
 	"log"
 	"net"
 	"net/http"
+	"regexp"
 	"strings"
 )
 
@@ -23,6 +24,25 @@ type HTTPServer struct {
 
 	authRequired bool
 	tlsRequired  bool
+	routeRegex   *regexp.Regexp
+}
+
+func (s *HTTPServer) needRoute(req *http.Request) bool {
+	hostname := req.URL.Hostname()
+	if s.routeRegex == nil {
+		s.routeRegex = regexp.MustCompile(`^([^\.]*)\.`)
+	}
+
+	for strings.Count(hostname, ".") > 0 {
+		if s.config.RouteHosts[hostname] {
+			log.Printf("route %s", hostname)
+			return true
+		}
+
+		hostname = s.routeRegex.ReplaceAllString(hostname, "")
+	}
+
+	return false
 }
 
 func (s *HTTPServer) authenticate(req *http.Request) (int, error) {
@@ -57,7 +77,11 @@ func (s *HTTPServer) handleConn(req *http.Request, conn net.Conn) (peer net.Conn
 		addr = net.JoinHostPort(addr, port)
 	}
 
-	peer, err = s.dial("tcp", addr)
+	if s.needRoute(req) {
+		peer, err = s.dial("tcp", addr)
+	} else {
+		peer, err = net.Dial("tcp", addr)
+	}
 	if err != nil {
 		return peer, fmt.Errorf("tun tcp dial failed: %w", err)
 	}
@@ -78,7 +102,12 @@ func (s *HTTPServer) handle(req *http.Request) (peer net.Conn, err error) {
 		addr = net.JoinHostPort(addr, port)
 	}
 
-	peer, err = s.dial("tcp", addr)
+	if s.needRoute(req) {
+		peer, err = s.dial("tcp", addr)
+	} else {
+		peer, err = net.Dial("tcp", addr)
+	}
+
 	if err != nil {
 		return peer, fmt.Errorf("tun tcp dial failed: %w", err)
 	}