glib/tests/child-test.c
2014-01-31 14:31:55 +01:00

199 lines
4.3 KiB
C

/* GLIB - Library of useful routines for C programming
* Copyright (C) 1995-1997 Peter Mattis, Spencer Kimball and Josh MacDonald
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, see <http://www.gnu.org/licenses/>.
*/
/*
* Modified by the GLib Team and others 1997-2000. See the AUTHORS
* file for a list of people on the GLib Team. See the ChangeLog
* files for a list of changes. These files are distributed with
* GLib at ftp://ftp.gtk.org/pub/gtk/.
*/
#include <sys/types.h>
#include <stdlib.h>
#include <glib.h>
#ifdef G_OS_UNIX
#include <unistd.h>
#endif
#ifdef G_OS_WIN32
#include <windows.h>
#endif
#ifdef G_OS_WIN32
#define GPID_FORMAT "%p"
#else
#define GPID_FORMAT "%d"
#endif
GMainLoop *main_loop;
gint alive;
#ifdef G_OS_WIN32
char *argv0;
#endif
static GPid
get_a_child (gint ttl)
{
GPid pid;
#ifdef G_OS_WIN32
STARTUPINFO si;
PROCESS_INFORMATION pi;
gchar *cmdline;
memset (&si, 0, sizeof (si));
si.cb = sizeof (&si);
memset (&pi, 0, sizeof (pi));
cmdline = g_strdup_printf( "child-test -c%d", ttl);
if (!CreateProcess (argv0, cmdline, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi))
g_error ("CreateProcess failed: %s\n", g_win32_error_message (GetLastError ()));
g_free(cmdline);
CloseHandle (pi.hThread);
pid = pi.hProcess;
return pid;
#else
pid = fork ();
if (pid < 0)
exit (1);
if (pid > 0)
return pid;
sleep (ttl);
_exit (0);
#endif /* G_OS_WIN32 */
}
static gboolean
child_watch_callback (GPid pid, gint status, gpointer data)
{
#ifdef VERBOSE
gint ttl = GPOINTER_TO_INT (data);
g_print ("child " GPID_FORMAT " (ttl %d) exited, status %d\n", pid, ttl, status);
#endif
g_spawn_close_pid (pid);
if (--alive == 0)
g_main_loop_quit (main_loop);
return TRUE;
}
static gboolean
quit_loop (gpointer data)
{
GMainLoop *main_loop = data;
g_main_loop_quit (main_loop);
return TRUE;
}
#ifdef TEST_THREAD
static gpointer
test_thread (gpointer data)
{
GMainLoop *new_main_loop;
GSource *source;
GPid pid;
gint ttl = GPOINTER_TO_INT (data);
new_main_loop = g_main_loop_new (NULL, FALSE);
pid = get_a_child (ttl);
source = g_child_watch_source_new (pid);
g_source_set_callback (source, (GSourceFunc) child_watch_callback, data, NULL);
g_source_attach (source, g_main_loop_get_context (new_main_loop));
g_source_unref (source);
#ifdef VERBOSE
g_print ("whee! created pid: " GPID_FORMAT " (ttl %d)\n", pid, ttl);
#endif
g_main_loop_run (new_main_loop);
return NULL;
}
#endif
int
main (int argc, char *argv[])
{
#ifndef TEST_THREAD
GPid pid;
#endif
#ifdef G_OS_WIN32
argv0 = argv[0];
if (argc > 1 && argv[1][0] == '-' && argv[1][1] == 'c')
{
int ttl = atoi (argv[1] + 2);
Sleep (ttl * 1000);
/* Exit on purpose with STILL_ACTIVE (which isn't a very common
* exit status) to verify that g_child_watch_check() in gmain.c
* doesn't believe a child still to be active if it happens to
* exit with that status.
*/
exit (STILL_ACTIVE);
}
#endif
main_loop = g_main_loop_new (NULL, FALSE);
#ifdef G_OS_WIN32
system ("ipconfig /all");
#else
system ("true");
#endif
alive = 2;
g_timeout_add_seconds (30, quit_loop, main_loop);
#ifdef TEST_THREAD
g_thread_create (test_thread, GINT_TO_POINTER (10), FALSE, NULL);
g_thread_create (test_thread, GINT_TO_POINTER (20), FALSE, NULL);
#else
pid = get_a_child (10);
g_child_watch_add (pid, (GChildWatchFunc) child_watch_callback,
GINT_TO_POINTER (10));
pid = get_a_child (20);
g_child_watch_add (pid, (GChildWatchFunc) child_watch_callback,
GINT_TO_POINTER (20));
#endif
g_main_loop_run (main_loop);
g_main_loop_unref (main_loop);
if (alive > 0)
{
g_warning ("%d children still alive\n", alive);
return 1;
}
return 0;
}