1 | /* |
2 | Copyright (c) 2007-2016 Contributors as noted in the AUTHORS file |
3 | |
4 | This file is part of libzmq, the ZeroMQ core engine in C++. |
5 | |
6 | libzmq is free software; you can redistribute it and/or modify it under |
7 | the terms of the GNU Lesser General Public License (LGPL) as published |
8 | by the Free Software Foundation; either version 3 of the License, or |
9 | (at your option) any later version. |
10 | |
11 | As a special exception, the Contributors give you permission to link |
12 | this library with independent modules to produce an executable, |
13 | regardless of the license terms of these independent modules, and to |
14 | copy and distribute the resulting executable under terms of your choice, |
15 | provided that you also meet, for each linked independent module, the |
16 | terms and conditions of the license of that module. An independent |
17 | module is a module which is not derived from or based on this library. |
18 | If you modify this library, you must extend this exception to your |
19 | version of the library. |
20 | |
21 | libzmq is distributed in the hope that it will be useful, but WITHOUT |
22 | ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
23 | FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public |
24 | License for more details. |
25 | |
26 | You should have received a copy of the GNU Lesser General Public License |
27 | along with this program. If not, see <http://www.gnu.org/licenses/>. |
28 | */ |
29 | |
30 | #include "testutil.hpp" |
31 | #include "testutil_unity.hpp" |
32 | |
33 | #include <string.h> |
34 | |
35 | SETUP_TEARDOWN_TESTCONTEXT |
36 | |
37 | void test () |
38 | { |
39 | // Create a publisher |
40 | void *pub = test_context_socket (ZMQ_PUB); |
41 | TEST_ASSERT_SUCCESS_ERRNO (zmq_bind (pub, "inproc://soname" )); |
42 | |
43 | // Create two subscribers |
44 | void *sub1 = test_context_socket (ZMQ_SUB); |
45 | TEST_ASSERT_SUCCESS_ERRNO (zmq_connect (sub1, "inproc://soname" )); |
46 | |
47 | void *sub2 = test_context_socket (ZMQ_SUB); |
48 | TEST_ASSERT_SUCCESS_ERRNO (zmq_connect (sub2, "inproc://soname" )); |
49 | |
50 | // Subscribe pub1 to one prefix |
51 | // and pub2 to another prefix. |
52 | const char prefi_x1[] = "prefix1" ; |
53 | const char prefi_x2[] = "p2" ; |
54 | |
55 | TEST_ASSERT_SUCCESS_ERRNO ( |
56 | zmq_setsockopt (sub1, ZMQ_SUBSCRIBE, prefi_x1, strlen (prefi_x1))); |
57 | TEST_ASSERT_SUCCESS_ERRNO ( |
58 | zmq_setsockopt (sub2, ZMQ_SUBSCRIBE, prefi_x2, strlen (prefi_x2))); |
59 | |
60 | // Send a message with the first prefix |
61 | send_string_expect_success (pub, prefi_x1, 0); |
62 | msleep (SETTLE_TIME); |
63 | |
64 | // sub1 should receive it, but not sub2 |
65 | recv_string_expect_success (sub1, prefi_x1, ZMQ_DONTWAIT); |
66 | |
67 | TEST_ASSERT_FAILURE_ERRNO (EAGAIN, zmq_recv (sub2, NULL, 0, ZMQ_DONTWAIT)); |
68 | |
69 | // Send a message with the second prefix |
70 | send_string_expect_success (pub, prefi_x2, 0); |
71 | msleep (SETTLE_TIME); |
72 | |
73 | // sub2 should receive it, but not sub1 |
74 | recv_string_expect_success (sub2, prefi_x2, ZMQ_DONTWAIT); |
75 | |
76 | TEST_ASSERT_FAILURE_ERRNO (EAGAIN, zmq_recv (sub1, NULL, 0, ZMQ_DONTWAIT)); |
77 | |
78 | // Now invert the matching |
79 | int invert = 1; |
80 | TEST_ASSERT_SUCCESS_ERRNO ( |
81 | zmq_setsockopt (pub, ZMQ_INVERT_MATCHING, &invert, sizeof (invert))); |
82 | |
83 | // ... on both sides, otherwise the SUB socket will filter the messages out |
84 | TEST_ASSERT_SUCCESS_ERRNO ( |
85 | zmq_setsockopt (sub1, ZMQ_INVERT_MATCHING, &invert, sizeof (invert))); |
86 | TEST_ASSERT_SUCCESS_ERRNO ( |
87 | zmq_setsockopt (sub2, ZMQ_INVERT_MATCHING, &invert, sizeof (invert))); |
88 | |
89 | // Send a message with the first prefix |
90 | send_string_expect_success (pub, prefi_x1, 0); |
91 | msleep (SETTLE_TIME); |
92 | |
93 | // sub2 should receive it, but not sub1 |
94 | recv_string_expect_success (sub2, prefi_x1, ZMQ_DONTWAIT); |
95 | |
96 | TEST_ASSERT_FAILURE_ERRNO (EAGAIN, zmq_recv (sub1, NULL, 0, ZMQ_DONTWAIT)); |
97 | |
98 | // Send a message with the second prefix |
99 | send_string_expect_success (pub, prefi_x2, 0); |
100 | msleep (SETTLE_TIME); |
101 | |
102 | // sub1 should receive it, but not sub2 |
103 | recv_string_expect_success (sub1, prefi_x2, ZMQ_DONTWAIT); |
104 | |
105 | TEST_ASSERT_FAILURE_ERRNO (EAGAIN, zmq_recv (sub2, NULL, 0, ZMQ_DONTWAIT)); |
106 | |
107 | // Clean up. |
108 | test_context_socket_close (pub); |
109 | test_context_socket_close (sub1); |
110 | test_context_socket_close (sub2); |
111 | } |
112 | |
113 | int main () |
114 | { |
115 | setup_test_environment (); |
116 | |
117 | UNITY_BEGIN (); |
118 | RUN_TEST (test); |
119 | return UNITY_END (); |
120 | } |
121 | |