Merge pull request #16 from ildyria/multi-hostnames

Support multiple host names, comma separated

cmd/socket-proxy/handlehttprequest.go

@@ -6,6 +6,7 @@ import (
 	"log/slog"
 	"net"
 	"net/http"
+	"strings"
 )
 
 // handleHttpRequest checks if the request is allowed and sends it to the proxy.
@@ -61,14 +62,18 @@ func isAllowedClient(remoteAddr string) (bool, error) {
 		return allowedIPNet.Contains(clientIP), nil
 	} else {
 		// AllowFrom is not a valid CIDR, so try to resolve it via DNS
-		ips, err := net.LookupIP(cfg.AllowFrom)
-		if err != nil {
-			return false, errors.New("error looking up allowed client hostname: " + err.Error())
-		}
-		for _, ip := range ips {
-			// Check if IP address is one of the resolved IPs
-			if ip.Equal(clientIP) {
-				return true, nil
+		// split over comma to support multiple hostnames
+		allowFroms := strings.Split(cfg.AllowFrom, ",")
+		for _, allowFrom := range allowFroms {
+			ips, err := net.LookupIP(allowFrom)
+			if err != nil {
+				return false, errors.New("error looking up allowed client hostname: " + err.Error())
+			}
+			for _, ip := range ips {
+				// Check if IP address is one of the resolved IPs
+				if ip.Equal(clientIP) {
+					return true, nil
+				}
 			}
 		}
 		return false, nil