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是否將信號轉發給它的孩子?有沒有辦法解決這個問題?有沒有更簡單的方法來做到這一點?
感謝您的指點。