1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
#include <config.h>
#include <stdio.h>
#include <stdlib.h>
#include <dbus/dbus.h>
#include <glib.h>
#include <dbus/dbus-gmain.h>
#include "test/lib/util.h"
DBusConnection *bus;
GMainContext *main_context;
typedef struct _SpiReentrantCallClosure
{
GMainLoop *loop;
DBusMessage *reply;
} SpiReentrantCallClosure;
static void
set_reply (DBusPendingCall * pending, void *user_data)
{
SpiReentrantCallClosure* closure = (SpiReentrantCallClosure *) user_data;
closure->reply = dbus_pending_call_steal_reply (pending);
dbus_gmain_set_up_connection (bus, NULL);
g_main_loop_quit (closure->loop);
}
static DBusMessage *
send_and_allow_reentry (DBusConnection * bus, DBusMessage * message,
dbus_bool_t switch_after_send)
{
DBusPendingCall *pending;
SpiReentrantCallClosure closure;
closure.loop = g_main_loop_new (main_context, FALSE);
dbus_gmain_set_up_connection (bus, (switch_after_send ? NULL :
main_context));
if (!dbus_connection_send_with_reply (bus, message, &pending, 3000))
{
dbus_gmain_set_up_connection (bus, NULL);
return NULL;
}
dbus_pending_call_set_notify (pending, set_reply, (void *) &closure, NULL);
if (switch_after_send)
dbus_gmain_set_up_connection (bus, main_context);
g_main_loop_run (closure.loop);
g_main_loop_unref (closure.loop);
dbus_pending_call_unref (pending);
return closure.reply;
}
static void
send_test_message (dbus_bool_t switch_after_send)
{
DBusMessage *message, *reply;
const char *str;
DBusError error;
dbus_error_init (&error);
message = dbus_message_new_method_call ("org.freedesktop.DBus",
"/org/freedesktop/DBus",
DBUS_INTERFACE_DBUS, "GetId");
reply = send_and_allow_reentry (bus, message, switch_after_send);
if (!reply)
{
fprintf(stderr, "Got no reply from send_and_allow_reentry\n");
exit(1);
}
if (dbus_message_get_type (reply) == DBUS_MESSAGE_TYPE_ERROR)
{
char *err = NULL;
dbus_message_get_args (reply, NULL, DBUS_TYPE_STRING, &err, DBUS_TYPE_INVALID);
fprintf (stderr, "Got error: %s\n", err);
exit(1);
}
if (!dbus_message_get_args (reply, &error, DBUS_TYPE_STRING, &str, DBUS_TYPE_INVALID))
{
fprintf(stderr, "Sorry; can't communicate: %s\n", error.message);
exit(1);
}
dbus_message_unref (reply);
dbus_message_unref (message);
}
int
main(int argc, const char *argv[])
{
DBusError error;
main_context = g_main_context_new ();
dbus_error_init (&error);
bus = dbus_bus_get_private (DBUS_BUS_SESSION, &error);
if (!bus)
{
fprintf(stderr, "Couldn't connect to bus: %s\n", error.name);
return 1;
}
dbus_gmain_set_up_connection (bus, NULL);
send_test_message (FALSE);
send_test_message (FALSE);
send_test_message (TRUE);
test_run_until_disconnected (bus, NULL);
dbus_connection_unref (bus);
dbus_shutdown ();
g_main_context_unref (main_context);
return 0;
}
|
