2

include/common/SocketWrapper.h

@@ -0,0 +1,139 @@
+#pragma once
+#include <string>
+
+#ifdef _WIN32
+    #include <winsock2.h>
+    #include <ws2tcpip.h>
+    #pragma comment(lib, "ws2_32.lib")
+    typedef SOCKET SocketFd;
+    #define INVALID_SOCKET_FD INVALID_SOCKET
+    #define SOCKET_ERROR_CODE WSAGetLastError()
+#else
+    #include <sys/socket.h>
+    #include <netinet/in.h>
+    #include <arpa/inet.h>
+    #include <unistd.h>
+    #include <fcntl.h>
+    #include <errno.h>
+    typedef int SocketFd;
+    #define INVALID_SOCKET_FD -1
+    #define SOCKET_ERROR_CODE errno
+    #define closesocket close
+#endif
+
+/**
+ * @brief Socket 封装类
+ * 
+ * 技术点: 网络编程 (Network Programming) - 原生 Socket API 封装
+ * 跨平台支持: Windows (Winsock) 和 Linux (POSIX Sockets)
+ */
+class SocketWrapper {
+private:
+    SocketFd m_socket;
+    bool m_isValid;
+
+public:
+    /**
+     * @brief 构造函数
+     */
+    SocketWrapper();
+
+    /**
+     * @brief 构造函数 (使用现有socket)
+     */
+    explicit SocketWrapper(SocketFd socket);
+
+    /**
+     * @brief 析构函数
+     */
+    ~SocketWrapper();
+
+    // 禁止拷贝
+    SocketWrapper(const SocketWrapper&) = delete;
+    SocketWrapper& operator=(const SocketWrapper&) = delete;
+
+    /**
+     * @brief 初始化 Winsock (仅 Windows)
+     */
+    static bool initializeWinsock();
+
+    /**
+     * @brief 清理 Winsock (仅 Windows)
+     */
+    static void cleanupWinsock();
+
+    /**
+     * @brief 创建 Socket
+     */
+    bool create();
+
+    /**
+     * @brief 绑定地址和端口
+     */
+    bool bind(const std::string& address, int port);
+
+    /**
+     * @brief 监听连接
+     */
+    bool listen(int backlog = 10);
+
+    /**
+     * @brief 接受客户端连接
+     */
+    SocketFd accept(std::string& clientAddress, int& clientPort);
+
+    /**
+     * @brief 连接到服务器
+     */
+    bool connect(const std::string& address, int port);
+
+    /**
+     * @brief 发送数据
+     */
+    int send(const char* data, int length);
+
+    /**
+     * @brief 发送字符串
+     */
+    int send(const std::string& message);
+
+    /**
+     * @brief 接收数据
+     */
+    int recv(char* buffer, int length);
+
+    /**
+     * @brief 接收一行数据 (以 \n 结尾)
+     */
+    std::string recvLine();
+
+    /**
+     * @brief 设置为非阻塞模式
+     */
+    bool setNonBlocking(bool nonBlocking);
+
+    /**
+     * @brief 设置 Socket 选项
+     */
+    bool setReuseAddr(bool reuse);
+
+    /**
+     * @brief 关闭 Socket
+     */
+    void close();
+
+    /**
+     * @brief 检查 Socket 是否有效
+     */
+    bool isValid() const { return m_isValid; }
+
+    /**
+     * @brief 获取原始 Socket 描述符
+     */
+    SocketFd getSocket() const { return m_socket; }
+
+    /**
+     * @brief 获取最后的错误信息
+     */
+    static std::string getLastError();
+};