From d54007f79b364555137cfdcb24dfea2a7ab26711 Mon Sep 17 00:00:00 2001
From: YuQing <384681@qq.com>
Date: Sun, 25 Oct 2020 20:42:53 +0800
Subject: [PATCH] check poll result for POLLHUP and POLLERR

---
 src/sockopt.c | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/sockopt.c b/src/sockopt.c
index 2fba045..c8a8c60 100644
--- a/src/sockopt.c
+++ b/src/sockopt.c
@@ -177,11 +177,11 @@ int tcprecvdata_ex(int sock, void *data, const int size, \
 		}
 #else
 		res = poll(&pollfds, 1, 1000 * timeout);
-		if (pollfds.revents & POLLHUP)
-		{
-			ret_code = ENOTCONN;
-			break;
-		}
+		if (res > 0 && (pollfds.revents & (POLLHUP | POLLERR)))
+        {
+            ret_code = ENOTCONN;
+            break;
+        }
 #endif
 
 		if (res < 0)
@@ -265,7 +265,7 @@ int tcpsenddata(int sock, void* data, const int size, const int timeout)
 		}
 #else
 		result = poll(&pollfds, 1, 1000 * timeout);
-		if (pollfds.revents & POLLHUP)
+		if (result > 0 && (pollfds.revents & (POLLHUP | POLLERR)))
 		{
 			return ENOTCONN;
 		}
@@ -371,7 +371,7 @@ int tcprecvdata_nb_ms(int sock, void *data, const int size, \
 		}
 #else
 		res = poll(&pollfds, 1, timeout_ms);
-		if (pollfds.revents & POLLHUP)
+		if (res > 0 && (pollfds.revents & (POLLHUP | POLLERR)))
 		{
 			ret_code = ENOTCONN;
 			break;
@@ -455,7 +455,7 @@ int tcpsenddata_nb(int sock, void* data, const int size, const int timeout)
 		}
 #else
 		result = poll(&pollfds, 1, 1000 * timeout);
-		if (pollfds.revents & POLLHUP)
+		if (result > 0 && (pollfds.revents & (POLLHUP | POLLERR)))
 		{
 			return ENOTCONN;
 		}