/*
* processes.c: Process handles
*
* Author:
* Dick Porter (dick@ximian.com)
*
* (C) 2002-2006 Novell, Inc.
*/
#include <config.h>
#include <glib.h>
#include <string.h>
#include <pthread.h>
#include <sched.h>
#include <sys/time.h>
#include <errno.h>
#include <sys/types.h>
#include <unistd.h>
#include <signal.h>
#include <sys/wait.h>
#include <mono/io-layer/wapi.h>
#include <mono/io-layer/wapi-private.h>
#include <mono/io-layer/handles-private.h>
#include <mono/io-layer/misc-private.h>
#include <mono/io-layer/mono-mutex.h>
#include <mono/io-layer/process-private.h>
#include <mono/io-layer/threads.h>
#include <mono/utils/strenc.h>
#include <mono/io-layer/timefuncs-private.h>
/* The process' environment strings */
extern char **environ;
#undef DEBUG
static guint32 process_wait (gpointer handle, guint32 timeout);
struct _WapiHandleOps _wapi_process_ops = {
NULL, /* close_shared */
NULL, /* signal */
NULL, /* own */
NULL, /* is_owned */
process_wait, /* special_wait */
NULL /* prewait */
};
static mono_once_t process_current_once=MONO_ONCE_INIT;
static gpointer current_process=NULL;
static mono_once_t process_ops_once=MONO_ONCE_INIT;
static void process_ops_init (void)
{
_wapi_handle_register_capabilities (WAPI_HANDLE_PROCESS,
WAPI_HANDLE_CAP_WAIT |
WAPI_HANDLE_CAP_SPECIAL_WAIT);
}
static gboolean process_set_termination_details (gpointer handle, int status)
{
struct _WapiHandle_process *process_handle;
gboolean ok;
int thr_ret;
ok = _wapi_lookup_handle (handle, WAPI_HANDLE_PROCESS,
(gpointer *)&process_handle);
if (ok == FALSE) {
g_warning ("%s: error looking up process handle %p",
__func__, handle);
return(FALSE);
}
thr_ret = _wapi_handle_lock_shared_handles ();
g_assert (thr_ret == 0);
if (WIFSIGNALED(status)) {
process_handle->exitstatus = 128 + WTERMSIG(status);
} else {
process_handle->exitstatus = WEXITSTATUS(status);
}
_wapi_time_t_to_filetime (time(NULL), &process_handle->exit_time);
#ifdef DEBUG
g_message ("%s: Setting handle %p signalled", __func__, handle);
#endif
_wapi_shared_handle_set_signal_state (handle, TRUE);
_wapi_handle_unlock_shared_handles ();
/* Drop the reference we hold so we have somewhere to store
* the exit details, now the process has in fact exited
*/
_wapi_handle_unref (handle);
return (ok);
}
/* See if any child processes have terminated and wait() for them,
* updating process handle info. This function is called from the
* collection thread every few seconds.
*/
static gboolean waitfor_pid (gpointer test, gpointer user_data)
{
struct _WapiHandle_process *process;
gboolean ok;
int status;
pid_t ret;
if (_wapi_handle_issignalled (test)) {
/* We've already done this one */
return (FALSE);
}
ok = _wapi_lookup_handle (test, WAPI_HANDLE_PROCESS,
(gpointer *)&process);
if (ok == FALSE) {
/* The handle must have been too old and was reaped */
return (FALSE);
}
do {
ret = waitpid (process->id, &status, WNOHANG);
} while (errno == EINTR);
if (ret <= 0) {
/* Process not ready for wait */
#ifdef DEBUG
g_message ("%s: Process %d not ready for waiting for: %s",
__func__, process->id, g_strerror (errno));
#endif
return (FALSE);
}
#ifdef DEBUG
g_message ("%s: Process %d finished", __func__, ret);
#endif
*(int *)user_data = status;
return (TRUE);
}
void _wapi_process_reap (void)
{
gpointer proc;
int status;
#ifdef DEBUG
g_message ("%s: Reaping child processes", __func__);
#endif
do {
proc = _wapi_search_handle (WAPI_HANDLE_PROCESS, waitfor_pid,
&status, NULL, FALSE);
if (proc != NULL) {
#ifdef DEBUG
g_message ("%s: process handle %p exit code %d",
__func__, proc, status);
#endif
process_set_termination_details (proc, status);
/* _wapi_search_handle adds a reference, so
* drop it here
*/
_wapi_handle_unref (proc);
}
} while (proc != NULL);
}
/* Limitations: This can only wait for processes that are our own
* children. Fixing this means resurrecting a daemon helper to manage
* processes.
*/
static guint32 process_wait (gpointer handle, guint32 timeout)
{
struct _WapiHandle_process *process_handle;
gboolean ok;
pid_t pid, ret;
int status;
#ifdef DEBUG
g_message ("%s: Waiting for process %p", __func__, handle);
#endif
if (_wapi_handle_issignalled (handle)) {
/* We've already done this one */
#ifdef DEBUG
g_message ("%s: Process %p already signalled", __func__,
handle);
#endif
return (WAIT_OBJECT_0);
}
ok = _wapi_lookup_handle (handle, WAPI_HANDLE_PROCESS,
(gpointer *)&process_handle);
if (ok == FALSE) {
g_warning ("%s: error looking up process handle %p", __func__,
handle);
return(WAIT_FAILED);
}
pid = process_handle->id;
#ifdef DEBUG
g_message ("%s: PID is %d, timeout %d", __func__, pid, timeout);
#endif
if (timeout == INFINITE) {
while ((ret = waitpid (pid, &status, 0)) != pid) {
if (ret == (pid_t)-1 && errno != EINTR) {
return(WAIT_FAILED);
}
}
} else if (timeout == 0) {
/* Just poll */
ret = waitpid (pid, &status, WNOHANG);
if (ret != pid) {
return (WAIT_TIMEOUT);
}
} else {
/* Poll in a loop */
do {
ret = waitpid (pid, &status, WNOHANG);
if (ret == pid) {
break;
} else if (ret == (pid_t)-1 && errno != EINTR) {
return(WAIT_FAILED);
}
_wapi_handle_spin (100);
timeout -= 100;
} while (timeout > 0);
if (timeout <= 0) {
return(WAIT_TIMEOUT);
}
}
/* Process must have exited */
#ifdef DEBUG
g_message ("%s: Wait done, status %d", __func__, status);
#endif
ok = process_set_termination_details (handle, status);
if (ok == FALSE) {
SetLastError (ERROR_OUTOFMEMORY);
return (WAIT_FAILED);
}
return(WAIT_OBJECT_0);
}
void _wapi_process_signal_self ()
{
if (current_process != NULL) {
process_set_termination_details (current_process, 0);
}
}
static void process_set_defaults (struct _WapiHandle_process *process_handle)
{
/* These seem to be the defaults on w2k */
process_handle->min_working_set = 204800;
process_handle->max_working_set = 1413120;
_wapi_time_t_to_filetime (time (NULL), &process_handle->create_time);
}
static int
len16 (const gunichar2 *str)
{
int len = 0;
while (*str++ != 0)
len++;
return len;
}
static gunichar2 *
utf16_concat (const gunichar2 *first, ...)
{
va_list args;
int total = 0, i;
const gunichar2 *s;
gunichar2 *ret;
va_start (args, first);
total += len16 (first);
for (s = va_arg (args, gunichar2 *); s != NULL; s = va_arg(args, gunichar2 *)){
total += len16 (s);
}
va_end (args);
ret = g_new (gunichar2, total + 1);
if (ret == NULL)
return NULL;
ret [total] = 0;
i = 0;
for (s = first; *s != 0; s++)
ret [i++] = *s;
va_start (args, first);
for (s = va_arg (args, gunichar2 *); s != NULL; s = va_arg (args, gunichar2 *)){
const gunichar2 *p;
for (p = s; *p != 0; p++)
ret [i++] = *p;
}
va_end (args);
return ret;
}
static const gunichar2 utf16_space_bytes [2] = { 0x20, 0 };
static const gunichar2 *utf16_space = utf16_space_bytes;
/* Implemented as just a wrapper around CreateProcess () */
gboolean ShellExecuteEx (WapiShellExecuteInfo *sei)
{
gboolean ret;
WapiProcessInformation process_info;
gunichar2 *args;
if (sei == NULL) {
/* w2k just segfaults here, but we can do better than
* that
*/
SetLastError (ERROR_INVALID_PARAMETER);
return (FALSE);
}
if (sei->lpFile == NULL) {
/* w2k returns TRUE for this, for some reason. */
return (TRUE);
}
/* Put both executable and parameters into the second argument
* to CreateProcess (), so it searches $PATH. The conversion
* into and back out of utf8 is because there is no
* g_strdup_printf () equivalent for gunichar2 :-(
*/
args = utf16_concat (sei->lpFile, sei->lpParameters == NULL ? NULL : utf16_space, sei->lpParameters, NULL);
if (args == NULL){
SetLastError (ERROR_INVALID_DATA);
return (FALSE);
}
ret = CreateProcess (NULL, args, NULL, NULL, TRUE,
CREATE_UNICODE_ENVIRONMENT, NULL,
sei->lpDirectory, NULL, &process_info);
g_free (args);
if (!ret) {
static char *handler;
static gunichar2 *handler_utf16;
if (handler_utf16 == (gunichar2 *)-1)
return FALSE;
#ifdef PLATFORM_MACOSX
handler = "/usr/bin/open";
#else
/*
* On Linux, try: xdg-open, the FreeDesktop standard way of doing it,
* if that fails, try to use gnome-open, then kfmclient
*/
handler = g_find_program_in_path ("xdg-open");
if (handler == NULL){
handler = g_find_program_in_path ("gnome-open");
if (handler == NULL){
handler = g_find_program_in_path ("kfmclient");
if (handler == NULL){
handler_utf16 = (gunichar2 *) -1;
return (FALSE);
} else {
/* kfmclient needs exec argument */
char *old = handler;
handler = g_strconcat (old, " exec",
NULL);
g_free (old);
}
}
}
#endif
handler_utf16 = g_utf8_to_utf16 (handler, -1, NULL, NULL, NULL);
g_free (handler);
args = utf16_concat (handler_utf16, utf16_space, sei->lpFile,
sei->lpParameters == NULL ? NULL : utf16_space,
sei->lpParameters, NULL);
if (args == NULL){
SetLastError (ERROR_INVALID_DATA);
return FALSE;
}
ret = CreateProcess (NULL, args, NULL, NULL, TRUE,
CREATE_UNICODE_ENVIRONMENT, NULL,
sei->lpDirectory, NULL, &process_info);
g_free (args);
if (!ret){
SetLastError (ERROR_INVALID_DATA);
return FALSE;
}
}
if (sei->fMask & SEE_MASK_NOCLOSEPROCESS) {
sei->hProcess = process_info.hProcess;
} else {
CloseHandle (process_info.hProcess);
}
return (ret);
}
gboolean CreateProcess (const gunichar2 *appname, const gunichar2 *cmdline,
WapiSecurityAttributes *process_attrs G_GNUC_UNUSED,
WapiSecurityAttributes *thread_attrs G_GNUC_UNUSED,
gboolean inherit_handles, guint32 create_flags,
gpointer new_environ, const gunichar2 *cwd,
WapiStartupInfo *startup,
WapiProcessInformation *process_info)
{
gchar *cmd=NULL, *prog = NULL, *full_prog = NULL, *args = NULL, *args_after_prog = NULL, *dir = NULL, **env_strings = NULL, **argv;
guint32 i, env_count = 0;
gboolean ret = FALSE;
gpointer handle;
struct _WapiHandle_process process_handle = {0}, *process_handle_data;
GError *gerr = NULL;
int in_fd, out_fd, err_fd;
pid_t pid;
int thr_ret;
mono_once (&process_ops_once, process_ops_init);
/* appname and cmdline specify the executable and its args:
*
* If appname is not NULL, it is the name of the executable.
* Otherwise the executable is the first token in cmdline.
*
* Executable searching:
*
* If appname is not NULL, it can specify the full path and
* file name, or else a partial name and the current directory
* will be used. There is no additional searching.
*
* If appname is NULL, the first whitespace-delimited token in
* cmdline is used. If the name does not contain a full
* directory path, the search sequence is:
*
* 1) The directory containing the current process
* 2) The current working directory
* 3) The windows system directory (Ignored)
* 4) The windows directory (Ignored)
* 5) $PATH
*
* Just to make things more interesting, tokens can contain
* white space if they are surrounded by quotation marks. I'm
* beginning to understand just why windows apps are generally
* so crap, with an API like this :-(
*/
if (appname != NULL) {
cmd = mono_unicode_to_external (appname);
if (cmd == NULL) {
#ifdef DEBUG
g_message ("%s: unicode conversion returned NULL",
__func__);
#endif
SetLastError (ERROR_PATH_NOT_FOUND);
goto cleanup;
}
/* Turn all the slashes round the right way */
for (i = 0; i < strlen (cmd); i++) {
if (cmd[i] == '\\') {
cmd[i] = '/';
}
}
}
if (cmdline != NULL) {
args = mono_unicode_to_external (cmdline);
if (args == NULL) {
#ifdef DEBUG
g_message ("%s: unicode conversion returned NULL", __func__);
#endif
SetLastError (ERROR_PATH_NOT_FOUND);
goto cleanup;
}
}
if (cwd != NULL) {
dir = mono_unicode_to_external (cwd);
if (dir == NULL) {
#ifdef DEBUG
g_message ("%s: unicode conversion returned NULL", __func__);
#endif
SetLastError (ERROR_PATH_NOT_FOUND);
goto cleanup;
}
/* Turn all the slashes round the right way */
for (i = 0; i < strlen (dir); i++) {
if (dir[i] == '\\') {
dir[i] = '/';
}
}
}
/* We can't put off locating the executable any longer :-( */
if (cmd != NULL) {
gchar *unquoted;
if (g_ascii_isalpha (cmd[0]) && (cmd[1] == ':')) {
/* Strip off the drive letter. I can't
* believe that CP/M holdover is still
* visible...
*/
g_memmove (cmd, cmd+2, strlen (cmd)-2);
cmd[strlen (cmd)-2] = '\0';
}
unquoted = g_shell_unquote (cmd, NULL);
if (unquoted[0] == '/') {
/* Assume full path given */
prog = g_strdup (unquoted);
/* Executable existing ? */
if (access (prog, X_OK) != 0) {
#ifdef DEBUG
g_message ("%s: Couldn't find executable %s",
__func__, prog);
#endif
g_free (prog);
g_free (unquoted);
SetLastError (ERROR_FILE_NOT_FOUND);
goto cleanup;
}
} else {
/* Search for file named by cmd in the current
* directory
*/
char *curdir = g_get_current_dir ();
prog = g_strdup_printf ("%s/%s", curdir, unquoted);
g_free (unquoted);
g_free (curdir);
/* And make sure it's executable */
if (access (prog, X_OK) != 0) {
#ifdef DEBUG
g_message ("%s: Couldn't find executable %s",
__func__, prog);
#endif
g_free (prog);
SetLastError (ERROR_FILE_NOT_FOUND);
goto cleanup;
}
}
args_after_prog = args;
} else {
gchar *token = NULL;
char quote;
/* Dig out the first token from args, taking quotation
* marks into account
*/
/* First, strip off all leading whitespace */
args = g_strchug (args);
/* args_after_prog points to the contents of args
* after token has been set (otherwise argv[0] is
* duplicated)
*/
args_after_prog = args;
/* Assume the opening quote will always be the first
* character
*/
if (args[0] == '\"' || args [0] == '\'') {
quote = args [0];
for (i = 1; args[i] != '\0' && args[i] != quote; i++);
if (g_ascii_isspace (args[i+1])) {
/* We found the first token */
token = g_strndup (args+1, i-1);
args_after_prog = args + i;
} else {
/* Quotation mark appeared in the
* middle of the token. Just give the
* whole first token, quotes and all,
* to exec.
*/
}
}
if (token == NULL) {
/* No quote mark, or malformed */
for (i = 0; args[i] != '\0'; i++) {
if (g_ascii_isspace (args[i])) {
token = g_strndup (args, i);
args_after_prog = args + i + 1;
break;
}
}
}
if (token == NULL && args[0] != '\0') {
/* Must be just one token in the string */
token = g_strdup (args);
args_after_prog = NULL;
}
if (token == NULL) {
/* Give up */
#ifdef DEBUG
g_message ("%s: Couldn't find what to exec", __func__);
#endif
SetLastError (ERROR_PATH_NOT_FOUND);
goto cleanup;
}
/* Turn all the slashes round the right way. Only for
* the prg. name
*/
for (i = 0; i < strlen (token); i++) {
if (token[i] == '\\') {
token[i] = '/';
}
}
if (g_ascii_isalpha (token[0]) && (token[1] == ':')) {
/* Strip off the drive letter. I can't
* believe that CP/M holdover is still
* visible...
*/
g_memmove (token, token+2, strlen (token)-2);
token[strlen (token)-2] = '\0';
}
if (token[0] == '/') {
/* Assume full path given */
prog = g_strdup (token);
/* Executable existing ? */
if (access (prog, X_OK) != 0) {
g_free (prog);
#ifdef DEBUG
g_message ("%s: Couldn't find executable %s",
__func__, token);
#endif
g_free (token);
SetLastError (ERROR_FILE_NOT_FOUND);
goto cleanup;
}
} else {
char *curdir = g_get_current_dir ();
/* FIXME: Need to record the directory
* containing the current process, and check
* that for the new executable as the first
* place to look
*/
prog = g_strdup_printf ("%s/%s", curdir, token);
g_free (curdir);
/* I assume X_OK is the criterion to use,
* rather than F_OK
*/
if (access (prog, X_OK) != 0) {
g_free (prog);
prog = g_find_program_in_path (token);
if (prog == NULL) {
#ifdef DEBUG
g_message ("%s: Couldn't find executable %s", __func__, token);
#endif
g_free (token);
SetLastError (ERROR_FILE_NOT_FOUND);
goto cleanup;
}
}
}
g_free (token);
}
#ifdef DEBUG
g_message ("%s: Exec prog [%s] args [%s]", __func__, prog,
args_after_prog);
#endif
if (args_after_prog != NULL && *args_after_prog) {
gchar *qprog;
qprog = g_shell_quote (prog);
full_prog = g_strconcat (qprog, " ", args_after_prog, NULL);
g_free (qprog);
} else {
full_prog = g_shell_quote (prog);
}
ret = g_shell_parse_argv (full_prog, NULL, &argv, &gerr);
if (ret == FALSE) {
/* FIXME: Could do something with the GError here
*/
}
if (startup != NULL && startup->dwFlags & STARTF_USESTDHANDLES) {
in_fd = GPOINTER_TO_UINT (startup->hStdInput);
out_fd = GPOINTER_TO_UINT (startup->hStdOutput);
err_fd = GPOINTER_TO_UINT (startup->hStdError);
} else {
in_fd = GPOINTER_TO_UINT (GetStdHandle (STD_INPUT_HANDLE));
out_fd = GPOINTER_TO_UINT (GetStdHandle (STD_OUTPUT_HANDLE));
err_fd = GPOINTER_TO_UINT (GetStdHandle (STD_ERROR_HANDLE));
}
g_strlcpy (process_handle.proc_name, prog,
_WAPI_PROC_NAME_MAX_LEN - 1);
process_set_defaults (&process_handle);
handle = _wapi_handle_new (WAPI_HANDLE_PROCESS, &process_handle);
if (handle == _WAPI_HANDLE_INVALID) {
g_warning ("%s: error creating process handle", __func__);
SetLastError (ERROR_PATH_NOT_FOUND);
goto cleanup;
}
/* Hold another reference so the process has somewhere to
* store its exit data even if we drop this handle
*/
_wapi_handle_ref (handle);
/* new_environ is a block of NULL-terminated strings, which
* is itself NULL-terminated. Of course, passing an array of
* string pointers would have made things too easy :-(
*
* If new_environ is not NULL it specifies the entire set of
* environment variables in the new process. Otherwise the
* new process inherits the same environment.
*/
if (new_environ != NULL) {
gunichar2 *new_environp;
/* Count the number of strings */
for (new_environp = (gunichar2 *)new_environ; *new_environp;
new_environp++) {
env_count++;
while (*new_environp) {
new_environp++;
}
}
/* +2: one for the process handle value, and the last
* one is NULL
*/
env_strings = g_new0 (gchar *, env_count + 2);
/* Copy each environ string into 'strings' turning it
* into utf8 (or the requested encoding) at the same
* time
*/
env_count = 0;
for (new_environp = (gunichar2 *)new_environ; *new_environp;
new_environp++) {
env_strings[env_count] = mono_unicode_to_external (new_environp);
env_count++;
while (*new_environp) {
new_environp++;
}
}
} else {
for (i = 0; environ[i] != NULL; i++) {
env_count++;
}
/* +2: one for the process handle value, and the last
* one is NULL
*/
env_strings = g_new0 (gchar *, env_count + 2);
/* Copy each environ string into 'strings' turning it
* into utf8 (or the requested encoding) at the same
* time
*/
env_count = 0;
for (i = 0; environ[i] != NULL; i++) {
env_strings[env_count] = g_strdup (environ[i]);
env_count++;
}
}
/* pass process handle info to the child, so it doesn't have
* to do an expensive search over the whole list
*/
if (env_strings != NULL) {
struct _WapiHandleUnshared *handle_data;
struct _WapiHandle_shared_ref *ref;
handle_data = &_WAPI_PRIVATE_HANDLES(GPOINTER_TO_UINT(handle));
ref = &handle_data->u.shared;
env_strings[env_count] = g_strdup_printf ("_WAPI_PROCESS_HANDLE_OFFSET=%d", ref->offset);
}
thr_ret = _wapi_handle_lock_shared_handles ();
g_assert (thr_ret == 0);
pid = fork ();
if (pid == -1) {
/* Error */
SetLastError (ERROR_OUTOFMEMORY);
_wapi_handle_unref (handle);
goto cleanup;
} else if (pid == 0) {
/* Child */
/* Wait for the parent to finish setting up the
* handle. The semaphore lock is safe because the
* sem_undo structures of a semaphore aren't inherited
* across a fork ()
*/
thr_ret = _wapi_handle_lock_shared_handles ();
g_assert (thr_ret == 0);
_wapi_handle_unlock_shared_handles ();
/* should we detach from the process group? */
/* Connect stdin, stdout and stderr */
dup2 (in_fd, 0);
dup2 (out_fd, 1);
dup2 (err_fd, 2);
if (inherit_handles != TRUE) {
/* FIXME: do something here */
}
/* Close all file descriptors */
for (i = getdtablesize () - 1; i > 2; i--) {
close (i);
}
#ifdef DEBUG
g_message ("%s: exec()ing [%s] in dir [%s]", __func__, cmd,
dir==NULL?".":dir);
for (i = 0; argv[i] != NULL; i++) {
g_message ("arg %d: [%s]", i, argv[i]);
}
for (i = 0; env_strings[i] != NULL; i++) {
g_message ("env %d: [%s]", i, env_strings[i]);
}
#endif
/* set cwd */
if (dir != NULL && chdir (dir) == -1) {
/* set error */
_exit (-1);
}
/* exec */
execve (argv[0], argv, env_strings);
/* set error */
_exit (-1);
}
/* parent */
ret = _wapi_lookup_handle (handle, WAPI_HANDLE_PROCESS,
(gpointer *)&process_handle_data);
if (ret == FALSE) {
g_warning ("%s: error looking up process handle %p", __func__,
handle);
_wapi_handle_unref (handle);
goto cleanup;
}
process_handle_data->id = pid;
if (process_info != NULL) {
process_info->hProcess = handle;
process_info->dwProcessId = pid;
/* FIXME: we might need to handle the thread info some
* day
*/
process_info->hThread = INVALID_HANDLE_VALUE;
process_info->dwThreadId = 0;
}
cleanup:
_wapi_handle_unlock_shared_handles ();
if (cmd != NULL) {
g_free (cmd);
}
if (full_prog != NULL) {
g_free (prog);
}
if (args != NULL) {
g_free (args);
}
if (dir != NULL) {
g_free (dir);
}
if(env_strings != NULL) {
g_strfreev (env_strings);
}
return(ret);
}
static void process_set_name (struct _WapiHandle_process *process_handle)
{
gchar *progname, *utf8_progname, *slash;
progname=g_get_prgname ();
utf8_progname=mono_utf8_from_external (progname);
#ifdef DEBUG
g_message ("%s: using [%s] as prog name", __func__, progname);
#endif
if(utf8_progname!=NULL) {
slash=strrchr (utf8_progname, '/');
if(slash!=NULL) {
g_strlcpy (process_handle->proc_name, slash+1,
_WAPI_PROC_NAME_MAX_LEN - 1);
} else {
g_strlcpy (process_handle->proc_name, utf8_progname,
_WAPI_PROC_NAME_MAX_LEN - 1);
}
g_free (utf8_progname);
}
}
extern void _wapi_time_t_to_filetime (time_t timeval, WapiFileTime *filetime);
#if !GLIB_CHECK_VERSION (2,4,0)
#define g_setenv(a,b,c) setenv(a,b,c)
#define g_unsetenv(a) unsetenv(a)
#endif
static void process_set_current (void)
{
pid_t pid = _wapi_getpid ();
const char *handle_env;
struct _WapiHandle_process process_handle = {0};
handle_env = g_getenv ("_WAPI_PROCESS_HANDLE_OFFSET");
g_unsetenv ("_WAPI_PROCESS_HANDLE_OFFSET");
if (handle_env != NULL) {
struct _WapiHandle_process *process_handlep;
gchar *procname = NULL;
gboolean ok;
current_process = _wapi_handle_new_from_offset (WAPI_HANDLE_PROCESS, atoi (handle_env), TRUE);
#ifdef DEBUG
g_message ("%s: Found my process handle: %p (offset %d 0x%x)",
__func__, current_process, atoi (handle_env),
atoi (handle_env));
#endif
ok = _wapi_lookup_handle (current_process, WAPI_HANDLE_PROCESS,
(gpointer *)&process_handlep);
if (ok) {
/* This test will probably break on linuxthreads, but
* that should be ancient history on all distros we
* care about by now
*/
if (process_handlep->id == pid) {
procname = process_handlep->proc_name;
if (!strcmp (procname, "mono")) {
/* Set a better process name */
#ifdef DEBUG
g_message ("%s: Setting better process name", __func__);
#endif
process_set_name (process_handlep);
} else {
#ifdef DEBUG
g_message ("%s: Leaving process name: %s", __func__, procname);
#endif
}
return;
}
/* Wrong pid, so drop this handle and fall through to
* create a new one
*/
_wapi_handle_unref (current_process);
}
}
/* We get here if the handle wasn't specified in the
* environment, or if the process ID was wrong, or if the
* handle lookup failed (eg if the parent process forked and
* quit immediately, and deleted the shared data before the
* child got a chance to attach it.)
*/
#ifdef DEBUG
g_message ("%s: Need to create my own process handle", __func__);
#endif
process_handle.id = pid;
process_set_defaults (&process_handle);
process_set_name (&process_handle);
current_process = _wapi_handle_new (WAPI_HANDLE_PROCESS,
&process_handle);
if (current_process == _WAPI_HANDLE_INVALID) {
g_warning ("%s: error creating process handle", __func__);
return;
}
}
gpointer _wapi_process_duplicate ()
{
mono_once (&process_current_once, process_set_current);
_wapi_handle_ref (current_process);
return(current_process);
}
/* Returns a pseudo handle that doesn't need to be closed afterwards */
gpointer GetCurrentProcess (void)
{
mono_once (&process_current_once, process_set_current);
return(_WAPI_PROCESS_CURRENT);
}
guint32 GetProcessId (gpointer handle)
{
struct _WapiHandle_process *process_handle;
gboolean ok;
ok = _wapi_lookup_handle (handle, WAPI_HANDLE_PROCESS,
(gpointer *)&process_handle);
if (ok == FALSE) {
SetLastError (ERROR_INVALID_HANDLE);
return (0);
}
return (process_handle->id);
}
guint32 GetCurrentProcessId (void)
{
mono_once (&process_current_once, process_set_current);
return (GetProcessId (current_process));
}
/* Returns the process id as a convenience to the functions that call this */
static pid_t signal_process_if_gone (gpointer handle)
{
struct _WapiHandle_process *process_handle;
gboolean ok;
/* Make sure the process is signalled if it has exited - if
* the parent process didn't wait for it then it won't be
*/
ok = _wapi_lookup_handle (handle, WAPI_HANDLE_PROCESS,
(gpointer *)&process_handle);
if (ok == FALSE) {
/* It's possible that the handle has vanished during
* the _wapi_search_handle before it gets here, so
* don't spam the console with warnings.
*/
/* g_warning ("%s: error looking up process handle %p",
__func__, handle);*/
return (0);
}
#ifdef DEBUG
g_message ("%s: looking at process %d", __func__, process_handle->id);
#endif
if (kill (process_handle->id, 0) == -1 &&
(errno == ESRCH ||
errno == EPERM)) {
/* The process is dead, (EPERM tells us a new process
* has that ID, but as it's owned by someone else it
* can't be the one listed in our shared memory file)
*/
_wapi_shared_handle_set_signal_state (handle, TRUE);
}
return (process_handle->id);
}
static gboolean process_enum (gpointer handle, gpointer user_data)
{
GArray *processes=user_data;
pid_t pid = signal_process_if_gone (handle);
int i;
if (pid == 0) {
return (FALSE);
}
/* Ignore processes that have already exited (ie they are signalled) */
if (_wapi_handle_issignalled (handle) == FALSE) {
#ifdef DEBUG
g_message ("%s: process %d added to array", __func__, pid);
#endif
/* This ensures that duplicates aren't returned (see
* the comment above _wapi_search_handle () for why
* it's needed
*/
for (i = 0; i < processes->len; i++) {
if (g_array_index (processes, pid_t, i) == pid) {
/* We've already got this one, return
* FALSE to keep searching
*/
return (FALSE);
}
}
g_array_append_val (processes, pid);
}
/* Return false to keep searching */
return(FALSE);
}
gboolean EnumProcesses (guint32 *pids, guint32 len, guint32 *needed)
{
GArray *processes = g_array_new (FALSE, FALSE, sizeof(pid_t));
guint32 fit, i, j;
mono_once (&process_current_once, process_set_current);
_wapi_search_handle (WAPI_HANDLE_PROCESS, process_enum, processes,
NULL, TRUE);
fit=len/sizeof(guint32);
for (i = 0, j = 0; j < fit && i < processes->len; i++) {
pids[j++] = g_array_index (processes, pid_t, i);
}
g_array_free (processes, TRUE);
*needed = j * sizeof(guint32);
return(TRUE);
}
static gboolean process_open_compare (gpointer handle, gpointer user_data)
{
pid_t wanted_pid;
pid_t checking_pid = signal_process_if_gone (handle);
if (checking_pid == 0) {
return(FALSE);
}
wanted_pid = GPOINTER_TO_UINT (user_data);
/* It's possible to have more than one process handle with the
* same pid, but only the one running process can be
* unsignalled
*/
if (checking_pid == wanted_pid &&
_wapi_handle_issignalled (handle) == FALSE) {
/* If the handle is blown away in the window between
* returning TRUE here and _wapi_search_handle pinging
* the timestamp, the search will continue
*/
return(TRUE);
} else {
return(FALSE);
}
}
gpointer OpenProcess (guint32 access G_GNUC_UNUSED, gboolean inherit G_GNUC_UNUSED, guint32 pid)
{
/* Find the process handle that corresponds to pid */
gpointer handle;
mono_once (&process_current_once, process_set_current);
#ifdef DEBUG
g_message ("%s: looking for process %d", __func__, pid);
#endif
handle = _wapi_search_handle (WAPI_HANDLE_PROCESS,
process_open_compare,
GUINT_TO_POINTER (pid), NULL, TRUE);
if (handle == 0) {
#ifdef DEBUG
g_message ("%s: Can't find pid %d", __func__, pid);
#endif
SetLastError (ERROR_PROC_NOT_FOUND);
return(NULL);
}
_wapi_handle_ref (handle);
return(handle);
}
gboolean GetExitCodeProcess (gpointer process, guint32 *code)
{
struct _WapiHandle_process *process_handle;
gboolean ok;
mono_once (&process_current_once, process_set_current);
if(code==NULL) {
return(FALSE);
}
ok=_wapi_lookup_handle (process, WAPI_HANDLE_PROCESS,
(gpointer *)&process_handle);
if(ok==FALSE) {
#ifdef DEBUG
g_message ("%s: Can't find process %p", __func__, process);
#endif
return(FALSE);
}
/* A process handle is only signalled if the process has exited
* and has been waited for */
if (_wapi_handle_issignalled (process) == TRUE ||
process_wait (process, 0) == WAIT_OBJECT_0) {
*code = process_handle->exitstatus;
} else {
*code = STILL_ACTIVE;
}
return(TRUE);
}
gboolean GetProcessTimes (gpointer process, WapiFileTime *create_time,
WapiFileTime *exit_time, WapiFileTime *kernel_time,
WapiFileTime *user_time)
{
struct _WapiHandle_process *process_handle;
gboolean ok;
mono_once (&process_current_once, process_set_current);
if(create_time==NULL || exit_time==NULL || kernel_time==NULL ||
user_time==NULL) {
/* Not sure if w32 allows NULLs here or not */
return(FALSE);
}
ok=_wapi_lookup_handle (process, WAPI_HANDLE_PROCESS,
(gpointer *)&process_handle);
if(ok==FALSE) {
#ifdef DEBUG
g_message ("%s: Can't find process %p", __func__, process);
#endif
return(FALSE);
}
*create_time=process_handle->create_time;
/* A process handle is only signalled if the process has
* exited. Otherwise exit_time isn't set
*/
if(_wapi_handle_issignalled (process)==TRUE) {
*exit_time=process_handle->exit_time;
}
return(TRUE);
}
gboolean EnumProcessModules (gpointer process, gpointer *modules,
guint32 size, guint32 *needed)
{
/* Store modules in an array of pointers (main module as
* modules[0]), using the load address for each module as a
* token. (Use 'NULL' as an alternative for the main module
* so that the simple implementation can just return one item
* for now.) Get the info from /proc/<pid>/maps on linux,
* other systems will have to implement /dev/kmem reading or
* whatever other horrid technique is needed.
*/
if(size<sizeof(gpointer)) {
return(FALSE);
}
#ifdef linux
modules[0]=NULL;
*needed=sizeof(gpointer);
#else
modules[0]=NULL;
*needed=sizeof(gpointer);
#endif
return(TRUE);
}
guint32 GetModuleBaseName (gpointer process, gpointer module,
gunichar2 *basename, guint32 size)
{
struct _WapiHandle_process *process_handle;
gboolean ok;
mono_once (&process_current_once, process_set_current);
#ifdef DEBUG
g_message ("%s: Getting module base name, process handle %p module %p",
__func__, process, module);
#endif
if(basename==NULL || size==0) {
return(FALSE);
}
ok=_wapi_lookup_handle (process, WAPI_HANDLE_PROCESS,
(gpointer *)&process_handle);
if(ok==FALSE) {
#ifdef DEBUG
g_message ("%s: Can't find process %p", __func__, process);
#endif
return(FALSE);
}
if(module==NULL) {
/* Shorthand for the main module, which has the
* process name recorded in the handle data
*/
pid_t pid;
gunichar2 *procname;
gchar *procname_utf8 = NULL;
glong len, bytes;
#ifdef DEBUG
g_message ("%s: Returning main module name", __func__);
#endif
pid=process_handle->id;
procname_utf8 = process_handle->proc_name;
#ifdef DEBUG
g_message ("%s: Process name is [%s]", __func__,
procname_utf8);
#endif
procname = g_utf8_to_utf16 (procname_utf8, -1, NULL, &len,
NULL);
if (procname == NULL) {
/* bugger */
return(0);
}
/* Add the terminator, and convert chars to bytes */
bytes = (len + 1) * 2;
if (size < bytes) {
#ifdef DEBUG
g_message ("%s: Size %d smaller than needed (%ld); truncating", __func__, size, bytes);
#endif
memcpy (basename, procname, size);
} else {
#ifdef DEBUG
g_message ("%s: Size %d larger than needed (%ld)",
__func__, size, bytes);
#endif
memcpy (basename, procname, bytes);
}
g_free (procname);
return(len);
} else {
/* Look up the address in /proc/<pid>/maps */
}
return(0);
}
gboolean GetProcessWorkingSetSize (gpointer process, size_t *min, size_t *max)
{
struct _WapiHandle_process *process_handle;
gboolean ok;
mono_once (&process_current_once, process_set_current);
if(min==NULL || max==NULL) {
/* Not sure if w32 allows NULLs here or not */
return(FALSE);
}
ok=_wapi_lookup_handle (process, WAPI_HANDLE_PROCESS,
(gpointer *)&process_handle);
if(ok==FALSE) {
#ifdef DEBUG
g_message ("%s: Can't find process %p", __func__, process);
#endif
return(FALSE);
}
*min=process_handle->min_working_set;
*max=process_handle->max_working_set;
return(TRUE);
}
gboolean SetProcessWorkingSetSize (gpointer process, size_t min, size_t max)
{
struct _WapiHandle_process *process_handle;
gboolean ok;
mono_once (&process_current_once, process_set_current);
ok=_wapi_lookup_handle (process, WAPI_HANDLE_PROCESS,
(gpointer *)&process_handle);
if(ok==FALSE) {
#ifdef DEBUG
g_message ("%s: Can't find process %p", __func__, process);
#endif
return(FALSE);
}
process_handle->min_working_set=min;
process_handle->max_working_set=max;
return(TRUE);
}
gboolean
TerminateProcess (gpointer process, gint32 exitCode)
{
struct _WapiHandle_process *process_handle;
gboolean ok;
int signo;
int ret;
ok = _wapi_lookup_handle (process, WAPI_HANDLE_PROCESS,
(gpointer *) &process_handle);
if (ok == FALSE) {
#ifdef DEBUG
g_message ("%s: Can't find process %p", __func__, process);
#endif
SetLastError (ERROR_INVALID_HANDLE);
return FALSE;
}
signo = (exitCode == -1) ? SIGKILL : SIGTERM;
ret = kill (process_handle->id, signo);
if (ret == -1) {
switch (errno) {
case EINVAL:
SetLastError (ERROR_INVALID_PARAMETER);
break;
case EPERM:
SetLastError (ERROR_ACCESS_DENIED);
break;
case ESRCH:
SetLastError (ERROR_PROC_NOT_FOUND);
break;
default:
SetLastError (ERROR_GEN_FAILURE);
}
}
return (ret == 0);
}