From 366eb2e149ff1b1d3075d1e8279bcc71cf398a6b Mon Sep 17 00:00:00 2001
From: Yonghong Song <yhs@plumgrid.com>
Date: Wed, 7 Oct 2015 08:59:42 -0700
Subject: [PATCH] handle return value in kretprobe handler for tcpv4connect

Signed-off-by: Yonghong Song <yhs@plumgrid.com>
---
 tools/tcpv4connect | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tools/tcpv4connect b/tools/tcpv4connect
index fdada97c20b9..67a2894034c3 100755
--- a/tools/tcpv4connect
+++ b/tools/tcpv4connect
@@ -61,6 +61,13 @@ int kretprobe__tcp_v4_connect(struct pt_regs *ctx)
 		return 0;	// missed entry
 	}
 
+	if (ret != 0) {
+		// failed to send SYNC packet, socket __sk_common.{skc_rcv_saddr, ...}
+		// may not be populated properly.
+		currsock.delete(&pid);
+		return 0;
+	}
+
 	// pull in details
 	struct sock *skp = *skpp;
 	u32 saddr = 0, daddr = 0;