Added GKdbus to GDBusWorker struct
[platform/upstream/glib.git] / gio / gdbusprivate.c
1 /* GDBus - GLib D-Bus Library
2  *
3  * Copyright (C) 2008-2010 Red Hat, Inc.
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2 of the License, or (at your option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General
16  * Public License along with this library; if not, write to the
17  * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
18  * Boston, MA 02111-1307, USA.
19  *
20  * Author: David Zeuthen <davidz@redhat.com>
21  */
22
23 #include "config.h"
24
25 #include <stdlib.h>
26 #include <string.h>
27 #ifdef HAVE_UNISTD_H
28 #include <unistd.h>
29 #endif
30
31 #include "giotypes.h"
32 #include "gsocket.h"
33 #include "gdbusprivate.h"
34 #include "gdbusmessage.h"
35 #include "gdbuserror.h"
36 #include "gdbusintrospection.h"
37 #include "gasyncresult.h"
38 #include "gsimpleasyncresult.h"
39 #include "ginputstream.h"
40 #include "gmemoryinputstream.h"
41 #include "giostream.h"
42 #include "glib/gstdio.h"
43 #include "gsocketcontrolmessage.h"
44 #include "gsocketconnection.h"
45 #include "gkdbusconnection.h"
46 #include "gsocketoutputstream.h"
47
48 #ifdef G_OS_UNIX
49 #include "gunixfdmessage.h"
50 #include "gunixconnection.h"
51 #include "gunixcredentialsmessage.h"
52 #endif
53
54 #ifdef G_OS_WIN32
55 #include <windows.h>
56 #endif
57
58 #include "glibintl.h"
59
60 static gboolean _g_dbus_worker_do_initial_read (gpointer data);
61
62 /* ---------------------------------------------------------------------------------------------------- */
63
64 gchar *
65 _g_dbus_hexdump (const gchar *data, gsize len, guint indent)
66 {
67  guint n, m;
68  GString *ret;
69
70  ret = g_string_new (NULL);
71
72  for (n = 0; n < len; n += 16)
73    {
74      g_string_append_printf (ret, "%*s%04x: ", indent, "", n);
75
76      for (m = n; m < n + 16; m++)
77        {
78          if (m > n && (m%4) == 0)
79            g_string_append_c (ret, ' ');
80          if (m < len)
81            g_string_append_printf (ret, "%02x ", (guchar) data[m]);
82          else
83            g_string_append (ret, "   ");
84        }
85
86      g_string_append (ret, "   ");
87
88      for (m = n; m < len && m < n + 16; m++)
89        g_string_append_c (ret, g_ascii_isprint (data[m]) ? data[m] : '.');
90
91      g_string_append_c (ret, '\n');
92    }
93
94  return g_string_free (ret, FALSE);
95 }
96
97 /* ---------------------------------------------------------------------------------------------------- */
98
99 /* Unfortunately ancillary messages are discarded when reading from a
100  * socket using the GSocketInputStream abstraction. So we provide a
101  * very GInputStream-ish API that uses GSocket in this case (very
102  * similar to GSocketInputStream).
103  */
104
105 typedef struct
106 {
107   GSocket *socket;
108   GCancellable *cancellable;
109
110   void *buffer;
111   gsize count;
112
113   GSocketControlMessage ***messages;
114   gint *num_messages;
115
116   GSimpleAsyncResult *simple;
117
118   gboolean from_mainloop;
119 } ReadWithControlData;
120
121 static void
122 read_with_control_data_free (ReadWithControlData *data)
123 {
124   g_object_unref (data->socket);
125   if (data->cancellable != NULL)
126     g_object_unref (data->cancellable);
127   g_object_unref (data->simple);
128   g_free (data);
129 }
130
131 static gboolean
132 _g_socket_read_with_control_messages_ready (GSocket      *socket,
133                                             GIOCondition  condition,
134                                             gpointer      user_data)
135 {
136   ReadWithControlData *data = user_data;
137   GError *error;
138   gssize result;
139   GInputVector vector;
140
141   error = NULL;
142   vector.buffer = data->buffer;
143   vector.size = data->count;
144   result = g_socket_receive_message (data->socket,
145                                      NULL, /* address */
146                                      &vector,
147                                      1,
148                                      data->messages,
149                                      data->num_messages,
150                                      NULL,
151                                      data->cancellable,
152                                      &error);
153   if (result >= 0)
154     {
155       g_simple_async_result_set_op_res_gssize (data->simple, result);
156     }
157   else
158     {
159       g_assert (error != NULL);
160       g_simple_async_result_take_error (data->simple, error);
161     }
162
163   if (data->from_mainloop)
164     g_simple_async_result_complete (data->simple);
165   else
166     g_simple_async_result_complete_in_idle (data->simple);
167
168   return FALSE;
169 }
170
171 static void
172 _g_socket_read_with_control_messages (GSocket                 *socket,
173                                       void                    *buffer,
174                                       gsize                    count,
175                                       GSocketControlMessage ***messages,
176                                       gint                    *num_messages,
177                                       gint                     io_priority,
178                                       GCancellable            *cancellable,
179                                       GAsyncReadyCallback      callback,
180                                       gpointer                 user_data)
181 {
182   ReadWithControlData *data;
183
184   data = g_new0 (ReadWithControlData, 1);
185   data->socket = g_object_ref (socket);
186   data->cancellable = cancellable != NULL ? g_object_ref (cancellable) : NULL;
187   data->buffer = buffer;
188   data->count = count;
189   data->messages = messages;
190   data->num_messages = num_messages;
191
192   data->simple = g_simple_async_result_new (G_OBJECT (socket),
193                                             callback,
194                                             user_data,
195                                             _g_socket_read_with_control_messages);
196   g_simple_async_result_set_check_cancellable (data->simple, cancellable);
197
198   if (!g_socket_condition_check (socket, G_IO_IN))
199     {
200       GSource *source;
201       data->from_mainloop = TRUE;
202       source = g_socket_create_source (data->socket,
203                                        G_IO_IN | G_IO_HUP | G_IO_ERR,
204                                        cancellable);
205       g_source_set_callback (source,
206                              (GSourceFunc) _g_socket_read_with_control_messages_ready,
207                              data,
208                              (GDestroyNotify) read_with_control_data_free);
209       g_source_attach (source, g_main_context_get_thread_default ());
210       g_source_unref (source);
211     }
212   else
213     {
214       _g_socket_read_with_control_messages_ready (data->socket, G_IO_IN, data);
215       read_with_control_data_free (data);
216     }
217 }
218
219 static gssize
220 _g_socket_read_with_control_messages_finish (GSocket       *socket,
221                                              GAsyncResult  *result,
222                                              GError       **error)
223 {
224   GSimpleAsyncResult *simple = G_SIMPLE_ASYNC_RESULT (result);
225
226   g_return_val_if_fail (G_IS_SOCKET (socket), -1);
227   g_warn_if_fail (g_simple_async_result_get_source_tag (simple) == _g_socket_read_with_control_messages);
228
229   if (g_simple_async_result_propagate_error (simple, error))
230       return -1;
231   else
232     return g_simple_async_result_get_op_res_gssize (simple);
233 }
234
235 /* ---------------------------------------------------------------------------------------------------- */
236
237 /* Work-around for https://bugzilla.gnome.org/show_bug.cgi?id=627724 */
238
239 static GPtrArray *ensured_classes = NULL;
240
241 static void
242 ensure_type (GType gtype)
243 {
244   g_ptr_array_add (ensured_classes, g_type_class_ref (gtype));
245 }
246
247 static void
248 release_required_types (void)
249 {
250   g_ptr_array_foreach (ensured_classes, (GFunc) g_type_class_unref, NULL);
251   g_ptr_array_unref (ensured_classes);
252   ensured_classes = NULL;
253 }
254
255 static void
256 ensure_required_types (void)
257 {
258   g_assert (ensured_classes == NULL);
259   ensured_classes = g_ptr_array_new ();
260   ensure_type (G_TYPE_SIMPLE_ASYNC_RESULT);
261   ensure_type (G_TYPE_MEMORY_INPUT_STREAM);
262 }
263 /* ---------------------------------------------------------------------------------------------------- */
264
265 typedef struct
266 {
267   volatile gint refcount;
268   GThread *thread;
269   GMainContext *context;
270   GMainLoop *loop;
271 } SharedThreadData;
272
273 static gpointer
274 gdbus_shared_thread_func (gpointer user_data)
275 {
276   SharedThreadData *data = user_data;
277
278   g_main_context_push_thread_default (data->context);
279   g_main_loop_run (data->loop);
280   g_main_context_pop_thread_default (data->context);
281
282   release_required_types ();
283
284   return NULL;
285 }
286
287 /* ---------------------------------------------------------------------------------------------------- */
288
289 static SharedThreadData *
290 _g_dbus_shared_thread_ref (void)
291 {
292   static gsize shared_thread_data = 0;
293   SharedThreadData *ret;
294
295   if (g_once_init_enter (&shared_thread_data))
296     {
297       SharedThreadData *data;
298
299       /* Work-around for https://bugzilla.gnome.org/show_bug.cgi?id=627724 */
300       ensure_required_types ();
301
302       data = g_new0 (SharedThreadData, 1);
303       data->refcount = 0;
304       
305       data->context = g_main_context_new ();
306       data->loop = g_main_loop_new (data->context, FALSE);
307       data->thread = g_thread_new ("gdbus",
308                                    gdbus_shared_thread_func,
309                                    data);
310       /* We can cast between gsize and gpointer safely */
311       g_once_init_leave (&shared_thread_data, (gsize) data);
312     }
313
314   ret = (SharedThreadData*) shared_thread_data;
315   g_atomic_int_inc (&ret->refcount);
316   return ret;
317 }
318
319 static void
320 _g_dbus_shared_thread_unref (SharedThreadData *data)
321 {
322   /* TODO: actually destroy the shared thread here */
323 #if 0
324   g_assert (data != NULL);
325   if (g_atomic_int_dec_and_test (&data->refcount))
326     {
327       g_main_loop_quit (data->loop);
328       //g_thread_join (data->thread);
329       g_main_loop_unref (data->loop);
330       g_main_context_unref (data->context);
331     }
332 #endif
333 }
334
335 /* ---------------------------------------------------------------------------------------------------- */
336
337 typedef enum {
338     PENDING_NONE = 0,
339     PENDING_WRITE,
340     PENDING_FLUSH,
341     PENDING_CLOSE
342 } OutputPending;
343
344 struct GDBusWorker
345 {
346   volatile gint                       ref_count;
347
348   SharedThreadData                   *shared_thread_data;
349
350   /* really a boolean, but GLib 2.28 lacks atomic boolean ops */
351   volatile gint                       stopped;
352
353   /* TODO: frozen (e.g. G_DBUS_CONNECTION_FLAGS_DELAY_MESSAGE_PROCESSING) currently
354    * only affects messages received from the other peer (since GDBusServer is the
355    * only user) - we might want it to affect messages sent to the other peer too?
356    */
357   gboolean                            frozen;
358   GDBusCapabilityFlags                capabilities;
359   GQueue                             *received_messages_while_frozen;
360
361   GIOStream                          *stream;
362   GCancellable                       *cancellable;
363   GDBusWorkerMessageReceivedCallback  message_received_callback;
364   GDBusWorkerMessageAboutToBeSentCallback message_about_to_be_sent_callback;
365   GDBusWorkerDisconnectedCallback     disconnected_callback;
366   gpointer                            user_data;
367
368   /* if GSocket and GKdbus are NULL, stream is GSocketConnection */
369   GSocket *socket;
370   GKdbus  *kdbus;
371
372   /* used for reading */
373   GMutex                              read_lock;
374   gchar                              *read_buffer;
375   gsize                               read_buffer_allocated_size;
376   gsize                               read_buffer_cur_size;
377   gsize                               read_buffer_bytes_wanted;
378   GUnixFDList                        *read_fd_list;
379   GSocketControlMessage             **read_ancillary_messages;
380   gint                                read_num_ancillary_messages;
381
382   /* Whether an async write, flush or close, or none of those, is pending.
383    * Only the worker thread may change its value, and only with the write_lock.
384    * Other threads may read its value when holding the write_lock.
385    * The worker thread may read its value at any time.
386    */
387   OutputPending                       output_pending;
388   /* used for writing */
389   GMutex                              write_lock;
390   /* queue of MessageToWriteData, protected by write_lock */
391   GQueue                             *write_queue;
392   /* protected by write_lock */
393   guint64                             write_num_messages_written;
394   /* number of messages we'd written out last time we flushed;
395    * protected by write_lock
396    */
397   guint64                             write_num_messages_flushed;
398   /* list of FlushData, protected by write_lock */
399   GList                              *write_pending_flushes;
400   /* list of CloseData, protected by write_lock */
401   GList                              *pending_close_attempts;
402   /* no lock - only used from the worker thread */
403   gboolean                            close_expected;
404 };
405
406 static void _g_dbus_worker_unref (GDBusWorker *worker);
407
408 /* ---------------------------------------------------------------------------------------------------- */
409
410 typedef struct
411 {
412   GMutex  mutex;
413   GCond   cond;
414   guint64 number_to_wait_for;
415   GError *error;
416 } FlushData;
417
418 struct _MessageToWriteData ;
419 typedef struct _MessageToWriteData MessageToWriteData;
420
421 static void message_to_write_data_free (MessageToWriteData *data);
422
423 static void read_message_print_transport_debug (gssize bytes_read,
424                                                 GDBusWorker *worker);
425
426 static void write_message_print_transport_debug (gssize bytes_written,
427                                                  MessageToWriteData *data);
428
429 typedef struct {
430     GDBusWorker *worker;
431     GCancellable *cancellable;
432     GSimpleAsyncResult *result;
433 } CloseData;
434
435 static void close_data_free (CloseData *close_data)
436 {
437   if (close_data->cancellable != NULL)
438     g_object_unref (close_data->cancellable);
439
440   if (close_data->result != NULL)
441     g_object_unref (close_data->result);
442
443   _g_dbus_worker_unref (close_data->worker);
444   g_slice_free (CloseData, close_data);
445 }
446
447 /* ---------------------------------------------------------------------------------------------------- */
448
449 static GDBusWorker *
450 _g_dbus_worker_ref (GDBusWorker *worker)
451 {
452   g_atomic_int_inc (&worker->ref_count);
453   return worker;
454 }
455
456 static void
457 _g_dbus_worker_unref (GDBusWorker *worker)
458 {
459   if (g_atomic_int_dec_and_test (&worker->ref_count))
460     {
461       g_assert (worker->write_pending_flushes == NULL);
462
463       _g_dbus_shared_thread_unref (worker->shared_thread_data);
464
465       g_object_unref (worker->stream);
466
467       g_mutex_clear (&worker->read_lock);
468       g_object_unref (worker->cancellable);
469       if (worker->read_fd_list != NULL)
470         g_object_unref (worker->read_fd_list);
471
472       g_queue_free_full (worker->received_messages_while_frozen, (GDestroyNotify) g_object_unref);
473       g_mutex_clear (&worker->write_lock);
474       g_queue_free_full (worker->write_queue, (GDestroyNotify) message_to_write_data_free);
475       g_free (worker->read_buffer);
476
477       g_free (worker);
478     }
479 }
480
481 static void
482 _g_dbus_worker_emit_disconnected (GDBusWorker  *worker,
483                                   gboolean      remote_peer_vanished,
484                                   GError       *error)
485 {
486   if (!g_atomic_int_get (&worker->stopped))
487     worker->disconnected_callback (worker, remote_peer_vanished, error, worker->user_data);
488 }
489
490 static void
491 _g_dbus_worker_emit_message_received (GDBusWorker  *worker,
492                                       GDBusMessage *message)
493 {
494   if (!g_atomic_int_get (&worker->stopped))
495     worker->message_received_callback (worker, message, worker->user_data);
496 }
497
498 static GDBusMessage *
499 _g_dbus_worker_emit_message_about_to_be_sent (GDBusWorker  *worker,
500                                               GDBusMessage *message)
501 {
502   GDBusMessage *ret;
503   if (!g_atomic_int_get (&worker->stopped))
504     ret = worker->message_about_to_be_sent_callback (worker, message, worker->user_data);
505   else
506     ret = message;
507   return ret;
508 }
509
510 /* can only be called from private thread with read-lock held - takes ownership of @message */
511 static void
512 _g_dbus_worker_queue_or_deliver_received_message (GDBusWorker  *worker,
513                                                   GDBusMessage *message)
514 {
515   if (worker->frozen || g_queue_get_length (worker->received_messages_while_frozen) > 0)
516     {
517       /* queue up */
518       g_queue_push_tail (worker->received_messages_while_frozen, message);
519     }
520   else
521     {
522       /* not frozen, nor anything in queue */
523       _g_dbus_worker_emit_message_received (worker, message);
524       g_object_unref (message);
525     }
526 }
527
528 /* called in private thread shared by all GDBusConnection instances (without read-lock held) */
529 static gboolean
530 unfreeze_in_idle_cb (gpointer user_data)
531 {
532   GDBusWorker *worker = user_data;
533   GDBusMessage *message;
534
535   g_mutex_lock (&worker->read_lock);
536   if (worker->frozen)
537     {
538       while ((message = g_queue_pop_head (worker->received_messages_while_frozen)) != NULL)
539         {
540           _g_dbus_worker_emit_message_received (worker, message);
541           g_object_unref (message);
542         }
543       worker->frozen = FALSE;
544     }
545   else
546     {
547       g_assert (g_queue_get_length (worker->received_messages_while_frozen) == 0);
548     }
549   g_mutex_unlock (&worker->read_lock);
550   return FALSE;
551 }
552
553 /* can be called from any thread */
554 void
555 _g_dbus_worker_unfreeze (GDBusWorker *worker)
556 {
557   GSource *idle_source;
558   idle_source = g_idle_source_new ();
559   g_source_set_priority (idle_source, G_PRIORITY_DEFAULT);
560   g_source_set_callback (idle_source,
561                          unfreeze_in_idle_cb,
562                          _g_dbus_worker_ref (worker),
563                          (GDestroyNotify) _g_dbus_worker_unref);
564   g_source_attach (idle_source, worker->shared_thread_data->context);
565   g_source_unref (idle_source);
566 }
567
568 /* ---------------------------------------------------------------------------------------------------- */
569
570 static void _g_dbus_worker_do_read_unlocked (GDBusWorker *worker);
571
572 /* called in private thread shared by all GDBusConnection instances (without read-lock held) */
573 static void
574 _g_dbus_worker_do_read_cb (GInputStream  *input_stream,
575                            GAsyncResult  *res,
576                            gpointer       user_data)
577 {
578   GDBusWorker *worker = user_data;
579   GError *error;
580   gssize bytes_read;
581
582   g_mutex_lock (&worker->read_lock);
583
584   /* If already stopped, don't even process the reply */
585   if (g_atomic_int_get (&worker->stopped))
586     goto out;
587
588   error = NULL;
589   if (worker->socket == NULL)
590     bytes_read = g_input_stream_read_finish (g_io_stream_get_input_stream (worker->stream),
591                                              res,
592                                              &error);
593   else
594     bytes_read = _g_socket_read_with_control_messages_finish (worker->socket,
595                                                               res,
596                                                               &error);
597   if (worker->read_num_ancillary_messages > 0)
598     {
599       gint n;
600       for (n = 0; n < worker->read_num_ancillary_messages; n++)
601         {
602           GSocketControlMessage *control_message = G_SOCKET_CONTROL_MESSAGE (worker->read_ancillary_messages[n]);
603
604           if (FALSE)
605             {
606             }
607 #ifdef G_OS_UNIX
608           else if (G_IS_UNIX_FD_MESSAGE (control_message))
609             {
610               GUnixFDMessage *fd_message;
611               gint *fds;
612               gint num_fds;
613
614               fd_message = G_UNIX_FD_MESSAGE (control_message);
615               fds = g_unix_fd_message_steal_fds (fd_message, &num_fds);
616               if (worker->read_fd_list == NULL)
617                 {
618                   worker->read_fd_list = g_unix_fd_list_new_from_array (fds, num_fds);
619                 }
620               else
621                 {
622                   gint n;
623                   for (n = 0; n < num_fds; n++)
624                     {
625                       /* TODO: really want a append_steal() */
626                       g_unix_fd_list_append (worker->read_fd_list, fds[n], NULL);
627                       (void) g_close (fds[n], NULL);
628                     }
629                 }
630               g_free (fds);
631             }
632           else if (G_IS_UNIX_CREDENTIALS_MESSAGE (control_message))
633             {
634               /* do nothing */
635             }
636 #endif
637           else
638             {
639               if (error == NULL)
640                 {
641                   g_set_error (&error,
642                                G_IO_ERROR,
643                                G_IO_ERROR_FAILED,
644                                "Unexpected ancillary message of type %s received from peer",
645                                g_type_name (G_TYPE_FROM_INSTANCE (control_message)));
646                   _g_dbus_worker_emit_disconnected (worker, TRUE, error);
647                   g_error_free (error);
648                   g_object_unref (control_message);
649                   n++;
650                   while (n < worker->read_num_ancillary_messages)
651                     g_object_unref (worker->read_ancillary_messages[n++]);
652                   g_free (worker->read_ancillary_messages);
653                   goto out;
654                 }
655             }
656           g_object_unref (control_message);
657         }
658       g_free (worker->read_ancillary_messages);
659     }
660
661   if (bytes_read == -1)
662     {
663       if (G_UNLIKELY (_g_dbus_debug_transport ()))
664         {
665           _g_dbus_debug_print_lock ();
666           g_print ("========================================================================\n"
667                    "GDBus-debug:Transport:\n"
668                    "  ---- READ ERROR on stream of type %s:\n"
669                    "  ---- %s %d: %s\n",
670                    g_type_name (G_TYPE_FROM_INSTANCE (g_io_stream_get_input_stream (worker->stream))),
671                    g_quark_to_string (error->domain), error->code,
672                    error->message);
673           _g_dbus_debug_print_unlock ();
674         }
675
676       /* Every async read that uses this callback uses worker->cancellable
677        * as its GCancellable. worker->cancellable gets cancelled if and only
678        * if the GDBusConnection tells us to close (either via
679        * _g_dbus_worker_stop, which is called on last-unref, or directly),
680        * so a cancelled read must mean our connection was closed locally.
681        *
682        * If we're closing, other errors are possible - notably,
683        * G_IO_ERROR_CLOSED can be seen if we close the stream with an async
684        * read in-flight. It seems sensible to treat all read errors during
685        * closing as an expected thing that doesn't trip exit-on-close.
686        *
687        * Because close_expected can't be set until we get into the worker
688        * thread, but the cancellable is signalled sooner (from another
689        * thread), we do still need to check the error.
690        */
691       if (worker->close_expected ||
692           g_error_matches (error, G_IO_ERROR, G_IO_ERROR_CANCELLED))
693         _g_dbus_worker_emit_disconnected (worker, FALSE, NULL);
694       else
695         _g_dbus_worker_emit_disconnected (worker, TRUE, error);
696
697       g_error_free (error);
698       goto out;
699     }
700
701 #if 0
702   g_debug ("read %d bytes (is_closed=%d blocking=%d condition=0x%02x) stream %p, %p",
703            (gint) bytes_read,
704            g_socket_is_closed (g_socket_connection_get_socket (G_SOCKET_CONNECTION (worker->stream))),
705            g_socket_get_blocking (g_socket_connection_get_socket (G_SOCKET_CONNECTION (worker->stream))),
706            g_socket_condition_check (g_socket_connection_get_socket (G_SOCKET_CONNECTION (worker->stream)),
707                                      G_IO_IN | G_IO_OUT | G_IO_HUP),
708            worker->stream,
709            worker);
710 #endif
711
712   /* TODO: hmm, hmm... */
713   if (bytes_read == 0)
714     {
715       g_set_error (&error,
716                    G_IO_ERROR,
717                    G_IO_ERROR_FAILED,
718                    "Underlying GIOStream returned 0 bytes on an async read");
719       _g_dbus_worker_emit_disconnected (worker, TRUE, error);
720       g_error_free (error);
721       goto out;
722     }
723
724   read_message_print_transport_debug (bytes_read, worker);
725
726   worker->read_buffer_cur_size += bytes_read;
727   if (worker->read_buffer_bytes_wanted == worker->read_buffer_cur_size)
728     {
729       /* OK, got what we asked for! */
730       if (worker->read_buffer_bytes_wanted == 16)
731         {
732           gssize message_len;
733           /* OK, got the header - determine how many more bytes are needed */
734           error = NULL;
735           message_len = g_dbus_message_bytes_needed ((guchar *) worker->read_buffer,
736                                                      16,
737                                                      &error);
738           if (message_len == -1)
739             {
740               g_warning ("_g_dbus_worker_do_read_cb: error determining bytes needed: %s", error->message);
741               _g_dbus_worker_emit_disconnected (worker, FALSE, error);
742               g_error_free (error);
743               goto out;
744             }
745
746           worker->read_buffer_bytes_wanted = message_len;
747           _g_dbus_worker_do_read_unlocked (worker);
748         }
749       else
750         {
751           GDBusMessage *message;
752           error = NULL;
753
754           /* TODO: use connection->priv->auth to decode the message */
755
756           message = g_dbus_message_new_from_blob ((guchar *) worker->read_buffer,
757                                                   worker->read_buffer_cur_size,
758                                                   worker->capabilities,
759                                                   &error);
760           if (message == NULL)
761             {
762               gchar *s;
763               s = _g_dbus_hexdump (worker->read_buffer, worker->read_buffer_cur_size, 2);
764               g_warning ("Error decoding D-Bus message of %" G_GSIZE_FORMAT " bytes\n"
765                          "The error is: %s\n"
766                          "The payload is as follows:\n"
767                          "%s\n",
768                          worker->read_buffer_cur_size,
769                          error->message,
770                          s);
771               g_free (s);
772               _g_dbus_worker_emit_disconnected (worker, FALSE, error);
773               g_error_free (error);
774               goto out;
775             }
776
777 #ifdef G_OS_UNIX
778           if (worker->read_fd_list != NULL)
779             {
780               g_dbus_message_set_unix_fd_list (message, worker->read_fd_list);
781               g_object_unref (worker->read_fd_list);
782               worker->read_fd_list = NULL;
783             }
784 #endif
785
786           if (G_UNLIKELY (_g_dbus_debug_message ()))
787             {
788               gchar *s;
789               _g_dbus_debug_print_lock ();
790               g_print ("========================================================================\n"
791                        "GDBus-debug:Message:\n"
792                        "  <<<< RECEIVED D-Bus message (%" G_GSIZE_FORMAT " bytes)\n",
793                        worker->read_buffer_cur_size);
794               s = g_dbus_message_print (message, 2);
795               g_print ("%s", s);
796               g_free (s);
797               if (G_UNLIKELY (_g_dbus_debug_payload ()))
798                 {
799                   s = _g_dbus_hexdump (worker->read_buffer, worker->read_buffer_cur_size, 2);
800                   g_print ("%s\n", s);
801                   g_free (s);
802                 }
803               _g_dbus_debug_print_unlock ();
804             }
805
806           /* yay, got a message, go deliver it */
807           _g_dbus_worker_queue_or_deliver_received_message (worker, message);
808
809           /* start reading another message! */
810           worker->read_buffer_bytes_wanted = 0;
811           worker->read_buffer_cur_size = 0;
812           _g_dbus_worker_do_read_unlocked (worker);
813         }
814     }
815   else
816     {
817       /* didn't get all the bytes we requested - so repeat the request... */
818       _g_dbus_worker_do_read_unlocked (worker);
819     }
820
821  out:
822   g_mutex_unlock (&worker->read_lock);
823
824   /* gives up the reference acquired when calling g_input_stream_read_async() */
825   _g_dbus_worker_unref (worker);
826 }
827
828 /* called in private thread shared by all GDBusConnection instances (with read-lock held) */
829 static void
830 _g_dbus_worker_do_read_unlocked (GDBusWorker *worker)
831 {
832   /* Note that we do need to keep trying to read even if close_expected is
833    * true, because only failing a read causes us to signal 'closed'.
834    */
835
836   /* if bytes_wanted is zero, it means start reading a message */
837   if (worker->read_buffer_bytes_wanted == 0)
838     {
839       worker->read_buffer_cur_size = 0;
840       worker->read_buffer_bytes_wanted = 16;
841     }
842
843   /* ensure we have a (big enough) buffer */
844   if (worker->read_buffer == NULL || worker->read_buffer_bytes_wanted > worker->read_buffer_allocated_size)
845     {
846       /* TODO: 4096 is randomly chosen; might want a better chosen default minimum */
847       worker->read_buffer_allocated_size = MAX (worker->read_buffer_bytes_wanted, 4096);
848       worker->read_buffer = g_realloc (worker->read_buffer, worker->read_buffer_allocated_size);
849     }
850
851   if (worker->socket == NULL)
852     g_input_stream_read_async (g_io_stream_get_input_stream (worker->stream),
853                                worker->read_buffer + worker->read_buffer_cur_size,
854                                worker->read_buffer_bytes_wanted - worker->read_buffer_cur_size,
855                                G_PRIORITY_DEFAULT,
856                                worker->cancellable,
857                                (GAsyncReadyCallback) _g_dbus_worker_do_read_cb,
858                                _g_dbus_worker_ref (worker));
859   else
860     {
861       worker->read_ancillary_messages = NULL;
862       worker->read_num_ancillary_messages = 0;
863       _g_socket_read_with_control_messages (worker->socket,
864                                             worker->read_buffer + worker->read_buffer_cur_size,
865                                             worker->read_buffer_bytes_wanted - worker->read_buffer_cur_size,
866                                             &worker->read_ancillary_messages,
867                                             &worker->read_num_ancillary_messages,
868                                             G_PRIORITY_DEFAULT,
869                                             worker->cancellable,
870                                             (GAsyncReadyCallback) _g_dbus_worker_do_read_cb,
871                                             _g_dbus_worker_ref (worker));
872     }
873 }
874
875 /* called in private thread shared by all GDBusConnection instances (without read-lock held) */
876 static gboolean
877 _g_dbus_worker_do_initial_read (gpointer data)
878 {
879   GDBusWorker *worker = data;
880   g_mutex_lock (&worker->read_lock);
881   _g_dbus_worker_do_read_unlocked (worker);
882   g_mutex_unlock (&worker->read_lock);
883   return FALSE;
884 }
885
886 /* ---------------------------------------------------------------------------------------------------- */
887
888 struct _MessageToWriteData
889 {
890   GDBusWorker  *worker;
891   GDBusMessage *message;
892   gchar        *blob;
893   gsize         blob_size;
894
895   gsize               total_written;
896   GSimpleAsyncResult *simple;
897
898 };
899
900 static void
901 message_to_write_data_free (MessageToWriteData *data)
902 {
903   _g_dbus_worker_unref (data->worker);
904   if (data->message)
905     g_object_unref (data->message);
906   g_free (data->blob);
907   g_free (data);
908 }
909
910 /* ---------------------------------------------------------------------------------------------------- */
911
912 static void write_message_continue_writing (MessageToWriteData *data);
913
914 /* called in private thread shared by all GDBusConnection instances
915  *
916  * write-lock is not held on entry
917  * output_pending is PENDING_WRITE on entry
918  */
919 static void
920 write_message_async_cb (GObject      *source_object,
921                         GAsyncResult *res,
922                         gpointer      user_data)
923 {
924   MessageToWriteData *data = user_data;
925   GSimpleAsyncResult *simple;
926   gssize bytes_written;
927   GError *error;
928
929   /* Note: we can't access data->simple after calling g_async_result_complete () because the
930    * callback can free @data and we're not completing in idle. So use a copy of the pointer.
931    */
932   simple = data->simple;
933
934   error = NULL;
935   bytes_written = g_output_stream_write_finish (G_OUTPUT_STREAM (source_object),
936                                                 res,
937                                                 &error);
938   if (bytes_written == -1)
939     {
940       g_simple_async_result_take_error (simple, error);
941       g_simple_async_result_complete (simple);
942       g_object_unref (simple);
943       goto out;
944     }
945   g_assert (bytes_written > 0); /* zero is never returned */
946
947   write_message_print_transport_debug (bytes_written, data);
948
949   data->total_written += bytes_written;
950   g_assert (data->total_written <= data->blob_size);
951   if (data->total_written == data->blob_size)
952     {
953       g_simple_async_result_complete (simple);
954       g_object_unref (simple);
955       goto out;
956     }
957
958   write_message_continue_writing (data);
959
960  out:
961   ;
962 }
963
964 /* called in private thread shared by all GDBusConnection instances
965  *
966  * write-lock is not held on entry
967  * output_pending is PENDING_WRITE on entry
968  */
969 #ifdef G_OS_UNIX
970 static gboolean
971 on_socket_ready (GSocket      *socket,
972                  GIOCondition  condition,
973                  gpointer      user_data)
974 {
975   MessageToWriteData *data = user_data;
976   write_message_continue_writing (data);
977   return FALSE; /* remove source */
978 }
979 #endif
980
981 /* called in private thread shared by all GDBusConnection instances
982  *
983  * write-lock is not held on entry
984  * output_pending is PENDING_WRITE on entry
985  */
986 static void
987 write_message_continue_writing (MessageToWriteData *data)
988 {
989   GOutputStream *ostream;
990 #ifdef G_OS_UNIX
991   GSimpleAsyncResult *simple;
992   GUnixFDList *fd_list;
993 #endif
994
995 #ifdef G_OS_UNIX
996   /* Note: we can't access data->simple after calling g_async_result_complete () because the
997    * callback can free @data and we're not completing in idle. So use a copy of the pointer.
998    */
999   simple = data->simple;
1000 #endif
1001
1002   ostream = g_io_stream_get_output_stream (data->worker->stream);
1003 #ifdef G_OS_UNIX
1004   fd_list = g_dbus_message_get_unix_fd_list (data->message);
1005 #endif
1006
1007   g_assert (!g_output_stream_has_pending (ostream));
1008   g_assert_cmpint (data->total_written, <, data->blob_size);
1009
1010   if (FALSE)
1011     {
1012     }
1013 #ifdef G_OS_UNIX
1014   else if (G_IS_SOCKET_OUTPUT_STREAM (ostream) && data->total_written == 0)
1015     {
1016       GOutputVector vector;
1017       GSocketControlMessage *control_message;
1018       gssize bytes_written;
1019       GError *error;
1020
1021       vector.buffer = data->blob;
1022       vector.size = data->blob_size;
1023
1024       control_message = NULL;
1025       if (fd_list != NULL && g_unix_fd_list_get_length (fd_list) > 0)
1026         {
1027           if (!(data->worker->capabilities & G_DBUS_CAPABILITY_FLAGS_UNIX_FD_PASSING))
1028             {
1029               g_simple_async_result_set_error (simple,
1030                                                G_IO_ERROR,
1031                                                G_IO_ERROR_FAILED,
1032                                                "Tried sending a file descriptor but remote peer does not support this capability");
1033               g_simple_async_result_complete (simple);
1034               g_object_unref (simple);
1035               goto out;
1036             }
1037           control_message = g_unix_fd_message_new_with_fd_list (fd_list);
1038         }
1039
1040       error = NULL;
1041       bytes_written = g_socket_send_message (data->worker->socket,
1042                                              NULL, /* address */
1043                                              &vector,
1044                                              1,
1045                                              control_message != NULL ? &control_message : NULL,
1046                                              control_message != NULL ? 1 : 0,
1047                                              G_SOCKET_MSG_NONE,
1048                                              data->worker->cancellable,
1049                                              &error);
1050       if (control_message != NULL)
1051         g_object_unref (control_message);
1052
1053       if (bytes_written == -1)
1054         {
1055           /* Handle WOULD_BLOCK by waiting until there's room in the buffer */
1056           if (g_error_matches (error, G_IO_ERROR, G_IO_ERROR_WOULD_BLOCK))
1057             {
1058               GSource *source;
1059               source = g_socket_create_source (data->worker->socket,
1060                                                G_IO_OUT | G_IO_HUP | G_IO_ERR,
1061                                                data->worker->cancellable);
1062               g_source_set_callback (source,
1063                                      (GSourceFunc) on_socket_ready,
1064                                      data,
1065                                      NULL); /* GDestroyNotify */
1066               g_source_attach (source, g_main_context_get_thread_default ());
1067               g_source_unref (source);
1068               g_error_free (error);
1069               goto out;
1070             }
1071           g_simple_async_result_take_error (simple, error);
1072           g_simple_async_result_complete (simple);
1073           g_object_unref (simple);
1074           goto out;
1075         }
1076       g_assert (bytes_written > 0); /* zero is never returned */
1077
1078       write_message_print_transport_debug (bytes_written, data);
1079
1080       data->total_written += bytes_written;
1081       g_assert (data->total_written <= data->blob_size);
1082       if (data->total_written == data->blob_size)
1083         {
1084           g_simple_async_result_complete (simple);
1085           g_object_unref (simple);
1086           goto out;
1087         }
1088
1089       write_message_continue_writing (data);
1090     }
1091 #endif
1092   else
1093     {
1094 #ifdef G_OS_UNIX
1095       if (fd_list != NULL)
1096         {
1097           g_simple_async_result_set_error (simple,
1098                                            G_IO_ERROR,
1099                                            G_IO_ERROR_FAILED,
1100                                            "Tried sending a file descriptor on unsupported stream of type %s",
1101                                            g_type_name (G_TYPE_FROM_INSTANCE (ostream)));
1102           g_simple_async_result_complete (simple);
1103           g_object_unref (simple);
1104           goto out;
1105         }
1106 #endif
1107
1108       g_output_stream_write_async (ostream,
1109                                    (const gchar *) data->blob + data->total_written,
1110                                    data->blob_size - data->total_written,
1111                                    G_PRIORITY_DEFAULT,
1112                                    data->worker->cancellable,
1113                                    write_message_async_cb,
1114                                    data);
1115     }
1116 #ifdef G_OS_UNIX
1117  out:
1118 #endif
1119   ;
1120 }
1121
1122 /* called in private thread shared by all GDBusConnection instances
1123  *
1124  * write-lock is not held on entry
1125  * output_pending is PENDING_WRITE on entry
1126  */
1127 static void
1128 write_message_async (GDBusWorker         *worker,
1129                      MessageToWriteData  *data,
1130                      GAsyncReadyCallback  callback,
1131                      gpointer             user_data)
1132 {
1133   data->simple = g_simple_async_result_new (NULL,
1134                                             callback,
1135                                             user_data,
1136                                             write_message_async);
1137   data->total_written = 0;
1138   write_message_continue_writing (data);
1139 }
1140
1141 /* called in private thread shared by all GDBusConnection instances (with write-lock held) */
1142 static gboolean
1143 write_message_finish (GAsyncResult   *res,
1144                       GError        **error)
1145 {
1146   g_warn_if_fail (g_simple_async_result_get_source_tag (G_SIMPLE_ASYNC_RESULT (res)) == write_message_async);
1147   if (g_simple_async_result_propagate_error (G_SIMPLE_ASYNC_RESULT (res), error))
1148     return FALSE;
1149   else
1150     return TRUE;
1151 }
1152 /* ---------------------------------------------------------------------------------------------------- */
1153
1154 static void continue_writing (GDBusWorker *worker);
1155
1156 typedef struct
1157 {
1158   GDBusWorker *worker;
1159   GList *flushers;
1160 } FlushAsyncData;
1161
1162 static void
1163 flush_data_list_complete (const GList  *flushers,
1164                           const GError *error)
1165 {
1166   const GList *l;
1167
1168   for (l = flushers; l != NULL; l = l->next)
1169     {
1170       FlushData *f = l->data;
1171
1172       f->error = error != NULL ? g_error_copy (error) : NULL;
1173
1174       g_mutex_lock (&f->mutex);
1175       g_cond_signal (&f->cond);
1176       g_mutex_unlock (&f->mutex);
1177     }
1178 }
1179
1180 /* called in private thread shared by all GDBusConnection instances
1181  *
1182  * write-lock is not held on entry
1183  * output_pending is PENDING_FLUSH on entry
1184  */
1185 static void
1186 ostream_flush_cb (GObject      *source_object,
1187                   GAsyncResult *res,
1188                   gpointer      user_data)
1189 {
1190   FlushAsyncData *data = user_data;
1191   GError *error;
1192
1193   error = NULL;
1194   g_output_stream_flush_finish (G_OUTPUT_STREAM (source_object),
1195                                 res,
1196                                 &error);
1197
1198   if (error == NULL)
1199     {
1200       if (G_UNLIKELY (_g_dbus_debug_transport ()))
1201         {
1202           _g_dbus_debug_print_lock ();
1203           g_print ("========================================================================\n"
1204                    "GDBus-debug:Transport:\n"
1205                    "  ---- FLUSHED stream of type %s\n",
1206                    g_type_name (G_TYPE_FROM_INSTANCE (g_io_stream_get_output_stream (data->worker->stream))));
1207           _g_dbus_debug_print_unlock ();
1208         }
1209     }
1210
1211   g_assert (data->flushers != NULL);
1212   flush_data_list_complete (data->flushers, error);
1213   g_list_free (data->flushers);
1214
1215   if (error != NULL)
1216     g_error_free (error);
1217
1218   /* Make sure we tell folks that we don't have additional
1219      flushes pending */
1220   g_mutex_lock (&data->worker->write_lock);
1221   data->worker->write_num_messages_flushed = data->worker->write_num_messages_written;
1222   g_assert (data->worker->output_pending == PENDING_FLUSH);
1223   data->worker->output_pending = PENDING_NONE;
1224   g_mutex_unlock (&data->worker->write_lock);
1225
1226   /* OK, cool, finally kick off the next write */
1227   continue_writing (data->worker);
1228
1229   _g_dbus_worker_unref (data->worker);
1230   g_free (data);
1231 }
1232
1233 /* called in private thread shared by all GDBusConnection instances
1234  *
1235  * write-lock is not held on entry
1236  * output_pending is PENDING_FLUSH on entry
1237  */
1238 static void
1239 start_flush (FlushAsyncData *data)
1240 {
1241   g_output_stream_flush_async (g_io_stream_get_output_stream (data->worker->stream),
1242                                G_PRIORITY_DEFAULT,
1243                                data->worker->cancellable,
1244                                ostream_flush_cb,
1245                                data);
1246 }
1247
1248 /* called in private thread shared by all GDBusConnection instances
1249  *
1250  * write-lock is held on entry
1251  * output_pending is PENDING_NONE on entry
1252  */
1253 static void
1254 message_written_unlocked (GDBusWorker *worker,
1255                           MessageToWriteData *message_data)
1256 {
1257   if (G_UNLIKELY (_g_dbus_debug_message ()))
1258     {
1259       gchar *s;
1260       _g_dbus_debug_print_lock ();
1261       g_print ("========================================================================\n"
1262                "GDBus-debug:Message:\n"
1263                "  >>>> SENT D-Bus message (%" G_GSIZE_FORMAT " bytes)\n",
1264                message_data->blob_size);
1265       s = g_dbus_message_print (message_data->message, 2);
1266       g_print ("%s", s);
1267       g_free (s);
1268       if (G_UNLIKELY (_g_dbus_debug_payload ()))
1269         {
1270           s = _g_dbus_hexdump (message_data->blob, message_data->blob_size, 2);
1271           g_print ("%s\n", s);
1272           g_free (s);
1273         }
1274       _g_dbus_debug_print_unlock ();
1275     }
1276
1277   worker->write_num_messages_written += 1;
1278 }
1279
1280 /* called in private thread shared by all GDBusConnection instances
1281  *
1282  * write-lock is held on entry
1283  * output_pending is PENDING_NONE on entry
1284  *
1285  * Returns: non-%NULL, setting @output_pending, if we need to flush now
1286  */
1287 static FlushAsyncData *
1288 prepare_flush_unlocked (GDBusWorker *worker)
1289 {
1290   GList *l;
1291   GList *ll;
1292   GList *flushers;
1293
1294   flushers = NULL;
1295   for (l = worker->write_pending_flushes; l != NULL; l = ll)
1296     {
1297       FlushData *f = l->data;
1298       ll = l->next;
1299
1300       if (f->number_to_wait_for == worker->write_num_messages_written)
1301         {
1302           flushers = g_list_append (flushers, f);
1303           worker->write_pending_flushes = g_list_delete_link (worker->write_pending_flushes, l);
1304         }
1305     }
1306   if (flushers != NULL)
1307     {
1308       g_assert (worker->output_pending == PENDING_NONE);
1309       worker->output_pending = PENDING_FLUSH;
1310     }
1311
1312   if (flushers != NULL)
1313     {
1314       FlushAsyncData *data;
1315
1316       data = g_new0 (FlushAsyncData, 1);
1317       data->worker = _g_dbus_worker_ref (worker);
1318       data->flushers = flushers;
1319       return data;
1320     }
1321
1322   return NULL;
1323 }
1324
1325 /* called in private thread shared by all GDBusConnection instances
1326  *
1327  * write-lock is not held on entry
1328  * output_pending is PENDING_WRITE on entry
1329  */
1330 static void
1331 write_message_cb (GObject       *source_object,
1332                   GAsyncResult  *res,
1333                   gpointer       user_data)
1334 {
1335   MessageToWriteData *data = user_data;
1336   GError *error;
1337
1338   g_mutex_lock (&data->worker->write_lock);
1339   g_assert (data->worker->output_pending == PENDING_WRITE);
1340   data->worker->output_pending = PENDING_NONE;
1341
1342   error = NULL;
1343   if (!write_message_finish (res, &error))
1344     {
1345       g_mutex_unlock (&data->worker->write_lock);
1346
1347       /* TODO: handle */
1348       _g_dbus_worker_emit_disconnected (data->worker, TRUE, error);
1349       g_error_free (error);
1350
1351       g_mutex_lock (&data->worker->write_lock);
1352     }
1353
1354   message_written_unlocked (data->worker, data);
1355
1356   g_mutex_unlock (&data->worker->write_lock);
1357
1358   continue_writing (data->worker);
1359
1360   message_to_write_data_free (data);
1361 }
1362
1363 /* called in private thread shared by all GDBusConnection instances
1364  *
1365  * write-lock is not held on entry
1366  * output_pending is PENDING_CLOSE on entry
1367  */
1368 static void
1369 iostream_close_cb (GObject      *source_object,
1370                    GAsyncResult *res,
1371                    gpointer      user_data)
1372 {
1373   GDBusWorker *worker = user_data;
1374   GError *error = NULL;
1375   GList *pending_close_attempts, *pending_flush_attempts;
1376   GQueue *send_queue;
1377
1378   g_io_stream_close_finish (worker->stream, res, &error);
1379
1380   g_mutex_lock (&worker->write_lock);
1381
1382   pending_close_attempts = worker->pending_close_attempts;
1383   worker->pending_close_attempts = NULL;
1384
1385   pending_flush_attempts = worker->write_pending_flushes;
1386   worker->write_pending_flushes = NULL;
1387
1388   send_queue = worker->write_queue;
1389   worker->write_queue = g_queue_new ();
1390
1391   g_assert (worker->output_pending == PENDING_CLOSE);
1392   worker->output_pending = PENDING_NONE;
1393
1394   g_mutex_unlock (&worker->write_lock);
1395
1396   while (pending_close_attempts != NULL)
1397     {
1398       CloseData *close_data = pending_close_attempts->data;
1399
1400       pending_close_attempts = g_list_delete_link (pending_close_attempts,
1401                                                    pending_close_attempts);
1402
1403       if (close_data->result != NULL)
1404         {
1405           if (error != NULL)
1406             g_simple_async_result_set_from_error (close_data->result, error);
1407
1408           /* this must be in an idle because the result is likely to be
1409            * intended for another thread
1410            */
1411           g_simple_async_result_complete_in_idle (close_data->result);
1412         }
1413
1414       close_data_free (close_data);
1415     }
1416
1417   g_clear_error (&error);
1418
1419   /* all messages queued for sending are discarded */
1420   g_queue_free_full (send_queue, (GDestroyNotify) message_to_write_data_free);
1421   /* all queued flushes fail */
1422   error = g_error_new (G_IO_ERROR, G_IO_ERROR_CANCELLED,
1423                        _("Operation was cancelled"));
1424   flush_data_list_complete (pending_flush_attempts, error);
1425   g_list_free (pending_flush_attempts);
1426   g_clear_error (&error);
1427
1428   _g_dbus_worker_unref (worker);
1429 }
1430
1431 /* called in private thread shared by all GDBusConnection instances
1432  *
1433  * write-lock is not held on entry
1434  * output_pending must be PENDING_NONE on entry
1435  */
1436 static void
1437 continue_writing (GDBusWorker *worker)
1438 {
1439   MessageToWriteData *data;
1440   FlushAsyncData *flush_async_data;
1441
1442  write_next:
1443   /* we mustn't try to write two things at once */
1444   g_assert (worker->output_pending == PENDING_NONE);
1445
1446   g_mutex_lock (&worker->write_lock);
1447
1448   data = NULL;
1449   flush_async_data = NULL;
1450
1451   /* if we want to close the connection, that takes precedence */
1452   if (worker->pending_close_attempts != NULL)
1453     {
1454       worker->close_expected = TRUE;
1455       worker->output_pending = PENDING_CLOSE;
1456
1457       g_io_stream_close_async (worker->stream, G_PRIORITY_DEFAULT,
1458                                NULL, iostream_close_cb,
1459                                _g_dbus_worker_ref (worker));
1460     }
1461   else
1462     {
1463       flush_async_data = prepare_flush_unlocked (worker);
1464
1465       if (flush_async_data == NULL)
1466         {
1467           data = g_queue_pop_head (worker->write_queue);
1468
1469           if (data != NULL)
1470             worker->output_pending = PENDING_WRITE;
1471         }
1472     }
1473
1474   g_mutex_unlock (&worker->write_lock);
1475
1476   /* Note that write_lock is only used for protecting the @write_queue
1477    * and @output_pending fields of the GDBusWorker struct ... which we
1478    * need to modify from arbitrary threads in _g_dbus_worker_send_message().
1479    *
1480    * Therefore, it's fine to drop it here when calling back into user
1481    * code and then writing the message out onto the GIOStream since this
1482    * function only runs on the worker thread.
1483    */
1484
1485   if (flush_async_data != NULL)
1486     {
1487       start_flush (flush_async_data);
1488       g_assert (data == NULL);
1489     }
1490   else if (data != NULL)
1491     {
1492       GDBusMessage *old_message;
1493       guchar *new_blob;
1494       gsize new_blob_size;
1495       GError *error;
1496
1497       old_message = data->message;
1498       data->message = _g_dbus_worker_emit_message_about_to_be_sent (worker, data->message);
1499       if (data->message == old_message)
1500         {
1501           /* filters had no effect - do nothing */
1502         }
1503       else if (data->message == NULL)
1504         {
1505           /* filters dropped message */
1506           g_mutex_lock (&worker->write_lock);
1507           worker->output_pending = PENDING_NONE;
1508           g_mutex_unlock (&worker->write_lock);
1509           message_to_write_data_free (data);
1510           goto write_next;
1511         }
1512       else
1513         {
1514           /* filters altered the message -> reencode */
1515           error = NULL;
1516           new_blob = g_dbus_message_to_blob (data->message,
1517                                              &new_blob_size,
1518                                              worker->capabilities,
1519                                              &error);
1520           if (new_blob == NULL)
1521             {
1522               /* if filter make the GDBusMessage unencodeable, just complain on stderr and send
1523                * the old message instead
1524                */
1525               g_warning ("Error encoding GDBusMessage with serial %d altered by filter function: %s",
1526                          g_dbus_message_get_serial (data->message),
1527                          error->message);
1528               g_error_free (error);
1529             }
1530           else
1531             {
1532               g_free (data->blob);
1533               data->blob = (gchar *) new_blob;
1534               data->blob_size = new_blob_size;
1535             }
1536         }
1537
1538       write_message_async (worker,
1539                            data,
1540                            write_message_cb,
1541                            data);
1542     }
1543 }
1544
1545 /* called in private thread shared by all GDBusConnection instances
1546  *
1547  * write-lock is not held on entry
1548  * output_pending may be anything
1549  */
1550 static gboolean
1551 continue_writing_in_idle_cb (gpointer user_data)
1552 {
1553   GDBusWorker *worker = user_data;
1554
1555   /* Because this is the worker thread, we can read this struct member
1556    * without holding the lock: no other thread ever modifies it.
1557    */
1558   if (worker->output_pending == PENDING_NONE)
1559     continue_writing (worker);
1560
1561   return FALSE;
1562 }
1563
1564 /*
1565  * @write_data: (transfer full) (allow-none):
1566  * @flush_data: (transfer full) (allow-none):
1567  * @close_data: (transfer full) (allow-none):
1568  *
1569  * Can be called from any thread
1570  *
1571  * write_lock is held on entry
1572  * output_pending may be anything
1573  */
1574 static void
1575 schedule_writing_unlocked (GDBusWorker        *worker,
1576                            MessageToWriteData *write_data,
1577                            FlushData          *flush_data,
1578                            CloseData          *close_data)
1579 {
1580   if (write_data != NULL)
1581     g_queue_push_tail (worker->write_queue, write_data);
1582
1583   if (flush_data != NULL)
1584     worker->write_pending_flushes = g_list_prepend (worker->write_pending_flushes, flush_data);
1585
1586   if (close_data != NULL)
1587     worker->pending_close_attempts = g_list_prepend (worker->pending_close_attempts,
1588                                                      close_data);
1589
1590   /* If we had output pending, the next bit of output will happen
1591    * automatically when it finishes, so we only need to do this
1592    * if nothing was pending.
1593    *
1594    * The idle callback will re-check that output_pending is still
1595    * PENDING_NONE, to guard against output starting before the idle.
1596    */
1597   if (worker->output_pending == PENDING_NONE)
1598     {
1599       GSource *idle_source;
1600       idle_source = g_idle_source_new ();
1601       g_source_set_priority (idle_source, G_PRIORITY_DEFAULT);
1602       g_source_set_callback (idle_source,
1603                              continue_writing_in_idle_cb,
1604                              _g_dbus_worker_ref (worker),
1605                              (GDestroyNotify) _g_dbus_worker_unref);
1606       g_source_attach (idle_source, worker->shared_thread_data->context);
1607       g_source_unref (idle_source);
1608     }
1609 }
1610
1611 /* ---------------------------------------------------------------------------------------------------- */
1612
1613 /* can be called from any thread - steals blob
1614  *
1615  * write_lock is not held on entry
1616  * output_pending may be anything
1617  */
1618 void
1619 _g_dbus_worker_send_message (GDBusWorker    *worker,
1620                              GDBusMessage   *message,
1621                              gchar          *blob,
1622                              gsize           blob_len)
1623 {
1624   MessageToWriteData *data;
1625
1626   g_return_if_fail (G_IS_DBUS_MESSAGE (message));
1627   g_return_if_fail (blob != NULL);
1628   g_return_if_fail (blob_len > 16);
1629
1630   data = g_new0 (MessageToWriteData, 1);
1631   data->worker = _g_dbus_worker_ref (worker);
1632   data->message = g_object_ref (message);
1633   data->blob = blob; /* steal! */
1634   data->blob_size = blob_len;
1635
1636   g_mutex_lock (&worker->write_lock);
1637   schedule_writing_unlocked (worker, data, NULL, NULL);
1638   g_mutex_unlock (&worker->write_lock);
1639 }
1640
1641 /* ---------------------------------------------------------------------------------------------------- */
1642
1643 GDBusWorker *
1644 _g_dbus_worker_new (GIOStream                              *stream,
1645                     GDBusCapabilityFlags                    capabilities,
1646                     gboolean                                initially_frozen,
1647                     GDBusWorkerMessageReceivedCallback      message_received_callback,
1648                     GDBusWorkerMessageAboutToBeSentCallback message_about_to_be_sent_callback,
1649                     GDBusWorkerDisconnectedCallback         disconnected_callback,
1650                     gpointer                                user_data)
1651 {
1652   GDBusWorker *worker;
1653   GSource *idle_source;
1654
1655   g_return_val_if_fail (G_IS_IO_STREAM (stream), NULL);
1656   g_return_val_if_fail (message_received_callback != NULL, NULL);
1657   g_return_val_if_fail (message_about_to_be_sent_callback != NULL, NULL);
1658   g_return_val_if_fail (disconnected_callback != NULL, NULL);
1659
1660   worker = g_new0 (GDBusWorker, 1);
1661   worker->ref_count = 1;
1662
1663   g_mutex_init (&worker->read_lock);
1664   worker->message_received_callback = message_received_callback;
1665   worker->message_about_to_be_sent_callback = message_about_to_be_sent_callback;
1666   worker->disconnected_callback = disconnected_callback;
1667   worker->user_data = user_data;
1668   worker->stream = g_object_ref (stream);
1669   worker->capabilities = capabilities;
1670   worker->cancellable = g_cancellable_new ();
1671   worker->output_pending = PENDING_NONE;
1672
1673   worker->frozen = initially_frozen;
1674   worker->received_messages_while_frozen = g_queue_new ();
1675
1676   g_mutex_init (&worker->write_lock);
1677   worker->write_queue = g_queue_new ();
1678
1679   if (G_IS_SOCKET_CONNECTION (worker->stream))
1680     worker->socket = g_socket_connection_get_socket (G_SOCKET_CONNECTION (worker->stream));
1681
1682   if (G_IS_KDBUS_CONNECTION (worker->stream))
1683     worker->kdbus = g_kdbus_connection_get_kdbus (G_KDBUS_CONNECTION (worker->stream));
1684
1685   worker->shared_thread_data = _g_dbus_shared_thread_ref ();
1686
1687   /* begin reading */
1688   idle_source = g_idle_source_new ();
1689   g_source_set_priority (idle_source, G_PRIORITY_DEFAULT);
1690   g_source_set_callback (idle_source,
1691                          _g_dbus_worker_do_initial_read,
1692                          _g_dbus_worker_ref (worker),
1693                          (GDestroyNotify) _g_dbus_worker_unref);
1694   g_source_attach (idle_source, worker->shared_thread_data->context);
1695   g_source_unref (idle_source);
1696
1697   return worker;
1698 }
1699
1700 /* ---------------------------------------------------------------------------------------------------- */
1701
1702 /* can be called from any thread
1703  *
1704  * write_lock is not held on entry
1705  * output_pending may be anything
1706  */
1707 void
1708 _g_dbus_worker_close (GDBusWorker         *worker,
1709                       GCancellable        *cancellable,
1710                       GSimpleAsyncResult  *result)
1711 {
1712   CloseData *close_data;
1713
1714   close_data = g_slice_new0 (CloseData);
1715   close_data->worker = _g_dbus_worker_ref (worker);
1716   close_data->cancellable =
1717       (cancellable == NULL ? NULL : g_object_ref (cancellable));
1718   close_data->result = (result == NULL ? NULL : g_object_ref (result));
1719
1720   /* Don't set worker->close_expected here - we're in the wrong thread.
1721    * It'll be set before the actual close happens.
1722    */
1723   g_cancellable_cancel (worker->cancellable);
1724   g_mutex_lock (&worker->write_lock);
1725   schedule_writing_unlocked (worker, NULL, NULL, close_data);
1726   g_mutex_unlock (&worker->write_lock);
1727 }
1728
1729 /* This can be called from any thread - frees worker. Note that
1730  * callbacks might still happen if called from another thread than the
1731  * worker - use your own synchronization primitive in the callbacks.
1732  *
1733  * write_lock is not held on entry
1734  * output_pending may be anything
1735  */
1736 void
1737 _g_dbus_worker_stop (GDBusWorker *worker)
1738 {
1739   g_atomic_int_set (&worker->stopped, TRUE);
1740
1741   /* Cancel any pending operations and schedule a close of the underlying I/O
1742    * stream in the worker thread
1743    */
1744   _g_dbus_worker_close (worker, NULL, NULL);
1745
1746   /* _g_dbus_worker_close holds a ref until after an idle in the worker
1747    * thread has run, so we no longer need to unref in an idle like in
1748    * commit 322e25b535
1749    */
1750   _g_dbus_worker_unref (worker);
1751 }
1752
1753 /* ---------------------------------------------------------------------------------------------------- */
1754
1755 /* can be called from any thread (except the worker thread) - blocks
1756  * calling thread until all queued outgoing messages are written and
1757  * the transport has been flushed
1758  *
1759  * write_lock is not held on entry
1760  * output_pending may be anything
1761  */
1762 gboolean
1763 _g_dbus_worker_flush_sync (GDBusWorker    *worker,
1764                            GCancellable   *cancellable,
1765                            GError        **error)
1766 {
1767   gboolean ret;
1768   FlushData *data;
1769   guint64 pending_writes;
1770
1771   data = NULL;
1772   ret = TRUE;
1773
1774   g_mutex_lock (&worker->write_lock);
1775
1776   /* if the queue is empty, no write is in-flight and we haven't written
1777    * anything since the last flush, then there's nothing to wait for
1778    */
1779   pending_writes = g_queue_get_length (worker->write_queue);
1780
1781   /* if a write is in-flight, we shouldn't be satisfied until the first
1782    * flush operation that follows it
1783    */
1784   if (worker->output_pending == PENDING_WRITE)
1785     pending_writes += 1;
1786
1787   if (pending_writes > 0 ||
1788       worker->write_num_messages_written != worker->write_num_messages_flushed)
1789     {
1790       data = g_new0 (FlushData, 1);
1791       g_mutex_init (&data->mutex);
1792       g_cond_init (&data->cond);
1793       data->number_to_wait_for = worker->write_num_messages_written + pending_writes;
1794       g_mutex_lock (&data->mutex);
1795
1796       schedule_writing_unlocked (worker, NULL, data, NULL);
1797     }
1798   g_mutex_unlock (&worker->write_lock);
1799
1800   if (data != NULL)
1801     {
1802       g_cond_wait (&data->cond, &data->mutex);
1803       g_mutex_unlock (&data->mutex);
1804
1805       /* note:the element is removed from worker->write_pending_flushes in flush_cb() above */
1806       g_cond_clear (&data->cond);
1807       g_mutex_clear (&data->mutex);
1808       if (data->error != NULL)
1809         {
1810           ret = FALSE;
1811           g_propagate_error (error, data->error);
1812         }
1813       g_free (data);
1814     }
1815
1816   return ret;
1817 }
1818
1819 /* ---------------------------------------------------------------------------------------------------- */
1820
1821 #define G_DBUS_DEBUG_AUTHENTICATION (1<<0)
1822 #define G_DBUS_DEBUG_TRANSPORT      (1<<1)
1823 #define G_DBUS_DEBUG_MESSAGE        (1<<2)
1824 #define G_DBUS_DEBUG_PAYLOAD        (1<<3)
1825 #define G_DBUS_DEBUG_CALL           (1<<4)
1826 #define G_DBUS_DEBUG_SIGNAL         (1<<5)
1827 #define G_DBUS_DEBUG_INCOMING       (1<<6)
1828 #define G_DBUS_DEBUG_RETURN         (1<<7)
1829 #define G_DBUS_DEBUG_EMISSION       (1<<8)
1830 #define G_DBUS_DEBUG_ADDRESS        (1<<9)
1831
1832 static gint _gdbus_debug_flags = 0;
1833
1834 gboolean
1835 _g_dbus_debug_authentication (void)
1836 {
1837   _g_dbus_initialize ();
1838   return (_gdbus_debug_flags & G_DBUS_DEBUG_AUTHENTICATION) != 0;
1839 }
1840
1841 gboolean
1842 _g_dbus_debug_transport (void)
1843 {
1844   _g_dbus_initialize ();
1845   return (_gdbus_debug_flags & G_DBUS_DEBUG_TRANSPORT) != 0;
1846 }
1847
1848 gboolean
1849 _g_dbus_debug_message (void)
1850 {
1851   _g_dbus_initialize ();
1852   return (_gdbus_debug_flags & G_DBUS_DEBUG_MESSAGE) != 0;
1853 }
1854
1855 gboolean
1856 _g_dbus_debug_payload (void)
1857 {
1858   _g_dbus_initialize ();
1859   return (_gdbus_debug_flags & G_DBUS_DEBUG_PAYLOAD) != 0;
1860 }
1861
1862 gboolean
1863 _g_dbus_debug_call (void)
1864 {
1865   _g_dbus_initialize ();
1866   return (_gdbus_debug_flags & G_DBUS_DEBUG_CALL) != 0;
1867 }
1868
1869 gboolean
1870 _g_dbus_debug_signal (void)
1871 {
1872   _g_dbus_initialize ();
1873   return (_gdbus_debug_flags & G_DBUS_DEBUG_SIGNAL) != 0;
1874 }
1875
1876 gboolean
1877 _g_dbus_debug_incoming (void)
1878 {
1879   _g_dbus_initialize ();
1880   return (_gdbus_debug_flags & G_DBUS_DEBUG_INCOMING) != 0;
1881 }
1882
1883 gboolean
1884 _g_dbus_debug_return (void)
1885 {
1886   _g_dbus_initialize ();
1887   return (_gdbus_debug_flags & G_DBUS_DEBUG_RETURN) != 0;
1888 }
1889
1890 gboolean
1891 _g_dbus_debug_emission (void)
1892 {
1893   _g_dbus_initialize ();
1894   return (_gdbus_debug_flags & G_DBUS_DEBUG_EMISSION) != 0;
1895 }
1896
1897 gboolean
1898 _g_dbus_debug_address (void)
1899 {
1900   _g_dbus_initialize ();
1901   return (_gdbus_debug_flags & G_DBUS_DEBUG_ADDRESS) != 0;
1902 }
1903
1904 G_LOCK_DEFINE_STATIC (print_lock);
1905
1906 void
1907 _g_dbus_debug_print_lock (void)
1908 {
1909   G_LOCK (print_lock);
1910 }
1911
1912 void
1913 _g_dbus_debug_print_unlock (void)
1914 {
1915   G_UNLOCK (print_lock);
1916 }
1917
1918 /*
1919  * _g_dbus_initialize:
1920  *
1921  * Does various one-time init things such as
1922  *
1923  *  - registering the G_DBUS_ERROR error domain
1924  *  - parses the G_DBUS_DEBUG environment variable
1925  */
1926 void
1927 _g_dbus_initialize (void)
1928 {
1929   static volatile gsize initialized = 0;
1930
1931   if (g_once_init_enter (&initialized))
1932     {
1933       volatile GQuark g_dbus_error_domain;
1934       const gchar *debug;
1935
1936       g_dbus_error_domain = G_DBUS_ERROR;
1937       (g_dbus_error_domain); /* To avoid -Wunused-but-set-variable */
1938
1939       debug = g_getenv ("G_DBUS_DEBUG");
1940       if (debug != NULL)
1941         {
1942           const GDebugKey keys[] = {
1943             { "authentication", G_DBUS_DEBUG_AUTHENTICATION },
1944             { "transport",      G_DBUS_DEBUG_TRANSPORT      },
1945             { "message",        G_DBUS_DEBUG_MESSAGE        },
1946             { "payload",        G_DBUS_DEBUG_PAYLOAD        },
1947             { "call",           G_DBUS_DEBUG_CALL           },
1948             { "signal",         G_DBUS_DEBUG_SIGNAL         },
1949             { "incoming",       G_DBUS_DEBUG_INCOMING       },
1950             { "return",         G_DBUS_DEBUG_RETURN         },
1951             { "emission",       G_DBUS_DEBUG_EMISSION       },
1952             { "address",        G_DBUS_DEBUG_ADDRESS        }
1953           };
1954
1955           _gdbus_debug_flags = g_parse_debug_string (debug, keys, G_N_ELEMENTS (keys));
1956           if (_gdbus_debug_flags & G_DBUS_DEBUG_PAYLOAD)
1957             _gdbus_debug_flags |= G_DBUS_DEBUG_MESSAGE;
1958         }
1959
1960       g_once_init_leave (&initialized, 1);
1961     }
1962 }
1963
1964 /* ---------------------------------------------------------------------------------------------------- */
1965
1966 GVariantType *
1967 _g_dbus_compute_complete_signature (GDBusArgInfo **args)
1968 {
1969   const GVariantType *arg_types[256];
1970   guint n;
1971
1972   if (args)
1973     for (n = 0; args[n] != NULL; n++)
1974       {
1975         /* DBus places a hard limit of 255 on signature length.
1976          * therefore number of args must be less than 256.
1977          */
1978         g_assert (n < 256);
1979
1980         arg_types[n] = G_VARIANT_TYPE (args[n]->signature);
1981
1982         if G_UNLIKELY (arg_types[n] == NULL)
1983           return NULL;
1984       }
1985   else
1986     n = 0;
1987
1988   return g_variant_type_new_tuple (arg_types, n);
1989 }
1990
1991 /* ---------------------------------------------------------------------------------------------------- */
1992
1993 #ifdef G_OS_WIN32
1994
1995 extern BOOL WINAPI ConvertSidToStringSidA (PSID Sid, LPSTR *StringSid);
1996
1997 gchar *
1998 _g_dbus_win32_get_user_sid (void)
1999 {
2000   HANDLE h;
2001   TOKEN_USER *user;
2002   DWORD token_information_len;
2003   PSID psid;
2004   gchar *sid;
2005   gchar *ret;
2006
2007   ret = NULL;
2008   user = NULL;
2009   h = INVALID_HANDLE_VALUE;
2010
2011   if (!OpenProcessToken (GetCurrentProcess (), TOKEN_QUERY, &h))
2012     {
2013       g_warning ("OpenProcessToken failed with error code %d", (gint) GetLastError ());
2014       goto out;
2015     }
2016
2017   /* Get length of buffer */
2018   token_information_len = 0;
2019   if (!GetTokenInformation (h, TokenUser, NULL, 0, &token_information_len))
2020     {
2021       if (GetLastError () != ERROR_INSUFFICIENT_BUFFER)
2022         {
2023           g_warning ("GetTokenInformation() failed with error code %d", (gint) GetLastError ());
2024           goto out;
2025         }
2026     }
2027   user = g_malloc (token_information_len);
2028   if (!GetTokenInformation (h, TokenUser, user, token_information_len, &token_information_len))
2029     {
2030       g_warning ("GetTokenInformation() failed with error code %d", (gint) GetLastError ());
2031       goto out;
2032     }
2033
2034   psid = user->User.Sid;
2035   if (!IsValidSid (psid))
2036     {
2037       g_warning ("Invalid SID");
2038       goto out;
2039     }
2040
2041   if (!ConvertSidToStringSidA (psid, &sid))
2042     {
2043       g_warning ("Invalid SID");
2044       goto out;
2045     }
2046
2047   ret = g_strdup (sid);
2048   LocalFree (sid);
2049
2050 out:
2051   g_free (user);
2052   if (h != INVALID_HANDLE_VALUE)
2053     CloseHandle (h);
2054   return ret;
2055 }
2056 #endif
2057
2058 /* ---------------------------------------------------------------------------------------------------- */
2059
2060 gchar *
2061 _g_dbus_get_machine_id (GError **error)
2062 {
2063 #ifdef G_OS_WIN32
2064   HW_PROFILE_INFOA info;
2065   char *src, *dest, *res;
2066   int i;
2067
2068   if (!GetCurrentHwProfileA (&info))
2069     {
2070       char *message = g_win32_error_message (GetLastError ());
2071       g_set_error (error,
2072                    G_IO_ERROR,
2073                    G_IO_ERROR_FAILED,
2074                    _("Unable to get Hardware profile: %s"), message);
2075       g_free (message);
2076       return NULL;
2077     }
2078
2079   /* Form: {12340001-4980-1920-6788-123456789012} */
2080   src = &info.szHwProfileGuid[0];
2081
2082   res = g_malloc (32+1);
2083   dest = res;
2084
2085   src++; /* Skip { */
2086   for (i = 0; i < 8; i++)
2087     *dest++ = *src++;
2088   src++; /* Skip - */
2089   for (i = 0; i < 4; i++)
2090     *dest++ = *src++;
2091   src++; /* Skip - */
2092   for (i = 0; i < 4; i++)
2093     *dest++ = *src++;
2094   src++; /* Skip - */
2095   for (i = 0; i < 4; i++)
2096     *dest++ = *src++;
2097   src++; /* Skip - */
2098   for (i = 0; i < 12; i++)
2099     *dest++ = *src++;
2100   *dest = 0;
2101
2102   return res;
2103 #else
2104   gchar *ret;
2105   GError *first_error;
2106   /* TODO: use PACKAGE_LOCALSTATEDIR ? */
2107   ret = NULL;
2108   first_error = NULL;
2109   if (!g_file_get_contents ("/var/lib/dbus/machine-id",
2110                             &ret,
2111                             NULL,
2112                             &first_error) &&
2113       !g_file_get_contents ("/etc/machine-id",
2114                             &ret,
2115                             NULL,
2116                             NULL))
2117     {
2118       g_propagate_prefixed_error (error, first_error,
2119                                   _("Unable to load /var/lib/dbus/machine-id or /etc/machine-id: "));
2120     }
2121   else
2122     {
2123       /* ignore the error from the first try, if any */
2124       g_clear_error (&first_error);
2125       /* TODO: validate value */
2126       g_strstrip (ret);
2127     }
2128   return ret;
2129 #endif
2130 }
2131
2132 /* ---------------------------------------------------------------------------------------------------- */
2133
2134 gchar *
2135 _g_dbus_enum_to_string (GType enum_type, gint value)
2136 {
2137   gchar *ret;
2138   GEnumClass *klass;
2139   GEnumValue *enum_value;
2140
2141   klass = g_type_class_ref (enum_type);
2142   enum_value = g_enum_get_value (klass, value);
2143   if (enum_value != NULL)
2144     ret = g_strdup (enum_value->value_nick);
2145   else
2146     ret = g_strdup_printf ("unknown (value %d)", value);
2147   g_type_class_unref (klass);
2148   return ret;
2149 }
2150
2151 /* ---------------------------------------------------------------------------------------------------- */
2152
2153 static void
2154 write_message_print_transport_debug (gssize bytes_written,
2155                                      MessageToWriteData *data)
2156 {
2157   if (G_LIKELY (!_g_dbus_debug_transport ()))
2158     goto out;
2159
2160   _g_dbus_debug_print_lock ();
2161   g_print ("========================================================================\n"
2162            "GDBus-debug:Transport:\n"
2163            "  >>>> WROTE %" G_GSIZE_FORMAT " bytes of message with serial %d and\n"
2164            "       size %" G_GSIZE_FORMAT " from offset %" G_GSIZE_FORMAT " on a %s\n",
2165            bytes_written,
2166            g_dbus_message_get_serial (data->message),
2167            data->blob_size,
2168            data->total_written,
2169            g_type_name (G_TYPE_FROM_INSTANCE (g_io_stream_get_output_stream (data->worker->stream))));
2170   _g_dbus_debug_print_unlock ();
2171  out:
2172   ;
2173 }
2174
2175 /* ---------------------------------------------------------------------------------------------------- */
2176
2177 static void
2178 read_message_print_transport_debug (gssize bytes_read,
2179                                     GDBusWorker *worker)
2180 {
2181   gsize size;
2182   gint32 serial;
2183   gint32 message_length;
2184
2185   if (G_LIKELY (!_g_dbus_debug_transport ()))
2186     goto out;
2187
2188   size = bytes_read + worker->read_buffer_cur_size;
2189   serial = 0;
2190   message_length = 0;
2191   if (size >= 16)
2192     message_length = g_dbus_message_bytes_needed ((guchar *) worker->read_buffer, size, NULL);
2193   if (size >= 1)
2194     {
2195       switch (worker->read_buffer[0])
2196         {
2197         case 'l':
2198           if (size >= 12)
2199             serial = GUINT32_FROM_LE (((guint32 *) worker->read_buffer)[2]);
2200           break;
2201         case 'B':
2202           if (size >= 12)
2203             serial = GUINT32_FROM_BE (((guint32 *) worker->read_buffer)[2]);
2204           break;
2205         default:
2206           /* an error will be set elsewhere if this happens */
2207           goto out;
2208         }
2209     }
2210
2211     _g_dbus_debug_print_lock ();
2212   g_print ("========================================================================\n"
2213            "GDBus-debug:Transport:\n"
2214            "  <<<< READ %" G_GSIZE_FORMAT " bytes of message with serial %d and\n"
2215            "       size %d to offset %" G_GSIZE_FORMAT " from a %s\n",
2216            bytes_read,
2217            serial,
2218            message_length,
2219            worker->read_buffer_cur_size,
2220            g_type_name (G_TYPE_FROM_INSTANCE (g_io_stream_get_input_stream (worker->stream))));
2221   _g_dbus_debug_print_unlock ();
2222  out:
2223   ;
2224 }
2225
2226 /* ---------------------------------------------------------------------------------------------------- */
2227
2228 gboolean
2229 _g_signal_accumulator_false_handled (GSignalInvocationHint *ihint,
2230                                      GValue                *return_accu,
2231                                      const GValue          *handler_return,
2232                                      gpointer               dummy)
2233 {
2234   gboolean continue_emission;
2235   gboolean signal_return;
2236
2237   signal_return = g_value_get_boolean (handler_return);
2238   g_value_set_boolean (return_accu, signal_return);
2239   continue_emission = signal_return;
2240
2241   return continue_emission;
2242 }