timeout/timeout

79 lines
1.2 KiB
Plaintext
Raw Normal View History

2022-12-31 13:25:07 +01:00
#!/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() {
2023-01-01 12:07:17 +01:00
if [ "$2" -eq 0 ]; then
"$SCRIPT_FILE" >/dev/null 2>&1 &
else
"$SCRIPT_FILE" >&2 &
fi
2022-12-31 13:25:07 +01:00
printf "%d" $!
}
stop_delete_script() {
2022-12-31 13:35:14 +01:00
kill -9 "$1"
2022-12-31 13:25:07 +01:00
rm -f "$2"
}
check_pid() {
PID=$1
LOOP=$2
2023-01-01 11:52:13 +01:00
for _ in $(seq 1 "$((LOOP*2))"); do
2023-01-01 11:51:24 +01:00
test -d "/proc/$PID/" && sleep 0.5 || STATUS=0
2022-12-31 13:25:07 +01:00
done
printf "%d" ${STATUS:-1}
}
main() {
2023-01-01 12:07:17 +01:00
while getopts "ht:c:v" option; do
2022-12-31 13:25:07 +01:00
case "${option}" in
h) usage; exit 0;;
t) TIMEOUT="$OPTARG";;
c) COMMAND="$OPTARG";;
2023-01-01 12:07:17 +01:00
v) VERBOSE=1;;
2022-12-31 13:25:07 +01:00
*) usage; exit 1;;
esac
done
TIMEOUT="${TIMEOUT:-10}"
2023-01-01 12:07:17 +01:00
VERBOSE="${VERBOSE:-0}"
2022-12-31 13:25:07 +01:00
if [ -z "$COMMAND" ]; then
printf "Command is required!\n"
exit 1
fi
SCRIPT_FILE="$(mktemp)"
create_script "$COMMAND" "$SCRIPT_FILE"
2023-01-01 12:07:17 +01:00
SCRIPT_PID=$(run_script "$SCRIPT_FILE" "$VERBOSE")
2022-12-31 13:25:07 +01:00
EXIT_STATUS=$(check_pid "$SCRIPT_PID" "$TIMEOUT")
2023-01-01 11:51:24 +01:00
stop_delete_script "$SCRIPT_PID" "$SCRIPT_FILE"
2022-12-31 13:25:07 +01:00
2022-12-31 13:34:22 +01:00
exit "$EXIT_STATUS"
2022-12-31 13:25:07 +01:00
}
trap on_interrupt INT
main "$@"