• 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 ", 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 ", 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... ", errno);
    67.         exit(EXIT_FAILURE);
    68.     }
    69.     //!> 设置非阻塞模式
    70.     //!>
    71.     if (setnonblocking(listen_fd) == -1) {
    72.         printf("Setnonblocking Error : %d ", 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 ", errno);
    79.         exit(EXIT_FAILURE);
    80.     }
    81.     //!> 监听
    82.     //!>
    83.     if (listen(listen_fd, MAXBACK) == -1) {
    84.         printf("Listen Error : %d ", 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 ", 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 ", 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 ", errno);
    110.                     exit(EXIT_FAILURE);
    111.                 }

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

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

    68.             memset(send, 0, sizeof(send));
    69.             if (fgets(send, MAXLINE, fp) == NULL) {
    70.                 printf("End... ");
    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.. ");
    79.                     return;
    80.                 }

    82.                 printf("Client send : %s ", 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 ! ");
    100.         exit(EXIT_FAILURE);
    101.     }
    102.     //!> 建立套接字
    103.     if ((connfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
    104.         printf("Socket Error... ", 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.. ");
    116.         exit(EXIT_FAILURE);
    117.     }
    118.     /*else
    119.      {
    120.      printf("Connet ok.. ");
    121.      } */

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

    127.     //!>

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

    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

    相关热门文章
    给主人留下些什么吧！~~
    评论热议
  • 相关阅读:
    1 TKinter小窗口及标题
    css之padding,marging
    css之opacity
    css之position
    Leetcode 1368 使网格图至少有一条有效路径的最小代价
    Leetcode 137 只出现一次的数字II
    Leetcode 135分发糖果
    Leetcode 134加油站
    Leetcode 124 二叉树的最大路径和
    Leetcode 5346 二叉树中的列表
  • 原文地址：https://www.cnblogs.com/ztguang/p/12649512.html
Copyright © 2020-2023  润新知