C++实现epoll回射服务器

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;
}