From 72b7aebb0022f8e391c999348763acd5f7a16133 Mon Sep 17 00:00:00 2001 From: wangzhengquan <wangzhengquan85@126.com> Date: 星期四, 26 十一月 2020 18:56:33 +0800 Subject: [PATCH] update --- test_socket/dgram_mod_bus.c | 37 ++++++++++++++++++------------------- 1 files changed, 18 insertions(+), 19 deletions(-) diff --git a/test_socket/dgram_mod_bus.c b/test_socket/dgram_mod_bus.c index f800f10..9f4795b 100644 --- a/test_socket/dgram_mod_bus.c +++ b/test_socket/dgram_mod_bus.c @@ -9,14 +9,14 @@ exit(0); } -void server(int port, bool restart) { +void server(int key, bool restart) { server_socket = dgram_mod_open_socket(); if(restart) { - dgram_mod_force_bind(server_socket, port); + dgram_mod_force_bind(server_socket, key); } else { - dgram_mod_bind(server_socket, port); + dgram_mod_bind(server_socket, key); } @@ -28,18 +28,17 @@ pthread_detach(pthread_self()); void *recvbuf; int size; - int port; - while (dgram_mod_recvfrom( socket, &recvbuf, &size, &port) == 0) { + int key; + while (dgram_mod_recvfrom( socket, &recvbuf, &size, &key) == 0) { printf("鏀跺埌璁㈤槄娑堟伅:%s\n", recvbuf); free(recvbuf); } } -void client(int port) { +void client(int key) { void *socket = dgram_mod_open_socket(); - pthread_t tid; pthread_create(&tid, NULL, run_recv, socket); int size; @@ -56,7 +55,7 @@ if(strcmp(action, "sub") == 0) { printf("Please input topic!\n"); scanf("%s", topic); - if (dgram_mod_sub(socket, topic, strlen(topic), port) == 0) { + if (dgram_mod_sub(socket, topic, strlen(topic), key) == 0) { printf("%d Sub success!\n", dgram_mod_get_port(socket)); } else { printf("Sub failture!\n"); @@ -66,7 +65,7 @@ } else if(strcmp(action, "desub") == 0) { printf("Please input topic!\n"); scanf("%s", topic); - if (dgram_mod_desub(socket, topic, strlen(topic), port) == 0) { + if (dgram_mod_desub(socket, topic, strlen(topic), key) == 0) { printf("%d Desub success!\n", dgram_mod_get_port(socket)); } else { printf("Desub failture!\n"); @@ -77,7 +76,7 @@ // printf("%s %s %s\n", action, topic, content); printf("Please input topic and content\n"); scanf("%s %s", topic, content); - if(dgram_mod_pub(socket, topic, strlen(topic)+1, content, strlen(content)+1, port) == 0){ + if(dgram_mod_pub(socket, topic, strlen(topic)+1, content, strlen(content)+1, key) == 0){ printf("%d Pub success!\n", dgram_mod_get_port(socket)); } else { printf("Pub failture!\n"); @@ -100,29 +99,29 @@ int main(int argc, char *argv[]) { shm_init(512); - int port; + int key; if (argc < 3) { - fprintf(stderr, "Usage: %s %s|%s|rmkey <PORT> ...\n", argv[0], "server", "client"); + fprintf(stderr, "Usage: %s %s|%s|rmkey <key> ...\n", argv[0], "server", "client"); return 1; } - port = atoi(argv[2]); + key = atoi(argv[2]); if (strcmp("server", argv[1]) == 0) { if(argc >= 4 && strcmp("restart", argv[3]) == 0) { - server(port, true); + server(key, true); } else{ - server(port, false); + server(key, false); } } else if (strcmp("client", argv[1]) == 0) { - client(port); + client(key); } else if(strcmp("rmkey", argv[1]) == 0) { for(int i = 2; i < argc; i++) { - port = atoi(argv[i]); - dgram_mod_remove_key(port); - // printf("%d\n", port); + key = atoi(argv[i]); + dgram_mod_remove_key(key); + // printf("%d\n", key); } } -- Gitblit v1.8.0