码迷,mamicode.com
首页 > 其他好文 > 详细

更改paramiko 源码 记录命令实现堡垒机功能

时间:2016-09-01 14:39:11      阅读:170      评论:0      收藏:0      [点我收藏+]

标签:

利用paramiko 下的demo可以很容易的实现记录客户在操作客户机时的命令,修改\demos\interactive.py

 1 def posix_shell(chan):
 2     import select
 3     oldtty = termios.tcgetattr(sys.stdin)
 4     try:
 5         tty.setraw(sys.stdin.fileno())
 6         tty.setcbreak(sys.stdin.fileno())
 7         chan.settimeout(0.0)
 8 
 9 
10         #这里就是主要改造的地方了
11         cmd_list=[]
#定义日志输出的位置
12 log_file=/home/myshare/demos/test.log 13 f=open(log_file,rb+) 14 while True: 15 r, w, e = select.select([chan, sys.stdin], [], []) 16 17 if chan in r: 18 try: 19 x = u(chan.recv(1024)) 20 if len(x) == 0: 21 sys.stdout.write(\r\n*** EOF\r\n) 22 break 23 sys.stdout.write(x) 24 sys.stdout.flush() 25 except socket.timeout: 26 pass 27 if sys.stdin in r: 28 x = sys.stdin.read(1)
#这个x每次敲击键盘就会记录一次,然后就将输入的内容交给输出的地方,这样才完成1个字符1个字符的显示过程。这个地方仔细想一下或者打印结X果就明白什么意思了
#这里将字符串放入列表中,然后当用户敲回车的时候在将列表中的字符串拼接起来
29 cmd_list.append(x) 30 if x == \r: 31 cmd=‘‘.join(cmd_list).replace(\r,\n) 32 f.write(cmd) 33 f.flush() 34 cmd_list=[] 35 if len(x) == 0: 36 break 37 chan.send(x) 38 39 finally: 40 termios.tcsetattr(sys.stdin, termios.TCSADRAIN, oldtty)

此时方法

[root@controller demos]# ./demo.py root@10.64.8.10  

更改paramiko 源码 记录命令实现堡垒机功能

标签:

原文地址:http://www.cnblogs.com/menkeyi/p/5829565.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!