improved host lookup

pull/194/head
t4skforce 2019-05-27 19:35:51 +02:00
parent 1950439b78
commit a187d23e3b
1 changed files with 2 additions and 8 deletions

View File

@ -90,17 +90,11 @@ func (s *Service) parseHost() string {
if s.Type == "tcp" || s.Type == "udp" { if s.Type == "tcp" || s.Type == "udp" {
return s.Domain return s.Domain
} else { } else {
domain := s.Domain u, err := url.Parse(s.Domain)
hasPort, _ := regexp.MatchString(`\:([0-9]+)`, domain)
if hasPort {
splitDomain := strings.Split(s.Domain, ":")
domain = splitDomain[len(splitDomain)-2]
}
host, err := url.Parse(domain)
if err != nil { if err != nil {
return s.Domain return s.Domain
} }
return host.Host return strings.Split(u.Host, ":")[0]
} }
} }