check for EINTR

2 files changed, 9 insertions, 4 deletions

diff --git a/unix.c b/unix.c
index 6669216..a7bd048 100644
--- a/unix.c
+++ b/unix.c
@@ -501,10 +501,14 @@ enet_socket_receive (ENetSocket socket,
 
     if (recvLength == -1)
     {
-       if (errno == EWOULDBLOCK)
-         return 0;
-
-       return -1;
+        switch (errno)
+        {
+            case EWOULDBLOCK:
+            case EINTR:
+                return 0;
+            default:
+                return -1;
+        }
     }
 
 #ifdef HAS_MSGHDR_FLAGS
diff --git a/win32.c b/win32.c
index f395c09..462e242 100644
--- a/win32.c
+++ b/win32.c
@@ -381,6 +381,7 @@ enet_socket_receive (ENetSocket socket,
        {
        case WSAEWOULDBLOCK:
        case WSAECONNRESET:
+       case WSAEINTR:
           return 0;
        case WSAEMSGSIZE:
           return -2;