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