Implemented poll() support for Process tasks.

Fixes #3035

lib/base/process.cpp

@@ -143,34 +143,36 @@ void Process::Run(void)
 void Process::WorkerThreadProc(int taskFd)
 {
 	map<int, Process::Ptr> tasks;
+	pollfd *pfds;
 
 	for (;;) {
 		map<int, Process::Ptr>::iterator it, prev;
 
 #ifndef _MSC_VER
-		fd_set readfds;
-		int nfds = 0;
+		pfds = (pollfd *)realloc(pfds, (1 + tasks.size()) * sizeof(pollfd));
 
-		FD_ZERO(&readfds);
+		if (pfds == NULL)
+			BOOST_THROW_EXCEPTION(PosixException("realloc() failed.", errno));
 
-		if (tasks.size() < MaxTasksPerThread)
-			FD_SET(taskFd, &readfds);
+		int idx = 0;
 
-		if (taskFd > nfds)
-			nfds = taskFd;
+		if (tasks.size() < MaxTasksPerThread) {
+			pfds[idx].fd = taskFd;
+			pfds[idx].events = POLLIN;
+			idx++;
+		}
 
 		int fd;
 		BOOST_FOREACH(tie(fd, tuples::ignore), tasks) {
-			if (fd > nfds)
-				nfds = fd;
-
-			FD_SET(fd, &readfds);
+			pfds[idx].fd = fd;
+			pfds[idx].events = POLLIN;
+			idx++;
 		}
 
-		int rc = select(nfds + 1, &readfds, NULL, NULL, NULL);
+		int rc = poll(pfds, idx, -1);
 
 		if (rc < 0 && errno != EINTR)
-			BOOST_THROW_EXCEPTION(PosixException("select() failed.", errno));
+			BOOST_THROW_EXCEPTION(PosixException("poll() failed.", errno));
 
 		if (rc == 0)
 			continue;
@@ -180,7 +182,11 @@ void Process::WorkerThreadProc(int taskFd)
 #endif /* _MSC_VER */
 
 #ifndef _MSC_VER
-		if (FD_ISSET(taskFd, &readfds)) {
+		for (int i = 0; i < idx; i++) {
+			if ((pfds[i].revents & POLLIN) == 0)
+				continue;
+
+			if (pfds[i].fd == taskFd) {
 #endif /* _MSC_VER */
 
 				while (tasks.size() < MaxTasksPerThread) {
@@ -189,6 +195,7 @@ void Process::WorkerThreadProc(int taskFd)
 					{
 						boost::mutex::scoped_lock lock(m_Mutex);
 
+#ifndef _MSC_VER
 						/* Read one byte for every task we take from the pending tasks list. */
 						char buffer;
 						int rc = read(taskFd, &buffer, sizeof(buffer));
@@ -201,6 +208,10 @@ void Process::WorkerThreadProc(int taskFd)
 						}
 
 						assert(!m_Tasks.empty());
+#else /* _MSC_VER */
+						if (m_Tasks.empty())
+							break;
+#endif /* _MSC_VER */
 
 						task = m_Tasks.front();
 						m_Tasks.pop_front();
@@ -217,30 +228,38 @@ void Process::WorkerThreadProc(int taskFd)
 					}
 				}
 #ifndef _MSC_VER
-		}
-#endif /* _MSC_VER */
 
-		for (it = tasks.begin(); it != tasks.end(); ) {
-			int fd = it->first;
-			Process::Ptr task = it->second;
-
-#ifndef _MSC_VER
-			if (!FD_ISSET(fd, &readfds)) {
-				it++;
 				continue;
 			}
+
+			it = tasks.find(pfds[i].fd);
+
+			if (it == tasks.end())
+				continue;
+#else /* _MSC_VER */
+			for (it = tasks.begin(); it != tasks.end(); ) {
+				int fd = it->first;
 #endif /* _MSC_VER */
+				Process::Ptr task = it->second;
 
 				if (!task->RunTask()) {
 					prev = it;
+#ifdef _MSC_VER
 					it++;
+#endif /* _MSC_VER */
 					tasks.erase(prev);
 
 					Event::Post(boost::bind(&Process::FinishResult, task, task->m_Result));
+#ifdef _MSC_VER
 				} else {
 					it++;
+#endif /* _MSC_VER */
 				}
+#ifdef _MSC_VER
 			}
+#else /* _MSC_VER */
+		}
+#endif /* _MSC_VER */
 	}
 }
 

lib/base/unix.h

@@ -35,6 +35,7 @@
 #include <syslog.h>
 #include <sys/file.h>
 #include <sys/wait.h>
+#include <poll.h>
 #include <glob.h>
 #include <ltdl.h>