Merge pull request #29 from HorayNarea/master

use net.JoinHostPort() instead of string-join directly
This commit is contained in:
kenshin 2017-02-08 10:20:36 +08:00 committed by GitHub
commit 7d1f6af220
2 changed files with 5 additions and 4 deletions

View File

@ -2,6 +2,7 @@ package main
import (
"fmt"
"net"
"strings"
"sync"
"time"
@ -93,9 +94,9 @@ func (r *Resolver) Lookup(net string, req *dns.Msg) (message *dns.Msg, err error
func (r *Resolver) Nameservers() (ns []string) {
for _, server := range r.config.Servers {
if i := strings.IndexByte(server, '#'); i > 0 {
server = server[:i] + ":" + server[i+1:]
server = net.JoinHostPort(server[:i], server[i+1:])
} else {
server = server + ":" + r.config.Port
server = net.JoinHostPort(server, r.config.Port)
}
ns = append(ns, server)
}

View File

@ -1,6 +1,7 @@
package main
import (
"net"
"strconv"
"time"
@ -15,11 +16,10 @@ type Server struct {
}
func (s *Server) Addr() string {
return s.host + ":" + strconv.Itoa(s.port)
return net.JoinHostPort(s.host, strconv.Itoa(s.port))
}
func (s *Server) Run() {
Handler := NewHandler()
tcpHandler := dns.NewServeMux()