'svn propset -R svn:eol-style CR *' Set to UNIX-style always. Binaries are auto skipped by svn.

git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@7454 e03df62e-2008-0410-955e-edbf42e46eb7

1 files changed, 1 insertions, 226 deletions

diff --git a/include/socketengine_iocp.h b/include/socketengine_iocp.h
index f4825c6b4..ac27c86e3 100644
--- a/include/socketengine_iocp.h
+++ b/include/socketengine_iocp.h
@@ -1,226 +1 @@
-/*       +------------------------------------+
- *       | Inspire Internet Relay Chat Daemon |
- *       +------------------------------------+
- *
- *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
- * See: http://www.inspircd.org/wiki/index.php/Credits
- *
- * This program is free but copyrighted software; see
- *            the file COPYING for details.
- *
- * ---------------------------------------------------
- */
-
-#ifndef __SOCKETENGINE_IOCP__
-#define __SOCKETENGINE_IOCP__
-
-#define READ_BUFFER_SIZE 600
-#define USING_IOCP 1
-
-#include "inspircd_config.h"
-#include "inspircd_win32wrapper.h"
-#include "globals.h"
-#include "inspircd.h"
-#include "socketengine.h"
-
-/** Socket overlapped event types
- */
-enum SocketIOEvent
-{
-	/** Read ready */
-	SOCKET_IO_EVENT_READ_READY			= 0,
-	/** Write ready */
-	SOCKET_IO_EVENT_WRITE_READY			= 1,
-	/** Accept ready */
-	SOCKET_IO_EVENT_ACCEPT				= 2,
-	/** Error occured */
-	SOCKET_IO_EVENT_ERROR				= 3,
-	/** Number of events */
-	NUM_SOCKET_IO_EVENTS				= 4,
-};
-
-/** Represents a windows overlapped IO event
- */
-class Overlapped
-{
- public:
-	/** Overlap event */
-	OVERLAPPED m_overlap;
-	/** Type of event */
-	SocketIOEvent m_event;
-#ifdef WIN64
-	/** Parameters */
-	unsigned __int64 m_params;
-#else
-	/** Parameters */
-	unsigned long m_params;
-#endif
-	/** Create an overlapped event
-	 */
-	Overlapped(SocketIOEvent ev, int params) : m_event(ev), m_params(params)
-	{
-		memset(&m_overlap, 0, sizeof(OVERLAPPED));
-	}
-};
-
-/** Specific to UDP sockets with overlapped IO
- */
-struct udp_overlap
-{
-	unsigned char udp_buffer[600];
-	unsigned long udp_len;
-	sockaddr udp_sockaddr[2];
-	unsigned long udp_sockaddr_len;
-};
-
-/** Specific to accepting sockets with overlapped IO
- */
-struct accept_overlap
-{
-	int socket;
-	char buf[1024];
-};
-
-/** Implementation of SocketEngine that implements windows IO Completion Ports
- */
-class IOCPEngine : public SocketEngine
-{
-	/** Creates a "fake" file descriptor for use with an IOCP socket.
-	 * This is a little slow, but it isnt called too much. We'll fix it
-	 * in a future release.
-	 * @return -1 if there are no free slots, and an integer if it finds one.
-	 */
-	__inline int GenerateFd(int RealFd)
-	{
-		int index_hash = RealFd % MAX_DESCRIPTORS;
-		if(ref[index_hash] == 0)
-			return index_hash;
-		else
-		{
-			register int i = 0;
-			for(; i < MAX_DESCRIPTORS; ++i)
-				if(ref[i] == 0)
-					return i;
-		}
-		return -1;
-	}
-	
-	/** Global I/O completion port that sockets attach to.
-	 */
-	HANDLE m_completionPort;
-
-	/** This is kinda shitty... :/ for getting an address from a real fd. 
-	 */
-	map<int, EventHandler*> m_binding;
-
-public:
-	/** Creates an IOCP Socket Engine
-	 * @param Instance The creator of this object
-	 */
-	IOCPEngine(InspIRCd* Instance);
-
-	/** Deletes an IOCP socket engine and all the attached sockets
-	 */
-	~IOCPEngine();
-
-	/** Adds an event handler to the completion port, and sets up initial events.
-	 * @param eh EventHandler to add
-	 * @return True if success, false if no room
-	 */
-	bool AddFd(EventHandler* eh);
-
-	/** Gets the maximum number of file descriptors that this engine can handle.
-	 * @return The number of file descriptors
-	 */
-	__inline int GetMaxFds() { return MAX_DESCRIPTORS; }
-
-	/** Gets the number of free/remaining file descriptors under this engine.
-	 * @return Remaining count
-	 */
-	__inline int GetRemainingFds()
-	{
-		register int count = 0;
-		register int i = 0;			
-		for(; i < MAX_DESCRIPTORS; ++i)
-			if(ref[i] == 0)
-				++count;
-		return count;
-	}
-
-	/** Removes a file descriptor from the set, preventing it from receiving any more events
-	 * @return True if remove was successful, false otherwise
-	 */
-	bool DelFd(EventHandler* eh, bool force = false);
-
-	/** Called every loop to handle input/output events for all sockets under this engine
-	 * @return The number of "changed" sockets.
-	 */
-	int DispatchEvents();
-
-	/** Gets the name of this socket engine as a string.
-	 * @return string of socket engine name
-	 */
-	std::string GetName();
-
-	/** Queues a Write event on the specified event handler.
-	 * @param eh EventHandler that needs data sent on
-	 */
-	void WantWrite(EventHandler* eh);
-
-	/** Posts a completion event on the specified socket.
-	 * @param eh EventHandler for message
-	 * @param type Event Type
-	 * @param param Event Parameter
-	 * @return True if added, false if not
-	 */
-	bool PostCompletionEvent(EventHandler* eh, SocketIOEvent type, int param);
-
-	/** Posts a read event on the specified socket
-	 * @param eh EventHandler (socket)
-	 */
-	void PostReadEvent(EventHandler* eh);
-
-	/** Posts an accept event on the specified socket
-	 * @param eh EventHandler (socket)
-	 */
-	void PostAcceptEvent(EventHandler* eh);
-
-	/** Returns the EventHandler attached to a specific fd.
-	 * If the fd isnt in the socketengine, returns NULL.
-	 * @param fd The event handler to look for
-	 * @return A pointer to the event handler, or NULL
-	 */
-	EventHandler* GetRef(int fd);
-
-	/** Returns true if a file descriptor exists in
-	 * the socket engine's list.
-	 * @param fd The event handler to look for
-	 * @return True if this fd has an event handler
-	 */
-	bool HasFd(int fd);
-
-	/** Returns the EventHandler attached to a specific fd.
-	 * If the fd isnt in the socketengine, returns NULL.
-	 * @param fd The event handler to look for
-	 * @return A pointer to the event handler, or NULL
-	 */
-	EventHandler* GetIntRef(int fd);
-
-	/** Holds the preallocated buffer passed to WSARecvFrom
-	 * function. Yes, I know, it's a dirty hack.
-	 */
-	udp_overlap * udp_ov;
-};
-
-/** Creates a SocketEngine
- */
-class SocketEngineFactory
-{
-public:
-	/** Create a new instance of SocketEngine based on IOCPEngine
-	 */
-	SocketEngine* Create(InspIRCd* Instance) { return new IOCPEngine(Instance); }
-};
-
-#endif
-
+/*       +------------------------------------+
 *       | Inspire Internet Relay Chat Daemon |
 *       +------------------------------------+
 *
 *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
 * See: http://www.inspircd.org/wiki/index.php/Credits
 *
 * This program is free but copyrighted software; see
 *            the file COPYING for details.
 *
 * ---------------------------------------------------
 */

#ifndef __SOCKETENGINE_IOCP__
#define __SOCKETENGINE_IOCP__

#define READ_BUFFER_SIZE 600
#define USING_IOCP 1

#include "inspircd_config.h"
#include "inspircd_win32wrapper.h"
#include "globals.h"
#include "inspircd.h"
#include "socketengine.h"

/** Socket overlapped event types
 */
enum SocketIOEvent
{
	/** Read ready */
	SOCKET_IO_EVENT_READ_READY			= 0,
	/** Write ready */
	SOCKET_IO_EVENT_WRITE_READY			= 1,
	/** Accept ready */
	SOCKET_IO_EVENT_ACCEPT				= 2,
	/** Error occured */
	SOCKET_IO_EVENT_ERROR				= 3,
	/** Number of events */
	NUM_SOCKET_IO_EVENTS				= 4,
};

/** Represents a windows overlapped IO event
 */
class Overlapped
{
 public:
	/** Overlap event */
	OVERLAPPED m_overlap;
	/** Type of event */
	SocketIOEvent m_event;
#ifdef WIN64
	/** Parameters */
	unsigned __int64 m_params;
#else
	/** Parameters */
	unsigned long m_params;
#endif
	/** Create an overlapped event
	 */
	Overlapped(SocketIOEvent ev, int params) : m_event(ev), m_params(params)
	{
		memset(&m_overlap, 0, sizeof(OVERLAPPED));
	}
};

/** Specific to UDP sockets with overlapped IO
 */
struct udp_overlap
{
	unsigned char udp_buffer[600];
	unsigned long udp_len;
	sockaddr udp_sockaddr[2];
	unsigned long udp_sockaddr_len;
};

/** Specific to accepting sockets with overlapped IO
 */
struct accept_overlap
{
	int socket;
	char buf[1024];
};

/** Implementation of SocketEngine that implements windows IO Completion Ports
 */
class IOCPEngine : public SocketEngine
{
	/** Creates a "fake" file descriptor for use with an IOCP socket.
	 * This is a little slow, but it isnt called too much. We'll fix it
	 * in a future release.
	 * @return -1 if there are no free slots, and an integer if it finds one.
	 */
	__inline int GenerateFd(int RealFd)
	{
		int index_hash = RealFd % MAX_DESCRIPTORS;
		if(ref[index_hash] == 0)
			return index_hash;
		else
		{
			register int i = 0;
			for(; i < MAX_DESCRIPTORS; ++i)
				if(ref[i] == 0)
					return i;
		}
		return -1;
	}
	
	/** Global I/O completion port that sockets attach to.
	 */
	HANDLE m_completionPort;

	/** This is kinda shitty... :/ for getting an address from a real fd. 
	 */
	map<int, EventHandler*> m_binding;

public:
	/** Creates an IOCP Socket Engine
	 * @param Instance The creator of this object
	 */
	IOCPEngine(InspIRCd* Instance);

	/** Deletes an IOCP socket engine and all the attached sockets
	 */
	~IOCPEngine();

	/** Adds an event handler to the completion port, and sets up initial events.
	 * @param eh EventHandler to add
	 * @return True if success, false if no room
	 */
	bool AddFd(EventHandler* eh);

	/** Gets the maximum number of file descriptors that this engine can handle.
	 * @return The number of file descriptors
	 */
	__inline int GetMaxFds() { return MAX_DESCRIPTORS; }

	/** Gets the number of free/remaining file descriptors under this engine.
	 * @return Remaining count
	 */
	__inline int GetRemainingFds()
	{
		register int count = 0;
		register int i = 0;			
		for(; i < MAX_DESCRIPTORS; ++i)
			if(ref[i] == 0)
				++count;
		return count;
	}

	/** Removes a file descriptor from the set, preventing it from receiving any more events
	 * @return True if remove was successful, false otherwise
	 */
	bool DelFd(EventHandler* eh, bool force = false);

	/** Called every loop to handle input/output events for all sockets under this engine
	 * @return The number of "changed" sockets.
	 */
	int DispatchEvents();

	/** Gets the name of this socket engine as a string.
	 * @return string of socket engine name
	 */
	std::string GetName();

	/** Queues a Write event on the specified event handler.
	 * @param eh EventHandler that needs data sent on
	 */
	void WantWrite(EventHandler* eh);

	/** Posts a completion event on the specified socket.
	 * @param eh EventHandler for message
	 * @param type Event Type
	 * @param param Event Parameter
	 * @return True if added, false if not
	 */
	bool PostCompletionEvent(EventHandler* eh, SocketIOEvent type, int param);

	/** Posts a read event on the specified socket
	 * @param eh EventHandler (socket)
	 */
	void PostReadEvent(EventHandler* eh);

	/** Posts an accept event on the specified socket
	 * @param eh EventHandler (socket)
	 */
	void PostAcceptEvent(EventHandler* eh);

	/** Returns the EventHandler attached to a specific fd.
	 * If the fd isnt in the socketengine, returns NULL.
	 * @param fd The event handler to look for
	 * @return A pointer to the event handler, or NULL
	 */
	EventHandler* GetRef(int fd);

	/** Returns true if a file descriptor exists in
	 * the socket engine's list.
	 * @param fd The event handler to look for
	 * @return True if this fd has an event handler
	 */
	bool HasFd(int fd);

	/** Returns the EventHandler attached to a specific fd.
	 * If the fd isnt in the socketengine, returns NULL.
	 * @param fd The event handler to look for
	 * @return A pointer to the event handler, or NULL
	 */
	EventHandler* GetIntRef(int fd);

	/** Holds the preallocated buffer passed to WSARecvFrom
	 * function. Yes, I know, it's a dirty hack.
	 */
	udp_overlap * udp_ov;
};

/** Creates a SocketEngine
 */
class SocketEngineFactory
{
public:
	/** Create a new instance of SocketEngine based on IOCPEngine
	 */
	SocketEngine* Create(InspIRCd* Instance) { return new IOCPEngine(Instance); }
};

#endif

\ No newline at end of file