diff options
Diffstat (limited to 'source3/lib')
-rw-r--r-- | source3/lib/events.c | 1 | ||||
-rw-r--r-- | source3/lib/g_lock.c | 1 | ||||
-rw-r--r-- | source3/lib/packet.c | 1 | ||||
-rw-r--r-- | source3/lib/readline.c | 1 | ||||
-rw-r--r-- | source3/lib/select.c | 207 | ||||
-rw-r--r-- | source3/lib/util_sock.c | 1 |
6 files changed, 5 insertions, 207 deletions
diff --git a/source3/lib/events.c b/source3/lib/events.c index 21a13b63fb..70a0d7c302 100644 --- a/source3/lib/events.c +++ b/source3/lib/events.c @@ -20,6 +20,7 @@ #include "includes.h" #include <tevent_internal.h> +#include "../lib/util/select.h" /* * Return if there's something in the queue diff --git a/source3/lib/g_lock.c b/source3/lib/g_lock.c index 2c065029e7..378e464b5c 100644 --- a/source3/lib/g_lock.c +++ b/source3/lib/g_lock.c @@ -21,6 +21,7 @@ #include "g_lock.h" #include "librpc/gen_ndr/messaging.h" #include "ctdbd_conn.h" +#include "../lib/util/select.h" static NTSTATUS g_lock_force_unlock(struct g_lock_ctx *ctx, const char *name, struct server_id pid); diff --git a/source3/lib/packet.c b/source3/lib/packet.c index c131b973bc..fefb74a0c5 100644 --- a/source3/lib/packet.c +++ b/source3/lib/packet.c @@ -18,6 +18,7 @@ */ #include "includes.h" +#include "../lib/util/select.h" struct packet_context { int fd; diff --git a/source3/lib/readline.c b/source3/lib/readline.c index 4b82291eca..1be0e05f7d 100644 --- a/source3/lib/readline.c +++ b/source3/lib/readline.c @@ -19,6 +19,7 @@ */ #include "includes.h" +#include "../lib/util/select.h" #ifdef HAVE_LIBREADLINE # ifdef HAVE_READLINE_READLINE_H diff --git a/source3/lib/select.c b/source3/lib/select.c deleted file mode 100644 index 846e6af1de..0000000000 --- a/source3/lib/select.c +++ /dev/null @@ -1,207 +0,0 @@ -/* - Unix SMB/Netbios implementation. - Version 3.0 - Samba select/poll implementation - Copyright (C) Andrew Tridgell 1992-1998 - - This program is free software; you can redistribute it and/or modify - it under the terms of the GNU General Public License as published by - the Free Software Foundation; either version 3 of the License, or - (at your option) any later version. - - This program is distributed in the hope that it will be useful, - but WITHOUT ANY WARRANTY; without even the implied warranty of - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - GNU General Public License for more details. - - You should have received a copy of the GNU General Public License - along with this program. If not, see <http://www.gnu.org/licenses/>. -*/ - -#include "includes.h" - -/* This is here because it allows us to avoid a nasty race in signal handling. - We need to guarantee that when we get a signal we get out of a select immediately - but doing that involves a race condition. We can avoid the race by getting the - signal handler to write to a pipe that is in the select/poll list - - This means all Samba signal handlers should call sys_select_signal(). -*/ - -static pid_t initialised; -static int select_pipe[2]; -static volatile unsigned pipe_written, pipe_read; - -/******************************************************************* - Call this from all Samba signal handlers if you want to avoid a - nasty signal race condition. -********************************************************************/ - -void sys_select_signal(char c) -{ - int saved_errno = errno; - - if (!initialised) return; - - if (pipe_written > pipe_read+256) return; - - if (write(select_pipe[1], &c, 1) == 1) pipe_written++; - - errno = saved_errno; -} - -/******************************************************************* - Like select() but avoids the signal race using a pipe - it also guuarantees that fds on return only ever contains bits set - for file descriptors that were readable. -********************************************************************/ - -int sys_select(int maxfd, fd_set *readfds, fd_set *writefds, fd_set *errorfds, struct timeval *tval) -{ - int ret, saved_errno; - fd_set *readfds2, readfds_buf; - - if (initialised != sys_getpid()) { - if (pipe(select_pipe) == -1) - { - DEBUG(0, ("sys_select: pipe failed (%s)\n", - strerror(errno))); - if (readfds != NULL) - FD_ZERO(readfds); - if (writefds != NULL) - FD_ZERO(writefds); - if (errorfds != NULL) - FD_ZERO(errorfds); - return -1; - } - - /* - * These next two lines seem to fix a bug with the Linux - * 2.0.x kernel (and probably other UNIXes as well) where - * the one byte read below can block even though the - * select returned that there is data in the pipe and - * the pipe_written variable was incremented. Thanks to - * HP for finding this one. JRA. - */ - - if(set_blocking(select_pipe[0],0)==-1) - smb_panic("select_pipe[0]: O_NONBLOCK failed"); - if(set_blocking(select_pipe[1],0)==-1) - smb_panic("select_pipe[1]: O_NONBLOCK failed"); - - initialised = sys_getpid(); - } - - maxfd = MAX(select_pipe[0]+1, maxfd); - - /* If readfds is NULL we need to provide our own set. */ - if (readfds) { - readfds2 = readfds; - } else { - readfds2 = &readfds_buf; - FD_ZERO(readfds2); - } - FD_SET(select_pipe[0], readfds2); - - errno = 0; - ret = select(maxfd,readfds2,writefds,errorfds,tval); - - if (ret <= 0) { - FD_ZERO(readfds2); - if (writefds) - FD_ZERO(writefds); - if (errorfds) - FD_ZERO(errorfds); - } else if (FD_ISSET(select_pipe[0], readfds2)) { - char c; - saved_errno = errno; - if (read(select_pipe[0], &c, 1) == 1) { - pipe_read++; - /* Mark Weaver <mark-clist@npsl.co.uk> pointed out a critical - fix to ensure we don't lose signals. We must always - return -1 when the select pipe is set, otherwise if another - fd is also ready (so ret == 2) then we used to eat the - byte in the pipe and lose the signal. JRA. - */ - ret = -1; -#if 0 - /* JRA - we can use this to debug the signal messaging... */ - DEBUG(0,("select got %u signal\n", (unsigned int)c)); -#endif - errno = EINTR; - } else { - FD_CLR(select_pipe[0], readfds2); - ret--; - errno = saved_errno; - } - } - - return ret; -} - -/******************************************************************* - Similar to sys_select() but catch EINTR and continue. - This is what sys_select() used to do in Samba. -********************************************************************/ - -int sys_select_intr(int maxfd, fd_set *readfds, fd_set *writefds, fd_set *errorfds, struct timeval *tval) -{ - int ret; - fd_set *readfds2, readfds_buf, *writefds2, writefds_buf, *errorfds2, errorfds_buf; - struct timeval tval2, *ptval; - struct timespec end_time; - - readfds2 = (readfds ? &readfds_buf : NULL); - writefds2 = (writefds ? &writefds_buf : NULL); - errorfds2 = (errorfds ? &errorfds_buf : NULL); - if (tval) { - clock_gettime_mono(&end_time); - end_time.tv_sec += tval->tv_sec; - end_time.tv_nsec += tval->tv_usec *1000; - end_time.tv_sec += end_time.tv_nsec / 1000000000; - end_time.tv_nsec %= 1000000000; - errno = 0; - tval2 = *tval; - ptval = &tval2; - } else { - ptval = NULL; - } - - do { - if (readfds) - readfds_buf = *readfds; - if (writefds) - writefds_buf = *writefds; - if (errorfds) - errorfds_buf = *errorfds; - if (ptval && (errno == EINTR)) { - struct timespec now_time; - int64_t tdif; - - clock_gettime_mono(&now_time); - tdif = nsec_time_diff(&end_time,&now_time); - if (tdif <= 0) { - ret = 0; /* time expired. */ - break; - } - ptval->tv_sec = tdif / 1000000000; - ptval->tv_usec = (tdif % 1000000000) / 1000; - } - - /* We must use select and not sys_select here. If we use - sys_select we'd lose the fact a signal occurred when sys_select - read a byte from the pipe. Fix from Mark Weaver - <mark-clist@npsl.co.uk> - */ - ret = select(maxfd, readfds2, writefds2, errorfds2, ptval); - } while (ret == -1 && errno == EINTR); - - if (readfds) - *readfds = readfds_buf; - if (writefds) - *writefds = writefds_buf; - if (errorfds) - *errorfds = errorfds_buf; - - return ret; -} diff --git a/source3/lib/util_sock.c b/source3/lib/util_sock.c index 2b33816293..188e11702e 100644 --- a/source3/lib/util_sock.c +++ b/source3/lib/util_sock.c @@ -22,6 +22,7 @@ #include "includes.h" #include "memcache.h" #include "../lib/async_req/async_sock.h" +#include "../lib/util/select.h" /**************************************************************************** Get a port number in host byte order from a sockaddr_storage. |