C++实现epoll回射服务器,下载地址
------------------------------------------------------------------------------------------------------------------------------------------
#include <iostream>
#include <string>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/epoll.h>
#define MAXSIZE 1024
using namespace std;
class server
{
private:
int m_port;
int m_sfd;
struct sockaddr_in serveraddr;
int epfd;
public:
server(int port = 9999);
~server();
//启动
void running();
};
/* -------------------------------------------*/
/**
* @brief running
*/
/* -------------------------------------------*/
void server::running()
{
struct epoll_event events[MAXSIZE];
int nready = 0;
while(1)
{
nready = epoll_wait(this->epfd,events,MAXSIZE,-1);
if(nready > 0)
{
int i;
for(i = 0;i<nready;i++)
{
int tmpfd = events[i].data.fd;
if(tmpfd == m_sfd)//监听
{
struct sockaddr_in clientaddr;
socklen_t c_len = sizeof(clientaddr);
int cfd = accept(tmpfd,(struct sockaddr*)&clientaddr,&c_len);
struct epoll_event event;
event.events = EPOLLIN;
event.data.fd = cfd;
//join tree
epoll_ctl(this->epfd,EPOLL_CTL_ADD,cfd,&event);
}
else
{
char buf[1024] = {0};
int len = read(tmpfd,buf,sizeof(buf));
if(0 >= len)
{
epoll_ctl(this->epfd,EPOLL_CTL_DEL,tmpfd,NULL);
shutdown(tmpfd,SHUT_RDWR);
}
write(tmpfd,buf,len);
}
}
}
}
}
/* -------------------------------------------*/
/**
* @brief init socket
*
* @param port
*/
/* -------------------------------------------*/
server::server(int port):m_port(port),m_sfd(-1)
{
//创建套接字
m_sfd = socket(AF_INET,SOCK_STREAM,0);
int ret = 0;
if(-1 == m_sfd)
{
throw "创建套接字\n";
}
//端口复用
int opt = 1;
setsockopt(m_sfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
serveraddr.sin_family = AF_INET;
serveraddr.sin_port = htons(m_port);
serveraddr.sin_addr.s_addr = htonl(INADDR_ANY);
//绑定
ret = bind(m_sfd,(struct sockaddr*)&serveraddr,sizeof(serveraddr));
if(-1 == ret)
{
throw "绑定失败\n";
}
//监听
ret = listen(m_sfd,128);
if(-1 == ret)
{
throw "监听失败";
}
//创建树根
this->epfd = epoll_create(MAXSIZE);
if(-1 == this->epfd)
{
throw "epoll_create is wrong";
}
struct epoll_event event;
event.events = EPOLLIN;
event.data.fd = m_sfd;
//加入树中
epoll_ctl(this->epfd,EPOLLIN,m_sfd,&event);
}
/* -------------------------------------------*/
/**
* @brief close socket
*/
/* -------------------------------------------*/
server::~server()
{
//关闭
shutdown(this->m_sfd, SHUT_RDWR);
}
int main(int argc, char *argv[])
{
try
{
server *s = new server(8888);
s->running();
delete s;
}
catch(const char *str)
{
cout << str << endl;
}
return 0;
}