socketengine.cpp

Go to the documentation of this file.

/*       +------------------------------------+
 *       | Inspire Internet Relay Chat Daemon |
 *       +------------------------------------+
 *
 *  Inspire is copyright (C) 2002-2005 ChatSpike-Dev.
 *                       E-mail:
 *                <brain@chatspike.net>
 *                <Craig@chatspike.net>
 *
 * Written by Craig Edwards, Craig McLure, and others.
 * This program is free but copyrighted software; see
 *            the file COPYING for details.
 *
 * ---------------------------------------------------
 */

#include "inspircd_config.h"
#include "globals.h"
#include "inspircd.h"
#ifdef USE_EPOLL
#include <sys/epoll.h>
#define EP_DELAY 5
#endif
#ifdef USE_KQUEUE
#include <sys/types.h>
#include <sys/event.h>
#include <sys/time.h>
#endif
#include <vector>
#include <string>
#include "socketengine.h"

char ref[65535];

SocketEngine::SocketEngine()
{
        log(DEBUG,"SocketEngine::SocketEngine()");
#ifdef USE_EPOLL
        EngineHandle = epoll_create(65535);
#endif
#ifdef USE_KQUEUE
        EngineHandle = kqueue();
#endif
}

SocketEngine::~SocketEngine()
{
        log(DEBUG,"SocketEngine::~SocketEngine()");
#ifdef USE_EPOLL
        close(EngineHandle);
#endif
#ifdef USE_KQUEUE
        close(EngineHandle);
#endif
}

char SocketEngine::GetType(int fd)
{
        if ((fd < 0) || (fd > 65535))
                return X_EMPTY_SLOT;
        /* Mask off the top bit used for 'read/write' state */
        return (ref[fd] & ~0x80);
}

bool SocketEngine::AddFd(int fd, bool readable, char type)
{
        if ((fd < 0) || (fd > 65535))
                return false;
        this->fds.push_back(fd);
        ref[fd] = type;
        if (readable)
        {
                log(DEBUG,"Set readbit");
                ref[fd] |= X_READBIT;
        }
        log(DEBUG,"Add socket %d",fd);
#ifdef USE_EPOLL
        struct epoll_event ev;
        log(DEBUG,"epoll: Add socket to events, ep=%d socket=%d",EngineHandle,fd);
        readable ? ev.events = EPOLLIN | EPOLLET : ev.events = EPOLLOUT | EPOLLET;
        ev.data.fd = fd;
        int i = epoll_ctl(EngineHandle, EPOLL_CTL_ADD, fd, &ev);
        if (i < 0)
        {
                log(DEBUG,"epoll: List insertion failure!");
                return false;
        }
#endif
#ifdef USE_KQUEUE
        struct kevent ke;
        log(DEBUG,"kqueue: Add socket to events, kq=%d socket=%d",EngineHandle,fd);
        EV_SET(&ke, fd, readable ? EVFILT_READ : EVFILT_WRITE, EV_ADD, 0, 0, NULL);
        int i = kevent(EngineHandle, &ke, 1, 0, 0, NULL);
        if (i == -1)
        {
                log(DEBUG,"kqueue: List insertion failure!");
                return false;
        }
#endif
return true;
}

bool SocketEngine::DelFd(int fd)
{
        log(DEBUG,"SocketEngine::DelFd(%d)",fd);

        if ((fd < 0) || (fd > 65535))
                return false;

        bool found = false;
        for (std::vector<int>::iterator i = fds.begin(); i != fds.end(); i++)
        {
                if (*i == fd)
                {
                        fds.erase(i);
                        log(DEBUG,"Deleted fd %d",fd);
                        found = true;
                        break;
                }
        }
#ifdef USE_KQUEUE
        struct kevent ke;
        EV_SET(&ke, fd, ref[fd] && X_READBIT ? EVFILT_READ : EVFILT_WRITE, EV_DELETE, 0, 0, NULL);
        int i = kevent(EngineHandle, &ke, 1, 0, 0, NULL);
        if (i == -1)
        {
                log(DEBUG,"kqueue: Failed to remove socket from queue!");
                return false;
        }
#endif
#ifdef USE_EPOLL
        struct epoll_event ev;
        ref[fd] && X_READBIT ? ev.events = EPOLLIN | EPOLLET : ev.events = EPOLLOUT | EPOLLET;
        ev.data.fd = fd;
        int i = epoll_ctl(EngineHandle, EPOLL_CTL_DEL, fd, &ev);
        if (i < 0)
        {
                log(DEBUG,"epoll: List deletion failure!");
                return false;
        }
#endif
        ref[fd] = 0;
        return found;
}

bool SocketEngine::Wait(std::vector<int> &fdlist)
{
        fdlist.clear();
#ifdef USE_SELECT
        FD_ZERO(&wfdset);
        FD_ZERO(&rfdset);
        timeval tval;
        int sresult;
        for (unsigned int a = 0; a < fds.size(); a++)
        {
                if (ref[fds[a]] & X_READBIT)
                {
                        FD_SET (fds[a], &rfdset);
                }
                else
                {
                        FD_SET (fds[a], &wfdset);
                }
                
        }
        tval.tv_sec = 0;
        tval.tv_usec = 100L;
        sresult = select(FD_SETSIZE, &rfdset, &wfdset, NULL, &tval);
        if (sresult > 0)
        {
                for (unsigned int a = 0; a < fds.size(); a++)
                {
                        if ((FD_ISSET (fds[a], &rfdset)) || (FD_ISSET (fds[a], &wfdset)))
                        {
                                log(DEBUG,"...Adding active %d",fds[a]);
                                fdlist.push_back(fds[a]);
                        }
                }
        }
#endif
#ifdef USE_KQUEUE
        ts.tv_nsec = 10000L;
        ts.tv_sec = 0;
        int i = kevent(EngineHandle, NULL, 0, &ke_list[0], 65535, &ts);
        for (int j = 0; j < i; j++)
                fdlist.push_back(ke_list[j].ident);
#endif
#ifdef USE_EPOLL
        int i = epoll_wait(EngineHandle, events, 65535, 100);
        for (int j = 0; j < i; j++)
                fdlist.push_back(events[j].data.fd);
#endif
        return true;
}

std::string SocketEngine::GetName()
{
#ifdef USE_SELECT
        return "select";
#endif
#ifdef USE_KQUEUE
        return "kqueue";
#endif
#ifdef USE_EPOLL
        return "epoll";
#endif
        return "misconfigured";
}

---