• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /* Simple sanity-check for loopback through TCP and Unix sockets.
2  *
3  * Author: Simon McVittie <simon.mcvittie@collabora.co.uk>
4  * Copyright © 2010-2011 Nokia Corporation
5  *
6  * Permission is hereby granted, free of charge, to any person
7  * obtaining a copy of this software and associated documentation files
8  * (the "Software"), to deal in the Software without restriction,
9  * including without limitation the rights to use, copy, modify, merge,
10  * publish, distribute, sublicense, and/or sell copies of the Software,
11  * and to permit persons to whom the Software is furnished to do so,
12  * subject to the following conditions:
13  *
14  * The above copyright notice and this permission notice shall be
15  * included in all copies or substantial portions of the Software.
16  *
17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
18  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
19  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
20  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
21  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
22  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
23  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
24  * SOFTWARE.
25  */
26 
27 #include <config.h>
28 
29 #include <glib.h>
30 
31 #include <dbus/dbus.h>
32 #include <dbus/dbus-glib-lowlevel.h>
33 
34 typedef struct {
35     DBusError e;
36 
37     DBusServer *server;
38     DBusConnection *server_conn;
39     /* queue of DBusMessage */
40     GQueue server_messages;
41 
42     DBusConnection *client_conn;
43 } Fixture;
44 
45 static void
assert_no_error(const DBusError * e)46 assert_no_error (const DBusError *e)
47 {
48   if (G_UNLIKELY (dbus_error_is_set (e)))
49     g_error ("expected success but got error: %s: %s", e->name, e->message);
50 }
51 
52 static DBusHandlerResult
server_message_cb(DBusConnection * server_conn,DBusMessage * message,void * data)53 server_message_cb (DBusConnection *server_conn,
54     DBusMessage *message,
55     void *data)
56 {
57   Fixture *f = data;
58 
59   g_assert (server_conn == f->server_conn);
60   g_queue_push_tail (&f->server_messages, dbus_message_ref (message));
61 
62   return DBUS_HANDLER_RESULT_HANDLED;
63 }
64 
65 static void
new_conn_cb(DBusServer * server,DBusConnection * server_conn,void * data)66 new_conn_cb (DBusServer *server,
67     DBusConnection *server_conn,
68     void *data)
69 {
70   Fixture *f = data;
71   dbus_bool_t have_mem;
72 
73   g_assert (f->server_conn == NULL);
74   f->server_conn = dbus_connection_ref (server_conn);
75   dbus_connection_setup_with_g_main (server_conn, NULL);
76 
77   have_mem = dbus_connection_add_filter (server_conn,
78       server_message_cb, f, NULL);
79   g_assert (have_mem);
80 }
81 
82 static void
setup(Fixture * f,gconstpointer addr)83 setup (Fixture *f,
84     gconstpointer addr)
85 {
86   dbus_error_init (&f->e);
87   g_queue_init (&f->server_messages);
88 
89   f->server = dbus_server_listen (addr, &f->e);
90   assert_no_error (&f->e);
91   g_assert (f->server != NULL);
92 
93   dbus_server_set_new_connection_function (f->server,
94       new_conn_cb, f, NULL);
95   dbus_server_setup_with_g_main (f->server, NULL);
96 }
97 
98 static void
test_connect(Fixture * f,gconstpointer addr G_GNUC_UNUSED)99 test_connect (Fixture *f,
100     gconstpointer addr G_GNUC_UNUSED)
101 {
102   g_assert (f->server_conn == NULL);
103 
104   f->client_conn = dbus_connection_open_private (
105       dbus_server_get_address (f->server), &f->e);
106   assert_no_error (&f->e);
107   g_assert (f->client_conn != NULL);
108   dbus_connection_setup_with_g_main (f->client_conn, NULL);
109 
110   while (f->server_conn == NULL)
111     {
112       g_print (".");
113       g_main_context_iteration (NULL, TRUE);
114     }
115 }
116 
117 static void
test_message(Fixture * f,gconstpointer addr)118 test_message (Fixture *f,
119     gconstpointer addr)
120 {
121   dbus_bool_t have_mem;
122   dbus_uint32_t serial;
123   DBusMessage *outgoing, *incoming;
124 
125   test_connect (f, addr);
126 
127   outgoing = dbus_message_new_signal ("/com/example/Hello",
128       "com.example.Hello", "Greeting");
129   g_assert (outgoing != NULL);
130 
131   have_mem = dbus_connection_send (f->client_conn, outgoing, &serial);
132   g_assert (have_mem);
133   g_assert (serial != 0);
134 
135   while (g_queue_is_empty (&f->server_messages))
136     {
137       g_print (".");
138       g_main_context_iteration (NULL, TRUE);
139     }
140 
141   g_assert_cmpuint (g_queue_get_length (&f->server_messages), ==, 1);
142 
143   incoming = g_queue_pop_head (&f->server_messages);
144 
145   g_assert (!dbus_message_contains_unix_fds (incoming));
146   g_assert_cmpstr (dbus_message_get_destination (incoming), ==, NULL);
147   g_assert_cmpstr (dbus_message_get_error_name (incoming), ==, NULL);
148   g_assert_cmpstr (dbus_message_get_interface (incoming), ==,
149       "com.example.Hello");
150   g_assert_cmpstr (dbus_message_get_member (incoming), ==, "Greeting");
151   g_assert_cmpstr (dbus_message_get_sender (incoming), ==, NULL);
152   g_assert_cmpstr (dbus_message_get_signature (incoming), ==, "");
153   g_assert_cmpstr (dbus_message_get_path (incoming), ==, "/com/example/Hello");
154   g_assert_cmpuint (dbus_message_get_serial (incoming), ==, serial);
155 
156   dbus_message_unref (incoming);
157 
158   dbus_message_unref (outgoing);
159 }
160 
161 static void
teardown(Fixture * f,gconstpointer addr G_GNUC_UNUSED)162 teardown (Fixture *f,
163     gconstpointer addr G_GNUC_UNUSED)
164 {
165   if (f->client_conn != NULL)
166     {
167       dbus_connection_close (f->client_conn);
168       dbus_connection_unref (f->client_conn);
169       f->client_conn = NULL;
170     }
171 
172   if (f->server_conn != NULL)
173     {
174       dbus_connection_close (f->server_conn);
175       dbus_connection_unref (f->server_conn);
176       f->server_conn = NULL;
177     }
178 
179   if (f->server != NULL)
180     {
181       dbus_server_disconnect (f->server);
182       dbus_server_unref (f->server);
183       f->server = NULL;
184     }
185 }
186 
187 int
main(int argc,char ** argv)188 main (int argc,
189     char **argv)
190 {
191   g_test_init (&argc, &argv, NULL);
192 
193   g_test_add ("/connect/tcp", Fixture, "tcp:host=127.0.0.1", setup,
194       test_connect, teardown);
195   g_test_add ("/message/tcp", Fixture, "tcp:host=127.0.0.1", setup,
196       test_message, teardown);
197 
198   g_test_add ("/connect/nonce-tcp", Fixture, "nonce-tcp:host=127.0.0.1", setup,
199       test_connect, teardown);
200   g_test_add ("/message/nonce-tcp", Fixture, "nonce-tcp:host=127.0.0.1", setup,
201       test_message, teardown);
202 
203 #ifdef DBUS_UNIX
204   g_test_add ("/connect/unix", Fixture, "unix:tmpdir=/tmp", setup,
205       test_connect, teardown);
206   g_test_add ("/message/unix", Fixture, "unix:tmpdir=/tmp", setup,
207       test_message, teardown);
208 #endif
209 
210   return g_test_run ();
211 }
212