mirror of https://github.com/0xERR0R/blocky.git
687 lines
21 KiB
Go
687 lines
21 KiB
Go
package server
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base64"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"net"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/0xERR0R/blocky/api"
|
|
"github.com/0xERR0R/blocky/config"
|
|
. "github.com/0xERR0R/blocky/helpertest"
|
|
. "github.com/0xERR0R/blocky/log"
|
|
"github.com/0xERR0R/blocky/model"
|
|
"github.com/0xERR0R/blocky/resolver"
|
|
"github.com/0xERR0R/blocky/util"
|
|
"github.com/creasty/defaults"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
var (
|
|
upstreamGoogle, upstreamFritzbox, upstreamClient config.Upstream
|
|
mockClientName string
|
|
sut *Server
|
|
err error
|
|
resp *dns.Msg
|
|
)
|
|
|
|
var _ = BeforeSuite(func() {
|
|
upstreamGoogle = resolver.TestUDPUpstream(func(request *dns.Msg) *dns.Msg {
|
|
if request.Question[0].Name == "error." {
|
|
return nil
|
|
}
|
|
response, err := util.NewMsgWithAnswer(
|
|
util.ExtractDomain(request.Question[0]), 123, dns.Type(dns.TypeA), "123.124.122.122",
|
|
)
|
|
|
|
Expect(err).Should(Succeed())
|
|
return response
|
|
})
|
|
upstreamFritzbox = resolver.TestUDPUpstream(func(request *dns.Msg) *dns.Msg {
|
|
response, err := util.NewMsgWithAnswer(
|
|
util.ExtractDomain(request.Question[0]), 3600, dns.Type(dns.TypeA), "192.168.178.2",
|
|
)
|
|
|
|
Expect(err).Should(Succeed())
|
|
return response
|
|
})
|
|
|
|
upstreamClient = resolver.TestUDPUpstream(func(request *dns.Msg) *dns.Msg {
|
|
response, err := util.NewMsgWithAnswer(
|
|
util.ExtractDomain(request.Question[0]), 3600, dns.Type(dns.TypePTR), mockClientName,
|
|
)
|
|
|
|
Expect(err).Should(Succeed())
|
|
return response
|
|
})
|
|
|
|
// create server
|
|
sut, err = NewServer(&config.Config{
|
|
CustomDNS: config.CustomDNSConfig{
|
|
CustomTTL: config.Duration(time.Duration(3600) * time.Second),
|
|
Mapping: config.CustomDNSMapping{
|
|
HostIPs: map[string][]net.IP{
|
|
"custom.lan": {net.ParseIP("192.168.178.55")},
|
|
"lan.home": {net.ParseIP("192.168.178.56")},
|
|
},
|
|
},
|
|
},
|
|
Conditional: config.ConditionalUpstreamConfig{
|
|
Mapping: config.ConditionalUpstreamMapping{
|
|
Upstreams: map[string][]config.Upstream{
|
|
"net.cn": {upstreamClient},
|
|
"fritz.box": {upstreamFritzbox},
|
|
},
|
|
},
|
|
},
|
|
Blocking: config.BlockingConfig{
|
|
BlackLists: map[string][]string{
|
|
"ads": {
|
|
"../testdata/doubleclick.net.txt",
|
|
"../testdata/www.bild.de.txt",
|
|
"../testdata/heise.de.txt"},
|
|
"youtube": {"../testdata/youtube.com.txt"}},
|
|
WhiteLists: map[string][]string{
|
|
"ads": {"../testdata/heise.de.txt"},
|
|
"whitelist": {"../testdata/heise.de.txt"},
|
|
},
|
|
ClientGroupsBlock: map[string][]string{
|
|
"default": {"ads"},
|
|
"clWhitelistOnly": {"whitelist"},
|
|
"clAdsAndYoutube": {"ads", "youtube"},
|
|
"clYoutubeOnly": {"youtube"},
|
|
},
|
|
BlockType: "zeroIp",
|
|
BlockTTL: config.Duration(6 * time.Hour),
|
|
},
|
|
Upstream: config.UpstreamConfig{
|
|
ExternalResolvers: map[string][]config.Upstream{"default": {upstreamGoogle}},
|
|
},
|
|
ClientLookup: config.ClientLookupConfig{
|
|
Upstream: upstreamClient,
|
|
},
|
|
|
|
DNSPorts: config.ListenConfig{"55555"},
|
|
TLSPorts: config.ListenConfig{"8853"},
|
|
CertFile: "../testdata/cert.pem",
|
|
KeyFile: "../testdata/key.pem",
|
|
HTTPPorts: config.ListenConfig{"4000"},
|
|
HTTPSPorts: config.ListenConfig{"4443"},
|
|
Prometheus: config.PrometheusConfig{
|
|
Enable: true,
|
|
Path: "/metrics",
|
|
},
|
|
})
|
|
|
|
Expect(err).Should(Succeed())
|
|
|
|
// start server
|
|
go func() {
|
|
sut.Start()
|
|
}()
|
|
time.Sleep(100 * time.Millisecond)
|
|
})
|
|
|
|
var _ = AfterSuite(func() {
|
|
sut.Stop()
|
|
})
|
|
|
|
var _ = Describe("Running DNS server", func() {
|
|
Describe("performing DNS request with running server", func() {
|
|
|
|
BeforeEach(func() {
|
|
mockClientName = ""
|
|
// reset client cache
|
|
res := sut.queryResolver
|
|
for res != nil {
|
|
if t, ok := res.(*resolver.ClientNamesResolver); ok {
|
|
t.FlushCache()
|
|
break
|
|
}
|
|
if c, ok := res.(resolver.ChainedResolver); ok {
|
|
res = c.GetNext()
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
})
|
|
|
|
AfterEach(func() {
|
|
Expect(resp.Rcode).Should(Equal(dns.RcodeSuccess))
|
|
})
|
|
Context("DNS query is resolvable via external DNS", func() {
|
|
It("should return valid answer", func() {
|
|
resp = requestServer(util.NewMsgWithQuestion("google.de.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("google.de.", dns.TypeA, 123, "123.124.122.122"))
|
|
})
|
|
})
|
|
Context("Custom DNS entry with exact match", func() {
|
|
It("should return valid answer", func() {
|
|
resp = requestServer(util.NewMsgWithQuestion("custom.lan.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("custom.lan.", dns.TypeA, 3600, "192.168.178.55"))
|
|
})
|
|
})
|
|
Context("Custom DNS entry with sub domain", func() {
|
|
It("should return valid answer", func() {
|
|
resp = requestServer(util.NewMsgWithQuestion("host.lan.home.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("host.lan.home.", dns.TypeA, 3600, "192.168.178.56"))
|
|
})
|
|
})
|
|
Context("Conditional upstream", func() {
|
|
It("should resolve query via conditional upstream resolver", func() {
|
|
resp = requestServer(util.NewMsgWithQuestion("host.fritz.box.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("host.fritz.box.", dns.TypeA, 3600, "192.168.178.2"))
|
|
})
|
|
})
|
|
Context("Conditional upstream blocking", func() {
|
|
It("Query should be blocked, domain is in default group", func() {
|
|
resp = requestServer(util.NewMsgWithQuestion("doubleclick.net.cn.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("doubleclick.net.cn.", dns.TypeA, 21600, "0.0.0.0"))
|
|
})
|
|
})
|
|
Context("Blocking default group", func() {
|
|
It("Query should be blocked, domain is in default group", func() {
|
|
resp = requestServer(util.NewMsgWithQuestion("doubleclick.net.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("doubleclick.net.", dns.TypeA, 21600, "0.0.0.0"))
|
|
})
|
|
})
|
|
Context("Blocking default group with sub domain", func() {
|
|
It("Query with subdomain should be blocked, domain is in default group", func() {
|
|
resp = requestServer(util.NewMsgWithQuestion("www.bild.de.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("www.bild.de.", dns.TypeA, 21600, "0.0.0.0"))
|
|
})
|
|
})
|
|
Context("no blocking default group with sub domain", func() {
|
|
It("Query with should not be blocked, sub domain is not in blacklist", func() {
|
|
resp = requestServer(util.NewMsgWithQuestion("bild.de.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("bild.de.", dns.TypeA, 0, "123.124.122.122"))
|
|
})
|
|
})
|
|
Context("domain is on white and blacklist default group", func() {
|
|
It("Query with should not be blocked, domain is on white and blacklist", func() {
|
|
resp = requestServer(util.NewMsgWithQuestion("heise.de.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("heise.de.", dns.TypeA, 0, "123.124.122.122"))
|
|
})
|
|
})
|
|
Context("domain is on client specific white list", func() {
|
|
It("Query with should not be blocked, domain is on client's white list", func() {
|
|
mockClientName = "clWhitelistOnly"
|
|
resp = requestServer(util.NewMsgWithQuestion("heise.de.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("heise.de.", dns.TypeA, 0, "123.124.122.122"))
|
|
})
|
|
})
|
|
Context("block client whitelist only", func() {
|
|
It("Query with should be blocked, client has only whitelist, domain is not on client's white list", func() {
|
|
mockClientName = "clWhitelistOnly"
|
|
resp = requestServer(util.NewMsgWithQuestion("google.de.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("google.de.", dns.TypeA, 0, "0.0.0.0"))
|
|
})
|
|
})
|
|
Context("block client with 2 groups", func() {
|
|
It("Query with should be blocked, domain is on black list", func() {
|
|
mockClientName = "clAdsAndYoutube"
|
|
resp = requestServer(util.NewMsgWithQuestion("www.bild.de.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("www.bild.de.", dns.TypeA, 0, "0.0.0.0"))
|
|
|
|
resp = requestServer(util.NewMsgWithQuestion("youtube.com.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("youtube.com.", dns.TypeA, 0, "0.0.0.0"))
|
|
})
|
|
})
|
|
Context("client with 1 group: no block if domain in other group", func() {
|
|
It("Query with should not be blocked, domain is on black list in another group", func() {
|
|
mockClientName = "clYoutubeOnly"
|
|
resp = requestServer(util.NewMsgWithQuestion("www.bild.de.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("www.bild.de.", dns.TypeA, 0, "123.124.122.122"))
|
|
})
|
|
})
|
|
Context("block client with 1 group", func() {
|
|
It("Query with should not blocked, domain is on black list in client's group", func() {
|
|
mockClientName = "clYoutubeOnly"
|
|
resp = requestServer(util.NewMsgWithQuestion("youtube.com.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeDNSRecord("youtube.com.", dns.TypeA, 0, "0.0.0.0"))
|
|
})
|
|
})
|
|
Context("health check", func() {
|
|
It("Should always return dummy response", func() {
|
|
resp = requestServer(util.NewMsgWithQuestion("healthcheck.blocky.", dns.Type(dns.TypeA)))
|
|
|
|
Expect(resp.Answer).Should(BeEmpty())
|
|
})
|
|
})
|
|
|
|
})
|
|
|
|
Describe("Prometheus endpoint", func() {
|
|
When("Prometheus URL is called", func() {
|
|
It("should return prometheus data", func() {
|
|
r, err := http.Get("http://localhost:4000/metrics")
|
|
Expect(err).Should(Succeed())
|
|
Expect(r.StatusCode).Should(Equal(http.StatusOK))
|
|
})
|
|
})
|
|
})
|
|
Describe("Root endpoint", func() {
|
|
When("Root URL is called", func() {
|
|
It("should return root page", func() {
|
|
r, err := http.Get("http://localhost:4000/")
|
|
Expect(err).Should(Succeed())
|
|
Expect(r.StatusCode).Should(Equal(http.StatusOK))
|
|
})
|
|
})
|
|
})
|
|
|
|
Describe("Query Rest API", func() {
|
|
When("Query API is called", func() {
|
|
It("Should process the query", func() {
|
|
req := api.QueryRequest{
|
|
Query: "google.de",
|
|
Type: "A",
|
|
}
|
|
jsonValue, err := json.Marshal(req)
|
|
Expect(err).Should(Succeed())
|
|
|
|
resp, err := http.Post("http://localhost:4000/api/query", "application/json", bytes.NewBuffer(jsonValue))
|
|
|
|
Expect(err).Should(Succeed())
|
|
defer resp.Body.Close()
|
|
|
|
Expect(resp.StatusCode).Should(Equal(http.StatusOK))
|
|
|
|
var result api.QueryResult
|
|
err = json.NewDecoder(resp.Body).Decode(&result)
|
|
Expect(err).Should(Succeed())
|
|
Expect(result.Response).Should(Equal("A (123.124.122.122)"))
|
|
})
|
|
})
|
|
When("Wrong request type is used", func() {
|
|
It("Should return internal error", func() {
|
|
req := api.QueryRequest{
|
|
Query: "google.de",
|
|
Type: "WrongType",
|
|
}
|
|
jsonValue, err := json.Marshal(req)
|
|
Expect(err).Should(Succeed())
|
|
|
|
resp, err := http.Post("http://localhost:4000/api/query", "application/json", bytes.NewBuffer(jsonValue))
|
|
|
|
Expect(err).Should(Succeed())
|
|
defer resp.Body.Close()
|
|
|
|
Expect(resp.StatusCode).Should(Equal(http.StatusInternalServerError))
|
|
})
|
|
})
|
|
When("Internal error occurs", func() {
|
|
It("Should return internal error", func() {
|
|
req := api.QueryRequest{
|
|
Query: "error.",
|
|
Type: "A",
|
|
}
|
|
jsonValue, err := json.Marshal(req)
|
|
Expect(err).Should(Succeed())
|
|
|
|
resp, err := http.Post("http://localhost:4000/api/query", "application/json", bytes.NewBuffer(jsonValue))
|
|
Expect(err).Should(Succeed())
|
|
Expect(resp.StatusCode).Should(Equal(http.StatusInternalServerError))
|
|
_ = resp.Body.Close()
|
|
})
|
|
})
|
|
When("Request is malformed", func() {
|
|
It("Should return internal error", func() {
|
|
jsonValue := []byte("")
|
|
|
|
resp, err := http.Post("http://localhost:4000/api/query", "application/json", bytes.NewBuffer(jsonValue))
|
|
|
|
Expect(err).Should(Succeed())
|
|
defer resp.Body.Close()
|
|
|
|
Expect(resp.StatusCode).Should(Equal(http.StatusInternalServerError))
|
|
})
|
|
})
|
|
})
|
|
|
|
Describe("DOH endpoint", func() {
|
|
Context("DOH over GET (RFC 8484)", func() {
|
|
When("DOH get request with 'example.com' is performed", func() {
|
|
It("should get a valid response", func() {
|
|
resp, err := http.Get("http://localhost:4000/dns-query?dns=AAABAAABAAAAAAAAA3d3dwdleGFtcGxlA2NvbQAAAQAB")
|
|
Expect(err).Should(Succeed())
|
|
defer resp.Body.Close()
|
|
|
|
Expect(resp).Should(HaveHTTPStatus(http.StatusOK))
|
|
rawMsg, err := ioutil.ReadAll(resp.Body)
|
|
Expect(err).Should(Succeed())
|
|
|
|
msg := new(dns.Msg)
|
|
err = msg.Unpack(rawMsg)
|
|
Expect(err).Should(Succeed())
|
|
|
|
Expect(msg.Answer).Should(BeDNSRecord("www.example.com.", dns.TypeA, 0, "123.124.122.122"))
|
|
})
|
|
})
|
|
When("Request does not contain a valid DNS message", func() {
|
|
It("should return 'Bad Request'", func() {
|
|
resp, err := http.Get("http://localhost:4000/dns-query?dns=xxxx")
|
|
Expect(err).Should(Succeed())
|
|
defer resp.Body.Close()
|
|
|
|
Expect(resp).Should(HaveHTTPStatus(http.StatusBadRequest))
|
|
})
|
|
})
|
|
When("Request's parameter does not contain a valid base64'", func() {
|
|
It("should return 'Bad Request'", func() {
|
|
resp, err := http.Get("http://localhost:4000/dns-query?dns=äöä")
|
|
Expect(err).Should(Succeed())
|
|
defer resp.Body.Close()
|
|
|
|
Expect(resp).Should(HaveHTTPStatus(http.StatusBadRequest))
|
|
})
|
|
})
|
|
When("Request does not contain a dns parameter", func() {
|
|
It("should return 'Bad Request'", func() {
|
|
resp, err := http.Get("http://localhost:4000/dns-query?test")
|
|
Expect(err).Should(Succeed())
|
|
defer resp.Body.Close()
|
|
|
|
Expect(resp).Should(HaveHTTPStatus(http.StatusBadRequest))
|
|
})
|
|
})
|
|
When("Request's dns parameter is too long'", func() {
|
|
It("should return 'URI Too Long'", func() {
|
|
longBase64msg := base64.StdEncoding.EncodeToString([]byte(strings.Repeat("t", 513)))
|
|
|
|
resp, err := http.Get("http://localhost:4000/dns-query?dns=" + longBase64msg)
|
|
Expect(err).Should(Succeed())
|
|
defer resp.Body.Close()
|
|
|
|
Expect(resp).Should(HaveHTTPStatus(http.StatusRequestURITooLong))
|
|
})
|
|
})
|
|
|
|
})
|
|
Context("DOH over POST (RFC 8484)", func() {
|
|
When("DOH post request with 'example.com' is performed", func() {
|
|
It("should get a valid response", func() {
|
|
msg := util.NewMsgWithQuestion("www.example.com.", dns.Type(dns.TypeA))
|
|
rawDNSMessage, err := msg.Pack()
|
|
Expect(err).Should(Succeed())
|
|
|
|
resp, err := http.Post("http://localhost:4000/dns-query",
|
|
"application/dns-message", bytes.NewReader(rawDNSMessage))
|
|
Expect(err).Should(Succeed())
|
|
defer resp.Body.Close()
|
|
Expect(resp).Should(HaveHTTPStatus(http.StatusOK))
|
|
rawMsg, err := ioutil.ReadAll(resp.Body)
|
|
Expect(err).Should(Succeed())
|
|
|
|
msg = new(dns.Msg)
|
|
err = msg.Unpack(rawMsg)
|
|
Expect(err).Should(Succeed())
|
|
|
|
Expect(msg.Answer).Should(BeDNSRecord("www.example.com.", dns.TypeA, 0, "123.124.122.122"))
|
|
})
|
|
It("should get a valid response, clientId is passed", func() {
|
|
msg := util.NewMsgWithQuestion("www.example.com.", dns.Type(dns.TypeA))
|
|
rawDNSMessage, err := msg.Pack()
|
|
Expect(err).Should(Succeed())
|
|
|
|
resp, err := http.Post("http://localhost:4000/dns-query/client123",
|
|
"application/dns-message", bytes.NewReader(rawDNSMessage))
|
|
Expect(err).Should(Succeed())
|
|
defer resp.Body.Close()
|
|
Expect(resp).Should(HaveHTTPStatus(http.StatusOK))
|
|
rawMsg, err := ioutil.ReadAll(resp.Body)
|
|
Expect(err).Should(Succeed())
|
|
|
|
msg = new(dns.Msg)
|
|
err = msg.Unpack(rawMsg)
|
|
Expect(err).Should(Succeed())
|
|
|
|
Expect(msg.Answer).Should(BeDNSRecord("www.example.com.", dns.TypeA, 0, "123.124.122.122"))
|
|
})
|
|
})
|
|
When("POST payload exceeds 512 bytes", func() {
|
|
It("should return 'Payload Too Large'", func() {
|
|
largeMessage := []byte(strings.Repeat("t", 513))
|
|
|
|
resp, err := http.Post("http://localhost:4000/dns-query", "application/dns-message", bytes.NewReader(largeMessage))
|
|
Expect(err).Should(Succeed())
|
|
defer resp.Body.Close()
|
|
|
|
Expect(resp).Should(HaveHTTPStatus(http.StatusRequestEntityTooLarge))
|
|
})
|
|
})
|
|
When("Request has wrong type", func() {
|
|
It("should return 'Unsupported Media Type'", func() {
|
|
resp, err := http.Post("http://localhost:4000/dns-query", "application/text", bytes.NewReader([]byte("a")))
|
|
Expect(err).Should(Succeed())
|
|
defer resp.Body.Close()
|
|
|
|
Expect(resp).Should(HaveHTTPStatus(http.StatusUnsupportedMediaType))
|
|
})
|
|
})
|
|
When("Internal error occurs", func() {
|
|
It("should return 'Internal server error'", func() {
|
|
msg := util.NewMsgWithQuestion("error.", dns.Type(dns.TypeA))
|
|
rawDNSMessage, err := msg.Pack()
|
|
Expect(err).Should(Succeed())
|
|
|
|
resp, err := http.Post("http://localhost:4000/dns-query",
|
|
"application/dns-message", bytes.NewReader(rawDNSMessage))
|
|
Expect(err).Should(Succeed())
|
|
defer resp.Body.Close()
|
|
Expect(resp).Should(HaveHTTPStatus(http.StatusInternalServerError))
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
Describe("Server create", func() {
|
|
var (
|
|
cfg config.Config
|
|
cErr error
|
|
)
|
|
BeforeEach(func() {
|
|
cErr = defaults.Set(&cfg)
|
|
|
|
Expect(cErr).Should(Succeed())
|
|
|
|
cfg.Upstream.ExternalResolvers = map[string][]config.Upstream{
|
|
"default": {config.Upstream{Net: config.NetProtocolTcpUdp, Host: "4.4.4.4", Port: 53}}}
|
|
|
|
cfg.Redis.Address = "test-fail"
|
|
})
|
|
When("Server is created", func() {
|
|
It("is created without redis connection", func() {
|
|
defer func() { Log().ExitFunc = nil }()
|
|
|
|
_, err := NewServer(&cfg)
|
|
|
|
Expect(err).Should(Succeed())
|
|
})
|
|
It("can't be created if redis server is unavailable", func() {
|
|
defer func() { Log().ExitFunc = nil }()
|
|
|
|
cfg.Redis.Required = true
|
|
|
|
_, err := NewServer(&cfg)
|
|
|
|
Expect(err).ShouldNot(Succeed())
|
|
})
|
|
})
|
|
})
|
|
|
|
Describe("Server start", func() {
|
|
When("Server start is called", func() {
|
|
It("start was called 2 times, start should fail", func() {
|
|
defer func() { Log().ExitFunc = nil }()
|
|
|
|
var fatal bool
|
|
|
|
Log().ExitFunc = func(int) { fatal = true }
|
|
|
|
// create server
|
|
server, err := NewServer(&config.Config{
|
|
Upstream: config.UpstreamConfig{
|
|
ExternalResolvers: map[string][]config.Upstream{
|
|
"default": {config.Upstream{Net: config.NetProtocolTcpUdp, Host: "4.4.4.4", Port: 53}}}},
|
|
CustomDNS: config.CustomDNSConfig{
|
|
Mapping: config.CustomDNSMapping{
|
|
HostIPs: map[string][]net.IP{
|
|
"custom.lan": {net.ParseIP("192.168.178.55")},
|
|
"lan.home": {net.ParseIP("192.168.178.56")},
|
|
},
|
|
}},
|
|
Blocking: config.BlockingConfig{BlockType: "zeroIp"},
|
|
DNSPorts: config.ListenConfig{":55556"},
|
|
})
|
|
|
|
Expect(err).Should(Succeed())
|
|
|
|
// start server
|
|
go func() {
|
|
server.Start()
|
|
}()
|
|
|
|
defer server.Stop()
|
|
|
|
Eventually(func() bool {
|
|
return fatal
|
|
}, "100ms").Should(BeFalse())
|
|
|
|
Expect(fatal).Should(BeFalse())
|
|
|
|
// start again -> should fail
|
|
server.Start()
|
|
|
|
Eventually(func() bool {
|
|
return fatal
|
|
}, "100ms").Should(BeTrue())
|
|
})
|
|
})
|
|
})
|
|
Describe("Server stop", func() {
|
|
When("Stop is called", func() {
|
|
It("stop was called 2 times, start should fail", func() {
|
|
defer func() { Log().ExitFunc = nil }()
|
|
|
|
var fatal bool
|
|
|
|
Log().ExitFunc = func(int) { fatal = true }
|
|
|
|
// create server
|
|
server, err := NewServer(&config.Config{
|
|
Upstream: config.UpstreamConfig{
|
|
ExternalResolvers: map[string][]config.Upstream{
|
|
"default": {config.Upstream{Net: config.NetProtocolTcpUdp, Host: "4.4.4.4", Port: 53}}}},
|
|
CustomDNS: config.CustomDNSConfig{
|
|
Mapping: config.CustomDNSMapping{
|
|
HostIPs: map[string][]net.IP{
|
|
"custom.lan": {net.ParseIP("192.168.178.55")},
|
|
"lan.home": {net.ParseIP("192.168.178.56")},
|
|
},
|
|
}},
|
|
Blocking: config.BlockingConfig{BlockType: "zeroIp"},
|
|
DNSPorts: config.ListenConfig{"127.0.0.1:55557"},
|
|
})
|
|
|
|
Expect(err).Should(Succeed())
|
|
|
|
// start server
|
|
go func() {
|
|
server.Start()
|
|
}()
|
|
|
|
defer server.Stop()
|
|
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
server.Stop()
|
|
|
|
// stop server, should be ok
|
|
Expect(fatal).Should(BeFalse())
|
|
|
|
// stop again, should raise fatal error
|
|
server.Stop()
|
|
|
|
Expect(fatal).Should(BeTrue())
|
|
})
|
|
})
|
|
})
|
|
|
|
Describe("resolve client IP", func() {
|
|
Context("UDP address", func() {
|
|
It("should correct resolve client IP", func() {
|
|
ip, protocol := resolveClientIPAndProtocol(&net.UDPAddr{IP: net.ParseIP("192.168.178.88")})
|
|
Expect(ip).Should(Equal(net.ParseIP("192.168.178.88")))
|
|
Expect(protocol).Should(Equal(model.RequestProtocolUDP))
|
|
})
|
|
})
|
|
Context("TCP address", func() {
|
|
It("should correct resolve client IP", func() {
|
|
ip, protocol := resolveClientIPAndProtocol(&net.TCPAddr{IP: net.ParseIP("192.168.178.88")})
|
|
Expect(ip).Should(Equal(net.ParseIP("192.168.178.88")))
|
|
Expect(protocol).Should(Equal(model.RequestProtocolTCP))
|
|
})
|
|
})
|
|
})
|
|
|
|
})
|
|
|
|
func requestServer(request *dns.Msg) *dns.Msg {
|
|
conn, err := net.Dial("udp", ":55555")
|
|
if err != nil {
|
|
Log().Fatal("could not connect to server: ", err)
|
|
}
|
|
defer conn.Close()
|
|
|
|
msg, err := request.Pack()
|
|
if err != nil {
|
|
Log().Fatal("can't pack request: ", err)
|
|
}
|
|
|
|
_, err = conn.Write(msg)
|
|
if err != nil {
|
|
Log().Fatal("can't send request to server: ", err)
|
|
}
|
|
|
|
out := make([]byte, 1024)
|
|
|
|
if _, err := conn.Read(out); err == nil {
|
|
response := new(dns.Msg)
|
|
err := response.Unpack(out)
|
|
|
|
if err != nil {
|
|
Log().Fatal("can't unpack response: ", err)
|
|
}
|
|
|
|
return response
|
|
}
|
|
|
|
Log().Fatal("could not read from connection", err)
|
|
|
|
return nil
|
|
}
|