feat(dns): support use multiple root domains (#51)

pkg/dns/dns.go

@@ -17,11 +17,11 @@ import (
 
 type Server struct {
 	Config
-	serverDomain string
-	serverIP     string
-	rules        []*Rule
-	rulesLock    sync.RWMutex
-	livingLock   sync.Mutex
+	serverDomains []string
+	serverIP      string
+	rules         []*Rule
+	rulesLock     sync.RWMutex
+	livingLock    sync.Mutex
 }
 
 var (
@@ -37,8 +37,8 @@ func GetServer() *Server {
 	return server
 }
 
-func (s *Server) SetServerDomain(serverDomain string) *Server {
-	s.serverDomain = serverDomain
+func (s *Server) SetServerDomain(serverDomains []string) *Server {
+	s.serverDomains = serverDomains
 	return s
 }
 
@@ -217,39 +217,42 @@ func (s *Server) Run() {
 	// create server
 	server := newdns.NewServer(newdns.Config{
 		Handler: func(name string) (*newdns.Zone, error) {
-			if name == s.serverDomain+"." {
-				return &newdns.Zone{
-					Name:             getZoneName(s.serverDomain),
-					MasterNameServer: "ns1.hostmaster.com.",
-					AllNameServers: []string{
-						"ns1.hostmaster.com.",
-						"ns2.hostmaster.com.",
-						"ns3.hostmaster.com.",
-					},
-					Handler: func(_, remoteAddr string) ([]newdns.Set, error) {
-						return []newdns.Set{
-							{
-								Name: name,
-								Type: newdns.A,
-								TTL:  10,
-								Records: []newdns.Record{
-									{
-										Address: s.serverIP,
-									},
-								}}}, nil
-					}}, nil
+			for _, serverDomain := range s.serverDomains {
+				if name == serverDomain+"." {
+					return &newdns.Zone{
+						Name:             getZoneName(serverDomain),
+						MasterNameServer: "ns1.hostmaster.com.",
+						AllNameServers: []string{
+							"ns1.hostmaster.com.",
+							"ns2.hostmaster.com.",
+							"ns3.hostmaster.com.",
+						},
+						Handler: func(_, remoteAddr string) ([]newdns.Set, error) {
+							return []newdns.Set{
+								{
+									Name: name,
+									Type: newdns.A,
+									TTL:  10,
+									Records: []newdns.Record{
+										{
+											Address: s.serverIP,
+										},
+									}}}, nil
+						}}, nil
+				}
 			}
 			return s.newZone(name), nil
 		},
 	})
 
 	// run server
-	log.Info("Starting DNS Server at :53, resolve %s to %s", s.serverDomain, s.serverIP)
+	log.Info("Starting DNS Server at :53, resolve %v to %s", s.serverDomains, s.serverIP)
 	go func() {
 		s.livingLock.Lock()
 		if !s.Enable {
 			server.Close()
 		}
+		s.livingLock.Unlock()
 	}()
 
 	err := server.Run(s.Addr)

pkg/ftp/ftp.go

@@ -393,6 +393,7 @@ func (s *Server) Run() {
 		if !s.Enable {
 			_ = listener.Close()
 		}
+		s.livingLock.Unlock()
 	}()
 
 	for {

pkg/ldap/ldap.go

@@ -180,6 +180,7 @@ func (s *Server) Run() {
 		if !s.Enable {
 			_ = listener.Close()
 		}
+		s.livingLock.Unlock()
 	}()
 
 	for {

pkg/mysql/mysql.go

@@ -310,6 +310,7 @@ func (s *Server) Run() {
 		if !s.Enable {
 			s.listener.Close()
 		}
+		s.livingLock.Unlock()
 	}()
 
 	s.listener.Accept()

pkg/rmi/rmi.go

@@ -189,6 +189,7 @@ func (s *Server) Run() {
 		if !s.Enable {
 			_ = listener.Close()
 		}
+		s.livingLock.Unlock()
 	}()
 
 	for {

pkg/server/config.go

@@ -21,7 +21,7 @@ type Config struct {
 	Version            float64
 	Addr               string
 	Token              string
-	Domain             string
+	Domains            []string
 	ExternalIP         string `yaml:"external_ip"`
 	AdminPathPrefix    string `yaml:"admin_path_prefix"`
 	Database           string

pkg/server/server.go

@@ -195,8 +195,8 @@ func New(c *Config) *Revsuit {
 	if c.HTTP.IpHeader != "" {
 		s.http.SetIpHeader(c.HTTP.IpHeader)
 	}
-	if c.Domain != "" {
-		s.dns.SetServerDomain(c.Domain)
+	if len(c.Domains) != 0 {
+		s.dns.SetServerDomain(c.Domains)
 	}
 	if c.ExternalIP != "" {
 		s.dns.SetServerIP(c.ExternalIP)

pkg/server/settings.go

@@ -3,6 +3,7 @@ package server
 import (
 	"fmt"
 	"io"
+	"strings"
 	"time"
 
 	"github.com/gin-gonic/gin"
@@ -166,7 +167,7 @@ func (revsuit *Revsuit) getPlatformConfig(c *gin.Context) {
 	var res = make(map[string]string)
 	res["Addr"] = revsuit.config.Addr
 	res["Token"] = revsuit.config.Token
-	res["Domain"] = revsuit.config.Domain
+	res["Domains"] = strings.Join(revsuit.config.Domains, ",")
 	res["AdminPathPrefix"] = revsuit.config.AdminPathPrefix
 	res["ExternalIP"] = revsuit.config.ExternalIP
 	res["Database"] = revsuit.config.Database
@@ -215,10 +216,10 @@ func (revsuit *Revsuit) updatePlatformConfig(c *gin.Context) {
 		log.Info("Update http config [ip_header] to %s", form["IpHeader"])
 	}
 
-	if form["Domain"] != revsuit.config.Domain {
-		revsuit.config.Domain = form["Domain"]
-		revsuit.dns.SetServerDomain(form["Domain"])
-		log.Info("Update platform config [domain] to %s", form["Domain"])
+	if form["Domains"] != strings.Join(revsuit.config.Domains, ",") {
+		revsuit.config.Domains = strings.Split(form["Domains"], ",")
+		revsuit.dns.SetServerDomain(revsuit.config.Domains)
+		log.Info("Update platform config [domain] to %v", revsuit.config.Domains)
 		if revsuit.dns.Enable {
 			revsuit.dns.Restart()
 		}
@@ -307,6 +308,11 @@ func (revsuit *Revsuit) updateDnsConfig(c *gin.Context) {
 		return
 	}
 
+	if form.Addr != revsuit.dns.Addr {
+		revsuit.dns.Addr = form.Addr
+		log.Info("Update dns config [addr] to %s", form.Addr)
+	}
+
 	if form.Enable != revsuit.dns.Enable {
 		log.Info("Update dns config [enable] to %v", form.Enable)
 		if form.Enable {
@@ -316,6 +322,11 @@ func (revsuit *Revsuit) updateDnsConfig(c *gin.Context) {
 		}
 		return
 	}
+
+	if revsuit.dns.Enable {
+		revsuit.dns.Restart()
+	}
+
 }
 
 func (revsuit *Revsuit) getMySQLConfig(c *gin.Context) {