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网络编程的一个功能。
for linux and python