使用Popen运行程序超时就kill

最新推荐文章于 2024-06-26 00:45:00 发布

原创最新推荐文章于 2024-06-26 00:45:00 发布 · 5.3k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #popen #子进程 #输入重定向 #超时kill

python 专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一个Python函数popenKillTimeout，该函数利用Popen启动由gcc编译的小程序，并能在指定时间内向其输入字符串。若程序运行超时，则会强制结束进程；此外，函数还会捕获并返回程序的标准输出与错误信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用Popen运行gcc生成的小程序,超时后就杀掉

使用字符串作为小程序的输入,并返回输出.

def popenKillTimeout(popenargs, stdinStr='', timeOut=1, **kwargs):
    process = sp.Popen(popenargs, stdin=sp.PIPE, stdout=sp.PIPE,
                       stderr=sp.PIPE, preexec_fn=os.setsid, **kwargs)
    process.stdin.write(stdinStr + '\n')
    timeUnit = 0.001
    timeStepNum = int(1 / timeUnit * timeOut)
    for i in xrange(timeStepNum):
        time.sleep(timeUnit)
        if process.poll() is None:
            if i == (timeStepNum - 1):
                print 'kill ' + str(process.pid)
                os.killpg(process.pid, sp.signal.SIGUSR1)
                return -1, '#!timeout!#', 'timeout'
            else:
                continue
        else:
            break
    output, err = process.communicate()
    return process.poll(), output, err