2011-06-14 66 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手冊頁)否則信號傳播給孩子