UNIX TCP回射服务器/客户端之使用epoll模型的服务器

程序简介：这是一个运用epoll系列函数进行IO复用的服务器模型。它是目前UNIX与LINUX平台上效率最高，最受欢迎的IO复用传输模型。

其他的不说了，直接粘贴代码吧！

服务器端：

#include "my_unp.h"  
  
int main(void)    
{    
    int listenfd, connfd, sockfd, epfd;  
    int i, maxi, nfds;  
    ssize_t n;    
    char buf[MAXLINE];    
    socklen_t clilen;    
    struct sockaddr_in cliaddr;    
    struct sockaddr_in servaddr;  
  
    //声明epoll_event结构体的变量,ev用于注册事件,数组用于回传要处理的事件    
    struct epoll_event ev, events[256];    
  
    //创建一个epoll的句柄，size用来告诉内核这个监听的数目一共有多大    
    epfd = Epoll_create(256);  
  
    //创建用于TCP协议的套接字      
    listenfd = Socket(AF_INET, SOCK_STREAM, 0);    
    memset(&servaddr, 0, sizeof(servaddr));    
    servaddr.sin_family = AF_INET;    
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);    
    servaddr.sin_port = htons(SERV_PORT);    
  
    //把socket和socket地址结构联系起来    
    Bind(listenfd, (SA*)&servaddr, sizeof(servaddr));  
    //开始监听LISTENQ端口    
    Listen(listenfd, LISTENQ);  
  
    //设置与要处理的事件相关的文件描述符和事件  
    ev.data.fd = listenfd;    
    ev.events = EPOLLIN|EPOLLET;    
  
    //注册epoll事件    
    Epoll_ctl(epfd, EPOLL_CTL_ADD, listenfd,&ev);    
    maxi = 0;  
  
    while(1)  
    {    
        //等待epoll事件的发生    
        //返回需要处理的事件数目nfds，如返回0表示已超时。    
        nfds = Epoll_wait(epfd, events, 20, 500);    
  
        //处理所发生的所有事件    
        for(i=0; i < nfds; ++i)    
        {    
            //如果新监测到一个SOCKET用户连接到了绑定的SOCKET端口，建立新的连接。    
            if(events[i].data.fd == listenfd)    
            {    
                connfd = Accept(listenfd,(SA*)&cliaddr, &clilen);    
                printf("connection from %s, port %d.\n",        
                    Inet_ntop(AF_INET, (void*)&cliaddr.sin_addr, buf, sizeof(buf)),        
                    ntohs(cliaddr.sin_port));    
  
                //设置用于读操作的文件描述符和事件  
                ev.data.fd = connfd;    
                ev.events = EPOLLIN|EPOLLET;    
  
                //注册事件  
                Epoll_ctl(epfd,EPOLL_CTL_ADD,connfd,&ev);  
            }    
  
            //如果是已经连接的用户，并且收到数据，那么进行读入。    
            else if(events[i].events & EPOLLIN)    
            {    
                sockfd = events[i].data.fd;  
                if ( sockfd < 0 )    
                    continue;    
                n = read(sockfd, buf, MAXLINE);  
                if ( n < 0)    
                {      
                    // Connection Reset:你连接的那一端已经断开了，  
                    //而你却还试着在对方已断开的socketfd上读写数据！    
                    if (errno == ECONNRESET)    
                    {    
                        Close(sockfd);    
                        events[i].data.fd = -1;    
                    }    
                    else    
                        error_quit("read error");  
                }    
                //如果读入的数据为空    
                else if ( n == 0 )  
                {    
                    Close(sockfd);    
                    events[i].data.fd = -1;    
                }    
                else  
                {  
                    //设置用于写操作的文件描述符和事件  
                    ev.data.fd = sockfd;  
                    ev.events = EPOLLOUT|EPOLLET;  
                    //注册事件  
                    Epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev);  
                }  
            }  
  
            //如果有数据发送  
            else if(events[i].events & EPOLLOUT)  
            {  
                sockfd = events[i].data.fd;  
                Writen(sockfd, buf, n);    
  
                //设置用于读操作的文件描述符和事件  
                ev.data.fd = sockfd;  
                ev.events = EPOLLIN|EPOLLET;  
  
                //注册事件  
                Epoll_ctl(epfd, EPOLL_CTL_MOD, sockfd, &ev);  
            }  
        }  
    }  
    return 0;    
} 

客户端的测试例子：

#include "my_unp.h" 

void str_cli(FILE *fp, int sockfd)  
{  
 char sendline[MAXLINE], recvline[MAXLINE];  
 int len;

 //从终端获取一行字符串，将其写入套接字  
 //然后从套接字一行字符串，将其写入终端  
 while( Fgets(sendline, MAXLINE, fp) != NULL )  
 {  
  Writen(sockfd, sendline, strlen(sendline));  
  len = Read(sockfd, recvline, MAXLINE);
  if( len == 0 )  
   error_quit("str_cli: server terminated prematurely");  
  recvline[len] = 0;
  Fputs(recvline, stdout);  
 }  
} 

int main(int argc, char **argv)  
{  
 int sockfd;  
 struct sockaddr_in servaddr;  
 if( argc != 2 )  
  error_quit("usage: client <IPAddress>"); 

 //创建用于TCP协议的套接字  
 sockfd = Socket(AF_INET, SOCK_STREAM, 0);  
 memset(&servaddr, 0, sizeof(servaddr));  
 servaddr.sin_family = AF_INET;  
 servaddr.sin_port = htons(SERV_PORT); 

 //将程序的参数1(argv[1])转换成套接字地址结构  
 Inet_pton(AF_INET, argv[1], &servaddr.sin_addr);  
 向服务器发起连接,连接成功后client_socket代表了客户机和服务器的一个socket连接  
 Connect(sockfd, (SA*)&servaddr, sizeof(servaddr)); 

 str_cli(stdin, sockfd);  
 return 0;  
}

运行示例（红色字体的为输入）（假设某个回射字符串的服务器已经启动）

www.linuxidc.com@ubuntu:~/code$ gcc my_unp_v1.c client.c -o client
www.linuxidc.com@ubuntu:~/code$ ./client 127.0.0.1
ABCD
ABCD
Ctrl+D