X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Futil%2Fos_priority.c;h=5946f2b3d4b32ae4ad256a0859bb64631c906742;hb=72c8645af31896829b674b575c5375706f362a30;hp=b20c3272d449892113dd17663c8566ff08849142;hpb=50ee6f12db1596d610f7c56a635d3d985352f6c9;p=oweals%2Fgnunet.git diff --git a/src/util/os_priority.c b/src/util/os_priority.c index b20c3272d..5946f2b3d 100644 --- a/src/util/os_priority.c +++ b/src/util/os_priority.c @@ -57,14 +57,10 @@ struct GNUNET_OS_Process #endif /** - * Pipe we use to signal the process (if used). + * Pipe we use to signal the process. + * NULL if unused, or if process was deemed uncontrollable. */ struct GNUNET_DISK_FileHandle *control_pipe; - - /** - * Name of the pipe, NULL for none. - */ - char *childpipename; }; @@ -74,248 +70,6 @@ struct GNUNET_OS_Process static struct GNUNET_OS_Process current_process; -/* MinGW version of named pipe API */ -#ifdef MINGW -/** - * Creates a named pipe/FIFO and opens it - * - * @param fn pointer to the name of the named pipe or to NULL - * @param flags open flags - * @param perm access permissions - * @return pipe handle on success, NULL on error - */ -static struct GNUNET_DISK_FileHandle * -npipe_create (char **fn, enum GNUNET_DISK_OpenFlags flags, - enum GNUNET_DISK_AccessPermissions perm) -{ - struct GNUNET_DISK_FileHandle *ret; - HANDLE h = NULL; - DWORD openMode; - char *name; - - openMode = 0; - if (flags & GNUNET_DISK_OPEN_READWRITE) - openMode = PIPE_ACCESS_DUPLEX; - else if (flags & GNUNET_DISK_OPEN_READ) - openMode = PIPE_ACCESS_INBOUND; - else if (flags & GNUNET_DISK_OPEN_WRITE) - openMode = PIPE_ACCESS_OUTBOUND; - if (flags & GNUNET_DISK_OPEN_FAILIFEXISTS) - openMode |= FILE_FLAG_FIRST_PIPE_INSTANCE; - - while (h == NULL) - { - DWORD error_code; - - name = NULL; - if (*fn != NULL) - { - GNUNET_asprintf (&name, "\\\\.\\pipe\\%.246s", fn); - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Trying to create an instance of named pipe `%s'\n", name); - /* 1) This might work just fine with UTF-8 strings as it is. - * 2) This is only used by GNUnet itself, and only with latin names. - */ - h = CreateNamedPipe (name, openMode | FILE_FLAG_OVERLAPPED, - PIPE_TYPE_BYTE | PIPE_READMODE_BYTE, 2, 1, 1, 0, - NULL); - } - else - { - GNUNET_asprintf (fn, "\\\\.\\pipe\\gnunet-%llu", - GNUNET_CRYPTO_random_u64 (GNUNET_CRYPTO_QUALITY_WEAK, - UINT64_MAX)); - LOG (GNUNET_ERROR_TYPE_DEBUG, "Trying to create unique named pipe `%s'\n", - *fn); - h = CreateNamedPipe (*fn, - openMode | FILE_FLAG_OVERLAPPED | - FILE_FLAG_FIRST_PIPE_INSTANCE, - PIPE_TYPE_BYTE | PIPE_READMODE_BYTE, 2, 1, 1, 0, - NULL); - } - error_code = GetLastError (); - if (name) - GNUNET_free (name); - /* don't re-set name to NULL yet */ - if (h == INVALID_HANDLE_VALUE) - { - SetErrnoFromWinError (error_code); - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Pipe creation have failed because of %d, errno is %d\n", error_code, - errno); - if (name == NULL) - { - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Pipe was to be unique, considering re-creation\n"); - GNUNET_free (*fn); - *fn = NULL; - if (error_code != ERROR_ACCESS_DENIED && error_code != ERROR_PIPE_BUSY) - { - return NULL; - } - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Pipe name was not unique, trying again\n"); - h = NULL; - } - else - return NULL; - } - } - errno = 0; - - ret = GNUNET_malloc (sizeof (*ret)); - ret->h = h; - ret->type = GNUNET_PIPE; - ret->oOverlapRead = GNUNET_malloc (sizeof (OVERLAPPED)); - ret->oOverlapWrite = GNUNET_malloc (sizeof (OVERLAPPED)); - ret->oOverlapRead->hEvent = CreateEvent (NULL, FALSE, FALSE, NULL); - ret->oOverlapWrite->hEvent = CreateEvent (NULL, FALSE, FALSE, NULL); - return ret; -} - - -/** - * Opens already existing named pipe/FIFO - * - * @param fn name of an existing named pipe - * @param flags open flags - * @return pipe handle on success, NULL on error - */ -static struct GNUNET_DISK_FileHandle * -npipe_open (const char *fn, enum GNUNET_DISK_OpenFlags flags) -{ - struct GNUNET_DISK_FileHandle *ret; - HANDLE h; - DWORD openMode; - - openMode = 0; - if (flags & GNUNET_DISK_OPEN_READWRITE) - openMode = GENERIC_WRITE | GENERIC_READ; - else if (flags & GNUNET_DISK_OPEN_READ) - openMode = GENERIC_READ; - else if (flags & GNUNET_DISK_OPEN_WRITE) - openMode = GENERIC_WRITE; - - h = CreateFile (fn, openMode, 0, NULL, OPEN_EXISTING, - FILE_FLAG_OVERLAPPED | FILE_READ_ATTRIBUTES, NULL); - if (h == INVALID_HANDLE_VALUE) - { - SetErrnoFromWinError (GetLastError ()); - return NULL; - } - - ret = GNUNET_malloc (sizeof (*ret)); - ret->h = h; - ret->type = GNUNET_PIPE; - ret->oOverlapRead = GNUNET_malloc (sizeof (OVERLAPPED)); - ret->oOverlapWrite = GNUNET_malloc (sizeof (OVERLAPPED)); - ret->oOverlapRead->hEvent = CreateEvent (NULL, FALSE, FALSE, NULL); - ret->oOverlapWrite->hEvent = CreateEvent (NULL, FALSE, FALSE, NULL); - - return ret; -} - -#else -/* UNIX version of named-pipe API */ - -/** - * Clean up a named pipe and the directory it was placed in. - * - * @param fn name of the pipe - */ -static void -cleanup_npipe (const char *fn) -{ - char *dn; - char *dp; - - if (0 != unlink (fn)) - GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "unlink", fn); - dn = GNUNET_strdup (fn); - dp = dirname (dn); - if (0 != rmdir (dp)) - GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "rmdir", dp); - GNUNET_free (dn); -} - - -/** - * Setup a named pipe. - * - * @param fn where to store the name of the new pipe, - * if *fn is non-null, the name of the pipe to setup - * @return GNUNET_OK on success - */ -static int -npipe_setup (char **fn) -{ - if (NULL == *fn) - { - /* FIXME: hardwired '/tmp' path... is bad */ - char dir[] = "/tmp/gnunet-pipe-XXXXXX"; - - if (NULL == mkdtemp (dir)) - { - LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "mkdtemp"); - return GNUNET_SYSERR; - } - GNUNET_asprintf (fn, "%s/child-control", dir); - } - if (-1 == mkfifo (*fn, S_IRUSR | S_IWUSR)) - return GNUNET_SYSERR; - return GNUNET_OK; -} - - -/** - * Open an existing named pipe. - * - * @param fn name of the file - * @param flags flags to use - * @return NULL on error - */ -static struct GNUNET_DISK_FileHandle * -npipe_open (const char *fn, - enum GNUNET_DISK_OpenFlags flags) -{ - struct GNUNET_DISK_FileHandle *ret; - int fd; - struct timespec req; - int i; - - /* 200 * 5ms = 1s at most */ - for (i=0;i<200;i++) - { - fd = open (fn, O_NONBLOCK | ((flags == GNUNET_DISK_OPEN_READ) ? O_RDONLY : O_WRONLY)); - if ( (-1 != fd) || (9 == i) || (flags == GNUNET_DISK_OPEN_READ)) - break; - /* as this is for killing a child process via pipe and it is conceivable that - the child process simply didn't finish starting yet, we do some sleeping - (which is obviously usually not allowed). We can't select on the FD as - 'open' fails, and we probably shouldn't just "ignore" the error, so wait - and retry a few times is likely the best method; our process API doesn't - support continuations, so we need to sleep directly... */ - req.tv_sec = 0; - req.tv_nsec = 5000000; /* 5ms */ - (void) nanosleep (&req, NULL); - } - if (-1 == fd) - { - GNUNET_log (GNUNET_ERROR_TYPE_INFO, - (flags == GNUNET_DISK_OPEN_READ) - ? _("Failed to open named pipe `%s' for reading: %s\n") - : _("Failed to open named pipe `%s' for writing: %s\n"), - fn, - STRERROR (errno)); - return NULL; - } - ret = GNUNET_malloc (sizeof (struct GNUNET_DISK_FileHandle)); - ret->fd = fd; - return ret; -} -#endif - - /** * This handler is called when there are control data to be read on the pipe * @@ -328,6 +82,7 @@ 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__, @@ -344,11 +99,15 @@ parent_control_handler (void *cls, { if (-1 == ret) LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "GNUNET_DISK_file_read"); + LOG (GNUNET_ERROR_TYPE_WARNING, "Closing control pipe\n"); GNUNET_DISK_file_close (control_pipe); control_pipe = NULL; return; } - LOG (GNUNET_ERROR_TYPE_DEBUG, "Got control code %d from parent\n", sig); + pipe_fd = getenv (GNUNET_OS_CONTROL_PIPE); + 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); @@ -371,30 +130,55 @@ GNUNET_OS_install_parent_control_handler (void *cls, GNUNET_SCHEDULER_TaskContext *tc) { const char *env_buf; + char *env_buf_end; struct GNUNET_DISK_FileHandle *control_pipe; + uint64_t pipe_fd; env_buf = getenv (GNUNET_OS_CONTROL_PIPE); - if ( (env_buf == NULL) || (strlen (env_buf) <= 0) ) + if ( (NULL == env_buf) || (strlen (env_buf) <= 0) ) { LOG (GNUNET_ERROR_TYPE_DEBUG, "Not installing a handler because $%s is empty\n", GNUNET_OS_CONTROL_PIPE); - putenv ("GNUNET_OS_CONTROL_PIPE="); + putenv (GNUNET_OS_CONTROL_PIPE "="); + return; + } + errno = 0; + 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); + putenv (GNUNET_OS_CONTROL_PIPE "="); + return; + } +#if !defined (WINDOWS) + if (pipe_fd >= FD_SETSIZE) +#else + if ((FILE_TYPE_UNKNOWN == GetFileType ((HANDLE) (uintptr_t) pipe_fd)) + && (0 != GetLastError ())) +#endif + { + LOG (GNUNET_ERROR_TYPE_ERROR, + "GNUNET_OS_CONTROL_PIPE `%s' contains garbage?\n", env_buf); + putenv (GNUNET_OS_CONTROL_PIPE "="); return; } - control_pipe = - npipe_open (env_buf, GNUNET_DISK_OPEN_READ); +#if WINDOWS + control_pipe = GNUNET_DISK_get_handle_from_w32_handle ((HANDLE) (uintptr_t) pipe_fd); +#else + control_pipe = GNUNET_DISK_get_handle_from_int_fd ((int) pipe_fd); +#endif if (NULL == control_pipe) { LOG_STRERROR_FILE (GNUNET_ERROR_TYPE_WARNING, "open", env_buf); - putenv ("GNUNET_OS_CONTROL_PIPE="); + 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); - putenv ("GNUNET_OS_CONTROL_PIPE="); + putenv (GNUNET_OS_CONTROL_PIPE "="); } @@ -433,28 +217,23 @@ GNUNET_OS_process_kill (struct GNUNET_OS_Process *proc, int sig) char csig; csig = (char) sig; -#if !WINDOWS - if ( (NULL == proc->control_pipe) && - (NULL != proc->childpipename) ) - proc->control_pipe = npipe_open (proc->childpipename, - GNUNET_DISK_OPEN_WRITE); -#endif 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)); - if (ret == sizeof (csig)) + if (sizeof (csig) == ret) return 0; } /* pipe failed or non-existent, try other methods */ switch (sig) { -#if !WINDOWS +#if !defined (WINDOWS) case SIGHUP: #endif case SIGINT: case SIGKILL: case SIGTERM: -#if WINDOWS && !defined(__CYGWIN__) +#if defined(WINDOWS) && !defined(__CYGWIN__) { DWORD exitcode; int must_kill = GNUNET_YES; @@ -476,7 +255,7 @@ GNUNET_OS_process_kill (struct GNUNET_OS_Process *proc, int sig) * is what will happen in process_wait() in that case) is * a valid option. */ - if (error_code == ERROR_ACCESS_DENIED) + if (ERROR_ACCESS_DENIED == error_code) { errno = 0; } @@ -490,13 +269,15 @@ 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); return PLIBC_KILL (proc->pid, sig); #endif default: -#if WINDOWS +#if defined (WINDOWS) errno = EINVAL; return -1; #else + 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 } @@ -526,22 +307,13 @@ GNUNET_OS_process_destroy (struct GNUNET_OS_Process *proc) { if (NULL != proc->control_pipe) GNUNET_DISK_file_close (proc->control_pipe); -// FIXME NILS -#ifdef WINDOWS +#if defined (WINDOWS) if (proc->handle != NULL) CloseHandle (proc->handle); #endif - if (NULL != proc->childpipename) - { -#if !WINDOWS - cleanup_npipe (proc->childpipename); -#endif - GNUNET_free (proc->childpipename); - } GNUNET_free (proc); } -// FIXME NILS #if WINDOWS #include "gnunet_signal_lib.h" @@ -571,112 +343,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 (prio == GNUNET_SCHEDULER_PRIORITY_KEEP) - 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) @@ -697,7 +363,7 @@ CreateCustomEnvTable (char **vars) char *val; win32_env_table = GetEnvironmentStringsA (); - if (win32_env_table == NULL) + if (NULL == win32_env_table) return NULL; for (c = 0, var_ptr = vars; *var_ptr; var_ptr += 2, c++) ; n_var = c; @@ -847,7 +513,9 @@ start_process (int pipe_control, char lpid[16]; char fds[16]; struct GNUNET_OS_Process *gnunet_proc; - char *childpipename = NULL; + struct GNUNET_DISK_FileHandle *childpipe_read; + struct GNUNET_DISK_FileHandle *childpipe_write; + int childpipe_read_fd; int i; int j; int k; @@ -860,11 +528,40 @@ start_process (int pipe_control, int fd_stdin_read; int fd_stdin_write; - if (GNUNET_SYSERR == GNUNET_OS_check_helper_binary (filename)) + if (GNUNET_SYSERR == GNUNET_OS_check_helper_binary (filename, GNUNET_NO, NULL)) return NULL; /* not executable */ - if ( (GNUNET_YES == pipe_control) && - (GNUNET_OK != npipe_setup (&childpipename)) ) - return NULL; + 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); + 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); + 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) + GNUNET_DISK_file_close (childpipe_read); + if (NULL != childpipe_write) + GNUNET_DISK_file_close (childpipe_write); + if (0 <= dup_childpipe_read_fd) + close (dup_childpipe_read_fd); + return NULL; + } + childpipe_read_fd = dup_childpipe_read_fd; + GNUNET_DISK_file_close (childpipe_read); + } + else + { + childpipe_write = NULL; + childpipe_read_fd = -1; + } if (NULL != pipe_stdout) { GNUNET_assert (GNUNET_OK == @@ -897,30 +594,49 @@ start_process (int pipe_control, GNUNET_array_append (lscp, ls, k); GNUNET_array_append (lscp, ls, -1); } +#if DARWIN + /* see https://gnunet.org/vfork */ + ret = vfork (); +#else ret = fork (); +#endif if (-1 == ret) { int eno = errno; - LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "fork"); - GNUNET_free_non_null (childpipename); GNUNET_array_grow (lscp, ls, 0); + if (NULL != childpipe_write) + GNUNET_DISK_file_close (childpipe_write); + if (0 <= childpipe_read_fd) + close (childpipe_read_fd); errno = eno; return NULL; } if (0 != ret) { + unsetenv (GNUNET_OS_CONTROL_PIPE); gnunet_proc = GNUNET_malloc (sizeof (struct GNUNET_OS_Process)); gnunet_proc->pid = ret; - gnunet_proc->childpipename = childpipename; + gnunet_proc->control_pipe = childpipe_write; + if (GNUNET_YES == pipe_control) + { + close (childpipe_read_fd); + } GNUNET_array_grow (lscp, ls, 0); return gnunet_proc; } - if (NULL != childpipename) + if (0 <= childpipe_read_fd) { - setenv (GNUNET_OS_CONTROL_PIPE, childpipename, 1); - GNUNET_free (childpipename); + char fdbuf[100]; +#ifndef DARWIN + /* due to vfork, we must NOT free memory on DARWIN! */ + GNUNET_DISK_file_close (childpipe_write); +#endif + snprintf (fdbuf, 100, "%x", childpipe_read_fd); + setenv (GNUNET_OS_CONTROL_PIPE, fdbuf, 1); } + else + unsetenv (GNUNET_OS_CONTROL_PIPE); if (NULL != pipe_stdin) { GNUNET_break (0 == close (fd_stdin_write)); @@ -992,13 +708,17 @@ start_process (int pipe_control, GNUNET_snprintf (fds, sizeof (fds), "%u", i); setenv ("LISTEN_FDS", fds, 1); } +#ifndef DARWIN + /* due to vfork, we must NOT free memory on DARWIN! */ GNUNET_array_grow (lscp, ls, 0); +#endif execvp (filename, argv); LOG_STRERROR_FILE (GNUNET_ERROR_TYPE_ERROR, "execvp", filename); _exit (1); #else - struct GNUNET_DISK_FileHandle *control_pipe; - char *childpipename = NULL; + struct GNUNET_DISK_FileHandle *childpipe_read; + struct GNUNET_DISK_FileHandle *childpipe_write; + HANDLE childpipe_read_handle; char **arg; char **non_const_argv; unsigned int cmdlen; @@ -1009,7 +729,7 @@ start_process (int pipe_control, int argcount = 0; struct GNUNET_OS_Process *gnunet_proc; char path[MAX_PATH + 1]; - char *our_env[5] = { NULL, NULL, NULL, NULL, NULL }; + char *our_env[7] = { NULL, NULL, NULL, NULL, NULL, NULL, NULL }; char *env_block = NULL; char *pathbuf; DWORD pathbuf_len; @@ -1038,7 +758,7 @@ start_process (int pipe_control, BOOL bresult; DWORD error_code; - if (GNUNET_SYSERR == GNUNET_OS_check_helper_binary (filename)) + 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 @@ -1071,7 +791,7 @@ start_process (int pipe_control, } cmdlen = strlen (filename); - if (cmdlen < 5 || strcmp (&filename[cmdlen - 4], ".exe") != 0) + if ( (cmdlen < 5) || (0 != strcmp (&filename[cmdlen - 4], ".exe")) ) GNUNET_asprintf (&non_const_filename, "%s.exe", filename); else GNUNET_asprintf (&non_const_filename, "%s", filename); @@ -1093,7 +813,11 @@ start_process (int pipe_control, * Remember: arguments to _start_process() are technically in UTF-8... */ if (non_const_filename[1] == ':') + { snprintf (path, sizeof (path) / sizeof (char), "%s", non_const_filename); + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Using path `%s' as-is. PATH is %s\n", path, ptr); + } else if (!SearchPathA (pathbuf, non_const_filename, NULL, sizeof (path) / sizeof (char), path, NULL)) @@ -1105,6 +829,9 @@ start_process (int pipe_control, GNUNET_free (pathbuf); return NULL; } + else + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Found `%s' in PATH `%s'\n", path, pathbuf); GNUNET_free (pathbuf); GNUNET_free (non_const_filename); @@ -1171,7 +898,7 @@ start_process (int pipe_control, &stdin_handle, sizeof (HANDLE)); start.hStdInput = stdin_handle; } - if (stdih) + else if (stdih) { if (std_inheritance & GNUNET_OS_INHERIT_STD_IN) { @@ -1194,7 +921,7 @@ start_process (int pipe_control, &stdout_handle, sizeof (HANDLE)); start.hStdOutput = stdout_handle; } - if (stdoh) + else if (stdoh) { if (std_inheritance & GNUNET_OS_INHERIT_STD_OUT) { @@ -1221,19 +948,35 @@ start_process (int pipe_control, if (GNUNET_YES == pipe_control) { - control_pipe = - npipe_create (&childpipename, GNUNET_DISK_OPEN_WRITE, - GNUNET_DISK_PERM_USER_READ | - GNUNET_DISK_PERM_USER_WRITE); - if (control_pipe == NULL) + struct GNUNET_DISK_PipeHandle *childpipe; + 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); + GNUNET_DISK_pipe_close (childpipe); + if ((NULL == childpipe_read) || (NULL == childpipe_write) || + (GNUNET_OK != GNUNET_DISK_internal_file_handle_ (childpipe_read, + &childpipe_read_handle, sizeof (HANDLE)))) { + if (childpipe_read) + GNUNET_DISK_file_close (childpipe_read); + if (childpipe_write) + GNUNET_DISK_file_close (childpipe_write); GNUNET_free (cmd); - GNUNET_free (path); return NULL; } + /* Unlike *nix variant, we don't dup the handle, so can't close + * filehandle right now. + */ + SetHandleInformation (childpipe_read_handle, HANDLE_FLAG_INHERIT, 1); } else - control_pipe = NULL; + { + childpipe_read = NULL; + childpipe_write = NULL; + } + if (lsocks != NULL && lsocks[0] != INVALID_SOCKET) { lsocks_pipe = GNUNET_DISK_pipe (GNUNET_YES, GNUNET_YES, GNUNET_YES, GNUNET_NO); @@ -1241,8 +984,12 @@ start_process (int pipe_control, if (lsocks_pipe == NULL) { GNUNET_free (cmd); - GNUNET_free (path); GNUNET_DISK_pipe_close (lsocks_pipe); + if (GNUNET_YES == pipe_control) + { + GNUNET_DISK_file_close (childpipe_write); + GNUNET_DISK_file_close (childpipe_read); + } return NULL; } lsocks_write_fd = GNUNET_DISK_pipe_handle (lsocks_pipe, @@ -1253,15 +1000,14 @@ start_process (int pipe_control, (lsocks_pipe, GNUNET_DISK_PIPE_END_READ), &lsocks_read, sizeof (HANDLE)); } + else + lsocks_pipe = NULL; env_off = 0; - if (NULL != childpipename) + if (GNUNET_YES == pipe_control) { - LOG (GNUNET_ERROR_TYPE_DEBUG, "Opened the parent end of the pipe `%s'\n", - childpipename); GNUNET_asprintf (&our_env[env_off++], "%s=", GNUNET_OS_CONTROL_PIPE); - GNUNET_asprintf (&our_env[env_off++], "%s", childpipename); - GNUNET_free (childpipename); + GNUNET_asprintf (&our_env[env_off++], "%p", childpipe_read_handle); } if ( (lsocks != NULL) && (lsocks[0] != INVALID_SOCKET)) { @@ -1281,6 +1027,13 @@ start_process (int pipe_control, "Failed to convert `%s' from UTF-8 to UTF-16: %d\n", path, errno); GNUNET_free (env_block); GNUNET_free (cmd); + if (lsocks_pipe) + GNUNET_DISK_pipe_close (lsocks_pipe); + if (GNUNET_YES == pipe_control) + { + GNUNET_DISK_file_close (childpipe_write); + GNUNET_DISK_file_close (childpipe_read); + } return NULL; } @@ -1292,10 +1045,17 @@ start_process (int pipe_control, GNUNET_free (env_block); GNUNET_free (cmd); free (wpath); + if (lsocks_pipe) + GNUNET_DISK_pipe_close (lsocks_pipe); + if (GNUNET_YES == pipe_control) + { + GNUNET_DISK_file_close (childpipe_write); + GNUNET_DISK_file_close (childpipe_read); + } return NULL; } - bresult = CreateProcessW (wpath, wcmd, NULL, NULL, TRUE, + bresult = CreateProcessW (wpath, wcmd, NULL, NULL, GNUNET_YES, DETACHED_PROCESS | CREATE_SUSPENDED, env_block, NULL, &start, &proc); error_code = GetLastError (); @@ -1309,33 +1069,41 @@ start_process (int pipe_control, if (stdeh) SetHandleInformation (stdeh, HANDLE_FLAG_INHERIT, stdef); + if (!bresult) + LOG (GNUNET_ERROR_TYPE_ERROR, "CreateProcess(%s, %s) failed: %lu\n", path, cmd, error_code); + GNUNET_free (env_block); GNUNET_free (cmd); free (wpath); free (wcmd); + if (GNUNET_YES == pipe_control) + { + GNUNET_DISK_file_close (childpipe_read); + } if (!bresult) { - SetErrnoFromWinError (error_code); - LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "CreateProcess"); - if (NULL != control_pipe) - GNUNET_DISK_file_close (control_pipe); + if (GNUNET_YES == pipe_control) + { + GNUNET_DISK_file_close (childpipe_write); + } if (NULL != lsocks) GNUNET_DISK_pipe_close (lsocks_pipe); + SetErrnoFromWinError (error_code); return NULL; } gnunet_proc = GNUNET_malloc (sizeof (struct GNUNET_OS_Process)); gnunet_proc->pid = proc.dwProcessId; gnunet_proc->handle = proc.hProcess; - gnunet_proc->control_pipe = control_pipe; + gnunet_proc->control_pipe = childpipe_write; CreateThread (NULL, 64000, &child_wait_thread, (void *) gnunet_proc, 0, NULL); ResumeThread (proc.hThread); CloseHandle (proc.hThread); - if (lsocks == NULL || lsocks[0] == INVALID_SOCKET) + if ( (NULL == lsocks) || (INVALID_SOCKET == lsocks[0]) ) return gnunet_proc; GNUNET_DISK_pipe_close_end (lsocks_pipe, GNUNET_DISK_PIPE_END_READ); @@ -1344,16 +1112,20 @@ start_process (int pipe_control, fail = 1; do { - int wrote; - uint64_t size, count, i; + ssize_t wrote; + uint64_t size; + uint64_t count; + unsigned int i; /* Tell the number of sockets */ for (count = 0; lsocks && lsocks[count] != INVALID_SOCKET; count++); wrote = GNUNET_DISK_file_write (lsocks_write_fd, &count, sizeof (count)); - if (wrote != sizeof (count)) + if (sizeof (count) != wrote) { - GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Failed to write %u count bytes to the child: %u\n", sizeof (count), GetLastError ()); + GNUNET_log (GNUNET_ERROR_TYPE_ERROR, + "Failed to write %u count bytes to the child: %u\n", + sizeof (count), GetLastError ()); break; } for (i = 0; lsocks && lsocks[i] != INVALID_SOCKET; i++) @@ -1362,8 +1134,9 @@ 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, GetLastError ()); - LOG_STRERROR (GNUNET_ERROR_TYPE_ERROR, "CreateProcess"); + GNUNET_log (GNUNET_ERROR_TYPE_ERROR, + "Failed to duplicate an socket[%llu]: %u\n", i, + GetLastError ()); break; } /* Synchronous I/O is not nice, but we can't schedule this: @@ -1376,16 +1149,20 @@ start_process (int pipe_control, */ size = sizeof (pi); wrote = GNUNET_DISK_file_write (lsocks_write_fd, &size, sizeof (size)); - if (wrote != sizeof (size)) + if (sizeof (size) != wrote) { - GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Failed to write %u size[%llu] bytes to the child: %u\n", sizeof (size), i, GetLastError ()); + GNUNET_log (GNUNET_ERROR_TYPE_ERROR, + "Failed to write %u size[%llu] bytes to the child: %u\n", + sizeof (size), i, GetLastError ()); break; } /* Finally! Send the data */ wrote = GNUNET_DISK_file_write (lsocks_write_fd, &pi, sizeof (pi)); - if (wrote != sizeof (pi)) + if (sizeof (pi) != wrote) { - GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Failed to write %u socket[%llu] bytes to the child: %u\n", sizeof (pi), i, GetLastError ()); + GNUNET_log (GNUNET_ERROR_TYPE_ERROR, + "Failed to write %u socket[%llu] bytes to the child: %u\n", + sizeof (pi), i, GetLastError ()); break; } } @@ -1495,7 +1272,6 @@ GNUNET_OS_start_process_va (int pipe_control, } - /** * Start a process. * @@ -1505,9 +1281,7 @@ GNUNET_OS_start_process_va (int pipe_control, * @param pipe_stdout pipe to use to get output from child process (or NULL) * @param filename name of the binary * @param ... NULL-terminated list of arguments to the process - * * @return pointer to process structure of the new process, NULL on error - * */ struct GNUNET_OS_Process * GNUNET_OS_start_process (int pipe_control, @@ -1521,7 +1295,7 @@ GNUNET_OS_start_process (int pipe_control, va_start (ap, filename); ret = GNUNET_OS_start_process_va (pipe_control, std_inheritance, pipe_stdin, - pipe_stdout, filename, ap); + pipe_stdout, filename, ap); va_end (ap); return ret; } @@ -1531,6 +1305,10 @@ GNUNET_OS_start_process (int pipe_control, * Start a process. * * @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 controlling which + * std handles of the parent are inherited by the child. + * pipe_stdin and pipe_stdout take priority over std_inheritance + * (when they are non-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 @@ -1657,13 +1435,13 @@ GNUNET_OS_process_status (struct GNUNET_OS_Process *proc, /** * Wait for a process + * * @param proc pointer to process structure * @return GNUNET_OK on success, GNUNET_SYSERR otherwise */ int GNUNET_OS_process_wait (struct GNUNET_OS_Process *proc) { - #ifndef MINGW pid_t pid = proc->pid; pid_t ret; @@ -1678,7 +1456,6 @@ GNUNET_OS_process_wait (struct GNUNET_OS_Process *proc) return GNUNET_OK; #else HANDLE h; - int ret; h = proc->handle; if (NULL == h) @@ -1687,18 +1464,15 @@ GNUNET_OS_process_wait (struct GNUNET_OS_Process *proc) proc->pid, h); return GNUNET_SYSERR; } - if (h == NULL) + if (NULL == h) h = GetCurrentProcess (); if (WAIT_OBJECT_0 != WaitForSingleObject (h, INFINITE)) { SetErrnoFromWinError (GetLastError ()); - ret = GNUNET_SYSERR; + return GNUNET_SYSERR; } - else - ret = GNUNET_OK; - - return ret; + return GNUNET_OK; #endif } @@ -1766,8 +1540,7 @@ struct GNUNET_OS_CommandHandle void GNUNET_OS_command_stop (struct GNUNET_OS_CommandHandle *cmd) { - - if (cmd->proc != NULL) + if (NULL != cmd->proc) { GNUNET_assert (GNUNET_SCHEDULER_NO_TASK != cmd->rtask); GNUNET_SCHEDULER_cancel (cmd->rtask); @@ -1820,7 +1593,7 @@ cmd_read (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) } end = memchr (&cmd->buf[cmd->off], '\n', ret); cmd->off += ret; - while (end != NULL) + while (NULL != end) { *end = '\0'; cmd->proc (cmd->proc_cls, cmd->buf); @@ -1880,6 +1653,4 @@ GNUNET_OS_command_run (GNUNET_OS_LineProcessor proc, void *proc_cls, } - - /* end of os_priority.c */