utils: tvh_write - add special socket error check

tvheadend · Jun 2, 2015 · 2b31be5 · 2b31be5
1 parent 2fdfe48
commit 2b31be5
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/wrappers.c b/src/wrappers.c
@@ -3,6 +3,7 @@
 #include <fcntl.h>
 #include <sys/types.h>          /* See NOTES */
 #include <sys/socket.h>
+#include <sys/stat.h>
 #include <unistd.h>
 #include <signal.h>
 #include <pthread.h>
@@ -72,11 +73,19 @@ int
 tvh_write(int fd, const void *buf, size_t len)
 {
   ssize_t c;
+  struct stat st;
+  int err;
+  socklen_t errlen;
 
   while (len) {
     c = write(fd, buf, len);
     if (c < 0) {
       if (ERRNO_AGAIN(errno)) {
+        fstat(fd, &st);
+        errlen = sizeof(err);
+        if (S_ISSOCK(st.st_mode) &&
+            (getsockopt(fd, SOL_SOCKET, SO_ERROR, (char *)&err, &errlen) || err))
+          break;
         usleep(100);
         continue;
       }