2007-06-12 Havoc Pennington <hp@redhat.com>
[platform/upstream/dbus.git] / dbus / dbus-connection.c
1 /* -*- mode: C; c-file-style: "gnu" -*- */
2 /* dbus-connection.c DBusConnection object
3  *
4  * Copyright (C) 2002-2006  Red Hat Inc.
5  *
6  * Licensed under the Academic Free License version 2.1
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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
21  *
22  */
23
24 #include <config.h>
25 #include "dbus-shared.h"
26 #include "dbus-connection.h"
27 #include "dbus-list.h"
28 #include "dbus-timeout.h"
29 #include "dbus-transport.h"
30 #include "dbus-watch.h"
31 #include "dbus-connection-internal.h"
32 #include "dbus-pending-call-internal.h"
33 #include "dbus-list.h"
34 #include "dbus-hash.h"
35 #include "dbus-message-internal.h"
36 #include "dbus-threads.h"
37 #include "dbus-protocol.h"
38 #include "dbus-dataslot.h"
39 #include "dbus-string.h"
40 #include "dbus-pending-call.h"
41 #include "dbus-object-tree.h"
42 #include "dbus-threads-internal.h"
43 #include "dbus-bus.h"
44
45 #ifdef DBUS_DISABLE_CHECKS
46 #define TOOK_LOCK_CHECK(connection)
47 #define RELEASING_LOCK_CHECK(connection)
48 #define HAVE_LOCK_CHECK(connection)
49 #else
50 #define TOOK_LOCK_CHECK(connection) do {                \
51     _dbus_assert (!(connection)->have_connection_lock); \
52     (connection)->have_connection_lock = TRUE;          \
53   } while (0)
54 #define RELEASING_LOCK_CHECK(connection) do {            \
55     _dbus_assert ((connection)->have_connection_lock);   \
56     (connection)->have_connection_lock = FALSE;          \
57   } while (0)
58 #define HAVE_LOCK_CHECK(connection)        _dbus_assert ((connection)->have_connection_lock)
59 /* A "DO_NOT_HAVE_LOCK_CHECK" is impossible since we need the lock to check the flag */
60 #endif
61
62 #define TRACE_LOCKS 1
63
64 #define CONNECTION_LOCK(connection)   do {                                      \
65     if (TRACE_LOCKS) { _dbus_verbose ("  LOCK: %s\n", _DBUS_FUNCTION_NAME); }   \
66     _dbus_mutex_lock ((connection)->mutex);                                      \
67     TOOK_LOCK_CHECK (connection);                                               \
68   } while (0)
69
70 #define CONNECTION_UNLOCK(connection) do {                                              \
71     if (TRACE_LOCKS) { _dbus_verbose ("  UNLOCK: %s\n", _DBUS_FUNCTION_NAME);  }        \
72     RELEASING_LOCK_CHECK (connection);                                                  \
73     _dbus_mutex_unlock ((connection)->mutex);                                            \
74   } while (0)
75
76 #define DISPATCH_STATUS_NAME(s)                                            \
77                      ((s) == DBUS_DISPATCH_COMPLETE ? "complete" :         \
78                       (s) == DBUS_DISPATCH_DATA_REMAINS ? "data remains" : \
79                       (s) == DBUS_DISPATCH_NEED_MEMORY ? "need memory" :   \
80                       "???")
81
82 /**
83  * @defgroup DBusConnection DBusConnection
84  * @ingroup  DBus
85  * @brief Connection to another application
86  *
87  * A DBusConnection represents a connection to another
88  * application. Messages can be sent and received via this connection.
89  * The other application may be a message bus; for convenience, the
90  * function dbus_bus_get() is provided to automatically open a
91  * connection to the well-known message buses.
92  * 
93  * In brief a DBusConnection is a message queue associated with some
94  * message transport mechanism such as a socket.  The connection
95  * maintains a queue of incoming messages and a queue of outgoing
96  * messages.
97  *
98  * Several functions use the following terms:
99  * <ul>
100  * <li><b>read</b> means to fill the incoming message queue by reading from the socket</li>
101  * <li><b>write</b> means to drain the outgoing queue by writing to the socket</li>
102  * <li><b>dispatch</b> means to drain the incoming queue by invoking application-provided message handlers</li>
103  * </ul>
104  *
105  * The function dbus_connection_read_write_dispatch() for example does all
106  * three of these things, offering a simple alternative to a main loop.
107  *
108  * In an application with a main loop, the read/write/dispatch
109  * operations are usually separate.
110  *
111  * The connection provides #DBusWatch and #DBusTimeout objects to
112  * the main loop. These are used to know when reading, writing, or
113  * dispatching should be performed.
114  * 
115  * Incoming messages are processed
116  * by calling dbus_connection_dispatch(). dbus_connection_dispatch()
117  * runs any handlers registered for the topmost message in the message
118  * queue, then discards the message, then returns.
119  * 
120  * dbus_connection_get_dispatch_status() indicates whether
121  * messages are currently in the queue that need dispatching.
122  * dbus_connection_set_dispatch_status_function() allows
123  * you to set a function to be used to monitor the dispatch status.
124  * 
125  * If you're using GLib or Qt add-on libraries for D-Bus, there are
126  * special convenience APIs in those libraries that hide
127  * all the details of dispatch and watch/timeout monitoring.
128  * For example, dbus_connection_setup_with_g_main().
129  *
130  * If you aren't using these add-on libraries, but want to process
131  * messages asynchronously, you must manually call
132  * dbus_connection_set_dispatch_status_function(),
133  * dbus_connection_set_watch_functions(),
134  * dbus_connection_set_timeout_functions() providing appropriate
135  * functions to integrate the connection with your application's main
136  * loop. This can be tricky to get right; main loops are not simple.
137  *
138  * If you don't need to be asynchronous, you can ignore #DBusWatch,
139  * #DBusTimeout, and dbus_connection_dispatch().  Instead,
140  * dbus_connection_read_write_dispatch() can be used.
141  *
142  * Or, in <em>very</em> simple applications,
143  * dbus_connection_pop_message() may be all you need, allowing you to
144  * avoid setting up any handler functions (see
145  * dbus_connection_add_filter(),
146  * dbus_connection_register_object_path() for more on handlers).
147  * 
148  * When you use dbus_connection_send() or one of its variants to send
149  * a message, the message is added to the outgoing queue.  It's
150  * actually written to the network later; either in
151  * dbus_watch_handle() invoked by your main loop, or in
152  * dbus_connection_flush() which blocks until it can write out the
153  * entire outgoing queue. The GLib/Qt add-on libraries again
154  * handle the details here for you by setting up watch functions.
155  *
156  * When a connection is disconnected, you are guaranteed to get a
157  * signal "Disconnected" from the interface
158  * #DBUS_INTERFACE_LOCAL, path
159  * #DBUS_PATH_LOCAL.
160  *
161  * You may not drop the last reference to a #DBusConnection
162  * until that connection has been disconnected.
163  *
164  * You may dispatch the unprocessed incoming message queue even if the
165  * connection is disconnected. However, "Disconnected" will always be
166  * the last message in the queue (obviously no messages are received
167  * after disconnection).
168  *
169  * After calling dbus_threads_init(), #DBusConnection has thread
170  * locks and drops them when invoking user callbacks, so in general is
171  * transparently threadsafe. However, #DBusMessage does NOT have
172  * thread locks; you must not send the same message to multiple
173  * #DBusConnection if those connections will be used from different threads,
174  * for example.
175  *
176  * Also, if you dispatch or pop messages from multiple threads, it
177  * may work in the sense that it won't crash, but it's tough to imagine
178  * sane results; it will be completely unpredictable which messages
179  * go to which threads.
180  *
181  * It's recommended to dispatch from a single thread.
182  *
183  * The most useful function to call from multiple threads at once
184  * is dbus_connection_send_with_reply_and_block(). That is,
185  * multiple threads can make method calls at the same time.
186  *
187  * If you aren't using threads, you can use a main loop and
188  * dbus_pending_call_set_notify() to achieve a similar result.
189  */
190
191 /**
192  * @defgroup DBusConnectionInternals DBusConnection implementation details
193  * @ingroup  DBusInternals
194  * @brief Implementation details of DBusConnection
195  *
196  * @{
197  */
198
199 /**
200  * Internal struct representing a message filter function 
201  */
202 typedef struct DBusMessageFilter DBusMessageFilter;
203
204 /**
205  * Internal struct representing a message filter function 
206  */
207 struct DBusMessageFilter
208 {
209   DBusAtomic refcount; /**< Reference count */
210   DBusHandleMessageFunction function; /**< Function to call to filter */
211   void *user_data; /**< User data for the function */
212   DBusFreeFunction free_user_data_function; /**< Function to free the user data */
213 };
214
215
216 /**
217  * Internals of DBusPreallocatedSend
218  */
219 struct DBusPreallocatedSend
220 {
221   DBusConnection *connection; /**< Connection we'd send the message to */
222   DBusList *queue_link;       /**< Preallocated link in the queue */
223   DBusList *counter_link;     /**< Preallocated link in the resource counter */
224 };
225
226 static dbus_bool_t _dbus_modify_sigpipe = TRUE;
227
228 /**
229  * Implementation details of DBusConnection. All fields are private.
230  */
231 struct DBusConnection
232 {
233   DBusAtomic refcount; /**< Reference count. */
234
235   DBusMutex *mutex; /**< Lock on the entire DBusConnection */
236
237   DBusMutex *dispatch_mutex;     /**< Protects dispatch_acquired */
238   DBusCondVar *dispatch_cond;    /**< Notify when dispatch_acquired is available */
239   DBusMutex *io_path_mutex;      /**< Protects io_path_acquired */
240   DBusCondVar *io_path_cond;     /**< Notify when io_path_acquired is available */
241   
242   DBusList *outgoing_messages; /**< Queue of messages we need to send, send the end of the list first. */
243   DBusList *incoming_messages; /**< Queue of messages we have received, end of the list received most recently. */
244
245   DBusMessage *message_borrowed; /**< Filled in if the first incoming message has been borrowed;
246                                   *   dispatch_acquired will be set by the borrower
247                                   */
248   
249   int n_outgoing;              /**< Length of outgoing queue. */
250   int n_incoming;              /**< Length of incoming queue. */
251
252   DBusCounter *outgoing_counter; /**< Counts size of outgoing messages. */
253   
254   DBusTransport *transport;    /**< Object that sends/receives messages over network. */
255   DBusWatchList *watches;      /**< Stores active watches. */
256   DBusTimeoutList *timeouts;   /**< Stores active timeouts. */
257   
258   DBusList *filter_list;        /**< List of filters. */
259
260   DBusDataSlotList slot_list;   /**< Data stored by allocated integer ID */
261
262   DBusHashTable *pending_replies;  /**< Hash of message serials to #DBusPendingCall. */  
263   
264   dbus_uint32_t client_serial;       /**< Client serial. Increments each time a message is sent  */
265   DBusList *disconnect_message_link; /**< Preallocated list node for queueing the disconnection message */
266
267   DBusWakeupMainFunction wakeup_main_function; /**< Function to wake up the mainloop  */
268   void *wakeup_main_data; /**< Application data for wakeup_main_function */
269   DBusFreeFunction free_wakeup_main_data; /**< free wakeup_main_data */
270
271   DBusDispatchStatusFunction dispatch_status_function; /**< Function on dispatch status changes  */
272   void *dispatch_status_data; /**< Application data for dispatch_status_function */
273   DBusFreeFunction free_dispatch_status_data; /**< free dispatch_status_data */
274
275   DBusDispatchStatus last_dispatch_status; /**< The last dispatch status we reported to the application. */
276
277   DBusList *link_cache; /**< A cache of linked list links to prevent contention
278                          *   for the global linked list mempool lock
279                          */
280   DBusObjectTree *objects; /**< Object path handlers registered with this connection */
281
282   char *server_guid; /**< GUID of server if we are in shared_connections, #NULL if server GUID is unknown or connection is private */
283
284   unsigned int shareable : 1; /**< #TRUE if libdbus owns a reference to the connection and can return it from dbus_connection_open() more than once */
285
286   unsigned int dispatch_acquired : 1; /**< Someone has dispatch path (can drain incoming queue) */
287   unsigned int io_path_acquired : 1;  /**< Someone has transport io path (can use the transport to read/write messages) */
288   
289   unsigned int exit_on_disconnect : 1; /**< If #TRUE, exit after handling disconnect signal */
290
291   unsigned int route_peer_messages : 1; /**< If #TRUE, if org.freedesktop.DBus.Peer messages have a bus name, don't handle them automatically */
292
293   unsigned int disconnected_message_arrived : 1;   /**< We popped or are dispatching the disconnected message.
294                                                     * if the disconnect_message_link is NULL then we queued it, but
295                                                     * this flag is whether it got to the head of the queue.
296                                                     */
297   unsigned int disconnected_message_processed : 1; /**< We did our default handling of the disconnected message,
298                                                     * such as closing the connection.
299                                                     */
300   
301 #ifndef DBUS_DISABLE_CHECKS
302   unsigned int have_connection_lock : 1; /**< Used to check locking */
303 #endif
304   
305 #ifndef DBUS_DISABLE_CHECKS
306   int generation; /**< _dbus_current_generation that should correspond to this connection */
307 #endif 
308 };
309
310 static DBusDispatchStatus _dbus_connection_get_dispatch_status_unlocked      (DBusConnection     *connection);
311 static void               _dbus_connection_update_dispatch_status_and_unlock (DBusConnection     *connection,
312                                                                               DBusDispatchStatus  new_status);
313 static void               _dbus_connection_last_unref                        (DBusConnection     *connection);
314 static void               _dbus_connection_acquire_dispatch                  (DBusConnection     *connection);
315 static void               _dbus_connection_release_dispatch                  (DBusConnection     *connection);
316 static DBusDispatchStatus _dbus_connection_flush_unlocked                    (DBusConnection     *connection);
317 static void               _dbus_connection_close_possibly_shared_and_unlock  (DBusConnection     *connection);
318 static dbus_bool_t        _dbus_connection_get_is_connected_unlocked         (DBusConnection     *connection);
319
320 static DBusMessageFilter *
321 _dbus_message_filter_ref (DBusMessageFilter *filter)
322 {
323   _dbus_assert (filter->refcount.value > 0);
324   _dbus_atomic_inc (&filter->refcount);
325
326   return filter;
327 }
328
329 static void
330 _dbus_message_filter_unref (DBusMessageFilter *filter)
331 {
332   _dbus_assert (filter->refcount.value > 0);
333
334   if (_dbus_atomic_dec (&filter->refcount) == 1)
335     {
336       if (filter->free_user_data_function)
337         (* filter->free_user_data_function) (filter->user_data);
338       
339       dbus_free (filter);
340     }
341 }
342
343 /**
344  * Acquires the connection lock.
345  *
346  * @param connection the connection.
347  */
348 void
349 _dbus_connection_lock (DBusConnection *connection)
350 {
351   CONNECTION_LOCK (connection);
352 }
353
354 /**
355  * Releases the connection lock.
356  *
357  * @param connection the connection.
358  */
359 void
360 _dbus_connection_unlock (DBusConnection *connection)
361 {
362   CONNECTION_UNLOCK (connection);
363 }
364
365 /**
366  * Wakes up the main loop if it is sleeping
367  * Needed if we're e.g. queueing outgoing messages
368  * on a thread while the mainloop sleeps.
369  *
370  * @param connection the connection.
371  */
372 static void
373 _dbus_connection_wakeup_mainloop (DBusConnection *connection)
374 {
375   if (connection->wakeup_main_function)
376     (*connection->wakeup_main_function) (connection->wakeup_main_data);
377 }
378
379 #ifdef DBUS_BUILD_TESTS
380 /* For now this function isn't used */
381 /**
382  * Adds a message to the incoming message queue, returning #FALSE
383  * if there's insufficient memory to queue the message.
384  * Does not take over refcount of the message.
385  *
386  * @param connection the connection.
387  * @param message the message to queue.
388  * @returns #TRUE on success.
389  */
390 dbus_bool_t
391 _dbus_connection_queue_received_message (DBusConnection *connection,
392                                          DBusMessage    *message)
393 {
394   DBusList *link;
395
396   link = _dbus_list_alloc_link (message);
397   if (link == NULL)
398     return FALSE;
399
400   dbus_message_ref (message);
401   _dbus_connection_queue_received_message_link (connection, link);
402
403   return TRUE;
404 }
405
406 /**
407  * Gets the locks so we can examine them
408  *
409  * @param connection the connection.
410  * @param mutex_loc return for the location of the main mutex pointer
411  * @param dispatch_mutex_loc return location of the dispatch mutex pointer
412  * @param io_path_mutex_loc return location of the io_path mutex pointer
413  * @param dispatch_cond_loc return location of the dispatch conditional 
414  *        variable pointer
415  * @param io_path_cond_loc return location of the io_path conditional 
416  *        variable pointer
417  */ 
418 void 
419 _dbus_connection_test_get_locks (DBusConnection *connection,
420                                  DBusMutex     **mutex_loc,
421                                  DBusMutex     **dispatch_mutex_loc,
422                                  DBusMutex     **io_path_mutex_loc,
423                                  DBusCondVar   **dispatch_cond_loc,
424                                  DBusCondVar   **io_path_cond_loc)
425 {
426   *mutex_loc = connection->mutex;
427   *dispatch_mutex_loc = connection->dispatch_mutex;
428   *io_path_mutex_loc = connection->io_path_mutex; 
429   *dispatch_cond_loc = connection->dispatch_cond;
430   *io_path_cond_loc = connection->io_path_cond;
431 }
432 #endif
433
434 /**
435  * Adds a message-containing list link to the incoming message queue,
436  * taking ownership of the link and the message's current refcount.
437  * Cannot fail due to lack of memory.
438  *
439  * @param connection the connection.
440  * @param link the message link to queue.
441  */
442 void
443 _dbus_connection_queue_received_message_link (DBusConnection  *connection,
444                                               DBusList        *link)
445 {
446   DBusPendingCall *pending;
447   dbus_int32_t reply_serial;
448   DBusMessage *message;
449   
450   _dbus_assert (_dbus_transport_get_is_authenticated (connection->transport));
451   
452   _dbus_list_append_link (&connection->incoming_messages,
453                           link);
454   message = link->data;
455
456   /* If this is a reply we're waiting on, remove timeout for it */
457   reply_serial = dbus_message_get_reply_serial (message);
458   if (reply_serial != -1)
459     {
460       pending = _dbus_hash_table_lookup_int (connection->pending_replies,
461                                              reply_serial);
462       if (pending != NULL)
463         {
464           if (_dbus_pending_call_is_timeout_added_unlocked (pending))
465             _dbus_connection_remove_timeout_unlocked (connection,
466                                                       _dbus_pending_call_get_timeout_unlocked (pending));
467
468           _dbus_pending_call_set_timeout_added_unlocked (pending, FALSE);
469         }
470     }
471   
472   
473
474   connection->n_incoming += 1;
475
476   _dbus_connection_wakeup_mainloop (connection);
477   
478   _dbus_verbose ("Message %p (%d %s %s %s '%s' reply to %u) added to incoming queue %p, %d incoming\n",
479                  message,
480                  dbus_message_get_type (message),
481                  dbus_message_get_path (message) ?
482                  dbus_message_get_path (message) :
483                  "no path",
484                  dbus_message_get_interface (message) ?
485                  dbus_message_get_interface (message) :
486                  "no interface",
487                  dbus_message_get_member (message) ?
488                  dbus_message_get_member (message) :
489                  "no member",
490                  dbus_message_get_signature (message),
491                  dbus_message_get_reply_serial (message),
492                  connection,
493                  connection->n_incoming);}
494
495 /**
496  * Adds a link + message to the incoming message queue.
497  * Can't fail. Takes ownership of both link and message.
498  *
499  * @param connection the connection.
500  * @param link the list node and message to queue.
501  *
502  */
503 void
504 _dbus_connection_queue_synthesized_message_link (DBusConnection *connection,
505                                                  DBusList *link)
506 {
507   HAVE_LOCK_CHECK (connection);
508   
509   _dbus_list_append_link (&connection->incoming_messages, link);
510
511   connection->n_incoming += 1;
512
513   _dbus_connection_wakeup_mainloop (connection);
514   
515   _dbus_verbose ("Synthesized message %p added to incoming queue %p, %d incoming\n",
516                  link->data, connection, connection->n_incoming);
517 }
518
519
520 /**
521  * Checks whether there are messages in the outgoing message queue.
522  * Called with connection lock held.
523  *
524  * @param connection the connection.
525  * @returns #TRUE if the outgoing queue is non-empty.
526  */
527 dbus_bool_t
528 _dbus_connection_has_messages_to_send_unlocked (DBusConnection *connection)
529 {
530   HAVE_LOCK_CHECK (connection);
531   return connection->outgoing_messages != NULL;
532 }
533
534 /**
535  * Checks whether there are messages in the outgoing message queue.
536  * Use dbus_connection_flush() to block until all outgoing
537  * messages have been written to the underlying transport
538  * (such as a socket).
539  * 
540  * @param connection the connection.
541  * @returns #TRUE if the outgoing queue is non-empty.
542  */
543 dbus_bool_t
544 dbus_connection_has_messages_to_send (DBusConnection *connection)
545 {
546   dbus_bool_t v;
547   
548   _dbus_return_val_if_fail (connection != NULL, FALSE);
549
550   CONNECTION_LOCK (connection);
551   v = _dbus_connection_has_messages_to_send_unlocked (connection);
552   CONNECTION_UNLOCK (connection);
553
554   return v;
555 }
556
557 /**
558  * Gets the next outgoing message. The message remains in the
559  * queue, and the caller does not own a reference to it.
560  *
561  * @param connection the connection.
562  * @returns the message to be sent.
563  */ 
564 DBusMessage*
565 _dbus_connection_get_message_to_send (DBusConnection *connection)
566 {
567   HAVE_LOCK_CHECK (connection);
568   
569   return _dbus_list_get_last (&connection->outgoing_messages);
570 }
571
572 /**
573  * Notifies the connection that a message has been sent, so the
574  * message can be removed from the outgoing queue.
575  * Called with the connection lock held.
576  *
577  * @param connection the connection.
578  * @param message the message that was sent.
579  */
580 void
581 _dbus_connection_message_sent (DBusConnection *connection,
582                                DBusMessage    *message)
583 {
584   DBusList *link;
585
586   HAVE_LOCK_CHECK (connection);
587   
588   /* This can be called before we even complete authentication, since
589    * it's called on disconnect to clean up the outgoing queue.
590    * It's also called as we successfully send each message.
591    */
592   
593   link = _dbus_list_get_last_link (&connection->outgoing_messages);
594   _dbus_assert (link != NULL);
595   _dbus_assert (link->data == message);
596
597   /* Save this link in the link cache */
598   _dbus_list_unlink (&connection->outgoing_messages,
599                      link);
600   _dbus_list_prepend_link (&connection->link_cache, link);
601   
602   connection->n_outgoing -= 1;
603
604   _dbus_verbose ("Message %p (%d %s %s %s '%s') removed from outgoing queue %p, %d left to send\n",
605                  message,
606                  dbus_message_get_type (message),
607                  dbus_message_get_path (message) ?
608                  dbus_message_get_path (message) :
609                  "no path",
610                  dbus_message_get_interface (message) ?
611                  dbus_message_get_interface (message) :
612                  "no interface",
613                  dbus_message_get_member (message) ?
614                  dbus_message_get_member (message) :
615                  "no member",
616                  dbus_message_get_signature (message),
617                  connection, connection->n_outgoing);
618
619   /* Save this link in the link cache also */
620   _dbus_message_remove_size_counter (message, connection->outgoing_counter,
621                                      &link);
622   _dbus_list_prepend_link (&connection->link_cache, link);
623   
624   dbus_message_unref (message);
625 }
626
627 /** Function to be called in protected_change_watch() with refcount held */
628 typedef dbus_bool_t (* DBusWatchAddFunction)     (DBusWatchList *list,
629                                                   DBusWatch     *watch);
630 /** Function to be called in protected_change_watch() with refcount held */
631 typedef void        (* DBusWatchRemoveFunction)  (DBusWatchList *list,
632                                                   DBusWatch     *watch);
633 /** Function to be called in protected_change_watch() with refcount held */
634 typedef void        (* DBusWatchToggleFunction)  (DBusWatchList *list,
635                                                   DBusWatch     *watch,
636                                                   dbus_bool_t    enabled);
637
638 static dbus_bool_t
639 protected_change_watch (DBusConnection         *connection,
640                         DBusWatch              *watch,
641                         DBusWatchAddFunction    add_function,
642                         DBusWatchRemoveFunction remove_function,
643                         DBusWatchToggleFunction toggle_function,
644                         dbus_bool_t             enabled)
645 {
646   DBusWatchList *watches;
647   dbus_bool_t retval;
648   
649   HAVE_LOCK_CHECK (connection);
650
651   /* This isn't really safe or reasonable; a better pattern is the "do everything, then
652    * drop lock and call out" one; but it has to be propagated up through all callers
653    */
654   
655   watches = connection->watches;
656   if (watches)
657     {
658       connection->watches = NULL;
659       _dbus_connection_ref_unlocked (connection);
660       CONNECTION_UNLOCK (connection);
661
662       if (add_function)
663         retval = (* add_function) (watches, watch);
664       else if (remove_function)
665         {
666           retval = TRUE;
667           (* remove_function) (watches, watch);
668         }
669       else
670         {
671           retval = TRUE;
672           (* toggle_function) (watches, watch, enabled);
673         }
674       
675       CONNECTION_LOCK (connection);
676       connection->watches = watches;
677       _dbus_connection_unref_unlocked (connection);
678
679       return retval;
680     }
681   else
682     return FALSE;
683 }
684      
685
686 /**
687  * Adds a watch using the connection's DBusAddWatchFunction if
688  * available. Otherwise records the watch to be added when said
689  * function is available. Also re-adds the watch if the
690  * DBusAddWatchFunction changes. May fail due to lack of memory.
691  * Connection lock should be held when calling this.
692  *
693  * @param connection the connection.
694  * @param watch the watch to add.
695  * @returns #TRUE on success.
696  */
697 dbus_bool_t
698 _dbus_connection_add_watch_unlocked (DBusConnection *connection,
699                                      DBusWatch      *watch)
700 {
701   return protected_change_watch (connection, watch,
702                                  _dbus_watch_list_add_watch,
703                                  NULL, NULL, FALSE);
704 }
705
706 /**
707  * Removes a watch using the connection's DBusRemoveWatchFunction
708  * if available. It's an error to call this function on a watch
709  * that was not previously added.
710  * Connection lock should be held when calling this.
711  *
712  * @param connection the connection.
713  * @param watch the watch to remove.
714  */
715 void
716 _dbus_connection_remove_watch_unlocked (DBusConnection *connection,
717                                         DBusWatch      *watch)
718 {
719   protected_change_watch (connection, watch,
720                           NULL,
721                           _dbus_watch_list_remove_watch,
722                           NULL, FALSE);
723 }
724
725 /**
726  * Toggles a watch and notifies app via connection's
727  * DBusWatchToggledFunction if available. It's an error to call this
728  * function on a watch that was not previously added.
729  * Connection lock should be held when calling this.
730  *
731  * @param connection the connection.
732  * @param watch the watch to toggle.
733  * @param enabled whether to enable or disable
734  */
735 void
736 _dbus_connection_toggle_watch_unlocked (DBusConnection *connection,
737                                         DBusWatch      *watch,
738                                         dbus_bool_t     enabled)
739 {
740   _dbus_assert (watch != NULL);
741
742   protected_change_watch (connection, watch,
743                           NULL, NULL,
744                           _dbus_watch_list_toggle_watch,
745                           enabled);
746 }
747
748 /** Function to be called in protected_change_timeout() with refcount held */
749 typedef dbus_bool_t (* DBusTimeoutAddFunction)    (DBusTimeoutList *list,
750                                                    DBusTimeout     *timeout);
751 /** Function to be called in protected_change_timeout() with refcount held */
752 typedef void        (* DBusTimeoutRemoveFunction) (DBusTimeoutList *list,
753                                                    DBusTimeout     *timeout);
754 /** Function to be called in protected_change_timeout() with refcount held */
755 typedef void        (* DBusTimeoutToggleFunction) (DBusTimeoutList *list,
756                                                    DBusTimeout     *timeout,
757                                                    dbus_bool_t      enabled);
758
759 static dbus_bool_t
760 protected_change_timeout (DBusConnection           *connection,
761                           DBusTimeout              *timeout,
762                           DBusTimeoutAddFunction    add_function,
763                           DBusTimeoutRemoveFunction remove_function,
764                           DBusTimeoutToggleFunction toggle_function,
765                           dbus_bool_t               enabled)
766 {
767   DBusTimeoutList *timeouts;
768   dbus_bool_t retval;
769   
770   HAVE_LOCK_CHECK (connection);
771
772   /* This isn't really safe or reasonable; a better pattern is the "do everything, then
773    * drop lock and call out" one; but it has to be propagated up through all callers
774    */
775   
776   timeouts = connection->timeouts;
777   if (timeouts)
778     {
779       connection->timeouts = NULL;
780       _dbus_connection_ref_unlocked (connection);
781       CONNECTION_UNLOCK (connection);
782
783       if (add_function)
784         retval = (* add_function) (timeouts, timeout);
785       else if (remove_function)
786         {
787           retval = TRUE;
788           (* remove_function) (timeouts, timeout);
789         }
790       else
791         {
792           retval = TRUE;
793           (* toggle_function) (timeouts, timeout, enabled);
794         }
795       
796       CONNECTION_LOCK (connection);
797       connection->timeouts = timeouts;
798       _dbus_connection_unref_unlocked (connection);
799
800       return retval;
801     }
802   else
803     return FALSE;
804 }
805
806 /**
807  * Adds a timeout using the connection's DBusAddTimeoutFunction if
808  * available. Otherwise records the timeout to be added when said
809  * function is available. Also re-adds the timeout if the
810  * DBusAddTimeoutFunction changes. May fail due to lack of memory.
811  * The timeout will fire repeatedly until removed.
812  * Connection lock should be held when calling this.
813  *
814  * @param connection the connection.
815  * @param timeout the timeout to add.
816  * @returns #TRUE on success.
817  */
818 dbus_bool_t
819 _dbus_connection_add_timeout_unlocked (DBusConnection *connection,
820                                        DBusTimeout    *timeout)
821 {
822   return protected_change_timeout (connection, timeout,
823                                    _dbus_timeout_list_add_timeout,
824                                    NULL, NULL, FALSE);
825 }
826
827 /**
828  * Removes a timeout using the connection's DBusRemoveTimeoutFunction
829  * if available. It's an error to call this function on a timeout
830  * that was not previously added.
831  * Connection lock should be held when calling this.
832  *
833  * @param connection the connection.
834  * @param timeout the timeout to remove.
835  */
836 void
837 _dbus_connection_remove_timeout_unlocked (DBusConnection *connection,
838                                           DBusTimeout    *timeout)
839 {
840   protected_change_timeout (connection, timeout,
841                             NULL,
842                             _dbus_timeout_list_remove_timeout,
843                             NULL, FALSE);
844 }
845
846 /**
847  * Toggles a timeout and notifies app via connection's
848  * DBusTimeoutToggledFunction if available. It's an error to call this
849  * function on a timeout that was not previously added.
850  * Connection lock should be held when calling this.
851  *
852  * @param connection the connection.
853  * @param timeout the timeout to toggle.
854  * @param enabled whether to enable or disable
855  */
856 void
857 _dbus_connection_toggle_timeout_unlocked (DBusConnection   *connection,
858                                           DBusTimeout      *timeout,
859                                           dbus_bool_t       enabled)
860 {
861   protected_change_timeout (connection, timeout,
862                             NULL, NULL,
863                             _dbus_timeout_list_toggle_timeout,
864                             enabled);
865 }
866
867 static dbus_bool_t
868 _dbus_connection_attach_pending_call_unlocked (DBusConnection  *connection,
869                                                DBusPendingCall *pending)
870 {
871   dbus_uint32_t reply_serial;
872   DBusTimeout *timeout;
873
874   HAVE_LOCK_CHECK (connection);
875
876   reply_serial = _dbus_pending_call_get_reply_serial_unlocked (pending);
877
878   _dbus_assert (reply_serial != 0);
879
880   timeout = _dbus_pending_call_get_timeout_unlocked (pending);
881
882   if (!_dbus_connection_add_timeout_unlocked (connection, timeout))
883     return FALSE;
884   
885   if (!_dbus_hash_table_insert_int (connection->pending_replies,
886                                     reply_serial,
887                                     pending))
888     {
889       _dbus_connection_remove_timeout_unlocked (connection, timeout);
890
891       _dbus_pending_call_set_timeout_added_unlocked (pending, FALSE);
892       HAVE_LOCK_CHECK (connection);
893       return FALSE;
894     }
895   
896   _dbus_pending_call_set_timeout_added_unlocked (pending, TRUE);
897
898   _dbus_pending_call_ref_unlocked (pending);
899
900   HAVE_LOCK_CHECK (connection);
901   
902   return TRUE;
903 }
904
905 static void
906 free_pending_call_on_hash_removal (void *data)
907 {
908   DBusPendingCall *pending;
909   DBusConnection  *connection;
910   
911   if (data == NULL)
912     return;
913
914   pending = data;
915
916   connection = _dbus_pending_call_get_connection_unlocked (pending);
917
918   HAVE_LOCK_CHECK (connection);
919   
920   if (_dbus_pending_call_is_timeout_added_unlocked (pending))
921     {
922       _dbus_connection_remove_timeout_unlocked (connection,
923                                                 _dbus_pending_call_get_timeout_unlocked (pending));
924       
925       _dbus_pending_call_set_timeout_added_unlocked (pending, FALSE);
926     }
927
928   /* FIXME 1.0? this is sort of dangerous and undesirable to drop the lock 
929    * here, but the pending call finalizer could in principle call out to 
930    * application code so we pretty much have to... some larger code reorg 
931    * might be needed.
932    */
933   _dbus_connection_ref_unlocked (connection);
934   _dbus_pending_call_unref_and_unlock (pending);
935   CONNECTION_LOCK (connection);
936   _dbus_connection_unref_unlocked (connection);
937 }
938
939 static void
940 _dbus_connection_detach_pending_call_unlocked (DBusConnection  *connection,
941                                                DBusPendingCall *pending)
942 {
943   /* This ends up unlocking to call the pending call finalizer, which is unexpected to
944    * say the least.
945    */
946   _dbus_hash_table_remove_int (connection->pending_replies,
947                                _dbus_pending_call_get_reply_serial_unlocked (pending));
948 }
949
950 static void
951 _dbus_connection_detach_pending_call_and_unlock (DBusConnection  *connection,
952                                                  DBusPendingCall *pending)
953 {
954   /* The idea here is to avoid finalizing the pending call
955    * with the lock held, since there's a destroy notifier
956    * in pending call that goes out to application code.
957    *
958    * There's an extra unlock inside the hash table
959    * "free pending call" function FIXME...
960    */
961   _dbus_pending_call_ref_unlocked (pending);
962   _dbus_hash_table_remove_int (connection->pending_replies,
963                                _dbus_pending_call_get_reply_serial_unlocked (pending));
964   _dbus_pending_call_unref_and_unlock (pending);
965 }
966
967 /**
968  * Removes a pending call from the connection, such that
969  * the pending reply will be ignored. May drop the last
970  * reference to the pending call.
971  *
972  * @param connection the connection
973  * @param pending the pending call
974  */
975 void
976 _dbus_connection_remove_pending_call (DBusConnection  *connection,
977                                       DBusPendingCall *pending)
978 {
979   CONNECTION_LOCK (connection);
980   _dbus_connection_detach_pending_call_and_unlock (connection, pending);
981 }
982
983 /**
984  * Acquire the transporter I/O path. This must be done before
985  * doing any I/O in the transporter. May sleep and drop the
986  * IO path mutex while waiting for the I/O path.
987  *
988  * @param connection the connection.
989  * @param timeout_milliseconds maximum blocking time, or -1 for no limit.
990  * @returns TRUE if the I/O path was acquired.
991  */
992 static dbus_bool_t
993 _dbus_connection_acquire_io_path (DBusConnection *connection,
994                                   int             timeout_milliseconds)
995 {
996   dbus_bool_t we_acquired;
997   
998   HAVE_LOCK_CHECK (connection);
999
1000   /* We don't want the connection to vanish */
1001   _dbus_connection_ref_unlocked (connection);
1002
1003   /* We will only touch io_path_acquired which is protected by our mutex */
1004   CONNECTION_UNLOCK (connection);
1005   
1006   _dbus_verbose ("%s locking io_path_mutex\n", _DBUS_FUNCTION_NAME);
1007   _dbus_mutex_lock (connection->io_path_mutex);
1008
1009   _dbus_verbose ("%s start connection->io_path_acquired = %d timeout = %d\n",
1010                  _DBUS_FUNCTION_NAME, connection->io_path_acquired, timeout_milliseconds);
1011
1012   we_acquired = FALSE;
1013   
1014   if (connection->io_path_acquired)
1015     {
1016       if (timeout_milliseconds != -1)
1017         {
1018           _dbus_verbose ("%s waiting %d for IO path to be acquirable\n",
1019                          _DBUS_FUNCTION_NAME, timeout_milliseconds);
1020
1021           if (!_dbus_condvar_wait_timeout (connection->io_path_cond,
1022                                            connection->io_path_mutex,
1023                                            timeout_milliseconds))
1024             {
1025               /* We timed out before anyone signaled. */
1026               /* (writing the loop to handle the !timedout case by
1027                * waiting longer if needed is a pain since dbus
1028                * wraps pthread_cond_timedwait to take a relative
1029                * time instead of absolute, something kind of stupid
1030                * on our part. for now it doesn't matter, we will just
1031                * end up back here eventually.)
1032                */
1033             }
1034         }
1035       else
1036         {
1037           while (connection->io_path_acquired)
1038             {
1039               _dbus_verbose ("%s waiting for IO path to be acquirable\n", _DBUS_FUNCTION_NAME);
1040               _dbus_condvar_wait (connection->io_path_cond, 
1041                                   connection->io_path_mutex);
1042             }
1043         }
1044     }
1045   
1046   if (!connection->io_path_acquired)
1047     {
1048       we_acquired = TRUE;
1049       connection->io_path_acquired = TRUE;
1050     }
1051   
1052   _dbus_verbose ("%s end connection->io_path_acquired = %d we_acquired = %d\n",
1053                  _DBUS_FUNCTION_NAME, connection->io_path_acquired, we_acquired);
1054
1055   _dbus_verbose ("%s unlocking io_path_mutex\n", _DBUS_FUNCTION_NAME);
1056   _dbus_mutex_unlock (connection->io_path_mutex);
1057
1058   CONNECTION_LOCK (connection);
1059   
1060   HAVE_LOCK_CHECK (connection);
1061
1062   _dbus_connection_unref_unlocked (connection);
1063   
1064   return we_acquired;
1065 }
1066
1067 /**
1068  * Release the I/O path when you're done with it. Only call
1069  * after you've acquired the I/O. Wakes up at most one thread
1070  * currently waiting to acquire the I/O path.
1071  *
1072  * @param connection the connection.
1073  */
1074 static void
1075 _dbus_connection_release_io_path (DBusConnection *connection)
1076 {
1077   HAVE_LOCK_CHECK (connection);
1078   
1079   _dbus_verbose ("%s locking io_path_mutex\n", _DBUS_FUNCTION_NAME);
1080   _dbus_mutex_lock (connection->io_path_mutex);
1081   
1082   _dbus_assert (connection->io_path_acquired);
1083
1084   _dbus_verbose ("%s start connection->io_path_acquired = %d\n",
1085                  _DBUS_FUNCTION_NAME, connection->io_path_acquired);
1086   
1087   connection->io_path_acquired = FALSE;
1088   _dbus_condvar_wake_one (connection->io_path_cond);
1089
1090   _dbus_verbose ("%s unlocking io_path_mutex\n", _DBUS_FUNCTION_NAME);
1091   _dbus_mutex_unlock (connection->io_path_mutex);
1092 }
1093
1094 /**
1095  * Queues incoming messages and sends outgoing messages for this
1096  * connection, optionally blocking in the process. Each call to
1097  * _dbus_connection_do_iteration_unlocked() will call select() or poll() one
1098  * time and then read or write data if possible.
1099  *
1100  * The purpose of this function is to be able to flush outgoing
1101  * messages or queue up incoming messages without returning
1102  * control to the application and causing reentrancy weirdness.
1103  *
1104  * The flags parameter allows you to specify whether to
1105  * read incoming messages, write outgoing messages, or both,
1106  * and whether to block if no immediate action is possible.
1107  *
1108  * The timeout_milliseconds parameter does nothing unless the
1109  * iteration is blocking.
1110  *
1111  * If there are no outgoing messages and DBUS_ITERATION_DO_READING
1112  * wasn't specified, then it's impossible to block, even if
1113  * you specify DBUS_ITERATION_BLOCK; in that case the function
1114  * returns immediately.
1115  *
1116  * Called with connection lock held.
1117  * 
1118  * @param connection the connection.
1119  * @param flags iteration flags.
1120  * @param timeout_milliseconds maximum blocking time, or -1 for no limit.
1121  */
1122 void
1123 _dbus_connection_do_iteration_unlocked (DBusConnection *connection,
1124                                         unsigned int    flags,
1125                                         int             timeout_milliseconds)
1126 {
1127   _dbus_verbose ("%s start\n", _DBUS_FUNCTION_NAME);
1128   
1129   HAVE_LOCK_CHECK (connection);
1130   
1131   if (connection->n_outgoing == 0)
1132     flags &= ~DBUS_ITERATION_DO_WRITING;
1133
1134   if (_dbus_connection_acquire_io_path (connection,
1135                                         (flags & DBUS_ITERATION_BLOCK) ? timeout_milliseconds : 0))
1136     {
1137       HAVE_LOCK_CHECK (connection);
1138       
1139       _dbus_transport_do_iteration (connection->transport,
1140                                     flags, timeout_milliseconds);
1141       _dbus_connection_release_io_path (connection);
1142     }
1143
1144   HAVE_LOCK_CHECK (connection);
1145
1146   _dbus_verbose ("%s end\n", _DBUS_FUNCTION_NAME);
1147 }
1148
1149 /**
1150  * Creates a new connection for the given transport.  A transport
1151  * represents a message stream that uses some concrete mechanism, such
1152  * as UNIX domain sockets. May return #NULL if insufficient
1153  * memory exists to create the connection.
1154  *
1155  * @param transport the transport.
1156  * @returns the new connection, or #NULL on failure.
1157  */
1158 DBusConnection*
1159 _dbus_connection_new_for_transport (DBusTransport *transport)
1160 {
1161   DBusConnection *connection;
1162   DBusWatchList *watch_list;
1163   DBusTimeoutList *timeout_list;
1164   DBusHashTable *pending_replies;
1165   DBusList *disconnect_link;
1166   DBusMessage *disconnect_message;
1167   DBusCounter *outgoing_counter;
1168   DBusObjectTree *objects;
1169   
1170   watch_list = NULL;
1171   connection = NULL;
1172   pending_replies = NULL;
1173   timeout_list = NULL;
1174   disconnect_link = NULL;
1175   disconnect_message = NULL;
1176   outgoing_counter = NULL;
1177   objects = NULL;
1178   
1179   watch_list = _dbus_watch_list_new ();
1180   if (watch_list == NULL)
1181     goto error;
1182
1183   timeout_list = _dbus_timeout_list_new ();
1184   if (timeout_list == NULL)
1185     goto error;  
1186
1187   pending_replies =
1188     _dbus_hash_table_new (DBUS_HASH_INT,
1189                           NULL,
1190                           (DBusFreeFunction)free_pending_call_on_hash_removal);
1191   if (pending_replies == NULL)
1192     goto error;
1193   
1194   connection = dbus_new0 (DBusConnection, 1);
1195   if (connection == NULL)
1196     goto error;
1197
1198   _dbus_mutex_new_at_location (&connection->mutex);
1199   if (connection->mutex == NULL)
1200     goto error;
1201
1202   _dbus_mutex_new_at_location (&connection->io_path_mutex);
1203   if (connection->io_path_mutex == NULL)
1204     goto error;
1205
1206   _dbus_mutex_new_at_location (&connection->dispatch_mutex);
1207   if (connection->dispatch_mutex == NULL)
1208     goto error;
1209   
1210   _dbus_condvar_new_at_location (&connection->dispatch_cond);
1211   if (connection->dispatch_cond == NULL)
1212     goto error;
1213   
1214   _dbus_condvar_new_at_location (&connection->io_path_cond);
1215   if (connection->io_path_cond == NULL)
1216     goto error;
1217
1218   disconnect_message = dbus_message_new_signal (DBUS_PATH_LOCAL,
1219                                                 DBUS_INTERFACE_LOCAL,
1220                                                 "Disconnected");
1221   
1222   if (disconnect_message == NULL)
1223     goto error;
1224
1225   disconnect_link = _dbus_list_alloc_link (disconnect_message);
1226   if (disconnect_link == NULL)
1227     goto error;
1228
1229   outgoing_counter = _dbus_counter_new ();
1230   if (outgoing_counter == NULL)
1231     goto error;
1232
1233   objects = _dbus_object_tree_new (connection);
1234   if (objects == NULL)
1235     goto error;
1236   
1237   if (_dbus_modify_sigpipe)
1238     _dbus_disable_sigpipe ();
1239   
1240   connection->refcount.value = 1;
1241   connection->transport = transport;
1242   connection->watches = watch_list;
1243   connection->timeouts = timeout_list;
1244   connection->pending_replies = pending_replies;
1245   connection->outgoing_counter = outgoing_counter;
1246   connection->filter_list = NULL;
1247   connection->last_dispatch_status = DBUS_DISPATCH_COMPLETE; /* so we're notified first time there's data */
1248   connection->objects = objects;
1249   connection->exit_on_disconnect = FALSE;
1250   connection->shareable = FALSE;
1251   connection->route_peer_messages = FALSE;
1252   connection->disconnected_message_arrived = FALSE;
1253   connection->disconnected_message_processed = FALSE;
1254   
1255 #ifndef DBUS_DISABLE_CHECKS
1256   connection->generation = _dbus_current_generation;
1257 #endif
1258   
1259   _dbus_data_slot_list_init (&connection->slot_list);
1260
1261   connection->client_serial = 1;
1262
1263   connection->disconnect_message_link = disconnect_link;
1264
1265   CONNECTION_LOCK (connection);
1266   
1267   if (!_dbus_transport_set_connection (transport, connection))
1268     {
1269       CONNECTION_UNLOCK (connection);
1270
1271       goto error;
1272     }
1273
1274   _dbus_transport_ref (transport);
1275
1276   CONNECTION_UNLOCK (connection);
1277   
1278   return connection;
1279   
1280  error:
1281   if (disconnect_message != NULL)
1282     dbus_message_unref (disconnect_message);
1283   
1284   if (disconnect_link != NULL)
1285     _dbus_list_free_link (disconnect_link);
1286   
1287   if (connection != NULL)
1288     {
1289       _dbus_condvar_free_at_location (&connection->io_path_cond);
1290       _dbus_condvar_free_at_location (&connection->dispatch_cond);
1291       _dbus_mutex_free_at_location (&connection->mutex);
1292       _dbus_mutex_free_at_location (&connection->io_path_mutex);
1293       _dbus_mutex_free_at_location (&connection->dispatch_mutex);
1294       dbus_free (connection);
1295     }
1296   if (pending_replies)
1297     _dbus_hash_table_unref (pending_replies);
1298   
1299   if (watch_list)
1300     _dbus_watch_list_free (watch_list);
1301
1302   if (timeout_list)
1303     _dbus_timeout_list_free (timeout_list);
1304
1305   if (outgoing_counter)
1306     _dbus_counter_unref (outgoing_counter);
1307
1308   if (objects)
1309     _dbus_object_tree_unref (objects);
1310   
1311   return NULL;
1312 }
1313
1314 /**
1315  * Increments the reference count of a DBusConnection.
1316  * Requires that the caller already holds the connection lock.
1317  *
1318  * @param connection the connection.
1319  * @returns the connection.
1320  */
1321 DBusConnection *
1322 _dbus_connection_ref_unlocked (DBusConnection *connection)
1323 {  
1324   _dbus_assert (connection != NULL);
1325   _dbus_assert (connection->generation == _dbus_current_generation);
1326
1327   HAVE_LOCK_CHECK (connection);
1328   
1329 #ifdef DBUS_HAVE_ATOMIC_INT
1330   _dbus_atomic_inc (&connection->refcount);
1331 #else
1332   _dbus_assert (connection->refcount.value > 0);
1333   connection->refcount.value += 1;
1334 #endif
1335
1336   return connection;
1337 }
1338
1339 /**
1340  * Decrements the reference count of a DBusConnection.
1341  * Requires that the caller already holds the connection lock.
1342  *
1343  * @param connection the connection.
1344  */
1345 void
1346 _dbus_connection_unref_unlocked (DBusConnection *connection)
1347 {
1348   dbus_bool_t last_unref;
1349
1350   HAVE_LOCK_CHECK (connection);
1351   
1352   _dbus_assert (connection != NULL);
1353
1354   /* The connection lock is better than the global
1355    * lock in the atomic increment fallback
1356    */
1357   
1358 #ifdef DBUS_HAVE_ATOMIC_INT
1359   last_unref = (_dbus_atomic_dec (&connection->refcount) == 1);
1360 #else
1361   _dbus_assert (connection->refcount.value > 0);
1362
1363   connection->refcount.value -= 1;
1364   last_unref = (connection->refcount.value == 0);  
1365 #if 0
1366   printf ("unref_unlocked() connection %p count = %d\n", connection, connection->refcount.value);
1367 #endif
1368 #endif
1369   
1370   if (last_unref)
1371     _dbus_connection_last_unref (connection);
1372 }
1373
1374 static dbus_uint32_t
1375 _dbus_connection_get_next_client_serial (DBusConnection *connection)
1376 {
1377   int serial;
1378
1379   serial = connection->client_serial++;
1380
1381   if (connection->client_serial < 0)
1382     connection->client_serial = 1;
1383   
1384   return serial;
1385 }
1386
1387 /**
1388  * A callback for use with dbus_watch_new() to create a DBusWatch.
1389  * 
1390  * @todo This is basically a hack - we could delete _dbus_transport_handle_watch()
1391  * and the virtual handle_watch in DBusTransport if we got rid of it.
1392  * The reason this is some work is threading, see the _dbus_connection_handle_watch()
1393  * implementation.
1394  *
1395  * @param watch the watch.
1396  * @param condition the current condition of the file descriptors being watched.
1397  * @param data must be a pointer to a #DBusConnection
1398  * @returns #FALSE if the IO condition may not have been fully handled due to lack of memory
1399  */
1400 dbus_bool_t
1401 _dbus_connection_handle_watch (DBusWatch                   *watch,
1402                                unsigned int                 condition,
1403                                void                        *data)
1404 {
1405   DBusConnection *connection;
1406   dbus_bool_t retval;
1407   DBusDispatchStatus status;
1408
1409   connection = data;
1410
1411   _dbus_verbose ("%s start\n", _DBUS_FUNCTION_NAME);
1412   
1413   CONNECTION_LOCK (connection);
1414   _dbus_connection_acquire_io_path (connection, -1);
1415   HAVE_LOCK_CHECK (connection);
1416   retval = _dbus_transport_handle_watch (connection->transport,
1417                                          watch, condition);
1418
1419   _dbus_connection_release_io_path (connection);
1420
1421   HAVE_LOCK_CHECK (connection);
1422
1423   _dbus_verbose ("%s middle\n", _DBUS_FUNCTION_NAME);
1424   
1425   status = _dbus_connection_get_dispatch_status_unlocked (connection);
1426
1427   /* this calls out to user code */
1428   _dbus_connection_update_dispatch_status_and_unlock (connection, status);
1429
1430   _dbus_verbose ("%s end\n", _DBUS_FUNCTION_NAME);
1431   
1432   return retval;
1433 }
1434
1435 _DBUS_DEFINE_GLOBAL_LOCK (shared_connections);
1436 static DBusHashTable *shared_connections = NULL;
1437
1438 static void
1439 shared_connections_shutdown (void *data)
1440 {
1441   int n_entries;
1442   
1443   _DBUS_LOCK (shared_connections);
1444   
1445   /* This is a little bit unpleasant... better ideas? */
1446   while ((n_entries = _dbus_hash_table_get_n_entries (shared_connections)) > 0)
1447     {
1448       DBusConnection *connection;
1449       DBusMessage *message;
1450       DBusHashIter iter;
1451       
1452       _dbus_hash_iter_init (shared_connections, &iter);
1453       _dbus_hash_iter_next (&iter);
1454        
1455       connection = _dbus_hash_iter_get_value (&iter);
1456
1457       _DBUS_UNLOCK (shared_connections);
1458
1459       dbus_connection_ref (connection);
1460       _dbus_connection_close_possibly_shared (connection);
1461
1462       /* Churn through to the Disconnected message */
1463       while ((message = dbus_connection_pop_message (connection)))
1464         {
1465           dbus_message_unref (message);
1466         }
1467       dbus_connection_unref (connection);
1468       
1469       _DBUS_LOCK (shared_connections);
1470
1471       /* The connection should now be dead and not in our hash ... */
1472       _dbus_assert (_dbus_hash_table_get_n_entries (shared_connections) < n_entries);
1473     }
1474
1475   _dbus_assert (_dbus_hash_table_get_n_entries (shared_connections) == 0);
1476   
1477   _dbus_hash_table_unref (shared_connections);
1478   shared_connections = NULL;
1479   
1480   _DBUS_UNLOCK (shared_connections);
1481 }
1482
1483 static dbus_bool_t
1484 connection_lookup_shared (DBusAddressEntry  *entry,
1485                           DBusConnection   **result)
1486 {
1487   _dbus_verbose ("checking for existing connection\n");
1488   
1489   *result = NULL;
1490   
1491   _DBUS_LOCK (shared_connections);
1492
1493   if (shared_connections == NULL)
1494     {
1495       _dbus_verbose ("creating shared_connections hash table\n");
1496       
1497       shared_connections = _dbus_hash_table_new (DBUS_HASH_STRING,
1498                                                  dbus_free,
1499                                                  NULL);
1500       if (shared_connections == NULL)
1501         {
1502           _DBUS_UNLOCK (shared_connections);
1503           return FALSE;
1504         }
1505
1506       if (!_dbus_register_shutdown_func (shared_connections_shutdown, NULL))
1507         {
1508           _dbus_hash_table_unref (shared_connections);
1509           shared_connections = NULL;
1510           _DBUS_UNLOCK (shared_connections);
1511           return FALSE;
1512         }
1513
1514       _dbus_verbose ("  successfully created shared_connections\n");
1515       
1516       _DBUS_UNLOCK (shared_connections);
1517       return TRUE; /* no point looking up in the hash we just made */
1518     }
1519   else
1520     {
1521       const char *guid;
1522
1523       guid = dbus_address_entry_get_value (entry, "guid");
1524       
1525       if (guid != NULL)
1526         {
1527           DBusConnection *connection;
1528           
1529           connection = _dbus_hash_table_lookup_string (shared_connections,
1530                                                        guid);
1531
1532           if (connection)
1533             {
1534               /* The DBusConnection can't be finalized without taking
1535                * the shared_connections lock to remove it from the
1536                * hash.  So it's safe to ref the connection here.
1537                * However, it may be disconnected if the Disconnected
1538                * message hasn't been processed yet, in which case we
1539                * want to pretend it isn't in the hash and avoid
1540                * returning it.
1541                *
1542                * The idea is to avoid ever returning a disconnected connection
1543                * from dbus_connection_open(). We could just synchronously
1544                * drop our shared ref to the connection on connection disconnect,
1545                * and then assert here that the connection is connected, but
1546                * that causes reentrancy headaches.
1547                */
1548               CONNECTION_LOCK (connection);
1549               if (_dbus_connection_get_is_connected_unlocked (connection))
1550                 {
1551                   _dbus_connection_ref_unlocked (connection);
1552                   *result = connection;
1553                   _dbus_verbose ("looked up existing connection to server guid %s\n",
1554                                  guid);
1555                 }
1556               else
1557                 {
1558                   _dbus_verbose ("looked up existing connection to server guid %s but it was disconnected so ignoring it\n",
1559                                  guid);
1560                 }
1561               CONNECTION_UNLOCK (connection);
1562             }
1563         }
1564       
1565       _DBUS_UNLOCK (shared_connections);
1566       return TRUE;
1567     }
1568 }
1569
1570 static dbus_bool_t
1571 connection_record_shared_unlocked (DBusConnection *connection,
1572                                    const char     *guid)
1573 {
1574   char *guid_key;
1575   char *guid_in_connection;
1576
1577   HAVE_LOCK_CHECK (connection);
1578   _dbus_assert (connection->server_guid == NULL);
1579   _dbus_assert (connection->shareable);
1580
1581   /* get a hard ref on this connection, even if
1582    * we won't in fact store it in the hash, we still
1583    * need to hold a ref on it until it's disconnected.
1584    */
1585   _dbus_connection_ref_unlocked (connection);
1586
1587   if (guid == NULL)
1588     return TRUE; /* don't store in the hash */
1589   
1590   /* A separate copy of the key is required in the hash table, because
1591    * we don't have a lock on the connection when we are doing a hash
1592    * lookup.
1593    */
1594   
1595   guid_key = _dbus_strdup (guid);
1596   if (guid_key == NULL)
1597     return FALSE;
1598
1599   guid_in_connection = _dbus_strdup (guid);
1600   if (guid_in_connection == NULL)
1601     {
1602       dbus_free (guid_key);
1603       return FALSE;
1604     }
1605   
1606   _DBUS_LOCK (shared_connections);
1607   _dbus_assert (shared_connections != NULL);
1608   
1609   if (!_dbus_hash_table_insert_string (shared_connections,
1610                                        guid_key, connection))
1611     {
1612       dbus_free (guid_key);
1613       dbus_free (guid_in_connection);
1614       _DBUS_UNLOCK (shared_connections);
1615       return FALSE;
1616     }
1617
1618   connection->server_guid = guid_in_connection;
1619
1620   _dbus_verbose ("stored connection to %s to be shared\n",
1621                  connection->server_guid);
1622   
1623   _DBUS_UNLOCK (shared_connections);
1624
1625   _dbus_assert (connection->server_guid != NULL);
1626   
1627   return TRUE;
1628 }
1629
1630 static void
1631 connection_forget_shared_unlocked (DBusConnection *connection)
1632 {
1633   HAVE_LOCK_CHECK (connection);
1634
1635   if (!connection->shareable)
1636     return;
1637   
1638   if (connection->server_guid != NULL)
1639     {
1640       _dbus_verbose ("dropping connection to %s out of the shared table\n",
1641                      connection->server_guid);
1642       
1643       _DBUS_LOCK (shared_connections);
1644       
1645       if (!_dbus_hash_table_remove_string (shared_connections,
1646                                            connection->server_guid))
1647         _dbus_assert_not_reached ("connection was not in the shared table");
1648       
1649       dbus_free (connection->server_guid);
1650       connection->server_guid = NULL;
1651       _DBUS_UNLOCK (shared_connections);
1652     }
1653   
1654   /* remove our reference held on all shareable connections */
1655   _dbus_connection_unref_unlocked (connection);
1656 }
1657
1658 static DBusConnection*
1659 connection_try_from_address_entry (DBusAddressEntry *entry,
1660                                    DBusError        *error)
1661 {
1662   DBusTransport *transport;
1663   DBusConnection *connection;
1664
1665   transport = _dbus_transport_open (entry, error);
1666
1667   if (transport == NULL)
1668     {
1669       _DBUS_ASSERT_ERROR_IS_SET (error);
1670       return NULL;
1671     }
1672
1673   connection = _dbus_connection_new_for_transport (transport);
1674
1675   _dbus_transport_unref (transport);
1676   
1677   if (connection == NULL)
1678     {
1679       _DBUS_SET_OOM (error);
1680       return NULL;
1681     }
1682
1683 #ifndef DBUS_DISABLE_CHECKS
1684   _dbus_assert (!connection->have_connection_lock);
1685 #endif
1686   return connection;
1687 }
1688
1689 /*
1690  * If the shared parameter is true, then any existing connection will
1691  * be used (and if a new connection is created, it will be available
1692  * for use by others). If the shared parameter is false, a new
1693  * connection will always be created, and the new connection will
1694  * never be returned to other callers.
1695  *
1696  * @param address the address
1697  * @param shared whether the connection is shared or private
1698  * @param error error return
1699  * @returns the connection or #NULL on error
1700  */
1701 static DBusConnection*
1702 _dbus_connection_open_internal (const char     *address,
1703                                 dbus_bool_t     shared,
1704                                 DBusError      *error)
1705 {
1706   DBusConnection *connection;
1707   DBusAddressEntry **entries;
1708   DBusError tmp_error;
1709   DBusError first_error;
1710   int len, i;
1711
1712   _DBUS_ASSERT_ERROR_IS_CLEAR (error);
1713
1714   _dbus_verbose ("opening %s connection to: %s\n",
1715                  shared ? "shared" : "private", address);
1716   
1717   if (!dbus_parse_address (address, &entries, &len, error))
1718     return NULL;
1719
1720   _DBUS_ASSERT_ERROR_IS_CLEAR (error);
1721   
1722   connection = NULL;
1723
1724   dbus_error_init (&tmp_error);
1725   dbus_error_init (&first_error);
1726   for (i = 0; i < len; i++)
1727     {
1728       if (shared)
1729         {
1730           if (!connection_lookup_shared (entries[i], &connection))
1731             _DBUS_SET_OOM (&tmp_error);
1732         }
1733
1734       if (connection == NULL)
1735         {
1736           connection = connection_try_from_address_entry (entries[i],
1737                                                           &tmp_error);
1738
1739           if (connection != NULL && shared)
1740             {
1741               const char *guid;
1742                   
1743               connection->shareable = TRUE;
1744                   
1745               /* guid may be NULL */
1746               guid = dbus_address_entry_get_value (entries[i], "guid");
1747                   
1748               CONNECTION_LOCK (connection);
1749           
1750               if (!connection_record_shared_unlocked (connection, guid))
1751                 {
1752                   _DBUS_SET_OOM (&tmp_error);
1753                   _dbus_connection_close_possibly_shared_and_unlock (connection);
1754                   dbus_connection_unref (connection);
1755                   connection = NULL;
1756                 }
1757               else
1758                 CONNECTION_UNLOCK (connection);
1759             }
1760         }
1761       
1762       if (connection)
1763         break;
1764
1765       _DBUS_ASSERT_ERROR_IS_SET (&tmp_error);
1766       
1767       if (i == 0)
1768         dbus_move_error (&tmp_error, &first_error);
1769       else
1770         dbus_error_free (&tmp_error);
1771     }
1772   
1773   _DBUS_ASSERT_ERROR_IS_CLEAR (error);
1774   _DBUS_ASSERT_ERROR_IS_CLEAR (&tmp_error);
1775   
1776   if (connection == NULL)
1777     {
1778       _DBUS_ASSERT_ERROR_IS_SET (&first_error);
1779       dbus_move_error (&first_error, error);
1780     }
1781   else
1782     dbus_error_free (&first_error);
1783   
1784   dbus_address_entries_free (entries);
1785   return connection;
1786 }
1787
1788 /**
1789  * Closes a shared OR private connection, while dbus_connection_close() can
1790  * only be used on private connections. Should only be called by the
1791  * dbus code that owns the connection - an owner must be known,
1792  * the open/close state is like malloc/free, not like ref/unref.
1793  * 
1794  * @param connection the connection
1795  */
1796 void
1797 _dbus_connection_close_possibly_shared (DBusConnection *connection)
1798 {
1799   _dbus_assert (connection != NULL);
1800   _dbus_assert (connection->generation == _dbus_current_generation);
1801
1802   CONNECTION_LOCK (connection);
1803   _dbus_connection_close_possibly_shared_and_unlock (connection);
1804 }
1805
1806 static DBusPreallocatedSend*
1807 _dbus_connection_preallocate_send_unlocked (DBusConnection *connection)
1808 {
1809   DBusPreallocatedSend *preallocated;
1810
1811   HAVE_LOCK_CHECK (connection);
1812   
1813   _dbus_assert (connection != NULL);
1814   
1815   preallocated = dbus_new (DBusPreallocatedSend, 1);
1816   if (preallocated == NULL)
1817     return NULL;
1818
1819   if (connection->link_cache != NULL)
1820     {
1821       preallocated->queue_link =
1822         _dbus_list_pop_first_link (&connection->link_cache);
1823       preallocated->queue_link->data = NULL;
1824     }
1825   else
1826     {
1827       preallocated->queue_link = _dbus_list_alloc_link (NULL);
1828       if (preallocated->queue_link == NULL)
1829         goto failed_0;
1830     }
1831   
1832   if (connection->link_cache != NULL)
1833     {
1834       preallocated->counter_link =
1835         _dbus_list_pop_first_link (&connection->link_cache);
1836       preallocated->counter_link->data = connection->outgoing_counter;
1837     }
1838   else
1839     {
1840       preallocated->counter_link = _dbus_list_alloc_link (connection->outgoing_counter);
1841       if (preallocated->counter_link == NULL)
1842         goto failed_1;
1843     }
1844
1845   _dbus_counter_ref (preallocated->counter_link->data);
1846
1847   preallocated->connection = connection;
1848   
1849   return preallocated;
1850   
1851  failed_1:
1852   _dbus_list_free_link (preallocated->queue_link);
1853  failed_0:
1854   dbus_free (preallocated);
1855   
1856   return NULL;
1857 }
1858
1859 /* Called with lock held, does not update dispatch status */
1860 static void
1861 _dbus_connection_send_preallocated_unlocked_no_update (DBusConnection       *connection,
1862                                                        DBusPreallocatedSend *preallocated,
1863                                                        DBusMessage          *message,
1864                                                        dbus_uint32_t        *client_serial)
1865 {
1866   dbus_uint32_t serial;
1867   const char *sig;
1868
1869   preallocated->queue_link->data = message;
1870   _dbus_list_prepend_link (&connection->outgoing_messages,
1871                            preallocated->queue_link);
1872
1873   _dbus_message_add_size_counter_link (message,
1874                                        preallocated->counter_link);
1875
1876   dbus_free (preallocated);
1877   preallocated = NULL;
1878   
1879   dbus_message_ref (message);
1880   
1881   connection->n_outgoing += 1;
1882
1883   sig = dbus_message_get_signature (message);
1884   
1885   _dbus_verbose ("Message %p (%d %s %s %s '%s') for %s added to outgoing queue %p, %d pending to send\n",
1886                  message,
1887                  dbus_message_get_type (message),
1888                  dbus_message_get_path (message) ?
1889                  dbus_message_get_path (message) :
1890                  "no path",
1891                  dbus_message_get_interface (message) ?
1892                  dbus_message_get_interface (message) :
1893                  "no interface",
1894                  dbus_message_get_member (message) ?
1895                  dbus_message_get_member (message) :
1896                  "no member",
1897                  sig,
1898                  dbus_message_get_destination (message) ?
1899                  dbus_message_get_destination (message) :
1900                  "null",
1901                  connection,
1902                  connection->n_outgoing);
1903
1904   if (dbus_message_get_serial (message) == 0)
1905     {
1906       serial = _dbus_connection_get_next_client_serial (connection);
1907       _dbus_message_set_serial (message, serial);
1908       if (client_serial)
1909         *client_serial = serial;
1910     }
1911   else
1912     {
1913       if (client_serial)
1914         *client_serial = dbus_message_get_serial (message);
1915     }
1916
1917   _dbus_verbose ("Message %p serial is %u\n",
1918                  message, dbus_message_get_serial (message));
1919   
1920   _dbus_message_lock (message);
1921
1922   /* Now we need to run an iteration to hopefully just write the messages
1923    * out immediately, and otherwise get them queued up
1924    */
1925   _dbus_connection_do_iteration_unlocked (connection,
1926                                           DBUS_ITERATION_DO_WRITING,
1927                                           -1);
1928
1929   /* If stuff is still queued up, be sure we wake up the main loop */
1930   if (connection->n_outgoing > 0)
1931     _dbus_connection_wakeup_mainloop (connection);
1932 }
1933
1934 static void
1935 _dbus_connection_send_preallocated_and_unlock (DBusConnection       *connection,
1936                                                DBusPreallocatedSend *preallocated,
1937                                                DBusMessage          *message,
1938                                                dbus_uint32_t        *client_serial)
1939 {
1940   DBusDispatchStatus status;
1941
1942   HAVE_LOCK_CHECK (connection);
1943   
1944   _dbus_connection_send_preallocated_unlocked_no_update (connection,
1945                                                          preallocated,
1946                                                          message, client_serial);
1947
1948   _dbus_verbose ("%s middle\n", _DBUS_FUNCTION_NAME);
1949   status = _dbus_connection_get_dispatch_status_unlocked (connection);
1950
1951   /* this calls out to user code */
1952   _dbus_connection_update_dispatch_status_and_unlock (connection, status);
1953 }
1954
1955 /**
1956  * Like dbus_connection_send(), but assumes the connection
1957  * is already locked on function entry, and unlocks before returning.
1958  *
1959  * @param connection the connection
1960  * @param message the message to send
1961  * @param client_serial return location for client serial of sent message
1962  * @returns #FALSE on out-of-memory
1963  */
1964 dbus_bool_t
1965 _dbus_connection_send_and_unlock (DBusConnection *connection,
1966                                   DBusMessage    *message,
1967                                   dbus_uint32_t  *client_serial)
1968 {
1969   DBusPreallocatedSend *preallocated;
1970
1971   _dbus_assert (connection != NULL);
1972   _dbus_assert (message != NULL);
1973   
1974   preallocated = _dbus_connection_preallocate_send_unlocked (connection);
1975   if (preallocated == NULL)
1976     {
1977       CONNECTION_UNLOCK (connection);
1978       return FALSE;
1979     }
1980
1981   _dbus_connection_send_preallocated_and_unlock (connection,
1982                                                  preallocated,
1983                                                  message,
1984                                                  client_serial);
1985   return TRUE;
1986 }
1987
1988 /**
1989  * Used internally to handle the semantics of dbus_server_set_new_connection_function().
1990  * If the new connection function does not ref the connection, we want to close it.
1991  *
1992  * A bit of a hack, probably the new connection function should have returned a value
1993  * for whether to close, or should have had to close the connection itself if it
1994  * didn't want it.
1995  *
1996  * But, this works OK as long as the new connection function doesn't do anything
1997  * crazy like keep the connection around without ref'ing it.
1998  *
1999  * We have to lock the connection across refcount check and close in case
2000  * the new connection function spawns a thread that closes and unrefs.
2001  * In that case, if the app thread
2002  * closes and unrefs first, we'll harmlessly close again; if the app thread
2003  * still has the ref, we'll close and then the app will close harmlessly.
2004  * If the app unrefs without closing, the app is broken since if the
2005  * app refs from the new connection function it is supposed to also close.
2006  *
2007  * If we didn't atomically check the refcount and close with the lock held
2008  * though, we could screw this up.
2009  * 
2010  * @param connection the connection
2011  */
2012 void
2013 _dbus_connection_close_if_only_one_ref (DBusConnection *connection)
2014 {
2015   CONNECTION_LOCK (connection);
2016   
2017   _dbus_assert (connection->refcount.value > 0);
2018
2019   if (connection->refcount.value == 1)
2020     _dbus_connection_close_possibly_shared_and_unlock (connection);
2021   else
2022     CONNECTION_UNLOCK (connection);
2023 }
2024
2025
2026 /**
2027  * When a function that blocks has been called with a timeout, and we
2028  * run out of memory, the time to wait for memory is based on the
2029  * timeout. If the caller was willing to block a long time we wait a
2030  * relatively long time for memory, if they were only willing to block
2031  * briefly then we retry for memory at a rapid rate.
2032  *
2033  * @timeout_milliseconds the timeout requested for blocking
2034  */
2035 static void
2036 _dbus_memory_pause_based_on_timeout (int timeout_milliseconds)
2037 {
2038   if (timeout_milliseconds == -1)
2039     _dbus_sleep_milliseconds (1000);
2040   else if (timeout_milliseconds < 100)
2041     ; /* just busy loop */
2042   else if (timeout_milliseconds <= 1000)
2043     _dbus_sleep_milliseconds (timeout_milliseconds / 3);
2044   else
2045     _dbus_sleep_milliseconds (1000);
2046 }
2047
2048 static DBusMessage *
2049 generate_local_error_message (dbus_uint32_t serial, 
2050                               char *error_name, 
2051                               char *error_msg)
2052 {
2053   DBusMessage *message;
2054   message = dbus_message_new (DBUS_MESSAGE_TYPE_ERROR);
2055   if (!message)
2056     goto out;
2057
2058   if (!dbus_message_set_error_name (message, error_name))
2059     {
2060       dbus_message_unref (message);
2061       message = NULL;
2062       goto out; 
2063     }
2064
2065   dbus_message_set_no_reply (message, TRUE); 
2066
2067   if (!dbus_message_set_reply_serial (message,
2068                                       serial))
2069     {
2070       dbus_message_unref (message);
2071       message = NULL;
2072       goto out;
2073     }
2074
2075   if (error_msg != NULL)
2076     {
2077       DBusMessageIter iter;
2078
2079       dbus_message_iter_init_append (message, &iter);
2080       if (!dbus_message_iter_append_basic (&iter,
2081                                            DBUS_TYPE_STRING,
2082                                            &error_msg))
2083         {
2084           dbus_message_unref (message);
2085           message = NULL;
2086           goto out;
2087         }
2088     }
2089
2090  out:
2091   return message;
2092 }
2093
2094
2095 /* This is slightly strange since we can pop a message here without
2096  * the dispatch lock.
2097  */
2098 static DBusMessage*
2099 check_for_reply_unlocked (DBusConnection *connection,
2100                           dbus_uint32_t   client_serial)
2101 {
2102   DBusList *link;
2103
2104   HAVE_LOCK_CHECK (connection);
2105   
2106   link = _dbus_list_get_first_link (&connection->incoming_messages);
2107
2108   while (link != NULL)
2109     {
2110       DBusMessage *reply = link->data;
2111
2112       if (dbus_message_get_reply_serial (reply) == client_serial)
2113         {
2114           _dbus_list_remove_link (&connection->incoming_messages, link);
2115           connection->n_incoming  -= 1;
2116           return reply;
2117         }
2118       link = _dbus_list_get_next_link (&connection->incoming_messages, link);
2119     }
2120
2121   return NULL;
2122 }
2123
2124 static void
2125 connection_timeout_and_complete_all_pending_calls_unlocked (DBusConnection *connection)
2126 {
2127    /* We can't iterate over the hash in the normal way since we'll be
2128     * dropping the lock for each item. So we restart the
2129     * iter each time as we drain the hash table.
2130     */
2131    
2132    while (_dbus_hash_table_get_n_entries (connection->pending_replies) > 0)
2133     {
2134       DBusPendingCall *pending;
2135       DBusHashIter iter;
2136       
2137       _dbus_hash_iter_init (connection->pending_replies, &iter);
2138       _dbus_hash_iter_next (&iter);
2139        
2140       pending = _dbus_hash_iter_get_value (&iter);
2141       _dbus_pending_call_ref_unlocked (pending);
2142        
2143       _dbus_pending_call_queue_timeout_error_unlocked (pending, 
2144                                                        connection);
2145       _dbus_connection_remove_timeout_unlocked (connection,
2146                                                 _dbus_pending_call_get_timeout_unlocked (pending));
2147       _dbus_pending_call_set_timeout_added_unlocked (pending, FALSE);       
2148       _dbus_hash_iter_remove_entry (&iter);
2149
2150       _dbus_pending_call_unref_and_unlock (pending);
2151       CONNECTION_LOCK (connection);
2152     }
2153   HAVE_LOCK_CHECK (connection);
2154 }
2155
2156 static void
2157 complete_pending_call_and_unlock (DBusConnection  *connection,
2158                                   DBusPendingCall *pending,
2159                                   DBusMessage     *message)
2160 {
2161   _dbus_pending_call_set_reply_unlocked (pending, message);
2162   _dbus_pending_call_ref_unlocked (pending); /* in case there's no app with a ref held */
2163   _dbus_connection_detach_pending_call_and_unlock (connection, pending);
2164  
2165   /* Must be called unlocked since it invokes app callback */
2166   _dbus_pending_call_complete (pending);
2167   dbus_pending_call_unref (pending);
2168 }
2169
2170 static dbus_bool_t
2171 check_for_reply_and_update_dispatch_unlocked (DBusConnection  *connection,
2172                                               DBusPendingCall *pending)
2173 {
2174   DBusMessage *reply;
2175   DBusDispatchStatus status;
2176
2177   reply = check_for_reply_unlocked (connection, 
2178                                     _dbus_pending_call_get_reply_serial_unlocked (pending));
2179   if (reply != NULL)
2180     {
2181       _dbus_verbose ("%s checked for reply\n", _DBUS_FUNCTION_NAME);
2182
2183       _dbus_verbose ("dbus_connection_send_with_reply_and_block(): got reply\n");
2184
2185       complete_pending_call_and_unlock (connection, pending, reply);
2186       dbus_message_unref (reply);
2187
2188       CONNECTION_LOCK (connection);
2189       status = _dbus_connection_get_dispatch_status_unlocked (connection);
2190       _dbus_connection_update_dispatch_status_and_unlock (connection, status);
2191       dbus_pending_call_unref (pending);
2192
2193       return TRUE;
2194     }
2195
2196   return FALSE;
2197 }
2198
2199 /**
2200  * Blocks until a pending call times out or gets a reply.
2201  *
2202  * Does not re-enter the main loop or run filter/path-registered
2203  * callbacks. The reply to the message will not be seen by
2204  * filter callbacks.
2205  *
2206  * Returns immediately if pending call already got a reply.
2207  * 
2208  * @todo could use performance improvements (it keeps scanning
2209  * the whole message queue for example)
2210  *
2211  * @param pending the pending call we block for a reply on
2212  */
2213 void
2214 _dbus_connection_block_pending_call (DBusPendingCall *pending)
2215 {
2216   long start_tv_sec, start_tv_usec;
2217   long end_tv_sec, end_tv_usec;
2218   long tv_sec, tv_usec;
2219   DBusDispatchStatus status;
2220   DBusConnection *connection;
2221   dbus_uint32_t client_serial;
2222   int timeout_milliseconds;
2223
2224   _dbus_assert (pending != NULL);
2225
2226   if (dbus_pending_call_get_completed (pending))
2227     return;
2228
2229   dbus_pending_call_ref (pending); /* necessary because the call could be canceled */
2230
2231   connection = _dbus_pending_call_get_connection_and_lock (pending);
2232   
2233   /* Flush message queue - note, can affect dispatch status */
2234   _dbus_connection_flush_unlocked (connection);
2235
2236   client_serial = _dbus_pending_call_get_reply_serial_unlocked (pending);
2237
2238   /* note that timeout_milliseconds is limited to a smallish value
2239    * in _dbus_pending_call_new() so overflows aren't possible
2240    * below
2241    */
2242   timeout_milliseconds = dbus_timeout_get_interval (_dbus_pending_call_get_timeout_unlocked (pending));
2243   
2244   _dbus_get_current_time (&start_tv_sec, &start_tv_usec);
2245   end_tv_sec = start_tv_sec + timeout_milliseconds / 1000;
2246   end_tv_usec = start_tv_usec + (timeout_milliseconds % 1000) * 1000;
2247   end_tv_sec += end_tv_usec / _DBUS_USEC_PER_SECOND;
2248   end_tv_usec = end_tv_usec % _DBUS_USEC_PER_SECOND;
2249
2250   _dbus_verbose ("dbus_connection_send_with_reply_and_block(): will block %d milliseconds for reply serial %u from %ld sec %ld usec to %ld sec %ld usec\n",
2251                  timeout_milliseconds,
2252                  client_serial,
2253                  start_tv_sec, start_tv_usec,
2254                  end_tv_sec, end_tv_usec);
2255
2256   /* check to see if we already got the data off the socket */
2257   /* from another blocked pending call */
2258   if (check_for_reply_and_update_dispatch_unlocked (connection, pending))
2259     return;
2260
2261   /* Now we wait... */
2262   /* always block at least once as we know we don't have the reply yet */
2263   _dbus_connection_do_iteration_unlocked (connection,
2264                                           DBUS_ITERATION_DO_READING |
2265                                           DBUS_ITERATION_BLOCK,
2266                                           timeout_milliseconds);
2267
2268  recheck_status:
2269
2270   _dbus_verbose ("%s top of recheck\n", _DBUS_FUNCTION_NAME);
2271   
2272   HAVE_LOCK_CHECK (connection);
2273   
2274   /* queue messages and get status */
2275
2276   status = _dbus_connection_get_dispatch_status_unlocked (connection);
2277
2278   /* the get_completed() is in case a dispatch() while we were blocking
2279    * got the reply instead of us.
2280    */
2281   if (_dbus_pending_call_get_completed_unlocked (pending))
2282     {
2283       _dbus_verbose ("Pending call completed by dispatch in %s\n", _DBUS_FUNCTION_NAME);
2284       _dbus_connection_update_dispatch_status_and_unlock (connection, status);
2285       dbus_pending_call_unref (pending);
2286       return;
2287     }
2288   
2289   if (status == DBUS_DISPATCH_DATA_REMAINS) {
2290     if (check_for_reply_and_update_dispatch_unlocked (connection, pending))
2291       return;
2292   }
2293   
2294   _dbus_get_current_time (&tv_sec, &tv_usec);
2295   
2296   if (!_dbus_connection_get_is_connected_unlocked (connection))
2297     {
2298       DBusMessage *error_msg;
2299
2300       error_msg = generate_local_error_message (client_serial,
2301                                                 DBUS_ERROR_DISCONNECTED, 
2302                                                 "Connection was disconnected before a reply was received"); 
2303
2304       /* on OOM error_msg is set to NULL */
2305       complete_pending_call_and_unlock (connection, pending, error_msg);
2306       dbus_pending_call_unref (pending);
2307       return;
2308     }
2309   else if (tv_sec < start_tv_sec)
2310     _dbus_verbose ("dbus_connection_send_with_reply_and_block(): clock set backward\n");
2311   else if (connection->disconnect_message_link == NULL)
2312     _dbus_verbose ("dbus_connection_send_with_reply_and_block(): disconnected\n");
2313   else if (tv_sec < end_tv_sec ||
2314            (tv_sec == end_tv_sec && tv_usec < end_tv_usec))
2315     {
2316       timeout_milliseconds = (end_tv_sec - tv_sec) * 1000 +
2317         (end_tv_usec - tv_usec) / 1000;
2318       _dbus_verbose ("dbus_connection_send_with_reply_and_block(): %d milliseconds remain\n", timeout_milliseconds);
2319       _dbus_assert (timeout_milliseconds >= 0);
2320       
2321       if (status == DBUS_DISPATCH_NEED_MEMORY)
2322         {
2323           /* Try sleeping a bit, as we aren't sure we need to block for reading,
2324            * we may already have a reply in the buffer and just can't process
2325            * it.
2326            */
2327           _dbus_verbose ("dbus_connection_send_with_reply_and_block() waiting for more memory\n");
2328
2329           _dbus_memory_pause_based_on_timeout (timeout_milliseconds);
2330         }
2331       else
2332         {          
2333           /* block again, we don't have the reply buffered yet. */
2334           _dbus_connection_do_iteration_unlocked (connection,
2335                                                   DBUS_ITERATION_DO_READING |
2336                                                   DBUS_ITERATION_BLOCK,
2337                                                   timeout_milliseconds);
2338         }
2339
2340       goto recheck_status;
2341     }
2342
2343   _dbus_verbose ("dbus_connection_send_with_reply_and_block(): Waited %ld milliseconds and got no reply\n",
2344                  (tv_sec - start_tv_sec) * 1000 + (tv_usec - start_tv_usec) / 1000);
2345
2346   _dbus_assert (!_dbus_pending_call_get_completed_unlocked (pending));
2347   
2348   /* unlock and call user code */
2349   complete_pending_call_and_unlock (connection, pending, NULL);
2350
2351   /* update user code on dispatch status */
2352   CONNECTION_LOCK (connection);
2353   status = _dbus_connection_get_dispatch_status_unlocked (connection);
2354   _dbus_connection_update_dispatch_status_and_unlock (connection, status);
2355   dbus_pending_call_unref (pending);
2356 }
2357
2358 /** @} */
2359
2360 /**
2361  * @addtogroup DBusConnection
2362  *
2363  * @{
2364  */
2365
2366 /**
2367  * Gets a connection to a remote address. If a connection to the given
2368  * address already exists, returns the existing connection with its
2369  * reference count incremented.  Otherwise, returns a new connection
2370  * and saves the new connection for possible re-use if a future call
2371  * to dbus_connection_open() asks to connect to the same server.
2372  *
2373  * Use dbus_connection_open_private() to get a dedicated connection
2374  * not shared with other callers of dbus_connection_open().
2375  *
2376  * If the open fails, the function returns #NULL, and provides a
2377  * reason for the failure in the error parameter. Pass #NULL for the
2378  * error parameter if you aren't interested in the reason for
2379  * failure.
2380  *
2381  * Because this connection is shared, no user of the connection
2382  * may call dbus_connection_close(). However, when you are done with the
2383  * connection you should call dbus_connection_unref().
2384  *
2385  * @note Prefer dbus_connection_open() to dbus_connection_open_private()
2386  * unless you have good reason; connections are expensive enough
2387  * that it's wasteful to create lots of connections to the same
2388  * server.
2389  * 
2390  * @param address the address.
2391  * @param error address where an error can be returned.
2392  * @returns new connection, or #NULL on failure.
2393  */
2394 DBusConnection*
2395 dbus_connection_open (const char     *address,
2396                       DBusError      *error)
2397 {
2398   DBusConnection *connection;
2399
2400   _dbus_return_val_if_fail (address != NULL, NULL);
2401   _dbus_return_val_if_error_is_set (error, NULL);
2402
2403   connection = _dbus_connection_open_internal (address,
2404                                                TRUE,
2405                                                error);
2406
2407   return connection;
2408 }
2409
2410 /**
2411  * Opens a new, dedicated connection to a remote address. Unlike
2412  * dbus_connection_open(), always creates a new connection.
2413  * This connection will not be saved or recycled by libdbus.
2414  *
2415  * If the open fails, the function returns #NULL, and provides a
2416  * reason for the failure in the error parameter. Pass #NULL for the
2417  * error parameter if you aren't interested in the reason for
2418  * failure.
2419  *
2420  * When you are done with this connection, you must
2421  * dbus_connection_close() to disconnect it,
2422  * and dbus_connection_unref() to free the connection object.
2423  * 
2424  * (The dbus_connection_close() can be skipped if the
2425  * connection is already known to be disconnected, for example
2426  * if you are inside a handler for the Disconnected signal.)
2427  *
2428  * @note Prefer dbus_connection_open() to dbus_connection_open_private()
2429  * unless you have good reason; connections are expensive enough
2430  * that it's wasteful to create lots of connections to the same
2431  * server.
2432  *
2433  * @param address the address.
2434  * @param error address where an error can be returned.
2435  * @returns new connection, or #NULL on failure.
2436  */
2437 DBusConnection*
2438 dbus_connection_open_private (const char     *address,
2439                               DBusError      *error)
2440 {
2441   DBusConnection *connection;
2442
2443   _dbus_return_val_if_fail (address != NULL, NULL);
2444   _dbus_return_val_if_error_is_set (error, NULL);
2445
2446   connection = _dbus_connection_open_internal (address,
2447                                                FALSE,
2448                                                error);
2449
2450   return connection;
2451 }
2452
2453 /**
2454  * Increments the reference count of a DBusConnection.
2455  *
2456  * @param connection the connection.
2457  * @returns the connection.
2458  */
2459 DBusConnection *
2460 dbus_connection_ref (DBusConnection *connection)
2461 {
2462   _dbus_return_val_if_fail (connection != NULL, NULL);
2463   _dbus_return_val_if_fail (connection->generation == _dbus_current_generation, NULL);
2464   
2465   /* The connection lock is better than the global
2466    * lock in the atomic increment fallback
2467    */
2468   
2469 #ifdef DBUS_HAVE_ATOMIC_INT
2470   _dbus_atomic_inc (&connection->refcount);
2471 #else
2472   CONNECTION_LOCK (connection);
2473   _dbus_assert (connection->refcount.value > 0);
2474
2475   connection->refcount.value += 1;
2476   CONNECTION_UNLOCK (connection);
2477 #endif
2478
2479   return connection;
2480 }
2481
2482 static void
2483 free_outgoing_message (void *element,
2484                        void *data)
2485 {
2486   DBusMessage *message = element;
2487   DBusConnection *connection = data;
2488
2489   _dbus_message_remove_size_counter (message,
2490                                      connection->outgoing_counter,
2491                                      NULL);
2492   dbus_message_unref (message);
2493 }
2494
2495 /* This is run without the mutex held, but after the last reference
2496  * to the connection has been dropped we should have no thread-related
2497  * problems
2498  */
2499 static void
2500 _dbus_connection_last_unref (DBusConnection *connection)
2501 {
2502   DBusList *link;
2503
2504   _dbus_verbose ("Finalizing connection %p\n", connection);
2505   
2506   _dbus_assert (connection->refcount.value == 0);
2507   
2508   /* You have to disconnect the connection before unref:ing it. Otherwise
2509    * you won't get the disconnected message.
2510    */
2511   _dbus_assert (!_dbus_transport_get_is_connected (connection->transport));
2512   _dbus_assert (connection->server_guid == NULL);
2513   
2514   /* ---- We're going to call various application callbacks here, hope it doesn't break anything... */
2515   _dbus_object_tree_free_all_unlocked (connection->objects);
2516   
2517   dbus_connection_set_dispatch_status_function (connection, NULL, NULL, NULL);
2518   dbus_connection_set_wakeup_main_function (connection, NULL, NULL, NULL);
2519   dbus_connection_set_unix_user_function (connection, NULL, NULL, NULL);
2520   
2521   _dbus_watch_list_free (connection->watches);
2522   connection->watches = NULL;
2523   
2524   _dbus_timeout_list_free (connection->timeouts);
2525   connection->timeouts = NULL;
2526
2527   _dbus_data_slot_list_free (&connection->slot_list);
2528   
2529   link = _dbus_list_get_first_link (&connection->filter_list);
2530   while (link != NULL)
2531     {
2532       DBusMessageFilter *filter = link->data;
2533       DBusList *next = _dbus_list_get_next_link (&connection->filter_list, link);
2534
2535       filter->function = NULL;
2536       _dbus_message_filter_unref (filter); /* calls app callback */
2537       link->data = NULL;
2538       
2539       link = next;
2540     }
2541   _dbus_list_clear (&connection->filter_list);
2542   
2543   /* ---- Done with stuff that invokes application callbacks */
2544
2545   _dbus_object_tree_unref (connection->objects);  
2546
2547   _dbus_hash_table_unref (connection->pending_replies);
2548   connection->pending_replies = NULL;
2549   
2550   _dbus_list_clear (&connection->filter_list);
2551   
2552   _dbus_list_foreach (&connection->outgoing_messages,
2553                       free_outgoing_message,
2554                       connection);
2555   _dbus_list_clear (&connection->outgoing_messages);
2556   
2557   _dbus_list_foreach (&connection->incoming_messages,
2558                       (DBusForeachFunction) dbus_message_unref,
2559                       NULL);
2560   _dbus_list_clear (&connection->incoming_messages);
2561
2562   _dbus_counter_unref (connection->outgoing_counter);
2563
2564   _dbus_transport_unref (connection->transport);
2565
2566   if (connection->disconnect_message_link)
2567     {
2568       DBusMessage *message = connection->disconnect_message_link->data;
2569       dbus_message_unref (message);
2570       _dbus_list_free_link (connection->disconnect_message_link);
2571     }
2572
2573   _dbus_list_clear (&connection->link_cache);
2574   
2575   _dbus_condvar_free_at_location (&connection->dispatch_cond);
2576   _dbus_condvar_free_at_location (&connection->io_path_cond);
2577
2578   _dbus_mutex_free_at_location (&connection->io_path_mutex);
2579   _dbus_mutex_free_at_location (&connection->dispatch_mutex);
2580
2581   _dbus_mutex_free_at_location (&connection->mutex);
2582   
2583   dbus_free (connection);
2584 }
2585
2586 /**
2587  * Decrements the reference count of a DBusConnection, and finalizes
2588  * it if the count reaches zero.
2589  *
2590  * Note: it is a bug to drop the last reference to a connection that
2591  * is still connected.
2592  *
2593  * For shared connections, libdbus will own a reference
2594  * as long as the connection is connected, so you can know that either
2595  * you don't have the last reference, or it's OK to drop the last reference.
2596  * Most connections are shared. dbus_connection_open() and dbus_bus_get()
2597  * return shared connections.
2598  *
2599  * For private connections, the creator of the connection must arrange for
2600  * dbus_connection_close() to be called prior to dropping the last reference.
2601  * Private connections come from dbus_connection_open_private() or dbus_bus_get_private().
2602  *
2603  * @param connection the connection.
2604  */
2605 void
2606 dbus_connection_unref (DBusConnection *connection)
2607 {
2608   dbus_bool_t last_unref;
2609
2610   _dbus_return_if_fail (connection != NULL);
2611   _dbus_return_if_fail (connection->generation == _dbus_current_generation);
2612   
2613   /* The connection lock is better than the global
2614    * lock in the atomic increment fallback
2615    */
2616   
2617 #ifdef DBUS_HAVE_ATOMIC_INT
2618   last_unref = (_dbus_atomic_dec (&connection->refcount) == 1);
2619 #else
2620   CONNECTION_LOCK (connection);
2621   
2622   _dbus_assert (connection->refcount.value > 0);
2623
2624   connection->refcount.value -= 1;
2625   last_unref = (connection->refcount.value == 0);
2626
2627 #if 0
2628   printf ("unref() connection %p count = %d\n", connection, connection->refcount.value);
2629 #endif
2630   
2631   CONNECTION_UNLOCK (connection);
2632 #endif
2633   
2634   if (last_unref)
2635     {
2636 #ifndef DBUS_DISABLE_CHECKS
2637       if (_dbus_transport_get_is_connected (connection->transport))
2638         {
2639           _dbus_warn_check_failed ("The last reference on a connection was dropped without closing the connection. This is a bug in an application. See dbus_connection_unref() documentation for details.\n%s",
2640                                    connection->shareable ?
2641                                    "Most likely, the application called unref() too many times and removed a reference belonging to libdbus, since this is a shared connection.\n" : 
2642                                     "Most likely, the application was supposed to call dbus_connection_close(), since this is a private connection.\n");
2643           return;
2644         }
2645 #endif
2646       _dbus_connection_last_unref (connection);
2647     }
2648 }
2649
2650 /*
2651  * Note that the transport can disconnect itself (other end drops us)
2652  * and in that case this function never runs. So this function must
2653  * not do anything more than disconnect the transport and update the
2654  * dispatch status.
2655  * 
2656  * If the transport self-disconnects, then we assume someone will
2657  * dispatch the connection to cause the dispatch status update.
2658  */
2659 static void
2660 _dbus_connection_close_possibly_shared_and_unlock (DBusConnection *connection)
2661 {
2662   DBusDispatchStatus status;
2663
2664   HAVE_LOCK_CHECK (connection);
2665   
2666   _dbus_verbose ("Disconnecting %p\n", connection);
2667
2668   /* We need to ref because update_dispatch_status_and_unlock will unref
2669    * the connection if it was shared and libdbus was the only remaining
2670    * refcount holder.
2671    */
2672   _dbus_connection_ref_unlocked (connection);
2673   
2674   _dbus_transport_disconnect (connection->transport);
2675
2676   /* This has the side effect of queuing the disconnect message link
2677    * (unless we don't have enough memory, possibly, so don't assert it).
2678    * After the disconnect message link is queued, dbus_bus_get/dbus_connection_open
2679    * should never again return the newly-disconnected connection.
2680    *
2681    * However, we only unref the shared connection and exit_on_disconnect when
2682    * the disconnect message reaches the head of the message queue,
2683    * NOT when it's first queued.
2684    */
2685   status = _dbus_connection_get_dispatch_status_unlocked (connection);
2686
2687   /* This calls out to user code */
2688   _dbus_connection_update_dispatch_status_and_unlock (connection, status);
2689
2690   /* Could also call out to user code */
2691   dbus_connection_unref (connection);
2692 }
2693
2694 /**
2695  * Closes a private connection, so no further data can be sent or received.
2696  * This disconnects the transport (such as a socket) underlying the
2697  * connection.
2698  *
2699  * Attempts to send messages after closing a connection are safe, but will result in
2700  * error replies generated locally in libdbus.
2701  * 
2702  * This function does not affect the connection's reference count.  It's
2703  * safe to close a connection more than once; all calls after the
2704  * first do nothing. It's impossible to "reopen" a connection, a
2705  * new connection must be created. This function may result in a call
2706  * to the DBusDispatchStatusFunction set with
2707  * dbus_connection_set_dispatch_status_function(), as the disconnect
2708  * message it generates needs to be dispatched.
2709  *
2710  * If a connection is dropped by the remote application, it will
2711  * close itself. 
2712  * 
2713  * You must close a connection prior to releasing the last reference to
2714  * the connection. If you dbus_connection_unref() for the last time
2715  * without closing the connection, the results are undefined; it
2716  * is a bug in your program and libdbus will try to print a warning.
2717  *
2718  * You may not close a shared connection. Connections created with
2719  * dbus_connection_open() or dbus_bus_get() are shared.
2720  * These connections are owned by libdbus, and applications should
2721  * only unref them, never close them. Applications can know it is
2722  * safe to unref these connections because libdbus will be holding a
2723  * reference as long as the connection is open. Thus, either the
2724  * connection is closed and it is OK to drop the last reference,
2725  * or the connection is open and the app knows it does not have the
2726  * last reference.
2727  *
2728  * Connections created with dbus_connection_open_private() or
2729  * dbus_bus_get_private() are not kept track of or referenced by
2730  * libdbus. The creator of these connections is responsible for
2731  * calling dbus_connection_close() prior to releasing the last
2732  * reference, if the connection is not already disconnected.
2733  *
2734  * @param connection the private (unshared) connection to close
2735  */
2736 void
2737 dbus_connection_close (DBusConnection *connection)
2738 {
2739   _dbus_return_if_fail (connection != NULL);
2740   _dbus_return_if_fail (connection->generation == _dbus_current_generation);
2741
2742   CONNECTION_LOCK (connection);
2743
2744 #ifndef DBUS_DISABLE_CHECKS
2745   if (connection->shareable)
2746     {
2747       CONNECTION_UNLOCK (connection);
2748
2749       _dbus_warn_check_failed ("Applications must not close shared connections - see dbus_connection_close() docs. This is a bug in the application.\n");
2750       return;
2751     }
2752 #endif
2753   
2754   _dbus_connection_close_possibly_shared_and_unlock (connection);
2755 }
2756
2757 static dbus_bool_t
2758 _dbus_connection_get_is_connected_unlocked (DBusConnection *connection)
2759 {
2760   HAVE_LOCK_CHECK (connection);
2761   return _dbus_transport_get_is_connected (connection->transport);
2762 }
2763
2764 /**
2765  * Gets whether the connection is currently open.  A connection may
2766  * become disconnected when the remote application closes its end, or
2767  * exits; a connection may also be disconnected with
2768  * dbus_connection_close().
2769  * 
2770  * There are not separate states for "closed" and "disconnected," the two
2771  * terms are synonymous. This function should really be called
2772  * get_is_open() but for historical reasons is not.
2773  *
2774  * @param connection the connection.
2775  * @returns #TRUE if the connection is still alive.
2776  */
2777 dbus_bool_t
2778 dbus_connection_get_is_connected (DBusConnection *connection)
2779 {
2780   dbus_bool_t res;
2781
2782   _dbus_return_val_if_fail (connection != NULL, FALSE);
2783   
2784   CONNECTION_LOCK (connection);
2785   res = _dbus_connection_get_is_connected_unlocked (connection);
2786   CONNECTION_UNLOCK (connection);
2787   
2788   return res;
2789 }
2790
2791 /**
2792  * Gets whether the connection was authenticated. (Note that
2793  * if the connection was authenticated then disconnected,
2794  * this function still returns #TRUE)
2795  *
2796  * @param connection the connection
2797  * @returns #TRUE if the connection was ever authenticated
2798  */
2799 dbus_bool_t
2800 dbus_connection_get_is_authenticated (DBusConnection *connection)
2801 {
2802   dbus_bool_t res;
2803
2804   _dbus_return_val_if_fail (connection != NULL, FALSE);
2805   
2806   CONNECTION_LOCK (connection);
2807   res = _dbus_transport_get_is_authenticated (connection->transport);
2808   CONNECTION_UNLOCK (connection);
2809   
2810   return res;
2811 }
2812
2813 /**
2814  * Gets whether the connection is not authenticated as a specific
2815  * user.  If the connection is not authenticated, this function
2816  * returns #TRUE, and if it is authenticated but as an anonymous user,
2817  * it returns #TRUE.  If it is authenticated as a specific user, then
2818  * this returns #FALSE. (Note that if the connection was authenticated
2819  * as anonymous then disconnected, this function still returns #TRUE.)
2820  *
2821  * If the connection is not anonymous, you can use
2822  * dbus_connection_get_unix_user() and
2823  * dbus_connection_get_windows_user() to see who it's authorized as.
2824  *
2825  * If you want to prevent non-anonymous authorization, use
2826  * dbus_server_set_auth_mechanisms() to remove the mechanisms that
2827  * allow proving user identity (i.e. only allow the ANONYMOUS
2828  * mechanism).
2829  * 
2830  * @param connection the connection
2831  * @returns #TRUE if not authenticated or authenticated as anonymous 
2832  */
2833 dbus_bool_t
2834 dbus_connection_get_is_anonymous (DBusConnection *connection)
2835 {
2836   dbus_bool_t res;
2837
2838   _dbus_return_val_if_fail (connection != NULL, FALSE);
2839   
2840   CONNECTION_LOCK (connection);
2841   res = _dbus_transport_get_is_anonymous (connection->transport);
2842   CONNECTION_UNLOCK (connection);
2843   
2844   return res;
2845 }
2846
2847 /**
2848  * Set whether _exit() should be called when the connection receives a
2849  * disconnect signal. The call to _exit() comes after any handlers for
2850  * the disconnect signal run; handlers can cancel the exit by calling
2851  * this function.
2852  *
2853  * By default, exit_on_disconnect is #FALSE; but for message bus
2854  * connections returned from dbus_bus_get() it will be toggled on
2855  * by default.
2856  *
2857  * @param connection the connection
2858  * @param exit_on_disconnect #TRUE if _exit() should be called after a disconnect signal
2859  */
2860 void
2861 dbus_connection_set_exit_on_disconnect (DBusConnection *connection,
2862                                         dbus_bool_t     exit_on_disconnect)
2863 {
2864   _dbus_return_if_fail (connection != NULL);
2865
2866   CONNECTION_LOCK (connection);
2867   connection->exit_on_disconnect = exit_on_disconnect != FALSE;
2868   CONNECTION_UNLOCK (connection);
2869 }
2870
2871 /**
2872  * Preallocates resources needed to send a message, allowing the message 
2873  * to be sent without the possibility of memory allocation failure.
2874  * Allows apps to create a future guarantee that they can send
2875  * a message regardless of memory shortages.
2876  *
2877  * @param connection the connection we're preallocating for.
2878  * @returns the preallocated resources, or #NULL
2879  */
2880 DBusPreallocatedSend*
2881 dbus_connection_preallocate_send (DBusConnection *connection)
2882 {
2883   DBusPreallocatedSend *preallocated;
2884
2885   _dbus_return_val_if_fail (connection != NULL, NULL);
2886
2887   CONNECTION_LOCK (connection);
2888   
2889   preallocated =
2890     _dbus_connection_preallocate_send_unlocked (connection);
2891
2892   CONNECTION_UNLOCK (connection);
2893
2894   return preallocated;
2895 }
2896
2897 /**
2898  * Frees preallocated message-sending resources from
2899  * dbus_connection_preallocate_send(). Should only
2900  * be called if the preallocated resources are not used
2901  * to send a message.
2902  *
2903  * @param connection the connection
2904  * @param preallocated the resources
2905  */
2906 void
2907 dbus_connection_free_preallocated_send (DBusConnection       *connection,
2908                                         DBusPreallocatedSend *preallocated)
2909 {
2910   _dbus_return_if_fail (connection != NULL);
2911   _dbus_return_if_fail (preallocated != NULL);  
2912   _dbus_return_if_fail (connection == preallocated->connection);
2913
2914   _dbus_list_free_link (preallocated->queue_link);
2915   _dbus_counter_unref (preallocated->counter_link->data);
2916   _dbus_list_free_link (preallocated->counter_link);
2917   dbus_free (preallocated);
2918 }
2919
2920 /**
2921  * Sends a message using preallocated resources. This function cannot fail.
2922  * It works identically to dbus_connection_send() in other respects.
2923  * Preallocated resources comes from dbus_connection_preallocate_send().
2924  * This function "consumes" the preallocated resources, they need not
2925  * be freed separately.
2926  *
2927  * @param connection the connection
2928  * @param preallocated the preallocated resources
2929  * @param message the message to send
2930  * @param client_serial return location for client serial assigned to the message
2931  */
2932 void
2933 dbus_connection_send_preallocated (DBusConnection       *connection,
2934                                    DBusPreallocatedSend *preallocated,
2935                                    DBusMessage          *message,
2936                                    dbus_uint32_t        *client_serial)
2937 {
2938   _dbus_return_if_fail (connection != NULL);
2939   _dbus_return_if_fail (preallocated != NULL);
2940   _dbus_return_if_fail (message != NULL);
2941   _dbus_return_if_fail (preallocated->connection == connection);
2942   _dbus_return_if_fail (dbus_message_get_type (message) != DBUS_MESSAGE_TYPE_METHOD_CALL ||
2943                         dbus_message_get_member (message) != NULL);
2944   _dbus_return_if_fail (dbus_message_get_type (message) != DBUS_MESSAGE_TYPE_SIGNAL ||
2945                         (dbus_message_get_interface (message) != NULL &&
2946                          dbus_message_get_member (message) != NULL));
2947   
2948   CONNECTION_LOCK (connection);
2949   _dbus_connection_send_preallocated_and_unlock (connection,
2950                                                  preallocated,
2951                                                  message, client_serial);
2952 }
2953
2954 static dbus_bool_t
2955 _dbus_connection_send_unlocked_no_update (DBusConnection *connection,
2956                                           DBusMessage    *message,
2957                                           dbus_uint32_t  *client_serial)
2958 {
2959   DBusPreallocatedSend *preallocated;
2960
2961   _dbus_assert (connection != NULL);
2962   _dbus_assert (message != NULL);
2963   
2964   preallocated = _dbus_connection_preallocate_send_unlocked (connection);
2965   if (preallocated == NULL)
2966     return FALSE;
2967
2968   _dbus_connection_send_preallocated_unlocked_no_update (connection,
2969                                                          preallocated,
2970                                                          message,
2971                                                          client_serial);
2972   return TRUE;
2973 }
2974
2975 /**
2976  * Adds a message to the outgoing message queue. Does not block to
2977  * write the message to the network; that happens asynchronously. To
2978  * force the message to be written, call dbus_connection_flush().
2979  * Because this only queues the message, the only reason it can
2980  * fail is lack of memory. Even if the connection is disconnected,
2981  * no error will be returned.
2982  *
2983  * If the function fails due to lack of memory, it returns #FALSE.
2984  * The function will never fail for other reasons; even if the
2985  * connection is disconnected, you can queue an outgoing message,
2986  * though obviously it won't be sent.
2987  *
2988  * The message serial is used by the remote application to send a
2989  * reply; see dbus_message_get_serial() or the D-Bus specification.
2990  * 
2991  * @param connection the connection.
2992  * @param message the message to write.
2993  * @param serial return location for message serial, or #NULL if you don't care
2994  * @returns #TRUE on success.
2995  */
2996 dbus_bool_t
2997 dbus_connection_send (DBusConnection *connection,
2998                       DBusMessage    *message,
2999                       dbus_uint32_t  *serial)
3000 {
3001   _dbus_return_val_if_fail (connection != NULL, FALSE);
3002   _dbus_return_val_if_fail (message != NULL, FALSE);
3003
3004   CONNECTION_LOCK (connection);
3005
3006   return _dbus_connection_send_and_unlock (connection,
3007                                            message,
3008                                            serial);
3009 }
3010
3011 static dbus_bool_t
3012 reply_handler_timeout (void *data)
3013 {
3014   DBusConnection *connection;
3015   DBusDispatchStatus status;
3016   DBusPendingCall *pending = data;
3017
3018   connection = _dbus_pending_call_get_connection_and_lock (pending);
3019
3020   _dbus_pending_call_queue_timeout_error_unlocked (pending, 
3021                                                    connection);
3022   _dbus_connection_remove_timeout_unlocked (connection,
3023                                             _dbus_pending_call_get_timeout_unlocked (pending));
3024   _dbus_pending_call_set_timeout_added_unlocked (pending, FALSE);
3025
3026   _dbus_verbose ("%s middle\n", _DBUS_FUNCTION_NAME);
3027   status = _dbus_connection_get_dispatch_status_unlocked (connection);
3028
3029   /* Unlocks, and calls out to user code */
3030   _dbus_connection_update_dispatch_status_and_unlock (connection, status);
3031   
3032   return TRUE;
3033 }
3034
3035 /**
3036  * Queues a message to send, as with dbus_connection_send(),
3037  * but also returns a #DBusPendingCall used to receive a reply to the
3038  * message. If no reply is received in the given timeout_milliseconds,
3039  * this function expires the pending reply and generates a synthetic
3040  * error reply (generated in-process, not by the remote application)
3041  * indicating that a timeout occurred.
3042  *
3043  * A #DBusPendingCall will see a reply message before any filters or
3044  * registered object path handlers. See dbus_connection_dispatch() for
3045  * details on when handlers are run.
3046  *
3047  * A #DBusPendingCall will always see exactly one reply message,
3048  * unless it's cancelled with dbus_pending_call_cancel().
3049  * 
3050  * If #NULL is passed for the pending_return, the #DBusPendingCall
3051  * will still be generated internally, and used to track
3052  * the message reply timeout. This means a timeout error will
3053  * occur if no reply arrives, unlike with dbus_connection_send().
3054  *
3055  * If -1 is passed for the timeout, a sane default timeout is used. -1
3056  * is typically the best value for the timeout for this reason, unless
3057  * you want a very short or very long timeout.  There is no way to
3058  * avoid a timeout entirely, other than passing INT_MAX for the
3059  * timeout to mean "very long timeout." libdbus clamps an INT_MAX
3060  * timeout down to a few hours timeout though.
3061  *
3062  * @warning if the connection is disconnected, the #DBusPendingCall
3063  * will be set to #NULL, so be careful with this.
3064  * 
3065  * @param connection the connection
3066  * @param message the message to send
3067  * @param pending_return return location for a #DBusPendingCall object, or #NULL if connection is disconnected
3068  * @param timeout_milliseconds timeout in milliseconds or -1 for default
3069  * @returns #FALSE if no memory, #TRUE otherwise.
3070  *
3071  */
3072 dbus_bool_t
3073 dbus_connection_send_with_reply (DBusConnection     *connection,
3074                                  DBusMessage        *message,
3075                                  DBusPendingCall   **pending_return,
3076                                  int                 timeout_milliseconds)
3077 {
3078   DBusPendingCall *pending;
3079   dbus_int32_t serial = -1;
3080   DBusDispatchStatus status;
3081
3082   _dbus_return_val_if_fail (connection != NULL, FALSE);
3083   _dbus_return_val_if_fail (message != NULL, FALSE);
3084   _dbus_return_val_if_fail (timeout_milliseconds >= 0 || timeout_milliseconds == -1, FALSE);
3085
3086   if (pending_return)
3087     *pending_return = NULL;
3088
3089   CONNECTION_LOCK (connection);
3090
3091    if (!_dbus_connection_get_is_connected_unlocked (connection))
3092     {
3093       CONNECTION_UNLOCK (connection);
3094
3095       *pending_return = NULL;
3096
3097       return TRUE;
3098     }
3099
3100   pending = _dbus_pending_call_new_unlocked (connection,
3101                                              timeout_milliseconds,
3102                                              reply_handler_timeout);
3103
3104   if (pending == NULL)
3105     {
3106       CONNECTION_UNLOCK (connection);
3107       return FALSE;
3108     }
3109
3110   /* Assign a serial to the message */
3111   serial = dbus_message_get_serial (message);
3112   if (serial == 0)
3113     {
3114       serial = _dbus_connection_get_next_client_serial (connection);
3115       _dbus_message_set_serial (message, serial);
3116     }
3117
3118   if (!_dbus_pending_call_set_timeout_error_unlocked (pending, message, serial))
3119     goto error;
3120     
3121   /* Insert the serial in the pending replies hash;
3122    * hash takes a refcount on DBusPendingCall.
3123    * Also, add the timeout.
3124    */
3125   if (!_dbus_connection_attach_pending_call_unlocked (connection,
3126                                                       pending))
3127     goto error;
3128  
3129   if (!_dbus_connection_send_unlocked_no_update (connection, message, NULL))
3130     {
3131       _dbus_connection_detach_pending_call_and_unlock (connection,
3132                                                        pending);
3133       goto error_unlocked;
3134     }
3135
3136   if (pending_return)
3137     *pending_return = pending; /* hand off refcount */
3138   else
3139     {
3140       _dbus_connection_detach_pending_call_unlocked (connection, pending);
3141       /* we still have a ref to the pending call in this case, we unref
3142        * after unlocking, below
3143        */
3144     }
3145
3146   status = _dbus_connection_get_dispatch_status_unlocked (connection);
3147
3148   /* this calls out to user code */
3149   _dbus_connection_update_dispatch_status_and_unlock (connection, status);
3150
3151   if (pending_return == NULL)
3152     dbus_pending_call_unref (pending);
3153   
3154   return TRUE;
3155
3156  error:
3157   CONNECTION_UNLOCK (connection);
3158  error_unlocked:
3159   dbus_pending_call_unref (pending);
3160   return FALSE;
3161 }
3162
3163 /**
3164  * Sends a message and blocks a certain time period while waiting for
3165  * a reply.  This function does not reenter the main loop,
3166  * i.e. messages other than the reply are queued up but not
3167  * processed. This function is used to invoke method calls on a
3168  * remote object.
3169  * 
3170  * If a normal reply is received, it is returned, and removed from the
3171  * incoming message queue. If it is not received, #NULL is returned
3172  * and the error is set to #DBUS_ERROR_NO_REPLY.  If an error reply is
3173  * received, it is converted to a #DBusError and returned as an error,
3174  * then the reply message is deleted and #NULL is returned. If
3175  * something else goes wrong, result is set to whatever is
3176  * appropriate, such as #DBUS_ERROR_NO_MEMORY or
3177  * #DBUS_ERROR_DISCONNECTED.
3178  *
3179  * @warning While this function blocks the calling thread will not be
3180  * processing the incoming message queue. This means you can end up
3181  * deadlocked if the application you're talking to needs you to reply
3182  * to a method. To solve this, either avoid the situation, block in a
3183  * separate thread from the main connection-dispatching thread, or use
3184  * dbus_pending_call_set_notify() to avoid blocking.
3185  *
3186  * @param connection the connection
3187  * @param message the message to send
3188  * @param timeout_milliseconds timeout in milliseconds or -1 for default
3189  * @param error return location for error message
3190  * @returns the message that is the reply or #NULL with an error code if the
3191  * function fails.
3192  */
3193 DBusMessage*
3194 dbus_connection_send_with_reply_and_block (DBusConnection     *connection,
3195                                            DBusMessage        *message,
3196                                            int                 timeout_milliseconds,
3197                                            DBusError          *error)
3198 {
3199   DBusMessage *reply;
3200   DBusPendingCall *pending;
3201   
3202   _dbus_return_val_if_fail (connection != NULL, NULL);
3203   _dbus_return_val_if_fail (message != NULL, NULL);
3204   _dbus_return_val_if_fail (timeout_milliseconds >= 0 || timeout_milliseconds == -1, NULL);
3205   _dbus_return_val_if_error_is_set (error, NULL);
3206   
3207   if (!dbus_connection_send_with_reply (connection, message,
3208                                         &pending, timeout_milliseconds))
3209     {
3210       _DBUS_SET_OOM (error);
3211       return NULL;
3212     }
3213
3214   if (pending == NULL)
3215     {
3216       dbus_set_error (error, DBUS_ERROR_DISCONNECTED, "Connection is closed");
3217       return NULL;
3218     }
3219   
3220   dbus_pending_call_block (pending);
3221
3222   reply = dbus_pending_call_steal_reply (pending);
3223   dbus_pending_call_unref (pending);
3224
3225   /* call_complete_and_unlock() called from pending_call_block() should
3226    * always fill this in.
3227    */
3228   _dbus_assert (reply != NULL);
3229   
3230    if (dbus_set_error_from_message (error, reply))
3231     {
3232       dbus_message_unref (reply);
3233       return NULL;
3234     }
3235   else
3236     return reply;
3237 }
3238
3239 /**
3240  * Blocks until the outgoing message queue is empty.
3241  * Assumes connection lock already held.
3242  *
3243  * If you call this, you MUST call update_dispatch_status afterword...
3244  * 
3245  * @param connection the connection.
3246  */
3247 DBusDispatchStatus
3248 _dbus_connection_flush_unlocked (DBusConnection *connection)
3249 {
3250   /* We have to specify DBUS_ITERATION_DO_READING here because
3251    * otherwise we could have two apps deadlock if they are both doing
3252    * a flush(), and the kernel buffers fill up. This could change the
3253    * dispatch status.
3254    */
3255   DBusDispatchStatus status;
3256
3257   HAVE_LOCK_CHECK (connection);
3258   
3259   while (connection->n_outgoing > 0 &&
3260          _dbus_connection_get_is_connected_unlocked (connection))
3261     {
3262       _dbus_verbose ("doing iteration in %s\n", _DBUS_FUNCTION_NAME);
3263       HAVE_LOCK_CHECK (connection);
3264       _dbus_connection_do_iteration_unlocked (connection,
3265                                               DBUS_ITERATION_DO_READING |
3266                                               DBUS_ITERATION_DO_WRITING |
3267                                               DBUS_ITERATION_BLOCK,
3268                                               -1);
3269     }
3270
3271   HAVE_LOCK_CHECK (connection);
3272   _dbus_verbose ("%s middle\n", _DBUS_FUNCTION_NAME);
3273   status = _dbus_connection_get_dispatch_status_unlocked (connection);
3274
3275   HAVE_LOCK_CHECK (connection);
3276   return status;
3277 }
3278
3279 /**
3280  * Blocks until the outgoing message queue is empty.
3281  *
3282  * @param connection the connection.
3283  */
3284 void
3285 dbus_connection_flush (DBusConnection *connection)
3286 {
3287   /* We have to specify DBUS_ITERATION_DO_READING here because
3288    * otherwise we could have two apps deadlock if they are both doing
3289    * a flush(), and the kernel buffers fill up. This could change the
3290    * dispatch status.
3291    */
3292   DBusDispatchStatus status;
3293
3294   _dbus_return_if_fail (connection != NULL);
3295   
3296   CONNECTION_LOCK (connection);
3297
3298   status = _dbus_connection_flush_unlocked (connection);
3299   
3300   HAVE_LOCK_CHECK (connection);
3301   /* Unlocks and calls out to user code */
3302   _dbus_connection_update_dispatch_status_and_unlock (connection, status);
3303
3304   _dbus_verbose ("%s end\n", _DBUS_FUNCTION_NAME);
3305 }
3306
3307 /**
3308  * This function implements dbus_connection_read_write_dispatch() and
3309  * dbus_connection_read_write() (they pass a different value for the
3310  * dispatch parameter).
3311  * 
3312  * @param connection the connection
3313  * @param timeout_milliseconds max time to block or -1 for infinite
3314  * @param dispatch dispatch new messages or leave them on the incoming queue
3315  * @returns #TRUE if the disconnect message has not been processed
3316  */
3317 static dbus_bool_t
3318 _dbus_connection_read_write_dispatch (DBusConnection *connection,
3319                                      int             timeout_milliseconds, 
3320                                      dbus_bool_t     dispatch)
3321 {
3322   DBusDispatchStatus dstatus;
3323   dbus_bool_t no_progress_possible;
3324   
3325   dstatus = dbus_connection_get_dispatch_status (connection);
3326
3327   if (dispatch && dstatus == DBUS_DISPATCH_DATA_REMAINS)
3328     {
3329       _dbus_verbose ("doing dispatch in %s\n", _DBUS_FUNCTION_NAME);
3330       dbus_connection_dispatch (connection);
3331       CONNECTION_LOCK (connection);
3332     }
3333   else if (dstatus == DBUS_DISPATCH_NEED_MEMORY)
3334     {
3335       _dbus_verbose ("pausing for memory in %s\n", _DBUS_FUNCTION_NAME);
3336       _dbus_memory_pause_based_on_timeout (timeout_milliseconds);
3337       CONNECTION_LOCK (connection);
3338     }
3339   else
3340     {
3341       CONNECTION_LOCK (connection);
3342       if (_dbus_connection_get_is_connected_unlocked (connection))
3343         {
3344           _dbus_verbose ("doing iteration in %s\n", _DBUS_FUNCTION_NAME);
3345           _dbus_connection_do_iteration_unlocked (connection,
3346                                                   DBUS_ITERATION_DO_READING |
3347                                                   DBUS_ITERATION_DO_WRITING |
3348                                                   DBUS_ITERATION_BLOCK,
3349                                                   timeout_milliseconds);
3350         }
3351     }
3352   
3353   HAVE_LOCK_CHECK (connection);
3354   /* If we can dispatch, we can make progress until the Disconnected message
3355    * has been processed; if we can only read/write, we can make progress
3356    * as long as the transport is open.
3357    */
3358   if (dispatch)
3359     no_progress_possible = connection->n_incoming == 0 &&
3360       connection->disconnect_message_link == NULL;
3361   else
3362     no_progress_possible = _dbus_connection_get_is_connected_unlocked (connection);
3363   CONNECTION_UNLOCK (connection);
3364   return !no_progress_possible; /* TRUE if we can make more progress */
3365 }
3366
3367
3368 /**
3369  * This function is intended for use with applications that don't want
3370  * to write a main loop and deal with #DBusWatch and #DBusTimeout. An
3371  * example usage would be:
3372  * 
3373  * @code
3374  *   while (dbus_connection_read_write_dispatch (connection, -1))
3375  *     ; // empty loop body
3376  * @endcode
3377  * 
3378  * In this usage you would normally have set up a filter function to look
3379  * at each message as it is dispatched. The loop terminates when the last
3380  * message from the connection (the disconnected signal) is processed.
3381  * 
3382  * If there are messages to dispatch, this function will
3383  * dbus_connection_dispatch() once, and return. If there are no
3384  * messages to dispatch, this function will block until it can read or
3385  * write, then read or write, then return.
3386  *
3387  * The way to think of this function is that it either makes some sort
3388  * of progress, or it blocks. Note that, while it is blocked on I/O, it
3389  * cannot be interrupted (even by other threads), which makes this function
3390  * unsuitable for applications that do more than just react to received
3391  * messages.
3392  *
3393  * The return value indicates whether the disconnect message has been
3394  * processed, NOT whether the connection is connected. This is
3395  * important because even after disconnecting, you want to process any
3396  * messages you received prior to the disconnect.
3397  *
3398  * @param connection the connection
3399  * @param timeout_milliseconds max time to block or -1 for infinite
3400  * @returns #TRUE if the disconnect message has not been processed
3401  */
3402 dbus_bool_t
3403 dbus_connection_read_write_dispatch (DBusConnection *connection,
3404                                      int             timeout_milliseconds)
3405 {
3406   _dbus_return_val_if_fail (connection != NULL, FALSE);
3407   _dbus_return_val_if_fail (timeout_milliseconds >= 0 || timeout_milliseconds == -1, FALSE);
3408    return _dbus_connection_read_write_dispatch(connection, timeout_milliseconds, TRUE);
3409 }
3410
3411 /** 
3412  * This function is intended for use with applications that don't want to
3413  * write a main loop and deal with #DBusWatch and #DBusTimeout. See also
3414  * dbus_connection_read_write_dispatch().
3415  * 
3416  * As long as the connection is open, this function will block until it can
3417  * read or write, then read or write, then return #TRUE.
3418  *
3419  * If the connection is closed, the function returns #FALSE.
3420  *
3421  * The return value indicates whether reading or writing is still
3422  * possible, i.e. whether the connection is connected.
3423  *
3424  * Note that even after disconnection, messages may remain in the
3425  * incoming queue that need to be
3426  * processed. dbus_connection_read_write_dispatch() dispatches
3427  * incoming messages for you; with dbus_connection_read_write() you
3428  * have to arrange to drain the incoming queue yourself.
3429  * 
3430  * @param connection the connection 
3431  * @param timeout_milliseconds max time to block or -1 for infinite 
3432  * @returns #TRUE if still connected
3433  */
3434 dbus_bool_t 
3435 dbus_connection_read_write (DBusConnection *connection, 
3436                             int             timeout_milliseconds) 
3437
3438   _dbus_return_val_if_fail (connection != NULL, FALSE);
3439   _dbus_return_val_if_fail (timeout_milliseconds >= 0 || timeout_milliseconds == -1, FALSE);
3440    return _dbus_connection_read_write_dispatch(connection, timeout_milliseconds, FALSE);
3441 }
3442
3443 /* We need to call this anytime we pop the head of the queue, and then
3444  * update_dispatch_status_and_unlock needs to be called afterward
3445  * which will "process" the disconnected message and set
3446  * disconnected_message_processed.
3447  */
3448 static void
3449 check_disconnected_message_arrived_unlocked (DBusConnection *connection,
3450                                              DBusMessage    *head_of_queue)
3451 {
3452   HAVE_LOCK_CHECK (connection);
3453
3454   /* checking that the link is NULL is an optimization to avoid the is_signal call */
3455   if (connection->disconnect_message_link == NULL &&
3456       dbus_message_is_signal (head_of_queue,
3457                               DBUS_INTERFACE_LOCAL,
3458                               "Disconnected"))
3459     {
3460       connection->disconnected_message_arrived = TRUE;
3461     }
3462 }
3463
3464 /**
3465  * Returns the first-received message from the incoming message queue,
3466  * leaving it in the queue. If the queue is empty, returns #NULL.
3467  * 
3468  * The caller does not own a reference to the returned message, and
3469  * must either return it using dbus_connection_return_message() or
3470  * keep it after calling dbus_connection_steal_borrowed_message(). No
3471  * one can get at the message while its borrowed, so return it as
3472  * quickly as possible and don't keep a reference to it after
3473  * returning it. If you need to keep the message, make a copy of it.
3474  *
3475  * dbus_connection_dispatch() will block if called while a borrowed
3476  * message is outstanding; only one piece of code can be playing with
3477  * the incoming queue at a time. This function will block if called
3478  * during a dbus_connection_dispatch().
3479  *
3480  * @param connection the connection.
3481  * @returns next message in the incoming queue.
3482  */
3483 DBusMessage*
3484 dbus_connection_borrow_message (DBusConnection *connection)
3485 {
3486   DBusDispatchStatus status;
3487   DBusMessage *message;
3488
3489   _dbus_return_val_if_fail (connection != NULL, NULL);
3490
3491   _dbus_verbose ("%s start\n", _DBUS_FUNCTION_NAME);
3492   
3493   /* this is called for the side effect that it queues
3494    * up any messages from the transport
3495    */
3496   status = dbus_connection_get_dispatch_status (connection);
3497   if (status != DBUS_DISPATCH_DATA_REMAINS)
3498     return NULL;
3499   
3500   CONNECTION_LOCK (connection);
3501
3502   _dbus_connection_acquire_dispatch (connection);
3503
3504   /* While a message is outstanding, the dispatch lock is held */
3505   _dbus_assert (connection->message_borrowed == NULL);
3506
3507   connection->message_borrowed = _dbus_list_get_first (&connection->incoming_messages);
3508   
3509   message = connection->message_borrowed;
3510
3511   check_disconnected_message_arrived_unlocked (connection, message);
3512   
3513   /* Note that we KEEP the dispatch lock until the message is returned */
3514   if (message == NULL)
3515     _dbus_connection_release_dispatch (connection);
3516
3517   CONNECTION_UNLOCK (connection);
3518
3519   /* We don't update dispatch status until it's returned or stolen */
3520   
3521   return message;
3522 }
3523
3524 /**
3525  * Used to return a message after peeking at it using
3526  * dbus_connection_borrow_message(). Only called if
3527  * message from dbus_connection_borrow_message() was non-#NULL.
3528  *
3529  * @param connection the connection
3530  * @param message the message from dbus_connection_borrow_message()
3531  */
3532 void
3533 dbus_connection_return_message (DBusConnection *connection,
3534                                 DBusMessage    *message)
3535 {
3536   DBusDispatchStatus status;
3537   
3538   _dbus_return_if_fail (connection != NULL);
3539   _dbus_return_if_fail (message != NULL);
3540   _dbus_return_if_fail (message == connection->message_borrowed);
3541   _dbus_return_if_fail (connection->dispatch_acquired);
3542   
3543   CONNECTION_LOCK (connection);
3544   
3545   _dbus_assert (message == connection->message_borrowed);
3546   
3547   connection->message_borrowed = NULL;
3548
3549   _dbus_connection_release_dispatch (connection); 
3550
3551   status = _dbus_connection_get_dispatch_status_unlocked (connection);
3552   _dbus_connection_update_dispatch_status_and_unlock (connection, status);
3553 }
3554
3555 /**
3556  * Used to keep a message after peeking at it using
3557  * dbus_connection_borrow_message(). Before using this function, see
3558  * the caveats/warnings in the documentation for
3559  * dbus_connection_pop_message().
3560  *
3561  * @param connection the connection
3562  * @param message the message from dbus_connection_borrow_message()
3563  */
3564 void
3565 dbus_connection_steal_borrowed_message (DBusConnection *connection,
3566                                         DBusMessage    *message)
3567 {
3568   DBusMessage *pop_message;
3569   DBusDispatchStatus status;
3570
3571   _dbus_return_if_fail (connection != NULL);
3572   _dbus_return_if_fail (message != NULL);
3573   _dbus_return_if_fail (message == connection->message_borrowed);
3574   _dbus_return_if_fail (connection->dispatch_acquired);
3575   
3576   CONNECTION_LOCK (connection);
3577  
3578   _dbus_assert (message == connection->message_borrowed);
3579
3580   pop_message = _dbus_list_pop_first (&connection->incoming_messages);
3581   _dbus_assert (message == pop_message);
3582   
3583   connection->n_incoming -= 1;
3584  
3585   _dbus_verbose ("Incoming message %p stolen from queue, %d incoming\n",
3586                  message, connection->n_incoming);
3587  
3588   connection->message_borrowed = NULL;
3589
3590   _dbus_connection_release_dispatch (connection);
3591
3592   status = _dbus_connection_get_dispatch_status_unlocked (connection);
3593   _dbus_connection_update_dispatch_status_and_unlock (connection, status);
3594 }
3595
3596 /* See dbus_connection_pop_message, but requires the caller to own
3597  * the lock before calling. May drop the lock while running.
3598  */
3599 static DBusList*
3600 _dbus_connection_pop_message_link_unlocked (DBusConnection *connection)
3601 {
3602   HAVE_LOCK_CHECK (connection);
3603   
3604   _dbus_assert (connection->message_borrowed == NULL);
3605   
3606   if (connection->n_incoming > 0)
3607     {
3608       DBusList *link;
3609
3610       link = _dbus_list_pop_first_link (&connection->incoming_messages);
3611       connection->n_incoming -= 1;
3612
3613       _dbus_verbose ("Message %p (%d %s %s %s '%s') removed from incoming queue %p, %d incoming\n",
3614                      link->data,
3615                      dbus_message_get_type (link->data),
3616                      dbus_message_get_path (link->data) ?
3617                      dbus_message_get_path (link->data) :
3618                      "no path",
3619                      dbus_message_get_interface (link->data) ?
3620                      dbus_message_get_interface (link->data) :
3621                      "no interface",
3622                      dbus_message_get_member (link->data) ?
3623                      dbus_message_get_member (link->data) :
3624                      "no member",
3625                      dbus_message_get_signature (link->data),
3626                      connection, connection->n_incoming);
3627
3628       check_disconnected_message_arrived_unlocked (connection, link->data);
3629       
3630       return link;
3631     }
3632   else
3633     return NULL;
3634 }
3635
3636 /* See dbus_connection_pop_message, but requires the caller to own
3637  * the lock before calling. May drop the lock while running.
3638  */
3639 static DBusMessage*
3640 _dbus_connection_pop_message_unlocked (DBusConnection *connection)
3641 {
3642   DBusList *link;
3643
3644   HAVE_LOCK_CHECK (connection);
3645   
3646   link = _dbus_connection_pop_message_link_unlocked (connection);
3647
3648   if (link != NULL)
3649     {
3650       DBusMessage *message;
3651       
3652       message = link->data;
3653       
3654       _dbus_list_free_link (link);
3655       
3656       return message;
3657     }
3658   else
3659     return NULL;
3660 }
3661
3662 static void
3663 _dbus_connection_putback_message_link_unlocked (DBusConnection *connection,
3664                                                 DBusList       *message_link)
3665 {
3666   HAVE_LOCK_CHECK (connection);
3667   
3668   _dbus_assert (message_link != NULL);
3669   /* You can't borrow a message while a link is outstanding */
3670   _dbus_assert (connection->message_borrowed == NULL);
3671   /* We had to have the dispatch lock across the pop/putback */
3672   _dbus_assert (connection->dispatch_acquired);
3673
3674   _dbus_list_prepend_link (&connection->incoming_messages,
3675                            message_link);
3676   connection->n_incoming += 1;
3677
3678   _dbus_verbose ("Message %p (%d %s %s '%s') put back into queue %p, %d incoming\n",
3679                  message_link->data,
3680                  dbus_message_get_type (message_link->data),
3681                  dbus_message_get_interface (message_link->data) ?
3682                  dbus_message_get_interface (message_link->data) :
3683                  "no interface",
3684                  dbus_message_get_member (message_link->data) ?
3685                  dbus_message_get_member (message_link->data) :
3686                  "no member",
3687                  dbus_message_get_signature (message_link->data),
3688                  connection, connection->n_incoming);
3689 }
3690
3691 /**
3692  * Returns the first-received message from the incoming message queue,
3693  * removing it from the queue. The caller owns a reference to the
3694  * returned message. If the queue is empty, returns #NULL.
3695  *
3696  * This function bypasses any message handlers that are registered,
3697  * and so using it is usually wrong. Instead, let the main loop invoke
3698  * dbus_connection_dispatch(). Popping messages manually is only
3699  * useful in very simple programs that don't share a #DBusConnection
3700  * with any libraries or other modules.
3701  *
3702  * There is a lock that covers all ways of accessing the incoming message
3703  * queue, so dbus_connection_dispatch(), dbus_connection_pop_message(),
3704  * dbus_connection_borrow_message(), etc. will all block while one of the others
3705  * in the group is running.
3706  * 
3707  * @param connection the connection.
3708  * @returns next message in the incoming queue.
3709  */
3710 DBusMessage*
3711 dbus_connection_pop_message (DBusConnection *connection)
3712 {
3713   DBusMessage *message;
3714   DBusDispatchStatus status;
3715
3716   _dbus_verbose ("%s start\n", _DBUS_FUNCTION_NAME);
3717   
3718   /* this is called for the side effect that it queues
3719    * up any messages from the transport
3720    */
3721   status = dbus_connection_get_dispatch_status (connection);
3722   if (status != DBUS_DISPATCH_DATA_REMAINS)
3723     return NULL;
3724   
3725   CONNECTION_LOCK (connection);
3726   _dbus_connection_acquire_dispatch (connection);
3727   HAVE_LOCK_CHECK (connection);
3728   
3729   message = _dbus_connection_pop_message_unlocked (connection);
3730
3731   _dbus_verbose ("Returning popped message %p\n", message);    
3732
3733   _dbus_connection_release_dispatch (connection);
3734
3735   status = _dbus_connection_get_dispatch_status_unlocked (connection);
3736   _dbus_connection_update_dispatch_status_and_unlock (connection, status);
3737   
3738   return message;
3739 }
3740
3741 /**
3742  * Acquire the dispatcher. This is a separate lock so the main
3743  * connection lock can be dropped to call out to application dispatch
3744  * handlers.
3745  *
3746  * @param connection the connection.
3747  */
3748 static void
3749 _dbus_connection_acquire_dispatch (DBusConnection *connection)
3750 {
3751   HAVE_LOCK_CHECK (connection);
3752
3753   _dbus_connection_ref_unlocked (connection);
3754   CONNECTION_UNLOCK (connection);
3755   
3756   _dbus_verbose ("%s locking dispatch_mutex\n", _DBUS_FUNCTION_NAME);
3757   _dbus_mutex_lock (connection->dispatch_mutex);
3758
3759   while (connection->dispatch_acquired)
3760     {
3761       _dbus_verbose ("%s waiting for dispatch to be acquirable\n", _DBUS_FUNCTION_NAME);
3762       _dbus_condvar_wait (connection->dispatch_cond, 
3763                           connection->dispatch_mutex);
3764     }
3765   
3766   _dbus_assert (!connection->dispatch_acquired);
3767
3768   connection->dispatch_acquired = TRUE;
3769
3770   _dbus_verbose ("%s unlocking dispatch_mutex\n", _DBUS_FUNCTION_NAME);
3771   _dbus_mutex_unlock (connection->dispatch_mutex);
3772   
3773   CONNECTION_LOCK (connection);
3774   _dbus_connection_unref_unlocked (connection);
3775 }
3776
3777 /**
3778  * Release the dispatcher when you're done with it. Only call
3779  * after you've acquired the dispatcher. Wakes up at most one
3780  * thread currently waiting to acquire the dispatcher.
3781  *
3782  * @param connection the connection.
3783  */
3784 static void
3785 _dbus_connection_release_dispatch (DBusConnection *connection)
3786 {
3787   HAVE_LOCK_CHECK (connection);
3788   
3789   _dbus_verbose ("%s locking dispatch_mutex\n", _DBUS_FUNCTION_NAME);
3790   _dbus_mutex_lock (connection->dispatch_mutex);
3791   
3792   _dbus_assert (connection->dispatch_acquired);
3793
3794   connection->dispatch_acquired = FALSE;
3795   _dbus_condvar_wake_one (connection->dispatch_cond);
3796
3797   _dbus_verbose ("%s unlocking dispatch_mutex\n", _DBUS_FUNCTION_NAME);
3798   _dbus_mutex_unlock (connection->dispatch_mutex);
3799 }
3800
3801 static void
3802 _dbus_connection_failed_pop (DBusConnection *connection,
3803                              DBusList       *message_link)
3804 {
3805   _dbus_list_prepend_link (&connection->incoming_messages,
3806                            message_link);
3807   connection->n_incoming += 1;
3808 }
3809
3810 /* Note this may be called multiple times since we don't track whether we already did it */
3811 static void
3812 notify_disconnected_unlocked (DBusConnection *connection)
3813 {
3814   HAVE_LOCK_CHECK (connection);
3815
3816   /* Set the weakref in dbus-bus.c to NULL, so nobody will get a disconnected
3817    * connection from dbus_bus_get(). We make the same guarantee for
3818    * dbus_connection_open() but in a different way since we don't want to
3819    * unref right here; we instead check for connectedness before returning
3820    * the connection from the hash.
3821    */
3822   _dbus_bus_notify_shared_connection_disconnected_unlocked (connection);
3823
3824   /* Dump the outgoing queue, we aren't going to be able to
3825    * send it now, and we'd like accessors like
3826    * dbus_connection_get_outgoing_size() to be accurate.
3827    */
3828   if (connection->n_outgoing > 0)
3829     {
3830       DBusList *link;
3831       
3832       _dbus_verbose ("Dropping %d outgoing messages since we're disconnected\n",
3833                      connection->n_outgoing);
3834       
3835       while ((link = _dbus_list_get_last_link (&connection->outgoing_messages)))
3836         {
3837           _dbus_connection_message_sent (connection, link->data);
3838         }
3839     } 
3840 }
3841
3842 /* Note this may be called multiple times since we don't track whether we already did it */
3843 static DBusDispatchStatus
3844 notify_disconnected_and_dispatch_complete_unlocked (DBusConnection *connection)
3845 {
3846   HAVE_LOCK_CHECK (connection);
3847   
3848   if (connection->disconnect_message_link != NULL)
3849     {
3850       _dbus_verbose ("Sending disconnect message from %s\n",
3851                      _DBUS_FUNCTION_NAME);
3852       
3853       /* If we have pending calls, queue their timeouts - we want the Disconnected
3854        * to be the last message, after these timeouts.
3855        */
3856       connection_timeout_and_complete_all_pending_calls_unlocked (connection);
3857       
3858       /* We haven't sent the disconnect message already,
3859        * and all real messages have been queued up.
3860        */
3861       _dbus_connection_queue_synthesized_message_link (connection,
3862                                                        connection->disconnect_message_link);
3863       connection->disconnect_message_link = NULL;
3864
3865       return DBUS_DISPATCH_DATA_REMAINS;
3866     }
3867
3868   return DBUS_DISPATCH_COMPLETE;
3869 }
3870
3871 static DBusDispatchStatus
3872 _dbus_connection_get_dispatch_status_unlocked (DBusConnection *connection)
3873 {
3874   HAVE_LOCK_CHECK (connection);
3875   
3876   if (connection->n_incoming > 0)
3877     return DBUS_DISPATCH_DATA_REMAINS;
3878   else if (!_dbus_transport_queue_messages (connection->transport))
3879     return DBUS_DISPATCH_NEED_MEMORY;
3880   else
3881     {
3882       DBusDispatchStatus status;
3883       dbus_bool_t is_connected;
3884       
3885       status = _dbus_transport_get_dispatch_status (connection->transport);
3886       is_connected = _dbus_transport_get_is_connected (connection->transport);
3887
3888       _dbus_verbose ("dispatch status = %s is_connected = %d\n",
3889                      DISPATCH_STATUS_NAME (status), is_connected);
3890       
3891       if (!is_connected)
3892         {
3893           /* It's possible this would be better done by having an explicit
3894            * notification from _dbus_transport_disconnect() that would
3895            * synchronously do this, instead of waiting for the next dispatch
3896            * status check. However, probably not good to change until it causes
3897            * a problem.
3898            */
3899           notify_disconnected_unlocked (connection);
3900
3901           /* I'm not sure this is needed; the idea is that we want to
3902            * queue the Disconnected only after we've read all the
3903            * messages, but if we're disconnected maybe we are guaranteed
3904            * to have read them all ?
3905            */
3906           if (status == DBUS_DISPATCH_COMPLETE)
3907             status = notify_disconnected_and_dispatch_complete_unlocked (connection);
3908         }
3909       
3910       if (status != DBUS_DISPATCH_COMPLETE)
3911         return status;
3912       else if (connection->n_incoming > 0)
3913         return DBUS_DISPATCH_DATA_REMAINS;
3914       else
3915         return DBUS_DISPATCH_COMPLETE;
3916     }
3917 }
3918
3919 static void
3920 _dbus_connection_update_dispatch_status_and_unlock (DBusConnection    *connection,
3921                                                     DBusDispatchStatus new_status)
3922 {
3923   dbus_bool_t changed;
3924   DBusDispatchStatusFunction function;
3925   void *data;
3926
3927   HAVE_LOCK_CHECK (connection);
3928
3929   _dbus_connection_ref_unlocked (connection);
3930
3931   changed = new_status != connection->last_dispatch_status;
3932
3933   connection->last_dispatch_status = new_status;
3934
3935   function = connection->dispatch_status_function;
3936   data = connection->dispatch_status_data;
3937
3938   if (connection->disconnected_message_arrived &&
3939       !connection->disconnected_message_processed)
3940     {
3941       connection->disconnected_message_processed = TRUE;
3942       
3943       /* this does an unref, but we have a ref
3944        * so we should not run the finalizer here
3945        * inside the lock.
3946        */
3947       connection_forget_shared_unlocked (connection);
3948
3949       if (connection->exit_on_disconnect)
3950         {
3951           CONNECTION_UNLOCK (connection);            
3952           
3953           _dbus_verbose ("Exiting on Disconnected signal\n");
3954           _dbus_exit (1);
3955           _dbus_assert_not_reached ("Call to exit() returned");
3956         }
3957     }
3958   
3959   /* We drop the lock */
3960   CONNECTION_UNLOCK (connection);
3961   
3962   if (changed && function)
3963     {
3964       _dbus_verbose ("Notifying of change to dispatch status of %p now %d (%s)\n",
3965                      connection, new_status,
3966                      DISPATCH_STATUS_NAME (new_status));
3967       (* function) (connection, new_status, data);      
3968     }
3969   
3970   dbus_connection_unref (connection);
3971 }
3972
3973 /**
3974  * Gets the current state of the incoming message queue.
3975  * #DBUS_DISPATCH_DATA_REMAINS indicates that the message queue
3976  * may contain messages. #DBUS_DISPATCH_COMPLETE indicates that the
3977  * incoming queue is empty. #DBUS_DISPATCH_NEED_MEMORY indicates that
3978  * there could be data, but we can't know for sure without more
3979  * memory.
3980  *
3981  * To process the incoming message queue, use dbus_connection_dispatch()
3982  * or (in rare cases) dbus_connection_pop_message().
3983  *
3984  * Note, #DBUS_DISPATCH_DATA_REMAINS really means that either we
3985  * have messages in the queue, or we have raw bytes buffered up
3986  * that need to be parsed. When these bytes are parsed, they
3987  * may not add up to an entire message. Thus, it's possible
3988  * to see a status of #DBUS_DISPATCH_DATA_REMAINS but not
3989  * have a message yet.
3990  *
3991  * In particular this happens on initial connection, because all sorts
3992  * of authentication protocol stuff has to be parsed before the
3993  * first message arrives.
3994  * 
3995  * @param connection the connection.
3996  * @returns current dispatch status
3997  */
3998 DBusDispatchStatus
3999 dbus_connection_get_dispatch_status (DBusConnection *connection)
4000 {
4001   DBusDispatchStatus status;
4002
4003   _dbus_return_val_if_fail (connection != NULL, DBUS_DISPATCH_COMPLETE);
4004
4005   _dbus_verbose ("%s start\n", _DBUS_FUNCTION_NAME);
4006   
4007   CONNECTION_LOCK (connection);
4008
4009   status = _dbus_connection_get_dispatch_status_unlocked (connection);
4010   
4011   CONNECTION_UNLOCK (connection);
4012
4013   return status;
4014 }
4015
4016 /**
4017  * Filter funtion for handling the Peer standard interface.
4018  */
4019 static DBusHandlerResult
4020 _dbus_connection_peer_filter_unlocked_no_update (DBusConnection *connection,
4021                                                  DBusMessage    *message)
4022 {
4023   if (connection->route_peer_messages && dbus_message_get_destination (message) != NULL)
4024     {
4025       /* This means we're letting the bus route this message */
4026       return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
4027     }
4028   else if (dbus_message_is_method_call (message,
4029                                         DBUS_INTERFACE_PEER,
4030                                         "Ping"))
4031     {
4032       DBusMessage *ret;
4033       dbus_bool_t sent;
4034       
4035       ret = dbus_message_new_method_return (message);
4036       if (ret == NULL)
4037         return DBUS_HANDLER_RESULT_NEED_MEMORY;
4038      
4039       sent = _dbus_connection_send_unlocked_no_update (connection, ret, NULL);
4040
4041       dbus_message_unref (ret);
4042
4043       if (!sent)
4044         return DBUS_HANDLER_RESULT_NEED_MEMORY;
4045       
4046       return DBUS_HANDLER_RESULT_HANDLED;
4047     }
4048   else if (dbus_message_is_method_call (message,
4049                                         DBUS_INTERFACE_PEER,
4050                                         "GetMachineId"))
4051     {
4052       DBusMessage *ret;
4053       dbus_bool_t sent;
4054       DBusString uuid;
4055       
4056       ret = dbus_message_new_method_return (message);
4057       if (ret == NULL)
4058         return DBUS_HANDLER_RESULT_NEED_MEMORY;
4059
4060       sent = FALSE;
4061       _dbus_string_init (&uuid);
4062       if (_dbus_get_local_machine_uuid_encoded (&uuid))
4063         {
4064           const char *v_STRING = _dbus_string_get_const_data (&uuid);
4065           if (dbus_message_append_args (ret,
4066                                         DBUS_TYPE_STRING, &v_STRING,
4067                                         DBUS_TYPE_INVALID))
4068             {
4069               sent = _dbus_connection_send_unlocked_no_update (connection, ret, NULL);
4070             }
4071         }
4072       _dbus_string_free (&uuid);
4073       
4074       dbus_message_unref (ret);
4075
4076       if (!sent)
4077         return DBUS_HANDLER_RESULT_NEED_MEMORY;
4078       
4079       return DBUS_HANDLER_RESULT_HANDLED;
4080     }
4081   else if (dbus_message_has_interface (message, DBUS_INTERFACE_PEER))
4082     {
4083       /* We need to bounce anything else with this interface, otherwise apps
4084        * could start extending the interface and when we added extensions
4085        * here to DBusConnection we'd break those apps.
4086        */
4087       
4088       DBusMessage *ret;
4089       dbus_bool_t sent;
4090       
4091       ret = dbus_message_new_error (message,
4092                                     DBUS_ERROR_UNKNOWN_METHOD,
4093                                     "Unknown method invoked on org.freedesktop.DBus.Peer interface");
4094       if (ret == NULL)
4095         return DBUS_HANDLER_RESULT_NEED_MEMORY;
4096       
4097       sent = _dbus_connection_send_unlocked_no_update (connection, ret, NULL);
4098       
4099       dbus_message_unref (ret);
4100       
4101       if (!sent)
4102         return DBUS_HANDLER_RESULT_NEED_MEMORY;
4103       
4104       return DBUS_HANDLER_RESULT_HANDLED;
4105     }
4106   else
4107     {
4108       return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
4109     }
4110 }
4111
4112 /**
4113 * Processes all builtin filter functions
4114 *
4115 * If the spec specifies a standard interface
4116 * they should be processed from this method
4117 **/
4118 static DBusHandlerResult
4119 _dbus_connection_run_builtin_filters_unlocked_no_update (DBusConnection *connection,
4120                                                            DBusMessage    *message)
4121 {
4122   /* We just run one filter for now but have the option to run more
4123      if the spec calls for it in the future */
4124
4125   return _dbus_connection_peer_filter_unlocked_no_update (connection, message);
4126 }
4127
4128 /**
4129  * Processes any incoming data.
4130  *
4131  * If there's incoming raw data that has not yet been parsed, it is
4132  * parsed, which may or may not result in adding messages to the
4133  * incoming queue.
4134  *
4135  * The incoming data buffer is filled when the connection reads from
4136  * its underlying transport (such as a socket).  Reading usually
4137  * happens in dbus_watch_handle() or dbus_connection_read_write().
4138  * 
4139  * If there are complete messages in the incoming queue,
4140  * dbus_connection_dispatch() removes one message from the queue and
4141  * processes it. Processing has three steps.
4142  *
4143  * First, any method replies are passed to #DBusPendingCall or
4144  * dbus_connection_send_with_reply_and_block() in order to
4145  * complete the pending method call.
4146  * 
4147  * Second, any filters registered with dbus_connection_add_filter()
4148  * are run. If any filter returns #DBUS_HANDLER_RESULT_HANDLED
4149  * then processing stops after that filter.
4150  *
4151  * Third, if the message is a method call it is forwarded to
4152  * any registered object path handlers added with
4153  * dbus_connection_register_object_path() or
4154  * dbus_connection_register_fallback().
4155  *
4156  * A single call to dbus_connection_dispatch() will process at most
4157  * one message; it will not clear the entire message queue.
4158  *
4159  * Be careful about calling dbus_connection_dispatch() from inside a
4160  * message handler, i.e. calling dbus_connection_dispatch()
4161  * recursively.  If threads have been initialized with a recursive
4162  * mutex function, then this will not deadlock; however, it can
4163  * certainly confuse your application.
4164  * 
4165  * @todo some FIXME in here about handling DBUS_HANDLER_RESULT_NEED_MEMORY
4166  * 
4167  * @param connection the connection
4168  * @returns dispatch status, see dbus_connection_get_dispatch_status()
4169  */
4170 DBusDispatchStatus
4171 dbus_connection_dispatch (DBusConnection *connection)
4172 {
4173   DBusMessage *message;
4174   DBusList *link, *filter_list_copy, *message_link;
4175   DBusHandlerResult result;
4176   DBusPendingCall *pending;
4177   dbus_int32_t reply_serial;
4178   DBusDispatchStatus status;
4179
4180   _dbus_return_val_if_fail (connection != NULL, DBUS_DISPATCH_COMPLETE);
4181
4182   _dbus_verbose ("%s\n", _DBUS_FUNCTION_NAME);
4183   
4184   CONNECTION_LOCK (connection);
4185   status = _dbus_connection_get_dispatch_status_unlocked (connection);
4186   if (status != DBUS_DISPATCH_DATA_REMAINS)
4187     {
4188       /* unlocks and calls out to user code */
4189       _dbus_connection_update_dispatch_status_and_unlock (connection, status);
4190       return status;
4191     }
4192   
4193   /* We need to ref the connection since the callback could potentially
4194    * drop the last ref to it
4195    */
4196   _dbus_connection_ref_unlocked (connection);
4197
4198   _dbus_connection_acquire_dispatch (connection);
4199   HAVE_LOCK_CHECK (connection);
4200
4201   message_link = _dbus_connection_pop_message_link_unlocked (connection);
4202   if (message_link == NULL)
4203     {
4204       /* another thread dispatched our stuff */
4205
4206       _dbus_verbose ("another thread dispatched message (during acquire_dispatch above)\n");
4207       
4208       _dbus_connection_release_dispatch (connection);
4209
4210       status = _dbus_connection_get_dispatch_status_unlocked (connection);
4211
4212       _dbus_connection_update_dispatch_status_and_unlock (connection, status);
4213       
4214       dbus_connection_unref (connection);
4215       
4216       return status;
4217     }
4218
4219   message = message_link->data;
4220
4221   _dbus_verbose (" dispatching message %p (%d %s %s '%s')\n",
4222                  message,
4223                  dbus_message_get_type (message),
4224                  dbus_message_get_interface (message) ?
4225                  dbus_message_get_interface (message) :
4226                  "no interface",
4227                  dbus_message_get_member (message) ?
4228                  dbus_message_get_member (message) :
4229                  "no member",
4230                  dbus_message_get_signature (message));
4231
4232   result = DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
4233   
4234   /* Pending call handling must be first, because if you do
4235    * dbus_connection_send_with_reply_and_block() or
4236    * dbus_pending_call_block() then no handlers/filters will be run on
4237    * the reply. We want consistent semantics in the case where we
4238    * dbus_connection_dispatch() the reply.
4239    */
4240   
4241   reply_serial = dbus_message_get_reply_serial (message);
4242   pending = _dbus_hash_table_lookup_int (connection->pending_replies,
4243                                          reply_serial);
4244   if (pending)
4245     {
4246       _dbus_verbose ("Dispatching a pending reply\n");
4247       complete_pending_call_and_unlock (connection, pending, message);
4248       pending = NULL; /* it's probably unref'd */
4249       
4250       CONNECTION_LOCK (connection);
4251       _dbus_verbose ("pending call completed in dispatch\n");
4252       result = DBUS_HANDLER_RESULT_HANDLED;
4253       goto out;
4254     }
4255
4256   result = _dbus_connection_run_builtin_filters_unlocked_no_update (connection, message);
4257   if (result != DBUS_HANDLER_RESULT_NOT_YET_HANDLED)
4258     goto out;
4259  
4260   if (!_dbus_list_copy (&connection->filter_list, &filter_list_copy))
4261     {
4262       _dbus_connection_release_dispatch (connection);
4263       HAVE_LOCK_CHECK (connection);
4264       
4265       _dbus_connection_failed_pop (connection, message_link);
4266
4267       /* unlocks and calls user code */
4268       _dbus_connection_update_dispatch_status_and_unlock (connection,
4269                                                           DBUS_DISPATCH_NEED_MEMORY);
4270
4271       if (pending)
4272         dbus_pending_call_unref (pending);
4273       dbus_connection_unref (connection);
4274       
4275       return DBUS_DISPATCH_NEED_MEMORY;
4276     }
4277   
4278   _dbus_list_foreach (&filter_list_copy,
4279                       (DBusForeachFunction)_dbus_message_filter_ref,
4280                       NULL);
4281
4282   /* We're still protected from dispatch() reentrancy here
4283    * since we acquired the dispatcher
4284    */
4285   CONNECTION_UNLOCK (connection);
4286   
4287   link = _dbus_list_get_first_link (&filter_list_copy);
4288   while (link != NULL)
4289     {
4290       DBusMessageFilter *filter = link->data;
4291       DBusList *next = _dbus_list_get_next_link (&filter_list_copy, link);
4292
4293       if (filter->function == NULL)
4294         {
4295           _dbus_verbose ("  filter was removed in a callback function\n");
4296           link = next;
4297           continue;
4298         }
4299
4300       _dbus_verbose ("  running filter on message %p\n", message);
4301       result = (* filter->function) (connection, message, filter->user_data);
4302
4303       if (result != DBUS_HANDLER_RESULT_NOT_YET_HANDLED)
4304         break;
4305
4306       link = next;
4307     }
4308
4309   _dbus_list_foreach (&filter_list_copy,
4310                       (DBusForeachFunction)_dbus_message_filter_unref,
4311                       NULL);
4312   _dbus_list_clear (&filter_list_copy);
4313   
4314   CONNECTION_LOCK (connection);
4315
4316   if (result == DBUS_HANDLER_RESULT_NEED_MEMORY)
4317     {
4318       _dbus_verbose ("No memory in %s\n", _DBUS_FUNCTION_NAME);
4319       goto out;
4320     }
4321   else if (result == DBUS_HANDLER_RESULT_HANDLED)
4322     {
4323       _dbus_verbose ("filter handled message in dispatch\n");
4324       goto out;
4325     }
4326
4327   /* We're still protected from dispatch() reentrancy here
4328    * since we acquired the dispatcher
4329    */
4330   _dbus_verbose ("  running object path dispatch on message %p (%d %s %s '%s')\n",
4331                  message,
4332                  dbus_message_get_type (message),
4333                  dbus_message_get_interface (message) ?
4334                  dbus_message_get_interface (message) :
4335                  "no interface",
4336                  dbus_message_get_member (message) ?
4337                  dbus_message_get_member (message) :
4338                  "no member",
4339                  dbus_message_get_signature (message));
4340
4341   HAVE_LOCK_CHECK (connection);
4342   result = _dbus_object_tree_dispatch_and_unlock (connection->objects,
4343                                                   message);
4344   
4345   CONNECTION_LOCK (connection);
4346
4347   if (result != DBUS_HANDLER_RESULT_NOT_YET_HANDLED)
4348     {
4349       _dbus_verbose ("object tree handled message in dispatch\n");
4350       goto out;
4351     }
4352
4353   if (dbus_message_get_type (message) == DBUS_MESSAGE_TYPE_METHOD_CALL)
4354     {
4355       DBusMessage *reply;
4356       DBusString str;
4357       DBusPreallocatedSend *preallocated;
4358
4359       _dbus_verbose ("  sending error %s\n",
4360                      DBUS_ERROR_UNKNOWN_METHOD);
4361       
4362       if (!_dbus_string_init (&str))
4363         {
4364           result = DBUS_HANDLER_RESULT_NEED_MEMORY;
4365           _dbus_verbose ("no memory for error string in dispatch\n");
4366           goto out;
4367         }
4368               
4369       if (!_dbus_string_append_printf (&str,
4370                                        "Method \"%s\" with signature \"%s\" on interface \"%s\" doesn't exist\n",
4371                                        dbus_message_get_member (message),
4372                                        dbus_message_get_signature (message),
4373                                        dbus_message_get_interface (message)))
4374         {
4375           _dbus_string_free (&str);
4376           result = DBUS_HANDLER_RESULT_NEED_MEMORY;
4377           _dbus_verbose ("no memory for error string in dispatch\n");
4378           goto out;
4379         }
4380       
4381       reply = dbus_message_new_error (message,
4382                                       DBUS_ERROR_UNKNOWN_METHOD,
4383                                       _dbus_string_get_const_data (&str));
4384       _dbus_string_free (&str);
4385
4386       if (reply == NULL)
4387         {
4388           result = DBUS_HANDLER_RESULT_NEED_MEMORY;
4389           _dbus_verbose ("no memory for error reply in dispatch\n");
4390           goto out;
4391         }
4392       
4393       preallocated = _dbus_connection_preallocate_send_unlocked (connection);
4394
4395       if (preallocated == NULL)
4396         {
4397           dbus_message_unref (reply);
4398           result = DBUS_HANDLER_RESULT_NEED_MEMORY;
4399           _dbus_verbose ("no memory for error send in dispatch\n");
4400           goto out;
4401         }
4402
4403       _dbus_connection_send_preallocated_unlocked_no_update (connection, preallocated,
4404                                                              reply, NULL);
4405
4406       dbus_message_unref (reply);
4407       
4408       result = DBUS_HANDLER_RESULT_HANDLED;
4409     }
4410   
4411   _dbus_verbose ("  done dispatching %p (%d %s %s '%s') on connection %p\n", message,
4412                  dbus_message_get_type (message),
4413                  dbus_message_get_interface (message) ?
4414                  dbus_message_get_interface (message) :
4415                  "no interface",
4416                  dbus_message_get_member (message) ?
4417                  dbus_message_get_member (message) :
4418                  "no member",
4419                  dbus_message_get_signature (message),
4420                  connection);
4421   
4422  out:
4423   if (result == DBUS_HANDLER_RESULT_NEED_MEMORY)
4424     {
4425       _dbus_verbose ("out of memory in %s\n", _DBUS_FUNCTION_NAME);
4426       
4427       /* Put message back, and we'll start over.
4428        * Yes this means handlers must be idempotent if they
4429        * don't return HANDLED; c'est la vie.
4430        */
4431       _dbus_connection_putback_message_link_unlocked (connection,
4432                                                       message_link);
4433     }
4434   else
4435     {
4436       _dbus_verbose (" ... done dispatching in %s\n", _DBUS_FUNCTION_NAME);
4437       
4438       _dbus_list_free_link (message_link);
4439       dbus_message_unref (message); /* don't want the message to count in max message limits
4440                                      * in computing dispatch status below
4441                                      */
4442     }
4443   
4444   _dbus_connection_release_dispatch (connection);
4445   HAVE_LOCK_CHECK (connection);
4446
4447   _dbus_verbose ("%s before final status update\n", _DBUS_FUNCTION_NAME);
4448   status = _dbus_connection_get_dispatch_status_unlocked (connection);
4449
4450   /* unlocks and calls user code */
4451   _dbus_connection_update_dispatch_status_and_unlock (connection, status);
4452   
4453   dbus_connection_unref (connection);
4454   
4455   return status;
4456 }
4457
4458 /**
4459  * Sets the watch functions for the connection. These functions are
4460  * responsible for making the application's main loop aware of file
4461  * descriptors that need to be monitored for events, using select() or
4462  * poll(). When using Qt, typically the DBusAddWatchFunction would
4463  * create a QSocketNotifier. When using GLib, the DBusAddWatchFunction
4464  * could call g_io_add_watch(), or could be used as part of a more
4465  * elaborate GSource. Note that when a watch is added, it may
4466  * not be enabled.
4467  *
4468  * The DBusWatchToggledFunction notifies the application that the
4469  * watch has been enabled or disabled. Call dbus_watch_get_enabled()
4470  * to check this. A disabled watch should have no effect, and enabled
4471  * watch should be added to the main loop. This feature is used
4472  * instead of simply adding/removing the watch because
4473  * enabling/disabling can be done without memory allocation.  The
4474  * toggled function may be NULL if a main loop re-queries
4475  * dbus_watch_get_enabled() every time anyway.
4476  * 
4477  * The DBusWatch can be queried for the file descriptor to watch using
4478  * dbus_watch_get_fd(), and for the events to watch for using
4479  * dbus_watch_get_flags(). The flags returned by
4480  * dbus_watch_get_flags() will only contain DBUS_WATCH_READABLE and
4481  * DBUS_WATCH_WRITABLE, never DBUS_WATCH_HANGUP or DBUS_WATCH_ERROR;
4482  * all watches implicitly include a watch for hangups, errors, and
4483  * other exceptional conditions.
4484  *
4485  * Once a file descriptor becomes readable or writable, or an exception
4486  * occurs, dbus_watch_handle() should be called to
4487  * notify the connection of the file descriptor's condition.
4488  *
4489  * dbus_watch_handle() cannot be called during the
4490  * DBusAddWatchFunction, as the connection will not be ready to handle
4491  * that watch yet.
4492  * 
4493  * It is not allowed to reference a DBusWatch after it has been passed
4494  * to remove_function.
4495  *
4496  * If #FALSE is returned due to lack of memory, the failure may be due
4497  * to a #FALSE return from the new add_function. If so, the
4498  * add_function may have been called successfully one or more times,
4499  * but the remove_function will also have been called to remove any
4500  * successful adds. i.e. if #FALSE is returned the net result
4501  * should be that dbus_connection_set_watch_functions() has no effect,
4502  * but the add_function and remove_function may have been called.
4503  *
4504  * @todo We need to drop the lock when we call the
4505  * add/remove/toggled functions which can be a side effect
4506  * of setting the watch functions.
4507  * 
4508  * @param connection the connection.
4509  * @param add_function function to begin monitoring a new descriptor.
4510  * @param remove_function function to stop monitoring a descriptor.
4511  * @param toggled_function function to notify of enable/disable
4512  * @param data data to pass to add_function and remove_function.
4513  * @param free_data_function function to be called to free the data.
4514  * @returns #FALSE on failure (no memory)
4515  */
4516 dbus_bool_t
4517 dbus_connection_set_watch_functions (DBusConnection              *connection,
4518                                      DBusAddWatchFunction         add_function,
4519                                      DBusRemoveWatchFunction      remove_function,
4520                                      DBusWatchToggledFunction     toggled_function,
4521                                      void                        *data,
4522                                      DBusFreeFunction             free_data_function)
4523 {
4524   dbus_bool_t retval;
4525   DBusWatchList *watches;
4526
4527   _dbus_return_val_if_fail (connection != NULL, FALSE);
4528   
4529   CONNECTION_LOCK (connection);
4530
4531 #ifndef DBUS_DISABLE_CHECKS
4532   if (connection->watches == NULL)
4533     {
4534       _dbus_warn_check_failed ("Re-entrant call to %s is not allowed\n",
4535                                _DBUS_FUNCTION_NAME);
4536       return FALSE;
4537     }
4538 #endif
4539   
4540   /* ref connection for slightly better reentrancy */
4541   _dbus_connection_ref_unlocked (connection);
4542
4543   /* This can call back into user code, and we need to drop the
4544    * connection lock when it does. This is kind of a lame
4545    * way to do it.
4546    */
4547   watches = connection->watches;
4548   connection->watches = NULL;
4549   CONNECTION_UNLOCK (connection);
4550
4551   retval = _dbus_watch_list_set_functions (watches,
4552                                            add_function, remove_function,
4553                                            toggled_function,
4554                                            data, free_data_function);
4555   CONNECTION_LOCK (connection);
4556   connection->watches = watches;
4557   
4558   CONNECTION_UNLOCK (connection);
4559   /* drop our paranoid refcount */
4560   dbus_connection_unref (connection);
4561   
4562   return retval;
4563 }
4564
4565 /**
4566  * Sets the timeout functions for the connection. These functions are
4567  * responsible for making the application's main loop aware of timeouts.
4568  * When using Qt, typically the DBusAddTimeoutFunction would create a
4569  * QTimer. When using GLib, the DBusAddTimeoutFunction would call
4570  * g_timeout_add.
4571  * 
4572  * The DBusTimeoutToggledFunction notifies the application that the
4573  * timeout has been enabled or disabled. Call
4574  * dbus_timeout_get_enabled() to check this. A disabled timeout should
4575  * have no effect, and enabled timeout should be added to the main
4576  * loop. This feature is used instead of simply adding/removing the
4577  * timeout because enabling/disabling can be done without memory
4578  * allocation. With Qt, QTimer::start() and QTimer::stop() can be used
4579  * to enable and disable. The toggled function may be NULL if a main
4580  * loop re-queries dbus_timeout_get_enabled() every time anyway.
4581  * Whenever a timeout is toggled, its interval may change.
4582  *
4583  * The DBusTimeout can be queried for the timer interval using
4584  * dbus_timeout_get_interval(). dbus_timeout_handle() should be called
4585  * repeatedly, each time the interval elapses, starting after it has
4586  * elapsed once. The timeout stops firing when it is removed with the
4587  * given remove_function.  The timer interval may change whenever the
4588  * timeout is added, removed, or toggled.
4589  *
4590  * @param connection the connection.
4591  * @param add_function function to add a timeout.
4592  * @param remove_function function to remove a timeout.
4593  * @param toggled_function function to notify of enable/disable
4594  * @param data data to pass to add_function and remove_function.
4595  * @param free_data_function function to be called to free the data.
4596  * @returns #FALSE on failure (no memory)
4597  */
4598 dbus_bool_t
4599 dbus_connection_set_timeout_functions   (DBusConnection            *connection,
4600                                          DBusAddTimeoutFunction     add_function,
4601                                          DBusRemoveTimeoutFunction  remove_function,
4602                                          DBusTimeoutToggledFunction toggled_function,
4603                                          void                      *data,
4604                                          DBusFreeFunction           free_data_function)
4605 {
4606   dbus_bool_t retval;
4607   DBusTimeoutList *timeouts;
4608
4609   _dbus_return_val_if_fail (connection != NULL, FALSE);
4610   
4611   CONNECTION_LOCK (connection);
4612
4613 #ifndef DBUS_DISABLE_CHECKS
4614   if (connection->timeouts == NULL)
4615     {
4616       _dbus_warn_check_failed ("Re-entrant call to %s is not allowed\n",
4617                                _DBUS_FUNCTION_NAME);
4618       return FALSE;
4619     }
4620 #endif
4621   
4622   /* ref connection for slightly better reentrancy */
4623   _dbus_connection_ref_unlocked (connection);
4624
4625   timeouts = connection->timeouts;
4626   connection->timeouts = NULL;
4627   CONNECTION_UNLOCK (connection);
4628   
4629   retval = _dbus_timeout_list_set_functions (timeouts,
4630                                              add_function, remove_function,
4631                                              toggled_function,
4632                                              data, free_data_function);
4633   CONNECTION_LOCK (connection);
4634   connection->timeouts = timeouts;
4635   
4636   CONNECTION_UNLOCK (connection);
4637   /* drop our paranoid refcount */
4638   dbus_connection_unref (connection);
4639
4640   return retval;
4641 }
4642
4643 /**
4644  * Sets the mainloop wakeup function for the connection. This function
4645  * is responsible for waking up the main loop (if its sleeping in
4646  * another thread) when some some change has happened to the
4647  * connection that the mainloop needs to reconsider (e.g. a message
4648  * has been queued for writing).  When using Qt, this typically
4649  * results in a call to QEventLoop::wakeUp().  When using GLib, it
4650  * would call g_main_context_wakeup().
4651  *
4652  * @param connection the connection.
4653  * @param wakeup_main_function function to wake up the mainloop
4654  * @param data data to pass wakeup_main_function
4655  * @param free_data_function function to be called to free the data.
4656  */
4657 void
4658 dbus_connection_set_wakeup_main_function (DBusConnection            *connection,
4659                                           DBusWakeupMainFunction     wakeup_main_function,
4660                                           void                      *data,
4661                                           DBusFreeFunction           free_data_function)
4662 {
4663   void *old_data;
4664   DBusFreeFunction old_free_data;
4665
4666   _dbus_return_if_fail (connection != NULL);
4667   
4668   CONNECTION_LOCK (connection);
4669   old_data = connection->wakeup_main_data;
4670   old_free_data = connection->free_wakeup_main_data;
4671
4672   connection->wakeup_main_function = wakeup_main_function;
4673   connection->wakeup_main_data = data;
4674   connection->free_wakeup_main_data = free_data_function;
4675   
4676   CONNECTION_UNLOCK (connection);
4677
4678   /* Callback outside the lock */
4679   if (old_free_data)
4680     (*old_free_data) (old_data);
4681 }
4682
4683 /**
4684  * Set a function to be invoked when the dispatch status changes.
4685  * If the dispatch status is #DBUS_DISPATCH_DATA_REMAINS, then
4686  * dbus_connection_dispatch() needs to be called to process incoming
4687  * messages. However, dbus_connection_dispatch() MUST NOT BE CALLED
4688  * from inside the DBusDispatchStatusFunction. Indeed, almost
4689  * any reentrancy in this function is a bad idea. Instead,
4690  * the DBusDispatchStatusFunction should simply save an indication
4691  * that messages should be dispatched later, when the main loop
4692  * is re-entered.
4693  *
4694  * If you don't set a dispatch status function, you have to be sure to
4695  * dispatch on every iteration of your main loop, especially if
4696  * dbus_watch_handle() or dbus_timeout_handle() were called.
4697  *
4698  * @param connection the connection
4699  * @param function function to call on dispatch status changes
4700  * @param data data for function
4701  * @param free_data_function free the function data
4702  */
4703 void
4704 dbus_connection_set_dispatch_status_function (DBusConnection             *connection,
4705                                               DBusDispatchStatusFunction  function,
4706                                               void                       *data,
4707                                               DBusFreeFunction            free_data_function)
4708 {
4709   void *old_data;
4710   DBusFreeFunction old_free_data;
4711
4712   _dbus_return_if_fail (connection != NULL);
4713   
4714   CONNECTION_LOCK (connection);
4715   old_data = connection->dispatch_status_data;
4716   old_free_data = connection->free_dispatch_status_data;
4717
4718   connection->dispatch_status_function = function;
4719   connection->dispatch_status_data = data;
4720   connection->free_dispatch_status_data = free_data_function;
4721   
4722   CONNECTION_UNLOCK (connection);
4723
4724   /* Callback outside the lock */
4725   if (old_free_data)
4726     (*old_free_data) (old_data);
4727 }
4728
4729 /**
4730  * Get the UNIX file descriptor of the connection, if any.  This can
4731  * be used for SELinux access control checks with getpeercon() for
4732  * example. DO NOT read or write to the file descriptor, or try to
4733  * select() on it; use DBusWatch for main loop integration. Not all
4734  * connections will have a file descriptor. So for adding descriptors
4735  * to the main loop, use dbus_watch_get_fd() and so forth.
4736  *
4737  * If the connection is socket-based, you can also use
4738  * dbus_connection_get_socket(), which will work on Windows too.
4739  * This function always fails on Windows.
4740  *
4741  * Right now the returned descriptor is always a socket, but
4742  * that is not guaranteed.
4743  * 
4744  * @param connection the connection
4745  * @param fd return location for the file descriptor.
4746  * @returns #TRUE if fd is successfully obtained.
4747  */
4748 dbus_bool_t
4749 dbus_connection_get_unix_fd (DBusConnection *connection,
4750                              int            *fd)
4751 {
4752   _dbus_return_val_if_fail (connection != NULL, FALSE);
4753   _dbus_return_val_if_fail (connection->transport != NULL, FALSE);
4754
4755 #ifdef DBUS_WIN
4756   /* FIXME do this on a lower level */
4757   return FALSE;
4758 #endif
4759   
4760   return dbus_connection_get_socket(connection, fd);
4761 }
4762
4763 /**
4764  * Gets the underlying Windows or UNIX socket file descriptor
4765  * of the connection, if any. DO NOT read or write to the file descriptor, or try to
4766  * select() on it; use DBusWatch for main loop integration. Not all
4767  * connections will have a socket. So for adding descriptors
4768  * to the main loop, use dbus_watch_get_fd() and so forth.
4769  *
4770  * If the connection is not socket-based, this function will return FALSE,
4771  * even if the connection does have a file descriptor of some kind.
4772  * i.e. this function always returns specifically a socket file descriptor.
4773  * 
4774  * @param connection the connection
4775  * @param fd return location for the file descriptor.
4776  * @returns #TRUE if fd is successfully obtained.
4777  */
4778 dbus_bool_t
4779 dbus_connection_get_socket(DBusConnection              *connection,
4780                            int                         *fd)
4781 {
4782   dbus_bool_t retval;
4783
4784   _dbus_return_val_if_fail (connection != NULL, FALSE);
4785   _dbus_return_val_if_fail (connection->transport != NULL, FALSE);
4786   
4787   CONNECTION_LOCK (connection);
4788   
4789   retval = _dbus_transport_get_socket_fd (connection->transport,
4790                                           fd);
4791
4792   CONNECTION_UNLOCK (connection);
4793
4794   return retval;
4795 }
4796
4797
4798 /**
4799  * Gets the UNIX user ID of the connection if known.  Returns #TRUE if
4800  * the uid is filled in.  Always returns #FALSE on non-UNIX platforms
4801  * for now, though in theory someone could hook Windows to NIS or
4802  * something.  Always returns #FALSE prior to authenticating the
4803  * connection.
4804  *
4805  * The UID is only read by servers from clients; clients can't usually
4806  * get the UID of servers, because servers do not authenticate to
4807  * clients.  The returned UID is the UID the connection authenticated
4808  * as.
4809  *
4810  * The message bus is a server and the apps connecting to the bus
4811  * are clients.
4812  *
4813  * You can ask the bus to tell you the UID of another connection though
4814  * if you like; this is done with dbus_bus_get_unix_user().
4815  *
4816  * @param connection the connection
4817  * @param uid return location for the user ID
4818  * @returns #TRUE if uid is filled in with a valid user ID
4819  */
4820 dbus_bool_t
4821 dbus_connection_get_unix_user (DBusConnection *connection,
4822                                unsigned long  *uid)
4823 {
4824   dbus_bool_t result;
4825
4826   _dbus_return_val_if_fail (connection != NULL, FALSE);
4827   _dbus_return_val_if_fail (uid != NULL, FALSE);
4828   
4829   CONNECTION_LOCK (connection);
4830
4831   if (!_dbus_transport_get_is_authenticated (connection->transport))
4832     result = FALSE;
4833   else
4834     result = _dbus_transport_get_unix_user (connection->transport,
4835                                             uid);
4836
4837 #ifdef DBUS_WIN
4838   _dbus_assert (!result);
4839 #endif
4840   
4841   CONNECTION_UNLOCK (connection);
4842
4843   return result;
4844 }
4845
4846 /**
4847  * Gets the process ID of the connection if any.
4848  * Returns #TRUE if the pid is filled in.
4849  * Always returns #FALSE prior to authenticating the
4850  * connection.
4851  *
4852  * @param connection the connection
4853  * @param pid return location for the process ID
4854  * @returns #TRUE if uid is filled in with a valid process ID
4855  */
4856 dbus_bool_t
4857 dbus_connection_get_unix_process_id (DBusConnection *connection,
4858                                      unsigned long  *pid)
4859 {
4860   dbus_bool_t result;
4861
4862   _dbus_return_val_if_fail (connection != NULL, FALSE);
4863   _dbus_return_val_if_fail (pid != NULL, FALSE);
4864   
4865   CONNECTION_LOCK (connection);
4866
4867   if (!_dbus_transport_get_is_authenticated (connection->transport))
4868     result = FALSE;
4869   else
4870     result = _dbus_transport_get_unix_process_id (connection->transport,
4871                                                   pid);
4872 #ifdef DBUS_WIN
4873   _dbus_assert (!result);
4874 #endif
4875   
4876   CONNECTION_UNLOCK (connection);
4877
4878   return result;
4879 }
4880
4881 /**
4882  * Sets a predicate function used to determine whether a given user ID
4883  * is allowed to connect. When an incoming connection has
4884  * authenticated with a particular user ID, this function is called;
4885  * if it returns #TRUE, the connection is allowed to proceed,
4886  * otherwise the connection is disconnected.
4887  *
4888  * If the function is set to #NULL (as it is by default), then
4889  * only the same UID as the server process will be allowed to
4890  * connect. Also, root is always allowed to connect.
4891  *
4892  * On Windows, the function will be set and its free_data_function will
4893  * be invoked when the connection is freed or a new function is set.
4894  * However, the function will never be called, because there are
4895  * no UNIX user ids to pass to it, or at least none of the existing
4896  * auth protocols would allow authenticating as a UNIX user on Windows.
4897  * 
4898  * @param connection the connection
4899  * @param function the predicate
4900  * @param data data to pass to the predicate
4901  * @param free_data_function function to free the data
4902  */
4903 void
4904 dbus_connection_set_unix_user_function (DBusConnection             *connection,
4905                                         DBusAllowUnixUserFunction   function,
4906                                         void                       *data,
4907                                         DBusFreeFunction            free_data_function)
4908 {
4909   void *old_data = NULL;
4910   DBusFreeFunction old_free_function = NULL;
4911
4912   _dbus_return_if_fail (connection != NULL);
4913   
4914   CONNECTION_LOCK (connection);
4915   _dbus_transport_set_unix_user_function (connection->transport,
4916                                           function, data, free_data_function,
4917                                           &old_data, &old_free_function);
4918   CONNECTION_UNLOCK (connection);
4919
4920   if (old_free_function != NULL)
4921     (* old_free_function) (old_data);
4922 }
4923
4924 /**
4925  * Gets the Windows user SID of the connection if known.  Returns
4926  * #TRUE if the ID is filled in.  Always returns #FALSE on non-Windows
4927  * platforms for now, though in theory someone could hook UNIX to
4928  * Active Directory or something.  Always returns #FALSE prior to
4929  * authenticating the connection.
4930  *
4931  * The user is only read by servers from clients; clients can't usually
4932  * get the user of servers, because servers do not authenticate to
4933  * clients. The returned user is the user the connection authenticated
4934  * as.
4935  *
4936  * The message bus is a server and the apps connecting to the bus
4937  * are clients.
4938  *
4939  * The returned user string has to be freed with dbus_free().
4940  *
4941  * The return value indicates whether the user SID is available;
4942  * if it's available but we don't have the memory to copy it,
4943  * then the return value is #TRUE and #NULL is given as the SID.
4944  * 
4945  * @todo We would like to be able to say "You can ask the bus to tell
4946  * you the user of another connection though if you like; this is done
4947  * with dbus_bus_get_windows_user()." But this has to be implemented
4948  * in bus/driver.c and dbus/dbus-bus.c, and is pointless anyway
4949  * since on Windows we only use the session bus for now.
4950  *
4951  * @param connection the connection
4952  * @param windows_sid_p return location for an allocated copy of the user ID, or #NULL if no memory
4953  * @returns #TRUE if user is available (returned value may be #NULL anyway if no memory)
4954  */
4955 dbus_bool_t
4956 dbus_connection_get_windows_user (DBusConnection             *connection,
4957                                   char                      **windows_sid_p)
4958 {
4959   dbus_bool_t result;
4960
4961   _dbus_return_val_if_fail (connection != NULL, FALSE);
4962   _dbus_return_val_if_fail (windows_sid_p != NULL, FALSE);
4963   
4964   CONNECTION_LOCK (connection);
4965
4966   if (!_dbus_transport_get_is_authenticated (connection->transport))
4967     result = FALSE;
4968   else
4969     result = _dbus_transport_get_windows_user (connection->transport,
4970                                                windows_sid_p);
4971
4972 #ifdef DBUS_UNIX
4973   _dbus_assert (!result);
4974 #endif
4975   
4976   CONNECTION_UNLOCK (connection);
4977
4978   return result;
4979 }
4980
4981 /**
4982  * Sets a predicate function used to determine whether a given user ID
4983  * is allowed to connect. When an incoming connection has
4984  * authenticated with a particular user ID, this function is called;
4985  * if it returns #TRUE, the connection is allowed to proceed,
4986  * otherwise the connection is disconnected.
4987  *
4988  * If the function is set to #NULL (as it is by default), then
4989  * only the same user owning the server process will be allowed to
4990  * connect.
4991  *
4992  * On UNIX, the function will be set and its free_data_function will
4993  * be invoked when the connection is freed or a new function is set.
4994  * However, the function will never be called, because there is no
4995  * way right now to authenticate as a Windows user on UNIX.
4996  * 
4997  * @param connection the connection
4998  * @param function the predicate
4999  * @param data data to pass to the predicate
5000  * @param free_data_function function to free the data
5001  */
5002 void
5003 dbus_connection_set_windows_user_function (DBusConnection              *connection,
5004                                            DBusAllowWindowsUserFunction function,
5005                                            void                        *data,
5006                                            DBusFreeFunction             free_data_function)
5007 {
5008   void *old_data = NULL;
5009   DBusFreeFunction old_free_function = NULL;
5010
5011   _dbus_return_if_fail (connection != NULL);
5012   
5013   CONNECTION_LOCK (connection);
5014   _dbus_transport_set_windows_user_function (connection->transport,
5015                                              function, data, free_data_function,
5016                                              &old_data, &old_free_function);
5017   CONNECTION_UNLOCK (connection);
5018
5019   if (old_free_function != NULL)
5020     (* old_free_function) (old_data);
5021 }
5022
5023 /**
5024  * This function must be called on the server side of a connection when the
5025  * connection is first seen in the #DBusNewConnectionFunction. If set to
5026  * #TRUE (the default is #FALSE), then the connection can proceed even if
5027  * the client does not authenticate as some user identity, i.e. clients
5028  * can connect anonymously.
5029  * 
5030  * This setting interacts with the available authorization mechanisms
5031  * (see dbus_server_set_auth_mechanisms()). Namely, an auth mechanism
5032  * such as ANONYMOUS that supports anonymous auth must be included in
5033  * the list of available mechanisms for anonymous login to work.
5034  *
5035  * This setting also changes the default rule for connections
5036  * authorized as a user; normally, if a connection authorizes as
5037  * a user identity, it is permitted if the user identity is
5038  * root or the user identity matches the user identity of the server
5039  * process. If anonymous connections are allowed, however,
5040  * then any user identity is allowed.
5041  *
5042  * You can override the rules for connections authorized as a
5043  * user identity with dbus_connection_set_unix_user_function()
5044  * and dbus_connection_set_windows_user_function().
5045  * 
5046  * @param connection the connection
5047  * @param value whether to allow authentication as an anonymous user
5048  */
5049 void
5050 dbus_connection_set_allow_anonymous (DBusConnection             *connection,
5051                                      dbus_bool_t                 value)
5052 {
5053   _dbus_return_if_fail (connection != NULL);
5054   
5055   CONNECTION_LOCK (connection);
5056   _dbus_transport_set_allow_anonymous (connection->transport, value);
5057   CONNECTION_UNLOCK (connection);
5058 }
5059
5060 /**
5061  *
5062  * Normally #DBusConnection automatically handles all messages to the
5063  * org.freedesktop.DBus.Peer interface. However, the message bus wants
5064  * to be able to route methods on that interface through the bus and
5065  * to other applications. If routing peer messages is enabled, then
5066  * messages with the org.freedesktop.DBus.Peer interface that also
5067  * have a bus destination name set will not be automatically
5068  * handled by the #DBusConnection and instead will be dispatched
5069  * normally to the application.
5070  *
5071  * If a normal application sets this flag, it can break things badly.
5072  * So don't set this unless you are the message bus.
5073  *
5074  * @param connection the connection
5075  * @param value #TRUE to pass through org.freedesktop.DBus.Peer messages with a bus name set
5076  */
5077 void
5078 dbus_connection_set_route_peer_messages (DBusConnection             *connection,
5079                                          dbus_bool_t                 value)
5080 {
5081   _dbus_return_if_fail (connection != NULL);
5082   
5083   CONNECTION_LOCK (connection);
5084   connection->route_peer_messages = TRUE;
5085   CONNECTION_UNLOCK (connection);
5086 }
5087
5088 /**
5089  * Adds a message filter. Filters are handlers that are run on all
5090  * incoming messages, prior to the objects registered with
5091  * dbus_connection_register_object_path().  Filters are run in the
5092  * order that they were added.  The same handler can be added as a
5093  * filter more than once, in which case it will be run more than once.
5094  * Filters added during a filter callback won't be run on the message
5095  * being processed.
5096  *
5097  * @todo we don't run filters on messages while blocking without
5098  * entering the main loop, since filters are run as part of
5099  * dbus_connection_dispatch(). This is probably a feature, as filters
5100  * could create arbitrary reentrancy. But kind of sucks if you're
5101  * trying to filter METHOD_RETURN for some reason.
5102  *
5103  * @param connection the connection
5104  * @param function function to handle messages
5105  * @param user_data user data to pass to the function
5106  * @param free_data_function function to use for freeing user data
5107  * @returns #TRUE on success, #FALSE if not enough memory.
5108  */
5109 dbus_bool_t
5110 dbus_connection_add_filter (DBusConnection            *connection,
5111                             DBusHandleMessageFunction  function,
5112                             void                      *user_data,
5113                             DBusFreeFunction           free_data_function)
5114 {
5115   DBusMessageFilter *filter;
5116   
5117   _dbus_return_val_if_fail (connection != NULL, FALSE);
5118   _dbus_return_val_if_fail (function != NULL, FALSE);
5119
5120   filter = dbus_new0 (DBusMessageFilter, 1);
5121   if (filter == NULL)
5122     return FALSE;
5123
5124   filter->refcount.value = 1;
5125   
5126   CONNECTION_LOCK (connection);
5127
5128   if (!_dbus_list_append (&connection->filter_list,
5129                           filter))
5130     {
5131       _dbus_message_filter_unref (filter);
5132       CONNECTION_UNLOCK (connection);
5133       return FALSE;
5134     }
5135
5136   /* Fill in filter after all memory allocated,
5137    * so we don't run the free_user_data_function
5138    * if the add_filter() fails
5139    */
5140   
5141   filter->function = function;
5142   filter->user_data = user_data;
5143   filter->free_user_data_function = free_data_function;
5144         
5145   CONNECTION_UNLOCK (connection);
5146   return TRUE;
5147 }
5148
5149 /**
5150  * Removes a previously-added message filter. It is a programming
5151  * error to call this function for a handler that has not been added
5152  * as a filter. If the given handler was added more than once, only
5153  * one instance of it will be removed (the most recently-added
5154  * instance).
5155  *
5156  * @param connection the connection
5157  * @param function the handler to remove
5158  * @param user_data user data for the handler to remove
5159  *
5160  */
5161 void
5162 dbus_connection_remove_filter (DBusConnection            *connection,
5163                                DBusHandleMessageFunction  function,
5164                                void                      *user_data)
5165 {
5166   DBusList *link;
5167   DBusMessageFilter *filter;
5168   
5169   _dbus_return_if_fail (connection != NULL);
5170   _dbus_return_if_fail (function != NULL);
5171   
5172   CONNECTION_LOCK (connection);
5173
5174   filter = NULL;
5175   
5176   link = _dbus_list_get_last_link (&connection->filter_list);
5177   while (link != NULL)
5178     {
5179       filter = link->data;
5180
5181       if (filter->function == function &&
5182           filter->user_data == user_data)
5183         {
5184           _dbus_list_remove_link (&connection->filter_list, link);
5185           filter->function = NULL;
5186           
5187           break;
5188         }
5189         
5190       link = _dbus_list_get_prev_link (&connection->filter_list, link);
5191     }
5192   
5193   CONNECTION_UNLOCK (connection);
5194
5195 #ifndef DBUS_DISABLE_CHECKS
5196   if (filter == NULL)
5197     {
5198       _dbus_warn_check_failed ("Attempt to remove filter function %p user data %p, but no such filter has been added\n",
5199                                function, user_data);
5200       return;
5201     }
5202 #endif
5203   
5204   /* Call application code */
5205   if (filter->free_user_data_function)
5206     (* filter->free_user_data_function) (filter->user_data);
5207
5208   filter->free_user_data_function = NULL;
5209   filter->user_data = NULL;
5210   
5211   _dbus_message_filter_unref (filter);
5212 }
5213
5214 /**
5215  * Registers a handler for a given path in the object hierarchy.
5216  * The given vtable handles messages sent to exactly the given path.
5217  *
5218  *
5219  * @param connection the connection
5220  * @param path a '/' delimited string of path elements
5221  * @param vtable the virtual table
5222  * @param user_data data to pass to functions in the vtable
5223  * @returns #FALSE if not enough memory
5224  */
5225 dbus_bool_t
5226 dbus_connection_register_object_path (DBusConnection              *connection,
5227                                       const char                  *path,
5228                                       const DBusObjectPathVTable  *vtable,
5229                                       void                        *user_data)
5230 {
5231   char **decomposed_path;
5232   dbus_bool_t retval;
5233   
5234   _dbus_return_val_if_fail (connection != NULL, FALSE);
5235   _dbus_return_val_if_fail (path != NULL, FALSE);
5236   _dbus_return_val_if_fail (path[0] == '/', FALSE);
5237   _dbus_return_val_if_fail (vtable != NULL, FALSE);
5238
5239   if (!_dbus_decompose_path (path, strlen (path), &decomposed_path, NULL))
5240     return FALSE;
5241
5242   CONNECTION_LOCK (connection);
5243
5244   retval = _dbus_object_tree_register (connection->objects,
5245                                        FALSE,
5246                                        (const char **) decomposed_path, vtable,
5247                                        user_data);
5248
5249   CONNECTION_UNLOCK (connection);
5250
5251   dbus_free_string_array (decomposed_path);
5252
5253   return retval;
5254 }
5255
5256 /**
5257  * Registers a fallback handler for a given subsection of the object
5258  * hierarchy.  The given vtable handles messages at or below the given
5259  * path. You can use this to establish a default message handling
5260  * policy for a whole "subdirectory."
5261  *
5262  * @param connection the connection
5263  * @param path a '/' delimited string of path elements
5264  * @param vtable the virtual table
5265  * @param user_data data to pass to functions in the vtable
5266  * @returns #FALSE if not enough memory
5267  */
5268 dbus_bool_t
5269 dbus_connection_register_fallback (DBusConnection              *connection,
5270                                    const char                  *path,
5271                                    const DBusObjectPathVTable  *vtable,
5272                                    void                        *user_data)
5273 {
5274   char **decomposed_path;
5275   dbus_bool_t retval;
5276   
5277   _dbus_return_val_if_fail (connection != NULL, FALSE);
5278   _dbus_return_val_if_fail (path != NULL, FALSE);
5279   _dbus_return_val_if_fail (path[0] == '/', FALSE);
5280   _dbus_return_val_if_fail (vtable != NULL, FALSE);
5281
5282   if (!_dbus_decompose_path (path, strlen (path), &decomposed_path, NULL))
5283     return FALSE;
5284
5285   CONNECTION_LOCK (connection);
5286
5287   retval = _dbus_object_tree_register (connection->objects,
5288                                        TRUE,
5289                                        (const char **) decomposed_path, vtable,
5290                                        user_data);
5291
5292   CONNECTION_UNLOCK (connection);
5293
5294   dbus_free_string_array (decomposed_path);
5295
5296   return retval;
5297 }
5298
5299 /**
5300  * Unregisters the handler registered with exactly the given path.
5301  * It's a bug to call this function for a path that isn't registered.
5302  * Can unregister both fallback paths and object paths.
5303  *
5304  * @param connection the connection
5305  * @param path a '/' delimited string of path elements
5306  * @returns #FALSE if not enough memory
5307  */
5308 dbus_bool_t
5309 dbus_connection_unregister_object_path (DBusConnection              *connection,
5310                                         const char                  *path)
5311 {
5312   char **decomposed_path;
5313
5314   _dbus_return_val_if_fail (connection != NULL, FALSE);
5315   _dbus_return_val_if_fail (path != NULL, FALSE);
5316   _dbus_return_val_if_fail (path[0] == '/', FALSE);
5317
5318   if (!_dbus_decompose_path (path, strlen (path), &decomposed_path, NULL))
5319       return FALSE;
5320
5321   CONNECTION_LOCK (connection);
5322
5323   _dbus_object_tree_unregister_and_unlock (connection->objects, (const char **) decomposed_path);
5324
5325   dbus_free_string_array (decomposed_path);
5326
5327   return TRUE;
5328 }
5329
5330 /**
5331  * Gets the user data passed to dbus_connection_register_object_path()
5332  * or dbus_connection_register_fallback(). If nothing was registered
5333  * at this path, the data is filled in with #NULL.
5334  *
5335  * @param connection the connection
5336  * @param path the path you registered with
5337  * @param data_p location to store the user data, or #NULL
5338  * @returns #FALSE if not enough memory
5339  */
5340 dbus_bool_t
5341 dbus_connection_get_object_path_data (DBusConnection *connection,
5342                                       const char     *path,
5343                                       void          **data_p)
5344 {
5345   char **decomposed_path;
5346
5347   _dbus_return_val_if_fail (connection != NULL, FALSE);
5348   _dbus_return_val_if_fail (path != NULL, FALSE);
5349   _dbus_return_val_if_fail (data_p != NULL, FALSE);
5350
5351   *data_p = NULL;
5352   
5353   if (!_dbus_decompose_path (path, strlen (path), &decomposed_path, NULL))
5354     return FALSE;
5355   
5356   CONNECTION_LOCK (connection);
5357
5358   *data_p = _dbus_object_tree_get_user_data_unlocked (connection->objects, (const char**) decomposed_path);
5359
5360   CONNECTION_UNLOCK (connection);
5361
5362   dbus_free_string_array (decomposed_path);
5363
5364   return TRUE;
5365 }
5366
5367 /**
5368  * Lists the registered fallback handlers and object path handlers at
5369  * the given parent_path. The returned array should be freed with
5370  * dbus_free_string_array().
5371  *
5372  * @param connection the connection
5373  * @param parent_path the path to list the child handlers of
5374  * @param child_entries returns #NULL-terminated array of children
5375  * @returns #FALSE if no memory to allocate the child entries
5376  */
5377 dbus_bool_t
5378 dbus_connection_list_registered (DBusConnection              *connection,
5379                                  const char                  *parent_path,
5380                                  char                      ***child_entries)
5381 {
5382   char **decomposed_path;
5383   dbus_bool_t retval;
5384   _dbus_return_val_if_fail (connection != NULL, FALSE);
5385   _dbus_return_val_if_fail (parent_path != NULL, FALSE);
5386   _dbus_return_val_if_fail (parent_path[0] == '/', FALSE);
5387   _dbus_return_val_if_fail (child_entries != NULL, FALSE);
5388
5389   if (!_dbus_decompose_path (parent_path, strlen (parent_path), &decomposed_path, NULL))
5390     return FALSE;
5391
5392   CONNECTION_LOCK (connection);
5393
5394   retval = _dbus_object_tree_list_registered_and_unlock (connection->objects,
5395                                                          (const char **) decomposed_path,
5396                                                          child_entries);
5397   dbus_free_string_array (decomposed_path);
5398
5399   return retval;
5400 }
5401
5402 static DBusDataSlotAllocator slot_allocator;
5403 _DBUS_DEFINE_GLOBAL_LOCK (connection_slots);
5404
5405 /**
5406  * Allocates an integer ID to be used for storing application-specific
5407  * data on any DBusConnection. The allocated ID may then be used
5408  * with dbus_connection_set_data() and dbus_connection_get_data().
5409  * The passed-in slot must be initialized to -1, and is filled in
5410  * with the slot ID. If the passed-in slot is not -1, it's assumed
5411  * to be already allocated, and its refcount is incremented.
5412  * 
5413  * The allocated slot is global, i.e. all DBusConnection objects will
5414  * have a slot with the given integer ID reserved.
5415  *
5416  * @param slot_p address of a global variable storing the slot
5417  * @returns #FALSE on failure (no memory)
5418  */
5419 dbus_bool_t
5420 dbus_connection_allocate_data_slot (dbus_int32_t *slot_p)
5421 {
5422   return _dbus_data_slot_allocator_alloc (&slot_allocator,
5423                                           &_DBUS_LOCK_NAME (connection_slots),
5424                                           slot_p);
5425 }
5426
5427 /**
5428  * Deallocates a global ID for connection data slots.
5429  * dbus_connection_get_data() and dbus_connection_set_data() may no
5430  * longer be used with this slot.  Existing data stored on existing
5431  * DBusConnection objects will be freed when the connection is
5432  * finalized, but may not be retrieved (and may only be replaced if
5433  * someone else reallocates the slot).  When the refcount on the
5434  * passed-in slot reaches 0, it is set to -1.
5435  *
5436  * @param slot_p address storing the slot to deallocate
5437  */
5438 void
5439 dbus_connection_free_data_slot (dbus_int32_t *slot_p)
5440 {
5441   _dbus_return_if_fail (*slot_p >= 0);
5442   
5443   _dbus_data_slot_allocator_free (&slot_allocator, slot_p);
5444 }
5445
5446 /**
5447  * Stores a pointer on a DBusConnection, along
5448  * with an optional function to be used for freeing
5449  * the data when the data is set again, or when
5450  * the connection is finalized. The slot number
5451  * must have been allocated with dbus_connection_allocate_data_slot().
5452  *
5453  * @param connection the connection
5454  * @param slot the slot number
5455  * @param data the data to store
5456  * @param free_data_func finalizer function for the data
5457  * @returns #TRUE if there was enough memory to store the data
5458  */
5459 dbus_bool_t
5460 dbus_connection_set_data (DBusConnection   *connection,
5461                           dbus_int32_t      slot,
5462                           void             *data,
5463                           DBusFreeFunction  free_data_func)
5464 {
5465   DBusFreeFunction old_free_func;
5466   void *old_data;
5467   dbus_bool_t retval;
5468
5469   _dbus_return_val_if_fail (connection != NULL, FALSE);
5470   _dbus_return_val_if_fail (slot >= 0, FALSE);
5471   
5472   CONNECTION_LOCK (connection);
5473
5474   retval = _dbus_data_slot_list_set (&slot_allocator,
5475                                      &connection->slot_list,
5476                                      slot, data, free_data_func,
5477                                      &old_free_func, &old_data);
5478   
5479   CONNECTION_UNLOCK (connection);
5480
5481   if (retval)
5482     {
5483       /* Do the actual free outside the connection lock */
5484       if (old_free_func)
5485         (* old_free_func) (old_data);
5486     }
5487
5488   return retval;
5489 }
5490
5491 /**
5492  * Retrieves data previously set with dbus_connection_set_data().
5493  * The slot must still be allocated (must not have been freed).
5494  *
5495  * @param connection the connection
5496  * @param slot the slot to get data from
5497  * @returns the data, or #NULL if not found
5498  */
5499 void*
5500 dbus_connection_get_data (DBusConnection   *connection,
5501                           dbus_int32_t      slot)
5502 {
5503   void *res;
5504
5505   _dbus_return_val_if_fail (connection != NULL, NULL);
5506   
5507   CONNECTION_LOCK (connection);
5508
5509   res = _dbus_data_slot_list_get (&slot_allocator,
5510                                   &connection->slot_list,
5511                                   slot);
5512   
5513   CONNECTION_UNLOCK (connection);
5514
5515   return res;
5516 }
5517
5518 /**
5519  * This function sets a global flag for whether dbus_connection_new()
5520  * will set SIGPIPE behavior to SIG_IGN.
5521  *
5522  * @param will_modify_sigpipe #TRUE to allow sigpipe to be set to SIG_IGN
5523  */
5524 void
5525 dbus_connection_set_change_sigpipe (dbus_bool_t will_modify_sigpipe)
5526 {  
5527   _dbus_modify_sigpipe = will_modify_sigpipe != FALSE;
5528 }
5529
5530 /**
5531  * Specifies the maximum size message this connection is allowed to
5532  * receive. Larger messages will result in disconnecting the
5533  * connection.
5534  * 
5535  * @param connection a #DBusConnection
5536  * @param size maximum message size the connection can receive, in bytes
5537  */
5538 void
5539 dbus_connection_set_max_message_size (DBusConnection *connection,
5540                                       long            size)
5541 {
5542   _dbus_return_if_fail (connection != NULL);
5543   
5544   CONNECTION_LOCK (connection);
5545   _dbus_transport_set_max_message_size (connection->transport,
5546                                         size);
5547   CONNECTION_UNLOCK (connection);
5548 }
5549
5550 /**
5551  * Gets the value set by dbus_connection_set_max_message_size().
5552  *
5553  * @param connection the connection
5554  * @returns the max size of a single message
5555  */
5556 long
5557 dbus_connection_get_max_message_size (DBusConnection *connection)
5558 {
5559   long res;
5560
5561   _dbus_return_val_if_fail (connection != NULL, 0);
5562   
5563   CONNECTION_LOCK (connection);
5564   res = _dbus_transport_get_max_message_size (connection->transport);
5565   CONNECTION_UNLOCK (connection);
5566   return res;
5567 }
5568
5569 /**
5570  * Sets the maximum total number of bytes that can be used for all messages
5571  * received on this connection. Messages count toward the maximum until
5572  * they are finalized. When the maximum is reached, the connection will
5573  * not read more data until some messages are finalized.
5574  *
5575  * The semantics of the maximum are: if outstanding messages are
5576  * already above the maximum, additional messages will not be read.
5577  * The semantics are not: if the next message would cause us to exceed
5578  * the maximum, we don't read it. The reason is that we don't know the
5579  * size of a message until after we read it.
5580  *
5581  * Thus, the max live messages size can actually be exceeded
5582  * by up to the maximum size of a single message.
5583  * 
5584  * Also, if we read say 1024 bytes off the wire in a single read(),
5585  * and that contains a half-dozen small messages, we may exceed the
5586  * size max by that amount. But this should be inconsequential.
5587  *
5588  * This does imply that we can't call read() with a buffer larger
5589  * than we're willing to exceed this limit by.
5590  *
5591  * @param connection the connection
5592  * @param size the maximum size in bytes of all outstanding messages
5593  */
5594 void
5595 dbus_connection_set_max_received_size (DBusConnection *connection,
5596                                        long            size)
5597 {
5598   _dbus_return_if_fail (connection != NULL);
5599   
5600   CONNECTION_LOCK (connection);
5601   _dbus_transport_set_max_received_size (connection->transport,
5602                                          size);
5603   CONNECTION_UNLOCK (connection);
5604 }
5605
5606 /**
5607  * Gets the value set by dbus_connection_set_max_received_size().
5608  *
5609  * @param connection the connection
5610  * @returns the max size of all live messages
5611  */
5612 long
5613 dbus_connection_get_max_received_size (DBusConnection *connection)
5614 {
5615   long res;
5616
5617   _dbus_return_val_if_fail (connection != NULL, 0);
5618   
5619   CONNECTION_LOCK (connection);
5620   res = _dbus_transport_get_max_received_size (connection->transport);
5621   CONNECTION_UNLOCK (connection);
5622   return res;
5623 }
5624
5625 /**
5626  * Gets the approximate size in bytes of all messages in the outgoing
5627  * message queue. The size is approximate in that you shouldn't use
5628  * it to decide how many bytes to read off the network or anything
5629  * of that nature, as optimizations may choose to tell small white lies
5630  * to avoid performance overhead.
5631  *
5632  * @param connection the connection
5633  * @returns the number of bytes that have been queued up but not sent
5634  */
5635 long
5636 dbus_connection_get_outgoing_size (DBusConnection *connection)
5637 {
5638   long res;
5639
5640   _dbus_return_val_if_fail (connection != NULL, 0);
5641   
5642   CONNECTION_LOCK (connection);
5643   res = _dbus_counter_get_value (connection->outgoing_counter);
5644   CONNECTION_UNLOCK (connection);
5645   return res;
5646 }
5647
5648 /** @} */