Project description.
[connman] / gdbus / object.c
1 /*
2  *
3  *  D-Bus helper library
4  *
5  *  Copyright (C) 2004-2009  Marcel Holtmann <marcel@holtmann.org>
6  *
7  *
8  *  This program is free software; you can redistribute it and/or modify
9  *  it under the terms of the GNU General Public License as published by
10  *  the Free Software Foundation; either version 2 of the License, or
11  *  (at your option) any later version.
12  *
13  *  This program is distributed in the hope that it will be useful,
14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  *  GNU General Public License for more details.
17  *
18  *  You should have received a copy of the GNU General Public License
19  *  along with this program; if not, write to the Free Software
20  *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
21  *
22  */
23
24 #ifdef HAVE_CONFIG_H
25 #include <config.h>
26 #endif
27
28 #include <stdio.h>
29 #include <string.h>
30
31 #include <glib.h>
32 #include <dbus/dbus.h>
33
34 #include "gdbus.h"
35
36 #define info(fmt...)
37 #define error(fmt...)
38 #define debug(fmt...)
39
40 struct generic_data {
41         unsigned int refcount;
42         GSList *interfaces;
43         char *introspect;
44 };
45
46 struct interface_data {
47         char *name;
48         GDBusMethodTable *methods;
49         GDBusSignalTable *signals;
50         GDBusPropertyTable *properties;
51         void *user_data;
52         GDBusDestroyFunction destroy;
53 };
54
55 static void print_arguments(GString *gstr, const char *sig,
56                                                 const char *direction)
57 {
58         int i;
59
60         for (i = 0; sig[i]; i++) {
61                 char type[32];
62                 int struct_level, dict_level;
63                 unsigned int len;
64                 gboolean complete;
65
66                 complete = FALSE;
67                 struct_level = dict_level = 0;
68                 memset(type, 0, sizeof(type));
69
70                 /* Gather enough data to have a single complete type */
71                 for (len = 0; len < (sizeof(type) - 1) && sig[i]; len++, i++) {
72                         switch (sig[i]){
73                         case '(':
74                                 struct_level++;
75                                 break;
76                         case ')':
77                                 struct_level--;
78                                 if (struct_level <= 0 && dict_level <= 0)
79                                         complete = TRUE;
80                                 break;
81                         case '{':
82                                 dict_level++;
83                                 break;
84                         case '}':
85                                 dict_level--;
86                                 if (struct_level <= 0 && dict_level <= 0)
87                                         complete = TRUE;
88                                 break;
89                         case 'a':
90                                 break;
91                         default:
92                                 if (struct_level <= 0 && dict_level <= 0)
93                                         complete = TRUE;
94                                 break;
95                         }
96
97                         type[len] = sig[i];
98
99                         if (complete)
100                                 break;
101                 }
102
103
104                 if (direction)
105                         g_string_append_printf(gstr,
106                                         "\t\t\t<arg type=\"%s\" direction=\"%s\"/>\n",
107                                         type, direction);
108                 else
109                         g_string_append_printf(gstr,
110                                         "\t\t\t<arg type=\"%s\"/>\n",
111                                         type);
112         }
113 }
114
115 static void generate_interface_xml(GString *gstr, struct interface_data *iface)
116 {
117         GDBusMethodTable *method;
118         GDBusSignalTable *signal;
119
120         for (method = iface->methods; method && method->name; method++) {
121                 if (!strlen(method->signature) && !strlen(method->reply))
122                         g_string_append_printf(gstr, "\t\t<method name=\"%s\"/>\n",
123                                                                 method->name);
124                 else {
125                         g_string_append_printf(gstr, "\t\t<method name=\"%s\">\n",
126                                                                 method->name);
127                         print_arguments(gstr, method->signature, "in");
128                         print_arguments(gstr, method->reply, "out");
129                         g_string_append_printf(gstr, "\t\t</method>\n");
130                 }
131         }
132
133         for (signal = iface->signals; signal && signal->name; signal++) {
134                 if (!strlen(signal->signature))
135                         g_string_append_printf(gstr, "\t\t<signal name=\"%s\"/>\n",
136                                                                 signal->name);
137                 else {
138                         g_string_append_printf(gstr, "\t\t<signal name=\"%s\">\n",
139                                                                 signal->name);
140                         print_arguments(gstr, signal->signature, NULL);
141                         g_string_append_printf(gstr, "\t\t</signal>\n");
142                 }
143         }
144 }
145
146 static void generate_introspection_xml(DBusConnection *conn,
147                                 struct generic_data *data, const char *path)
148 {
149         GSList *list;
150         GString *gstr;
151         char **children;
152         int i;
153
154         g_free(data->introspect);
155
156         gstr = g_string_new(DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE);
157
158         g_string_append_printf(gstr, "<node name=\"%s\">\n", path);
159
160         for (list = data->interfaces; list; list = list->next) {
161                 struct interface_data *iface = list->data;
162
163                 g_string_append_printf(gstr, "\t<interface name=\"%s\">\n",
164                                                                 iface->name);
165
166                 generate_interface_xml(gstr, iface);
167
168                 g_string_append_printf(gstr, "\t</interface>\n");
169         }
170
171         if (!dbus_connection_list_registered(conn, path, &children))
172                 goto done;
173
174         for (i = 0; children[i]; i++)
175                 g_string_append_printf(gstr, "\t<node name=\"%s\"/>\n",
176                                                                 children[i]);
177
178         dbus_free_string_array(children);
179
180 done:
181         g_string_append_printf(gstr, "</node>\n");
182
183         data->introspect = g_string_free(gstr, FALSE);
184 }
185
186 static DBusHandlerResult introspect(DBusConnection *connection,
187                                 DBusMessage *message, struct generic_data *data)
188 {
189         DBusMessage *reply;
190
191         if (!dbus_message_has_signature(message, DBUS_TYPE_INVALID_AS_STRING)) {
192                 error("Unexpected signature to introspect call");
193                 return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
194         }
195
196         if (!data->introspect)
197                 generate_introspection_xml(connection, data,
198                                                 dbus_message_get_path(message));
199
200         reply = dbus_message_new_method_return(message);
201         if (!reply)
202                 return DBUS_HANDLER_RESULT_NEED_MEMORY;
203
204         dbus_message_append_args(reply, DBUS_TYPE_STRING, &data->introspect,
205                                         DBUS_TYPE_INVALID);
206
207         dbus_connection_send(connection, reply, NULL);
208
209         dbus_message_unref(reply);
210
211         return DBUS_HANDLER_RESULT_HANDLED;
212 }
213
214 static void generic_unregister(DBusConnection *connection, void *user_data)
215 {
216         struct generic_data *data = user_data;
217
218         g_free(data->introspect);
219         g_free(data);
220 }
221
222 static struct interface_data *find_interface(GSList *interfaces,
223                                                 const char *name)
224 {
225         GSList *list;
226
227         if (!name)
228                 return NULL;
229
230         for (list = interfaces; list; list = list->next) {
231                 struct interface_data *iface = list->data;
232                 if (!strcmp(name, iface->name))
233                         return iface;
234         }
235
236         return NULL;
237 }
238
239 static DBusHandlerResult generic_message(DBusConnection *connection,
240                                         DBusMessage *message, void *user_data)
241 {
242         struct generic_data *data = user_data;
243         struct interface_data *iface;
244         GDBusMethodTable *method;
245         const char *interface;
246
247         if (dbus_message_is_method_call(message,
248                                         DBUS_INTERFACE_INTROSPECTABLE,
249                                                                 "Introspect"))
250                 return introspect(connection, message, data);
251
252         interface = dbus_message_get_interface(message);
253
254         iface = find_interface(data->interfaces, interface);
255         if (!iface)
256                 return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
257
258         for (method = iface->methods; method &&
259                         method->name && method->function; method++) {
260                 DBusMessage *reply;
261
262                 if (dbus_message_is_method_call(message, iface->name,
263                                                         method->name) == FALSE)
264                         continue;
265
266                 if (dbus_message_has_signature(message,
267                                                 method->signature) == FALSE)
268                         continue;
269
270                 reply = method->function(connection, message, iface->user_data);
271
272                 if (method->flags & G_DBUS_METHOD_FLAG_NOREPLY) {
273                         if (reply != NULL)
274                                 dbus_message_unref(reply);
275                         return DBUS_HANDLER_RESULT_HANDLED;
276                 }
277
278                 if (method->flags & G_DBUS_METHOD_FLAG_ASYNC) {
279                         if (reply == NULL)
280                                 return DBUS_HANDLER_RESULT_HANDLED;
281                 }
282
283                 if (reply == NULL)
284                         return DBUS_HANDLER_RESULT_NEED_MEMORY;
285
286                 dbus_connection_send(connection, reply, NULL);
287                 dbus_message_unref(reply);
288
289                 return DBUS_HANDLER_RESULT_HANDLED;
290         }
291
292         return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
293 }
294
295 static DBusObjectPathVTable generic_table = {
296         .unregister_function    = generic_unregister,
297         .message_function       = generic_message,
298 };
299
300 static void invalidate_parent_data(DBusConnection *conn, const char *child_path)
301 {
302         struct generic_data *data = NULL;
303         char *parent_path, *slash;
304
305         parent_path = g_strdup(child_path);
306         slash = strrchr(parent_path, '/');
307         if (!slash)
308                 goto done;
309
310         if (slash == parent_path && parent_path[1] != '\0')
311                 parent_path[1] = '\0';
312         else
313                 *slash = '\0';
314
315         if (!strlen(parent_path))
316                 goto done;
317
318         if (!dbus_connection_get_object_path_data(conn, parent_path,
319                                                         (void *) &data))
320                 goto done;
321
322         if (!data)
323                 goto done;
324
325         g_free(data->introspect);
326         data->introspect = NULL;
327
328 done:
329         g_free(parent_path);
330 }
331
332 static struct generic_data *object_path_ref(DBusConnection *connection,
333                                                         const char *path)
334 {
335         struct generic_data *data;
336
337         if (dbus_connection_get_object_path_data(connection, path,
338                                                 (void *) &data) == TRUE) {
339                 if (data != NULL) {
340                         data->refcount++;
341                         return data;
342                 }
343         }
344
345         data = g_new0(struct generic_data, 1);
346
347         data->introspect = g_strdup(DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE "<node></node>");
348
349         data->refcount = 1;
350
351         if (!dbus_connection_register_object_path(connection, path,
352                                                 &generic_table, data)) {
353                 g_free(data->introspect);
354                 g_free(data);
355                 return NULL;
356         }
357
358         invalidate_parent_data(connection, path);
359
360         return data;
361 }
362
363 static void object_path_unref(DBusConnection *connection, const char *path)
364 {
365         struct generic_data *data = NULL;
366
367         if (dbus_connection_get_object_path_data(connection, path,
368                                                 (void *) &data) == FALSE)
369                 return;
370
371         if (data == NULL)
372                 return;
373
374         data->refcount--;
375
376         if (data->refcount > 0)
377                 return;
378
379         invalidate_parent_data(connection, path);
380
381         dbus_connection_unregister_object_path(connection, path);
382 }
383
384 static gboolean check_signal(DBusConnection *conn, const char *path,
385                                 const char *interface, const char *name,
386                                 const char **args)
387 {
388         struct generic_data *data = NULL;
389         struct interface_data *iface;
390         GDBusSignalTable *signal;
391
392         *args = NULL;
393         if (!dbus_connection_get_object_path_data(conn, path,
394                                         (void *) &data) || !data) {
395                 error("dbus_connection_emit_signal: path %s isn't registered",
396                                 path);
397                 return FALSE;
398         }
399
400         iface = find_interface(data->interfaces, interface);
401         if (!iface) {
402                 error("dbus_connection_emit_signal: %s does not implement %s",
403                                 path, interface);
404                 return FALSE;
405         }
406
407         for (signal = iface->signals; signal && signal->name; signal++) {
408                 if (!strcmp(signal->name, name)) {
409                         *args = signal->signature;
410                         break;
411                 }
412         }
413
414         if (!*args) {
415                 error("No signal named %s on interface %s", name, interface);
416                 return FALSE;
417         }
418
419         return TRUE;
420 }
421
422 static dbus_bool_t emit_signal_valist(DBusConnection *conn,
423                                                 const char *path,
424                                                 const char *interface,
425                                                 const char *name,
426                                                 int first,
427                                                 va_list var_args)
428 {
429         DBusMessage *signal;
430         dbus_bool_t ret;
431         const char *signature, *args;
432
433         if (!check_signal(conn, path, interface, name, &args))
434                 return FALSE;
435
436         signal = dbus_message_new_signal(path, interface, name);
437         if (!signal) {
438                 error("Unable to allocate new %s.%s signal", interface,  name);
439                 return FALSE;
440         }
441
442         ret = dbus_message_append_args_valist(signal, first, var_args);
443         if (!ret)
444                 goto fail;
445
446         signature = dbus_message_get_signature(signal);
447         if (strcmp(args, signature) != 0) {
448                 error("%s.%s: expected signature'%s' but got '%s'",
449                                 interface, name, args, signature);
450                 ret = FALSE;
451                 goto fail;
452         }
453
454         ret = dbus_connection_send(conn, signal, NULL);
455
456 fail:
457         dbus_message_unref(signal);
458
459         return ret;
460 }
461
462 gboolean g_dbus_register_interface(DBusConnection *connection,
463                                         const char *path, const char *name,
464                                         GDBusMethodTable *methods,
465                                         GDBusSignalTable *signals,
466                                         GDBusPropertyTable *properties,
467                                         void *user_data,
468                                         GDBusDestroyFunction destroy)
469 {
470         struct generic_data *data;
471         struct interface_data *iface;
472
473         data = object_path_ref(connection, path);
474         if (data == NULL)
475                 return FALSE;
476
477         if (find_interface(data->interfaces, name))
478                 return FALSE;
479
480         iface = g_new0(struct interface_data, 1);
481
482         iface->name = g_strdup(name);
483         iface->methods = methods;
484         iface->signals = signals;
485         iface->properties = properties;
486         iface->user_data = user_data;
487         iface->destroy = destroy;
488
489         data->interfaces = g_slist_append(data->interfaces, iface);
490
491         g_free(data->introspect);
492         data->introspect = NULL;
493
494         return TRUE;
495 }
496
497 gboolean g_dbus_unregister_interface(DBusConnection *connection,
498                                         const char *path, const char *name)
499 {
500         struct generic_data *data = NULL;
501         struct interface_data *iface;
502
503         if (!path)
504                 return FALSE;
505
506         if (dbus_connection_get_object_path_data(connection, path,
507                                                 (void *) &data) == FALSE)
508                 return FALSE;
509
510         if (data == NULL)
511                 return FALSE;
512
513         iface = find_interface(data->interfaces, name);
514         if (!iface)
515                 return FALSE;
516
517         data->interfaces = g_slist_remove(data->interfaces, iface);
518
519         if (iface->destroy)
520                 iface->destroy(iface->user_data);
521
522         g_free(iface->name);
523         g_free(iface);
524
525         g_free(data->introspect);
526         data->introspect = NULL;
527
528         object_path_unref(connection, path);
529
530         return TRUE;
531 }
532
533 DBusMessage *g_dbus_create_error_valist(DBusMessage *message, const char *name,
534                                         const char *format, va_list args)
535 {
536         char str[1024];
537
538         vsnprintf(str, sizeof(str), format, args);
539
540         return dbus_message_new_error(message, name, str);
541 }
542
543 DBusMessage *g_dbus_create_error(DBusMessage *message, const char *name,
544                                                 const char *format, ...)
545 {
546         va_list args;
547         DBusMessage *reply;
548
549         va_start(args, format);
550
551         reply = g_dbus_create_error_valist(message, name, format, args);
552
553         va_end(args);
554
555         return reply;
556 }
557
558 DBusMessage *g_dbus_create_reply_valist(DBusMessage *message,
559                                                 int type, va_list args)
560 {
561         DBusMessage *reply;
562
563         reply = dbus_message_new_method_return(message);
564         if (reply == NULL)
565                 return NULL;
566
567         if (dbus_message_append_args_valist(reply, type, args) == FALSE) {
568                 dbus_message_unref(reply);
569                 return NULL;
570         }
571
572         return reply;
573 }
574
575 DBusMessage *g_dbus_create_reply(DBusMessage *message, int type, ...)
576 {
577         va_list args;
578         DBusMessage *reply;
579
580         va_start(args, type);
581
582         reply = g_dbus_create_reply_valist(message, type, args);
583
584         va_end(args);
585
586         return reply;
587 }
588
589 gboolean g_dbus_send_message(DBusConnection *connection, DBusMessage *message)
590 {
591         dbus_bool_t result;
592
593         if (dbus_message_get_type(message) == DBUS_MESSAGE_TYPE_METHOD_CALL)
594                 dbus_message_set_no_reply(message, TRUE);
595
596         result = dbus_connection_send(connection, message, NULL);
597
598         dbus_message_unref(message);
599
600         return result;
601 }
602
603 gboolean g_dbus_send_reply_valist(DBusConnection *connection,
604                                 DBusMessage *message, int type, va_list args)
605 {
606         DBusMessage *reply;
607
608         reply = dbus_message_new_method_return(message);
609         if (reply == NULL)
610                 return FALSE;
611
612         if (dbus_message_append_args_valist(reply, type, args) == FALSE) {
613                 dbus_message_unref(reply);
614                 return FALSE;
615         }
616
617         return g_dbus_send_message(connection, reply);
618 }
619
620 gboolean g_dbus_send_reply(DBusConnection *connection,
621                                 DBusMessage *message, int type, ...)
622 {
623         va_list args;
624         gboolean result;
625
626         va_start(args, type);
627
628         result = g_dbus_send_reply_valist(connection, message, type, args);
629
630         va_end(args);
631
632         return result;
633 }
634
635 gboolean g_dbus_emit_signal(DBusConnection *connection,
636                                 const char *path, const char *interface,
637                                 const char *name, int type, ...)
638 {
639         va_list args;
640         gboolean result;
641
642         va_start(args, type);
643
644         result = emit_signal_valist(connection, path, interface,
645                                                         name, type, args);
646
647         va_end(args);
648
649         return result;
650 }
651
652 gboolean g_dbus_emit_signal_valist(DBusConnection *connection,
653                                 const char *path, const char *interface,
654                                 const char *name, int type, va_list args)
655 {
656         return emit_signal_valist(connection, path, interface,
657                                                         name, type, args);
658 }