python中模拟进行ssh命令的执行

时间：2023-03-09 19:18:44
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网络编程的一个功能。

相关文章