internal.h

Go to the documentation of this file.

#pragma once
#include "../../DCS_Utils/include/DCS_ModuleUtils.h"
#include "../../DCS_Core/include/DCS_ModuleCore.h"
#include "DCS_ModuleNetwork.h"
#include "../config/registry.h"

#define _WINSOCK_DEPRECATED_NO_WARNINGS
#include <WinSock2.h>
#include <ws2tcpip.h>

#include <atomic>
#include <queue>

#define _GET_F_NAME() __FUNCTION__

#define GET_F_NAME() _GET_F_NAME()

#define DCS_EMIT_NAMED_EVT(name, data, size) DCS::Network::Message::EmitEvent(name, data, size)

#define DCS_EMIT_EVT(data, size) DCS::Network::Message::EmitEvent(DCS::Registry::GetEvent(GET_F_NAME()), data, size)

namespace DCS
{
    namespace Network
    {
        struct DCS_INTERNAL_TEST WindowsSocketInformation
        {
            WSADATA wsa;
        };

        inline static bool is_inited = false;

        DCS_INTERNAL_TEST WSADATA InitWinSock();

        DCS_INTERNAL_TEST void CleanupWinSock();

        DCS_INTERNAL_TEST SOCKET CreateServerSocket(i32 listen_port);

        DCS_INTERNAL_TEST SOCKET CreateClientSocket(const char *host, i32 port);

        DCS_INTERNAL_TEST void ServerListen(SOCKET server);

        DCS_INTERNAL_TEST SOCKET ServerAcceptConnection(SOCKET server);

        DCS_INTERNAL_TEST void CloseSocketConnection(SOCKET client);

        DCS_INTERNAL_TEST bool ValidateSocket(SOCKET s);

        DCS_INTERNAL_TEST i32 ReceiveData(SOCKET client, unsigned char *buffer, i32 buff_len);

        DCS_INTERNAL_TEST i32 SendData(SOCKET client, const unsigned char *buffer, i32 buff_len);

        DCS_INTERNAL_TEST i32 SendPriorityData(SOCKET client, const unsigned char *buffer, i32 buff_len);

        DCS_INTERNAL_TEST void GetSocketIpAddress(SOCKET s, char* buffer);

        namespace Message
        {

#define MESSAGE_XTRA_SPACE 9
#pragma pack(push, 1)

            struct DCS_INTERNAL_TEST DefaultMessage
            {
                u8 op;
                u64 id; // used for sync call message identify
                i64 size;
                u8 *ptr;
            };
#pragma pack(pop)

            extern std::mutex message_m;
            extern std::condition_variable lsync;
            extern DefaultMessage lmessage;

            enum class DCS_INTERNAL_TEST InternalOperation
            {
                NO_OP = 0,          
                SYNC_REQUEST = 2,   
                ASYNC_REQUEST = 3,  
                SYNC_RESPONSE = 5,  
                ASYNC_RESPONSE = 6, 
                EVT_SUB = 7,        
                EVT_RESPONSE = 8,   
                EVT_UNSUB = 9,      
                OP_ERROR,           
                CON_VALID,          
                DATA                
            };

            DCS_INTERNAL_TEST Registry::SVReturn WaitForId(u64 id);

            DCS_INTERNAL_TEST void SetMsgIdCondition(DefaultMessage &msg);

            DCS_INTERNAL_TEST void NotifyPromise(DefaultMessage &msg);

            DCS_INTERNAL_TEST DefaultMessage Alloc(i32 size);

            DCS_INTERNAL_TEST void SetCopyIdAndCode(DefaultMessage &msg, u8 *data);

            DCS_INTERNAL_TEST void SetCopyId(DefaultMessage &msg, u8 opcode, u64 id, u8 *data);

            DCS_INTERNAL_TEST void SetNew(DefaultMessage &msg, u8 opcode, u8 *data);

            DCS_INTERNAL_TEST DefaultMessage Copy(DefaultMessage &msg);

            DCS_INTERNAL_TEST void Delete(DefaultMessage &msg);

            DCS_INTERNAL_TEST void ScheduleTransmission(DefaultMessage msg);

            DCS_INTERNAL_TEST void EmitEvent(u8 EVT_ID, u8 *evtData, i32 size);
        }
    }
}