tests/child-test.c

   1 /* GLIB - Library of useful routines for C programming
   2  * Copyright (C) 1995-1997  Peter Mattis, Spencer Kimball and Josh MacDonald
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  */
  19
  20 /*
  21  * Modified by the GLib Team and others 1997-2000.  See the AUTHORS
  22  * file for a list of people on the GLib Team.  See the ChangeLog
  23  * files for a list of changes.  These files are distributed with
  24  * GLib at ftp://ftp.gtk.org/pub/gtk/.
  25  */
  26
  27 #include "config.h"
  28
  29 #include <sys/types.h>
  30 #ifdef HAVE_UNISTD_H
  31 #include <unistd.h>
  32 #endif
  33 #include <stdlib.h>
  34
  35 #include <glib.h>
  36
  37 #ifdef G_OS_WIN32
  38 #include <windows.h>
  39 #endif
  40
  41 #ifdef G_OS_WIN32
  42 #define GPID_FORMAT "%p"
  43 #else
  44 #define GPID_FORMAT "%d"
  45 #endif
  46
  47 GMainLoop *main_loop;
  48 gint alive;
  49
  50 #ifdef G_OS_WIN32
  51 char *argv0;
  52 #endif
  53
  54 GPid
  55 get_a_child (gint ttl)
  56 {
  57   GPid pid;
  58
  59 #ifdef G_OS_WIN32
  60   STARTUPINFO si;
  61   PROCESS_INFORMATION pi;
  62   gchar *cmdline;
  63
  64   memset (&si, 0, sizeof (si));
  65   si.cb = sizeof (&si);
  66   memset (&pi, 0, sizeof (pi));
  67
  68   cmdline = g_strdup_printf( "child-test -c%d", ttl);
  69
  70   if (!CreateProcess (argv0, cmdline, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi))
  71     g_error ("CreateProcess failed: %s\n", g_win32_error_message (GetLastError ()));
  72
  73   g_free(cmdline);
  74
  75   CloseHandle (pi.hThread);
  76   pid = pi.hProcess;
  77
  78   return pid;
  79 #else
  80   pid = fork ();
  81   if (pid < 0)
  82     exit (1);
  83
  84   if (pid > 0)
  85     return pid;
  86
  87   sleep (ttl);
  88   _exit (0);
  89 #endif /* G_OS_WIN32 */
  90 }
  91
  92 gboolean
  93 child_watch_callback (GPid pid, gint status, gpointer data)
  94 {
  95 #ifdef VERBOSE
  96   gint ttl = GPOINTER_TO_INT (data);
  97
  98   g_print ("child " GPID_FORMAT " (ttl %d) exited, status %d\n", pid, ttl, status);
  99 #endif
 100
 101   g_spawn_close_pid (pid);
 102
 103   if (--alive == 0)
 104     g_main_loop_quit (main_loop);
 105
 106   return TRUE;
 107 }
 108
 109 static gboolean
 110 quit_loop (gpointer data)
 111 {
 112   GMainLoop *main_loop = data;
 113
 114   g_main_loop_quit (main_loop);
 115
 116   return TRUE;
 117 }
 118
 119 #ifdef TEST_THREAD
 120 static gpointer
 121 test_thread (gpointer data)
 122 {
 123   GMainLoop *new_main_loop;
 124   GSource *source;
 125   GPid pid;
 126   gint ttl = GPOINTER_TO_INT (data);
 127
 128   new_main_loop = g_main_loop_new (NULL, FALSE);
 129
 130   pid = get_a_child (ttl);
 131   source = g_child_watch_source_new (pid);
 132   g_source_set_callback (source, (GSourceFunc) child_watch_callback, data, NULL);
 133   g_source_attach (source, g_main_loop_get_context (new_main_loop));
 134   g_source_unref (source);
 135
 136 #ifdef VERBOSE
 137   g_print ("whee! created pid: " GPID_FORMAT " (ttl %d)\n", pid, ttl);
 138 #endif
 139
 140   g_main_loop_run (new_main_loop);
 141
 142   return NULL;
 143 }
 144 #endif
 145
 146 int
 147 main (int argc, char *argv[])
 148 {
 149 #ifndef TEST_THREAD
 150   GPid pid;
 151 #endif
 152 #ifdef G_OS_WIN32
 153   argv0 = argv[0];
 154   if (argc > 1 && argv[1][0] == '-' && argv[1][1] == 'c')
 155     {
 156       int ttl = atoi (argv[1] + 2);
 157       Sleep (ttl * 1000);
 158       /* Exit on purpose with STILL_ACTIVE (which isn't a very common
 159        * exit status) to verify that g_child_watch_check() in gmain.c
 160        * doesn't believe a child still to be active if it happens to
 161        * exit with that status.
 162        */
 163       exit (STILL_ACTIVE);
 164     }
 165 #endif
 166
 167 #ifdef TEST_THREAD
 168   g_thread_init (NULL);
 169 #endif
 170   main_loop = g_main_loop_new (NULL, FALSE);
 171
 172 #ifdef G_OS_WIN32
 173   system ("ipconfig /all");
 174 #else
 175   system ("true");
 176 #endif
 177
 178   alive = 2;
 179   g_timeout_add (30000, quit_loop, main_loop);
 180
 181 #ifdef TEST_THREAD
 182   g_thread_create (test_thread, GINT_TO_POINTER (10), FALSE, NULL);
 183   g_thread_create (test_thread, GINT_TO_POINTER (20), FALSE, NULL);
 184 #else
 185   pid = get_a_child (10);
 186   g_child_watch_add (pid, (GChildWatchFunc) child_watch_callback,
 187                      GINT_TO_POINTER (10));
 188   pid = get_a_child (20);
 189   g_child_watch_add (pid, (GChildWatchFunc) child_watch_callback,
 190                      GINT_TO_POINTER (20));
 191 #endif
 192
 193   g_main_loop_run (main_loop);
 194
 195   if (alive > 0)
 196     {
 197       g_warning ("%d children still alive\n", alive);
 198       return 1;
 199     }
 200
 201    return 0;
 202 }