gst/tcp/gstfdset.c: Add more locking and bounds checking.

Original commit message from CVS:
* gst/tcp/gstfdset.c: (gst_fdset_add_fd), (gst_fdset_remove_fd),
(gst_fdset_fd_ctl_write), (gst_fdset_fd_ctl_read),
(gst_fdset_fd_has_closed), (gst_fdset_fd_has_error),
(gst_fdset_fd_can_read), (gst_fdset_fd_can_write),
(gst_fdset_wait):
Add more locking and bounds checking.
This commit is contained in:
Wim Taymans 2004-08-18 11:57:56 +00:00
parent d5c88bc058
commit 36db5cb890
2 changed files with 50 additions and 12 deletions

View file

@ -1,3 +1,18 @@
2004-08-18 Wim Taymans <wim@fluendo.com>
* gst/tcp/gstfdset.c: (gst_fdset_add_fd), (gst_fdset_remove_fd),
(gst_fdset_fd_ctl_write), (gst_fdset_fd_ctl_read),
(gst_fdset_fd_has_closed), (gst_fdset_fd_has_error),
(gst_fdset_fd_can_read), (gst_fdset_fd_can_write),
(gst_fdset_wait):
Add more locking and bounds checking.
2004-08-18 Wim Taymans <wim@fluendo.com>
* gst/tcp/gstfdset.c: (ensure_size), (gst_fdset_wait):
Realloc test fdset in the lock and right before starting
the poll call. Bump the limit to 4096.
2004-08-17 David Schleef <ds@schleef.org> 2004-08-17 David Schleef <ds@schleef.org>
* sys/sunaudio/Makefile.am: * sys/sunaudio/Makefile.am:

View file

@ -55,6 +55,7 @@ struct _GstFDSet
/* for poll */ /* for poll */
struct pollfd *testpollfds; struct pollfd *testpollfds;
gint last_testpollfds;
gint testsize; gint testsize;
struct pollfd *pollfds; struct pollfd *pollfds;
@ -183,7 +184,7 @@ gst_fdset_add_fd (GstFDSet * set, GstFD * fd)
nfd = &set->pollfds[idx]; nfd = &set->pollfds[idx];
nfd->fd = fd->fd; nfd->fd = fd->fd;
nfd->events = 0; nfd->events = POLLERR | POLLNVAL | POLLHUP;
nfd->revents = 0; nfd->revents = 0;
/* see if we have one fd more */ /* see if we have one fd more */
@ -211,6 +212,7 @@ gst_fdset_remove_fd (GstFDSet * set, GstFD * fd)
case GST_FDSET_MODE_POLL: case GST_FDSET_MODE_POLL:
{ {
g_mutex_lock (set->poll_lock); g_mutex_lock (set->poll_lock);
set->pollfds[fd->idx].fd = -1; set->pollfds[fd->idx].fd = -1;
set->pollfds[fd->idx].events = 0; set->pollfds[fd->idx].events = 0;
set->pollfds[fd->idx].revents = 0; set->pollfds[fd->idx].revents = 0;
@ -246,7 +248,14 @@ gst_fdset_fd_ctl_write (GstFDSet * set, GstFD * fd, gboolean active)
break; break;
case GST_FDSET_MODE_POLL: case GST_FDSET_MODE_POLL:
{ {
set->pollfds[fd->idx].events = (active ? POLLOUT : 0); gint events = set->pollfds[fd->idx].events;
if (active)
events |= POLLOUT;
else
events &= ~POLLOUT;
set->pollfds[fd->idx].events = events;
break; break;
} }
case GST_FDSET_MODE_EPOLL: case GST_FDSET_MODE_EPOLL:
@ -266,7 +275,14 @@ gst_fdset_fd_ctl_read (GstFDSet * set, GstFD * fd, gboolean active)
break; break;
case GST_FDSET_MODE_POLL: case GST_FDSET_MODE_POLL:
{ {
set->pollfds[fd->idx].events = (active ? (POLLIN | POLLPRI) : 0); gint events = set->pollfds[fd->idx].events;
if (active)
events |= (POLLIN | POLLPRI);
else
events &= ~(POLLIN | POLLPRI);
set->pollfds[fd->idx].events = events;
break; break;
} }
case GST_FDSET_MODE_EPOLL: case GST_FDSET_MODE_EPOLL:
@ -287,8 +303,10 @@ gst_fdset_fd_has_closed (GstFDSet * set, GstFD * fd)
{ {
gint idx = fd->idx; gint idx = fd->idx;
if (idx >= 0) g_mutex_lock (set->poll_lock);
if (idx >= 0 && idx < set->last_testpollfds)
res = (set->testpollfds[idx].revents & POLLHUP) != 0; res = (set->testpollfds[idx].revents & POLLHUP) != 0;
g_mutex_unlock (set->poll_lock);
break; break;
} }
case GST_FDSET_MODE_EPOLL: case GST_FDSET_MODE_EPOLL:
@ -310,8 +328,10 @@ gst_fdset_fd_has_error (GstFDSet * set, GstFD * fd)
{ {
gint idx = fd->idx; gint idx = fd->idx;
if (idx >= 0) g_mutex_lock (set->poll_lock);
if (idx >= 0 && idx < set->last_testpollfds)
res = (set->testpollfds[idx].revents & (POLLERR | POLLNVAL)) != 0; res = (set->testpollfds[idx].revents & (POLLERR | POLLNVAL)) != 0;
g_mutex_unlock (set->poll_lock);
break; break;
} }
case GST_FDSET_MODE_EPOLL: case GST_FDSET_MODE_EPOLL:
@ -333,8 +353,10 @@ gst_fdset_fd_can_read (GstFDSet * set, GstFD * fd)
{ {
gint idx = fd->idx; gint idx = fd->idx;
if (idx >= 0) g_mutex_lock (set->poll_lock);
if (idx >= 0 && idx < set->last_testpollfds)
res = (set->testpollfds[idx].revents & (POLLIN | POLLPRI)) != 0; res = (set->testpollfds[idx].revents & (POLLIN | POLLPRI)) != 0;
g_mutex_unlock (set->poll_lock);
break; break;
} }
case GST_FDSET_MODE_EPOLL: case GST_FDSET_MODE_EPOLL:
@ -356,8 +378,10 @@ gst_fdset_fd_can_write (GstFDSet * set, GstFD * fd)
{ {
gint idx = fd->idx; gint idx = fd->idx;
if (idx >= 0) g_mutex_lock (set->poll_lock);
if (idx >= 0 && idx < set->last_testpollfds)
res = (set->testpollfds[idx].revents & POLLOUT) != 0; res = (set->testpollfds[idx].revents & POLLOUT) != 0;
g_mutex_unlock (set->poll_lock);
break; break;
} }
case GST_FDSET_MODE_EPOLL: case GST_FDSET_MODE_EPOLL:
@ -393,19 +417,18 @@ gst_fdset_wait (GstFDSet * set, int timeout)
} }
case GST_FDSET_MODE_POLL: case GST_FDSET_MODE_POLL:
{ {
gint last_pollfds;
g_mutex_lock (set->poll_lock); g_mutex_lock (set->poll_lock);
if (set->testsize != set->size) { if (set->testsize != set->size) {
set->testpollfds = g_realloc (set->testpollfds, set->size); set->testpollfds = g_realloc (set->testpollfds, set->size);
set->testsize = set->size; set->testsize = set->size;
} }
last_pollfds = set->last_pollfds; set->last_testpollfds = set->last_pollfds;
memcpy (set->testpollfds, set->pollfds, memcpy (set->testpollfds, set->pollfds,
sizeof (struct pollfd) * last_pollfds); sizeof (struct pollfd) * set->last_testpollfds);
g_mutex_unlock (set->poll_lock); g_mutex_unlock (set->poll_lock);
res = poll (set->testpollfds, last_pollfds, timeout); res = poll (set->testpollfds, set->last_testpollfds, timeout);
break; break;
} }
case GST_FDSET_MODE_EPOLL: case GST_FDSET_MODE_EPOLL: