python中模擬進行ssh命令的執行

在進行socket編程的時候，可以實現遠程執行命令，然后返回相關的結果，但是這種。。。很容易就把服務器搞掛了。

在這里需要用到commands模塊，commands模塊中有一個方法為getstatusoutput，主要就是返回執行的返回碼和結果，如下所示：

>>> import commands
>>> commands.getstatusoutput('ls')
(0, 'client.py\nserver.py')

>>> commands.getstatusoutput('kel')
(32512, 'sh: kel: command not found')
>>> 

返回的結果是一個元組，第一個為返回碼，第一個返回值0表示命令成功執行，其他的表示為未能正確執行；第二個返回值表示返回的結果，如果命令執行失敗，那就是返回命令執行的錯誤信息。

服務器端代碼如下：

#!/usr/bin/env python

import commands
import SocketServer

class MyHandler(SocketServer.BaseRequestHandler):
    def handle(self):
        while True:
            data = self.request.recv(1024)
            if not data: break
            cmd = data.strip()
            status,output = commands.getstatusoutput(cmd)

            #send the output size
            self.request.sendall(str(len(output)))

            #sent the content
            self.request.sendall(output)

HOST = '192.168.1.60'
PORT = 9999

s = SocketServer.ThreadingTCPServer((HOST,PORT),MyHandler)
print 'waiting for connection...'
s.serve_forever()

在服務器端代碼中，在發送的時候，需要發送需要發送的結果的大小，從而發送了兩次信息，第一次發送的是文件的大小，第二次發送的是文件的內容。

客戶端代碼如下：

#!/usr/bin/env python
import socket

s = socket.socket()
HOST = '192.168.1.60'
PORT = 9999

def recv_all(s,file_size):
    results = ''
    file_size = int(file_size)
    while file_size > 0:
        if file_size <= 1024:
            data = s.recv(1024)
            results += data
            break
        elif file_size > 1024:
            data = s.recv(1024)
            results += data
            file_size -= 1024
    return results

s.connect((HOST,PORT))
while True:
    data = raw_input('>>>')
    if not data:break
    if data.strip() == 'exit':
        break
    s.sendall(data)
    if not data:break
    #first time get the size of the file
    file_size = s.recv(1024)
    contens = recv_all(s,file_size)
    print contens
s.close()

在進行接收的時候，第一次接收了發過來的文件的大小，然后在進行調用函數recv_all接收所有的文件，在此函數中，主要就是根據緩沖區的大小，將文件進行接收，在這里緩沖區的大小是有設置的，設置為1024，從而每次判斷文件的大小，進行接收剩下的文件內容。

在這里基本實現了ssh的命令執行功能，也就是在服務器端執行相關的命令，然后返回結果。

主要在進行命令執行的時候，直接使用了commands模塊的getstatusoutput方法。

其實在進行連接ssh時候，python也是可以做到的，密碼登錄驗證，使用paramiko進行連接，好像都是可以直接進行遠程操作，不過這也就是socket網絡編程的一個功能。