summaryrefslogtreecommitdiff
path: root/tests/test_stream.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_stream.cpp')
-rw-r--r--tests/test_stream.cpp227
1 files changed, 227 insertions, 0 deletions
diff --git a/tests/test_stream.cpp b/tests/test_stream.cpp
new file mode 100644
index 0000000..5e80f01
--- /dev/null
+++ b/tests/test_stream.cpp
@@ -0,0 +1,227 @@
+/*
+ Copyright (c) 2007-2013 Contributors as noted in the AUTHORS file
+
+ This file is part of 0MQ.
+
+ 0MQ is free software; you can redistribute it and/or modify it under
+ the terms of the GNU Lesser General Public License as published by
+ the Free Software Foundation; either version 3 of the License, or
+ (at your option) any later version.
+
+ 0MQ is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU Lesser General Public License for more details.
+
+ You should have received a copy of the GNU Lesser General Public License
+ along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "testutil.hpp"
+
+// ZMTP protocol greeting structure
+
+typedef unsigned char byte;
+typedef struct {
+ byte signature [10]; // 0xFF 8*0x00 0x7F
+ byte version [2]; // 0x03 0x00 for ZMTP/3.0
+ byte mechanism [20]; // "NULL"
+ byte as_server;
+ byte filler [31];
+} zmtp_greeting_t;
+
+#define ZMTP_DEALER 5 // Socket type constants
+
+// This is a greeting matching what 0MQ will send us; note the
+// 8-byte size is set to 1 for backwards compatibility
+
+static zmtp_greeting_t greeting
+ = { { 0xFF, 0, 0, 0, 0, 0, 0, 0, 1, 0x7F }, { 3, 0 }, { 'N', 'U', 'L', 'L'} };
+
+static void
+test_stream_to_dealer (void)
+{
+ int rc;
+
+ // Set up our context and sockets
+ void *ctx = zmq_ctx_new ();
+ assert (ctx);
+
+ // We'll be using this socket in raw mode
+ void *stream = zmq_socket (ctx, ZMQ_STREAM);
+ assert (stream);
+
+ int zero = 0;
+ rc = zmq_setsockopt (stream, ZMQ_LINGER, &zero, sizeof (zero));
+ assert (rc == 0);
+ rc = zmq_bind (stream, "tcp://127.0.0.1:5556");
+ assert (rc == 0);
+
+ // We'll be using this socket as the other peer
+ void *dealer = zmq_socket (ctx, ZMQ_DEALER);
+ assert (dealer);
+ rc = zmq_setsockopt (dealer, ZMQ_LINGER, &zero, sizeof (zero));
+ assert (rc == 0);
+ rc = zmq_connect (dealer, "tcp://localhost:5556");
+
+ // Send a message on the dealer socket
+ rc = zmq_send (dealer, "Hello", 5, 0);
+ assert (rc == 5);
+
+ // First frame is identity
+ zmq_msg_t identity;
+ rc = zmq_msg_init (&identity);
+ assert (rc == 0);
+ rc = zmq_msg_recv (&identity, stream, 0);
+ assert (rc > 0);
+ assert (zmq_msg_more (&identity));
+
+ // Second frame is greeting signature
+ byte buffer [255];
+ rc = zmq_recv (stream, buffer, 255, 0);
+ assert (rc == 10);
+ assert (memcmp (buffer, greeting.signature, 10) == 0);
+
+ // Send our own protocol greeting
+ rc = zmq_msg_send (&identity, stream, ZMQ_SNDMORE);
+ assert (rc > 0);
+ rc = zmq_send (stream, &greeting, sizeof (greeting), 0);
+ assert (rc == sizeof (greeting));
+
+ // Now we expect the data from the DEALER socket
+ // We want the rest of greeting along with the Ready command
+ int bytes_read = 0;
+ while (bytes_read < 97) {
+ // First frame is the identity of the connection (each time)
+ rc = zmq_msg_recv (&identity, stream, 0);
+ assert (rc > 0);
+ assert (zmq_msg_more (&identity));
+ // Second frame contains the next chunk of data
+ rc = zmq_recv (stream, buffer + bytes_read, 255 - bytes_read, 0);
+ assert (rc >= 0);
+ bytes_read += rc;
+ }
+
+ // First two bytes are major and minor version numbers.
+ assert (buffer [0] == 3); // ZMTP/3.0
+ assert (buffer [1] == 0);
+
+ // Mechanism is "NULL"
+ assert (memcmp (buffer + 2, "NULL\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0", 20) == 0);
+ assert (memcmp (buffer + 54, "\4\51\5READY", 8) == 0);
+ assert (memcmp (buffer + 62, "\13Socket-Type\0\0\0\6DEALER", 22) == 0);
+ assert (memcmp (buffer + 84, "\10Identity\0\0\0\0", 13) == 0);
+
+ // Announce we are ready
+ memcpy (buffer, "\4\51\5READY", 8);
+ memcpy (buffer + 8, "\13Socket-Type\0\0\0\6ROUTER", 22);
+ memcpy (buffer + 30, "\10Identity\0\0\0\0", 13);
+
+ // Send Ready command
+ rc = zmq_msg_send (&identity, stream, ZMQ_SNDMORE);
+ assert (rc > 0);
+ rc = zmq_send (stream, buffer, 43, 0);
+ assert (rc == 43);
+
+ // Now we expect the data from the DEALER socket
+ // First frame is, again, the identity of the connection
+ rc = zmq_msg_recv (&identity, stream, 0);
+ assert (rc > 0);
+ assert (zmq_msg_more (&identity));
+
+ // Third frame contains Hello message from DEALER
+ rc = zmq_recv (stream, buffer, sizeof buffer, 0);
+ assert (rc == 7);
+
+ // Then we have a 5-byte message "Hello"
+ assert (buffer [0] == 0); // Flags = 0
+ assert (buffer [1] == 5); // Size = 5
+ assert (memcmp (buffer + 2, "Hello", 5) == 0);
+
+ // Send "World" back to DEALER
+ rc = zmq_msg_send (&identity, stream, ZMQ_SNDMORE);
+ assert (rc > 0);
+ byte world [] = { 0, 5, 'W', 'o', 'r', 'l', 'd' };
+ rc = zmq_send (stream, world, sizeof (world), 0);
+ assert (rc == sizeof (world));
+
+ // Expect response on DEALER socket
+ rc = zmq_recv (dealer, buffer, 255, 0);
+ assert (rc == 5);
+ assert (memcmp (buffer, "World", 5) == 0);
+
+ rc = zmq_close (dealer);
+ assert (rc == 0);
+
+ rc = zmq_close (stream);
+ assert (rc == 0);
+
+ rc = zmq_ctx_term (ctx);
+ assert (rc == 0);
+}
+
+
+static void
+test_stream_to_stream (void)
+{
+ int rc;
+ // Set-up our context and sockets
+ void *ctx = zmq_ctx_new ();
+ assert (ctx);
+
+ void *server = zmq_socket (ctx, ZMQ_STREAM);
+ assert (server);
+ rc = zmq_bind (server, "tcp://127.0.0.1:9080");
+ assert (rc == 0);
+
+ void *client = zmq_socket (ctx, ZMQ_STREAM);
+ assert (client);
+ rc = zmq_connect (client, "tcp://localhost:9080");
+ assert (rc == 0);
+ // It would be less surprising to get an empty message instead
+ // of having to fetch the identity like this [PH 2013/06/27]
+ uint8_t id [256];
+ size_t id_size = 256;
+ rc = zmq_getsockopt (client, ZMQ_IDENTITY, id, &id_size);
+ assert (rc == 0);
+
+ // Sent HTTP request on client socket
+ // First frame is server identity
+ rc = zmq_send (client, id, id_size, ZMQ_SNDMORE);
+ assert (rc == (int) id_size);
+ // Second frame is HTTP GET request
+ rc = zmq_send (client, "GET /\n\n", 7, 0);
+ assert (rc == 7);
+
+ // Get HTTP request; ID frame and then request
+ id_size = zmq_recv (server, id, 256, 0);
+ assert (id_size > 0);
+ uint8_t buffer [256];
+ rc = zmq_recv (server, buffer, 256, 0);
+ assert (rc > 0);
+ assert (memcmp (buffer, "GET /\n\n", 7) == 0);
+
+ // Send reply back to client
+ char http_response [] =
+ "HTTP/1.0 200 OK\r\n"
+ "Content-Type: text/plain\r\n"
+ "\r\n"
+ "Hello, World!";
+ zmq_send (server, id, id_size, ZMQ_SNDMORE);
+ zmq_send (server, http_response, sizeof (http_response), 0);
+
+ // Get reply at client and check that it's complete
+ id_size = zmq_recv (client, id, 256, 0);
+ assert (id_size > 0);
+ rc = zmq_recv (client, buffer, 256, 0);
+ assert (rc == sizeof (http_response));
+ assert (memcmp (buffer, http_response, sizeof (http_response)) == 0);
+}
+
+
+int main (void)
+{
+ setup_test_environment();
+ test_stream_to_dealer ();
+ test_stream_to_stream ();
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-05-03 18:51:52 +0000