C 语言回调函数客户端简单示例

鸣涧

C 语言回调函数客户端简单示例

回调函数就是一个被作为参数传递的函数。在C语言中，回调函数只能使用函数指针实现，在C++、Python、ECMAScript等更现代的编程语言中还可以使用仿函数或匿名函数。

回调函数的使用可以大大提升编程的效率，这使得它在现代编程中被非常多地使用。同时，有一些需求必须要使用回调函数来实现。使用回调函数就会让别的程序员调用我们的代码时操作简单。

client.h：

1. #ifndef _HY_TCP_H
   
2. #define _HY_TCP_H
   
3. 
   
4. typedef void (* recv_callback)(char *data, int len);//定义回调
   
5. typedef struct param{
   
6.         recv_callback callback;
   
7.     } callback_param;
   
8. 
   
9. int tcp_dns_change(char *dns, char *str, int size);
   
10. int tcp_connect(char *IP,int PORT);
    
11. void tcp_disconnect(void);
    
12. int tcp_send(char *buf);
    
13. int tcp_recv(char *buf, int size);
    
14. void *thread_recv(void *param);
    
15. void tcp_register_callback(recv_callback cb); //调用回调函数
    
16. 
    
17. #endif

复制代码

client.c：

1. #include <stdio.h>
   
2. #include <unistd.h>
   
3. #include <sys/socket.h>
   
4. #include <stdlib.h>
   
5. #include <sys/types.h>
   
6. #include <netinet/in.h>
   
7. #include <arpa/inet.h>
   
8. #include <string.h>
   
9. #include <netdb.h>
   
10. #include <syslog.h>
    
11. #include <pthread.h>
    
12. #include <poll.h>
    
13. #include "hy_tcp.h"
    
14. 
    
15. int sockfd;
    
16. 
    
17. enum {
    
18.     STATE_NOCONNECTED,  //未连接
    
19.     STATE_CONNECTED,    //已连接
    
20.     STATE_DISCONNECTE   //连接断开
    
21. };
    
22. 
    
23. int connect_state = STATE_NOCONNECTED;
    
24. 
    
25. int tcp_dns_change(char *dns, char *str, int size)
    
26. {
    
27.     struct hostent *hptr;
    
28.     hptr = gethostbyname(dns);
    
29.     if (hptr == NULL) {
    
30.         syslog(LOG_ERR, "gethostbyname error");
    
31.         return -1;
    
32.     }
    
33.     syslog(LOG_INFO, "ipaddr:%s\n", inet_ntop(hptr->h_addrtype, hptr->h_addr, str, size));
    
34.     return 0;
    
35. }
    
36. 
    
37. int tcp_connect(char *IP,int PORT)
    
38. {
    
39.     if (connect_state == STATE_NOCONNECTED) {
    
40.         struct sockaddr_in server_addr;
    
41.         sockfd = socket(AF_INET, SOCK_STREAM, 0);
    
42.         if (sockfd < 0) {
    
43.             syslog(LOG_ERR, "create socket failed!\n");
    
44.             return -1;
    
45.         }
    
46.         memset(&server_addr, 0, sizeof(server_addr));
    
47.         server_addr.sin_family = AF_INET;
    
48.         server_addr.sin_port = htons(PORT);
    
49.         server_addr.sin_addr.s_addr = inet_addr(IP);
    
50.         if (connect(sockfd, (struct sockaddr*)&server_addr, sizeof(server_addr)) < 0) {
    
51.             syslog(LOG_ERR, "connet error!\n");
    
52.             return -1;
    
53.         }
    
54.         connect_state = STATE_CONNECTED;
    
55.         syslog(LOG_INFO, "connect success!\n");
    
56.         return 0;
    
57.     }
    
58.     else
    
59.         return -1;
    
60. }
    
61. 
    
62. void tcp_disconnect(void)
    
63. {
    
64.     if (connect_state != STATE_NOCONNECTED) {
    
65.         close(sockfd);
    
66.         syslog(LOG_INFO, "connect is break!\n");
    
67.     }
    
68. }
    
69. 
    
70. int tcp_send(char *buf)
    
71. {
    
72.     if (connect_state == STATE_CONNECTED) {
    
73.         return send(sockfd, buf, strlen(buf)+1, 0);
    
74.     }
    
75.     else
    
76.         return -1;
    
77. }
    
78. 
    
79. void *thread_recv(void *param)
    
80. {
    
81.     int ret;
    
82.     static char buf[2048] = {0};
    
83.     char heartbeat_buf[] = "heartbeat data";
    
84.     callback_param *p = (callback_param *)param;
    
85.     struct pollfd c_poll;
    
86.     c_poll.fd = sockfd;
    
87.     c_poll.events = POLLIN;
    
88.     memset(buf, 0, sizeof(buf));
    
89.     while (1) {
    
90. 
    
91.         if (connect_state == STATE_CONNECTED) {
    
92.             ret = poll(&c_poll, 1, 5000);
    
93.             if (ret < 0) {
    
94.                 syslog(LOG_ERR, "poll error!\n");
    
95.                 break;
    
96.             }
    
97.             else if (0 == ret) {
    
98.                 if (send(sockfd, heartbeat_buf, sizeof(heartbeat_buf), 0) < 0) {
    
99.                     connect_state = STATE_DISCONNECTE;
    
100.                     syslog(LOG_ERR, "disconnect!\n");
     
101.                     break;
     
102.                 }
     
103.             }
     
104.             else {
     
105.                 if (recv(sockfd, buf, sizeof(buf), 0) > 0) {
     
106.                     syslog(LOG_INFO, "recv:%s\n", buf);
     
107.                     p->callback(buf, sizeof(buf));
     
108.                 }
     
109.             }
     
110.         }
     
111.         else
     
112.             break;
     
113.     }
     
114. }
     
115. 
     
116. void tcp_register_callback(recv_callback cb)
     
117. {
     
118.     pthread_t recv_t;
     
119.     static callback_param param;
     
120.     param.callback = cb;
     
121.     pthread_create(&recv_t, NULL, thread_recv, (void *)param);
     
122.     pthread_detach(recv_t);
     
123. }

复制代码

main.c：

1. #include <stdio.h>
   
2. #include <unistd.h>
   
3. #include <sys/socket.h>
   
4. #include <netinet/in.h>
   
5. #include <string.h>
   
6. #include <arpa/inet.h>
   
7. #include "hy_tcp.h"
   
8. #include "hy_udp.h"
   
9. 
   
10. #define PORT 6000   //服务器端口号
    
11. #if 1
    
12. 
    
13. void r_callback(char *msg, int len) {
    
14.    
    
15. }
    
16. 
    
17. int main(int argc, char *argv[])
    
18. {
    
19.     char recv_buf[1024] = {0};
    
20.     char send_buf[1024] = {0};
    
21.     char str[16];
    
22.     recv_callback func = r_callback;
    
23.     if(argc < 2){
    
24.         printf("required parameter missing\n");
    
25.         return -1;
    
26.     }
    
27.     //tcp_dns_change(argv[1], str, sizeof(str));
    
28.     tcp_connect(argv[1],PORT);
    
29.     tcp_register_callback(func);
    
30.    
    
31.     while(1){
    
32.         memset(send_buf,0,sizeof(send_buf));
    
33.         printf("please input something\n");
    
34.         scanf("%s",send_buf);
    
35.         tcp_send(send_buf);
    
36.     }
    
37.     tcp_disconnect();
    
38.     return 0;
    
39. }
    
40. #endif
    

复制代码