1: #include "test-utils.h"
    2: 
    3: typedef struct
    4: {
    5:   DBusLoop *loop;
    6:   DBusConnection *connection;
    7: 
    8: } CData;
    9: 
   10: static dbus_bool_t
   11: connection_watch_callback (DBusWatch     *watch,
   12:                            unsigned int   condition,
   13:                            void          *data)
   14: {
   15:   return dbus_watch_handle (watch, condition);
   16: }
   17: 
   18: static dbus_bool_t
   19: add_watch (DBusWatch *watch,
   20:            void      *data)
   21: {
   22:   CData *cd = data;
   23: 
   24:   return _dbus_loop_add_watch (cd->loop,
   25:                                watch,
   26:                                connection_watch_callback,
   27:                                cd, NULL);
   28: }
   29: 
   30: static void
   31: remove_watch (DBusWatch *watch,
   32:               void      *data)
   33: {
   34:   CData *cd = data;
   35:   
   36:   _dbus_loop_remove_watch (cd->loop,
   37:                            watch, connection_watch_callback, cd);  
   38: }
   39: 
   40: static void
   41: connection_timeout_callback (DBusTimeout   *timeout,
   42:                              void          *data)
   43: {
   44:   /* Can return FALSE on OOM but we just let it fire again later */
   45:   dbus_timeout_handle (timeout);
   46: }
   47: 
   48: static dbus_bool_t
   49: add_timeout (DBusTimeout *timeout,
   50:              void        *data)
   51: {
   52:   CData *cd = data;
   53: 
   54:   return _dbus_loop_add_timeout (cd->loop,
   55:                                  timeout, connection_timeout_callback, cd, NULL);
   56: }
   57: 
   58: static void
   59: remove_timeout (DBusTimeout *timeout,
   60:                 void        *data)
   61: {
   62:   CData *cd = data;
   63: 
   64:   _dbus_loop_remove_timeout (cd->loop,
   65:                              timeout, connection_timeout_callback, cd);
   66: }
   67: 
   68: static void
   69: dispatch_status_function (DBusConnection    *connection,
   70:                           DBusDispatchStatus new_status,
   71:                           void              *data)
   72: {
   73:   DBusLoop *loop = data;
   74:   
   75:   if (new_status != DBUS_DISPATCH_COMPLETE)
   76:     {
   77:       while (!_dbus_loop_queue_dispatch (loop, connection))
   78:         _dbus_wait_for_memory ();
   79:     }
   80: }
   81: 
   82: static void
   83: cdata_free (void *data)
   84: {
   85:   CData *cd = data;
   86: 
   87:   dbus_connection_unref (cd->connection);
   88:   _dbus_loop_unref (cd->loop);
   89:   
   90:   dbus_free (cd);
   91: }
   92: 
   93: static CData*
   94: cdata_new (DBusLoop       *loop,
   95:            DBusConnection *connection)
   96: {
   97:   CData *cd;
   98: 
   99:   cd = dbus_new0 (CData, 1);
  100:   if (cd == NULL)
  101:     return NULL;
  102: 
  103:   cd->loop = loop;
  104:   cd->connection = connection;
  105: 
  106:   dbus_connection_ref (cd->connection);
  107:   _dbus_loop_ref (cd->loop);
  108: 
  109:   return cd;
  110: }
  111: 
  112: dbus_bool_t
  113: test_connection_setup (DBusLoop       *loop,
  114:                        DBusConnection *connection)
  115: {
  116:   CData *cd;
  117: 
  118:   cd = NULL;
  119:   
  120:   dbus_connection_set_dispatch_status_function (connection, dispatch_status_function,
  121:                                                 loop, NULL);
  122:   
  123:   cd = cdata_new (loop, connection);
  124:   if (cd == NULL)
  125:     goto nomem;
  126: 
  127:   /* Because dbus-mainloop.c checks dbus_timeout_get_enabled(),
  128:    * dbus_watch_get_enabled() directly, we don't have to provide
  129:    * "toggled" callbacks.
  130:    */
  131:   
  132:   if (!dbus_connection_set_watch_functions (connection,
  133:                                             add_watch,
  134:                                             remove_watch,
  135:                                             NULL,
  136:                                             cd, cdata_free))
  137:     goto nomem;
  138: 
  139: 
  140:   cd = cdata_new (loop, connection);
  141:   if (cd == NULL)
  142:     goto nomem;
  143:   
  144:   if (!dbus_connection_set_timeout_functions (connection,
  145:                                               add_timeout,
  146:                                               remove_timeout,
  147:                                               NULL,
  148:                                               cd, cdata_free))
  149:     goto nomem;
  150: 
  151:   if (dbus_connection_get_dispatch_status (connection) != DBUS_DISPATCH_COMPLETE)
  152:     {
  153:       if (!_dbus_loop_queue_dispatch (loop, connection))
  154:         goto nomem;
  155:     }
  156:   
  157:   return TRUE;
  158:   
  159:  nomem:
  160:   if (cd)
  161:     cdata_free (cd);
  162:   
  163:   dbus_connection_set_dispatch_status_function (connection, NULL, NULL, NULL);
  164:   dbus_connection_set_watch_functions (connection, NULL, NULL, NULL, NULL, NULL);
  165:   dbus_connection_set_timeout_functions (connection, NULL, NULL, NULL, NULL, NULL);
  166:   
  167:   return FALSE;
  168: }
  169: 
  170: void
  171: test_connection_shutdown (DBusLoop       *loop,
  172:                           DBusConnection *connection)
  173: {
  174:   if (!dbus_connection_set_watch_functions (connection,
  175:                                             NULL,
  176:                                             NULL,
  177:                                             NULL,
  178:                                             NULL, NULL))
  179:     _dbus_assert_not_reached ("setting watch functions to NULL failed");
  180:   
  181:   if (!dbus_connection_set_timeout_functions (connection,
  182:                                               NULL,
  183:                                               NULL,
  184:                                               NULL,
  185:                                               NULL, NULL))
  186:     _dbus_assert_not_reached ("setting timeout functions to NULL failed");
  187: 
  188:   dbus_connection_set_dispatch_status_function (connection, NULL, NULL, NULL);
  189: }