From 4e6a1e5508151d308a53124e4733ae1c1f7b3fb2 Mon Sep 17 00:00:00 2001 From: Martin Sustrik Date: Thu, 3 May 2012 16:19:44 +0200 Subject: Test for backlog exhaustion added Signed-off-by: Martin Sustrik --- builds/msvc/tests/tests.vcxproj | 4 + builds/msvc/tests/tests.vcxproj.filters | 3 + tests/Makefile.am | 4 +- tests/backlog.cpp | 134 ++++++++++++++++++++++++++++++++ tests/tests.cpp | 6 ++ 5 files changed, 150 insertions(+), 1 deletion(-) create mode 100644 tests/backlog.cpp diff --git a/builds/msvc/tests/tests.vcxproj b/builds/msvc/tests/tests.vcxproj index 20d2554..5c5df84 100644 --- a/builds/msvc/tests/tests.vcxproj +++ b/builds/msvc/tests/tests.vcxproj @@ -83,6 +83,10 @@ + + true + true + true true diff --git a/builds/msvc/tests/tests.vcxproj.filters b/builds/msvc/tests/tests.vcxproj.filters index ea97fed..030aed7 100644 --- a/builds/msvc/tests/tests.vcxproj.filters +++ b/builds/msvc/tests/tests.vcxproj.filters @@ -71,6 +71,9 @@ Header Files + + Header Files + diff --git a/tests/Makefile.am b/tests/Makefile.am index 4e08ae7..71442a2 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -25,7 +25,8 @@ noinst_PROGRAMS = pair_inproc \ libzmq21 \ resubscribe \ survey \ - shutdown + shutdown \ + backlog pair_inproc_SOURCES = pair_inproc.cpp testutil.hpp pair_tcp_SOURCES = pair_tcp.cpp testutil.hpp @@ -50,5 +51,6 @@ libzmq21_SOURCES = libzmq21.cpp resubscribe_SOURCES = resubscribe.cpp survey_SOURCES = survey.cpp shutdown_SOURCES = shutdown.cpp +backlog_SOURCES = backlog.cpp TESTS = $(noinst_PROGRAMS) diff --git a/tests/backlog.cpp b/tests/backlog.cpp new file mode 100644 index 0000000..7ae9ed8 --- /dev/null +++ b/tests/backlog.cpp @@ -0,0 +1,134 @@ +/* + Copyright (c) 2012 250bpm s.r.o. + Copyright (c) 2012 Other contributors as noted in the AUTHORS file + + This file is part of Crossroads I/O project. + + Crossroads I/O 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. + + Crossroads 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 . +*/ + +#include "testutil.hpp" + +#define BACKLOG 10 +#define PARALLEL_CONNECTS 50 + +extern "C" +{ + +#if !defined XS_HAVE_WINDOWS && !defined XS_HAVE_OPENVMS + static void ipc_worker (void *ctx_) + { + int rc; + void *push; + + push = xs_socket (ctx_, XS_PUSH); + assert (push); + rc = xs_connect (push, "ipc:///tmp/test-backlog"); + assert (rc >= 0); + rc = xs_send (push, NULL, 0, 0); + assert (rc == 0); + rc = xs_close (push); + assert (rc == 0); + } +#endif + + static void tcp_worker (void *ctx_) + { + int rc; + void *push; + + push = xs_socket (ctx_, XS_PUSH); + assert (push); + rc = xs_connect (push, "tcp://127.0.0.1:5560"); + assert (rc >= 0); + rc = xs_send (push, NULL, 0, 0); + assert (rc == 0); + rc = xs_close (push); + assert (rc == 0); + } +} + +int XS_TEST_MAIN () +{ + fprintf (stderr, "backlog test running...\n"); + + int rc; + void *ctx; + void *pull; + int i; + int backlog; + void *threads [PARALLEL_CONNECTS]; + + // Test the exhaustion on IPC backlog. +#if !defined XS_HAVE_WINDOWS && !defined XS_HAVE_OPENVMS + ctx = xs_init (); + assert (ctx); + + pull = xs_socket (ctx, XS_PULL); + assert (pull); + backlog = BACKLOG; + rc = xs_setsockopt (pull, XS_BACKLOG, &backlog, sizeof (backlog)); + assert (rc == 0); + rc = xs_bind (pull, "ipc:///tmp/test-backlog"); + assert (rc >= 0); + + + for (i = 0; i < PARALLEL_CONNECTS; i++) + threads [i] = thread_create (ipc_worker, ctx); + + for (i = 0; i < PARALLEL_CONNECTS; i++) { + rc = xs_recv (pull, NULL, 0, 0); + assert (rc == 0); + } + + rc = xs_close (pull); + assert (rc == 0); + rc = xs_term (ctx); + assert (rc == 0); + + for (i = 0; i < PARALLEL_CONNECTS; i++) + thread_join (threads [i]); +#endif + + // Test the exhaustion on TCP backlog. + + ctx = xs_init (); + assert (ctx); + + pull = xs_socket (ctx, XS_PULL); + assert (pull); + backlog = BACKLOG; + rc = xs_setsockopt (pull, XS_BACKLOG, &backlog, sizeof (backlog)); + assert (rc == 0); + rc = xs_bind (pull, "tcp://127.0.0.1:5560"); + assert (rc >= 0); + + for (i = 0; i < PARALLEL_CONNECTS; i++) + threads [i] = thread_create (tcp_worker, ctx); + + for (i = 0; i < PARALLEL_CONNECTS; i++) { + rc = xs_recv (pull, NULL, 0, 0); + assert (rc == 0); + } + + rc = xs_close (pull); + assert (rc == 0); + rc = xs_term (ctx); + assert (rc == 0); + + for (i = 0; i < PARALLEL_CONNECTS; i++) + thread_join (threads [i]); + + return 0; +} diff --git a/tests/tests.cpp b/tests/tests.cpp index 6474300..4071056 100644 --- a/tests/tests.cpp +++ b/tests/tests.cpp @@ -119,6 +119,10 @@ #include "shutdown.cpp" #undef XS_TEST_MAIN +#define XS_TEST_MAIN backlog +#include "backlog.cpp" +#undef XS_TEST_MAIN + int main () { int rc; @@ -167,6 +171,8 @@ int main () assert (rc == 0); rc = shutdown (); assert (rc == 0); + rc = backlog (); + assert (rc == 0); fprintf (stderr, "SUCCESS\n"); sleep (1); -- cgit v1.2.3