linux epoll的介紹,操作和具體實現

Epoll基本介紹
 
在linux的網絡編程中,很長的時間都在使用select來做事件觸發。在linux新的內核中,有了一種替換它的機制,就是epoll。相比於 select,epoll最大的好處在於它不會隨着監聽fd數目的增長而降低效率。因爲在內核中的select實現中,它是採用輪詢來處理的,輪詢的fd 數目越多,自然耗時越多。並且,在linux/posix_types.h頭文件有這樣的聲明:
#define __FD_SETSIZE    1024
表示select最多同時監聽1024個fd,當然,可以通過修改頭文件再重編譯內核來擴大這個數目,但這似乎並不治本。
epoll的接口非常簡單,一共就三個函數:
1.int epoll_create( int size ); //int close(int epfd);
創建一個epoll的句柄,size用來告訴內核這個監聽的數目一共有多大。這個參數不同於select()中的第一個參數,給出最大監聽的fd+1的值。需要注意的是,當創建好epoll句柄後,它就是會佔用一個fd值,在linux下如果查看/proc/進程id/fd/,是能夠看到這個fd的,所以在使用完epoll後,必須調用close()關閉,否則可能導致fd被耗盡。
2.int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event );
epoll的事件註冊函數,它不同與select()是在監聽事件時告訴內核要監聽什麼類型的事件,而是在這裏先註冊要監聽的事件類型。第一個參數是epoll_create()的返回值,第二個參數表示動作,用三個宏來表示:
EPOLL_CTL_ADD:註冊新的fd到epfd中;
EPOLL_CTL_MOD:修改已經註冊的fd的監聽事件;
EPOLL_CTL_DEL:從epfd中刪除一個fd;
第三個參數是需要監聽的fd,第四個參數是告訴內核需要監聽什麼事,struct epoll_event結構如下:
struct epoll_event
{
       __uint32_t events; /* Epoll events */
    epoll_data_t data; /* User data variable */
}
events可以是以下幾個宏的集合:
EPOLLIN :表示對應的文件描述符可以讀(包括對端SOCKET正常關閉);
EPOLLOUT:表示對應的文件描述符可以寫;
EPOLLPRI:表示對應的文件描述符有緊急的數據可讀(這裏應該表示有帶外數據到來);
EPOLLERR:表示對應的文件描述符發生錯誤;
EPOLLHUP:表示對應的文件描述符被掛斷;
EPOLLET: 將EPOLL設爲邊緣觸發(Edge Triggered)模式,這是相對於水平觸發(Level Triggered)來說的。
EPOLLONESHOT:只監聽一次事件,當監聽完這次事件之後,如果還需要繼續監聽這個socket的話,需要再次把這個socket加入到EPOLL隊列裏
3. int epoll_wait(int epfd, struct epoll_event * events, int maxevents, int timeout);
等 待事件的產生,類似於select()調用。參數events用來從內核得到事件的集合,maxevents告之內核這個events有多大,這個 maxevents的值不能大於創建epoll_create()時的size,參數timeout是超時時間(毫秒,0會立即返回,-1將不確定,也有說法說是永久阻塞)。該函數返回需要處理的事件數目,如返回0表示已超時。
從man手冊中,得到ET和LT的具體描述如下

EPOLL事件有兩種模型:
Edge Triggered (ET)
Level Triggered (LT)

假如有這樣一個例子:
1. 我們已經把一個用來從管道中讀取數據的文件句柄(RFD)添加到epoll描述符
2. 這個時候從管道的另一端被寫入了2KB的數據
3. 調用epoll_wait(2),並且它會返回RFD,說明它已經準備好讀取操作
4. 然後我們讀取了1KB的數據
5. 調用epoll_wait(2)......

Edge Triggered 工作模式:
如果我們在第1步將RFD添加到epoll描述符的時候使用了EPOLLET標誌,那麼在第5步調用epoll_wait(2)之後將有可能會掛起,因爲剩餘的數據還存在於文件的輸入緩衝區內,而且數據發出端還在等待一個針對已經發出數據的反饋信息。只有在監視的文件句柄上發生了某個事件的時候 ET 工作模式纔會彙報事件。因此在第5步的時候,調用者可能會放棄等待仍在存在於文件輸入緩衝區內的剩餘數據。在上面的例子中,會有一個事件產生在RFD句柄上,因爲在第2步執行了一個寫操作,然後,事件將會在第3步被銷燬。因爲第4步的讀取操作沒有讀空文件輸入緩衝區內的數據,因此我們在第5步調用 epoll_wait(2)完成後,是否掛起是不確定的。epoll工作在ET模式的時候,必須使用非阻塞套接口,以避免由於一個文件句柄的阻塞讀/阻塞寫操作把處理多個文件描述符的任務餓死。最好以下面的方式調用ET模式的epoll接口,在後面會介紹避免可能的缺陷。
   i    基於非阻塞文件句柄
   ii   只有當read(2)或者write(2)返回EAGAIN時才需要掛起,等待。但這並不是說每次read()時都需要循環讀,直到讀到產生一個EAGAIN才認爲此次事件處理完成,當read()返回的讀到的數據長度小於請求的數據長度時,就可以確定此時緩衝中已沒有數據了,也就可以認爲此事讀事件已處理完成。

Level Triggered 工作模式
相反的,以LT方式調用epoll接口的時候,它就相當於一個速度比較快的poll(2),並且無論後面的數據是否被使用,因此他們具有同樣的職能。因爲即使使用ET模式的epoll,在收到多個chunk的數據的時候仍然會產生多個事件。調用者可以設定EPOLLONESHOT標誌,在 epoll_wait(2)收到事件後epoll會與事件關聯的文件句柄從epoll描述符中禁止掉。因此當EPOLLONESHOT設定後,使用帶有 EPOLL_CTL_MOD標誌的epoll_ctl(2)處理文件句柄就成爲調用者必須作的事情。


然後詳細解釋ET, LT:

LT(level triggered)是缺省的工作方式,並且同時支持block和no-block socket.在這種做法中,內核告訴你一個文件描述符是否就緒了,然後你可以對這個就緒的fd進行IO操作。如果你不作任何操作,內核還是會繼續通知你的,所以,這種模式編程出錯誤可能性要小一點。傳統的select/poll都是這種模型的代表.

ET(edge-triggered) 是高速工作方式,只支持no-block socket。在這種模式下,當描述符從未就緒變爲就緒時,內核通過epoll告訴你。然後它會假設你知道文件描述符已經就緒,並且不會再爲那個文件描述符發送更多的就緒通知,直到你做了某些操作導致那個文件描述符不再爲就緒狀態了(比如,你在發送,接收或者接收請求,或者發送接收的數據少於一定量時導致了一個EWOULDBLOCK 錯誤)。但是請注意,如果一直不對這個fd作IO操作(從而導致它再次變成未就緒),內核不會發送更多的通知(only once),不過在TCP協議中,ET模式的加速效用仍需要更多的benchmark確認(這句話不理解)。

在 許多測試中我們會看到如果沒有大量的idle -connection或者dead-connection,epoll的效率並不會比select/poll高很多,但是當我們遇到大量的idle- connection(例如WAN環境中存在大量的慢速連接),就會發現epoll的效率大大高於select/poll。(未測試)

 

另外,當使用epoll的ET模型來工作時,當產生了一個EPOLLIN事件後,
讀數據的時候需要考慮的是當recv()返回的大小如果等於請求的大小,那麼很有可能是緩衝區還有數據未讀完,也意味着該次事件還沒有處理完,所以還需要再次讀取:
while(rs)
{
buflen = recv(activeevents[i].data.fd, buf, sizeof(buf), 0);
if(buflen < 0)
{
    // 由於是非阻塞的模式,所以當errno爲EAGAIN時,表示當前緩衝區已無數據可讀
    // 在這裏就當作是該次事件已處理處.
    if(errno == EAGAIN)
     break;
    else
     return;
   }
   else if(buflen == 0)
   {
     // 這裏表示對端的socket已正常關閉.
   }
   if(buflen == sizeof(buf)
     rs = 1;   // 需要再次讀取
   else
     rs = 0;
}


還有,假如發送端流量大於接收端的流量(意思是epoll所在的程序讀比轉發的socket要快),由於是非阻塞的socket,那麼send()函數雖然返回,但實際緩衝區的數據並未真正發給接收端,這樣不斷的讀和發,當緩衝區滿後會產生EAGAIN錯誤(參考man send),同時,不理會這次請求發送的數據.所以,需要封裝socket_send()的函數用來處理這種情況,該函數會盡量將數據寫完再返回,返回- 1表示出錯。在socket_send()內部,當寫緩衝已滿(send()返回-1,且errno爲EAGAIN),那麼會等待後再重試.這種方式並不很完美,在理論上可能會長時間的阻塞在socket_send()內部,但暫沒有更好的辦法.

ssize_t socket_send(int sockfd, const char* buffer, size_t buflen)
{
ssize_t tmp;
size_t total = buflen;
const char *p = buffer;

while(1)
{
    tmp = send(sockfd, p, total, 0);
    if(tmp < 0)
    {
      // 當send收到信號時,可以繼續寫,但這裏返回-1.
      if(errno == EINTR)
        return -1;

      // 當socket是非阻塞時,如返回此錯誤,表示寫緩衝隊列已滿,
      // 在這裏做延時後再重試.
      if(errno == EAGAIN)
      {
        usleep(1000);
        continue;
      }

      return -1;
    }

    if((size_t)tmp == total)
      return buflen;

    total -= tmp;
    p += tmp;
}

return tmp;
}


epoll用到的所有函數都是在頭文件sys/epoll.h中聲明,有什麼地方不明白或函數忘記了可以去看一下。
epoll和select相比,最大不同在於:

1epoll返回時已經明確的知道哪個sokcet fd發生了事件,不用再一個個比對。這樣就提高了效率。
2select的FD_SETSIZE是有限止的,而epoll是沒有限止的只與系統資源有關。

1、epoll_create函數
函數聲明:int epoll_create(int size)
該函數生成一個epoll專用的文件描述符。它其實是在內核申請一空間,用來存放你想關注的socket fd上是否發生以及發生了什麼事件。size就是你在這個epoll fd上能關注的最大socket fd數。隨你定好了。只要你有空間。可參見上面與select之不同2.

22、epoll_ctl函數
函數聲明:int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event)
該函數用於控制某個epoll文件描述符上的事件,可以註冊事件,修改事件,刪除事件。
參數:
epfd:由 epoll_create 生成的epoll專用的文件描述符;
op:要進行的操作例如註冊事件,可能的取值EPOLL_CTL_ADD 註冊、EPOLL_CTL_MOD 修 改、EPOLL_CTL_DEL 刪除

fd:關聯的文件描述符;
event:指向epoll_event的指針;
如果調用成功返回0,不成功返回-1

用到的數據結構
typedef union epoll_data {
void *ptr;
int fd;
__uint32_t u32;
__uint64_t u64;
} epoll_data_t;

struct epoll_event {
__uint32_t events; /* Epoll events */
epoll_data_t data; /* User data variable */
};


如:
struct epoll_event ev;
//設置與要處理的事件相關的文件描述符
ev.data.fd=listenfd;
//設置要處理的事件類型
ev.events=EPOLLIN|EPOLLET;
//註冊epoll事件
epoll_ctl(epfd,EPOLL_CTL_ADD,listenfd,&ev);


常用的事件類型:
EPOLLIN :表示對應的文件描述符可以讀;
EPOLLOUT:表示對應的文件描述符可以寫;
EPOLLPRI:表示對應的文件描述符有緊急的數據可讀
EPOLLERR:表示對應的文件描述符發生錯誤;
EPOLLHUP:表示對應的文件描述符被掛斷;
EPOLLET:表示對應的文件描述符有事件發生;


3、epoll_wait函數
函數聲明:int epoll_wait(int epfd,struct epoll_event * events,int maxevents,int timeout)
該函數用於輪詢I/O事件的發生;
參數:
epfd:由epoll_create 生成的epoll專用的文件描述符;
epoll_event:用於回傳代處理事件的數組;
maxevents:每次能處理的事件數;
timeout:等待I/O事件發生的超時值(單位我也不太清楚);-1相當於阻塞,0相當於非阻塞。一般用-1即可
返回發生事件數。


用法如下:

/*build the epoll enent for recall */
struct epoll_event ev_read[20];
int nfds = 0; //return the events count
nfds=epoll_wait(epoll_fd,ev_read,20, -1);
for(i=0; i
{
if(ev_read[i].data.fd == sock)// the listener port hava data
......

epoll_wait運行的原理是
等侍註冊在epfd上的socket fd的事件的發生,如果發生則將發生的sokct fd和事件類型放入到events數組中。
並且將註冊在epfd上的socket fd的事件類型給清空,所以如果下一個循環你還要關注這個socket fd的話,則需要用epoll_ctl(epfd,EPOLL_CTL_MOD,listenfd,&ev)來重新設置socket fd的事件類型。這時不用EPOLL_CTL_ADD,因爲socket fd並未清空,只是事件類型清空。這一步非常重要。
俺最開始就是沒有加這個,白搞了一個上午。

4單個epoll並不能解決所有問題,特別是你的每個操作都比較費時的時候,因爲epoll是串行處理的。
所以你有還是必要建立線程池來發揮更大的效能。
 
例子:
 
#include <iostream>
#include <sys/socket.h>
#include <sys/epoll.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdio.h>
#define MAXLINE 10
#define OPEN_MAX 100
#define LISTENQ 20
#define SERV_PORT 5555
#define INFTIM 1000
void setnonblocking(int sock)
{
int opts;
opts=fcntl(sock,F_GETFL);
if(opts<0)
{
perror("fcntl(sock,GETFL)");
exit(1);
}
opts = opts|O_NONBLOCK;
if(fcntl(sock,F_SETFL,opts)<0)
{
perror("fcntl(sock,SETFL,opts)");
exit(1);
}
}

int main()
{
int i, maxi, listenfd, connfd, sockfd,epfd,nfds;
ssize_t n;
char line[MAXLINE];
socklen_t clilen;
//聲明epoll_event結構體的變量,ev用於註冊事件,數組用於回傳要處理的事件
struct epoll_event ev,events[20];
//生成用於處理accept的epoll專用的文件描述符
epfd=epoll_create(256);
struct sockaddr_in clientaddr;
struct sockaddr_in serveraddr;
listenfd = socket(AF_INET, SOCK_STREAM, 0);
//把socket設置爲非阻塞方式
setnonblocking(listenfd);
//設置與要處理的事件相關的文件描述符
ev.data.fd=listenfd;
//設置要處理的事件類型
ev.events=EPOLLIN|EPOLLET;
//註冊epoll事件
epoll_ctl(epfd,EPOLL_CTL_ADD,listenfd,&ev);
bzero(&serveraddr, sizeof(serveraddr));
serveraddr.sin_family = AF_INET;
char *local_addr="200.200.200.204";
inet_aton(local_addr,&(serveraddr.sin_addr));//htons(SERV_PORT);
serveraddr.sin_port=htons(SERV_PORT);
bind(listenfd,(sockaddr *)&serveraddr, sizeof(serveraddr));
listen(listenfd, LISTENQ);
maxi = 0;
for ( ; ; ) {
//等待epoll事件的發生
nfds=epoll_wait(epfd,events,20,500);
//處理所發生的所有事件
for(i=0;i<nfds;++i)
{
if(events[i].data.fd==listenfd)
{
connfd = accept(listenfd,(sockaddr *)&clientaddr, &clilen);
if(connfd<0){
perror("connfd<0");
exit(1);
}
setnonblocking(connfd);
char *str = inet_ntoa(clientaddr.sin_addr);
std::cout<<"connect from "<_u115 ?tr<<std::endl;
//設置用於讀操作的文件描述符
ev.data.fd=connfd;
//設置用於注測的讀操作事件
ev.events=EPOLLIN|EPOLLET;
//註冊ev
epoll_ctl(epfd,EPOLL_CTL_ADD,connfd,&ev);
}
else if(events[i].events&EPOLLIN)
{
if ( (sockfd = events[i].data.fd) < 0) continue;
if ( (n = read(sockfd, line, MAXLINE)) < 0) {
if (errno == ECONNRESET) {
close(sockfd);
events[i].data.fd = -1;
} else
std::cout<<"readline error"<<std::endl;
} else if (n == 0) {
close(sockfd);
events[i].data.fd = -1;
}
//設置用於寫操作的文件描述符
ev.data.fd=sockfd;
//設置用於注測的寫操作事件
ev.events=EPOLLOUT|EPOLLET;
//修改sockfd上要處理的事件爲EPOLLOUT
epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev);
}
else if(events[i].events&EPOLLOUT)
{
sockfd = events[i].data.fd;
write(sockfd, line, n);
//設置用於讀操作的文件描述符
ev.data.fd=sockfd;
//設置用於注測的讀操作事件
ev.events=EPOLLIN|EPOLLET;
//修改sockfd上要處理的事件爲EPOLIN
epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev);
}
}
}
}
 
 
 
Epoll的具體實現
 
在ACE 中有一個 TP_Reactor 是基於select的..就是可以多線程輪循select
1. 線程1 用select 去輪循fd_set , 結束後.獲得可讀的所有句柄.然後將這些句柄從集合中刪掉. 然後去操作這些可讀句柄
2. 線程2 用select 去輪循fd_set (不包含剛纔已經可讀的句柄).
3. .....

因爲select 是每次循環都要重新加載fd_set.所以可以這樣實現多線程輪循.
但epoll的話是一次性的, 如果將每次可讀的fd 都調用EPOLL_CTL_DEL將其從events隊列中行色匆除.我覺得可能得不到預期的高效. 因爲使用epoll的環境畢竟是fd量比較大的情況
 
 
 
2008-3-27 12:12cofish
 
搜一下就能搜到了
我再貼一次吧

[code]
 
#include <iostream>
 
#include <sys/socket.h>
 
#include <sys/epoll.h>
 
#include <netinet/in.h>
 
#include <arpa/inet.h>
 
#include <fcntl.h>
 
#include <unistd.h>
 
#include <stdio.h>
 
#include <pthread.h>
 
 
 
#define MAXLINE 10
 
 
 
#define OPEN_MAX 100
 
 
 
#define LISTENQ 20
 
 
 
#define SERV_PORT 5555
 
 
 
#define INFTIM 1000
 
 
 
//線程池任務隊列結構體
 
 
 
struct task
 
{
 
int fd; //需要讀寫的文件描述符
 
struct task *next; //下一個任務
 
};
 
 
 
//用於讀寫兩個的兩個方面傳遞參數
 
 
 
struct user_data
 
{
 
int fd;
 
unsigned int n_size;
 
char line[MAXLINE];
 
};
 
 
 
//線程的任務函數
 
 
 
void * readtask(void *args);
 
void * writetask(void *args);
 
 
 
//聲明epoll_event結構體的變量,ev用於註冊事件,數組用於回傳要處理的事件
 
 
 
struct epoll_event ev, events[20];
 
int epfd;
 
pthread_mutex_t mutex;
 
pthread_cond_t cond1;
 
struct task *readhead = NULL, *readtail = NULL, *writehead = NULL;
 
 
 
void setnonblocking(int sock)
 
{
 
int opts;
 
opts = fcntl(sock, F_GETFL);
 
if (opts < 0)
 
{
 
perror("fcntl(sock,GETFL)");
 
exit(1);
 
}
 
opts = opts | O_NONBLOCK;
 
if (fcntl(sock, F_SETFL, opts) < 0)
 
{
 
perror("fcntl(sock,SETFL,opts)");
 
exit(1);
 
}
 
}
 
 
 
int main()
 
{
 
int i, maxi, listenfd, connfd, sockfd, nfds;
 
pthread_t tid1, tid2;
 
 
 
struct task *new_task = NULL;
 
struct user_data *rdata = NULL;
 
socklen_t clilen;
 
 
 
pthread_mutex_init(&mutex, NULL);
 
pthread_cond_init(&cond1, NULL);
 
//初始化用於讀線程池的線程
 
 
 
pthread_create(&tid1, NULL, readtask, NULL);
 
pthread_create(&tid2, NULL, readtask, NULL);
 
 
 
//生成用於處理accept的epoll專用的文件描述符 
 
epfd = epoll_create(256);
 
 
 
struct sockaddr_in clientaddr;
 
struct sockaddr_in serveraddr;
 
listenfd = socket(AF_INET, SOCK_STREAM, 0);
 
//把socket設置爲非阻塞方式
 
 
 
setnonblocking(listenfd);
 
//設置與要處理的事件相關的文件描述符
 
 
 
ev.data.fd = listenfd;
 
//設置要處理的事件類型
 
 
 
ev.events = EPOLLIN | EPOLLET;
 
//註冊epoll事件
 
 
 
epoll_ctl(epfd, EPOLL_CTL_ADD, listenfd, &ev);
 
 
 
bzero(&serveraddr, sizeof(serveraddr));
 
serveraddr.sin_family = AF_INET;
 
 
 
char *local_addr = "200.200.200.222";
 
inet_aton(local_addr, &(serveraddr.sin_addr));//htons(SERV_PORT);
 
serveraddr.sin_port = htons(SERV_PORT);
 
bind(listenfd, (sockaddr *) &serveraddr, sizeof(serveraddr));
 
listen(listenfd, LISTENQ);
 
 
 
maxi = 0;
 
for (;;)
 
{
 
//等待epoll事件的發生
 
 
 
nfds = epoll_wait(epfd, events, 20, 500);
 
//處理所發生的所有事件   
 
for (i = 0; i < nfds; ++i)
 
{
 
if (events[i].data.fd == listenfd)
 
{
 
connfd = accept(listenfd, (sockaddr *) &clientaddr, &clilen);
 
if (connfd < 0)
 
{
 
perror("connfd<0");
 
exit(1);
 
}
 
setnonblocking(connfd);
 
 
 
char *str = inet_ntoa(clientaddr.sin_addr);
 
std::cout << "connec_ from >>" << str << std::endl;
 
//設置用於讀操作的文件描述符
 
 
 
ev.data.fd = connfd;
 
//設置用於注測的讀操作事件
 
 
 
ev.events = EPOLLIN | EPOLLET;
 
//註冊ev
 
 
 
epoll_ctl(epfd, EPOLL_CTL_ADD, connfd, &ev);
 
}
 
else if (events[i].events & EPOLLIN)
 
{
 
printf("reading!\n");
 
if ((sockfd = events[i].data.fd) < 0)
 
continue;
 
new_task = new task();
 
new_task->fd = sockfd;
 
new_task->next = NULL;
 
//添加新的讀任務
 
 
 
pthread_mutex_lock(&mutex);
 
if (readhead == NULL)
 
{
 
readhead = new_task;
 
readtail = new_task;
 
}
 
else
 
{
 
readtail->next = new_task;
 
readtail = new_task;
 
}
 
//喚醒所有等待cond1條件的線程
 
 
 
pthread_cond_broadcast(&cond1);
 
pthread_mutex_unlock(&mutex);
 
}
 
else if (events[i].events & EPOLLOUT)
 
{
 
rdata = (struct user_data *) events[i].data.ptr;
 
sockfd = rdata->fd;
 
write(sockfd, rdata->line, rdata->n_size);
 
delete rdata;
 
//設置用於讀操作的文件描述符
 
 
 
ev.data.fd = sockfd;
 
//設置用於注測的讀操作事件
 
 
 
ev.events = EPOLLIN | EPOLLET;
 
//修改sockfd上要處理的事件爲EPOLIN
 
 
 
epoll_ctl(epfd, EPOLL_CTL_MOD, sockfd, &ev);
 
}
 
}
 
}
 
}
 
 
 
void * readtask(void *args)
 
{
 
 
 
int fd = -1;
 
unsigned int n;
 
//用於把讀出來的數據傳遞出去
 
 
 
struct user_data *data = NULL;
 
while (1)
 
{
 
pthread_mutex_lock(&mutex);
 
//等待到任務隊列不爲空
 
 
 
while (readhead == NULL)
 
pthread_cond_wait(&cond1, &mutex);
 
 
 
fd = readhead->fd;
 
//從任務隊列取出一個讀任務
 
 
 
struct task *tmp = readhead;
 
readhead = readhead->next;
 
delete tmp;
 
pthread_mutex_unlock(&mutex);
 
data = new user_data();
 
data->fd = fd;
 
if ((n = read(fd, data->line, MAXLINE)) < 0)
 
{
 
if (errno == ECONNRESET)
 
{
 
close(fd);
 
 
 
} else
 
std::cout << "readline error" << std::endl;
 
if (data != NULL)
 
delete data;
 
}
 
else if (n == 0)
 
{
 
close(fd);
 
printf("Client close connect!\n");
 
if (data != NULL)
 
delete data;
 
}
 
else
 
{
 
data->n_size = n;
 
//設置需要傳遞出去的數據
 
ev.data.ptr = data;
 
//設置用於注測的寫操作事件
 
ev.events = EPOLLOUT | EPOLLET;
 
//修改sockfd上要處理的事件爲EPOLLOUT
 
epoll_ctl(epfd, EPOLL_CTL_MOD, fd, &ev);
 
}
 
}
 
}
 
 
 
[/code]
 
 
 
[code]
 
/*-------------------------------------------------------------------------------------------------
 
 gcc -o epolld epoll.c -lpthread
 
 author: wyezl
 
 2006.4.28
 
 ---------------------------------------------------------------------------------------------------*/
 
 
 
#include <sys/socket.h>
 
#include <sys/epoll.h>
 
#include <netinet/in.h>
 
#include <arpa/inet.h>
 
#include <fcntl.h>
 
#include <unistd.h>
 
#include <stdio.h>
 
#include <pthread.h>
 
#include <errno.h>
 
#include <string.h>
 
 
 
#define PORT 8888
 
#define MAXFDS 5000
 
#define EVENTSIZE 100
 
 
 
#define BUFFER "HTTP/1.1 200 OK\r\nContent-Length: 5\r\nConnection: close\r\nContent-Type: text/html\r\n\r\nHello"
 
 
 
int epfd;
 
void *serv_epoll(void *p);
 
void setnonblocking(int fd)
 
{
 
int opts;
 
opts = fcntl(fd, F_GETFL);
 
if (opts < 0)
 
{
 
fprintf(stderr, "fcntl failed\n");
 
return;
 
}
 
opts = opts | O_NONBLOCK;
 
if (fcntl(fd, F_SETFL, opts) < 0)
 
{
 
fprintf(stderr, "fcntl failed\n");
 
return;
 
}
 
return;
 
}
 
 
 
int main(int argc, char *argv[])
 
{
 
int fd, cfd, opt = 1;
 
struct epoll_event ev;
 
struct sockaddr_in sin, cin;
 
socklen_t sin_len = sizeof(struct sockaddr_in);
 
pthread_t tid;
 
pthread_attr_t attr;
 
epfd = epoll_create(MAXFDS);
 
if ((fd = socket(AF_INET, SOCK_STREAM, 0)) <= 0)
 
{
 
fprintf(stderr, "socket failed\n");
 
return -1;
 
}
 
setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (const void*) &opt, sizeof(opt));
 
memset(&sin, 0, sizeof(struct sockaddr_in));
 
sin.sin_family = AF_INET;
 
sin.sin_port = htons((short) (PORT));
 
sin.sin_addr.s_addr = INADDR_ANY;
 
if (bind(fd, (struct sockaddr *) &sin, sizeof(sin)) != 0)
 
{
 
fprintf(stderr, "bind failed\n");
 
return -1;
 
}
 
if (listen(fd, 32) != 0)
 
{
 
fprintf(stderr, "listen failed\n");
 
return -1;
 
}
 
pthread_attr_init(&attr);
 
pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
 
if (pthread_create(&tid, &attr, serv_epoll, NULL) != 0)
 
{
 
fprintf(stderr, "pthread_create failed\n");
 
return -1;
 
}
 
while (1)
 
{
 
cfd = accept(fd, (struct sockaddr *) &cin, &sin_len);
 
if (cfd <= 0)
 
{
 
sleep(1);
 
continue;
 
}
 
setnonblocking(cfd);
 
ev.data.fd = cfd;
 
ev.events = EPOLLIN | EPOLLET;
 
if (epoll_ctl(epfd, EPOLL_CTL_ADD, cfd, &ev) < 0)
 
{
 
close(cfd);
 
}
 
//printf("connect from %s\n",inet_ntoa(cin.sin_addr));
 
//printf("cfd=%d\n",cfd);
 
}
 
if (fd > 0)
 
close(fd);
 
return 0;
 
}
 
void *serv_epoll(void *p)
 
{
 
int i, ret, cfd, nfds;
 
struct epoll_event ev, events[EVENTSIZE];
 
char buffer[512];
 
while (1) {
 
nfds = epoll_wait(epfd, events, EVENTSIZE, -1);
 
//printf("nfds ........... %d\n",nfds);
 
for (i = 0; i < nfds; i++)
 
{
 
if (events[i].events & EPOLLIN)
 
{
 
cfd = events[i].data.fd;
 
ret = recv(cfd, buffer, sizeof(buffer), 0);
 
//printf("read ret..........= %d\n",ret);
 
ev.data.fd = cfd;
 
ev.events = EPOLLOUT | EPOLLET;
 
epoll_ctl(epfd, EPOLL_CTL_MOD, cfd, &ev);
 
}
 
else if (events[i].events & EPOLLOUT)
 
{
 
cfd = events[i].data.fd;
 
ret = send(cfd, buffer, strlen(buffer), 0);
 
//printf("send ret...........= %d\n", ret);
 
ev.data.fd = cfd;
 
epoll_ctl(epfd, EPOLL_CTL_DEL, cfd, &ev);
 
//shutdown(cfd, 1);
 
if (cfd > 0)
 
close(cfd);
 
}
 
}
 
}
 
return NULL;
 
}
 
[/code]
 
 
 
 
 
 
 
Epoll的簡單操作
 
那麼究竟如何來使用epoll呢?其實非常簡單。
 
  
 
通過在包含一個頭文件#include <sys/epoll.h>以及幾個簡單的API將可以大大的提高你的網絡服務器的支持人數。
 
  首先通過epoll_create(int maxfds)來創建一個epoll的句柄,其中maxfds爲你epoll所支持的最大句柄數。這個函數會返回一個新的epoll句柄,之後的所有操作將通過這個句柄來進行操作。在用完之後,記得用close()來關閉這個創建出來的epoll句柄。
 
  之後在你的網絡主循環裏面,每一幀的調用epoll_wait(int epfd, epoll_event events, int max events, int timeout)來查詢所有的網絡接口,看哪一個可以讀,哪一個可以寫了。基本的語法爲:
 
  nfds = epoll_wait(kdpfd, events, maxevents, -1);
 
  其中kdpfd爲用epoll_create創建之後的句柄,events是一個epoll_event*的指針,當epoll_wait這個函數操作成功之後,epoll_events裏面將儲存所有的讀寫事件。max_events是當前需要監聽的所有socket句柄數。最後一個timeout是epoll_wait的超時,爲0的時候表示馬上返回,爲-1的時候表示一直等下去,直到有事件範圍,爲任意正整數的時候表示等這麼長的時間,如果一直沒有事件,則範圍。一般如果網絡主循環是單獨的線程的話,可以用-1來等,這樣可以保證一些效率,如果是和主邏輯在同一個線程的話,則可以用0來保證主循環的效率。
 
  epoll_wait範圍之後應該是一個循環,遍利所有的事件:
 
  for(n = 0; n < nfds; ++n) {
 
  if(events[n].data.fd == listener) { //如果是主socket的事件的話,則表示有新連接進入了,進行新連接的處理。
 
  client = accept(listener, (struct sockaddr *) &local,
 
  &addrlen);
 
  if(client < 0){
 
  perror("accept");
 
  continue;
 
  }
 
  setnonblocking(client); // 將新連接置於非阻塞模式
 
  ev.events = EPOLLIN | EPOLLET; // 並且將新連接也加入EPOLL的監聽隊列。
 
  注意,這裏的參數EPOLLIN | EPOLLET並沒有設置對寫socket的監聽,如果有寫操作的話,這個時候epoll是不會返回事件的,如果要對寫操作也監聽的話,應該是EPOLLIN | EPOLLOUT | EPOLLET
 
  ev.data.fd = client;
 
  if (epoll_ctl(kdpfd, EPOLL_CTL_ADD, client, &ev) < 0) {
 
  // 設置好event之後,將這個新的event通過epoll_ctl加入到epoll的監聽隊列裏面,這裏用EPOLL_CTL_ADD來加一個新的epoll事件,通過EPOLL_CTL_DEL來減少一個epoll事件,通過EPOLL_CTL_MOD來改變一個事件的監聽方式。
 
  fprintf(stderr, "epoll set insertion error: fd=%d0,
 
  client);
 
  return -1;
 
  }
 
  }
 
  else // 如果不是主socket的事件的話,則代表是一個用戶socket的事件,則來處理這個用戶socket的事情,比如說read(fd,xxx)之類的,或者一些其他的處理。
 
  do_use_fd(events[n].data.fd);
 
  }
 
  對,epoll的操作就這麼簡單,總共不過4個API:epoll_create, epoll_ctl, epoll_wait和close。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章