eventmachine 1.2.0.dev.2-x64-mingw32

data/ext/em.h

@@ -0,0 +1,308 @@
+/*****************************************************************************
+
+$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>
+  #ifdef HAVE_RB_THREAD_FD_SELECT
+    #define EmSelect rb_thread_fd_select
+  #else
+    // ruby 1.9.1 and below
+    #define EmSelect rb_thread_select
+  #endif
+
+  #ifdef HAVE_RB_THREAD_CALL_WITHOUT_GVL
+   #include <ruby/thread.h>
+  #endif
+
+  #ifdef HAVE_RB_WAIT_FOR_SINGLE_FD
+    #include <ruby/io.h>
+  #endif
+
+  #if defined(HAVE_RB_TRAP_IMMEDIATE)
+    #include <rubysig.h>
+  #elif defined(HAVE_RB_ENABLE_INTERRUPT)
+    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
+
+#if !defined(HAVE_TYPE_RB_FDSET_T)
+#define fd_check(n) (((n) < FD_SETSIZE) ? 1 : 0*fprintf(stderr, "fd %d too large for select\n", (n)))
+// These definitions are cribbed from include/ruby/intern.h in Ruby 1.9.3,
+// with this change: any macros that read or write the nth element of an
+// fdset first call fd_check to make sure n is in bounds.
+typedef fd_set rb_fdset_t;
+#define rb_fd_zero(f) FD_ZERO(f)
+#define rb_fd_set(n, f) do { if (fd_check(n)) FD_SET((n), (f)); } while(0)
+#define rb_fd_clr(n, f) do { if (fd_check(n)) FD_CLR((n), (f)); } while(0)
+#define rb_fd_isset(n, f) (fd_check(n) ? FD_ISSET((n), (f)) : 0)
+#define rb_fd_copy(d, s, n) (*(d) = *(s))
+#define rb_fd_dup(d, s) (*(d) = *(s))
+#define rb_fd_resize(n, f)  ((void)(f))
+#define rb_fd_ptr(f)  (f)
+#define rb_fd_init(f) FD_ZERO(f)
+#define rb_fd_init_copy(d, s) (*(d) = *(s))
+#define rb_fd_term(f) ((void)(f))
+#define rb_fd_max(f)  FD_SETSIZE
+#define rb_fd_select(n, rfds, wfds, efds, timeout)  \
+  select(fd_check((n)-1) ? (n) : FD_SETSIZE, (rfds), (wfds), (efds), (timeout))
+#define rb_thread_fd_select(n, rfds, wfds, efds, timeout)  \
+  rb_thread_select(fd_check((n)-1) ? (n) : FD_SETSIZE, (rfds), (wfds), (efds), (timeout))
+#endif
+
+
+// This Solaris fix is adapted from eval_intern.h in Ruby 1.9.3:
+// Solaris sys/select.h switches select to select_large_fdset to support larger
+// file descriptors if FD_SETSIZE is larger than 1024 on 32bit environment.
+// But Ruby doesn't change FD_SETSIZE because fd_set is allocated dynamically.
+// So following definition is required to use select_large_fdset.
+#ifdef HAVE_SELECT_LARGE_FDSET
+#define select(n, r, w, e, t) select_large_fdset((n), (r), (w), (e), (t))
+extern "C" {
+  int select_large_fdset(int, fd_set *, fd_set *, fd_set *, struct timeval *);
+}
+#endif
+
+class EventableDescriptor;
+class InotifyDescriptor;
+struct SelectData_t;
+
+/*************
+enum Poller_t
+*************/
+enum Poller_t {
+	Poller_Default, // typically Select
+	Poller_Epoll,
+	Poller_Kqueue
+};
+
+
+/********************
+class EventMachine_t
+********************/
+
+class EventMachine_t
+{
+	public:
+		static int GetMaxTimerCount();
+		static void SetMaxTimerCount (int);
+
+		static int GetSimultaneousAcceptCount();
+		static void SetSimultaneousAcceptCount (int);
+
+	public:
+		EventMachine_t (EMCallback, Poller_t);
+		virtual ~EventMachine_t();
+
+		bool RunOnce();
+		void Run();
+		void ScheduleHalt();
+		bool Stopping();
+		void SignalLoopBreaker();
+		const uintptr_t InstallOneshotTimer (int);
+		const uintptr_t ConnectToServer (const char *, int, const char *, int);
+		const uintptr_t ConnectToUnixServer (const char *);
+
+		const uintptr_t CreateTcpServer (const char *, int);
+		const uintptr_t OpenDatagramSocket (const char *, int);
+		const uintptr_t CreateUnixDomainServer (const char*);
+		const uintptr_t AttachSD (SOCKET);
+		const uintptr_t OpenKeyboard();
+		//const char *Popen (const char*, const char*);
+		const uintptr_t Socketpair (char* const*);
+
+		void Add (EventableDescriptor*);
+		void Modify (EventableDescriptor*);
+		void Deregister (EventableDescriptor*);
+
+		const uintptr_t AttachFD (SOCKET, 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 uintptr_t WatchFile (const char*);
+		void UnwatchFile (int);
+		void UnwatchFile (const uintptr_t);
+
+		#ifdef HAVE_KQUEUE
+		void _HandleKqueueFileEvent (struct kevent*);
+		void _RegisterKqueueFileEvent(int);
+		#endif
+
+		const uintptr_t WatchPid (int);
+		void UnwatchPid (int);
+		void UnwatchPid (const uintptr_t);
+
+		#ifdef HAVE_KQUEUE
+		void _HandleKqueuePidEvent (struct kevent*);
+		#endif
+
+		uint64_t GetCurrentLoopTime() { return MyCurrentLoopTime; }
+
+		void QueueHeartbeat(EventableDescriptor*);
+		void ClearHeartbeat(uint64_t, EventableDescriptor*);
+
+		uint64_t GetRealTime();
+
+		Poller_t GetPoller() { return Poller; }
+
+		static bool name2address (const char *server, int port, struct sockaddr *addr, size_t *addr_len);
+
+	private:
+		void _RunTimers();
+		void _UpdateTime();
+		void _AddNewDescriptors();
+		void _ModifyDescriptors();
+		void _InitializeLoopBreaker();
+		void _CleanupSockets();
+
+		void _RunSelectOnce();
+		void _RunEpollOnce();
+		void _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;
+
+		SOCKET LoopBreakerReader;
+		SOCKET LoopBreakerWriter;
+		#ifdef OS_WIN32
+		struct sockaddr_in LoopBreakerTarget;
+		#endif
+
+		timeval Quantum;
+
+		uint64_t MyCurrentLoopTime;
+
+		#ifdef OS_WIN32
+		unsigned TickCountTickover;
+		unsigned LastTickCount;
+		#endif
+
+		#ifdef OS_DARWIN
+		mach_timebase_info_data_t mach_timebase;
+		#endif
+
+	private:
+		bool bTerminateSignalReceived;
+		SelectData_t *SelectData;
+
+		Poller_t Poller;
+
+		int epfd; // Epoll file-descriptor
+		#ifdef HAVE_EPOLL
+		struct epoll_event epoll_events [MaxEvents];
+		#endif
+
+		int kqfd; // Kqueue file-descriptor
+		#ifdef HAVE_KQUEUE
+		struct kevent Karray [MaxEvents];
+		#endif
+
+		#ifdef HAVE_INOTIFY
+		InotifyDescriptor *inotify; // pollable descriptor for our inotify instance
+		#endif
+};
+
+
+/*******************
+struct SelectData_t
+*******************/
+
+struct SelectData_t
+{
+	SelectData_t();
+	~SelectData_t();
+
+	int _Select();
+	void _Clear();
+
+	SOCKET maxsocket;
+	rb_fdset_t fdreads;
+	rb_fdset_t fdwrites;
+	rb_fdset_t fderrors;
+	timeval tv;
+	int nSockets;
+};
+
+#endif // __EventMachine__H_

data/ext/eventmachine.h

@@ -0,0 +1,143 @@
+/*****************************************************************************
+
+$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
+	};
+
+	enum { // SSL/TLS Protocols
+		EM_PROTO_SSLv2 = 2,
+		EM_PROTO_SSLv3 = 4,
+		EM_PROTO_TLSv1 = 8,
+		EM_PROTO_TLSv1_1 = 16,
+		EM_PROTO_TLSv1_2 = 32
+	};
+
+	void evma_initialize_library (EMCallback);
+	bool evma_run_machine_once();
+	void evma_run_machine();
+	void evma_release_library();
+	const uintptr_t evma_install_oneshot_timer (int seconds);
+	const uintptr_t evma_connect_to_server (const char *bind_addr, int bind_port, const char *server, int port);
+	const uintptr_t evma_connect_to_unix_server (const char *server);
+
+	const uintptr_t evma_attach_fd (int file_descriptor, int watch_mode);
+	int evma_detach_fd (const uintptr_t binding);
+	int evma_get_file_descriptor (const uintptr_t binding);
+	int evma_is_notify_readable (const uintptr_t binding);
+	void evma_set_notify_readable (const uintptr_t binding, int mode);
+	int evma_is_notify_writable (const uintptr_t binding);
+	void evma_set_notify_writable (const uintptr_t binding, int mode);
+
+	int evma_pause(const uintptr_t binding);
+	int evma_is_paused(const uintptr_t binding);
+	int evma_resume(const uintptr_t binding);
+
+	int evma_num_close_scheduled();
+
+	void evma_stop_tcp_server (const uintptr_t binding);
+	const uintptr_t evma_create_tcp_server (const char *address, int port);
+	const uintptr_t evma_create_unix_domain_server (const char *filename);
+	const uintptr_t evma_attach_sd (int sd);
+	const uintptr_t evma_open_datagram_socket (const char *server, int port);
+	const uintptr_t evma_open_keyboard();
+	void evma_set_tls_parms (const uintptr_t binding, const char *privatekey_filename, const char *certchain_filenane, int verify_peer, int fail_if_no_peer_cert, const char *sni_hostname, const char *cipherlist, const char *ecdh_curve, const char *dhparam, int protocols);
+	void evma_start_tls (const uintptr_t binding);
+
+	#ifdef WITH_SSL
+	X509 *evma_get_peer_cert (const uintptr_t binding);
+	int evma_get_cipher_bits (const uintptr_t binding);
+	const char *evma_get_cipher_name (const uintptr_t binding);
+	const char *evma_get_cipher_protocol (const uintptr_t binding);
+	const char *evma_get_sni_hostname (const uintptr_t binding);
+	void evma_accept_ssl_peer (const uintptr_t binding);
+	#endif
+
+	int evma_get_peername (const uintptr_t binding, struct sockaddr*, socklen_t*);
+	int evma_get_sockname (const uintptr_t binding, struct sockaddr*, socklen_t*);
+	int evma_get_subprocess_pid (const uintptr_t binding, pid_t*);
+	int evma_get_subprocess_status (const uintptr_t binding, int*);
+	int evma_get_connection_count();
+	int evma_send_data_to_connection (const uintptr_t binding, const char *data, int data_length);
+	int evma_send_datagram (const uintptr_t binding, const char *data, int data_length, const char *address, int port);
+	float evma_get_comm_inactivity_timeout (const uintptr_t binding);
+	int evma_set_comm_inactivity_timeout (const uintptr_t binding, float value);
+	float evma_get_pending_connect_timeout (const uintptr_t binding);
+	int evma_set_pending_connect_timeout (const uintptr_t binding, float value);
+	int evma_get_outbound_data_size (const uintptr_t binding);
+	uint64_t evma_get_last_activity_time (const uintptr_t binding);
+	int evma_send_file_data_to_connection (const uintptr_t binding, const char *filename);
+
+	void evma_close_connection (const uintptr_t binding, int after_writing);
+	int evma_report_connection_error_status (const uintptr_t binding);
+	void evma_signal_loopbreak();
+	void evma_set_timer_quantum (int);
+	int evma_get_max_timer_count();
+	void evma_set_max_timer_count (int);
+	int evma_get_simultaneous_accept_count();
+	void evma_set_simultaneous_accept_count (int);
+	void evma_setuid_string (const char *username);
+	void evma_stop_machine();
+	bool evma_stopping();
+	float evma_get_heartbeat_interval();
+	int evma_set_heartbeat_interval(float);
+
+	const uintptr_t evma_popen (char * const*cmd_strings);
+
+	const uintptr_t evma_watch_filename (const char *fname);
+	void evma_unwatch_filename (const uintptr_t binding);
+
+	const uintptr_t evma_watch_pid (int);
+	void evma_unwatch_pid (const uintptr_t binding);
+
+	void evma_start_proxy(const uintptr_t from, const uintptr_t to, const unsigned long bufsize, const unsigned long length);
+	void evma_stop_proxy(const uintptr_t from);
+	unsigned long evma_proxied_bytes(const uintptr_t from);
+
+	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_
+