GNU/Linux >> Znalost Linux >  >> Linux

Ekvivalent Waitpid s časovým limitem?

Fork prostřední dítě, které forkne skutečné dítě a časový limit a čeká na všechny (obě) své děti. Když jeden odejde, zabije druhého a odejde.

pid_t intermediate_pid = fork();
if (intermediate_pid == 0) {
    pid_t worker_pid = fork();
    if (worker_pid == 0) {
        do_work();
        _exit(0);
    }

    pid_t timeout_pid = fork();
    if (timeout_pid == 0) {
        sleep(timeout_time);
        _exit(0);
    }

    pid_t exited_pid = wait(NULL);
    if (exited_pid == worker_pid) {
        kill(timeout_pid, SIGKILL);
    } else {
        kill(worker_pid, SIGKILL); // Or something less violent if you prefer
    }
    wait(NULL); // Collect the other process
    _exit(0); // Or some more informative status
}
waitpid(intermediate_pid, 0, 0);

Překvapivě jednoduché :)

Můžete dokonce vynechat zprostředkujícího potomka, pokud jste si jisti, že žádný jiný modul v programu nevytvoří vlastní dceřiné procesy.


Nesměšujte alarm() s wait() . Tímto způsobem můžete ztratit informace o chybě.

Použijte trik s vlastním potrubím. Tím se jakýkoli signál změní na select() možná událost:

int selfpipe[2];
void selfpipe_sigh(int n)
{
    int save_errno = errno;
    (void)write(selfpipe[1], "",1);
    errno = save_errno;
}
void selfpipe_setup(void)
{
    static struct sigaction act;
    if (pipe(selfpipe) == -1) { abort(); }

    fcntl(selfpipe[0],F_SETFL,fcntl(selfpipe[0],F_GETFL)|O_NONBLOCK);
    fcntl(selfpipe[1],F_SETFL,fcntl(selfpipe[1],F_GETFL)|O_NONBLOCK);
    memset(&act, 0, sizeof(act));
    act.sa_handler = selfpipe_sigh;
    sigaction(SIGCHLD, &act, NULL);
}

Vaše funkce podobná waitpid pak vypadá takto:

int selfpipe_waitpid(void)
{
    static char dummy[4096];
    fd_set rfds;
    struct timeval tv;
    int died = 0, st;

    tv.tv_sec = 5;
    tv.tv_usec = 0;
    FD_ZERO(&rfds);
    FD_SET(selfpipe[0], &rfds);
    if (select(selfpipe[0]+1, &rfds, NULL, NULL, &tv) > 0) {
       while (read(selfpipe[0],dummy,sizeof(dummy)) > 0);
       while (waitpid(-1, &st, WNOHANG) != -1) died++;
    }
    return died;
}

Můžete vidět v selfpipe_waitpid() jak můžete ovládat časový limit a dokonce jej kombinovat s ostatními select() -založená IO.


To je zajímavá otázka. Zjistil jsem, že to umí sigtimedwait.

EDIT 2016/08/29:Díky za návrh Marka Edingtona. Testoval jsem váš příklad na Ubuntu 16.04, funguje podle očekávání.

Poznámka:Toto funguje pouze pro podřízené procesy. Je škoda, že se zdá, že neexistuje ekvivalentní způsob WaitForSingleObject (unrelated_process_handle, časový limit) v Linuxu/Unixu, jak dostat upozornění na ukončení nesouvisejícího procesu během časového limitu.

OK, ukázkový kód Marka Edingtona je zde:

/* The program creates a child process and waits for it to finish. If a timeout
 * elapses the child is killed. Waiting is done using sigtimedwait(). Race
 * condition is avoided by blocking the SIGCHLD signal before fork().
 */
#include <sys/types.h>
#include <sys/wait.h>
#include <signal.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>

static pid_t fork_child (void)
{
    int p = fork ();

    if (p == -1) {
        perror ("fork");
        exit (1);
    }

    if (p == 0) {
        puts ("child: sleeping...");
        sleep (10);
        puts ("child: exiting");
        exit (0);
    }

    return p;
}

int main (int argc, char *argv[])
{
    sigset_t mask;
    sigset_t orig_mask;
    struct timespec timeout;
    pid_t pid;

    sigemptyset (&mask);
    sigaddset (&mask, SIGCHLD);

    if (sigprocmask(SIG_BLOCK, &mask, &orig_mask) < 0) {
        perror ("sigprocmask");
        return 1;
    }

    pid = fork_child ();

    timeout.tv_sec = 5;
    timeout.tv_nsec = 0;

    do {
        if (sigtimedwait(&mask, NULL, &timeout) < 0) {
            if (errno == EINTR) {
                /* Interrupted by a signal other than SIGCHLD. */
                continue;
            }
            else if (errno == EAGAIN) {
                printf ("Timeout, killing child\n");
                kill (pid, SIGKILL);
            }
            else {
                perror ("sigtimedwait");
                return 1;
            }
        }

        break;
    } while (1);

    if (waitpid(pid, NULL, 0) < 0) {
        perror ("waitpid");
        return 1;
    }

    return 0;
}

Linux
  1. Oprava binárního kódu pomocí Dd?

  2. Co dělá poll() s časovým limitem 0?

  3. Windows ekvivalent inet_aton

  1. UNIX / Linux :Jak odstranit ekvivalentního uživatele root (uživatel jiného typu než root s UID 0)

  2. Linux:existuje čtení nebo recv ze socketu s časovým limitem?

  3. Bash:počkejte s časovým limitem

  1. Trasování jádra pomocí trace-cmd

  2. ekvivalent příkazu fallocate() v OS X?

  3. Existuje ekvivalent cd - pro cp nebo mv?