eventmachine-le 1.1.0.beta.1

data/ext/em.h

@@ -0,0 +1,243 @@
+/*****************************************************************************
+
+$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.
+
+*****************************************************************************/
+
+#ifndef __EventMachine__H_
+#define __EventMachine__H_
+
+#ifdef BUILD_FOR_RUBY
+  #include <ruby.h>
+  #define EmSelect rb_thread_select
+
+  #if defined(HAVE_RBTRAP)
+    #include <rubysig.h>
+  #elif defined(HAVE_RB_THREAD_CHECK_INTS)
+    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)
+  #else
+    #define TRAP_BEG
+    #define TRAP_END
+  #endif
+
+  // 1.9.0 compat
+  #ifndef RUBY_UBF_IO
+    #define RUBY_UBF_IO RB_UBF_DFL
+  #endif
+  #ifndef RSTRING_PTR
+    #define RSTRING_PTR(str) RSTRING(str)->ptr
+  #endif
+  #ifndef RSTRING_LEN
+    #define RSTRING_LEN(str) RSTRING(str)->len
+  #endif
+  #ifndef RSTRING_LENINT
+    #define RSTRING_LENINT(str) RSTRING_LEN(str)
+  #endif
+#else
+  #define EmSelect select
+#endif
+
+class EventableDescriptor;
+class InotifyDescriptor;
+
+
+/********************
+class EventMachine_t
+********************/
+
+class EventMachine_t
+{
+	public:
+                /* Internal helper to convert strings to internet addresses.
+                 * IPv6-aware.
+                 * Not reentrant or threadsafe, optimized for speed.
+                 */
+                static struct sockaddr *name2address (const char *server, int port, int *family, int *bind_size);
+
+	public:
+		EventMachine_t (EMCallback);
+		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 OpenKeyboard();
+		//const char *Popen (const char*, const char*);
+		const unsigned long Socketpair (char* const*);
+
+		void Add (EventableDescriptor*);
+		void Modify (EventableDescriptor*);
+		void Deregister (EventableDescriptor*);
+
+		const unsigned long AttachFD (int, bool);
+		int DetachFD (EventableDescriptor*);
+		const unsigned long AttachServerFD (int);
+
+		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
+
+		uint64_t GetCurrentLoopTime() { return MyCurrentLoopTime; }
+
+		// Temporary:
+		void _UseEpoll();
+		void _UseKqueue();
+
+		bool UsingKqueue() { return bKqueue; }
+		bool UsingEpoll() { return bEpoll; }
+
+		void QueueHeartbeat(EventableDescriptor*);
+		void ClearHeartbeat(uint64_t, EventableDescriptor*);
+
+		uint64_t GetRealTime();
+
+	private:
+		bool _RunOnce();
+		void _RunTimers();
+		void _UpdateTime();
+		void _AddNewDescriptors();
+		void _ModifyDescriptors();
+		void _InitializeLoopBreaker();
+		void _CleanupSockets();
+
+		bool _RunSelectOnce();
+		bool _RunEpollOnce();
+		bool _RunKqueueOnce();
+
+		void _ModifyEpollEvent (EventableDescriptor*);
+		void _DispatchHeartbeats();
+		timeval _TimeTilNextEvent();
+		void _CleanBadDescriptors();
+
+	public:
+		void _ReadLoopBreaker();
+		void _ReadInotifyEvents();
+        int NumCloseScheduled;
+
+	private:
+		enum {
+			MaxEpollDescriptors = 64*1024,
+			MaxEvents = 4096
+		};
+		int HeartbeatInterval;
+		EMCallback EventCallback;
+
+		class Timer_t: public Bindable_t {
+		};
+
+		multimap<uint64_t, Timer_t> Timers;
+		multimap<uint64_t, EventableDescriptor*> Heartbeats;
+		map<int, Bindable_t*> Files;
+		map<int, Bindable_t*> Pids;
+		vector<EventableDescriptor*> Descriptors;
+		vector<EventableDescriptor*> NewDescriptors;
+		set<EventableDescriptor*> ModifiedDescriptors;
+
+		uint64_t NextHeartbeatTime;
+
+		int LoopBreakerReader;
+		int LoopBreakerWriter;
+		#ifdef OS_WIN32
+		struct sockaddr_in LoopBreakerTarget;
+		#endif
+
+		timeval Quantum;
+
+		uint64_t MyCurrentLoopTime;
+
+		#ifdef OS_WIN32
+		unsigned TickCountTickover;
+		unsigned LastTickCount;
+		#endif
+
+	private:
+		bool bTerminateSignalReceived;
+
+		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;
+	fd_set fderrors;
+	timeval tv;
+	int nSockets;
+};
+
+#endif // __EventMachine__H_

data/ext/eventmachine.h

@@ -0,0 +1,126 @@
+/*****************************************************************************
+
+$Id$
+
+File:     eventmachine.h
+Date:     15Apr06
+
+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.
+
+*****************************************************************************/
+
+#ifndef __EVMA_EventMachine__H_
+#define __EVMA_EventMachine__H_
+
+#if __cplusplus
+extern "C" {
+#endif
+
+	enum { // Event names
+		EM_TIMER_FIRED = 100,
+		EM_CONNECTION_READ = 101,
+		EM_CONNECTION_UNBOUND = 102,
+		EM_CONNECTION_ACCEPTED = 103,
+		EM_CONNECTION_COMPLETED = 104,
+		EM_LOOPBREAK_SIGNAL = 105,
+		EM_CONNECTION_NOTIFY_READABLE = 106,
+		EM_CONNECTION_NOTIFY_WRITABLE = 107,
+		EM_SSL_HANDSHAKE_COMPLETED = 108,
+		EM_SSL_VERIFY = 109,
+		EM_PROXY_TARGET_UNBOUND = 110,
+		EM_PROXY_COMPLETED = 111,
+		EM_CONNECTION_SENDERROR = 112
+
+	};
+
+	void evma_initialize_library (EMCallback);
+	void evma_run_machine();
+	void evma_release_library();
+	const unsigned long evma_install_oneshot_timer (int seconds);
+	const unsigned long evma_connect_to_server (const char *bind_addr, int bind_port, const char *server, int port);
+	const unsigned long evma_connect_to_unix_server (const char *server);
+
+	const unsigned long evma_attach_fd (int file_descriptor, int watch_mode);
+	int evma_detach_fd (const unsigned long binding);
+	const unsigned long evma_attach_server_fd (int file_descriptor);
+	int evma_get_file_descriptor (const unsigned long binding);
+	int evma_is_notify_readable (const unsigned long binding);
+	void evma_set_notify_readable (const unsigned long binding, int mode);
+	int evma_is_notify_writable (const unsigned long binding);
+	void evma_set_notify_writable (const unsigned long binding, int mode);
+	void evma_set_error_handling (const unsigned long binding, int mode);
+
+	int evma_pause(const unsigned long binding);
+	int evma_is_paused(const unsigned long binding);
+	int evma_resume(const unsigned long binding);
+
+    int evma_num_close_scheduled();
+
+	void evma_stop_tcp_server (const unsigned long signature);
+	const unsigned long evma_create_tcp_server (const char *address, int port);
+	const unsigned long evma_create_unix_domain_server (const char *filename);
+	const unsigned long evma_open_datagram_socket (const char *server, int port);
+	const unsigned long evma_open_keyboard();
+	void evma_set_tls_parms (const unsigned long binding, const char *privatekey_filename, const char *certchain_filenane, int verify_peer);
+	void evma_start_tls (const unsigned long binding);
+
+	#ifdef WITH_SSL
+	X509 *evma_get_peer_cert (const unsigned long binding);
+	void evma_accept_ssl_peer (const unsigned long binding);
+	#endif
+
+	int evma_get_peername (const unsigned long binding, struct sockaddr_storage*, socklen_t *len);
+	int evma_get_sockname (const unsigned long binding, struct sockaddr_storage*, socklen_t *len);
+	int evma_get_subprocess_pid (const unsigned long binding, pid_t*);
+	int evma_get_subprocess_status (const unsigned long binding, int*);
+	int evma_get_connection_count();
+	int evma_send_data_to_connection (const unsigned long binding, const char *data, int data_length);
+	int evma_send_datagram (const unsigned long binding, const char *data, int data_length, const char *address, int port);
+	float evma_get_comm_inactivity_timeout (const unsigned long binding);
+	int evma_set_comm_inactivity_timeout (const unsigned long binding, float value);
+	float evma_get_pending_connect_timeout (const unsigned long binding);
+	int evma_set_pending_connect_timeout (const unsigned long binding, float value);
+	int evma_get_outbound_data_size (const unsigned long binding);
+	int evma_send_file_data_to_connection (const unsigned long binding, const char *filename);
+
+	void evma_close_connection (const unsigned long binding, int after_writing);
+	int evma_report_connection_error_status (const unsigned long binding);
+	void evma_signal_loopbreak();
+	void evma_set_timer_quantum (int);
+	void evma_setuid_string (const char *username);
+	void evma_stop_machine();
+	float evma_get_heartbeat_interval();
+	int evma_set_heartbeat_interval(float);
+
+	const unsigned long evma_popen (char * const*cmd_strings);
+
+	const unsigned long evma_watch_filename (const char *fname);
+	void evma_unwatch_filename (const unsigned long);
+
+	const unsigned long evma_watch_pid (int);
+	void evma_unwatch_pid (const unsigned long);
+
+	void evma_start_proxy(const unsigned long, const unsigned long, const unsigned long, const unsigned long);
+	void evma_stop_proxy(const unsigned long);
+
+	int evma_set_rlimit_nofile (int n_files);
+
+	void evma_set_epoll (int use);
+	void evma_set_kqueue (int use);
+
+	uint64_t evma_get_current_loop_time();
+#if __cplusplus
+}
+#endif
+
+
+#endif // __EventMachine__H_
+

data/ext/extconf.rb

@@ -0,0 +1,166 @@
+require 'fileutils'
+require 'mkmf'
+
+def check_libs libs = [], fatal = false
+  libs.all? { |lib| have_library(lib) || (abort("could not find library: #{lib}") if fatal) }
+end
+
+def check_heads heads = [], fatal = false
+  heads.all? { |head| have_header(head) || (abort("could not find header: #{head}") if fatal)}
+end
+
+def add_define(name)
+  $defs.push("-D#{name}")
+end
+
+##
+# OpenSSL:
+
+# override append_library, so it actually appends (instead of prepending)
+# this fixes issues with linking ssl, since libcrypto depends on symbols in libssl
+def append_library(libs, lib)
+  libs + " " + format(LIBARG, lib)
+end
+
+def manual_ssl_config
+  ssl_libs_heads_args = {
+    :unix => [%w[ssl crypto], %w[openssl/ssl.h openssl/err.h]],
+    :mswin => [%w[ssleay32 eay32], %w[openssl/ssl.h openssl/err.h]],
+  }
+
+  dc_flags = ['ssl']
+  dc_flags += ["#{ENV['OPENSSL']}/include", ENV['OPENSSL']] if /linux/ =~ RUBY_PLATFORM and ENV['OPENSSL']
+
+  libs, heads = case RUBY_PLATFORM
+  when /mswin/    ; ssl_libs_heads_args[:mswin]
+  else              ssl_libs_heads_args[:unix]
+  end
+  dir_config(*dc_flags)
+  check_libs(libs) and check_heads(heads)
+end
+
+if ENV['CROSS_COMPILING']
+  openssl_dir = File.expand_path("~/.rake-compiler/builds/openssl-1.0.0a/")
+  if File.exists?(openssl_dir)
+    FileUtils.mkdir_p Dir.pwd+"/openssl/"
+    FileUtils.cp Dir[openssl_dir+"/include/openssl/*.h"], Dir.pwd+"/openssl/", :verbose => true
+    FileUtils.cp Dir[openssl_dir+"/lib*.a"], Dir.pwd, :verbose => true
+    $INCFLAGS << " -I#{Dir.pwd}" # for the openssl headers
+  else
+    STDERR.puts
+    STDERR.puts "**************************************************************************************"
+    STDERR.puts "**** Cross-compiled OpenSSL not found"
+    STDERR.puts "**** Run: hg clone http://bitbucket.org/ged/ruby-pg && cd ruby-pg && rake openssl_libs"
+    STDERR.puts "**************************************************************************************"
+    STDERR.puts
+  end
+end
+
+# Try to use pkg_config first, fixes #73
+if (!ENV['CROSS_COMPILING'] and pkg_config('openssl')) || manual_ssl_config
+  add_define "WITH_SSL"
+else
+  add_define "WITHOUT_SSL"
+end
+
+add_define 'BUILD_FOR_RUBY'
+add_define 'HAVE_RBTRAP' if have_var('rb_trap_immediate', ['ruby.h', 'rubysig.h'])
+add_define "HAVE_TBR" if have_func('rb_thread_blocking_region')# and have_macro('RUBY_UBF_IO', 'ruby.h')
+add_define "HAVE_INOTIFY" if inotify = have_func('inotify_init', 'sys/inotify.h')
+add_define "HAVE_OLD_INOTIFY" if !inotify && have_macro('__NR_inotify_init', 'sys/syscall.h')
+add_define 'HAVE_WRITEV' if have_func('writev', 'sys/uio.h')
+
+have_func('rb_thread_check_ints')
+have_func('rb_time_new')
+
+# Minor platform details between *nix and Windows:
+
+if RUBY_PLATFORM =~ /(mswin|mingw|bccwin)/
+  GNU_CHAIN = ENV['CROSS_COMPILING'] or $1 == 'mingw'
+  OS_WIN32 = true
+  add_define "OS_WIN32"
+else
+  GNU_CHAIN = true
+  OS_UNIX = true
+  add_define 'OS_UNIX'
+
+  add_define "HAVE_KQUEUE" if have_header("sys/event.h") and have_header("sys/queue.h")
+end
+
+# Main platform invariances:
+
+case RUBY_PLATFORM
+when /mswin32/, /mingw32/, /bccwin32/
+  check_heads(%w[windows.h winsock.h], true)
+  check_libs(%w[kernel32 rpcrt4 gdi32], true)
+
+  if GNU_CHAIN
+    CONFIG['LDSHARED'] = "$(CXX) -shared -lstdc++"
+  else
+    $defs.push "-EHs"
+    $defs.push "-GR"
+  end
+
+when /solaris/
+  add_define 'OS_SOLARIS8'
+  check_libs(%w[nsl socket], true)
+
+  if CONFIG['CC'] == 'cc' and `cc -flags 2>&1` =~ /Sun/ # detect SUNWspro compiler
+    # SUN CHAIN
+    add_define 'CC_SUNWspro'
+    $preload = ["\nCXX = CC"] # hack a CXX= line into the makefile
+    $CFLAGS = CONFIG['CFLAGS'] = "-KPIC"
+    CONFIG['CCDLFLAGS'] = "-KPIC"
+    CONFIG['LDSHARED'] = "$(CXX) -G -KPIC -lCstd"
+  else
+    # GNU CHAIN
+    # on Unix we need a g++ link, not gcc.
+    CONFIG['LDSHARED'] = "$(CXX) -shared"
+  end
+
+when /openbsd/
+  # OpenBSD branch contributed by Guillaume Sellier.
+
+  # on Unix we need a g++ link, not gcc. On OpenBSD, linking against libstdc++ have to be explicitly done for shared libs
+  CONFIG['LDSHARED'] = "$(CXX) -shared -lstdc++ -fPIC"
+  CONFIG['LDSHAREDXX'] = "$(CXX) -shared -lstdc++ -fPIC"
+
+when /darwin/
+  # on Unix we need a g++ link, not gcc.
+  # Ff line contributed by Daniel Harple.
+  CONFIG['LDSHARED'] = "$(CXX) " + CONFIG['LDSHARED'].split[1..-1].join(' ')
+
+when /linux/
+  add_define 'HAVE_EPOLL' if have_func('epoll_create', 'sys/epoll.h')
+
+  # on Unix we need a g++ link, not gcc.
+  CONFIG['LDSHARED'] = "$(CXX) -shared"
+
+when /aix/
+  CONFIG['LDSHARED'] = "$(CXX) -shared -Wl,-G -Wl,-brtl"
+
+when /cygwin/
+  # For rubies built with Cygwin, CXX may be set to CC, which is just
+  # a wrapper for gcc.
+  # This will compile, but it will not link to the C++ std library.
+  # Explicitly set CXX to use g++.
+  CONFIG['CXX'] = "g++"
+  # on Unix we need a g++ link, not gcc.
+  CONFIG['LDSHARED'] = "$(CXX) -shared"
+
+else
+  # on Unix we need a g++ link, not gcc.
+  CONFIG['LDSHARED'] = "$(CXX) -shared"
+end
+
+
+# solaris c++ compiler doesn't have make_pair()
+TRY_LINK.sub!('$(CC)', '$(CXX)')
+add_define 'HAVE_MAKE_PAIR' if try_link(<<SRC, '-lstdc++')
+  #include <utility>
+  using namespace std;
+  int main(){ pair<int,int> tuple = make_pair(1,2); }
+SRC
+TRY_LINK.sub!('$(CXX)', '$(CC)')
+
+create_makefile "rubyeventmachine"