mqtt
This commit is contained in:
@@ -1,411 +1,426 @@
|
|||||||
#pragma once
|
#pragma once
|
||||||
#include "QCL_Include.hpp"
|
#include "QCL_Include.hpp"
|
||||||
|
|
||||||
namespace QCL
|
namespace QCL
|
||||||
{
|
{
|
||||||
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||||
/**
|
/**
|
||||||
* @class TcpServer
|
* @class TcpServer
|
||||||
* @brief 简单的多线程TCP服务器类，支持多个客户端连接，数据收发及断开处理
|
* @brief 简单的多线程TCP服务器类，支持多个客户端连接，数据收发及断开处理
|
||||||
*
|
*
|
||||||
* 该类使用一个线程专门用于监听客户端连接，
|
* 该类使用一个线程专门用于监听客户端连接，
|
||||||
* 每当有客户端连接成功时，为其创建一个独立线程处理该客户端的数据收发。
|
* 每当有客户端连接成功时，为其创建一个独立线程处理该客户端的数据收发。
|
||||||
* 线程安全地管理所有客户端Socket句柄。
|
* 线程安全地管理所有客户端Socket句柄。
|
||||||
*/
|
*/
|
||||||
class TcpServer
|
class TcpServer
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
/**
|
/**
|
||||||
* @brief 构造函数，指定监听端口
|
* @brief 构造函数，指定监听端口
|
||||||
* @param port 服务器监听端口号
|
* @param port 服务器监听端口号
|
||||||
*/
|
*/
|
||||||
TcpServer(int port);
|
TcpServer(int port);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief 析构函数，自动调用 stop() 停止服务器并清理资源
|
* @brief 析构函数，自动调用 stop() 停止服务器并清理资源
|
||||||
*/
|
*/
|
||||||
~TcpServer();
|
~TcpServer();
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief 启动服务器，创建监听socket，开启监听线程
|
* @brief 启动服务器，创建监听socket，开启监听线程
|
||||||
* @return 启动成功返回true，失败返回false
|
* @return 启动成功返回true，失败返回false
|
||||||
*/
|
*/
|
||||||
bool start();
|
bool start();
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief 停止服务器，关闭所有连接，释放资源，等待所有线程退出
|
* @brief 停止服务器，关闭所有连接，释放资源，等待所有线程退出
|
||||||
*/
|
*/
|
||||||
void stop();
|
void stop();
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief 发送消息给指定客户端
|
* @brief 发送消息给指定客户端
|
||||||
* @param clientSock 客户端Socket描述符
|
* @param clientSock 客户端Socket描述符
|
||||||
* @param message 发送的字符串消息
|
* @param message 发送的字符串消息
|
||||||
*/
|
*/
|
||||||
void sendToClient(int clientSock, const std::string &message);
|
void sendToClient(int clientSock, const std::string &message);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief 从指定客户端接收数据（单次调用）
|
* @brief 从指定客户端接收数据（单次调用）
|
||||||
* @param clientSock 客户端Socket描述符
|
* @param clientSock 客户端Socket描述符
|
||||||
* @param flag:false 非阻塞模式,true 阻塞模式
|
* @param flag:false 非阻塞模式,true 阻塞模式
|
||||||
*/
|
*/
|
||||||
std::string receiveFromClient(int clientSock, bool flag = true);
|
std::string receiveFromClient(int clientSock, bool flag = true);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief 获取连接客户端的IP和端口
|
* @brief 获取连接客户端的IP和端口
|
||||||
* @param clientSock 客户端Socket描述符
|
* @param clientSock 客户端Socket描述符
|
||||||
*/
|
*/
|
||||||
char *getClientIPAndPort(int clientSock);
|
char *getClientIPAndPort(int clientSock);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief 获取当前所有已连接客户端Socket的副本
|
* @brief 获取当前所有已连接客户端Socket的副本
|
||||||
* @return 包含所有客户端Socket的vector，线程安全
|
* @return 包含所有客户端Socket的vector，线程安全
|
||||||
*/
|
*/
|
||||||
std::vector<int> getClientSockets();
|
std::vector<int> getClientSockets();
|
||||||
|
|
||||||
private:
|
/**
|
||||||
/**
|
* @brief 从服务器的客户端列表中移除并关闭一个客户端socket
|
||||||
* @brief 监听并接受新的客户端连接（运行在独立线程中）
|
* @param clientSock 客户端Socket描述符
|
||||||
*/
|
*/
|
||||||
void acceptClients();
|
void removeClient(int clientSock);
|
||||||
|
|
||||||
private:
|
/**
|
||||||
int serverSock_; ///< 服务器监听Socket描述符
|
* @brief 非阻塞探测客户端是否已断开（不消耗数据）
|
||||||
int port_; ///< 服务器监听端口
|
* @param clientSock 客户端Socket描述符
|
||||||
std::atomic<bool> running_; ///< 服务器运行状态标志（线程安全）
|
* @return true 已断开或发生致命错误；false 仍然存活或暂无数据
|
||||||
std::vector<std::thread> clientThreads_; ///< 用于处理每个客户端的线程集合
|
*/
|
||||||
std::thread acceptThread_; ///< 负责监听新连接的线程
|
bool isClientDisconnected(int clientSock);
|
||||||
std::mutex clientsMutex_; ///< 保护clientSockets_的互斥锁
|
|
||||||
std::vector<int> clientSockets_; ///< 当前所有连接的客户端Socket集合
|
private:
|
||||||
};
|
/**
|
||||||
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
* @brief 监听并接受新的客户端连接（运行在独立线程中）
|
||||||
/**
|
*/
|
||||||
* @brief 文件写入工具类（线程安全）
|
void acceptClients();
|
||||||
*
|
|
||||||
* 该类支持多种文件写入方式：
|
private:
|
||||||
* - 覆盖写文本
|
int serverSock_; ///< 服务器监听Socket描述符
|
||||||
* - 追加写文本
|
int port_; ///< 服务器监听端口
|
||||||
* - 按位置覆盖原文写入
|
std::atomic<bool> running_; ///< 服务器运行状态标志（线程安全）
|
||||||
* - 二进制覆盖写
|
std::vector<std::thread> clientThreads_; ///< 用于处理每个客户端的线程集合
|
||||||
* - 二进制追加写
|
std::thread acceptThread_; ///< 负责监听新连接的线程
|
||||||
*
|
std::mutex clientsMutex_; ///< 保护clientSockets_的互斥锁
|
||||||
* 特点：
|
std::vector<int> clientSockets_; ///< 当前所有连接的客户端Socket集合
|
||||||
* - 文件不存在时自动创建
|
};
|
||||||
* - 支持覆盖和追加两种模式
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||||
* - 支持二进制模式，适合写入非文本数据
|
/**
|
||||||
* - 内部使用 std::mutex 实现线程安全
|
* @brief 文件写入工具类（线程安全）
|
||||||
*/
|
*
|
||||||
class WriteFile
|
* 该类支持多种文件写入方式：
|
||||||
{
|
* - 覆盖写文本
|
||||||
public:
|
* - 追加写文本
|
||||||
/**
|
* - 按位置覆盖原文写入
|
||||||
* @brief 构造函数
|
* - 二进制覆盖写
|
||||||
* @param filePath 文件路径
|
* - 二进制追加写
|
||||||
*/
|
*
|
||||||
explicit WriteFile(const std::string &filePath);
|
* 特点：
|
||||||
|
* - 文件不存在时自动创建
|
||||||
/**
|
* - 支持覆盖和追加两种模式
|
||||||
* @brief 覆盖写文本文件（线程安全）
|
* - 支持二进制模式，适合写入非文本数据
|
||||||
* @param content 要写入的文本内容
|
* - 内部使用 std::mutex 实现线程安全
|
||||||
* @return true 写入成功
|
*/
|
||||||
* @return false 写入失败
|
class WriteFile
|
||||||
*/
|
{
|
||||||
bool overwriteText(const std::string &content);
|
public:
|
||||||
|
/**
|
||||||
/**
|
* @brief 构造函数
|
||||||
* @brief 追加写文本文件（线程安全）
|
* @param filePath 文件路径
|
||||||
* @param content 要写入的文本内容
|
*/
|
||||||
* @return true 写入成功
|
explicit WriteFile(const std::string &filePath);
|
||||||
* @return false 写入失败
|
|
||||||
*/
|
/**
|
||||||
bool appendText(const std::string &content);
|
* @brief 覆盖写文本文件（线程安全）
|
||||||
|
* @param content 要写入的文本内容
|
||||||
/**
|
* @return true 写入成功
|
||||||
* @brief 覆盖写二进制文件（线程安全）
|
* @return false 写入失败
|
||||||
* @param data 要写入的二进制数据
|
*/
|
||||||
* @return true 写入成功
|
bool overwriteText(const std::string &content);
|
||||||
* @return false 写入失败
|
|
||||||
*/
|
/**
|
||||||
bool overwriteBinary(const std::vector<char> &data);
|
* @brief 追加写文本文件（线程安全）
|
||||||
|
* @param content 要写入的文本内容
|
||||||
/**
|
* @return true 写入成功
|
||||||
* @brief 追加写二进制文件（线程安全）
|
* @return false 写入失败
|
||||||
* @param data 要写入的二进制数据
|
*/
|
||||||
* @return true 写入成功
|
bool appendText(const std::string &content);
|
||||||
* @return false 写入失败
|
|
||||||
*/
|
/**
|
||||||
bool appendBinary(const std::vector<char> &data);
|
* @brief 覆盖写二进制文件（线程安全）
|
||||||
|
* @param data 要写入的二进制数据
|
||||||
/**
|
* @return true 写入成功
|
||||||
* @brief 计算第一个指定字节序列前的字节数
|
* @return false 写入失败
|
||||||
* @param pattern 要查找的字节序列
|
*/
|
||||||
* @param includePattern true 表示返回值包含 pattern 自身长度，false 表示不包含
|
bool overwriteBinary(const std::vector<char> &data);
|
||||||
* @return size_t 字节数，如果文件没打开或 pattern 为空则返回 0
|
|
||||||
*/
|
/**
|
||||||
size_t countBytesPattern(const std::string &pattern, bool includePattern = false);
|
* @brief 追加写二进制文件（线程安全）
|
||||||
|
* @param data 要写入的二进制数据
|
||||||
/**
|
* @return true 写入成功
|
||||||
* @brief 在文件中查找指定字节序列并在其后写入内容，如果不存在则追加到文件末尾
|
* @return false 写入失败
|
||||||
* @param pattern 要查找的字节序列
|
*/
|
||||||
* @param content 要写入的内容
|
bool appendBinary(const std::vector<char> &data);
|
||||||
* @return true 写入成功，false 文件打开失败
|
|
||||||
*
|
/**
|
||||||
* 功能说明：
|
* @brief 计算第一个指定字节序列前的字节数
|
||||||
* 1. 若文件中存在 pattern，则删除 pattern 之后的所有内容，并在其后插入 content。
|
* @param pattern 要查找的字节序列
|
||||||
* 2. 若文件中不存在 pattern，则在文件末尾追加 content，若末尾无换行符则先补充换行。
|
* @param includePattern true 表示返回值包含 pattern 自身长度，false 表示不包含
|
||||||
*/
|
* @return size_t 字节数，如果文件没打开或 pattern 为空则返回 0
|
||||||
bool writeAfterPatternOrAppend(const std::string &pattern, const std::string &content);
|
*/
|
||||||
|
size_t countBytesPattern(const std::string &pattern, bool includePattern = false);
|
||||||
/**
|
|
||||||
* @brief 在文件指定位置之后插入内容
|
/**
|
||||||
* @param content 要插入的内容
|
* @brief 在文件中查找指定字节序列并在其后写入内容，如果不存在则追加到文件末尾
|
||||||
* @param pos 插入位置（从文件开头算起的字节偏移量）
|
* @param pattern 要查找的字节序列
|
||||||
* @param length 插入的长度（>= content.size() 时，多余部分用空字节填充；< content.size() 时只截取前 length 个字节）
|
* @param content 要写入的内容
|
||||||
* @return true 插入成功，false 文件打开失败或参数不合法
|
* @return true 写入成功，false 文件打开失败
|
||||||
*
|
*
|
||||||
* 功能说明：
|
* 功能说明：
|
||||||
* 1. 不会覆盖原有数据，而是将 pos 之后的内容整体向后移动 length 个字节。
|
* 1. 若文件中存在 pattern，则删除 pattern 之后的所有内容，并在其后插入 content。
|
||||||
* 2. 如果 length > content.size()，则在 content 后补充 '\0'（或空格，可按需求改）。
|
* 2. 若文件中不存在 pattern，则在文件末尾追加 content，若末尾无换行符则先补充换行。
|
||||||
* 3. 如果 length < content.size()，则只写入 content 的前 length 个字节。
|
*/
|
||||||
* 4. 文件整体大小会增加 length 个字节。
|
bool writeAfterPatternOrAppend(const std::string &pattern, const std::string &content);
|
||||||
*
|
|
||||||
* 举例：
|
/**
|
||||||
* 原始文件内容: "ABCDEFG"
|
* @brief 在文件指定位置之后插入内容
|
||||||
* insertAfterPos("XY", 2, 3) // 在索引 2 后插入
|
* @param content 要插入的内容
|
||||||
* 结果: "ABX Y\0CDEFG" (这里 \0 代表补充的空字节)
|
* @param pos 插入位置（从文件开头算起的字节偏移量）
|
||||||
*/
|
* @param length 插入的长度（>= content.size() 时，多余部分用空字节填充；< content.size() 时只截取前 length 个字节）
|
||||||
bool insertAfterPos(const std::string &content, size_t pos, size_t length);
|
* @return true 插入成功，false 文件打开失败或参数不合法
|
||||||
|
*
|
||||||
/**
|
* 功能说明：
|
||||||
* @brief 在文件指定位置覆盖写入内容
|
* 1. 不会覆盖原有数据，而是将 pos 之后的内容整体向后移动 length 个字节。
|
||||||
* @param content 要写入的内容
|
* 2. 如果 length > content.size()，则在 content 后补充 '\0'（或空格，可按需求改）。
|
||||||
* @param pos 覆盖起始位置（从文件开头算起的字节偏移量）
|
* 3. 如果 length < content.size()，则只写入 content 的前 length 个字节。
|
||||||
* @param length 覆盖长度
|
* 4. 文件整体大小会增加 length 个字节。
|
||||||
* @return true 覆盖成功，false 文件打开失败或 pos 越界
|
*
|
||||||
*
|
* 举例：
|
||||||
* 功能说明：
|
* 原始文件内容: "ABCDEFG"
|
||||||
* 1. 从 pos 开始覆盖 length 个字节，不会移动或增加文件大小。
|
* insertAfterPos("XY", 2, 3) // 在索引 2 后插入
|
||||||
* 2. 如果 content.size() >= length，则只写入前 length 个字节。
|
* 结果: "ABX Y\0CDEFG" (这里 \0 代表补充的空字节)
|
||||||
* 3. 如果 content.size() < length，则写入 content，并用 '\0' 补齐至 length。
|
*/
|
||||||
* 4. 如果 pos + length 超过文件末尾，则只覆盖到文件尾部，不会越界。
|
bool insertAfterPos(const std::string &content, size_t pos, size_t length);
|
||||||
*
|
|
||||||
* 举例：
|
/**
|
||||||
* 原始文件内容: "ABCDEFG"
|
* @brief 在文件指定位置覆盖写入内容
|
||||||
* overwriteAtPos("XY", 2, 3)
|
* @param content 要写入的内容
|
||||||
* 结果: "ABXYEFG" （原 "CDE" 被 "XY\0" 覆盖，\0 实际不可见）
|
* @param pos 覆盖起始位置（从文件开头算起的字节偏移量）
|
||||||
*/
|
* @param length 覆盖长度
|
||||||
bool overwriteAtPos(const std::string &content, size_t pos, size_t length);
|
* @return true 覆盖成功，false 文件打开失败或 pos 越界
|
||||||
|
*
|
||||||
private:
|
* 功能说明：
|
||||||
std::string filePath_; ///< 文件路径
|
* 1. 从 pos 开始覆盖 length 个字节，不会移动或增加文件大小。
|
||||||
std::mutex writeMutex_; ///< 线程锁，保证多线程写入安全
|
* 2. 如果 content.size() >= length，则只写入前 length 个字节。
|
||||||
|
* 3. 如果 content.size() < length，则写入 content，并用 '\0' 补齐至 length。
|
||||||
/**
|
* 4. 如果 pos + length 超过文件末尾，则只覆盖到文件尾部，不会越界。
|
||||||
* @brief 通用文本写入接口（线程安全）
|
*
|
||||||
* @param content 要写入的内容
|
* 举例：
|
||||||
* @param mode 打开模式（追加/覆盖等）
|
* 原始文件内容: "ABCDEFG"
|
||||||
* @return true 写入成功
|
* overwriteAtPos("XY", 2, 3)
|
||||||
* @return false 写入失败
|
* 结果: "ABXYEFG" （原 "CDE" 被 "XY\0" 覆盖，\0 实际不可见）
|
||||||
*/
|
*/
|
||||||
bool writeToFile(const std::string &content, std::ios::openmode mode);
|
bool overwriteAtPos(const std::string &content, size_t pos, size_t length);
|
||||||
|
|
||||||
/**
|
void close();
|
||||||
* @brief 通用二进制写入接口（线程安全）
|
|
||||||
* @param data 要写入的二进制数据
|
private:
|
||||||
* @param mode 打开模式（追加/覆盖等）
|
std::string filePath_; ///< 文件路径
|
||||||
* @return true 写入成功
|
std::mutex writeMutex_; ///< 线程锁，保证多线程写入安全
|
||||||
* @return false 写入失败
|
|
||||||
*/
|
/**
|
||||||
bool writeBinary(const std::vector<char> &data, std::ios::openmode mode);
|
* @brief 通用文本写入接口（线程安全）
|
||||||
};
|
* @param content 要写入的内容
|
||||||
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
* @param mode 打开模式（追加/覆盖等）
|
||||||
/**
|
* @return true 写入成功
|
||||||
* @brief ReadFile 类 - 读文件操作工具类
|
* @return false 写入失败
|
||||||
*
|
*/
|
||||||
* 功能：
|
bool writeToFile(const std::string &content, std::ios::openmode mode);
|
||||||
* 1. 读取全文（文本 / 二进制）
|
|
||||||
* 2. 按行读取
|
/**
|
||||||
* 3. 按字节数读取
|
* @brief 通用二进制写入接口（线程安全）
|
||||||
* 4. 获取指定字节序列前的字节数（包含该字节序列）
|
* @param data 要写入的二进制数据
|
||||||
* 5. 检查文件是否存在
|
* @param mode 打开模式（追加/覆盖等）
|
||||||
* 6. 获取文件大小
|
* @return true 写入成功
|
||||||
*
|
* @return false 写入失败
|
||||||
* 设计：
|
*/
|
||||||
* - 与 WriteFile 类风格保持一致
|
bool writeBinary(const std::vector<char> &data, std::ios::openmode mode);
|
||||||
* - 支持文本文件与二进制文件
|
};
|
||||||
* - 自动关闭文件（析构时）
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||||
* - 内部使用 std::mutex 实现线程安全
|
/**
|
||||||
*/
|
* @brief ReadFile 类 - 读文件操作工具类
|
||||||
class ReadFile
|
*
|
||||||
{
|
* 功能：
|
||||||
public:
|
* 1. 读取全文（文本 / 二进制）
|
||||||
/**
|
* 2. 按行读取
|
||||||
* @brief 构造函数
|
* 3. 按字节数读取
|
||||||
* @param filename 文件路径
|
* 4. 获取指定字节序列前的字节数（包含该字节序列）
|
||||||
*/
|
* 5. 检查文件是否存在
|
||||||
explicit ReadFile(const std::string &filename);
|
* 6. 获取文件大小
|
||||||
|
*
|
||||||
/**
|
* 设计：
|
||||||
* @brief 析构函数，自动关闭文件
|
* - 与 WriteFile 类风格保持一致
|
||||||
*/
|
* - 支持文本文件与二进制文件
|
||||||
~ReadFile();
|
* - 自动关闭文件（析构时）
|
||||||
|
* - 内部使用 std::mutex 实现线程安全
|
||||||
/**
|
*/
|
||||||
* @brief 打开文件（以二进制方式）
|
class ReadFile
|
||||||
* @return true 打开成功
|
{
|
||||||
* @return false 打开失败
|
public:
|
||||||
*/
|
/**
|
||||||
bool Open();
|
* @brief 构造函数
|
||||||
|
* @param filename 文件路径
|
||||||
/**
|
*/
|
||||||
* @brief 关闭文件
|
explicit ReadFile(const std::string &filename);
|
||||||
*/
|
|
||||||
void Close();
|
/**
|
||||||
|
* @brief 析构函数，自动关闭文件
|
||||||
/**
|
*/
|
||||||
* @brief 文件是否已经打开
|
~ReadFile();
|
||||||
*/
|
|
||||||
bool IsOpen() const;
|
/**
|
||||||
|
* @brief 打开文件（以二进制方式）
|
||||||
/**
|
* @return true 打开成功
|
||||||
* @brief 读取全文（文本模式）
|
* @return false 打开失败
|
||||||
* @return 文件内容字符串
|
*/
|
||||||
*/
|
bool Open();
|
||||||
std::string ReadAllText();
|
|
||||||
|
/**
|
||||||
/**
|
* @brief 关闭文件
|
||||||
* @brief 读取全文（二进制模式）
|
*/
|
||||||
* @return 文件内容字节数组
|
void Close();
|
||||||
*/
|
|
||||||
std::vector<char> ReadAllBinary();
|
/**
|
||||||
|
* @brief 文件是否已经打开
|
||||||
/**
|
*/
|
||||||
* @brief 按行读取文本
|
bool IsOpen() const;
|
||||||
* @return 每行作为一个字符串的 vector
|
|
||||||
*/
|
/**
|
||||||
std::vector<std::string> ReadLines();
|
* @brief 读取全文（文本模式）
|
||||||
|
* @return 文件内容字符串
|
||||||
/**
|
*/
|
||||||
* @brief 读取指定字节数
|
std::string ReadAllText();
|
||||||
* @param count 要读取的字节数
|
|
||||||
* @return 实际读取到的字节数据
|
/**
|
||||||
*/
|
* @brief 读取全文（二进制模式）
|
||||||
std::vector<char> ReadBytes(size_t count);
|
* @return 文件内容字节数组
|
||||||
|
*/
|
||||||
/**
|
std::vector<char> ReadAllBinary();
|
||||||
* @brief 获取指定字节序列前的字节数（包含该字节序列）
|
|
||||||
* @param marker 要查找的字节序列（可能不止一个字节）
|
/**
|
||||||
* @return 如果找到，返回前面部分字节数；找不到返回0
|
* @brief 按行读取文本
|
||||||
*/
|
* @return 每行作为一个字符串的 vector
|
||||||
size_t GetBytesBefore(const std::string &marker, bool includeMarker = false);
|
*/
|
||||||
|
std::vector<std::string> ReadLines();
|
||||||
/**
|
|
||||||
* @brief 从指定位置读取指定字节数，默认读取到文件末尾
|
/**
|
||||||
* @param pos 起始位置（字节偏移）
|
* @brief 读取指定字节数
|
||||||
* @param count 要读取的字节数，默认为0表示读取到文件末尾
|
* @param count 要读取的字节数
|
||||||
* @return 读取到的字节数据
|
* @return 实际读取到的字节数据
|
||||||
*/
|
*/
|
||||||
std::vector<char> ReadBytesFrom(size_t pos, size_t count = 0);
|
std::vector<char> ReadBytes(size_t count);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief 检查文件是否存在
|
* @brief 获取指定字节序列前的字节数（包含该字节序列）
|
||||||
*/
|
* @param marker 要查找的字节序列（可能不止一个字节）
|
||||||
bool FileExists() const;
|
* @return 如果找到，返回前面部分字节数；找不到返回0
|
||||||
|
*/
|
||||||
/**
|
size_t GetBytesBefore(const std::string &marker, bool includeMarker = false);
|
||||||
* @brief 获取文件大小（字节数）
|
|
||||||
*/
|
/**
|
||||||
size_t GetFileSize() const;
|
* @brief 从指定位置读取指定字节数，默认读取到文件末尾
|
||||||
|
* @param pos 起始位置（字节偏移）
|
||||||
/**
|
* @param count 要读取的字节数，默认为0表示读取到文件末尾
|
||||||
* @brief 重置读取位置到文件开头
|
* @return 读取到的字节数据
|
||||||
*/
|
*/
|
||||||
void Reset();
|
std::vector<char> ReadBytesFrom(size_t pos, size_t count = 0);
|
||||||
|
|
||||||
private:
|
/**
|
||||||
std::string filename_; // 文件路径
|
* @brief 检查文件是否存在
|
||||||
std::ifstream file_; // 文件流对象
|
*/
|
||||||
mutable std::mutex mtx_; // 可变，保证 const 方法也能加锁
|
bool FileExists() const;
|
||||||
};
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
/**
|
||||||
|
* @brief 获取文件大小（字节数）
|
||||||
// 屏蔽所有信号
|
*/
|
||||||
void blockAllSignals();
|
size_t GetFileSize() const;
|
||||||
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
// 字符串操作
|
/**
|
||||||
// 去除字符串的左空格
|
* @brief 重置读取位置到文件开头
|
||||||
std::string Ltrim(const std::string &s);
|
*/
|
||||||
|
void Reset();
|
||||||
// 去除字符串右侧的空格
|
|
||||||
std::string Rtrim(const std::string &s);
|
private:
|
||||||
|
std::string filename_; // 文件路径
|
||||||
// 去除字符串左右两侧的空格
|
std::ifstream file_; // 文件流对象
|
||||||
std::string LRtrim(const std::string &s);
|
mutable std::mutex mtx_; // 可变，保证 const 方法也能加锁
|
||||||
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
};
|
||||||
// c++进行格式化输出
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||||
// 通用类型转字符串
|
|
||||||
template <typename T>
|
// 屏蔽所有信号
|
||||||
std::string to_string_any(const T &value)
|
void blockAllSignals();
|
||||||
{
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||||
std::ostringstream oss;
|
// 字符串操作
|
||||||
oss << value;
|
// 去除字符串的左空格
|
||||||
return oss.str();
|
std::string Ltrim(const std::string &s);
|
||||||
}
|
|
||||||
|
// 去除字符串右侧的空格
|
||||||
// 递归获取 tuple 中 index 对应参数
|
std::string Rtrim(const std::string &s);
|
||||||
template <std::size_t I = 0, typename Tuple>
|
|
||||||
std::string get_tuple_arg(const Tuple &tup, std::size_t index)
|
// 去除字符串左右两侧的空格
|
||||||
{
|
std::string LRtrim(const std::string &s);
|
||||||
if constexpr (I < std::tuple_size_v<Tuple>)
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||||
{
|
// c++进行格式化输出
|
||||||
if (I == index)
|
// 通用类型转字符串
|
||||||
return to_string_any(std::get<I>(tup));
|
template <typename T>
|
||||||
else
|
std::string to_string_any(const T &value)
|
||||||
return get_tuple_arg<I + 1>(tup, index);
|
{
|
||||||
}
|
std::ostringstream oss;
|
||||||
else
|
oss << value;
|
||||||
{
|
return oss.str();
|
||||||
throw std::runtime_error("Too few arguments for format string");
|
}
|
||||||
}
|
|
||||||
}
|
// 递归获取 tuple 中 index 对应参数
|
||||||
|
template <std::size_t I = 0, typename Tuple>
|
||||||
// format 函数
|
std::string get_tuple_arg(const Tuple &tup, std::size_t index)
|
||||||
template <typename... Args>
|
{
|
||||||
std::string format(const std::string &fmt, const Args &...args)
|
if constexpr (I < std::tuple_size_v<Tuple>)
|
||||||
{
|
{
|
||||||
std::ostringstream oss;
|
if (I == index)
|
||||||
std::tuple<const Args &...> tup(args...);
|
return to_string_any(std::get<I>(tup));
|
||||||
size_t pos = 0;
|
else
|
||||||
size_t arg_idx = 0;
|
return get_tuple_arg<I + 1>(tup, index);
|
||||||
|
}
|
||||||
while (pos < fmt.size())
|
else
|
||||||
{
|
{
|
||||||
if (fmt[pos] == '{' && pos + 1 < fmt.size() && fmt[pos + 1] == '{')
|
throw std::runtime_error("Too few arguments for format string");
|
||||||
{
|
}
|
||||||
oss << '{';
|
}
|
||||||
pos += 2;
|
|
||||||
}
|
// format 函数
|
||||||
else if (fmt[pos] == '}' && pos + 1 < fmt.size() && fmt[pos + 1] == '}')
|
template <typename... Args>
|
||||||
{
|
std::string format(const std::string &fmt, const Args &...args)
|
||||||
oss << '}';
|
{
|
||||||
pos += 2;
|
std::ostringstream oss;
|
||||||
}
|
std::tuple<const Args &...> tup(args...);
|
||||||
else if (fmt[pos] == '{' && pos + 1 < fmt.size() && fmt[pos + 1] == '}')
|
size_t pos = 0;
|
||||||
{
|
size_t arg_idx = 0;
|
||||||
oss << get_tuple_arg(tup, arg_idx++);
|
|
||||||
pos += 2;
|
while (pos < fmt.size())
|
||||||
}
|
{
|
||||||
else
|
if (fmt[pos] == '{' && pos + 1 < fmt.size() && fmt[pos + 1] == '{')
|
||||||
{
|
{
|
||||||
oss << fmt[pos++];
|
oss << '{';
|
||||||
}
|
pos += 2;
|
||||||
}
|
}
|
||||||
|
else if (fmt[pos] == '}' && pos + 1 < fmt.size() && fmt[pos + 1] == '}')
|
||||||
if (arg_idx < sizeof...(Args))
|
{
|
||||||
throw std::runtime_error("Too many arguments for format string");
|
oss << '}';
|
||||||
|
pos += 2;
|
||||||
return oss.str();
|
}
|
||||||
}
|
else if (fmt[pos] == '{' && pos + 1 < fmt.size() && fmt[pos + 1] == '}')
|
||||||
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
{
|
||||||
|
oss << get_tuple_arg(tup, arg_idx++);
|
||||||
}
|
pos += 2;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
oss << fmt[pos++];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (arg_idx < sizeof...(Args))
|
||||||
|
throw std::runtime_error("Too many arguments for format string");
|
||||||
|
|
||||||
|
return oss.str();
|
||||||
|
}
|
||||||
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
}
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
Binary file not shown.
@@ -10,6 +10,8 @@
|
|||||||
#include <string>
|
#include <string>
|
||||||
#include <thread>
|
#include <thread>
|
||||||
|
|
||||||
|
#include <mqtt/async_client.h>
|
||||||
|
|
||||||
#include <boost/process.hpp>
|
#include <boost/process.hpp>
|
||||||
|
|
||||||
#include <nlohmann/json.hpp>
|
#include <nlohmann/json.hpp>
|
||||||
@@ -34,7 +36,12 @@ string filepath = "/home/orangepi/InitAuth/conf/.env";
|
|||||||
string cameraPath = "/opt/rknn-yolov11/.env";
|
string cameraPath = "/opt/rknn-yolov11/.env";
|
||||||
string passwd = "/home/orangepi/InitAuth/pwd/.env";
|
string passwd = "/home/orangepi/InitAuth/pwd/.env";
|
||||||
|
|
||||||
string url = "http://116.147.36.110:8095/device/validateDevice";
|
// 云端Web认证接口
|
||||||
|
const string url = "http://116.147.36.110:8095/device/validateDevice";
|
||||||
|
const string mqtt_url = "tcp://192.168.12.1:1883";
|
||||||
|
const string clientId = "RK3588_SubTest";
|
||||||
|
const string Topic = "/Test";
|
||||||
|
const int Qos = 1;
|
||||||
|
|
||||||
std::atomic<bool> isRunning(true); // 全局运行标志
|
std::atomic<bool> isRunning(true); // 全局运行标志
|
||||||
std::atomic<bool> confirm(true); // 发送卡和ID
|
std::atomic<bool> confirm(true); // 发送卡和ID
|
||||||
@@ -54,6 +61,8 @@ struct OutSignal
|
|||||||
bool inPutMode; // 触发输入模式 true--高电平,false -- 低电平
|
bool inPutMode; // 触发输入模式 true--高电平,false -- 低电平
|
||||||
} algor{};
|
} algor{};
|
||||||
|
|
||||||
|
mqtt::async_client client(mqtt_url, clientId);
|
||||||
|
|
||||||
// 确认是否已经进行过认证
|
// 确认是否已经进行过认证
|
||||||
bool ConfirmInit();
|
bool ConfirmInit();
|
||||||
|
|
||||||
@@ -94,6 +103,12 @@ void StartNet();
|
|||||||
// 开启服务
|
// 开启服务
|
||||||
void StartService();
|
void StartService();
|
||||||
|
|
||||||
|
// mqtt初始化
|
||||||
|
void mqttInit();
|
||||||
|
|
||||||
|
// 接收消息回调
|
||||||
|
void getMsgCallback(mqtt::const_message_ptr msg);
|
||||||
|
|
||||||
/*
|
/*
|
||||||
parames:
|
parames:
|
||||||
argv[1] - SSID of the WiFi network to connect to
|
argv[1] - SSID of the WiFi network to connect to
|
||||||
@@ -110,6 +125,14 @@ int main(int argc, char *argv[])
|
|||||||
blockAllSignals();
|
blockAllSignals();
|
||||||
signal(SIGINT, Exit); // 捕获Ctrl+C信号
|
signal(SIGINT, Exit); // 捕获Ctrl+C信号
|
||||||
|
|
||||||
|
// 初始化mqtt服务器
|
||||||
|
mqttInit();
|
||||||
|
|
||||||
|
while (1)
|
||||||
|
{
|
||||||
|
this_thread::sleep_for(chrono::seconds(1));
|
||||||
|
}
|
||||||
|
|
||||||
// 开启服务器
|
// 开启服务器
|
||||||
MyServer = new TcpServer(8848);
|
MyServer = new TcpServer(8848);
|
||||||
|
|
||||||
@@ -149,6 +172,27 @@ int main(int argc, char *argv[])
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void mqttInit()
|
||||||
|
{
|
||||||
|
|
||||||
|
client.set_connected_handler([](const string &cause)
|
||||||
|
{
|
||||||
|
cout << cause << endl;
|
||||||
|
cout << "Connected Successed!\n"; });
|
||||||
|
client.set_message_callback(getMsgCallback);
|
||||||
|
|
||||||
|
// 连接服务器
|
||||||
|
client.connect()->wait();
|
||||||
|
client.subscribe(Topic, Qos)->wait();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 接收消息回调
|
||||||
|
void getMsgCallback(mqtt::const_message_ptr msg)
|
||||||
|
{
|
||||||
|
cout << "收到消息" << endl;
|
||||||
|
cout << "recv:" << msg->to_string() << endl;
|
||||||
|
}
|
||||||
|
|
||||||
// 开启服务
|
// 开启服务
|
||||||
void StartService()
|
void StartService()
|
||||||
{
|
{
|
||||||
@@ -263,9 +307,11 @@ void ReceiveData()
|
|||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
int index = client.size();
|
int index = client.size();
|
||||||
for (int ii = 0; ii < index; ii++)
|
// cout << "index: " << index << endl;
|
||||||
|
for (int ii = 0; ii < index; ++ii) // 从新到旧，优先处理新连接
|
||||||
{
|
{
|
||||||
buffer = MyServer->receiveFromClient(client[ii], false); // 非阻塞模式接受数据
|
buffer = MyServer->receiveFromClient(client[ii], false); // 非阻塞模式接受数据
|
||||||
|
// cout << buffer << endl;
|
||||||
if (buffer.empty() == false)
|
if (buffer.empty() == false)
|
||||||
{
|
{
|
||||||
cout << "已收到" << buffer << endl;
|
cout << "已收到" << buffer << endl;
|
||||||
@@ -313,14 +359,15 @@ void ReceiveData()
|
|||||||
double safe = toDouble(safe_json);
|
double safe = toDouble(safe_json);
|
||||||
|
|
||||||
// 整文件读取，逐行替换，保持注释不变
|
// 整文件读取，逐行替换，保持注释不变
|
||||||
ReadFile rf(cameraPath);
|
ReadFile *rf = new ReadFile(cameraPath);
|
||||||
if (!rf.Open())
|
if (!rf->Open())
|
||||||
{
|
{
|
||||||
cerr << "文件打开失败: " << cameraPath << "\n";
|
cerr << "文件打开失败: " << cameraPath << "\n";
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
auto lines = rf.ReadLines();
|
auto lines = rf->ReadLines();
|
||||||
|
delete rf;
|
||||||
for (auto &line : lines)
|
for (auto &line : lines)
|
||||||
{
|
{
|
||||||
if (line.rfind("NEAR_THRESHOLD=", 0) == 0)
|
if (line.rfind("NEAR_THRESHOLD=", 0) == 0)
|
||||||
@@ -346,32 +393,90 @@ void ReceiveData()
|
|||||||
if (i + 1 < lines.size())
|
if (i + 1 < lines.size())
|
||||||
out += "\n";
|
out += "\n";
|
||||||
}
|
}
|
||||||
WriteFile wf(cameraPath);
|
WriteFile *wf = new WriteFile(cameraPath);
|
||||||
wf.overwriteText(out);
|
wf->overwriteText(out);
|
||||||
|
delete wf;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else
|
else if (buffer.find("media") != string::npos)
|
||||||
{
|
{
|
||||||
// 写摄像头画面设置
|
// 写摄像头参数(与距离相同方式: 整体读取-逐行替换-整体写回)
|
||||||
string conf = "";
|
|
||||||
cout << buffer << endl;
|
cout << buffer << endl;
|
||||||
if (buffer.find("media") != string::npos)
|
CalculateInfo(media, buffer);
|
||||||
{ // 写摄像头参数
|
ReadFile *rf = new ReadFile(cameraPath);
|
||||||
cout << buffer << endl;
|
if (!rf->Open())
|
||||||
WriteFile cam(cameraPath);
|
|
||||||
CalculateInfo(media, buffer);
|
|
||||||
conf = format("\nMEDIA_MIRROR={}\nMEDIA_FLIP={}\nMEDIA_OCCLUSION={}\n", media.mirror == 1 ? "true" : "false", media.flip == 1 ? "true" : "false", media.occlusion == 1 ? "true" : "false");
|
|
||||||
cout << conf;
|
|
||||||
cam.writeAfterPatternOrAppend("***---***", conf);
|
|
||||||
}
|
|
||||||
if (buffer.find("algorithm") != string::npos)
|
|
||||||
{
|
{
|
||||||
// 写输入输出参数
|
cerr << "文件打开失败: " << cameraPath << "\n";
|
||||||
WriteFile mode(filepath);
|
}
|
||||||
int pos = mode.countBytesPattern("***---***", true);
|
else
|
||||||
CalculateInfo(algor, buffer);
|
{
|
||||||
conf = format("\noutPutMode:{}\ninPutMode:{}", algor.outPutMode == 1 ? "true" : "false", algor.inPutMode == 1 ? "true" : "false");
|
auto lines = rf->ReadLines();
|
||||||
mode.writeAfterPatternOrAppend("***---***", conf);
|
delete rf;
|
||||||
|
for (auto &line : lines)
|
||||||
|
{
|
||||||
|
if (line.rfind("MEDIA_MIRROR=", 0) == 0)
|
||||||
|
{
|
||||||
|
line = format("MEDIA_MIRROR={}", media.mirror ? "true" : "false");
|
||||||
|
}
|
||||||
|
else if (line.rfind("MEDIA_FLIP=", 0) == 0)
|
||||||
|
{
|
||||||
|
line = format("MEDIA_FLIP={}", media.flip ? "true" : "false");
|
||||||
|
}
|
||||||
|
else if (line.rfind("MEDIA_OCCLUSION=", 0) == 0)
|
||||||
|
{
|
||||||
|
line = format("MEDIA_OCCLUSION={}", media.occlusion ? "true" : "false");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
string out;
|
||||||
|
out.reserve(4096);
|
||||||
|
for (size_t i = 0; i < lines.size(); ++i)
|
||||||
|
{
|
||||||
|
out += lines[i];
|
||||||
|
if (i + 1 < lines.size())
|
||||||
|
out += "\n";
|
||||||
|
}
|
||||||
|
WriteFile *wf = new WriteFile(cameraPath);
|
||||||
|
wf->overwriteText(out);
|
||||||
|
delete wf;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (buffer.find("algorithm") != string::npos)
|
||||||
|
{
|
||||||
|
// 写输入输出参数(与距离相同方式: 整体读取-逐行替换-整体写回)
|
||||||
|
CalculateInfo(algor, buffer);
|
||||||
|
ReadFile *rf2 = new ReadFile(filepath);
|
||||||
|
if (!rf2->Open())
|
||||||
|
{
|
||||||
|
cerr << "文件打开失败: " << filepath << "\n";
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
auto lines = rf2->ReadLines();
|
||||||
|
delete rf2;
|
||||||
|
for (auto &line : lines)
|
||||||
|
{
|
||||||
|
if (line.rfind("outPutMode:", 0) == 0)
|
||||||
|
{
|
||||||
|
line = format("outPutMode:{}", algor.outPutMode ? "true" : "false");
|
||||||
|
}
|
||||||
|
else if (line.rfind("inPutMode:", 0) == 0)
|
||||||
|
{
|
||||||
|
line = format("inPutMode:{}", algor.inPutMode ? "true" : "false");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
string out;
|
||||||
|
out.reserve(4096);
|
||||||
|
for (size_t i = 0; i < lines.size(); ++i)
|
||||||
|
{
|
||||||
|
out += lines[i];
|
||||||
|
if (i + 1 < lines.size())
|
||||||
|
out += "\n";
|
||||||
|
}
|
||||||
|
WriteFile *wf2 = new WriteFile(filepath);
|
||||||
|
wf2->overwriteText(out);
|
||||||
|
delete wf2;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -390,7 +495,8 @@ void OpenRTSP()
|
|||||||
// 视频推流
|
// 视频推流
|
||||||
void VideoStream()
|
void VideoStream()
|
||||||
{
|
{
|
||||||
string commnd = "ffmpeg -f v4l2 -i /dev/video0 -c:v h264_rkmpp -rtsp_transport tcp -f rtsp rtsp://192.168.12.1:8554/stream";
|
// 静音ffmpeg的统计输出，保留错误；避免污染日志
|
||||||
|
string commnd = "ffmpeg -nostats -hide_banner -loglevel error -f v4l2 -i /dev/video0 -c:v h264_rkmpp -rtsp_transport tcp -f rtsp rtsp://192.168.12.1:8554/stream 2>/dev/null";
|
||||||
video_proc = bp::child("/bin/bash", bp::args = {"-c", commnd});
|
video_proc = bp::child("/bin/bash", bp::args = {"-c", commnd});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
all:wifi
|
all:wifi
|
||||||
|
|
||||||
wifi:main.cpp
|
wifi:main.cpp
|
||||||
g++ -g -o wifi main.cpp /home/orangepi/RKApp/ApCreate/NetraLib/src/Netra.cpp /home/orangepi/RKApp/ApCreate/NetraLib/src/encrypt.cpp /home/orangepi/RKApp/ApCreate/NetraLib/src/NetRequest.cpp -lpthread -I/home/orangepi/RKApp/ApCreate/NetraLib/include
|
g++ -g -o wifi main.cpp /home/orangepi/RKApp/ApCreate/NetraLib/src/Netra.cpp /home/orangepi/RKApp/ApCreate/NetraLib/src/encrypt.cpp /home/orangepi/RKApp/ApCreate/NetraLib/src/NetRequest.cpp -I/home/orangepi/RKApp/ApCreate/NetraLib/include -lpaho-mqttpp3 -lpaho-mqtt3a -lpthread
|
||||||
mv ./wifi ../bin/wifi
|
mv ./wifi ../bin/wifi
|
||||||
|
|
||||||
clean:
|
clean:
|
||||||
|
|||||||
Reference in New Issue
Block a user