zoukankan      html  css  js  c++  java

  • (OK) C/S—心跳检测—heartbeat

    /root/桌面/server-client-pthread-c

    heartbeat-server.c

    点击(此处)折叠或打开

    1. // gcc heartbeat-server.c -o heartbeat-server
    2. // indent -npro -kr -i8 -ts8 -sob -l280 -ss -ncs -cp1 *
    3. /* heartbeat-server.c
    4.  *
    5.  * improved by ztguang, 2015.12.20
    6.  *
    7.  * Copyright (c) 2000 Sean Walton and Macmillan Publishers. Use may be in
    8.  * whole or in part in accordance to the General Public License (GPL).
    9.  *
    10.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
    11.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
    12.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
    13.  * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
    14.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
    15.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
    16.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
    17.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
    18.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
    19.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
    20.  * SUCH DAMAGE.
    21. */
    22. /*****************************************************************************/
    23. /*** heartbeat-server.c ***/
    24. /*** ***/
    25. /*** Demonstrates how to keep track of the server using a "heartbeat". If ***/
    26. /*** the heartbeat is lost, the connection can be reestablished and the ***/
    27. /*** session resumed. ***/
    28. /*****************************************************************************/
    29. #include <stdio.h>
    30. #include <stdlib.h>
    31. #include <unistd.h>
    32. #include <strings.h>
    33. #include <errno.h>
    34. #include <fcntl.h>
    35. #include <resolv.h>
    36. #include <signal.h>
    37. #include <sys/wait.h>
    38. #include <netinet/in.h>
    39. #include <arpa/inet.h>
    40. #include <sys/socket.h>

    42. #define BUF_SIZE 1024            //默认缓冲区
    43. #define SERVER_PORT 11111        //监听端口
    44. #define SERVER_HOST "127.0.0.1"    //服务器IP地址
    45. #define LISTEN_SIZE 10            //监听队列长度

    47. //两个有用的宏定义：检查和赋值并且检测
    48. #define CHK(eval) if(eval < 0){perror("eval"); exit(-1);}
    49. #define CHK2(res, eval) if((res = eval) < 0){perror("eval"); exit(-1);}

    51. int client;
    52. struct sigaction act;
    53. /*---------------------------------------------------------------------
    54.     sig_handler - catch and send heartbeat.
    55.  ---------------------------------------------------------------------*/
    56. void sig_handler(int signum)
    57. {
    58.     if (signum == SIGURG) {
    59.         char c;
    60.         recv(client, &c, sizeof(c), MSG_OOB);
    61.         if (c == '?')
    62.             /* Are you alive? */
    63.             send(client, "Y", 1, MSG_OOB);    /* */
    64.     } else if (signum == SIGCHLD)
    65.         wait(0);
    66. }

    68. /*---------------------------------------------------------------------
    69.     servlet - process requests
    70.  ---------------------------------------------------------------------*/
    71. void servlet(void)
    72. {
    73.     int bytes;
    74.     char buffer[BUF_SIZE];
    75.     bzero(&act, sizeof(act));
    76.     act.sa_handler = sig_handler;
    77.     act.sa_flags = SA_RESTART;
    78.     sigaction(SIGURG, &act, 0);    /* connect SIGURG signal */
    79.     if (fcntl(client, F_SETOWN, getpid()) != 0)
    80.         perror("Can't claim SIGIO and SIGURG");
    81.     do {
    82.         bzero(buffer, BUF_SIZE);
    83.         bytes = recv(client, buffer, sizeof(buffer), 0);
    84.         printf("recieve [%s] from client-%d ", buffer, client);
    85.         if (bytes > 0)
    86.             send(client, buffer, bytes, 0);
    87.     } while (bytes > 0);

    89.     printf("end while ");

    91.     close(client);
    92.     exit(0);
    93. }

    95. /*---------------------------------------------------------------------
    96.     main - set up client and begin the heartbeat.
    97.  ---------------------------------------------------------------------*/
    98. int main(int count, char *strings[])
    99. {
    100.     bzero(&act, sizeof(act));
    101.     act.sa_handler = sig_handler;
    102.     act.sa_flags = SA_NOCLDSTOP | SA_RESTART;
    103.     if (sigaction(SIGCHLD, &act, 0) != 0)
    104.         perror("sigaction()");

    106.     int listener;        //监听socket
    107.     struct sockaddr_in addr, peer;
    108.     addr.sin_family = PF_INET;
    109.     addr.sin_port = htons(SERVER_PORT);
    110.     addr.sin_addr.s_addr = inet_addr(SERVER_HOST);
    111.     socklen_t socklen;
    112.     socklen = sizeof(struct sockaddr_in);
    113.     CHK2(listener, socket(PF_INET, SOCK_STREAM, 0));    //初始化监听socket

    115.     // 设置套接字选项避免地址使用错误
    116.     int on = 1;
    117.     if ((setsockopt(listener, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on))) < 0) {
    118.         perror("server: setsockopt failed");
    119.         exit(EXIT_FAILURE);
    120.     }

    122.     CHK(bind(listener, (struct sockaddr *)&addr, sizeof(addr)));    //绑定监听socket

    124.     //printf("listen ");
    125.     CHK(listen(listener, LISTEN_SIZE));    //设置监听
    126.     printf("server: listening ");

    128.     while (1) {
    129.         client = accept(listener, (struct sockaddr *)&peer, &socklen);
    130.         printf("accept client-%d ", client);
    131.         if (client > 0) {
    132.             if (fork() == 0) {
    133.                 close(listener);
    134.                 servlet();
    135.             } else
    136.                 close(client);
    137.         } else
    138.             perror("accept()");
    139.     }
    140.     close(client);
    141.     return 0;
    142. }

    heartbeat-client.c

    点击(此处)折叠或打开

    1. // gcc heartbeat-client.c -o heartbeat-client
    2. // indent -npro -kr -i8 -ts8 -sob -l280 -ss -ncs -cp1 *
    3. /* heartbeat-client.c
    4.  *
    5.  * improved by ztguang, 2015.12.20
    6.  *
    7.  * Copyright (c) 2000 Sean Walton and Macmillan Publishers. Use may be in
    8.  * whole or in part in accordance to the General Public License (GPL).
    9.  *
    10.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
    11.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
    12.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
    13.  * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
    14.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
    15.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
    16.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
    17.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
    18.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
    19.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
    20.  * SUCH DAMAGE.
    21. */
    22. /*****************************************************************************/
    23. /*** heartbeat-client.c ***/
    24. /*** ***/
    25. /*** Demonstrates how to keep track of the server using a "heartbeat". If ***/
    26. /*** the heartbeat is lost, the connection can be reestablished and the ***/
    27. /*** session resumed. ***/
    28. /*****************************************************************************/
    29. #include <stdio.h>
    30. #include <stdlib.h>
    31. #include <unistd.h>
    32. #include <string.h>
    33. #include <errno.h>
    34. #include <fcntl.h>
    35. #include <signal.h>
    36. #include <resolv.h>
    37. #include <netinet/tcp.h>
    38. #include <netinet/in.h>
    39. #include <arpa/inet.h>
    40. #include <sys/socket.h>

    42. #define SERVER_PORT 11111    //监听端口
    43. #define SERVER_HOST "127.0.0.1"    //服务器IP地址
    44. #define DELAY 2            /*seconds */

    46. //两个有用的宏定义：检查和赋值并且检测
    47. #define CHK(eval) if(eval < 0){perror("eval"); exit(-1);}
    48. #define CHK2(res, eval) if((res = eval) < 0){perror("eval"); exit(-1);}

    50. int serverfd, got_reply = 1;

    52. /*---------------------------------------------------------------------
    53.     sig_handler - if the single is OOB, set flag. If ALARM, send heartbeat.
    54.  ---------------------------------------------------------------------*/
    55. void sig_handler(int signum)
    56. {
    57.     int bytes, len;

    59.     struct tcp_info info;
    60.     int leng = sizeof(info);

    62.     if (signum == SIGURG) {
    63. //*
    64.         getsockopt(serverfd, IPPROTO_TCP, TCP_INFO, &info, (socklen_t *) &leng);
    65.         if ((info.tcpi_state == TCP_ESTABLISHED)) {        // socket connected
    66.             ; //return 1;
    67.         } else {        // socket disconnected
    68.             return;
    69.         }
    70. //*/
    71.         char c;
    72.         CHK2(bytes, recv(serverfd, &c, sizeof(c), MSG_OOB));
    73.         if (bytes == 0) return;

    75.         got_reply = (c == 'Y');    /* Got reply */
    76.         fprintf(stderr, "[server is alive] ");
    77.     } else if (signum == SIGALRM) {
    78.         printf("reply = %d ", got_reply);
    79.         if (got_reply) {
    80.             alarm(DELAY);    // Wait a while
    81. //*
    82.             getsockopt(serverfd, IPPROTO_TCP, TCP_INFO, &info, (socklen_t *) &leng);
    83.             if ((info.tcpi_state == TCP_ESTABLISHED)) {        // socket connected
    84.                 ; //return 1;
    85.             } else {        // socket disconnected
    86.                 return;
    87.             }
    88. //*/
    89.             CHK2(len, send(serverfd, "?", 1, MSG_OOB));        // Alive??
    90.             if (len == 0) return;
    91.             got_reply = 0;
    92.         } else {
    93.             fprintf(stderr, "Lost connection to server! ");
    94.             alarm(DELAY);    /* Wait a while */
    95.         }
    96.     }
    97. }

    99. /*---------------------------------------------------------------------
    100.     main - set up client and begin the heartbeat.
    101.  ---------------------------------------------------------------------*/
    102. int main(int count, char *strings[])
    103. {

    105.     int bytes, len;
    106.     char line[100];

    108.     struct tcp_info info;
    109.     int leng = sizeof(info);

    111.     struct sigaction act;
    112.     bzero(&act, sizeof(act));
    113.     act.sa_handler = sig_handler;
    114.     act.sa_flags = SA_RESTART;
    115.     sigaction(SIGURG, &act, 0);
    116.     sigaction(SIGALRM, &act, 0);

    118.     struct sockaddr_in seraddr;
    119.     seraddr.sin_family = PF_INET;
    120.     seraddr.sin_port = htons(SERVER_PORT);
    121.     seraddr.sin_addr.s_addr = inet_addr(SERVER_HOST);

    123.     CHK2(serverfd, socket(PF_INET, SOCK_STREAM, 0));
    124.     if (fcntl(serverfd, F_SETOWN, getpid()) != 0)    // claim SIGIO/SIGURG signals
    125.         perror("Can't claim SIGURG and SIGIO");

    127.     while (1)
    128.     {
    129.         got_reply = 1;

    131.         close(serverfd);
    132.         CHK2(serverfd, socket(PF_INET, SOCK_STREAM, 0));
    133.         if (fcntl(serverfd, F_SETOWN, getpid()) != 0) {        // claim SIGIO/SIGURG signals
    134.             perror("Can't claim SIGURG and SIGIO");
    135.             continue;
    136.         }

    138.         if (connect(serverfd, (struct sockaddr *)&seraddr, sizeof(seraddr)) == 0) {
    139.             printf("connect [OK] ");
    140.             alarm(DELAY);
    141.             do {
    142. //*
    143.                 getsockopt(serverfd, IPPROTO_TCP, TCP_INFO, &info, (socklen_t *) &leng);
    144.                 if ((info.tcpi_state == TCP_ESTABLISHED)) {        // socket connected
    145.                     ; //return 1;
    146.                 } else {        // socket disconnected
    147.                     break;
    148.                 }
    149. //*/
    150.                 //CHK2(len, send(serverfd, "?", 1, MSG_OOB));        // Alive??
    151.                 //got_reply = 0;
    152.                 //CHK2(len, send(serverfd, "restart", 7, 0));
    153.                 //printf("send [restart] ");
    154.                 memset(line, 0x00, 100);
    155.                 gets(line);
    156.                 printf("send [%s] ", line);
    157.                 //CHK2(len, send(serverfd, "adsf", 4, 0)); sleep(1);

    159.                 CHK2(len, send(serverfd, line, strlen(line), 0));        // send
    160.                 if (len <= 0) break;

    162.                 CHK2(bytes, recv(serverfd, line, sizeof(line), 0));        // recieve
    163.                 if (bytes <= 0) break;

    165.             } while (bytes > 0);

    167.             printf("end while ");

    169.         } else {
    170.             perror("connect failed, retry! ");
    171.             sleep(2);
    172.         }
    173.     }
    174.     //close(serverfd);
    175.     //return 0;
    176. }






    <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>
    阅读(49) | 评论(0) | 转发(0) |
    0

    上一篇：如何判断SOCKET已经断开

    下一篇：遗产

    相关热门文章
    给主人留下些什么吧！~~
    评论热议
  • 相关阅读:
    慕课网-安卓攻城狮视频学习及练习（二）
    慕课网-安卓攻城狮视频学习及练习（一）
    1126 Eulerian Path
    1127 ZigZagging on a Tree
    1128 N Queens Puzzle
    1129 Recommendation System
    1130 Infix Expression
    1131 Subway Map
    1132 Cut Integer
    1133 Splitting A Linked List
  • 原文地址：https://www.cnblogs.com/ztguang/p/12649392.html
Copyright © 2011-2022 走看看