GNU/Linux >> Linux Esercitazione >  >> Linux

Waitpid equivalente con timeout?

Fork un bambino intermedio, che esegue il fork del bambino reale e un processo di timeout e attende tutti (entrambi) i suoi figli. Quando uno esce, uccide l'altro ed esce.

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);

Sorprendentemente semplice :)

Puoi anche tralasciare il figlio intermedio se sei sicuro che nessun altro modulo nel programma stia generando processi figli propri.


Non mischiare alarm() con wait() . In questo modo puoi perdere le informazioni sull'errore.

Usa il trucco dell'autopipa. Questo trasforma qualsiasi segnale in un select() evento capace:

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);
}

Quindi, la tua funzione simile a waitpid avrà questo aspetto:

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;
}

Puoi vedere in selfpipe_waitpid() come puoi controllare il timeout e persino mescolarlo con altri select() IO basato su.


Questa è una domanda interessante. Ho scoperto che sigtimedwait può farlo.

EDIT 29/08/2016:Grazie per il suggerimento di Mark Edington. Ho testato il tuo esempio su Ubuntu 16.04, funziona come previsto.

Nota:questo funziona solo per i processi figlio. È un peccato che non sembri un modo equivalente di WaitForSingleObject(unrelated_process_handle, timeout) di Windows in Linux/Unix per ricevere una notifica della terminazione di un processo non correlato entro il timeout.

OK, il codice di esempio di Mark Edington è qui:

/* 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. UNIX / Linux:come eliminare l'utente root equivalente (utente non root con UID 0)

  2. Cosa fa poll() con un timeout pari a 0?

  3. windows equivalente di inet_aton

  4. Linux:c'è una lettura o una ricezione dal socket con timeout?

  5. Bash:aspetta con timeout

Comando IP Linux con esempi

Comando Netcat (nc) con esempi

Comando di timeout in Linux

Spiegazione del comando di timeout di Linux per i principianti (con esempi)

15 Comando Linux ps con esempi

Metti un timer sui tuoi comandi in esecuzione con il comando di timeout in Linux