309 lines
7.9 KiB
C++
309 lines
7.9 KiB
C++
/*****************************************************************************
|
|
|
|
$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 (uint64_t);
|
|
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 int name2address (const char *server, int port, int socktype, 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 {
|
|
};
|
|
|
|
std::multimap<uint64_t, Timer_t> Timers;
|
|
std::multimap<uint64_t, EventableDescriptor*> Heartbeats;
|
|
std::map<int, Bindable_t*> Files;
|
|
std::map<int, Bindable_t*> Pids;
|
|
std::vector<EventableDescriptor*> Descriptors;
|
|
std::vector<EventableDescriptor*> NewDescriptors;
|
|
std::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_
|