diff options
Diffstat (limited to '')
| -rw-r--r-- | src/libmdsserver/util.c | 39 | ||||
| -rw-r--r-- | src/libmdsserver/util.h | 12 | 
2 files changed, 51 insertions, 0 deletions
| diff --git a/src/libmdsserver/util.c b/src/libmdsserver/util.c index 01f5cbe..83641f2 100644 --- a/src/libmdsserver/util.c +++ b/src/libmdsserver/util.c @@ -28,6 +28,8 @@  #include <sys/socket.h>  #include <errno.h>  #include <ctype.h> +#include <time.h> +#include <sys/wait.h> @@ -294,3 +296,40 @@ int startswith_n(const char* haystack, const char* needle, size_t haystack_n, si    return 1;  } + +/** + * Wrapper around `waitpid` that never returns on an interruption unless + * it is interrupted 100 times within the same second + *  + * @param   pid  See description of `pid` in the documentation for `waitpid` + * @param   pid  See description of `status` in the documentation for `waitpid` + * @param   pid  See description of `options` in the documentation for `waitpid` + * @return       See the documentation for `waitpid` + */ +pid_t uninterruptable_waitpid(pid_t pid, int* restrict status, int options) +{ +  struct timespec time_start; +  struct timespec time_intr; +  int intr_count = 0, have_time; +  pid_t rc; +   +  have_time = (monotone(&time_start) >= 0); +   + rewait: +  rc = waitpid(pid, status, options); +  if (rc == (pid_t)-1) +    { +      if (errno != EINTR) +	goto fail; +      if (have_time && (monotone(&time_intr) >= 0)) +	if (time_start.tv_sec != time_intr.tv_sec) +	  intr_count = 0; +      if (intr_count++ < 100) +	goto rewait; +      /* Don't let the CPU catch fire! */ +      errno = EINTR; +    } + fail: +  return rc; +} + diff --git a/src/libmdsserver/util.h b/src/libmdsserver/util.h index 00dfb58..3220adb 100644 --- a/src/libmdsserver/util.h +++ b/src/libmdsserver/util.h @@ -21,6 +21,7 @@  #include <stddef.h>  #include <stdint.h> +#include <sys/types.h>  /** @@ -113,6 +114,17 @@ char* full_read(int fd);   */  int startswith_n(const char* haystack, const char* needle, size_t haystack_n, size_t needle_n) __attribute__((pure)); +/** + * Wrapper around `waitpid` that never returns on an interruption unless + * it is interrupted 100 times within the same second + *  + * @param   pid  See description of `pid` in the documentation for `waitpid` + * @param   pid  See description of `status` in the documentation for `waitpid` + * @param   pid  See description of `options` in the documentation for `waitpid` + * @return       See the documentation for `waitpid` + */ +pid_t uninterruptable_waitpid(pid_t pid, int* restrict status, int options); +  #endif | 
