Python 远程执行命令(Socket)

一、远程执行命令

1.服务器端：

import subprocess
import socket
sk=socket.socket()
address=('127.0.0.1',8080)
sk.bind(address)
sk.listen(3)
while True:
    conn,add=sk.accept()
    print(conn,add)
    while True:
        try:
            data=conn.recv(1024)
        except Exception:
            break

        print('%s'%(str(data,'utf8')))
        if not data:
            conn.close()
            conn,add=sk.accept()
            print(add)
            continue
        # inp=input('>>>')
        a=subprocess.Popen(str(data,'utf8'),shell=True,stdout=subprocess.PIPE)
        cmd_shell=a.stdout.read()
        conn.send(cmd_shell)
sk.close()

解析：使用subprocess模块可以远程执行shell命令，但是会启动另外一个进程不与主进程同步，使用stdout=subprocess.PIPE参数可以使其返回主进程

a.stdout.read()返回值就是bytes类型，可以直接发送，不用做类型转换

2.客户端：

import socket
sk=socket.socket()
address=('127.0.0.1',8080)
sk.connect(address)
while True:
    inp=input('>>>')
    if inp=='exit':
        break
    sk.send(bytes(inp,'utf8'))
    data=sk.recv(1024)
    # print('%s'%(str(data,'utf8')))
    print(str(data,'gbk'))
sk.close()

当客户端接收到服务器端返回的值时，需要使用str(data,'gbk')来编码，因为windows中cmd窗口默认为GBK编码

注意：当传送数据超过1024字节时，会出现接收不完整，串行接收的情况

二、优化版本

如何确保完整接收服务器端返回的数据

优化方案：首先发送数据长度，客户端按照服务器端首次返回的数据长度去接收第二次返回的数据，

当分批次接收的数据累加长度为第一次返回的数据时，则意味着已经接收完毕

1.服务器端

#__author: Lobin
#__date:   2018/1/19
import subprocess
import socket
sk=socket.socket()
address=('127.0.0.1',8080)
sk.bind(address)
sk.listen(3)
while True:
    conn,add=sk.accept()
    print(conn,add)
    while True:
        try:
            data=conn.recv(1024)
        except Exception:
            break
        print('%s'%(str(data,'utf8')))
        if not data:
            conn.close()
            conn,add=sk.accept()
            print(add)
            continue
        a=subprocess.Popen(str(data,'utf8'),shell=True,stdout=subprocess.PIPE)
        cmd_shell=a.stdout.read()
        result_len=bytes(str(len(cmd_shell)),'utf8')
        conn.send(result_len)
        conn.send(cmd_shell)
sk.close()

传送的数据一定为bytes类型，分两次发送数据，一次数据为二次数据的长度

2.客户端

#__author: Lobin
#__date:   2018/1/19
import socket
sk=socket.socket()
address=('127.0.0.1',8080)
sk.connect(address)
while True:
    inp=input('>>>')
    if inp=='exit':
        break
    sk.send(bytes(inp,'utf8'))
    result_len=int(str(sk.recv(1024),'utf8'))
    print(result_len)
    data=bytes()
    while result_len!=len(data):
        recv=sk.recv(1024)
        data+=recv
    print(str(data,'gbk'))
sk.close()

分两次接收，第一次数据为第二次接收数据的长度，初始化一个接收数据bytes类型

3.粘包现象

当连续两次send数据时，CPU可能将两次合并为一次来发送，故上述优化方法可能存在报错

解决方案为：在两次send中间插一条recv，当然客户端需要在第一次recv后加一个send任意数据