1 files changed, 38 insertions, 101 deletions
diff --git a/src/event/MultiSocketMonitor.cxx b/src/event/MultiSocketMonitor.cxx
index bd1aa6fef..5ef8b98af 100644
--- a/src/event/MultiSocketMonitor.cxx
+++ b/src/event/MultiSocketMonitor.cxx
@@ -20,12 +20,12 @@
 #include "config.h"
 #include "MultiSocketMonitor.hxx"
 #include "Loop.hxx"
-#include "system/fd_util.h"
-#include "Compiler.h"
 
-#include <assert.h>
+#include <algorithm>
 
-#ifdef USE_EPOLL
+#ifndef WIN32
+#include <poll.h>
+#endif
 
 MultiSocketMonitor::MultiSocketMonitor(EventLoop &_loop)
 	:IdleMonitor(_loop), TimeoutMonitor(_loop), ready(false) {
@@ -37,6 +37,40 @@ MultiSocketMonitor::~MultiSocketMonitor()
 }
 
 void
+MultiSocketMonitor::ClearSocketList()
+{
+	assert(GetEventLoop().IsInsideOrNull());
+
+	fds.clear();
+}
+
+#ifndef WIN32
+
+void
+MultiSocketMonitor::ReplaceSocketList(pollfd *pfds, unsigned n)
+{
+	pollfd *const end = pfds + n;
+
+	UpdateSocketList([pfds, end](int fd) -> unsigned {
+			auto i = std::find_if(pfds, end, [fd](const struct pollfd &pfd){
+					return pfd.fd == fd;
+				});
+			if (i == end)
+				return 0;
+
+			auto events = i->events;
+			i->events = 0;
+			return events;
+		});
+
+	for (auto i = pfds; i != end; ++i)
+		if (i->events != 0)
+			AddSocket(i->fd, i->events);
+}
+
+#endif
+
+void
 MultiSocketMonitor::Prepare()
 {
 	int timeout_ms = PrepareSockets();
@@ -64,100 +98,3 @@ MultiSocketMonitor::OnIdle()
 		Prepare();
 	}
 }
-
-#else
-
-/**
- * The vtable for our GSource implementation.  Unfortunately, we
- * cannot declare it "const", because g_source_new() takes a non-const
- * pointer, for whatever reason.
- */
-static GSourceFuncs multi_socket_monitor_source_funcs = {
-	MultiSocketMonitor::Prepare,
-	MultiSocketMonitor::Check,
-	MultiSocketMonitor::Dispatch,
-	nullptr,
-	nullptr,
-	nullptr,
-};
-
-MultiSocketMonitor::MultiSocketMonitor(EventLoop &_loop)
-	:loop(_loop),
-	 source((Source *)g_source_new(&multi_socket_monitor_source_funcs,
-				       sizeof(*source))),
-	 absolute_timeout_us(-1) {
-	source->monitor = this;
-
-	g_source_attach(&source->base, loop.GetContext());
-}
-
-MultiSocketMonitor::~MultiSocketMonitor()
-{
-	g_source_destroy(&source->base);
-	g_source_unref(&source->base);
-	source = nullptr;
-}
-
-bool
-MultiSocketMonitor::Prepare(gint *timeout_r)
-{
-	int timeout_ms = *timeout_r = PrepareSockets();
-	absolute_timeout_us = timeout_ms < 0
-		? uint64_t(-1)
-		: GetTime() + uint64_t(timeout_ms) * 1000;
-
-	return false;
-}
-
-bool
-MultiSocketMonitor::Check() const
-{
-	if (GetTime() >= absolute_timeout_us)
-		return true;
-
-	for (const auto &i : fds)
-		if (i.GetReturnedEvents() != 0)
-			return true;
-
-	return false;
-}
-
-/*
- * GSource methods
- *
- */
-
-gboolean
-MultiSocketMonitor::Prepare(GSource *_source, gint *timeout_r)
-{
-	Source &source = *(Source *)_source;
-	MultiSocketMonitor &monitor = *source.monitor;
-	assert(_source == &monitor.source->base);
-
-	return monitor.Prepare(timeout_r);
-}
-
-gboolean
-MultiSocketMonitor::Check(GSource *_source)
-{
-	const Source &source = *(const Source *)_source;
-	const MultiSocketMonitor &monitor = *source.monitor;
-	assert(_source == &monitor.source->base);
-
-	return monitor.Check();
-}
-
-gboolean
-MultiSocketMonitor::Dispatch(GSource *_source,
-			     gcc_unused GSourceFunc callback,
-			     gcc_unused gpointer user_data)
-{
-	Source &source = *(Source *)_source;
-	MultiSocketMonitor &monitor = *source.monitor;
-	assert(_source == &monitor.source->base);
-
-	monitor.Dispatch();
-	return true;
-}
-
-#endif