143 lines
3.5 KiB
C++
143 lines
3.5 KiB
C++
#ifndef UDPC_DEFINES_HPP
|
|
#define UDPC_DEFINES_HPP
|
|
|
|
#define UDPC_CONTEXT_IDENTIFIER 0x902F4DB3
|
|
#define UDPC_TIMEOUT_SECONDS 10.0f
|
|
#define UDPC_GOOD_MODE_SEND_INTERVAL (1.0f / 30.0f)
|
|
#define UDPC_BAD_MODE_SEND_INTERVAL (1.0f / 10.0f)
|
|
#define UDPC_SENT_PKTS_MAX_SIZE 33
|
|
#define UDPC_QUEUED_PKTS_MAX_SIZE 32
|
|
#define UDPC_GOOD_RTT_LIMIT_SEC 0.25f
|
|
#define UDPC_PACKET_TIMEOUT_SEC 1.0f
|
|
|
|
#define UDPC_ID_CONNECT 0x80000000
|
|
#define UDPC_ID_PING 0x40000000
|
|
#define UDPC_ID_NO_REC_CHK 0x20000000
|
|
#define UDPC_ID_RESENDING 0x10000000
|
|
|
|
#include <atomic>
|
|
#include <bitset>
|
|
#include <chrono>
|
|
#include <cstdint>
|
|
#include <deque>
|
|
#include <unordered_map>
|
|
#include <queue>
|
|
#include <random>
|
|
#include <memory>
|
|
|
|
#include "TSQueue.hpp"
|
|
#include "UDPConnection.h"
|
|
|
|
namespace UDPC {
|
|
|
|
static uint32_t LOCAL_ADDR = 0;
|
|
static const auto INIT_PKT_INTERVAL_DT = std::chrono::seconds(5);
|
|
static const auto HEARTBEAT_PKT_INTERVAL_DT = std::chrono::milliseconds(150);
|
|
static const auto PACKET_TIMEOUT_TIME = std::chrono::seconds(1);
|
|
|
|
// forward declaration
|
|
struct Context;
|
|
|
|
struct SentPktInfo {
|
|
typedef std::shared_ptr<SentPktInfo> Ptr;
|
|
|
|
SentPktInfo();
|
|
|
|
uint32_t id;
|
|
std::chrono::steady_clock::time_point sentTime;
|
|
};
|
|
|
|
struct ConnectionData {
|
|
ConnectionData();
|
|
ConnectionData(bool isServer, Context *ctx);
|
|
|
|
// copy
|
|
ConnectionData(const ConnectionData& other) = delete;
|
|
ConnectionData& operator=(const ConnectionData& other) = delete;
|
|
|
|
// move
|
|
ConnectionData(ConnectionData&& other) = default;
|
|
ConnectionData& operator=(ConnectionData&& other) = default;
|
|
|
|
void cleanupSentPkts();
|
|
|
|
/*
|
|
* 0 - trigger send
|
|
* 1 - is good mode
|
|
* 2 - is good rtt
|
|
* 3 - initiating connection
|
|
* 4 - is id set
|
|
*/
|
|
std::bitset<32> flags;
|
|
uint32_t id;
|
|
uint32_t lseq;
|
|
uint32_t rseq;
|
|
uint32_t ack;
|
|
float timer;
|
|
float toggleT;
|
|
float toggleTimer;
|
|
float toggledTimer;
|
|
uint32_t addr; // in network order
|
|
uint16_t port; // in native order
|
|
std::deque<UDPC_PacketInfo> sentPkts;
|
|
TSQueue<UDPC_PacketInfo> sendPkts;
|
|
TSQueue<UDPC_PacketInfo> priorityPkts;
|
|
// pkt id to pkt shared_ptr
|
|
std::unordered_map<uint32_t, SentPktInfo::Ptr> sentInfoMap;
|
|
std::chrono::steady_clock::time_point received;
|
|
std::chrono::steady_clock::time_point sent;
|
|
float rtt;
|
|
}; // struct ConnectionData
|
|
|
|
struct Context {
|
|
Context(bool isThreaded);
|
|
|
|
uint_fast32_t _contextIdentifier;
|
|
char recvBuf[UDPC_PACKET_MAX_SIZE];
|
|
/*
|
|
* 0 - is threaded
|
|
* 1 - is client
|
|
* 2 - is accepting new connections
|
|
*/
|
|
std::bitset<32> flags;
|
|
std::atomic_bool isAcceptNewConnections;
|
|
std::atomic_uint32_t protocolID;
|
|
std::atomic_uint_fast8_t loggingType;
|
|
char atostrBuf[16];
|
|
|
|
int socketHandle;
|
|
struct sockaddr_in socketInfo;
|
|
|
|
std::chrono::steady_clock::time_point lastUpdated;
|
|
// ipv4 address to ConnectionData
|
|
std::unordered_map<uint32_t, ConnectionData> conMap;
|
|
// id to ipv4 address
|
|
std::unordered_map<uint32_t, uint32_t> idMap;
|
|
|
|
std::default_random_engine rng_engine;
|
|
|
|
}; // struct Context
|
|
|
|
Context *verifyContext(void *ctx);
|
|
|
|
bool isBigEndian();
|
|
|
|
/*
|
|
* flags:
|
|
* - 0x1 - connect
|
|
* - 0x2 - ping
|
|
* - 0x4 - no_rec_chk
|
|
* - 0x8 - resending
|
|
*/
|
|
void preparePacket(char *data, uint32_t protocolID, uint32_t conID,
|
|
uint32_t rseq, uint32_t ack, uint32_t *seqID, int flags);
|
|
|
|
uint32_t generateConnectionID(Context &ctx);
|
|
|
|
float durationToFSec(
|
|
const std::chrono::steady_clock::time_point& older,
|
|
const std::chrono::steady_clock::time_point& newer);
|
|
|
|
} // namespace UDPC
|
|
|
|
#endif
|