1 #include <config.h>
2
3 #include "test-utils.h"
4
5 static DBusLoop *loop;
6 static dbus_bool_t already_quit = FALSE;
7 static const char* echo_path = "/org/freedesktop/TestSuite";
8
9 typedef struct
10 {
11 int argc;
12 char **argv;
13 } EchoData;
14
15 static void
quit(void)16 quit (void)
17 {
18 if (!already_quit)
19 {
20 _dbus_loop_quit (loop);
21 already_quit = TRUE;
22 }
23 }
24
25 static void
die(const char * message)26 die (const char *message)
27 {
28 fprintf (stderr, "*** test-service: %s", message);
29 exit (1);
30 }
31
32 static DBusHandlerResult
handle_echo(DBusConnection * connection,DBusMessage * message)33 handle_echo (DBusConnection *connection,
34 DBusMessage *message)
35 {
36 DBusError error;
37 DBusMessage *reply;
38 DBusMessageIter iter;
39 int i;
40 EchoData *d;
41
42 _dbus_verbose ("sending reply to Echo method\n");
43
44 if (!dbus_connection_get_object_path_data (connection, echo_path, (void **)&d))
45 die ("No memory");
46
47
48 dbus_error_init (&error);
49
50 reply = dbus_message_new_method_return (message);
51 if (reply == NULL)
52 die ("No memory\n");
53
54 dbus_message_iter_init_append (reply, &iter);
55 for (i = 0; i < d->argc; ++i)
56 if (!dbus_message_iter_append_basic (&iter, DBUS_TYPE_STRING, &(d->argv[i])))
57 die ("No memory\n");
58
59 if (!dbus_connection_send (connection, reply, NULL))
60 die ("No memory\n");
61
62 fprintf (stderr, "Shell echo service echoed the command line\n");
63
64 dbus_message_unref (reply);
65
66 return DBUS_HANDLER_RESULT_HANDLED;
67 }
68
69 static void
path_unregistered_func(DBusConnection * connection,void * user_data)70 path_unregistered_func (DBusConnection *connection,
71 void *user_data)
72 {
73 /* connection was finalized */
74 }
75
76 static DBusHandlerResult
path_message_func(DBusConnection * connection,DBusMessage * message,void * user_data)77 path_message_func (DBusConnection *connection,
78 DBusMessage *message,
79 void *user_data)
80 {
81 if (dbus_message_is_method_call (message,
82 "org.freedesktop.TestSuite",
83 "Echo"))
84 return handle_echo (connection, message);
85 else if (dbus_message_is_method_call (message,
86 "org.freedesktop.TestSuite",
87 "Exit"))
88 {
89 quit ();
90 return DBUS_HANDLER_RESULT_HANDLED;
91 }
92 else
93 return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
94 }
95
96 static DBusObjectPathVTable
97 echo_vtable = {
98 path_unregistered_func,
99 path_message_func,
100 NULL,
101 };
102
103 static DBusHandlerResult
filter_func(DBusConnection * connection,DBusMessage * message,void * user_data)104 filter_func (DBusConnection *connection,
105 DBusMessage *message,
106 void *user_data)
107 {
108 if (dbus_message_is_signal (message,
109 DBUS_INTERFACE_LOCAL,
110 "Disconnected"))
111 {
112 quit ();
113 return DBUS_HANDLER_RESULT_HANDLED;
114 }
115 else
116 {
117 return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
118 }
119 }
120
121 int
main(int argc,char ** argv)122 main (int argc,
123 char **argv)
124 {
125 DBusConnection *connection;
126 DBusError error;
127 EchoData echo_data;
128 int result;
129
130 echo_data.argc = argc;
131 echo_data.argv = argv;
132
133 dbus_error_init (&error);
134 connection = dbus_bus_get (DBUS_BUS_STARTER, &error);
135 if (connection == NULL)
136 {
137 fprintf (stderr, "*** Failed to open connection to activating message bus: %s\n",
138 error.message);
139 dbus_error_free (&error);
140 return 1;
141 }
142
143 loop = _dbus_loop_new ();
144 if (loop == NULL)
145 die ("No memory\n");
146
147 if (!test_connection_setup (loop, connection))
148 die ("No memory\n");
149
150 if (!dbus_connection_add_filter (connection,
151 filter_func, NULL, NULL))
152 die ("No memory");
153
154 if (!dbus_connection_register_object_path (connection,
155 echo_path,
156 &echo_vtable,
157 (void*) &echo_data))
158 die ("No memory");
159
160 {
161 void *d;
162 if (!dbus_connection_get_object_path_data (connection, echo_path, &d))
163 die ("No memory");
164 if (d != (void*) &echo_data)
165 die ("dbus_connection_get_object_path_data() doesn't seem to work right\n");
166 }
167
168 result = dbus_bus_request_name (connection, "org.freedesktop.DBus.TestSuiteShellEchoServiceSuccess",
169 0, &error);
170 if (dbus_error_is_set (&error))
171 {
172 fprintf (stderr, "Error %s\n", error.message);
173 _dbus_verbose ("*** Failed to acquire service: %s\n",
174 error.message);
175 dbus_error_free (&error);
176 exit (1);
177 }
178
179 if (result != DBUS_REQUEST_NAME_REPLY_PRIMARY_OWNER)
180 {
181 fprintf (stderr, "Unable to acquire service: code %d\n", result);
182 _dbus_verbose ("*** Failed to acquire service: %d\n", result);
183 exit (1);
184 }
185
186 _dbus_verbose ("*** Test service entering main loop\n");
187 _dbus_loop_run (loop);
188
189 test_connection_shutdown (loop, connection);
190
191 dbus_connection_remove_filter (connection, filter_func, NULL);
192
193 dbus_connection_unref (connection);
194
195 _dbus_loop_unref (loop);
196 loop = NULL;
197
198 dbus_shutdown ();
199
200 _dbus_verbose ("*** Test service exiting\n");
201
202 return 0;
203 }
204