UE4 Sockets多线程TCP通信

不念不忘少年蓝@ 2022-06-09 04:55 452阅读 0赞

本文转自：http://blog.csdn.net/zilisen/article/details/75007447，感谢博主。

## 一、简介 ##

UE4引擎是提供了Sockets模块和Networking模块的，博主在研究此功能时也是参考的Sockets模块和Networking模块的源码，其中引擎为我们提供了一些实例类很有参考价值，比如Sockets模块中的MultichannelTcpReceiver.h和MultichannelTcpSender.h，Networking模块中的UdpSocketReceiver.h和UdpSocketSender.h。博主的例子就是研究参考了以上代码写出来的。

### 编译与运行环境 ###

UnrealEngine 4.15 ， Visual Studio 2015

## 实现功能介绍 ##

博主的例子实现的是一个使用Socket多线程TCP通信的客户端。在主线程中发消息，子线程中收消息。当然也能类似的实现两个子线程分别收发消息。Socket相关函数都定义在了GameInstance中，以便我们能在不同场景都能调用。

## 二、代码实现 ##

### 服务器代码 ###

此处使用了一个有简单收发功能的服务器，用VS2015新建空项目即可：

//SocketTest.cpp: Defines the entry point for the console application
    
    #include <stdio.h>
    #include <string>
    #include <iostream>
    #include <WinSock2.h>
    #include <WS2tcpip.h>
    #pragma comment(lib, "WS2_32.lib")
    
    using namespace std;
    
    int main()
    {
        WSADATA wsaData;
        SOCKET serverSock;
        SOCKADDR_IN serverAddr;
        SOCKET clientSock;
        SOCKADDR_IN clientAddr;
    
    
        cout << "Server Start!" << endl;
        //加载套接字库
        int err = WSAStartup(MAKEWORD(2, 2), &wsaData);
        if (0 != err)
        {
            cout << "WSAStartup failed!" << endl;
            return 1;
        }
    
        //构造监听SOCKET，流式SOCKET
        serverSock = socket(AF_INET, SOCK_STREAM, 0);
        //配置监听地址和端口
        serverAddr.sin_family = AF_INET;
        serverAddr.sin_port = htons(4399);  //本地监听端口：4399
        serverAddr.sin_addr.S_un.S_addr = INADDR_ANY;
    
        //绑定SOCKET
        int retVal = bind(serverSock, (SOCKADDR*)&serverAddr, sizeof(serverAddr));
        if (SOCKET_ERROR == retVal)
        {
            cout << "bind failed!" << endl;
            closesocket(serverSock);
            WSACleanup();
            return -1;
        }
    
        retVal = listen(serverSock, 5);
        if (SOCKET_ERROR == retVal)
        {
            cout << "listen failed!" << endl;
            closesocket(serverSock);
            WSACleanup();
            return -1;
        }
    
        char IPdot[20] = { '\0' };
        inet_ntop(AF_INET, (void*)&serverAddr.sin_addr, IPdot, 16);
        printf("Welcome,the Host %s is running!Now Wating for someone comes in!\n", IPdot);
    
        int addrClientlen = sizeof(clientAddr);
        //等待客户端连接
        clientSock = accept(serverSock, (SOCKADDR*)&clientAddr, &addrClientlen);
        while (1)
        {
            //发送数据
            char sendBuff[50];
            //sprintf_s(sendBuff, "welcome %s to here", inet_ntoa(clientAddr.sin_addr));
            char IPdotdec[20] = { '\0' };
            inet_ntop(AF_INET, (void*)&clientAddr.sin_addr, IPdotdec, 16);
            sprintf_s(sendBuff, "From Server: welcome %s to here", IPdotdec);
            send(clientSock, sendBuff, strlen(sendBuff) + 1, 0);
    
            //接收数据
            char recvBuff[50];
            recv(clientSock, recvBuff, 50, 0);
            printf(" %s \n\n", recvBuff);
    
            Sleep(1000);
        }
        //关闭套接字，关闭加载的套接字库
        closesocket(serverSock);
        WSACleanup();
        return 0;
    }
       
        
         1 
         2 
         3 
         4 
         5 
         6 
         7 
         8 
         9 
         10 
         11 
         12 
         13 
         14 
         15 
         16 
         17 
         18 
         19 
         20 
         21 
         22 
         23 
         24 
         25 
         26 
         27 
         28 
         29 
         30 
         31 
         32 
         33 
         34 
         35 
         36 
         37 
         38 
         39 
         40 
         41 
         42 
         43 
         44 
         45 
         46 
         47 
         48 
         49 
         50 
         51 
         52 
         53 
         54 
         55 
         56 
         57 
         58 
         59 
         60 
         61 
         62 
         63 
         64 
         65 
         66 
         67 
         68 
         69 
         70 
         71 
         72 
         73 
         74 
         75 
         76 
         77 
         78 
         79 
         80 
         81 
         82 
         83 
         84 
        
       
        
         1 
         2 
         3 
         4 
         5 
         6 
         7 
         8 
         9 
         10 
         11 
         12 
         13 
         14 
         15 
         16 
         17 
         18 
         19 
         20 
         21 
         22 
         23 
         24 
         25 
         26 
         27 
         28 
         29 
         30 
         31 
         32 
         33 
         34 
         35 
         36 
         37 
         38 
         39 
         40 
         41 
         42 
         43 
         44 
         45 
         46 
         47 
         48 
         49 
         50 
         51 
         52 
         53 
         54 
         55 
         56 
         57 
         58 
         59 
         60 
         61 
         62 
         63 
         64 
         65 
         66 
         67 
         68 
         69 
         70 
         71 
         72 
         73 
         74 
         75 
         76 
         77 
         78 
         79 
         80 
         81 
         82 
         83 
         84

### 模块添加 ###

新建项目，在.Build.cs文件中添加Sockets模块和Networking模块：

// Copyright 1998-2017 Epic Games, Inc. All Rights Reserved.
    
    using UnrealBuildTool;
    
    public class SocketThread : ModuleRules
    {
        public SocketThread(TargetInfo Target)
        {
            PublicDependencyModuleNames.AddRange(new string[] { "Core", "CoreUObject", "Engine", "InputCore", "HeadMountedDisplay","Sockets", "Networking" });
        }
    }
       
        
         1 
         2 
         3 
         4 
         5 
         6 
         7 
         8 
         9 
         10 
         11 
        
       
        
         1 
         2 
         3 
         4 
         5 
         6 
         7 
         8 
         9 
         10 
         11

### Socket相关函数定义 ###

#### STGameInstance.h ####

//this is the STGameInstance.h
    
    #pragma once
    
    #include "Engine/GameInstance.h"
    #include "Networking.h"
    #include "ReceiveThread.h"
    #include "STGameInstance.generated.h"
    
    UCLASS()
    class SOCKETTHREAD_API USTGameInstance : public UGameInstance
    {
          
        GENERATED_BODY()
    
    public:
        //创建Socket并连接到服务器（主线程）
        UFUNCTION(BlueprintCallable, Category = "MySocket")
        bool SocketCreate(FString IPStr, int32 port);
    
        //发消息(主线程)
        UFUNCTION(BlueprintCallable, Category = "MySocket")
        bool SocketSend(FString message);
    
        //收消息(子线程)
        UFUNCTION(BlueprintCallable, Category = "MySocket")
        bool SocketReceive();
    
        UFUNCTION(BlueprintCallable, Category = "MySocket")
        bool ThreadEnd();
    
        FString StringFromBinaryArray(TArray<uint8> BinaryArray);
    
    public:
        FSocket *Host;
        FIPv4Address ip;    
        FRunnableThread* m_RecvThread;
    };
    
       
        
         1 
         2 
         3 
         4 
         5 
         6 
         7 
         8 
         9 
         10 
         11 
         12 
         13 
         14 
         15 
         16 
         17 
         18 
         19 
         20 
         21 
         22 
         23 
         24 
         25 
         26 
         27 
         28 
         29 
         30 
         31 
         32 
         33 
         34 
         35 
         36 
         37 
         38 
        
       
        
         1 
         2 
         3 
         4 
         5 
         6 
         7 
         8 
         9 
         10 
         11 
         12 
         13 
         14 
         15 
         16 
         17 
         18 
         19 
         20 
         21 
         22 
         23 
         24 
         25 
         26 
         27 
         28 
         29 
         30 
         31 
         32 
         33 
         34 
         35 
         36 
         37 
         38

#### STGameInstance.cpp ####

//this is the STGameInstance.cpp
    
    #include "SocketThread.h"
    #include "STGameInstance.h"
    
    bool USTGameInstance::SocketCreate(FString IPStr, int32 port)
    {
        FIPv4Address::Parse(IPStr, ip);     //将传入的IPStr转为IPv4地址
    
        //创建一个addr存放ip地址和端口
        TSharedPtr<FInternetAddr> addr = ISocketSubsystem::Get(PLATFORM_SOCKETSUBSYSTEM)->CreateInternetAddr();
        addr->SetIp(ip.Value);
        addr->SetPort(port);
    
        //创建客户端socket
        Host = ISocketSubsystem::Get(PLATFORM_SOCKETSUBSYSTEM)->CreateSocket(NAME_Stream, TEXT("default"), false);
    
        //连接成功
        if (Host->Connect(*addr))
        {
            UE_LOG(LogTemp, Warning, TEXT("Connect Succeed!"));
            return true;
        }
        //连接失败
        else
        {
            UE_LOG(LogTemp, Warning, TEXT("Connect Failed!"));
            return false;
        }
    }
    
    bool USTGameInstance::SocketSend(FString message)
    {
        TCHAR *seriallizedChar = message.GetCharArray().GetData();
        int32 size = FCString::Strlen(seriallizedChar) + 1;
        int32 sent = 0;
    
        if (Host->Send((uint8*)TCHAR_TO_UTF8(seriallizedChar), size, sent))
        {
            UE_LOG(LogTemp, Warning, TEXT("___Send Succeed!"));
            return true;
        }
        else
        {
            UE_LOG(LogTemp, Warning, TEXT("___Send Failed!"));
            return false;
        }
    }
    
    bool USTGameInstance::SocketReceive()
    {
        m_RecvThread = FRunnableThread::Create(new FReceiveThread(Host), TEXT("RecvThread"));
        return true;
    }
    
    bool USTGameInstance::ThreadEnd()
    {
        if (m_RecvThread != nullptr)
        {
            m_RecvThread->Kill(true);
            delete m_RecvThread;
        }
        return true;
    }
    
    FString USTGameInstance::StringFromBinaryArray(TArray<uint8> BinaryArray)
    {
        return FString(ANSI_TO_TCHAR(reinterpret_cast<const char*>(BinaryArray.GetData())));
    }
       
        
         1 
         2 
         3 
         4 
         5 
         6 
         7 
         8 
         9 
         10 
         11 
         12 
         13 
         14 
         15 
         16 
         17 
         18 
         19 
         20 
         21 
         22 
         23 
         24 
         25 
         26 
         27 
         28 
         29 
         30 
         31 
         32 
         33 
         34 
         35 
         36 
         37 
         38 
         39 
         40 
         41 
         42 
         43 
         44 
         45 
         46 
         47 
         48 
         49 
         50 
         51 
         52 
         53 
         54 
         55 
         56 
         57 
         58 
         59 
         60 
         61 
         62 
         63 
         64 
         65 
         66 
         67 
         68 
         69 
        
       
        
         1 
         2 
         3 
         4 
         5 
         6 
         7 
         8 
         9 
         10 
         11 
         12 
         13 
         14 
         15 
         16 
         17 
         18 
         19 
         20 
         21 
         22 
         23 
         24 
         25 
         26 
         27 
         28 
         29 
         30 
         31 
         32 
         33 
         34 
         35 
         36 
         37 
         38 
         39 
         40 
         41 
         42 
         43 
         44 
         45 
         46 
         47 
         48 
         49 
         50 
         51 
         52 
         53 
         54 
         55 
         56 
         57 
         58 
         59 
         60 
         61 
         62 
         63 
         64 
         65 
         66 
         67 
         68 
         69

### 收消息线程 ###

#### ReceiveThread.h ####

//this is the ReceiveThread.h
    #pragma once
    
    #include "ThreadingBase.h"
    #include "Networking.h"
    
    class SOCKETTHREAD_API FReceiveThread : public FRunnable
    {
    public:
        FReceiveThread(FSocket* client): m_Client(client)
        {}
        ~FReceiveThread()
        {
            stopping = true;
        }
    
        virtual bool Init() override
        {
            stopping = false;
            return true;
        }
    
        virtual uint32 Run() override
        {
            if (!m_Client)
            {
                return 0;
            }
    
            TArray<uint8> ReceiveData;
            uint8 element = 0;
            //接收数据包
            while (!stopping)   //线程计数器控制
            {
                ReceiveData.Init(element, 1024u);
                int32 read = 0;
                m_Client->Recv(ReceiveData.GetData(), ReceiveData.Num(), read);
                const FString ReceivedUE4String = FString(ANSI_TO_TCHAR(reinterpret_cast<const char*>(ReceiveData.GetData())));
                FString log = "Server:" + ReceivedUE4String;
                UE_LOG(LogTemp, Warning, TEXT("*** %s"), *log);
    
                FPlatformProcess::Sleep(0.01f);
            }
    
    
            return 1;
        }
    
        virtual void Stop() override
        {
            stopping = true;    //计数器-1
        }
    
    private:
        FSocket* m_Client;  //客户端套接字
        bool stopping;      //循环控制
        FThreadSafeCounter m_StopTaskCounter;   //线程引用计数器
    };
       
        
         1 
         2 
         3 
         4 
         5 
         6 
         7 
         8 
         9 
         10 
         11 
         12 
         13 
         14 
         15 
         16 
         17 
         18 
         19 
         20 
         21 
         22 
         23 
         24 
         25 
         26 
         27 
         28 
         29 
         30 
         31 
         32 
         33 
         34 
         35 
         36 
         37 
         38 
         39 
         40 
         41 
         42 
         43 
         44 
         45 
         46 
         47 
         48 
         49 
         50 
         51 
         52 
         53 
         54 
         55 
         56 
         57 
         58 
        
       
        
         1 
         2 
         3 
         4 
         5 
         6 
         7 
         8 
         9 
         10 
         11 
         12 
         13 
         14 
         15 
         16 
         17 
         18 
         19 
         20 
         21 
         22 
         23 
         24 
         25 
         26 
         27 
         28 
         29 
         30 
         31 
         32 
         33 
         34 
         35 
         36 
         37 
         38 
         39 
         40 
         41 
         42 
         43 
         44 
         45 
         46 
         47 
         48 
         49 
         50 
         51 
         52 
         53 
         54 
         55 
         56 
         57 
         58

## 三、运行 ##

在编辑器中，创建我们之前定义的GameInstance的蓝图类，并在Project Setting中设其为默认GameInstance。创建一个GameMode，在他的EventGraph中添加如下蓝图逻辑：  
![这里写图片描述][SouthEast]  
其中Succe是一个bool类型变量，Index是一个Int类型变量，不改默认值。  
之后将这个GameMode绑定到当前场景中，编译运行服务器，然后Play，大功告成！博主运行结果如下：  
![这里写图片描述][SouthEast 1]  
可以看到，收发消息都成功了。

[SouthEast]: /images/20220609/04f337800daa413f919e6b377e1bae0d.png
[SouthEast 1]: /images/20220609/95a74a989247417f8a3e5cb08b0f0b21.png