Aguarde que um processo termine OU que o usuário pressione uma tecla

3

Eu preciso de duas maneiras de terminar uma parte do meu script bash.

Um contador atinge um número predefinido, ou o usuário força manualmente o script a continuar com qualquer valor que o contador tenha atualmente.

Especificamente - estou listando unidades USB. Se houver 15 deles, a função que conta para eles sai e o script pode continuar.

Meu código parece um pouco assim:

scannew(){
    NEW=0
    OLD=$NEW
    while [ true ]; do
        # count the new drives
        lsblk -r > drives.new
        diff drives.old drives.new | grep disk | cut -d' ' -f 2 | sort > drives.all
        NEW=$(wc -l drives.all | cut -d' ' -f1)
        echo -en "   Detected drives: $NEW    \r"
        sleep 0.01
        if [ "$NEW" -eq "15" ]; then # exit if we reach the limit
            break
        fi
    done
}

# SOME CODE...

lsblk -r > drives.old

scannew & # start live device counter in the background
SCAN_PID=$! # remember it's PID
wait $SCAN_PID 2>/dev/null # wait until it dies
echo "It's on!"

# REST OF THE CODE...

Eu tentei várias coisas com o comando read , mas o resultado é que o script sempre esperará que a leitura saia (depois de pressionar ENTER) e eu não posso fazer a condição "15 limite" para sobrescrever isso. / p>

Por exemplo, tentei usar read -t em vez de sleep na função scannew() :

scannew(){
    NEW=0
    OLD=$NEW
    while [ true ]; do

        # count the new drives
        lsblk -r > drives.new
        diff drives.old drives.new | grep disk | cut -d' ' -f 2 | sort > drives.all
        NEW=$(wc -l drives.all | cut -d' ' -f1)
        echo -en "   Detected drives: $NEW    \r"
        read -t 0.01 -n 1 && break # read instead of sleep
        if [ "$NEW" -eq "15" ]; then
            break
        fi
    done
}

No entanto - parece que o subprocesso de funções não tem acesso a stdin, e usar read -t 0.01 -n 1 < /dev/stdin && break também não funcionou.

Como posso fazer isso funcionar?

    
por unfa 15.03.2017 / 14:04

2 respostas

3

Deixe-me começar dizendo, você poderia apenas in-line todas as coisas que você tem em scannew , já que você está wait ing, a menos que você pretenda digitalizar novamente em algum outro ponto no seu script. É realmente a chamada para wc que você está preocupado pode levar muito tempo, o que, se acontecer, você pode terminá-lo. Esta é uma maneira simples de configurar isso usando trap , que permite capturar sinais enviados para um processo e definir seu próprio manipulador para ele:

#! /usr/bin/env bash

# print a line just before we run our subshell, so we know when that happens
printf "Lets do something foolish...\n"

# trap SIGINT since it will be sent to the entire process group and we only
# want the subshell killed
trap "" SIGINT

# run something that takes ages to complete
BAD_IDEA=$( trap "exit 1" SIGINT; ls -laR / )

# remove the trap because we might want to actually terminate the script
# after this point
trap - SIGINT

# if the script gets here, we know only 'ls' got killed
printf "Got here! Only 'ls' got killed.\n"

exit 0

No entanto, se você quiser manter a maneira como faz as coisas, com scannew sendo uma função executada como um trabalho em segundo plano, é necessário um pouco mais de trabalho.

Como você deseja a entrada do usuário, a maneira correta de fazer isso é usar read , mas ainda precisamos que o script continue se scannew for concluído e não apenas aguardar a entrada do usuário para sempre. read torna isso um pouco complicado, porque bash espera que o comando atual seja concluído antes de permitir que trap s funcione nos sinais. A única solução para isso que eu conheço, sem refatorar o script inteiro, é colocar read em um loop while true e dar a ele um tempo limite de 1 segundo, usando read -t 1 . Dessa forma, sempre levará pelo menos um segundo para o processo ser concluído, mas isso pode ser aceitável em uma situação como a sua, na qual você deseja essencialmente executar um daemon de pesquisa que liste dispositivos usb.

#! /usr/bin/env bash

function slow_background_work {
    # condition can be anything of course
    # for testing purposes, we're just checking if the variable has anything in it
    while [[ -z $BAD_IDEA ]]
    do
        BAD_IDEA=$( ls -laR / 2>&1 | wc )
    done

    # '$$' normally gives us our own PID
    # but in a subshell, it is inherited and thus
    # gives the parent's PID
    printf "\nI'm done!\n"
    kill -s SIGUSR1 -- $$
    return 0
}

# trap SIGUSR1, which we're expecting from the background job
# once it's done with the work we gave it
trap "break" SIGUSR1

slow_background_work &

while true
do
    # rewinding the line with printf instead of the prompt string because
    # read doesn't understand backslash escapes in the prompt string
    printf "\r"
    # must check return value instead of the variable
    # because a return value of 0 always means there was
    # input of _some_ sort, including <enter> and <space>
    # otherwise, it's really tricky to test the empty variable
    # since read apparently defines it even if it doesn't get input
    read -st1 -n1 -p "prompt: " useless_variable && {
                              printf "Keypress! Quick, kill the background job w/ fire!\n"
                              # make sure we don't die as we kill our only child
                              trap "" SIGINT
                              kill -s SIGINT -- "$!"
                              trap - SIGINT
                              break
                            }
done

trap - SIGUSR1

printf "Welcome to the start of the rest of your script.\n"

exit 0

Claro, se o que você realmente quer é um daemon que esteja atento a mudanças no número de dispositivos usb ou algo assim, você deve procurar em systemd , que pode fornecer algo mais elegante.

    
por 15.03.2017 / 19:01
2

Uma solução (um pouco) geral para executar um determinado comando e eliminá-lo, caso haja uma entrada do usuário, caso contrário, sair. A essência é executar a leitura em um modo de terminal um pouco bruto procurando a tecla "qualquer" e manipular o caso run-program-exits via o sinal SIGCHLD que deve ser enviado quando a criança sair (assumindo que não é engraçado negócios do processo filho). Código e documentos (e eventuais testes) .

#ifdef __linux__
#define _POSIX_SOURCE
#include <sys/types.h>
#endif

#include <err.h>
#include <fcntl.h>
#include <getopt.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sysexits.h>
#include <termios.h>
#include <unistd.h>

int Flag_UserOkay;              // -U

struct termios Original_Termios;
pid_t Child_Pid;

void child_signal(int unused);
void emit_help(void);
void reset_term(void);

int main(int argc, char *argv[])
{
    int ch, status;
    char anykey;
    struct termios terminfo;

    while ((ch = getopt(argc, argv, "h?U")) != -1) {
        switch (ch) {
        case 'U':
            Flag_UserOkay = 1;
            break;
        case 'h':
        case '?':
        default:
            emit_help();
            /* NOTREACHED */
        }
    }
    argc -= optind;
    argv += optind;

    if (argc == 0)
        emit_help();

    if (!isatty(STDIN_FILENO))
        errx(1, "must have tty to read from");

    if (tcgetattr(STDIN_FILENO, &terminfo) < 0)
        err(EX_OSERR, "could not tcgetattr() on stdin");

    Original_Termios = terminfo;

    // cfmakeraw(3) is a tad too raw and influences output from child;
    // per termios(5) use "Case B" for quick "any" key reads with
    // canonical mode (line-based processing) and echo turned off.
    terminfo.c_cc[VMIN] = 1;
    terminfo.c_cc[VTIME] = 0;
    terminfo.c_lflag &= ~(ICANON | ECHO);

    tcsetattr(STDIN_FILENO, TCSAFLUSH, &terminfo);
    atexit(reset_term);

    signal(SIGCHLD, child_signal);

    Child_Pid = fork();

    if (Child_Pid == 0) {       // child
        close(STDIN_FILENO);
        signal(SIGCHLD, SIG_DFL);
        status = execvp(*argv, argv);
        warn("could not exec '%s' (%d)", *argv, status);
        _exit(EX_OSERR);

    } else if (Child_Pid > 0) { // parent
        if ((status = read(STDIN_FILENO, &anykey, 1)) < 0)
            err(EX_IOERR, "read() failed??");
        kill(Child_Pid, SIGTERM);

    } else {
        err(EX_OSERR, "could not fork");
    }

    exit(Flag_UserOkay ? 0 : 1);
}

void child_signal(int unused)
{
    // might try to pass along the exit status of the child, but that's
    // extra work and complication...
    exit(0);
}

void emit_help(void)
{
    fprintf(stderr, "Usage: waitornot [-U] command [args ..]\n");
    exit(EX_USAGE);
}

void reset_term(void)
{
    tcsetattr(STDIN_FILENO, TCSAFLUSH, &Original_Termios);
}
    
por 15.03.2017 / 18:57