X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Futil%2Fos_priority.c;h=03d4bde9e12956e51079d96ea6f34d3f426e2882;hb=31adcb0e020b7861ee7f306a6e2fac0df9b10290;hp=e0e1614f4ff1953e46a476988a6679c1fb8128d9;hpb=8a6448fb4f4dbfab07995028ce8f2516c563cac3;p=oweals%2Fgnunet.git diff --git a/src/util/os_priority.c b/src/util/os_priority.c index e0e1614f4..03d4bde9e 100644 --- a/src/util/os_priority.c +++ b/src/util/os_priority.c @@ -1,21 +1,21 @@ /* This file is part of GNUnet - (C) 2002, 2003, 2004, 2005, 2006, 2011 Christian Grothoff (and other contributing authors) + Copyright (C) 2002, 2003, 2004, 2005, 2006, 2011 GNUnet e.V. - GNUnet is free software; you can redistribute it and/or modify - it under the terms of the GNU General Public License as published - by the Free Software Foundation; either version 2, or (at your - option) any later version. + GNUnet is free software: you can redistribute it and/or modify it + under the terms of the GNU Affero General Public License as published + by the Free Software Foundation, either version 3 of the License, + or (at your option) any later version. GNUnet 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 - General Public License for more details. + Affero General Public License for more details. + + You should have received a copy of the GNU Affero General Public License + along with this program. If not, see . - You should have received a copy of the GNU General Public License - along with GNUnet; see the file COPYING. If not, write to the - Free Software Foundation, Inc., 59 Temple Place - Suite 330, - Boston, MA 02111-1307, USA. + SPDX-License-Identifier: AGPL3.0-or-later */ /** @@ -25,19 +25,15 @@ */ #include "platform.h" -#include "gnunet_common.h" -#include "gnunet_os_lib.h" -#include "gnunet_scheduler_lib.h" -#include "gnunet_strings_lib.h" -#include "gnunet_crypto_lib.h" +#include "gnunet_util_lib.h" #include "disk.h" #include -#define LOG(kind,...) GNUNET_log_from (kind, "util", __VA_ARGS__) +#define LOG(kind,...) GNUNET_log_from (kind, "util-os-priority", __VA_ARGS__) -#define LOG_STRERROR(kind,syscall) GNUNET_log_from_strerror (kind, "util", syscall) +#define LOG_STRERROR(kind,syscall) GNUNET_log_from_strerror (kind, "util-os-priority", syscall) -#define LOG_STRERROR_FILE(kind,syscall,filename) GNUNET_log_from_strerror_file (kind, "util", syscall, filename) +#define LOG_STRERROR_FILE(kind,syscall,filename) GNUNET_log_from_strerror_file (kind, "util-os-priority", syscall, filename) #define GNUNET_OS_CONTROL_PIPE "GNUNET_OS_CONTROL_PIPE" @@ -69,71 +65,102 @@ struct GNUNET_OS_Process */ static struct GNUNET_OS_Process current_process; +/** + * Handle for the #parent_control_handler() Task. + */ +static struct GNUNET_SCHEDULER_Task *pch; + +/** + * Handle for the #shutdown_pch() Task. + */ +static struct GNUNET_SCHEDULER_Task *spch; + + +/** + * This handler is called on shutdown to remove the #pch. + * + * @param cls the `struct GNUNET_DISK_FileHandle` of the control pipe + */ +static void +shutdown_pch (void *cls) +{ + struct GNUNET_DISK_FileHandle *control_pipe = cls; + + GNUNET_SCHEDULER_cancel (pch); + pch = NULL; + GNUNET_DISK_file_close (control_pipe); + control_pipe = NULL; +} + /** * This handler is called when there are control data to be read on the pipe * - * @param cls the 'struct GNUNET_DISK_FileHandle' of the control pipe - * @param tc scheduler context + * @param cls the `struct GNUNET_DISK_FileHandle` of the control pipe */ static void -parent_control_handler (void *cls, - const struct GNUNET_SCHEDULER_TaskContext *tc) +parent_control_handler (void *cls) { struct GNUNET_DISK_FileHandle *control_pipe = cls; char sig; char *pipe_fd; ssize_t ret; - - LOG (GNUNET_ERROR_TYPE_DEBUG, "`%s' invoked because of %d\n", __FUNCTION__, - tc->reason); - if (0 != (tc->reason & - (GNUNET_SCHEDULER_REASON_SHUTDOWN | GNUNET_SCHEDULER_REASON_TIMEOUT))) - { - GNUNET_DISK_file_close (control_pipe); - control_pipe = NULL; - return; - } - ret = GNUNET_DISK_file_read (control_pipe, &sig, sizeof (sig)); + + pch = NULL; + ret = GNUNET_DISK_file_read (control_pipe, + &sig, + sizeof (sig)); if (sizeof (sig) != ret) { if (-1 == ret) - LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "GNUNET_DISK_file_read"); - LOG (GNUNET_ERROR_TYPE_WARNING, "Closing control pipe\n"); + LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, + "GNUNET_DISK_file_read"); + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Closing control pipe\n"); GNUNET_DISK_file_close (control_pipe); control_pipe = NULL; + GNUNET_SCHEDULER_cancel (spch); + spch = NULL; return; } pipe_fd = getenv (GNUNET_OS_CONTROL_PIPE); - GNUNET_assert ( (NULL == pipe_fd) || (strlen (pipe_fd) <= 0) ); + GNUNET_assert ( (NULL == pipe_fd) || + (strlen (pipe_fd) <= 0) ); LOG (GNUNET_ERROR_TYPE_DEBUG, - "Got control code %d from parent via pipe %s\n", sig, pipe_fd); - GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL, - control_pipe, &parent_control_handler, - control_pipe); - raise ((int) sig); + "Got control code %d from parent via pipe %s\n", + sig, + pipe_fd); + pch = GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL, + control_pipe, + &parent_control_handler, + control_pipe); + GNUNET_SIGNAL_raise ((int) sig); } /** * Task that connects this process to its parent via pipe; * essentially, the parent control handler will read signal numbers - * from the 'GNUNET_OS_CONTROL_PIPE' (as given in an environment + * from the #GNUNET_OS_CONTROL_PIPE (as given in an environment * variable) and raise those signals. * * @param cls closure (unused) - * @param tc scheduler context (unused) */ void -GNUNET_OS_install_parent_control_handler (void *cls, - const struct - GNUNET_SCHEDULER_TaskContext *tc) +GNUNET_OS_install_parent_control_handler (void *cls) { const char *env_buf; char *env_buf_end; struct GNUNET_DISK_FileHandle *control_pipe; uint64_t pipe_fd; + (void) cls; + if (NULL != pch) + { + /* already done, we've been called twice... */ + GNUNET_break (0); + return; + } env_buf = getenv (GNUNET_OS_CONTROL_PIPE); if ( (NULL == env_buf) || (strlen (env_buf) <= 0) ) { @@ -147,7 +174,9 @@ GNUNET_OS_install_parent_control_handler (void *cls, pipe_fd = strtoull (env_buf, &env_buf_end, 16); if ((0 != errno) || (env_buf == env_buf_end)) { - LOG_STRERROR_FILE (GNUNET_ERROR_TYPE_WARNING, "strtoull", env_buf); + LOG_STRERROR_FILE (GNUNET_ERROR_TYPE_WARNING, + "strtoull", + env_buf); putenv (GNUNET_OS_CONTROL_PIPE "="); return; } @@ -159,7 +188,8 @@ GNUNET_OS_install_parent_control_handler (void *cls, #endif { LOG (GNUNET_ERROR_TYPE_ERROR, - "GNUNET_OS_CONTROL_PIPE `%s' contains garbage?\n", env_buf); + "GNUNET_OS_CONTROL_PIPE `%s' contains garbage?\n", + env_buf); putenv (GNUNET_OS_CONTROL_PIPE "="); return; } @@ -170,14 +200,21 @@ GNUNET_OS_install_parent_control_handler (void *cls, #endif if (NULL == control_pipe) { - LOG_STRERROR_FILE (GNUNET_ERROR_TYPE_WARNING, "open", env_buf); + LOG_STRERROR_FILE (GNUNET_ERROR_TYPE_WARNING, + "open", + env_buf); putenv (GNUNET_OS_CONTROL_PIPE "="); return; } LOG (GNUNET_ERROR_TYPE_DEBUG, - "Adding parent control handler pipe `%s' to the scheduler\n", env_buf); - GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL, control_pipe, - &parent_control_handler, control_pipe); + "Adding parent control handler pipe `%s' to the scheduler\n", + env_buf); + pch = GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL, + control_pipe, + &parent_control_handler, + control_pipe); + spch = GNUNET_SCHEDULER_add_shutdown (&shutdown_pch, + control_pipe); putenv (GNUNET_OS_CONTROL_PIPE "="); } @@ -185,8 +222,8 @@ GNUNET_OS_install_parent_control_handler (void *cls, /** * Get process structure for current process * - * The pointer it returns points to static memory location and must not be - * deallocated/closed + * The pointer it returns points to static memory location and must + * not be deallocated/closed. * * @return pointer to the process sturcutre for this process */ @@ -211,7 +248,8 @@ GNUNET_OS_process_current () * @return 0 on success, -1 on error */ int -GNUNET_OS_process_kill (struct GNUNET_OS_Process *proc, int sig) +GNUNET_OS_process_kill (struct GNUNET_OS_Process *proc, + int sig) { int ret; char csig; @@ -219,8 +257,13 @@ GNUNET_OS_process_kill (struct GNUNET_OS_Process *proc, int sig) csig = (char) sig; if (NULL != proc->control_pipe) { - LOG (GNUNET_ERROR_TYPE_DEBUG, "Sending signal %d to pid: %u via pipe\n", sig, proc->pid); - ret = GNUNET_DISK_file_write (proc->control_pipe, &csig, sizeof (csig)); + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Sending signal %d to pid: %u via pipe\n", + sig, + proc->pid); + ret = GNUNET_DISK_file_write (proc->control_pipe, + &csig, + sizeof (csig)); if (sizeof (csig) == ret) return 0; } @@ -233,6 +276,9 @@ GNUNET_OS_process_kill (struct GNUNET_OS_Process *proc, int sig) case SIGINT: case SIGKILL: case SIGTERM: +#if (SIGTERM != GNUNET_TERM_SIG) + case GNUNET_TERM_SIG: +#endif #if defined(WINDOWS) && !defined(__CYGWIN__) { DWORD exitcode; @@ -240,14 +286,17 @@ GNUNET_OS_process_kill (struct GNUNET_OS_Process *proc, int sig) if (0 != GetExitCodeProcess (proc->handle, &exitcode)) must_kill = (exitcode == STILL_ACTIVE) ? GNUNET_YES : GNUNET_NO; if (GNUNET_YES == must_kill) + { if (0 == SafeTerminateProcess (proc->handle, 0, 0)) { DWORD error_code = GetLastError (); - if ((error_code != WAIT_TIMEOUT) && (error_code != ERROR_PROCESS_ABORTED)) + if ( (error_code != WAIT_TIMEOUT) && + (error_code != ERROR_PROCESS_ABORTED) ) { LOG ((error_code == ERROR_ACCESS_DENIED) ? - GNUNET_ERROR_TYPE_INFO : GNUNET_ERROR_TYPE_WARNING, - "SafeTermiateProcess failed with code %lu\n", error_code); + GNUNET_ERROR_TYPE_INFO : GNUNET_ERROR_TYPE_WARNING, + "SafeTermiateProcess failed with code %lu\n", + error_code); /* The problem here is that a process that is already dying * might cause SafeTerminateProcess to fail with * ERROR_ACCESS_DENIED, but the process WILL die eventually. @@ -266,10 +315,14 @@ GNUNET_OS_process_kill (struct GNUNET_OS_Process *proc, int sig) } } } + } } return 0; #else - LOG (GNUNET_ERROR_TYPE_DEBUG, "Sending signal %d to pid: %u via system call\n", sig, proc->pid); + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Sending signal %d to pid: %u via system call\n", + sig, + proc->pid); return PLIBC_KILL (proc->pid, sig); #endif default: @@ -277,12 +330,16 @@ GNUNET_OS_process_kill (struct GNUNET_OS_Process *proc, int sig) errno = EINVAL; return -1; #else - LOG (GNUNET_ERROR_TYPE_DEBUG, "Sending signal %d to pid: %u via system call\n", sig, proc->pid); + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Sending signal %d to pid: %u via system call\n", + sig, + proc->pid); return PLIBC_KILL (proc->pid, sig); -#endif +#endif } } + /** * Get the pid of the process in question * @@ -298,7 +355,8 @@ GNUNET_OS_process_get_pid (struct GNUNET_OS_Process * proc) /** - * Cleans up process structure contents (OS-dependent) and deallocates it + * Cleans up process structure contents (OS-dependent) and deallocates + * it. * * @param proc pointer to process structure */ @@ -308,12 +366,13 @@ GNUNET_OS_process_destroy (struct GNUNET_OS_Process *proc) if (NULL != proc->control_pipe) GNUNET_DISK_file_close (proc->control_pipe); #if defined (WINDOWS) - if (proc->handle != NULL) + if (NULL != proc->handle) CloseHandle (proc->handle); #endif GNUNET_free (proc); } + #if WINDOWS #include "gnunet_signal_lib.h" @@ -324,6 +383,7 @@ extern GNUNET_SIGNAL_Handler w32_sigchld_handler; */ #define DWORD_WINAPI DWORD WINAPI + /** * @brief Waits for a process to terminate and invokes the SIGCHLD handler * @param proc pointer to process structure @@ -343,112 +403,6 @@ child_wait_thread (void *arg) #endif -/** - * Set process priority - * - * @param proc pointer to process structure - * @param prio priority value - * @return GNUNET_OK on success, GNUNET_SYSERR on error - */ -int -GNUNET_OS_set_process_priority (struct GNUNET_OS_Process *proc, - enum GNUNET_SCHEDULER_Priority prio) -{ - int rprio; - - GNUNET_assert (prio < GNUNET_SCHEDULER_PRIORITY_COUNT); - if (GNUNET_SCHEDULER_PRIORITY_KEEP == prio) - return GNUNET_OK; - - /* convert to MINGW/Unix values */ - switch (prio) - { - case GNUNET_SCHEDULER_PRIORITY_UI: - case GNUNET_SCHEDULER_PRIORITY_URGENT: -#ifdef MINGW - rprio = HIGH_PRIORITY_CLASS; -#else - rprio = 0; -#endif - break; - - case GNUNET_SCHEDULER_PRIORITY_HIGH: -#ifdef MINGW - rprio = ABOVE_NORMAL_PRIORITY_CLASS; -#else - rprio = 5; -#endif - break; - - case GNUNET_SCHEDULER_PRIORITY_DEFAULT: -#ifdef MINGW - rprio = NORMAL_PRIORITY_CLASS; -#else - rprio = 7; -#endif - break; - - case GNUNET_SCHEDULER_PRIORITY_BACKGROUND: -#ifdef MINGW - rprio = BELOW_NORMAL_PRIORITY_CLASS; -#else - rprio = 10; -#endif - break; - - case GNUNET_SCHEDULER_PRIORITY_IDLE: -#ifdef MINGW - rprio = IDLE_PRIORITY_CLASS; -#else - rprio = 19; -#endif - break; - default: - GNUNET_assert (0); - return GNUNET_SYSERR; - } - - /* Set process priority */ -#ifdef MINGW - { - HANDLE h = proc->handle; - - GNUNET_assert (h != NULL); - SetPriorityClass (h, rprio); - } -#elif LINUX - pid_t pid; - - pid = proc->pid; - if ((0 == pid) || (pid == getpid ())) - { - int have = nice (0); - int delta = rprio - have; - - errno = 0; - if ((delta != 0) && (rprio == nice (delta)) && (errno != 0)) - { - LOG_STRERROR (GNUNET_ERROR_TYPE_WARNING | GNUNET_ERROR_TYPE_BULK, "nice"); - return GNUNET_SYSERR; - } - } - else - { - if (0 != setpriority (PRIO_PROCESS, pid, rprio)) - { - LOG_STRERROR (GNUNET_ERROR_TYPE_WARNING | GNUNET_ERROR_TYPE_BULK, - "setpriority"); - return GNUNET_SYSERR; - } - } -#else - LOG (GNUNET_ERROR_TYPE_DEBUG | GNUNET_ERROR_TYPE_BULK, - "Priority management not availabe for this platform\n"); -#endif - return GNUNET_OK; -} - - #if MINGW static char * CreateCustomEnvTable (char **vars) @@ -573,7 +527,9 @@ open_dev_null (int target_fd, fd = open ("/dev/null", flags); if (-1 == fd) { - GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_ERROR, "open", "/dev/null"); + GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_ERROR, + "open", + "/dev/null"); return; } if (fd == target_fd) @@ -599,6 +555,7 @@ open_dev_null (int target_fd, * (when they are non-NULL). * @param pipe_stdin pipe to use to send input to child process (or NULL) * @param pipe_stdout pipe to use to get output from child process (or NULL) + * @param pipe_stderr pipe to use for stderr for child process (or NULL) * @param lsocks array of listen sockets to dup systemd-style (or NULL); * must be NULL on platforms where dup is not supported * @param filename name of the binary @@ -610,13 +567,13 @@ start_process (int pipe_control, enum GNUNET_OS_InheritStdioFlags std_inheritance, struct GNUNET_DISK_PipeHandle *pipe_stdin, struct GNUNET_DISK_PipeHandle *pipe_stdout, + struct GNUNET_DISK_PipeHandle *pipe_stderr, const SOCKTYPE *lsocks, const char *filename, char *const argv[]) { #ifndef MINGW pid_t ret; - char lpid[16]; char fds[16]; struct GNUNET_OS_Process *gnunet_proc; struct GNUNET_DISK_FileHandle *childpipe_read; @@ -631,29 +588,39 @@ start_process (int pipe_control, unsigned int ls; int fd_stdout_write; int fd_stdout_read; + int fd_stderr_write; + int fd_stderr_read; int fd_stdin_read; int fd_stdin_write; - if (GNUNET_SYSERR == GNUNET_OS_check_helper_binary (filename, GNUNET_NO, NULL)) + if (GNUNET_SYSERR == + GNUNET_OS_check_helper_binary (filename, GNUNET_NO, NULL)) return NULL; /* not executable */ if (GNUNET_YES == pipe_control) { struct GNUNET_DISK_PipeHandle *childpipe; int dup_childpipe_read_fd = -1; - childpipe = GNUNET_DISK_pipe (GNUNET_NO, GNUNET_NO, GNUNET_YES, GNUNET_NO); + + childpipe = GNUNET_DISK_pipe (GNUNET_NO, GNUNET_NO, + GNUNET_YES, GNUNET_NO); if (NULL == childpipe) return NULL; - childpipe_read = GNUNET_DISK_pipe_detach_end (childpipe, GNUNET_DISK_PIPE_END_READ); - childpipe_write = GNUNET_DISK_pipe_detach_end (childpipe, GNUNET_DISK_PIPE_END_WRITE); + childpipe_read = GNUNET_DISK_pipe_detach_end (childpipe, + GNUNET_DISK_PIPE_END_READ); + childpipe_write = GNUNET_DISK_pipe_detach_end (childpipe, + GNUNET_DISK_PIPE_END_WRITE); GNUNET_DISK_pipe_close (childpipe); - if ((NULL == childpipe_read) || (NULL == childpipe_write) || - (GNUNET_OK != GNUNET_DISK_internal_file_handle_ (childpipe_read, - &childpipe_read_fd, sizeof (int))) || - (-1 == (dup_childpipe_read_fd = dup (childpipe_read_fd)))) + if ( (NULL == childpipe_read) || + (NULL == childpipe_write) || + (GNUNET_OK != + GNUNET_DISK_internal_file_handle_ (childpipe_read, + &childpipe_read_fd, + sizeof (int))) || + (-1 == (dup_childpipe_read_fd = dup (childpipe_read_fd)))) { - if (childpipe_read) + if (NULL != childpipe_read) GNUNET_DISK_file_close (childpipe_read); - if (childpipe_write) + if (NULL != childpipe_write) GNUNET_DISK_file_close (childpipe_write); if (0 <= dup_childpipe_read_fd) close (dup_childpipe_read_fd); @@ -664,10 +631,20 @@ start_process (int pipe_control, } else { - childpipe_read = NULL; childpipe_write = NULL; childpipe_read_fd = -1; } + if (NULL != pipe_stdin) + { + GNUNET_assert (GNUNET_OK == + GNUNET_DISK_internal_file_handle_ (GNUNET_DISK_pipe_handle + (pipe_stdin, GNUNET_DISK_PIPE_END_READ), + &fd_stdin_read, sizeof (int))); + GNUNET_assert (GNUNET_OK == + GNUNET_DISK_internal_file_handle_ (GNUNET_DISK_pipe_handle + (pipe_stdin, GNUNET_DISK_PIPE_END_WRITE), + &fd_stdin_write, sizeof (int))); + } if (NULL != pipe_stdout) { GNUNET_assert (GNUNET_OK == @@ -680,16 +657,18 @@ start_process (int pipe_control, (pipe_stdout, GNUNET_DISK_PIPE_END_READ), &fd_stdout_read, sizeof (int))); } - if (NULL != pipe_stdin) + if (NULL != pipe_stderr) { GNUNET_assert (GNUNET_OK == - GNUNET_DISK_internal_file_handle_ (GNUNET_DISK_pipe_handle - (pipe_stdin, GNUNET_DISK_PIPE_END_READ), - &fd_stdin_read, sizeof (int))); + GNUNET_DISK_internal_file_handle_ (GNUNET_DISK_pipe_handle + (pipe_stderr, + GNUNET_DISK_PIPE_END_READ), + &fd_stderr_read, sizeof (int))); GNUNET_assert (GNUNET_OK == - GNUNET_DISK_internal_file_handle_ (GNUNET_DISK_pipe_handle - (pipe_stdin, GNUNET_DISK_PIPE_END_WRITE), - &fd_stdin_write, sizeof (int))); + GNUNET_DISK_internal_file_handle_ (GNUNET_DISK_pipe_handle + (pipe_stderr, + GNUNET_DISK_PIPE_END_WRITE), + &fd_stderr_write, sizeof (int))); } lscp = NULL; ls = 0; @@ -711,9 +690,7 @@ start_process (int pipe_control, int eno = errno; LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "fork"); GNUNET_array_grow (lscp, ls, 0); - if (childpipe_read) - GNUNET_DISK_file_close (childpipe_read); - if (childpipe_write) + if (NULL != childpipe_write) GNUNET_DISK_file_close (childpipe_write); if (0 <= childpipe_read_fd) close (childpipe_read_fd); @@ -722,7 +699,8 @@ start_process (int pipe_control, } if (0 != ret) { - gnunet_proc = GNUNET_malloc (sizeof (struct GNUNET_OS_Process)); + unsetenv (GNUNET_OS_CONTROL_PIPE); + gnunet_proc = GNUNET_new (struct GNUNET_OS_Process); gnunet_proc->pid = ret; gnunet_proc->control_pipe = childpipe_write; if (GNUNET_YES == pipe_control) @@ -768,7 +746,14 @@ start_process (int pipe_control, GNUNET_break (0 == close (1)); open_dev_null (1, O_WRONLY); } - if (0 == (std_inheritance & GNUNET_OS_INHERIT_STD_ERR)) + if (NULL != pipe_stderr) + { + GNUNET_break (0 == close (fd_stderr_read)); + if (-1 == dup2 (fd_stderr_write, 2)) + LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "dup2"); + GNUNET_break (0 == close (fd_stderr_write)); + } + else if (0 == (std_inheritance & GNUNET_OS_INHERIT_STD_ERR)) { GNUNET_break (0 == close (2)); open_dev_null (2, O_WRONLY); @@ -776,8 +761,6 @@ start_process (int pipe_control, if (NULL != lscp) { /* read systemd documentation... */ - GNUNET_snprintf (lpid, sizeof (lpid), "%u", getpid ()); - setenv ("LISTEN_PID", lpid, 1); i = 0; tgt = 3; while (-1 != lscp[i]) @@ -864,10 +847,11 @@ start_process (int pipe_control, DWORD stdif, stdof, stdef; BOOL bresult; DWORD error_code; + DWORD create_no_window; if (GNUNET_SYSERR == GNUNET_OS_check_helper_binary (filename, GNUNET_NO, NULL)) return NULL; /* not executable */ - + /* Search in prefix dir (hopefully - the directory from which * the current module was loaded), bindir and libdir, then in PATH */ @@ -1016,7 +1000,7 @@ start_process (int pipe_control, else SetHandleInformation (stdih, HANDLE_FLAG_INHERIT, 0); } - + stdoh = GetStdHandle (STD_OUTPUT_HANDLE); GetHandleInformation (stdoh, &stdof); @@ -1108,7 +1092,10 @@ start_process (int pipe_control, &lsocks_read, sizeof (HANDLE)); } else + { lsocks_pipe = NULL; + lsocks_write_fd = NULL; + } env_off = 0; if (GNUNET_YES == pipe_control) @@ -1148,7 +1135,9 @@ start_process (int pipe_control, if (NULL == (wcmd = u8_to_u16 ((uint8_t *) cmd, 1 + strlen (cmd), NULL, &wcmd_len))) { LOG (GNUNET_ERROR_TYPE_DEBUG, - "Failed to convert `%s' from UTF-8 to UTF-16: %d\n", cmd, errno); + "Failed to convert `%s' from UTF-8 to UTF-16: %d\n", + cmd, + errno); GNUNET_free (env_block); GNUNET_free (cmd); free (wpath); @@ -1162,13 +1151,22 @@ start_process (int pipe_control, return NULL; } + create_no_window = 0; + { + HANDLE console_input = CreateFile ("CONIN$", GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL); + if (INVALID_HANDLE_VALUE == console_input) + create_no_window = CREATE_NO_WINDOW; + else + CloseHandle (console_input); + } + bresult = CreateProcessW (wpath, wcmd, NULL, NULL, GNUNET_YES, - DETACHED_PROCESS | CREATE_SUSPENDED, env_block, NULL, &start, &proc); + create_no_window | CREATE_SUSPENDED, env_block, NULL, &start, &proc); error_code = GetLastError (); if ((NULL == pipe_stdin) && (stdih)) SetHandleInformation (stdih, HANDLE_FLAG_INHERIT, stdif); - + if ((NULL == pipe_stdout) && (stdoh)) SetHandleInformation (stdoh, HANDLE_FLAG_INHERIT, stdof); @@ -1177,7 +1175,11 @@ start_process (int pipe_control, SetHandleInformation (stdeh, HANDLE_FLAG_INHERIT, stdef); if (!bresult) - LOG (GNUNET_ERROR_TYPE_ERROR, "CreateProcess(%s, %s) failed: %lu\n", path, cmd, error_code); + LOG (GNUNET_ERROR_TYPE_ERROR, + "CreateProcess(%s, %s) failed: %lu\n", + path, + cmd, + error_code); GNUNET_free (env_block); GNUNET_free (cmd); @@ -1200,7 +1202,7 @@ start_process (int pipe_control, return NULL; } - gnunet_proc = GNUNET_malloc (sizeof (struct GNUNET_OS_Process)); + gnunet_proc = GNUNET_new (struct GNUNET_OS_Process); gnunet_proc->pid = proc.dwProcessId; gnunet_proc->handle = proc.hProcess; gnunet_proc->control_pipe = childpipe_write; @@ -1230,8 +1232,8 @@ start_process (int pipe_control, wrote = GNUNET_DISK_file_write (lsocks_write_fd, &count, sizeof (count)); if (sizeof (count) != wrote) { - GNUNET_log (GNUNET_ERROR_TYPE_ERROR, - "Failed to write %u count bytes to the child: %u\n", + GNUNET_log (GNUNET_ERROR_TYPE_ERROR, + "Failed to write %u count bytes to the child: %lu\n", sizeof (count), GetLastError ()); break; } @@ -1241,8 +1243,8 @@ start_process (int pipe_control, /* Get a socket duplication info */ if (SOCKET_ERROR == WSADuplicateSocketA (lsocks[i], gnunet_proc->pid, &pi)) { - GNUNET_log (GNUNET_ERROR_TYPE_ERROR, - "Failed to duplicate an socket[%llu]: %u\n", i, + GNUNET_log (GNUNET_ERROR_TYPE_ERROR, + "Failed to duplicate an socket[%u]: %lu\n", i, GetLastError ()); break; } @@ -1258,8 +1260,8 @@ start_process (int pipe_control, wrote = GNUNET_DISK_file_write (lsocks_write_fd, &size, sizeof (size)); if (sizeof (size) != wrote) { - GNUNET_log (GNUNET_ERROR_TYPE_ERROR, - "Failed to write %u size[%llu] bytes to the child: %u\n", + GNUNET_log (GNUNET_ERROR_TYPE_ERROR, + "Failed to write %u size[%u] bytes to the child: %lu\n", sizeof (size), i, GetLastError ()); break; } @@ -1267,8 +1269,8 @@ start_process (int pipe_control, wrote = GNUNET_DISK_file_write (lsocks_write_fd, &pi, sizeof (pi)); if (sizeof (pi) != wrote) { - GNUNET_log (GNUNET_ERROR_TYPE_ERROR, - "Failed to write %u socket[%llu] bytes to the child: %u\n", + GNUNET_log (GNUNET_ERROR_TYPE_ERROR, + "Failed to write %u socket[%u] bytes to the child: %lu\n", sizeof (pi), i, GetLastError ()); break; } @@ -1303,8 +1305,6 @@ start_process (int pipe_control, } - - /** * Start a process. * @@ -1312,6 +1312,7 @@ start_process (int pipe_control, * @param std_inheritance a set of GNUNET_OS_INHERIT_STD_* flags * @param pipe_stdin pipe to use to send input to child process (or NULL) * @param pipe_stdout pipe to use to get output from child process (or NULL) + * @param pipe_stderr pipe to use to get output from child process (or NULL) * @param filename name of the binary * @param argv NULL-terminated array of arguments to the process * @return pointer to process structure of the new process, NULL on error @@ -1321,13 +1322,15 @@ GNUNET_OS_start_process_vap (int pipe_control, enum GNUNET_OS_InheritStdioFlags std_inheritance, struct GNUNET_DISK_PipeHandle *pipe_stdin, struct GNUNET_DISK_PipeHandle *pipe_stdout, - const char *filename, + struct GNUNET_DISK_PipeHandle *pipe_stderr, + const char *filename, char *const argv[]) { return start_process (pipe_control, std_inheritance, pipe_stdin, pipe_stdout, + pipe_stderr, NULL, filename, argv); @@ -1341,6 +1344,7 @@ GNUNET_OS_start_process_vap (int pipe_control, * @param std_inheritance a set of GNUNET_OS_INHERIT_STD_* flags * @param pipe_stdin pipe to use to send input to child process (or NULL) * @param pipe_stdout pipe to use to get output from child process (or NULL) + * @param pipe_stderr pipe to use to get output from child process (or NULL) * @param filename name of the binary * @param va NULL-terminated list of arguments to the process * @return pointer to process structure of the new process, NULL on error @@ -1350,6 +1354,7 @@ GNUNET_OS_start_process_va (int pipe_control, enum GNUNET_OS_InheritStdioFlags std_inheritance, struct GNUNET_DISK_PipeHandle *pipe_stdin, struct GNUNET_DISK_PipeHandle *pipe_stdout, + struct GNUNET_DISK_PipeHandle *pipe_stderr, const char *filename, va_list va) { struct GNUNET_OS_Process *ret; @@ -1372,6 +1377,7 @@ GNUNET_OS_start_process_va (int pipe_control, std_inheritance, pipe_stdin, pipe_stdout, + pipe_stderr, filename, argv); GNUNET_free (argv); @@ -1395,14 +1401,20 @@ GNUNET_OS_start_process (int pipe_control, enum GNUNET_OS_InheritStdioFlags std_inheritance, struct GNUNET_DISK_PipeHandle *pipe_stdin, struct GNUNET_DISK_PipeHandle *pipe_stdout, + struct GNUNET_DISK_PipeHandle *pipe_stderr, const char *filename, ...) { struct GNUNET_OS_Process *ret; va_list ap; va_start (ap, filename); - ret = GNUNET_OS_start_process_va (pipe_control, std_inheritance, pipe_stdin, - pipe_stdout, filename, ap); + ret = GNUNET_OS_start_process_va (pipe_control, + std_inheritance, + pipe_stdin, + pipe_stdout, + pipe_stderr, + filename, + ap); va_end (ap); return ret; } @@ -1433,6 +1445,7 @@ GNUNET_OS_start_process_v (int pipe_control, std_inheritance, NULL, NULL, + NULL, lsocks, filename, argv); @@ -1440,28 +1453,168 @@ GNUNET_OS_start_process_v (int pipe_control, /** - * Retrieve the status of a process, waiting on him if dead. + * Start a process. This function is similar to the GNUNET_OS_start_process_* + * except that the filename and arguments can have whole strings which contain + * the arguments. These arguments are to be separated by spaces and are parsed + * in the order they appear. Arguments containing spaces can be used by + * quoting them with @em ". + * + * @param pipe_control should a pipe be used to send signals to the child? + * @param std_inheritance a set of GNUNET_OS_INHERIT_STD_* flags + * @param lsocks array of listen sockets to dup systemd-style (or NULL); + * must be NULL on platforms where dup is not supported + * @param filename name of the binary. It is valid to have the arguments + * in this string when they are separated by spaces. + * @param ... more arguments. Should be of type `char *`. It is valid + * to have the arguments in these strings when they are separated by + * spaces. The last argument MUST be NULL. + * @return pointer to process structure of the new process, NULL on error + */ +struct GNUNET_OS_Process * +GNUNET_OS_start_process_s (int pipe_control, + unsigned int std_inheritance, + const SOCKTYPE * lsocks, + const char *filename, ...) +{ + va_list ap; + char **argv; + unsigned int argv_size; + const char *arg; + const char *rpos; + char *pos; + char *cp; + const char *last; + struct GNUNET_OS_Process *proc; + char *binary_path; + int quote_on; + unsigned int i; + size_t len; + + argv_size = 1; + va_start (ap, filename); + arg = filename; + last = NULL; + do + { + rpos = arg; + quote_on = 0; + while ('\0' != *rpos) + { + if ('"' == *rpos) + { + if (1 == quote_on) + quote_on = 0; + else + quote_on = 1; + } + if ( (' ' == *rpos) && (0 == quote_on) ) + { + if (NULL != last) + argv_size++; + last = NULL; + rpos++; + while (' ' == *rpos) + rpos++; + } + if ( (NULL == last) && ('\0' != *rpos) ) // FIXME: == or !=? + last = rpos; + if ('\0' != *rpos) + rpos++; + } + if (NULL != last) + argv_size++; + } + while (NULL != (arg = (va_arg (ap, const char*)))); + va_end (ap); + + argv = GNUNET_malloc (argv_size * sizeof (char *)); + argv_size = 0; + va_start (ap, filename); + arg = filename; + last = NULL; + do + { + cp = GNUNET_strdup (arg); + quote_on = 0; + pos = cp; + while ('\0' != *pos) + { + if ('"' == *pos) + { + if (1 == quote_on) + quote_on = 0; + else + quote_on = 1; + } + if ( (' ' == *pos) && (0 == quote_on) ) + { + *pos = '\0'; + if (NULL != last) + argv[argv_size++] = GNUNET_strdup (last); + last = NULL; + pos++; + while (' ' == *pos) + pos++; + } + if ( (NULL == last) && ('\0' != *pos)) // FIXME: == or !=? + last = pos; + if ('\0' != *pos) + pos++; + } + if (NULL != last) + argv[argv_size++] = GNUNET_strdup (last); + last = NULL; + GNUNET_free (cp); + } + while (NULL != (arg = (va_arg (ap, const char*)))); + va_end (ap); + argv[argv_size] = NULL; + + for(i = 0; i < argv_size; i++) + { + len = strlen (argv[i]); + if ( (argv[i][0] == '"') && (argv[i][len-1] == '"')) + { + memmove (&argv[i][0], &argv[i][1], len - 2); + argv[i][len-2] = '\0'; + } + } + binary_path = argv[0]; + proc = GNUNET_OS_start_process_v (pipe_control, std_inheritance, lsocks, + binary_path, argv); + while (argv_size > 0) + GNUNET_free (argv[--argv_size]); + GNUNET_free (argv); + return proc; +} + + +/** + * Retrieve the status of a process, waiting on it if dead. * Nonblocking version. - * + * * @param proc process ID * @param type status type * @param code return code/signal number - * @return GNUNET_OK on success, GNUNET_NO if the process is still running, GNUNET_SYSERR otherwise + * @param options WNOHANG if non-blocking is desired + * @return #GNUNET_OK on success, #GNUNET_NO if the process is still running, #GNUNET_SYSERR otherwise */ -int -GNUNET_OS_process_status (struct GNUNET_OS_Process *proc, - enum GNUNET_OS_ProcessStatusType *type, - unsigned long *code) +static int +process_status (struct GNUNET_OS_Process *proc, + enum GNUNET_OS_ProcessStatusType *type, + unsigned long *code, + int options) { #ifndef MINGW int status; int ret; GNUNET_assert (0 != proc); - ret = waitpid (proc->pid, &status, WNOHANG); + ret = waitpid (proc->pid, &status, options); if (ret < 0) { - LOG_STRERROR (GNUNET_ERROR_TYPE_WARNING, "waitpid"); + LOG_STRERROR (GNUNET_ERROR_TYPE_WARNING, + "waitpid"); return GNUNET_SYSERR; } if (0 == ret) @@ -1503,6 +1656,10 @@ GNUNET_OS_process_status (struct GNUNET_OS_Process *proc, *code = 0; } #else +#ifndef WNOHANG +#define WNOHANG 42 /* just a flag for W32, purely internal at this point */ +#endif + HANDLE h; DWORD c, error_code, ret; @@ -1510,13 +1667,22 @@ GNUNET_OS_process_status (struct GNUNET_OS_Process *proc, ret = proc->pid; if (h == NULL || ret == 0) { - LOG (GNUNET_ERROR_TYPE_WARNING, "Invalid process information {%d, %08X}\n", + LOG (GNUNET_ERROR_TYPE_WARNING, + "Invalid process information {%d, %08X}\n", ret, h); return GNUNET_SYSERR; } if (h == NULL) h = GetCurrentProcess (); + if (WNOHANG != options) + { + if (WAIT_OBJECT_0 != WaitForSingleObject (h, INFINITE)) + { + SetErrnoFromWinError (GetLastError ()); + return GNUNET_SYSERR; + } + } SetLastError (0); ret = GetExitCodeProcess (h, &c); error_code = GetLastError (); @@ -1541,10 +1707,56 @@ GNUNET_OS_process_status (struct GNUNET_OS_Process *proc, /** - * Wait for a process + * Retrieve the status of a process, waiting on it if dead. + * Nonblocking version. + * + * @param proc process ID + * @param type status type + * @param code return code/signal number + * @return #GNUNET_OK on success, #GNUNET_NO if the process is still running, #GNUNET_SYSERR otherwise + */ +int +GNUNET_OS_process_status (struct GNUNET_OS_Process *proc, + enum GNUNET_OS_ProcessStatusType *type, + unsigned long *code) +{ + return process_status (proc, + type, + code, + WNOHANG); +} + + +/** + * Retrieve the status of a process, waiting on it if dead. + * Blocking version. * * @param proc pointer to process structure - * @return GNUNET_OK on success, GNUNET_SYSERR otherwise + * @param type status type + * @param code return code/signal number + * @return #GNUNET_OK on success, #GNUNET_NO if the process is still running, #GNUNET_SYSERR otherwise + */ +int +GNUNET_OS_process_wait_status (struct GNUNET_OS_Process *proc, + enum GNUNET_OS_ProcessStatusType *type, + unsigned long *code) +{ + return process_status (proc, + type, + code, + 0); +} + + +/** + * Wait for a process to terminate. The return code is discarded. + * You must not use #GNUNET_OS_process_status() on the same process + * after calling this function! This function is blocking and should + * thus only be used if the child process is known to have terminated + * or to terminate very soon. + * + * @param proc pointer to process structure + * @return #GNUNET_OK on success, #GNUNET_SYSERR otherwise */ int GNUNET_OS_process_wait (struct GNUNET_OS_Process *proc) @@ -1555,9 +1767,10 @@ GNUNET_OS_process_wait (struct GNUNET_OS_Process *proc) while ( (pid != (ret = waitpid (pid, NULL, 0))) && (EINTR == errno) ) ; - if (pid != ret) + if (pid != ret) { - LOG_STRERROR (GNUNET_ERROR_TYPE_WARNING, "waitpid"); + LOG_STRERROR (GNUNET_ERROR_TYPE_WARNING, + "waitpid"); return GNUNET_SYSERR; } return GNUNET_OK; @@ -1567,7 +1780,8 @@ GNUNET_OS_process_wait (struct GNUNET_OS_Process *proc) h = proc->handle; if (NULL == h) { - LOG (GNUNET_ERROR_TYPE_WARNING, "Invalid process information {%d, %08X}\n", + LOG (GNUNET_ERROR_TYPE_WARNING, + "Invalid process information {%d, %08X}\n", proc->pid, h); return GNUNET_SYSERR; } @@ -1611,7 +1825,7 @@ struct GNUNET_OS_CommandHandle GNUNET_OS_LineProcessor proc; /** - * Closure for 'proc'. + * Closure for @e proc. */ void *proc_cls; @@ -1623,7 +1837,7 @@ struct GNUNET_OS_CommandHandle /** * Task reading from pipe. */ - GNUNET_SCHEDULER_TaskIdentifier rtask; + struct GNUNET_SCHEDULER_Task *rtask; /** * When to time out. @@ -1649,7 +1863,7 @@ GNUNET_OS_command_stop (struct GNUNET_OS_CommandHandle *cmd) { if (NULL != cmd->proc) { - GNUNET_assert (GNUNET_SCHEDULER_NO_TASK != cmd->rtask); + GNUNET_assert (NULL != cmd->rtask); GNUNET_SCHEDULER_cancel (cmd->rtask); } (void) GNUNET_OS_process_kill (cmd->eip, SIGKILL); @@ -1663,29 +1877,32 @@ GNUNET_OS_command_stop (struct GNUNET_OS_CommandHandle *cmd) /** * Read from the process and call the line processor. * - * @param cls the 'struct GNUNET_OS_CommandHandle' - * @param tc scheduler context + * @param cls the `struct GNUNET_OS_CommandHandle *` */ static void -cmd_read (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +cmd_read (void *cls) { struct GNUNET_OS_CommandHandle *cmd = cls; + const struct GNUNET_SCHEDULER_TaskContext *tc; GNUNET_OS_LineProcessor proc; char *end; ssize_t ret; - cmd->rtask = GNUNET_SCHEDULER_NO_TASK; - if (GNUNET_YES != GNUNET_NETWORK_fdset_handle_isset (tc->read_ready, cmd->r)) + cmd->rtask = NULL; + tc = GNUNET_SCHEDULER_get_task_context (); + if (GNUNET_YES != + GNUNET_NETWORK_fdset_handle_isset (tc->read_ready, + cmd->r)) { - /* timeout, shutdown, etc. */ + /* timeout */ proc = cmd->proc; cmd->proc = NULL; proc (cmd->proc_cls, NULL); return; } - ret = - GNUNET_DISK_file_read (cmd->r, &cmd->buf[cmd->off], - sizeof (cmd->buf) - cmd->off); + ret = GNUNET_DISK_file_read (cmd->r, + &cmd->buf[cmd->off], + sizeof (cmd->buf) - cmd->off); if (ret <= 0) { if ((cmd->off > 0) && (cmd->off < sizeof (cmd->buf))) @@ -1708,9 +1925,11 @@ cmd_read (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) cmd->off -= (end + 1 - cmd->buf); end = memchr (cmd->buf, '\n', cmd->off); } - cmd->rtask = - GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_absolute_get_remaining - (cmd->timeout), cmd->r, &cmd_read, cmd); + cmd->rtask + = GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_absolute_get_remaining + (cmd->timeout), + cmd->r, + &cmd_read, cmd); } @@ -1719,15 +1938,17 @@ cmd_read (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) * for each line of the output. * * @param proc function to call for each line of the output - * @param proc_cls closure for proc + * @param proc_cls closure for @a proc * @param timeout when to time out * @param binary command to run * @param ... arguments to command * @return NULL on error */ struct GNUNET_OS_CommandHandle * -GNUNET_OS_command_run (GNUNET_OS_LineProcessor proc, void *proc_cls, - struct GNUNET_TIME_Relative timeout, const char *binary, +GNUNET_OS_command_run (GNUNET_OS_LineProcessor proc, + void *proc_cls, + struct GNUNET_TIME_Relative timeout, + const char *binary, ...) { struct GNUNET_OS_CommandHandle *cmd; @@ -1735,12 +1956,15 @@ GNUNET_OS_command_run (GNUNET_OS_LineProcessor proc, void *proc_cls, struct GNUNET_DISK_PipeHandle *opipe; va_list ap; - opipe = GNUNET_DISK_pipe (GNUNET_YES, GNUNET_YES, GNUNET_NO, GNUNET_YES); + opipe = GNUNET_DISK_pipe (GNUNET_YES, GNUNET_YES, + GNUNET_NO, GNUNET_YES); if (NULL == opipe) return NULL; va_start (ap, binary); /* redirect stdout, don't inherit stderr/stdin */ - eip = GNUNET_OS_start_process_va (GNUNET_NO, 0, NULL, opipe, binary, ap); + eip = GNUNET_OS_start_process_va (GNUNET_NO, 0, NULL, + opipe, NULL, binary, + ap); va_end (ap); if (NULL == eip) { @@ -1748,14 +1972,18 @@ GNUNET_OS_command_run (GNUNET_OS_LineProcessor proc, void *proc_cls, return NULL; } GNUNET_DISK_pipe_close_end (opipe, GNUNET_DISK_PIPE_END_WRITE); - cmd = GNUNET_malloc (sizeof (struct GNUNET_OS_CommandHandle)); + cmd = GNUNET_new (struct GNUNET_OS_CommandHandle); cmd->timeout = GNUNET_TIME_relative_to_absolute (timeout); cmd->eip = eip; cmd->opipe = opipe; cmd->proc = proc; cmd->proc_cls = proc_cls; - cmd->r = GNUNET_DISK_pipe_handle (opipe, GNUNET_DISK_PIPE_END_READ); - cmd->rtask = GNUNET_SCHEDULER_add_read_file (timeout, cmd->r, &cmd_read, cmd); + cmd->r = GNUNET_DISK_pipe_handle (opipe, + GNUNET_DISK_PIPE_END_READ); + cmd->rtask = GNUNET_SCHEDULER_add_read_file (timeout, + cmd->r, + &cmd_read, + cmd); return cmd; }