Implemented tls probe #173

Author: illarion

config/gobetween.toml

@@ -210,7 +210,7 @@ protocol = "udp"
 #  ## could be also matched using regular expression.
 #  ##
 #  kind = "probe"
-#  probe_protocol = "udp"         # (required) "udp" | "tcp"
+#  probe_protocol = "udp"         # (required) "udp" | "tcp" | "tls"
 #  probe_strategy = "starts_with" # (optional) "starts_with" | "regexp"
 #  probe_send    = 'PING\xFF\n'   # (required) string to send
 #  probe_recv    = 'PONG\xAA\n'   # (required) string to expect in response (response starts with or regexp)

src/healthcheck/probe.go

@@ -1,14 +1,15 @@
 package healthcheck
 
 /**
- * probe.go - TCP/UDP probe healthcheck
+ * probe.go - TCP/UDP and TLS probe healthcheck
  *
  * @author Yousong Zhou <[email protected]>
  * @author Illarion Kovalchuk <[email protected]>
  */
 
 import (
 	"bytes"
+	"crypto/tls"
 	"io"
 	"net"
 	"regexp"
@@ -37,7 +38,17 @@ func probe(t core.Target, cfg config.HealthcheckConfig, result chan<- CheckResul
 		}
 	}()
 
-	conn, err := net.DialTimeout(cfg.ProbeProtocol, t.Address(), timeout)
+	var conn net.Conn
+	var err error
+
+	switch cfg.ProbeProtocol {
+	case "tls":
+		conn, err = tls.DialWithDialer(&net.Dialer{
+			Timeout: timeout,
+		}, "tcp", t.Address(), &tls.Config{})
+	default:
+		conn, err = net.DialTimeout(cfg.ProbeProtocol, t.Address(), timeout)
+	}
 	if err != nil {
 		checkResult.Live = false
 		return

src/manager/manager.go

@@ -316,7 +316,7 @@ func prepareConfig(name string, server config.Server, defaults config.Connection
 	if server.Healthcheck.Kind == "probe" {
 
 		switch server.Healthcheck.ProbeProtocol {
-		case "tcp", "udp":
+		case "tcp", "udp", "tls":
 		default:
 			return config.Server{}, errors.New("Unsupported probe_protocol")
 		}