Rename coredns.go to dns.go

dns.go

@@ -0,0 +1,89 @@
+package main
+
+import (
+	"fmt"
+	"log"
+	"net"
+
+	"github.com/AdguardTeam/AdGuardHome/dnsfilter"
+	"github.com/AdguardTeam/AdGuardHome/dnsforward"
+	"github.com/joomcode/errorx"
+)
+
+var dnsServer = dnsforward.Server{}
+
+func isRunning() bool {
+	return dnsServer.IsRunning()
+}
+
+func generateServerConfig() dnsforward.ServerConfig {
+	filters := []dnsfilter.Filter{}
+	userFilter := userFilter()
+	filters = append(filters, dnsfilter.Filter{
+		ID:    userFilter.ID,
+		Rules: userFilter.Rules,
+	})
+	for _, filter := range config.Filters {
+		filters = append(filters, dnsfilter.Filter{
+			ID:    filter.ID,
+			Rules: filter.Rules,
+		})
+	}
+
+	newconfig := dnsforward.ServerConfig{
+		UDPListenAddr:   &net.UDPAddr{Port: config.DNS.Port},
+		FilteringConfig: config.DNS.FilteringConfig,
+		Filters:         filters,
+	}
+
+	for _, u := range config.DNS.UpstreamDNS {
+		upstream, err := dnsforward.AddressToUpstream(u, config.DNS.BootstrapDNS)
+		if err != nil {
+			log.Printf("Couldn't get upstream: %s", err)
+			// continue, just ignore the upstream
+			continue
+		}
+		newconfig.Upstreams = append(newconfig.Upstreams, upstream)
+	}
+	return newconfig
+}
+
+func startDNSServer() error {
+	if isRunning() {
+		return fmt.Errorf("Unable to start forwarding DNS server: Already running")
+	}
+
+	newconfig := generateServerConfig()
+	err := dnsServer.Start(&newconfig)
+	if err != nil {
+		return errorx.Decorate(err, "Couldn't start forwarding DNS server")
+	}
+
+	return nil
+}
+
+func reconfigureDNSServer() error {
+	if !isRunning() {
+		return fmt.Errorf("Refusing to reconfigure forwarding DNS server: not running")
+	}
+
+	err := dnsServer.Reconfigure(generateServerConfig())
+	if err != nil {
+		return errorx.Decorate(err, "Couldn't start forwarding DNS server")
+	}
+
+	return nil
+}
+
+func stopDNSServer() error {
+	if !isRunning() {
+		return fmt.Errorf("Refusing to stop forwarding DNS server: not running")
+	}
+
+	err := dnsServer.Stop()
+	if err != nil {
+		return errorx.Decorate(err, "Couldn't stop forwarding DNS server")
+	}
+
+	return nil
+}