Fix linter warnings and go fmt

This commit is contained in:
George
2020-09-23 14:49:57 -04:00
committed by George
parent ae6bd725d6
commit 1f68d1b213
5 changed files with 25 additions and 25 deletions
-2
View File
@@ -10,7 +10,6 @@ import (
"github.com/miekg/dns"
)
func GenConf(target, zone string) (string, error) {
ip := net.ParseIP(target)
if ip == nil {
@@ -50,4 +49,3 @@ func ParseZoneConfigMap(zoneConfigMap map[string][]string) (map[string][]dns.RR,
}
return zoneConfig, nil
}
+3 -3
View File
@@ -1,10 +1,10 @@
package meshname
import (
import (
"bytes"
"fmt"
"net"
"testing"
"fmt"
"github.com/zhoreeq/meshname/pkg/meshname"
)
@@ -16,7 +16,7 @@ func TestIPFromDomain(t *testing.T) {
if ip, err := meshname.IPFromDomain(&test_subdomain); err != nil {
t.Fatal(err)
} else if bytes.Compare(ip, test_ip) != 0 {
t.Fatalf("Decoding IP error %s != %s", ip.String(), test_ip.String())
t.Fatalf("Decoding IP error %s != %s", ip.String(), test_ip.String())
}
}
+13 -11
View File
@@ -30,24 +30,23 @@ func New(log *log.Logger, listenAddr string) *MeshnameServer {
dnsClient.Timeout = 5000000000 // increased 5 seconds timeout
return &MeshnameServer{
log: log,
log: log,
listenAddr: listenAddr,
zoneConfig: make(map[string][]dns.RR),
networks: make(map[string]*net.IPNet),
dnsClient: dnsClient,
networks: make(map[string]*net.IPNet),
dnsClient: dnsClient,
}
}
func (s *MeshnameServer) Stop() error {
func (s *MeshnameServer) Stop() {
s.startedLock.Lock()
defer s.startedLock.Unlock()
if s.started == true {
s.dnsServer.Shutdown()
if s.started {
if err := s.dnsServer.Shutdown(); err != nil {
s.log.Debugln(err)
}
s.started = false
return nil
} else {
return errors.New("MeshnameServer is not running")
}
}
@@ -55,7 +54,7 @@ func (s *MeshnameServer) Start() error {
s.startedLock.Lock()
defer s.startedLock.Unlock()
if s.started == false {
if !s.started {
s.dnsServer = &dns.Server{Addr: s.listenAddr, Net: "udp"}
for tld, subnet := range s.networks {
dns.HandleFunc(tld, s.handleRequest)
@@ -129,7 +128,10 @@ func (s *MeshnameServer) handleRequest(w dns.ResponseWriter, r *dns.Msg) {
}
m.Answer = append(m.Answer, resp.Answer...)
}
w.WriteMsg(m)
if err := w.WriteMsg(m); err != nil {
s.log.Debugln("Error writing response:", err)
}
}
func (s *MeshnameServer) isRemoteLookupAllowed(addr net.Addr) bool {
+1 -2
View File
@@ -23,7 +23,7 @@ func TestServerLocalDomain(t *testing.T) {
exampleConfig := make(map[string][]string)
exampleConfig["aiarnf2wpqjxkp6rhivuxbondy"] = append(exampleConfig["aiarnf2wpqjxkp6rhivuxbondy"],
"test.aiarnf2wpqjxkp6rhivuxbondy.meshname. AAAA 201:1697:567c:1375:3fd1:3a2b:4b85:cd1e")
"test.aiarnf2wpqjxkp6rhivuxbondy.meshname. AAAA 201:1697:567c:1375:3fd1:3a2b:4b85:cd1e")
if zoneConfig, err := meshname.ParseZoneConfigMap(exampleConfig); err == nil {
ts.SetZoneConfig(zoneConfig)
@@ -31,7 +31,6 @@ func TestServerLocalDomain(t *testing.T) {
t.Fatalf("meshname: Failed to parse Meshname config: %s", err)
}
ts.Start()
tc := new(dns.Client)