Python编写的ssh客户端[类似putty]

转载请注明出处：http://blog.csdn.net/jmppok/article/details/17588381

windows下可以通过putty以ssh方式连接linux主机。但putty也有一些问题：比如没有保存session,乱码等等。

偶然发现Python也有ssh模块，学习的同时尝试使用python编写一个类似putty的ssh客户端。

1.环境准备

python2.7 + PyCrypto + paramiko + ecdsa

后面三个是python的扩展模块，实现了ssh连接功能。

需要分别下载这三个模块，安装。

PyCrypto下载地址：https://www.dlitz.net/software/pycrypto/

paramiko下载地址：http://www.lag.net/paramiko/

edssa下载地址：https://pypi.python.org/pypi/ecdsa/0.9

安装十分简单，进入模块根目录执行 python setup.py build install命令即可。

说明：linux下可按此安装。windows下我未编译通过，可使用编译好的版本，直接拷贝至Lib\site-packages下即可。

下面是一个windows下编译好的版本：win7下编译好的python ssh模块

2.原理

1）通过paramiko的ssh模块连接指定主机；

2) 通过SSHClient.exec_command在远程主机上执行命令；

3）通过exec_command返回的stdout,stdin,stderr进行交互；

4）保存成功连接的主机信息（session），可以通过ls命令查看， session id命令，直接启动新连接；

5）可在windows和linux下运行，写程序时需要注意他们的差别。

3.代码ssh.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

import os,sys
import paramiko  
import threading  
import platform

curr_ssh = None	
curr_prompt = ">>"

#使用说明		
def printUsage():
	print "    !ls                     :list sessions."
	print "    !session id             :connect session."
	print "    !conn host user password:connect host with user."
	print "    !exit                   :exit."

#连接	
def conn(ip,username,passwd):
	try:
		ssh = paramiko.SSHClient()  
		ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())  
		ssh.connect(ip,22,username,passwd,timeout=5)  
		print "Connect to ",ip," with ",username
		global curr_prompt
		curr_prompt=username+"@"+ip+">>"
		return ssh
	except:
		return None
		
#加载以前的连接信息
sessions=[]
def loadSessions():
	global sessions
	try:
		f = open("sessions")
		sessions = f.readlines()
		f.close()
	except:
		pass

#执行本地命令,ssh.py的命令		
def exe_cmd_local(cmd):
	if(cmd == "!ls"):
		loadSessions()
		global sessions
		i=0
		print "Sessions:"
		for s in sessions:
			print"[%d] %s" %(i,s)
			i+=1
	else:
		vals = cmd.split(' ')
		if(vals[0]=="!session"):
			id = (int)(vals[1])
			if(id
4.使用

可以直接ssh.py,进入主菜单：



通过!conn   hZ喎?http://www.Bkjia.com/kf/ware/vc/" target="_blank" class="keylink">vc3QgIHVzcm5hbWUgIHBhc3N3ZMGsvdOjujwvcD4KPHA+PGltZyBzcmM9"http://www.Bkjia.com/uploadfile/Collfiles/20131227/20131227134751392.jpg" alt="\">


也可以ssh.py  host  usrname  passwd，直接连接指定的客户机。



可以通过!ls 查看保存的Session列表：



通过!session id 直接启动新窗口并连接：



通过!exit退出。



连接后可远程执行shell 命令：



5.问题
1)部分ssh命令无法执行；
2)界面显示太弱，需要改进。