2012-05-04 03:32:46 +02:00
|
|
|
/*
|
2017-05-01 13:11:11 +02:00
|
|
|
Copyright (c) 2007-2017 Contributors as noted in the AUTHORS file
|
2012-05-04 03:32:46 +02:00
|
|
|
|
2015-06-02 22:33:55 +02:00
|
|
|
This file is part of libzmq, the ZeroMQ core engine in C++.
|
2012-05-04 03:32:46 +02:00
|
|
|
|
2015-06-02 22:33:55 +02:00
|
|
|
libzmq is free software; you can redistribute it and/or modify it under
|
|
|
|
the terms of the GNU Lesser General Public License (LGPL) as published
|
|
|
|
by the Free Software Foundation; either version 3 of the License, or
|
2012-05-04 03:32:46 +02:00
|
|
|
(at your option) any later version.
|
|
|
|
|
2015-06-02 22:33:55 +02:00
|
|
|
As a special exception, the Contributors give you permission to link
|
|
|
|
this library with independent modules to produce an executable,
|
|
|
|
regardless of the license terms of these independent modules, and to
|
|
|
|
copy and distribute the resulting executable under terms of your choice,
|
|
|
|
provided that you also meet, for each linked independent module, the
|
|
|
|
terms and conditions of the license of that module. An independent
|
|
|
|
module is a module which is not derived from or based on this library.
|
|
|
|
If you modify this library, you must extend this exception to your
|
|
|
|
version of the library.
|
|
|
|
|
|
|
|
libzmq 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.
|
2012-05-04 03:32:46 +02:00
|
|
|
|
|
|
|
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/>.
|
|
|
|
*/
|
|
|
|
|
2012-08-23 22:40:30 +02:00
|
|
|
#include "testutil.hpp"
|
2019-02-02 11:32:27 +01:00
|
|
|
#include "testutil_monitoring.hpp"
|
2012-05-04 03:32:46 +02:00
|
|
|
|
2019-02-01 17:22:43 +01:00
|
|
|
#include "testutil_unity.hpp"
|
|
|
|
|
|
|
|
void setUp ()
|
2012-05-04 03:32:46 +02:00
|
|
|
{
|
2019-02-01 17:22:43 +01:00
|
|
|
setup_test_context ();
|
|
|
|
}
|
2012-09-21 13:53:31 +02:00
|
|
|
|
2019-02-01 17:22:43 +01:00
|
|
|
void tearDown ()
|
|
|
|
{
|
|
|
|
teardown_test_context ();
|
|
|
|
}
|
2018-02-01 11:46:09 +01:00
|
|
|
|
2019-02-01 17:22:43 +01:00
|
|
|
void test_monitor_invalid_protocol_fails ()
|
|
|
|
{
|
|
|
|
void *client = test_context_socket (ZMQ_DEALER);
|
2014-04-28 11:30:04 +02:00
|
|
|
|
|
|
|
// Socket monitoring only works over inproc://
|
2019-02-01 17:22:43 +01:00
|
|
|
TEST_ASSERT_FAILURE_ERRNO (
|
|
|
|
EPROTONOSUPPORT, zmq_socket_monitor (client, "tcp://127.0.0.1:*", 0));
|
2019-02-02 17:15:27 +01:00
|
|
|
|
|
|
|
test_context_socket_close_zero_linger (client);
|
2019-02-01 17:22:43 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void test_monitor_basic ()
|
|
|
|
{
|
|
|
|
char my_endpoint[MAX_SOCKET_STRING];
|
|
|
|
|
|
|
|
// We'll monitor these two sockets
|
|
|
|
void *client = test_context_socket (ZMQ_DEALER);
|
|
|
|
void *server = test_context_socket (ZMQ_DEALER);
|
2012-09-21 13:53:31 +02:00
|
|
|
|
2014-04-28 11:30:04 +02:00
|
|
|
// Monitor all events on client and server sockets
|
2019-02-01 17:22:43 +01:00
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (
|
|
|
|
zmq_socket_monitor (client, "inproc://monitor-client", ZMQ_EVENT_ALL));
|
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (
|
|
|
|
zmq_socket_monitor (server, "inproc://monitor-server", ZMQ_EVENT_ALL));
|
2012-05-04 03:32:46 +02:00
|
|
|
|
2014-04-28 11:30:04 +02:00
|
|
|
// Create two sockets for collecting monitor events
|
2019-02-01 17:22:43 +01:00
|
|
|
void *client_mon = test_context_socket (ZMQ_PAIR);
|
|
|
|
void *server_mon = test_context_socket (ZMQ_PAIR);
|
2013-06-22 11:58:16 +02:00
|
|
|
|
2014-04-28 11:30:04 +02:00
|
|
|
// Connect these to the inproc endpoints so they'll get events
|
2019-02-01 17:22:43 +01:00
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (
|
|
|
|
zmq_connect (client_mon, "inproc://monitor-client"));
|
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (
|
|
|
|
zmq_connect (server_mon, "inproc://monitor-server"));
|
2018-02-01 11:46:09 +01:00
|
|
|
|
2014-04-28 11:30:04 +02:00
|
|
|
// Now do a basic ping test
|
2019-02-01 17:22:43 +01:00
|
|
|
bind_loopback_ipv4 (server, my_endpoint, sizeof my_endpoint);
|
|
|
|
|
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_connect (client, my_endpoint));
|
2016-05-01 22:40:08 +02:00
|
|
|
bounce (server, client);
|
2012-09-21 13:53:31 +02:00
|
|
|
|
2014-04-28 11:30:04 +02:00
|
|
|
// Close client and server
|
2019-02-01 17:22:43 +01:00
|
|
|
// TODO why does this use zero_linger?
|
|
|
|
test_context_socket_close_zero_linger (client);
|
|
|
|
test_context_socket_close_zero_linger (server);
|
2018-02-01 11:46:09 +01:00
|
|
|
|
2014-04-28 11:30:04 +02:00
|
|
|
// Now collect and check events from both sockets
|
|
|
|
int event = get_monitor_event (client_mon, NULL, NULL);
|
|
|
|
if (event == ZMQ_EVENT_CONNECT_DELAYED)
|
|
|
|
event = get_monitor_event (client_mon, NULL, NULL);
|
|
|
|
assert (event == ZMQ_EVENT_CONNECTED);
|
2017-08-20 16:21:39 +02:00
|
|
|
expect_monitor_event (client_mon, ZMQ_EVENT_HANDSHAKE_SUCCEEDED);
|
|
|
|
expect_monitor_event (client_mon, ZMQ_EVENT_MONITOR_STOPPED);
|
2014-04-28 11:30:04 +02:00
|
|
|
|
|
|
|
// This is the flow of server events
|
2017-08-20 16:21:39 +02:00
|
|
|
expect_monitor_event (server_mon, ZMQ_EVENT_LISTENING);
|
|
|
|
expect_monitor_event (server_mon, ZMQ_EVENT_ACCEPTED);
|
|
|
|
expect_monitor_event (server_mon, ZMQ_EVENT_HANDSHAKE_SUCCEEDED);
|
2014-04-28 11:30:04 +02:00
|
|
|
event = get_monitor_event (server_mon, NULL, NULL);
|
2015-07-28 22:42:46 +02:00
|
|
|
// Sometimes the server sees the client closing before it gets closed.
|
|
|
|
if (event != ZMQ_EVENT_DISCONNECTED) {
|
2019-02-01 17:22:43 +01:00
|
|
|
TEST_ASSERT_EQUAL_INT (ZMQ_EVENT_CLOSED, event);
|
2018-02-01 11:46:09 +01:00
|
|
|
event = get_monitor_event (server_mon, NULL, NULL);
|
2015-07-28 22:42:46 +02:00
|
|
|
}
|
|
|
|
if (event != ZMQ_EVENT_DISCONNECTED) {
|
2019-02-01 17:22:43 +01:00
|
|
|
TEST_ASSERT_EQUAL_INT (ZMQ_EVENT_MONITOR_STOPPED, event);
|
2015-07-28 22:42:46 +02:00
|
|
|
}
|
2018-02-01 11:46:09 +01:00
|
|
|
|
2014-04-28 11:30:04 +02:00
|
|
|
// Close down the sockets
|
2019-02-01 17:22:43 +01:00
|
|
|
// TODO why does this use zero_linger?
|
|
|
|
test_context_socket_close_zero_linger (client_mon);
|
|
|
|
test_context_socket_close_zero_linger (server_mon);
|
|
|
|
}
|
|
|
|
|
2019-02-02 12:11:11 +01:00
|
|
|
#ifdef ZMQ_BUILD_DRAFT_API
|
|
|
|
void test_monitor_versioned_basic ()
|
|
|
|
{
|
2019-02-02 16:34:19 +01:00
|
|
|
char server_endpoint[MAX_SOCKET_STRING];
|
2019-02-02 12:11:11 +01:00
|
|
|
|
|
|
|
// We'll monitor these two sockets
|
|
|
|
void *client = test_context_socket (ZMQ_DEALER);
|
|
|
|
void *server = test_context_socket (ZMQ_DEALER);
|
|
|
|
|
|
|
|
// Monitor all events on client and server sockets
|
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_socket_monitor_versioned (
|
|
|
|
client, "inproc://monitor-client", ZMQ_EVENT_ALL_V2, 2));
|
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_socket_monitor_versioned (
|
|
|
|
server, "inproc://monitor-server", ZMQ_EVENT_ALL_V2, 2));
|
|
|
|
|
|
|
|
// Create two sockets for collecting monitor events
|
|
|
|
void *client_mon = test_context_socket (ZMQ_PAIR);
|
|
|
|
void *server_mon = test_context_socket (ZMQ_PAIR);
|
|
|
|
|
|
|
|
// Connect these to the inproc endpoints so they'll get events
|
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (
|
|
|
|
zmq_connect (client_mon, "inproc://monitor-client"));
|
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (
|
|
|
|
zmq_connect (server_mon, "inproc://monitor-server"));
|
|
|
|
|
|
|
|
// Now do a basic ping test
|
2019-02-02 16:34:19 +01:00
|
|
|
bind_loopback_ipv4 (server, server_endpoint, sizeof server_endpoint);
|
2019-02-02 12:11:11 +01:00
|
|
|
|
2019-02-02 16:34:19 +01:00
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_connect (client, server_endpoint));
|
2019-02-02 12:11:11 +01:00
|
|
|
bounce (server, client);
|
|
|
|
|
|
|
|
// Close client and server
|
|
|
|
// TODO why does this use zero_linger?
|
|
|
|
test_context_socket_close_zero_linger (client);
|
|
|
|
test_context_socket_close_zero_linger (server);
|
|
|
|
|
2019-02-02 16:34:19 +01:00
|
|
|
char *client_local_address = NULL;
|
|
|
|
char *client_remote_address = NULL;
|
|
|
|
|
2019-02-02 12:11:11 +01:00
|
|
|
// Now collect and check events from both sockets
|
2019-02-02 16:34:19 +01:00
|
|
|
int64_t event = get_monitor_event_v2 (
|
|
|
|
client_mon, NULL, &client_local_address, &client_remote_address);
|
|
|
|
if (event == ZMQ_EVENT_CONNECT_DELAYED) {
|
|
|
|
free (client_local_address);
|
|
|
|
free (client_remote_address);
|
|
|
|
event = get_monitor_event_v2 (client_mon, NULL, &client_local_address,
|
|
|
|
&client_remote_address);
|
|
|
|
}
|
|
|
|
TEST_ASSERT_EQUAL (ZMQ_EVENT_CONNECTED, event);
|
|
|
|
TEST_ASSERT_EQUAL_STRING (server_endpoint, client_remote_address);
|
|
|
|
static const char prefix[] = "tcp://127.0.0.1:";
|
|
|
|
TEST_ASSERT_EQUAL_STRING_LEN (prefix, client_local_address,
|
|
|
|
strlen (prefix));
|
|
|
|
TEST_ASSERT_NOT_EQUAL (
|
|
|
|
0, strcmp (client_local_address, client_remote_address));
|
|
|
|
|
|
|
|
expect_monitor_event_v2 (client_mon, ZMQ_EVENT_HANDSHAKE_SUCCEEDED,
|
|
|
|
client_local_address, client_remote_address);
|
2019-02-02 18:29:33 +01:00
|
|
|
expect_monitor_event_v2 (client_mon, ZMQ_EVENT_MONITOR_STOPPED, "", "");
|
2019-02-02 12:11:11 +01:00
|
|
|
|
|
|
|
// This is the flow of server events
|
2019-02-02 18:29:33 +01:00
|
|
|
expect_monitor_event_v2 (server_mon, ZMQ_EVENT_LISTENING,
|
|
|
|
client_remote_address, "");
|
2019-02-02 16:34:19 +01:00
|
|
|
expect_monitor_event_v2 (server_mon, ZMQ_EVENT_ACCEPTED,
|
|
|
|
client_remote_address, client_local_address);
|
|
|
|
expect_monitor_event_v2 (server_mon, ZMQ_EVENT_HANDSHAKE_SUCCEEDED,
|
|
|
|
client_remote_address, client_local_address);
|
2019-02-02 12:11:11 +01:00
|
|
|
event = get_monitor_event_v2 (server_mon, NULL, NULL, NULL);
|
|
|
|
// Sometimes the server sees the client closing before it gets closed.
|
|
|
|
if (event != ZMQ_EVENT_DISCONNECTED) {
|
|
|
|
TEST_ASSERT_EQUAL_INT (ZMQ_EVENT_CLOSED, event);
|
|
|
|
event = get_monitor_event_v2 (server_mon, NULL, NULL, NULL);
|
|
|
|
}
|
|
|
|
if (event != ZMQ_EVENT_DISCONNECTED) {
|
|
|
|
TEST_ASSERT_EQUAL_INT (ZMQ_EVENT_MONITOR_STOPPED, event);
|
|
|
|
}
|
2019-02-02 16:34:19 +01:00
|
|
|
free (client_local_address);
|
|
|
|
free (client_remote_address);
|
2019-02-02 12:11:11 +01:00
|
|
|
|
|
|
|
// Close down the sockets
|
|
|
|
// TODO why does this use zero_linger?
|
|
|
|
test_context_socket_close_zero_linger (client_mon);
|
|
|
|
test_context_socket_close_zero_linger (server_mon);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2019-02-01 17:22:43 +01:00
|
|
|
int main ()
|
|
|
|
{
|
|
|
|
setup_test_environment ();
|
2012-05-21 21:47:11 +02:00
|
|
|
|
2019-02-01 17:22:43 +01:00
|
|
|
UNITY_BEGIN ();
|
|
|
|
RUN_TEST (test_monitor_invalid_protocol_fails);
|
|
|
|
RUN_TEST (test_monitor_basic);
|
2019-02-02 12:11:11 +01:00
|
|
|
|
|
|
|
#ifdef ZMQ_BUILD_DRAFT_API
|
|
|
|
RUN_TEST (test_monitor_versioned_basic);
|
|
|
|
#endif
|
|
|
|
|
2019-02-01 17:22:43 +01:00
|
|
|
return UNITY_END ();
|
2012-06-05 16:22:22 +02:00
|
|
|
}
|