From a187d23e3b7c84f556762d17544e49c89c6a2be0 Mon Sep 17 00:00:00 2001
From: t4skforce <t4skforce1337@gmail.com>
Date: Mon, 27 May 2019 19:35:51 +0200
Subject: [PATCH] improved host lookup

---
 core/checker.go | 10 ++--------
 1 file changed, 2 insertions(+), 8 deletions(-)

diff --git a/core/checker.go b/core/checker.go
index b95451c3..12148969 100644
--- a/core/checker.go
+++ b/core/checker.go
@@ -90,17 +90,11 @@ func (s *Service) parseHost() string {
 	if s.Type == "tcp" || s.Type == "udp" {
 		return s.Domain
 	} else {
-		domain := 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)
+		u, err := url.Parse(s.Domain)
 		if err != nil {
 			return s.Domain
 		}
-		return host.Host
+		return strings.Split(u.Host, ":")[0]
 	}
 }