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

 

 

本节将开始着手用Python编写远程控制工具。因篇幅限制,这里主要编写两 个常用的功能:命令执行和文件传输。当然,好的远程控制工具的功能远远不只 这些,读者可以在此基础上增加新的功能。具体步骤如下。

1)编写主函数并导入相关模块。我们让被控端主动连接主控端(反向连

接),连接成功时会将自己的主机名发送给主控端。同时,主控端可以进行功能 选择(命令执行和文件传输),所以我们还要写一个死循环,让被控端根据主控 端的回馈信息进入相应的模块:

 

#!/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{ … }语句,当被控端报错时能及时捕获并 处理错误,重新进入命令执行的循环,提高被控端的可靠性:

 

 

 

 

 

 

# 命令执行函数

def Execommand(clientSocket) :

while True:

try:

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

# 将接收到的命令进行命令、参数分割

commList = command.split()

# 接收到exit时退出命令执行功能

if commList[0] == 'exit ' :

break

# 执行cd时不能直接通过subprocess进行目录切换,

# 否则会出现[Errno 2] No such file or directory错误,

# 要通过os .chdir来切换目录

elif commList[0] == 'cd ' :

os .chdir(commList[1])

# 切换完毕后,将当前被控端的工作路径发给主控端

clientSocket.sendall(os .getcwd() .encode())

else:

clientSocket.sendall(subprocess .check_output(command,

shell=True))

# 出现异常时进行捕获,并通知主控端

except Exception as message:

clientSocket.sendall("Failed to execute, please check your

command!!!" .encode())

# 报错跳出循环时,通过continue重新进入循环

continue

 

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模块将需要传输 的文件信息(文件名、文件大小)进行打包发送给接收端,接收端根据传输文件 的大小来对接收到的数据进行计算,防止粘包。当文件信息成功发送后,再真正 进行文件的传输。对文件进行多次分块读取发送,防止因为文件过大导致读取文 件内容时内存不足:

# 文件上传函数

def Upload File(clientSocket, filepath) :

while True:

upload FilePath = filepath

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)

clientSocket.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 Upload Over !!!")

break

clientSocket.sendall(data)

break

 

5)编写文件传输的下载函数。在下载文件前,需要先接收传输文件信息的  包,并拆包得到文件的文件名和大小,这里需要注意的是,拆包后的文件名信息 可能会出现多余的字符,此时需要使用strip方法进行清除。在得到传输文件的大 小信息后,就可以以此为依据,进行文件数据的分块写入:

# 文件下载函数

def Download File(clientSocket) :

while True:

# 先接收文件的信息,进行解析

# 长度自定义,先接收文件信息的主要原因是防止粘包

# 接收长度为128sl

fileInfo = clientSocket.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 = clientSocket.recv(1024)

f.write(data)

recvdSize += len(data)

else:

# 剩下的内容不足1024时,则把剩余的全部内容都接收写入

data = clientSocket.recv(fileSize - recvdSize)

f.write(data)

recvdSize = fileSize

break

print("[+]File Receive over !!!")

break

 

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/676821.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

每日一题——LeetCode1417.重新格式化字符串

方法一 个人方法: s里的字符只有小写字母和数字两种情况,我们可以把s里的字母和数字分隔成两个字符串, 比较两个字符串的长度,只有当两个字符串的长度差值的绝对值为1或0才能满足题意。 长度更长的要放在结果字符串的第一位&am…

写一个python基于线程池的多线程

下面是一个示例程序,它使用Python的concurrent.futures模块中的线程池来实现多线程操作: import concurrent.futuresdef worker(arg):print(fWorking on {arg})return arg * 2if __name__ __main__:# 创建线程池with concurrent.futures.ThreadPoolExe…

嵌入式学习之Linux入门篇笔记——18,makefile基本语法(下)

配套视频学习链接:http://【【北京迅为】嵌入式学习之Linux入门篇】 https://www.bilibili.com/video/BV1M7411m7wT/?p4&share_sourcecopy_web&vd_sourcea0ef2c4953d33a9260910aaea45eaec8 1.wildcard 函数 格式:$(wildcard PAT…

SQL如何实现数据表行转列、列转行?

SQL行转列、列转行可以帮助我们更方便地处理数据,生成需要的报表和结果集。本文将介绍在SQL中如何实现数据表地行转列、列转行操作,以及实际应用示例。 这里通过表下面三张表进行举例 SQL创建数据库和数据表 数据表示例数据分别如下: data_…

数据湖的整体思路

湖本质上是一个集中化,中心化的,一体化的存储技术,并且在其之上追求技术架构的统一化,如流批一体,服务分析一体化。 当数据湖成为中心,那么就可以围湖而建“数据服务环”,环上的服务包括了数仓、…

鸿蒙开发(六)布局概述

迄今为止,我还没有正式提到布局的概念。但其实我之前的demo里面,已经默认使用到了一种布局,那就是线性布局(Row、Column),这也是DevEco创建项目默认页面里面默认采用的布局。那么本篇,带着大家一…

微服务介绍、使用 Nacos 实现远程调用以及 OpenFeign 的使用

1 微服务的概念 区别于单体项目 单体项目拆分成微服务项目的目标:高内聚、低耦合 拆分思路 纵向拆分:根据功能模块 横向拆分:抽取可复用模块 2 微服务拆分——远程调用 背景:微服务单一职责,每个服务只有自己的功能…

[PYthon] 字典

如题,这篇博客将带大家来学习Python中的字典~ 那么,是我们想象中的那种字典嘛?接下来跟着可莉去一探究竟吧~ 可莉将这篇文章收录在了:《Python》 可莉推荐的优质博主主页:Keve ’ s blog Python 中的字典(D…

-转换流-

它是字节流和字符流之间转换的桥梁 转换流本身其实是字符流; 转换流的位置: 分为两个:字符转换输入流:InputStreamReader将InputStream转换为Reader字符转换输出流:OutputStreamWriter将Writer转换为OutputStream 作用…

4.2 Verilog 过程赋值

关键词:阻塞赋值,非阻塞赋值,并行 过程性赋值是在 initial 或 always 语句块里的赋值,赋值对象是寄存器、整数、实数等类型。 这些变量在被赋值后,其值将保持不变,直到重新被赋予新值。 连续性赋值总是处…

C++ 11新特性之可变参数模板

概述 随着C 11标准的发布,C语言获得了许多强大的新特性,其中一项显著提升灵活性和实用性的创新便是可变参数模板。这一特性极大地扩展了模板在处理不定数量类型或值参数时的能力,为开发者提供了更为强大且灵活的泛型编程工具。 工作机制 在C …

LeetCode Python - 6.Z字形变换

文章目录 题目答案运行结果 题目 将一个给定字符串 s 根据给定的行数 numRows ,以从上往下、从左到右进行 Z 字形排列。 比如输入字符串为 “PAYPALISHIRING” 行数为 3 时,排列如下: P A H N A P L S I I G Y I R 之后,你的输…

Qualcomm 蓝牙耳机 FAQ(41)---------Audio 问题分析之 ACAT Tools安装

大家好! 新的一年,在此祝大家:新年快乐!工作上步步高升!!龙年大吉!!! 也欢迎大家登录大大通平台,春节期间正常更新文章,期待你的到来&#xff0…

语义分割任务的准确率计算:基于PyTorch实现

语义分割任务的准确率计算:基于PyTorch实现 🌵文章目录🌵 🌳引言🌳🌳语义分割任务概述🌳🌳准确率的定义与计算方法🌳🌳实践应用与优化策略🌳&…

《Django+React前后端分离项目开发实战:爱计划》 03 理解项目结构

03 Understanding the Project Structure 《Django+React前后端分离项目开发实战:爱计划》 03 理解项目结构 Let’s look the project files created for us, Open the project folder backend in the code editor. 让我们看看为我们创建的项目文件,在代码编辑器中打开项目…

跟我学C++中级篇——函数模板的匹配

一、介绍和说明 在本文会尝试着把函数的重载以及模板自动推导等方法结合起来,一起分析模板函数的匹配的方式和原则。在普通的函数重载和普通的模板函数中,都比较容易理解调用哪一类,但在一些较为少用或者复杂的情况下,可能会发现…

《MySQL 简易速速上手小册》第7章:MySQL监控和日志分析(2024 最新版)

文章目录 7.1 配置和使用 MySQL 监控工具7.1.1 基础知识7.1.2 重点案例:使用 Python 和 Prometheus 监控 MySQL 性能7.1.3 拓展案例 1:自动化 MySQL 慢查询日志分析7.1.4 拓展案例 2:实时警报系统 7.2 解读 MySQL 日志文件7.2.1 基础知识7.2.…

使用django构建一个多级评论功能

,评论系统是交流和反馈的重要工具,尤其是多级评论系统,它允许用户回复特定评论,形成丰富的对话结构。这个文章是使用Django框架从零开始构建一个多级评论系统。Django是一个高级Python Web框架,它鼓励快速开发和干净、…

FastAPI使用异步motor实现对mongodb的增删改查

一、环境准备: 需要有mongodb和poetry以及Python3.10 二、克隆示例代码到本地 git clone gitgithub.com:waketzheng/fastapi_async_mongo_demo.git 三、安装依赖 cd fastapi_async_mongo_demo poetry shell poetry install 四、启动服务 python app/main.py …

Qt中程序发布及常见问题

1、引言 当我们写好一个程序时通常需要发布给用户使用,那么在Qt中程序又是如何实现发布的呢,这里我就来浅谈一下qt中如何发布程序,以及发布程序时的常见问题。 2、发布过程 2.1、切换为release模式 当我们写qt程序时默认是debug模式&#x…