QT 将单线程改为线程池端口扫描3.5 - qidao123.com技术社区-IT企服评测·应用市场

#ifndef SCANTASK_H
#define SCANTASK_H
#include <QTcpSocket>
#include <QDebug>
#include <QRunnable>
#include <QObject>
class ScanTask : public QObject, public QRunnable // 需要继承两个类
{
Q_OBJECT
public:
ScanTask(QString strIP, int intPort);
~ScanTask();
void setPaused(bool paused); // 设置暂停状态
void closeThread(); // 关闭线程
protected:
void run() override; // 重写 run 方法
private:
volatile bool isStop; // 控制任务停止
volatile bool isPaused; // 控制任务暂停
QString m_strIP; // IP 地址
int m_intPort; // 端口号
signals:
void send_scan_signal(int port, bool isOpen); // 发送扫描结果信号
void finished(int port, bool isOpen); // 扫描完成信号
};
#endif // SCANTASK_H

复制代码

#include "scantask.h"
ScanTask::ScanTask(QString strIP, int intPort)
: m_strIP(strIP), m_intPort(intPort), isStop(false), isPaused(false) // 初始化成员变量
{
}
ScanTask::~ScanTask()
{
qDebug("ScanTask deleted.");
}
void ScanTask::setPaused(bool paused)
{
isPaused = paused; // 更新暂停状态
}
void ScanTask::closeThread()
{
isStop = true; // 设置为停止状态
}
void ScanTask::run()
{
QTcpSocket socket;
socket.abort(); // 取消任何现有连接
// 执行端口扫描
if (isStop) return; // 检查是否停止
socket.connectToHost(m_strIP, m_intPort); // 连接到指定的 IP 和端口
if (socket.waitForConnected(1000)) // 等待连接
{
emit send_scan_signal(m_intPort, true); // 发送信号，表示端口打开
qDebug("%d: %s", m_intPort, "opened"); // 调试信息
}
else
{
emit send_scan_signal(m_intPort, false); // 发送信号，表示端口关闭
qDebug("%d: %s", m_intPort, "closed");
}
}

复制代码

#ifndef MAINWINDOW_H
#define MAINWINDOW_H
#include <QMessageBox>
#include <QTreeWidget>
#include <QMainWindow>
#include <QThreadPool>
#include "scantask.h"
QT_BEGIN_NAMESPACE
namespace Ui { class MainWindow; }
QT_END_NAMESPACE
class MainWindow : public QMainWindow
{
Q_OBJECT
public:
MainWindow(QWidget *parent = nullptr);
~MainWindow();
protected:
private:
Ui::MainWindow *ui;
QTreeWidgetItem *itemRoot; // 树形控件的条目，用于表示目标 IP
QTreeWidgetItem *itemLeaf; // 树形控件的条目，用于表示目标端口
QThreadPool *threadpool; // 声明一个线程池
bool isPaused; // 控制暂停状态
private slots:
void on_pushButton_Scan_clicked();
void on_stopButton_clicked();
void on_pushButton_Quit_clicked();
void recv_result(int port, bool isOpen); // 自定义槽函数，用于接收线程发送的消息
void on_pushButton_Stop_clicked(); // 暂停按钮点击事件
void on_pushButton_Continue_clicked(); // 继续按钮点击事件
};
#endif // MAINWINDOW_H

复制代码

#include "mainwindow.h"
#include "ui_mainwindow.h"
MainWindow::MainWindow(QWidget *parent)
: QMainWindow(parent)
, ui(new Ui::MainWindow)
{
ui->setupUi(this);
// 给 QTreeWidget 初始化表头
QStringList head;
head << "扫描结果";
ui->treeWidget->setHeaderLabels(head);
threadpool = new QThreadPool; // 创建线程池
threadpool->setMaxThreadCount(10); // 设置最大线程数
isPaused = false; // 初始化暂停状态
}
MainWindow::~MainWindow()
{
delete ui;
}
void MainWindow::on_pushButton_Scan_clicked()
{
ui->pushButton_Scan->setEnabled(false); // 正在扫描时，禁用扫描按钮
QString strIP = ui->lineEdit_IP->text(); // 获取用户输入的 IP 地址
int startPort = ui->spinBox_Port->value(); // 获取用户输入的起始端口
int endPort = startPort + 10; // 假设扫描 10 个端口
// 验证 IP 地址是否为空
if (strIP.isEmpty()) {
QMessageBox::information(this, "Error", "请输入 IP", QMessageBox::Ok);
return; // 退出函数
}
ui->treeWidget->clear(); // 清空 QTreeWidget 中的先前结果
itemRoot = new QTreeWidgetItem(ui->treeWidget, QStringList(strIP)); // 创建根节点，显示 IP 地址
for (int i = startPort; i < endPort; i++) { // 扫描指定范围的端口
ScanTask *task = new ScanTask(strIP, i); // 创建新的扫描任务
connect(task, &ScanTask::send_scan_signal, this, &MainWindow::recv_result); // 连接信号
threadpool->start(task); // 将任务提交到线程池
QThread::msleep(10); // 短暂休眠，避免过快提交任务
}
}
void MainWindow::recv_result(int port, bool isOpen)
{
QString strPort = QString::number(port); // 将端口号转换为字符串
// 根据端口状态创建叶子节点
if (isOpen)
itemLeaf = new QTreeWidgetItem(itemRoot, QStringList(strPort + " opened")); // 端口打开
else
itemLeaf = new QTreeWidgetItem(itemRoot, QStringList(strPort + " closed")); // 端口关闭
ui->treeWidget->expandAll(); // 展开所有项
}
void MainWindow::on_stopButton_clicked()
{
// 停止所有正在运行的任务
threadpool->clear(); // 清理线程池中未执行完的任务
ui->pushButton_Scan->setEnabled(true); // 重新启用扫描按钮
}
void MainWindow::on_pushButton_Quit_clicked()
{
QApplication::quit(); // 退出应用程序
}
void MainWindow::on_pushButton_Stop_clicked()
{
isPaused = true; // 设置为暂停状态
ui->pushButton_Stop->setEnabled(false); // 禁用暂停按钮
ui->pushButton_Continue->setVisible(true); // 显示继续按钮
}
void MainWindow::on_pushButton_Continue_clicked()
{
isPaused = false; // 设置为继续状态
ui->pushButton_Stop->setEnabled(true); // 启用暂停按钮
ui->pushButton_Continue->setVisible(false); // 隐藏继续按钮
}

复制代码