python获取交互式ssh shell的方法


Posted in Python onFebruary 14, 2019

更新,最近在学unix环境编程,了解一下进程的创建过程,用最原始的方式实现了一个ssh命令的执行。

#coding=utf8
 
'''
用python实现了一个简单的shell,了解进程创建
类unix 环境下 fork和exec 两个系统调用完成进程的创建
'''
 
import sys, os
 
 
def myspawn(cmdline):
 argv = cmdline.split()
 if len(argv) == 0:
  return 
 program_file = argv[0]
 pid = os.fork()
 if pid < 0:
  sys.stderr.write("fork error")
 elif pid == 0:
  # child
  os.execvp(program_file, argv)
  sys.stderr.write("cannot exec: "+ cmdline)
  sys.exit(127)
 # parent
 pid, status = os.waitpid(pid, 0)
 ret = status >> 8 # 返回值是一个16位的二进制数字,高8位为退出状态码,低8位为程序结束系统信号的编号
 signal_num = status & 0x0F
 sys.stdout.write("ret: %s, signal: %s\n" % (ret, signal_num))
 return ret
 
 
def ssh(host, user, port=22, password=None):
 if password:
  sys.stdout.write("password is: '%s' , plz paste it into ssh\n" % (password))
 cmdline = "ssh %s@%s -p %s " % (user, host, port)
 ret = myspawn(cmdline)
 
 
if __name__ == "__main__":
 host = ''
 user = ''
 password = ''
 ssh(host, user, password=password)

最近在做一个项目,需要在客户端集成一个交互式ssh功能,大概就是客户端跟服务器申请个可用的机器,服务端返回个ip,端口,密码, 然后客户端就可以直接登录到机器上操做了。该程序基于paramiko模块。

经查找,从paramiko的源码包demos目录下,可以看到交互式shell的实现,就是那个demo.py。但是用起来有些bug,于是我给修改了一下interactive.py(我把windows的代码删掉了,剩下的只能在linux下用)。代码如下:

#coding=utf-8
import socket
import sys
import os
import termios
import tty
import fcntl
import signal
import struct
import select
 
now_channel = None
 
def interactive_shell(chan):
 posix_shell(chan)
 
 
def ioctl_GWINSZ(fd):
 try:
  cr = struct.unpack('hh', fcntl.ioctl(fd, termios.TIOCGWINSZ,'aaaa'))
 except:
  return
 return cr
 
 
def getTerminalSize():
 cr = ioctl_GWINSZ(0) or ioctl_GWINSZ(1) or ioctl_GWINSZ(2)
 return int(cr[1]), int(cr[0])
 
 
def resize_pty(signum=0, frame=0):
 width, height = getTerminalSize()
 if now_channel is not None:
  now_channel.resize_pty(width=width, height=height)
 
 
 
def posix_shell(chan):
 global now_channel
 now_channel = chan
 resize_pty()
 signal.signal(signal.SIGWINCH, resize_pty) # 终端大小改变时,修改pty终端大小
 stdin = os.fdopen(sys.stdin.fileno(), 'r', 0) # stdin buff置为空,否则粘贴多字节或者按方向键的时候显示不正确
 fd = stdin.fileno()
 oldtty = termios.tcgetattr(fd)
 newtty = termios.tcgetattr(fd)
 newtty[3] = newtty[3] | termios.ICANON
 try:
  termios.tcsetattr(fd, termios.TCSANOW, newtty)
  tty.setraw(fd)
  tty.setcbreak(fd)
  chan.settimeout(0.0)
  while True:
   try:
    r, w, e = select.select([chan, stdin], [], [])
   except:
    # 解决SIGWINCH信号将休眠的select系统调用唤醒引发的系统中断,忽略中断重新调用解决。
    continue
   if chan in r:
    try:
     x = chan.recv(1024)
     if len(x) == 0:
      print 'rn*** EOFrn',
      break
     sys.stdout.write(x)
     sys.stdout.flush()
    except socket.timeout:
     pass
   if stdin in r:
    x = stdin.read(1)
    if len(x) == 0:
     break
    chan.send(x)
 finally:
  termios.tcsetattr(sys.stdin, termios.TCSADRAIN, oldtty)

使用示例:

#coding=utf8
import paramiko
import interactive
 
 
#记录日志
paramiko.util.log_to_file('/tmp/aaa')
#建立ssh连接
ssh=paramiko.SSHClient()
ssh.load_system_host_keys()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect('192.168.1.11',port=22,username='hahaha',password='********',compress=True)
 
#建立交互式shell连接
channel=ssh.invoke_shell()
#建立交互式管道
interactive.interactive_shell(channel)
#关闭连接
channel.close()
ssh.close()

interactive.py代码中主要修复了几个问题:

1、当读取键盘输入时,方向键会有问题,因为按一次方向键会产生3个字节数据,我的理解是按键一次会被select捕捉一次标准输入有变化,但是我每次只处理1个字节的数据,其他的数据会存放在输入缓冲区中,等待下次按键的时候一起发过去。这就导致了本来3个字节才能完整定义一个方向键的行为,但是我只发过去一个字节,所以终端并不知道我要干什么。所以没有变化,当下次触发按键,才会把上一次的信息完整发过去,看起来就是按一下方向键有延迟。多字节的粘贴也是一个原理。解决办法是将输入缓冲区置为0,这样就没有缓冲,有多少发过去多少,这样就不会有那种显示的延迟问题了。

2、终端大小适应。paramiko.channel会创建一个pty(伪终端),有个默认的大小(width=80, height=24),所以登录过去会发现能显示的区域很小,并且是固定的。编辑vim的时候尤其痛苦。channel中有resize_pty方法,但是需要获取到当前终端的大小。经查找,当终端窗口发生变化时,系统会给前台进程组发送SIGWINCH信号,也就是当进程收到该信号时,获取一下当前size,然后再同步到pty中,那pty中的进程等于也感受到了窗口变化,也会收到SIGWINCH信号。

3、读写‘慢'设备(包括pipe,终端设备,网络连接等)。读时,数据不存在,需要等待;写时,缓冲区满或其他原因,需要等待。ssh通道属于这一类的。本来进程因为网络没有通信,select调用为阻塞中的状态,但是当终端窗口大小变化,接收到SIGWINCH信号被唤醒。此时select会出现异常,触发系统中断(4, 'Interrupted system call'),但是这种情况只会出现一次,当重新调用select方法又会恢复正常。所以捕获到select异常后重新进行select可以解决该问题。

以上这篇python获取交互式ssh shell的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python解惑之整数比较详解
Apr 24 Python
浅谈Python中带_的变量或函数命名
Dec 04 Python
python数字图像处理实现直方图与均衡化
May 04 Python
Python+selenium 获取浏览器窗口坐标、句柄的方法
Oct 14 Python
使用Python实现在Windows下安装Django
Oct 17 Python
Python时间和字符串转换操作实例分析
Mar 16 Python
Python面向对象程序设计类的多态用法详解
Apr 12 Python
Python约瑟夫生者死者小游戏实例讲解
Jan 04 Python
jupyter 添加不同内核的操作
Feb 06 Python
Python WSGI 规范简介
Apr 11 Python
python编写五子棋游戏
May 25 Python
Python实现单例模式的5种方法
Jun 15 Python
对python多线程SSH登录并发脚本详解
Feb 14 #Python
Python Selenium 之关闭窗口close与quit的方法
Feb 13 #Python
python 实现selenium断言和验证的方法
Feb 13 #Python
使用Python自动化破解自定义字体混淆信息的方法实例
Feb 13 #Python
python selenium执行所有测试用例并生成报告的方法
Feb 13 #Python
对python_discover方法遍历所有执行的用例详解
Feb 13 #Python
django2.0扩展用户字段示例
Feb 13 #Python
You might like
discuz authcode 经典php加密解密函数解析
2020/07/12 PHP
php防止恶意刷新与刷票的方法
2014/11/21 PHP
php实现比较全的数据库操作类
2015/06/18 PHP
从性能方面考虑PHP下载远程文件的3种方法
2015/12/29 PHP
PHP实现单文件、多个单文件、多文件上传函数的封装示例
2019/09/02 PHP
麻雀虽小五脏俱全 Dojo自定义控件应用
2010/09/04 Javascript
js中事件的处理与浏览器对象示例介绍
2013/11/29 Javascript
javascript实现全角与半角字符的转换
2015/01/07 Javascript
jQuery焦点图切换特效代码分享
2015/09/15 Javascript
学习javascript文件加载优化
2016/02/19 Javascript
Easyui Treegrid改变默认图标的方法
2016/04/29 Javascript
Bootstrap实现弹性搜索框
2016/07/11 Javascript
JS实现一个简单的日历
2017/02/22 Javascript
vue中子组件传递数据给父组件的讲解
2019/01/27 Javascript
JS页面获取 session 值,作用域和闭包学习笔记
2019/10/16 Javascript
bootstrap-paginator服务器端分页使用方法详解
2020/02/13 Javascript
合并Excel工作薄中成绩表的VBA代码,非常适合教育一线的朋友
2009/04/09 Python
Python抓取Discuz!用户名脚本代码
2013/12/30 Python
Python机器学习之SVM支持向量机
2017/12/27 Python
安装python时MySQLdb报错的问题描述及解决方法
2018/03/20 Python
Python实现判断一行代码是否为注释的方法
2018/05/23 Python
Python使用修饰器进行异常日志记录操作示例
2019/03/19 Python
详解python校验SQL脚本命名规则
2019/03/22 Python
python 字符串常用方法汇总详解
2019/09/16 Python
Unineed旗下时尚轻奢网站:FABHunt
2019/05/13 全球购物
康拓普公司Java笔面试
2016/09/23 面试题
上学迟到的检讨书
2014/01/11 职场文书
关于爱情的广播稿
2014/01/16 职场文书
水毁工程实施方案
2014/04/01 职场文书
软件售后服务方案
2014/05/29 职场文书
预防煤气中毒方案
2014/06/16 职场文书
教师辞职书范文
2015/02/26 职场文书
公司客户答谢酒会祝酒词
2015/08/11 职场文书
农村房屋租赁合同(范本)
2019/07/23 职场文书
详解MySQL的主键查询为什么这么快
2022/04/03 MySQL
使用CSS实现六边形的图片效果
2022/08/05 HTML / CSS