]> git.sesse.net Git - vlc/blobdiff - src/network/tcp.c
misc/objects.c: Don't rely on vlc_object_destroy() to destroy objects, but expects...
[vlc] / src / network / tcp.c
index c7766f9575f0caaeb0ef3643304ddb7314b709d2..2dbaa64487d2617177c809ff9422da131758b08d 100644 (file)
 /*****************************************************************************
  * Preamble
  *****************************************************************************/
+#ifdef HAVE_CONFIG_H
+# include "config.h"
+#endif
+
 #include <vlc/vlc.h>
 
 #include <errno.h>
+#include <assert.h>
 
 #ifdef HAVE_FCNTL_H
 #   include <fcntl.h>
@@ -76,9 +81,14 @@ int __net_Connect( vlc_object_t *p_this, const char *psz_host, int i_port,
     char            *psz_socks;
     int             i_realport, i_val, i_handle = -1;
 
+    int evfd = vlc_object_waitpipe (p_this);
+    if (evfd == -1)
+        return -1;
+
     if( i_port == 0 )
         i_port = 80; /* historical VLC thing */
 
+
     memset( &hints, 0, sizeof( hints ) );
     hints.ai_socktype = SOCK_STREAM;
 
@@ -153,9 +163,7 @@ int __net_Connect( vlc_object_t *p_this, const char *psz_host, int i_port,
 
         if( connect( fd, ptr->ai_addr, ptr->ai_addrlen ) )
         {
-            socklen_t i_val_size = sizeof( i_val );
-            div_t d;
-            vlc_value_t timeout;
+            int timeout, val;
 
             if( net_errno != EINPROGRESS )
             {
@@ -164,62 +172,47 @@ int __net_Connect( vlc_object_t *p_this, const char *psz_host, int i_port,
             }
             msg_Dbg( p_this, "connection: %m" );
 
-            var_Create( p_this, "ipv4-timeout",
-                        VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
-            var_Get( p_this, "ipv4-timeout", &timeout );
-            if( timeout.i_int < 0 )
+            timeout = var_CreateGetInteger (p_this, "ipv4-timeout");
+            if (timeout < 0)
             {
                 msg_Err( p_this, "invalid negative value for ipv4-timeout" );
-                timeout.i_int = 0;
+                timeout = 0;
             }
-            d = div( timeout.i_int, 100 );
 
-            for (;;)
+            struct pollfd ufd[2] = {
+                { .fd = fd,   .events = POLLOUT },
+                { .fd = evfd, .events = POLLIN },
+            };
+
+            do
+                /* NOTE: timeout screwed up if we catch a signal (EINTR) */
+                val = poll (ufd, sizeof (ufd) / sizeof (ufd[0]), timeout);
+            while ((val == -1) && (net_errno == EINTR));
+
+            switch (val)
             {
-                struct pollfd ufd = { .fd = fd, .events = POLLOUT };
-                int i_ret;
-
-                if( p_this->b_die )
-                {
-                    msg_Dbg( p_this, "connection aborted" );
-                    net_Close( fd );
-                    vlc_freeaddrinfo( res );
-                    return -1;
-                }
-
-                /*
-                 * We'll wait 0.1 second if nothing happens
-                 * NOTE:
-                 * time out will be shortened if we catch a signal (EINTR)
-                 */
-                i_ret = poll (&ufd, 1, (d.quot > 0) ? 100 : d.rem);
-                if( i_ret == 1 )
-                    break;
-
-                if( ( i_ret == -1 ) && ( net_errno != EINTR ) )
-                {
-                    msg_Err( p_this, "connection polling error: %m" );
-                    goto next_ai;
-                }
-
-                if( d.quot <= 0 )
-                {
-                    msg_Warn( p_this, "connection timed out" );
-                    goto next_ai;
-                }
-
-                d.quot--;
+                 case -1: /* error */
+                     msg_Err (p_this, "connection polling error: %m");
+                     goto next_ai;
+
+                 case 0: /* timeout */
+                     msg_Warn (p_this, "connection timed out");
+                     goto next_ai;
+
+                 default: /* something happended */
+                     if (ufd[1].revents)
+                         goto next_ai; /* LibVLC object killed */
             }
 
-#if !defined( SYS_BEOS ) && !defined( UNDER_CE )
-            if( getsockopt( fd, SOL_SOCKET, SO_ERROR, (void*)&i_val,
-                            &i_val_size ) == -1 || i_val != 0 )
+            /* There is NO WAY around checking SO_ERROR.
+             * Don't ifdef it out!!! */
+            if (getsockopt (fd, SOL_SOCKET, SO_ERROR, &val,
+                            &(socklen_t){ sizeof (val) }) || val)
             {
-                errno = i_val;
-                msg_Err( p_this, "connection failed: %m" );
+                errno = val;
+                msg_Err (p_this, "connection failed: %m");
                 goto next_ai;
             }
-#endif
         }
 
         msg_Dbg( p_this, "connection succeeded (socket = %d)", fd );
@@ -258,6 +251,22 @@ next_ai: /* failure */
 }
 
 
+int net_AcceptSingle (vlc_object_t *obj, int lfd)
+{
+    int fd = accept (lfd, NULL, NULL);
+    if (fd == -1)
+    {
+        if (net_errno != EAGAIN)
+            msg_Err (obj, "accept failed (from socket %d): %m", lfd);
+        return -1;
+    }
+
+    msg_Dbg (obj, "accepted socket %d (from socket %d)", fd, lfd);
+    net_SetupSocket (fd);
+    return fd;
+}
+
+
 /*****************************************************************************
  * __net_Accept:
  *****************************************************************************
@@ -266,14 +275,14 @@ next_ai: /* failure */
 int __net_Accept( vlc_object_t *p_this, int *pi_fd, mtime_t i_wait )
 {
     int timeout = (i_wait < 0) ? -1 : i_wait / 1000;
-    int evfd;
+    int evfd = vlc_object_waitpipe (p_this);
 
-    assert( pi_fd != NULL );
+    if (evfd == -1)
+        return -1;
 
-    vlc_object_lock (p_this);
-    evfd = vlc_object_waitpipe (p_this);
+    assert( pi_fd != NULL );
 
-    while (vlc_object_alive (p_this))
+    for (;;)
     {
         unsigned n = 0;
         while (pi_fd[n] != -1)
@@ -287,8 +296,9 @@ int __net_Accept( vlc_object_t *p_this, int *pi_fd, mtime_t i_wait )
             ufd[i].events = POLLIN;
             ufd[i].revents = 0;
         }
+        if (evfd == -1)
+            n--; /* avoid EBADF */
 
-        vlc_object_unlock (p_this);
         switch (poll (ufd, n, timeout))
         {
             case -1:
@@ -297,11 +307,9 @@ int __net_Accept( vlc_object_t *p_this, int *pi_fd, mtime_t i_wait )
             case 0:
                 return -1; /* NOTE: p_this already unlocked */
         }
-        vlc_object_lock (p_this);
 
         if (ufd[n].revents)
         {
-            vlc_object_wait (p_this);
             errno = EINTR;
             break;
         }
@@ -312,13 +320,9 @@ int __net_Accept( vlc_object_t *p_this, int *pi_fd, mtime_t i_wait )
                 continue;
 
             int sfd = ufd[i].fd;
-            int fd = accept (sfd, NULL, NULL);
+            int fd = net_AcceptSingle (p_this, sfd);
             if (fd == -1)
-            {
-                msg_Err (p_this, "accept failed (%m)");
                 continue;
-            }
-            net_SetupSocket (fd);
 
             /*
              * Move listening socket to the end to let the others in the
@@ -326,12 +330,9 @@ int __net_Accept( vlc_object_t *p_this, int *pi_fd, mtime_t i_wait )
              */
             memmove (pi_fd + i, pi_fd + i + 1, n - (i + 1));
             pi_fd[n - 1] = sfd;
-            vlc_object_unlock (p_this);
-            msg_Dbg (p_this, "accepted socket %d (from socket %d)", fd, sfd);
             return fd;
         }
     }
-    vlc_object_unlock (p_this);
     return -1;
 }