2006-09-30 Havoc Pennington <hp@redhat.com>
[platform/upstream/dbus.git] / bus / connection.c
1 /* -*- mode: C; c-file-style: "gnu" -*- */
2 /* connection.c  Client connections
3  *
4  * Copyright (C) 2003  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 #include "connection.h"
24 #include "dispatch.h"
25 #include "policy.h"
26 #include "services.h"
27 #include "utils.h"
28 #include "signals.h"
29 #include "expirelist.h"
30 #include "selinux.h"
31 #include <dbus/dbus-list.h>
32 #include <dbus/dbus-hash.h>
33 #include <dbus/dbus-timeout.h>
34
35 static void bus_connection_remove_transactions (DBusConnection *connection);
36
37 typedef struct
38 {
39   BusExpireItem expire_item;
40
41   DBusConnection *will_get_reply;
42   DBusConnection *will_send_reply;
43
44   dbus_uint32_t reply_serial;
45   
46 } BusPendingReply;
47
48 struct BusConnections
49 {
50   int refcount;
51   DBusList *completed;  /**< List of all completed connections */
52   int n_completed;      /**< Length of completed list */
53   DBusList *incomplete; /**< List of all not-yet-active connections */
54   int n_incomplete;     /**< Length of incomplete list */
55   BusContext *context;
56   DBusHashTable *completed_by_user; /**< Number of completed connections for each UID */
57   DBusTimeout *expire_timeout; /**< Timeout for expiring incomplete connections. */
58   int stamp;                   /**< Incrementing number */
59   BusExpireList *pending_replies; /**< List of pending replies */
60 };
61
62 static dbus_int32_t connection_data_slot = -1;
63
64 typedef struct
65 {
66   BusConnections *connections;
67   DBusList *link_in_connection_list;
68   DBusConnection *connection;
69   DBusList *services_owned;
70   int n_services_owned;
71   DBusList *match_rules;
72   int n_match_rules;
73   char *name;
74   DBusList *transaction_messages; /**< Stuff we need to send as part of a transaction */
75   DBusMessage *oom_message;
76   DBusPreallocatedSend *oom_preallocated;
77   BusClientPolicy *policy;
78
79   BusSELinuxID *selinux_id;
80
81   long connection_tv_sec;  /**< Time when we connected (seconds component) */
82   long connection_tv_usec; /**< Time when we connected (microsec component) */
83   int stamp;               /**< connections->stamp last time we were traversed */
84 } BusConnectionData;
85
86 static dbus_bool_t bus_pending_reply_expired (BusExpireList *list,
87                                               DBusList      *link,
88                                               void          *data);
89
90 static void bus_connection_drop_pending_replies (BusConnections  *connections,
91                                                  DBusConnection  *connection);
92
93 static dbus_bool_t expire_incomplete_timeout (void *data);
94
95 #define BUS_CONNECTION_DATA(connection) (dbus_connection_get_data ((connection), connection_data_slot))
96
97 static DBusLoop*
98 connection_get_loop (DBusConnection *connection)
99 {
100   BusConnectionData *d;
101
102   d = BUS_CONNECTION_DATA (connection);
103
104   return bus_context_get_loop (d->connections->context);
105 }
106
107
108 static int
109 get_connections_for_uid (BusConnections *connections,
110                          dbus_uid_t      uid)
111 {
112   void *val;
113   int current_count;
114
115   /* val is NULL is 0 when it isn't in the hash yet */
116   
117   val = _dbus_hash_table_lookup_ulong (connections->completed_by_user,
118                                        uid);
119
120   current_count = _DBUS_POINTER_TO_INT (val);
121
122   return current_count;
123 }
124
125 static dbus_bool_t
126 adjust_connections_for_uid (BusConnections *connections,
127                             dbus_uid_t      uid,
128                             int             adjustment)
129 {
130   int current_count;
131
132   current_count = get_connections_for_uid (connections, uid);
133
134   _dbus_verbose ("Adjusting connection count for UID " DBUS_UID_FORMAT
135                  ": was %d adjustment %d making %d\n",
136                  uid, current_count, adjustment, current_count + adjustment);
137   
138   _dbus_assert (current_count >= 0);
139   
140   current_count += adjustment;
141
142   _dbus_assert (current_count >= 0);
143
144   if (current_count == 0)
145     {
146       _dbus_hash_table_remove_ulong (connections->completed_by_user, uid);
147       return TRUE;
148     }
149   else
150     {
151       dbus_bool_t retval;
152       
153       retval = _dbus_hash_table_insert_ulong (connections->completed_by_user,
154                                               uid, _DBUS_INT_TO_POINTER (current_count));
155
156       /* only positive adjustment can fail as otherwise
157        * a hash entry should already exist
158        */
159       _dbus_assert (adjustment > 0 ||
160                     (adjustment <= 0 && retval));
161
162       return retval;
163     }
164 }
165
166 void
167 bus_connection_disconnected (DBusConnection *connection)
168 {
169   BusConnectionData *d;
170   BusService *service;
171   BusMatchmaker *matchmaker;
172   
173   d = BUS_CONNECTION_DATA (connection);
174   _dbus_assert (d != NULL);
175
176   _dbus_verbose ("%s disconnected, dropping all service ownership and releasing\n",
177                  d->name ? d->name : "(inactive)");
178
179   /* Delete our match rules */
180   if (d->n_match_rules > 0)
181     {
182       matchmaker = bus_context_get_matchmaker (d->connections->context);
183       bus_matchmaker_disconnected (matchmaker, connection);
184     }
185   
186   /* Drop any service ownership. Unfortunately, this requires
187    * memory allocation and there doesn't seem to be a good way to
188    * handle it other than sleeping; we can't "fail" the operation of
189    * disconnecting a client, and preallocating a broadcast "service is
190    * now gone" message for every client-service pair seems kind of
191    * involved.
192    */
193   while ((service = _dbus_list_get_last (&d->services_owned)))
194     {
195       BusTransaction *transaction;
196       DBusError error;
197
198     retry:
199       
200       dbus_error_init (&error);
201         
202       while ((transaction = bus_transaction_new (d->connections->context)) == NULL)
203         _dbus_wait_for_memory ();
204         
205       if (!bus_service_remove_owner (service, connection,
206                                      transaction, &error))
207         {
208           _DBUS_ASSERT_ERROR_IS_SET (&error);
209           
210           if (dbus_error_has_name (&error, DBUS_ERROR_NO_MEMORY))
211             {
212               dbus_error_free (&error);
213               bus_transaction_cancel_and_free (transaction);
214               _dbus_wait_for_memory ();
215               goto retry;
216             }
217           else
218             {
219               _dbus_verbose ("Failed to remove service owner: %s %s\n",
220                              error.name, error.message);
221               _dbus_assert_not_reached ("Removing service owner failed for non-memory-related reason");
222             }
223         }
224         
225       bus_transaction_execute_and_free (transaction);
226     }
227
228   bus_dispatch_remove_connection (connection);
229   
230   /* no more watching */
231   if (!dbus_connection_set_watch_functions (connection,
232                                             NULL, NULL, NULL,
233                                             connection,
234                                             NULL))
235     _dbus_assert_not_reached ("setting watch functions to NULL failed");
236
237   if (!dbus_connection_set_timeout_functions (connection,
238                                               NULL, NULL, NULL,
239                                               connection,
240                                               NULL))
241     _dbus_assert_not_reached ("setting timeout functions to NULL failed");
242   
243   dbus_connection_set_unix_user_function (connection,
244                                           NULL, NULL, NULL);
245
246   dbus_connection_set_dispatch_status_function (connection,
247                                                 NULL, NULL, NULL);
248   
249   bus_connection_remove_transactions (connection);
250
251   if (d->link_in_connection_list != NULL)
252     {
253       if (d->name != NULL)
254         {
255           unsigned long uid;
256           
257           _dbus_list_remove_link (&d->connections->completed, d->link_in_connection_list);
258           d->link_in_connection_list = NULL;
259           d->connections->n_completed -= 1;
260
261           if (dbus_connection_get_unix_user (connection, &uid))
262             {
263               if (!adjust_connections_for_uid (d->connections,
264                                                uid, -1))
265                 _dbus_assert_not_reached ("adjusting downward should never fail");
266             }
267         }
268       else
269         {
270           _dbus_list_remove_link (&d->connections->incomplete, d->link_in_connection_list);
271           d->link_in_connection_list = NULL;
272           d->connections->n_incomplete -= 1;
273         }
274       
275       _dbus_assert (d->connections->n_incomplete >= 0);
276       _dbus_assert (d->connections->n_completed >= 0);
277     }
278
279   bus_connection_drop_pending_replies (d->connections, connection);
280   
281   /* frees "d" as side effect */
282   dbus_connection_set_data (connection,
283                             connection_data_slot,
284                             NULL, NULL);
285   
286   dbus_connection_unref (connection);
287 }
288
289 static dbus_bool_t
290 connection_watch_callback (DBusWatch     *watch,
291                            unsigned int   condition,
292                            void          *data)
293 {
294  /* FIXME this can be done in dbus-mainloop.c
295   * if the code in activation.c for the babysitter
296   * watch handler is fixed.
297   */
298   
299 #if 0
300   _dbus_verbose ("Calling handle_watch\n");
301 #endif
302   return dbus_watch_handle (watch, condition);
303 }
304
305 static dbus_bool_t
306 add_connection_watch (DBusWatch      *watch,
307                       void           *data)
308 {
309   DBusConnection *connection = data;
310
311   return _dbus_loop_add_watch (connection_get_loop (connection),
312                                watch, connection_watch_callback, connection,
313                                NULL);
314 }
315
316 static void
317 remove_connection_watch (DBusWatch      *watch,
318                          void           *data)
319 {
320   DBusConnection *connection = data;
321   
322   _dbus_loop_remove_watch (connection_get_loop (connection),
323                            watch, connection_watch_callback, connection);
324 }
325
326 static void
327 connection_timeout_callback (DBusTimeout   *timeout,
328                              void          *data)
329 {
330   /* DBusConnection *connection = data; */
331
332   /* can return FALSE on OOM but we just let it fire again later */
333   dbus_timeout_handle (timeout);
334 }
335
336 static dbus_bool_t
337 add_connection_timeout (DBusTimeout    *timeout,
338                         void           *data)
339 {
340   DBusConnection *connection = data;
341   
342   return _dbus_loop_add_timeout (connection_get_loop (connection),
343                                  timeout, connection_timeout_callback, connection, NULL);
344 }
345
346 static void
347 remove_connection_timeout (DBusTimeout    *timeout,
348                            void           *data)
349 {
350   DBusConnection *connection = data;
351   
352   _dbus_loop_remove_timeout (connection_get_loop (connection),
353                              timeout, connection_timeout_callback, connection);
354 }
355
356 static void
357 dispatch_status_function (DBusConnection    *connection,
358                           DBusDispatchStatus new_status,
359                           void              *data)
360 {
361   DBusLoop *loop = data;
362   
363   if (new_status != DBUS_DISPATCH_COMPLETE)
364     {
365       while (!_dbus_loop_queue_dispatch (loop, connection))
366         _dbus_wait_for_memory ();
367     }
368 }
369
370 static dbus_bool_t
371 allow_user_function (DBusConnection *connection,
372                      unsigned long   uid,
373                      void           *data)
374 {
375   BusConnectionData *d;
376     
377   d = BUS_CONNECTION_DATA (connection);
378
379   _dbus_assert (d != NULL);
380   
381   return bus_context_allow_user (d->connections->context, uid);
382 }
383
384 static void
385 free_connection_data (void *data)
386 {
387   BusConnectionData *d = data;
388
389   /* services_owned should be NULL since we should be disconnected */
390   _dbus_assert (d->services_owned == NULL);
391   _dbus_assert (d->n_services_owned == 0);
392   /* similarly */
393   _dbus_assert (d->transaction_messages == NULL);
394
395   if (d->oom_preallocated)
396     dbus_connection_free_preallocated_send (d->connection, d->oom_preallocated);
397
398   if (d->oom_message)
399     dbus_message_unref (d->oom_message);
400
401   if (d->policy)
402     bus_client_policy_unref (d->policy);
403
404   if (d->selinux_id)
405     bus_selinux_id_unref (d->selinux_id);
406   
407   dbus_free (d->name);
408   
409   dbus_free (d);
410 }
411
412 static void
413 call_timeout_callback (DBusTimeout   *timeout,
414                        void          *data)
415 {
416   /* can return FALSE on OOM but we just let it fire again later */
417   dbus_timeout_handle (timeout);
418 }
419
420 BusConnections*
421 bus_connections_new (BusContext *context)
422 {
423   BusConnections *connections;
424
425   if (!dbus_connection_allocate_data_slot (&connection_data_slot))
426     goto failed_0;
427
428   connections = dbus_new0 (BusConnections, 1);
429   if (connections == NULL)
430     goto failed_1;
431
432   connections->completed_by_user = _dbus_hash_table_new (DBUS_HASH_ULONG,
433                                                          NULL, NULL);
434   if (connections->completed_by_user == NULL)
435     goto failed_2;
436
437   connections->expire_timeout = _dbus_timeout_new (100, /* irrelevant */
438                                                    expire_incomplete_timeout,
439                                                    connections, NULL);
440   if (connections->expire_timeout == NULL)
441     goto failed_3;
442
443   _dbus_timeout_set_enabled (connections->expire_timeout, FALSE);
444
445   connections->pending_replies = bus_expire_list_new (bus_context_get_loop (context),
446                                                       bus_context_get_reply_timeout (context),
447                                                       bus_pending_reply_expired,
448                                                       connections);
449   if (connections->pending_replies == NULL)
450     goto failed_4;
451   
452   if (!_dbus_loop_add_timeout (bus_context_get_loop (context),
453                                connections->expire_timeout,
454                                call_timeout_callback, NULL, NULL))
455     goto failed_5;
456   
457   connections->refcount = 1;
458   connections->context = context;
459   
460   return connections;
461
462  failed_5:
463   bus_expire_list_free (connections->pending_replies);
464  failed_4:
465   _dbus_timeout_unref (connections->expire_timeout);
466  failed_3:
467   _dbus_hash_table_unref (connections->completed_by_user);
468  failed_2:
469   dbus_free (connections);
470  failed_1:
471   dbus_connection_free_data_slot (&connection_data_slot);
472  failed_0:
473   return NULL;
474 }
475
476 BusConnections *
477 bus_connections_ref (BusConnections *connections)
478 {
479   _dbus_assert (connections->refcount > 0);
480   connections->refcount += 1;
481
482   return connections;
483 }
484
485 void
486 bus_connections_unref (BusConnections *connections)
487 {
488   _dbus_assert (connections->refcount > 0);
489   connections->refcount -= 1;
490   if (connections->refcount == 0)
491     {
492       /* drop all incomplete */
493       while (connections->incomplete != NULL)
494         {
495           DBusConnection *connection;
496
497           connection = connections->incomplete->data;
498
499           dbus_connection_ref (connection);
500           dbus_connection_close (connection);
501           bus_connection_disconnected (connection);
502           dbus_connection_unref (connection);
503         }
504
505       _dbus_assert (connections->n_incomplete == 0);
506       
507       /* drop all real connections */
508       while (connections->completed != NULL)
509         {
510           DBusConnection *connection;
511
512           connection = connections->completed->data;
513
514           dbus_connection_ref (connection);
515           dbus_connection_close (connection);
516           bus_connection_disconnected (connection);
517           dbus_connection_unref (connection);
518         }
519
520       _dbus_assert (connections->n_completed == 0);
521
522       bus_expire_list_free (connections->pending_replies);
523       
524       _dbus_loop_remove_timeout (bus_context_get_loop (connections->context),
525                                  connections->expire_timeout,
526                                  call_timeout_callback, NULL);
527       
528       _dbus_timeout_unref (connections->expire_timeout);
529       
530       _dbus_hash_table_unref (connections->completed_by_user);
531       
532       dbus_free (connections);
533
534       dbus_connection_free_data_slot (&connection_data_slot);
535     }
536 }
537
538 dbus_bool_t
539 bus_connections_setup_connection (BusConnections *connections,
540                                   DBusConnection *connection)
541 {
542   BusConnectionData *d;
543   dbus_bool_t retval;
544   DBusError error;
545   
546   d = dbus_new0 (BusConnectionData, 1);
547   
548   if (d == NULL)
549     return FALSE;
550
551   d->connections = connections;
552   d->connection = connection;
553   
554   _dbus_get_current_time (&d->connection_tv_sec,
555                           &d->connection_tv_usec);
556   
557   _dbus_assert (connection_data_slot >= 0);
558   
559   if (!dbus_connection_set_data (connection,
560                                  connection_data_slot,
561                                  d, free_connection_data))
562     {
563       dbus_free (d);
564       return FALSE;
565     }
566
567   dbus_connection_set_route_peer_messages (connection, TRUE);
568   
569   retval = FALSE;
570
571   dbus_error_init (&error);
572   d->selinux_id = bus_selinux_init_connection_id (connection,
573                                                   &error);
574   if (dbus_error_is_set (&error))
575     {
576       /* This is a bit bogus because we pretend all errors
577        * are OOM; this is done because we know that in bus.c
578        * an OOM error disconnects the connection, which is
579        * the same thing we want on any other error.
580        */
581       dbus_error_free (&error);
582       goto out;
583     }
584   
585   if (!dbus_connection_set_watch_functions (connection,
586                                             add_connection_watch,
587                                             remove_connection_watch,
588                                             NULL,
589                                             connection,
590                                             NULL))
591     goto out;
592   
593   if (!dbus_connection_set_timeout_functions (connection,
594                                               add_connection_timeout,
595                                               remove_connection_timeout,
596                                               NULL,
597                                               connection, NULL))
598     goto out;
599   
600   dbus_connection_set_unix_user_function (connection,
601                                           allow_user_function,
602                                           NULL, NULL);
603
604   dbus_connection_set_dispatch_status_function (connection,
605                                                 dispatch_status_function,
606                                                 bus_context_get_loop (connections->context),
607                                                 NULL);
608
609   d->link_in_connection_list = _dbus_list_alloc_link (connection);
610   if (d->link_in_connection_list == NULL)
611     goto out;
612   
613   /* Setup the connection with the dispatcher */
614   if (!bus_dispatch_add_connection (connection))
615     goto out;
616
617   if (dbus_connection_get_dispatch_status (connection) != DBUS_DISPATCH_COMPLETE)
618     {
619       if (!_dbus_loop_queue_dispatch (bus_context_get_loop (connections->context), connection))
620         {
621           bus_dispatch_remove_connection (connection);
622           goto out;
623         }
624     }
625
626   _dbus_list_append_link (&connections->incomplete, d->link_in_connection_list);
627   connections->n_incomplete += 1;
628   
629   dbus_connection_ref (connection);
630
631   /* Note that we might disconnect ourselves here, but it only takes
632    * effect on return to the main loop. We call this to free up
633    * expired connections if possible, and to queue the timeout for our
634    * own expiration.
635    */
636   bus_connections_expire_incomplete (connections);
637   
638   /* And we might also disconnect ourselves here, but again it
639    * only takes effect on return to main loop.
640    */
641   if (connections->n_incomplete >
642       bus_context_get_max_incomplete_connections (connections->context))
643     {
644       _dbus_verbose ("Number of incomplete connections exceeds max, dropping oldest one\n");
645       
646       _dbus_assert (connections->incomplete != NULL);
647       /* Disconnect the oldest unauthenticated connection.  FIXME
648        * would it be more secure to drop a *random* connection?  This
649        * algorithm seems to mean that if someone can create new
650        * connections quickly enough, they can keep anyone else from
651        * completing authentication. But random may or may not really
652        * help with that, a more elaborate solution might be required.
653        */
654       dbus_connection_close (connections->incomplete->data);
655     }
656   
657   retval = TRUE;
658
659  out:
660   if (!retval)
661     {
662       if (d->selinux_id)
663         bus_selinux_id_unref (d->selinux_id);
664       d->selinux_id = NULL;
665       
666       if (!dbus_connection_set_watch_functions (connection,
667                                                 NULL, NULL, NULL,
668                                                 connection,
669                                                 NULL))
670         _dbus_assert_not_reached ("setting watch functions to NULL failed");
671       
672       if (!dbus_connection_set_timeout_functions (connection,
673                                                   NULL, NULL, NULL,
674                                                   connection,
675                                                   NULL))
676         _dbus_assert_not_reached ("setting timeout functions to NULL failed");
677
678       dbus_connection_set_unix_user_function (connection,
679                                               NULL, NULL, NULL);
680
681       dbus_connection_set_dispatch_status_function (connection,
682                                                     NULL, NULL, NULL);
683
684       if (d->link_in_connection_list != NULL)
685         {
686           _dbus_assert (d->link_in_connection_list->next == NULL);
687           _dbus_assert (d->link_in_connection_list->prev == NULL);
688           _dbus_list_free_link (d->link_in_connection_list);
689           d->link_in_connection_list = NULL;
690         }
691       
692       if (!dbus_connection_set_data (connection,
693                                      connection_data_slot,
694                                      NULL, NULL))
695         _dbus_assert_not_reached ("failed to set connection data to null");
696
697       /* "d" has now been freed */
698     }
699   
700   return retval;
701 }
702
703 void
704 bus_connections_expire_incomplete (BusConnections *connections)
705 {    
706   int next_interval;
707
708   next_interval = -1;
709   
710   if (connections->incomplete != NULL)
711     {
712       long tv_sec, tv_usec;
713       DBusList *link;
714       int auth_timeout;
715       
716       _dbus_get_current_time (&tv_sec, &tv_usec);
717       auth_timeout = bus_context_get_auth_timeout (connections->context);
718   
719       link = _dbus_list_get_first_link (&connections->incomplete);
720       while (link != NULL)
721         {
722           DBusList *next = _dbus_list_get_next_link (&connections->incomplete, link);
723           DBusConnection *connection;
724           BusConnectionData *d;
725           double elapsed;
726       
727           connection = link->data;
728       
729           d = BUS_CONNECTION_DATA (connection);
730       
731           _dbus_assert (d != NULL);
732       
733           elapsed = ELAPSED_MILLISECONDS_SINCE (d->connection_tv_sec,
734                                                 d->connection_tv_usec,
735                                                 tv_sec, tv_usec);
736
737           if (elapsed >= (double) auth_timeout)
738             {
739               _dbus_verbose ("Timing out authentication for connection %p\n", connection);
740               dbus_connection_close (connection);
741             }
742           else
743             {
744               /* We can end the loop, since the connections are in oldest-first order */
745               next_interval = ((double)auth_timeout) - elapsed;
746               _dbus_verbose ("Connection %p authentication expires in %d milliseconds\n",
747                              connection, next_interval);
748           
749               break;
750             }
751       
752           link = next;
753         }
754     }
755
756   bus_expire_timeout_set_interval (connections->expire_timeout,
757                                    next_interval);
758 }
759
760 static dbus_bool_t
761 expire_incomplete_timeout (void *data)
762 {
763   BusConnections *connections = data;
764
765   _dbus_verbose ("Running %s\n", _DBUS_FUNCTION_NAME);
766   
767   /* note that this may remove the timeout */
768   bus_connections_expire_incomplete (connections);
769
770   return TRUE;
771 }
772
773 dbus_bool_t
774 bus_connection_get_groups  (DBusConnection   *connection,
775                             unsigned long   **groups,
776                             int              *n_groups,
777                             DBusError        *error)
778 {
779   BusConnectionData *d;
780   unsigned long uid;
781   DBusUserDatabase *user_database;
782   
783   d = BUS_CONNECTION_DATA (connection);
784
785   _dbus_assert (d != NULL);
786
787   user_database = bus_context_get_user_database (d->connections->context);
788   
789   *groups = NULL;
790   *n_groups = 0;
791
792   if (dbus_connection_get_unix_user (connection, &uid))
793     {
794       if (!_dbus_user_database_get_groups (user_database,
795                                            uid, groups, n_groups,
796                                            error))
797         {
798           _DBUS_ASSERT_ERROR_IS_SET (error);
799           _dbus_verbose ("Did not get any groups for UID %lu\n",
800                          uid);
801           return FALSE;
802         }
803       else
804         {
805           _dbus_verbose ("Got %d groups for UID %lu\n",
806                          *n_groups, uid);
807           return TRUE;
808         }
809     }
810   else
811     return TRUE; /* successfully got 0 groups */
812 }
813
814 dbus_bool_t
815 bus_connection_is_in_group (DBusConnection *connection,
816                             unsigned long   gid)
817 {
818   int i;
819   unsigned long *group_ids;
820   int n_group_ids;
821
822   if (!bus_connection_get_groups (connection, &group_ids, &n_group_ids,
823                                   NULL))
824     return FALSE;
825
826   i = 0;
827   while (i < n_group_ids)
828     {
829       if (group_ids[i] == gid)
830         {
831           dbus_free (group_ids);
832           return TRUE;
833         }
834       ++i;
835     }
836
837   dbus_free (group_ids);
838   return FALSE;
839 }
840
841 BusClientPolicy*
842 bus_connection_get_policy (DBusConnection *connection)
843 {
844   BusConnectionData *d;
845     
846   d = BUS_CONNECTION_DATA (connection);
847
848   _dbus_assert (d != NULL);
849   _dbus_assert (d->policy != NULL);
850   
851   return d->policy;
852 }
853
854 static dbus_bool_t
855 foreach_active (BusConnections               *connections,
856                 BusConnectionForeachFunction  function,
857                 void                         *data)
858 {
859   DBusList *link;
860   
861   link = _dbus_list_get_first_link (&connections->completed);
862   while (link != NULL)
863     {
864       DBusConnection *connection = link->data;
865       DBusList *next = _dbus_list_get_next_link (&connections->completed, link);
866
867       if (!(* function) (connection, data))
868         return FALSE;
869       
870       link = next;
871     }
872
873   return TRUE;
874 }
875
876 static dbus_bool_t
877 foreach_inactive (BusConnections               *connections,
878                   BusConnectionForeachFunction  function,
879                   void                         *data)
880 {
881   DBusList *link;
882   
883   link = _dbus_list_get_first_link (&connections->incomplete);
884   while (link != NULL)
885     {
886       DBusConnection *connection = link->data;
887       DBusList *next = _dbus_list_get_next_link (&connections->incomplete, link);
888
889       if (!(* function) (connection, data))
890         return FALSE;
891       
892       link = next;
893     }
894
895   return TRUE;
896 }
897
898 /**
899  * Calls function on each active connection; if the function returns
900  * #FALSE, stops iterating. Active connections are authenticated
901  * and have sent a Hello message.
902  *
903  * @param connections the connections object
904  * @param function the function
905  * @param data data to pass to it as a second arg
906  */
907 void
908 bus_connections_foreach_active (BusConnections               *connections,
909                                 BusConnectionForeachFunction  function,
910                                 void                         *data)
911 {
912   foreach_active (connections, function, data);
913 }
914
915 /**
916  * Calls function on each connection; if the function returns
917  * #FALSE, stops iterating.
918  *
919  * @param connections the connections object
920  * @param function the function
921  * @param data data to pass to it as a second arg
922  */
923 void
924 bus_connections_foreach (BusConnections               *connections,
925                          BusConnectionForeachFunction  function,
926                          void                         *data)
927 {
928   if (!foreach_active (connections, function, data))
929     return;
930
931   foreach_inactive (connections, function, data);
932 }
933
934 BusContext*
935 bus_connections_get_context (BusConnections *connections)
936 {
937   return connections->context;
938 }
939
940 /*
941  * This is used to avoid covering the same connection twice when
942  * traversing connections. Note that it assumes we will
943  * bus_connection_mark_stamp() each connection at least once per
944  * INT_MAX increments of the global stamp, or wraparound would break
945  * things.
946  */
947 void
948 bus_connections_increment_stamp (BusConnections *connections)
949 {
950   connections->stamp += 1;
951 }
952
953 /* Mark connection with current stamp, return TRUE if it
954  * didn't already have that stamp
955  */
956 dbus_bool_t
957 bus_connection_mark_stamp (DBusConnection *connection)
958 {
959   BusConnectionData *d;
960   
961   d = BUS_CONNECTION_DATA (connection);
962   
963   _dbus_assert (d != NULL);
964
965   if (d->stamp == d->connections->stamp)
966     return FALSE;
967   else
968     {
969       d->stamp = d->connections->stamp;
970       return TRUE;
971     }
972 }
973
974 BusContext*
975 bus_connection_get_context (DBusConnection *connection)
976 {
977   BusConnectionData *d;
978
979   d = BUS_CONNECTION_DATA (connection);
980
981   _dbus_assert (d != NULL);
982
983   return d->connections->context;
984 }
985
986 BusConnections*
987 bus_connection_get_connections (DBusConnection *connection)
988 {
989   BusConnectionData *d;
990     
991   d = BUS_CONNECTION_DATA (connection);
992
993   _dbus_assert (d != NULL);
994
995   return d->connections;
996 }
997
998 BusRegistry*
999 bus_connection_get_registry (DBusConnection *connection)
1000 {
1001   BusConnectionData *d;
1002
1003   d = BUS_CONNECTION_DATA (connection);
1004
1005   _dbus_assert (d != NULL);
1006
1007   return bus_context_get_registry (d->connections->context);
1008 }
1009
1010 BusActivation*
1011 bus_connection_get_activation (DBusConnection *connection)
1012 {
1013   BusConnectionData *d;
1014
1015   d = BUS_CONNECTION_DATA (connection);
1016
1017   _dbus_assert (d != NULL);
1018
1019   return bus_context_get_activation (d->connections->context);
1020 }
1021
1022 BusMatchmaker*
1023 bus_connection_get_matchmaker (DBusConnection *connection)
1024 {
1025   BusConnectionData *d;
1026
1027   d = BUS_CONNECTION_DATA (connection);
1028
1029   _dbus_assert (d != NULL);
1030
1031   return bus_context_get_matchmaker (d->connections->context);
1032 }
1033
1034 BusSELinuxID*
1035 bus_connection_get_selinux_id (DBusConnection *connection)
1036 {
1037   BusConnectionData *d;
1038
1039   d = BUS_CONNECTION_DATA (connection);
1040
1041   _dbus_assert (d != NULL);
1042
1043   return d->selinux_id;
1044 }
1045
1046 /**
1047  * Checks whether the connection is registered with the message bus.
1048  *
1049  * @param connection the connection
1050  * @returns #TRUE if we're an active message bus participant
1051  */
1052 dbus_bool_t
1053 bus_connection_is_active (DBusConnection *connection)
1054 {
1055   BusConnectionData *d;
1056
1057   d = BUS_CONNECTION_DATA (connection);
1058   
1059   return d != NULL && d->name != NULL;
1060 }
1061
1062 dbus_bool_t
1063 bus_connection_preallocate_oom_error (DBusConnection *connection)
1064 {
1065   DBusMessage *message;
1066   DBusPreallocatedSend *preallocated;
1067   BusConnectionData *d;
1068
1069   d = BUS_CONNECTION_DATA (connection);  
1070
1071   _dbus_assert (d != NULL);
1072
1073   if (d->oom_preallocated != NULL)
1074     return TRUE;
1075   
1076   preallocated = dbus_connection_preallocate_send (connection);
1077   if (preallocated == NULL)
1078     return FALSE;
1079
1080   message = dbus_message_new (DBUS_MESSAGE_TYPE_ERROR);
1081
1082   if (message == NULL)
1083     {
1084       dbus_connection_free_preallocated_send (connection, preallocated);
1085       return FALSE;
1086     }
1087
1088   /* d->name may be NULL, but that is OK */
1089   if (!dbus_message_set_error_name (message, DBUS_ERROR_NO_MEMORY) ||
1090       !dbus_message_set_destination (message, d->name) ||
1091       !dbus_message_set_sender (message,
1092                                 DBUS_SERVICE_DBUS))
1093     {
1094       dbus_connection_free_preallocated_send (connection, preallocated);
1095       dbus_message_unref (message);
1096       return FALSE;
1097     }
1098   
1099   /* set reply serial to placeholder value just so space is already allocated
1100    * for it.
1101    */
1102   if (!dbus_message_set_reply_serial (message, 14))
1103     {
1104       dbus_connection_free_preallocated_send (connection, preallocated);
1105       dbus_message_unref (message);
1106       return FALSE;
1107     }
1108
1109   d->oom_message = message;
1110   d->oom_preallocated = preallocated;
1111   
1112   return TRUE;
1113 }
1114
1115 void
1116 bus_connection_send_oom_error (DBusConnection *connection,
1117                                DBusMessage    *in_reply_to)
1118 {
1119   BusConnectionData *d;
1120
1121   d = BUS_CONNECTION_DATA (connection);  
1122
1123   _dbus_assert (d != NULL);  
1124   _dbus_assert (d->oom_message != NULL);
1125
1126   /* should always succeed since we set it to a placeholder earlier */
1127   if (!dbus_message_set_reply_serial (d->oom_message,
1128                                       dbus_message_get_serial (in_reply_to)))
1129     _dbus_assert_not_reached ("Failed to set reply serial for preallocated oom message");
1130
1131   _dbus_assert (dbus_message_get_sender (d->oom_message) != NULL);
1132   
1133   dbus_connection_send_preallocated (connection, d->oom_preallocated,
1134                                      d->oom_message, NULL);
1135
1136   dbus_message_unref (d->oom_message);
1137   d->oom_message = NULL;
1138   d->oom_preallocated = NULL;
1139 }
1140
1141 void
1142 bus_connection_add_match_rule_link (DBusConnection *connection,
1143                                     DBusList       *link)
1144 {
1145   BusConnectionData *d;
1146
1147   d = BUS_CONNECTION_DATA (connection);
1148   _dbus_assert (d != NULL);
1149
1150   _dbus_list_append_link (&d->match_rules, link);
1151
1152   d->n_match_rules += 1;
1153 }
1154
1155 dbus_bool_t
1156 bus_connection_add_match_rule (DBusConnection *connection,
1157                                BusMatchRule   *rule)
1158 {
1159     DBusList *link;
1160
1161   link = _dbus_list_alloc_link (rule);
1162
1163   if (link == NULL)
1164     return FALSE;
1165
1166   bus_connection_add_match_rule_link (connection, link);
1167
1168   return TRUE;
1169 }
1170
1171 void
1172 bus_connection_remove_match_rule (DBusConnection *connection,
1173                                   BusMatchRule   *rule)
1174 {
1175   BusConnectionData *d;
1176
1177   d = BUS_CONNECTION_DATA (connection);
1178   _dbus_assert (d != NULL);
1179
1180   _dbus_list_remove_last (&d->match_rules, rule);
1181
1182   d->n_match_rules -= 1;
1183   _dbus_assert (d->n_match_rules >= 0);
1184 }
1185
1186 int
1187 bus_connection_get_n_match_rules (DBusConnection *connection)
1188 {
1189   BusConnectionData *d;
1190
1191   d = BUS_CONNECTION_DATA (connection);
1192   _dbus_assert (d != NULL);
1193   
1194   return d->n_match_rules;
1195 }
1196
1197 void
1198 bus_connection_add_owned_service_link (DBusConnection *connection,
1199                                        DBusList       *link)
1200 {
1201   BusConnectionData *d;
1202
1203   d = BUS_CONNECTION_DATA (connection);
1204   _dbus_assert (d != NULL);
1205
1206   _dbus_list_append_link (&d->services_owned, link);
1207
1208   d->n_services_owned += 1;
1209 }
1210
1211 dbus_bool_t
1212 bus_connection_add_owned_service (DBusConnection *connection,
1213                                   BusService     *service)
1214 {
1215   DBusList *link;
1216
1217   link = _dbus_list_alloc_link (service);
1218
1219   if (link == NULL)
1220     return FALSE;
1221
1222   bus_connection_add_owned_service_link (connection, link);
1223
1224   return TRUE;
1225 }
1226
1227 void
1228 bus_connection_remove_owned_service (DBusConnection *connection,
1229                                      BusService     *service)
1230 {
1231   BusConnectionData *d;
1232
1233   d = BUS_CONNECTION_DATA (connection);
1234   _dbus_assert (d != NULL);
1235
1236   _dbus_list_remove_last (&d->services_owned, service);
1237
1238   d->n_services_owned -= 1;
1239   _dbus_assert (d->n_services_owned >= 0);
1240 }
1241
1242 int
1243 bus_connection_get_n_services_owned (DBusConnection *connection)
1244 {
1245   BusConnectionData *d;
1246
1247   d = BUS_CONNECTION_DATA (connection);
1248   _dbus_assert (d != NULL);
1249   
1250   return d->n_services_owned;
1251 }
1252
1253 dbus_bool_t
1254 bus_connection_complete (DBusConnection   *connection,
1255                          const DBusString *name,
1256                          DBusError        *error)
1257 {
1258   BusConnectionData *d;
1259   unsigned long uid;
1260   
1261   d = BUS_CONNECTION_DATA (connection);
1262   _dbus_assert (d != NULL);
1263   _dbus_assert (d->name == NULL);
1264   _dbus_assert (d->policy == NULL);
1265
1266   _dbus_assert (!bus_connection_is_active (connection));
1267   
1268   if (!_dbus_string_copy_data (name, &d->name))
1269     {
1270       BUS_SET_OOM (error);
1271       return FALSE;
1272     }
1273
1274   _dbus_assert (d->name != NULL);
1275   
1276   _dbus_verbose ("Name %s assigned to %p\n", d->name, connection);
1277
1278   d->policy = bus_context_create_client_policy (d->connections->context,
1279                                                 connection,
1280                                                 error);
1281
1282   /* we may have a NULL policy on OOM or error getting list of
1283    * groups for a user. In the latter case we don't handle it so
1284    * well currently, as it will just keep failing over and over.
1285    */
1286
1287   if (d->policy == NULL)
1288     {
1289       _dbus_verbose ("Failed to create security policy for connection %p\n",
1290                      connection);
1291       _DBUS_ASSERT_ERROR_IS_SET (error);
1292       dbus_free (d->name);
1293       d->name = NULL;
1294       return FALSE;
1295     }
1296   
1297   if (dbus_connection_get_unix_user (connection, &uid))
1298     {
1299       if (!adjust_connections_for_uid (d->connections,
1300                                        uid, 1))
1301         {
1302           BUS_SET_OOM (error);
1303           dbus_free (d->name);
1304           d->name = NULL;
1305           return FALSE;
1306         }
1307     }
1308   
1309   /* Now the connection is active, move it between lists */
1310   _dbus_list_unlink (&d->connections->incomplete,
1311                      d->link_in_connection_list);
1312   d->connections->n_incomplete -= 1;
1313   _dbus_list_append_link (&d->connections->completed,
1314                           d->link_in_connection_list);
1315   d->connections->n_completed += 1;
1316
1317   _dbus_assert (d->connections->n_incomplete >= 0);
1318   _dbus_assert (d->connections->n_completed > 0);
1319
1320   /* See if we can remove the timeout */
1321   bus_connections_expire_incomplete (d->connections);
1322
1323   _dbus_assert (bus_connection_is_active (connection));
1324   
1325   return TRUE;
1326 }
1327
1328 const char *
1329 bus_connection_get_name (DBusConnection *connection)
1330 {
1331   BusConnectionData *d;
1332   
1333   d = BUS_CONNECTION_DATA (connection);
1334   _dbus_assert (d != NULL);
1335   
1336   return d->name;
1337 }
1338
1339 /**
1340  * Check whether completing the passed-in connection would
1341  * exceed limits, and if so set error and return #FALSE
1342  */
1343 dbus_bool_t
1344 bus_connections_check_limits (BusConnections  *connections,
1345                               DBusConnection  *requesting_completion,
1346                               DBusError       *error)
1347 {
1348   BusConnectionData *d;
1349   unsigned long uid;
1350   
1351   d = BUS_CONNECTION_DATA (requesting_completion);
1352   _dbus_assert (d != NULL);
1353
1354   _dbus_assert (d->name == NULL);
1355
1356   if (connections->n_completed >=
1357       bus_context_get_max_completed_connections (connections->context))
1358     {
1359       dbus_set_error (error, DBUS_ERROR_LIMITS_EXCEEDED,
1360                       "The maximum number of active connections has been reached");
1361       return FALSE;
1362     }
1363   
1364   if (dbus_connection_get_unix_user (requesting_completion, &uid))
1365     {
1366       if (get_connections_for_uid (connections, uid) >=
1367           bus_context_get_max_connections_per_user (connections->context))
1368         {
1369           dbus_set_error (error, DBUS_ERROR_LIMITS_EXCEEDED,
1370                           "The maximum number of active connections for UID %lu has been reached",
1371                           uid);
1372           return FALSE;
1373         }
1374     }
1375   
1376   return TRUE;
1377 }
1378
1379 static void
1380 bus_pending_reply_free (BusPendingReply *pending)
1381 {
1382   _dbus_verbose ("Freeing pending reply %p, replier %p receiver %p serial %u\n",
1383                  pending,
1384                  pending->will_send_reply,
1385                  pending->will_get_reply,
1386                  pending->reply_serial);
1387
1388   dbus_free (pending);
1389 }
1390
1391 static dbus_bool_t
1392 bus_pending_reply_send_no_reply (BusConnections  *connections,
1393                                  BusTransaction  *transaction,
1394                                  BusPendingReply *pending)
1395 {
1396   DBusMessage *message;
1397   DBusMessageIter iter;
1398   dbus_bool_t retval;
1399   const char *errmsg;
1400
1401   retval = FALSE;
1402   
1403   message = dbus_message_new (DBUS_MESSAGE_TYPE_ERROR);
1404   if (message == NULL)
1405     return FALSE;
1406   
1407   dbus_message_set_no_reply (message, TRUE);
1408   
1409   if (!dbus_message_set_reply_serial (message,
1410                                       pending->reply_serial))
1411     goto out;
1412
1413   if (!dbus_message_set_error_name (message,
1414                                     DBUS_ERROR_NO_REPLY))
1415     goto out;
1416
1417   errmsg = "Message did not receive a reply (timeout by message bus)";
1418   dbus_message_iter_init_append (message, &iter);
1419   if (!dbus_message_iter_append_basic (&iter, DBUS_TYPE_STRING, &errmsg))
1420     goto out;
1421     
1422   if (!bus_transaction_send_from_driver (transaction, pending->will_get_reply,
1423                                          message))
1424     goto out;
1425
1426   retval = TRUE;
1427
1428  out:
1429   dbus_message_unref (message);
1430   return retval;
1431 }
1432
1433 static dbus_bool_t
1434 bus_pending_reply_expired (BusExpireList *list,
1435                            DBusList      *link,
1436                            void          *data)
1437 {
1438   BusPendingReply *pending = link->data;
1439   BusConnections *connections = data;
1440   BusTransaction *transaction;
1441   
1442   /* No reply is forthcoming. So nuke it if we can. If not,
1443    * leave it in the list to try expiring again later when we
1444    * get more memory.
1445    */
1446
1447   _dbus_verbose ("Expiring pending reply %p, replier %p receiver %p serial %u\n",
1448                  pending,
1449                  pending->will_send_reply,
1450                  pending->will_get_reply,
1451                  pending->reply_serial);
1452   
1453   transaction = bus_transaction_new (connections->context);
1454   if (transaction == NULL)
1455     return FALSE;
1456   
1457   if (!bus_pending_reply_send_no_reply (connections,
1458                                         transaction,
1459                                         pending))
1460     {
1461       bus_transaction_cancel_and_free (transaction);
1462       return FALSE;
1463     }
1464   
1465   _dbus_list_remove_link (&connections->pending_replies->items,
1466                           link);
1467   bus_pending_reply_free (pending);
1468   bus_transaction_execute_and_free (transaction);
1469
1470   return TRUE;
1471 }
1472
1473 static void
1474 bus_connection_drop_pending_replies (BusConnections  *connections,
1475                                      DBusConnection  *connection)
1476 {
1477   /* The DBusConnection is almost 100% finalized here, so you can't
1478    * do anything with it except check for pointer equality
1479    */
1480   DBusList *link;
1481
1482   _dbus_verbose ("Dropping pending replies that involve connection %p\n",
1483                  connection);
1484   
1485   link = _dbus_list_get_first_link (&connections->pending_replies->items);
1486   while (link != NULL)
1487     {
1488       DBusList *next;
1489       BusPendingReply *pending;
1490
1491       next = _dbus_list_get_next_link (&connections->pending_replies->items,
1492                                        link);
1493       pending = link->data;
1494
1495       if (pending->will_get_reply == connection)
1496         {
1497           /* We don't need to track this pending reply anymore */
1498
1499           _dbus_verbose ("Dropping pending reply %p, replier %p receiver %p serial %u\n",
1500                          pending,
1501                          pending->will_send_reply,
1502                          pending->will_get_reply,
1503                          pending->reply_serial);
1504           
1505           _dbus_list_remove_link (&connections->pending_replies->items,
1506                                   link);
1507           bus_pending_reply_free (pending);
1508         }
1509       else if (pending->will_send_reply == connection)
1510         {
1511           /* The reply isn't going to be sent, so set things
1512            * up so it will be expired right away
1513            */
1514           _dbus_verbose ("Will expire pending reply %p, replier %p receiver %p serial %u\n",
1515                          pending,
1516                          pending->will_send_reply,
1517                          pending->will_get_reply,
1518                          pending->reply_serial);
1519           
1520           pending->will_send_reply = NULL;
1521           pending->expire_item.added_tv_sec = 0;
1522           pending->expire_item.added_tv_usec = 0;
1523
1524           bus_expire_timeout_set_interval (connections->pending_replies->timeout,
1525                                            0);
1526         }
1527       
1528       link = next;
1529     }
1530 }
1531
1532
1533 typedef struct
1534 {
1535   BusPendingReply *pending;
1536   BusConnections  *connections;
1537 } CancelPendingReplyData;
1538
1539 static void
1540 cancel_pending_reply (void *data)
1541 {
1542   CancelPendingReplyData *d = data;
1543
1544   _dbus_verbose ("%s: d = %p\n", _DBUS_FUNCTION_NAME, d);
1545   
1546   if (!_dbus_list_remove (&d->connections->pending_replies->items,
1547                           d->pending))
1548     _dbus_assert_not_reached ("pending reply did not exist to be cancelled");
1549
1550   bus_pending_reply_free (d->pending); /* since it's been cancelled */
1551 }
1552
1553 static void
1554 cancel_pending_reply_data_free (void *data)
1555 {
1556   CancelPendingReplyData *d = data;
1557
1558   _dbus_verbose ("%s: d = %p\n", _DBUS_FUNCTION_NAME, d);
1559   
1560   /* d->pending should be either freed or still
1561    * in the list of pending replies (owned by someone
1562    * else)
1563    */
1564   
1565   dbus_free (d);
1566 }
1567
1568 /*
1569  * Record that a reply is allowed; return TRUE on success.
1570  */
1571 dbus_bool_t
1572 bus_connections_expect_reply (BusConnections  *connections,
1573                               BusTransaction  *transaction,
1574                               DBusConnection  *will_get_reply,
1575                               DBusConnection  *will_send_reply,
1576                               DBusMessage     *reply_to_this,
1577                               DBusError       *error)
1578 {
1579   BusPendingReply *pending;
1580   dbus_uint32_t reply_serial;
1581   DBusList *link;
1582   CancelPendingReplyData *cprd;
1583   int count;
1584
1585   _dbus_assert (will_get_reply != NULL);
1586   _dbus_assert (will_send_reply != NULL);
1587   _dbus_assert (reply_to_this != NULL);
1588   
1589   if (dbus_message_get_no_reply (reply_to_this))
1590     return TRUE; /* we won't allow a reply, since client doesn't care for one. */
1591   
1592   reply_serial = dbus_message_get_serial (reply_to_this);
1593
1594   link = _dbus_list_get_first_link (&connections->pending_replies->items);
1595   count = 0;
1596   while (link != NULL)
1597     {
1598       pending = link->data;
1599
1600       if (pending->reply_serial == reply_serial &&
1601           pending->will_get_reply == will_get_reply &&
1602           pending->will_send_reply == will_send_reply)
1603         {
1604           dbus_set_error (error, DBUS_ERROR_ACCESS_DENIED,
1605                           "Message has the same reply serial as a currently-outstanding existing method call");
1606           return FALSE;
1607         }
1608       
1609       link = _dbus_list_get_next_link (&connections->pending_replies->items,
1610                                        link);
1611       if (pending->will_get_reply == will_get_reply)
1612         ++count;
1613     }
1614   
1615   if (count >=
1616       bus_context_get_max_replies_per_connection (connections->context))
1617     {
1618       dbus_set_error (error, DBUS_ERROR_LIMITS_EXCEEDED,
1619                       "The maximum number of pending replies per connection has been reached");
1620       return FALSE;
1621     }
1622
1623   pending = dbus_new0 (BusPendingReply, 1);
1624   if (pending == NULL)
1625     {
1626       BUS_SET_OOM (error);
1627       return FALSE;
1628     }
1629
1630 #ifdef DBUS_ENABLE_VERBOSE_MODE
1631   /* so we can see a not-yet-added pending reply */
1632   pending->expire_item.added_tv_sec = 1;
1633   pending->expire_item.added_tv_usec = 1;
1634 #endif
1635
1636   pending->will_get_reply = will_get_reply;
1637   pending->will_send_reply = will_send_reply;
1638   pending->reply_serial = reply_serial;
1639   
1640   cprd = dbus_new0 (CancelPendingReplyData, 1);
1641   if (cprd == NULL)
1642     {
1643       BUS_SET_OOM (error);
1644       bus_pending_reply_free (pending);
1645       return FALSE;
1646     }
1647   
1648   if (!_dbus_list_prepend (&connections->pending_replies->items,
1649                            pending))
1650     {
1651       BUS_SET_OOM (error);
1652       dbus_free (cprd);
1653       bus_pending_reply_free (pending);
1654       return FALSE;
1655     }
1656
1657   if (!bus_transaction_add_cancel_hook (transaction,
1658                                         cancel_pending_reply,
1659                                         cprd,
1660                                         cancel_pending_reply_data_free))
1661     {
1662       BUS_SET_OOM (error);
1663       _dbus_list_remove (&connections->pending_replies->items, pending);
1664       dbus_free (cprd);
1665       bus_pending_reply_free (pending);
1666       return FALSE;
1667     }
1668                                         
1669   cprd->pending = pending;
1670   cprd->connections = connections;
1671   
1672   _dbus_get_current_time (&pending->expire_item.added_tv_sec,
1673                           &pending->expire_item.added_tv_usec);
1674
1675   _dbus_verbose ("Added pending reply %p, replier %p receiver %p serial %u\n",
1676                  pending,
1677                  pending->will_send_reply,
1678                  pending->will_get_reply,
1679                  pending->reply_serial);
1680   
1681   return TRUE;
1682 }
1683
1684 typedef struct
1685 {
1686   DBusList        *link;
1687   BusConnections  *connections;
1688 } CheckPendingReplyData;
1689
1690 static void
1691 cancel_check_pending_reply (void *data)
1692 {
1693   CheckPendingReplyData *d = data;
1694
1695   _dbus_verbose ("%s: d = %p\n", _DBUS_FUNCTION_NAME, d);
1696   
1697   _dbus_list_prepend_link (&d->connections->pending_replies->items,
1698                            d->link);
1699   d->link = NULL;
1700 }
1701
1702 static void
1703 check_pending_reply_data_free (void *data)
1704 {
1705   CheckPendingReplyData *d = data;
1706
1707   _dbus_verbose ("%s: d = %p\n", _DBUS_FUNCTION_NAME, d);
1708   
1709   if (d->link != NULL)
1710     {
1711       BusPendingReply *pending = d->link->data;
1712       
1713       _dbus_assert (_dbus_list_find_last (&d->connections->pending_replies->items,
1714                                           pending) == NULL);
1715       
1716       bus_pending_reply_free (pending);
1717       _dbus_list_free_link (d->link);
1718     }
1719   
1720   dbus_free (d);
1721 }
1722
1723 /*
1724  * Check whether a reply is allowed, remove BusPendingReply
1725  * if so, return TRUE if so.
1726  */
1727 dbus_bool_t
1728 bus_connections_check_reply (BusConnections *connections,
1729                              BusTransaction *transaction,
1730                              DBusConnection *sending_reply,
1731                              DBusConnection *receiving_reply,
1732                              DBusMessage    *reply,
1733                              DBusError      *error)
1734 {
1735   CheckPendingReplyData *cprd;
1736   DBusList *link;
1737   dbus_uint32_t reply_serial;
1738   
1739   _dbus_assert (sending_reply != NULL);
1740   _dbus_assert (receiving_reply != NULL);
1741
1742   reply_serial = dbus_message_get_reply_serial (reply);
1743
1744   link = _dbus_list_get_first_link (&connections->pending_replies->items);
1745   while (link != NULL)
1746     {
1747       BusPendingReply *pending = link->data;
1748
1749       if (pending->reply_serial == reply_serial &&
1750           pending->will_get_reply == receiving_reply &&
1751           pending->will_send_reply == sending_reply)
1752         {
1753           _dbus_verbose ("Found pending reply with serial %u\n", reply_serial);
1754           break;
1755         }
1756       
1757       link = _dbus_list_get_next_link (&connections->pending_replies->items,
1758                                        link);
1759     }
1760
1761   if (link == NULL)
1762     {
1763       _dbus_verbose ("No pending reply expected\n");
1764
1765       return FALSE;
1766     }
1767
1768   cprd = dbus_new0 (CheckPendingReplyData, 1);
1769   if (cprd == NULL)
1770     {
1771       BUS_SET_OOM (error);
1772       return FALSE;
1773     }
1774   
1775   if (!bus_transaction_add_cancel_hook (transaction,
1776                                         cancel_check_pending_reply,
1777                                         cprd,
1778                                         check_pending_reply_data_free))
1779     {
1780       BUS_SET_OOM (error);
1781       dbus_free (cprd);
1782       return FALSE;
1783     }
1784
1785   cprd->link = link;
1786   cprd->connections = connections;
1787   
1788   _dbus_list_unlink (&connections->pending_replies->items,
1789                      link);
1790   
1791   _dbus_assert (_dbus_list_find_last (&connections->pending_replies->items,
1792                                       link->data) == NULL);
1793
1794   return TRUE;
1795 }
1796
1797 /*
1798  * Transactions
1799  *
1800  * Note that this is fairly fragile; in particular, don't try to use
1801  * one transaction across any main loop iterations.
1802  */
1803
1804 typedef struct
1805 {
1806   BusTransaction *transaction;
1807   DBusMessage    *message;
1808   DBusPreallocatedSend *preallocated;
1809 } MessageToSend;
1810
1811 typedef struct
1812 {
1813   BusTransactionCancelFunction cancel_function;
1814   DBusFreeFunction free_data_function;
1815   void *data;
1816 } CancelHook;
1817
1818 struct BusTransaction
1819 {
1820   DBusList *connections;
1821   BusContext *context;
1822   DBusList *cancel_hooks;
1823 };
1824
1825 static void
1826 message_to_send_free (DBusConnection *connection,
1827                       MessageToSend  *to_send)
1828 {
1829   if (to_send->message)
1830     dbus_message_unref (to_send->message);
1831
1832   if (to_send->preallocated)
1833     dbus_connection_free_preallocated_send (connection, to_send->preallocated);
1834
1835   dbus_free (to_send);
1836 }
1837
1838 static void
1839 cancel_hook_cancel (void *element,
1840                     void *data)
1841 {
1842   CancelHook *ch = element;
1843
1844   _dbus_verbose ("Running transaction cancel hook\n");
1845   
1846   if (ch->cancel_function)
1847     (* ch->cancel_function) (ch->data);  
1848 }
1849
1850 static void
1851 cancel_hook_free (void *element,
1852                   void *data)
1853 {
1854   CancelHook *ch = element;
1855
1856   if (ch->free_data_function)
1857     (* ch->free_data_function) (ch->data);
1858
1859   dbus_free (ch);
1860 }
1861
1862 static void
1863 free_cancel_hooks (BusTransaction *transaction)
1864 {
1865   _dbus_list_foreach (&transaction->cancel_hooks,
1866                       cancel_hook_free, NULL);
1867   
1868   _dbus_list_clear (&transaction->cancel_hooks);
1869 }
1870
1871 BusTransaction*
1872 bus_transaction_new (BusContext *context)
1873 {
1874   BusTransaction *transaction;
1875
1876   transaction = dbus_new0 (BusTransaction, 1);
1877   if (transaction == NULL)
1878     return NULL;
1879
1880   transaction->context = context;
1881   
1882   return transaction;
1883 }
1884
1885 BusContext*
1886 bus_transaction_get_context (BusTransaction  *transaction)
1887 {
1888   return transaction->context;
1889 }
1890
1891 BusConnections*
1892 bus_transaction_get_connections (BusTransaction  *transaction)
1893 {
1894   return bus_context_get_connections (transaction->context);
1895 }
1896
1897 dbus_bool_t
1898 bus_transaction_send_from_driver (BusTransaction *transaction,
1899                                   DBusConnection *connection,
1900                                   DBusMessage    *message)
1901 {
1902   /* We have to set the sender to the driver, and have
1903    * to check security policy since it was not done in
1904    * dispatch.c
1905    */
1906   _dbus_verbose ("Sending %s %s %s from driver\n",
1907                  dbus_message_get_interface (message) ?
1908                  dbus_message_get_interface (message) : "(no interface)",
1909                  dbus_message_get_member (message) ?
1910                  dbus_message_get_member (message) : "(no member)",
1911                  dbus_message_get_error_name (message) ?
1912                  dbus_message_get_error_name (message) : "(no error name)");
1913                  
1914   if (!dbus_message_set_sender (message, DBUS_SERVICE_DBUS))
1915     return FALSE;
1916
1917   if (bus_connection_is_active (connection))
1918     {
1919       if (!dbus_message_set_destination (message,
1920                                          bus_connection_get_name (connection)))
1921         return FALSE;
1922     }
1923   
1924   /* bus driver never wants a reply */
1925   dbus_message_set_no_reply (message, TRUE);
1926   
1927   /* If security policy doesn't allow the message, we silently
1928    * eat it; the driver doesn't care about getting a reply.
1929    */
1930   if (!bus_context_check_security_policy (bus_transaction_get_context (transaction),
1931                                           transaction,
1932                                           NULL, connection, connection, message, NULL))
1933     return TRUE;
1934
1935   return bus_transaction_send (transaction, connection, message);
1936 }
1937
1938 dbus_bool_t
1939 bus_transaction_send (BusTransaction *transaction,
1940                       DBusConnection *connection,
1941                       DBusMessage    *message)
1942 {
1943   MessageToSend *to_send;
1944   BusConnectionData *d;
1945   DBusList *link;
1946
1947   _dbus_verbose ("  trying to add %s interface=%s member=%s error=%s to transaction%s\n",
1948                  dbus_message_get_type (message) == DBUS_MESSAGE_TYPE_ERROR ? "error" :
1949                  dbus_message_get_reply_serial (message) != 0 ? "reply" :
1950                  "message",
1951                  dbus_message_get_interface (message) ?
1952                  dbus_message_get_interface (message) : "(unset)",
1953                  dbus_message_get_member (message) ?
1954                  dbus_message_get_member (message) : "(unset)",
1955                  dbus_message_get_error_name (message) ?
1956                  dbus_message_get_error_name (message) : "(unset)",
1957                  dbus_connection_get_is_connected (connection) ?
1958                  "" : " (disconnected)");
1959
1960   _dbus_assert (dbus_message_get_sender (message) != NULL);
1961   
1962   if (!dbus_connection_get_is_connected (connection))
1963     return TRUE; /* silently ignore disconnected connections */
1964   
1965   d = BUS_CONNECTION_DATA (connection);
1966   _dbus_assert (d != NULL);
1967   
1968   to_send = dbus_new (MessageToSend, 1);
1969   if (to_send == NULL)
1970     {
1971       return FALSE;
1972     }
1973
1974   to_send->preallocated = dbus_connection_preallocate_send (connection);
1975   if (to_send->preallocated == NULL)
1976     {
1977       dbus_free (to_send);
1978       return FALSE;
1979     }  
1980   
1981   dbus_message_ref (message);
1982   to_send->message = message;
1983   to_send->transaction = transaction;
1984
1985   _dbus_verbose ("about to prepend message\n");
1986   
1987   if (!_dbus_list_prepend (&d->transaction_messages, to_send))
1988     {
1989       message_to_send_free (connection, to_send);
1990       return FALSE;
1991     }
1992
1993   _dbus_verbose ("prepended message\n");
1994   
1995   /* See if we already had this connection in the list
1996    * for this transaction. If we have a pending message,
1997    * then we should already be in transaction->connections
1998    */
1999   link = _dbus_list_get_first_link (&d->transaction_messages);
2000   _dbus_assert (link->data == to_send);
2001   link = _dbus_list_get_next_link (&d->transaction_messages, link);
2002   while (link != NULL)
2003     {
2004       MessageToSend *m = link->data;
2005       DBusList *next = _dbus_list_get_next_link (&d->transaction_messages, link);
2006       
2007       if (m->transaction == transaction)
2008         break;
2009         
2010       link = next;
2011     }
2012
2013   if (link == NULL)
2014     {
2015       if (!_dbus_list_prepend (&transaction->connections, connection))
2016         {
2017           _dbus_list_remove (&d->transaction_messages, to_send);
2018           message_to_send_free (connection, to_send);
2019           return FALSE;
2020         }
2021     }
2022
2023   return TRUE;
2024 }
2025
2026 static void
2027 connection_cancel_transaction (DBusConnection *connection,
2028                                BusTransaction *transaction)
2029 {
2030   DBusList *link;
2031   BusConnectionData *d;
2032   
2033   d = BUS_CONNECTION_DATA (connection);
2034   _dbus_assert (d != NULL);
2035   
2036   link = _dbus_list_get_first_link (&d->transaction_messages);
2037   while (link != NULL)
2038     {
2039       MessageToSend *m = link->data;
2040       DBusList *next = _dbus_list_get_next_link (&d->transaction_messages, link);
2041       
2042       if (m->transaction == transaction)
2043         {
2044           _dbus_list_remove_link (&d->transaction_messages,
2045                                   link);
2046           
2047           message_to_send_free (connection, m);
2048         }
2049         
2050       link = next;
2051     }
2052 }
2053
2054 void
2055 bus_transaction_cancel_and_free (BusTransaction *transaction)
2056 {
2057   DBusConnection *connection;
2058
2059   _dbus_verbose ("TRANSACTION: cancelled\n");
2060   
2061   while ((connection = _dbus_list_pop_first (&transaction->connections)))
2062     connection_cancel_transaction (connection, transaction);
2063
2064   _dbus_assert (transaction->connections == NULL);
2065
2066   _dbus_list_foreach (&transaction->cancel_hooks,
2067                       cancel_hook_cancel, NULL);
2068
2069   free_cancel_hooks (transaction);
2070   
2071   dbus_free (transaction);
2072 }
2073
2074 static void
2075 connection_execute_transaction (DBusConnection *connection,
2076                                 BusTransaction *transaction)
2077 {
2078   DBusList *link;
2079   BusConnectionData *d;
2080   
2081   d = BUS_CONNECTION_DATA (connection);
2082   _dbus_assert (d != NULL);
2083
2084   /* Send the queue in order (FIFO) */
2085   link = _dbus_list_get_last_link (&d->transaction_messages);
2086   while (link != NULL)
2087     {
2088       MessageToSend *m = link->data;
2089       DBusList *prev = _dbus_list_get_prev_link (&d->transaction_messages, link);
2090       
2091       if (m->transaction == transaction)
2092         {
2093           _dbus_list_remove_link (&d->transaction_messages,
2094                                   link);
2095
2096           _dbus_assert (dbus_message_get_sender (m->message) != NULL);
2097           
2098           dbus_connection_send_preallocated (connection,
2099                                              m->preallocated,
2100                                              m->message,
2101                                              NULL);
2102
2103           m->preallocated = NULL; /* so we don't double-free it */
2104           
2105           message_to_send_free (connection, m);
2106         }
2107         
2108       link = prev;
2109     }
2110 }
2111
2112 void
2113 bus_transaction_execute_and_free (BusTransaction *transaction)
2114 {
2115   /* For each connection in transaction->connections
2116    * send the messages
2117    */
2118   DBusConnection *connection;
2119
2120   _dbus_verbose ("TRANSACTION: executing\n");
2121   
2122   while ((connection = _dbus_list_pop_first (&transaction->connections)))
2123     connection_execute_transaction (connection, transaction);
2124
2125   _dbus_assert (transaction->connections == NULL);
2126
2127   free_cancel_hooks (transaction);
2128   
2129   dbus_free (transaction);
2130 }
2131
2132 static void
2133 bus_connection_remove_transactions (DBusConnection *connection)
2134 {
2135   MessageToSend *to_send;
2136   BusConnectionData *d;
2137   
2138   d = BUS_CONNECTION_DATA (connection);
2139   _dbus_assert (d != NULL);
2140   
2141   while ((to_send = _dbus_list_get_first (&d->transaction_messages)))
2142     {
2143       /* only has an effect for the first MessageToSend listing this transaction */
2144       _dbus_list_remove (&to_send->transaction->connections,
2145                          connection);
2146
2147       _dbus_list_remove (&d->transaction_messages, to_send);
2148       message_to_send_free (connection, to_send);
2149     }
2150 }
2151
2152 /**
2153  * Converts the DBusError to a message reply
2154  */
2155 dbus_bool_t
2156 bus_transaction_send_error_reply (BusTransaction  *transaction,
2157                                   DBusConnection  *connection,
2158                                   const DBusError *error,
2159                                   DBusMessage     *in_reply_to)
2160 {
2161   DBusMessage *reply;
2162   
2163   _dbus_assert (error != NULL);
2164   _DBUS_ASSERT_ERROR_IS_SET (error);
2165   
2166   _dbus_verbose ("Sending error reply %s \"%s\"\n",
2167                  error->name, error->message);
2168
2169   reply = dbus_message_new_error (in_reply_to,
2170                                   error->name,
2171                                   error->message);
2172   if (reply == NULL)
2173     return FALSE;
2174
2175   if (!bus_transaction_send_from_driver (transaction, connection, reply))
2176     {
2177       dbus_message_unref (reply);
2178       return FALSE;
2179     }
2180
2181   dbus_message_unref (reply);
2182   
2183   return TRUE;
2184 }
2185
2186 dbus_bool_t
2187 bus_transaction_add_cancel_hook (BusTransaction               *transaction,
2188                                  BusTransactionCancelFunction  cancel_function,
2189                                  void                         *data,
2190                                  DBusFreeFunction              free_data_function)
2191 {
2192   CancelHook *ch;
2193
2194   ch = dbus_new (CancelHook, 1);
2195   if (ch == NULL)
2196     return FALSE;
2197
2198   _dbus_verbose ("     adding cancel hook function = %p data = %p\n",
2199                  cancel_function, data);
2200   
2201   ch->cancel_function = cancel_function;
2202   ch->data = data;
2203   ch->free_data_function = free_data_function;
2204
2205   /* It's important that the hooks get run in reverse order that they
2206    * were added
2207    */
2208   if (!_dbus_list_prepend (&transaction->cancel_hooks, ch))
2209     {
2210       dbus_free (ch);
2211       return FALSE;
2212     }
2213
2214   return TRUE;
2215 }