paramiko库exec_command长时间执行命令后通道失活的问题

使用paramiko库远程连接服务器并执行命令,发现exec_command长时间运行后,channel.recv无法获取服务器数据的问题。

确认服务器脚本可以正常运行完毕,但channel.recv始终拉取不到最后的返回数据。

尝试增加缓存以及设置非阻塞模式运行后,依旧无法解决此问题。

调用命令监控服务器 TCP / IP 网络(grep :22,仅查看ssh连接的网络端口)

netstat -nt | grep :22

发现程序阻塞期间,服务器Send-Q存在数据,且并未被拉取。

可推测此时channel通道应该是出现了问题,导致无法获取服务器最后的返回数据。

考虑到exec_command调用服务器脚本命令期间有超过10分钟缓存队列是没有数据回显的,推测是通道长时间无数据返回后,通道失活导致的问题。

查阅paramiko官方使用文档https://docs.paramiko.org/en/latest/
发现transport模块下有一个属性set_keepalive,可以设置每隔一段时间发送保活数据包使通道保持活跃状态。

set_keepalive (间隔) 打开/关闭保活数据包(默认为关闭)。
参数:间隔( int ) – 发送保活数据包之前等待的秒数(或 0 禁用保活)。
如果设置了此项,interval几秒钟后没有通过连接发送任何数据,将发送一个“keepalive”数据包(并被远程主机忽略)。
例如,这对于保持 NAT上的连接处于活动状态非常有用。

优化代码如下

# 创建SSHClient 实例对象
ssh = paramiko.SSHClient()
# 设置信任远程机器,允许连接不在know_hosts文件中的主机,允许访问
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
# 连接服务器,地址、端口、用户名、密码,设置连接超时
ssh.connect(ipaddr, port=port, username=username, password=pwd, timeout=10)
# stdout 为正确输出,stderr为错误输出,同时仅有1个变量有值
stdin, stdout, stderr = ssh.exec_command(cmd)
# 获取通道并设置非阻塞模式
channel = stdout.channel
channel.setblocking(False)
# 获取运输,设置每隔60秒发送keepalive 
transport = channel.get_transport()
transport.set_keepalive(60)
while not channel.exit_status_ready():
    try:
        time.sleep(1)
        # 接收到的服务器返回值,设置缓存65535
        output = channel.recv(65535)
        # 查看命令执行的情况,防止中文乱码使用'utf-8'解码,忽略错误的字节
        output = output.decode('utf-8', 'ignore')
        print(output, end='')
    except TimeoutError:
        print(datetime.datetime.now(), " 没有数据可读(TimeoutError)")
        time.sleep(10)
        pass
    except paramiko.SSHException as e:
        print("读取服务器连接异常 ", e)
        break
    except Exception as e:
        print("读取服务器未知异常 ", e)
        break
print('指令执行完毕(channel.exit_status_ready())')
# 关闭SSH连接
transport.close()
channel.close()
ssh.close()
print('ssh连接已关闭')

注意:
channel 是否设置非阻塞模式视个人情况而定。
channel.setblocking(False) 非阻塞模式下,若channel.recv无数据返回则会报TimeoutError。

  • 8
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值