Added an epoll socket engine

6 files changed, 175 insertions, 93 deletions

diff --git a/include/config.h b/include/config.h
index f6f794a33..8b1d42527 100644
--- a/include/config.h
+++ b/include/config.h
@@ -802,6 +802,9 @@ class ServerConfig
 	/* Reason to akill clients for defcon */
 	char *DefConAkillReason;
 
+	/* The socket engine in use */
+	ci::string SocketEngine;
+
 	/* User keys to use for generating random hashes for pass codes etc */
 	long unsigned int UserKey1;
 	long unsigned int UserKey2;
diff --git a/include/extern.h b/include/extern.h
index e8019a0b4..a59ebc2b4 100644
--- a/include/extern.h
+++ b/include/extern.h
@@ -400,7 +400,7 @@ E int exception_add(User *u, const char *mask, const int limit, const char *reas
 
 /**** sockets.cpp ****/
 
-E SocketEngine socketEngine;
+E SocketEngineBase *SocketEngine;
 E int32 TotalRead;
 E int32 TotalWritten;
 
diff --git a/include/modules.h b/include/modules.h
index cb5e6dd28..d2f20dff4 100644
--- a/include/modules.h
+++ b/include/modules.h
@@ -155,7 +155,7 @@ enum ModuleReturn
 /** Priority types which can be returned from Module::Prioritize()
  */
 enum Priority { PRIORITY_FIRST, PRIORITY_DONTCARE, PRIORITY_LAST, PRIORITY_BEFORE, PRIORITY_AFTER };
-enum MODType { CORE, PROTOCOL, THIRD, SUPPORTED, QATESTED, ENCRYPTION, DATABASE };
+enum MODType { CORE, PROTOCOL, THIRD, SUPPORTED, QATESTED, ENCRYPTION, DATABASE, SOCKETENGINE };
 
 struct Message;
 extern CoreExport std::multimap<std::string, Message *> MessageMap;
diff --git a/include/services.h b/include/services.h
index 34041df2f..36a0937c3 100644
--- a/include/services.h
+++ b/include/services.h
@@ -300,10 +300,6 @@ class DatabaseException : public CoreException
 
 /*************************************************************************/
 
-#include "sockets.h"
-
-/*************************************************************************/
-
 /** Class with the ability to keep flags on items, they should extend from this
  * where T is an enum.
  */
@@ -354,6 +350,9 @@ template<typename T, size_t Size = 32> class Flags
 	}
 };
 
+#include "sockets.h"
+#include "socketengine.h"
+
 /*************************************************************************/
 
 template<typename T>
diff --git a/include/socketengine.h b/include/socketengine.h
new file mode 100644
index 000000000..042d4b410
--- /dev/null
+++ b/include/socketengine.h
@@ -0,0 +1,54 @@
+/*
+ *
+ * (C) 2003-2010 Anope Team
+ * Contact us at team@anope.org
+ *
+ * Please read COPYING and README for furhter details.
+ *
+ * Based on the original code of Epona by Lara.
+ * Based on the original code of Services by Andy Church.
+ */
+
+#ifndef SOCKETENGINE_H
+#define SOCKETENGINE_H
+
+class CoreExport SocketEngineBase
+{
+ public:
+	/* Map of sockets */
+	std::map<int, Socket *> Sockets;
+
+	/** Default constructor
+	 */
+	SocketEngineBase() { }
+
+	/** Default destructor
+	 */
+	virtual ~SocketEngineBase() { }
+
+	/** Add a socket to the internal list
+	 * @param s The socket
+	 */
+	virtual void AddSocket(Socket *s) { }
+
+	/** Delete a socket from the internal list
+	 * @param s The socket
+	 */
+	virtual void DelSocket(Socket *s) { }
+
+	/** Mark a socket as writeable
+	 * @param s The socket
+	 */
+	virtual void MarkWriteable(Socket *s) { }
+
+	/** Unmark a socket as writeable
+	 * @param s The socket
+	 */
+	virtual void ClearWriteable(Socket *s) { }
+
+	/** Read from sockets and do things
+	 */
+	virtual void Process() { }
+};
+
+#endif // SOCKETENGINE_H
diff --git a/include/sockets.h b/include/sockets.h
index a705baab6..92279f81f 100644
--- a/include/sockets.h
+++ b/include/sockets.h
@@ -1,5 +1,6 @@
-/*
- * (C) 2004-2010 Anope Team
+/*                                               
+ *
+ * (C) 2003-2010 Anope Team
  * Contact us at team@anope.org
  *
  * Please read COPYING and README for furhter details.
@@ -11,14 +12,14 @@
 #ifndef SOCKETS_H
 #define SOCKETS_H
 
+#define NET_BUFSIZE 65535
+
 #ifdef _WIN32
 # define CloseSocket closesocket
 #else
 # define CloseSocket close
 #endif
 
-#define NET_BUFSIZE 65536
-
 class SocketException : public CoreException
 {
  public:
@@ -33,152 +34,177 @@ class SocketException : public CoreException
 	virtual ~SocketException() throw() { }
 };
 
-class CoreExport Socket
+enum SocketType
+{
+	SOCKTYPE_CLIENT,
+	SOCKTYPE_LISTEN
+};
+
+enum SocketFlag
+{
+	SF_DEAD
+};
+
+class CoreExport Socket : public Flags<SocketFlag, 1>
 {
  private:
-	/** Read from the socket
-	 * @param buf Buffer to read to
+	/** Really recieve something from the buffer
+	 * @param buf The buf to read to
 	 * @param sz How much to read
 	 * @return Number of bytes recieved
 	 */
-	virtual int RecvInternal(char *buf, size_t sz) const;
+	virtual const int RecvInternal(char *buf, size_t sz) const;
 
-	/** Write to the socket
+	/** Really write something to the socket
 	 * @param buf What to write
-	 * @return Number of bytes sent, -1 on error
+	 * @return Number of bytes written
 	 */
-	virtual int SendInternal(const std::string &buf) const;
+	virtual const int SendInternal(const std::string &buf) const;
 
  protected:
-	/* Socket FD */
-	int Sock;
-	/* Host this socket is connected to */
-	std::string TargetHost;
-	/* Port we're connected to */
-	int Port;
-	/* IP this socket is bound to */
-	std::string BindHost;
-	/* Is this an IPv6 socket? */
+ 	/* Socket FD */
+	int sock;
+	/* IPv6? */
 	bool IPv6;
-
-	/* Messages to be written to the socket */
+	/* Things to be written to the socket */
 	std::string WriteBuffer;
-	/* Part of a message not totally yet recieved */
+	/* Part of a message sent from the server, but not totally recieved */
 	std::string extrabuf;
-	/* How much data was recieved from the socket */
+	/* How much data was received from this socket */
 	size_t RecvLen;
 
  public:
+	/* Type this socket is */
+	SocketType Type;
+
 	/** Default constructor
-	 * @param nTargetHost Hostname to connect to
-	 * @param nPort Port to connect to
-	 * @param nBindHos Host to bind to when connecting
-	 * @param nIPv6 true to use IPv6
+	 * @param nsock The socket to use, 0 if we need to create our own
+	 * @param nIPv6 true if using ipv6
 	 */
-	Socket(const std::string &nTargetHost, int nPort, const std::string &nBindHost = "", bool nIPv6 = false);
-
+ 	Socket(int nsock, bool nIPv6);
+	
 	/** Default destructor
 	 */
 	virtual ~Socket();
 
 	/** Get the socket FD for this socket
-	 * @return The fd
+	 * @return the fd
 	 */
-	virtual int GetSock() const;
+	int GetSock() const;
 
 	/** Check if this socket is IPv6
 	 * @return true or false
 	 */
 	bool IsIPv6() const;
 
-	/** Called when there is something to be read from thie socket
-	 * @return true on success, false to kill this socket
+	/** Get the length of the read buffer
+	 * @return The length of the read buffer
+	 */
+	size_t ReadBufferLen() const;
+
+	/** Get the length of the write buffer
+	 * @return The length of the write buffer
+	 */
+	size_t WriteBufferLen() const;
+
+	/** Called when there is something to be recieved for this socket
+	 * @return true on success, false to drop this socket
 	 */
 	virtual bool ProcessRead();
 
-	/** Called when this socket becomes writeable
+	/** Called when there is something to be written to this socket
 	 * @return true on success, false to drop this socket
 	 */
 	virtual bool ProcessWrite();
 
-	/** Called when there is an error on this socket
+	/** Called when there is an error for this socket
+	 * @return true on success, false to drop this socket
 	 */
 	virtual void ProcessError();
 
-	/** Called with a message recieved from the socket
-	 * @param buf The message
-	 * @return true on success, false to kill this socket
+	/** Called with a line recieved from the socket
+	 * @param buf The line
+	 * @return true to continue reading, false to drop the socket
 	 */
 	virtual bool Read(const std::string &buf);
 
 	/** Write to the socket
-	 * @param message The message to write
+	 * @param message The message
 	 */
 	void Write(const char *message, ...);
-	void Write(std::string &message);
+	void Write(const std::string &message);
+};
 
-	/** Get the length of the read buffer
-	 * @return The length of the read buffer
-	 */
-	size_t ReadBufferLen() const;
+class CoreExport ClientSocket : public Socket
+{
+ protected:
+	/* Target host we're connected to */
+	std::string TargetHost;
+	/* Target port we're connected to */
+	int Port;
+	/* The host to bind to */
+	std::string BindHost;
 
-	/** Get the length of the write buffer
-	 * @return The length of the write buffer
-	 */
-	size_t WriteBufferLen() const;
+ public:
+
+       /** Constructor
+	* @param nTargetHost The target host to connect to
+	* @param nPort The target port to connect to
+	* @param nBindHost The host to bind to for connecting
+	* @param nIPv6 true to use IPv6
+	*/
+       ClientSocket(const std::string &nTargetHost, int nPort, const std::string &nBindHost, bool nIPv6);
+
+       /** Default destructor
+	*/
+       virtual ~ClientSocket();
+
+      /** Called with a line recieved from the socket
+	* @param buf The line
+	* @return true to continue reading, false to drop the socket
+	*/
+	virtual bool Read(const std::string &buf);
 };
 
-class CoreExport SocketEngine
+class CoreExport ListenSocket : public Socket
 {
- private:
-	/* List of sockets that need to be deleted */
-	std::set<Socket *> OldSockets;
-	/* FDs to read */
-	fd_set ReadFDs;
-	/* FDs that want writing */
-	fd_set WriteFDs;
-	/* Max FD */
-	int MaxFD;
-
-	/** Unmark a socket as writeable
-	 * @param s The socket
-	 */
-	void ClearWriteable(Socket *s);
- public:
-	/* Set of sockets */
-	std::set<Socket *> Sockets;
+ protected:
+	/* Bind IP */
+	std::string BindIP;
+	/* Port to bind to */
+	int Port;
 
+ public:
 	/** Constructor
+	 * @param bind The IP to bind to
+	 * @param port The port to listen on
 	 */
-	SocketEngine();
+	ListenSocket(const std::string &bind, int port);
 
 	/** Destructor
 	 */
-	virtual ~SocketEngine();
+	virtual ~ListenSocket();
 
-	/** Add a socket to the socket engine
-	 * @param s The socket
+	/** Process what has come in from the connection
+	 * @return false to destory this socket
 	 */
-	void AddSocket(Socket *s);
+	bool ProcessRead();
 
-	/** Delete a socket from the socket engine
-	 * @param s The socket
+	/** Called when a connection is accepted
+	 * @param s The socket for the new connection
+	 * @return true if the listen socket should remain alive
 	 */
-	void DelSocket(Socket *s);
+	virtual bool OnAccept(Socket *s);
 
-	/** Mark a socket as wanting to be written to
-	 * @param s The socket
+	/** Get the bind IP for this socket
+	 * @return the bind ip
 	 */
-	void MarkWriteable(Socket *s);
+	const std::string &GetBindIP() const;
 
-	/** Called to iterate through each socket and check for activity
-	 */
-	void Process();
-
-	/** Get the last socket error
-	 * @return The error
-	 */
-	const std::string GetError() const;
+	/** Get the port this socket is bound to
+	 * @return The port
+	*/
+	const int GetPort() const;
 };
 
-#endif // SOCKETS_H
+#endif // SOCKET_H