C++之基于Winsock2封装UDPServer与UDPClient

Socket过程

• UDPServer

• UDPClient

UDPServer.h

#ifndef UDPSERVER_H_INCLUDED
#define UDPSERVER_H_INCLUDED#include <iostream>
#include <string>
#include <Winsock2.h>class UDPServer
{
private:WORD sockVersion;WSADATA wsaData;SOCKET udpSocket;sockaddr_in localAddr;sockaddr_in remoteAddr; // Clientconst int MSG_SIZE = 1024;public:UDPServer();~UDPServer();void Bind(u_short port);void Receive();bool Send(std::string message);
};#endif // UDPSERVER_H_INCLUDED

UDPServer.cpp

#include "UDPServer.h"UDPServer::UDPServer()
{sockVersion = MAKEWORD(2, 2);WSAStartup(sockVersion, &wsaData);udpSocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
}UDPServer::~UDPServer()
{closesocket(udpSocket);WSACleanup();
}void UDPServer::Bind(u_short port)
{localAddr.sin_family = AF_INET;localAddr.sin_port = htons(port);// localAddr.sin_addr.S_un.S_addr = INADDR_ANY; // 0.0.0.0localAddr.sin_addr.S_un.S_addr = inet_addr("0.0.0.0");bind(udpSocket, (const sockaddr *)&localAddr, sizeof(localAddr));
}void UDPServer::Receive()
{while (true){char recvBuffer[MSG_SIZE];int localAddrSize = sizeof(localAddr);int lenRemoteAddr = sizeof(remoteAddr);int len = recvfrom(udpSocket, recvBuffer, MSG_SIZE, 0, (SOCKADDR *)&remoteAddr, &lenRemoteAddr);if (len > 0){recvBuffer[len] = '\0';std::cout << "[" << inet_ntoa(remoteAddr.sin_addr) << ":" << remoteAddr.sin_port << "] -> " << recvBuffer << std::endl;}}
}bool UDPServer::Send(std::string message)
{int ret = sendto(udpSocket, message.c_str(), message.length(), 0, (sockaddr *)&remoteAddr, sizeof(remoteAddr));if (ret == SOCKET_ERROR){return false;}else{return true;}
}

UDPClient.h

#ifndef UDPCLIENT_H_INCLUDED
#define UDPCLIENT_H_INCLUDED#include <iostream>
#include <string>
#include <Winsock2.h>class UDPClient
{
private:WORD sockVersion;WSADATA wsaData;SOCKET udpSocket;sockaddr_in localAddr;sockaddr_in remoteAddr; // Clientconst int MSG_SIZE = 1024;public:UDPClient();~UDPClient();void Bind(std::string ip, u_short port);void Receive();bool Send(std::string message);
};#endif // UDPCLIENT_H_INCLUDED

UDPClient.cpp

#include "UDPClient.h"UDPClient::UDPClient()
{sockVersion = MAKEWORD(2, 2);WSAStartup(sockVersion, &wsaData);udpSocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
}UDPClient::~UDPClient()
{closesocket(udpSocket);WSACleanup();
}void UDPClient::Bind(std::string ip, u_short port)
{localAddr.sin_family = AF_INET;localAddr.sin_port = htons(port);localAddr.sin_addr.S_un.S_addr = INADDR_ANY; // 0.0.0.0remoteAddr = localAddr;remoteAddr.sin_addr.S_un.S_addr = inet_addr(ip.c_str());bind(udpSocket, (const sockaddr *)&localAddr, sizeof(localAddr));Send("Hello");
}void UDPClient::Receive()
{while (true){char recvBuffer[MSG_SIZE];int localAddrSize = sizeof(localAddr);int lenRemoteAddr = sizeof(remoteAddr);int len = recvfrom(udpSocket, recvBuffer, MSG_SIZE, 0, (SOCKADDR *)&remoteAddr, &lenRemoteAddr);if (len > 0){recvBuffer[len] = '\0';std::cout << "[" << inet_ntoa(remoteAddr.sin_addr) << ":" << remoteAddr.sin_port << "] -> " << recvBuffer << std::endl;}}
}bool UDPClient::Send(std::string message)
{int ret = sendto(udpSocket, message.c_str(), message.length(), 0, (sockaddr *)&remoteAddr, sizeof(remoteAddr));if (ret == SOCKET_ERROR){return false;}else{return true;}
}

main.cpp

#include <iostream>
#include <thread>
#include <string>#include "UDPClient.h"
#include "UDPServer.h"int main()
{std::cout << "[0] -> UDPServer"<< "\n"<< "[1] -> UDPClient" << std::endl;u_int flag = 2;std::cin >> flag;switch (flag){case 0:{std::cout << "[UDPServer] -> Start" << std::endl;UDPServer udpServer;udpServer.Bind(6789);std::thread recvThread(&UDPServer::Receive, &udpServer);// 主线程while (true){std::cout << "\n[Input] -> ";std::string msg;std::cin >> msg;if (udpServer.Send(msg)){std::cout << "[UDPServer] -> " << msg << std::endl;}else{std::cout << "[ErrorCode] -> " << GetLastError() << std::endl;}}// 等待线程结束recvThread.join();std::cout << "[UDPServer] -> Stop" << std::endl;break;}case 1:{std::cout << "[UDPClient] -> Start" << std::endl;UDPClient udpClient;udpClient.Bind("127.0.0.1", 6789);std::thread recvThread(&UDPClient::Receive, &udpClient);// 主线程while (true){std::cout << "\n[Input] -> ";std::string msg;std::cin >> msg;if (udpClient.Send(msg)){std::cout << "[UDPClient] -> " << msg << std::endl;}else{std::cout << "[ErrorCode] -> " << GetLastError() << std::endl;}}// 等待线程结束recvThread.join();std::cout << "[UDPClient] -> Stop" << std::endl;break;}default:break;}system("pause");return 0;
}

CMakeLists.txt

cmake_minimum_required(VERSION 3.0.0)
project(UDPProtocol VERSION 0.1.0 LANGUAGES CXX)add_executable(${PROJECT_NAME} UDPClient.cppUDPServer.cppmain.cpp)if(WIN32)target_link_libraries(${PROJECT_NAME} PRIVATE ws2_32)
endif()

测试截图