sl@0: /* -*- mode: C; c-file-style: "gnu" -*- */
sl@0: /* test.c  unit test routines
sl@0:  *
sl@0:  * Copyright (C) 2003 Red Hat, Inc.
sl@0:  * Portion Copyright © 2008 Nokia Corporation and/or its subsidiary(-ies). All rights reserved.
sl@0:  * Licensed under the Academic Free License version 2.1
sl@0:  * 
sl@0:  * This program is free software; you can redistribute it and/or modify
sl@0:  * it under the terms of the GNU General Public License as published by
sl@0:  * the Free Software Foundation; either version 2 of the License, or
sl@0:  * (at your option) any later version.
sl@0:  *
sl@0:  * This program is distributed in the hope that it will be useful,
sl@0:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
sl@0:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
sl@0:  * GNU General Public License for more details.
sl@0:  * 
sl@0:  * You should have received a copy of the GNU General Public License
sl@0:  * along with this program; if not, write to the Free Software
sl@0:  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
sl@0:  *
sl@0:  */
sl@0: 
sl@0: #ifndef __SYMBIAN32__
sl@0: #include <config.h>
sl@0: #else
sl@0: #include "config.h"
sl@0: #endif //__SYMBIAN32__
sl@0: 
sl@0: #ifdef DBUS_BUILD_TESTS
sl@0: #include "test.h"
sl@0: #ifndef __SYMBIAN32__
sl@0: #include <dbus/dbus-internals.h>
sl@0: #include <dbus/dbus-list.h>
sl@0: #else
sl@0: #include "dbus-internals.h"
sl@0: #include "dbus-list.h"
sl@0: #endif //__SYMBIAN32__
sl@0: 
sl@0: /* The "debug client" watch/timeout handlers don't dispatch messages,
sl@0:  * as we manually pull them in order to verify them. This is why they
sl@0:  * are different from the real handlers in connection.c
sl@0:  */
sl@0: static DBusList *clients = NULL;
sl@0: static DBusLoop *client_loop = NULL;
sl@0: 
sl@0: static dbus_bool_t
sl@0: client_watch_callback (DBusWatch     *watch,
sl@0:                        unsigned int   condition,
sl@0:                        void          *data)
sl@0: {
sl@0:   /* FIXME this can be done in dbus-mainloop.c
sl@0:    * if the code in activation.c for the babysitter
sl@0:    * watch handler is fixed.
sl@0:    */
sl@0:  
sl@0:   return dbus_watch_handle (watch, condition);
sl@0: }
sl@0: 
sl@0: static dbus_bool_t
sl@0: add_client_watch (DBusWatch      *watch,
sl@0:                   void           *data)
sl@0: {
sl@0:   DBusConnection *connection = data;
sl@0:   
sl@0:   return _dbus_loop_add_watch (client_loop,
sl@0:                                watch, client_watch_callback, connection,
sl@0:                                NULL);
sl@0: }
sl@0: 
sl@0: static void
sl@0: remove_client_watch (DBusWatch      *watch,
sl@0:                      void           *data)
sl@0: {
sl@0:   DBusConnection *connection = data;
sl@0:   
sl@0:   _dbus_loop_remove_watch (client_loop,
sl@0:                            watch, client_watch_callback, connection);
sl@0: }
sl@0: 
sl@0: static void
sl@0: client_timeout_callback (DBusTimeout   *timeout,
sl@0:                          void          *data)
sl@0: {
sl@0:   DBusConnection *connection = data;
sl@0: 
sl@0:   dbus_connection_ref (connection);
sl@0: 
sl@0:   /* can return FALSE on OOM but we just let it fire again later */
sl@0:   dbus_timeout_handle (timeout);
sl@0: 
sl@0:   dbus_connection_unref (connection);
sl@0: }
sl@0: 
sl@0: static dbus_bool_t
sl@0: add_client_timeout (DBusTimeout    *timeout,
sl@0:                     void           *data)
sl@0: {
sl@0:   DBusConnection *connection = data;
sl@0:   
sl@0:   return _dbus_loop_add_timeout (client_loop, timeout, client_timeout_callback, connection, NULL);
sl@0: }
sl@0: 
sl@0: static void
sl@0: remove_client_timeout (DBusTimeout    *timeout,
sl@0:                        void           *data)
sl@0: {
sl@0:   DBusConnection *connection = data;
sl@0:   
sl@0:   _dbus_loop_remove_timeout (client_loop, timeout, client_timeout_callback, connection);
sl@0: }
sl@0: 
sl@0: static DBusHandlerResult
sl@0: client_disconnect_filter (DBusConnection     *connection,
sl@0:                           DBusMessage        *message,
sl@0:                           void               *user_data)
sl@0: {
sl@0:   if (!dbus_message_is_signal (message,
sl@0:                                DBUS_INTERFACE_LOCAL,
sl@0:                                "Disconnected"))
sl@0:     return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
sl@0:     
sl@0:   _dbus_verbose ("Removing client %p in disconnect handler\n",
sl@0:                  connection);
sl@0:   
sl@0:   _dbus_list_remove (&clients, connection);
sl@0: 
sl@0:   dbus_connection_unref (connection);
sl@0:   
sl@0:   if (clients == NULL)
sl@0:     {
sl@0:       _dbus_loop_unref (client_loop);
sl@0:       client_loop = NULL;
sl@0:     }
sl@0:   
sl@0:   return DBUS_HANDLER_RESULT_HANDLED;
sl@0: }
sl@0: 
sl@0: dbus_bool_t
sl@0: bus_setup_debug_client (DBusConnection *connection)
sl@0: {
sl@0:   dbus_bool_t retval;  
sl@0: 
sl@0:   if (!dbus_connection_add_filter (connection,
sl@0:                                    client_disconnect_filter,
sl@0:                                    NULL, NULL))
sl@0:     return FALSE;
sl@0: 
sl@0:   retval = FALSE;
sl@0: 
sl@0:   if (client_loop == NULL)
sl@0:     {
sl@0:       client_loop = _dbus_loop_new ();
sl@0:       if (client_loop == NULL)
sl@0:         goto out;
sl@0:     }
sl@0:   
sl@0:   if (!dbus_connection_set_watch_functions (connection,
sl@0:                                             add_client_watch,
sl@0:                                             remove_client_watch,
sl@0:                                             NULL,
sl@0:                                             connection,
sl@0:                                             NULL))
sl@0:     goto out;
sl@0:       
sl@0:   if (!dbus_connection_set_timeout_functions (connection,
sl@0:                                               add_client_timeout,
sl@0:                                               remove_client_timeout,
sl@0:                                               NULL,
sl@0:                                               connection, NULL))
sl@0:     goto out;
sl@0: 
sl@0:   if (!_dbus_list_append (&clients, connection))
sl@0:     goto out;
sl@0:   
sl@0:   retval = TRUE;
sl@0:   
sl@0:  out:
sl@0:   if (!retval)
sl@0:     {
sl@0:       dbus_connection_remove_filter (connection,
sl@0:                                      client_disconnect_filter,
sl@0:                                      NULL);
sl@0:       
sl@0:       dbus_connection_set_watch_functions (connection,
sl@0:                                            NULL, NULL, NULL, NULL, NULL);
sl@0:       dbus_connection_set_timeout_functions (connection,
sl@0:                                              NULL, NULL, NULL, NULL, NULL);
sl@0: 
sl@0:       _dbus_list_remove_last (&clients, connection);
sl@0: 
sl@0:       if (clients == NULL)
sl@0:         {
sl@0:           _dbus_loop_unref (client_loop);
sl@0:           client_loop = NULL;
sl@0:         }
sl@0:     }
sl@0:       
sl@0:   return retval;
sl@0: }
sl@0: 
sl@0: void
sl@0: bus_test_clients_foreach (BusConnectionForeachFunction  function,
sl@0:                           void                         *data)
sl@0: {
sl@0:   DBusList *link;
sl@0:   
sl@0:   link = _dbus_list_get_first_link (&clients);
sl@0:   while (link != NULL)
sl@0:     {
sl@0:       DBusConnection *connection = link->data;
sl@0:       DBusList *next = _dbus_list_get_next_link (&clients, link);
sl@0: 
sl@0:       if (!(* function) (connection, data))
sl@0:         break;
sl@0:       
sl@0:       link = next;
sl@0:     }
sl@0: }
sl@0: 
sl@0: dbus_bool_t
sl@0: bus_test_client_listed (DBusConnection *connection)
sl@0: {
sl@0:   DBusList *link;
sl@0:   
sl@0:   link = _dbus_list_get_first_link (&clients);
sl@0:   while (link != NULL)
sl@0:     {
sl@0:       DBusConnection *c = link->data;
sl@0:       DBusList *next = _dbus_list_get_next_link (&clients, link);
sl@0: 
sl@0:       if (c == connection)
sl@0:         return TRUE;
sl@0:       
sl@0:       link = next;
sl@0:     }
sl@0: 
sl@0:   return FALSE;
sl@0: }
sl@0: 
sl@0: void
sl@0: bus_test_run_clients_loop (dbus_bool_t block_once)
sl@0: {  
sl@0:   if (client_loop == NULL)
sl@0:     return;
sl@0: 
sl@0:   _dbus_verbose ("---> Dispatching on \"client side\"\n");
sl@0:   
sl@0:   /* dispatch before we block so pending dispatches
sl@0:    * won't make our block return early
sl@0:    */
sl@0:   _dbus_loop_dispatch (client_loop);
sl@0:   
sl@0:   /* Do one blocking wait, since we're expecting data */
sl@0:   if (block_once)
sl@0:     {
sl@0:       _dbus_verbose ("---> blocking on \"client side\"\n");
sl@0:       _dbus_loop_iterate (client_loop, TRUE);
sl@0:     }
sl@0: 
sl@0:   /* Then mop everything up */
sl@0:   while (_dbus_loop_iterate (client_loop, FALSE))
sl@0:     ;
sl@0: 
sl@0:   _dbus_verbose ("---> Done dispatching on \"client side\"\n");
sl@0: }
sl@0: 
sl@0: void
sl@0: bus_test_run_bus_loop (BusContext *context,
sl@0:                        dbus_bool_t block_once)
sl@0: {
sl@0:   _dbus_verbose ("---> Dispatching on \"server side\"\n");
sl@0:   
sl@0:   /* dispatch before we block so pending dispatches
sl@0:    * won't make our block return early
sl@0:    */
sl@0:   _dbus_loop_dispatch (bus_context_get_loop (context));
sl@0:   
sl@0:   /* Do one blocking wait, since we're expecting data */
sl@0:   if (block_once)
sl@0:     {
sl@0:       _dbus_verbose ("---> blocking on \"server side\"\n");
sl@0:       _dbus_loop_iterate (bus_context_get_loop (context), TRUE);
sl@0:     }
sl@0: 
sl@0:   /* Then mop everything up */
sl@0:   while (_dbus_loop_iterate (bus_context_get_loop (context), FALSE))
sl@0:     ;
sl@0: 
sl@0:   _dbus_verbose ("---> Done dispatching on \"server side\"\n");
sl@0: }
sl@0: 
sl@0: void
sl@0: bus_test_run_everything (BusContext *context)
sl@0: {
sl@0:   while (_dbus_loop_iterate (bus_context_get_loop (context), FALSE) ||
sl@0:          (client_loop == NULL || _dbus_loop_iterate (client_loop, FALSE)))
sl@0:     ;
sl@0: }
sl@0: 
sl@0: BusContext*
sl@0: bus_context_new_test (const DBusString *test_data_dir,
sl@0:                       const char       *filename)
sl@0: {
sl@0:   DBusError error;
sl@0:   DBusString config_file;
sl@0:   DBusString relative;
sl@0:   BusContext *context;
sl@0:   
sl@0:   if (!_dbus_string_init (&config_file))
sl@0:     {
sl@0:       _dbus_warn ("No memory\n");
sl@0:       return NULL;
sl@0:     }
sl@0: 
sl@0:   if (!_dbus_string_copy (test_data_dir, 0,
sl@0:                           &config_file, 0))
sl@0:     {
sl@0:       _dbus_warn ("No memory\n");
sl@0:       _dbus_string_free (&config_file);
sl@0:       return NULL;
sl@0:     }
sl@0: 
sl@0:   _dbus_string_init_const (&relative, filename);
sl@0: 
sl@0:   if (!_dbus_concat_dir_and_file (&config_file, &relative))
sl@0:     {
sl@0:       _dbus_warn ("No memory\n");
sl@0:       _dbus_string_free (&config_file);
sl@0:       return NULL;
sl@0:     }
sl@0:   
sl@0:   dbus_error_init (&error);
sl@0:   context = bus_context_new (&config_file, FALSE, -1, -1, &error);
sl@0:   if (context == NULL)
sl@0:     {
sl@0:       _DBUS_ASSERT_ERROR_IS_SET (&error);
sl@0:       
sl@0:       _dbus_warn ("Failed to create debug bus context from configuration file %s: %s\n",
sl@0:                   filename, error.message);
sl@0: 
sl@0:       dbus_error_free (&error);
sl@0:       
sl@0:       _dbus_string_free (&config_file);
sl@0:       
sl@0:       return NULL;
sl@0:     }
sl@0: 
sl@0:   _dbus_string_free (&config_file);
sl@0:   
sl@0:   return context;
sl@0: }
sl@0: 
sl@0: #endif