使用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。