c++ 封装socket和epoll编程

编程实例：

CSocket.h文件请参考前一篇文章

CSocketEvent.h

#include <sys/epoll.h>
#include <iostream>
#include <cstdio>
#include "CSocket.h"

const int MaxEvents = 256;
const int EpollSize = 500;

class CSocketEvent
{
public:
CSocketEvent();
virtual ~CSocketEvent();
bool eventSet(CSocket *socket, bool readable, bool writeable);
bool eventAdd(CSocket *socket, bool readable, bool writeable);
bool eventDel(CSocket *socket);
int getEvents(int timeout);
struct epoll_event events[MaxEvents];
private:
int epfd;
CSocketEvent(const CSocketEvent&);
CSocketEvent& operator=(const CSocketEvent&);
};

CSocketEvent::CSocketEvent()
{
if((epfd = epoll_create(EpollSize)) < 0)
{
std::cout << "create epoll handle err\n";
return;
}
memset(&events, 0, sizeof(struct epoll_event)*MaxEvents);
}

CSocketEvent::~CSocketEvent()
{
if(epfd > 0)
{
close(epfd);
epfd = -1;
}
}

bool CSocketEvent::eventSet(CSocket *socket, bool readable, bool writeable)
{
int	sockfd = socket->getSockfd();
if(sockfd < 0)
{
std::cout << "eventSet the socket fd err\n";
return false;
}
struct epoll_event ev;
memset(&ev, 0, sizeof(ev));
ev.events |= EPOLLET; //ET modle as default
if(readable)
ev.events |= EPOLLIN;
if(writeable)
ev.events |= EPOLLOUT;
//ev.data.ptr = (void*)sockfd;
return ( 0 == epoll_ctl(epfd, EPOLL_CTL_MOD, sockfd, &ev));
}

bool CSocketEvent::eventAdd(CSocket *socket, bool readable, bool writeable)
{
int sockfd = -1;
if(socket->setNonBlock(true))
sockfd = socket->getSockfd();
if(sockfd < 0)
{
std::cout << "eventAdd the socket fd err\n";
return false;
}
struct epoll_event ev;
memset(&ev, 0, sizeof(ev));
ev.events |= EPOLLET; //ET modle as default
if(readable)
ev.events |= EPOLLIN;
if(writeable)
ev.events |= EPOLLOUT;
//ev.data.ptr = sockfd;
return ( 0 == epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &ev));
}

bool CSocketEvent::eventDel(CSocket *socket)
{
int sockfd = socket->getSockfd();
if(sockfd < 0)
{
std::cout << "eventDel the socket fd err\n";
return false;
}
struct epoll_event ev;
memset(&ev, 0, sizeof(ev));
ev.events = 0;
return (0==epoll_ctl(epfd, EPOLL_CTL_DEL, sockfd, &ev));
}

int CSocketEvent::getEvents(int timeout)
{
int ret = epoll_wait(epfd, events, MaxEvents, timeout);
return ret;
}

TestSocketEvent.cpp

#include "CSocketEvent.h"

int main(int argc, char *argv[])
{
CServSocket servSocket;
CSocketEvent socketEvent;
char buf[256];
memset(buf,0, sizeof(buf));
if(!servSocket.Listen(9999))
{
std::cout << "listen err\n";
}
for(;;)
{
int ret = socketEvent.getEvents(1000);
for(int i=0; i< ret; ++i)
{
if(socketEvent.events[i].data.fd == servSocket.getSockfd())
{
CSocket *socket = NULL;
if((socket = servSocket.Accept()) == NULL)
{
std::cout << "accpt error\n";
break;
}
socketEvent.eventAdd(socket, true, true);
delete socket;
}
else if(socketEvent.events[i].events & EPOLLIN)
{
std::cout << "event trigger ...\n";
int sockfd = socketEvent.events[i].data.fd;
CSocket *socket = new CSocket();
socket->setUp(sockfd, NULL);
socket->Recv(buf, 10, 0);
std::cout << buf << "\n";
socketEvent.eventSet(socket, true, true);
delete socket;
}
}
}
return 0;
}

编译：

g++ -o TestSocketEvent TestSocketEvent.cpp CSocketEvent.h CSocket.h