Merge pull request #22 from TrekkieCoder/main

Use given hint to select laddr in case of tcp probes
loxilb-io · Aug 26, 2024 · 6929c7b · 6929c7b
2 parents 7ac692a + 4bf2b99
commit 6929c7b
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/serviceprobe.go b/serviceprobe.go
@@ -104,7 +104,20 @@ func L4ServiceProber(sType string, sName string, sHint, req, resp string) bool {
 		return sOk
 	}
 
-	c, err := net.DialTimeout(sType, sName, timeout)
+	var c net.Conn
+	if sHint == "" {
+		c, err = net.DialTimeout(sType, sName, timeout)
+	} else {
+		dialer := &net.Dialer{
+			LocalAddr: &net.TCPAddr{
+				IP:   net.ParseIP(sHint),
+				Port: 0,
+			},
+		}
+		ctx, cancel := context.WithTimeout(context.Background(), 300*time.Millisecond)
+		defer cancel()
+		c, err = dialer.DialContext(ctx, sType, sName)
+	}
 	if err != nil {
 		sOk = false
 	} else {