From f75a37444e7a97fb58439077eea4d1b3c576bfe3 Mon Sep 17 00:00:00 2001 From: wangzhengquan <wangzhengquan85@126.com> Date: 星期一, 20 七月 2020 11:17:33 +0800 Subject: [PATCH] update --- test2/pub_sub.c | 41 +++++++++++++++++++---------------------- 1 files changed, 19 insertions(+), 22 deletions(-) diff --git a/test2/pub_sub.c b/test2/pub_sub.c index 669783f..78aff69 100644 --- a/test2/pub_sub.c +++ b/test2/pub_sub.c @@ -1,41 +1,37 @@ -#include "socket.h" +#include "mod_socket.h" +#include "shm_mm.h" +#include "usg_common.h" void server(int port) { - void *socket = shm_open_socket(PUB_SUB); - shm_bind(socket, port); - shm_listen(socket); + void *socket = mod_open_socket(PUB_SUB); + mod_socket_bind(socket, port); + mod_listen(socket); int size; void *recvbuf; char sendbuf[512]; while(true) { - shm_recv(socket, &recvbuf, &size); - sprintf(sendbuf, "pub: %s", recvbuf); - puts(sendbuf); - shm_send(socket, sendbuf, strlen(sendbuf)+1) ; - shm_free(recvbuf); + printf("璇疯緭鍏ュ彂甯冩秷鎭�:"); + scanf("%s", sendbuf); + mod_send(socket, sendbuf, strlen(sendbuf)+1) ; + free(recvbuf); } - shm_close_socket(socket); + mod_close_socket(socket); } void client(int port) { - void *socket = shm_open_socket(PUB_SUB); - shm_connect(socket, port); + void *socket = mod_open_socket(PUB_SUB); + mod_connect(socket, port); int size; void *recvbuf; - char sendbuf[512]; - - sprintf(sendbuf, "sub"); - shm_send(socket, sendbuf, strlen(sendbuf)+1) ; - while(true) { - - shm_recv(socket, &recvbuf, &size); - printf("received sub message: %s\n", (char *)recvbuf); - shm_free(recvbuf); + + while(mod_recv(socket, &recvbuf, &size) == 0) { + printf("鏀跺埌璁㈤槄娑堟伅: %s\n", (char *)recvbuf); + free(recvbuf); } - shm_close_socket(socket); + mod_close_socket(socket); } int main(int argc, char *argv[]) { @@ -54,6 +50,7 @@ if (strcmp("client", argv[1]) == 0) client(port); + shm_destroy(); // fprintf(stderr, "Usage: reqrep %s|%s <URL> ...\n", "server", "client"); return 0; -- Gitblit v1.8.0