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是否将信号转发给它的孩子？有办法解决这个问题吗？有更简单的方法吗？

感谢您的任何指示。

Answer 1

os.setpgrp（查看setpgrp联机帮助页）否则信号将传播给孩子

Catch＆amp;忽略信号

1 个答案: