int XS_TEST_MAIN ()
{
fprintf (stderr, "reqrep_ipc test running...\n");
void *ctx = xs_init ();
assert (ctx);
void *sb = xs_socket (ctx, XS_REP);
assert (sb);
int rc = xs_bind (sb, "ipc:///tmp/tester");
assert (rc == 0);
void *sc = xs_socket (ctx, XS_REQ);
assert (sc);
rc = xs_connect (sc, "ipc:///tmp/tester");
assert (rc == 0);
bounce (sb, sc);
rc = xs_close (sc);
assert (rc == 0);
rc = xs_close (sb);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
return 0 ;
}
int XS_TEST_MAIN ()
{
fprintf (stderr, "pair_tcp test running...\n");
void *ctx = xs_init ();
assert (ctx);
void *sb = xs_socket (ctx, XS_PAIR);
assert (sb);
int rc = xs_bind (sb, "tcp://127.0.0.1:5560");
assert (rc == 0);
void *sc = xs_socket (ctx, XS_PAIR);
assert (sc);
rc = xs_connect (sc, "tcp://127.0.0.1:5560");
assert (rc == 0);
bounce (sb, sc);
rc = xs_close (sc);
assert (rc == 0);
rc = xs_close (sb);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
return 0 ;
}
int XS_TEST_MAIN ()
{
fprintf (stderr, "hwm test running...\n");
void *ctx = xs_init ();
assert (ctx);
// Create pair of socket, each with high watermark of 2. Thus the total
// buffer space should be 4 messages.
void *sb = xs_socket (ctx, XS_PULL);
assert (sb);
int hwm = 2;
int rc = xs_setsockopt (sb, XS_RCVHWM, &hwm, sizeof (hwm));
assert (rc == 0);
rc = xs_bind (sb, "inproc://a");
assert (rc != -1);
void *sc = xs_socket (ctx, XS_PUSH);
assert (sc);
rc = xs_setsockopt (sc, XS_SNDHWM, &hwm, sizeof (hwm));
assert (rc == 0);
rc = xs_connect (sc, "inproc://a");
assert (rc != -1);
// Try to send 10 messages. Only 4 should succeed.
for (int i = 0; i < 10; i++)
{
rc = xs_send (sc, NULL, 0, XS_DONTWAIT);
if (i < 4)
assert (rc == 0);
else
assert (rc < 0 && errno == EAGAIN);
}
// There should be now 4 messages pending, consume them.
for (int i = 0; i != 4; i++) {
rc = xs_recv (sb, NULL, 0, 0);
assert (rc == 0);
}
// Now it should be possible to send one more.
rc = xs_send (sc, NULL, 0, 0);
assert (rc == 0);
// Consume the remaining message.
rc = xs_recv (sb, NULL, 0, 0);
assert (rc == 0);
rc = xs_close (sc);
assert (rc == 0);
rc = xs_close (sb);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
return 0;
}
int XS_TEST_MAIN ()
{
fprintf (stderr, "msg_flags test running...\n");
// Create the infrastructure
void *ctx = xs_init ();
assert (ctx);
void *sb = xs_socket (ctx, XS_XREP);
assert (sb);
int rc = xs_bind (sb, "inproc://a");
assert (rc == 0);
void *sc = xs_socket (ctx, XS_XREQ);
assert (sc);
rc = xs_connect (sc, "inproc://a");
assert (rc == 0);
// Send 2-part message.
rc = xs_send (sc, "A", 1, XS_SNDMORE);
assert (rc == 1);
rc = xs_send (sc, "B", 1, 0);
assert (rc == 1);
// Identity comes first.
xs_msg_t msg;
rc = xs_msg_init (&msg);
assert (rc == 0);
rc = xs_recvmsg (sb, &msg, 0);
assert (rc >= 0);
int more;
size_t more_size = sizeof (more);
rc = xs_getmsgopt (&msg, XS_MORE, &more, &more_size);
assert (rc == 0);
assert (more == 1);
// Then the first part of the message body.
rc = xs_recvmsg (sb, &msg, 0);
assert (rc == 1);
more_size = sizeof (more);
rc = xs_getmsgopt (&msg, XS_MORE, &more, &more_size);
assert (rc == 0);
assert (more == 1);
// And finally, the second part of the message body.
rc = xs_recvmsg (sb, &msg, 0);
assert (rc == 1);
more_size = sizeof (more);
rc = xs_getmsgopt (&msg, XS_MORE, &more, &more_size);
assert (rc == 0);
assert (more == 0);
// Deallocate the infrastructure.
rc = xs_close (sc);
assert (rc == 0);
rc = xs_close (sb);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
return 0 ;
}
static void *worker (void *ctx_)
#endif
{
void *s;
int rc;
int i;
xs_msg_t msg;
s = xs_socket (ctx_, XS_REP);
if (!s) {
printf ("error in xs_socket: %s\n", xs_strerror (errno));
exit (1);
}
rc = xs_connect (s, "inproc://lat_test");
if (rc == -1) {
printf ("error in xs_connect: %s\n", xs_strerror (errno));
exit (1);
}
rc = xs_msg_init (&msg);
if (rc != 0) {
printf ("error in xs_msg_init: %s\n", xs_strerror (errno));
exit (1);
}
for (i = 0; i != roundtrip_count; i++) {
rc = xs_recvmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_recvmsg: %s\n", xs_strerror (errno));
exit (1);
}
rc = xs_sendmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_sendmsg: %s\n", xs_strerror (errno));
exit (1);
}
}
rc = xs_msg_close (&msg);
if (rc != 0) {
printf ("error in xs_msg_close: %s\n", xs_strerror (errno));
exit (1);
}
rc = xs_close (s);
if (rc != 0) {
printf ("error in xs_close: %s\n", xs_strerror (errno));
exit (1);
}
#if defined XS_HAVE_WINDOWS
return 0;
#else
return NULL;
#endif
}
static void *worker (void *ctx_)
#endif
{
void *s;
int rc;
int i;
xs_msg_t msg;
s = xs_socket (ctx_, XS_PUSH);
if (!s) {
printf ("error in xs_socket: %s\n", xs_strerror (errno));
exit (1);
}
rc = xs_connect (s, "inproc://thr_test");
if (rc == -1) {
printf ("error in xs_connect: %s\n", xs_strerror (errno));
exit (1);
}
for (i = 0; i != message_count; i++) {
rc = xs_msg_init_size (&msg, message_size);
if (rc != 0) {
printf ("error in xs_msg_init_size: %s\n", xs_strerror (errno));
exit (1);
}
#if defined XS_MAKE_VALGRIND_HAPPY
memset (xs_msg_data (&msg), 0, message_size);
#endif
rc = xs_sendmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_sendmsg: %s\n", xs_strerror (errno));
exit (1);
}
rc = xs_msg_close (&msg);
if (rc != 0) {
printf ("error in xs_msg_close: %s\n", xs_strerror (errno));
exit (1);
}
}
rc = xs_close (s);
if (rc != 0) {
printf ("error in xs_close: %s\n", xs_strerror (errno));
exit (1);
}
#if defined XS_HAVE_WINDOWS
return 0;
#else
return NULL;
#endif
}
static void shutdown_stress_worker (void *s_)
{
int rc;
rc = xs_connect (s_, "tcp://127.0.0.1:5560");
assert (rc == 0);
// Start closing the socket while the connecting process is underway.
rc = xs_close (s_);
assert (rc == 0);
}
void polltimeo_worker(void *ctx_)
{
// Worker thread connects after delay of 1 second. Then it waits
// for 1 more second, so that async connect has time to succeed.
sleep (1);
void *sc = xs_socket (ctx_, XS_PUSH);
assert (sc);
int rc = xs_connect (sc, "inproc://timeout_test");
assert (rc == 0);
sleep (1);
rc = xs_close (sc);
assert (rc == 0);
}
int XS_TEST_MAIN ()
{
fprintf (stderr, "linger test running...\n");
// Create socket.
void *ctx = xs_init ();
assert (ctx);
void *s = xs_socket (ctx, XS_PUSH);
assert (s);
// Set linger to 0.1 second.
int linger = 100;
int rc = xs_setsockopt (s, XS_LINGER, &linger, sizeof (int));
// Connect to non-existent endpoing.
assert (rc == 0);
rc = xs_connect (s, "tcp://127.0.0.1:5560");
assert (rc == 0);
// Send a message.
rc = xs_send (s, "r", 1, 0);
assert (rc == 1);
// Close the socket.
rc = xs_close (s);
assert (rc == 0);
// Terminate the context. This should take 0.1 second.
void *watch = xs_stopwatch_start ();
rc = xs_term (ctx);
assert (rc == 0);
int ms = (int) xs_stopwatch_stop (watch) / 1000;
#if !defined _WIN32 || !defined _DEBUG
assert (ms > 50 && ms < 150);
#endif
return 0;
}
int zmq_connect (void *s, const char *addr)
{
int rc = xs_connect (s, addr);
return rc < 0 ? -1 : 0;
}
int XS_TEST_MAIN ()
{
fprintf (stderr, "sub_forward test running...\n");
void *ctx = xs_init ();
assert (ctx);
// First, create an intermediate device.
void *xpub = xs_socket (ctx, XS_XPUB);
assert (xpub);
int rc = xs_bind (xpub, "tcp://127.0.0.1:5560");
assert (rc == 0);
void *xsub = xs_socket (ctx, XS_XSUB);
assert (xsub);
rc = xs_bind (xsub, "tcp://127.0.0.1:5561");
assert (rc == 0);
// Create a publisher.
void *pub = xs_socket (ctx, XS_PUB);
assert (pub);
rc = xs_connect (pub, "tcp://127.0.0.1:5561");
assert (rc == 0);
// Create a subscriber.
void *sub = xs_socket (ctx, XS_SUB);
assert (sub);
rc = xs_connect (sub, "tcp://127.0.0.1:5560");
assert (rc == 0);
// Subscribe for all messages.
rc = xs_setsockopt (sub, XS_SUBSCRIBE, "", 0);
assert (rc == 0);
// Pass the subscription upstream through the device.
char buff [32];
rc = xs_recv (xpub, buff, sizeof (buff), 0);
assert (rc >= 0);
rc = xs_send (xsub, buff, rc, 0);
assert (rc >= 0);
// Wait a bit till the subscription gets to the publisher.
sleep (1);
// Send an empty message.
rc = xs_send (pub, NULL, 0, 0);
assert (rc == 0);
// Pass the message downstream through the device.
rc = xs_recv (xsub, buff, sizeof (buff), 0);
assert (rc >= 0);
rc = xs_send (xpub, buff, rc, 0);
assert (rc >= 0);
// Receive the message in the subscriber.
rc = xs_recv (sub, buff, sizeof (buff), 0);
assert (rc == 0);
// Clean up.
rc = xs_close (xpub);
assert (rc == 0);
rc = xs_close (xsub);
assert (rc == 0);
rc = xs_close (pub);
assert (rc == 0);
rc = xs_close (sub);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
return 0 ;
}
int XS_TEST_MAIN ()
{
fprintf (stderr, "reqrep_device test running...\n");
void *ctx = xs_init ();
errno_assert (ctx);
// Create a req/rep device.
void *xreq = xs_socket (ctx, XS_XREQ);
errno_assert (xreq);
int rc = xs_bind (xreq, "tcp://127.0.0.1:5560");
errno_assert (rc != -1);
void *xrep = xs_socket (ctx, XS_XREP);
errno_assert (xrep);
rc = xs_bind (xrep, "tcp://127.0.0.1:5561");
errno_assert (rc != -1);
// Create a worker.
void *rep = xs_socket (ctx, XS_REP);
errno_assert (rep);
rc = xs_connect (rep, "tcp://127.0.0.1:5560");
errno_assert (rc != -1);
// Create a client.
void *req = xs_socket (ctx, XS_REQ);
errno_assert (req);
rc = xs_connect (req, "tcp://127.0.0.1:5561");
errno_assert (rc != -1);
// Send a request.
rc = xs_send (req, "ABC", 3, XS_SNDMORE);
errno_assert (rc == 3);
rc = xs_send (req, "DEF", 3, 0);
errno_assert (rc == 3);
// Pass the request through the device.
for (int i = 0; i != 4; i++) {
xs_msg_t msg;
rc = xs_msg_init (&msg);
errno_assert (rc == 0);
rc = xs_recvmsg (xrep, &msg, 0);
errno_assert (rc >= 0);
int rcvmore;
size_t sz = sizeof (rcvmore);
rc = xs_getsockopt (xrep, XS_RCVMORE, &rcvmore, &sz);
errno_assert (rc == 0);
rc = xs_sendmsg (xreq, &msg, rcvmore ? XS_SNDMORE : 0);
errno_assert (rc >= 0);
}
// Receive the request.
char buff [3];
rc = xs_recv (rep, buff, 3, 0);
errno_assert (rc == 3);
assert (memcmp (buff, "ABC", 3) == 0);
int rcvmore;
size_t sz = sizeof (rcvmore);
rc = xs_getsockopt (rep, XS_RCVMORE, &rcvmore, &sz);
errno_assert (rc == 0);
assert (rcvmore);
rc = xs_recv (rep, buff, 3, 0);
errno_assert (rc == 3);
assert (memcmp (buff, "DEF", 3) == 0);
rc = xs_getsockopt (rep, XS_RCVMORE, &rcvmore, &sz);
errno_assert (rc == 0);
assert (!rcvmore);
// Send the reply.
rc = xs_send (rep, "GHI", 3, XS_SNDMORE);
errno_assert (rc == 3);
rc = xs_send (rep, "JKL", 3, 0);
errno_assert (rc == 3);
// Pass the reply through the device.
for (int i = 0; i != 4; i++) {
xs_msg_t msg;
rc = xs_msg_init (&msg);
errno_assert (rc == 0);
rc = xs_recvmsg (xreq, &msg, 0);
errno_assert (rc >= 0);
rc = xs_getsockopt (xreq, XS_RCVMORE, &rcvmore, &sz);
errno_assert (rc == 0);
rc = xs_sendmsg (xrep, &msg, rcvmore ? XS_SNDMORE : 0);
errno_assert (rc >= 0);
}
// Receive the reply.
rc = xs_recv (req, buff, 3, 0);
errno_assert (rc == 3);
assert (memcmp (buff, "GHI", 3) == 0);
rc = xs_getsockopt (req, XS_RCVMORE, &rcvmore, &sz);
errno_assert (rc == 0);
assert (rcvmore);
rc = xs_recv (req, buff, 3, 0);
errno_assert (rc == 3);
assert (memcmp (buff, "JKL", 3) == 0);
rc = xs_getsockopt (req, XS_RCVMORE, &rcvmore, &sz);
errno_assert (rc == 0);
assert (!rcvmore);
// Clean up.
rc = xs_close (req);
errno_assert (rc == 0);
rc = xs_close (rep);
errno_assert (rc == 0);
rc = xs_close (xrep);
errno_assert (rc == 0);
rc = xs_close (xreq);
errno_assert (rc == 0);
rc = xs_term (ctx);
errno_assert (rc == 0);
return 0 ;
}
int XS_TEST_MAIN ()
{
int rc;
char buf [32];
fprintf (stderr, "survey test running...\n");
// Create the basic infrastructure.
void *ctx = xs_init ();
assert (ctx);
void *xsurveyor = xs_socket (ctx, XS_XSURVEYOR);
assert (xsurveyor);
rc = xs_bind (xsurveyor, "inproc://a");
assert (rc != -1);
void *xrespondent = xs_socket (ctx, XS_XRESPONDENT);
assert (xrespondent);
rc = xs_bind (xrespondent, "inproc://b");
assert (rc != -1);
void *surveyor = xs_socket (ctx, XS_SURVEYOR);
assert (surveyor);
rc = xs_connect (surveyor, "inproc://b");
assert (rc != -1);
void *respondent1 = xs_socket (ctx, XS_RESPONDENT);
assert (respondent1);
rc = xs_connect (respondent1, "inproc://a");
assert (rc != -1);
void *respondent2 = xs_socket (ctx, XS_RESPONDENT);
assert (respondent2);
rc = xs_connect (respondent2, "inproc://a");
assert (rc != -1);
// Send the survey.
rc = xs_send (surveyor, "ABC", 3, 0);
assert (rc == 3);
// Forward the survey through the intermediate device.
// Survey consist of identity (4 bytes), survey ID (4 bytes) and the body.
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xsurveyor, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xsurveyor, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 3);
rc = xs_send (xsurveyor, buf, 3, 0);
assert (rc == 3);
// Respondent 1 responds to the survey.
rc = xs_recv (respondent1, buf, sizeof (buf), 0);
assert (rc == 3);
rc = xs_send (respondent1, "DE", 2, 0);
assert (rc == 2);
// Forward the response through the intermediate device.
rc = xs_recv (xsurveyor, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xsurveyor, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xsurveyor, buf, sizeof (buf), 0);
assert (rc == 2);
rc = xs_send (xrespondent, buf, 2, 0);
assert (rc == 2);
// Surveyor gets the response.
rc = xs_recv (surveyor, buf, sizeof (buf), 0);
assert (rc == 2);
// Respondent 2 responds to the survey.
rc = xs_recv (respondent2, buf, sizeof (buf), 0);
assert (rc == 3);
rc = xs_send (respondent2, "FGHI", 4, 0);
assert (rc == 4);
// Forward the response through the intermediate device.
rc = xs_recv (xsurveyor, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xsurveyor, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xsurveyor, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, 0);
assert (rc == 4);
// Surveyor gets the response.
rc = xs_recv (surveyor, buf, sizeof (buf), 0);
assert (rc == 4);
// Now let's test whether survey timeout works as expected.
int timeout = 100;
rc = xs_setsockopt (surveyor, XS_SURVEY_TIMEOUT, &timeout, sizeof (int));
assert (rc == 0);
rc = xs_send (surveyor, "ABC", 3, 0);
assert (rc == 3);
void *watch = xs_stopwatch_start ();
rc = xs_recv (surveyor, buf, sizeof (buf), 0);
assert (rc == - 1 && errno == EAGAIN);
unsigned long elapsed = xs_stopwatch_stop (watch) / 1000;
time_assert (elapsed, (unsigned long) timeout);
// Test whether responses for old surveys are discarded. First,
// initiate new survey.
rc = xs_setsockopt (surveyor, XS_SURVEY_TIMEOUT, &timeout, sizeof (int));
assert (rc == 0);
rc = xs_send (surveyor, "DE", 2, 0);
assert (rc == 2);
// Read, process and reply to the old survey.
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 3);
rc = xs_send (xrespondent, buf, 3, 0);
assert (rc == 3);
// Read, process and reply to the new survey.
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 2);
rc = xs_send (xrespondent, buf, 2, 0);
assert (rc == 2);
// Get the response and check it's the response to the new survey and
// that response to the old survey was silently discarded.
rc = xs_recv (surveyor, buf, sizeof (buf), 0);
assert (rc == 2);
rc = xs_close (respondent2);
assert (rc == 0);
rc = xs_close (respondent1);
assert (rc == 0);
rc = xs_close (surveyor);
assert (rc == 0);
rc = xs_close (xrespondent);
assert (rc == 0);
rc = xs_close (xsurveyor);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
return 0 ;
}
int main (int argc, char *argv [])
{
const char *connect_to;
int roundtrip_count;
size_t message_size;
void *ctx;
void *s;
int rc;
int i;
xs_msg_t msg;
void *watch;
unsigned long elapsed;
double latency;
if (argc != 4) {
printf ("usage: remote_lat <connect-to> <message-size> "
"<roundtrip-count>\n");
return 1;
}
connect_to = argv [1];
message_size = atoi (argv [2]);
roundtrip_count = atoi (argv [3]);
ctx = xs_init ();
if (!ctx) {
printf ("error in xs_init: %s\n", xs_strerror (errno));
return -1;
}
s = xs_socket (ctx, XS_REQ);
if (!s) {
printf ("error in xs_socket: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_connect (s, connect_to);
if (rc == -1) {
printf ("error in xs_connect: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_msg_init_size (&msg, message_size);
if (rc != 0) {
printf ("error in xs_msg_init_size: %s\n", xs_strerror (errno));
return -1;
}
memset (xs_msg_data (&msg), 0, message_size);
watch = xs_stopwatch_start ();
for (i = 0; i != roundtrip_count; i++) {
rc = xs_sendmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_sendmsg: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_recvmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_recvmsg: %s\n", xs_strerror (errno));
return -1;
}
if (xs_msg_size (&msg) != message_size) {
printf ("message of incorrect size received\n");
return -1;
}
}
elapsed = xs_stopwatch_stop (watch);
rc = xs_msg_close (&msg);
if (rc != 0) {
printf ("error in xs_msg_close: %s\n", xs_strerror (errno));
return -1;
}
latency = (double) elapsed / (roundtrip_count * 2);
printf ("message size: %d [B]\n", (int) message_size);
printf ("roundtrip count: %d\n", (int) roundtrip_count);
printf ("average latency: %.3f [us]\n", (double) latency);
rc = xs_close (s);
if (rc != 0) {
printf ("error in xs_close: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_term (ctx);
if (rc != 0) {
printf ("error in xs_term: %s\n", xs_strerror (errno));
return -1;
}
return 0;
}
