捕获并忽略信号

Question

我正在尝试写一个程序，想通过ssh在远程机器上运行另一个程序，并且希望能发送SIGINT信号，而不影响我的ssh连接。如果我在终端上运行，这个操作就简单多了，可以用类似下面的方式：

ssh -t -t host "command to run"

我用以下脚本尝试过：

#!/bin/bash

ssh -t -t host "program $@"

当我从终端运行这个脚本时，一切正常，但当proofgeneral运行这个脚本并发送SIGINT信号时，它却直接结束了程序（我猜是因为它无法分配一个终端？）。

我写了以下一堆代码：

CTRL_C = "<CTRL-C>"

def endpoint(proc, handler):
    signal.signal(2, signal.SIG_IGN)

    inp = ""
    out = ""
    err = ""

    inp_from_fd = sys.stdin.fileno()
    inp_to_fd   = proc.stdin.fileno()
    out_from_fd = proc.stdout.fileno()
    out_to_fd   = sys.stdout.fileno()
    err_from_fd = proc.stderr.fileno()
    err_to_fd   = sys.stderr.fileno()

    while True:
       try:
           ins,outs,_ = select.select([inp_from_fd, out_from_fd, err_from_fd],
                                      [inp_to_fd, out_to_fd, err_to_fd],
                                      [])
           for fd in ins:
               if fd == inp_from_fd:
                   k   = os.read(fd, 1024)
                   if k == "":
                       os.close(proc.stdin.fileno())
                       return
                   inp = inp + k
               elif fd == out_from_fd:
                   k   = os.read(fd, 1024)
                   out = out + k
               elif fd == err_from_fd:
                   k   = os.read(fd, 1024)
                   err = err + k
               else:
                   assert False
           for fd in outs:
               if fd == inp_to_fd:
                   while CTRL_C in inp:
                       proc.send_signal(2)
                       p = inp.find(CTRL_C)
                       inp = inp[0:p] + inp[p+len(CTRL_C):]
                   k = os.write(fd, inp)
                   inp = inp[k:]
               elif fd == out_to_fd:
                   k = os.write(fd, out)
                   out = out[k:]
               elif fd == err_to_fd:
                   k = os.write(fd, err)
                   err = err[k:]
               else:
                   assert False
       except select.error:
           pass
       except KeyboardInterrupt:
           handler()
       except IOError, e:
           pass

def usage(args):
    print "must specify --client or --server" 

if __name__ == '__main__':
    if len(sys.argv) == 1:
        usage(sys.argv)
    elif sys.argv[1] == '--server':
        proc = subprocess.Popen(sys.argv[2:],
                                stdin=subprocess.PIPE,
                                stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        def INT():
            proc.stdin.write(CTRL_C)
            proc.stdin.flush()
        endpoint(proc, INT)
    elif '--client' in sys.argv:
        proc = subprocess.Popen(sys.argv[2:],
                                stdin=subprocess.PIPE,
                                stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        import time
        time.sleep(1)
        def INT():
            pass
        endpoint(proc, INT)
    else:
        usage(sys.argv)

我用类似下面的方式调用它：

remote.py --client ssh -t -t host "remote.py --server <program-to-run> <args>"

我在处理信号时是不是做错了什么？我试着在信号2的处理函数里加了个打印，确实能打印出来，但同时也杀掉了ssh连接（我在控制台上看到“被信号2杀死。”的提示）。是python把信号转发给它的子进程了吗？有没有办法解决这个问题？有没有更简单的方法可以做到这一点？

谢谢任何建议。

信号处理程序终止子进程管理控制台输出 ssh连接远程执行 sigint信号终端分配

捕获并忽略信号

1 个回答

撰写回答