Python编写远程控制工具--被控端的编写

#!/usr/bin/python3

# -*- coding: utf-8 -*-

import socket

import struct

import os

import subprocess

if __name__ == '__main__ ' :

# 连接主控端

clientSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

clientSocket.connect(('127.0.0.1 ', 6666))

# 发送被控端的主机名

hostName = subprocess .check_output("hostname")

clientSocket.sendall(hostName)

# 等待主控端指令

print("[*]Waiting instruction . . .")

while True:

# 接收主控端的指令，并进入相应的模块

# 接收到的内容为bytes型，需要将decode转换为str型

instruction = clientSocket.recv(10) .decode()

if instruction == '1 ' :

Execommand(clientSocket)

elif instruction == '2 ' :

TransferFiles(clientSocket)

elif instruction == 'exit ' :

break

else:

pass

clientSocket.close()

2）编写命令执行函数。被控端接收主控端的命令执行，并将命令执行的指  令进行命令、参数分割。因为subprocess模块不能跨工作目录执行命令，所以对一 些命令需要通过其他手段实现。例如，cd命令是无法通过subprocess实现的，执行 cd命令会导致被控端报错，所以需要用os.chdir来代替cd命令。

同时，我们还要增加try{ … }expect{ … }语句，当被控端报错时能及时捕获并 处理错误，重新进入命令执行的循环，提高被控端的可靠性：

3）编写文件传输主函数。因为文件传输包含文件上传和文件下载。我们让 文件传输主函数对文件传输命令进行命令、参数的分割。若命令为upload ，则调 用文件上传函数，若命令为download ，则调用文件下载函数：

# 文件传输函数

def TransferFiles(clientSocket) :

while True:

command = clientSocket.recv(1024) .decode()

# 进行命令、参数的分割

commList = command.split()

if commList[0] == 'exit ' :

break

# 若方法为download，则表示主控端需要获取被控端的文件

if commList[0] == 'download ' :

Upload File(clientSocket, commList[1])

if commList[0] == 'upload ' :

Download File(clientSocket)

4）编写文件传输的上传函数。在传输文件前，要通过struct模块将需要传输 的文件信息（文件名、文件大小）进行打包发送给接收端，接收端根据传输文件 的大小来对接收到的数据进行计算，防止粘包。当文件信息成功发送后，再真正 进行文件的传输。对文件进行多次分块读取发送，防止因为文件过大导致读取文 件内容时内存不足：