glib/gtestutils.c

   1 /* GLib testing utilities
   2  * Copyright (C) 2007 Imendio AB
   3  * Authors: Tim Janik, Sven Herzberg
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the
  17  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18  * Boston, MA 02111-1307, USA.
  19  */
  20 #include "config.h"
  21 #include "gtestutils.h"
  22 #include "galias.h"
  23 #include <sys/types.h>
  24 #ifdef G_OS_UNIX
  25 #include <sys/wait.h>
  26 #include <sys/time.h>
  27 #include <fcntl.h>
  28 #endif
  29 #include <string.h>
  30 #include <stdlib.h>
  31 #ifdef HAVE_UNISTD_H
  32 #include <unistd.h>
  33 #endif
  34 #ifdef G_OS_WIN32
  35 #include <io.h>
  36 #endif
  37 #include <errno.h>
  38 #include <signal.h>
  39 #ifdef HAVE_SYS_SELECT_H
  40 #include <sys/select.h>
  41 #endif /* HAVE_SYS_SELECT_H */
  42
  43 /* --- structures --- */
  44 struct GTestCase
  45 {
  46   gchar  *name;
  47   guint   fixture_size;
  48   void   (*fixture_setup)    (void*, gconstpointer);
  49   void   (*fixture_test)     (void*, gconstpointer);
  50   void   (*fixture_teardown) (void*, gconstpointer);
  51   gpointer test_data;
  52 };
  53 struct GTestSuite
  54 {
  55   gchar  *name;
  56   GSList *suites;
  57   GSList *cases;
  58 };
  59 typedef struct DestroyEntry DestroyEntry;
  60 struct DestroyEntry
  61 {
  62   DestroyEntry *next;
  63   GDestroyNotify destroy_func;
  64   gpointer       destroy_data;
  65 };
  66
  67 /* --- prototypes --- */
  68 static void     test_run_seed                   (const gchar *rseed);
  69 static void     test_trap_clear                 (void);
  70 static guint8*  g_test_log_dump                 (GTestLogMsg *msg,
  71                                                  guint       *len);
  72 static void     gtest_default_log_handler       (const gchar    *log_domain,
  73                                                  GLogLevelFlags  log_level,
  74                                                  const gchar    *message,
  75                                                  gpointer        unused_data);
  76
  77
  78 /* --- variables --- */
  79 static int         test_log_fd = -1;
  80 static gboolean    test_mode_fatal = TRUE;
  81 static gboolean    g_test_run_once = TRUE;
  82 static gboolean    test_run_list = FALSE;
  83 static gchar      *test_run_seedstr = NULL;
  84 static GRand      *test_run_rand = NULL;
  85 static gchar      *test_run_name = "";
  86 static guint       test_run_forks = 0;
  87 static guint       test_run_count = 0;
  88 static guint       test_skip_count = 0;
  89 static GTimer     *test_user_timer = NULL;
  90 static double      test_user_stamp = 0;
  91 static GSList     *test_paths = NULL;
  92 static GTestSuite *test_suite_root = NULL;
  93 static int         test_trap_last_status = 0;
  94 static int         test_trap_last_pid = 0;
  95 static char       *test_trap_last_stdout = NULL;
  96 static char       *test_trap_last_stderr = NULL;
  97 static char       *test_uri_base = NULL;
  98 static gboolean    test_debug_log = FALSE;
  99 static DestroyEntry *test_destroy_queue = NULL;
 100 static GTestConfig mutable_test_config_vars = {
 101   FALSE,        /* test_initialized */
 102   TRUE,         /* test_quick */
 103   FALSE,        /* test_perf */
 104   FALSE,        /* test_verbose */
 105   FALSE,        /* test_quiet */
 106 };
 107 const GTestConfig * const g_test_config_vars = &mutable_test_config_vars;
 108
 109 /* --- functions --- */
 110 const char*
 111 g_test_log_type_name (GTestLogType log_type)
 112 {
 113   switch (log_type)
 114     {
 115     case G_TEST_LOG_NONE:               return "none";
 116     case G_TEST_LOG_ERROR:              return "error";
 117     case G_TEST_LOG_START_BINARY:       return "binary";
 118     case G_TEST_LOG_LIST_CASE:          return "list";
 119     case G_TEST_LOG_SKIP_CASE:          return "skip";
 120     case G_TEST_LOG_START_CASE:         return "start";
 121     case G_TEST_LOG_STOP_CASE:          return "stop";
 122     case G_TEST_LOG_MIN_RESULT:         return "minperf";
 123     case G_TEST_LOG_MAX_RESULT:         return "maxperf";
 124     case G_TEST_LOG_MESSAGE:            return "message";
 125     }
 126   return "???";
 127 }
 128
 129 static void
 130 g_test_log_send (guint         n_bytes,
 131                  const guint8 *buffer)
 132 {
 133   if (test_log_fd >= 0)
 134     {
 135       int r;
 136       do
 137         r = write (test_log_fd, buffer, n_bytes);
 138       while (r < 0 && errno == EINTR);
 139     }
 140   if (test_debug_log)
 141     {
 142       GTestLogBuffer *lbuffer = g_test_log_buffer_new();
 143       GTestLogMsg *msg;
 144       guint ui;
 145       g_test_log_buffer_push (lbuffer, n_bytes, buffer);
 146       msg = g_test_log_buffer_pop (lbuffer);
 147       g_assert (msg != NULL); /* FIXME: should be g_awrn_if_fail */
 148       g_assert (lbuffer->data->len == 0); /* FIXME: should be g_awrn_if_fail */
 149       g_test_log_buffer_free (lbuffer);
 150       /* print message */
 151       g_printerr ("{*LOG(%s)", g_test_log_type_name (msg->log_type));
 152       for (ui = 0; ui < msg->n_strings; ui++)
 153         g_printerr (":{%s}", msg->strings[ui]);
 154       if (msg->n_nums)
 155         {
 156           g_printerr (":(");
 157           for (ui = 0; ui < msg->n_nums; ui++)
 158             g_printerr ("%s%.16Lg", ui ? ";" : "", msg->nums[ui]);
 159           g_printerr (")");
 160         }
 161       g_printerr (":LOG*}\n");
 162       g_test_log_msg_free (msg);
 163     }
 164 }
 165
 166 static void
 167 g_test_log (GTestLogType lbit,
 168             const gchar *string1,
 169             const gchar *string2,
 170             guint        n_args,
 171             long double *largs)
 172 {
 173   gboolean fail = lbit == G_TEST_LOG_STOP_CASE && largs[0] != 0;
 174   GTestLogMsg msg;
 175   gchar *astrings[3] = { NULL, NULL, NULL };
 176   guint8 *dbuffer;
 177   guint32 dbufferlen;
 178
 179   switch (lbit)
 180     {
 181     case G_TEST_LOG_START_BINARY:
 182       if (g_test_verbose())
 183         g_print ("GTest: random seed: %s\n", string2);
 184       break;
 185     case G_TEST_LOG_STOP_CASE:
 186       if (g_test_verbose())
 187         g_print ("GTest: result: %s\n", fail ? "FAIL" : "OK");
 188       else if (!g_test_quiet())
 189         g_print ("%s\n", fail ? "FAIL" : "OK");
 190       if (fail && test_mode_fatal)
 191         abort();
 192       break;
 193     case G_TEST_LOG_MIN_RESULT:
 194       if (g_test_verbose())
 195         g_print ("(MINPERF:%s)\n", string1);
 196       break;
 197     case G_TEST_LOG_MAX_RESULT:
 198       if (g_test_verbose())
 199         g_print ("(MAXPERF:%s)\n", string1);
 200       break;
 201     case G_TEST_LOG_MESSAGE:
 202       if (g_test_verbose())
 203         g_print ("(MSG: %s)\n", string1);
 204       break;
 205     default: ;
 206     }
 207
 208   msg.log_type = lbit;
 209   msg.n_strings = (string1 != NULL) + (string1 && string2);
 210   msg.strings = astrings;
 211   astrings[0] = (gchar*) string1;
 212   astrings[1] = astrings[0] ? (gchar*) string2 : NULL;
 213   msg.n_nums = n_args;
 214   msg.nums = largs;
 215   dbuffer = g_test_log_dump (&msg, &dbufferlen);
 216   g_test_log_send (dbufferlen, dbuffer);
 217   g_free (dbuffer);
 218
 219   switch (lbit)
 220     {
 221     case G_TEST_LOG_START_CASE:
 222       if (g_test_verbose())
 223         g_print ("GTest: run: %s\n", string1);
 224       else if (!g_test_quiet())
 225         g_print ("%s: ", string1);
 226       break;
 227     default: ;
 228     }
 229 }
 230
 231 static void
 232 parse_args (gint    *argc_p,
 233             gchar ***argv_p)
 234 {
 235   guint argc = *argc_p;
 236   gchar **argv = *argv_p;
 237   guint i, e;
 238   /* parse known args */
 239   for (i = 1; i < argc; i++)
 240     {
 241       if (strcmp (argv[i], "--g-fatal-warnings") == 0)
 242         {
 243           GLogLevelFlags fatal_mask = (GLogLevelFlags) g_log_set_always_fatal ((GLogLevelFlags) G_LOG_FATAL_MASK);
 244           fatal_mask = (GLogLevelFlags) (fatal_mask | G_LOG_LEVEL_WARNING | G_LOG_LEVEL_CRITICAL);
 245           g_log_set_always_fatal (fatal_mask);
 246           argv[i] = NULL;
 247         }
 248       else if (strcmp (argv[i], "--keep-going") == 0 ||
 249                strcmp (argv[i], "-k") == 0)
 250         {
 251           test_mode_fatal = FALSE;
 252           argv[i] = NULL;
 253         }
 254       else if (strcmp (argv[i], "--debug-log") == 0)
 255         {
 256           test_debug_log = TRUE;
 257           argv[i] = NULL;
 258         }
 259       else if (strcmp ("--GTestLogFD", argv[i]) == 0 || strncmp ("--GTestLogFD=", argv[i], 13) == 0)
 260         {
 261           gchar *equal = argv[i] + 12;
 262           if (*equal == '=')
 263             test_log_fd = g_ascii_strtoull (equal + 1, NULL, 0);
 264           else if (i + 1 < argc)
 265             {
 266               argv[i++] = NULL;
 267               test_log_fd = g_ascii_strtoull (argv[i], NULL, 0);
 268             }
 269           argv[i] = NULL;
 270         }
 271       else if (strcmp ("--GTestSkipCount", argv[i]) == 0 || strncmp ("--GTestSkipCount=", argv[i], 17) == 0)
 272         {
 273           gchar *equal = argv[i] + 16;
 274           if (*equal == '=')
 275             test_skip_count = g_ascii_strtoull (equal + 1, NULL, 0);
 276           else if (i + 1 < argc)
 277             {
 278               argv[i++] = NULL;
 279               test_skip_count = g_ascii_strtoull (argv[i], NULL, 0);
 280             }
 281           argv[i] = NULL;
 282         }
 283       else if (strcmp ("-p", argv[i]) == 0 || strncmp ("-p=", argv[i], 3) == 0)
 284         {
 285           gchar *equal = argv[i] + 2;
 286           if (*equal == '=')
 287             test_paths = g_slist_prepend (test_paths, equal + 1);
 288           else if (i + 1 < argc)
 289             {
 290               argv[i++] = NULL;
 291               test_paths = g_slist_prepend (test_paths, argv[i]);
 292             }
 293           argv[i] = NULL;
 294         }
 295       else if (strcmp ("-m", argv[i]) == 0 || strncmp ("-m=", argv[i], 3) == 0)
 296         {
 297           gchar *equal = argv[i] + 2;
 298           const gchar *mode = "";
 299           if (*equal == '=')
 300             mode = equal + 1;
 301           else if (i + 1 < argc)
 302             {
 303               argv[i++] = NULL;
 304               mode = argv[i];
 305             }
 306           if (strcmp (mode, "perf") == 0)
 307             mutable_test_config_vars.test_perf = TRUE;
 308           else if (strcmp (mode, "slow") == 0)
 309             mutable_test_config_vars.test_quick = FALSE;
 310           else if (strcmp (mode, "thorough") == 0)
 311             mutable_test_config_vars.test_quick = FALSE;
 312           else if (strcmp (mode, "quick") == 0)
 313             {
 314               mutable_test_config_vars.test_quick = TRUE;
 315               mutable_test_config_vars.test_perf = FALSE;
 316             }
 317           else
 318             g_error ("unknown test mode: -m %s", mode);
 319           argv[i] = NULL;
 320         }
 321       else if (strcmp ("-q", argv[i]) == 0 || strcmp ("--quiet", argv[i]) == 0)
 322         {
 323           mutable_test_config_vars.test_quiet = TRUE;
 324           mutable_test_config_vars.test_verbose = FALSE;
 325           argv[i] = NULL;
 326         }
 327       else if (strcmp ("--verbose", argv[i]) == 0)
 328         {
 329           mutable_test_config_vars.test_quiet = FALSE;
 330           mutable_test_config_vars.test_verbose = TRUE;
 331           argv[i] = NULL;
 332         }
 333       else if (strcmp ("-l", argv[i]) == 0)
 334         {
 335           test_run_list = TRUE;
 336           argv[i] = NULL;
 337         }
 338       else if (strcmp ("--seed", argv[i]) == 0 || strncmp ("--seed=", argv[i], 7) == 0)
 339         {
 340           gchar *equal = argv[i] + 6;
 341           if (*equal == '=')
 342             test_run_seedstr = equal + 1;
 343           else if (i + 1 < argc)
 344             {
 345               argv[i++] = NULL;
 346               test_run_seedstr = argv[i];
 347             }
 348           argv[i] = NULL;
 349         }
 350     }
 351   /* collapse argv */
 352   e = 1;
 353   for (i = 1; i < argc; i++)
 354     if (argv[i])
 355       {
 356         argv[e++] = argv[i];
 357         if (i >= e)
 358           argv[i] = NULL;
 359       }
 360   *argc_p = e;
 361 }
 362
 363 /**
 364  * g_test_init:
 365  * @argc: Address of the @argc parameter of the main() function.
 366  *        Changed if any arguments were handled.
 367  * @argv: Address of the @argv parameter of main().
 368  *        Any parameters understood by g_test_init() stripped before return.
 369  *
 370  * Initialize the GLib testing framework, e.g. by seeding the
 371  * test random number generator, the name for g_get_prgname()
 372  * and parsing test related command line args.
 373  * So far, the following arguments are understood:
 374  * <informalexample>
 375  * -l                   list test cases available in a test executable.
 376  * --seed RANDOMSEED    provide a random seed to reproduce test runs using random numbers.
 377  * --verbose            run tests verbosely.
 378  * -q, --quiet          run tests quietly.
 379  * -p TESTPATH          execute all tests matching TESTPATH.
 380  * -m {perf|slow|thorough|quick}
 381  *                      execute tests according to these test modes:
 382  *                      perf - performance tests, may take long and report results.
 383  *                      slow - slow and thorough tests, may take quite long and maximize coverage.
 384  *                      thorough - currently an alias for "slow".
 385  *                      quick - quick tests, should run really quickly and give good coverage.
 386  * --debug-log          debug test logging output.
 387  * -k, --keep-going     gtester specific argument.
 388  * --GTestLogFD N       gtester specific argument.
 389  * --GTestSkipCount N   gtester specific argument.
 390  * </informalexample>
 391  */
 392 void
 393 g_test_init (int    *argc,
 394              char ***argv,
 395              ...)
 396 {
 397   static char seedstr[4 + 4 * 8 + 1];
 398   va_list args;
 399   gpointer vararg1;
 400   /* make warnings and criticals fatal for all test programs */
 401   GLogLevelFlags fatal_mask = (GLogLevelFlags) g_log_set_always_fatal ((GLogLevelFlags) G_LOG_FATAL_MASK);
 402   fatal_mask = (GLogLevelFlags) (fatal_mask | G_LOG_LEVEL_WARNING | G_LOG_LEVEL_CRITICAL);
 403   g_log_set_always_fatal (fatal_mask);
 404   /* check caller args */
 405   g_return_if_fail (argc != NULL);
 406   g_return_if_fail (argv != NULL);
 407   g_return_if_fail (g_test_config_vars->test_initialized == FALSE);
 408   mutable_test_config_vars.test_initialized = TRUE;
 409
 410   va_start (args, argv);
 411   vararg1 = va_arg (args, gpointer); /* reserved for future extensions */
 412   va_end (args);
 413   g_return_if_fail (vararg1 == NULL);
 414
 415   /* setup random seed string */
 416   g_snprintf (seedstr, sizeof (seedstr), "R02S%08x%08x%08x%08x", g_random_int(), g_random_int(), g_random_int(), g_random_int());
 417   test_run_seedstr = seedstr;
 418
 419   /* parse args, sets up mode, changes seed, etc. */
 420   parse_args (argc, argv);
 421   if (!g_get_prgname())
 422     g_set_prgname ((*argv)[0]);
 423
 424   /* verify GRand reliability, needed for reliable seeds */
 425   if (1)
 426     {
 427       GRand *rg = g_rand_new_with_seed (0xc8c49fb6);
 428       guint32 t1 = g_rand_int (rg), t2 = g_rand_int (rg), t3 = g_rand_int (rg), t4 = g_rand_int (rg);
 429       /* g_print ("GRand-current: 0x%x 0x%x 0x%x 0x%x\n", t1, t2, t3, t4); */
 430       if (t1 != 0xfab39f9b || t2 != 0xb948fb0e || t3 != 0x3d31be26 || t4 != 0x43a19d66)
 431         g_warning ("random numbers are not GRand-2.2 compatible, seeds may be broken (check $G_RANDOM_VERSION)");
 432       g_rand_free (rg);
 433     }
 434
 435   /* check rand seed */
 436   test_run_seed (test_run_seedstr);
 437
 438   /* report program start */
 439   g_log_set_default_handler (gtest_default_log_handler, NULL);
 440   g_test_log (G_TEST_LOG_START_BINARY, g_get_prgname(), test_run_seedstr, 0, NULL);
 441 }
 442
 443 static void
 444 test_run_seed (const gchar *rseed)
 445 {
 446   guint seed_failed = 0;
 447   if (test_run_rand)
 448     g_rand_free (test_run_rand);
 449   test_run_rand = NULL;
 450   while (strchr (" \t\v\r\n\f", *rseed))
 451     rseed++;
 452   if (strncmp (rseed, "R02S", 4) == 0)  /* seed for random generator 02 (GRand-2.2) */
 453     {
 454       const char *s = rseed + 4;
 455       if (strlen (s) >= 32)             /* require 4 * 8 chars */
 456         {
 457           guint32 seedarray[4];
 458           gchar *p, hexbuf[9] = { 0, };
 459           memcpy (hexbuf, s + 0, 8);
 460           seedarray[0] = g_ascii_strtoull (hexbuf, &p, 16);
 461           seed_failed += p != NULL && *p != 0;
 462           memcpy (hexbuf, s + 8, 8);
 463           seedarray[1] = g_ascii_strtoull (hexbuf, &p, 16);
 464           seed_failed += p != NULL && *p != 0;
 465           memcpy (hexbuf, s + 16, 8);
 466           seedarray[2] = g_ascii_strtoull (hexbuf, &p, 16);
 467           seed_failed += p != NULL && *p != 0;
 468           memcpy (hexbuf, s + 24, 8);
 469           seedarray[3] = g_ascii_strtoull (hexbuf, &p, 16);
 470           seed_failed += p != NULL && *p != 0;
 471           if (!seed_failed)
 472             {
 473               test_run_rand = g_rand_new_with_seed_array (seedarray, 4);
 474               return;
 475             }
 476         }
 477     }
 478   g_error ("Unknown or invalid random seed: %s", rseed);
 479 }
 480
 481 /**
 482  * g_test_rand_int:
 483  *
 484  * Get a reproducable random integer number.
 485  * The random numbers generate by the g_test_rand_*() family of functions
 486  * change with every new test program start, unless the --seed option is
 487  * given when starting test programs.
 488  * For individual test cases however, the random number generator is
 489  * reseeded, to avoid dependencies between tests and to make --seed
 490  * effective for all test cases.
 491  *
 492  * Returns: a random number from the seeded random number generator.
 493  */
 494 gint32
 495 g_test_rand_int (void)
 496 {
 497   return g_rand_int (test_run_rand);
 498 }
 499
 500 /**
 501  * g_test_rand_int_range:
 502  * @begin: the minimum value returned by this function
 503  * @end:   the smallest value not to be returned by this function
 504  *
 505  * Get a reproducable random integer number out of a specified range,
 506  * see g_test_rand_int() for details on test case random numbers.
 507  *
 508  * Returns: a number with @begin <= number < @end.
 509  */
 510 gint32
 511 g_test_rand_int_range (gint32          begin,
 512                        gint32          end)
 513 {
 514   return g_rand_int_range (test_run_rand, begin, end);
 515 }
 516
 517 /**
 518  * g_test_rand_double:
 519  *
 520  * Get a reproducable random floating point number,
 521  * see g_test_rand_int() for details on test case random numbers.
 522  *
 523  * Return a random number from the seeded random number generator.
 524  */
 525 double
 526 g_test_rand_double (void)
 527 {
 528   return g_rand_double (test_run_rand);
 529 }
 530
 531 /**
 532  * g_test_rand_double_range:
 533  * @range_start: the minimum value returned by this function
 534  * @range_end: the minimum value not returned by this function
 535  *
 536  * Get a reproducable random floating pointer number out of a specified range,
 537  * see g_test_rand_int() for details on test case random numbers.
 538  *
 539  * Returns: a number with @range_start <= number < @range_end.
 540  */
 541 double
 542 g_test_rand_double_range (double          range_start,
 543                           double          range_end)
 544 {
 545   return g_rand_double_range (test_run_rand, range_start, range_end);
 546 }
 547
 548 /**
 549  * g_test_timer_start:
 550  *
 551  * Start a timing test. Call g_test_timer_elapsed() when the task is supposed
 552  * to be done. Call this function again to restart the timer.
 553  */
 554 void
 555 g_test_timer_start (void)
 556 {
 557   if (!test_user_timer)
 558     test_user_timer = g_timer_new();
 559   test_user_stamp = 0;
 560   g_timer_start (test_user_timer);
 561 }
 562
 563 /**
 564  * g_test_timer_elapsed:
 565  *
 566  * Get the time since the last start of the timer with g_test_timer_start().
 567  *
 568  * Returns: the time since the last start of the timer, as a double
 569  */
 570 double
 571 g_test_timer_elapsed (void)
 572 {
 573   test_user_stamp = test_user_timer ? g_timer_elapsed (test_user_timer, NULL) : 0;
 574   return test_user_stamp;
 575 }
 576
 577 /**
 578  * g_test_timer_last:
 579  *
 580  * Report the last result of g_test_timer_elapsed().
 581  *
 582  * Returns: the last result of g_test_timer_elapsed(), as a double
 583  */
 584 double
 585 g_test_timer_last (void)
 586 {
 587   return test_user_stamp;
 588 }
 589
 590 /**
 591  * g_test_minimized_result:
 592  * @minimized_quantity: the reported value
 593  * @format: the format string of the report message
 594  *
 595  * Report the result of a performance or measurement test.
 596  * The test should generally strive to minimize the reported
 597  * quantities (smaller values are better than larger ones),
 598  * this and @minimized_quantity can determine sorting
 599  * order for test result reports.
 600  */
 601 void
 602 g_test_minimized_result (double          minimized_quantity,
 603                          const char     *format,
 604                          ...)
 605 {
 606   long double largs = minimized_quantity;
 607   gchar *buffer;
 608   va_list args;
 609   va_start (args, format);
 610   buffer = g_strdup_vprintf (format, args);
 611   va_end (args);
 612   g_test_log (G_TEST_LOG_MIN_RESULT, buffer, NULL, 1, &largs);
 613   g_free (buffer);
 614 }
 615
 616 /**
 617  * g_test_minimized_result:
 618  * @maximized_quantity: the reported value
 619  * @format: the format string of the report message
 620  *
 621  * Report the result of a performance or measurement test.
 622  * The test should generally strive to maximize the reported
 623  * quantities (larger values are better than smaller ones),
 624  * this and @maximized_quantity can determine sorting
 625  * order for test result reports.
 626  */
 627 void
 628 g_test_maximized_result (double          maximized_quantity,
 629                          const char     *format,
 630                          ...)
 631 {
 632   long double largs = maximized_quantity;
 633   gchar *buffer;
 634   va_list args;
 635   va_start (args, format);
 636   buffer = g_strdup_vprintf (format, args);
 637   va_end (args);
 638   g_test_log (G_TEST_LOG_MAX_RESULT, buffer, NULL, 1, &largs);
 639   g_free (buffer);
 640 }
 641
 642 /**
 643  * g_test_message:
 644  * @format: the format string
 645  * @...:    printf-like arguments to @format
 646  *
 647  * Add a message to the test report.
 648  */
 649 void
 650 g_test_message (const char *format,
 651                 ...)
 652 {
 653   gchar *buffer;
 654   va_list args;
 655   va_start (args, format);
 656   buffer = g_strdup_vprintf (format, args);
 657   va_end (args);
 658   g_test_log (G_TEST_LOG_MESSAGE, buffer, NULL, 0, NULL);
 659   g_free (buffer);
 660 }
 661
 662 /**
 663  * g_test_bug_base:
 664  * @uri_pattern: the base pattern for bug URIs
 665  *
 666  * Specify the base URI for bug reports.
 667  * The base URI is used to construct bug report messages for
 668  * g_test_message() when g_test_bug() is called.
 669  * Calling this function outside of a test case sets the
 670  * default base URI for all test cases. Calling it from within
 671  * a test case changes the base URI for the scope of the test
 672  * case only.
 673  * Bug URIs are constructed by appending a bug specific URI
 674  * portion to @uri_pattern, or by replacing the special string
 675  * '%s' within @uri_pattern if that is present.
 676  */
 677 void
 678 g_test_bug_base (const char *uri_pattern)
 679 {
 680   g_free (test_uri_base);
 681   test_uri_base = g_strdup (uri_pattern);
 682 }
 683
 684 /**
 685  * g_test_bug:
 686  * @bug_uri_snippet: Bug specific bug tracker URI portion.
 687  *
 688  * This function adds a message to test reports that
 689  * associates a bug URI with a test case.
 690  * Bug URIs are constructed from a base URI set with g_test_bug_base()
 691  * and @bug_uri_snippet.
 692  */
 693 void
 694 g_test_bug (const char *bug_uri_snippet)
 695 {
 696   char *c;
 697   g_return_if_fail (test_uri_base != NULL);
 698   g_return_if_fail (bug_uri_snippet != NULL);
 699   c = strstr (test_uri_base, "%s");
 700   if (c)
 701     {
 702       char *b = g_strndup (test_uri_base, c - test_uri_base);
 703       char *s = g_strconcat (b, bug_uri_snippet, c + 2, NULL);
 704       g_free (b);
 705       g_test_message ("Bug Reference: %s", s);
 706       g_free (s);
 707     }
 708   else
 709     g_test_message ("Bug Reference: %s%s", test_uri_base, bug_uri_snippet);
 710 }
 711
 712 /**
 713  * g_test_get_root:
 714  *
 715  * Get the toplevel test suite for the test path API.
 716  *
 717  * Returns: the toplevel #GTestSuite
 718  */
 719 GTestSuite*
 720 g_test_get_root (void)
 721 {
 722   if (!test_suite_root)
 723     {
 724       test_suite_root = g_test_create_suite ("root");
 725       g_free (test_suite_root->name);
 726       test_suite_root->name = g_strdup ("");
 727     }
 728   return test_suite_root;
 729 }
 730
 731 /**
 732  * g_test_run:
 733  *
 734  * Runs all tests under the toplevel suite which can be retrieved
 735  * with g_test_get_root(). Similar to g_test_run_suite(), the test
 736  * cases to be run are filtered according to
 737  * test path arguments (-p <testpath>) as parsed by g_test_init().
 738  * g_test_run_suite() or g_test_run() may only be called once
 739  * in a program.
 740  *
 741  * Returns: 0 on success
 742  */
 743 int
 744 g_test_run (void)
 745 {
 746   return g_test_run_suite (g_test_get_root());
 747 }
 748
 749 /**
 750  * g_test_create_case:
 751  * @test_name:     the name for the test case
 752  * @data_size:     the size of the fixture data structure
 753  * @test_data:     test data argument for the test functions
 754  * @data_setup:    the function to set up the fixture data
 755  * @data_test:     the actual test function
 756  * @data_teardown: the function to teardown the fixture data
 757  *
 758  * Create a new #GTestCase, named @test_name, this API is fairly
 759  * low level, calling g_test_add() or g_test_add_func() is preferable.
 760  * When this test is executed, a fixture structure of size @data_size
 761  * will be allocated and filled with 0s. Then data_setup() is called
 762  * to initialize the fixture. After fixture setup, the actual test
 763  * function data_test() is called. Once the test run completed, the
 764  * fixture structure is torn down  by calling data_teardown() and
 765  * after that the memory is released.
 766  * Splitting up a test run into fixture setup, test function and
 767  * fixture teardown is most usful if the same fixture is used for
 768  * multiple tests. In this cases, g_test_create_case() will be
 769  * called with the same fixture, but varying @test_name and
 770  * @data_test arguments.
 771  *
 772  * Returns: a newly allocated #GTestCase.
 773  */
 774 GTestCase*
 775 g_test_create_case (const char     *test_name,
 776                     gsize           data_size,
 777                     gconstpointer   test_data,
 778                     void          (*data_setup) (void),
 779                     void          (*data_test) (void),
 780                     void          (*data_teardown) (void))
 781 {
 782   GTestCase *tc;
 783   g_return_val_if_fail (test_name != NULL, NULL);
 784   g_return_val_if_fail (strchr (test_name, '/') == NULL, NULL);
 785   g_return_val_if_fail (test_name[0] != 0, NULL);
 786   g_return_val_if_fail (data_test != NULL, NULL);
 787   tc = g_slice_new0 (GTestCase);
 788   tc->name = g_strdup (test_name);
 789   tc->test_data = (gpointer) test_data;
 790   tc->fixture_size = data_size;
 791   tc->fixture_setup = (void*) data_setup;
 792   tc->fixture_test = (void*) data_test;
 793   tc->fixture_teardown = (void*) data_teardown;
 794   return tc;
 795 }
 796
 797 void
 798 g_test_add_vtable (const char     *testpath,
 799                    gsize           data_size,
 800                    gconstpointer   test_data,
 801                    void          (*data_setup)    (void),
 802                    void          (*fixture_test_func) (void),
 803                    void          (*data_teardown) (void))
 804 {
 805   gchar **segments;
 806   guint ui;
 807   GTestSuite *suite;
 808
 809   g_return_if_fail (testpath != NULL);
 810   g_return_if_fail (testpath[0] == '/');
 811   g_return_if_fail (fixture_test_func != NULL);
 812
 813   suite = g_test_get_root();
 814   segments = g_strsplit (testpath, "/", -1);
 815   for (ui = 0; segments[ui] != NULL; ui++)
 816     {
 817       const char *seg = segments[ui];
 818       gboolean islast = segments[ui + 1] == NULL;
 819       if (islast && !seg[0])
 820         g_error ("invalid test case path: %s", testpath);
 821       else if (!seg[0])
 822         continue;       /* initial or duplicate slash */
 823       else if (!islast)
 824         {
 825           GTestSuite *csuite = g_test_create_suite (seg);
 826           g_test_suite_add_suite (suite, csuite);
 827           suite = csuite;
 828         }
 829       else /* islast */
 830         {
 831           GTestCase *tc = g_test_create_case (seg, data_size, test_data, data_setup, fixture_test_func, data_teardown);
 832           g_test_suite_add (suite, tc);
 833         }
 834     }
 835   g_strfreev (segments);
 836 }
 837
 838 /**
 839  * g_test_add_func:
 840  * @testpath:   Slash seperated test case path name for the test.
 841  * @test_func:  The test function to invoke for this test.
 842  *
 843  * Create a new test case, similar to g_test_create_case(). However
 844  * the test is assumed to use no fixture, and test suites are automatically
 845  * created on the fly and added to the root fixture, based on the
 846  * slash seperated portions of @testpath.
 847  */
 848 void
 849 g_test_add_func (const char     *testpath,
 850                  void          (*test_func) (void))
 851 {
 852   g_return_if_fail (testpath != NULL);
 853   g_return_if_fail (testpath[0] == '/');
 854   g_return_if_fail (test_func != NULL);
 855   g_test_add_vtable (testpath, 0, NULL, NULL, test_func, NULL);
 856 }
 857
 858 /**
 859  * g_test_add_data_func:
 860  * @testpath:   Slash seperated test case path name for the test.
 861  * @test_data:  Test data argument for the test function.
 862  * @test_func:  The test function to invoke for this test.
 863  *
 864  * Create a new test case, similar to g_test_create_case(). However
 865  * the test is assumed to use no fixture, and test suites are automatically
 866  * created on the fly and added to the root fixture, based on the
 867  * slash seperated portions of @testpath. The @test_data argument
 868  * will be passed as first argument to @test_func.
 869  */
 870 void
 871 g_test_add_data_func (const char     *testpath,
 872                       gconstpointer   test_data,
 873                       void          (*test_func) (gconstpointer))
 874 {
 875   g_return_if_fail (testpath != NULL);
 876   g_return_if_fail (testpath[0] == '/');
 877   g_return_if_fail (test_func != NULL);
 878   g_test_add_vtable (testpath, 0, test_data, NULL, (void(*)(void)) test_func, NULL);
 879 }
 880
 881 /**
 882  * g_test_create_suite:
 883  * @suite_name: a name for the suite
 884  *
 885  * Create a new test suite with the name @suite_name.
 886  *
 887  * Returns: A newly allocated #GTestSuite instance.
 888  */
 889 GTestSuite*
 890 g_test_create_suite (const char *suite_name)
 891 {
 892   GTestSuite *ts;
 893   g_return_val_if_fail (suite_name != NULL, NULL);
 894   g_return_val_if_fail (strchr (suite_name, '/') == NULL, NULL);
 895   g_return_val_if_fail (suite_name[0] != 0, NULL);
 896   ts = g_slice_new0 (GTestSuite);
 897   ts->name = g_strdup (suite_name);
 898   return ts;
 899 }
 900
 901 /**
 902  * g_test_suite_add:
 903  * @suite: a #GTestSuite
 904  * @test_case: a #GTestCase
 905  *
 906  * Adds @test_case to @suite.
 907  */
 908 void
 909 g_test_suite_add (GTestSuite     *suite,
 910                   GTestCase      *test_case)
 911 {
 912   g_return_if_fail (suite != NULL);
 913   g_return_if_fail (test_case != NULL);
 914   suite->cases = g_slist_prepend (suite->cases, test_case);
 915 }
 916
 917 /**
 918  * g_test_suite_add_suite:
 919  * @suite:       a #GTestSuite
 920  * @nestedsuite: another #GTestSuite
 921  *
 922  * Adds @nestedsuite to @suite.
 923  */
 924 void
 925 g_test_suite_add_suite (GTestSuite     *suite,
 926                         GTestSuite     *nestedsuite)
 927 {
 928   g_return_if_fail (suite != NULL);
 929   g_return_if_fail (nestedsuite != NULL);
 930   suite->suites = g_slist_prepend (suite->suites, nestedsuite);
 931 }
 932
 933 /**
 934  * g_test_queue_free:
 935  * @gfree_pointer: the pointer to be stored.
 936  *
 937  * Enqueue a pointer to be released with g_free() during the next
 938  * teardown phase. This is equivalent to calling g_test_queue_destroy()
 939  * with a destroy callback of g_free().
 940  */
 941 void
 942 g_test_queue_free (gpointer gfree_pointer)
 943 {
 944   if (gfree_pointer)
 945     g_test_queue_destroy (g_free, gfree_pointer);
 946 }
 947
 948 /**
 949  * g_test_queue_destroy:
 950  * @destroy_func:       Destroy callback for teardown phase.
 951  * @destroy_data:       Destroy callback data.
 952  *
 953  * This function enqueus a callback @destroy_func() to be executed
 954  * during the next test case teardown phase. This is most useful
 955  * to auto destruct allocted test resources at the end of a test run.
 956  * Resources are released in reverse queue order, that means enqueueing
 957  * callback A before callback B will cause B() to be called before
 958  * A() during teardown.
 959  */
 960 void
 961 g_test_queue_destroy (GDestroyNotify destroy_func,
 962                       gpointer       destroy_data)
 963 {
 964   DestroyEntry *dentry;
 965   g_return_if_fail (destroy_func != NULL);
 966   dentry = g_slice_new0 (DestroyEntry);
 967   dentry->destroy_func = destroy_func;
 968   dentry->destroy_data = destroy_data;
 969   dentry->next = test_destroy_queue;
 970   test_destroy_queue = dentry;
 971 }
 972
 973 static int
 974 test_case_run (GTestCase *tc)
 975 {
 976   gchar *old_name = test_run_name, *old_base = g_strdup (test_uri_base);
 977   test_run_name = g_strconcat (old_name, "/", tc->name, NULL);
 978   if (++test_run_count <= test_skip_count)
 979     g_test_log (G_TEST_LOG_SKIP_CASE, test_run_name, NULL, 0, NULL);
 980   else if (test_run_list)
 981     {
 982       g_print ("%s\n", test_run_name);
 983       g_test_log (G_TEST_LOG_LIST_CASE, test_run_name, NULL, 0, NULL);
 984     }
 985   else
 986     {
 987       GTimer *test_run_timer = g_timer_new();
 988       long double largs[3];
 989       void *fixture;
 990       g_test_log (G_TEST_LOG_START_CASE, test_run_name, NULL, 0, NULL);
 991       test_run_forks = 0;
 992       g_timer_start (test_run_timer);
 993       fixture = tc->fixture_size ? g_malloc0 (tc->fixture_size) : tc->test_data;
 994       test_run_seed (test_run_seedstr);
 995       if (tc->fixture_setup)
 996         tc->fixture_setup (fixture, tc->test_data);
 997       tc->fixture_test (fixture, tc->test_data);
 998       test_trap_clear();
 999       while (test_destroy_queue)
1000         {
1001           DestroyEntry *dentry = test_destroy_queue;
1002           test_destroy_queue = dentry->next;
1003           dentry->destroy_func (dentry->destroy_data);
1004           g_slice_free (DestroyEntry, dentry);
1005         }
1006       if (tc->fixture_teardown)
1007         tc->fixture_teardown (fixture, tc->test_data);
1008       if (tc->fixture_size)
1009         g_free (fixture);
1010       g_timer_stop (test_run_timer);
1011       largs[0] = 0; /* OK */
1012       largs[1] = test_run_forks;
1013       largs[2] = g_timer_elapsed (test_run_timer, NULL);
1014       g_test_log (G_TEST_LOG_STOP_CASE, NULL, NULL, G_N_ELEMENTS (largs), largs);
1015       g_timer_destroy (test_run_timer);
1016     }
1017   g_free (test_run_name);
1018   test_run_name = old_name;
1019   g_free (test_uri_base);
1020   test_uri_base = old_base;
1021   return 0;
1022 }
1023
1024 static int
1025 g_test_run_suite_internal (GTestSuite *suite,
1026                            const char *path)
1027 {
1028   guint n_bad = 0, n_good = 0, bad_suite = 0, l;
1029   gchar *rest, *old_name = test_run_name;
1030   GSList *slist, *reversed;
1031   g_return_val_if_fail (suite != NULL, -1);
1032   while (path[0] == '/')
1033     path++;
1034   l = strlen (path);
1035   rest = strchr (path, '/');
1036   l = rest ? MIN (l, rest - path) : l;
1037   test_run_name = suite->name[0] == 0 ? g_strdup (test_run_name) : g_strconcat (old_name, "/", suite->name, NULL);
1038   reversed = g_slist_reverse (g_slist_copy (suite->cases));
1039   for (slist = reversed; slist; slist = slist->next)
1040     {
1041       GTestCase *tc = slist->data;
1042       guint n = l ? strlen (tc->name) : 0;
1043       if (l == n && strncmp (path, tc->name, n) == 0)
1044         {
1045           n_good++;
1046           n_bad += test_case_run (tc) != 0;
1047         }
1048     }
1049   g_slist_free (reversed);
1050   reversed = g_slist_reverse (g_slist_copy (suite->suites));
1051   for (slist = reversed; slist; slist = slist->next)
1052     {
1053       GTestSuite *ts = slist->data;
1054       guint n = l ? strlen (ts->name) : 0;
1055       if (l == n && strncmp (path, ts->name, n) == 0)
1056         bad_suite += g_test_run_suite_internal (ts, rest ? rest : "") != 0;
1057     }
1058   g_slist_free (reversed);
1059   g_free (test_run_name);
1060   test_run_name = old_name;
1061   return n_bad || bad_suite;
1062 }
1063
1064 /**
1065  * g_test_run_suite:
1066  * @suite: a #GTestSuite
1067  *
1068  * Execute the tests within @suite and all nested #GTestSuites.
1069  * The test suites to be executed are filtered according to
1070  * test path arguments (-p <testpath>) as parsed by g_test_init().
1071  * g_test_run_suite() or g_test_run() may only be called once
1072  * in a program.
1073  *
1074  * Returns: 0 on success
1075  */
1076 int
1077 g_test_run_suite (GTestSuite *suite)
1078 {
1079   guint n_bad = 0;
1080   g_return_val_if_fail (g_test_config_vars->test_initialized, -1);
1081   g_return_val_if_fail (g_test_run_once == TRUE, -1);
1082   g_test_run_once = FALSE;
1083   if (!test_paths)
1084     test_paths = g_slist_prepend (test_paths, "");
1085   while (test_paths)
1086     {
1087       const char *rest, *path = test_paths->data;
1088       guint l, n = strlen (suite->name);
1089       test_paths = g_slist_delete_link (test_paths, test_paths);
1090       while (path[0] == '/')
1091         path++;
1092       if (!n) /* root suite, run unconditionally */
1093         {
1094           n_bad += 0 != g_test_run_suite_internal (suite, path);
1095           continue;
1096         }
1097       /* regular suite, match path */
1098       rest = strchr (path, '/');
1099       l = strlen (path);
1100       l = rest ? MIN (l, rest - path) : l;
1101       if ((!l || l == n) && strncmp (path, suite->name, n) == 0)
1102         n_bad += 0 != g_test_run_suite_internal (suite, rest ? rest : "");
1103     }
1104   return n_bad;
1105 }
1106
1107 static void
1108 gtest_default_log_handler (const gchar    *log_domain,
1109                            GLogLevelFlags  log_level,
1110                            const gchar    *message,
1111                            gpointer        unused_data)
1112 {
1113   const gchar *strv[16];
1114   gchar *msg;
1115   guint i = 0;
1116   if (log_domain)
1117     {
1118       strv[i++] = log_domain;
1119       strv[i++] = "-";
1120     }
1121   if (log_level & G_LOG_FLAG_FATAL)
1122     strv[i++] = "FATAL-";
1123   if (log_level & G_LOG_FLAG_RECURSION)
1124     strv[i++] = "RECURSIVE-";
1125   if (log_level & G_LOG_LEVEL_ERROR)
1126     strv[i++] = "ERROR";
1127   if (log_level & G_LOG_LEVEL_CRITICAL)
1128     strv[i++] = "CRITICAL";
1129   if (log_level & G_LOG_LEVEL_WARNING)
1130     strv[i++] = "WARNING";
1131   if (log_level & G_LOG_LEVEL_MESSAGE)
1132     strv[i++] = "MESSAGE";
1133   if (log_level & G_LOG_LEVEL_INFO)
1134     strv[i++] = "INFO";
1135   if (log_level & G_LOG_LEVEL_DEBUG)
1136     strv[i++] = "DEBUG";
1137   strv[i++] = ": ";
1138   strv[i++] = message;
1139   strv[i++] = NULL;
1140   msg = g_strjoinv ("", (gchar**) strv);
1141   g_test_log (G_TEST_LOG_ERROR, msg, NULL, 0, NULL);
1142   g_log_default_handler (log_domain, log_level, message, unused_data);
1143   g_free (msg);
1144 }
1145
1146 void
1147 g_assertion_message (const char     *domain,
1148                      const char     *file,
1149                      int             line,
1150                      const char     *func,
1151                      const char     *message)
1152 {
1153   char lstr[32];
1154   char *s;
1155   if (!message)
1156     message = "code should not be reached";
1157   g_snprintf (lstr, 32, "%d", line);
1158   s = g_strconcat (domain ? domain : "", domain && domain[0] ? ":" : "",
1159                    "ERROR:(", file, ":", lstr, "):",
1160                    func, func[0] ? ":" : "",
1161                    " ", message, NULL);
1162   g_printerr ("**\n** %s\n", s);
1163   g_test_log (G_TEST_LOG_ERROR, s, NULL, 0, NULL);
1164   g_free (s);
1165   abort();
1166 }
1167
1168 void
1169 g_assertion_message_expr (const char     *domain,
1170                           const char     *file,
1171                           int             line,
1172                           const char     *func,
1173                           const char     *expr)
1174 {
1175   char *s = g_strconcat ("assertion failed: (", expr, ")", NULL);
1176   g_assertion_message (domain, file, line, func, s);
1177   g_free (s);
1178 }
1179
1180 void
1181 g_assertion_message_cmpnum (const char     *domain,
1182                             const char     *file,
1183                             int             line,
1184                             const char     *func,
1185                             const char     *expr,
1186                             long double     arg1,
1187                             const char     *cmp,
1188                             long double     arg2,
1189                             char            numtype)
1190 {
1191   char *s = NULL;
1192   switch (numtype)
1193     {
1194     case 'i':   s = g_strdup_printf ("assertion failed (%s): (%.0Lf %s %.0Lf)", expr, arg1, cmp, arg2); break;
1195     case 'x':   s = g_strdup_printf ("assertion failed (%s): (0x%08" G_GUINT64_FORMAT " %s 0x%08" G_GUINT64_FORMAT ")", expr, (guint64) arg1, cmp, (guint64) arg2); break;
1196     case 'f':   s = g_strdup_printf ("assertion failed (%s): (%.9Lg %s %.9Lg)", expr, arg1, cmp, arg2); break;
1197       /* ideally use: floats=%.7g double=%.17g */
1198     }
1199   g_assertion_message (domain, file, line, func, s);
1200   g_free (s);
1201 }
1202
1203 void
1204 g_assertion_message_cmpstr (const char     *domain,
1205                             const char     *file,
1206                             int             line,
1207                             const char     *func,
1208                             const char     *expr,
1209                             const char     *arg1,
1210                             const char     *cmp,
1211                             const char     *arg2)
1212 {
1213   char *a1, *a2, *s, *t1 = NULL, *t2 = NULL;
1214   a1 = arg1 ? g_strconcat ("\"", t1 = g_strescape (arg1, NULL), "\"", NULL) : g_strdup ("NULL");
1215   a2 = arg2 ? g_strconcat ("\"", t2 = g_strescape (arg2, NULL), "\"", NULL) : g_strdup ("NULL");
1216   g_free (t1);
1217   g_free (t2);
1218   s = g_strdup_printf ("assertion failed (%s): (%s %s %s)", expr, a1, cmp, a2);
1219   g_free (a1);
1220   g_free (a2);
1221   g_assertion_message (domain, file, line, func, s);
1222   g_free (s);
1223 }
1224
1225 /**
1226  * g_strcmp0:
1227  * @str1: a C string or %NULL
1228  * @str2: another C string or %NULL
1229  *
1230  * Compares @str1 and @str2 like strcmp(). Handles %NULL strings gracefully.
1231  */
1232 int
1233 g_strcmp0 (const char     *str1,
1234            const char     *str2)
1235 {
1236   if (!str1)
1237     return -(str1 != str2);
1238   if (!str2)
1239     return str1 != str2;
1240   return strcmp (str1, str2);
1241 }
1242
1243 #ifdef G_OS_UNIX
1244 static int /* 0 on success */
1245 kill_child (int  pid,
1246             int *status,
1247             int  patience)
1248 {
1249   int wr;
1250   if (patience >= 3)    /* try graceful reap */
1251     {
1252       if (waitpid (pid, status, WNOHANG) > 0)
1253         return 0;
1254     }
1255   if (patience >= 2)    /* try SIGHUP */
1256     {
1257       kill (pid, SIGHUP);
1258       if (waitpid (pid, status, WNOHANG) > 0)
1259         return 0;
1260       g_usleep (20 * 1000); /* give it some scheduling/shutdown time */
1261       if (waitpid (pid, status, WNOHANG) > 0)
1262         return 0;
1263       g_usleep (50 * 1000); /* give it some scheduling/shutdown time */
1264       if (waitpid (pid, status, WNOHANG) > 0)
1265         return 0;
1266       g_usleep (100 * 1000); /* give it some scheduling/shutdown time */
1267       if (waitpid (pid, status, WNOHANG) > 0)
1268         return 0;
1269     }
1270   if (patience >= 1)    /* try SIGTERM */
1271     {
1272       kill (pid, SIGTERM);
1273       if (waitpid (pid, status, WNOHANG) > 0)
1274         return 0;
1275       g_usleep (200 * 1000); /* give it some scheduling/shutdown time */
1276       if (waitpid (pid, status, WNOHANG) > 0)
1277         return 0;
1278       g_usleep (400 * 1000); /* give it some scheduling/shutdown time */
1279       if (waitpid (pid, status, WNOHANG) > 0)
1280         return 0;
1281     }
1282   /* finish it off */
1283   kill (pid, SIGKILL);
1284   do
1285     wr = waitpid (pid, status, 0);
1286   while (wr < 0 && errno == EINTR);
1287   return wr;
1288 }
1289 #endif
1290
1291 static inline int
1292 g_string_must_read (GString *gstring,
1293                     int      fd)
1294 {
1295 #define STRING_BUFFER_SIZE     4096
1296   char buf[STRING_BUFFER_SIZE];
1297   gssize bytes;
1298  again:
1299   bytes = read (fd, buf, sizeof (buf));
1300   if (bytes == 0)
1301     return 0; /* EOF, calling this function assumes data is available */
1302   else if (bytes > 0)
1303     {
1304       g_string_append_len (gstring, buf, bytes);
1305       return 1;
1306     }
1307   else if (bytes < 0 && errno == EINTR)
1308     goto again;
1309   else /* bytes < 0 */
1310     {
1311       g_warning ("failed to read() from child process (%d): %s", test_trap_last_pid, g_strerror (errno));
1312       return 1; /* ignore error after warning */
1313     }
1314 }
1315
1316 static inline void
1317 g_string_write_out (GString *gstring,
1318                     int      outfd,
1319                     int     *stringpos)
1320 {
1321   if (*stringpos < gstring->len)
1322     {
1323       int r;
1324       do
1325         r = write (outfd, gstring->str + *stringpos, gstring->len - *stringpos);
1326       while (r < 0 && errno == EINTR);
1327       *stringpos += MAX (r, 0);
1328     }
1329 }
1330
1331 static int
1332 sane_dup2 (int fd1,
1333            int fd2)
1334 {
1335   int ret;
1336   do
1337     ret = dup2 (fd1, fd2);
1338   while (ret < 0 && errno == EINTR);
1339   return ret;
1340 }
1341
1342 static void
1343 test_trap_clear (void)
1344 {
1345   test_trap_last_status = 0;
1346   test_trap_last_pid = 0;
1347   g_free (test_trap_last_stdout);
1348   test_trap_last_stdout = NULL;
1349   g_free (test_trap_last_stderr);
1350   test_trap_last_stderr = NULL;
1351 }
1352
1353 static guint64
1354 test_time_stamp (void)
1355 {
1356   GTimeVal tv;
1357   guint64 stamp;
1358   g_get_current_time (&tv);
1359   stamp = tv.tv_sec;
1360   stamp = stamp * 1000000 + tv.tv_usec;
1361   return stamp;
1362 }
1363
1364 /**
1365  * g_test_trap_fork:
1366  * @usec_timeout:    Timeout for the forked test in micro seconds.
1367  * @test_trap_flags: Flags to modify forking behaviour.
1368  *
1369  * Fork the current test program to execute a test case that might
1370  * not return or that might abort. The forked test case is aborted
1371  * and considered failing if its run time exceeds @usec_timeout.
1372  * The forking behavior can be configured with the following flags:
1373  * %G_TEST_TRAP_SILENCE_STDOUT - redirect stdout of the test child
1374  * to /dev/null so it cannot be observed on the console during test
1375  * runs. The actual output is still captured though to allow later
1376  * tests with g_test_trap_assert_stdout().
1377  * %G_TEST_TRAP_SILENCE_STDERR - redirect stderr of the test child
1378  * to /dev/null so it cannot be observed on the console during test
1379  * runs. The actual output is still captured though to allow later
1380  * tests with g_test_trap_assert_stderr().
1381  * %G_TEST_TRAP_INHERIT_STDIN - if this flag is given, stdin of the
1382  * forked child process is shared with stdin of its parent process.
1383  * It is redirected to /dev/null otherwise.
1384  *
1385  * In the following example, the test code forks, the forked child
1386  * process produces some sample output and exits successfully.
1387  * The forking parent process then asserts successfull child program
1388  * termination and validates cihld program outputs.
1389  *
1390  * |[
1391  *   static void
1392  *   test_fork_patterns (void)
1393  *   {
1394  *     if (g_test_trap_fork (0, G_TEST_TRAP_SILENCE_STDOUT | G_TEST_TRAP_SILENCE_STDERR))
1395  *       {
1396  *         g_print ("some stdout text: somagic17\n");
1397  *         g_printerr ("some stderr text: semagic43\n");
1398  *         exit (0); /&ast; successful test run &ast;/
1399  *       }
1400  *     g_test_trap_assert_passed();
1401  *     g_test_trap_assert_stdout ("*somagic17*");
1402  *     g_test_trap_assert_stderr ("*semagic43*");
1403  *   }
1404  * ]|
1405  *
1406  * Returns: %TRUE for the forked child and %FALSE for the executing parent process.
1407  */
1408 gboolean
1409 g_test_trap_fork (guint64        usec_timeout,
1410                   GTestTrapFlags test_trap_flags)
1411 {
1412 #ifdef G_OS_UNIX
1413   gboolean pass_on_forked_log = FALSE;
1414   int stdout_pipe[2] = { -1, -1 };
1415   int stderr_pipe[2] = { -1, -1 };
1416   int stdtst_pipe[2] = { -1, -1 };
1417   test_trap_clear();
1418   if (pipe (stdout_pipe) < 0 || pipe (stderr_pipe) < 0 || pipe (stdtst_pipe) < 0)
1419     g_error ("failed to create pipes to fork test program: %s", g_strerror (errno));
1420   signal (SIGCHLD, SIG_DFL);
1421   test_trap_last_pid = fork ();
1422   if (test_trap_last_pid < 0)
1423     g_error ("failed to fork test program: %s", g_strerror (errno));
1424   if (test_trap_last_pid == 0)  /* child */
1425     {
1426       int fd0 = -1;
1427       close (stdout_pipe[0]);
1428       close (stderr_pipe[0]);
1429       close (stdtst_pipe[0]);
1430       if (!(test_trap_flags & G_TEST_TRAP_INHERIT_STDIN))
1431         fd0 = open ("/dev/null", O_RDONLY);
1432       if (sane_dup2 (stdout_pipe[1], 1) < 0 || sane_dup2 (stderr_pipe[1], 2) < 0 || (fd0 >= 0 && sane_dup2 (fd0, 0) < 0))
1433         g_error ("failed to dup2() in forked test program: %s", g_strerror (errno));
1434       if (fd0 >= 3)
1435         close (fd0);
1436       if (stdout_pipe[1] >= 3)
1437         close (stdout_pipe[1]);
1438       if (stderr_pipe[1] >= 3)
1439         close (stderr_pipe[1]);
1440       test_log_fd = stdtst_pipe[1];
1441       return TRUE;
1442     }
1443   else                          /* parent */
1444     {
1445       GString *sout = g_string_new (NULL);
1446       GString *serr = g_string_new (NULL);
1447       guint64 sstamp;
1448       int soutpos = 0, serrpos = 0, wr, need_wait = TRUE;
1449       test_run_forks++;
1450       close (stdout_pipe[1]);
1451       close (stderr_pipe[1]);
1452       close (stdtst_pipe[1]);
1453       sstamp = test_time_stamp();
1454       /* read data until we get EOF on all pipes */
1455       while (stdout_pipe[0] >= 0 || stderr_pipe[0] >= 0 || stdtst_pipe[0] > 0)
1456         {
1457           fd_set fds;
1458           struct timeval tv;
1459           FD_ZERO (&fds);
1460           if (stdout_pipe[0] >= 0)
1461             FD_SET (stdout_pipe[0], &fds);
1462           if (stderr_pipe[0] >= 0)
1463             FD_SET (stderr_pipe[0], &fds);
1464           if (stdtst_pipe[0] >= 0)
1465             FD_SET (stdtst_pipe[0], &fds);
1466           tv.tv_sec = 0;
1467           tv.tv_usec = MIN (usec_timeout ? usec_timeout : 1000000, 100 * 1000); /* sleep at most 0.5 seconds to catch clock skews, etc. */
1468           int ret = select (MAX (MAX (stdout_pipe[0], stderr_pipe[0]), stdtst_pipe[0]) + 1, &fds, NULL, NULL, &tv);
1469           if (ret < 0 && errno != EINTR)
1470             {
1471               g_warning ("Unexpected error in select() while reading from child process (%d): %s", test_trap_last_pid, g_strerror (errno));
1472               break;
1473             }
1474           if (stdout_pipe[0] >= 0 && FD_ISSET (stdout_pipe[0], &fds) &&
1475               g_string_must_read (sout, stdout_pipe[0]) == 0)
1476             {
1477               close (stdout_pipe[0]);
1478               stdout_pipe[0] = -1;
1479             }
1480           if (stderr_pipe[0] >= 0 && FD_ISSET (stderr_pipe[0], &fds) &&
1481               g_string_must_read (serr, stderr_pipe[0]) == 0)
1482             {
1483               close (stderr_pipe[0]);
1484               stderr_pipe[0] = -1;
1485             }
1486           if (stdtst_pipe[0] >= 0 && FD_ISSET (stdtst_pipe[0], &fds))
1487             {
1488               guint8 buffer[4096];
1489               gint l, r = read (stdtst_pipe[0], buffer, sizeof (buffer));
1490               if (r > 0 && test_log_fd > 0)
1491                 do
1492                   l = write (pass_on_forked_log ? test_log_fd : -1, buffer, r);
1493                 while (l < 0 && errno == EINTR);
1494               if (r == 0 || (r < 0 && errno != EINTR && errno != EAGAIN))
1495                 {
1496                   close (stdtst_pipe[0]);
1497                   stdtst_pipe[0] = -1;
1498                 }
1499             }
1500           if (!(test_trap_flags & G_TEST_TRAP_SILENCE_STDOUT))
1501             g_string_write_out (sout, 1, &soutpos);
1502           if (!(test_trap_flags & G_TEST_TRAP_SILENCE_STDERR))
1503             g_string_write_out (serr, 2, &serrpos);
1504           if (usec_timeout)
1505             {
1506               guint64 nstamp = test_time_stamp();
1507               int status = 0;
1508               sstamp = MIN (sstamp, nstamp); /* guard against backwards clock skews */
1509               if (usec_timeout < nstamp - sstamp)
1510                 {
1511                   /* timeout reached, need to abort the child now */
1512                   kill_child (test_trap_last_pid, &status, 3);
1513                   test_trap_last_status = 1024; /* timeout */
1514                   if (0 && WIFSIGNALED (status))
1515                     g_printerr ("%s: child timed out and received: %s\n", G_STRFUNC, g_strsignal (WTERMSIG (status)));
1516                   need_wait = FALSE;
1517                   break;
1518                 }
1519             }
1520         }
1521       close (stdout_pipe[0]);
1522       close (stderr_pipe[0]);
1523       close (stdtst_pipe[0]);
1524       if (need_wait)
1525         {
1526           int status = 0;
1527           do
1528             wr = waitpid (test_trap_last_pid, &status, 0);
1529           while (wr < 0 && errno == EINTR);
1530           if (WIFEXITED (status)) /* normal exit */
1531             test_trap_last_status = WEXITSTATUS (status); /* 0..255 */
1532           else if (WIFSIGNALED (status))
1533             test_trap_last_status = (WTERMSIG (status) << 12); /* signalled */
1534           else /* WCOREDUMP (status) */
1535             test_trap_last_status = 512; /* coredump */
1536         }
1537       test_trap_last_stdout = g_string_free (sout, FALSE);
1538       test_trap_last_stderr = g_string_free (serr, FALSE);
1539       return FALSE;
1540     }
1541 #else
1542   g_error ("Not implemented: g_test_trap_fork");
1543 #endif
1544 }
1545
1546 /**
1547  * g_test_trap_has_passed:
1548  *
1549  * Check the reuslt of the last g_test_trap_fork() call.
1550  *
1551  * Returns: %TRUE if the last forked child terminated successfully.
1552  */
1553 gboolean
1554 g_test_trap_has_passed (void)
1555 {
1556   return test_trap_last_status == 0; /* exit_status == 0 && !signal && !coredump */
1557 }
1558
1559 /**
1560  * g_test_trap_reached_timeout:
1561  *
1562  * Check the reuslt of the last g_test_trap_fork() call.
1563  *
1564  * Returns: %TRUE if the last forked child got killed due to a fork timeout.
1565  */
1566 gboolean
1567 g_test_trap_reached_timeout (void)
1568 {
1569   return 0 != (test_trap_last_status & 1024); /* timeout flag */
1570 }
1571
1572 void
1573 g_test_trap_assertions (const char     *domain,
1574                         const char     *file,
1575                         int             line,
1576                         const char     *func,
1577                         guint64         assertion_flags, /* 0-pass, 1-fail, 2-outpattern, 4-errpattern */
1578                         const char     *pattern)
1579 {
1580   gboolean must_pass = assertion_flags == 0;
1581   gboolean must_fail = assertion_flags == 1;
1582   gboolean match_result = 0 == (assertion_flags & 1);
1583   const char *stdout_pattern = (assertion_flags & 2) ? pattern : NULL;
1584   const char *stderr_pattern = (assertion_flags & 4) ? pattern : NULL;
1585   const char *match_error = match_result ? "failed to match" : "contains invalid match";
1586   if (test_trap_last_pid == 0)
1587     g_error ("child process failed to exit after g_test_trap_fork() and before g_test_trap_assert*()");
1588   if (must_pass && !g_test_trap_has_passed())
1589     {
1590       char *msg = g_strdup_printf ("child process (%d) of test trap failed unexpectedly", test_trap_last_pid);
1591       g_assertion_message (domain, file, line, func, msg);
1592       g_free (msg);
1593     }
1594   if (must_fail && g_test_trap_has_passed())
1595     {
1596       char *msg = g_strdup_printf ("child process (%d) did not fail as expected", test_trap_last_pid);
1597       g_assertion_message (domain, file, line, func, msg);
1598       g_free (msg);
1599     }
1600   if (stdout_pattern && match_result == !g_pattern_match_simple (stdout_pattern, test_trap_last_stdout))
1601     {
1602       char *msg = g_strdup_printf ("stdout of child process (%d) %s: %s", test_trap_last_pid, match_error, stdout_pattern);
1603       g_assertion_message (domain, file, line, func, msg);
1604       g_free (msg);
1605     }
1606   if (stderr_pattern && match_result == !g_pattern_match_simple (stderr_pattern, test_trap_last_stderr))
1607     {
1608       char *msg = g_strdup_printf ("stderr of child process (%d) %s: %s", test_trap_last_pid, match_error, stderr_pattern);
1609       g_assertion_message (domain, file, line, func, msg);
1610       g_free (msg);
1611     }
1612 }
1613
1614 static void
1615 gstring_overwrite_int (GString *gstring,
1616                        guint    pos,
1617                        guint32  vuint)
1618 {
1619   vuint = g_htonl (vuint);
1620   g_string_overwrite_len (gstring, pos, (const gchar*) &vuint, 4);
1621 }
1622
1623 static void
1624 gstring_append_int (GString *gstring,
1625                     guint32  vuint)
1626 {
1627   vuint = g_htonl (vuint);
1628   g_string_append_len (gstring, (const gchar*) &vuint, 4);
1629 }
1630
1631 static void
1632 gstring_append_double (GString *gstring,
1633                        double   vdouble)
1634 {
1635   union { double vdouble; guint64 vuint64; } u;
1636   u.vdouble = vdouble;
1637   u.vuint64 = GUINT64_TO_BE (u.vuint64);
1638   g_string_append_len (gstring, (const gchar*) &u.vuint64, 8);
1639 }
1640
1641 static guint8*
1642 g_test_log_dump (GTestLogMsg *msg,
1643                  guint       *len)
1644 {
1645   GString *gstring = g_string_sized_new (1024);
1646   guint ui;
1647   gstring_append_int (gstring, 0);              /* message length */
1648   gstring_append_int (gstring, msg->log_type);
1649   gstring_append_int (gstring, msg->n_strings);
1650   gstring_append_int (gstring, msg->n_nums);
1651   gstring_append_int (gstring, 0);      /* reserved */
1652   for (ui = 0; ui < msg->n_strings; ui++)
1653     {
1654       guint l = strlen (msg->strings[ui]);
1655       gstring_append_int (gstring, l);
1656       g_string_append_len (gstring, msg->strings[ui], l);
1657     }
1658   for (ui = 0; ui < msg->n_nums; ui++)
1659     gstring_append_double (gstring, msg->nums[ui]);
1660   *len = gstring->len;
1661   gstring_overwrite_int (gstring, 0, *len);     /* message length */
1662   return (guint8*) g_string_free (gstring, FALSE);
1663 }
1664
1665 static inline long double
1666 net_double (const gchar **ipointer)
1667 {
1668   union { guint64 vuint64; double vdouble; } u;
1669   guint64 aligned_int64;
1670   memcpy (&aligned_int64, *ipointer, 8);
1671   *ipointer += 8;
1672   u.vuint64 = GUINT64_FROM_BE (aligned_int64);
1673   return u.vdouble;
1674 }
1675
1676 static inline guint32
1677 net_int (const gchar **ipointer)
1678 {
1679   guint32 aligned_int;
1680   memcpy (&aligned_int, *ipointer, 4);
1681   *ipointer += 4;
1682   return g_ntohl (aligned_int);
1683 }
1684
1685 static gboolean
1686 g_test_log_extract (GTestLogBuffer *tbuffer)
1687 {
1688   const gchar *p = tbuffer->data->str;
1689   GTestLogMsg msg;
1690   guint mlength;
1691   if (tbuffer->data->len < 4 * 5)
1692     return FALSE;
1693   mlength = net_int (&p);
1694   if (tbuffer->data->len < mlength)
1695     return FALSE;
1696   msg.log_type = net_int (&p);
1697   msg.n_strings = net_int (&p);
1698   msg.n_nums = net_int (&p);
1699   if (net_int (&p) == 0)
1700     {
1701       guint ui;
1702       msg.strings = g_new0 (gchar*, msg.n_strings + 1);
1703       msg.nums = g_new0 (long double, msg.n_nums);
1704       for (ui = 0; ui < msg.n_strings; ui++)
1705         {
1706           guint sl = net_int (&p);
1707           msg.strings[ui] = g_strndup (p, sl);
1708           p += sl;
1709         }
1710       for (ui = 0; ui < msg.n_nums; ui++)
1711         msg.nums[ui] = net_double (&p);
1712       if (p <= tbuffer->data->str + mlength)
1713         {
1714           g_string_erase (tbuffer->data, 0, mlength);
1715           tbuffer->msgs = g_slist_prepend (tbuffer->msgs, g_memdup (&msg, sizeof (msg)));
1716           return TRUE;
1717         }
1718     }
1719   g_free (msg.nums);
1720   g_strfreev (msg.strings);
1721   g_error ("corrupt log stream from test program");
1722   return FALSE;
1723 }
1724
1725 /**
1726  * g_test_log_buffer_new:
1727  *
1728  * Internal function for gtester to decode test log messages, no ABI guarantees provided.
1729  */
1730 GTestLogBuffer*
1731 g_test_log_buffer_new (void)
1732 {
1733   GTestLogBuffer *tb = g_new0 (GTestLogBuffer, 1);
1734   tb->data = g_string_sized_new (1024);
1735   return tb;
1736 }
1737
1738 /**
1739  * g_test_log_buffer_free
1740  *
1741  * Internal function for gtester to free test log messages, no ABI guarantees provided.
1742  */
1743 void
1744 g_test_log_buffer_free (GTestLogBuffer *tbuffer)
1745 {
1746   g_return_if_fail (tbuffer != NULL);
1747   while (tbuffer->msgs)
1748     g_test_log_msg_free (g_test_log_buffer_pop (tbuffer));
1749   g_string_free (tbuffer->data, TRUE);
1750   g_free (tbuffer);
1751 }
1752
1753 /**
1754  * g_test_log_buffer_push
1755  *
1756  * Internal function for gtester to decode test log messages, no ABI guarantees provided.
1757  */
1758 void
1759 g_test_log_buffer_push (GTestLogBuffer *tbuffer,
1760                         guint           n_bytes,
1761                         const guint8   *bytes)
1762 {
1763   g_return_if_fail (tbuffer != NULL);
1764   if (n_bytes)
1765     {
1766       gboolean more_messages;
1767       g_return_if_fail (bytes != NULL);
1768       g_string_append_len (tbuffer->data, (const gchar*) bytes, n_bytes);
1769       do
1770         more_messages = g_test_log_extract (tbuffer);
1771       while (more_messages);
1772     }
1773 }
1774
1775 /**
1776  * g_test_log_buffer_pop:
1777  *
1778  * Internal function for gtester to retrieve test log messages, no ABI guarantees provided.
1779  */
1780 GTestLogMsg*
1781 g_test_log_buffer_pop (GTestLogBuffer *tbuffer)
1782 {
1783   GTestLogMsg *msg = NULL;
1784   g_return_val_if_fail (tbuffer != NULL, NULL);
1785   if (tbuffer->msgs)
1786     {
1787       GSList *slist = g_slist_last (tbuffer->msgs);
1788       msg = slist->data;
1789       tbuffer->msgs = g_slist_delete_link (tbuffer->msgs, slist);
1790     }
1791   return msg;
1792 }
1793
1794 /**
1795  * g_test_log_msg_free:
1796  *
1797  * Internal function for gtester to free test log messages, no ABI guarantees provided.
1798  */
1799 void
1800 g_test_log_msg_free (GTestLogMsg *tmsg)
1801 {
1802   g_return_if_fail (tmsg != NULL);
1803   g_strfreev (tmsg->strings);
1804   g_free (tmsg->nums);
1805   g_free (tmsg);
1806 }
1807
1808 /* --- macros docs START --- */
1809 /**
1810  * g_test_add:
1811  * @testpath:  The test path for a new test case.
1812  * @Fixture:   The type of a fixture data structure.
1813  * @fsetup:    The function to set up the fixture data.
1814  * @ftest:     The actual test function.
1815  * @fteardown: The function to tear down the fixture data.
1816  *
1817  * Hook up a new test case at @testpath, similar to g_test_add_func().
1818  * A fixture data structure with setup and teardown function may be provided
1819  * though, simmilar to g_test_create_case().
1820  * g_test_add() is implemented as a macro, so that the fsetup(), ftest() and
1821  * fteardown() callbacks can expect a @Fixture pointer as first argument in
1822  * a type safe manner.
1823  **/
1824 /* --- macros docs END --- */
1825
1826 #define __G_TEST_UTILS_C__
1827 #include "galiasdef.c"