From 1b94589dcb8d497d2d8a208efd61a54631f6b84e Mon Sep 17 00:00:00 2001 From: wangzhengquan <wangzhengquan85@126.com> Date: 星期三, 23 十二月 2020 16:08:33 +0800 Subject: [PATCH] update --- test_net_socket/heart_beat.c | 41 +++++++++++++++++++++++++++++------------ 1 files changed, 29 insertions(+), 12 deletions(-) diff --git a/test_socket/dgram_mod_survey.c b/test_net_socket/heart_beat.c similarity index 69% rename from test_socket/dgram_mod_survey.c rename to test_net_socket/heart_beat.c index da3260f..562cb23 100644 --- a/test_socket/dgram_mod_survey.c +++ b/test_net_socket/heart_beat.c @@ -1,6 +1,10 @@ -#include "dgram_mod_socket.h" +#include "net_mod_server_socket_wrapper.h" +#include "net_mod_socket_wrapper.h" +#include "bus_server_socket_wrapper.h" + #include "shm_mm_wraper.h" #include "usg_common.h" +#include <getopt.h> typedef struct Targ { @@ -10,43 +14,50 @@ }Targ; void sigint_handler(int sig) { - //dgram_mod_close_socket(server_socket); + // net_mod_socket_close(server_socket); printf("===Catch sigint======================\n"); shm_mm_wrapper_destroy(); exit(0); } void server(int port) { - void *socket = dgram_mod_open_socket(); - dgram_mod_bind(socket, port); + void *serv = net_mod_socket_open(); + net_mod_socket_bind(serv, port); int size; void *recvbuf; char sendbuf[512]; int rv; int remote_port; while (true) { - if ((rv = dgram_mod_recvfrom_timeout(socket, &recvbuf, &size, &remote_port, 15, 0) ) == 0) { + if ((rv = net_mod_socket_recvfrom(serv, &recvbuf, &size, &remote_port) ) == 0) { printf( "RECEIVED HREARTBEAT FROM %d: %s\n", remote_port, recvbuf); + net_mod_socket_sendto(serv, "suc", strlen("suc")+1, remote_port); free(recvbuf); } } - dgram_mod_close_socket(socket); + net_mod_socket_close(serv); } void client(int port) { - void *socket = dgram_mod_open_socket(); + int rv; + void *client = net_mod_socket_open(); int size; char sendbuf[512]; long i = 0; + net_node_t node_arr[] = {"", 0, 100}; + int node_arr_size = 1; + + int recv_arr_size; + net_mod_recv_msg_t *recv_arr; while (true) { sprintf(sendbuf, "%d", i); printf("SEND HEART:%s\n", sendbuf); - dgram_mod_sendto(socket, sendbuf, strlen(sendbuf) + 1, port); + rv = net_mod_socket_sendandrecv(client, node_arr, node_arr_size, sendbuf, strlen(sendbuf)+1, NULL, NULL); // sleep(1); i++; } - dgram_mod_close_socket(socket); + net_mod_socket_close(client); } @@ -54,20 +65,26 @@ signal(SIGINT, sigint_handler); Targ *targ = (Targ *)arg; int port = targ->port; - void *socket = dgram_mod_open_socket(); + void *socket = net_mod_socket_open(); int size; char sendbuf[512]; long scale = 10; long i = 0; + net_node_t node_arr[] = {"", 0, 100}; + int node_arr_size = 1; + + int recv_arr_size; + net_mod_recv_msg_t *recv_arr; + while (i < scale) { sprintf(sendbuf, "%d", i); printf("%d SEND HEART:%s\n", targ->id, sendbuf); - dgram_mod_sendto(socket, sendbuf, strlen(sendbuf) + 1, port); + net_mod_socket_sendto(socket, sendbuf, strlen(sendbuf) + 1, port); sleep(1); i++; } - dgram_mod_close_socket(socket); + net_mod_socket_close(socket); return (void *)i; } -- Gitblit v1.8.0