MyLib

Socket/NetworkRelated.h

@@ -0,0 +1,51 @@
+#ifndef LXZL_NETWORKRELATED_H
+#define LXZL_NETWORKRELATED_H
+
+#include <iostream>
+// 平台检测
+#ifdef _WIN32
+#include <winsock2.h>
+#include <ws2tcpip.h>
+#else
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <sys/ioctl.h>
+#endif
+
+// 平台兼容类型定义
+#ifdef _WIN32
+using socklen_t = int;
+using socket_t = SOCKET;
+constexpr socket_t INVALID_SOCKET_VALUE = INVALID_SOCKET;
+#else
+using socket_t = int;
+constexpr socket_t INVALID_SOCKET_VALUE = -1;
+#define SOCKET_ERROR (-1)
+#endif
+
+
+
+// Windows平台初始化
+#ifdef _WIN32
+
+class WSAInitializer {
+public:
+	WSAInitializer() {
+		WSADATA wsaData;
+		if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
+			throw std::runtime_error("WSAStartup failed");
+		}
+	}
+	
+	~WSAInitializer() {
+		WSACleanup();
+	}
+};
+
+static WSAInitializer wsa_initializer;
+#endif
+
+#endif //LXZL_NETWORKRELATED_H