Branch data Line data Source code
1 : : /* GLib testing framework examples and tests
2 : : *
3 : : * Copyright (C) 2008-2010 Red Hat, Inc.
4 : : *
5 : : * SPDX-License-Identifier: LGPL-2.1-or-later
6 : : *
7 : : * This library is free software; you can redistribute it and/or
8 : : * modify it under the terms of the GNU Lesser General Public
9 : : * License as published by the Free Software Foundation; either
10 : : * version 2.1 of the License, or (at your option) any later version.
11 : : *
12 : : * This library is distributed in the hope that it will be useful,
13 : : * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 : : * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 : : * Lesser General Public License for more details.
16 : : *
17 : : * You should have received a copy of the GNU Lesser General
18 : : * Public License along with this library; if not, see <http://www.gnu.org/licenses/>.
19 : : *
20 : : * Author: David Zeuthen <davidz@redhat.com>
21 : : */
22 : :
23 : : #include <gio/gio.h>
24 : : #include <stdlib.h>
25 : : #include <unistd.h>
26 : : #include <string.h>
27 : :
28 : : #include "gdbusprivate.h"
29 : : #include "gdbus-tests.h"
30 : :
31 : : /* all tests rely on a shared mainloop */
32 : : static GMainLoop *loop = NULL;
33 : :
34 : : /* ---------------------------------------------------------------------------------------------------- */
35 : :
36 : : typedef struct {
37 : : const gchar *name;
38 : : const gchar *bug;
39 : : enum {
40 : : EXPLICITLY_FALSE = FALSE,
41 : : EXPLICITLY_TRUE = TRUE,
42 : : IMPLICITLY_TRUE
43 : : } exit_on_close;
44 : : enum {
45 : : LOCAL,
46 : : REMOTE
47 : : } who_closes;
48 : : } TestData;
49 : :
50 : : static const TestData cases[] = {
51 : : { "default", NULL, IMPLICITLY_TRUE, REMOTE },
52 : : { "true", NULL, EXPLICITLY_TRUE, REMOTE },
53 : : { "false", NULL, EXPLICITLY_FALSE, REMOTE },
54 : : { "we-close", "662100", EXPLICITLY_TRUE, LOCAL },
55 : : { NULL, NULL, 0, 0 }
56 : : };
57 : :
58 : : static gboolean
59 : 4 : quit_later_cb (gpointer data G_GNUC_UNUSED)
60 : : {
61 : 4 : g_main_loop_quit (loop);
62 : :
63 : 4 : return G_SOURCE_REMOVE;
64 : : }
65 : :
66 : : static void
67 : 2 : closed_cb (GDBusConnection *c G_GNUC_UNUSED,
68 : : gboolean remote_peer_vanished,
69 : : GError *error,
70 : : gpointer test_data)
71 : : {
72 : 2 : const TestData *td = test_data;
73 : :
74 : 2 : if (error == NULL)
75 : 1 : g_debug ("closed (%d, no error)", remote_peer_vanished);
76 : : else
77 : 1 : g_debug ("closed (%d, %s %d \"%s\")", remote_peer_vanished,
78 : : g_quark_to_string (error->domain), error->code, error->message);
79 : :
80 : 2 : g_assert_cmpint (remote_peer_vanished, ==, (td->who_closes == REMOTE));
81 : 2 : g_assert_cmpint ((error == NULL), ==, (td->who_closes == LOCAL));
82 : :
83 : : /* we delay this so that if exit-on-close was going to happen, it will
84 : : * win the race
85 : : */
86 : 2 : g_timeout_add (50, quit_later_cb, NULL);
87 : 2 : }
88 : :
89 : : static void
90 : 1 : close_async_cb (GObject *source G_GNUC_UNUSED,
91 : : GAsyncResult *res G_GNUC_UNUSED,
92 : : gpointer nil G_GNUC_UNUSED)
93 : : {
94 : 1 : GError *error = NULL;
95 : :
96 : 1 : if (g_dbus_connection_close_finish (G_DBUS_CONNECTION (source),
97 : : res,
98 : : &error))
99 : : {
100 : 1 : g_debug ("closed connection");
101 : : }
102 : : else
103 : : {
104 : 0 : g_warning ("failed to close connection: %s (%s #%d)",
105 : : error->message, g_quark_to_string (error->domain),
106 : : error->code);
107 : : }
108 : 1 : }
109 : :
110 : : static void
111 : 2 : test_exit_on_close_subprocess (gconstpointer test_data)
112 : : {
113 : 2 : const TestData *td = test_data;
114 : : GDBusConnection *c;
115 : :
116 : 2 : loop = g_main_loop_new (NULL, FALSE);
117 : :
118 : 2 : session_bus_up ();
119 : 2 : c = g_bus_get_sync (G_BUS_TYPE_SESSION, NULL, NULL);
120 : :
121 : 2 : g_assert (c != NULL);
122 : :
123 : : /* the default is meant to be TRUE */
124 : 2 : if (td->exit_on_close != IMPLICITLY_TRUE)
125 : 2 : g_dbus_connection_set_exit_on_close (c, td->exit_on_close);
126 : :
127 : 2 : g_assert_cmpint (g_dbus_connection_get_exit_on_close (c), ==,
128 : : (td->exit_on_close != EXPLICITLY_FALSE));
129 : 2 : g_assert (!g_dbus_connection_is_closed (c));
130 : :
131 : 2 : g_timeout_add (50, quit_later_cb, NULL);
132 : 2 : g_main_loop_run (loop);
133 : :
134 : 2 : g_signal_connect (c, "closed", G_CALLBACK (closed_cb), (gpointer) td);
135 : :
136 : 2 : if (td->who_closes == LOCAL)
137 : : {
138 : : GVariant *v;
139 : 1 : GError *error = NULL;
140 : :
141 : 1 : v = g_dbus_connection_call_sync (c, DBUS_SERVICE_DBUS,
142 : : DBUS_PATH_DBUS,
143 : : DBUS_INTERFACE_DBUS,
144 : : "ListNames",
145 : : NULL,
146 : : G_VARIANT_TYPE ("(as)"),
147 : : G_DBUS_CALL_FLAGS_NONE,
148 : : -1,
149 : : NULL,
150 : : &error);
151 : 1 : g_assert_no_error (error);
152 : 1 : g_assert (v != NULL);
153 : 1 : g_variant_unref (v);
154 : :
155 : 1 : g_dbus_connection_close (c, NULL, close_async_cb, NULL);
156 : : }
157 : : else
158 : : {
159 : 1 : session_bus_stop ();
160 : : }
161 : :
162 : 2 : g_main_loop_run (loop);
163 : : /* this is only reached when we turn off exit-on-close */
164 : 2 : g_main_loop_unref (loop);
165 : 2 : g_object_unref (c);
166 : :
167 : 2 : session_bus_down ();
168 : :
169 : 2 : exit (0);
170 : : }
171 : :
172 : : static void
173 : 4 : test_exit_on_close (gconstpointer test_data)
174 : : {
175 : 4 : const TestData *td = test_data;
176 : : GTestSubprocessFlags flags;
177 : : char *child_name;
178 : :
179 : 4 : g_test_dbus_unset ();
180 : :
181 : 4 : if (g_test_verbose ())
182 : 0 : flags = G_TEST_SUBPROCESS_INHERIT_STDOUT | G_TEST_SUBPROCESS_INHERIT_STDERR;
183 : : else
184 : 4 : flags = 0;
185 : :
186 : 4 : child_name = g_strdup_printf ("/gdbus/exit-on-close/%s/subprocess", td->name);
187 : 4 : g_test_trap_subprocess (child_name, 0, flags);
188 : 4 : g_free (child_name);
189 : :
190 : 4 : if (td->exit_on_close == EXPLICITLY_FALSE ||
191 : 3 : td->who_closes == LOCAL)
192 : 2 : g_test_trap_assert_passed ();
193 : : else
194 : 2 : g_test_trap_assert_failed();
195 : 4 : }
196 : :
197 : : /* ---------------------------------------------------------------------------------------------------- */
198 : :
199 : : int
200 : 3 : main (int argc,
201 : : char *argv[])
202 : : {
203 : : gint i;
204 : :
205 : 3 : g_test_init (&argc, &argv, G_TEST_OPTION_ISOLATE_DIRS, NULL);
206 : :
207 : 15 : for (i = 0; cases[i].name != NULL; i++)
208 : : {
209 : : gchar *name;
210 : :
211 : 12 : name = g_strdup_printf ("/gdbus/exit-on-close/%s", cases[i].name);
212 : 12 : g_test_add_data_func (name, &cases[i], test_exit_on_close);
213 : 12 : g_free (name);
214 : :
215 : 12 : name = g_strdup_printf ("/gdbus/exit-on-close/%s/subprocess", cases[i].name);
216 : 12 : g_test_add_data_func (name, &cases[i], test_exit_on_close_subprocess);
217 : 12 : g_free (name);
218 : : }
219 : :
220 : 3 : return g_test_run();
221 : : }
|
