timeout/timeout
Simon Vieille dd06a74bc4
Some checks failed
ci/woodpecker/push/woodpecker Pipeline failed
fix pid variable
2023-01-01 11:51:24 +01:00

73 lines
1.1 KiB
Bash
Executable file

#!/bin/sh
usage() {
printf "Usage: %s [-t TIMEOUT] [-c COMMAND]\n\n" "$0"
printf "TIMEOUT and COMMAND can be set as env vars.\n"
}
on_interrupt() {
print "Process aborted!\n"
exit 130
}
create_script() {
printf "#!/bin/sh\n%s\n" "$1" > "$2"
chmod +x "$SCRIPT_FILE"
}
run_script() {
"$SCRIPT_FILE" >/dev/null 2>&1 &
printf "%d" $!
}
stop_delete_script() {
kill -9 "$1"
rm -f "$2"
}
check_pid() {
PID=$1
LOOP=$2
for _ in $(seq 1 "$(($LOOP*2))"); do
test -d "/proc/$PID/" && sleep 0.5 || STATUS=0
done
printf "%d" ${STATUS:-1}
}
main() {
while getopts "ht:c:" option; do
case "${option}" in
h) usage; exit 0;;
t) TIMEOUT="$OPTARG";;
c) COMMAND="$OPTARG";;
*) usage; exit 1;;
esac
done
TIMEOUT="${TIMEOUT:-10}"
if [ -z "$COMMAND" ]; then
printf "Command is required!\n"
exit 1
fi
SCRIPT_FILE="$(mktemp)"
create_script "$COMMAND" "$SCRIPT_FILE"
SCRIPT_PID=$(run_script "$SCRIPT_FILE")
EXIT_STATUS=$(check_pid "$SCRIPT_PID" "$TIMEOUT")
stop_delete_script "$SCRIPT_PID" "$SCRIPT_FILE"
exit "$EXIT_STATUS"
}
trap on_interrupt INT
main "$@"