bus/test-main.c

   1 /* -*- mode: C; c-file-style: "gnu"; indent-tabs-mode: nil; -*- */
   2 /* test-main.c  main() for make check
   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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  21  *
  22  */
  23
  24 #include <config.h>
  25 #include "test.h"
  26 #include <stdio.h>
  27 #include <stdlib.h>
  28 #include <dbus/dbus-string.h>
  29 #include <dbus/dbus-sysdeps.h>
  30 #include <dbus/dbus-internals.h>
  31 #include <dbus/dbus-message-internal.h>
  32 #include "selinux.h"
  33
  34 #ifdef DBUS_UNIX
  35 # include <dbus/dbus-sysdeps-unix.h>
  36 #endif
  37
  38 #ifdef DBUS_ENABLE_EMBEDDED_TESTS
  39 static void
  40 die (const char *failure)
  41 {
  42   fprintf (stderr, "Unit test failed: %s\n", failure);
  43   exit (1);
  44 }
  45
  46 static void
  47 check_memleaks (const char *name)
  48 {
  49   dbus_shutdown ();
  50
  51   printf ("%s: checking for memleaks\n", name);
  52   if (_dbus_get_malloc_blocks_outstanding () != 0)
  53     {
  54       _dbus_warn ("%d dbus_malloc blocks were not freed\n",
  55                   _dbus_get_malloc_blocks_outstanding ());
  56       die ("memleaks");
  57     }
  58 }
  59 #endif /* DBUS_ENABLE_EMBEDDED_TESTS */
  60
  61 static DBusInitialFDs *initial_fds = NULL;
  62
  63 static void
  64 test_pre_hook (void)
  65 {
  66
  67   if (_dbus_getenv ("DBUS_TEST_SELINUX")
  68       && (!bus_selinux_pre_init ()
  69           || !bus_selinux_full_init ()))
  70     die ("could not init selinux support");
  71
  72   initial_fds = _dbus_check_fdleaks_enter ();
  73 }
  74
  75 static char *progname = "";
  76
  77 static void
  78 test_post_hook (void)
  79 {
  80   if (_dbus_getenv ("DBUS_TEST_SELINUX"))
  81     bus_selinux_shutdown ();
  82   check_memleaks (progname);
  83
  84   _dbus_check_fdleaks_leave (initial_fds);
  85   initial_fds = NULL;
  86 }
  87
  88 int
  89 main (int argc, char **argv)
  90 {
  91 #ifdef DBUS_ENABLE_EMBEDDED_TESTS
  92   const char *dir;
  93   const char *only;
  94   DBusString test_data_dir;
  95
  96   progname = argv[0];
  97
  98   if (argc > 1)
  99     dir = argv[1];
 100   else
 101     dir = _dbus_getenv ("DBUS_TEST_DATA");
 102
 103   if (argc > 2)
 104     only = argv[2];
 105   else
 106     only = NULL;
 107
 108   if (dir == NULL)
 109     {
 110       fprintf (stderr, "Must specify test data directory as argv[1] or in DBUS_TEST_DATA env variable\n");
 111       return 1;
 112     }
 113
 114   _dbus_string_init_const (&test_data_dir, dir);
 115
 116 #ifdef DBUS_UNIX
 117   /* close any inherited fds so dbus-spawn's check for close-on-exec works */
 118   _dbus_close_all ();
 119 #endif
 120
 121   if (!_dbus_threads_init_debug ())
 122     die ("initializing debug threads");
 123
 124   if (only == NULL || strcmp (only, "expire-list") == 0)
 125     {
 126       test_pre_hook ();
 127       printf ("%s: Running expire list test\n", argv[0]);
 128       if (!bus_expire_list_test (&test_data_dir))
 129         die ("expire list");
 130       test_post_hook ();
 131     }
 132
 133   if (only == NULL || strcmp (only, "config-parser") == 0)
 134     {
 135       test_pre_hook ();
 136       printf ("%s: Running config file parser test\n", argv[0]);
 137       if (!bus_config_parser_test (&test_data_dir))
 138         die ("parser");
 139       test_post_hook ();
 140     }
 141
 142   if (only == NULL || strcmp (only, "signals") == 0)
 143     {
 144       test_pre_hook ();
 145       printf ("%s: Running signals test\n", argv[0]);
 146       if (!bus_signals_test (&test_data_dir))
 147         die ("signals");
 148       test_post_hook ();
 149     }
 150
 151   if (only == NULL || strcmp (only, "dispatch-sha1") == 0)
 152     {
 153       test_pre_hook ();
 154       printf ("%s: Running SHA1 connection test\n", argv[0]);
 155       if (!bus_dispatch_sha1_test (&test_data_dir))
 156         die ("sha1");
 157       test_post_hook ();
 158     }
 159
 160   if (only == NULL || strcmp (only, "dispatch") == 0)
 161     {
 162       test_pre_hook ();
 163       printf ("%s: Running message dispatch test\n", argv[0]);
 164       if (!bus_dispatch_test (&test_data_dir))
 165         die ("dispatch");
 166       test_post_hook ();
 167     }
 168
 169   if (only == NULL || strcmp (only, "activation-service-reload") == 0)
 170     {
 171       test_pre_hook ();
 172       printf ("%s: Running service files reloading test\n", argv[0]);
 173       if (!bus_activation_service_reload_test (&test_data_dir))
 174         die ("service reload");
 175       test_post_hook ();
 176     }
 177
 178 #ifdef HAVE_UNIX_FD_PASSING
 179   if (only == NULL || strcmp (only, "unix-fds-passing") == 0)
 180     {
 181       test_pre_hook ();
 182       printf ("%s: Running unix fd passing test\n", argv[0]);
 183       if (!bus_unix_fds_passing_test (&test_data_dir))
 184         die ("unix fd passing");
 185       test_post_hook ();
 186     }
 187 #endif
 188
 189   printf ("%s: Success\n", argv[0]);
 190
 191
 192   return 0;
 193 #else /* DBUS_ENABLE_EMBEDDED_TESTS */
 194
 195   printf ("Not compiled with test support\n");
 196
 197   return 0;
 198 #endif
 199 }