1. 程式人生 > >python中模擬進行ssh命令的執行

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網路程式設計的一個功能。