libevent源码分析-TCP服务端代码

  1 #include <stdio.h>    
  2 #include <string.h>    
  3 #include <iostream>    
  4 #include <sys/socket.h>        
  5 #include <netinet/in.h>        
  6 #include <arpa/inet.h>        
  7 #include <netdb.h>    
  8     
  9 #include <event.h>    
 10 using namespace std;    
 11     
 12 // 事件base    
 13 struct event_base* base;    
 14     
 15 // 读事件回调函数    
 16 void onRead(int iCliFd, short iEvent, void *arg)    
 17 {    
 18     int iLen;    
 19     char buf[1500];    
 20     
 21     iLen = recv(iCliFd, buf, 1500, 0);    
 22     
 23     if (iLen <= 0) {    
 24         cout << "Client Close" << endl;    
 25     
 26         // 连接结束(=0)或连接错误(<0)，将事件删除并释放内存空间    
 27         struct event *pEvRead = (struct event*)arg;    
 28         event_del(pEvRead);    
 29         delete pEvRead;    
 30     
 31         close(iCliFd);    
 32         return;    
 33     }    
 34     
 35     buf[iLen] = 0;    
 36     cout << "Client Info:" << buf << endl;    
 37   
 38   
 39   
 40     struct bufferevent* buf_ev;  
 41     buf_ev = bufferevent_new(iCliFd, NULL, NULL, NULL, NULL);  
 42     buf_ev->wm_read.high = 4096;  
 43     char MESSAGE[]="welcome to server..";  
 44     bufferevent_write(buf_ev, MESSAGE, strlen(MESSAGE));  
 45 }    
 46     
 47 // 连接请求事件回调函数    
 48 void onAccept(int iSvrFd, short iEvent, void *arg)    
 49 {    
 50     int iCliFd;    
 51     struct sockaddr_in sCliAddr;    
 52     
 53     socklen_t iSinSize = sizeof(sCliAddr);    
 54     iCliFd = accept(iSvrFd, (struct sockaddr*)&sCliAddr, &iSinSize);    
 55     
 56     // 连接注册为新事件 (EV_PERSIST为事件触发后不默认删除)    
 57     struct event *pEvRead = new event;    
 58     event_set(pEvRead, iCliFd, EV_READ|EV_PERSIST, onRead, pEvRead);    
 59     event_base_set(base, pEvRead);    
 60     event_add(pEvRead, NULL);   
 61   
 62   
 63     struct bufferevent* buf_ev;    
 64      buf_ev = bufferevent_new(iCliFd, NULL, NULL, NULL, NULL);      
 65      buf_ev->wm_read.high = 4096;  
 66      char MESSAGE[]="welcome to server..";  
 67     bufferevent_write(buf_ev, MESSAGE, strlen(MESSAGE));   
 68   
 69     cout<<"a client connect:"<<iCliFd<<endl;  
 70 }    
 71     
 72 int main()    
 73 {    
 74     
 75     int iSvrFd;      
 76     struct sockaddr_in sSvrAddr;    
 77                   
 78     memset(&sSvrAddr, 0, sizeof(sSvrAddr));      
 79     sSvrAddr.sin_family = AF_INET;      
 80     sSvrAddr.sin_addr.s_addr = inet_addr("127.0.0.1");        
 81     sSvrAddr.sin_port = htons(8888);       
 82                                 
 83     // 创建tcpSocket（iSvrFd），监听本机8888端口      
 84     iSvrFd = socket(AF_INET, SOCK_STREAM, 0);      
 85     bind(iSvrFd, (struct sockaddr*)&sSvrAddr, sizeof(sSvrAddr));      
 86     listen(iSvrFd, 10);    
 87     
 88     // 初始化base    
 89     base = (struct event_base*)event_init();    
 90         
 91     struct event evListen;    
 92     // 设置事件    
 93     event_set(&evListen, iSvrFd, EV_READ|EV_PERSIST, onAccept, NULL);    
 94     // 设置为base事件    
 95     event_base_set(base, &evListen);    
 96     // 添加事件    
 97     event_add(&evListen, NULL);    
 98         
 99     // 事件循环    
100     event_base_dispatch(base);    
101     
102     return 0;    
103 }