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<len(sessions)):
				os_name = platform.system()
				
				new_console_cmd = ""
				if(os_name == "Linux"):
					new_console_cmd="gnome-terminal -e \"./ssh.py " + sessions[id]+"\""
				elif(os_name == "Windows"):
					new_console_cmd="start ssh.py " + sessions[id]
				os.system(new_console_cmd)
			else:
				print "Didn't hava sessoin ",vals[1]
		elif(vals[0]=="!conn"):
			global curr_ssh
			curr_ssh = conn(vals[1],vals[2],vals[3])
			f = open("sessions","a")
			line = vals[1]+" "+vals[2]+" "+vals[3]+"\n"
			f.write(line)
			f.close()

#在ssh鏈接的主機上執行命令			
def exe_cmd_ssh(ssh,cmd):
	if(ssh == None):
		print "Didn't connect to a server. Use '!conn' to connect please."
		return
	stdin, stdout, stderr = ssh.exec_command(cmd)  
	#stdin.write("Y")   #簡單交互，輸入 ‘Y’   
	#屏幕輸出  
	print stdout.read()
	print stderr.read()
		
#入口函數
if __name__=='__main__':
	loadSessions()
	if(len(sys.argv)==4):
		curr_ssh = conn(sys.argv[1],sys.argv[2],sys.argv[3])
	else:
		printUsage()
	while True:
		cmd = raw_input(curr_prompt)
		if(len(cmd)==0):
			continue
		
		if(cmd == "!exit"):
			if(curr_ssh != None):
				curr_ssh.close();
			break
		else:
			if(cmd[0] == '!'):
				exe_cmd_local(cmd)
			else:
				exe_cmd_ssh(curr_ssh,cmd)

4.使用

能夠直接ssh.py,進入主菜單：

經過!conn   host  usrname  passwd鏈接：

也能夠ssh.py  host  usrname  passwd，直接鏈接指定的客戶機。

能夠經過!ls 查看保存的Session列表：

經過!session id 直接啓動新窗口並鏈接：

經過!exit退出。

鏈接後可遠程執行shell 命令：

5.問題

1)部分ssh命令沒法執行；

2)界面顯示太弱，須要改進。