5 #if !defined(SPAWN_DEBUG) || defined(_MSC_VER)
8 #define PING() fprintf (stderr, "%s:%s:%d\n", __FILE__, __FUNCTION__, __LINE__); fflush (stderr)
16 /* -*- mode: C; c-file-style: "gnu"; indent-tabs-mode: nil; -*- */
17 /* dbus-spawn-win32.c Wrapper around g_spawn
19 * Copyright (C) 2002, 2003, 2004 Red Hat, Inc.
20 * Copyright (C) 2003 CodeFactory AB
21 * Copyright (C) 2005 Novell, Inc.
23 * Licensed under the Academic Free License version 2.1
25 * This program is free software; you can redistribute it and/or modify
26 * it under the terms of the GNU General Public License as published by
27 * the Free Software Foundation; either version 2 of the License, or
28 * (at your option) any later version.
30 * This program is distributed in the hope that it will be useful,
31 * but WITHOUT ANY WARRANTY; without even the implied warranty of
32 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
33 * GNU General Public License for more details.
35 * You should have received a copy of the GNU General Public License
36 * along with this program; if not, write to the Free Software
37 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
40 #include "dbus-spawn.h"
41 #include "dbus-sysdeps.h"
42 #include "dbus-sysdeps-win.h"
43 #include "dbus-internals.h"
44 #include "dbus-test.h"
45 #include "dbus-protocol.h"
47 #define WIN32_LEAN_AND_MEAN
49 //#include <windows.h>
59 * Babysitter implementation details
65 HANDLE start_sync_event;
66 #ifdef DBUS_BUILD_TESTS
68 HANDLE end_sync_event;
72 DBusSpawnChildSetupFunc child_setup;
80 int socket_to_babysitter; /* Connection to the babysitter thread */
83 DBusWatchList *watches;
84 DBusWatch *sitter_watch;
86 dbus_bool_t have_spawn_errno;
88 dbus_bool_t have_child_status;
92 static DBusBabysitter*
93 _dbus_babysitter_new (void)
95 DBusBabysitter *sitter;
97 sitter = dbus_new0 (DBusBabysitter, 1);
101 sitter->refcount = 1;
103 sitter->start_sync_event = CreateEvent (NULL, FALSE, FALSE, NULL);
104 if (sitter->start_sync_event == NULL)
106 _dbus_babysitter_unref (sitter);
110 #ifdef DBUS_BUILD_TESTS
111 sitter->end_sync_event = CreateEvent (NULL, FALSE, FALSE, NULL);
112 if (sitter->end_sync_event == NULL)
114 _dbus_babysitter_unref (sitter);
119 sitter->child_handle = NULL;
121 sitter->socket_to_babysitter = sitter->socket_to_main = -1;
127 sitter->watches = _dbus_watch_list_new ();
128 if (sitter->watches == NULL)
130 _dbus_babysitter_unref (sitter);
134 sitter->have_spawn_errno = FALSE;
135 sitter->have_child_status = FALSE;
141 * Increment the reference count on the babysitter object.
143 * @param sitter the babysitter
144 * @returns the babysitter
147 _dbus_babysitter_ref (DBusBabysitter *sitter)
150 _dbus_assert (sitter != NULL);
151 _dbus_assert (sitter->refcount > 0);
153 sitter->refcount += 1;
159 * Decrement the reference count on the babysitter object.
161 * @param sitter the babysitter
164 _dbus_babysitter_unref (DBusBabysitter *sitter)
169 _dbus_assert (sitter != NULL);
170 _dbus_assert (sitter->refcount > 0);
172 sitter->refcount -= 1;
174 if (sitter->refcount == 0)
176 if (sitter->socket_to_babysitter != -1)
178 _dbus_close_socket (sitter->socket_to_babysitter, NULL);
179 sitter->socket_to_babysitter = -1;
182 if (sitter->socket_to_main != -1)
184 _dbus_close_socket (sitter->socket_to_main, NULL);
185 sitter->socket_to_main = -1;
189 if (sitter->argv != NULL)
191 for (i = 0; i < sitter->argc; i++)
192 if (sitter->argv[i] != NULL)
194 dbus_free (sitter->argv[i]);
195 sitter->argv[i] = NULL;
197 dbus_free (sitter->argv);
201 if (sitter->envp != NULL)
203 char **e = sitter->envp;
207 dbus_free (sitter->envp);
211 if (sitter->child_handle != NULL)
213 CloseHandle (sitter->child_handle);
214 sitter->child_handle = NULL;
217 if (sitter->sitter_watch)
219 _dbus_watch_invalidate (sitter->sitter_watch);
220 _dbus_watch_unref (sitter->sitter_watch);
221 sitter->sitter_watch = NULL;
225 _dbus_watch_list_free (sitter->watches);
227 if (sitter->start_sync_event != NULL)
230 CloseHandle (sitter->start_sync_event);
231 sitter->start_sync_event = NULL;
234 #ifdef DBUS_BUILD_TESTS
235 if (sitter->end_sync_event != NULL)
237 CloseHandle (sitter->end_sync_event);
238 sitter->end_sync_event = NULL;
242 dbus_free (sitter->executable);
249 _dbus_babysitter_kill_child (DBusBabysitter *sitter)
252 if (sitter->child_handle == NULL)
253 return; /* child is already dead, or we're so hosed we'll never recover */
256 TerminateProcess (sitter->child_handle, 12345);
260 * Checks whether the child has exited, without blocking.
262 * @param sitter the babysitter
265 _dbus_babysitter_get_child_exited (DBusBabysitter *sitter)
268 return (sitter->child_handle == NULL);
272 * Gets the exit status of the child. We do this so implementation specific
273 * detail is not cluttering up dbus, for example the system launcher code.
274 * This can only be called if the child has exited, i.e. call
275 * _dbus_babysitter_get_child_exited(). It returns FALSE if the child
276 * did not return a status code, e.g. because the child was signaled
277 * or we failed to ever launch the child in the first place.
279 * @param sitter the babysitter
280 * @param status the returned status code
281 * @returns #FALSE on failure
284 _dbus_babysitter_get_child_exit_status (DBusBabysitter *sitter,
287 if (!_dbus_babysitter_get_child_exited (sitter))
288 _dbus_assert_not_reached ("Child has not exited");
290 if (!sitter->have_child_status ||
291 sitter->child_status == STILL_ACTIVE)
294 *status = sitter->child_status;
299 * Sets the #DBusError with an explanation of why the spawned
300 * child process exited (on a signal, or whatever). If
301 * the child process has not exited, does nothing (error
302 * will remain unset).
304 * @param sitter the babysitter
305 * @param error an error to fill in
308 _dbus_babysitter_set_child_exit_error (DBusBabysitter *sitter,
312 if (!_dbus_babysitter_get_child_exited (sitter))
316 if (sitter->have_spawn_errno)
318 dbus_set_error (error, DBUS_ERROR_SPAWN_EXEC_FAILED,
319 "Failed to execute program %s: %s",
320 sitter->executable, _dbus_strerror (sitter->spawn_errno));
322 else if (sitter->have_child_status)
325 dbus_set_error (error, DBUS_ERROR_SPAWN_CHILD_EXITED,
326 "Process %s exited with status %d",
327 sitter->executable, sitter->child_status);
332 dbus_set_error (error, DBUS_ERROR_FAILED,
333 "Process %s exited, status unknown",
340 _dbus_babysitter_set_watch_functions (DBusBabysitter *sitter,
341 DBusAddWatchFunction add_function,
342 DBusRemoveWatchFunction remove_function,
343 DBusWatchToggledFunction toggled_function,
345 DBusFreeFunction free_data_function)
348 return _dbus_watch_list_set_functions (sitter->watches,
357 handle_watch (DBusWatch *watch,
358 unsigned int condition,
361 DBusBabysitter *sitter = data;
363 /* On Unix dbus-spawn uses a babysitter *process*, thus it has to
364 * actually send the exit statuses, error codes and whatnot through
365 * sockets and/or pipes. On Win32, the babysitter is jus a thread,
366 * so it can set the status fields directly in the babysitter struct
367 * just fine. The socket pipe is used just so we can watch it with
368 * select(), as soon as anything is written to it we know that the
369 * babysitter thread has recorded the status in the babysitter
374 _dbus_close_socket (sitter->socket_to_babysitter, NULL);
376 sitter->socket_to_babysitter = -1;
381 /* protect_argv lifted from GLib, relicensed by author, Tor Lillqvist */
383 protect_argv (char **argv,
391 *new_argv = dbus_malloc ((argc + 1) * sizeof (char *));
392 if (*new_argv == NULL)
395 for (i = 0; i < argc; i++)
396 (*new_argv)[i] = NULL;
398 /* Quote each argv element if necessary, so that it will get
399 * reconstructed correctly in the C runtime startup code. Note that
400 * the unquoting algorithm in the C runtime is really weird, and
401 * rather different than what Unix shells do. See stdargv.c in the C
402 * runtime sources (in the Platform SDK, in src/crt).
404 * Note that an new_argv[0] constructed by this function should
405 * *not* be passed as the filename argument to a spawn* or exec*
406 * family function. That argument should be the real file name
407 * without any quoting.
409 for (i = 0; i < argc; i++)
414 int need_dblquotes = FALSE;
417 if (*p == ' ' || *p == '\t')
418 need_dblquotes = TRUE;
424 while (*pp && *pp == '\\')
433 q = (*new_argv)[i] = dbus_malloc (len + need_dblquotes*2 + 1);
451 while (*pp && *pp == '\\')
463 /* printf ("argv[%d]:%s, need_dblquotes:%s len:%d => %s\n", i, argv[i], need_dblquotes?"TRUE":"FALSE", len, (*new_argv)[i]); */
465 (*new_argv)[argc] = NULL;
470 static unsigned __stdcall
471 babysitter (void *parameter)
473 DBusBabysitter *sitter = (DBusBabysitter *) parameter;
476 _dbus_babysitter_ref (sitter);
478 if (sitter->child_setup)
481 (*sitter->child_setup) (sitter->user_data);
484 _dbus_verbose ("babysitter: spawning %s\n", sitter->executable);
487 if (sitter->envp != NULL)
488 sitter->child_handle = (HANDLE) spawnve (P_NOWAIT, sitter->executable,
489 (const char * const *) sitter->argv,
490 (const char * const *) sitter->envp);
492 sitter->child_handle = (HANDLE) spawnv (P_NOWAIT, sitter->executable,
493 (const char * const *) sitter->argv);
496 if (sitter->child_handle == (HANDLE) -1)
498 sitter->child_handle = NULL;
499 sitter->have_spawn_errno = TRUE;
500 sitter->spawn_errno = errno;
504 SetEvent (sitter->start_sync_event);
506 if (sitter->child_handle != NULL)
512 WaitForSingleObject (sitter->child_handle, INFINITE);
515 ret = GetExitCodeProcess (sitter->child_handle, &status);
517 sitter->child_status = status;
518 sitter->have_child_status = TRUE;
520 CloseHandle (sitter->child_handle);
521 sitter->child_handle = NULL;
524 #ifdef DBUS_BUILD_TESTS
525 SetEvent (sitter->end_sync_event);
529 send (sitter->socket_to_main, " ", 1, 0);
531 _dbus_babysitter_unref (sitter);
537 _dbus_spawn_async_with_babysitter (DBusBabysitter **sitter_p,
540 DBusSpawnChildSetupFunc child_setup,
544 DBusBabysitter *sitter;
545 HANDLE sitter_thread;
546 int sitter_thread_id;
548 _DBUS_ASSERT_ERROR_IS_CLEAR (error);
553 sitter = _dbus_babysitter_new ();
556 _DBUS_SET_OOM (error);
560 sitter->child_setup = child_setup;
561 sitter->user_data = user_data;
563 sitter->executable = _dbus_strdup (argv[0]);
564 if (sitter->executable == NULL)
566 _DBUS_SET_OOM (error);
571 if (!_dbus_full_duplex_pipe (&sitter->socket_to_babysitter,
572 &sitter->socket_to_main,
576 sitter->sitter_watch = _dbus_watch_new (sitter->socket_to_babysitter,
578 TRUE, handle_watch, sitter, NULL);
580 if (sitter->sitter_watch == NULL)
582 _DBUS_SET_OOM (error);
587 if (!_dbus_watch_list_add_watch (sitter->watches, sitter->sitter_watch))
589 _DBUS_SET_OOM (error);
593 sitter->argc = protect_argv (argv, &sitter->argv);
594 if (sitter->argc == -1)
596 _DBUS_SET_OOM (error);
602 sitter_thread = (HANDLE) _beginthreadex (NULL, 0, babysitter,
603 sitter, 0, &sitter_thread_id);
605 if (sitter_thread == 0)
608 dbus_set_error_const (error, DBUS_ERROR_SPAWN_FORK_FAILED,
609 "Failed to create new thread");
612 CloseHandle (sitter_thread);
615 WaitForSingleObject (sitter->start_sync_event, INFINITE);
618 if (sitter_p != NULL)
621 _dbus_babysitter_unref (sitter);
623 _DBUS_ASSERT_ERROR_IS_CLEAR (error);
629 _dbus_babysitter_unref (sitter);
634 #ifdef DBUS_BUILD_TESTS
636 #define LIVE_CHILDREN(sitter) ((sitter)->child_handle != NULL)
639 _dbus_babysitter_block_for_child_exit (DBusBabysitter *sitter)
641 if (sitter->child_handle == NULL)
644 WaitForSingleObject (sitter->end_sync_event, INFINITE);
648 check_spawn_nonexistent (void *data)
650 char *argv[4] = { NULL, NULL, NULL, NULL };
651 DBusBabysitter *sitter;
656 dbus_error_init (&error);
658 /*** Test launching nonexistent binary */
660 argv[0] = "/this/does/not/exist/32542sdgafgafdg";
661 if (_dbus_spawn_async_with_babysitter (&sitter, argv, NULL,
665 _dbus_babysitter_block_for_child_exit (sitter);
666 _dbus_babysitter_set_child_exit_error (sitter, &error);
670 _dbus_babysitter_unref (sitter);
672 if (!dbus_error_is_set (&error))
674 _dbus_warn ("Did not get an error launching nonexistent executable\n");
678 if (!(dbus_error_has_name (&error, DBUS_ERROR_NO_MEMORY) ||
679 dbus_error_has_name (&error, DBUS_ERROR_SPAWN_EXEC_FAILED)))
681 _dbus_warn ("Not expecting error when launching nonexistent executable: %s: %s\n",
682 error.name, error.message);
683 dbus_error_free (&error);
687 dbus_error_free (&error);
693 check_spawn_segfault (void *data)
695 char *argv[4] = { NULL, NULL, NULL, NULL };
696 DBusBabysitter *sitter;
701 dbus_error_init (&error);
703 /*** Test launching segfault binary */
705 argv[0] = TEST_SEGFAULT_BINARY;
706 if (_dbus_spawn_async_with_babysitter (&sitter, argv, NULL,
710 _dbus_babysitter_block_for_child_exit (sitter);
711 _dbus_babysitter_set_child_exit_error (sitter, &error);
715 _dbus_babysitter_unref (sitter);
717 if (!dbus_error_is_set (&error))
719 _dbus_warn ("Did not get an error launching segfaulting binary\n");
723 if (!(dbus_error_has_name (&error, DBUS_ERROR_NO_MEMORY) ||
724 dbus_error_has_name (&error, DBUS_ERROR_SPAWN_CHILD_EXITED)))
726 _dbus_warn ("Not expecting error when launching segfaulting executable: %s: %s\n",
727 error.name, error.message);
728 dbus_error_free (&error);
732 dbus_error_free (&error);
738 check_spawn_exit (void *data)
740 char *argv[4] = { NULL, NULL, NULL, NULL };
741 DBusBabysitter *sitter;
746 dbus_error_init (&error);
748 /*** Test launching exit failure binary */
750 argv[0] = TEST_EXIT_BINARY;
751 if (_dbus_spawn_async_with_babysitter (&sitter, argv, NULL,
755 _dbus_babysitter_block_for_child_exit (sitter);
756 _dbus_babysitter_set_child_exit_error (sitter, &error);
760 _dbus_babysitter_unref (sitter);
762 if (!dbus_error_is_set (&error))
764 _dbus_warn ("Did not get an error launching binary that exited with failure code\n");
768 if (!(dbus_error_has_name (&error, DBUS_ERROR_NO_MEMORY) ||
769 dbus_error_has_name (&error, DBUS_ERROR_SPAWN_CHILD_EXITED)))
771 _dbus_warn ("Not expecting error when launching exiting executable: %s: %s\n",
772 error.name, error.message);
773 dbus_error_free (&error);
777 dbus_error_free (&error);
783 check_spawn_and_kill (void *data)
785 char *argv[4] = { NULL, NULL, NULL, NULL };
786 DBusBabysitter *sitter;
791 dbus_error_init (&error);
793 /*** Test launching sleeping binary then killing it */
795 argv[0] = TEST_SLEEP_FOREVER_BINARY;
796 if (_dbus_spawn_async_with_babysitter (&sitter, argv, NULL,
800 _dbus_babysitter_kill_child (sitter);
802 _dbus_babysitter_block_for_child_exit (sitter);
804 _dbus_babysitter_set_child_exit_error (sitter, &error);
808 _dbus_babysitter_unref (sitter);
810 if (!dbus_error_is_set (&error))
812 _dbus_warn ("Did not get an error after killing spawned binary\n");
816 if (!(dbus_error_has_name (&error, DBUS_ERROR_NO_MEMORY) ||
817 dbus_error_has_name (&error, DBUS_ERROR_SPAWN_CHILD_EXITED)))
819 _dbus_warn ("Not expecting error when killing executable: %s: %s\n",
820 error.name, error.message);
821 dbus_error_free (&error);
825 dbus_error_free (&error);
831 _dbus_spawn_test (const char *test_data_dir)
833 if (!_dbus_test_oom_handling ("spawn_nonexistent",
834 check_spawn_nonexistent,
838 /* Don't run the obnoxious segfault test by default,
839 * it's a pain to have to click all those error boxes.
841 if (getenv ("DO_SEGFAULT_TEST"))
842 if (!_dbus_test_oom_handling ("spawn_segfault",
843 check_spawn_segfault,
847 if (!_dbus_test_oom_handling ("spawn_exit",
852 if (!_dbus_test_oom_handling ("spawn_and_kill",
853 check_spawn_and_kill,