]> git.sesse.net Git - vlc/blobdiff - src/network/io.c
Removes trailing spaces. Removes tabs.
[vlc] / src / network / io.c
index d22f59537366b91ce8a95b8c5859319e1e4e05cd..e83f985dafb53ff9f0c021d9f5c01e72d1d25bd1 100644 (file)
@@ -97,12 +97,8 @@ int net_Socket (vlc_object_t *p_this, int family, int socktype,
 
 #ifdef IPV6_V6ONLY
     /*
-     * Accepts only IPv6 connections on IPv6 sockets
-     * (and open an IPv4 socket later as well if needed).
-     * Only Linux and FreeBSD can map IPv4 connections on IPv6 sockets,
-     * so this allows for more uniform handling across platforms. Besides,
-     * it makes sure that IPv4 addresses will be printed as w.x.y.z rather
-     * than ::ffff:w.x.y.z
+     * Accepts only IPv6 connections on IPv6 sockets.
+     * If possible, we should open two sockets, but it is not always possible.
      */
     if (family == AF_INET6)
         setsockopt (fd, IPPROTO_IPV6, IPV6_V6ONLY, &(int){ 1 }, sizeof (int));
@@ -215,6 +211,9 @@ int *net_Listen (vlc_object_t *p_this, const char *psz_host,
             case SOCK_STREAM:
             case SOCK_RDM:
             case SOCK_SEQPACKET:
+#ifdef SOCK_DCCP
+            case SOCK_DCCP:
+#endif
                 if (listen (fd, INT_MAX))
                 {
                     msg_Err (p_this, "socket listen error (%s)",
@@ -243,118 +242,90 @@ int *net_Listen (vlc_object_t *p_this, const char *psz_host,
 }
 
 
-int net_ListenSingle (vlc_object_t *obj, const char *host, int port,
-                      int family, int socktype, int protocol)
-{
-    int *fdv = net_Listen (obj, host, port, family, socktype, protocol);
-    if (fdv == NULL)
-        return -1;
-
-    for (unsigned i = 1; fdv[i] != -1; i++)
-    {
-        msg_Warn (obj, "Multiple sockets opened. Dropping extra ones!");
-        net_Close (fdv[i]);
-    }
-
-    int fd = fdv[0];
-    assert (fd != -1);
-
-    free (fdv);
-    return fd;
-}
-
-
-
-/*****************************************************************************
- * __net_Close:
- *****************************************************************************
- * Close a network handle
- *****************************************************************************/
-void net_Close (int fd)
-{
-#ifdef UNDER_CE
-    CloseHandle ((HANDLE)fd);
-#elif defined (WIN32)
-    closesocket (fd);
-#else
-    (void)close (fd);
-#endif
-}
-
-
 static ssize_t
-net_ReadInnervlc_object_t *restrict p_this, unsigned fdc, const int *fdv,
+net_ReadInner (vlc_object_t *restrict p_this, unsigned fdc, const int *fdv,
                const v_socket_t *const *restrict vsv,
-               uint8_t *restrict p_buf, size_t i_buflen,
-               vlc_bool_t dontwait, vlc_bool_t waitall )
+               uint8_t *restrict p_buf, size_t i_buflen, vlc_bool_t waitall)
 {
     size_t i_total = 0;
 
     while (i_buflen > 0)
     {
-        unsigned i;
-        ssize_t n;
-        struct pollfd ufd[fdc];
-
-        int delay_ms = dontwait ? 0 : 500;
-        if (p_this->b_die)
+        if( ( p_this->b_die ) || ( p_this->p_libvlc->b_die ) )
         {
 #if defined(WIN32) || defined(UNDER_CE)
-            WSASetLastError(WSAEINTR);
+            WSASetLastError (WSAEINTR);
 #else
             errno = EINTR;
 #endif
             goto error;
         }
 
-        memset (ufd, 0, sizeof (ufd));
+        struct pollfd ufd[fdc];
 
-        for( i = 0; i < fdc; i++ )
+        for (unsigned i = 0; i < fdc; i++)
         {
             ufd[i].fd = fdv[i];
             ufd[i].events = POLLIN;
+            ufd[i].revents = 0;
         }
 
-        n = poll( ufd, fdc, delay_ms );
-        if( n == -1 )
-            goto error;
-
-        assert ((unsigned)n <= fdc);
+        switch (poll (ufd, fdc, 500))
+        {
+            case -1:
+                goto error;
 
-        if (n == 0) // timeout
-            continue;
+            case 0: // timeout
+                continue;
+        }
 
-        for (i = 0;; i++)
+        for (unsigned i = 0;; i++)
         {
-            if ((i_total > 0) && (ufd[i].revents & POLLERR))
-                return i_total; // error will be dequeued on next run
+            assert (i < fdc); /* no events found = bug ! */
 
-            if ((ufd[i].revents & POLLIN) == 0)
+            if (ufd[i].revents == 0)
                 continue;
 
+#ifndef POLLRDHUP /* This is nice but non-portable */
+# define POLLRDHUP 0
+#endif
+            if (i_total > 0)
+            {
+                // Errors (-1) and EOF (0) will be returned on next run
+                if (ufd[i].revents & (POLLERR|POLLNVAL|POLLRDHUP))
+                    return i_total;
+            }
+            else
+            {
+                if (ufd[i].revents & POLLRDHUP)
+                    return 0; // EOF, read() would yield 0
+            }
+
             fdc = 1;
             fdv += i;
             vsv += i;
+
             break;
         }
 
-        if( (*vsv) != NULL )
+        ssize_t n;
+        if (*vsv != NULL)
         {
-            n = (*vsv)->pf_recv( (*vsv)->p_sys, p_buf, i_buflen );
+            n = (*vsv)->pf_recv ((*vsv)->p_sys, p_buf, i_buflen);
         }
         else
         {
-#if defined(WIN32) || defined(UNDER_CE)
-            n = recv( *fdv, p_buf, i_buflen, 0 );
+#ifdef WIN32
+            n = recv (*fdv, p_buf, i_buflen, 0);
 #else
-            n = read( *fdv, p_buf, i_buflen );
+            n = read (*fdv, p_buf, i_buflen);
 #endif
         }
 
-        if( n == -1 )
+        if (n == -1)
         {
 #if defined(WIN32) || defined(UNDER_CE)
-            switch( WSAGetLastError() )
+            switch (WSAGetLastError ())
             {
                 case WSAEWOULDBLOCK:
                 /* only happens with vs != NULL (SSL) - not really an error */
@@ -365,8 +336,8 @@ net_ReadInner( vlc_object_t *restrict p_this, unsigned fdc, const int *fdv,
                 /* On Win32, recv() fails if the datagram doesn't fit inside
                  * the passed buffer, even though the buffer will be filled
                  * with the first part of the datagram. */
-                    msg_Errp_this, "Receive error: "
-                                     "Increase the mtu size (--mtu option)" );
+                    msg_Err (p_this, "Receive error: "
+                                     "Increase the mtu size (--mtu option)");
                     n = i_buflen;
                     break;
 
@@ -374,26 +345,24 @@ net_ReadInner( vlc_object_t *restrict p_this, unsigned fdc, const int *fdv,
                     goto error;
             }
 #else
-            if( errno == EAGAIN ) /* spurious wake-up (sucks if fdc > 1) */
+            /* spurious wake-up or TLS did not yield any actual data */
+            if (errno == EAGAIN)
                 continue;
             goto error;
 #endif
         }
 
-        if (n == 0) // EOF
-            break;
-
         i_total += n;
         p_buf += n;
         i_buflen -= n;
 
-        if (dontwait || !waitall)
+        if ((n == 0) || !waitall)
             break;
     }
     return i_total;
 
 error:
-    msg_Err( p_this, "Read error: %s", net_strerror (net_errno) );
+    msg_Err (p_this, "Read error: %s", net_strerror (net_errno));
     return i_total ? (ssize_t)i_total : -1;
 }
 
@@ -411,24 +380,7 @@ ssize_t __net_Read( vlc_object_t *restrict p_this, int fd,
 {
     return net_ReadInner( p_this, 1, &(int){ fd },
                           &(const v_socket_t *){ p_vs },
-                          buf, len, VLC_FALSE, b_retry );
-}
-
-
-/*****************************************************************************
- * __net_ReadNonBlock:
- *****************************************************************************
- * Read from a network socket, non blocking mode.
- * This function should only be used after a poll() (or select(), but you
- * should use poll instead of select()) invocation to avoid busy loops.
- *****************************************************************************/
-ssize_t __net_ReadNonBlock( vlc_object_t *restrict p_this, int fd,
-                            const v_socket_t *restrict p_vs,
-                            uint8_t *restrict buf, size_t len )
-{
-    return net_ReadInner (p_this, 1, &(int){ fd },
-                          &(const v_socket_t *){ p_vs },
-                          buf, len, VLC_TRUE, VLC_FALSE);
+                          buf, len, b_retry );
 }
 
 
@@ -445,7 +397,7 @@ ssize_t __net_Select( vlc_object_t *restrict p_this,
     memset( vsv, 0, sizeof (vsv) );
 
     return net_ReadInner( p_this, nfd, fds, vsv,
-                          buf, len, VLC_FALSE, VLC_FALSE );
+                          buf, len, VLC_FALSE );
 }
 
 
@@ -476,13 +428,13 @@ ssize_t __net_Write( vlc_object_t *p_this, int fd, const v_socket_t *p_vs,
                 continue;
         }
 
-        if ((ufd[0].revents & POLLERR) && (i_total > 0))
+        if ((ufd[0].revents & (POLLERR|POLLNVAL|POLLHUP)) && (i_total > 0))
             return i_total; // error will be dequeued separately on next call
 
         if (p_vs != NULL)
             val = p_vs->pf_send (p_vs->p_sys, p_data, i_data);
         else
-#if defined(WIN32) || defined(UNDER_CE)
+#ifdef WIN32
             val = send (fd, p_data, i_data, 0);
 #else
             val = write (fd, p_data, i_data);
@@ -493,8 +445,6 @@ ssize_t __net_Write( vlc_object_t *p_this, int fd, const v_socket_t *p_vs,
             msg_Err (p_this, "Write error: %s", net_strerror (net_errno));
             break;
         }
-        if (val == 0)
-            return i_total;
 
         p_data += val;
         i_data -= val;
@@ -655,8 +605,8 @@ const char *inet_ntop(int af, const void * src,
                 memset(&addr, 0, sizeof(addr));
                 addr.sin6_family = AF_INET6;
                 addr.sin6_addr = *((struct in6_addr*)src);
-                if( 0 == WSAAddressToStringA((LPSOCKADDR)&addr, 
-                                             sizeof(struct sockaddr_in6), 
+                if( 0 == WSAAddressToStringA((LPSOCKADDR)&addr,
+                                             sizeof(struct sockaddr_in6),
                                              NULL, dst, &cnt) )
                 {
                     dst[cnt] = '\0';