程序简介:这是一个运用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