【C++】 socket UDP网络编程(简单聊天室)

0.通信步骤流程图（左：服务器；右：客户端；）

1.服务器代码
1.1服务器类头文件(CServer_UDP.h)

#pragma once
 
#include <winsock2.h>
 
class CServer_UDP
{
public:
    CServer_UDP();
    void SendMsg(const char sendBuf[]);
    void RecMsg();
    ~CServer_UDP();
 
private:
    SOCKET m_sServer;
    struct sockaddr_in m_SocAddrClient;        //建立连接时，用于保存客户端信息
    bool m_terminal;
};

 
1.2服务器类源文件(CServer_UDP.cpp)

#define _WINSOCK_DEPRECATED_NO_WARNINGS
#include "CServer_UDP.h"
 
#include <iostream>
 
#pragma comment(lib, "ws2_32.lib")
 
 
CServer_UDP::CServer_UDP():m_terminal(false)
{
    //必须进行如下初始化， 否则socket()会返回10093错误
    //初始化WSA  
    WORD sockVersion = MAKEWORD(2, 2);
    WSADATA wsaData;
    if (WSAStartup(sockVersion, &wsaData) != 0) //通过一个进程初始化ws2_32.dll
    {
        std::cout << "Initialize WSA failed" << std::endl;
        return;
    }
    //初始化UDDP套接字
    m_sServer = socket(AF_INET, SOCK_DGRAM, 0);
 
    struct sockaddr_in m_SocAddrserver;
    m_SocAddrserver.sin_addr.S_un.S_addr = 0;//htonl(INADDR_ANY);
    m_SocAddrserver.sin_family = AF_INET;
    m_SocAddrserver.sin_port = htons(8090);
 
    int ret = bind(m_sServer, (sockaddr*)&m_SocAddrserver, sizeof(m_SocAddrserver));
    if (ret == -1)
    {
        std::cout << "bind failed!" << std::endl;
        WSACleanup();
    }
    else
    {
        //此处必须赋初值，不然会导致服务器端无法正常发送
        int len_Client = sizeof(sockaddr);  
        char recBuf[1025];
        int len = recvfrom(m_sServer, recBuf, 1024, 0, (sockaddr*)&m_SocAddrClient, &len_Client);
        if (len > 0)
        {
            recBuf[len] = '\0';
            std::cout << "Client say:" << recBuf << std::endl;
        }
    }
 
 
}
void CServer_UDP::SendMsg(const char sendBuf[])
{
    int ret = sendto(m_sServer, sendBuf, strlen(sendBuf), 0, (sockaddr*)&m_SocAddrClient, sizeof(m_SocAddrClient));
    if (ret == -1)
    {
        std::cout << "send failed" << std::endl;
        std::cout << GetLastError()<< std::endl;
    }
}
void CServer_UDP::RecMsg()
{
    char recBuf[1025];
    while (!m_terminal)
    {
        //std::cout << "Begin rec...(server)" << std::endl;
        int len = recvfrom(m_sServer, recBuf, 1024, 0, 0, 0);
        if (len > 0)
        {
            recBuf[len] = '\0';
            std::cout << "Client say:" << recBuf << std::endl;
        }
    }
    
}
CServer_UDP::~CServer_UDP()
{
    closesocket(m_sServer);
    WSACleanup();
 
}

1.3服务器主函数

#include <iostream>
#include <thread>
#include <string>
 
#include "CServer_UDP.h"
 
using namespace std;
 
int main()
{
    CServer_UDP server_UDP;
 
    thread recProc(&CServer_UDP::RecMsg, &server_UDP);
 
    while (1)
    {
        //cout << "Pleaes input content:" << endl;
        string content;
        cin >> content;
        server_UDP.SendMsg(content.c_str());
    }
 
    recProc.join();
    cout << "I love china!" << endl;
    system("pause");
    return 0;
}

2.客户端代码
2.1客户端类头文件(CClient.h)

#pragma once
 
#include <winsock2.h>
 
class CClient
{
public:
    CClient();
    void RecMsg();
    void SendMsg(const char sendBuf[]);
    ~CClient();
private:
    SOCKET m_sockClient;
    sockaddr_in m_TargetServer;
};

 
2.2客户端类源文件(CClient.cpp)

#define _WINSOCK_DEPRECATED_NO_WARNINGS
#include "CClient.h"
 
#include <iostream>
 
#pragma comment(lib, "ws2_32.lib")
 
CClient::CClient()
{
    //必须进行如下初始化， 否则socket()会返回10093错误
    //初始化WSA  
    WORD sockVersion = MAKEWORD(2, 2);
    WSADATA wsaData;
    if (WSAStartup(sockVersion, &wsaData) != 0) //通过一个进程初始化ws2_32.dll
    {
        std::cout << "Initialize WSA failed" << std::endl;
        return;
    }
 
    m_sockClient = socket(AF_INET, SOCK_DGRAM, 0);
 
    m_TargetServer.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
    m_TargetServer.sin_family = AF_INET;
    m_TargetServer.sin_port = htons(8090);
 
    if (m_sockClient == -1)
    {
        std::cout << "Create socket failed!" << std::endl;
        WSACleanup();
    }
    else
    {
        //发送信息与服务器建立连接(必须加)
        sendto(m_sockClient, "hello server", strlen("hello server"), 0, (sockaddr*)&m_TargetServer, sizeof(m_TargetServer));
    }
}
 
void CClient::SendMsg(const char sendBuf[])
{
    sendto(m_sockClient, sendBuf, strlen(sendBuf), 0, (sockaddr*)&m_TargetServer, sizeof(m_TargetServer));
}
void CClient::RecMsg()
{
    char recBuf[1025];
 
    while (1)
    {
        //std::cout << "Begin rec...(client)" << std::endl;
        int len = recvfrom(m_sockClient, recBuf, 1024, 0, 0, 0);
        if (len > 0)
        {
            recBuf[len] = '\0';
            std::cout << "Server say: " << recBuf << std::endl;
        }
    }
 
}
CClient::~CClient()
{
    closesocket(m_sockClient);
    WSACleanup();
}

2.3客户端主函数

#include <iostream>
#include <string>
#include <thread>
 
#include "CClient.h"
 
using namespace std;
 
int main()
{
    CClient client_UDP;
 
    thread RecProc(&CClient::RecMsg, &client_UDP);
 
    while (1)
    {
        //cout << "Please input content:" << endl;
        string content;
        cin >> content;
 
        client_UDP.SendMsg(content.c_str());
    }
 
 
    RecProc.join();
    cout << "I love china!" << endl;
    system("pause");
    return 0;
}

3.效果图(win7+VS2017)
3.1服务端

3.2客户端