155 lines
3.6 KiB
C++
155 lines
3.6 KiB
C++
#ifndef TCP_UTILITY_HPP
|
|
#define TCP_UTILITY_HPP
|
|
|
|
#include <cstdint>
|
|
#include <vector>
|
|
#include <string>
|
|
#include <functional>
|
|
#include <memory>
|
|
#include <thread>
|
|
#include <mutex>
|
|
#include <atomic>
|
|
|
|
namespace tcp {
|
|
|
|
// 错误码枚举
|
|
enum class ErrorCode {
|
|
Success = 0,
|
|
SocketCreationFailed,
|
|
BindFailed,
|
|
ListenFailed,
|
|
AcceptFailed,
|
|
ConnectFailed,
|
|
SendFailed,
|
|
ReceiveFailed,
|
|
Disconnected,
|
|
InvalidArgument
|
|
};
|
|
|
|
// 回调函数类型定义
|
|
using MessageCallback = std::function<void(class TcpSocket&, const std::vector<unsigned char>&)>;
|
|
using ErrorCallback = std::function<void(class TcpSocket&, ErrorCode)>;
|
|
using ConnectionCallback = std::function<void(class TcpServer&, std::shared_ptr<class TcpSocket>)>;
|
|
|
|
// 前向声明
|
|
class TcpSocket;
|
|
class TcpClient;
|
|
class TcpServer;
|
|
|
|
// TCP套接字类
|
|
class TcpSocket {
|
|
public:
|
|
TcpSocket(int socket_fd = -1);
|
|
~TcpSocket();
|
|
|
|
// 禁止拷贝，允许移动
|
|
TcpSocket(const TcpSocket&) = delete;
|
|
TcpSocket& operator=(const TcpSocket&) = delete;
|
|
TcpSocket(TcpSocket&&) noexcept;
|
|
TcpSocket& operator=(TcpSocket&&) noexcept;
|
|
|
|
// 连接到服务器
|
|
ErrorCode connect(const std::string& host, uint16_t port);
|
|
|
|
// 发送数据
|
|
ErrorCode send(const std::vector<unsigned char>& data);
|
|
ErrorCode send(const std::string& data);
|
|
|
|
// 接收数据
|
|
ErrorCode receive(std::vector<unsigned char>& buffer, size_t max_size);
|
|
|
|
// 关闭连接
|
|
void close();
|
|
|
|
// 获取状态
|
|
bool isConnected() const;
|
|
std::string getPeerAddress() const;
|
|
uint16_t getPeerPort() const;
|
|
|
|
// 设置回调函数
|
|
void setMessageCallback(const MessageCallback& callback);
|
|
void setErrorCallback(const ErrorCallback& callback);
|
|
|
|
// 开始异步接收
|
|
void startReceiving();
|
|
|
|
private:
|
|
int m_socket_fd;
|
|
bool m_connected;
|
|
std::string m_peer_address;
|
|
uint16_t m_peer_port;
|
|
MessageCallback m_message_callback;
|
|
ErrorCallback m_error_callback;
|
|
};
|
|
|
|
// TCP客户端类
|
|
class TcpClient {
|
|
public:
|
|
TcpClient();
|
|
~TcpClient();
|
|
|
|
// 连接到服务器
|
|
ErrorCode connect(const std::string& host, uint16_t port);
|
|
|
|
// 断开连接
|
|
void disconnect();
|
|
|
|
// 发送数据
|
|
ErrorCode send(const std::vector<unsigned char>& data);
|
|
ErrorCode send(const std::string& data);
|
|
|
|
// 获取状态
|
|
bool isConnected() const;
|
|
std::shared_ptr<TcpSocket> getSocket() const;
|
|
|
|
// 设置回调函数
|
|
void setMessageCallback(const MessageCallback& callback);
|
|
void setErrorCallback(const ErrorCallback& callback);
|
|
|
|
private:
|
|
std::shared_ptr<TcpSocket> m_socket;
|
|
MessageCallback m_message_callback;
|
|
ErrorCallback m_error_callback;
|
|
};
|
|
|
|
// TCP服务器类
|
|
class TcpServer {
|
|
public:
|
|
TcpServer();
|
|
~TcpServer();
|
|
|
|
// 启动服务器
|
|
ErrorCode start(uint16_t port, size_t max_connections = 10);
|
|
|
|
// 停止服务器
|
|
void stop();
|
|
|
|
// 获取状态
|
|
bool isRunning() const;
|
|
uint16_t getPort() const;
|
|
|
|
// 设置回调函数
|
|
void setConnectionCallback(const ConnectionCallback& callback);
|
|
void setMessageCallback(const MessageCallback& callback);
|
|
void setErrorCallback(const ErrorCallback& callback);
|
|
|
|
private:
|
|
int m_listen_fd;
|
|
uint16_t m_port;
|
|
bool m_running;
|
|
std::vector<std::shared_ptr<TcpSocket>> m_clients;
|
|
ConnectionCallback m_connection_callback;
|
|
MessageCallback m_message_callback;
|
|
ErrorCallback m_error_callback;
|
|
|
|
// 接受新连接的线程函数
|
|
void acceptThread();
|
|
};
|
|
|
|
// 错误处理辅助函数
|
|
std::string errorCodeToString(ErrorCode error);
|
|
|
|
} // namespace tcp
|
|
|
|
#endif // TCP_UTILITY_HPP
|