From bf17db04a093ca277c421ea7d24fe2a050963dca Mon Sep 17 00:00:00 2001
From: wangzhengquan <wangzhengquan85@126.com>
Date: 星期三, 10 六月 2020 18:25:38 +0800
Subject: [PATCH] version 1
---
test/nng/reqrep.c | 211 +++++++++++++++++++++++++++++++++-------------------
1 files changed, 133 insertions(+), 78 deletions(-)
diff --git a/test/nng/reqrep.c b/test/nng/reqrep.c
index e6f17c7..805afe6 100644
--- a/test/nng/reqrep.c
+++ b/test/nng/reqrep.c
@@ -1,106 +1,161 @@
-#include <stdlib.h>
+//
+// Copyright 2018 Staysail Systems, Inc. <info@staysail.tech>
+// Copyright 2018 Capitar IT Group BV <info@capitar.com>
+//
+// This software is supplied under the terms of the MIT License, a
+// copy of which should be located in the distribution where this
+// file was obtained (LICENSE.txt). A copy of the license may also be
+// found online at https://opensource.org/licenses/MIT.
+//
+
+//
+// This is just a simple REQ/REP demonstration application. It is derived
+// from the legacy nanomsg demonstration program of the same name, written
+// by Tim Dysinger, but updated for nng. I've also updated it to pass simpler
+// binary data rather than strings over the network.
+//
+// The program implements a simple RPC style service, which just returns
+// the date in UNIX time (seconds since 1970).
+//
+#include <stdint.h>
#include <stdio.h>
+#include <stdlib.h>
#include <string.h>
#include <time.h>
-#include <nanomsg/nn.h>
-#include <nanomsg/reqrep.h>
-#define NODE0 "node0"
-#define NODE1 "node1"
-#define DATE "DATE"
+#include <nng/nng.h>
+#include <nng/protocol/reqrep0/rep.h>
+#include <nng/protocol/reqrep0/req.h>
+
+#define CLIENT "client"
+#define SERVER "server"
+#define DATECMD 1
+
+#define PUT64(ptr, u) \
+ do { \
+ (ptr)[0] = (uint8_t)(((uint64_t)(u)) >> 56); \
+ (ptr)[1] = (uint8_t)(((uint64_t)(u)) >> 48); \
+ (ptr)[2] = (uint8_t)(((uint64_t)(u)) >> 40); \
+ (ptr)[3] = (uint8_t)(((uint64_t)(u)) >> 32); \
+ (ptr)[4] = (uint8_t)(((uint64_t)(u)) >> 24); \
+ (ptr)[5] = (uint8_t)(((uint64_t)(u)) >> 16); \
+ (ptr)[6] = (uint8_t)(((uint64_t)(u)) >> 8); \
+ (ptr)[7] = (uint8_t)((uint64_t)(u)); \
+ } while (0)
+
+#define GET64(ptr, v) \
+ v = (((uint64_t)((uint8_t)(ptr)[0])) << 56) + \
+ (((uint64_t)((uint8_t)(ptr)[1])) << 48) + \
+ (((uint64_t)((uint8_t)(ptr)[2])) << 40) + \
+ (((uint64_t)((uint8_t)(ptr)[3])) << 32) + \
+ (((uint64_t)((uint8_t)(ptr)[4])) << 24) + \
+ (((uint64_t)((uint8_t)(ptr)[5])) << 16) + \
+ (((uint64_t)((uint8_t)(ptr)[6])) << 8) + \
+ (((uint64_t)(uint8_t)(ptr)[7]))
void
-fatal(const char *func)
+fatal(const char *func, int rv)
{
- fprintf(stderr, "%s: %s\n", func, nn_strerror(nn_errno()));
- exit(1);
+ fprintf(stderr, "%s: %s\n", func, nng_strerror(rv));
+ exit(1);
}
-char *
-date(void)
+void
+showdate(time_t now)
{
- time_t now = time(&now);
- struct tm *info = localtime(&now);
- char *text = asctime(info);
- text[strlen(text)-1] = '\0'; // remove '\n'
- return (text);
+ struct tm *info = localtime(&now);
+ printf("%s", asctime(info));
}
int
-node0(const char *url)
+server(const char *url)
{
- int sz_date = strlen(DATE) + 1; // '\0' too
- int sock;
- int rv;
+ nng_socket sock;
+ int rv;
- if ((sock = nn_socket(AF_SP, NN_REP)) < 0) {
- fatal("nn_socket");
- }
- if ((rv = nn_bind(sock, url)) < 0) {
- fatal("nn_bind");
- }
- for (;;) {
- char *buf = NULL;
- int bytes;
- if ((bytes = nn_recv(sock, &buf, NN_MSG, 0)) < 0) {
- fatal("nn_recv");
- }
- if ((bytes == (strlen(DATE) + 1)) && (strcmp(DATE, buf) == 0)) {
- printf("NODE0: RECEIVED DATE REQUEST\n");
- char *d = date();
- int sz_d = strlen(d) + 1; // '\0' too
- printf("NODE0: SENDING DATE %s\n", d);
- if ((bytes = nn_send(sock, d, sz_d, 0)) < 0) {
- fatal("nn_send");
- }
- }
- nn_freemsg(buf);
- }
+ if ((rv = nng_rep0_open(&sock)) != 0) {
+ fatal("nng_rep0_open", rv);
+ }
+ if ((rv = nng_listen(sock, url, NULL, 0)) != 0) {
+ fatal("nng_listen", rv);
+ }
+ for (;;) {
+ char * buf = NULL;
+ size_t sz;
+ uint64_t val;
+ if ((rv = nng_recv(sock, &buf, &sz, NNG_FLAG_ALLOC)) != 0) {
+ fatal("nng_recv", rv);
+ }
+ if ((sz == sizeof(uint64_t)) &&
+ ((GET64(buf, val)) == DATECMD)) {
+ time_t now;
+ printf("SERVER: RECEIVED DATE REQUEST\n");
+ now = time(&now);
+ printf("SERVER: SENDING DATE: ");
+ showdate(now);
+
+ // Reuse the buffer. We know it is big enough.
+ PUT64(buf, (uint64_t) now);
+ rv = nng_send(sock, buf, sz, NNG_FLAG_ALLOC);
+ if (rv != 0) {
+ fatal("nng_send", rv);
+ }
+ continue;
+ }
+ // Unrecognized command, so toss the buffer.
+ nng_free(buf, sz);
+ }
}
int
-node1(const char *url)
+client(const char *url)
{
- int sz_date = strlen(DATE) + 1; // '\0' too
- char *buf = NULL;
- int bytes = -1;
- int sock;
- int rv;
+ nng_socket sock;
+ int rv;
+ size_t sz;
+ char * buf = NULL;
+ uint8_t cmd[sizeof(uint64_t)];
- if ((sock = nn_socket(AF_SP, NN_REQ)) < 0) {
- fatal("nn_socket");
- }
- if ((rv = nn_connect (sock, url)) < 0) {
- fatal("nn_connect");
- }
- printf("NODE1: SENDING DATE REQUEST %s\n", DATE);
- if ((bytes = nn_send(sock, DATE, sz_date, 0)) < 0) {
- fatal("nn_send");
- }
- if ((bytes = nn_recv(sock, &buf, NN_MSG, 0)) < 0) {
- fatal("nn_recv");
- }
- printf("NODE1: RECEIVED DATE %s\n", buf);
- nn_freemsg(buf);
- return (nn_shutdown(sock, 0));
+ PUT64(cmd, DATECMD);
+
+ if ((rv = nng_req0_open(&sock)) != 0) {
+ fatal("nng_socket", rv);
+ }
+ if ((rv = nng_dial(sock, url, NULL, 0)) != 0) {
+ fatal("nng_dial", rv);
+ }
+ printf("CLIENT: SENDING DATE REQUEST\n");
+ if ((rv = nng_send(sock, cmd, sizeof(cmd), 0)) != 0) {
+ fatal("nng_send", rv);
+ }
+ if ((rv = nng_recv(sock, &buf, &sz, NNG_FLAG_ALLOC)) != 0) {
+ fatal("nng_recv", rv);
+ }
+
+ if (sz == sizeof(uint64_t)) {
+ uint64_t now;
+ GET64(buf, now);
+ printf("CLIENT: RECEIVED DATE: ");
+ showdate((time_t) now);
+ } else {
+ printf("CLIENT: GOT WRONG SIZE!\n");
+ }
+
+ // This assumes that buf is ASCIIZ (zero terminated).
+ nng_free(buf, sz);
+ nng_close(sock);
+ return (0);
}
int
main(const int argc, const char **argv)
{
- if ((argc > 1) && (strcmp(NODE0, argv[1]) == 0))
- return (node0(argv[2]));
+ if ((argc > 1) && (strcmp(CLIENT, argv[1]) == 0))
+ return (client(argv[2]));
- if ((argc > 1) && (strcmp(NODE1, argv[1]) == 0))
- return (node1(argv[2]));
+ if ((argc > 1) && (strcmp(SERVER, argv[1]) == 0))
+ return (server(argv[2]));
- fprintf(stderr, "Usage: reqrep %s|%s <URL> ...\n", NODE0, NODE1);
- return (1);
+ fprintf(stderr, "Usage: reqrep %s|%s <URL> ...\n", CLIENT, SERVER);
+ return (1);
}
-
-/*
-
-./reqrep node0 ipc:///tmp/reqrep.ipc & node0=$! && sleep 1
-./reqrep node1 ipc:///tmp/reqrep.ipc
-kill $node0
-*/
\ No newline at end of file
--
Gitblit v1.8.0