1 files changed, 69 insertions, 1 deletions
diff --git a/src/socketengines/socketengine_poll.cpp b/src/socketengines/socketengine_poll.cpp
index a72d21d1f..16917ffbc 100644
--- a/src/socketengines/socketengine_poll.cpp
+++ b/src/socketengines/socketengine_poll.cpp
@@ -13,7 +13,71 @@
 
 #include "inspircd.h"
 #include "exitcodes.h"
-#include "socketengines/socketengine_poll.h"
+/*       +------------------------------------+
+ *       | Inspire Internet Relay Chat Daemon |
+ *       +------------------------------------+
+ *
+ *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ * See: http://wiki.inspircd.org/Credits
+ *
+ * This program is free but copyrighted software; see
+ *            the file COPYING for details.
+ *
+ * ---------------------------------------------------
+ */
+
+#ifndef __SOCKETENGINE_POLL__
+#define __SOCKETENGINE_POLL__
+
+#include <vector>
+#include <string>
+#include <map>
+#include "inspircd_config.h"
+#include "inspircd.h"
+#include "socketengine.h"
+
+#ifndef WINDOWS
+	#ifndef __USE_XOPEN
+    	    #define __USE_XOPEN /* fuck every fucking OS ever made. needed by poll.h to work.*/
+	#endif
+	#include <poll.h>
+	#include <sys/poll.h>
+#else
+	/* *grumble* */
+	#define struct pollfd WSAPOLLFD
+	#define poll WSAPoll
+#endif
+
+class InspIRCd;
+
+/** A specialisation of the SocketEngine class, designed to use poll().
+ */
+class PollEngine : public SocketEngine
+{
+private:
+	/** These are used by poll() to hold socket events
+	 */
+	struct pollfd *events;
+	/** This map maps fds to an index in the events array.
+	 */
+	std::map<int, unsigned int> fd_mappings;
+public:
+	/** Create a new PollEngine
+	 */
+	PollEngine();
+	/** Delete a PollEngine
+	 */
+	virtual ~PollEngine();
+	virtual bool AddFd(EventHandler* eh, int event_mask);
+	virtual void OnSetEvent(EventHandler* eh, int old_mask, int new_mask);
+	virtual EventHandler* GetRef(int fd);
+	virtual bool DelFd(EventHandler* eh, bool force = false);
+	virtual int DispatchEvents();
+	virtual std::string GetName();
+};
+
+#endif
+
 #include <ulimit.h>
 #ifdef __FreeBSD__
 	#include <sys/sysctl.h>
@@ -225,3 +289,7 @@ std::string PollEngine::GetName()
 	return "poll";
 }
 
+SocketEngine* CreateSocketEngine()
+{
+	return new PollEngine;
+}