layout: Reorganize TCP client/server code

2025-06-29 21:25:59 +08:00
parent 4ca2b45242
commit 017a8c8fa2
65 changed files with 1074 additions and 843 deletions
--- a/server/2015Remote/IOCPUDPServer.cpp
+++ b/server/2015Remote/IOCPUDPServer.cpp
@@ -0,0 +1,164 @@
+#include "stdafx.h"
+#include "IOCPUDPServer.h"
+#include <thread>
+#include <iostream>
+
+IOCPUDPServer::IOCPUDPServer() {
+	WSADATA wsaData;
+	WSAStartup(MAKEWORD(2, 2), &wsaData);
+}
+
+IOCPUDPServer::~IOCPUDPServer() {
+	WSACleanup();
+}
+
+UINT IOCPUDPServer::StartServer(pfnNotifyProc NotifyProc, pfnOfflineProc OffProc, USHORT uPort) {
+	if (m_running) return 1;
+
+	m_notify = NotifyProc;
+	m_offline = OffProc;
+
+	m_socket = WSASocket(AF_INET, SOCK_DGRAM, IPPROTO_UDP, NULL, 0, WSA_FLAG_OVERLAPPED);
+	if (m_socket == INVALID_SOCKET) return 2;
+
+	sockaddr_in addr{};
+	addr.sin_family = AF_INET;
+	addr.sin_port = htons(uPort);
+	addr.sin_addr.s_addr = INADDR_ANY;
+
+	if (bind(m_socket, (sockaddr*)&addr, sizeof(addr)) == SOCKET_ERROR) return 3;
+
+	m_hIOCP = CreateIoCompletionPort((HANDLE)m_socket, NULL, 0, 0);
+	if (!m_hIOCP) return 4;
+
+	m_running = true;
+
+	// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>߳<EFBFBD>
+	m_hThread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)+[](LPVOID param) -> DWORD {
+		((IOCPUDPServer*)param)->WorkerThread();
+		return 0;
+		}, this, 0, NULL);
+
+	// <20>ύ<EFBFBD><E1BDBB><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʼ<EFBFBD><CABC><EFBFBD><EFBFBD>
+	for (int i = 0; i < 4; ++i)
+		PostRecv();
+
+	return 0; // <20>ɹ<EFBFBD>
+}
+
+void IOCPUDPServer::PostRecv() {
+	if (!m_running) return;
+
+	AddCount(1);
+	CONTEXT_OBJECT* ctx = new CONTEXT_OBJECT();
+	ctx->InitMember(m_socket);
+
+	IO_CONTEXT* ioCtx = new IO_CONTEXT();
+	ZeroMemory(&ioCtx->ol, sizeof(OVERLAPPED));
+	ioCtx->pContext = ctx;
+
+	ctx->wsaInBuf.buf = ctx->szBuffer;
+	ctx->wsaInBuf.len = sizeof(ctx->szBuffer);
+	ctx->addrLen = sizeof(sockaddr_in);
+
+	DWORD flags = 0;
+	int err = WSARecvFrom(
+		m_socket,
+		&ctx->wsaInBuf,
+		1,
+		NULL,
+		&flags,
+		(sockaddr*)&ctx->clientAddr,
+		&ctx->addrLen,
+		&ioCtx->ol,
+		NULL
+	);
+
+	if (err == SOCKET_ERROR && WSAGetLastError() != WSA_IO_PENDING) {
+		DWORD err = WSAGetLastError();
+		Mprintf("[IOCP] PostRecv error: %d\n", err);
+		delete ioCtx;
+		delete ctx;
+		AddCount(-1);
+	}
+}
+
+void IOCPUDPServer::WorkerThread() {
+	while (m_running) {
+		DWORD bytes = 0;
+		ULONG_PTR key = 0;
+		LPOVERLAPPED pOverlapped = nullptr;
+
+		BOOL ok = GetQueuedCompletionStatus(m_hIOCP, &bytes, &key, &pOverlapped, INFINITE);
+		if (!ok) {
+			DWORD err = WSAGetLastError();
+			Mprintf("[IOCP] PostRecv error: %d\n", err);
+			if (pOverlapped) {
+				IO_CONTEXT* ioCtx = CONTAINING_RECORD(pOverlapped, IO_CONTEXT, ol);
+				delete ioCtx->pContext;
+				delete ioCtx;
+				AddCount(-1);
+			}
+			continue;
+		}
+		if (!pOverlapped) continue;
+
+		IO_CONTEXT* ioCtx = CONTAINING_RECORD(pOverlapped, IO_CONTEXT, ol);
+		CONTEXT_OBJECT* ctx = ioCtx->pContext;
+
+		if (m_notify)
+			m_notify(ctx);
+
+		// <20>ͷ<EFBFBD>
+		delete ioCtx;
+		delete ctx;
+		AddCount(-1);
+
+		PostRecv(); // <20><><EFBFBD><EFBFBD><EFBFBD>ύ
+	}
+}
+
+VOID IOCPUDPServer::Send2Client(CONTEXT_OBJECT* ContextObject, PBYTE szBuffer, ULONG ulOriginalLength) {
+	WSABUF buf = { ulOriginalLength, (CHAR*)szBuffer };
+	DWORD sent = 0;
+	int err = WSASendTo(
+		ContextObject->sClientSocket,
+		&buf,
+		1,
+		&sent,
+		0,
+		(sockaddr*)&ContextObject->clientAddr,
+		sizeof(sockaddr_in),
+		NULL,
+		NULL
+	);
+	if (err == SOCKET_ERROR) {
+		DWORD err = WSAGetLastError();
+		Mprintf("[IOCP] Send2Client error: %d\n", err);
+	}
+}
+
+VOID IOCPUDPServer::Destroy() {
+	if (m_socket != INVALID_SOCKET) {
+		CancelIoEx((HANDLE)m_socket, NULL); // ȡ<><C8A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD>IO<49><4F><EFBFBD><EFBFBD>
+		closesocket(m_socket);
+		m_socket = INVALID_SOCKET;
+	}
+
+	while (GetCount())
+		Sleep(200);
+
+	m_running = false;
+	PostQueuedCompletionStatus(m_hIOCP, 0, 0, NULL); // <20><><EFBFBD>ڻ<EFBFBD><DABB><EFBFBD><EFBFBD>߳<EFBFBD><DFB3>˳<EFBFBD>
+
+	if (m_hThread) {
+		WaitForSingleObject(m_hThread, INFINITE);
+		CloseHandle(m_hThread);
+		m_hThread = NULL;
+	}
+
+	if (m_hIOCP) {
+		CloseHandle(m_hIOCP);
+		m_hIOCP = NULL;
+	}
+}