python中模擬進行ssh命令的執行
阿新 • • 發佈:2019-02-14
在進行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()
客戶端程式碼如下:
在進行接收的時候,第一次接收了發過來的檔案的大小,然後在進行呼叫函式recv_all接收所有的檔案,在此函式中,主要就是根據緩衝區的大小,將檔案進行接收,在這裡緩衝區的大小是有設定的,設定為1024,從而每次判斷檔案的大小,進行接收剩下的檔案內容。#!/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()
在這裡基本實現了ssh的命令執行功能,也就是在伺服器端執行相關的命令,然後返回結果。
主要在進行命令執行的時候,直接使用了commands模組的getstatusoutput方法。
其實在進行連線ssh時候,python也是可以做到的,密碼登入驗證,使用paramiko進行連線,好像都是可以直接進行遠端操作,不過這也就是socket網路程式設計的一個功能。