Paramiko 和 exec_command - 杀死远程进程？

Question

我正在使用 Paramiko tail -f 远程服务器上的文件。

以前，我们通过 ssh -t 运行此命令，但事实证明这很不稳定，而且 -t 导致我们的远程调度系统出现问题。

我的问题是当脚本捕获到 SIGINT 时如何杀死 tail？

我的脚本（基于长-在 python paramiko 模块中运行 ssh 命令（以及如何结束它们））

#!/usr/bin/env python2
import paramiko
import select

client = paramiko.SSHClient()
client.load_system_host_keys()
client.connect('someserver', username='victorhooi', password='blahblah')
transport = client.get_transport()
channel = transport.open_session()

channel.exec_command("tail -f /home/victorhooi/macbeth.txt")
while True:
    try:
        rl, wl, xl = select.select([channel],[],[],0.0)
        if len(rl) > 0:
            # Must be stdout
            print channel.recv(1024)
    except KeyboardInterrupt:
        print("Caught control-C")
        client.close()
        channel.close()
        exit(0)

脚本成功捕获我的 Ctrl-C 并结束。但是，它会使 tail -f 进程在远程系统上运行。

client.close() 和channel.close() 似乎都没有终止它。

我可以在 except 块中发出什么命令来杀死它？

远程服务器正在运行 Solaris 10。

Answer 1

您应该使用 ssh keepalives...您遇到的问题是远程 shell 无法知道（默认情况下）您的 ssh 会话已被终止。 Keepalives 将使远程 shell 能够检测到您终止了会话

client = paramiko.SSHClient()
client.load_system_host_keys()
client.connect('someserver', username='victorhooi', password='blahblah')
transport = client.get_transport()
transport.set_keepalive(1)   # <------------------------------
# ... carry on as usual...

将 keepalive 值设置为您喜欢的最低值（甚至 1 秒）...几秒钟后，远程 shell 将看到 ssh 登录已终止，并将终止任何会话由它产生的进程。

Answer 2

有一种方法可以做到这一点。它的工作方式就像在 shell 上一样。

ssh -t commandname

选项 -t 打开一个伪 pty 来帮助 ssh 跟踪这个过程应该持续多长时间。同样的事情可以通过 pormiko

channel.get_pty()

在execute_command(...)之前完成。这不会像使用channel.invoke_shell()那样打开shell，它只是请求这样一个伪接口来绑定所有进程。如果在远程计算机上发出 ps aux，也可以看到效果，该进程现在通过 ptxXY 接口连接到 sshd。

Answer 3

我刚刚遇到这个问题，但无法最终发出 pkill 来关闭该进程。

更好的解决方案是将您正在运行的命令更改为：

tail -f /path/to/file & { read ; kill %1; }

这将允许您根据需要运行 tail 命令。一旦您向远程进程发送换行符，kill %1 就会执行并停止您后台运行的 tail 命令。 （供参考：%1 是一个作业规范，用于描述会话中第一个后台进程，即 tail 命令）

Answer 4

以下是获取远程进程 ID 的方法：

def execute(channel, command):
    command = 'echo $; exec ' + command
    stdin, stdout, stderr = channel.exec_command(command)
    pid = int(stdout.readline())
    return pid, stdin, stdout, stderr

以下是如何使用它（将 ... 替换为原始问题中的位）：

pid, _, _, _ = execute(channel, "tail -f /home/victorhooi/macbeth.txt")
while True:
    try:
        # ...
    except KeyboardInterrupt:
        client.exec_command("kill %d" % pid)
        # ...

Answer 5

虽然不是最有效的方法，但这应该可行。 CTRL+C之后；在 KeyboardInterrupt 处理程序中，您可以像这样 exec_command("killall -u %s tail" % uname) ：

#!/usr/bin/env python2

import paramiko
import select

import time
ltime = time.time()

# Or use random:
# import random
# ltime = random.randint(0, 500)

uname = "victorhooi"
client = paramiko.SSHClient()
client.load_system_host_keys()
client.connect('someserver', username=uname, password='blahblah')
transport = client.get_transport()
channel = transport.open_session()

channel.exec_command("tail -%df /home/victorhooi/macbeth.txt" % ltime)
while True:
    try:
        rl, wl, xl = select.select([channel],[],[],0.0)
        if len(rl) > 0:
            # Must be stdout
            print channel.recv(1024)
    except KeyboardInterrupt:
        print("Caught control-C")
        channel.close()
        try:
            # open new socket and kill the proc..
            client.get_transport().open_session().exec_command("kill -9 `ps -fu %s | grep 'tail -%df /home/victorhooi/macbeth.txt' | grep -v grep | awk '{print $2}'`" % (uname, ltime))
        except:
            pass
    
        client.close()
        exit(0)

这将杀死所有名为 tail 的打开进程。如果您打开了 tail 并且不想关闭，这可能会导致问题，如果是这种情况，您可以 grep a ps，得到pid 并 kill -9 它。

首先，将 tail 设置为在跟随之前从文件末尾读取 n 行。将 n 设置为一个唯一的数字，例如 time.time()，因为 tail 不关心该数字是否大于文件中的行数，即大数字来自 time.time() 不应引起问题并且是唯一的。然后 grep 查找 ps 中的唯一编号：

   client.get_transport().open_session().exec_command("kill -9 `ps -fu %s | grep 'tail -%df /home/victorhooi/macbeth.txt' | grep -v grep | awk '{print $2}'`" % (uname, ltime))

Answer 6

ssh -t 也有同样的问题。有一个名为 closer 的库 - 它通过 ssh 运行远程进程并自动为您关闭。一探究竟。

Answer 7

特别是对于“tail”，您可以使用 --pid=PID 参数并让 tail 处理它：

  --pid=PID  with -f, terminate after process ID, PID dies

Answer 8

您可以按照 https://stackoverflow.com/a/38883662/565212 中的说明使用 get_pty。

例如场景 - 何时调用 client/channel.close():

第 1 步：执行写入日志文件的远程命令。

第2步：生成一个执行tail命令并阻塞在readline循环中的线程

第三步：在主线程中，当命令返回时，您知道不会有更多日志，杀死尾线程。