MattHulse-eventmachine 0.0.1

data/ext/em.h

@@ -0,0 +1,228 @@
+/*****************************************************************************
+
+$Id$
+
+File:     em.h
+Date:     06Apr06
+
+Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+Gmail: blackhedd
+
+This program is free software; you can redistribute it and/or modify
+it under the terms of either: 1) the GNU General Public License
+as published by the Free Software Foundation; either version 2 of the
+License, or (at your option) any later version; or 2) Ruby's License.
+
+See the file COPYING for complete licensing information.
+
+*****************************************************************************/
+
+
+
+#ifdef OS_WIN32
+#include "emwin.h"
+#endif
+
+
+// THIS ENTIRE FILE WILL EVENTUALLY BE FOR UNIX BUILDS ONLY.
+//#ifdef OS_UNIX
+
+#ifndef __EventMachine__H_
+#define __EventMachine__H_
+
+#ifdef BUILD_FOR_RUBY
+  #include <ruby.h>
+  #define EmSelect rb_thread_select
+
+  #ifdef HAVE_RBTRAP
+    #include <rubysig.h>
+  #else
+    extern "C" {
+      void rb_enable_interrupt(void);
+      void rb_disable_interrupt(void);
+    }
+
+    #define TRAP_BEG rb_enable_interrupt()
+    #define TRAP_END do { rb_disable_interrupt(); rb_thread_check_ints(); } while(0)
+  #endif
+
+  // 1.9.0 compat
+  #ifndef RUBY_UBF_IO
+    #define RUBY_UBF_IO RB_UBF_DFL
+  #endif
+#else
+  #define EmSelect select
+#endif
+
+
+#ifdef OS_UNIX
+typedef long long Int64;
+#endif
+#ifdef OS_WIN32
+typedef __int64 Int64;
+#endif
+
+extern Int64 gCurrentLoopTime;
+
+class EventableDescriptor;
+class InotifyDescriptor;
+
+
+/********************
+class EventMachine_t
+********************/
+
+class EventMachine_t
+{
+	public:
+		static int GetMaxTimerCount();
+		static void SetMaxTimerCount (int);
+
+	public:
+		EventMachine_t (void(*event_callback)(const unsigned long, int, const char*, const unsigned long));
+		virtual ~EventMachine_t();
+
+		void Run();
+		void ScheduleHalt();
+		void SignalLoopBreaker();
+		const unsigned long InstallOneshotTimer (int);
+		const unsigned long ConnectToServer (const char *, int, const char *, int);
+		const unsigned long ConnectToUnixServer (const char *);
+
+		const unsigned long CreateTcpServer (const char *, int);
+		const unsigned long OpenDatagramSocket (const char *, int);
+		const unsigned long CreateUnixDomainServer (const char*);
+		const unsigned long _OpenFileForWriting (const char*);
+		const unsigned long OpenKeyboard();
+		//const char *Popen (const char*, const char*);
+		const unsigned long Socketpair (char* const*);
+
+		void Add (EventableDescriptor*);
+		void Modify (EventableDescriptor*);
+
+		const unsigned long AttachFD (int, bool);
+		int DetachFD (EventableDescriptor*);
+
+		void ArmKqueueWriter (EventableDescriptor*);
+		void ArmKqueueReader (EventableDescriptor*);
+
+		void SetTimerQuantum (int);
+		static void SetuidString (const char*);
+		static int SetRlimitNofile (int);
+
+		pid_t SubprocessPid;
+		int SubprocessExitStatus;
+
+		int GetConnectionCount();
+		float GetHeartbeatInterval();
+		int SetHeartbeatInterval(float);
+
+		const unsigned long WatchFile (const char*);
+		void UnwatchFile (int);
+		void UnwatchFile (const unsigned long);
+
+		#ifdef HAVE_KQUEUE
+		void _HandleKqueueFileEvent (struct kevent*);
+		void _RegisterKqueueFileEvent(int);
+		#endif
+
+		const unsigned long WatchPid (int);
+		void UnwatchPid (int);
+		void UnwatchPid (const unsigned long);
+
+		#ifdef HAVE_KQUEUE
+		void _HandleKqueuePidEvent (struct kevent*);
+		#endif
+
+		// Temporary:
+		void _UseEpoll();
+		void _UseKqueue();
+
+		bool UsingKqueue() { return bKqueue; }
+		bool UsingEpoll() { return bEpoll; }
+
+	private:
+		bool _RunOnce();
+		bool _RunTimers();
+		void _UpdateTime();
+		void _AddNewDescriptors();
+		void _ModifyDescriptors();
+		void _InitializeLoopBreaker();
+
+		bool _RunSelectOnce();
+		bool _RunEpollOnce();
+		bool _RunKqueueOnce();
+
+		void _ModifyEpollEvent (EventableDescriptor*);
+
+	public:
+		void _ReadLoopBreaker();
+		void _ReadInotifyEvents();
+
+	private:
+		enum {
+			MaxEpollDescriptors = 64*1024,
+			MaxEvents = 4096
+		};
+		int HeartbeatInterval;
+		void (*EventCallback)(const unsigned long, int, const char*, const unsigned long);
+
+		class Timer_t: public Bindable_t {
+		};
+
+		multimap<Int64, Timer_t> Timers;
+		map<int, Bindable_t*> Files;
+		map<int, Bindable_t*> Pids;
+		vector<EventableDescriptor*> Descriptors;
+		vector<EventableDescriptor*> NewDescriptors;
+		set<EventableDescriptor*> ModifiedDescriptors;
+
+		Int64 NextHeartbeatTime;
+
+		int LoopBreakerReader;
+		int LoopBreakerWriter;
+		#ifdef OS_WIN32
+		struct sockaddr_in LoopBreakerTarget;
+		#endif
+
+		timeval Quantum;
+
+	private:
+		bool bEpoll;
+		int epfd; // Epoll file-descriptor
+		#ifdef HAVE_EPOLL
+		struct epoll_event epoll_events [MaxEvents];
+		#endif
+
+		bool bKqueue;
+		int kqfd; // Kqueue file-descriptor
+		#ifdef HAVE_KQUEUE
+		struct kevent Karray [MaxEvents];
+		#endif
+
+		InotifyDescriptor *inotify; // pollable descriptor for our inotify instance
+};
+
+
+/*******************
+struct SelectData_t
+*******************/
+
+struct SelectData_t
+{
+	SelectData_t();
+
+	int _Select();
+
+	int maxsocket;
+	fd_set fdreads;
+	fd_set fdwrites;
+	timeval tv;
+	int nSockets;
+};
+
+
+
+#endif // __EventMachine__H_
+
+//#endif // OS_UNIX

data/ext/emwin.cpp

@@ -0,0 +1,300 @@
+/*****************************************************************************
+
+$Id$
+
+File:     emwin.cpp
+Date:     05May06
+
+Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+Gmail: blackhedd
+
+This program is free software; you can redistribute it and/or modify
+it under the terms of either: 1) the GNU General Public License
+as published by the Free Software Foundation; either version 2 of the
+License, or (at your option) any later version; or 2) Ruby's License.
+
+See the file COPYING for complete licensing information.
+
+*****************************************************************************/
+
+
+// THIS ENTIRE FILE IS FOR WINDOWS BUILDS ONLY
+// INCOMPLETE AND DISABLED FOR NOW.
+#ifdef xOS_WIN32
+
+#include "project.h"
+
+
+// Keep a global variable floating around
+// with the current loop time as set by the Event Machine.
+// This avoids the need for frequent expensive calls to time(NULL);
+time_t gCurrentLoopTime;
+
+
+/******************************
+EventMachine_t::EventMachine_t
+******************************/
+
+EventMachine_t::EventMachine_t (void (*event_callback)(const char*, int, const char*, int)):
+	EventCallback (event_callback),
+	NextHeartbeatTime (0)
+{
+	gTerminateSignalReceived = false;
+	Iocp = NULL;
+}
+
+
+/*******************************
+EventMachine_t::~EventMachine_t
+*******************************/
+
+EventMachine_t::~EventMachine_t()
+{
+	cerr << "EM __dt\n";
+	if (Iocp)
+		CloseHandle (Iocp);
+}
+
+
+/****************************
+EventMachine_t::ScheduleHalt
+****************************/
+
+void EventMachine_t::ScheduleHalt()
+{
+  /* This is how we stop the machine.
+   * This can be called by clients. Signal handlers will probably
+   * set the global flag.
+   * For now this means there can only be one EventMachine ever running at a time.
+   */
+	gTerminateSignalReceived = true;
+}
+
+
+
+/*******************
+EventMachine_t::Run
+*******************/
+
+void EventMachine_t::Run()
+{
+	HookControlC (true);
+
+	Iocp = CreateIoCompletionPort (INVALID_HANDLE_VALUE, NULL, 0, 0);
+	if (Iocp == NULL)
+		throw std::runtime_error ("no completion port");
+
+
+	DWORD nBytes, nCompletionKey;
+	LPOVERLAPPED Overlapped;
+
+	do {
+		gCurrentLoopTime = time(NULL);
+		// Have some kind of strategy that will dequeue maybe up to 10 completions 
+		// without running the timers as long as they are available immediately.
+		// Otherwise in a busy server we're calling them every time through the loop.
+		if (!_RunTimers())
+			break;
+		if (GetQueuedCompletionStatus (Iocp, &nBytes, &nCompletionKey, &Overlapped, 1000)) {
+		}
+		cerr << "+";
+	} while (!gTerminateSignalReceived);
+
+
+	/*
+	while (true) {
+		gCurrentLoopTime = time(NULL);
+		if (!_RunTimers())
+			break;
+		_AddNewDescriptors();
+		if (!_RunOnce())
+			break;
+		if (gTerminateSignalReceived)
+			break;
+	}
+	*/
+
+	HookControlC (false);
+}
+
+
+/**************************
+EventMachine_t::_RunTimers
+**************************/
+
+bool EventMachine_t::_RunTimers()
+{
+	// These are caller-defined timer handlers.
+	// Return T/F to indicate whether we should continue the main loop.
+	// We rely on the fact that multimaps sort by their keys to avoid
+	// inspecting the whole list every time we come here.
+	// Just keep inspecting and processing the list head until we hit
+	// one that hasn't expired yet.
+
+	while (true) {
+		multimap<time_t,Timer_t>::iterator i = Timers.begin();
+		if (i == Timers.end())
+			break;
+		if (i->first > gCurrentLoopTime)
+			break;
+		if (EventCallback)
+			(*EventCallback) (NULL, EM_TIMER_FIRED, NULL, i->second.GetBinding());
+		Timers.erase (i);
+	}
+	return true;
+}
+
+
+/***********************************
+EventMachine_t::InstallOneshotTimer
+***********************************/
+
+const char *EventMachine_t::InstallOneshotTimer (int seconds)
+{
+	if (Timers.size() > MaxOutstandingTimers)
+		return false;
+	// Don't use the global loop-time variable here, because we might
+	// get called before the main event machine is running.
+
+	Timer_t t;
+	Timers.insert (make_pair (time(NULL) + seconds, t));
+	return t.GetBinding();
+}
+
+
+/**********************************
+EventMachine_t::OpenDatagramSocket
+**********************************/
+
+const char *EventMachine_t::OpenDatagramSocket (const char *address, int port)
+{
+	cerr << "OPEN DATAGRAM SOCKET\n";
+	return "Unimplemented";
+}
+
+
+/*******************************
+EventMachine_t::CreateTcpServer
+*******************************/
+
+const char *EventMachine_t::CreateTcpServer (const char *server, int port)
+{
+	/* Create a TCP-acceptor (server) socket and add it to the event machine.
+	 * Return the binding of the new acceptor to the caller.
+	 * This binding will be referenced when the new acceptor sends events
+	 * to indicate accepted connections.
+	 */
+
+	const char *output_binding = NULL;
+
+	struct sockaddr_in sin;
+
+	SOCKET sd_accept = socket (AF_INET, SOCK_STREAM, 0);
+	if (sd_accept == INVALID_SOCKET) {
+		goto fail;
+	}
+
+	memset (&sin, 0, sizeof(sin));
+	sin.sin_family = AF_INET;
+	sin.sin_addr.s_addr = INADDR_ANY;
+	sin.sin_port = htons (port);
+
+	if (server && *server) {
+		sin.sin_addr.s_addr = inet_addr (server);
+		if (sin.sin_addr.s_addr == INADDR_NONE) {
+			hostent *hp = gethostbyname (server);
+			if (hp == NULL) {
+				//__warning ("hostname not resolved: ", server);
+				goto fail;
+			}
+			sin.sin_addr.s_addr = ((in_addr*)(hp->h_addr))->s_addr;
+		}
+	}
+
+
+	// No need to set reuseaddr on Windows.
+
+
+	if (bind (sd_accept, (struct sockaddr*)&sin, sizeof(sin))) {
+		//__warning ("binding failed");
+		goto fail;
+	}
+
+	if (listen (sd_accept, 100)) {
+		//__warning ("listen failed");
+		goto fail;
+	}
+
+	{ // Looking good.
+		AcceptorDescriptor *ad = new AcceptorDescriptor (this, sd_accept);
+		if (!ad)
+			throw std::runtime_error ("unable to allocate acceptor");
+		Add (ad);
+		output_binding = ad->GetBinding();
+
+		CreateIoCompletionPort ((HANDLE)sd_accept, Iocp, NULL, 0);
+		SOCKET sd = socket (AF_INET, SOCK_STREAM, 0);
+		CreateIoCompletionPort ((HANDLE)sd, Iocp, NULL, 0);
+		AcceptEx (sd_accept, sd, 
+	}
+
+	return output_binding;
+
+	fail:
+	if (sd_accept != INVALID_SOCKET)
+		closesocket (sd_accept);
+	return NULL;
+}
+
+
+/*******************************
+EventMachine_t::ConnectToServer
+*******************************/
+
+const char *EventMachine_t::ConnectToServer (const char *server, int port)
+{
+	if (!server || !*server || !port)
+		return NULL;
+
+	sockaddr_in pin;
+	unsigned long HostAddr;
+
+	HostAddr = inet_addr (server);
+	if (HostAddr == INADDR_NONE) {
+		hostent *hp = gethostbyname (server);
+		if (!hp)
+			return NULL;
+		HostAddr = ((in_addr*)(hp->h_addr))->s_addr;
+	}
+
+	memset (&pin, 0, sizeof(pin));
+	pin.sin_family = AF_INET;
+	pin.sin_addr.s_addr = HostAddr;
+	pin.sin_port = htons (port);
+
+	int sd = socket (AF_INET, SOCK_STREAM, 0);
+	if (sd == INVALID_SOCKET)
+		return NULL;
+
+
+	LPOVERLAPPED olap = (LPOVERLAPPED) calloc (1, sizeof (OVERLAPPED));
+	cerr << "I'm dying now\n";
+	throw runtime_error ("UNIMPLEMENTED!!!\n");
+
+}
+
+
+
+/*******************
+EventMachine_t::Add
+*******************/
+
+void EventMachine_t::Add (EventableDescriptor *ed)
+{
+	cerr << "ADD\n";
+}
+
+
+
+#endif // OS_WIN32
+