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

主控端的编写

 

主控端的部分函数与被控端无异，因此笔者对相同的函数不再赘述，下面是 实现主要功能的具体步骤。

1）编写主函数并导入相关模块。主控端需要开启监听等待被控端的回连，

当接收到被控端的回连时，输出被控端的主机名，并给出功能选择的提示。在主 控端选择功能的同时，也要将交互数据传送给被控端，使其根据我们的操作在相 应的功能函数中待命：

#!/usr/bin/python3

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

 

import socket

import os

import struct

 

if __name__ == '__main__ ' :

# 主控端监听地址

serverIP = '127 .0 .0 .1 '

# 主控端监听端口

serverPort = 6666

serverAddr = (serverIP, serverPort)

# 主控端开始监听

try:

serverSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) serverSocket.bind(serverAddr)

serverSocket.listen(1)

except socket.error as message:

print(message)

os ._exit(0)

print("[*]Server is up!!!")

conn, addr = serverSocket.accept()

# 接收并打印上线主机的主机名、地址和端口

hostName = conn .recv(1024)

print("[+]Host is up! \n ============ \n name:{0} ip:{1} \n port :{2} \n

============ \n" .format(bytes .decode(hostName), addr[0], addr[1])) try:

while True:

print("Functional selection:\n")

print("[1]ExecCommand \n[2]TransferFiles\n")

choice = input( '[None]>>> ')

# 给被控端发送指令，主控端进入相应的功能模块

if choice == '1 ' :

# 发送的命令为str型，需要用encode函数把命令转换为bytes型

conn .sendall( '1 ' .encode())

ExecCommand(conn, addr)

elif choice == '2 ' :

conn .sendall( '2 ' .encode())

TransferFiles(conn, addr)

elif choice == 'exit ' :

conn .sendall( 'exit ' .encode())

serverSocket.close()

break

except :

 

 

 

 

 

serverSocket.close()

 

2）编写命令执行函数。只需要把用户的命令传送给被控端执行，并接收被 控端回传的命令执行结果即可：

# 命令执行函数 def ExecCommand(conn, addr) : while True: command = input("[ExecCommand]>>> ") if command == 'exit ' : # 主控端退出相应模块时，也要通知客户端退出对应的功能模块 conn .sendall( 'exit ' .encode()) break conn .sendall(command.encode()) result = conn .recv(10000) .decode() print(result)

3）编写文件传输函数的主函数。这里的主函数与被控端的文件传输主函数  差别不大。主控端在上传文件时，需要传递upload关键字以及需要传送到被控端 文件的文件路径。在进行文件下载时，需要传递download关键字，以及需要下载 的被控端文件的文件路径：

# 文件传输函数 def TransferFiles(conn, addr) : print("Usage: method filepath") print("Example: upload /root/ms08067 | download /root/ms08067") while True: command = input("[TransferFiles]>>> ") #对输入进行命令和参数分割 command List = command.split() if command List[0] == 'exit ' : # 主控端退出相应模块时，也要通知被控端退出对应的功能模块 conn .sendall( 'exit ' .encode()) break # 若方法为download，则表示主控端需要获取被控端的文件 if command List[0] == 'download ' : Download File(conn, addr, command) if command List[0] == 'upload ' : Upload File(conn, addr, command)

4）编写文件传输的上传函数。传输的过程与被控端无异，只是主控端需要 先把命令发送给被控端，然后再执行文件上传：

# 文件上传函数

def Upload File(conn, addr, command) :

# 把主控端的命令发送给被控端

conn .sendall(command.encode())

#从命令中分离出要上传的文件的路径

command List = command.split()

while True:

upload FilePath = command List[1]

if os .path.isfile(upload FilePath) :

# 先传输文件信息，用来防止粘包

# 定义文件信息，128s表示文件名长度为128字节，l表示用in t类型表示文件大小 # 把文件名和文件大小信息进行封装，发送给接收端

fileInfo = struct.pack( '128sl ', bytes(os .path.basename(upload FilePath) .

 

 

 

 

 

encode( 'utf-8 ')), os .stat(upload FilePath) .st_size)

conn .sendall(fileInfo)

print( '[+]FileInfo send success ! name:{0}  size:{1} ' .format(os . path.basename(upload FilePath), os .stat(upload FilePath) .st_  size))

# 开始传输文件的内容

print( '[+]start uploading . . . ')

with open(upload FilePath, 'rb ') as f :

while True:

# 分块多次读，防止文件过大时一次性读完导致内存不足

data = f.read(1024)

if not data:

print("File Send Over !")

break

conn .sendall(data)

break

 

5）编写文件传输的下载函数。传输的过程与被控端无异，只是主控端需要 先把文件下载命令发送给被控端，然后再执行文件下载：

# 文件下载函数 def Download File(conn, addr, command) : # 把主控端的命令发送给被控端 conn .sendall(command.encode()) while True: # 先接收文件的信息，进行解析 # 长度自定义，先接收文件信息的主要原因是防止粘包 # 接收长度为128sl fileInfo = conn .recv(struct.calcsize( '128sl ')) if fileInfo: # 按照同样的格式（128sl）进行拆包 fileName, fileSize = struct.unpack( '128sl ', fileInfo) # 要把文件名后面的多余空字符去除 fileName = fileName .decode() .strip( '\00 ') # 定义上传文件的存放路径， ./表示当前目录下 newFilename = os .path.join( ' ./ ', fileName) print( 'Fileinfo Receive over ! name:{0}  size:{1} ' .format (fileName, fileSize)) # 接下来开始接收文件的内容 # 表示已经接收到的文件内容大小 recvdSize = 0 print( 'start receiving . . . ') with open(newFilename, 'wb ') as f : # 分次分块写入 while not recvdSize == fileSize: if fileSize - recvdSize > 1024: data = conn .recv(1024) f.write(data) recvdSize += len(data) else: # 当剩余内容不足1024时，则把剩下的全部内容都接收写入 data = conn .recv(fileSize - recvdSize) f.write(data) recvdSize = fileSize break print("File Receive over !!!") break