2011-06-14 69 views
3

我正在尝试编写一个程序,该程序将使用ssh在远程计算机上运行程序,并传递它的SIGINT信号而不会终止我的ssh连接。如果我是一个终端上运行,那么这将是很容易的东西,如Catch&Ignore Signal

ssh -t -t host "command to run" 

我用下面的脚本尝试这样做:

#!/bin/bash 

ssh -t -t host "program [email protected]" 

,当我运行从终端能正常工作,但是当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是否将信号转发给它的孩子?有没有办法解决这个问题?有没有更简单的方法来做到这一点?

感谢您的指点。

回答

1

os.setpgrp(看setpgrp手册页)否则信号传播给孩子