epoll—linux c language —TCP unicast

http://blog.csdn.net/shanshanpt/article/details/7383400

server.c

点击(此处)折叠或打开

  1. // gcc server.c -o server
  2. // indent -npro -kr -i8 -ts8 -sob -l280 -ss -ncs -cp1 *

  4. #include <stdio.h>
  5. #include <unistd.h>
  6. #include <stdlib.h>
  7. #include <string.h>
  8. #include <sys/types.h>
  9. #include <errno.h>
  10. #include <sys/socket.h>
  11. #include <netinet/in.h>        /* socket类定义需要 */
  12. #include <sys/epoll.h>        /* epoll头文件 */
  13. #include <fcntl.h>        /* nonblocking需要 */
  14. #include <sys/resource.h>    /* 设置最大的连接数需要setrlimit */

  16. #define MAXEPOLL 10000    /* 对于服务器来说，这个值可以很大的！ */
  17. #define MAXLINE 1024
  18. #define PORT 6000
  19. #define MAXBACK 1000

  21. //!> 设置非阻塞
  22. //!>
  23. int setnonblocking(int fd)
  24. {
  25.     if (fcntl(fd, F_SETFL, fcntl(fd, F_GETFD, 0) | O_NONBLOCK) == -1) {
  26.         printf("Set blocking error : %d\n", errno);
  27.         return -1;
  28.     }
  29.     return 0;
  30. }

  32. int main(int argc, char **argv)
  33. {
  34.     int listen_fd;
  35.     int conn_fd;
  36.     int epoll_fd;
  37.     int nread;
  38.     int cur_fds;        //!> 当前已经存在的数量
  39.     int wait_fds;        //!> epoll_wait 的返回值
  40.     int i;
  41.     struct sockaddr_in servaddr;
  42.     struct sockaddr_in cliaddr;
  43.     struct epoll_event ev;
  44.     struct epoll_event evs[MAXEPOLL];
  45.     struct rlimit rlt;    //!> 设置连接数所需
  46.     char buf[MAXLINE];
  47.     socklen_t len = sizeof(struct sockaddr_in);

  49.     //!> 设置每个进程允许打开的最大文件数
  50.     //!> 每个主机是不一样的哦，一般服务器应该很大吧！
  51.     //!>
  52.     rlt.rlim_max = rlt.rlim_cur = MAXEPOLL;
  53.     if (setrlimit(RLIMIT_NOFILE, &rlt) == -1) {
  54.         printf("Setrlimit Error : %d\n", errno);
  55.         exit(EXIT_FAILURE);
  56.     }
  57.     //!> server 套接口
  58.     //!>
  59.     bzero(&servaddr, sizeof(servaddr));
  60.     servaddr.sin_family = AF_INET;
  61.     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
  62.     servaddr.sin_port = htons(PORT);

  64.     //!> 建立套接字
  65.     if ((listen_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
  66.         printf("Socket Error...\n", errno);
  67.         exit(EXIT_FAILURE);
  68.     }
  69.     //!> 设置非阻塞模式
  70.     //!>
  71.     if (setnonblocking(listen_fd) == -1) {
  72.         printf("Setnonblocking Error : %d\n", errno);
  73.         exit(EXIT_FAILURE);
  74.     }
  75.     //!> 绑定
  76.     //!>
  77.     if (bind(listen_fd, (struct sockaddr *)&servaddr, sizeof(struct sockaddr)) == -1) {
  78.         printf("Bind Error : %d\n", errno);
  79.         exit(EXIT_FAILURE);
  80.     }
  81.     //!> 监听
  82.     //!>
  83.     if (listen(listen_fd, MAXBACK) == -1) {
  84.         printf("Listen Error : %d\n", errno);
  85.         exit(EXIT_FAILURE);
  86.     }
  87.     //!> 创建epoll
  88.     //!>
  89.     epoll_fd = epoll_create(MAXEPOLL);    //!> create
  90.     ev.events = EPOLLIN | EPOLLET;    //!> accept
  91.     ev.data.fd = listen_fd;    //!> 将listen_fd 加入
  92.     if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, listen_fd, &ev) < 0) {
  93.         printf("Epoll Error : %d\n", errno);
  94.         exit(EXIT_FAILURE);
  95.     }
  96.     cur_fds = 1;

  98.     while (1) {
  99.         if ((wait_fds = epoll_wait(epoll_fd, evs, cur_fds, -1)) == -1) {
  100.             printf("Epoll Wait Error : %d\n", errno);
  101.             exit(EXIT_FAILURE);
  102.         }

  104.         for (i = 0; i < wait_fds; i++) {
  105.             if (evs[i].data.fd == listen_fd && cur_fds < MAXEPOLL)
  106.                 //!> if是监听端口有事
  107.             {
  108.                 if ((conn_fd = accept(listen_fd, (struct sockaddr *)&cliaddr, &len)) == -1) {
  109.                     printf("Accept Error : %d\n", errno);
  110.                     exit(EXIT_FAILURE);
  111.                 }

  113.                 printf("Server get from client !\n" /*, inet_ntoa(cliaddr.sin_addr), cliaddr.sin_port */ );

  115.                 ev.events = EPOLLIN | EPOLLET;    //!> accept
  116.                 ev.data.fd = conn_fd;    //!> 将conn_fd 加入
  117.                 if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, conn_fd, &ev) < 0) {
  118.                     printf("Epoll Error : %d\n", errno);
  119.                     exit(EXIT_FAILURE);
  120.                 }
  121.                 ++cur_fds;
  122.                 continue;
  123.             }
  124.             //!> 下面处理数据
  125.             //!>
  126.             nread = read(evs[i].data.fd, buf, sizeof(buf));
  127.             if (nread <= 0)    //!> 结束后者出错
  128.             {
  129.                 close(evs[i].data.fd);
  130.                 epoll_ctl(epoll_fd, EPOLL_CTL_DEL, evs[i].data.fd, &ev);    //!> 删除计入的fd
  131.                 --cur_fds;    //!> 减少一个呗！
  132.                 continue;
  133.             }

  135.             write(evs[i].data.fd, buf, nread);    //!> 回写

  137.         }
  138.     }

  140.     close(listen_fd);
  141.     return 0;
  142. }

client.c

点击(此处)折叠或打开

  1. // gcc client.c -o client
  2. // indent -npro -kr -i8 -ts8 -sob -l280 -ss -ncs -cp1 *

  4. #include <stdio.h>
  5. #include <unistd.h>
  6. #include <stdlib.h>
  7. #include <string.h>
  8. #include <errno.h>
  9. #include <netinet/in.h>
  10. #include <sys/types.h>
  11. #include <sys/socket.h>
  12. #include <arpa/inet.h>
  13. #include <sys/select.h>

  15. #define MAXLINE 1024
  16. #define SERV_PORT 6000

  18. //!> 注意输入是由stdin，接受是由server发送过来
  19. //!> 所以在client端也是需要select进行处理的
  20. void send_and_recv(int connfd)
  21. {
  22.     FILE *fp = stdin;
  23.     int lens;
  24.     char send[MAXLINE];
  25.     char recv[MAXLINE];
  26.     fd_set rset;
  27.     FD_ZERO(&rset);
  28.     int maxfd = (fileno(fp) > connfd ? fileno(fp) : connfd + 1);
  29.     //!> 输入和输出的最大值
  30.     int n;

  32.     while (1) {
  33.         FD_SET(fileno(fp), &rset);
  34.         FD_SET(connfd, &rset);    //!> 注意不要把rset看作是简单的一个变量
  35.         //!> 注意它其实是可以包含一组套接字的哦，
  36.         //!> 相当于是封装的数组！每次都要是新的哦！

  38.         if (select(maxfd, &rset, NULL, NULL, NULL) == -1) {
  39.             printf("Client Select Error..\n");
  40.             exit(EXIT_FAILURE);
  41.         }
  42.         //!> if 连接口有信息
  43.         if (FD_ISSET(connfd, &rset))    //!> if 连接端口有信息
  44.         {
  45.             printf("client get from server ...\n");
  46.             memset(recv, 0, sizeof(recv));
  47.             n = read(connfd, recv, MAXLINE);
  48.             if (n == 0) {
  49.                 printf("Recv ok...\n");
  50.                 break;
  51.             } else if (n == -1) {
  52.                 printf("Recv error...\n");
  53.                 break;
  54.             } else {
  55.                 lens = strlen(recv);
  56.                 recv[lens] = '\0';
  57.                 //!> 写到stdout
  58.                 write(STDOUT_FILENO, recv, MAXLINE);
  59.                 printf("\n");
  60.             }

  62.         }
  63.         //!> if 有stdin输入
  64.         if (FD_ISSET(fileno(fp), &rset))    //!> if 有输入
  65.         {
  66.             //!> printf("client stdin ...\n");

  68.             memset(send, 0, sizeof(send));
  69.             if (fgets(send, MAXLINE, fp) == NULL) {
  70.                 printf("End...\n");
  71.                 exit(EXIT_FAILURE);
  72.             } else {
  73.                 //!>if( str )
  74.                 lens = strlen(send);
  75.                 send[lens - 1] = '\0';    //!> 减一的原因是不要回车字符
  76.                 //!> 经验值：这一步非常重要的哦！！！！！！！！
  77.                 if (strcmp(send, "q") == 0) {
  78.                     printf("Bye..\n");
  79.                     return;
  80.                 }

  82.                 printf("Client send : %s\n", send);
  83.                 write(connfd, send, strlen(send));
  84.             }
  85.         }

  87.     }

  89. }

  91. int main(int argc, char **argv)
  92. {
  93.     //!> char * SERV_IP = "10.30.97.188";
  94.     char buf[MAXLINE];
  95.     int connfd;
  96.     struct sockaddr_in servaddr;

  98.     if (argc != 2) {
  99.         printf("Input server ip !\n");
  100.         exit(EXIT_FAILURE);
  101.     }
  102.     //!> 建立套接字
  103.     if ((connfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
  104.         printf("Socket Error...\n", errno);
  105.         exit(EXIT_FAILURE);
  106.     }
  107.     //!> 套接字信息
  108.     bzero(&servaddr, sizeof(servaddr));
  109.     servaddr.sin_family = AF_INET;
  110.     servaddr.sin_port = htons(SERV_PORT);
  111.     inet_pton(AF_INET, argv[1], &servaddr.sin_addr);

  113.     //!> 链接server
  114.     if (connect(connfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) < 0) {
  115.         printf("Connect error..\n");
  116.         exit(EXIT_FAILURE);
  117.     }
  118.     /*else
  119.      {
  120.      printf("Connet ok..\n");
  121.      } */

  123.     //!>
  124.     //!> send and recv
  125.     send_and_recv(connfd);

  127.     //!>

  129.     close(connfd);
  130.     printf("Exit\n");

  132.     return 0;
  133. }

<script>window._bd_share_config={"common":{"bdSnsKey":{},"bdText":"","bdMini":"2","bdMiniList":false,"bdPic":"","bdStyle":"0","bdSize":"16"},"share":{}};with(document)0[(getElementsByTagName('head')[0]||body).appendChild(createElement('script')).src='http://bdimg.share.baidu.com/static/api/js/share.js?v=89860593.js?cdnversion='+~(-new Date()/36e5)];</script>
阅读(98) | 评论(0) | 转发(0) |
0

上一篇：epoll—linux c language —UDP Broadcast

下一篇：select—linux c language —TCP broadcast—chat

相关热门文章
给主人留下些什么吧！~~
评论热议
posted @ 2016-02-01 00:00  张同光  阅读(93)  评论(0)  编辑  收藏  举报