头文件duye_epoll.h
/***********************************************************************************
**
* &＃64;copyright (c) 2010-2019, Technology Co., LTD. All Right Reserved.
*
************************************************************************************/
/**
* &＃64;file duye_epoll.h
* &＃64;version
* &＃64;brief
* &＃64;author
* &＃64;date 2013-03-29
* &＃64;note
*
* 1. 2013-03-29 Created this file
*/#pragma once#include
#include
#include
#include
#include namespace duye {class EpollEvent {
public:EpollEvent() : m_fd(-1), m_event(0){}EpollEvent(const int32 fd, const uint32 event) : m_fd(fd), m_event(event) {}~EpollEvent() {}/*** &＃64;brief get/set fd* &＃64;return fd*/int32 fd() const { return m_fd; }/*** &＃64;brief get/set events* &＃64;return events*/uint32 event() const { return m_event; }/*** &＃64;brief is received data, for network event* &＃64;return true/false*/bool isRecv() { return m_event & EPOLLIN; }bool isRecv() const { return m_event & EPOLLIN; }/*** &＃64;brief is sent data, for network event* &＃64;return true/false*/bool isSend() { return m_event & EPOLLOUT; }bool isSend() const { return m_event & EPOLLOUT; }/*** &＃64;brief is sent data, for network event* &＃64;return true/false*/ bool isClosed() { return (m_event & EPOLLERR) || (m_event & EPOLLHUP); }bool isClosed() const { return (m_event & EPOLLERR) || (m_event & EPOLLHUP); }/*** &＃64;brief is disconnection, for network event* &＃64;return true/false*/bool isDiscon() { return (m_event &＃61;&＃61; EPOLLERR || m_event &＃61;&＃61; EPOLLHUP) ? true : false; }private:int32 m_fd; uint32 m_event;
};/*** &＃64;brief epoll socket server*/
class Epoll {
public:typedef struct epoll_event SysEvent;typedef std::list EventList;public:Epoll();~Epoll();/*** &＃64;brief open epoll* &＃64;param [in] maxEvent : the number of the max events* &＃64;return true/false*/bool open(const uint32 maxEvents &＃61; 1024);/*** &＃64;brief close epoll* &＃64;return true/false*/bool close(); /*** &＃64;brief add fd* &＃64;param [in] fd : fd* &＃64;param [in] epollMode : epoll events(EPOLLIN, EPOLLOUT, EPOLLPRI, EPOLLERR, * EPOLLHUP, EPOLLET, EPOLLONESHOT), default is EPOLLIN | EPOLLOUT* &＃64;return true/false*/ bool addfd(const int32 fd, const uint32 epollMode &＃61; EPOLLIN | EPOLLOUT);/*** &＃64;brief modify fd* &＃64;param [in] fd : fd* &＃64;param [in] epollMode : epoll events(EPOLLIN, EPOLLOUT, EPOLLPRI, EPOLLERR, * EPOLLHUP, EPOLLET, EPOLLONESHOT), default is EPOLLIN | EPOLLOUT* &＃64;return true/false*/ bool modfd(const int32 fd, const uint32 epollMode &＃61; EPOLLIN | EPOLLOUT);/*** &＃64;brief delete fd* &＃64;param [in] fd : fd* &＃64;return true/false*/bool delfd(const int32 fd);/*** &＃64;brief wait event* &＃64;param [out] eventList : return event* &＃64;param [in] timeout : wait time out, default is -1, indicate block, millisecond* &＃64;return true/false*/bool wait(Epoll::EventList& eventList, const uint32 timeout &＃61; -1);/*** &＃64;brief get last error string* &＃64;return error string* &＃64;note */ uint8* error(); private:// create epollbool create();private:int32 m_epollfd; uint32 m_maxEvents;SysEvent* m_sysEvents;Error m_error;
};}
cpp文件duye_epoll.h
#include
#include <string.h>
#include
#include
#include namespace duye {static const int8* DUYE_LOG_PREFIX &＃61; "duye.system.epoll";Epoll::Epoll() : m_epollfd(-1), m_maxEvents(0), m_sysEvents(NULL) {m_error.setPrefix(DUYE_LOG_PREFIX);
}Epoll::~Epoll() {close();
}bool Epoll::open(const uint32 maxEvents) {close();m_maxEvents &＃61; maxEvents;m_epollfd &＃61; epoll_create(m_maxEvents);if (m_epollfd &＃61;&＃61; -1) {ERROR_DUYE_LOG("%s:%d > epoll_create failed \n", __FILE__, __LINE__);return false;} m_sysEvents &＃61; (struct epoll_event*)calloc(m_maxEvents, sizeof(struct epoll_event));if (m_sysEvents &＃61;&＃61; NULL) {ERROR_DUYE_LOG("%s:%d > calloc return NULL \n", __FILE__, __LINE__);return false;}return true;
}bool Epoll::close() {m_maxEvents &＃61; 0;if (m_epollfd !&＃61; -1) {::close(m_epollfd);m_epollfd &＃61; -1;}if (m_sysEvents !&＃61; NULL) {free(m_sysEvents);m_sysEvents &＃61; NULL;} return true;
}bool Epoll::addfd(const int32 fd, const uint32 epollMode) {if (m_epollfd &＃61;&＃61; -1) {ERROR_DUYE_LOG("%s:%d > m_epollfd &＃61;&＃61; -1", __FILE__, __LINE__);return false;}struct epoll_event epollEvent;bzero(&epollEvent, sizeof(struct epoll_event));epollEvent.data.fd &＃61; fd;epollEvent.events &＃61; epollMode;int32 ret &＃61; epoll_ctl(m_epollfd, EPOLL_CTL_ADD, fd, &epollEvent);if (ret !&＃61; 0) {ERROR_DUYE_LOG("[%s:%d] epoll_ctl() return &＃61; %d", __FILE__, __LINE__, ret);return false;}return true;
}bool Epoll::modfd(const int32 fd, const uint32 epollMode) {if (m_epollfd &＃61;&＃61; -1) {ERROR_DUYE_LOG("[%s:%d] m_epollfd &＃61;&＃61; -1", __FILE__, __LINE__);return false;}struct epoll_event epollEvent;bzero(&epollEvent, sizeof(struct epoll_event));epollEvent.data.fd &＃61; fd;epollEvent.events &＃61; epollMode;return epoll_ctl(m_epollfd, EPOLL_CTL_MOD, fd, &epollEvent) &＃61;&＃61; 0 ? true : false;
}bool Epoll::delfd(const int32 fd) {if (m_epollfd &＃61;&＃61; -1) {ERROR_DUYE_LOG("[%s:%d] m_epollfd &＃61;&＃61; -1", __FILE__, __LINE__);return false;}return epoll_ctl(m_epollfd, EPOLL_CTL_DEL, fd, NULL) &＃61;&＃61; 0 ? true : false;
}bool Epoll::wait(Epoll::EventList& eventList, const uint32 timeout) {if (m_epollfd &＃61;&＃61; -1) {ERROR_DUYE_LOG("[%s:%d] m_epollfd &＃61;&＃61; -1", __FILE__, __LINE__);return false;}int32 event_count &＃61; epoll_wait(m_epollfd, m_sysEvents, m_maxEvents, timeout);if (event_count <&＃61; 0) {return false;}for (uint32 i &＃61; 0; i <(uint32)event_count; i&＃43;&＃43;) {eventList.push_back(EpollEvent(m_sysEvents[i].data.fd, m_sysEvents[i].events));}return true;
}uint8* Epoll::error() {return m_error.error;
}}
京公网安备 11010802041100号