在进行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网络编程的一个功能。