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()

# 文件上传函数

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) .