Commit 590701d6 authored by Rémi Denis-Courmont's avatar Rémi Denis-Courmont

Remove net_Select. Simplify net_Read consequently.

parent 8c906d60
...@@ -123,9 +123,6 @@ struct virtual_socket_t ...@@ -123,9 +123,6 @@ struct virtual_socket_t
#define net_Read(a,b,c,d,e,f) __net_Read(VLC_OBJECT(a),b,c,d,e,f) #define net_Read(a,b,c,d,e,f) __net_Read(VLC_OBJECT(a),b,c,d,e,f)
VLC_EXPORT( ssize_t, __net_Read, ( vlc_object_t *p_this, int fd, const v_socket_t *, uint8_t *p_data, size_t i_data, vlc_bool_t b_retry ) ); VLC_EXPORT( ssize_t, __net_Read, ( vlc_object_t *p_this, int fd, const v_socket_t *, uint8_t *p_data, size_t i_data, vlc_bool_t b_retry ) );
#define net_Select(a,b,c,d,e) __net_Select(VLC_OBJECT(a),b,c,d,e)
VLC_EXPORT( ssize_t, __net_Select, ( vlc_object_t *p_this, const int *pi_fd, int i_fd, uint8_t *p_data, size_t i_data ) );
#define net_Write(a,b,c,d,e) __net_Write(VLC_OBJECT(a),b,c,d,e) #define net_Write(a,b,c,d,e) __net_Write(VLC_OBJECT(a),b,c,d,e)
VLC_EXPORT( ssize_t, __net_Write, ( vlc_object_t *p_this, int fd, const v_socket_t *, const uint8_t *p_data, size_t i_data ) ); VLC_EXPORT( ssize_t, __net_Write, ( vlc_object_t *p_this, int fd, const v_socket_t *, const uint8_t *p_data, size_t i_data ) );
......
...@@ -174,7 +174,6 @@ net_ListenClose ...@@ -174,7 +174,6 @@ net_ListenClose
__net_OpenDgram __net_OpenDgram
net_Printf net_Printf
__net_Read __net_Read
__net_Select
net_SetCSCov net_SetCSCov
__net_vaPrintf __net_vaPrintf
__net_Write __net_Write
......
...@@ -253,12 +253,22 @@ int *net_Listen (vlc_object_t *p_this, const char *psz_host, ...@@ -253,12 +253,22 @@ int *net_Listen (vlc_object_t *p_this, const char *psz_host,
} }
static ssize_t /*****************************************************************************
net_ReadInner (vlc_object_t *restrict p_this, unsigned fdc, const int *fdv, * __net_Read:
const v_socket_t *const *restrict vsv, *****************************************************************************
* Read from a network socket
* If waitall is true, then we repeat until we have read the right amount of
* data; in that case, a short count means EOF has been reached.
*****************************************************************************/
ssize_t
__net_Read (vlc_object_t *restrict p_this, int fd, const v_socket_t *vs,
uint8_t *restrict p_buf, size_t i_buflen, vlc_bool_t waitall) uint8_t *restrict p_buf, size_t i_buflen, vlc_bool_t waitall)
{ {
size_t i_total = 0; size_t i_total = 0;
struct pollfd ufd[1];
ufd[0].fd = fd;
ufd[0].events = POLLIN;
while (i_buflen > 0) while (i_buflen > 0)
{ {
...@@ -272,16 +282,9 @@ net_ReadInner (vlc_object_t *restrict p_this, unsigned fdc, const int *fdv, ...@@ -272,16 +282,9 @@ net_ReadInner (vlc_object_t *restrict p_this, unsigned fdc, const int *fdv,
goto error; goto error;
} }
struct pollfd ufd[fdc]; ufd[0].revents = 0;
/* TODO: don't use arbitrary timer just for b_die */
for (unsigned i = 0; i < fdc; i++) switch (poll (ufd, sizeof (ufd) / sizeof (ufd[0]), 500))
{
ufd[i].fd = fdv[i];
ufd[i].events = POLLIN;
ufd[i].revents = 0;
}
switch (poll (ufd, fdc, 500))
{ {
case -1: case -1:
goto error; goto error;
...@@ -290,64 +293,41 @@ net_ReadInner (vlc_object_t *restrict p_this, unsigned fdc, const int *fdv, ...@@ -290,64 +293,41 @@ net_ReadInner (vlc_object_t *restrict p_this, unsigned fdc, const int *fdv,
continue; continue;
} }
for (unsigned i = 0;; i++) assert (ufd[0].revents);
{
assert (i < fdc); /* no events found = bug ! */
if (ufd[i].revents == 0)
continue;
#ifndef POLLRDHUP /* This is nice but non-portable */ #ifndef POLLRDHUP /* This is nice but non-portable */
# define POLLRDHUP 0 # define POLLRDHUP 0
#endif #endif
if (i_total > 0) if (i_total > 0)
{ {
// Errors (-1) and EOF (0) will be returned on next run /* Errors (-1) and EOF (0) will be returned on next call,
if (ufd[i].revents & (POLLERR|POLLNVAL|POLLRDHUP)) * otherwise we'd "hide" the error from the caller, which is a
* bad idea™. */
if (ufd[0].revents & (POLLERR|POLLNVAL|POLLRDHUP))
return i_total; return i_total;
} }
else
{
if (ufd[i].revents & POLLRDHUP)
return 0; // EOF, read() would yield 0
}
fdc = 1;
fdv += i;
vsv += i;
break;
}
ssize_t n; ssize_t n;
if (*vsv != NULL) if (vs != NULL)
{ {
n = (*vsv)->pf_recv ((*vsv)->p_sys, p_buf, i_buflen); n = vs->pf_recv (vs->p_sys, p_buf, i_buflen);
} }
else else
{ {
#ifdef WIN32 #ifdef WIN32
n = recv (*fdv, p_buf, i_buflen, 0); n = recv (fd, p_buf, i_buflen, 0);
#else #else
n = read (*fdv, p_buf, i_buflen); n = read (fd, p_buf, i_buflen);
#endif #endif
} }
if (n == 0)
/* For streams, this means end of file, and there will not be any
* further data ever on the stream. For datagram sockets, this
* means empty datagram, and there could be more data coming.
* However, it makes no sense to set <waitall> with datagrams.
*/
break; // EOF
if (n == -1) if (n == -1)
{ {
#if defined(WIN32) || defined(UNDER_CE) #if defined(WIN32) || defined(UNDER_CE)
switch (WSAGetLastError ()) switch (WSAGetLastError ())
{ {
case WSAEWOULDBLOCK: case WSAEWOULDBLOCK:
/* only happens with vs != NULL (SSL) - not really an error */ /* only happens with vs != NULL (TLS) - not really an error */
continue; continue;
case WSAEMSGSIZE: case WSAEMSGSIZE:
...@@ -371,6 +351,15 @@ net_ReadInner (vlc_object_t *restrict p_this, unsigned fdc, const int *fdv, ...@@ -371,6 +351,15 @@ net_ReadInner (vlc_object_t *restrict p_this, unsigned fdc, const int *fdv,
#endif #endif
} }
if (n == 0)
/* For streams, this means end of file, and there will not be any
* further data ever on the stream. For datagram sockets, this
* means empty datagram, and there could be more data coming.
* However, it makes no sense to set <waitall> with datagrams in the
* first place.
*/
break; // EOF
i_total += n; i_total += n;
p_buf += n; p_buf += n;
i_buflen -= n; i_buflen -= n;
...@@ -386,40 +375,6 @@ error: ...@@ -386,40 +375,6 @@ error:
} }
/*****************************************************************************
* __net_Read:
*****************************************************************************
* Read from a network socket
* If b_retry is true, then we repeat until we have read the right amount of
* data; in that case, a short count means EOF has been reached.
*****************************************************************************/
ssize_t __net_Read( vlc_object_t *restrict p_this, int fd,
const v_socket_t *restrict p_vs,
uint8_t *restrict buf, size_t len, vlc_bool_t b_retry )
{
return net_ReadInner( p_this, 1, &(int){ fd },
&(const v_socket_t *){ p_vs },
buf, len, b_retry );
}
/*****************************************************************************
* __net_Select:
*****************************************************************************
* Read from several sockets. Takes data from the first socket that has some.
*****************************************************************************/
ssize_t __net_Select( vlc_object_t *restrict p_this,
const int *restrict fds, int nfd,
uint8_t *restrict buf, size_t len )
{
const v_socket_t *vsv[nfd];
memset( vsv, 0, sizeof (vsv) );
return net_ReadInner( p_this, nfd, fds, vsv,
buf, len, VLC_FALSE );
}
/* Write exact amount requested */ /* Write exact amount requested */
ssize_t __net_Write( vlc_object_t *p_this, int fd, const v_socket_t *p_vs, ssize_t __net_Write( vlc_object_t *p_this, int fd, const v_socket_t *p_vs,
const uint8_t *p_data, size_t i_data ) const uint8_t *p_data, size_t i_data )
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment