udp & tcp 对epoll的共用

最新推荐文章于 2022-10-05 10:27:03 发布

yarsen

最新推荐文章于 2022-10-05 10:27:03 发布

阅读量1k

点赞数

分类专栏：网络

网络专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一个使用epoll同时监听UDP和TCP连接的示例。通过设置非阻塞模式和利用epoll_ctl添加监听事件，实现了服务器端对两种不同类型的网络连接进行高效管理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

http://blog.youkuaiyun.com/lipengfei634626165/article/details/8136715

epoll 可以同时监听UDP 和 TCP的链接。以下为用EPOLL进行的UDP和TCP的共同监听。以下为服务器端server.c

[cpp]view plaincopy 
   
 #include <stdio.h>  
 #include <stdlib.h>  
 #include <errno.h>  
 #include <string.h>  
 #include <sys/types.h>  
 #include <netinet/in.h>  
 #include <sys/socket.h>  
 #include <sys/wait.h>  
 #include <unistd.h>  
 #include <arpa/inet.h>  
 #include <fcntl.h>  
 #include <sys/epoll.h>  
 #include <sys/time.h>  
 #include <sys/resource.h>  
   
   
 #define MAXLINE 100  
 #define OPEN_MAX 100  
 #define LISTENQ 20  
 #define SERV_TPORT 5000  
 #define SERV_UPORT 5001  
   
 #define INFTIM 1000  
   
   
 int setnonblocking(int sockfd)  
 {  
     if (fcntl(sockfd, F_SETFL, fcntl(sockfd, F_GETFD, 0)|O_NONBLOCK) == -1)  
     {  
         return -1;  
     }  
     return 0;  
 }  
   
 int main(int argc, char* argv[])  
 {  
     int i, tfd_listener, connfd, ufd_listener, sockfd,epfd,nfds, portnumber;  
     ssize_t n;  
     char line[MAXLINE] = "abcdefg";  
     socklen_t clilen;  
   
     int count = 0;  
   
     struct epoll_event ev,events[20];  
     struct sockaddr_in clientaddr;  
     struct sockaddr_in serveraddr;  
     struct sockaddr_in my_addr;  
   
     tfd_listener = socket(AF_INET, SOCK_STREAM, 0);  
     if(tfd_listener < 0)  
         perror("tfd_listener create error\n");  
   
     ufd_listener = socket(PF_INET,  SOCK_DGRAM, 0);  
     if(ufd_listener < 0)  
         perror("ufd_listener create error\n");  
       
     int opt=SO_REUSEADDR;  
     setsockopt(ufd_listener,SOL_SOCKET,SO_REUSEADDR,&opt,sizeof(opt));  
   
     setnonblocking(ufd_listener);  
     setnonblocking(tfd_listener);  
   
     bzero(&my_addr, sizeof(my_addr));  
     my_addr.sin_family = PF_INET;  
     my_addr.sin_port = htons(4000);  
     my_addr.sin_addr.s_addr = INADDR_ANY;  
   
   
     bzero(&serveraddr, sizeof(serveraddr));  
     serveraddr.sin_family = AF_INET;  
     serveraddr.sin_addr.s_addr = INADDR_ANY;  
     serveraddr.sin_port=htons(1235);  
   
     if (bind(ufd_listener, (struct sockaddr *) &my_addr, sizeof(struct sockaddr)) == -1)  
         perror("ufd_listener bind error\n");  
   
     if(bind(tfd_listener,(struct sockaddr *)&serveraddr, sizeof(struct sockaddr))== -1)  
         perror("tfd_listener bind error\n");  
   
     epfd=epoll_create(256);  
   
     ev.data.fd=tfd_listener;  
     ev.events=EPOLLIN|EPOLLET;    
     epoll_ctl(epfd,EPOLL_CTL_ADD,tfd_listener,&ev);  
   
     ev.data.fd = ufd_listener;  
     ev.events = EPOLLIN|EPOLLET;  
     epoll_ctl(epfd,EPOLL_CTL_ADD,ufd_listener,&ev);  
   
     listen(tfd_listener, LISTENQ);  
   
     for ( ; ; ) {  
   
         nfds=epoll_wait(epfd,events,20,500);  
   
         printf("wait end nfds = %d\n",nfds);  
   
         for(i=0;i<nfds;++i)  
         {  
             if(events[i].data.fd == tfd_listener)  
             {  
                 connfd = accept(tfd_listener,(struct sockaddr *)&clientaddr, &clilen);  
                 if(connfd<0){  
                     perror("connfd<0");  
                     exit(1);  
                 }  
                 setnonblocking(connfd);  
                 char *str = inet_ntoa(clientaddr.sin_addr);  
                     printf("accept a connection from %s\n", str);  
   
                     ev.data.fd=connfd;  
                     ev.events=EPOLLIN|EPOLLET;  
                     epoll_ctl(epfd,EPOLL_CTL_ADD,connfd,&ev);  
   
             }  
             else if(events[i].data.fd == ufd_listener)  
             {  
                 printf("udp\n");  
                 if( events[i].events && EPOLLIN )  
                     printf( "udp receive begin!\n" );  
                 n = recvfrom(ufd_listener, line,MAXLINE, 0,  
                                              (struct sockaddr *)&clientaddr, &clilen);                
                 if(n < 0)  
                     perror( "udp receive ERROR\n" );  
                 else  
                     printf( "udp receive right!\n" );  
           
             }  
             else if(events[i].events&EPOLLIN)//如果是已经连接的用户，并且收到数据，那么进行读入。  
                 {  
                 printf("count == %d\n",count);  
                 count ++;  
                     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  
                                 printf("readline error\n");  
                     }   
                 else if (n == 0) {  
                     printf("close tcp port\n");  
                     close(sockfd);  
                         events[i].data.fd = -1;  
                 }  
                 else  
                 {  
                     printf("read OK %s\n", line);  
                         line[n] = '/0';  
                     }             
                     ev.data.fd=sockfd;  
                     ev.events=EPOLLOUT|EPOLLET;  
                     epoll_ctl(epfd,EPOLL_CTL_ADD,connfd,&ev);                 
         }else if(events[i].events&EPOLLOUT) // 如果有数据发送  
   
             {  
                 printf("write \n");  
                 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);  
             }  
         }  
     }  
     return 0;  
 }  

下面是udp客户端 udp_client.c的实现

[cpp]view plaincopy 
   
 #include <sys/types.h>  
 #include <sys/socket.h>  
 #include <string.h>  
 #include <netinet/in.h>  
 #include <stdio.h>  
 #include <stdlib.h>  
 #include <arpa/inet.h>  
 #include <unistd.h>  
   
 #define MAXLINE 1024  
 #define SERV_PORT 4000  
   
 void do_cli(FILE *fp, int sockfd, struct sockaddr *pservaddr, socklen_t servlen)  
 {  
     int n;  
     char sendline[MAXLINE], recvline[MAXLINE + 1];  
   
   
     if(connect(sockfd, (struct sockaddr *)pservaddr, servlen) == -1)  
     {  
         perror("connect error");  
         exit(1);  
     }  
     else  
     {  
         printf("connect OK\n");  
     }  
   
     fgets(sendline, MAXLINE, fp);  
     n = write(sockfd, sendline, strlen(sendline)-1);  
     if ( n <= 0)  
         perror("write error\n");  
     printf("write %s \n",sendline);  
           
 }  
   
 int main(int argc, char **argv)  
 {  
     int sockfd;  
     struct sockaddr_in servaddr;  
   
     bzero(&servaddr, sizeof(servaddr));  
     servaddr.sin_family = AF_INET;  
     servaddr.sin_port = htons(SERV_PORT);  
     inet_aton("127.0.0.1",&(servaddr.sin_addr));  
   
     sockfd = socket(AF_INET, SOCK_DGRAM, 0);  
   
     do_cli(stdin, sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr));  
   
     return 0;  
 }  

下面是TCP 的客户端tcp_client.c的实现

[cpp]view plaincopy 
   
 #include <stdio.h>  
 #include <sys/socket.h>  
 #include <sys/epoll.h>  
 #include <netinet/in.h>  
 #include <arpa/inet.h>  
 #include <fcntl.h>  
 #include <unistd.h>  
 #include <errno.h>  
 #include <stdlib.h>  
 #include <string.h>  
 #include <errno.h>  
   
 int main(){  
     int fd;  
     struct sockaddr_in clientaddr;  
     char  recvline[1000] = "abcdef";  
     int n = 0;  
     if((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)  
         perror("socket create error");  
       
     bzero(&clientaddr, sizeof(clientaddr));  
     clientaddr.sin_family = AF_INET;  
     inet_aton("127.0.0.1",&(clientaddr.sin_addr));  
     clientaddr.sin_port = htons(1235);  
   
     if(connect(fd, (struct sockaddr *)&clientaddr,sizeof(clientaddr)) < 0)  
         perror("connect error");  
     printf("%d\n",fd);  
     while((n = write(fd, recvline,6))>0){  
         printf("recvline = %s",recvline);  
         if(n > 0)  
             break;  
     }  
     if (n < 0)  
         perror("write error");  
   
     exit(0);  
 }  