命令行命令在一定时间后自动终止命令

Question

我想在一定时间后自动终止命令。我想到了这样的界面：

% constrain 300 ./foo args

它会运行带有“args”的“./foo”，但如果它在 5 分钟后仍在运行，则会自动终止它。

将这个想法推广到其他约束可能很有用，例如如果进程使用过多内存则自动终止进程。

是否有任何现有的工具可以做到这一点，或者有没有人写过这样的东西？

添加：乔纳森的解决方案正是我的想法，它在 linux 上就像一个魅力，但我无法让它在 Mac OSX 上工作。我摆脱了 SIGRTMIN 让它编译得很好，但信号只是没有被发送到子进程。有人知道如何在 Mac 上进行这项工作吗？

[添加：请注意，Jonathan 提供了适用于 Mac 和其他地方的更新。]

Answer 1

我有一个名为 timeout 的程序可以做到这一点 - 用 C 语言编写，最初是在 1989 年，但此后定期更新。

---

更新：此代码无法在 MacOS X 上编译，因为未定义 SIGRTMIN，并且在 MacOS X 上运行时无法超时，因为那里的 `signal()` 函数在警报超时后恢复 `wait()` - 这不是所需的行为。我有一个新版本的 `timeout.c` 可以解决这两个问题（使用 `sigaction()` 而不是 `signal()`）。和以前一样，请与我联系以获取 10K gzipped tar 文件，其中包含源代码和手册页（请参阅我的个人资料）。

---

/*
@(#)File:           $RCSfile: timeout.c,v $
@(#)Version:        $Revision: 4.6 $
@(#)Last changed:   $Date: 2007/03/01 22:23:02 $
@(#)Purpose:        Run command with timeout monitor
@(#)Author:         J Leffler
@(#)Copyright:      (C) JLSS 1989,1997,2003,2005-07
*/

#define _POSIX_SOURCE       /* Enable kill() in <unistd.h> on Solaris 7 */
#define _XOPEN_SOURCE 500

#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <errno.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include "stderr.h"

#define CHILD       0
#define FORKFAIL    -1

static const char usestr[] = "[-vV] -t time [-s signal] cmd [arg ...]";

#ifndef lint
/* Prevent over-aggressive optimizers from eliminating ID string */
const char jlss_id_timeout_c[] = "@(#)$Id: timeout.c,v 4.6 2007/03/01 22:23:02 jleffler Exp $";
#endif /* lint */

static void catcher(int signum)
{
    return;
}

int main(int argc, char **argv)
{
    pid_t   pid;
    int     tm_out;
    int     kill_signal;
    pid_t   corpse;
    int     status;
    int     opt;
    int     vflag = 0;

    err_setarg0(argv[0]);

    opterr = 0;
    tm_out = 0;
    kill_signal = SIGTERM;
    while ((opt = getopt(argc, argv, "vVt:s:")) != -1)
    {
        switch(opt)
        {
        case 'V':
            err_version("TIMEOUT", &"@(#)$Revision: 4.6 $ ($Date: 2007/03/01 22:23:02 $)"[4]);
            break;
        case 's':
            kill_signal = atoi(optarg);
            if (kill_signal <= 0 || kill_signal >= SIGRTMIN)
                err_error("signal number must be between 1 and %d\n", SIGRTMIN - 1);
            break;
        case 't':
            tm_out = atoi(optarg);
            if (tm_out <= 0)
                err_error("time must be greater than zero (%s)\n", optarg);
            break;
        case 'v':
            vflag = 1;
            break;
        default:
            err_usage(usestr);
            break;
        }
    }

    if (optind >= argc || tm_out == 0)
        err_usage(usestr);

    if ((pid = fork()) == FORKFAIL)
        err_syserr("failed to fork\n");
    else if (pid == CHILD)
    {
        execvp(argv[optind], &argv[optind]);
        err_syserr("failed to exec command %s\n", argv[optind]);
    }

    /* Must be parent -- wait for child to die */
    if (vflag)
        err_remark("time %d, signal %d, child PID %u\n", tm_out, kill_signal, (unsigned)pid);
    signal(SIGALRM, catcher);
    alarm((unsigned int)tm_out);
    while ((corpse = wait(&status)) != pid && errno != ECHILD)
    {
        if (errno == EINTR)
        {
            /* Timed out -- kill child */
            if (vflag)
                err_remark("timed out - send signal %d to process %d\n", (int)kill_signal, (int)pid);
            if (kill(pid, kill_signal) != 0)
                err_syserr("sending signal %d to PID %d - ", kill_signal, pid);
            corpse = wait(&status);
            break;
        }
    }

    alarm(0);
    if (vflag)
    {
        if (corpse == (pid_t) -1)
            err_syserr("no valid PID from waiting - ");
        else
            err_remark("child PID %u status 0x%04X\n", (unsigned)corpse, (unsigned)status);
    }

    if (corpse != pid)
        status = 2; /* I don't know what happened! */
    else if (WIFEXITED(status))
        status = WEXITSTATUS(status);
    else if (WIFSIGNALED(status))
        status = WTERMSIG(status);
    else
        status = 2; /* I don't know what happened! */

    return(status);
}

如果您想要“stderr.h”和“stderr.c”的“官方”代码，请联系我（查看我的个人资料）。

Answer 2

GNU Coreutils 包含 timeout 命令，默认安装在许多系统上。

https://www.gnu.org/software/coreutils/manual/html_node/timeout-invocation.html

观看free -m 一分钟，然后通过发送 TERM 信号将其杀死：

timeout 1m watch free -m

Answer 3

我参加这个聚会很晚，但我没有在答案中看到我最喜欢的把戏。

在 *NIX 下，alarm(2) 继承于 execve(2)，默认情况下 SIGALRM 是致命的。因此，您通常可以简单地：

$ doalarm () { perl -e 'alarm shift; exec @ARGV' "$@"; } # define a helper function

$ doalarm 300 ./foo.sh args

或者安装一个trivial C wrapper 来为你做这件事。

优势 只涉及一个PID，机制简单。例如，如果./foo.sh 退出“太快”并且其 PID 被重新使用，您将不会杀死错误的进程。您不需要多个 shell 子进程协同工作，这可以正确完成，但比较容易竞争。

缺点 时间受限的进程无法操纵其闹钟（例如，alarm(2)、ualarm(2)、setitimer(2)），因为这可能会清除继承的闹钟。显然，它也不能阻止或忽略 SIGALRM，尽管对于其他一些方法，SIGINT、SIGTERM 等也可以这样说。

一些（我认为非常古老）系统以alarm(2) 的形式实现sleep(2)，即使在今天，一些程序员也使用alarm(2) 作为I/O 和其他操作的粗略的内部超时机制。但是，根据我的经验，这种技术适用于您想要限制时间的绝大多数流程。

Answer 4

有没有办法用“at”设置一个特定的时间来做到这一点？

$ at 05:00 PM kill -9 $pid

看起来简单多了。

如果您不知道 pid 号是多少，我认为有一种方法可以使用 ps aux 和 grep 编写脚本读取它，但不确定如何实现。

$   | grep someprogram
tony     11585  0.0  0.0   3116   720 pts/1    S+   11:39   0:00 grep someprogram
tony     22532  0.0  0.9  27344 14136 ?        S    Aug25   1:23 someprogram

您的脚本必须读取 pid 并为其分配一个变量。 我不是太熟练，但假设这是可行的。

Answer 5

#!/bin/sh
( some_slow_task ) & pid=$!
( sleep $TIMEOUT && kill -HUP $pid ) 2>/dev/null & watcher=$!
wait $pid 2>/dev/null && pkill -HUP -P $watcher

观察者在给定超时后杀死慢任务；脚本等待慢任务并终止观察者。

示例：

• 慢速任务运行超过 2 秒并被终止

慢任务被打断

( sleep 20 ) & pid=$!
( sleep 2 && kill -HUP $pid ) 2>/dev/null & watcher=$!
if wait $pid 2>/dev/null; then
    echo "Slow task finished"
    pkill -HUP -P $watcher
    wait $watcher
else
    echo "Slow task interrupted"
fi

• 这个缓慢的任务在给定的超时之前完成

慢任务完成

( sleep 2 ) & pid=$!
( sleep 20 && kill -HUP $pid ) 2>/dev/null & watcher=$!
if wait $pid 2>/dev/null; then
    echo "Slow task finished"
    pkill -HUP -P $watcher
    wait $watcher
else
    echo "Slow task interrupted"
fi

Answer 6

我对 perl 单行的变体为您提供了退出状态，而无需使用 fork() 和 wait() 进行处理，也没有杀死错误进程的风险：

#!/bin/sh
# Usage: timelimit.sh secs cmd [ arg ... ]
exec perl -MPOSIX -e '$SIG{ALRM} = sub { print "timeout: @ARGV\n"; kill(SIGTERM, -$$); }; alarm shift; $exit = system @ARGV; exit(WIFEXITED($exit) ? WEXITSTATUS($exit) : WTERMSIG($exit));' "$@"

基本上 fork() 和 wait() 隐藏在 system() 中。 SIGALRM 被传递给父进程，然后父进程通过将 SIGTERM 发送到整个进程组 (-$$) 来杀死它自己和它的子进程。万一子进程退出并且子进程的 pid 在 kill() 发生之前被重用，这不会杀死错误的进程，因为具有旧子进程 pid 的新进程不会在父 perl 进程的同一进程组中.

作为一个额外的好处，脚本也会以可能正确的退出状态退出。

Answer 7

从源代码编译以在 Mac 上工作时来自 Ubuntu/Debian 的超时命令。达尔文

10.4.*

http://packages.ubuntu.com/lucid/timeout

Answer 8

我使用“timelimit”，它是 debian 存储库中提供的一个包。

http://devel.ringlet.net/sysutils/timelimit/

Answer 9

纯 bash：

#!/bin/bash

if [[ $# < 2 ]]; then
  echo "Usage: $0 timeout cmd [options]"
  exit 1
fi

TIMEOUT="$1"
shift

BOSSPID=$$

(
  sleep $TIMEOUT
  kill -9 -$BOSSPID
)&
TIMERPID=$!

trap "kill -9 $TIMERPID" EXIT

eval "$@"

Answer 10

使用expect工具怎么样？

## run a command, aborting if timeout exceeded, e.g. timed-run 20 CMD ARGS ...
timed-run() {
  # timeout in seconds
  local tmout="$1"
  shift
  env CMD_TIMEOUT="$tmout" expect -f - "$@" <<"EOF"
# expect script follows
eval spawn -noecho $argv
set timeout $env(CMD_TIMEOUT)
expect {
   timeout {
      send_error "error: operation timed out\n"
      exit 1
   }
   eof
}
EOF
}

Answer 11

尝试类似：

# This function is called with a timeout (in seconds) and a pid.
# After the timeout expires, if the process still exists, it attempts
# to kill it.
function timeout() {
    sleep $1
    # kill -0 tests whether the process exists
    if kill -0 $2 > /dev/null 2>&1 ; then
        echo "killing process $2"
        kill $2 > /dev/null 2>&1
    else
        echo "process $2 already completed"
    fi
}

<your command> &
cpid=$!
timeout 3 $cpid
wait $cpid > /dev/null 2>&
exit $?

它的缺点是如果您的进程的 pid 在超时内被重用，它可能会杀死错误的进程。这不太可能，但您可能每秒启动 20000 多个进程。这可以修复。

Answer 12

对 perl 单行稍作修改将获得正确的退出状态。

perl -e '$s = shift; $SIG{ALRM} = sub { print STDERR "Timeout!\n"; kill INT => $p; exit 77 }; exec(@ARGV) unless $p = fork; alarm $s; waitpid $p, 0; exit ($? >> 8)' 10 yes foo

基本上exit($?>>8)会转发子进程的退出状态。我只是在退出状态选择了77超时。

Answer 13

也许我不理解这个问题，但这听起来直接可行，至少在 bash 中：

( /path/to/slow command with options ) & sleep 5 ; kill $!

这会在括号内运行第一个命令五秒钟，然后将其终止。整个操作同步运行，即当它忙于等待慢速命令时，您将无法使用您的 shell。如果这不是您想要的，应该可以添加另一个 &。

$! 变量是一个 Bash 内置函数，其中包含最近启动的子 shell 的进程 ID。括号内不要有 & 很重要，这样做会丢失进程 ID。

Answer 14

Perl 一个班轮，只是为了好玩：

perl -e '$s = shift; $SIG{ALRM} = sub { print STDERR "Timeout!\n"; kill INT => $p }; exec(@ARGV) unless $p = fork; alarm $s; waitpid $p, 0' 10 yes foo

这会打印 'foo' 十秒钟，然后超时。将“10”替换为任意秒数，将“yes foo”替换为任意命令。

Answer 15

还有ulimit，可以用来限制子进程可用的执行时间。

ulimit -t 10

将进程限制为 10 秒的 CPU 时间。

要实际使用它来限制新进程，而不是当前进程，您可能希望使用包装脚本：

#! /usr/bin/env python

import os
os.system("ulimit -t 10; other-command-here")

other-command 可以是任何工具。我正在运行不同排序算法的 Java、Python、C 和 Scheme 版本，并记录它们花费的时间，同时将执行时间限制为 30 秒。一个 Cocoa-Python 应用程序生成了各种命令行（包括参数）并将时间整理到一个 CSV 文件中，但它实际上只是在上面提供的命令之上的绒毛。